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/aes256cbc_decrypt_block.c b/test/aes256cbc_decrypt_block.c
index bac98e3..bb082d5 100644
--- a/test/aes256cbc_decrypt_block.c
+++ b/test/aes256cbc_decrypt_block.c
@@ -21,6 +21,7 @@ int main(int argc, char** argv)
 {
     __declspec(align(16)) AesBlock128 plain, cipher, iv;
     __declspec(align(16)) AesBlock256 key;
+    __declspec(align(16)) Aes256KeySchedule key_schedule, inverted_schedule;
 
     if (argc < 3)
         exit_with_usage(argv[0]);
@@ -37,6 +38,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 = 3; i < argc; ++i)
     {
         if (parse_aes_block128(&cipher, argv[i]) != 0)
@@ -44,7 +48,7 @@ int main(int argc, char** argv)
             fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
             continue;
         }
-        plain = aes256cbc_decrypt(cipher, &key, &iv);
+        plain = aes256cbc_decrypt(cipher, &inverted_schedule, &iv);
         print_aes_block128(&plain);
     }