]> git.cryptolib.org Git - avr-crypto-lib.git/blobdiff - test_src/main-dsa-test.c
stack usage measurement
[avr-crypto-lib.git] / test_src / main-dsa-test.c
diff --git a/test_src/main-dsa-test.c b/test_src/main-dsa-test.c
new file mode 100644 (file)
index 0000000..acbffa9
--- /dev/null
@@ -0,0 +1,191 @@
+/* main-dsa-test.c */
+/*
+    This file is part of the AVR-Crypto-Lib.
+    Copyright (C) 2010 Daniel Otte (daniel.otte@rub.de)
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+/*
+ * DSA test-suit
+ *
+*/
+
+#include "config.h"
+
+#include "uart_i.h"
+#include "debug.h"
+
+#include "noekeon.h"
+#include "noekeon_prng.h"
+#include "bigint.h"
+#include "bigint_io.h"
+#include "dsa.h"
+#include "dsa_key_blob.h"
+
+#include "cli.h"
+#include "performance_test.h"
+#include "hfal_sha1.h"
+#include "base64_enc.h"
+#include "base64_dec.h"
+#include <stdint.h>
+#include <string.h>
+#include <stdlib.h>
+
+char* algo_name = "DSA";
+
+/*****************************************************************************
+ *  additional validation-functions                                                                                     *
+ *****************************************************************************/
+
+dsa_ctx_t dsa_ctx;
+
+void load_fix_dsa(void){
+       load_dsa_key_blob(&dsa_ctx);
+}
+
+void dsa_print_item(bigint_t* a, PGM_P pstr){
+       uint8_t *p;
+       cli_putstr_P(PSTR("\r\n"));
+       cli_putstr_P(pstr);
+       cli_putstr_P(PSTR(": "));
+       uint16_t i;
+       p = a->wordv + a->length_B -1;
+       for(i=0; i<a->length_B-1; ++i){
+               if(i%16==0){
+                       cli_putstr_P(PSTR("\r\n    "));
+               }
+               cli_hexdump(p, 1);
+               cli_putc(':');
+               --p;
+       }
+       if(i%16==0){
+               cli_putstr_P(PSTR("\r\n    "));
+       }
+       cli_hexdump(p, 1);
+}
+
+void dsa_print_signature_b64(dsa_signature_t* s){
+       uint16_t size_r, size_s, size_o, i,j;
+       size_r = s->r.length_B +2;
+       size_s = s->s.length_B +2;
+       size_o = size_r + size_s +2;
+       uint8_t bin_b[size_o];
+       bin_b[0] = 0x30;
+       bin_b[1] = size_o -2;
+       bin_b[2] = 0x02;
+       bin_b[3] = size_r-2;
+       j=4;
+       for(i=s->r.length_B; i>0;  --i){
+               bin_b[j++] = s->r.wordv[i-1];
+       }
+       bin_b[j++] = 0x02;
+       bin_b[j++] = size_s -2;
+       for(i=s->s.length_B; i>0;  --i){
+               bin_b[j++] = s->s.wordv[i-1];
+       }
+       char b64_b[size_o*4/3+5];
+       base64enc(b64_b, bin_b, size_o);
+       cli_putstr(b64_b);
+}
+
+void dsa_print_ctx(dsa_ctx_t* ctx){
+       dsa_print_item(&(ctx->priv), PSTR("private"));
+       dsa_print_item(&(ctx->pub), PSTR("public"));
+       dsa_print_item(&(ctx->domain.p), PSTR("P"));
+       dsa_print_item(&(ctx->domain.q), PSTR("Q"));
+       dsa_print_item(&(ctx->domain.g), PSTR("G"));
+}
+
+void dsa_print_signature(const dsa_signature_t* sig){
+       cli_putstr_P(PSTR("\r\nDSA-Signature:\r\n r:"));
+       bigint_print_hex(&(sig->r));
+       cli_putstr_P(PSTR("\r\n s:"));
+       bigint_print_hex(&(sig->s));
+}
+
+void quick_test(void){
+       dsa_signature_t dsa_sig;
+       uint8_t i, t=0, message[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef};
+       load_fix_dsa();
+       uint8_t dsa_sig_s_b[dsa_ctx.domain.q.length_B],
+               dsa_sig_r_b[dsa_ctx.domain.q.length_B];
+       dsa_print_ctx(&dsa_ctx);
+       dsa_sig.r.wordv = dsa_sig_r_b;
+       dsa_sig.s.wordv = dsa_sig_s_b;
+       cli_putstr_P(PSTR("\r\n\r\n=== DSA QUICK TEST ==="));
+       for(i=0; i<8; ++i){
+               cli_putstr_P(PSTR("\r\n"));
+               cli_putc('1'+i);
+               cli_putstr_P(PSTR(": message: "));
+               if (i){
+                       cli_hexdump(message, i);
+               }else{
+                       cli_putstr_P(PSTR("<empty>"));
+               }
+               cli_putstr_P(PSTR("\r\n computing signature ... "));
+               dsa_sign_message(&dsa_sig, message, i*8, &sha1_desc, &dsa_ctx, random8);
+               dsa_print_signature(&dsa_sig);
+               cli_putstr_P(PSTR("\r\n base64:\r\n--- SIGNATURE ---\r\n "));
+               dsa_print_signature_b64(&dsa_sig);
+               cli_putstr_P(PSTR("\r\n verifying signature ... "));
+               t = dsa_verify_message(&dsa_sig, message, i*8, &sha1_desc, &dsa_ctx);
+               cli_putstr_P(PSTR("\r\n verification: "));
+               if(t==DSA_SIGNATURE_OK){
+                       cli_putstr_P(PSTR("[PASS]"));
+               }else{
+                       cli_putstr_P(PSTR("[FAIL]"));
+               }
+       }
+}
+
+void reset_prng(void){
+       uint8_t buf[16];
+       memset(buf, 0, 16);
+       random_seed(buf);
+       cli_putstr_P(PSTR("\r\nPRNG reset"));
+}
+
+void testrun_performance_bigint(void){
+
+}
+/*****************************************************************************
+ *  main                                                                                                                                        *
+ *****************************************************************************/
+
+const char echo_test_str[]        PROGMEM = "echo-test";
+const char reset_prng_str[]       PROGMEM = "reset-prng";
+const char quick_test_str[]       PROGMEM = "quick-test";
+const char performance_str[]      PROGMEM = "performance";
+const char echo_str[]             PROGMEM = "echo";
+
+cmdlist_entry_t cmdlist[] PROGMEM = {
+       { reset_prng_str,       NULL, reset_prng                    },
+       { quick_test_str,       NULL, quick_test                    },
+       { performance_str,      NULL, testrun_performance_bigint    },
+       { echo_str,         (void*)1, (void_fpt)echo_ctrl           },
+       { NULL,                 NULL, NULL                          }
+};
+
+int main (void){
+       DEBUG_INIT();
+
+       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);
+       }
+}