BOOL AESEncrypt(BYTE *pOriginalData, DWORD dwOriginalDataSize, BYTE *pAESKey, DWORD dwAESKeySize, BYTE **ppEncryptData, DWORD *pdwEncryptData)
{
	// 定義AES加密需要的數(shù)據(jù)塊
	AESEncryption aesEncryptor;
	// 加密原文數(shù)據(jù)塊
	unsigned char inBlock[AES::BLOCKSIZE];
	// 加密后密文數(shù)據(jù)塊
	unsigned char outBlock[AES::BLOCKSIZE];
	// 必須設(shè)定全為0
	unsigned char xorBlock[AES::BLOCKSIZE];

	DWORD dwOffset = 0;
	BYTE *pEncryptData = NULL;
	DWORD dwEncryptDataSize = 0;

	// 計算需要的加密數(shù)據(jù)塊大小, 并按 128位 即 16字節(jié) 對齊, 不夠則 填充0 對齊
	// 商
	DWORD dwQuotient = dwOriginalDataSize / AES::BLOCKSIZE;
	// 余數(shù)
	DWORD dwRemaind = dwOriginalDataSize % AES::BLOCKSIZE;
	if (0 != dwRemaind)
	{
		dwQuotient++;
	}

	// 申請動態(tài)內(nèi)存
	dwEncryptDataSize = dwQuotient * AES::BLOCKSIZE;

	// 分配加密后的數(shù)據(jù)緩沖區(qū)
	pEncryptData = new BYTE[dwEncryptDataSize];
	if (NULL == pEncryptData)
	{
		return FALSE;
	}

	// 設(shè)置AES加密密鑰
	aesEncryptor.SetKey(pAESKey, dwAESKeySize);

	do
	{
		// 初始化數(shù)據(jù)塊
		RtlZeroMemory(inBlock, AES::BLOCKSIZE);
		RtlZeroMemory(xorBlock, AES::BLOCKSIZE);
		RtlZeroMemory(outBlock, AES::BLOCKSIZE);

		// 獲取加密塊
		if (dwOffset <= (dwOriginalDataSize - AES::BLOCKSIZE))
		{
			RtlCopyMemory(inBlock, (PVOID)(pOriginalData + dwOffset), AES::BLOCKSIZE);
		}
		else
		{
			RtlCopyMemory(inBlock, (PVOID)(pOriginalData + dwOffset), (dwOriginalDataSize - dwOffset));
		}

		// 使用AES算法進行加密
		aesEncryptor.ProcessAndXorBlock(inBlock, xorBlock, outBlock);

		// 將加密后的數(shù)據(jù)塊拷貝到輸出緩沖區(qū)
		RtlCopyMemory((PVOID)(pEncryptData + dwOffset), outBlock, AES::BLOCKSIZE);

		// 更新數(shù)據(jù)
		dwOffset = dwOffset + AES::BLOCKSIZE;
		dwQuotient--;
	} while (0 < dwQuotient);

	// 返回數(shù)據(jù)
	*ppEncryptData = pEncryptData;
	*pdwEncryptData = dwEncryptDataSize;

	return TRUE;
}

BOOL AESDecrypt(BYTE *pEncryptData, DWORD dwEncryptData, BYTE *pAESKey, DWORD dwAESKeySize, BYTE **ppDecryptData, DWORD *pdwDecryptData)
{
	// 定義AES解密需要的數(shù)據(jù)塊
	AESDecryption aesDecryptor;				
	// 解密密文數(shù)據(jù)塊
	unsigned char inBlock[AES::BLOCKSIZE];		
	// 解密后后明文數(shù)據(jù)塊
	unsigned char outBlock[AES::BLOCKSIZE];			
	// 必須設(shè)定全為0
	unsigned char xorBlock[AES::BLOCKSIZE];						
	DWORD dwOffset = 0;
	BYTE *pDecryptData = NULL;
	DWORD dwDecryptDataSize = 0;

	// 計算密文長度, 并按 128位 即 16字節(jié) 對齊, 不夠則填充0對齊
	// 商
	DWORD dwQuotient = dwEncryptData / AES::BLOCKSIZE;
	// 余數(shù)
	DWORD dwRemaind = dwEncryptData % AES::BLOCKSIZE;		
	if (0 != dwRemaind)
	{
		dwQuotient++;
	}

	// 分配解密后的數(shù)據(jù)緩沖區(qū)
	dwDecryptDataSize = dwQuotient * AES::BLOCKSIZE;
	pDecryptData = new BYTE[dwDecryptDataSize];
	if (NULL == pDecryptData)
	{
		return FALSE;
	}

	// 設(shè)置AES解密密鑰
	aesDecryptor.SetKey(pAESKey, dwAESKeySize);

	do
	{
		// 初始化數(shù)據(jù)塊
		RtlZeroMemory(inBlock, AES::BLOCKSIZE);
		RtlZeroMemory(xorBlock, AES::BLOCKSIZE);
		RtlZeroMemory(outBlock, AES::BLOCKSIZE);

		// 將加密數(shù)據(jù)塊拷貝到輸入數(shù)據(jù)塊
		if (dwOffset <= (dwDecryptDataSize - AES::BLOCKSIZE))
		{
			RtlCopyMemory(inBlock, (PVOID)(pEncryptData + dwOffset), AES::BLOCKSIZE);
		}
		else
		{
			RtlCopyMemory(inBlock, (PVOID)(pEncryptData + dwOffset), (dwEncryptData - dwOffset));
		}

		// 使用AES算法進行解密
		aesDecryptor.ProcessAndXorBlock(inBlock, xorBlock, outBlock);

		// 將解密后的數(shù)據(jù)塊拷貝到輸出緩沖區(qū)
		RtlCopyMemory((PVOID)(pDecryptData + dwOffset), outBlock, AES::BLOCKSIZE);

		// 更新數(shù)據(jù)
		dwOffset = dwOffset + AES::BLOCKSIZE;
		dwQuotient--;
	} while (0 < dwQuotient);

	// 返回數(shù)據(jù)
	*ppDecryptData = pDecryptData;
	*pdwDecryptData = dwDecryptDataSize;

	return TRUE;
}

