使用AES加密算法

上几天大概了解了一下AES算法，具体的加密算法现在不想很清楚的了解，只是能够拿来用就好了。这里有份c++的AES。

Aes.h

 

#include <Windows.h>
//enum KeySize { Bits128, Bits192, Bits256 };  // key size, in bits, for construtor
#define Bits128    16
#define Bits192    24
#define Bits256    32

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

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

class Aes  // Advanced Encryption Standard
{
public:
~Aes();
Aes();
Aes(int keySize, unsigned char* keyBytes);
unsigned char State[4][4];
void Cipher(unsigned char* input, unsigned char* output);  // encipher 16-bit input
void InvCipher(unsigned char* input, unsigned char* output);  // decipher 16-bit input
private:
int Nb;         // block size in 32-bit words.  Always 4 for AES.  (128 bits).
int Nk;         // key size in 32-bit words.  4, 6, 8.  (128, 192, 256 bits).
int Nr;         // number of rounds. 10, 12, 14.

unsigned char key[32];
unsigned char w[16*15];

void SetNbNkNr(int keySize);
void AddRoundKey(int round);      //轮密钥加
void SubBytes();                  //S盒字节代换
void InvSubBytes();               //逆S盒字节代换
void ShiftRows();                 //行移位
void InvShiftRows();
void MixColumns();                //列混淆
void InvMixColumns();
unsigned char gfmultby01(unsigned char b);
unsigned char gfmultby02(unsigned char b);
unsigned char gfmultby03(unsigned char b);
unsigned char gfmultby09(unsigned char b);
unsigned char gfmultby0b(unsigned char b);
unsigned char gfmultby0d(unsigned char b);
unsigned char gfmultby0e(unsigned char b);
void KeyExpansion();              //密钥扩展
unsigned char* SubWord(unsigned char* word);         //密钥S盒字代换
unsigned char* RotWord(unsigned char* word);         //密钥移位
//Dump();
//DumpKey();
//    DumpTwoByTwo(char* a);
};

然后是Aes.cpp文件

//#include "StdAfx.h"    //注意在此 #include "Aes.h"不可以放在前面，否则出错，
#include "Aes.h"
Aes::~Aes()
{

}
Aes::Aes()
{

}

//构造函数
Aes::Aes(int keysize,unsigned char* keyBytes)
{
SetNbNkNr(keysize);                         //设置密钥块数，轮数
memcpy(key,keyBytes,keysize);                //字符串拷贝函数，把keyBytes的keysize个字符复制到key中
KeyExpansion();                                //密钥扩展，必须提前做的初始化
}

void Aes::SetNbNkNr(int keySize)
{
Nb=4;
if(keySize=Bits128)
{
Nk=4;    //4*4字节，128位密钥，10轮加密
Nr=10;
}
else if(keySize=Bits192)
{
Nk=6;    //6*4字节，192位密钥，12轮加密
Nr=12;
}
else if(keySize=Bits256)
{
Nk=8;    //8*4字节，256位密钥，14轮加密
Nr=14;
}
}

void Aes::KeyExpansion()
{
int row;
memset(w,0,16*15);
for(row=0;row<Nk;row++)       //拷贝seed 密钥
{
w[4*row+0] =  key[4*row];
w[4*row+1] =  key[4*row+1];
w[4*row+2] =  key[4*row+2];
w[4*row+3] =  key[4*row+3];
}
byte* temp = new byte[4];
for(row=Nk;row<4*(Nr+1);row++)
{
temp[0]=w[4*row-4];     //当前列的前一列
temp[1]=w[4*row-3];
temp[2]=w[4*row-2];
temp[3]=w[4*row-1];
if(row%Nk==0)           //逢nk时，对当前列的前一列作特殊处理
{
temp=SubWord(RotWord(temp));   //先移位，再代换，最后和轮常量异或
temp[0] = (byte)( (int)temp[0] ^ (int) AesRcon[4*(row/Nk)+0] );
temp[1] = (byte)( (int)temp[1] ^ (int) AesRcon[4*(row/Nk)+1] );
temp[2] = (byte)( (int)temp[2] ^ (int) AesRcon[4*(row/Nk)+2] );
temp[3] = (byte)( (int)temp[3] ^ (int) AesRcon[4*(row/Nk)+3] );
}
else if ( Nk > 6 && (row % Nk == 4) )  //这个还没有搞清楚
{
temp = SubWord(temp);
}

// w[row] = w[row-Nk] xor temp
w[4*row+0] = (byte) ( (int) w[4*(row-Nk)+0] ^ (int)temp[0] );
w[4*row+1] = (byte) ( (int) w[4*(row-Nk)+1] ^ (int)temp[1] );
w[4*row+2] = (byte) ( (int) w[4*(row-Nk)+2] ^ (int)temp[2] );
w[4*row+3] = (byte) ( (int) w[4*(row-Nk)+3] ^ (int)temp[3] );
}  // for loop

}

