aboutsummaryrefslogtreecommitdiffstats
path: root/replay/replay_test.go
diff options
context:
space:
mode:
authorJason A. Donenfeld <Jason@zx2c4.com>2018-05-23 02:32:02 +0200
committerJason A. Donenfeld <Jason@zx2c4.com>2018-05-23 03:58:27 +0200
commit5a2228a5c910ada948677f1dd3fcc59f74e5cb20 (patch)
treebafb31cbcc18221e2299f8fef21d9c2f4471f706 /replay/replay_test.go
parentMove tun to subpackage (diff)
downloadwireguard-go-5a2228a5c910ada948677f1dd3fcc59f74e5cb20.tar.xz
wireguard-go-5a2228a5c910ada948677f1dd3fcc59f74e5cb20.zip
Move replay into subpackage
Diffstat (limited to 'replay/replay_test.go')
-rw-r--r--replay/replay_test.go120
1 files changed, 120 insertions, 0 deletions
diff --git a/replay/replay_test.go b/replay/replay_test.go
new file mode 100644
index 0000000..da39498
--- /dev/null
+++ b/replay/replay_test.go
@@ -0,0 +1,120 @@
+/* SPDX-License-Identifier: GPL-2.0
+ *
+ * Copyright (C) 2017-2018 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
+ * Copyright (C) 2017-2018 Mathias N. Hall-Andersen <mathias@hall-andersen.dk>.
+ */
+
+package replay
+
+import (
+ "testing"
+)
+
+/* Ported from the linux kernel implementation
+ *
+ *
+ */
+
+const RejectAfterMessages = (1 << 64) - (1 << 4) - 1
+
+func TestReplay(t *testing.T) {
+ var filter ReplayFilter
+
+ T_LIM := CounterWindowSize + 1
+
+ testNumber := 0
+ T := func(n uint64, v bool) {
+ testNumber++
+ if filter.ValidateCounter(n, RejectAfterMessages) != v {
+ t.Fatal("Test", testNumber, "failed", n, v)
+ }
+ }
+
+ filter.Init()
+
+ T(0, true) /* 1 */
+ T(1, true) /* 2 */
+ T(1, false) /* 3 */
+ T(9, true) /* 4 */
+ T(8, true) /* 5 */
+ T(7, true) /* 6 */
+ T(7, false) /* 7 */
+ T(T_LIM, true) /* 8 */
+ T(T_LIM-1, true) /* 9 */
+ T(T_LIM-1, false) /* 10 */
+ T(T_LIM-2, true) /* 11 */
+ T(2, true) /* 12 */
+ T(2, false) /* 13 */
+ T(T_LIM+16, true) /* 14 */
+ T(3, false) /* 15 */
+ T(T_LIM+16, false) /* 16 */
+ T(T_LIM*4, true) /* 17 */
+ T(T_LIM*4-(T_LIM-1), true) /* 18 */
+ T(10, false) /* 19 */
+ T(T_LIM*4-T_LIM, false) /* 20 */
+ T(T_LIM*4-(T_LIM+1), false) /* 21 */
+ T(T_LIM*4-(T_LIM-2), true) /* 22 */
+ T(T_LIM*4+1-T_LIM, false) /* 23 */
+ T(0, false) /* 24 */
+ T(RejectAfterMessages, false) /* 25 */
+ T(RejectAfterMessages-1, true) /* 26 */
+ T(RejectAfterMessages, false) /* 27 */
+ T(RejectAfterMessages-1, false) /* 28 */
+ T(RejectAfterMessages-2, true) /* 29 */
+ T(RejectAfterMessages+1, false) /* 30 */
+ T(RejectAfterMessages+2, false) /* 31 */
+ T(RejectAfterMessages-2, false) /* 32 */
+ T(RejectAfterMessages-3, true) /* 33 */
+ T(0, false) /* 34 */
+
+ t.Log("Bulk test 1")
+ filter.Init()
+ testNumber = 0
+ for i := uint64(1); i <= CounterWindowSize; i++ {
+ T(i, true)
+ }
+ T(0, true)
+ T(0, false)
+
+ t.Log("Bulk test 2")
+ filter.Init()
+ testNumber = 0
+ for i := uint64(2); i <= CounterWindowSize+1; i++ {
+ T(i, true)
+ }
+ T(1, true)
+ T(0, false)
+
+ t.Log("Bulk test 3")
+ filter.Init()
+ testNumber = 0
+ for i := CounterWindowSize + 1; i > 0; i-- {
+ T(i, true)
+ }
+
+ t.Log("Bulk test 4")
+ filter.Init()
+ testNumber = 0
+ for i := CounterWindowSize + 2; i > 1; i-- {
+ T(i, true)
+ }
+ T(0, false)
+
+ t.Log("Bulk test 5")
+ filter.Init()
+ testNumber = 0
+ for i := CounterWindowSize; i > 0; i-- {
+ T(i, true)
+ }
+ T(CounterWindowSize+1, true)
+ T(0, false)
+
+ t.Log("Bulk test 6")
+ filter.Init()
+ testNumber = 0
+ for i := CounterWindowSize; i > 0; i-- {
+ T(i, true)
+ }
+ T(0, true)
+ T(CounterWindowSize+1, true)
+}