//  寫了一個AES的C++實現(xiàn)，僅支持128位密鑰， 寫得匆忙，不夠規(guī)范，僅供參考。

//    AES.h

#ifndef AES_H_
#define AES_H_

#include <bitset>
#include <utility>

using namespace std;

class AES
{
public:
    typedef unsigned char    byte;
    static const int KEY_SIZE = 16;    //    密鑰長度為128位
    static const int N_ROUND = 11;
    byte plainText[16];    //    明文
    byte state[16];    //    當(dāng)前分組。
    byte cipherKey[16];    //    密鑰
    byte roundKey[N_ROUND][16];    //輪密鑰
    byte cipherText[16];    //密文
    byte SBox[16][16];    //    S盒
    byte InvSBox[16][16];    //    逆S盒    
    void EncryptionProcess();    
    void DecryptionProcess();
    void Round(const int& round);
    void InvRound(const int& round);
    void FinalRound();
    void InvFinalRound();
    void KeyExpansion();
    void AddRoundKey(const int& round);    
    void SubBytes();    
    void InvSubBytes();
    void ShiftRows();    
    void InvShiftRows();
    void MixColumns();    
    void InvMixColumns();
    void BuildSBox();
    void BuildInvSBox();
    void InitialState(const byte* text);
    void InitialCipherText();    
    void InitialplainText();        
    byte GFMultplyByte(const byte& left, const byte& right);
    const byte* GFMultplyBytesMatrix(const byte* left, const byte* right);
public:    
    AES();    
    const byte* Cipher(const byte* text, const byte* key, const int& keySize);    
    const byte* InVCipher(const byte* text, const byte* key, const int& keySize);
};


void AES::EncryptionProcess()
{    //    加密過程
    InitialState(plainText);
    KeyExpansion();    //    密鑰擴展
    AddRoundKey(0);    //    輪密鑰加
    for(int i = 1; i < N_ROUND-1; ++i)
    {            
        Round(i);
    }
    FinalRound();
    InitialCipherText();
}

void AES::DecryptionProcess()
{    //    解密過程
    InitialState(cipherText);
    KeyExpansion();
    InvFinalRound();
    for(int i = N_ROUND-2; i > 0 ; --i)
    {            
        InvRound(i);
    }
    AddRoundKey(0);
    InitialplainText();
}

void AES::Round(const int& round)
{    //    正常輪
    SubBytes();
    ShiftRows();
    MixColumns();
    AddRoundKey(round);        
}

void AES::InvRound(const int& round)
{    //    正常輪的逆
    AddRoundKey(round);    
    InvMixColumns();
    InvShiftRows();
    InvSubBytes();            
}

void AES::FinalRound()
{    //    最后輪
    SubBytes();
    ShiftRows();
    AddRoundKey(N_ROUND - 1);
}

void AES::InvFinalRound()
{    //    最后輪的逆
    AddRoundKey(N_ROUND - 1);        
    InvShiftRows();
    InvSubBytes();        
}

void AES::KeyExpansion()
{    //    密鑰擴展
    const byte rcon[N_ROUND][4] = { 
        {0x00, 0x00, 0x00, 0x00},  
        {0x01, 0x00, 0x00, 0x00},
        {0x02, 0x00, 0x00, 0x00},
        {0x04, 0x00, 0x00, 0x00},
        {0x08, 0x00, 0x00, 0x00},
        {0x10, 0x00, 0x00, 0x00},
        {0x20, 0x00, 0x00, 0x00},
        {0x40, 0x00, 0x00, 0x00},
        {0x80, 0x00, 0x00, 0x00},
        {0x1b, 0x00, 0x00, 0x00},
        {0x36, 0x00, 0x00, 0x00} };        
        for(int i = 0; i < 16; ++i)
        {
            roundKey[0][i] = cipherKey[i];
        }
        for(int i = 0; i < 4; ++i)
        {    //  roundKey[0][16]為cipherKey的轉(zhuǎn)置矩陣
            for(int j = 0; j < 4; ++j)
            {
                roundKey[0][4*i + j] = cipherKey[4*j + i];
            }
        }
        for(int roundIndex = 1; roundIndex < N_ROUND; ++roundIndex)
        {
            byte rotWord[4] = {0x00};
            rotWord[0] = roundKey[roundIndex - 1][3];
            rotWord[1] = roundKey[roundIndex - 1][7];
            rotWord[2] = roundKey[roundIndex - 1][11];
            rotWord[3] = roundKey[roundIndex - 1][15];
            std::swap<byte>(rotWord[0], rotWord[1]);
            std::swap<byte>(rotWord[1], rotWord[2]);
            std::swap<byte>(rotWord[2], rotWord[3]);        
            for(int i = 0; i < 4; ++i)
            {
                rotWord[i] = SBox[ rotWord[i] >> 4][ rotWord[i] & 0x0f ];
                roundKey[roundIndex][4*i] = roundKey[roundIndex - 1][4*i] ^ rotWord[i] ^ rcon[roundIndex][i];
            }
            for(int j = 1; j < 4; ++j)
            {
                for(int i = 0; i < 4; ++i)
                {
                    roundKey[roundIndex][4*i + j] = roundKey[roundIndex - 1][4*i + j] ^ roundKey[roundIndex][4*i + j - 1];
                }
            }
        }
}

