xref: /OK3568_Linux_fs/external/security/librkcrypto/test/c_mode/aes_ecb.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 
7*4882a593Smuzhiyun 
8*4882a593Smuzhiyun 
rk_crypto_ecb128_encrypt(const unsigned char * in,unsigned long length,unsigned char * out,const RK_AES_KEY * ks,const int enc)9*4882a593Smuzhiyun static int rk_crypto_ecb128_encrypt(const unsigned char *in, unsigned long length, unsigned char *out,
10*4882a593Smuzhiyun 		const RK_AES_KEY *ks, const int enc)
11*4882a593Smuzhiyun {
12*4882a593Smuzhiyun 	unsigned long i = 0;
13*4882a593Smuzhiyun 
14*4882a593Smuzhiyun 	for(i=0; i<length/16; i++){
15*4882a593Smuzhiyun 		if (enc)
16*4882a593Smuzhiyun 			rk_aes_encrypt(in+i*16, out+i*16, ks);
17*4882a593Smuzhiyun 		else
18*4882a593Smuzhiyun 			rk_aes_decrypt(in+i*16, out+i*16, ks);
19*4882a593Smuzhiyun 
20*4882a593Smuzhiyun 	};
21*4882a593Smuzhiyun 	return 0;
22*4882a593Smuzhiyun }
23*4882a593Smuzhiyun 
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)24*4882a593Smuzhiyun int rk_aes_ecb_encrypt(const unsigned char *in, unsigned char *out,
25*4882a593Smuzhiyun         unsigned long length, const unsigned char *key, const int key_len,
26*4882a593Smuzhiyun         const int enc)
27*4882a593Smuzhiyun {
28*4882a593Smuzhiyun     RK_AES_KEY ks;
29*4882a593Smuzhiyun 
30*4882a593Smuzhiyun 	if (in == NULL || out ==NULL || key == NULL)
31*4882a593Smuzhiyun 		return -1;
32*4882a593Smuzhiyun 
33*4882a593Smuzhiyun 	if (key_len != 128/8 && key_len != 192/8 && key_len != 256/8)
34*4882a593Smuzhiyun 		return -2;
35*4882a593Smuzhiyun 
36*4882a593Smuzhiyun 	if(length % 16 != 0)
37*4882a593Smuzhiyun 		return -3;
38*4882a593Smuzhiyun 
39*4882a593Smuzhiyun     if (enc) {
40*4882a593Smuzhiyun             rk_aes_set_encrypt_key(key, key_len * 8, &ks);
41*4882a593Smuzhiyun     } else {
42*4882a593Smuzhiyun             rk_aes_set_decrypt_key(key, key_len * 8, &ks);
43*4882a593Smuzhiyun     }
44*4882a593Smuzhiyun 
45*4882a593Smuzhiyun 	 rk_crypto_ecb128_encrypt(in, length, out, &ks, enc);
46*4882a593Smuzhiyun 	 return 0;
47*4882a593Smuzhiyun }
48*4882a593Smuzhiyun 
49*4882a593Smuzhiyun 
50