1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#include <stdio.h>
#include "test_util.h"
#include "bitstring.h"
uint8_t test_append() {
uint8_t c[5] = {0};
NtruBitStr b0 = {{0}, 0, 0};;
c[0] = 78;
ntru_append(&b0, (uint8_t*)&c, 1);
uint8_t exp0[] = {78};
uint8_t valid = equals_arr(b0.buf, exp0, 1);
NtruBitStr b1 = {{0}, 0, 0};;
c[1] = 251;
ntru_append(&b1, (uint8_t*)&c, 2);
uint8_t exp1[] = {78, 251};
valid &= equals_arr(b1.buf, exp1, 2);
NtruBitStr b2 = {{0}, 0, 0};;
c[2] = 127;
ntru_append(&b2, (uint8_t*)&c, 3);
uint8_t exp2[] = {78, 251, 127};
valid &= equals_arr(b2.buf, exp2, 3);
NtruBitStr b3 = {{0}, 0, 0};;
c[3] = 0;
ntru_append(&b3, (uint8_t*)&c, 4);
uint8_t exp3[] = {78, 251, 127, 0};
valid &= equals_arr(b3.buf, exp3, 4);
NtruBitStr b4 = {{0}, 0, 0};;
c[4] = 100;
ntru_append(&b4, (uint8_t*)&c, 5);
uint8_t exp4[] = {78, 251, 127, 0, 100};
valid &= equals_arr(b4.buf, exp4, 5);
return valid;
}
uint8_t test_trailing() {
NtruBitStr b0 = {{0}, 0, 0};
uint8_t c0[] = {78};
ntru_append(&b0, (uint8_t*)&c0, 1);
NtruBitStr b0_trail;
ntru_trailing(&b0, 3, &b0_trail);
uint8_t exp0[] = {6};
uint8_t valid = equals_arr(b0_trail.buf, exp0, 1);
NtruBitStr b1 = {{0}, 0, 0};
uint8_t c1[] = {78, 251};
ntru_append(&b1, (uint8_t*)&c1, 2);
NtruBitStr b1_trail;
ntru_trailing(&b1, 9, &b1_trail);
uint8_t exp1[] = {78, 1};
valid &= equals_arr(b1_trail.buf, exp1, 2);
uint8_t c2[] = {100};
ntru_append(&b1_trail, (uint8_t*)&c2, 1);
uint8_t exp2[] = {78, 201};
valid &= equals_arr((uint8_t*)&b1_trail.buf, (uint8_t*)&exp2, 2);
NtruBitStr b2_trail;
ntru_trailing(&b1_trail, 13, &b2_trail);
uint8_t exp3[] = {78, 9};
valid &= equals_arr(b2_trail.buf, exp3, 2);
ntru_trailing(&b1_trail, 11, &b2_trail);
uint8_t exp4[] = {78, 1};
valid &= equals_arr(b2_trail.buf, exp4, 2);
uint8_t c3[] = {100};
ntru_append(&b2_trail, (uint8_t*)&c3, 1);
uint8_t exp5[] = {78, 33, 3};
valid &= equals_arr(b2_trail.buf, exp5, 3);
ntru_trailing(&b2_trail, 16, &b1_trail);
uint8_t exp6[] = {78, 33};
valid &= equals_arr(b1_trail.buf, exp6, 2);
return valid;
}
uint8_t test_leading() {
uint8_t valid = 1;
NtruBitStr b0 = {{0}, 0, 0};
uint8_t c0[] = {78, 42};
ntru_append(&b0, (uint8_t*)&c0, 2);
valid &= ntru_leading(&b0, 3) == 1;
valid &= ntru_leading(&b0, 9) == 84;
valid &= ntru_leading(&b0, 11) == 338;
NtruBitStr b1;
ntru_trailing(&b0, 11, &b1);
uint8_t exp0[] = {78, 2};
valid &= equals_arr(b1.buf, exp0, 2);
valid &= ntru_leading(&b1, 11) == 590;
valid &= ntru_leading(&b1, 5) == 9;
uint8_t c1[] = {115};
ntru_append(&b1, (uint8_t*)&c1, 1);
valid &= ntru_leading(&b1, 9) == 230;
valid &= ntru_leading(&b1, 11) == 922;
uint8_t c2[] = {220};
ntru_append(&b1, (uint8_t*)&c2, 1);
valid &= ntru_leading(&b1, 6) == 55;
return valid;
}
uint8_t test_bitstring() {
uint8_t valid = test_append();
valid &= test_trailing();
valid &= test_leading();
print_result("test_bitstring", valid);
return valid;
}
|
