38 lines
1.3 KiB
C++
38 lines
1.3 KiB
C++
|
#ifndef _RIJNDAEL_H_
|
||
|
|
#define _RIJNDAEL_H_
|
||
|
|
|
||
|
|
/**************************************************************************
|
||
|
|
* This code is based on Szymon Stefanek AES implementation: *
|
||
|
|
* http://www.esat.kuleuven.ac.be/~rijmen/rijndael/rijndael-cpplib.tar.gz *
|
||
|
|
* *
|
||
|
|
* Dynamic tables generation is based on the Brian Gladman's work: *
|
||
|
|
* http://fp.gladman.plus.com/cryptography_technology/rijndael *
|
||
|
|
**************************************************************************/
|
||
|
|
|
||
|
|
#define _MAX_KEY_COLUMNS (256/32)
|
||
|
|
#define _MAX_ROUNDS 14
|
||
|
|
#define MAX_IV_SIZE 16
|
||
|
|
|
||
|
|
class Rijndael
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
enum Direction { Encrypt , Decrypt };
|
||
|
|
private:
|
||
|
|
void keySched(byte key[_MAX_KEY_COLUMNS][4]);
|
||
|
|
void keyEncToDec();
|
||
|
|
void encrypt(const byte a[16], byte b[16]);
|
||
|
|
void decrypt(const byte a[16], byte b[16]);
|
||
|
|
void GenerateTables();
|
||
|
|
|
||
|
|
Direction m_direction;
|
||
|
|
byte m_initVector[MAX_IV_SIZE];
|
||
|
|
byte m_expandedKey[_MAX_ROUNDS+1][4][4];
|
||
|
|
public:
|
||
|
|
Rijndael();
|
||
|
|
void init(Direction dir,const byte *key,byte *initVector);
|
||
|
|
size_t blockEncrypt(const byte *input, size_t inputLen, byte *outBuffer);
|
||
|
|
size_t blockDecrypt(const byte *input, size_t inputLen, byte *outBuffer);
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // _RIJNDAEL_H_
|