//密钥移位函数
unsigned char* Aes::RotWord(unsigned char* word)
{
byte* temp = new byte[4];
temp[0] = word[1];
temp[1] = word[2];
temp[2] = word[3];
temp[3] = word[0];
return temp;
}

//密钥字代换函数
unsigned char* Aes::SubWord(unsigned char* word)
{
byte* temp = new byte[4];
for(int j=0;j<4;j++)
{
temp[j] = AesSbox[16*(word[j] >> 4)+(word[j] & 0x0f)];  //实际上也可以写成AesSbox[[j]];因为两者相等
}
return temp;

}

//Aes加密函数
void Aes::Cipher(unsigned char* input, unsigned char* output)
{
int i;
memset(&State[0][0],0,16);
for(i=0;i<4*Nb;i++)                        //这里是先写列后写行的，即输入是一列一列的进来的
{
State[i%4][i/4]=input[i];                    //换成先写行后写列也是可以的，只要在输出时也是这样就可以了
}
AddRoundKey(0);                                    //轮密钥加

for (int round = 1; round <= (Nr - 1); round++)  // main round loop
{
SubBytes();                                    //字节代换
ShiftRows();                                //行移位
MixColumns();                                //列混淆
AddRoundKey(round);                            //轮密钥加
}  // main round loop

SubBytes();                                        //字节代换
ShiftRows();                                    //行移位
AddRoundKey(Nr);                                //轮密钥加

// output = state
for (i = 0; i < (4 * Nb); i++)
{
output[i] =  State[i % 4][ i / 4];
}

}

//Aes解密函数
void Aes::InvCipher(unsigned char* input,unsigned char* output)
{
int i;
memset(&State[0][0],0,16);
for (i = 0; i < (4 * Nb); i++)
{
State[i % 4][ i / 4] = input[i];
}

AddRoundKey(Nr);

for (int round = Nr-1; round >= 1; round--)  // main round loop
{
InvShiftRows();
InvSubBytes();
AddRoundKey(round);
InvMixColumns();
}  // end main round loop for InvCipher

InvShiftRows();
InvSubBytes();
AddRoundKey(0);

// output = state
for (i = 0; i < (4 * Nb); i++)
{
output[i] =  State[i % 4][ i / 4];
}
}

//轮密钥加
void Aes::AddRoundKey(int round)
{
int i,j;  //i行 j列           //因为密钥w是一列一列排列的，即 k0 k4 k8 k12
for(j=0;j<4;j++)              //                              k1 k5 k9 k13
{                              //                              k2 k6 k10k14
for(i=0;i<4;i++)          //                              k3 k7 k11k15
{                          // 所以i行j列的下标是4*((round*4)+j)+i即16*round+4*j+i
State[i][j]=(unsigned char)((int)State[i][j]^(int)w[4*((round*4)+j)+i]);
}
}
}

//字节代换函数
void Aes::SubBytes()                              //Page 103
{
int i,j;
for(j=0;j<4;j++)
{
for(i=0;i<4;i++)
{
State[i][j]=AesSbox[State[i][j]];
//因为 16*(State[i][j]>>4)+State[i][j]&0x0f=State[i][j]

}
}
}
void Aes::InvSubBytes()
{
int i,j;
for(j=0;j<4;j++)
{
for(i=0;i<4;i++)
{
State[i][j]=AesiSbox[State[i][j]]; //因为 16*(State[i][j]>>4)+State[i][j]&0x0f=State[i][j]
}
}

}

