From 9d9f62444da664a1c4a2fd4163b03f9396fafc44 Mon Sep 17 00:00:00 2001
From: Stephen Heumann <stephenheumann@gmail.com>
Date: Tue, 27 Jun 2017 12:45:58 -0500
Subject: [PATCH] Add timing test.

---
 aestest.c | 67 ++++++++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 56 insertions(+), 11 deletions(-)

diff --git a/aestest.c b/aestest.c
index a128e8b..fe476c1 100644
--- a/aestest.c
+++ b/aestest.c
@@ -1,4 +1,7 @@
 #include <stdio.h>
+#include <MiscTool.h>
+#include <Memory.h>
+#include <orca.h>
 
 #include "aes.h"
 
@@ -24,11 +27,13 @@ void aes128_test(void) {
     print_hexbytes("Key:          ", aes_state.key, 16);
     
     aes_expandkey128(&aes_state);
-    
+
+#ifdef PRINT_ROUND_KEYS
     for (i = 1; i <= 10; i++) {
         printf("Round key %2i: ", i);
         print_hexbytes("", aes_state.key + i*16, 16);
     }
+#endif
     
     aes_encrypt(&aes_state);
     
@@ -41,7 +46,7 @@ void aes192_test(void) {
         {0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
         {0},
         {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f,
-	 0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17}
+         0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17}
     };
     
     print_hexbytes("Input:        ", aes_state.data, 16);
@@ -49,10 +54,12 @@ void aes192_test(void) {
     
     aes_expandkey192(&aes_state);
     
+#ifdef PRINT_ROUND_KEYS
     for (i = 1; i <= 12; i++) {
         printf("Round key %2i: ", i);
         print_hexbytes("", aes_state.key + i*16, 16);
     }
+#endif
     
     aes_encrypt(&aes_state);
     
@@ -65,7 +72,7 @@ void aes256_test(void) {
         {0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
         {0},
         {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f,
-	 0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17,0x18,0x19,0x1a,0x1b,0x1c,0x1d,0x1e,0x1f}
+         0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17,0x18,0x19,0x1a,0x1b,0x1c,0x1d,0x1e,0x1f}
     };
     
     print_hexbytes("Input:        ", aes_state.data, 16);
@@ -73,23 +80,61 @@ void aes256_test(void) {
     
     aes_expandkey256(&aes_state);
     
+#ifdef PRINT_ROUND_KEYS
     for (i = 1; i <= 14; i++) {
         printf("Round key %2i: ", i);
         print_hexbytes("", aes_state.key + i*16, 16);
     }
+#endif
     
     aes_encrypt(&aes_state);
     
     print_hexbytes("Output:       ", aes_state.data, 16);
 }
 
-int main(void) {
-	printf("AES-128 test:\n");
-	aes128_test();
-	
-	printf("AES-192 test:\n");
-	aes192_test();
+unsigned long aes128_time_test(unsigned int iters) {
+    unsigned int i;
+    unsigned long tick_count;
+    struct aes_state *aes_state, **aes_state_hndl;
+    static struct aes_state aes_state_init = {
+        {0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
+        {0},
+        {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
+    };
+    
+    aes_state_hndl = (struct aes_state **)NewHandle(sizeof(struct aes_state),
+    		userid(), attrFixed|attrPage|attrBank|attrNoCross, 0x000000);
+    if (toolerror())
+	return 0;
+    aes_state = *aes_state_hndl;
+    *aes_state = aes_state_init;
 
-	printf("AES-256 test:\n");
-	aes256_test();
+    aes_expandkey128(aes_state);
+    
+    tick_count = GetTick();
+    for (i = 0; i < iters; i++) {
+        aes_encrypt(aes_state);
+    }
+    return GetTick() - tick_count;
+}
+
+int main(void) {
+    unsigned long tick_count;
+    unsigned int test_iters = 1000;
+    long double bytes_per_sec;
+
+    printf("AES-128 test:\n");
+    aes128_test();
+    
+    printf("AES-192 test:\n");
+    aes192_test();
+
+    printf("AES-256 test:\n");
+    aes256_test();
+    
+    tick_count = aes128_time_test(test_iters);
+    printf("%u iterations takes %lu ticks", test_iters, tick_count);
+    fflush(stdout);
+    bytes_per_sec = (long double)test_iters * 16 * 60 / tick_count;
+    printf(" (%lf bytes/sec)\n", bytes_per_sec);
 }