You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.6 KiB
C++
67 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#ifndef _AES_H
|
|
#define _AES_H
|
|
#include <exception>
|
|
#include <cstring>
|
|
#include <string>
|
|
#define BLOCK_SIZE 16
|
|
using namespace std;
|
|
|
|
class AES
|
|
{
|
|
public:
|
|
enum
|
|
{
|
|
ECB = 0, CBC = 1, CFB = 2
|
|
};
|
|
|
|
private:
|
|
enum
|
|
{
|
|
DEFAULT_BLOCK_SIZE = 16
|
|
};
|
|
enum
|
|
{
|
|
MAX_BLOCK_SIZE = 32, MAX_ROUNDS = 14, MAX_KC = 8, MAX_BC = 8
|
|
};
|
|
public:
|
|
AFX_EXT_CLASS AES();
|
|
virtual ~AES();
|
|
private:
|
|
//Key Initialization Flag
|
|
bool m_bKeyInit;
|
|
//Encryption (m_Ke) round key
|
|
int m_Ke[MAX_ROUNDS + 1][MAX_BC];
|
|
//Decryption (m_Kd) round key
|
|
int m_Kd[MAX_ROUNDS + 1][MAX_BC];
|
|
//Key Length
|
|
int m_keylength;
|
|
//Block Size
|
|
int m_blockSize;
|
|
//Number of Rounds
|
|
int m_iROUNDS;
|
|
//Chain Block
|
|
char m_chain0[MAX_BLOCK_SIZE];
|
|
char m_chain[MAX_BLOCK_SIZE];
|
|
//Auxiliary private use buffers
|
|
int tk[MAX_KC];
|
|
int a[MAX_BC];
|
|
int t[MAX_BC];
|
|
private:
|
|
void Xor(char* buff, char const* chain);
|
|
void DefEncryptBlock(char const* in, char* result);
|
|
void DefDecryptBlock(char const* in, char* result);
|
|
void EncryptBlock(char const* in, char* result);
|
|
void DecryptBlock(char const* in, char* result);
|
|
public:
|
|
AFX_EXT_CLASS void MakeKey(char const* key, char const* chain, int keylength =
|
|
DEFAULT_BLOCK_SIZE, int blockSize = DEFAULT_BLOCK_SIZE);
|
|
AFX_EXT_CLASS void Encrypt(char const* in, char* result, size_t n, int iMode = ECB);
|
|
AFX_EXT_CLASS void Decrypt(char const* in, char* result, size_t n, int iMode = ECB);
|
|
AFX_EXT_CLASS string EncryptionAES(const string& strSrc, const char* pKey, const char* pIV);
|
|
AFX_EXT_CLASS string DecryptionAES(const string& strSrc, const char* pKey, const char* pIV);
|
|
};
|
|
|
|
#endif // __RIJNDAEL_H__
|