1*4882a593Smuzhiyun #include <stdio.h>
2*4882a593Smuzhiyun #include <stdlib.h>
3*4882a593Smuzhiyun #include <string.h>
4*4882a593Smuzhiyun #include "aes_core.h"
5*4882a593Smuzhiyun
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun typedef void (*block128_f)(const unsigned char in[16],
8*4882a593Smuzhiyun unsigned char out[16],
9*4882a593Smuzhiyun const void *key);
10*4882a593Smuzhiyun
rk_crypto_cfb128_encrypt(const unsigned char * in,unsigned char * out,size_t len,const void * key,unsigned char ivec[16],int * num,int enc,block128_f block)11*4882a593Smuzhiyun static void rk_crypto_cfb128_encrypt(const unsigned char *in, unsigned char *out,
12*4882a593Smuzhiyun size_t len, const void *key,
13*4882a593Smuzhiyun unsigned char ivec[16], int *num,
14*4882a593Smuzhiyun int enc, block128_f block)
15*4882a593Smuzhiyun {
16*4882a593Smuzhiyun unsigned int n;
17*4882a593Smuzhiyun size_t l = 0;
18*4882a593Smuzhiyun n = *num;
19*4882a593Smuzhiyun if (enc) {
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun while (l < len) {
22*4882a593Smuzhiyun if (n == 0) {
23*4882a593Smuzhiyun (*block)(ivec, ivec, key);
24*4882a593Smuzhiyun }
25*4882a593Smuzhiyun out[l] = ivec[n] ^= in[l];
26*4882a593Smuzhiyun ++l;
27*4882a593Smuzhiyun n = (n+1) % 16;
28*4882a593Smuzhiyun }
29*4882a593Smuzhiyun *num = n;
30*4882a593Smuzhiyun } else {
31*4882a593Smuzhiyun while (l < len) {
32*4882a593Smuzhiyun unsigned char c;
33*4882a593Smuzhiyun if (n == 0) {
34*4882a593Smuzhiyun (*block)(ivec, ivec, key);
35*4882a593Smuzhiyun }
36*4882a593Smuzhiyun out[l] = ivec[n] ^ (c = in[l]); ivec[n] = c;
37*4882a593Smuzhiyun ++l;
38*4882a593Smuzhiyun n = (n+1) % 16;
39*4882a593Smuzhiyun }
40*4882a593Smuzhiyun *num=n;
41*4882a593Smuzhiyun }
42*4882a593Smuzhiyun }
43*4882a593Smuzhiyun
rk_aes_cfb128_encrypt(const unsigned char * in,unsigned char * out,size_t length,const RK_AES_KEY * key,unsigned char * ivec,int * num,const int enc)44*4882a593Smuzhiyun static void rk_aes_cfb128_encrypt(const unsigned char *in, unsigned char *out,
45*4882a593Smuzhiyun size_t length, const RK_AES_KEY *key,
46*4882a593Smuzhiyun unsigned char *ivec, int *num, const int enc)
47*4882a593Smuzhiyun {
48*4882a593Smuzhiyun rk_crypto_cfb128_encrypt(in,out,length,key,ivec,num,enc,(block128_f)rk_aes_encrypt);
49*4882a593Smuzhiyun }
50*4882a593Smuzhiyun
rk_aes_cfb_encrypt(const unsigned char * in,unsigned char * out,unsigned long length,const unsigned char * key,const int key_len,unsigned char * ivec,const int enc)51*4882a593Smuzhiyun int rk_aes_cfb_encrypt(const unsigned char *in, unsigned char *out,
52*4882a593Smuzhiyun unsigned long length, const unsigned char *key, const int key_len,
53*4882a593Smuzhiyun unsigned char *ivec, const int enc)
54*4882a593Smuzhiyun {
55*4882a593Smuzhiyun RK_AES_KEY ks1;
56*4882a593Smuzhiyun int num = 0;
57*4882a593Smuzhiyun
58*4882a593Smuzhiyun if (in == NULL || out ==NULL || key == NULL)
59*4882a593Smuzhiyun return -1;
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun if (key_len != 128 / 8 && key_len != 192 / 8 && key_len != 256 / 8)
62*4882a593Smuzhiyun return -2;
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun rk_aes_set_encrypt_key(key, key_len * 8, &ks1);
65*4882a593Smuzhiyun rk_aes_cfb128_encrypt(in, out, length, &ks1, ivec, &num, enc);
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun return 0;
68*4882a593Smuzhiyun }
69*4882a593Smuzhiyun
70*4882a593Smuzhiyun
71