TheArchive/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2024-11-27 12:05:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

correctly update buffer sizs when buffers overflow, and returned freed packets to the byte store

Browse Source
This commit is contained in:
Arceliar 2018-07-06 17:27:04 -05:00
parent ad5dc9ea87
commit ba4047b51a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/yggdrasil/switch.go
View File

@ -631,8 +631,12 @@ func (b *switch_buffers) cleanup(t *switchTable) {
packet, buf.packets = buf.packets[0], buf.packets[1:]
buf.size -= uint64(len(packet.bytes))
b.size -= uint64(len(packet.bytes))
util_putBytes(packet.bytes)
if len(buf.packets) == 0 {
delete(b.bufs, streamID)
} else {
// Need to update the map, since buf was retrieved by value
b.bufs[streamID] = buf
}
break
}
@ -672,6 +676,7 @@ func (t *switchTable) handleIdle(port switchPort, bufs *switch_buffers) bool {
if len(buf.packets) == 0 {
delete(bufs.bufs, best)
} else {
// Need to update the map, since buf was retrieved by value
bufs.bufs[best] = buf
}
to.sendPacket(packet.bytes)