void Aes::ShiftRows()
{
unsigned char temp[4*4];                                        //Page105
int i,j;
for(j=0;j<4;j++)
{
for(i=0;i<4;i++)
{
temp[4*i+j]=State[i][j];
}
}
for(i=1;i<4;i++)
{
for(j=0;j<4;j++)
{
if(i==1)State[i][j]=temp[4*i+(j+1)%4];                    //第一行左移1位
else if(i==2)State[i][j]=temp[4*i+(j+2)%4];                //第二行左移2位
else if(i==3)State[i][j]=temp[4*i+(j+3)%4];                //第三行左移3位
}
}

}
void Aes::InvShiftRows()
{
unsigned char temp[4*4];
int i,j;
for(j=0;j<4;j++)
{
for(i=0;i<4;i++)
{
temp[4*i+j]=State[i][j];
}
}
for(i=1;i<4;i++)
{
for(j=0;j<4;j++)
{
//if(i==1)State[i][j]=temp[4*i+(j-1)%4];    在此犯了一个错误 -1%4=-1 而不是3，所以采用了下面再加一个4的做法
if(i==1)State[i][j]=temp[4*i+(j+3)%4];            //第一行右移1位 j-1+4=j+3
else if(i==2)State[i][j]=temp[4*i+(j+2)%4];        //第二行右移2位 j-2+4=j+2
else if(i==3)State[i][j]=temp[4*i+(j+1)%4];        //第三行右移3位 j-3+4=j+2
}
}

}

void Aes::MixColumns()
{
unsigned char temp[4*4];
int i,j;
for(j=0;j<4;j++)                                    //2 3 1 1  列混淆矩阵  Page107
{                                                    //1 2 3 1
for(i=0;i<4;i++)                                //1 1 2 3
{                                                //3 1 1 2
temp[4*i+j]=State[i][j];
}
}
for(j=0;j<4;j++)
{
State[0][j] = (unsigned char) ( (int)gfmultby02(temp[0+j]) ^ (int)gfmultby03(temp[4*1+j]) ^
(int)gfmultby01(temp[4*2+j]) ^ (int)gfmultby01(temp[4*3+j]) );
State[1][j] = (unsigned char) ( (int)gfmultby01(temp[0+j]) ^ (int)gfmultby02(temp[4*1+j]) ^
(int)gfmultby03(temp[4*2+j]) ^ (int)gfmultby01(temp[4*3+j]) );
State[2][j] = (unsigned char) ( (int)gfmultby01(temp[0+j]) ^ (int)gfmultby01(temp[4*1+j]) ^
(int)gfmultby02(temp[4*2+j]) ^ (int)gfmultby03(temp[4*3+j]) );
State[3][j] = (unsigned char) ( (int)gfmultby03(temp[0+j]) ^ (int)gfmultby01(temp[4*1+j]) ^
(int)gfmultby01(temp[4*2+j]) ^ (int)gfmultby02(temp[4*3+j]) );
}

}
void Aes::InvMixColumns()
{
unsigned char temp[4*4];
int i,j;
for (i = 0; i < 4; i++)  // copy State into temp[]
{
for (j = 0; j < 4; j++)                         //0e 0b 0d 09   逆变换矩阵 Page108
{                                                //09 0e 0b 0d
temp[4*i+j] =  State[i][j];                    //0d 09 0e 0b
}                                                //0b 0d 09 0e
}

for (j = 0; j < 4; j++)
{
State[0][j] = (unsigned char) ( (int)gfmultby0e(temp[j]) ^ (int)gfmultby0b(temp[4+j]) ^
(int)gfmultby0d(temp[4*2+j]) ^ (int)gfmultby09(temp[4*3+j]) );
State[1][j] = (unsigned char) ( (int)gfmultby09(temp[j]) ^ (int)gfmultby0e(temp[4+j]) ^
(int)gfmultby0b(temp[4*2+j]) ^ (int)gfmultby0d(temp[4*3+j]) );
State[2][j] = (unsigned char) ( (int)gfmultby0d(temp[j]) ^ (int)gfmultby09(temp[4+j]) ^
(int)gfmultby0e(temp[4*2+j]) ^ (int)gfmultby0b(temp[4*3+j]) );
State[3][j] = (unsigned char) ( (int)gfmultby0b(temp[j]) ^ (int)gfmultby0d(temp[4+j]) ^
(int)gfmultby09(temp[4*2+j]) ^ (int)gfmultby0e(temp[4*3+j]) );
}
}

