detach key expansion routines

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

1 files changed, 5 insertions, 1 deletions

diff --git a/test/aes256ecb_decrypt_block.c b/test/aes256ecb_decrypt_block.c
index f3126fe..a290eed 100644
--- a/test/aes256ecb_decrypt_block.c
+++ b/test/aes256ecb_decrypt_block.c
@@ -21,6 +21,7 @@ int main(int argc, char** argv)
 {
     __declspec(align(16)) AesBlock128 plain, cipher;
     __declspec(align(16)) AesBlock256 key;
+    __declspec(align(16)) Aes256KeySchedule key_schedule, inverted_schedule;
 
     if (argc < 2)
         exit_with_usage(argv[0]);
@@ -31,6 +32,9 @@ int main(int argc, char** argv)
         exit_with_usage(argv[0]);
     }
 
+    aes256_expand_key_schedule(&key, &key_schedule);
+    aes256_invert_key_schedule(&key_schedule, &inverted_schedule);
+
     for (int i = 2; i < argc; ++i)
     {
         if (parse_aes_block128(&cipher, argv[i]) != 0)
@@ -38,7 +42,7 @@ int main(int argc, char** argv)
             fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
             continue;
         }
-        plain = aes256ecb_decrypt(cipher, &key);
+        plain = aes256ecb_decrypt(cipher, &inverted_schedule);
         print_aes_block128(&plain);
     }