xref: /OK3568_Linux_fs/external/security/librkcrypto/test/c_mode/sm4_cfb.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun #include <stdio.h>
2*4882a593Smuzhiyun #include <stdlib.h>
3*4882a593Smuzhiyun #include <string.h>
4*4882a593Smuzhiyun #include "sm4_core.h"
5*4882a593Smuzhiyun 
rk_crypto_cfb128_encrypt(void * ctx,const unsigned char * in,unsigned char * out,size_t len,unsigned char ivec[SM4_BLOCK_SIZE],int * num,int enc,block128_f block)6*4882a593Smuzhiyun void rk_crypto_cfb128_encrypt(void *ctx, const unsigned char *in, unsigned char *out,
7*4882a593Smuzhiyun 			size_t len, unsigned char ivec[SM4_BLOCK_SIZE], int *num,
8*4882a593Smuzhiyun 			int enc, block128_f block)
9*4882a593Smuzhiyun {
10*4882a593Smuzhiyun     unsigned int n;
11*4882a593Smuzhiyun     size_t l = 0;
12*4882a593Smuzhiyun 
13*4882a593Smuzhiyun     n = *num;
14*4882a593Smuzhiyun     if (enc) {
15*4882a593Smuzhiyun 
16*4882a593Smuzhiyun 	while (l<len) {
17*4882a593Smuzhiyun 		if (n == 0) {
18*4882a593Smuzhiyun 			(*block)(ivec, ivec, ctx);
19*4882a593Smuzhiyun 		}
20*4882a593Smuzhiyun 		out[l] = ivec[n] ^= in[l];
21*4882a593Smuzhiyun 		++l;
22*4882a593Smuzhiyun 		n = (n+1) % SM4_BLOCK_SIZE;
23*4882a593Smuzhiyun 	}
24*4882a593Smuzhiyun 	*num = n;
25*4882a593Smuzhiyun     } else {
26*4882a593Smuzhiyun 	while (l<len) {
27*4882a593Smuzhiyun 		unsigned char c;
28*4882a593Smuzhiyun 		if (n == 0) {
29*4882a593Smuzhiyun 			(*block)(ivec, ivec, ctx);
30*4882a593Smuzhiyun 		}
31*4882a593Smuzhiyun 		out[l] = ivec[n] ^ (c = in[l]); ivec[n] = c;
32*4882a593Smuzhiyun 		++l;
33*4882a593Smuzhiyun 		n = (n+1) % SM4_BLOCK_SIZE;
34*4882a593Smuzhiyun 	}
35*4882a593Smuzhiyun 	*num=n;
36*4882a593Smuzhiyun     }
37*4882a593Smuzhiyun }
38*4882a593Smuzhiyun 
rk_sm4_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)39*4882a593Smuzhiyun int rk_sm4_cfb_encrypt(const unsigned char *in, unsigned char *out,
40*4882a593Smuzhiyun         unsigned long length, const unsigned char *key, const int key_len,
41*4882a593Smuzhiyun         unsigned char *ivec, const int enc)
42*4882a593Smuzhiyun {
43*4882a593Smuzhiyun 	sm4_context  ctx;
44*4882a593Smuzhiyun 	int num = 0;
45*4882a593Smuzhiyun 
46*4882a593Smuzhiyun 	if (in == NULL || out ==NULL || key == NULL)
47*4882a593Smuzhiyun 		return -1;
48*4882a593Smuzhiyun 
49*4882a593Smuzhiyun 	if (key_len != 16)
50*4882a593Smuzhiyun 		return -2;
51*4882a593Smuzhiyun 
52*4882a593Smuzhiyun 	if(length == 0)
53*4882a593Smuzhiyun 		return -3;
54*4882a593Smuzhiyun 
55*4882a593Smuzhiyun 	rk_sm4_setkey_enc(&ctx, key);
56*4882a593Smuzhiyun 
57*4882a593Smuzhiyun 	rk_crypto_cfb128_encrypt((void *)&ctx, in, out,length, ivec,
58*4882a593Smuzhiyun 				&num,enc, rk_rk_sm4_crypt_ecb);
59*4882a593Smuzhiyun 	return 0;
60*4882a593Smuzhiyun }
61*4882a593Smuzhiyun 
62*4882a593Smuzhiyun 
63