77 lines
1.5 KiB
C++
77 lines
1.5 KiB
C++
|
#include "aes.h"
|
||
|
|
#include "coAes.h"
|
||
|
|
#include <cstring>
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////
|
||
|
|
static void aes_encrypt_data( aes_context *ctx, const std::string& ins, std::string& outs)
|
||
|
|
{
|
||
|
|
int index = 0;
|
||
|
|
int size = ins.size();
|
||
|
|
int sizeN = size / 16 * 16;
|
||
|
|
|
||
|
|
if (sizeN < size)
|
||
|
|
sizeN += 16;
|
||
|
|
|
||
|
|
outs = ins;
|
||
|
|
outs.resize(sizeN);
|
||
|
|
|
||
|
|
while (index < sizeN)
|
||
|
|
{
|
||
|
|
aes_encrypt(ctx, (unsigned char*)&outs[index], (unsigned char*)&outs[index]);
|
||
|
|
index += 16;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
static void aes_decrypt_data( aes_context *ctx, const std::string& ins, std::string& outs)
|
||
|
|
{
|
||
|
|
int index = 0;
|
||
|
|
int size = ins.size();
|
||
|
|
int sizeN = size / 16 * 16;
|
||
|
|
|
||
|
|
if (sizeN < size)
|
||
|
|
sizeN += 16;
|
||
|
|
|
||
|
|
outs = ins;
|
||
|
|
outs.resize(sizeN);
|
||
|
|
|
||
|
|
while (index < sizeN)
|
||
|
|
{
|
||
|
|
aes_decrypt(ctx, (unsigned char*)&outs[index], (unsigned char*)&outs[index]);
|
||
|
|
index += 16;
|
||
|
|
}
|
||
|
|
|
||
|
|
//去掉000
|
||
|
|
int iZero = 0;
|
||
|
|
for (int i = outs.size() - 1; i >= 0; --i)
|
||
|
|
{
|
||
|
|
if (outs[i]=='\0')
|
||
|
|
iZero++;
|
||
|
|
else
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (iZero > 0)
|
||
|
|
outs = outs.substr(0, outs.size() - iZero);
|
||
|
|
};
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////
|
||
|
|
aes_context __gAesContext;
|
||
|
|
|
||
|
|
void coAesSetKey(const char* key)
|
||
|
|
{
|
||
|
|
unsigned char keyBuf[32];
|
||
|
|
memset(keyBuf, 0, sizeof(keyBuf));
|
||
|
|
strncpy((char*)keyBuf, (const char*)key, 32);
|
||
|
|
aes_set_key(&__gAesContext, keyBuf, 256);
|
||
|
|
}
|
||
|
|
|
||
|
|
void coAesEncript(const std::string& ins, std::string& outs)
|
||
|
|
{
|
||
|
|
aes_encrypt_data(&__gAesContext, ins, outs);
|
||
|
|
}
|
||
|
|
|
||
|
|
void coAesDecript(const std::string& ins, std::string& outs)
|
||
|
|
{
|
||
|
|
aes_decrypt_data(&__gAesContext, ins, outs);
|
||
|
|
}
|