bertos/sec/kdf/pbkdf2_test.c

   1 #include <string.h>
   2 #include <cpu/detect.h>
   3 #include <cfg/test.h>
   4 #include <cfg/debug.h>
   5 #include <sec/mac/hmac.h>
   6 #include <sec/hash/sha1.h>
   7 #include <sec/kdf/pbkdf2.h>
   8
   9 int PBKDF2_testSetup(void)
  10 {
  11         kdbg_init();
  12         return 0;
  13 }
  14
  15 int PBKDF2_testTearDown(void)
  16 {
  17         return 0;
  18 }
  19
  20 int PBKDF2_testRun(void)
  21 {
  22         Kdf *kdf = PBKDF2_stackinit(HMAC_stackinit(SHA1_stackinit()));
  23
  24         uint8_t res[32];
  25
  26         PBKDF2_set_iterations(kdf, 1);
  27         kdf_begin(kdf, "password", 8, (const uint8_t*)"salt", 4);
  28         kdf_read(kdf, res, 20);
  29         ASSERT(memcmp(res, "\x0c\x60\xc8\x0f\x96\x1f\x0e\x71\xf3\xa9\xb5\x24\xaf\x60\x12\x06\x2f\xe0\x37\xa6", 20) == 0);
  30
  31         PBKDF2_set_iterations(kdf, 2);
  32         kdf_begin(kdf, "password", 8, (const uint8_t*)"salt", 4);
  33         kdf_read(kdf, res, 20);
  34         ASSERT(memcmp(res, "\xea\x6c\x01\x4d\xc7\x2d\x6f\x8c\xcd\x1e\xd9\x2a\xce\x1d\x41\xf0\xd8\xde\x89\x57", 20) == 0);
  35
  36         PBKDF2_set_iterations(kdf, 4096);
  37         kdf_begin(kdf, "password", 8, (const uint8_t*)"salt", 4);
  38         kdf_read(kdf, res, 20);
  39         ASSERT(memcmp(res, "\x4b\x00\x79\x01\xb7\x65\x48\x9a\xbe\xad\x49\xd9\x26\xf7\x21\xd0\x65\xa4\x29\xc1", 20) == 0);
  40
  41 #if CPU_X86
  42         // Too slow for an embedded system...
  43         PBKDF2_set_iterations(kdf, 16777216);
  44         kdf_begin(kdf, "password", 8, (const uint8_t*)"salt", 4);
  45         kdf_read(kdf, res, 20);
  46         ASSERT(memcmp(res, "\xee\xfe\x3d\x61\xcd\x4d\xa4\xe4\xe9\x94\x5b\x3d\x6b\xa2\x15\x8c\x26\x34\xe9\x84", 20) == 0);
  47 #endif
  48
  49         PBKDF2_set_iterations(kdf, 4096);
  50         kdf_begin(kdf, "pass\0word", 9, (const uint8_t*)"sa\0lt", 5);
  51         kdf_read(kdf, res, 16);
  52         ASSERT(memcmp(res, "\x56\xfa\x6a\xa7\x55\x48\x09\x9d\xcc\x37\xd7\xf0\x34\x25\xe0\xc3", 16) == 0);
  53
  54         PBKDF2_set_iterations(kdf, 4096);
  55         kdf_begin(kdf, "passwordPASSWORDpassword", 24, (const uint8_t*)"saltSALTsaltSALTsaltSALTsaltSALTsalt", 36);
  56         kdf_read(kdf, res, 25);
  57         ASSERT(memcmp(res, "\x3d\x2e\xec\x4f\xe4\x1c\x84\x9b\x80\xc8\xd8\x36\x62\xc0\xe4\x4a\x8b\x29\x1a\x96\x4c\xf2\xf0\x70\x38", 25) == 0);
  58
  59         return 0;
  60 }
  61
  62 TEST_MAIN(PBKDF2);