git.ozlabs.org Git - ccan/blob - 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 }