shachain: remove unnecessary shachain_index_t
[ccan] / ccan / crypto / shachain / test / run-8bit.c
1 #define SHACHAIN_BITS 8
2
3 #include <ccan/crypto/shachain/shachain.h>
4 /* Include the C files directly. */
5 #include <ccan/crypto/shachain/shachain.c>
6 #include <ccan/tap/tap.h>
7
8 #include <stdio.h>
9
10 #define NUM_TESTS 255
11
12 int main(void)
13 {
14         struct sha256 seed;
15         struct shachain chain;
16         struct sha256 expect[NUM_TESTS+1];
17         int i, j;
18
19         /* This is how many tests you plan to run */
20         plan_tests(66559);
21
22         memset(&seed, 0, sizeof(seed));
23         /* Generate a whole heap; each should be different */
24         for (i = 0; i <= NUM_TESTS; i++) {
25                 shachain_from_seed(&seed, i, &expect[i]);
26                 if (i == 0)
27                         ok1(memcmp(&expect[i], &seed, sizeof(expect[i])) == 0);
28                 else
29                         ok1(memcmp(&expect[i], &expect[i-1], sizeof(expect[i])));
30         }
31
32         shachain_init(&chain);
33
34         for (i = NUM_TESTS; i > 0; i--) {
35                 struct sha256 hash;
36
37                 ok1(shachain_add_hash(&chain, i, &expect[i]));
38                 for (j = i; j <= NUM_TESTS; j++) {
39                         ok1(shachain_get_hash(&chain, j, &hash));
40                         ok1(memcmp(&hash, &expect[j], sizeof(hash)) == 0);
41                 }
42                 ok1(!shachain_get_hash(&chain, i-1, &hash));
43         }
44
45         /* Now add seed. */
46         ok1(shachain_add_hash(&chain, 0, &expect[0]));
47         for (j = 0; j <= NUM_TESTS; j++) {
48                 struct sha256 hash;
49                 ok1(shachain_get_hash(&chain, j, &hash));
50                 ok1(memcmp(&hash, &expect[j], sizeof(hash)) == 0);
51         }
52
53         return exit_status();
54 }