aes.h

 
#ifndef AES_H
#define AES_H
 
#include <string.h>
 
#include "defs.h"
#include "lut.h"
#include "aesMath.h"
 
// Logger
#ifdef DEBUG
#include "logger.h"
#endif
 
// Masking
#ifdef MASKING
#include "masking.h"
#endif
 
// Hiding
#if defined(SHUFFLING) || defined(DUMMY_OPS)
#include "hiding.h"
#endif
 
class AES
{
public:
    AES(const aes_key_t masterKey);
 
    void decrypt(uint8_t *cipher);
    
private:
    // *******************************************************************************
    // Private Attributes ************************************************************
    // *******************************************************************************
    sub_keys_t mSubkeys = {};       
    static uint8_t mRCs[ROUNDS];    
 
    // Logger 
    #ifdef DEBUG
    Logger mLog;                    
    #endif
 
    // Masking
    #ifdef MASKING
    Masking mMasking;               
    sub_keys_t mOriginalSubKeys;    
    #endif
 
    // Hiding
    #if defined(SHUFFLING) || defined(DUMMY_OPS)
    Hiding mHiding;                 
    #endif
 
    #ifdef SHUFFLING
    uint8_t mShuffledSBoxIndices[STATE_BYTES] = {}; 
    #endif
    
    // ******************************************************************************
    // Private Methods **************************************************************
    // ******************************************************************************
    // Key Schedule *****************************************************************
    void createKeySchedule(const aes_key_t masterKey, sub_keys_t subKeys) const;
    
    // Key Addition Layer ***********************************************************
    void addRoundKey(const aes_key_t roundKey, state_t state);
 
    // Diffusion Layer **************************************************************
    void invMixCols(state_t state);
 
    void invShiftRows(state_t state);
 
    // Byte Substitution layer ******************************************************
    void invByteSub(state_t state);
};
 
#endif // AES_H