luckfox-pico-sdk/media/security/librkcrypto/test/c_mode/aes_ecb.c
2023-08-08 20:36:47 +08:00

44 lines
1.1 KiB
C

#include "aes_core.h"
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
static int rk_crypto_ecb128_encrypt(const unsigned char *in,
unsigned long length, unsigned char *out,
const RK_AES_KEY *ks, const int enc) {
unsigned long i = 0;
for (i = 0; i < length / 16; i++) {
if (enc)
rk_aes_encrypt(in + i * 16, out + i * 16, ks);
else
rk_aes_decrypt(in + i * 16, out + i * 16, ks);
};
return 0;
}
int rk_aes_ecb_encrypt(const unsigned char *in, unsigned char *out,
unsigned long length, const unsigned char *key,
const int key_len, const int enc) {
RK_AES_KEY ks;
if (in == NULL || out == NULL || key == NULL)
return -1;
if (key_len != 128 / 8 && key_len != 192 / 8 && key_len != 256 / 8)
return -2;
if (length % 16 != 0)
return -3;
if (enc) {
rk_aes_set_encrypt_key(key, key_len * 8, &ks);
} else {
rk_aes_set_decrypt_key(key, key_len * 8, &ks);
}
rk_crypto_ecb128_encrypt(in, length, out, &ks, enc);
return 0;
}