luckfox-pico-sdk/media/security/librkcrypto/test/c_mode/aes_ctr.c
luckfox-eng29 8f34c2760d project:build.sh: Added fastboot support; custom modifications to U-Boot and kernel implemented using patches.
project:cfg:BoardConfig_IPC: Added fastboot BoardConfig file and firmware post-scripts, distinguishing between
the BoardConfigs for Luckfox Pico Pro and Luckfox Pico Max. project:app: Added fastboot_client and rk_smart_door
for quick boot applications; updated rkipc app to adapt to the latest media library. media:samples: Added more
usage examples. media:rockit: Fixed bugs; removed support for retrieving data frames from VPSS. media:isp:
Updated rkaiq library and related tools to support connection to RKISP_Tuner. sysdrv:Makefile: Added support for
compiling drv_ko on Luckfox Pico Ultra W using Ubuntu; added support for custom root filesystem.
sysdrv:tools:board: Updated Buildroot optional mirror sources, updated some software versions, and stored device
tree files and configuration files that undergo multiple modifications for U-Boot and kernel separately.
sysdrv:source:mcu: Used RISC-V MCU SDK with RT-Thread system, mainly for initializing camera AE during quick
boot. sysdrv:source:uboot: Added support for fastboot; added high baud rate DDR bin for serial firmware upgrades.
sysdrv:source:kernel: Upgraded to version 5.10.160; increased NPU frequency for RV1106G3; added support for
fastboot.

Signed-off-by: luckfox-eng29 <eng29@luckfox.com>
2024-10-14 09:47:04 +08:00

71 lines
1.6 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "aes_core.h"
static void rk_init_ctr(struct ctr_state * state, const unsigned char *iv)
{
state->num = 0;
memset(state->ecount, 0, 16);
memcpy(state->ivec, iv, 16);
}
/* increment counter (128-bit int) by 1 */
static void rk_ctr128_inc(unsigned char *counter) {
unsigned int n=16;
unsigned char c;
do {
--n;
c = counter[n];
++c;
counter[n] = c;
if (c) return;
} while (n);
}
static void rk_crypto_ctr128_encrypt(const unsigned char *in, unsigned char *out,
int len, const void *key, unsigned char *ivec, unsigned char *ecount_buf,
unsigned int *num)
{
int n, l=0;
n = *num;
while (l<len) {
if (n==0) {
rk_aes_encrypt(ivec, ecount_buf, key);
rk_ctr128_inc(ivec);
}
out[l] = in[l] ^ ecount_buf[n];
++l;
n = (n+1) % 16;
}
*num = n;
}
/* XTS makes use of two different keys, usually generated by splitting
* the supplied block cipher's key in half.
* Because of the splitting, users wanting AES 256 and AES 128 encryption
* will need to choose key sizes of 512 bits and 256 bits respectively.
*/
int rk_aes_ctr_encrypt(const unsigned char *in, unsigned char *out,
unsigned long length, const unsigned char *key, const int key_len,
const unsigned char *ivec, const int enc)
{
RK_AES_KEY ks;
struct ctr_state state;
if(key_len != 16 && key_len != 24 && key_len != 32)
return -1;
rk_init_ctr(&state, ivec);
rk_aes_set_encrypt_key(key, key_len * 8, &ks);
rk_crypto_ctr128_encrypt(in, out, length, &ks, state.ivec, state.ecount, &state.num);
return 0;
}