v / vlib / crypto / internal / subtle / comparison_test.v
65 lines · 58 sloc · 2.14 KB · 19f080ffb8f8f01976692f6b79d9f857c685e109
Raw
1module subtle
2
3fn test_constant_time_byte_eq() {
4 assert constant_time_byte_eq(0, 0) == 1
5 assert constant_time_byte_eq(1, 1) == 1
6 assert constant_time_byte_eq(255, 255) == 1
7 assert constant_time_byte_eq(255, 1) == 0
8 assert constant_time_byte_eq(1, 255) == 0
9 assert constant_time_byte_eq(2, 1) == 0
10}
11
12fn test_constant_time_eq() {
13 assert constant_time_eq(0, 0) == 1
14 assert constant_time_eq(255, 255) == 1
15 assert constant_time_eq(65536, 65536) == 1
16 assert constant_time_eq(-1, -1) == 1
17 assert constant_time_eq(-256, -256) == 1
18 assert constant_time_eq(0, 1) == 0
19}
20
21fn test_constant_time_select() {
22 assert constant_time_select(1, 1, 0) == 1
23 assert constant_time_select(1, 1, 255) == 1
24 assert constant_time_select(1, 1, 255 * 255) == 1
25 assert constant_time_select(1, 2, 0) == 2
26 assert constant_time_select(1, 2, 255) == 2
27 assert constant_time_select(1, 2, 255 * 255) == 2
28
29 assert constant_time_select(0, 1, 0) == 0
30 assert constant_time_select(0, 1, 255) == 255
31 assert constant_time_select(0, 1, 255 * 255) == 255 * 255
32 assert constant_time_select(0, 2, 0) == 0
33 assert constant_time_select(0, 2, 255) == 255
34 assert constant_time_select(0, 2, 255 * 255) == 255 * 255
35}
36
37fn test_constant_time_compare() {
38 assert constant_time_compare([u8(1), 2, 3], [u8(1), 2, 3]) == 1
39 assert constant_time_compare([u8(1), 2, 3], [u8(1), 2, 9]) == 0
40 assert constant_time_compare([u8(1), 2, 3], [u8(1), 2, 3, 4]) == 0
41 assert constant_time_compare([u8(1), 2, 3], [u8(1), 2]) == 0
42}
43
44fn test_constant_time_copy() {
45 y := [u8(3), 4, 5]
46 mut x := [u8(0), 0, 0]
47 constant_time_copy(0, mut x, y)
48 assert x == [u8(0), 0, 0]
49 constant_time_copy(1, mut x, y)
50 assert x == y
51 assert x == [u8(3), 4, 5]
52}
53
54fn test_constant_time_less_or_eq() {
55 assert constant_time_less_or_eq(1, 1) == 1
56 assert constant_time_less_or_eq(1, 2) == 1
57 assert constant_time_less_or_eq(1, 3) == 1
58 assert constant_time_less_or_eq(255, 255) == 1
59 assert constant_time_less_or_eq(255, 256) == 1
60 assert constant_time_less_or_eq(255, 257) == 1
61 assert constant_time_less_or_eq(1, 0) == 0
62 assert constant_time_less_or_eq(2, 1) == 0
63 assert constant_time_less_or_eq(3, 2) == 0
64 assert constant_time_less_or_eq(255, 3) == 0
65}
66