1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
4 */
5
6#ifdef DEBUG
7bool __init wg_packet_counter_selftest(void)
8{
9 struct noise_replay_counter *counter;
10 unsigned int test_num = 0, i;
11 bool success = true;
12
13 counter = kmalloc(size: sizeof(*counter), GFP_KERNEL);
14 if (unlikely(!counter)) {
15 pr_err("nonce counter self-test malloc: FAIL\n");
16 return false;
17 }
18
19#define T_INIT do { \
20 memset(counter, 0, sizeof(*counter)); \
21 spin_lock_init(&counter->lock); \
22 } while (0)
23#define T_LIM (COUNTER_WINDOW_SIZE + 1)
24#define T(n, v) do { \
25 ++test_num; \
26 if (counter_validate(counter, n) != (v)) { \
27 pr_err("nonce counter self-test %u: FAIL\n", \
28 test_num); \
29 success = false; \
30 } \
31 } while (0)
32
33 T_INIT;
34 /* 1 */ T(0, true);
35 /* 2 */ T(1, true);
36 /* 3 */ T(1, false);
37 /* 4 */ T(9, true);
38 /* 5 */ T(8, true);
39 /* 6 */ T(7, true);
40 /* 7 */ T(7, false);
41 /* 8 */ T(T_LIM, true);
42 /* 9 */ T(T_LIM - 1, true);
43 /* 10 */ T(T_LIM - 1, false);
44 /* 11 */ T(T_LIM - 2, true);
45 /* 12 */ T(2, true);
46 /* 13 */ T(2, false);
47 /* 14 */ T(T_LIM + 16, true);
48 /* 15 */ T(3, false);
49 /* 16 */ T(T_LIM + 16, false);
50 /* 17 */ T(T_LIM * 4, true);
51 /* 18 */ T(T_LIM * 4 - (T_LIM - 1), true);
52 /* 19 */ T(10, false);
53 /* 20 */ T(T_LIM * 4 - T_LIM, false);
54 /* 21 */ T(T_LIM * 4 - (T_LIM + 1), false);
55 /* 22 */ T(T_LIM * 4 - (T_LIM - 2), true);
56 /* 23 */ T(T_LIM * 4 + 1 - T_LIM, false);
57 /* 24 */ T(0, false);
58 /* 25 */ T(REJECT_AFTER_MESSAGES, false);
59 /* 26 */ T(REJECT_AFTER_MESSAGES - 1, true);
60 /* 27 */ T(REJECT_AFTER_MESSAGES, false);
61 /* 28 */ T(REJECT_AFTER_MESSAGES - 1, false);
62 /* 29 */ T(REJECT_AFTER_MESSAGES - 2, true);
63 /* 30 */ T(REJECT_AFTER_MESSAGES + 1, false);
64 /* 31 */ T(REJECT_AFTER_MESSAGES + 2, false);
65 /* 32 */ T(REJECT_AFTER_MESSAGES - 2, false);
66 /* 33 */ T(REJECT_AFTER_MESSAGES - 3, true);
67 /* 34 */ T(0, false);
68
69 T_INIT;
70 for (i = 1; i <= COUNTER_WINDOW_SIZE; ++i)
71 T(i, true);
72 T(0, true);
73 T(0, false);
74
75 T_INIT;
76 for (i = 2; i <= COUNTER_WINDOW_SIZE + 1; ++i)
77 T(i, true);
78 T(1, true);
79 T(0, false);
80
81 T_INIT;
82 for (i = COUNTER_WINDOW_SIZE + 1; i-- > 0;)
83 T(i, true);
84
85 T_INIT;
86 for (i = COUNTER_WINDOW_SIZE + 2; i-- > 1;)
87 T(i, true);
88 T(0, false);
89
90 T_INIT;
91 for (i = COUNTER_WINDOW_SIZE + 1; i-- > 1;)
92 T(i, true);
93 T(COUNTER_WINDOW_SIZE + 1, true);
94 T(0, false);
95
96 T_INIT;
97 for (i = COUNTER_WINDOW_SIZE + 1; i-- > 1;)
98 T(i, true);
99 T(0, true);
100 T(COUNTER_WINDOW_SIZE + 1, true);
101
102#undef T
103#undef T_LIM
104#undef T_INIT
105
106 if (success)
107 pr_info("nonce counter self-tests: pass\n");
108 kfree(objp: counter);
109 return success;
110}
111#endif
112

source code of linux/drivers/net/wireguard/selftest/counter.c