mirror of
https://github.com/wangyu-/udp2raw.git
synced 2025-01-19 14:29:34 +08:00
57 lines
1.4 KiB
C++
57 lines
1.4 KiB
C++
#include "aes.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#if defined(AES256) && (AES256 == 1)
|
|
#define AES_KEYSIZE 256
|
|
#elif defined(AES192) && (AES192 == 1)
|
|
#define AES_KEYSIZE 192
|
|
#else
|
|
#define AES_KEYSIZE 128
|
|
#endif
|
|
|
|
|
|
void AES_ECB_encrypt(const uint8_t* input, const uint8_t* key, uint8_t *output, const uint32_t length)
|
|
{
|
|
printf("AES_ECB_encrypt not implemented\n");
|
|
exit(-1);
|
|
}
|
|
void AES_ECB_decrypt(const uint8_t* input, const uint8_t* key, uint8_t *output, const uint32_t length)
|
|
{
|
|
printf("AES_ECB_encrypt not implemented\n");
|
|
exit(-1);
|
|
}
|
|
|
|
void AES_CBC_encrypt_buffer(uint8_t* output, uint8_t* input, uint32_t length, const uint8_t* key, const uint8_t* iv)
|
|
{
|
|
static aes_context ctx;
|
|
static int done=0;
|
|
if(done==0)
|
|
{
|
|
aes_init( &ctx);
|
|
done=1;
|
|
}
|
|
|
|
char tmp_iv[16];
|
|
if(key!=0) aes_setkey_enc(&ctx,key,AES_KEYSIZE);
|
|
memcpy(tmp_iv,iv,16);
|
|
aes_crypt_cbc( &ctx, AES_ENCRYPT, length, (unsigned char* )tmp_iv, (const unsigned char*)input,(unsigned char*) output );
|
|
return ;
|
|
}
|
|
void AES_CBC_decrypt_buffer(uint8_t* output, uint8_t* input, uint32_t length, const uint8_t* key, const uint8_t* iv)
|
|
{
|
|
static aes_context ctx;
|
|
static int done=0;
|
|
if(done==0)
|
|
{
|
|
aes_init( &ctx);
|
|
done=1;
|
|
}
|
|
|
|
char tmp_iv[16];
|
|
if(key!=0) aes_setkey_dec(&ctx,key,AES_KEYSIZE);
|
|
memcpy(tmp_iv,iv,16);
|
|
aes_crypt_cbc( &ctx,AES_DECRYPT, length, (unsigned char*)tmp_iv, (const unsigned char*)input, (unsigned char*) output );
|
|
return;
|
|
}
|