summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--device/channels.go69
-rw-r--r--device/device.go61
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.
*