aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2015-06-04 03:18:34 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2015-06-04 03:18:34 +0300
commit21b736e980709dad61c1412e22ec26265bfe4f0e (patch)
tree0a29fafff6a9b6c5a31d39c39480dca5084bbcb0
parentadd BE-to-LE conversion functions (diff)
downloadaes-tools-21b736e980709dad61c1412e22ec26265bfe4f0e.tar.gz
aes-tools-21b736e980709dad61c1412e22ec26265bfe4f0e.zip
add CTR mode
-rw-r--r--examples/CMakeLists.txt3
-rw-r--r--examples/aes128ctr_example.c51
-rw-r--r--examples/aes192ctr_example.c52
-rw-r--r--examples/aes256ctr_example.c52
-rw-r--r--include/aesni/api.h72
-rw-r--r--test/CMakeLists.txt3
-rw-r--r--test/aes128ctr_decrypt_block.c54
-rw-r--r--test/aes128ctr_encrypt_block.c54
-rw-r--r--test/aes192ctr_decrypt_block.c55
-rw-r--r--test/aes192ctr_encrypt_block.c55
-rw-r--r--test/aes256ctr_decrypt_block.c55
-rw-r--r--test/aes256ctr_encrypt_block.c55
12 files changed, 561 insertions, 0 deletions
diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
index 080e382..db2b2c4 100644
--- a/examples/CMakeLists.txt
+++ b/examples/CMakeLists.txt
@@ -9,11 +9,14 @@ example(aes128ecb)
example(aes128cbc)
example(aes128cfb)
example(aes128ofb)
+example(aes128ctr)
example(aes192ecb)
example(aes192cbc)
example(aes192cfb)
example(aes192ofb)
+example(aes192ctr)
example(aes256ecb)
example(aes256cbc)
example(aes256cfb)
example(aes256ofb)
+example(aes256ctr)
diff --git a/examples/aes128ctr_example.c b/examples/aes128ctr_example.c
new file mode 100644
index 0000000..b0d38ff
--- /dev/null
+++ b/examples/aes128ctr_example.c
@@ -0,0 +1,51 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+
+int main()
+{
+ __declspec(align(16)) AesBlock128 plain, key, cypher, decrypted, iv;
+ __declspec(align(16)) Aes128KeySchedule key_schedule;
+
+ plain = make_aes_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
+ key = make_aes_block128(0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
+ iv = make_aes_block128(0xfedcba98, 0x76543210, 0xfedcba98, 0x76543210);
+
+ printf("Plain: %s\n", format_aes_block128(&plain).str);
+ print_aes_block128_fips_matrix_style(&plain);
+
+ printf("\n");
+ printf("Key: %s\n", format_aes_block128(&key).str);
+ print_aes_block128_fips_matrix_style(&key);
+
+ printf("\n");
+ printf("Initialization vector: %s\n", format_aes_block128(&iv).str);
+ print_aes_block128_fips_matrix_style(&iv);
+
+ aes128_expand_key_schedule(key, &key_schedule);
+
+ printf("\n");
+ printf("Key schedule:\n");
+ for (int i = 0; i < 11; ++i)
+ printf("\t[%d]: %s\n", i, format_aes_block128(&key_schedule.keys[i]).str);
+
+ cypher = aes128ctr_encrypt(plain, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Cypher: %s\n", format_aes_block128(&cypher).str);
+ print_aes_block128_fips_matrix_style(&cypher);
+
+ decrypted = aes128ctr_decrypt(cypher, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Decrypted: %s\n", format_aes_block128(&decrypted).str);
+ print_aes_block128_fips_matrix_style(&decrypted);
+
+ return 0;
+}
diff --git a/examples/aes192ctr_example.c b/examples/aes192ctr_example.c
new file mode 100644
index 0000000..c438f55
--- /dev/null
+++ b/examples/aes192ctr_example.c
@@ -0,0 +1,52 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+
+int main()
+{
+ __declspec(align(16)) AesBlock128 plain, cypher, decrypted, iv;
+ __declspec(align(16)) AesBlock192 key;
+ __declspec(align(16)) Aes192KeySchedule key_schedule;
+
+ plain = make_aes_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
+ key = make_aes_block192(0x17161514, 0x13121110, 0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
+ iv = make_aes_block128(0xfedcba98, 0x76543210, 0xfedcba98, 0x76543210);
+
+ printf("Plain: %s\n", format_aes_block128(&plain).str);
+ print_aes_block128_fips_matrix_style(&plain);
+
+ printf("\n");
+ printf("Key: %s\n", format_aes_block192(&key).str);
+ print_aes_block192_fips_matrix_style(&key);
+
+ printf("\n");
+ printf("Initialization vector: %s\n", format_aes_block128(&iv).str);
+ print_aes_block128_fips_matrix_style(&iv);
+
+ aes192_expand_key_schedule(&key, &key_schedule);
+
+ printf("\n");
+ printf("Key schedule:\n");
+ for (int i = 0; i < 13; ++i)
+ printf("\t[%d]: %s\n", i, format_aes_block128(&key_schedule.keys[i]).str);
+
+ cypher = aes192ctr_encrypt(plain, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Cypher: %s\n", format_aes_block128(&cypher).str);
+ print_aes_block128_fips_matrix_style(&cypher);
+
+ decrypted = aes192ctr_decrypt(cypher, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Decrypted: %s\n", format_aes_block128(&decrypted).str);
+ print_aes_block128_fips_matrix_style(&decrypted);
+
+ return 0;
+}
diff --git a/examples/aes256ctr_example.c b/examples/aes256ctr_example.c
new file mode 100644
index 0000000..a6fcffe
--- /dev/null
+++ b/examples/aes256ctr_example.c
@@ -0,0 +1,52 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+
+int main()
+{
+ __declspec(align(16)) AesBlock128 plain, cypher, decrypted, iv;
+ __declspec(align(16)) AesBlock256 key;
+ __declspec(align(16)) Aes256KeySchedule key_schedule;
+
+ plain = make_aes_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
+ key = make_aes_block256(0x1f1e1d1c, 0x1b1a1918, 0x17161514, 0x13121110, 0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
+ iv = make_aes_block128(0xfedcba98, 0x76543210, 0xfedcba98, 0x76543210);
+
+ printf("Plain: %s\n", format_aes_block128(&plain).str);
+ print_aes_block128_fips_matrix_style(&plain);
+
+ printf("\n");
+ printf("Key: %s\n", format_aes_block256(&key).str);
+ print_aes_block256_fips_matrix_style(&key);
+
+ printf("\n");
+ printf("Initialization vector: %s\n", format_aes_block128(&iv).str);
+ print_aes_block128_fips_matrix_style(&iv);
+
+ aes256_expand_key_schedule(&key, &key_schedule);
+
+ printf("\n");
+ printf("Key schedule:\n");
+ for (int i = 0; i < 15; ++i)
+ printf("\t[%d]: %s\n", i, format_aes_block128(&key_schedule.keys[i]).str);
+
+ cypher = aes256ctr_encrypt(plain, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Cypher: %s\n", format_aes_block128(&cypher).str);
+ print_aes_block128_fips_matrix_style(&cypher);
+
+ decrypted = aes256ctr_decrypt(cypher, &key_schedule, iv, 0);
+ printf("\n");
+ printf("Decrypted: %s\n", format_aes_block128(&decrypted).str);
+ print_aes_block128_fips_matrix_style(&decrypted);
+
+ return 0;
+}
diff --git a/include/aesni/api.h b/include/aesni/api.h
index fff3672..3318aef 100644
--- a/include/aesni/api.h
+++ b/include/aesni/api.h
@@ -105,6 +105,30 @@ static __inline AesBlock128 __fastcall aes128ofb_decrypt(
return _mm_xor_si128(tmp, cypher);
}
+static __inline AesBlock128 __fastcall aes128ctr_encrypt(
+ AesBlock128 plain,
+ Aes128KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(plain, raw_aes128ecb_encrypt(init_vector, key_schedule));
+}
+
+static __inline AesBlock128 __fastcall aes128ctr_decrypt(
+ AesBlock128 cypher,
+ Aes128KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(cypher, raw_aes128ecb_encrypt(init_vector, key_schedule));
+}
+
static __inline void __fastcall aes192_expand_key_schedule(
AesBlock192* key,
Aes192KeySchedule* key_schedule)
@@ -199,6 +223,30 @@ static __inline AesBlock128 __fastcall aes192ofb_decrypt(
return _mm_xor_si128(tmp, cypher);
}
+static __inline AesBlock128 __fastcall aes192ctr_encrypt(
+ AesBlock128 plain,
+ Aes192KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(plain, raw_aes192ecb_encrypt(init_vector, key_schedule));
+}
+
+static __inline AesBlock128 __fastcall aes192ctr_decrypt(
+ AesBlock128 cypher,
+ Aes192KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(cypher, raw_aes192ecb_encrypt(init_vector, key_schedule));
+}
+
static __inline void __fastcall aes256_expand_key_schedule(
AesBlock256* key,
Aes256KeySchedule* key_schedule)
@@ -292,3 +340,27 @@ static __inline AesBlock128 __fastcall aes256ofb_decrypt(
_mm_storeu_si128(next_init_vector, tmp);
return _mm_xor_si128(tmp, cypher);
}
+
+static __inline AesBlock128 __fastcall aes256ctr_encrypt(
+ AesBlock128 plain,
+ Aes256KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(plain, raw_aes256ecb_encrypt(init_vector, key_schedule));
+}
+
+static __inline AesBlock128 __fastcall aes256ctr_decrypt(
+ AesBlock128 cypher,
+ Aes256KeySchedule* key_schedule,
+ AesBlock128 init_vector,
+ int counter)
+{
+ init_vector = aes128_le2be(init_vector);
+ init_vector = _mm_add_epi32(init_vector, make_aes_block128(0, 0, 0, counter));
+ init_vector = aes128_be2le(init_vector);
+ return _mm_xor_si128(cypher, raw_aes256ecb_encrypt(init_vector, key_schedule));
+}
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 2057b8f..f4460f0 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -12,11 +12,14 @@ test(aes128ecb)
test(aes128cbc)
test(aes128cfb)
test(aes128ofb)
+test(aes128ctr)
test(aes192ecb)
test(aes192cbc)
test(aes192cfb)
test(aes192ofb)
+test(aes192ctr)
test(aes256ecb)
test(aes256cbc)
test(aes256cfb)
test(aes256ofb)
+test(aes256ctr)
diff --git a/test/aes128ctr_decrypt_block.c b/test/aes128ctr_decrypt_block.c
new file mode 100644
index 0000000..313e5d9
--- /dev/null
+++ b/test/aes128ctr_decrypt_block.c
@@ -0,0 +1,54 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [CIPHER...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, key, cipher, iv;
+ __declspec(align(16)) Aes128KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ 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(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes128_expand_key_schedule(key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&cipher, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ plain = aes128ctr_decrypt(cipher, &key_schedule, iv, i - 3);
+ print_aes_block128(&plain);
+ }
+
+ return 0;
+}
diff --git a/test/aes128ctr_encrypt_block.c b/test/aes128ctr_encrypt_block.c
new file mode 100644
index 0000000..e5372c2
--- /dev/null
+++ b/test/aes128ctr_encrypt_block.c
@@ -0,0 +1,54 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [PLAIN...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, key, cipher, iv;
+ __declspec(align(16)) Aes128KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ 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(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes128_expand_key_schedule(key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&plain, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ cipher = aes128ctr_encrypt(plain, &key_schedule, iv, i - 3);
+ print_aes_block128(&cipher);
+ }
+
+ return 0;
+}
diff --git a/test/aes192ctr_decrypt_block.c b/test/aes192ctr_decrypt_block.c
new file mode 100644
index 0000000..01258ea
--- /dev/null
+++ b/test/aes192ctr_decrypt_block.c
@@ -0,0 +1,55 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [CIPHER...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, cipher, iv;
+ __declspec(align(16)) AesBlock192 key;
+ __declspec(align(16)) Aes192KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ if (parse_aes_block192(&key, argv[1]) != 0)
+ {
+ fprintf(stderr, "Invalid 192-bit AES block '%s'\n", argv[1]);
+ exit_with_usage(argv[0]);
+ }
+
+ if (parse_aes_block128(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes192_expand_key_schedule(&key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&cipher, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ plain = aes192ctr_decrypt(cipher, &key_schedule, iv, i - 3);
+ print_aes_block128(&plain);
+ }
+
+ return 0;
+}
diff --git a/test/aes192ctr_encrypt_block.c b/test/aes192ctr_encrypt_block.c
new file mode 100644
index 0000000..9a29005
--- /dev/null
+++ b/test/aes192ctr_encrypt_block.c
@@ -0,0 +1,55 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [PLAIN...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, cipher, iv;
+ __declspec(align(16)) AesBlock192 key;
+ __declspec(align(16)) Aes192KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ if (parse_aes_block192(&key, argv[1]) != 0)
+ {
+ fprintf(stderr, "Invalid 192-bit AES block '%s'\n", argv[1]);
+ exit_with_usage(argv[0]);
+ }
+
+ if (parse_aes_block128(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes192_expand_key_schedule(&key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&plain, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ cipher = aes192ctr_encrypt(plain, &key_schedule, iv, i - 3);
+ print_aes_block128(&cipher);
+ }
+
+ return 0;
+}
diff --git a/test/aes256ctr_decrypt_block.c b/test/aes256ctr_decrypt_block.c
new file mode 100644
index 0000000..db7ae5c
--- /dev/null
+++ b/test/aes256ctr_decrypt_block.c
@@ -0,0 +1,55 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [CIPHER...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, cipher, iv;
+ __declspec(align(16)) AesBlock256 key;
+ __declspec(align(16)) Aes256KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ if (parse_aes_block256(&key, argv[1]) != 0)
+ {
+ fprintf(stderr, "Invalid 256-bit AES block '%s'\n", argv[1]);
+ exit_with_usage(argv[0]);
+ }
+
+ if (parse_aes_block128(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes256_expand_key_schedule(&key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&cipher, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ plain = aes256ctr_decrypt(cipher, &key_schedule, iv, i - 3);
+ print_aes_block128(&plain);
+ }
+
+ return 0;
+}
diff --git a/test/aes256ctr_encrypt_block.c b/test/aes256ctr_encrypt_block.c
new file mode 100644
index 0000000..d812a86
--- /dev/null
+++ b/test/aes256ctr_encrypt_block.c
@@ -0,0 +1,55 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ * See LICENSE.txt for details.
+ */
+
+#include <aesni/all.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static void exit_with_usage(const char* argv0)
+{
+ printf("Usage: %s KEY INIT_VECTOR [PLAIN...]\n", argv0);
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char** argv)
+{
+ __declspec(align(16)) AesBlock128 plain, cipher, iv;
+ __declspec(align(16)) AesBlock256 key;
+ __declspec(align(16)) Aes256KeySchedule key_schedule;
+
+ if (argc < 3)
+ exit_with_usage(argv[0]);
+
+ if (parse_aes_block256(&key, argv[1]) != 0)
+ {
+ fprintf(stderr, "Invalid 256-bit AES block '%s'\n", argv[1]);
+ exit_with_usage(argv[0]);
+ }
+
+ if (parse_aes_block128(&iv, argv[2]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[2]);
+ exit_with_usage(argv[0]);
+ }
+
+ aes256_expand_key_schedule(&key, &key_schedule);
+
+ for (int i = 3; i < argc; ++i)
+ {
+ if (parse_aes_block128(&plain, argv[i]) != 0)
+ {
+ fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]);
+ continue;
+ }
+ cipher = aes256ctr_encrypt(plain, &key_schedule, iv, i - 3);
+ print_aes_block128(&cipher);
+ }
+
+ return 0;
+}