TheArchive/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2024-11-30 13:35:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a separate buffer per session for incoming packets, so 1 session that floods won't block other sessions

Browse Source
This commit is contained in:
Arceliar 2019-08-07 18:08:31 -05:00
parent 9ab08446ff
commit 5e81a0c421
1 changed files with 28 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/yggdrasil/session.go
View File

@ -508,6 +508,32 @@ func (sinfo *sessionInfo) recvWorker() {
util.WorkerGo(poolFunc)
callbacks = append(callbacks, ch)
}
fromHelper := make(chan wire_trafficPacket, 1)
go func() {
var buf []wire_trafficPacket
for {
for len(buf) > 0 {
select {
case <-sinfo.cancel.Finished():
return
case p := <-sinfo.fromRouter:
buf = append(buf, p)
for len(buf) > 64 { // Based on nonce window size
util.PutBytes(buf[0].Payload)
buf = buf[1:]
}
case fromHelper <- buf[0]:
buf = buf[1:]
}
}
select {
case <-sinfo.cancel.Finished():
return
case p := <-sinfo.fromRouter:
buf = append(buf, p)
}
}
}()
for {
for len(callbacks) > 0 {
select {
@ -516,14 +542,14 @@ func (sinfo *sessionInfo) recvWorker() {
f()
case <-sinfo.cancel.Finished():
return
case p := <-sinfo.fromRouter:
case p := <-fromHelper:
doRecv(p)
}
}
select {
case <-sinfo.cancel.Finished():
return
case p := <-sinfo.fromRouter:
case p := <-fromHelper:
doRecv(p)
}
}