mirror of
https://github.com/cwinfo/yggdrasil-go.git
synced 2024-11-26 06:01:37 +00:00
commit
f53699367b
@ -425,10 +425,33 @@ func (sinfo *sessionInfo) doSend(bs []byte) {
|
||||
if !sinfo.init {
|
||||
return
|
||||
} // To prevent using empty session keys
|
||||
// Now we append something to the coords
|
||||
// Specifically, we append a 0, and then arbitrary data
|
||||
// The 0 ensures that the destination node switch forwards to the self peer (router)
|
||||
// The rest is ignored, but it's still part as the coords, so it affects switch queues
|
||||
// This helps separate traffic streams (protocol + source + dest port) are queued independently
|
||||
var coords []byte
|
||||
addUint64 := func(bs []byte) {
|
||||
// Converts bytes to a uint64
|
||||
// Converts that back to variable length bytes
|
||||
// Appends it to coords
|
||||
var u uint64
|
||||
for _, b := range bs {
|
||||
u <<= 8
|
||||
u |= uint64(b)
|
||||
}
|
||||
coords = append(coords, wire_encode_uint64(u)...)
|
||||
}
|
||||
coords = append(coords, sinfo.coords...) // Start with the real coords
|
||||
coords = append(coords, 0) // Add an explicit 0 for the destination's self peer
|
||||
addUint64(bs[6:7]) // Byte 6, next header type (e.g. TCP vs UDP)
|
||||
// TODO parse headers, in case the next header isn't TCP/UDP for some reason
|
||||
addUint64(bs[40:42]) // Bytes 40-41, source port for TCP/UDP
|
||||
addUint64(bs[42:44]) // Bytes 42-43, destination port for TCP/UDP
|
||||
payload, nonce := boxSeal(&sinfo.sharedSesKey, bs, &sinfo.myNonce)
|
||||
defer util_putBytes(payload)
|
||||
p := wire_trafficPacket{
|
||||
Coords: sinfo.coords,
|
||||
Coords: coords,
|
||||
Handle: sinfo.theirHandle,
|
||||
Nonce: *nonce,
|
||||
Payload: payload,
|
||||
|
@ -529,24 +529,13 @@ func switch_getPacketCoords(packet []byte) []byte {
|
||||
}
|
||||
|
||||
// Returns a unique string for each stream of traffic
|
||||
// Equal to type+coords+handle for traffic packets
|
||||
// Equal to type+coords+toKey+fromKey for protocol traffic packets
|
||||
// Equal to coords
|
||||
// The sender may append arbitrary info to the end of coords (as long as it's begins with a 0x00) to designate separate traffic streams
|
||||
// Currently, it's the IPv6 next header type and the first 2 uint16 of the next header
|
||||
// This is equivalent to the TCP/UDP protocol numbers and the source / dest ports
|
||||
// TODO figure out if something else would make more sense (other transport protocols?)
|
||||
func switch_getPacketStreamID(packet []byte) string {
|
||||
pType, pTypeLen := wire_decode_uint64(packet)
|
||||
_, coordLen := wire_decode_coords(packet[pTypeLen:])
|
||||
end := pTypeLen + coordLen
|
||||
switch {
|
||||
case pType == wire_Traffic:
|
||||
end += handleLen // handle
|
||||
case pType == wire_ProtocolTraffic:
|
||||
end += 2 * boxPubKeyLen
|
||||
default:
|
||||
end = 0
|
||||
}
|
||||
if end > len(packet) {
|
||||
end = len(packet)
|
||||
}
|
||||
return string(packet[:end])
|
||||
return string(switch_getPacketCoords(packet))
|
||||
}
|
||||
|
||||
// Handle an incoming packet
|
||||
|
Loading…
Reference in New Issue
Block a user