aes.cpp

自编高级加密算法体制AES

#include "StdAfx.h"
#include "AES.h"

AES::AES(void)//初始化数据
{
	long firstKey[4*Nk] ={0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c};
	for(int i = 0; i < 4*Nk; i ++)
		m_Key[i] = firstKey[i];
	for(int i = 0; i < (EncryptionRounds+1)*Nb; i++)
		m_KeyW[i] = 0;
	m_Rcon[0] = 0x01000000;
	for(int i = 1; i < (EncryptionRounds*Nb+1)/Nk; i++)
	{
		m_Rcon[i] = 2*m_Rcon[i-1].to_ulong();
		if(m_Rcon[i].to_ulong()==0)
			m_Rcon[i] = 0x1b000000;
	}
	for(int i = 0; i < (EncryptionRounds+1)*Nb/4; i++)
		for(int j = 0; j < 4; j++)
			for(int k = 0; k < 4; k++)
				m_KeyB[i][j][k] = 0;
	
	long datasource[16][16] = {{0x63,0x7c,0x77,0x7b,0xf2,0x6b,0x6f,0xc5,0x30,0x01,0x67,0x2b,0xfe,0xd7,0xab,0x76},
	{0xca,0x82,0xc9,0x7d,0xfa,0x59,0x47,0xf0,0xad,0xd4,0xa2,0xaf,0x9c,0xa4,0x72,0xc0},
	{0xb7,0xfd,0x93,0x26,0x36,0x3f,0xf7,0xcc,0x34,0xa5,0xe5,0xf1,0x71,0xd8,0x31,0x15},
	{0x04,0xc7,0x23,0xc3,0x18,0x96,0x05,0x9a,0x07,0x12,0x80,0xe2,0xeb,0x27,0xb2,0x75},
	{0x09,0x83,0x2c,0x1a,0x1b,0x6e,0x5a,0xa0,0x52,0x3b,0xd6,0xb3,0x29,0xe3,0x2f,0x84},
	{0x53,0xd1,0x00,0xed,0x20,0xfc,0xb1,0x5b,0x6a,0xcb,0xbe,0x39,0x4a,0x4c,0x58,0xcf},
	{0xd0,0xef,0xaa,0xfb,0x43,0x4d,0x33,0x85,0x45,0xf9,0x02,0x7f,0x50,0x3c,0x9f,0xa8},
	{0x51,0xa3,0x40,0x8f,0x92,0x9d,0x38,0xf5,0xbc,0xb6,0xda,0x21,0x10,0xff,0xf3,0xd2},
	{0xcd,0x0c,0x13,0xec,0x5f,0x97,0x44,0x17,0xc4,0xa7,0x7e,0x3d,0x64,0x5d,0x19,0x73},
	{0x60,0x81,0x4f,0xdc,0x22,0x2a,0x90,0x88,0x46,0xee,0xb8,0x14,0xde,0x5e,0x0b,0xdb},
	{0xe0,0x32,0x3a,0x0a,0x49,0x06,0x24,0x5c,0xc2,0xd3,0xac,0x62,0x91,0x95,0xe4,0x79},
	{0xe7,0xc8,0x37,0x6d,0x8d,0xd5,0x4e,0xa9,0x6c,0x56,0xf4,0xea,0x65,0x7a,0xae,0x08},
	{0xba,0x78,0x25,0x2e,0x1c,0xa6,0xb4,0xc6,0xe8,0xdd,0x74,0x1f,0x4d,0xbd,0x8b,0x8a},
	{0x70,0x3e,0xb5,0x66,0x48,0x03,0xf6,0x0e,0x61,0x35,0x57,0xb9,0x86,0xc1,0x1d,0x9e},
	{0xe1,0xf8,0x98,0x11,0x69,0xd9,0x8e,0x94,0x9b,0x1e,0x87,0xe9,0xce,0x55,0x28,0xdf},
	{0x8c,0xa1,0x89,0x0d,0xbf,0xe6,0x42,0x68,0x41,0x99,0x2d,0x0f,0xb0,0x54,0xbb,0x16}};

	long logtableSource[16][16] = {{0x00,0x00,0x19,0x01,0x32,0x02,0x1a,0xc6,0x4b,0xc7,0x1b,0x68,0x33,0xee,0xdf,0x03},
	{0x64,0x04,0xe0,0x0e,0x34,0x8d,0x81,0xef,0x4c,0x71,0x08,0xc8,0xf8,0x69,0x1c,0xc1},
	{0x7d,0xc2,0x1d,0xb5,0xf9,0xb9,0x27,0x6a,0x4d,0xe4,0xa6,0x72,0x9a,0xc9,0x09,0x78},
	{0x65,0x2f,0x8a,0x05,0x21,0x0f,0xe1,0x24,0x12,0xf0,0x82,0x45,0x35,0x93,0xda,0x8e},
	{0x96,0x8f,0xdb,0xbd,0x36,0xd0,0xce,0x94,0x13,0x5c,0xd2,0xf1,0x40,0x46,0x83,0x38},
	{0x66,0xdd,0xfd,0x30,0xbf,0x06,0x8b,0x62,0xb3,0x25,0xe2,0x98,0x22,0x88,0x91,0x10},
	{0x7e,0x6e,0x48,0xc3,0xa3,0xb6,0x1e,0x42,0x3a,0x6b,0x28,0x54,0xfa,0x85,0x3d,0xba},
	{0x2b,0x79,0x0a,0x15,0x9b,0x9f,0x5e,0xca,0x4e,0xd4,0xac,0xe5,0xf3,0x73,0xa7,0x57},
	{0xaf,0x58,0xa8,0x50,0xf4,0xea,0xd6,0x74,0x4f,0xae,0xe9,0xd5,0xe7,0xe6,0xad,0xe8},
	{0x2c,0xd7,0x75,0x7a,0xeb,0x16,0x0b,0xf5,0x59,0xcb,0x5f,0xb0,0x9c,0xa9,0x51,0xa0},
	{0x7f,0x0c,0xf6,0x6f,0x17,0xc4,0x49,0xec,0xd8,0x43,0x1f,0x2d,0xa4,0x76,0x7b,0xb7},
	{0xcc,0xbb,0x3e,0x5a,0xfb,0x60,0xb1,0x86,0x3b,0x52,0xa1,0x6c,0xaa,0x55,0x29,0x9d},
	{0x97,0xb2,0x87,0x90,0x61,0xbe,0xdc,0xfc,0xbc,0x95,0xcf,0xcd,0x37,0x3f,0x5b,0xd1},
	{0x53,0x39,0x84,0x3c,0x41,0xa2,0x6d,0x47,0x14,0x2a,0x9e,0x5d,0x56,0xf2,0xd3,0xab},
	{0x44,0x11,0x92,0xd9,0x23,0x20,0x2e,0x89,0xb4,0x7c,0xb8,0x26,0x77,0x99,0xe3,0xa5},
	{0x67,0x4a,0xed,0xde,0xc5,0x31,0xfe,0x18,0x0d,0x63,0x8c,0x80,0xc0,0xf7,0x70,0x07}};

	long invlogtableSource[16][16] ={{0x01,0x03,0x05,0x0f,0x11,0x33,0x55,0xff,0x1a,0x2e,0x72,0x96,0xa1,0xf8,0x13,0x35},
	{0x5f,0xe1,0x38,0x48,0xd8,0x73,0x95,0xa4,0xf7,0x02,0x06,0x0a,0x1e,0x22,0x66,0xaa},
	{0xe5,0x34,0x5c,0xe4,0x37,0x59,0xeb,0x26,0x6a,0xbe,0xd9,0x70,0x90,0xab,0xe6,0x31},
	{0x53,0xf5,0x04,0x0c,0x14,0x3c,0x44,0xcc,0x4f,0xd1,0x68,0xb8,0xd3,0x6e,0xb2,0xcd},
	{0x4c,0xd4,0x67,0xa9,0xe0,0x3b,0x4d,0xd7,0x62,0xa6,0xf1,0x08,0x18,0x28,0x78,0x88},
	{0x83,0x9e,0xb9,0xd0,0x6b,0xbd,0xdc,0x7f,0x81,0x98,0xb3,0xce,0x49,0xdb,0x76,0x9a},
	{0xb5,0xc4,0x57,0xf9,0x10,0x30,0x50,0xf0,0x0b,0x1d,0x27,0x69,0xbb,0xd6,0x61,0xa3},
	{0xfe,0x19,0x2b,0x7d,0x87,0x92,0xad,0xec,0x2f,0x71,0x93,0xae,0xe9,0x20,0x60,0xa0},
	{0xfb,0x16,0x3a,0x4e,0xd2,0x6d,0xb7,0xc2,0x5d,0xe7,0x32,0x56,0xfa,0x15,0x3f,0x41},
	{0xc3,0x5e,0xe2,0x3d,0x47,0xc9,0x40,0xc0,0x5b,0xed,0x2c,0x74,0x9c,0xbf,0xda,0x75},
	{0x9f,0xba,0xd5,0x64,0xac,0xef,0x2a,0x7e,0x82,0x9d,0xbc,0xdf,0x7a,0x8e,0x89,0x80},
	{0x9b,0xb6,0xc1,0x58,0xe8,0x23,0x65,0xaf,0xea,0x25,0x6f,0xb1,0xc8,0x43,0xc5,0x54},
	{0xfc,0x1f,0x21,0x63,0xa5,0xf4,0x07,0x09,0x1b,0x2d,0x77,0x99,0xb0,0xcb,0x46,0xca},
	{0x45,0xcf,0x4a,0xde,0x79,0x8b,0x86,0x91,0xa8,0xe3,0x3e,0x42,0xc6,0x51,0xf3,0x0e},
	{0x12,0x36,0x5a,0xee,0x29,0x7b,0x8d,0x8c,0x8f,0x8a,0x85,0x94,0xa7,0xf2,0x0d,0x17},
	{0x39,0x4b,0xdd,0x7c,0x84,0x97,0xa2,0xfd,0x1c,0x24,0x6c,0xb4,0xc7,0x52,0xf6,0x01},};

	long invsdatasource[16][16] = {{0x52,0x09,0x6a,0xd5,0x30,0x36,0xa5,0x38,0xbf,0x40,0xa3,0x9e,0x81,0xf3,0xd7,0xfb},
	{0x7c,0xe3,0x39,0x82,0x9b,0x2f,0xff,0x87,0x34,0x8e,0x43,0x44,0xc4,0xde,0xe9,0xcb},
	{0x54,0x7b,0x94,0x32,0xa6,0xc2,0x23,0x3d,0xee,0x4c,0x95,0x0b,0x42,0xfa,0xc3,0x4e},
	{0x08,0x2e,0xa1,0x66,0x28,0xd9,0x24,0xb2,0x76,0x5b,0xa2,0x49,0x6d,0x8b,0xd1,0x25},
	{0x72,0xf8,0xf6,0x64,0x86,0x68,0x98,0x16,0xd4,0xa4,0x5c,0xcc,0x5d,0x65,0xb6,0x92},
	{0x6c,0x70,0x48,0x50,0xfd,0xed,0xb9,0xda,0x5e,0x15,0x46,0x57,0xa7,0x8d,0x9d,0x84},
	{0x90,0xd8,0xab,0x00,0x8c,0xbc,0xd3,0x0a,0xf7,0xe4,0x58,0x05,0xb8,0xb3,0x45,0x06},
	{0xd0,0x2c,0x1e,0x8f,0xca,0x3f,0x0f,0x02,0xc1,0xaf,0xbd,0x03,0x01,0x13,0x8a,0x6b},
	{0x3a,0x91,0x11,0x41,0x4f,0x67,0xdc,0xea,0x97,0xf2,0xcf,0xce,0xf0,0xb4,0xe6,0x73},
	{0x96,0xac,0x74,0x22,0xe7,0xad,0x35,0x85,0xe2,0xf9,0x37,0xe8,0x1c,0x75,0xdf,0x6e},
	{0x47,0xf1,0x1a,0x71,0x1d,0x29,0xc5,0x89,0x6f,0xb7,0x62,0x0e,0xaa,0x18,0xbe,0x1b},
	{0xfc,0x56,0x3e,0x4b,0xc6,0xd2,0x79,0x20,0x9a,0xdb,0xc0,0xfe,0x78,0xcd,0x5a,0xf4},
	{0x1f,0xdd,0xa8,0x33,0x88,0x07,0xc7,0x31,0xb1,0x12,0x10,0x59,0x27,0x80,0xec,0x5f},
	{0x60,0x51,0x7f,0xa9,0x19,0xb5,0x4a,0x0d,0x2d,0xe5,0x7a,0x9f,0x93,0xc9,0x9c,0xef},
	{0xa0,0xe0,0x3b,0x4d,0xac,0x2a,0xf5,0xb0,0xc8,0xeb,0xbb,0x3c,0x83,0x53,0x99,0x61},
	{0x17,0x2b,0x04,0x7e,0xba,0x77,0xd6,0x26,0xe1,0x69,0x14,0x63,0x55,0x21,0x0c,0x7d},};

			

			
	for(int i = 0; i < 16; i++)
		for(int j = 0; j < 16; j++)
		{
			InvLogTable[i][j] = invlogtableSource[i][j];
			logTable[i][j] = logtableSource[i][j];
			InvSDataSource[i][j] = invsdatasource[i][j];
			SDataSource[i][j] = datasource[i][j];
		}
			
	long input[4] = {0x3243f6a8,0x885a308d,0x313198a2,0xe0370734};
	for(int i = 0;i < 4; i++)
	{
		m_InputW[i] = input[i];
		m_CodeW[i] = 0;
		m_ClearW[i] = 0;
	}
			
	for(int i = 0;i < 4; i++)
		for(int j = 0; j < 4; j++)
		{
			m_Current[i][j] = 0;
			m_InputB[i][j] = 0;
			m_CodeB[i][j] = 0;
		}
	m_first = true;
	m_last = EncryptionRounds+1;
		
}