unsigned char Aes::gfmultby01(unsigned char b)
{
return b;
}
unsigned char Aes::gfmultby02(unsigned char b)
{
if (b < 0x80)
return (unsigned char)(int)(b <<1);
else
return (unsigned char)( (int)(b << 1) ^ (int)(0x1b) );
}

unsigned char Aes::gfmultby03(unsigned char b)
{
return (unsigned char) ( (int)gfmultby02(b) ^ (int)b );
}

unsigned char Aes::gfmultby09(unsigned char b)
{
return (unsigned char)( (int)gfmultby02(gfmultby02(gfmultby02(b))) ^ (int)b );
}

unsigned char Aes::gfmultby0b(unsigned char b)
{
return (unsigned char)( (int)gfmultby02(gfmultby02(gfmultby02(b))) ^
(int)gfmultby02(b) ^ (int)b );
}

unsigned char Aes::gfmultby0d(unsigned char b)
{
return (unsigned char)( (int)gfmultby02(gfmultby02(gfmultby02(b))) ^
(int)gfmultby02(gfmultby02(b)) ^ (int)(b) );
}

unsigned char Aes::gfmultby0e(unsigned char b)
{
return (unsigned char)( (int)gfmultby02(gfmultby02(gfmultby02(b))) ^
(int)gfmultby02(gfmultby02(b)) ^(int)gfmultby02(b) );
}

这里想说下Aes加密需要注意的地方。我们上面代码的时候，首先要传入密钥。

Aes(int keySize, unsigned char* keyBytes);

密钥的长度可以是128位， 192位， 256位。对应分别是16个字节， 24个字节和32个字节。

然后在加密和解密的时候，我们传入两个指针。

void Cipher(unsigned char* input, unsigned char* output);  // encipher 16-bit input
void InvCipher(unsigned char* input, unsigned char* output);  // decipher 16-bit input

注意每次都是传16位的。也就是说，如果你的数据字节很大，你必须把数据拆开来，拆成16个字节16个字节的进行加密。

这里就产生一个问题：如果我们的数据是17个字节，那该怎么办呢？

这里我百度了一下采用以下这个方法：

 

如果要加密的字符串不够16位，输入字符串就要补位，比方说：
源输入：
abcdefg
补位后:
abcdefg999999999

源输入：
123456789
补位后：
1234567897777777

也就是说差几个字节补几个字节，补的字节就是ascii值从1 到 15。

下面自己写了个测试程序来测试AES加密。在D盘目录下建立一个1.txt文件。输入17个字节的数据。运行下面的程序：

#include "Aes.h"
#include <stdio.h>

//对文件的AES加密操作
int FileCrypt();
//对文件的AES解密操作
int FileDeCrypt();

int main()
{
FileCrypt();
FileDeCrypt();
return 0;
}

