From ad5dc9ea87afbde2aee36c9a7f377460105a8589 Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Fri, 6 Jul 2018 00:55:00 -0500
Subject: [PATCH] Drop 1 packet instead of a whole queue when overflowing

---
 src/yggdrasil/switch.go | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/src/yggdrasil/switch.go b/src/yggdrasil/switch.go
index f47526f..ea52fdc 100644
--- a/src/yggdrasil/switch.go
+++ b/src/yggdrasil/switch.go
@@ -605,21 +605,16 @@ type switch_buffers struct {
 }
 
 func (b *switch_buffers) cleanup(t *switchTable) {
-	remove := func(streamID string) {
-		// Helper function to drop a queue
-		buf := b.bufs[streamID]
-		for _, packet := range buf.packets {
-			util_putBytes(packet.bytes)
-		}
-		b.size -= buf.size
-		delete(b.bufs, streamID)
-	}
 	for streamID, buf := range b.bufs {
 		// Remove queues for which we have no next hop
 		packet := buf.packets[0]
 		coords := switch_getPacketCoords(packet.bytes)
 		if t.selfIsClosest(coords) {
-			remove(streamID)
+			for _, packet := range buf.packets {
+				util_putBytes(packet.bytes)
+			}
+			b.size -= buf.size
+			delete(b.bufs, streamID)
 		}
 	}
 	const maxSize = 4 * 1048576 // Maximum 4 MB
@@ -632,7 +627,13 @@ func (b *switch_buffers) cleanup(t *switchTable) {
 			if size < target {
 				continue
 			}
-			remove(streamID)
+			var packet switch_packetInfo
+			packet, buf.packets = buf.packets[0], buf.packets[1:]
+			buf.size -= uint64(len(packet.bytes))
+			b.size -= uint64(len(packet.bytes))
+			if len(buf.packets) == 0 {
+				delete(b.bufs, streamID)
+			}
 			break
 		}
 	}