444 lines
15 KiB
C
444 lines
15 KiB
C
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <20><> <20><> <20><><EFBFBD><EFBFBD>wechat_aes.c
|
|||
|
// <20><> <20><><EFBFBD><EFBFBD>AES<45><53><EFBFBD><EFBFBD><EFBFBD>㷨
|
|||
|
// <20><> <20><> <20>ˣ<EFBFBD>Liangbofu
|
|||
|
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڣ<EFBFBD>2009-07-17
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
#include "wechat_aes.h"
|
|||
|
#include <string.h>
|
|||
|
#include <stdlib.h>
|
|||
|
#include <stdint.h>
|
|||
|
#include <stdio.h>
|
|||
|
|
|||
|
// Ϊ<><CEAA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>C51<35><31><EFBFBD><EFBFBD><EFBFBD>Ż<EFBFBD><C5BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ARM<52><4D>PC<50>Ȼ<EFBFBD><C8BB><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <20>ڷ<EFBFBD>C51<35><31><EFBFBD><EFBFBD><EFBFBD><EFBFBD>û<EFBFBD>ж<EFBFBD><D0B6><EFBFBD>__C51__<5F><5F><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҫ<EFBFBD><D2AA>C51<35>ض<EFBFBD><D8B6>Ĺؼ<C4B9><D8BC>ֶ<EFBFBD><D6B6><EFBFBD>Ϊ<EFBFBD><CEAA>
|
|||
|
#ifndef __C51__
|
|||
|
#define code
|
|||
|
#define data
|
|||
|
#define idata
|
|||
|
#define xdata
|
|||
|
#define pdata
|
|||
|
typedef unsigned char BOOL;
|
|||
|
#else
|
|||
|
typedef bit BOOL;
|
|||
|
#endif
|
|||
|
|
|||
|
|
|||
|
#define Nk (AES_KEY_LENGTH / 32) // <20>ԡ<EFBFBD><D4A1>֡<EFBFBD><D6A1><EFBFBD>4<EFBFBD>ֽڣ<D6BD>Ϊ<EFBFBD><CEAA>λ<EFBFBD><CEBB><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD>
|
|||
|
#define Nb 4 // <20>ԡ<EFBFBD><D4A1>֡<EFBFBD><D6A1><EFBFBD>4<EFBFBD>ֽڣ<D6BD>Ϊ<EFBFBD><CEAA>λ<EFBFBD>ļӽ<C4BC><D3BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݿ<EFBFBD><DDBF><EFBFBD>С<EFBFBD><D0A1><EFBFBD>̶<EFBFBD>Ϊ4
|
|||
|
|
|||
|
// Nr<4E><72><EFBFBD><EFBFBD><EFBFBD>ܵ<EFBFBD><DCB5><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
#if AES_KEY_LENGTH == 128
|
|||
|
#define Nr 10
|
|||
|
#elif AES_KEY_LENGTH == 192
|
|||
|
#define Nr 12
|
|||
|
#elif AES_KEY_LENGTH == 256
|
|||
|
#define Nr 14
|
|||
|
#else
|
|||
|
#error AES_KEY_LENGTH must be 128, 192 or 256 BOOLs!
|
|||
|
#endif
|
|||
|
|
|||
|
// GF(28) <20><><EFBFBD><EFBFBD>ʽ
|
|||
|
#define BPOLY 0x1B // Lower 8 BOOLs of (x^8 + x^4 + x^3 + x + 1), ie. (x^4 + x^3 + x + 1).
|
|||
|
|
|||
|
// AES<45><53><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD>Ϊ128λʱ<CEBB><CAB1>ռ<EFBFBD><D5BC>176<37>ֽڿռ<DABF>
|
|||
|
static xdata unsigned char g_roundKeyTable[4*Nb*(Nr+1)];
|
|||
|
|
|||
|
// <20><><EFBFBD><EFBFBD><EFBFBD>õ<EFBFBD>SBox
|
|||
|
static code const unsigned char SBox[256] =
|
|||
|
{
|
|||
|
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, 0x4b, 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
|
|||
|
};
|
|||
|
|
|||
|
// <20><><EFBFBD><EFBFBD><EFBFBD>õ<EFBFBD>SBox
|
|||
|
static code const unsigned char InvSBox[256] =
|
|||
|
{
|
|||
|
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
|
|||
|
};
|
|||
|
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> RotationWord
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>֡<EFBFBD><D6A1><EFBFBD><EFBFBD>ݽ<EFBFBD><DDBD><EFBFBD>ѭ<EFBFBD><D1AD><EFBFBD><EFBFBD><EFBFBD>ơ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pWord -- Ҫ<><D2AA><EFBFBD>Ƶ<EFBFBD>4<EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pWord -- <20><><EFBFBD>ƺ<EFBFBD><C6BA><EFBFBD>4<EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void RotationWord(unsigned char *pWord)
|
|||
|
{
|
|||
|
unsigned char temp = pWord[0];
|
|||
|
pWord[0] = pWord[1];
|
|||
|
pWord[1] = pWord[2];
|
|||
|
pWord[2] = pWord[3];
|
|||
|
pWord[3] = temp;
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> XorBytes
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pData1 -- Ҫ<><D2AA><EFBFBD><EFBFBD><EFBFBD>ĵ<EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// pData1 -- Ҫ<><D2AA><EFBFBD><EFBFBD><EFBFBD>ĵڶ<C4B5><DAB6><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// nCount -- Ҫ<><D2AA><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><DDB3>ȡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pData1 -- <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ľ<EFBFBD><C4BD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void XorBytes(unsigned char *pData1, const unsigned char *pData2, unsigned char nCount)
|
|||
|
{
|
|||
|
unsigned char i;
|
|||
|
|
|||
|
for (i = 0; i < nCount; i++)
|
|||
|
{
|
|||
|
pData1[i] ^= pData2[i];
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> AddRoundKey
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><> <20>м<EFBFBD>״̬<D7B4><CCAC><EFBFBD><EFBFBD> <20><><EFBFBD>ϣ<EFBFBD><CFA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><DDB3><EFBFBD>Ϊ16<31>ֽڡ<D6BD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// pRoundKey -- <20><><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// static void AddRoundKey(unsigned char *pState, const unsigned char *pRoundKey)
|
|||
|
// {
|
|||
|
// XorBytes(pState, pRoundKey, 4*Nb);
|
|||
|
// }
|
|||
|
|
|||
|
// AddRoundKey<65>ĺ<EFBFBD><C4BA><EFBFBD>ʽ<EFBFBD><CABD><EFBFBD>Ⱥ<EFBFBD><C8BA><EFBFBD><EFBFBD><EFBFBD>ʽ<EFBFBD><CABD><EFBFBD>Խ<EFBFBD>ʡ4<CAA1>ֽڵ<D6BD>data<74><61><EFBFBD><EFBFBD>
|
|||
|
#define AddRoundKey(pState, pRoundKey) \
|
|||
|
XorBytes((pState), (pRoundKey), 4*Nb)
|
|||
|
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> SubBytes
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> ͨ<><CDA8>S<EFBFBD><53><EFBFBD><EFBFBD><EFBFBD>û<EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// nCount -- ״̬<D7B4><CCAC><EFBFBD>ݳ<EFBFBD><DDB3>ȡ<EFBFBD>
|
|||
|
// bInvert -- <20>Ƿ<EFBFBD>ʹ<EFBFBD>÷<EFBFBD><C3B7><EFBFBD>S<EFBFBD><53><EFBFBD>ӣ<EFBFBD><D3A3><EFBFBD><EFBFBD><EFBFBD>ʱʹ<CAB1>ã<EFBFBD><C3A3><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20>û<EFBFBD><C3BB><EFBFBD><EFBFBD><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void SubBytes(unsigned char *pState, unsigned char nCount, BOOL bInvert)
|
|||
|
{
|
|||
|
unsigned char i;
|
|||
|
const unsigned char code *pSBox = bInvert ? InvSBox : SBox;
|
|||
|
|
|||
|
for (i = 0; i < nCount; i++)
|
|||
|
{
|
|||
|
pState[i] = pSBox[pState[i]];
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> ShiftRows
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><>״̬<D7B4><CCAC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>С<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// bInvert -- <20>Ƿ<EFBFBD><C7B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>У<EFBFBD><D0A3><EFBFBD><EFBFBD><EFBFBD>ʱʹ<CAB1>ã<EFBFBD><C3A3><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20><><EFBFBD>к<EFBFBD><D0BA><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void ShiftRows(unsigned char *pState, BOOL bInvert)
|
|||
|
{
|
|||
|
// ע<>⣺״̬<D7B4><CCAC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʽ<EFBFBD><CABD><EFBFBD>ţ<EFBFBD>
|
|||
|
|
|||
|
unsigned char r; // row<6F><77> <20><>
|
|||
|
unsigned char c; // column<6D><6E><EFBFBD><EFBFBD>
|
|||
|
unsigned char temp;
|
|||
|
unsigned char rowData[4];
|
|||
|
|
|||
|
for (r = 1; r < 4; r++)
|
|||
|
{
|
|||
|
// <20><><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
for (c = 0; c < 4; c++)
|
|||
|
{
|
|||
|
rowData[c] = pState[r + 4*c];
|
|||
|
}
|
|||
|
|
|||
|
temp = bInvert ? (4 - r) : r;
|
|||
|
for (c = 0; c < 4; c++)
|
|||
|
{
|
|||
|
pState[r + 4*c] = rowData[(c + temp) % 4];
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> GfMultBy02
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><>GF(28)<29><><EFBFBD><EFBFBD> <20><>2 <20><><EFBFBD>㡣
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> num -- <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09>ޡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> num<75><6D><EFBFBD><EFBFBD>2<EFBFBD>Ľ<EFBFBD><C4BD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static unsigned char GfMultBy02(unsigned char num)
|
|||
|
{
|
|||
|
if ((num & 0x80) == 0)
|
|||
|
{
|
|||
|
num = num << 1;
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
num = (num << 1) ^ BPOLY;
|
|||
|
}
|
|||
|
|
|||
|
return num;
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> MixColumns
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><><EFBFBD><EFBFBD>״̬<D7B4><CCAC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// bInvert -- <20>Ƿ<EFBFBD><C7B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ϣ<EFBFBD><CFA3><EFBFBD><EFBFBD><EFBFBD>ʱʹ<CAB1>ã<EFBFBD><C3A3><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20><><EFBFBD><EFBFBD><EFBFBD>к<EFBFBD><D0BA><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void MixColumns(unsigned char *pState, BOOL bInvert)
|
|||
|
{
|
|||
|
unsigned char i;
|
|||
|
unsigned char temp;
|
|||
|
unsigned char a0Pa2_M4; // 4(a0 + a2)
|
|||
|
unsigned char a1Pa3_M4; // 4(a1 + a3)
|
|||
|
unsigned char result[4];
|
|||
|
|
|||
|
for (i = 0; i < 4; i++, pState += 4)
|
|||
|
{
|
|||
|
// b0 = 2a0 + 3a1 + a2 + a3
|
|||
|
// = (a0 + a1 + a2 + a3) + 2(a0 + a1) + a0
|
|||
|
|
|||
|
temp = pState[0] ^ pState[1] ^ pState[2] ^ pState[3];
|
|||
|
result[0] = temp ^ pState[0] ^ GfMultBy02((unsigned char) (pState[0] ^ pState[1]));
|
|||
|
result[1] = temp ^ pState[1] ^ GfMultBy02((unsigned char) (pState[1] ^ pState[2]));
|
|||
|
result[2] = temp ^ pState[2] ^ GfMultBy02((unsigned char) (pState[2] ^ pState[3]));
|
|||
|
result[3] = temp ^ pState[3] ^ GfMultBy02((unsigned char) (pState[3] ^ pState[0]));
|
|||
|
|
|||
|
if (bInvert)
|
|||
|
{
|
|||
|
// b0' = 14a0 + 11a1 + 13a2 + 9a3
|
|||
|
// = (a0 + a1 + a2 + a3) + 2(a0 + a1) + a0 <09><><EFBFBD>ⲿ<EFBFBD><E2B2BF>Ϊb0<62><30>
|
|||
|
// + 2(4(a0 + a2) + 4(a1 + a3))
|
|||
|
// + 4(a0 + a2)
|
|||
|
|
|||
|
a0Pa2_M4 = GfMultBy02(GfMultBy02((unsigned char) (pState[0] ^ pState[2])));
|
|||
|
a1Pa3_M4 = GfMultBy02(GfMultBy02((unsigned char) (pState[1] ^ pState[3])));
|
|||
|
temp = GfMultBy02((unsigned char) (a0Pa2_M4 ^ a1Pa3_M4));
|
|||
|
result[0] ^= temp ^ a0Pa2_M4;
|
|||
|
result[1] ^= temp ^ a1Pa3_M4;
|
|||
|
result[2] ^= temp ^ a0Pa2_M4;
|
|||
|
result[3] ^= temp ^ a1Pa3_M4;
|
|||
|
}
|
|||
|
|
|||
|
memcpy(pState, result, 4);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> BlockEncrypt
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09>Ե<EFBFBD><D4B5><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݼ<EFBFBD><DDBC>ܡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20><><EFBFBD>ܺ<EFBFBD><DCBA><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void BlockEncrypt(unsigned char *pState)
|
|||
|
{
|
|||
|
unsigned char i;
|
|||
|
|
|||
|
AddRoundKey(pState, g_roundKeyTable);
|
|||
|
|
|||
|
for (i = 1; i <= Nr; i++) // i = [1, Nr]
|
|||
|
{
|
|||
|
SubBytes(pState, 4*Nb, 0);
|
|||
|
ShiftRows(pState, 0);
|
|||
|
|
|||
|
if (i != Nr)
|
|||
|
{
|
|||
|
MixColumns(pState, 0);
|
|||
|
}
|
|||
|
|
|||
|
AddRoundKey(pState, &g_roundKeyTable[4*Nb*i]);
|
|||
|
}
|
|||
|
|
|||
|
// Ϊ<>˽<EFBFBD>ʡ<EFBFBD><CAA1><EFBFBD>룬<EFBFBD>ϲ<EFBFBD><CFB2><EFBFBD>ѭ<EFBFBD><D1AD>ִ<EFBFBD><D6B4>
|
|||
|
// SubBytes(pState, 4*Nb);
|
|||
|
// ShiftRows(pState, 0);
|
|||
|
// AddRoundKey(pState, &g_roundKeyTable[4*Nb*Nr]);
|
|||
|
}
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> BlockDecrypt
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09>Ե<EFBFBD><D4B5><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݽ<EFBFBD><DDBD>ܡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- ״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pState -- <20><><EFBFBD>ܺ<EFBFBD><DCBA><EFBFBD>״̬<D7B4><CCAC><EFBFBD>ݡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
static void BlockDecrypt(unsigned char *pState)
|
|||
|
{
|
|||
|
unsigned char i;
|
|||
|
|
|||
|
AddRoundKey(pState, &g_roundKeyTable[4*Nb*Nr]);
|
|||
|
|
|||
|
for (i = Nr; i > 0; i--) // i = [Nr, 1]
|
|||
|
{
|
|||
|
ShiftRows(pState, 1);
|
|||
|
SubBytes(pState, 4*Nb, 1);
|
|||
|
AddRoundKey(pState, &g_roundKeyTable[4*Nb*(i-1)]);
|
|||
|
|
|||
|
if (i != 1)
|
|||
|
{
|
|||
|
MixColumns(pState, 1);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> AES_Init
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD>ڴ<EFBFBD>ִ<EFBFBD><D6B4><EFBFBD><EFBFBD>չ<EFBFBD><D5B9>Կ<EFBFBD><D4BF><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pKey -- ԭʼ<D4AD><CABC>Կ<EFBFBD><D4BF><EFBFBD>䳤<EFBFBD>ȱ<EFBFBD><C8B1><EFBFBD>Ϊ AES_KEY_LENGTH/8 <20>ֽڡ<D6BD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09>ޡ<EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
///////////////////////////////////////////////////////////////////////////////
|
|||
|
void AES_Init(const void *pKey)
|
|||
|
{
|
|||
|
// <20><>չ<EFBFBD><D5B9>Կ
|
|||
|
unsigned char i;
|
|||
|
unsigned char *pRoundKey;
|
|||
|
unsigned char Rcon[4] = {0x01, 0x00, 0x00, 0x00};
|
|||
|
|
|||
|
memcpy(g_roundKeyTable, pKey, 4*Nk);
|
|||
|
|
|||
|
pRoundKey = &g_roundKeyTable[4*Nk];
|
|||
|
|
|||
|
for (i = Nk; i < Nb*(Nr+1); pRoundKey += 4, i++)
|
|||
|
{
|
|||
|
memcpy(pRoundKey, pRoundKey - 4, 4);
|
|||
|
|
|||
|
if (i % Nk == 0)
|
|||
|
{
|
|||
|
RotationWord(pRoundKey);
|
|||
|
SubBytes(pRoundKey, 4, 0);
|
|||
|
XorBytes(pRoundKey, Rcon, 4);
|
|||
|
|
|||
|
Rcon[0] = GfMultBy02(Rcon[0]);
|
|||
|
}
|
|||
|
else if (Nk > 6 && i % Nk == Nb)
|
|||
|
{
|
|||
|
SubBytes(pRoundKey, 4, 0);
|
|||
|
}
|
|||
|
|
|||
|
XorBytes(pRoundKey, pRoundKey - 4*Nk, 4);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
unsigned int AES_get_length(unsigned int length)
|
|||
|
{
|
|||
|
return ((length>>4) + 1)<<4;
|
|||
|
}
|
|||
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> AES_Encrypt
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pPlainText -- <20><><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ܵ<EFBFBD><DCB5><EFBFBD><EFBFBD>ݣ<EFBFBD><DDA3>䳤<EFBFBD><E4B3A4>ΪnDataLen<65>ֽڡ<D6BD>
|
|||
|
// nDataLen -- <20><><EFBFBD>ݳ<EFBFBD><DDB3>ȣ<EFBFBD><C8A3><EFBFBD><EFBFBD>ֽ<EFBFBD>Ϊ<EFBFBD><CEAA>λ
|
|||
|
// pIV -- <20><>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9>ECBģʽ<C4A3><CABD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪNULL<4C><4C>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pCipherText -- <20><><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><C4BC>ܺ<EFBFBD><DCBA><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݣ<EFBFBD><DDA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>pPlainText<78><74>ͬ<EFBFBD><CDAC>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
unsigned int AES_Encrypt_PKCS7(const unsigned char *pPlainText, unsigned char *pCipherText,
|
|||
|
unsigned int nDataLen, const unsigned char *pIV)
|
|||
|
{
|
|||
|
unsigned int i;
|
|||
|
//<2F><><EFBFBD>ȵ<EFBFBD><C8B5><EFBFBD>
|
|||
|
unsigned int length = nDataLen;
|
|||
|
nDataLen = ((nDataLen>>4) + 1)<<4;
|
|||
|
uint8_t p=(Nb<<2)-(length%(Nb<<2));
|
|||
|
if (pPlainText != pCipherText){memcpy(pCipherText,pPlainText,length);}
|
|||
|
//<2F><><EFBFBD><EFBFBD>
|
|||
|
unsigned char temp[16];
|
|||
|
memset(temp,p,16);
|
|||
|
if(length<nDataLen)
|
|||
|
{
|
|||
|
memcpy(pCipherText + length,temp,nDataLen - length);
|
|||
|
}
|
|||
|
if(length == nDataLen)
|
|||
|
{
|
|||
|
memcpy(pCipherText + length,temp,16);
|
|||
|
}
|
|||
|
|
|||
|
for (i = nDataLen/(4*Nb); i > 0 ; i--, pCipherText += 4*Nb)
|
|||
|
{
|
|||
|
#if AES_MODE == AES_MODE_CBC
|
|||
|
XorBytes(pCipherText, pIV, 4*Nb);
|
|||
|
#endif
|
|||
|
BlockEncrypt(pCipherText);
|
|||
|
pIV = pCipherText;
|
|||
|
}
|
|||
|
return(nDataLen);
|
|||
|
}
|
|||
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> AES_Decrypt
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pCipherText -- <20><><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ܵ<EFBFBD><DCB5><EFBFBD><EFBFBD>ݣ<EFBFBD><DDA3>䳤<EFBFBD><E4B3A4>ΪnDataLen<65>ֽڡ<D6BD>
|
|||
|
// nDataLen -- <20><><EFBFBD>ݳ<EFBFBD><DDB3>ȣ<EFBFBD><C8A3><EFBFBD><EFBFBD>ֽ<EFBFBD>Ϊ<EFBFBD><CEAA>λ<EFBFBD><CEBB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪAES_KEY_LENGTH/8<><38><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
// pIV -- <20><>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʹ<EFBFBD><CAB9>ECBģʽ<C4A3><CABD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΪNULL<4C><4C>
|
|||
|
// <09><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> pPlainText -- <20><><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ľ<EFBFBD><C4BD>ܺ<EFBFBD><DCBA><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݣ<EFBFBD><DDA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>pCipherText<78><74>ͬ<EFBFBD><CDAC>
|
|||
|
// <09><><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5> <09>ޡ<EFBFBD>
|
|||
|
//////////////////////////////////////////////////////////////////////////
|
|||
|
void AES_Decrypt(unsigned char *pPlainText, const unsigned char *pCipherText,
|
|||
|
unsigned int nDataLen, const unsigned char *pIV)
|
|||
|
{
|
|||
|
unsigned int i;
|
|||
|
|
|||
|
if (pPlainText != pCipherText)
|
|||
|
{
|
|||
|
memcpy(pPlainText, pCipherText, nDataLen);
|
|||
|
}
|
|||
|
|
|||
|
// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>ݿ<EFBFBD>ʼ<EFBFBD><CABC><EFBFBD>ܣ<EFBFBD><DCA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ÿ<EFBFBD><C3BF>ٿռ<D9BF><D5BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>IV
|
|||
|
pPlainText += nDataLen - 4*Nb;
|
|||
|
for (i = nDataLen/(4*Nb); i > 0 ; i--, pPlainText -= 4*Nb)
|
|||
|
{
|
|||
|
BlockDecrypt(pPlainText);
|
|||
|
|
|||
|
#if AES_MODE == AES_MODE_CBC
|
|||
|
if (i == 1)
|
|||
|
{// <20><><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
|||
|
XorBytes(pPlainText, pIV, 4*Nb);
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
XorBytes(pPlainText, pPlainText - 4*Nb, 4*Nb);
|
|||
|
}
|
|||
|
#endif
|
|||
|
}
|
|||
|
}
|
|||
|
|