diff options
| -rw-r--r-- | sw/aes_tester.c | 169 |
1 files changed, 88 insertions, 81 deletions
diff --git a/sw/aes_tester.c b/sw/aes_tester.c index 80acc80..3509950 100644 --- a/sw/aes_tester.c +++ b/sw/aes_tester.c @@ -58,26 +58,18 @@ //------------------------------------------------------------------ // Robs macros. Scary scary. //------------------------------------------------------------------ -#define check(_expr_)\ - do {\ - if ((_expr_) != 0)\ - printf("%s failed\n", #_expr_), 1;\ - } while (0) - -#define set_and_check(_field_)\ - do {\ - printf("Setting %s\n", #_field_);\ - check(tc_write(_field_, one, sizeof(one)));\ - check(tc_read( _field_, res, sizeof(res)));\ - if (memcmp(one, res, sizeof(one)))\ - printf("MISMATCH\n");\ - printf("\n");\ +#define check(_expr_) \ + do { \ + if ((_expr_) != 0) { \ + printf("%s failed\n", #_expr_); \ + exit(1); \ + } \ } while (0) //------------------------------------------------------------------ //------------------------------------------------------------------ -void check_aes_access() +static void check_aes_access(void) { uint8_t name0[4], name1[4], version[4]; @@ -92,23 +84,23 @@ void check_aes_access() //------------------------------------------------------------------ //------------------------------------------------------------------ -void tc_w32(addr, data) +static void tc_w32(const off_t addr, const uint32_t data) { uint8_t w[4]; w[0] = data >> 24 & 0xff; w[1] = data >> 16 & 0xff; w[2] = data >> 8 & 0xff; w[3] = data & 0xff; - tc_write(addr, w, 4); + check(tc_write(addr, w, 4)); } //------------------------------------------------------------------ //------------------------------------------------------------------ -uint32_t tc_r32(addr) +static uint32_t tc_r32(const off_t addr) { uint8_t w[4]; - tc_read(addr, w, 4); + check(tc_read(addr, w, 4)); return (uint32_t)((w[0] << 24) + (w[1] << 16) + (w[2] << 8) + w[3]); } @@ -116,8 +108,8 @@ uint32_t tc_r32(addr) //------------------------------------------------------------------ // dual_block_test //------------------------------------------------------------------ -void dual_block_test(uint8_t keylength, uint32_t *key, uint32_t *block0, - uint32_t *block1, uint32_t *expected) +static void dual_block_test(const uint8_t keylength, const uint32_t *key, const uint32_t *block0, + const uint32_t *block1, const uint32_t *expected) { } @@ -128,22 +120,20 @@ void dual_block_test(uint8_t keylength, uint32_t *key, uint32_t *block0, // // Perform single block tests. //------------------------------------------------------------------ -void single_block_test(uint32_t keylength, uint32_t *key, uint32_t *block, - uint32_t *expected) +static void single_block_test(const uint32_t keylength, const uint32_t *key, + const uint32_t *block, const uint32_t *expected) { uint32_t enc_result[4]; uint32_t dec_result[4]; if (VERBOSE) { - if (keylength == 256) { + if (keylength == 256) printf("Writing key 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x\n", key[0], key[1], key[2], key[3], key[4], key[5], key[6], key[7]); - } - else { + else printf("Writing key 0x%08x 0x%08x 0x%08x 0x%08x\n", key[0], key[1], key[2], key[3]); - } } tc_w32(AES_ADDR_KEY0, key[0]); @@ -152,20 +142,28 @@ void single_block_test(uint32_t keylength, uint32_t *key, uint32_t *block, tc_w32(AES_ADDR_KEY3, key[3]); if (keylength == 256) { - tc_w32(AES_ADDR_KEY0, key[4]); - tc_w32(AES_ADDR_KEY1, key[5]); - tc_w32(AES_ADDR_KEY2, key[6]); - tc_w32(AES_ADDR_KEY3, key[7]); + tc_w32(AES_ADDR_KEY4, key[4]); + tc_w32(AES_ADDR_KEY5, key[5]); + tc_w32(AES_ADDR_KEY6, key[6]); + tc_w32(AES_ADDR_KEY7, key[7]); } - if (CHECK_WRITE) - { - printf("Reading back key: 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x\n", - tc_r32(AES_ADDR_KEY0), tc_r32(AES_ADDR_KEY1), - tc_r32(AES_ADDR_KEY2), tc_r32(AES_ADDR_KEY3), - tc_r32(AES_ADDR_KEY4), tc_r32(AES_ADDR_KEY5), - tc_r32(AES_ADDR_KEY6), tc_r32(AES_ADDR_KEY7)); - } + if (CHECK_WRITE) { + const uint32_t + k0 = tc_r32(AES_ADDR_KEY0), k1 = tc_r32(AES_ADDR_KEY1), + k2 = tc_r32(AES_ADDR_KEY2), k3 = tc_r32(AES_ADDR_KEY3), + k4 = tc_r32(AES_ADDR_KEY4), k5 = tc_r32(AES_ADDR_KEY5), + k6 = tc_r32(AES_ADDR_KEY6), k7 = tc_r32(AES_ADDR_KEY7); + const int + ok1 = k0 == key[0] && k1 == key[1] && k2 == key[2] && k3 == key[3], + ok2 = k4 == key[4] && k5 == key[5] && k6 == key[6] && k7 == key[7]; + if (keylength == 256) + printf("Reading back key: 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x 0x%08x %s\n", + k0, k1, k2, k3, k4, k5, k6, k7, (ok1 && ok2) ? "OK" : "BAD"); + else + printf("Reading back key: 0x%08x 0x%08x 0x%08x 0x%08x %s\n", + k0, k1, k2, k3, ok1 ? "OK" : "BAD"); + } // Performing init i.e. key expansion, printf("Doing key init\n"); @@ -178,23 +176,27 @@ void single_block_test(uint32_t keylength, uint32_t *key, uint32_t *block, if (VERBOSE) - printf("Writing block 0x%08x 0x%08x 0x%08x 0x%08x\n", - block[0], block[1], block[2], block[3]); + printf("Writing block 0x%08x 0x%08x 0x%08x 0x%08x\n", + block[0], block[1], block[2], block[3]); tc_w32(AES_ADDR_BLOCK0, block[0]); tc_w32(AES_ADDR_BLOCK1, block[1]); tc_w32(AES_ADDR_BLOCK2, block[2]); tc_w32(AES_ADDR_BLOCK3, block[3]); - if (CHECK_WRITE) - { - printf("Reading back block: 0x%08x 0x%08x 0x%08x 0x%08x\n", - tc_r32(AES_ADDR_BLOCK0), tc_r32(AES_ADDR_BLOCK1), - tc_r32(AES_ADDR_BLOCK2), tc_r32(AES_ADDR_BLOCK3)); - } + if (CHECK_WRITE) { + const uint32_t + b0 = tc_r32(AES_ADDR_BLOCK0), b1 = tc_r32(AES_ADDR_BLOCK1), + b2 = tc_r32(AES_ADDR_BLOCK2), b3 = tc_r32(AES_ADDR_BLOCK3); + const int + ok = b0 == block[0] && b1 == block[1] && b2 == block[2] && b3 == block[3]; + printf("Reading back block: 0x%08x 0x%08x 0x%08x 0x%08x %s\n", + b0, b1, b2, b3, ok ? "OK" : "BAD"); + } if (VERBOSE) printf("Starting single block encipher operation\n"); + if (keylength == 256) tc_w32(AES_ADDR_CONFIG, 0x00000003); else @@ -205,12 +207,13 @@ void single_block_test(uint32_t keylength, uint32_t *key, uint32_t *block, if (VERBOSE) printf("Checking ready: 0x%08x\n", tc_r32(AES_ADDR_STATUS)); - tc_wait_ready(AES_ADDR_STATUS); + + check(tc_wait_ready(AES_ADDR_STATUS)); if (VERBOSE) - printf("Ready seen. Result: 0x%08x 0x%08x 0x%08x 0x%08x\n", - tc_r32(AES_ADDR_RESULT0), tc_r32(AES_ADDR_RESULT1), - tc_r32(AES_ADDR_RESULT2), tc_r32(AES_ADDR_RESULT3)); + printf("Ready seen. Result: 0x%08x 0x%08x 0x%08x 0x%08x\n", + tc_r32(AES_ADDR_RESULT0), tc_r32(AES_ADDR_RESULT1), + tc_r32(AES_ADDR_RESULT2), tc_r32(AES_ADDR_RESULT3)); enc_result[0] = tc_r32(AES_ADDR_RESULT0); enc_result[1] = tc_r32(AES_ADDR_RESULT1); @@ -229,56 +232,60 @@ void single_block_test(uint32_t keylength, uint32_t *key, uint32_t *block, tc_w32(AES_ADDR_CONFIG, 0x00000000); tc_w32(AES_ADDR_CTRL, 0x00000002); - tc_wait_ready(AES_ADDR_STATUS); + check(tc_wait_ready(AES_ADDR_STATUS)); dec_result[0] = tc_r32(AES_ADDR_RESULT0); dec_result[1] = tc_r32(AES_ADDR_RESULT1); dec_result[2] = tc_r32(AES_ADDR_RESULT2); dec_result[3] = tc_r32(AES_ADDR_RESULT3); - - printf("Generated cipher block:\n"); - printf("0x%08x 0x%08x 0x%08x 0x%08x\n", + printf("Generated cipher block: 0x%08x 0x%08x 0x%08x 0x%08x\n", enc_result[0], enc_result[1], enc_result[2], enc_result[3]); - printf("Expected cipher block:\n"); - printf("0x%08x 0x%08x 0x%08x 0x%08x\n", + printf("Expected cipher block: 0x%08x 0x%08x 0x%08x 0x%08x\n", expected[0], expected[1], expected[2], expected[3]); + if (enc_result[0] == expected[0] && enc_result[1] == expected[1] && + enc_result[2] == expected[2] && enc_result[3] == expected[3]) + printf("OK\n"); + else + printf("BAD\n"); printf("\n"); - printf("Generated decipher block:\n"); - printf("0x%08x 0x%08x 0x%08x 0x%08x\n", + printf("Generated decipher block: 0x%08x 0x%08x 0x%08x 0x%08x\n", dec_result[0], dec_result[1], dec_result[2], dec_result[3]); - printf("Expected decipher block:\n"); - printf("0x%08x 0x%08x 0x%08x 0x%08x\n", + printf("Expected decipher block: 0x%08x 0x%08x 0x%08x 0x%08x\n", block[0], block[1], block[2], block[3]); + if (dec_result[0] == block[0] && dec_result[1] == block[1] && + dec_result[2] == block[2] && dec_result[3] == block[3]) + printf("OK\n"); + else + printf("BAD\n"); printf("\n"); - } //------------------------------------------------------------------ // run_nist_tests() //------------------------------------------------------------------ -void run_nist_tests() +static void run_nist_tests() { - uint32_t nist_aes128_key[4] = {0x2b7e1516, 0x28aed2a6, 0xabf71588, 0x09cf4f3c}; - uint32_t nist_aes256_key[8] = {0x603deb10, 0x15ca71be, 0x2b73aef0, 0x857d7781, - 0x1f352c07, 0x3b6108d7, 0x2d9810a3, 0x0914dff4}; - - uint32_t nist_plaintext0[4] = {0x6bc1bee2, 0x2e409f96, 0xe93d7e11, 0x7393172a}; - uint32_t nist_plaintext1[4] = {0xae2d8a57, 0x1e03ac9c, 0x9eb76fac, 0x45af8e51}; - uint32_t nist_plaintext2[4] = {0x30c81c46, 0xa35ce411, 0xe5fbc119, 0x1a0a52ef}; - uint32_t nist_plaintext3[4] = {0xf69f2445, 0xdf4f9b17, 0xad2b417b, 0xe66c3710}; - - uint32_t nist_ecb_128_enc_expected0[4] = {0x3ad77bb4, 0x0d7a3660, 0xa89ecaf3, 0x2466ef97}; - uint32_t nist_ecb_128_enc_expected1[4] = {0xf5d3d585, 0x03b9699d, 0xe785895a, 0x96fdbaaf}; - uint32_t nist_ecb_128_enc_expected2[4] = {0x43b1cd7f, 0x598ece23, 0x881b00e3, 0xed030688}; - uint32_t nist_ecb_128_enc_expected3[4] = {0x7b0c785e, 0x27e8ad3f, 0x82232071, 0x04725dd4}; - - uint32_t nist_ecb_256_enc_expected0[4] = {0xf3eed1bd, 0xb5d2a03c, 0x064b5a7e, 0x3db181f8}; - uint32_t nist_ecb_256_enc_expected1[4] = {0x591ccb10, 0xd410ed26, 0xdc5ba74a, 0x31362870}; - uint32_t nist_ecb_256_enc_expected2[4] = {0xb6ed21b9, 0x9ca6f4f9, 0xf153e7b1, 0xbeafed1d}; - uint32_t nist_ecb_256_enc_expected3[4] = {0x23304b7a, 0x39f9f3ff, 0x067d8d8f, 0x9e24ecc7}; + static const uint32_t nist_aes128_key[4] = {0x2b7e1516, 0x28aed2a6, 0xabf71588, 0x09cf4f3c}; + static const uint32_t nist_aes256_key[8] = {0x603deb10, 0x15ca71be, 0x2b73aef0, 0x857d7781, + 0x1f352c07, 0x3b6108d7, 0x2d9810a3, 0x0914dff4}; + + static const uint32_t nist_plaintext0[4] = {0x6bc1bee2, 0x2e409f96, 0xe93d7e11, 0x7393172a}; + static const uint32_t nist_plaintext1[4] = {0xae2d8a57, 0x1e03ac9c, 0x9eb76fac, 0x45af8e51}; + static const uint32_t nist_plaintext2[4] = {0x30c81c46, 0xa35ce411, 0xe5fbc119, 0x1a0a52ef}; + static const uint32_t nist_plaintext3[4] = {0xf69f2445, 0xdf4f9b17, 0xad2b417b, 0xe66c3710}; + + static const uint32_t nist_ecb_128_enc_expected0[4] = {0x3ad77bb4, 0x0d7a3660, 0xa89ecaf3, 0x2466ef97}; + static const uint32_t nist_ecb_128_enc_expected1[4] = {0xf5d3d585, 0x03b9699d, 0xe785895a, 0x96fdbaaf}; + static const uint32_t nist_ecb_128_enc_expected2[4] = {0x43b1cd7f, 0x598ece23, 0x881b00e3, 0xed030688}; + static const uint32_t nist_ecb_128_enc_expected3[4] = {0x7b0c785e, 0x27e8ad3f, 0x82232071, 0x04725dd4}; + + static const uint32_t nist_ecb_256_enc_expected0[4] = {0xf3eed1bd, 0xb5d2a03c, 0x064b5a7e, 0x3db181f8}; + static const uint32_t nist_ecb_256_enc_expected1[4] = {0x591ccb10, 0xd410ed26, 0xdc5ba74a, 0x31362870}; + static const uint32_t nist_ecb_256_enc_expected2[4] = {0xb6ed21b9, 0x9ca6f4f9, 0xf153e7b1, 0xbeafed1d}; + static const uint32_t nist_ecb_256_enc_expected3[4] = {0x23304b7a, 0x39f9f3ff, 0x067d8d8f, 0x9e24ecc7}; printf("Running NIST single block test.\n"); |