aboutsummaryrefslogtreecommitdiffstats
path: root/device
diff options
context:
space:
mode:
authorMatt Layher <mdlayher@gmail.com>2019-05-29 12:18:20 -0400
committerJason A. Donenfeld <Jason@zx2c4.com>2019-05-29 18:34:55 +0200
commit32912dc778a0b98637fbfe010b31bd0b746662e6 (patch)
tree7d9dfa99bcb6e0479f6b4e0358f5f6daa213929d /device
parentwintun: remove extra / (diff)
downloadwireguard-go-32912dc778a0b98637fbfe010b31bd0b746662e6.tar.xz
wireguard-go-32912dc778a0b98637fbfe010b31bd0b746662e6.zip
device, tun: rearrange code and fix device tests
Signed-off-by: Matt Layher <mdlayher@gmail.com>
Diffstat (limited to 'device')
-rw-r--r--device/device_test.go40
-rw-r--r--device/tun_test.go56
2 files changed, 86 insertions, 10 deletions
diff --git a/device/device_test.go b/device/device_test.go
index db5a3c0..b6212b5 100644
--- a/device/device_test.go
+++ b/device/device_test.go
@@ -9,21 +9,17 @@ package device
* without network dependencies
*/
-import "testing"
+import (
+ "bytes"
+ "testing"
+)
func TestDevice(t *testing.T) {
// prepare tun devices for generating traffic
- tun1, err := CreateDummyTUN("tun1")
- if err != nil {
- t.Error("failed to create tun:", err.Error())
- }
-
- tun2, err := CreateDummyTUN("tun2")
- if err != nil {
- t.Error("failed to create tun:", err.Error())
- }
+ tun1 := newDummyTUN("tun1")
+ tun2 := newDummyTUN("tun2")
_ = tun1
_ = tun2
@@ -46,3 +42,27 @@ func TestDevice(t *testing.T) {
// create binds
}
+
+func randDevice(t *testing.T) *Device {
+ sk, err := newPrivateKey()
+ if err != nil {
+ t.Fatal(err)
+ }
+ tun := newDummyTUN("dummy")
+ logger := NewLogger(LogLevelError, "")
+ device := NewDevice(tun, logger)
+ device.SetPrivateKey(sk)
+ return device
+}
+
+func assertNil(t *testing.T, err error) {
+ if err != nil {
+ t.Fatal(err)
+ }
+}
+
+func assertEqual(t *testing.T, a []byte, b []byte) {
+ if bytes.Compare(a, b) != 0 {
+ t.Fatal(a, "!=", b)
+ }
+}
diff --git a/device/tun_test.go b/device/tun_test.go
new file mode 100644
index 0000000..fbe4c1d
--- /dev/null
+++ b/device/tun_test.go
@@ -0,0 +1,56 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package device
+
+import (
+ "errors"
+ "os"
+
+ "golang.zx2c4.com/wireguard/tun"
+)
+
+// newDummyTUN creates a dummy TUN device with the specified name.
+func newDummyTUN(name string) tun.TUNDevice {
+ return &dummyTUN{
+ name: name,
+ packets: make(chan []byte, 100),
+ events: make(chan tun.TUNEvent, 10),
+ }
+}
+
+// A dummyTUN is a tun.TUNDevice which is used in unit tests.
+type dummyTUN struct {
+ name string
+ mtu int
+ packets chan []byte
+ events chan tun.TUNEvent
+}
+
+func (d *dummyTUN) Events() chan tun.TUNEvent { return d.events }
+func (*dummyTUN) File() *os.File { return nil }
+func (*dummyTUN) Flush() error { return nil }
+func (d *dummyTUN) MTU() (int, error) { return d.mtu, nil }
+func (d *dummyTUN) Name() (string, error) { return d.name, nil }
+
+func (d *dummyTUN) Close() error {
+ close(d.events)
+ close(d.packets)
+ return nil
+}
+
+func (d *dummyTUN) Read(b []byte, offset int) (int, error) {
+ buf, ok := <-d.packets
+ if !ok {
+ return 0, errors.New("device closed")
+ }
+ copy(b[offset:], buf)
+ return len(buf), nil
+}
+
+func (d *dummyTUN) Write(b []byte, offset int) (int, error) {
+ d.packets <- b[offset:]
+ return len(b), nil
+}