mirror of
https://git.zx2c4.com/wireguard-go
synced 2024-11-15 01:05:15 +01:00
device: receive: drain decryption queue before exiting RoutineDecryption
It's possible for RoutineSequentialReceiver to try to lock an elem after RoutineDecryption has exited. Before this meant we didn't then unlock the elem, so the whole program deadlocked. As well, it looks like the flush code (which is now potentially unnecessary?) wasn't properly dropping the buffers for the not-already-dropped case. Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
This commit is contained in:
parent
85b4950579
commit
29b0477585
@ -371,7 +371,10 @@ func (device *Device) FlushPacketQueues() {
|
|||||||
select {
|
select {
|
||||||
case elem, ok := <-device.queue.decryption:
|
case elem, ok := <-device.queue.decryption:
|
||||||
if ok {
|
if ok {
|
||||||
elem.Drop()
|
if !elem.IsDropped() {
|
||||||
|
elem.Drop()
|
||||||
|
device.PutMessageBuffer(elem.buffer)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
case <-device.queue.handshake:
|
case <-device.queue.handshake:
|
||||||
default:
|
default:
|
||||||
|
@ -251,7 +251,20 @@ func (device *Device) RoutineDecryption() {
|
|||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-device.signals.stop:
|
case <-device.signals.stop:
|
||||||
return
|
for {
|
||||||
|
select {
|
||||||
|
case elem, ok := <-device.queue.decryption:
|
||||||
|
if ok {
|
||||||
|
if !elem.IsDropped() {
|
||||||
|
elem.Drop()
|
||||||
|
device.PutMessageBuffer(elem.buffer)
|
||||||
|
}
|
||||||
|
elem.Unlock()
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
case elem, ok := <-device.queue.decryption:
|
case elem, ok := <-device.queue.decryption:
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user