aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/include/aes/aes.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/aes/aes.h')
-rw-r--r--include/aes/aes.h529
1 files changed, 529 insertions, 0 deletions
diff --git a/include/aes/aes.h b/include/aes/aes.h
new file mode 100644
index 0000000..ea859a3
--- /dev/null
+++ b/include/aes/aes.h
@@ -0,0 +1,529 @@
+/**
+ * \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.
+ */
+
+#pragma once
+
+#include "data.h"
+#include "error.h"
+#include "mode.h"
+
+#include <assert.h>
+
+#ifdef __cplusplus
+extern "C"
+{
+#endif
+
+typedef AesNI_Block128 AesNI_AES_Block;
+typedef AesNI_AES_Block AesNI_AES128_Block;
+typedef AesNI_AES_Block AesNI_AES192_Block;
+typedef AesNI_AES_Block AesNI_AES256_Block;
+
+typedef struct
+{
+ AesNI_AES_Block key;
+}
+AesNI_AES128_Key;
+
+typedef struct
+{
+ AesNI_AES_Block hi;
+ AesNI_AES_Block lo;
+}
+AesNI_AES192_Key;
+
+typedef struct
+{
+ AesNI_AES_Block hi;
+ AesNI_AES_Block lo;
+}
+AesNI_AES256_Key;
+
+static __inline void aesni_AES_make_block(AesNI_AES_Block* dest, int hi3, int hi2, int lo1, int lo0)
+{
+ *dest = aesni_make_block128(hi3, hi2, lo1, lo0);
+}
+
+static __inline void aesni_AES128_make_block(AesNI_AES128_Block* dest, int hi3, int hi2, int lo1, int lo0)
+{
+ aesni_AES_make_block(dest, hi3, hi2, lo1, lo0);
+}
+
+static __inline void aesni_AES192_make_block(AesNI_AES192_Block* dest, int hi3, int hi2, int lo1, int lo0)
+{
+ aesni_AES_make_block(dest, hi3, hi2, lo1, lo0);
+}
+
+static __inline void aesni_AES256_make_block(AesNI_AES256_Block* dest, int hi3, int hi2, int lo1, int lo0)
+{
+ aesni_AES_make_block(dest, hi3, hi2, lo1, lo0);
+}
+
+static __inline void aesni_AES128_make_key(AesNI_AES128_Key* dest, int hi3, int hi2, int lo1, int lo0)
+{
+ aesni_AES_make_block(&dest->key, hi3, hi2, lo1, lo0);
+}
+
+static __inline void aesni_AES192_make_key(AesNI_AES192_Key* dest, int hi5, int hi4, int lo3, int lo2, int lo1, int lo0)
+{
+ aesni_AES_make_block(&dest->hi, 0, 0, hi5, hi4);
+ aesni_AES_make_block(&dest->lo, lo3, lo2, lo1, lo0);
+}
+
+static __inline void aesni_AES256_make_key(AesNI_AES256_Key* dest, int hi7, int hi6, int hi5, int hi4, int lo3, int lo2, int lo1, int lo0)
+{
+ aesni_AES_make_block(&dest->hi, hi7, hi6, hi5, hi4);
+ aesni_AES_make_block(&dest->lo, lo3, lo2, lo1, lo0);
+}
+
+typedef struct { char str[33]; } AesNI_AES_BlockString;
+typedef AesNI_AES_BlockString AesNI_AES128_BlockString;
+typedef AesNI_AES_BlockString AesNI_AES192_BlockString;
+typedef AesNI_AES_BlockString AesNI_AES256_BlockString;
+
+typedef struct { char str[49]; } AesNI_AES_BlockMatrixString;
+typedef AesNI_AES_BlockMatrixString AesNI_AES128_BlockMatrixString;
+typedef AesNI_AES_BlockMatrixString AesNI_AES192_BlockMatrixString;
+typedef AesNI_AES_BlockMatrixString AesNI_AES256_BlockMatrixString;
+
+AesNI_StatusCode aesni_AES_format_block(
+ AesNI_AES_BlockString*,
+ const AesNI_AES_Block*,
+ AesNI_ErrorDetails*);
+
+static __inline AesNI_StatusCode aesni_AES128_format_block(
+ AesNI_AES128_BlockString* dest,
+ const AesNI_AES128_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES192_format_block(
+ AesNI_AES192_BlockString* dest,
+ const AesNI_AES192_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES256_format_block(
+ AesNI_AES256_BlockString* dest,
+ const AesNI_AES256_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block(dest, src, err_details);
+}
+
+AesNI_StatusCode aesni_AES_format_block_as_matrix(
+ AesNI_AES_BlockMatrixString*,
+ const AesNI_AES_Block*,
+ AesNI_ErrorDetails*);
+
+static __inline AesNI_StatusCode aesni_AES128_format_block_as_matrix(
+ AesNI_AES128_BlockMatrixString* dest,
+ const AesNI_AES128_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block_as_matrix(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES192_format_block_as_matrix(
+ AesNI_AES192_BlockMatrixString* dest,
+ const AesNI_AES192_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block_as_matrix(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES256_format_block_as_matrix(
+ AesNI_AES256_BlockMatrixString* dest,
+ const AesNI_AES256_Block* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_format_block_as_matrix(dest, src, err_details);
+}
+
+AesNI_StatusCode aesni_AES_print_block(
+ const AesNI_AES_Block*,
+ AesNI_ErrorDetails*);
+
+static __inline AesNI_StatusCode aesni_AES128_print_block(
+ const AesNI_AES128_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block(block, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES192_print_block(
+ const AesNI_AES192_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block(block, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES256_print_block(
+ const AesNI_AES256_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block(block, err_details);
+}
+
+AesNI_StatusCode aesni_AES_print_block_as_matrix(
+ const AesNI_AES_Block*,
+ AesNI_ErrorDetails*);
+
+static __inline AesNI_StatusCode aesni_AES128_print_block_as_matrix(
+ const AesNI_AES128_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block_as_matrix(block, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES192_print_block_as_matrix(
+ const AesNI_AES192_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block_as_matrix(block, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES256_print_block_as_matrix(
+ const AesNI_AES256_Block* block,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_print_block_as_matrix(block, err_details);
+}
+
+AesNI_StatusCode aesni_AES_parse_block(
+ AesNI_AES_Block* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details);
+
+static __inline AesNI_StatusCode aesni_AES128_parse_block(
+ AesNI_AES128_Block* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_parse_block(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES192_parse_block(
+ AesNI_AES192_Block* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_parse_block(dest, src, err_details);
+}
+
+static __inline AesNI_StatusCode aesni_AES256_parse_block(
+ AesNI_AES256_Block* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details)
+{
+ return aesni_AES_parse_block(dest, src, err_details);
+}
+
+typedef struct { char str[33]; } AesNI_AES128_KeyString;
+typedef struct { char str[49]; } AesNI_AES192_KeyString;
+typedef struct { char str[65]; } AesNI_AES256_KeyString;
+
+AesNI_StatusCode aesni_AES128_format_key(
+ AesNI_AES128_KeyString*,
+ const AesNI_AES128_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES192_format_key(
+ AesNI_AES192_KeyString*,
+ const AesNI_AES192_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES256_format_key(
+ AesNI_AES256_KeyString*,
+ const AesNI_AES256_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES128_print_key(
+ const AesNI_AES128_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES192_print_key(
+ const AesNI_AES192_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES256_print_key(
+ const AesNI_AES256_Key*,
+ AesNI_ErrorDetails*);
+
+AesNI_StatusCode aesni_AES128_parse_key(
+ AesNI_AES128_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details);
+
+AesNI_StatusCode aesni_AES192_parse_key(
+ AesNI_AES192_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details);
+
+AesNI_StatusCode aesni_AES256_parse_key(
+ AesNI_AES256_Key* dest,
+ const char* src,
+ AesNI_ErrorDetails* err_details);
+
+typedef struct
+{
+ AesNI_AES_Block keys[11];
+}
+AesNI_AES128_RoundKeys;
+
+typedef struct
+{
+ AesNI_AES_Block keys[13];
+}
+AesNI_AES192_RoundKeys;
+
+typedef struct
+{
+ AesNI_AES_Block keys[15];
+}
+AesNI_AES256_RoundKeys;
+
+void __fastcall aesni_AES128_expand_key_(
+ AesNI_AES_Block key,
+ AesNI_AES128_RoundKeys* encryption_keys);
+
+void __fastcall aesni_AES192_expand_key_(
+ AesNI_AES_Block key_lo,
+ AesNI_AES_Block key_hi,
+ AesNI_AES192_RoundKeys* encryption_keys);
+
+void __fastcall aesni_AES256_expand_key_(
+ AesNI_AES_Block key_lo,
+ AesNI_AES_Block key_hi,
+ AesNI_AES256_RoundKeys* encryption_keys);
+
+void __fastcall aesni_AES128_derive_decryption_keys_(
+ const AesNI_AES128_RoundKeys* encryption_keys,
+ AesNI_AES128_RoundKeys* decryption_keys);
+
+void __fastcall aesni_AES192_derive_decryption_keys_(
+ const AesNI_AES192_RoundKeys* encryption_keys,
+ AesNI_AES192_RoundKeys* decryption_keys);
+
+void __fastcall aesni_AES256_derive_decryption_keys_(
+ const AesNI_AES256_RoundKeys* encryption_keys,
+ AesNI_AES256_RoundKeys* decryption_keys);
+
+AesNI_AES_Block __fastcall aesni_AES128_encrypt_block_(
+ AesNI_AES_Block plaintext,
+ const AesNI_AES128_RoundKeys*);
+
+AesNI_AES_Block __fastcall aesni_AES192_encrypt_block_(
+ AesNI_AES_Block plaintext,
+ const AesNI_AES192_RoundKeys*);
+
+AesNI_AES_Block __fastcall aesni_AES256_encrypt_block_(
+ AesNI_AES_Block plaintext,
+ const AesNI_AES256_RoundKeys*);
+
+AesNI_AES_Block __fastcall aesni_AES128_decrypt_block_(
+ AesNI_AES_Block ciphertext,
+ const AesNI_AES128_RoundKeys*);
+
+AesNI_AES_Block __fastcall aesni_AES192_decrypt_block_(
+ AesNI_AES_Block ciphertext,
+ const AesNI_AES192_RoundKeys*);
+
+AesNI_AES_Block __fastcall aesni_AES256_decrypt_block_(
+ AesNI_AES_Block ciphertext,
+ const AesNI_AES256_RoundKeys*);
+
+static __inline AesNI_AES_Block __fastcall aesni_AES_xor_blocks(
+ AesNI_AES_Block a,
+ AesNI_AES_Block b)
+{
+ return aesni_xor_block128(a, b);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES128_xor_blocks(
+ AesNI_AES128_Block a,
+ AesNI_AES128_Block b)
+{
+ return aesni_AES_xor_blocks(a, b);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES192_xor_blocks(
+ AesNI_AES192_Block a,
+ AesNI_AES192_Block b)
+{
+ return aesni_AES_xor_blocks(a, b);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES256_xor_blocks(
+ AesNI_AES256_Block a,
+ AesNI_AES256_Block b)
+{
+ return aesni_AES_xor_blocks(a, b);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES_inc_block(
+ AesNI_AES_Block block)
+{
+ block = aesni_reverse_byte_order_block128(block);
+ block = aesni_inc_block128(block);
+ return aesni_reverse_byte_order_block128(block);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES128_inc_block(
+ AesNI_AES128_Block block)
+{
+ return aesni_AES_inc_block(block);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES192_inc_block(
+ AesNI_AES192_Block block)
+{
+ return aesni_AES_inc_block(block);
+}
+
+static __inline AesNI_AES_Block __fastcall aesni_AES256_inc_block(
+ AesNI_AES256_Block block)
+{
+ return aesni_AES_inc_block(block);
+}
+
+AESNI_ENCRYPT_BLOCK_ECB(AES128);
+AESNI_DECRYPT_BLOCK_ECB(AES128);
+AESNI_ENCRYPT_BLOCK_CBC(AES128);
+AESNI_DECRYPT_BLOCK_CBC(AES128);
+AESNI_ENCRYPT_BLOCK_CFB(AES128);
+AESNI_DECRYPT_BLOCK_CFB(AES128);
+AESNI_ENCRYPT_BLOCK_OFB(AES128);
+AESNI_DECRYPT_BLOCK_OFB(AES128);
+AESNI_ENCRYPT_BLOCK_CTR(AES128);
+AESNI_DECRYPT_BLOCK_CTR(AES128);
+
+AESNI_ENCRYPT_BLOCK_ECB(AES192);
+AESNI_DECRYPT_BLOCK_ECB(AES192);
+AESNI_ENCRYPT_BLOCK_CBC(AES192);
+AESNI_DECRYPT_BLOCK_CBC(AES192);
+AESNI_ENCRYPT_BLOCK_CFB(AES192);
+AESNI_DECRYPT_BLOCK_CFB(AES192);
+AESNI_ENCRYPT_BLOCK_OFB(AES192);
+AESNI_DECRYPT_BLOCK_OFB(AES192);
+AESNI_ENCRYPT_BLOCK_CTR(AES192);
+AESNI_DECRYPT_BLOCK_CTR(AES192);
+
+AESNI_ENCRYPT_BLOCK_ECB(AES256);
+AESNI_DECRYPT_BLOCK_ECB(AES256);
+AESNI_ENCRYPT_BLOCK_CBC(AES256);
+AESNI_DECRYPT_BLOCK_CBC(AES256);
+AESNI_ENCRYPT_BLOCK_CFB(AES256);
+AESNI_DECRYPT_BLOCK_CFB(AES256);
+AESNI_ENCRYPT_BLOCK_OFB(AES256);
+AESNI_DECRYPT_BLOCK_OFB(AES256);
+AESNI_ENCRYPT_BLOCK_CTR(AES256);
+AESNI_DECRYPT_BLOCK_CTR(AES256);
+
+/**
+ * \brief Expands an AES-128 key into 10 encryption round keys.
+ *
+ * \param[in] key The AES-128 key.
+ * \param[out] encryption_keys The AES-128 encryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES128_expand_key(
+ const AesNI_AES128_Key* key,
+ AesNI_AES128_RoundKeys* encryption_keys)
+{
+ assert(encryption_keys);
+
+ aesni_AES128_expand_key_(key->key, encryption_keys);
+}
+
+/**
+ * \brief Derives AES-128 decryption round keys from AES-128 encryption round keys.
+ *
+ * \param[in] encryption_keys The AES-128 encryption round keys. Must not be `NULL`.
+ * \param[out] decryption_keys The AES-128 decryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES128_derive_decryption_keys(
+ const AesNI_AES128_RoundKeys* encryption_keys,
+ AesNI_AES128_RoundKeys* decryption_keys)
+{
+ assert(encryption_keys);
+ assert(decryption_keys);
+
+ aesni_AES128_derive_decryption_keys_(encryption_keys, decryption_keys);
+}
+
+/**
+ * \brief Expands an AES-192 key into 12 encryption round keys.
+ *
+ * \param[in] key The AES-192 key.
+ * \param[out] encryption_keys The AES-192 encryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES192_expand_key(
+ const AesNI_AES192_Key* key,
+ AesNI_AES192_RoundKeys* encryption_keys)
+{
+ assert(key);
+ assert(encryption_keys);
+
+ aesni_AES192_expand_key_(key->lo, key->hi, encryption_keys);
+}
+
+/**
+ * \brief Derives AES-192 decryption round keys from AES-192 encryption round keys.
+ *
+ * \param[in] encryption_keys The AES-192 encryption round keys. Must not be `NULL`.
+ * \param[out] decryption_keys The AES-192 decryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES192_derive_decryption_keys(
+ const AesNI_AES192_RoundKeys* encryption_keys,
+ AesNI_AES192_RoundKeys* decryption_keys)
+{
+ assert(encryption_keys);
+ assert(decryption_keys);
+
+ aesni_AES192_derive_decryption_keys_(encryption_keys, decryption_keys);
+}
+
+/**
+ * \brief Expands an AES-256 key into 14 encryption round keys.
+ *
+ * \param[in] key The AES-256 key.
+ * \param[out] encryption_keys The AES-256 encryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES256_expand_key(
+ const AesNI_AES256_Key* key,
+ AesNI_AES256_RoundKeys* encryption_keys)
+{
+ assert(key);
+ assert(encryption_keys);
+
+ aesni_AES256_expand_key_(key->lo, key->hi, encryption_keys);
+}
+
+/**
+ * \brief Derives AES-256 decryption round keys from AES-256 encryption round keys.
+ *
+ * \param[in] encryption_keys The AES-256 encryption round keys. Must not be `NULL`.
+ * \param[out] decryption_keys The AES-256 decryption round keys. Must not be `NULL`.
+ */
+static __inline void __fastcall aesni_AES256_derive_decryption_keys(
+ const AesNI_AES256_RoundKeys* encryption_keys,
+ AesNI_AES256_RoundKeys* decryption_keys)
+{
+ assert(encryption_keys);
+ assert(decryption_keys);
+
+ aesni_AES256_derive_decryption_keys_(encryption_keys, decryption_keys);
+}
+
+#ifdef __cplusplus
+}
+#endif