transport: optimize stream read

named-data · Dec 26, 2024 · 2b37e4a · 2b37e4a
1 parent 3ef2202
commit 2b37e4a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/fw/face/stream-transport.go b/fw/face/stream-transport.go
@@ -13,7 +13,7 @@ func readTlvStream(
 	onFrame func([]byte),
 	ignoreError func(error) bool,
 ) error {
-	recvBuf := make([]byte, defn.MaxNDNPacketSize*32)
+	recvBuf := make([]byte, defn.MaxNDNPacketSize*8)
 	recvOff := 0
 	tlvOff := 0
 
@@ -60,8 +60,15 @@ func readTlvStream(
 			}
 		}
 
+		// If recvOff and tlvOff coincide, reset to beginning (for free!)
+		if recvOff == tlvOff {
+			recvOff = 0
+			tlvOff = 0
+			continue
+		}
+
 		// If less than one packet space remains in buffer, shift to beginning
-		if recvOff-tlvOff < defn.MaxNDNPacketSize {
+		if len(recvBuf)-recvOff < defn.MaxNDNPacketSize {
 			copy(recvBuf, recvBuf[tlvOff:recvOff])
 			recvOff -= tlvOff
 			tlvOff = 0