diff options
Diffstat (limited to 'test/aes256ofb_encrypt_block.c')
| -rw-r--r-- | test/aes256ofb_encrypt_block.c | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/test/aes256ofb_encrypt_block.c b/test/aes256ofb_encrypt_block.c index 0018e9a..7dc01b8 100644 --- a/test/aes256ofb_encrypt_block.c +++ b/test/aes256ofb_encrypt_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, iv; @@ -29,15 +36,15 @@ int main(int argc, char** argv) if (argc < 2) 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(); } - if (aesni_is_error(aesni_aes_parse_block(&iv, argv[1], NULL))) + if (aesni_is_error(status = aesni_aes_parse_block(&iv, argv[1], NULL))) { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[1]); + print_error(status); exit_with_usage(); } @@ -48,13 +55,19 @@ int main(int argc, char** argv) if (strcmp("--", *argv) == 0) break; - if (aesni_is_error(aesni_aes_parse_block(&plaintext, *argv, NULL))) + if (aesni_is_error(status = aesni_aes_parse_block(&plaintext, *argv, NULL))) { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", *argv); + print_error(status); continue; } + ciphertext = aesni_aes256_encrypt_block_ofb(plaintext, &encryption_keys, iv, &iv); - aesni_aes_print_block(&ciphertext, NULL); + + if (aesni_is_error(status = aesni_aes_print_block(&ciphertext, NULL))) + { + print_error(status); + continue; + } } } |