AES::~AES(void)
{

}

void AES::KeyExpansion(Bit key[4*Nk], Word w[(EncryptionRounds+1)*Nb])//扩展密钥
{
	int index = 0;
	while(index < Nk)
	{
		w[index] = key[4*index].to_ulong()*pow(2.0,24) + key[4*index+1].to_ulong()*pow(2.0,16) + key[4*index+2].to_ulong()*pow(2.0,8) + key[4*index+3].to_ulong();
		index++;
	}	
//	cout<<"First four w:"<<endl;测试
//	for(int i = 0; i < Nk; i++)
//		cout<<hex<<w[i].to_ulong()<<endl;测试
	index = Nk;
	while(index < Nb*(EncryptionRounds+1))
	{
		
//		cout<<dec<<"\norginal w["<<index<<"]:"<<hex<<w[index-1].to_ulong()<<endl;测试
		Word temp = w[index-1];
		if(index % Nk == 0)
		{
			temp = xor(SubWord(RotWord(temp)),m_Rcon[index/Nk-1]);
//			cout<<dec<<"Rcon["<<index/Nk<<"]:"<<hex<<m_Rcon[index/Nk].to_ulong()<<endl;测试
		}
		else if(Nk == 8 && index % Nk == 4)
			;
		w[index] = xor(temp,w[index-Nk]);
//		cout<<dec<<"w["<<index-Nk<<"]:"<<hex<<w[index-Nk].to_ulong()<<endl;测试
//		cout<<dec<<"w["<<index<<"]:"<<hex<<w[index].to_ulong()<<endl;测试
		index++;
	}
}

