diff options
Diffstat (limited to 'test/aes256ecb_decrypt_block.c')
-rw-r--r-- | test/aes256ecb_decrypt_block.c | 23 |
1 files changed, 18 insertions, 5 deletions
diff --git a/test/aes256ecb_decrypt_block.c b/test/aes256ecb_decrypt_block.c index 8439044..16ffe77 100644 --- a/test/aes256ecb_decrypt_block.c +++ b/test/aes256ecb_decrypt_block.c @@ -18,8 +18,15 @@ static void exit_with_usage() exit(EXIT_FAILURE); } +static void print_error(AesNI_StatusCode status) +{ + fprintf(stderr, "AesNI error: %s\n", aesni_strerror(status)); +} + int main(int argc, char** argv) { + AesNI_StatusCode status = AESNI_SUCCESS; + for (--argc, ++argv; argc > -1; --argc, ++argv) { AesNI_Block128 plaintext, ciphertext; @@ -29,9 +36,9 @@ int main(int argc, char** argv) if (argc < 1) exit_with_usage(); - if (aesni_is_error(aesni_aes256_parse_key(&key, *argv, NULL))) + if (aesni_is_error(status = aesni_aes256_parse_key(&key, *argv, NULL))) { - fprintf(stderr, "Invalid 256-bit AES block '%s'\n", *argv); + print_error(status); exit_with_usage(); } @@ -43,13 +50,19 @@ int main(int argc, char** argv) if (strcmp("--", *argv) == 0) break; - if (aesni_is_error(aesni_aes_parse_block(&ciphertext, *argv, NULL))) + if (aesni_is_error(status = aesni_aes_parse_block(&ciphertext, *argv, NULL))) { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", *argv); + print_error(status); continue; } + plaintext = aesni_aes256_decrypt_block_ecb(ciphertext, &decryption_keys); - aesni_aes_print_block(&plaintext, NULL); + + if (aesni_is_error(status = aesni_aes_print_block(&plaintext, NULL))) + { + print_error(status); + continue; + } } } |