/**
 * \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 "aes.h"
#include "error.h"

#include <stdlib.h>

#ifdef __cplusplus
extern "C"
{
#endif

typedef union
{
    AesNI_Aes128_Key aes128_key;
    AesNI_Aes192_Key aes192_key;
    AesNI_Aes256_Key aes256_key;
}
AesNI_BoxAlgorithmParams;

typedef enum
{
    AESNI_AES128,
    AESNI_AES192,
    AESNI_AES256,
}
AesNI_BoxAlgorithm;

typedef enum
{
    AESNI_ECB,
    AESNI_CBC,
    AESNI_CFB,
    AESNI_OFB,
    AESNI_CTR,
}
AesNI_BoxMode;

typedef union
{
    AesNI_Aes128_RoundKeys aes128_encryption_keys;
    AesNI_Aes192_RoundKeys aes192_encryption_keys;
    AesNI_Aes256_RoundKeys aes256_encryption_keys;
}
AesNI_BoxEncryptionParams;

typedef union
{
    AesNI_Aes128_RoundKeys aes128_decryption_keys;
    AesNI_Aes192_RoundKeys aes192_decryption_keys;
    AesNI_Aes256_RoundKeys aes256_decryption_keys;
}
AesNI_BoxDecryptionParams;

typedef union
{
    AesNI_Aes_Block aes_block;
}
AesNI_BoxBlock;

typedef AesNI_StatusCode (*AesNI_BoxDeriveParams)(
    const AesNI_BoxAlgorithmParams* params,
    AesNI_BoxEncryptionParams*,
    AesNI_BoxDecryptionParams*,
    AesNI_ErrorDetails* err_details);

typedef AesNI_StatusCode (*AesNI_BoxEncryptBlock)(
    const AesNI_BoxBlock* plaintext,
    const AesNI_BoxEncryptionParams* params,
    AesNI_BoxBlock* ciphertext,
    AesNI_ErrorDetails* err_details);

typedef AesNI_StatusCode (*AesNI_BoxDecryptBlock)(
    const AesNI_BoxBlock* ciphertext,
    const AesNI_BoxDecryptionParams* params,
    AesNI_BoxBlock* plaintext,
    AesNI_ErrorDetails* err_details);

typedef AesNI_StatusCode (*AesNI_BoxXorBlock)(
    AesNI_BoxBlock*,
    const AesNI_BoxBlock*,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxNextCounter)(
    AesNI_BoxBlock*,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxGetBlockSize)(
    size_t*,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxStoreBlock)(
    void*,
    const AesNI_BoxBlock*,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxStorePartialBlock)(
    void*,
    const AesNI_BoxBlock*,
    size_t,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxLoadBlock)(
    AesNI_BoxBlock*,
    const void*,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxLoadPartialBlock)(
    AesNI_BoxBlock*,
    const void*,
    size_t,
    AesNI_ErrorDetails*);

typedef AesNI_StatusCode (*AesNI_BoxLoadBlockWithPadding)(
    AesNI_BoxBlock*,
    const void*,
    size_t,
    AesNI_ErrorDetails*);

typedef struct
{
    AesNI_BoxDeriveParams derive_params;
    AesNI_BoxEncryptBlock encrypt_block;
    AesNI_BoxDecryptBlock decrypt_block;
    AesNI_BoxXorBlock xor_block;
    AesNI_BoxNextCounter next_counter;
    AesNI_BoxGetBlockSize get_block_size;
    AesNI_BoxStoreBlock store_block;
    AesNI_BoxStorePartialBlock store_partial_block;
    AesNI_BoxLoadBlock load_block;
    AesNI_BoxLoadPartialBlock load_partial_block;
    AesNI_BoxLoadBlockWithPadding load_block_with_padding;
}
AesNI_BoxAlgorithmInterface;

typedef struct
{
    const AesNI_BoxAlgorithmInterface* algorithm;
    AesNI_BoxEncryptionParams encrypt_params;
    AesNI_BoxDecryptionParams decrypt_params;
    AesNI_BoxMode mode;
    AesNI_BoxBlock iv;
}
AesNI_Box;

#ifdef __cplusplus
}
#endif