From a71256d250b1838f2db96db7a80a68a1a06d40de Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Wed, 20 Nov 2024 18:53:51 +0800
Subject: [PATCH] Add device.InputPacket

---
 device/send.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/device/send.go b/device/send.go
index 78a3108..64a4221 100644
--- a/device/send.go
+++ b/device/send.go
@@ -322,6 +322,30 @@ func (device *Device) RoutineReadFromTUN() {
 	}
 }
 
+func (device *Device) InputPacket(destination []byte, packetSlices [][]byte) {
+	peer := device.allowedips.Lookup(destination)
+	if peer == nil {
+		return
+	}
+	elem := device.NewOutboundElement()
+	packet := elem.buffer[MessageTransportHeaderSize:]
+	var n int
+	for _, packetSlice := range packetSlices {
+		n += copy(packet[n:], packetSlice)
+	}
+	elem.packet = packet[:n]
+	elemsForPeer := device.GetOutboundElementsContainer()
+	if peer.isRunning.Load() {
+		elemsForPeer.elems = append(elemsForPeer.elems, elem)
+		peer.StagePackets(elemsForPeer)
+		peer.SendStagedPackets()
+	} else {
+		device.PutMessageBuffer(elem.buffer)
+		device.PutOutboundElement(elem)
+		device.PutOutboundElementsContainer(elemsForPeer)
+	}
+}
+
 func (peer *Peer) StagePackets(elems *QueueOutboundElementsContainer) {
 	for {
 		select {