test_main.c

来自「This a good VPN source」· C语言 代码 · 共 42 行

#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include "aes_cbc.h"
#define AES_BLOCK_SIZE	16
#define KEY_SIZE 	128	/* bits */
#define KEY 		"1234567890123456"
#define STR 		"hola guaso como estaisss ... 012"
#define STRSZ		(sizeof(STR)-1)

#define EMT_AESCBC_BLKLEN AES_BLOCK_SIZE
#define AES_CONTEXT_T  aes_context
#define EMT_ESPAES_KEY_SZ 16
int pretty_print(const unsigned char *buf, int count) {
	int i=0;
	for (;i<count;i++) {
		if (i%8==0) putchar(' ');
		if (i%16==0) putchar('\n');
		printf ("%02hhx ", buf[i]);
	}
	putchar('\n');
	return i;
}
//#define SIZE STRSZ/2
#define SIZE STRSZ
int main() {
	int ret;
	char buf0[SIZE+1], buf1[SIZE+1];
	char IV[AES_BLOCK_SIZE]="\0\0\0\0\0\0\0\0" "\0\0\0\0\0\0\0\0";
	aes_context ac;	
	AES_set_key(&ac, KEY, KEY_SIZE);
	//pretty_print((char *)&ac.aes_e_key, sizeof(ac.aes_e_key));
	memset(buf0, 0, sizeof (buf0));
	memset(buf1, 0, sizeof (buf1));
	ret=AES_cbc_encrypt(&ac, STR, buf0, SIZE, IV, 1);
	pretty_print(buf0, SIZE);
	printf("size=%d ret=%d\n%s\n", SIZE, ret, buf0);
	ret=AES_cbc_encrypt(&ac, buf0, buf1, SIZE, IV, 0);
	printf("size=%d ret=%d\n%s\n", SIZE, ret, buf1);
	return 0;
}