aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/test/aes128ecb_encrypt.c
diff options
context:
space:
mode:
Diffstat (limited to 'test/aes128ecb_encrypt.c')
-rw-r--r--test/aes128ecb_encrypt.c45
1 files changed, 26 insertions, 19 deletions
diff --git a/test/aes128ecb_encrypt.c b/test/aes128ecb_encrypt.c
index b06f60b..6fde333 100644
--- a/test/aes128ecb_encrypt.c
+++ b/test/aes128ecb_encrypt.c
@@ -10,38 +10,45 @@
#include <stdio.h>
#include <stdlib.h>
+#include <string.h>
-static void exit_with_usage(const char* argv0)
+static void exit_with_usage()
{
- printf("Usage: %s KEY [PLAIN...]\n", argv0);
+ puts("Usage: aes128ecb_encrypt.exe KEY0 [PLAIN0...] [-- KEY1 [PLAIN1...]...]");
exit(EXIT_FAILURE);
}
int main(int argc, char** argv)
{
- AesBlock128 plain, key, cipher;
- Aes128KeySchedule key_schedule;
+ for (--argc, ++argv; argc > -1; --argc, ++argv)
+ {
+ AesBlock128 plain, key, cipher;
+ Aes128KeySchedule key_schedule;
- if (argc < 2)
- exit_with_usage(argv[0]);
+ if (argc < 1)
+ 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();
+ }
- aes128_expand_key_schedule(key, &key_schedule);
+ aes128_expand_key_schedule(key, &key_schedule);
- for (int i = 2; i < argc; ++i)
- {
- if (parse_aes_block128(&plain, argv[i]) != 0)
+ for (--argc, ++argv; 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(&plain, *argv) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", *argv);
+ continue;
+ }
+ cipher = aes128ecb_encrypt(plain, &key_schedule);
+ print_aes_block128(&cipher);
}
- cipher = aes128ecb_encrypt(plain, &key_schedule);
- print_aes_block128(&cipher);
}
return 0;