1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <assert.h>
5 #include "aes_core.h"
6
rk_crypto_cbc128_encrypt(const unsigned char * in,unsigned char * out,int len,const void * key,unsigned char * ivec)7 static void rk_crypto_cbc128_encrypt(const unsigned char *in, unsigned char *out,
8 int len, const void *key, unsigned char *ivec)
9 {
10 int n;
11 const unsigned char *iv = ivec;
12
13 while (len) {
14 for(n=0; n<16 && n<len; ++n)
15 out[n] = in[n] ^ iv[n];
16 for(; n<16; ++n)
17 out[n] = iv[n];
18 rk_aes_encrypt(out, out,key);
19 iv = out;
20 if (len<=16) break;
21 len -= 16;
22 in += 16;
23 out += 16;
24 }
25 memcpy(ivec,iv,16);
26 }
27
rk_crypto_cbc128_decrypt(const unsigned char * in,unsigned char * out,int len,const void * key,unsigned char * ivec)28 static void rk_crypto_cbc128_decrypt(const unsigned char *in, unsigned char *out,
29 int len, const void *key, unsigned char *ivec)
30 {
31 int n;
32 unsigned char c;
33 unsigned char tmp_buf[AES_BLOCK_SIZE];
34
35 memset(tmp_buf, 0x00, sizeof(tmp_buf));
36
37 while (len) {
38 rk_aes_decrypt(in, tmp_buf, key);
39 for(n=0; n<16 && n<len; ++n) {
40 c = in[n];
41 out[n] = tmp_buf[n] ^ ivec[n];
42 ivec[n] = c;
43 }
44 if (len<=16) {
45 for (; n<16; ++n)
46 ivec[n] = in[n];
47 break;
48 }
49 len -= 16;
50 in += 16;
51 out += 16;
52 }
53
54 }
55
rk_aes_cbc_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)56 int rk_aes_cbc_encrypt(const unsigned char *in, unsigned char *out,
57 unsigned long length, const unsigned char *key, const int key_len,
58 unsigned char *ivec, const int enc)
59 {
60 RK_AES_KEY ks;
61
62 if(key_len != 16 && key_len != 24 && key_len != 32)
63 return -1;
64
65 if(length%AES_BLOCK_SIZE)
66 return -1;
67
68 if (enc) {
69 rk_aes_set_encrypt_key(key, key_len * 8, &ks);
70 rk_crypto_cbc128_encrypt(in, out, length, &ks, ivec);
71 } else {
72 rk_aes_set_decrypt_key(key, key_len * 8, &ks);
73 rk_crypto_cbc128_decrypt(in, out, length, &ks, ivec);
74 }
75 return 0;
76 }
77
78
79