From d84140f48f9180fce594ee77e288bd6bbfc11fc9 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Mon, 25 May 2015 02:40:04 +0300 Subject: add test/ Includes simple block encryption/decryption programs for the supported modes (AES128/192/256 ECB + AES256 CBC). --- CMakeLists.txt | 1 + test/CMakeLists.txt | 14 ++++++++++++ test/aes128ecb_decrypt_block.c | 45 ++++++++++++++++++++++++++++++++++++ test/aes128ecb_encrypt_block.c | 45 ++++++++++++++++++++++++++++++++++++ test/aes192ecb_decrypt_block.c | 46 +++++++++++++++++++++++++++++++++++++ test/aes192ecb_encrypt_block.c | 46 +++++++++++++++++++++++++++++++++++++ test/aes256cbc_decrypt_block.c | 52 ++++++++++++++++++++++++++++++++++++++++++ test/aes256cbc_encrypt_block.c | 52 ++++++++++++++++++++++++++++++++++++++++++ test/aes256ecb_decrypt_block.c | 46 +++++++++++++++++++++++++++++++++++++ test/aes256ecb_encrypt_block.c | 46 +++++++++++++++++++++++++++++++++++++ 10 files changed, 393 insertions(+) create mode 100644 test/CMakeLists.txt create mode 100644 test/aes128ecb_decrypt_block.c create mode 100644 test/aes128ecb_encrypt_block.c create mode 100644 test/aes192ecb_decrypt_block.c create mode 100644 test/aes192ecb_encrypt_block.c create mode 100644 test/aes256cbc_decrypt_block.c create mode 100644 test/aes256cbc_encrypt_block.c create mode 100644 test/aes256ecb_decrypt_block.c create mode 100644 test/aes256ecb_encrypt_block.c diff --git a/CMakeLists.txt b/CMakeLists.txt index 702f157..5be65df 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -7,3 +7,4 @@ target_include_directories(${PROJECT_NAME} PUBLIC include/) target_compile_definitions(${PROJECT_NAME} PRIVATE _CRT_SECURE_NO_WARNINGS) add_subdirectory(examples) +add_subdirectory(test) diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt new file mode 100644 index 0000000..cba5d79 --- /dev/null +++ b/test/CMakeLists.txt @@ -0,0 +1,14 @@ +set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} /SAFESEH:NO") + +macro(test prefix) + add_executable(${prefix}_encrypt_block ${prefix}_encrypt_block.c) + target_link_libraries(${prefix}_encrypt_block libaesni) + + add_executable(${prefix}_decrypt_block ${prefix}_decrypt_block.c) + target_link_libraries(${prefix}_decrypt_block libaesni) +endmacro() + +test(aes128ecb) +test(aes192ecb) +test(aes256ecb) +test(aes256cbc) diff --git a/test/aes128ecb_decrypt_block.c b/test/aes128ecb_decrypt_block.c new file mode 100644 index 0000000..188524b --- /dev/null +++ b/test/aes128ecb_decrypt_block.c @@ -0,0 +1,45 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [CIPHER...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, key, cipher; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&cipher, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + plain = aes128ecb_decrypt(cipher, key); + print_aes_block128(&plain); + } + + return 0; +} diff --git a/test/aes128ecb_encrypt_block.c b/test/aes128ecb_encrypt_block.c new file mode 100644 index 0000000..27c7a0e --- /dev/null +++ b/test/aes128ecb_encrypt_block.c @@ -0,0 +1,45 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [PLAIN...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, key, cipher; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&plain, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + cipher = aes128ecb_encrypt(plain, key); + print_aes_block128(&cipher); + } + + return 0; +} diff --git a/test/aes192ecb_decrypt_block.c b/test/aes192ecb_decrypt_block.c new file mode 100644 index 0000000..73827a2 --- /dev/null +++ b/test/aes192ecb_decrypt_block.c @@ -0,0 +1,46 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [CIPHER...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, cipher; + __declspec(align(16)) AesBlock192 key; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&cipher, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + plain = aes192ecb_decrypt(cipher, &key); + print_aes_block128(&plain); + } + + return 0; +} diff --git a/test/aes192ecb_encrypt_block.c b/test/aes192ecb_encrypt_block.c new file mode 100644 index 0000000..9f2db50 --- /dev/null +++ b/test/aes192ecb_encrypt_block.c @@ -0,0 +1,46 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [PLAIN...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, cipher; + __declspec(align(16)) AesBlock192 key; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&plain, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + cipher = aes192ecb_encrypt(plain, &key); + print_aes_block128(&cipher); + } + + return 0; +} diff --git a/test/aes256cbc_decrypt_block.c b/test/aes256cbc_decrypt_block.c new file mode 100644 index 0000000..bac98e3 --- /dev/null +++ b/test/aes256cbc_decrypt_block.c @@ -0,0 +1,52 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +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; + + 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]); + } + + 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 = aes256cbc_decrypt(cipher, &key, &iv); + print_aes_block128(&plain); + } + + return 0; +} diff --git a/test/aes256cbc_encrypt_block.c b/test/aes256cbc_encrypt_block.c new file mode 100644 index 0000000..08ca11b --- /dev/null +++ b/test/aes256cbc_encrypt_block.c @@ -0,0 +1,52 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +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; + + 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]); + } + + 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 = aes256cbc_encrypt(plain, &key, &iv); + print_aes_block128(&cipher); + } + + return 0; +} diff --git a/test/aes256ecb_decrypt_block.c b/test/aes256ecb_decrypt_block.c new file mode 100644 index 0000000..f3126fe --- /dev/null +++ b/test/aes256ecb_decrypt_block.c @@ -0,0 +1,46 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [CIPHER...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, cipher; + __declspec(align(16)) AesBlock256 key; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&cipher, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + plain = aes256ecb_decrypt(cipher, &key); + print_aes_block128(&plain); + } + + return 0; +} diff --git a/test/aes256ecb_encrypt_block.c b/test/aes256ecb_encrypt_block.c new file mode 100644 index 0000000..3fb979d --- /dev/null +++ b/test/aes256ecb_encrypt_block.c @@ -0,0 +1,46 @@ +/** + * \file + * \author Egor Tensin + * \date 2015 + * \copyright This file is licensed under the terms of the MIT License. + * See LICENSE.txt for details. + */ + +#include + +#include +#include + +static void exit_with_usage(const char* argv0) +{ + printf("Usage: %s KEY [PLAIN...]\n", argv0); + exit(EXIT_FAILURE); +} + +int main(int argc, char** argv) +{ + __declspec(align(16)) AesBlock128 plain, cipher; + __declspec(align(16)) AesBlock256 key; + + if (argc < 2) + 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]); + } + + for (int i = 2; i < argc; ++i) + { + if (parse_aes_block128(&plain, argv[i]) != 0) + { + fprintf(stderr, "Invalid 128-bit AES block '%s'\n", argv[i]); + continue; + } + cipher = aes256ecb_encrypt(plain, &key); + print_aes_block128(&cipher); + } + + return 0; +} -- cgit v1.2.3