72 lines
1.5 KiB
C++
72 lines
1.5 KiB
C++
|
#include "aes.h"
|
|||
|
|
#include <stdlib.h>
|
|||
|
|
#include <string.h>
|
|||
|
|
#include "base64.h"
|
|||
|
|
|
|||
|
|
static int boInited = 0;
|
|||
|
|
static const int Bits128 = 16;
|
|||
|
|
|
|||
|
|
const char* EncryptPassword(char* sDest, int nDestSize, const char* sPassWord, const char* sKey128Bit)
|
|||
|
|
{
|
|||
|
|
struct aes_ctx ctx;
|
|||
|
|
int nInSize;
|
|||
|
|
char* sBuffer;
|
|||
|
|
char* sBase64Buffer;
|
|||
|
|
int nSize;
|
|||
|
|
|
|||
|
|
if ( !boInited )
|
|||
|
|
{
|
|||
|
|
boInited = 1;
|
|||
|
|
gen_tabs();
|
|||
|
|
}
|
|||
|
|
aes_set_key( &ctx, (const u8 *)sKey128Bit, Bits128 );
|
|||
|
|
|
|||
|
|
nInSize = (int)strlen(sPassWord) * sizeof(sPassWord[0]);
|
|||
|
|
sBuffer = (char*)malloc(nInSize * 8 + 4);
|
|||
|
|
sBase64Buffer = sBuffer + nInSize * 2;
|
|||
|
|
|
|||
|
|
nSize = (int)_AESEncrypt( &ctx, sBuffer, sPassWord, nInSize );
|
|||
|
|
nSize = base64_encode(sBase64Buffer, sBuffer, nSize);
|
|||
|
|
sBase64Buffer[nSize] = 0;
|
|||
|
|
|
|||
|
|
strncpy(sDest, sBase64Buffer, nDestSize-1);
|
|||
|
|
sDest[nDestSize-1] = 0;
|
|||
|
|
|
|||
|
|
free(sBuffer);
|
|||
|
|
return sDest;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
const char* DecryptPassword(char* sPassword, int nPasswordLen, const char* sEncrypted, const char* sKey128Bit)
|
|||
|
|
{
|
|||
|
|
struct aes_ctx ctx;
|
|||
|
|
int nInSize;
|
|||
|
|
char* sAESBuffer;
|
|||
|
|
char* sBuffer;
|
|||
|
|
int nSize;
|
|||
|
|
|
|||
|
|
if ( !boInited )
|
|||
|
|
{
|
|||
|
|
boInited = 1;
|
|||
|
|
gen_tabs();
|
|||
|
|
}
|
|||
|
|
aes_set_key( &ctx, (const u8 *)sKey128Bit, Bits128 );
|
|||
|
|
|
|||
|
|
|
|||
|
|
nInSize = (int)strlen(sEncrypted) * sizeof(sEncrypted[0]);
|
|||
|
|
sAESBuffer = (char*)malloc(nInSize * 8 + 4);
|
|||
|
|
sBuffer = sAESBuffer + nInSize * 2;
|
|||
|
|
|
|||
|
|
nSize = base64_decode(sAESBuffer, sEncrypted, nInSize);
|
|||
|
|
_AESDecrypt( &ctx, sBuffer, sAESBuffer, nSize );
|
|||
|
|
sBuffer[nSize] = 0;
|
|||
|
|
|
|||
|
|
strncpy(sPassword, sBuffer, nPasswordLen-1);
|
|||
|
|
sPassword[nPasswordLen-1] = 0;
|
|||
|
|
|
|||
|
|
free(sAESBuffer);
|
|||
|
|
return sPassword;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|