#ifndef AES_AES_H
#define AES_AES_H

#include <stdint.h>
#include "aes_types.h"

typedef aes_key AESKey;
AESKey *allocate_key128();
void generate_key128(AESKey *k, const uint8_t *bytes);
void free_key128(AESKey *k);
void encrypt_ecb(const AESKey *k, uint8_t *dst, const uint8_t *src, const uint32_t nr);
void decrypt_ecb(const AESKey *k, uint8_t *dst, const uint8_t *src, const uint32_t nr);
void aes_gcm_full_encrypt( const AESKey *k
                         , uint8_t *iv, uint32_t ivLen
                         , uint8_t *aad, uint32_t aadLen
                         , uint8_t *pt, uint32_t ptLen
                         , uint8_t *ct, uint8_t *tag);
void aes_gcm_full_decrypt( const AESKey *k
                         , uint8_t *iv, uint32_t ivLen
                         , uint8_t *aad, uint32_t aadLen
                         , uint8_t *ct, uint32_t ctLen
                         , uint8_t *pt, uint8_t *tag);
void free_gcm(aes_gcm *g);
aes_gcm *allocate_gcm();
void aes_gcm_init(aes_gcm *gcm, const aes_key *key);
void aes_gcm_init_ctx(aes_gcm_ctx *ctx, const aes_gcm *gcm, uint8_t *iv, uint32_t len);
void aes_gcm_encrypt(uint8_t *output, const aes_gcm *gcm, aes_gcm_ctx *ctx, const uint8_t *input, uint32_t length);
void aes_gcm_decrypt(uint8_t *output, const aes_gcm *gcm, aes_gcm_ctx *ctx, const uint8_t *input, uint32_t length);
void aes_gcm_aad(const aes_gcm *gcm, aes_gcm_ctx *ctx, uint8_t *input, uint32_t length);
void aes_gcm_finish(uint8_t *tag, const aes_gcm *gcm, aes_gcm_ctx *ctx);
void aes_gcm_enc_finish( uint8_t *output, uint8_t *tag
                       , const aes_gcm *gcm
                       , uint8_t *iv, uint32_t ivLen
                       , uint8_t *input, uint32_t inputLen
                       , uint8_t *aad, uint32_t aadLen);
void aes_gcm_dec_finish( uint8_t *output, uint8_t *tag
                       , const aes_gcm *gcm
                       , uint8_t *iv, uint32_t ivLen
                       , uint8_t *input, uint32_t inputLen
                       , uint8_t *aad, uint32_t aadLen);

void encrypt_ctr( const AESKey *key
                , const uint8_t *iv  /* 16 bytes buffer with the count */
                , uint8_t *newIV     /* the return buffer for the next IV (or NULL) */
                , uint8_t *dst       /* 'len' byte buffer for output */
                , const uint8_t *src /* 'len' bytes of input */
                , uint32_t len);     /* Length in bytes */

#endif