aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src/aes.c
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2015-06-17 23:25:52 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2015-06-17 23:25:52 +0300
commit44d58e0a10c7dfdb33899fa98c0eea790e0b90ce (patch)
tree646541b156ffdd88bbaa4f1fa860aed465dfce60 /src/aes.c
parentrefactoring (diff)
downloadaes-tools-44d58e0a10c7dfdb33899fa98c0eea790e0b90ce.tar.gz
aes-tools-44d58e0a10c7dfdb33899fa98c0eea790e0b90ce.zip
factoring out AES-specific stuff
Diffstat (limited to 'src/aes.c')
-rw-r--r--src/aes.c362
1 files changed, 362 insertions, 0 deletions
diff --git a/src/aes.c b/src/aes.c
new file mode 100644
index 0000000..98cc9cd
--- /dev/null
+++ b/src/aes.c
@@ -0,0 +1,362 @@
+/**
+ * \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 <assert.h>
+#include <stdio.h>
+#include <string.h>
+
+AesNI_StatusCode aesni_aes_format_block(
+ AesNI_Aes_BlockString* str,
+ const AesNI_Aes_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(str);
+ assert(block);
+
+ if (str == NULL)
+ return aesni_make_null_argument_error(err_details, "str");
+ if (block == NULL)
+ return aesni_make_null_argument_error(err_details, "block");
+
+ char* cursor = str->str;
+
+ __declspec(align(16)) unsigned char bytes[16];
+ aesni_store_block128_aligned(bytes, *block);
+
+ for (int i = 0; i < 16; ++i, cursor += 2)
+ sprintf(cursor, "%02x", bytes[i]);
+
+ *cursor = '\0';
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes_format_block_as_matrix(
+ AesNI_Aes_BlockMatrixString* str,
+ const AesNI_Aes_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(str);
+ assert(block);
+
+ if (str == NULL)
+ return aesni_make_null_argument_error(err_details, "str");
+ if (block == NULL)
+ return aesni_make_null_argument_error(err_details, "block");
+
+ char* cursor = str->str;
+
+ __declspec(align(16)) unsigned char bytes[4][4];
+ aesni_store_block128_aligned(bytes, *block);
+
+ for (int i = 0; i < 4; ++i, cursor += 3)
+ {
+ for (int j = 0; j < 3; ++j, cursor += 3)
+ sprintf(cursor, "%02x ", bytes[j][i]);
+ sprintf(cursor, "%02x\n", bytes[3][i]);
+ }
+
+ *cursor = '\0';
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes_print_block(
+ const AesNI_Aes_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(block);
+
+ if (block == NULL)
+ return aesni_make_null_argument_error(err_details, "block");
+
+ AesNI_StatusCode ec = AESNI_SUCCESS;
+ AesNI_Aes_BlockString str;
+
+ if (aesni_is_error(ec = aesni_aes_format_block(&str, block, err_details)))
+ return ec;
+
+ printf("%s\n", str.str);
+ return ec;
+}
+
+AesNI_StatusCode aesni_aes_print_block_as_matrix(
+ const AesNI_Aes_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(block);
+
+ if (block == NULL)
+ return aesni_make_null_argument_error(err_details, "block");
+
+ AesNI_StatusCode ec = AESNI_SUCCESS;
+ AesNI_Aes_BlockMatrixString str;
+
+ if (aesni_is_error(ec = aesni_aes_format_block_as_matrix(&str, block, err_details)))
+ return ec;
+
+ printf("%s", str.str);
+ return ec;
+}
+
+AesNI_StatusCode aesni_aes_parse_block(
+ AesNI_Aes_Block* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(dest);
+ assert(src);
+
+ if (dest == NULL)
+ return aesni_make_null_argument_error(err_details, "dest");
+ if (src == NULL)
+ return aesni_make_null_argument_error(err_details, "src");
+
+ __declspec(align(16)) unsigned char bytes[16];
+
+ for (int i = 0; i < 16; ++i)
+ {
+ int n;
+ unsigned int byte;
+ if (sscanf(src, "%2x%n", &byte, &n) != 1)
+ return aesni_make_parse_error(err_details, src);
+ bytes[i] = (unsigned char) byte;
+ src += n;
+ }
+
+ *dest = aesni_load_block128_aligned(bytes);
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes128_format_key(
+ AesNI_Aes128_KeyString* str,
+ const AesNI_Aes128_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_aes_format_block(str, &key->key, err_details);
+}
+
+AesNI_StatusCode aesni_aes192_format_key(
+ AesNI_Aes192_KeyString* str,
+ const AesNI_Aes192_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(str);
+ assert(key);
+
+ if (str == NULL)
+ return aesni_make_null_argument_error(err_details, "str");
+ if (key == NULL)
+ return aesni_make_null_argument_error(err_details, "key");
+
+ char* cursor = str->str;
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+ aesni_store_block128_aligned(bytes, key->lo);
+
+ for (int i = 0; i < 16; ++i, cursor += 2)
+ sprintf(cursor, "%02x", bytes[i]);
+ }
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+ aesni_store_block128_aligned(bytes, key->hi);
+
+ for (int i = 0; i < 8; ++i, cursor += 2)
+ sprintf(cursor, "%02x", bytes[i]);
+ }
+
+ *cursor = '\0';
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes256_format_key(
+ AesNI_Aes256_KeyString* str,
+ const AesNI_Aes256_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(str);
+ assert(key);
+
+ if (str == NULL)
+ return aesni_make_null_argument_error(err_details, "str");
+ if (key == NULL)
+ return aesni_make_null_argument_error(err_details, "key");
+
+ char* cursor = str->str;
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+ aesni_store_block128_aligned(bytes, key->lo);
+
+ for (int i = 0; i < 16; ++i, cursor += 2)
+ sprintf(cursor, "%02x", bytes[i]);
+ }
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+ aesni_store_block128_aligned(bytes, key->hi);
+
+ for (int i = 0; i < 16; ++i, cursor += 2)
+ sprintf(cursor, "%02x", bytes[i]);
+ }
+
+ *cursor = '\0';
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes128_print_key(
+ const AesNI_Aes128_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_aes_print_block(&key->key, err_details);
+}
+
+AesNI_StatusCode aesni_aes192_print_key(
+ const AesNI_Aes192_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(key);
+
+ if (key == NULL)
+ return aesni_make_null_argument_error(err_details, "key");
+
+ AesNI_StatusCode ec = AESNI_SUCCESS;
+ AesNI_Aes192_KeyString str;
+
+ if (aesni_is_error(ec = aesni_aes192_format_key(&str, key, err_details)))
+ return ec;
+
+ printf("%s\n", str.str);
+ return ec;
+}
+
+AesNI_StatusCode aesni_aes256_print_key(
+ const AesNI_Aes256_Key* key,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(key);
+
+ if (key == NULL)
+ return aesni_make_null_argument_error(err_details, "key");
+
+ AesNI_StatusCode ec = AESNI_SUCCESS;
+ AesNI_Aes256_KeyString str;
+
+ if (aesni_is_error(ec = aesni_aes256_format_key(&str, key, err_details)))
+ return ec;
+
+ printf("%s\n", str.str);
+ return ec;
+}
+
+AesNI_StatusCode aesni_aes128_parse_key(
+ AesNI_Aes128_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_aes_parse_block(&dest->key, src, err_details);
+}
+
+AesNI_StatusCode aesni_aes192_parse_key(
+ AesNI_Aes192_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(dest);
+ assert(src);
+
+ if (dest == NULL)
+ return aesni_make_null_argument_error(err_details, "dest");
+ if (src == NULL)
+ return aesni_make_null_argument_error(err_details, "src");
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+
+ for (int i = 0; i < 16; ++i)
+ {
+ int n;
+ unsigned int byte;
+ if (sscanf(src, "%2x%n", &byte, &n) != 1)
+ return aesni_make_parse_error(err_details, src);
+ bytes[i] = (unsigned char) byte;
+ src += n;
+ }
+
+ dest->lo = aesni_load_block128_aligned(bytes);
+ }
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+
+ for (int i = 0; i < 8; ++i)
+ {
+ int n;
+ unsigned int byte;
+ if (sscanf(src, "%2x%n", &byte, &n) != 1)
+ return aesni_make_parse_error(err_details, src);
+ bytes[i] = (unsigned char) byte;
+ src += n;
+ }
+
+ memset(bytes + 8, 0x00, 8);
+ dest->hi = aesni_load_block128_aligned(bytes);
+ }
+
+ return AESNI_SUCCESS;
+}
+
+AesNI_StatusCode aesni_aes256_parse_key(
+ AesNI_Aes256_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ assert(dest);
+ assert(src);
+
+ if (dest == NULL)
+ return aesni_make_null_argument_error(err_details, "dest");
+ if (src == NULL)
+ return aesni_make_null_argument_error(err_details, "src");
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+
+ for (int i = 0; i < 16; ++i)
+ {
+ int n;
+ unsigned int byte;
+ if (sscanf(src, "%2x%n", &byte, &n) != 1)
+ return aesni_make_parse_error(err_details, src);
+ bytes[i] = (unsigned char) byte;
+ src += n;
+ }
+
+ dest->lo = aesni_load_block128_aligned(bytes);
+ }
+
+ {
+ __declspec(align(16)) unsigned char bytes[16];
+
+ for (int i = 0; i < 16; ++i)
+ {
+ int n;
+ unsigned int byte;
+ if (sscanf(src, "%2x%n", &byte, &n) != 1)
+ return aesni_make_parse_error(err_details, src);
+ bytes[i] = (unsigned char) byte;
+ src += n;
+ }
+
+ dest->hi = aesni_load_block128_aligned(bytes);
+ }
+
+ return AESNI_SUCCESS;
+}