void AES::AddRoundKey(const int& round)
{    //    輪密鑰加
    for(int i = 0; i < 16; ++i)
    {    //    利用當(dāng)前分組state和第round組擴展密鑰進行按位異或
        state[i] ^= roundKey[round][i];
    }
}

void AES::SubBytes()
{    //    字節(jié)代換
    for(int i = 0; i < 16; ++i)
    {
        state[i] = SBox[ state[i] >> 4][ state[i] & 0x0f ];
    }
}

void AES::InvSubBytes()
{    //    逆字節(jié)代換
    for(int i = 0; i < 16; ++i)
    {
        state[i] = InvSBox[ state[i] >> 4][ state[i] & 0x0f ];
    }
}

void AES::ShiftRows()
{    //    行變換
    //state第一行保持不變
    // Do nothing.
    //state第二行循環(huán)左移一個字節(jié)
    std::swap<byte>(state[4], state[5]);
    std::swap<byte>(state[5], state[6]);
    std::swap<byte>(state[6], state[7]);        
    //state第三行循環(huán)左移兩個字節(jié)
    std::swap<byte>(state[8], state[10]);
    std::swap<byte>(state[9], state[11]);
    //state第三行循環(huán)左移三個字節(jié)
    std::swap<byte>(state[14], state[15]);
    std::swap<byte>(state[13], state[14]);
    std::swap<byte>(state[12], state[13]);    
}

void AES::InvShiftRows()
{    //    行變換反演
    //state第一行保持不變
    // Do nothing.
    //state第二行循環(huán)右移一個字節(jié)
    std::swap<byte>(state[6], state[7]);
    std::swap<byte>(state[5], state[6]);
    std::swap<byte>(state[4], state[5]);        
    //state第三行循環(huán)右移兩個字節(jié)
    std::swap<byte>(state[9], state[11]);
    std::swap<byte>(state[8], state[10]);
    //state第三行循環(huán)右移三個字節(jié)
    std::swap<byte>(state[12], state[13]);
    std::swap<byte>(state[13], state[14]);        
    std::swap<byte>(state[14], state[15]);    
}

void AES::MixColumns()
{    //    列混淆
    byte matrix[4][4] = {
        {0x02, 0x03, 0x01, 0x01},
        {0x01, 0x02, 0x03, 0x01},
        {0x01, 0x01, 0x02, 0x03},
        {0x03, 0x01, 0x01, 0x02}};
        const byte* temp = GFMultplyBytesMatrix((byte*)matrix, state);
        for(int i = 0; i < 16; ++i)
        {
            state[i] = temp[i];
        }
        delete[] temp;
}

void AES::InvMixColumns()
{    //    列混淆反演
    byte matrix[4][4] = {
        {0x0e, 0x0b, 0x0d, 0x09},
        {0x09, 0x0e, 0x0b, 0x0d},
        {0x0d, 0x09, 0x0e, 0x0b},
        {0x0b, 0x0d, 0x09, 0x0e} };
        const byte* temp = GFMultplyBytesMatrix((byte*)matrix, state);
        for(int i = 0; i < 16; ++i)
        {
            state[i] = temp[i];
        }
        delete[] temp;
}

void AES::BuildSBox()
{    //    構(gòu)建S盒
    byte box[16][16] = 
    {