BOOL AESEncrypt(
    BYTE *pOriginalData,     // [in] 原始數(shù)據(jù)的指針
    DWORD dwOriginalDataSize, // [in] 原始數(shù)據(jù)的大小
    BYTE *pAESKey,           // [in] AES加密密鑰的指針
    DWORD dwAESKeySize,       // [in] AES加密密鑰的大小
    BYTE **ppEncryptData,    // [out] 指向指針的指針，用于存儲加密后的數(shù)據(jù)
    DWORD *pdwEncryptData    // [out] 指向DWORD的指針，用于存儲加密后的數(shù)據(jù)大小
);

BOOL AESDecrypt(
    BYTE *pEncryptData,      // [in] 加密后的數(shù)據(jù)的指針
    DWORD dwEncryptDataSize, // [in] 加密后的數(shù)據(jù)的大小
    BYTE *pAESKey,           // [in] AES解密密鑰的指針
    DWORD dwAESKeySize,       // [in] AES解密密鑰的大小
    BYTE **ppDecryptData,    // [out] 指向指針的指針，用于存儲解密后的數(shù)據(jù)
    DWORD *pdwDecryptData    // [out] 指向DWORD的指針，用于存儲解密后的數(shù)據(jù)大小
);

void ShowData(BYTE *pData, DWORD dwSize)
{
	for (int i = 0; i < dwSize; i++)
	{
		if ((0 != i) &&
			(0 == i % 16))
		{
			printf("\n");
		}
		else if ((0 != i) &&
			(0 == i % 8))
		{
			printf(" ");
		}

		printf("%02X ", pData[i]);
	}
	printf("\n");
}

int main(int argc, char* argv[])
{
	BYTE *pEncryptData = NULL;
	DWORD dwEncryptDataSize = 0;
	BYTE *pDecryptData = NULL;
	DWORD dwDecryptDataSize = 0;
	char szOriginalData[] = "It's better to be alone than to be with someone you're not happy to be with.";

	char szAESKey[] = "ABCDEFGHIJKIMNOP";
	BOOL  bRet = FALSE;

	// 加密
	bRet = AESEncrypt((BYTE *)szOriginalData, (1 + ::lstrlen(szOriginalData)), (BYTE *)szAESKey, ::lstrlen(szAESKey), &pEncryptData, &dwEncryptDataSize);
	if (FALSE == bRet)
	{
		return 1;
	}

	// 解密
	bRet = AESDecrypt(pEncryptData, dwEncryptDataSize, (BYTE *)szAESKey, ::lstrlen(szAESKey), &pDecryptData, &dwDecryptDataSize);
	if (FALSE == bRet)
	{
		return 2;
	}

	// 顯示
	printf("原文數(shù)據(jù):\n");
	ShowData((BYTE *)szOriginalData, (1 + ::lstrlen(szOriginalData)));
	printf("密文數(shù)據(jù):\n");
	ShowData(pEncryptData, dwEncryptDataSize);
	printf("解密后數(shù)據(jù):\n");
	ShowData(pDecryptData, dwDecryptDataSize);

	// 釋放內(nèi)存
	delete[]pEncryptData;
	pEncryptData = NULL;
	delete[]pDecryptData;
	pDecryptData = NULL;

	system("pause");
	return 0;
}