diff options
-rw-r--r-- | device/channels.go | 69 | ||||
-rw-r--r-- | device/device.go | 61 |
2 files changed, 69 insertions, 61 deletions
diff --git a/device/channels.go b/device/channels.go new file mode 100644 index 0000000..4471477 --- /dev/null +++ b/device/channels.go @@ -0,0 +1,69 @@ +/* SPDX-License-Identifier: MIT + * + * Copyright (C) 2017-2021 WireGuard LLC. All Rights Reserved. + */ + +package device + +import "sync" + +// An outboundQueue is a channel of QueueOutboundElements awaiting encryption. +// An outboundQueue is ref-counted using its wg field. +// An outboundQueue created with newOutboundQueue has one reference. +// Every additional writer must call wg.Add(1). +// Every completed writer must call wg.Done(). +// When no further writers will be added, +// call wg.Done to remove the initial reference. +// When the refcount hits 0, the queue's channel is closed. +type outboundQueue struct { + c chan *QueueOutboundElement + wg sync.WaitGroup +} + +func newOutboundQueue() *outboundQueue { + q := &outboundQueue{ + c: make(chan *QueueOutboundElement, QueueOutboundSize), + } + q.wg.Add(1) + go func() { + q.wg.Wait() + close(q.c) + }() + return q +} + +// A inboundQueue is similar to an outboundQueue; see those docs. +type inboundQueue struct { + c chan *QueueInboundElement + wg sync.WaitGroup +} + +func newInboundQueue() *inboundQueue { + q := &inboundQueue{ + c: make(chan *QueueInboundElement, QueueInboundSize), + } + q.wg.Add(1) + go func() { + q.wg.Wait() + close(q.c) + }() + return q +} + +// A handshakeQueue is similar to an outboundQueue; see those docs. +type handshakeQueue struct { + c chan QueueHandshakeElement + wg sync.WaitGroup +} + +func newHandshakeQueue() *handshakeQueue { + q := &handshakeQueue{ + c: make(chan QueueHandshakeElement, QueueHandshakeSize), + } + q.wg.Add(1) + go func() { + q.wg.Wait() + close(q.c) + }() + return q +} diff --git a/device/device.go b/device/device.go index 52b0143..b93ad22 100644 --- a/device/device.go +++ b/device/device.go @@ -133,67 +133,6 @@ func (device *Device) isUp() bool { return device.deviceState() == deviceStateUp } -// An outboundQueue is a channel of QueueOutboundElements awaiting encryption. -// An outboundQueue is ref-counted using its wg field. -// An outboundQueue created with newOutboundQueue has one reference. -// Every additional writer must call wg.Add(1). -// Every completed writer must call wg.Done(). -// When no further writers will be added, -// call wg.Done to remove the initial reference. -// When the refcount hits 0, the queue's channel is closed. -type outboundQueue struct { - c chan *QueueOutboundElement - wg sync.WaitGroup -} - -func newOutboundQueue() *outboundQueue { - q := &outboundQueue{ - c: make(chan *QueueOutboundElement, QueueOutboundSize), - } - q.wg.Add(1) - go func() { - q.wg.Wait() - close(q.c) - }() - return q -} - -// A inboundQueue is similar to an outboundQueue; see those docs. -type inboundQueue struct { - c chan *QueueInboundElement - wg sync.WaitGroup -} - -func newInboundQueue() *inboundQueue { - q := &inboundQueue{ - c: make(chan *QueueInboundElement, QueueInboundSize), - } - q.wg.Add(1) - go func() { - q.wg.Wait() - close(q.c) - }() - return q -} - -// A handshakeQueue is similar to an outboundQueue; see those docs. -type handshakeQueue struct { - c chan QueueHandshakeElement - wg sync.WaitGroup -} - -func newHandshakeQueue() *handshakeQueue { - q := &handshakeQueue{ - c: make(chan QueueHandshakeElement, QueueHandshakeSize), - } - q.wg.Add(1) - go func() { - q.wg.Wait() - close(q.c) - }() - return q -} - /* Converts the peer into a "zombie", which remains in the peer map, * but processes no packets and does not exists in the routing table. * |