From 9cbcaf39acb97868bf34866acb4f784adbd2a335 Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Sat, 21 Jul 2018 18:59:29 -0500
Subject: [PATCH] Use coords for queue stream IDs in the switch, and append
 protocol/port information to coords when sending, to designate different
 streams

---
 src/yggdrasil/session.go | 25 ++++++++++++++++++++++++-
 src/yggdrasil/switch.go  | 23 ++++++-----------------
 2 files changed, 30 insertions(+), 18 deletions(-)

diff --git a/src/yggdrasil/session.go b/src/yggdrasil/session.go
index 8631ff2..f1ee92a 100644
--- a/src/yggdrasil/session.go
+++ b/src/yggdrasil/session.go
@@ -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,
diff --git a/src/yggdrasil/switch.go b/src/yggdrasil/switch.go
index 6fe50bf..63380da 100644
--- a/src/yggdrasil/switch.go
+++ b/src/yggdrasil/switch.go
@@ -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