]> git.cryptolib.org Git - avr-crypto-lib.git/blobdiff - test_src/main-aes-test.c
fixing some warnings (AES); simplifyning AES headers (now simply include "aes.h"...
[avr-crypto-lib.git] / test_src / main-aes-test.c
index ef4262662d8c5fc985f0b6a163c4d55828e7bdb6..71b6600a6dcee9f607c84d6649bd58df7f675226 100644 (file)
@@ -1,6 +1,6 @@
 /* main-aes-test.c */
 /*
-    This file is part of the Crypto-avr-lib/microcrypt-lib.
+    This file is part of the AVR-Crypto-Lib.
     Copyright (C) 2008  Daniel Otte (daniel.otte@rub.de)
 
     This program is free software: you can redistribute it and/or modify
 */
 /*
  * AES test-suit
- * 
+ *
 */
 
 #include "config.h"
-#include "serial-tools.h"
-#include "uart.h"
+
+#include "uart_i.h"
 #include "debug.h"
 
-#include "aes.h"
-#include "aes128_enc.h"
-#include "aes128_dec.h"
-#include "aes192_enc.h"
-#include "aes192_dec.h"
-#include "aes256_enc.h"
-#include "aes256_dec.h"
-#include "aes_keyschedule.h"
+#include "aes/aes.h"
 
 #include "nessie_bc_test.h"
 #include "cli.h"
@@ -42,8 +35,9 @@
 #include <stdint.h>
 #include <string.h>
 #include <stdlib.h>
+#include <avr/pgmspace.h>
 
-char* cipher_name = "AES";
+char* algo_name = "AES";
 
 /*****************************************************************************
  *  additional validation-functions                                                                                     *
@@ -52,132 +46,279 @@ char* cipher_name = "AES";
 void testrun_nessie_aes(void){
        nessie_bc_ctx.blocksize_B =  16;
        nessie_bc_ctx.keysize_b   = 128;
-       nessie_bc_ctx.name        = cipher_name;
+       nessie_bc_ctx.name        = algo_name;
        nessie_bc_ctx.ctx_size_B  = sizeof(aes128_ctx_t);
        nessie_bc_ctx.cipher_enc  = (nessie_bc_enc_fpt)aes128_enc;
        nessie_bc_ctx.cipher_dec  = (nessie_bc_dec_fpt)aes128_dec;
        nessie_bc_ctx.cipher_genctx  = (nessie_bc_gen_fpt)aes_init;
        nessie_bc_run();
-       
+
        nessie_bc_ctx.keysize_b   = 192;
        nessie_bc_ctx.ctx_size_B  = sizeof(aes192_ctx_t);
        nessie_bc_ctx.cipher_enc  = (nessie_bc_enc_fpt)aes192_enc;
        nessie_bc_ctx.cipher_dec  = (nessie_bc_dec_fpt)aes192_dec;
        nessie_bc_run();
-       
+
        nessie_bc_ctx.keysize_b   = 256;
        nessie_bc_ctx.ctx_size_B  = sizeof(aes256_ctx_t);
        nessie_bc_ctx.cipher_enc  = (nessie_bc_enc_fpt)aes256_enc;
        nessie_bc_ctx.cipher_dec  = (nessie_bc_dec_fpt)aes256_dec;
-       nessie_bc_run(); 
+       nessie_bc_run();
 }
 
 void testrun_test_aes(void){
-       uint8_t key[16] = { 0x2b, 0x7e, 0x15, 0x16, 
+       uint8_t key[16] = { 0x2b, 0x7e, 0x15, 0x16,
                            0x28, 0xae, 0xd2, 0xa6,
                            0xab, 0xf7, 0x15, 0x88,
                            0x09, 0xcf, 0x4f, 0x3c };
        uint8_t data[16] = { 0x32, 0x43, 0xf6, 0xa8,
-                            0x88, 0x5a, 0x30, 0x8d, 
-                            0x31, 0x31, 0x98, 0xa2, 
+                            0x88, 0x5a, 0x30, 0x8d,
+                            0x31, 0x31, 0x98, 0xa2,
                             0xe0, 0x37, 0x07, 0x34 };
        aes128_ctx_t ctx;
        aes128_init(key, &ctx);
-       uart_putstr_P(PSTR("\r\n\r\n cipher test (FIPS 197):\r\n key:        "));
-       uart_hexdump(key, 16);
-       uart_putstr_P(PSTR("\r\n plaintext:  "));
-       uart_hexdump(data, 16);
+       cli_putstr_P(PSTR("\r\n\r\n cipher test (FIPS 197):\r\n key:        "));
+       cli_hexdump(key, 16);
+       cli_putstr_P(PSTR("\r\n plaintext:  "));
+       cli_hexdump(data, 16);
        aes128_enc(data, &ctx);
-       uart_putstr_P(PSTR("\r\n ciphertext: "));
-       uart_hexdump(data, 16);
-       
-       
+       cli_putstr_P(PSTR("\r\n ciphertext: "));
+       cli_hexdump(data, 16);
+       aes128_dec(data, &ctx);
+       cli_putstr_P(PSTR("\r\n plaintext:  "));
+       cli_hexdump(data, 16);
+
+
 }
 
-void testrun_testkey_aes(void){
-       uint8_t key[16] = { 0x2b, 0x7e, 0x15, 0x16, 
+void testrun_testkey_aes128(void){
+       uint8_t key[16] = { 0x2b, 0x7e, 0x15, 0x16,
                            0x28, 0xae, 0xd2, 0xa6,
                            0xab, 0xf7, 0x15, 0x88,
                            0x09, 0xcf, 0x4f, 0x3c};
        aes128_ctx_t ctx;
        uint8_t i;
        aes128_init(key, &ctx);
-       uart_putstr_P(PSTR("\r\n\r\n keyschedule test (FIPS 197):\r\n key:   "));
-       uart_hexdump(key, 16);
+       cli_putstr_P(PSTR("\r\n\r\n keyschedule test (FIPS 197):\r\n key:   "));
+       cli_hexdump(key, 16);
        for(i=0; i<11; ++i){
-               uart_putstr_P(PSTR("\r\n index: "));
-               uart_putc('0'+i/10);
-               uart_putc('0'+i%10);
-               uart_putstr_P(PSTR(" roundkey "));
-               uart_hexdump(ctx.key[i].ks, 16);
+               cli_putstr_P(PSTR("\r\n index: "));
+               cli_putc('0'+i/10);
+               cli_putc('0'+i%10);
+               cli_putstr_P(PSTR(" roundkey "));
+               cli_hexdump(ctx.key[i].ks, 16);
        }
 }
 
-void testrun_performance_aes(void){
+void testrun_testkey_aes192(void){
+       uint8_t key[24] = { 0x8e, 0x73, 0xb0, 0xf7,
+                           0xda, 0x0e, 0x64, 0x52,
+                           0xc8, 0x10, 0xf3, 0x2b,
+                           0x80, 0x90, 0x79, 0xe5,
+                           0x62, 0xf8, 0xea, 0xd2,
+                           0x52, 0x2c, 0x6b, 0x7b};
+       aes192_ctx_t ctx;
+       uint8_t i;
+       memset(&ctx, 0, sizeof(aes192_ctx_t));
+       aes192_init(key, &ctx);
+       cli_putstr_P(PSTR("\r\n\r\n keyschedule test (FIPS 197):\r\n key:   "));
+       cli_hexdump(key, 24);
+       for(i=0; i<13; ++i){
+               cli_putstr_P(PSTR("\r\n index: "));
+               cli_putc('0'+i/10);
+               cli_putc('0'+i%10);
+               cli_putstr_P(PSTR(" roundkey "));
+               cli_hexdump(ctx.key[i].ks, 16);
+       }
+}
+
+
+void testrun_testkey_aes256(void){
+       uint8_t key[32] = { 0x60, 0x3d, 0xeb, 0x10,
+                           0x15, 0xca, 0x71, 0xbe,
+                           0x2b, 0x73, 0xae, 0xf0,
+                           0x85, 0x7d, 0x77, 0x81,
+                           0x1f, 0x35, 0x2c, 0x07,
+                           0x3b, 0x61, 0x08, 0xd7,
+                           0x2d, 0x98, 0x10, 0xa3,
+                           0x09, 0x14, 0xdf, 0xf4};
+       aes256_ctx_t ctx;
+       uint8_t i;
+       memset(&ctx, 0, sizeof(aes256_ctx_t));
+       aes256_init(key, &ctx);
+       cli_putstr_P(PSTR("\r\n\r\n keyschedule test (FIPS 197):\r\n key:   "));
+       cli_hexdump(key, 32);
+       for(i=0; i<15; ++i){
+               cli_putstr_P(PSTR("\r\n index: "));
+               cli_putc('0'+i/10);
+               cli_putc('0'+i%10);
+               cli_putstr_P(PSTR(" roundkey "));
+               cli_hexdump(ctx.key[i].ks, 16);
+       }
+}
+
+void testrun_testkey_aes(void){
+       testrun_testkey_aes128();
+       testrun_testkey_aes192();
+       testrun_testkey_aes256();
+}
+/*****************************************************************************/
+
+void testrun_performance_aes128(void){
        uint64_t t;
        char str[16];
        uint8_t key[32], data[16];
        aes128_ctx_t ctx;
-       
+
        calibrateTimer();
        print_overhead();
-       
+
        memset(key,  0, 32);
        memset(data, 0, 16);
-       
+
        startTimer(1);
        aes128_init(key, &ctx);
        t = stopTimer();
-       uart_putstr_P(PSTR("\r\n\tctx-gen time: "));
+       cli_putstr_P(PSTR("\r\n\tctx-gen time: "));
        ultoa((unsigned long)t, str, 10);
-       uart_putstr(str);
-       
-       
+       cli_putstr(str);
+
+
        startTimer(1);
        aes128_enc(data, &ctx);
        t = stopTimer();
-       uart_putstr_P(PSTR("\r\n\tencrypt time: "));
+       cli_putstr_P(PSTR("\r\n\tencrypt time: "));
        ultoa((unsigned long)t, str, 10);
-       uart_putstr(str);
-       
-       
+       cli_putstr(str);
+
+
        startTimer(1);
        aes128_dec(data, &ctx);
        t = stopTimer();
-       uart_putstr_P(PSTR("\r\n\tdecrypt time: "));
+       cli_putstr_P(PSTR("\r\n\tdecrypt time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+       cli_putstr_P(PSTR("\r\n"));
+}
+
+
+void testrun_performance_aes192(void){
+       uint64_t t;
+       char str[16];
+       uint8_t key[32], data[16];
+       aes192_ctx_t ctx;
+
+       calibrateTimer();
+       print_overhead();
+
+       memset(key,  0, 32);
+       memset(data, 0, 16);
+
+       startTimer(1);
+       aes192_init(key, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tctx-gen time: "));
        ultoa((unsigned long)t, str, 10);
-       uart_putstr(str);
-       
-       uart_putstr_P(PSTR("\r\n"));
+       cli_putstr(str);
+
+
+       startTimer(1);
+       aes192_enc(data, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tencrypt time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+
+       startTimer(1);
+       aes192_dec(data, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tdecrypt time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+       cli_putstr_P(PSTR("\r\n"));
+}
+
+
+void testrun_performance_aes256(void){
+       uint64_t t;
+       char str[16];
+       uint8_t key[32], data[16];
+       aes256_ctx_t ctx;
+
+       calibrateTimer();
+       print_overhead();
+
+       memset(key,  0, 32);
+       memset(data, 0, 16);
+
+       startTimer(1);
+       aes256_init(key, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tctx-gen time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+
+       startTimer(1);
+       aes256_enc(data, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tencrypt time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+
+       startTimer(1);
+       aes256_dec(data, &ctx);
+       t = stopTimer();
+       cli_putstr_P(PSTR("\r\n\tdecrypt time: "));
+       ultoa((unsigned long)t, str, 10);
+       cli_putstr(str);
+
+       cli_putstr_P(PSTR("\r\n"));
+}
+
+void testrun_performance_aes(void){
+       cli_putstr_P(PSTR("\r\n -=AES Performance Test=-\r\n"));
+       cli_putstr_P(PSTR("\r\n       AES-128\r\n"));
+       testrun_performance_aes128();
+       cli_putstr_P(PSTR("\r\n       AES-192\r\n"));
+       testrun_performance_aes192();
+       cli_putstr_P(PSTR("\r\n       AES-256\r\n"));
+       testrun_performance_aes256();
 }
 /*****************************************************************************
  *  main                                                                                                                                        *
  *****************************************************************************/
 
+const char nessie_str[]      PROGMEM = "nessie";
+const char test_str[]        PROGMEM = "test";
+const char testkey_str[]     PROGMEM = "testkey";
+const char performance_str[] PROGMEM = "performance";
+const char echo_str[]        PROGMEM = "echo";
+
+cmdlist_entry_t cmdlist[] PROGMEM = {
+       { nessie_str,      NULL, testrun_nessie_aes },
+       { test_str,        NULL, testrun_test_aes},
+       { testkey_str,     NULL, testrun_testkey_aes},
+       { performance_str, NULL, testrun_performance_aes},
+       { echo_str,    (void*)1, (void_fpt)echo_ctrl},
+       { NULL,            NULL, NULL}
+};
+
+
 int main (void){
-       char  str[20];
        DEBUG_INIT();
-       uart_putstr("\r\n");
-
-       uart_putstr_P(PSTR("\r\n\r\nCrypto-VS ("));
-       uart_putstr(cipher_name);
-       uart_putstr_P(PSTR(")\r\nloaded and running\r\n"));
-
-       PGM_P    u   = PSTR("nessie\0test\0testkey\0performance\0");
-       void_fpt v[] = {testrun_nessie_aes, 
-                       testrun_test_aes, 
-                       testrun_testkey_aes, 
-                       testrun_performance_aes};
-
-       while(1){ 
-               if (!getnextwordn(str,20)){DEBUG_S("DBG: W1\r\n"); goto error;}
-               if(execcommand_d0_P(str, u, v)<0){
-                       uart_putstr_P(PSTR("\r\nunknown command\r\n"));
-               }
-               continue;
-       error:
-               uart_putstr("ERROR\r\n");
+
+       cli_rx = (cli_rx_fpt)uart0_getc;
+       cli_tx = (cli_tx_fpt)uart0_putc;
+       for(;;){
+               cli_putstr_P(PSTR("\r\n\r\nCrypto-VS ("));
+               cli_putstr(algo_name);
+               cli_putstr_P(PSTR(")\r\nloaded and running\r\n"));
+               cmd_interface(cmdlist);
        }
-       
 }