1 files changed, 84 insertions, 0 deletions
diff --git a/device/kdf_test.go b/device/kdf_test.go
new file mode 100644
index 0000000..11ea8d5
--- /dev/null
+++ b/device/kdf_test.go
@@ -0,0 +1,84 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package device
+
+import (
+	"encoding/hex"
+	"golang.org/x/crypto/blake2s"
+	"testing"
+)
+
+type KDFTest struct {
+	key   string
+	input string
+	t0    string
+	t1    string
+	t2    string
+}
+
+func assertEquals(t *testing.T, a string, b string) {
+	if a != b {
+		t.Fatal("expected", a, "=", b)
+	}
+}
+
+func TestKDF(t *testing.T) {
+	tests := []KDFTest{
+		{
+			key:   "746573742d6b6579",
+			input: "746573742d696e707574",
+			t0:    "6f0e5ad38daba1bea8a0d213688736f19763239305e0f58aba697f9ffc41c633",
+			t1:    "df1194df20802a4fe594cde27e92991c8cae66c366e8106aaa937a55fa371e8a",
+			t2:    "fac6e2745a325f5dc5d11a5b165aad08b0ada28e7b4e666b7c077934a4d76c24",
+		},
+		{
+			key:   "776972656775617264",
+			input: "776972656775617264",
+			t0:    "491d43bbfdaa8750aaf535e334ecbfe5129967cd64635101c566d4caefda96e8",
+			t1:    "1e71a379baefd8a79aa4662212fcafe19a23e2b609a3db7d6bcba8f560e3d25f",
+			t2:    "31e1ae48bddfbe5de38f295e5452b1909a1b4e38e183926af3780b0c1e1f0160",
+		},
+		{
+			key:   "",
+			input: "",
+			t0:    "8387b46bf43eccfcf349552a095d8315c4055beb90208fb1be23b894bc2ed5d0",
+			t1:    "58a0e5f6faefccf4807bff1f05fa8a9217945762040bcec2f4b4a62bdfe0e86e",
+			t2:    "0ce6ea98ec548f8e281e93e32db65621c45eb18dc6f0a7ad94178610a2f7338e",
+		},
+	}
+
+	var t0, t1, t2 [blake2s.Size]byte
+
+	for _, test := range tests {
+		key, _ := hex.DecodeString(test.key)
+		input, _ := hex.DecodeString(test.input)
+		KDF3(&t0, &t1, &t2, key, input)
+		t0s := hex.EncodeToString(t0[:])
+		t1s := hex.EncodeToString(t1[:])
+		t2s := hex.EncodeToString(t2[:])
+		assertEquals(t, t0s, test.t0)
+		assertEquals(t, t1s, test.t1)
+		assertEquals(t, t2s, test.t2)
+	}
+
+	for _, test := range tests {
+		key, _ := hex.DecodeString(test.key)
+		input, _ := hex.DecodeString(test.input)
+		KDF2(&t0, &t1, key, input)
+		t0s := hex.EncodeToString(t0[:])
+		t1s := hex.EncodeToString(t1[:])
+		assertEquals(t, t0s, test.t0)
+		assertEquals(t, t1s, test.t1)
+	}
+
+	for _, test := range tests {
+		key, _ := hex.DecodeString(test.key)
+		input, _ := hex.DecodeString(test.input)
+		KDF1(&t0, key, input)
+		t0s := hex.EncodeToString(t0[:])
+		assertEquals(t, t0s, test.t0)
+	}
+}