int FileCrypt()
{
//CreateFile获得文件内核句柄
DWORD FileSize;
HANDLE hFile = CreateFile(L"D:\\1.txt", GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
printf("无效的句柄，尝试重新打开\n");
}

//获取文件大小
FileSize = GetFileSize(hFile, NULL);
if (FileSize == INVALID_FILE_SIZE)
{
printf("获取文件大小出错，请重新尝试\n");
}

//动态分配待加密字符串
char * Buff1 = (char*) malloc(sizeof(char) * FileSize);
memset(Buff1, 0, sizeof(Buff1));
BOOL b;
DWORD dwSizeOfRead = 0;
b = ReadFile(hFile, Buff1, FileSize, &dwSizeOfRead, NULL);

//读取文件内容
if ((dwSizeOfRead != FileSize) || (!b))
{
printf("读取文件出错\n");
}

//字符串补位操作
int add;
if (FileSize / 16 == 0)
add = 16;
else
add = 16 - FileSize % 16;

//动态分配补位后待加密字符串
char * Buff2 = (char*) malloc(sizeof(char) * (FileSize + add));
memset(Buff2, 0, sizeof(Buff2));
//进行补齐
for(int i = 0; i < FileSize + add; i++)
{
if(i < FileSize)
Buff2[i] = Buff1[i];
else
Buff2[i] = (char)add;
}

//加密操作
Aes aes(16, (unsigned char *)"1234567812345678");
//将缓存区里面的数据加密,放入新的缓存区里面
HANDLE hNewFile = CreateFile(L"D:\\1(加密).txt", GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
DWORD count = 0;
char Temp1[16];
char Temp2[16];
while (count <= FileSize)
{
memset(Temp1, 0 ,sizeof(Temp1));
memset(Temp2, 0, sizeof(Temp2));
for (int i = 0; i <=15; i++)
Temp1[i] = Buff2[i + count];
aes.Cipher((unsigned char*)Temp1, (unsigned char*)Temp2);
//将新缓存区的数据写入新的文件里面
DWORD dwSizeOfWrite = 0;
WriteFile(hNewFile, Temp2, sizeof(Temp2), &dwSizeOfWrite, 0);
SetFilePointer(hNewFile, 0, NULL, FILE_END);
count +=16;
}
CloseHandle(hFile);
CloseHandle(hNewFile);
return 0;
}

int FileDeCrypt()
{
//CreateFile获得文件内核句柄
DWORD FileSize;
HANDLE hFile = CreateFile(L"D:\\1(加密).txt", GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
printf("无效的句柄，尝试重新打开\n");
}

//获取文件大小
FileSize = GetFileSize(hFile, NULL);
if (FileSize == INVALID_FILE_SIZE)
{
printf("获取文件大小出错，请重新尝试\n");
return 1;
}
if (FileSize % 16 != 0)
{
printf("加密文件大小有误，请检查加密文件");
return 1;
}

//动态分配内存
char* Buff1 = (char*)malloc(sizeof(char) * FileSize);
char* Buff2 = (char*)malloc(sizeof(char) * FileSize);
BOOL b;
DWORD dwSizeOfRead = 0;
b = ReadFile(hFile, Buff1, FileSize, &dwSizeOfRead, NULL);
//读取文件内容
if ((dwSizeOfRead != FileSize) || (!b))
{
printf("读取文件出错\n");
return 1;
}
//解密操作
Aes aes(16, (unsigned char *)"1234567812345678");
//将缓存区里面的数据加密,放入新的缓存区里面
HANDLE hNewFile = CreateFile(L"D:\\1(解密).txt", GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
DWORD count = 0;
char Temp1[16];
char Temp2[16];
for (int j = 0; j < FileSize / 16; j++)
{
memset(Temp1, 0 ,sizeof(Temp1));
memset(Temp2, 0, sizeof(Temp2));
for (int i = 0; i <=15; i++)
Temp1[i] = Buff1[i + count];
aes.InvCipher((unsigned char*)Temp1, (unsigned char*)Temp2);
if (j == FileSize / 16 - 1) //最后一次写文件
{
int add = (int)Temp2[15];//获得最后一次要写入的16-add个字节
DWORD dwSizeOfWrite = 0;
WriteFile(hNewFile, Temp2, sizeof(char) * (16 - add), &dwSizeOfWrite, 0);
SetFilePointer(hNewFile, 0, NULL, FILE_END);
}
else
{
//将新缓存区的数据写入新的文件里面
DWORD dwSizeOfWrite = 0;
WriteFile(hNewFile, Temp2, sizeof(Temp2), &dwSizeOfWrite, 0);
SetFilePointer(hNewFile, 0, NULL, FILE_END);
count +=16;
}
}

CloseHandle(hFile);
CloseHandle(hNewFile);
return 0;
}