密码学课后作业实现AES/ECB算法的实现,所以这几天都有在查一些AES资料,在此整理下并附上C++代码(:
这里引用一下维基上关于AES的基本介绍:
AES加密过程是在一个4×4的字节矩阵上运作,这个矩阵又称为“体(state)”,其初值就是一个明文区块(矩阵中一个元素大小就是明文区块中的一个Byte)。(Rijndael加密法因支持更大的区块,其矩阵行数可视情况增加)加密时,各轮AES加密循环(除最后一轮外)均包含4个步骤:
AddRoundKey—矩阵中的每一个字节都与该次回合密钥(round key)做XOR运算;每个子密钥由密钥生成方案产生。
SubBytes—通过一个非线性的替换函数,用查找表的方式把每个字节替换成对应的字节。
ShiftRows—将矩阵中的每个横列进行循环式移位。
MixColumns—为了充分混合矩阵中各个直行的操作。这个步骤使用线性转换来混合每内联的四个字节。最后一个加密循环中省略MixColumns步骤,而以另一个AddRoundKey取代。
(找资料的时候发现这个视频讲加密还是很清楚的,附上链接)AddRoundKey步骤addroundkey步骤实现的是回合密钥(round key)与当前状态(state)的异或操作。此步骤加密解密阶段都是相同的,无需更改。具体代码如下:

/*
* 异或处理
*/
void Aes::addRoundKey(unsigned char state[4][4], unsigned char roundKey[4][4])// already test
{
for(int col=0 ; col<4 ; col++){
for(int row=0 ; row<4 ; row++){
state[row][col] ^= roundKey[row][col];
}
}
}View Code
SubBytes步骤
subbytes步骤实现的是根据S盒以及逆S盒进行替换操作。加密时用S盒,解密时用逆S盒。具体代码如下:

/*
* 替换 already test
*/
void Aes::subBytes(unsigned char state[4][4])
{
int x, y;
int value;
for(int i=0 ; i<4 ;i++){//列
for(int j=0 ; j<4 ;j++){//行
value = state[j][i];
x = value / 16 ;
y = value % 16 ;
state[j][i] = s_box[x*16+y];
}
}
}
/*
* 逆替换 already test
*/
void Aes::invSubBytes(unsigned char state[4][4])
{
int col,row;
int x,y;
int value;
for(col=0 ; col<4 ; col++){
for(row=0 ; row<4 ; row++){
value = state[row][col];
x = value / 16 ;
y = value % 16 ;
state[row][col] = Inv_S_Box[x][y];
}
}
}View Code
ShiftRows步骤
shiftrows步骤实现的是行移位操作,第一行不动,第二行循环左移一位,第三行循环左移二位,第四行循环左移三位(这里的位具体指的是一数组元素),解密操作则是把左移改为右移即可。

/*
* 移动行 already test
*/
void Aes::shiftRows(unsigned char state[4][4])
{
unsigned char temp;
int times = 0;
for(int i = 1; i < 4 ; i++)//行
{
times = i;
while(times--)
{
temp = state[i][0];//每一行的第一个元素 copy一份
for(int j = 1 ; j < 4 ; j++)//列
{
state[i][j-1] = state[i][j]; //左移一位
}
state[i][3] = temp;//填充最后一位
}
}
}
/*
* 逆移动行 already test
*/
void Aes::invShiftRows(unsigned char state[4][4])
{
unsigned char temp;
int times = 0;
int i,j;
for(i = 1; i < 4 ; i++)//行
{
times = i;
while(times--)
{
temp = state[i][3];//每一行的第一个元素 copy一份
for(j=2 ; j>=0 ; j--)//列
{
state[i][j+1] = state[i][j]; //左移一位
}
state[i][0] = temp;//填充最后一位
}
}
}View Code
MixColumns步骤
重点来了,AES与DES的最大区别,AES除了替换,位移,还有就是数学上操作(GF(28)),这里我就不详细说明了(好吧,其实是水平有限,大家可以看一下这个关于mixcolumns操作的介绍)

/*
* 列处理 already test
*/
void Aes::mixColumns(unsigned char state[4][4])
{
unsigned char copyState[4];
unsigned char state2[4]; //2倍state
unsigned char h;
for(int col=0 ; col<4 ; col++)//列
{
//每一列操作
for(int row=0 ; row<4 ; row++)//行
{
copyState[row] = state[row][col];//copy one col state
h = (unsigned char)((signed char)state[row][col] >> 7);//
state2[row] = state[row][col] << 1;
state2[row] ^= 0x1b & h;
}
state[0][col] = state2[0] ^ copyState[3] ^ copyState[2] ^ state2[1] ^ copyState[1];/* 2 * a0 + a3 + a2 + 3 * a1 */
state[1][col] = state2[1] ^ copyState[0] ^ copyState[3] ^ state2[2] ^ copyState[2];/* 2 * a1 + a0 + a3 + 3 * a2 */
state[2][col] = state2[2] ^ copyState[1] ^ copyState[0] ^ state2[3] ^ copyState[3];/* 2 * a2 + a1 + a0 + 3 * a3 */
state[3][col] = state2[3] ^ copyState[2] ^ copyState[1] ^ state2[0] ^ copyState[0];/* 2 * a3 + a2 + a1 + 3 * a0 */
}
}
/*
* 逆列处理 ?????
*/
void Aes::invMixColumns(unsigned char state[4][4])
{
unsigned char state1[4];//2倍
unsigned char state2[4];//4倍
unsigned char state3[4];//8倍
unsigned char copyState[4];//1倍
unsigned char h;
int col,row;
for(col=0; col<4 ; col++){
for(row=0 ; row<4 ; row++){
copyState[row] = state[row][col];
h = (unsigned char)((signed char)state[row][col] >> 7);//
state1[row] = state[row][col] << 1;
state1[row] ^= 0x1b & h;
h = (unsigned char)((signed char)state1[row] >> 7);//
state2[row] = state1[row] << 1;
state2[row] ^= 0x1b & h;
h = (unsigned char)((signed char)state2[row] >> 7);//
state3[row] = state2[row] << 1;
state3[row] ^= 0x1b & h;
}
state[0][col] = (state3[0]^state2[0]^state1[0]) ^ (state3[3]^copyState[3]) ^ (state3[2]^state2[2]^copyState[2]) ^ (state3[1]^state1[1]^copyState[1]);/* 14 * a0 + 9 * a3 + 13 * a2 + 11 * a1 */
state[1][col] = (state3[1]^state2[1]^state1[1]) ^ (state3[0]^copyState[0]) ^ (state3[3]^state2[3]^copyState[3]) ^ (state3[2]^state1[2]^copyState[2]);/* 14 * a1 + 9 * a0 + 13 * a3 + 11 * a2 */
state[2][col] = (state3[2]^state2[2]^state1[2]) ^ (state3[1]^copyState[1]) ^ (state3[0]^state2[0]^copyState[0]) ^ (state3[3]^state1[3]^copyState[3]);/* 14 * a2 + 9 * a1 + 13 * a0 + 11 * a3 */
state[3][col] = (state3[3]^state2[3]^state1[3]) ^ (state3[2]^copyState[2]) ^ (state3[1]^state2[1]^copyState[1]) ^ (state3[0]^state1[0]^copyState[0]);/* 14 * a3 + 9 * a2 + 13 * a1 + 11 * a0 */
}
}View Code
至此最主要的4个操作已经解决,接下来就可以编写加密解密函数了,加密过程在一开始给的视频链接中有动态过程,这里就不啰嗦了。
附上完整代码:
main.cpp


#include <iostream>
#include "Aes.h"
#include <string>
#include <stdlib.h>
#include <fstream>
#define FILEOPENERROR 1
using namespace std;
int main()
{
int choose;
bool glap = true;
bool sign = true;
Aes Encryption;
Aes Decryption;
while(glap)
{
choose = 0;
cout<<"choose one option.\n";
cout<<"1. Aes Encryption\n"
<<"2. Aes Decryption\n"
<<"3. Quit\n";
cin>>choose;
switch (choose)
{
case 1:
Encryption.inputKey();
Encryption.setFilePath();
Encryption.initAes();
sign = true;
do{
if(Encryption.encode() == FILEOPENERROR){
cout<<"file open fail, please input file path again!\n";
Encryption.setFilePath();
}else{//加密成功
cout<<"Encode success!\n";
sign = false;
}
}while(sign);
cout<<endl<<endl;
break;
case 2:
Decryption.inputKey();
Decryption.setFilePath();
Decryption.initAes();
sign = true;
do{
if(Decryption.decode() == FILEOPENERROR){
cout<<"file open fail, please input file path again!\n";
Decryption.setFilePath();
}else{//解密成功
cout<<"Decode success!\n";
sign = false;
}
}while(sign);
cout<<endl<<endl;
break;
case 3:
glap = false;
system("cls");
break;
default:
cout<<"option choose fail, choose again.\n";
cin>>choose;
}
}
return 0;
}View Code
Aes.h


#ifndef AES_H
#define AES_H
#include <string.h>
#include <fstream>
using namespace std;
class Aes
{
public:
Aes();
virtual ~Aes();
void subBytes(unsigned char state[4][4]);
void shiftRows(unsigned char state[4][4]);
void mixColumns(unsigned char state[4][4]);
void invSubBytes(unsigned char state[4][4]);
void invShiftRows(unsigned char state[4][4]);
void invMixColumns(unsigned char state[4][4]);
void addRoundKey(unsigned char state[4][4], unsigned char roundKey[4][4]);
void keySchedule(unsigned char roundKey[4][44]);
void inputKey();
int encode();
int decode();
void setFilePath();
void setRoundKey(unsigned char* key)
{
for(int col=0 ; col<4 ; col++)
for(int row=0 ; row<4 ; row++)
roundKey_temp[row][col] = key[row+col*4];
}
void initAes()
{
setRoundKey(initKey);
keySchedule(roundKey_temp);
for(int col=0 ; col<44 ; col++)
for(int row=0 ; row<4 ; row++)
roundKey[col/4][row][col%4] = roundKey_temp[row][col];
}
private:
char *filePath;
unsigned char roundKey_temp[4][44];
unsigned char roundKey[11][4][4];
unsigned char initKey[16];
};
#endif // AES_HView Code
Aes.cpp


#include "Aes.h"
#include <iostream>
#include <fstream>
#include <string>
#include <string.h>
#define FILEOPENERROR 1
unsigned char s_box[256] = {
/* 0 1 2 3 4 5 6 7 8 9 a b c d e f */
0x63,0x7c,0x77,0x7b,0xf2,0x6b,0x6f,0xc5,0x30,0x01,0x67,0x2b,0xfe,0xd7,0xab,0x76, /*0*/
0xca,0x82,0xc9,0x7d,0xfa,0x59,0x47,0xf0,0xad,0xd4,0xa2,0xaf,0x9c,0xa4,0x72,0xc0, /*1*/
0xb7,0xfd,0x93,0x26,0x36,0x3f,0xf7,0xcc,0x34,0xa5,0xe5,0xf1,0x71,0xd8,0x31,0x15, /*2*/
0x04,0xc7,0x23,0xc3,0x18,0x96,0x05,0x9a,0x07,0x12,0x80,0xe2,0xeb,0x27,0xb2,0x75, /*3*/
0x09,0x83,0x2c,0x1a,0x1b,0x6e,0x5a,0xa0,0x52,0x3b,0xd6,0xb3,0x29,0xe3,0x2f,0x84, /*4*/
0x53,0xd1,0x00,0xed,0x20,0xfc,0xb1,0x5b,0x6a,0xcb,0xbe,0x39,0x4a,0x4c,0x58,0xcf, /*5*/
0xd0,0xef,0xaa,0xfb,0x43,0x4d,0x33,0x85,0x45,0xf9,0x02,0x7f,0x50,0x3c,0x9f,0xa8, /*6*/
0x51,0xa3,0x40,0x8f,0x92,0x9d,0x38,0xf5,0xbc,0xb6,0xda,0x21,0x10,0xff,0xf3,0xd2, /*7*/
0xcd,0x0c,0x13,0xec,0x5f,0x97,0x44,0x17,0xc4,0xa7,0x7e,0x3d,0x64,0x5d,0x19,0x73, /*8*/
0x60,0x81,0x4f,0xdc,0x22,0x2a,0x90,0x88,0x46,0xee,0xb8,0x14,0xde,0x5e,0x0b,0xdb, /*9*/
0xe0,0x32,0x3a,0x0a,0x49,0x06,0x24,0x5c,0xc2,0xd3,0xac,0x62,0x91,0x95,0xe4,0x79, /*a*/
0xe7,0xc8,0x37,0x6d,0x8d,0xd5,0x4e,0xa9,0x6c,0x56,0xf4,0xea,0x65,0x7a,0xae,0x08, /*b*/
0xba,0x78,0x25,0x2e,0x1c,0xa6,0xb4,0xc6,0xe8,0xdd,0x74,0x1f,0x4b,0xbd,0x8b,0x8a, /*c*/
0x70,0x3e,0xb5,0x66,0x48,0x03,0xf6,0x0e,0x61,0x35,0x57,0xb9,0x86,0xc1,0x1d,0x9e, /*d*/
0xe1,0xf8,0x98,0x11,0x69,0xd9,0x8e,0x94,0x9b,0x1e,0x87,0xe9,0xce,0x55,0x28,0xdf, /*e*/
0x8c,0xa1,0x89,0x0d,0xbf,0xe6,0x42,0x68,0x41,0x99,0x2d,0x0f,0xb0,0x54,0xbb,0x16 /*f*/
};
unsigned char Rcon[4][10] = {
{0x01,0x02,0x04,0x08,0x10,0x20,0x40,0x80,0x1b,0x36},
{0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
{0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
{0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}
};
unsigned char Inv_S_Box[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, 0xAE, 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}
};
using namespace std;
Aes::Aes()
{
//ctor
}
Aes::~Aes()
{
//dtor
}
/*
* 替换 already test
*/
void Aes::subBytes(unsigned char state[4][4])
{
int x, y;
int value;
for(int i=0 ; i<4 ;i++){//列
for(int j=0 ; j<4 ;j++){//行
value = state[j][i];
x = value / 16 ;
y = value % 16 ;
state[j][i] = s_box[x*16+y];
}
}
}
/*
* 逆替换 already test
*/
void Aes::invSubBytes(unsigned char state[4][4])
{
int col,row;
int x,y;
int value;
for(col=0 ; col<4 ; col++){
for(row=0 ; row<4 ; row++){
value = state[row][col];
x = value / 16 ;
y = value % 16 ;
state[row][col] = Inv_S_Box[x][y];
}
}
}
/*
* 移动行 already test
*/
void Aes::shiftRows(unsigned char state[4][4])
{
unsigned char temp;
int times = 0;
for(int i = 1; i < 4 ; i++)//行
{
times = i;
while(times--)
{
temp = state[i][0];//每一行的第一个元素 copy一份
for(int j = 1 ; j < 4 ; j++)//列
{
state[i][j-1] = state[i][j]; //左移一位
}
state[i][3] = temp;//填充最后一位
}
}
}
/*
* 逆移动行 already test
*/
void Aes::invShiftRows(unsigned char state[4][4])
{
unsigned char temp;
int times = 0;
int i,j;
for(i = 1; i < 4 ; i++)//行
{
times = i;
while(times--)
{
temp = state[i][3];//每一行的第一个元素 copy一份
for(j=2 ; j>=0 ; j--)//列
{
state[i][j+1] = state[i][j]; //左移一位
}
state[i][0] = temp;//填充最后一位
}
}
}
/*
* 列处理 already test
*/
void Aes::mixColumns(unsigned char state[4][4])
{
unsigned char copyState[4];
unsigned char state2[4]; //2倍state
unsigned char h;
for(int col=0 ; col<4 ; col++)//列
{
//每一列操作
for(int row=0 ; row<4 ; row++)//行
{
copyState[row] = state[row][col];//copy one col state
h = (unsigned char)((signed char)state[row][col] >> 7);//
state2[row] = state[row][col] << 1;
state2[row] ^= 0x1b & h;
}
state[0][col] = state2[0] ^ copyState[3] ^ copyState[2] ^ state2[1] ^ copyState[1];/* 2 * a0 + a3 + a2 + 3 * a1 */
state[1][col] = state2[1] ^ copyState[0] ^ copyState[3] ^ state2[2] ^ copyState[2];/* 2 * a1 + a0 + a3 + 3 * a2 */
state[2][col] = state2[2] ^ copyState[1] ^ copyState[0] ^ state2[3] ^ copyState[3];/* 2 * a2 + a1 + a0 + 3 * a3 */
state[3][col] = state2[3] ^ copyState[2] ^ copyState[1] ^ state2[0] ^ copyState[0];/* 2 * a3 + a2 + a1 + 3 * a0 */
}
}
/*
* 逆列处理 ?????
*/
void Aes::invMixColumns(unsigned char state[4][4])
{
unsigned char state1[4];//2倍
unsigned char state2[4];//4倍
unsigned char state3[4];//8倍
unsigned char copyState[4];//1倍
unsigned char h;
int col,row;
for(col=0; col<4 ; col++){
for(row=0 ; row<4 ; row++){
copyState[row] = state[row][col];
h = (unsigned char)((signed char)state[row][col] >> 7);//
state1[row] = state[row][col] << 1;
state1[row] ^= 0x1b & h;
h = (unsigned char)((signed char)state1[row] >> 7);//
state2[row] = state1[row] << 1;
state2[row] ^= 0x1b & h;
h = (unsigned char)((signed char)state2[row] >> 7);//
state3[row] = state2[row] << 1;
state3[row] ^= 0x1b & h;
}
state[0][col] = (state3[0]^state2[0]^state1[0]) ^ (state3[3]^copyState[3]) ^ (state3[2]^state2[2]^copyState[2]) ^ (state3[1]^state1[1]^copyState[1]);/* 14 * a0 + 9 * a3 + 13 * a2 + 11 * a1 */
state[1][col] = (state3[1]^state2[1]^state1[1]) ^ (state3[0]^copyState[0]) ^ (state3[3]^state2[3]^copyState[3]) ^ (state3[2]^state1[2]^copyState[2]);/* 14 * a1 + 9 * a0 + 13 * a3 + 11 * a2 */
state[2][col] = (state3[2]^state2[2]^state1[2]) ^ (state3[1]^copyState[1]) ^ (state3[0]^state2[0]^copyState[0]) ^ (state3[3]^state1[3]^copyState[3]);/* 14 * a2 + 9 * a1 + 13 * a0 + 11 * a3 */
state[3][col] = (state3[3]^state2[3]^state1[3]) ^ (state3[2]^copyState[2]) ^ (state3[1]^state2[1]^copyState[1]) ^ (state3[0]^state1[0]^copyState[0]);/* 14 * a3 + 9 * a2 + 13 * a1 + 11 * a0 */
}
}
/*
* 异或处理 already test
*/
void Aes::addRoundKey(unsigned char state[4][4], unsigned char roundKey[4][4])// not test
{
for(int col=0 ; col<4 ; col++){
for(int row=0 ; row<4 ; row++){
state[row][col] ^= roundKey[row][col];
}
}
}
/*
* 轮密钥生成 already test
*/
void Aes::keySchedule(unsigned char roundKey[4][44])
{
int colAfter=3;
unsigned char subCol[4];
unsigned char downCol[4];
int x, y;
int value;
int t = 0;//轮数 为了使用Rcon
for(int colBefore=0; colBefore<40 ; colBefore++){//当前轮的列
if((colAfter+1)%4 == 0){
//down one bite
downCol[3] = roundKey[0][colAfter];
for(int row=1 ; row<4 ;row++){
downCol[row-1] = roundKey[row][colAfter];
}
//sub bytes
for(int i=0 ; i<4 ;i++){//
value = downCol[i];
x = value / 16 ;
y = value % 16 ;
subCol[i] = s_box[x*16+y];
}
//Xoring
for(int j=0 ; j<4 ; j++){
roundKey[j][colAfter+1] = roundKey[j][colBefore] ^ subCol[j] ^ Rcon[j][t];
}
t++;
}else {
//Xoring
for(int j=0 ; j<4 ; j++){
roundKey[j][colAfter+1] = roundKey[j][colBefore] ^ roundKey[j][colAfter];
}
}
colAfter++;
}//end for
}
/*
* 加密 right
*/
int Aes::encode()
{
//打开文件
ifstream fileRd(filePath, ios::binary);
ofstream fileWt("c:\\encode.txt", ios::binary);
if(fileRd.fail() || fileWt.fail()){//打开失败
fileRd.close();
fileWt.close();
return FILEOPENERROR;
}
cout<<"your CIPHERTEXT file will be created at C:\\encode.txt!\n";
unsigned char buf[16];//取文件内容
unsigned char state[4][4];//运算状态
unsigned char enCdText[16];//加密后的数据
int t = 0; //计数器
int i,j;
int col,row;
bool glap = false;
while(!fileRd.eof())
{
glap = false;
memset(buf,0x00,16*sizeof(char));//清空buf
fileRd.read((char *)buf, sizeof(buf));
//cout<<buf<<endl;
for(i=0 ; i<16 ; i++)
if(buf[i] != 0x00){
glap = true;
break;
}
if(!glap)break;
//buf转化为state
for(col=0 ; col<4 ; col++)
for(row=0 ; row<4 ; row++)
state[row][col] = buf[row+col*4];
//addRoundkey 第一轮
addRoundKey(state, roundKey[0]);
//9轮
for(i=1 ; i<=9 ;i++){
subBytes(state);
shiftRows(state);
mixColumns(state);
addRoundKey(state, roundKey[i]);
}
//最后一轮
subBytes(state);
shiftRows(state);
addRoundKey(state, roundKey[10]);
//state 转化为 密文串
t = 0;
for(i=0;i<4;i++){
for(j=0;j<4;j++){
enCdText[t++] = state[j][i];
}
}
for(i=0 ; i<16 ; i++)
fileWt.put(enCdText[i]);
}
fileRd.close();
fileWt.close();
return 0;
}
/*
* 解密
*/
int Aes::decode()
{
//打开文件
ifstream fileRd("c:\\encode.txt", ios::binary);
ofstream fileWt("c:\\decode.txt", ios::binary);
if(fileRd.fail() || fileWt.fail()){//打开失败
fileRd.close();
fileWt.close();
return FILEOPENERROR;
}
cout<<"Your PLAINTEXT file will be created at C:\\decode.txt!\n";
unsigned char buf[16];//取文件内容
unsigned char state[4][4];//运算状态
unsigned char deCdText[16];//加密后的数据
int t = 0; //计数器
int i,j;
int col,row;
bool glap = false;
while(!fileRd.eof())
{
glap = false;
memset(buf,0x00,16*sizeof(char));//清空buf
fileRd.read((char *)buf, sizeof(buf));
//cout<<buf<<endl;
for(i=0 ; i<16 ; i++)
if(buf[i] != 0x00){
glap = true;
break;
}
if(!glap)break;
//buf转化为state
for(col=0 ; col<4 ; col++)
for(row=0 ; row<4 ; row++)
state[row][col] = buf[row+col*4];
//addRoundkey 第一轮
addRoundKey(state, roundKey[10]);
//9轮
for(i=9 ; i>=1 ;i--){
invShiftRows(state);
invSubBytes(state);
addRoundKey(state, roundKey[i]);
invMixColumns(state);
}
//最后一轮
invShiftRows(state);
invSubBytes(state);
addRoundKey(state, roundKey[0]);
//state 转化为 密文串
t = 0;
for(i=0;i<4;i++){
for(j=0;j<4;j++){
deCdText[t++] = state[j][i];
}
}
for(i=0 ; i<16 ; i++)
fileWt.put(deCdText[i]);
}
fileRd.close();
fileWt.close();
return 0;
}
void Aes::inputKey()
{
string temp_key;
bool glap = true;
int i;
do{
cout<<"please input your key(16 characters) : ";
cin>>temp_key;
if(temp_key.length() == 16){
glap = false;
cout<<"key init success!\n";
}else {
glap = true;
cout<<"key init fail, input again!\n";
}
}while(glap);
for(i=0 ; i<16 ; i++){
initKey[i] = temp_key[i];
}
}
void Aes::setFilePath()
{
char path[30];
cout<<"input your PLAINTEXT file path : ";
cin>>path;
int size = strlen(path);
filePath = new char(size);
strcpy(filePath, path);
}View Code
在写完程序后,需要对程序进行测试,可以在google上搜AES TEST VECTOR,这里我列出几个测试向量:
KEY = 8d2e60365f17c7df1040d7501b4a7b5a PLAINTEXT = 59b5088e6dadc3ad5f27a460872d5929 CIPHERTEXT = a02600ecb8ea77625bba6641ed5f5920 KEY = 2d0860dae7fdb0bd4bfab111f615227a PLAINTEXT = a02600ecb8ea77625bba6641ed5f5920 CIPHERTEXT = 5241ead9a89ca31a7147f53a5bf6d96a KEY = 7f498a034f6113a73abd442bade3fb10 PLAINTEXT = 5241ead9a89ca31a7147f53a5bf6d96a CIPHERTEXT = 22f09171bc67d0661d1c25f181a69f33
















