detach key expansion routines

Block encryption/decryption routines now don't expand key schedules on
each call.

1 files changed, 4 insertions, 1 deletions

diff --git a/test/aes128cbc_encrypt_block.c b/test/aes128cbc_encrypt_block.c
index 31bb479..e35a1bf 100644
--- a/test/aes128cbc_encrypt_block.c
+++ b/test/aes128cbc_encrypt_block.c
@@ -20,6 +20,7 @@ static void exit_with_usage(const char* argv0)
 int main(int argc, char** argv)
 {
     __declspec(align(16)) AesBlock128 plain, key, cipher, iv;
+    __declspec(align(16)) Aes128KeySchedule key_schedule;
 
     if (argc < 3)
         exit_with_usage(argv[0]);
@@ -36,6 +37,8 @@ int main(int argc, char** argv)
         exit_with_usage(argv[0]);
     }
 
+    aes128_expand_key_schedule(key, &key_schedule);
+
     for (int i = 3; i < argc; ++i)
     {
         if (parse_aes_block128(&plain, argv[i]) != 0)
@@ -43,7 +46,7 @@ int main(int argc, char** argv)
             fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
             continue;
         }
-        cipher = aes128cbc_encrypt(plain, key, &iv);
+        cipher = aes128cbc_encrypt(plain, &key_schedule, &iv);
         print_aes_block128(&cipher);
     }