1 #define shachain_index_t uint8_t
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>
15 struct shachain chain;
16 struct sha256 expect[NUM_TESTS+1];
19 /* This is how many tests you plan to run */
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]);
27 ok1(memcmp(&expect[i], &seed, sizeof(expect[i])) == 0);
29 ok1(memcmp(&expect[i], &expect[i-1], sizeof(expect[i])));
32 shachain_init(&chain);
34 for (i = NUM_TESTS; i > 0; i--) {
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);
42 ok1(!shachain_get_hash(&chain, i-1, &hash));
46 ok1(shachain_add_hash(&chain, 0, &expect[0]));
47 for (j = 0; j <= NUM_TESTS; j++) {
49 ok1(shachain_get_hash(&chain, j, &hash));
50 ok1(memcmp(&hash, &expect[j], sizeof(hash)) == 0);