void AES::Encrypt(void)//加密算法
{
	int index = 0;
	KeyExpansion(m_Key,m_KeyW);
	for(int i = 0; i < (EncryptionRounds+1)*Nb/4; i++)
		for(int j = 0; j < 4; j++) 
			WordToBit(m_KeyW[index++],m_KeyB[i][j]);

	index = 0;
	m_first = true;
	for(int num = 0; num < EncryptionRounds+1; num++)
	{
		cout<<dec<<"加密第"<<index<<"轮的开始:"<<endl;
		if(m_first)
		{
			cout<<"输入:"<<endl;
			for(int i = 0; i < 4; i++)
				WordToBit(m_InputW[i],m_InputB[i]);

			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_InputB[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			cout<<"本轮密钥:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_KeyB[index][j][i].to_ulong()<<"\t";
				cout<<endl;
			}
	
			xor(m_InputB,m_KeyB[index++]);

			cout<<"AddRoundKey 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			m_first = false;
		}	
		else
		{

			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			for(int i = 0; i < 4; i++)
				for(int j = 0; j < 4; j++)
					m_Current[i][j] = SubBypes(m_Current[i][j]);

			cout<<"SubBytes 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			ShiftRows(m_Current);

			cout<<"ShiftRows 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			if(num == m_last-1)
				goto jumpto;
			MixColumns(m_Current);

			cout<<"MixColumn 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

jumpto:

			cout<<"本轮密钥:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_KeyB[index][j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			xor(m_Current,m_KeyB[index++]);

			cout<<"AddRoundKey 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			if(num == m_last-1)
			{
				for(int i = 0; i < 4; i++)
					BitToWord(m_Current[i],m_CodeW[i]);
				cout<<"输出得到密文:"<<endl;
				for(int i = 0; i < 4; i++)
					cout<<hex<<m_CodeW[i].to_ulong()<<endl;	
				cout<<"按任意键继续..."<<endl;
				getchar();
				return;
			}
			
		}
		cout<<"本轮结束\n"<<endl;
		cout<<"按任意键继续..."<<endl;
		getchar();
	}
}
void AES::InvEncrypt(void)//解密算法
{

	int index = 0;
	KeyExpansion(m_Key,m_KeyW);
	for(int i = 0; i < (EncryptionRounds+1)*Nb/4; i++)
		for(int j = 0; j < 4; j++) 
			WordToBit(m_KeyW[index++],m_KeyB[i][j]);

	index = 0;
	m_first = true;
	for(int num = 0; num < EncryptionRounds+1; num++)
	{
		cout<<dec<<"解密第"<<index<<"轮的开始:"<<endl;
		if(m_first)
		{
			cout<<"输入:"<<endl;
			for(int i = 0; i < 4; i++)
				WordToBit(m_CodeW[i],m_CodeB[i]);
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_CodeB[j][i].to_ulong()<<"\t";
				cout<<endl;
			}
			cout<<"本轮密钥:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_KeyB[EncryptionRounds-index][j][i].to_ulong()<<"\t";
				cout<<endl;
			}
	
			xor(m_CodeB,m_KeyB[EncryptionRounds-index]);
			cout<<"AddRoundKey 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}
			m_first = false;
			index++;
		}	
		else
		{

			for(int i = 0; i < 4; i++)
			{
				for(int j = 0; j < 4;j++)
					cout<<dec<<"bit["<<i<<"]:"<<hex<<m_Current[j][i].to_ulong()<<"\t";
				cout<<endl;
			}

			InvShiftRows(m_Current);
			cout<<"InvShiftRows 处理后:"<<endl;
			for(int i = 0; i < 4; i++)
			{