masking.h

 
#ifndef MASK_H
#define MASK_H
 
#include "defs.h"
#include "lut.h"
#include "aesMath.h"
#include "rng.h"
 
// Logger
#ifdef DEBUG
#include "logger.h"
#endif
 
class Masking
{
public:
    Masking() = default;
 
    void init();
 
    void maskSubKeys(const sub_keys_t subKeys, sub_keys_t maskedSubKeys) const;
    
    void invMaskState(state_t state) const;
 
    void invReMaskState(state_t state) const;
 
    void invUnMaskState(state_t state) const;
 
    uint8_t getInvMaskedSBoxValue(const uint8_t index) const { return mInvMaskedSBox[index]; }
 
private:
    // ******************************************************************************
    // Private Structures ***********************************************************
    // ******************************************************************************
    struct mask_t
    {
        uint8_t input;  
        uint8_t output; 
    };
 
    // ******************************************************************************
    // Private Attributes ***********************************************************
    // ******************************************************************************
    uint8_t mInvMaskedSBox[SBOX_BYTES]; 
    
    mask_t mSubByteMask     = {};
 
    mask_t mMixColMasks[4]  = {};
    
    RNG mRNG;       
    #ifdef DEBUG
    Logger mLog;    
    #endif
    // ******************************************************************************
    // Private Methods **************************************************************
    // ******************************************************************************
    void initInvMaskedSBox(uint8_t maskedSBox[], const mask_t &subByteMask) const;
 
    void initMixColInputMask(mask_t mixColMasks[]) const;
};
 
#endif // MASK_H