#ifndef AES_INTERNAL_H_
#define AES_INTERNAL_H_

#include <stdlib.h>

#include "aes.h"

#define WORD_LENGTH 4
#define ROUNDS 10

#define BOOL int
#define TRUE 1
#define FALSE 0

void aes__shift_rows(uint8_t* block, aes_encryption_mode_t mode);
uint8_t aes__xtime(uint8_t x);
void aes__mix_words(uint8_t* block, aes_encryption_mode_t mode);
uint8_t* aes__expand_key(uint8_t* key);
void aes__encrypt_block(uint8_t* block, uint8_t* round_keys);
void aes__decrypt_block(uint8_t* block, uint8_t* round_keys);
int aes__is_valid_padding(uint8_t* block);

#endif // AES_INTERNAL_H_