diff options
Diffstat (limited to '')
| -rw-r--r-- | test/aes256ecb_decrypt_block.c | 6 |
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); } |