xref: /OK3568_Linux_fs/external/security/librkcrypto/test/c_mode/aes_ofb.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun #include <stdio.h>
2*4882a593Smuzhiyun #include <stdlib.h>
3*4882a593Smuzhiyun #include <string.h>
4*4882a593Smuzhiyun #include <assert.h>
5*4882a593Smuzhiyun #include "aes_core.h"
6*4882a593Smuzhiyun 
rk_crypto_ofb128_encrypt(const unsigned char * in,unsigned char * out,int len,const void * key,unsigned char * ivec,unsigned int * num)7*4882a593Smuzhiyun static void rk_crypto_ofb128_encrypt(const unsigned char *in, unsigned char *out,
8*4882a593Smuzhiyun 			int len, const void *key, unsigned char *ivec, unsigned int *num)
9*4882a593Smuzhiyun {
10*4882a593Smuzhiyun 	int n, l=0;
11*4882a593Smuzhiyun 
12*4882a593Smuzhiyun 	n = *num;
13*4882a593Smuzhiyun 
14*4882a593Smuzhiyun 	while (l < len) {
15*4882a593Smuzhiyun 		if (n==0) {
16*4882a593Smuzhiyun 			rk_aes_encrypt(ivec, ivec, key);
17*4882a593Smuzhiyun 		}
18*4882a593Smuzhiyun 		out[l] = in[l] ^ ivec[n];
19*4882a593Smuzhiyun 		++l;
20*4882a593Smuzhiyun 		n = (n+1) % 16;
21*4882a593Smuzhiyun 	}
22*4882a593Smuzhiyun 
23*4882a593Smuzhiyun 	*num=n;
24*4882a593Smuzhiyun }
25*4882a593Smuzhiyun 
26*4882a593Smuzhiyun /* XTS makes use of two different keys, usually generated by splitting
27*4882a593Smuzhiyun  * the supplied block cipher's key in half.
28*4882a593Smuzhiyun  * Because of the splitting, users wanting AES 256 and AES 128 encryption
29*4882a593Smuzhiyun  * will need to choose key sizes of 512 bits and 256 bits respectively.
30*4882a593Smuzhiyun  */
rk_aes_ofb_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)31*4882a593Smuzhiyun int rk_aes_ofb_encrypt(const unsigned char *in, unsigned char *out,
32*4882a593Smuzhiyun         unsigned long length, const unsigned char *key, const int key_len,
33*4882a593Smuzhiyun         unsigned char *ivec, const int enc)
34*4882a593Smuzhiyun {
35*4882a593Smuzhiyun 	RK_AES_KEY ks;
36*4882a593Smuzhiyun 	unsigned int num = 0;
37*4882a593Smuzhiyun 
38*4882a593Smuzhiyun 	if(key_len != 16 && key_len != 24 && key_len != 32)
39*4882a593Smuzhiyun 		return -1;
40*4882a593Smuzhiyun 
41*4882a593Smuzhiyun 	if(length == 0)
42*4882a593Smuzhiyun 		return -1;
43*4882a593Smuzhiyun 
44*4882a593Smuzhiyun 	rk_aes_set_encrypt_key(key, key_len * 8, &ks);
45*4882a593Smuzhiyun 	rk_crypto_ofb128_encrypt(in, out, length, &ks, ivec, &num);
46*4882a593Smuzhiyun 	return 0;
47*4882a593Smuzhiyun }
48*4882a593Smuzhiyun 
49*4882a593Smuzhiyun 
50