From 8fdfe033a9cb297b80877894e9d3b148fc99d935 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sat, 6 Jun 2015 02:26:05 +0300 Subject: test: enable the utilities to accept many keys --- test/aes128cbc_decrypt.c | 57 +++++++++++++++++++++++++++--------------------- 1 file changed, 32 insertions(+), 25 deletions(-) (limited to 'test/aes128cbc_decrypt.c') diff --git a/test/aes128cbc_decrypt.c b/test/aes128cbc_decrypt.c index f5f2c35..a97d00b 100644 --- a/test/aes128cbc_decrypt.c +++ b/test/aes128cbc_decrypt.c @@ -10,45 +10,52 @@ #include #include +#include -static void exit_with_usage(const char* argv0) +static void exit_with_usage() { - printf("Usage: %s KEY INIT_VECTOR [CIPHER...]\n", argv0); + puts("Usage: aes128cbc_decrypt.exe KEY0 IV0 [CIPHER0...] [-- KEY1 IV1 [CIPHER1...]...]"); exit(EXIT_FAILURE); } int main(int argc, char** argv) { - AesBlock128 plain, key, cipher, iv; - Aes128KeySchedule key_schedule, inverted_schedule; + for (--argc, ++argv; argc > -1; --argc, ++argv) + { + AesBlock128 plain, key, cipher, iv; + Aes128KeySchedule key_schedule, inverted_schedule; - if (argc < 3) - exit_with_usage(argv[0]); + if (argc < 2) + exit_with_usage(); - if (parse_aes_block128(&key, argv[1]) != 0) - { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[1]); - exit_with_usage(argv[0]); - } + if (parse_aes_block128(&key, *argv) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", *argv); + exit_with_usage(); + } - if (parse_aes_block128(&iv, argv[2]) != 0) - { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]); - exit_with_usage(argv[0]); - } + if (parse_aes_block128(&iv, argv[1]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[1]); + exit_with_usage(); + } - aes128_expand_key_schedule(key, &key_schedule); - aes128_invert_key_schedule(&key_schedule, &inverted_schedule); + aes128_expand_key_schedule(key, &key_schedule); + aes128_invert_key_schedule(&key_schedule, &inverted_schedule); - for (int i = 3; i < argc; ++i) - { - if (parse_aes_block128(&cipher, argv[i]) != 0) + for (argc -= 2, argv += 2; argc > 0; --argc, ++argv) { - fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); - continue; + if (strcmp("--", *argv) == 0) + break; + + if (parse_aes_block128(&cipher, *argv) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", *argv); + continue; + } + plain = aes128cbc_decrypt(cipher, &inverted_schedule, iv, &iv); + print_aes_block128(&plain); } - plain = aes128cbc_decrypt(cipher, &inverted_schedule, iv, &iv); - print_aes_block128(&plain); } return 0; -- cgit v1.2.3