61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package node
|
|
|
|
import (
|
|
"log"
|
|
"sync/atomic"
|
|
)
|
|
|
|
func _sendControlPacket(pkt interface{ Marshal([]byte) []byte }, route peerRoute, buf1, buf2 []byte) {
|
|
buf := pkt.Marshal(buf2)
|
|
h := header{
|
|
StreamID: controlStreamID,
|
|
Counter: atomic.AddUint64(&sendCounters[route.IP], 1),
|
|
SourceIP: localIP,
|
|
DestIP: route.IP,
|
|
}
|
|
buf = route.ControlCipher.Encrypt(h, buf, buf1)
|
|
|
|
if route.RelayIP == 0 {
|
|
_conn.WriteTo(buf, route.RemoteAddr)
|
|
return
|
|
}
|
|
|
|
_relayPacket(route.RelayIP, route.IP, buf, buf2)
|
|
}
|
|
|
|
func _sendDataPacket(route *peerRoute, pkt, buf1, buf2 []byte) {
|
|
h := header{
|
|
StreamID: dataStreamID,
|
|
Counter: atomic.AddUint64(&sendCounters[route.IP], 1),
|
|
SourceIP: localIP,
|
|
DestIP: route.IP,
|
|
}
|
|
|
|
enc := route.DataCipher.Encrypt(h, pkt, buf1)
|
|
|
|
if route.RelayIP == 0 {
|
|
_conn.WriteTo(enc, route.RemoteAddr)
|
|
return
|
|
}
|
|
|
|
_relayPacket(route.RelayIP, route.IP, enc, buf2)
|
|
}
|
|
|
|
func _relayPacket(relayIP, destIP byte, data, buf []byte) {
|
|
relayRoute := routingTable[relayIP].Load()
|
|
if !relayRoute.Up || !relayRoute.Relay {
|
|
log.Print("Failed to send data packet: relay not available.")
|
|
return
|
|
}
|
|
|
|
h := header{
|
|
StreamID: dataStreamID,
|
|
Counter: atomic.AddUint64(&sendCounters[relayRoute.IP], 1),
|
|
SourceIP: localIP,
|
|
DestIP: destIP,
|
|
}
|
|
|
|
enc := relayRoute.DataCipher.Encrypt(h, data, buf)
|
|
_conn.WriteTo(enc, relayRoute.RemoteAddr)
|
|
}
|