58 lines
1.3 KiB
C
58 lines
1.3 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <string.h>
|
|||
|
|
#include <stdio.h>
|
|||
|
|
|
|||
|
|
class AES {
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
AES();
|
|||
|
|
AES(const unsigned char *key);
|
|||
|
|
virtual ~AES();
|
|||
|
|
void encrypt(const unsigned char data[16], unsigned char out[16]);
|
|||
|
|
void decrypt(const unsigned char data[16], unsigned char out[16]);
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
//
|
|||
|
|
int mNb;
|
|||
|
|
|
|||
|
|
//word length of the secret key used in one turn
|
|||
|
|
int mNk;
|
|||
|
|
|
|||
|
|
//number of turns
|
|||
|
|
int mNr;
|
|||
|
|
|
|||
|
|
//the secret key,which can be 16bytes，24bytes or 32bytes
|
|||
|
|
unsigned char mKey[32];
|
|||
|
|
|
|||
|
|
//the extended key,which can be 176bytes,208bytes,240bytes
|
|||
|
|
unsigned char mW[60][4];
|
|||
|
|
|
|||
|
|
static unsigned char sBox[];
|
|||
|
|
static unsigned char invSBox[];
|
|||
|
|
//constant
|
|||
|
|
static unsigned char rcon[];
|
|||
|
|
void setKey(const unsigned char *key);
|
|||
|
|
|
|||
|
|
void subBytes(unsigned char state[][4]);
|
|||
|
|
void shiftRows(unsigned char state[][4]);
|
|||
|
|
void mixColumns(unsigned char state[][4]);
|
|||
|
|
void addRoundKey(unsigned char state[][4], unsigned char w[][4]);
|
|||
|
|
|
|||
|
|
void invSubBytes(unsigned char state[][4]);
|
|||
|
|
void invShiftRows(unsigned char state[][4]);
|
|||
|
|
void invMixColumns(unsigned char state[][4]);
|
|||
|
|
|
|||
|
|
void keyExpansion();
|
|||
|
|
|
|||
|
|
//
|
|||
|
|
unsigned char GF28Multi(unsigned char s, unsigned char a);
|
|||
|
|
|
|||
|
|
void rotWord(unsigned char w[]);
|
|||
|
|
void subWord(unsigned char w[]);
|
|||
|
|
|
|||
|
|
//get the secret key
|
|||
|
|
void getKeyAt(unsigned char key[][4],int i);
|
|||
|
|
|
|||
|
|
};
|