lai/gk-sdk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
master
gk-sdk/sample/cipher/sample_cipher_via.c
T

451 lines
14 KiB
C
Raw Permalink Blame History

/*
* Copyright (c) XMEDIA. All rights reserved.
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include "xmedia_type.h"
#include "xmedia_cipher.h"
#include "sample_common.h"
#define AES_KEY_MAX_LENGTH 32
#define AES128_KEY_LENGTH 16
#define AES_IV_LENGTH 16
static xmedia_s32 set_config_info(xmedia_handle handle, xmedia_bool key_by_ca, xmedia_cipher_alg alg, xmedia_cipher_work_mode mode,
xmedia_cipher_key_length key_len, const xmedia_u8 *key, const xmedia_u8 iv[AES_IV_LENGTH])
{
xmedia_s32 ret = XMEDIA_SUCCESS;
xmedia_cipher_ctrl ctrl;
memset(&ctrl, 0, sizeof(xmedia_cipher_ctrl));
ctrl.alg = alg;
ctrl.mode = mode;
ctrl.key_by_ca = key_by_ca;
if (ctrl.alg == XMEDIA_CIPHER_ALG_AES) {
/*only support ECB/CBC/CFB/OFB/CTR*/
if (ctrl.mode <= XMEDIA_CIPHER_WORK_MODE_CTR && ctrl.mode >= XMEDIA_CIPHER_WORK_MODE_ECB) {
ctrl.aes_ctrl.bit_width = XMEDIA_CIPHER_BIT_WIDTH_128BIT;
ctrl.aes_ctrl.key_len = key_len;
if (ctrl.mode != XMEDIA_CIPHER_WORK_MODE_ECB) {
ctrl.aes_ctrl.iv_usage = 1; //must set for CBC , CFB mode
memcpy(ctrl.aes_ctrl.iv, iv, AES_IV_LENGTH);
}
if (key_len == XMEDIA_CIPHER_KEY_AES_128BIT) {
memcpy(ctrl.aes_ctrl.key, key, AES128_KEY_LENGTH);
} else {
memcpy(ctrl.aes_ctrl.key, key, AES_KEY_MAX_LENGTH);
}
}
}
ret = xmedia_cipher_config_handle(handle, &ctrl);
if (XMEDIA_SUCCESS != ret) {
return XMEDIA_FAILURE;
}
return XMEDIA_SUCCESS;
}
/* encrypt data using special chn*/
static xmedia_s32 cbc_aes128(xmedia_void)
{
xmedia_s32 ret = XMEDIA_SUCCESS;
xmedia_u32 data_len = 16;
xmedia_u8 *input_addr_vir = XMEDIA_NULL;
xmedia_u8 *output_addr_vir = XMEDIA_NULL;
xmedia_handle chnid = 0;
xmedia_u8 aes_key[16] = {0x2B,0x7E,0x15,0x16,0x28,0xAE,0xD2,0xA6,0xAB,0xF7,0x15,0x88,0x09,0xCF,0x4F,0x3C};
xmedia_u8 aes_iv[16] = {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0A,0x0B,0x0C,0x0D,0x0E,0x0F};
xmedia_u8 aes_src[16] = {0x6B,0xC1,0xBE,0xE2,0x2E,0x40,0x9F,0x96,0xE9,0x3D,0x7E,0x11,0x73,0x93,0x17,0x2A};
xmedia_u8 aes_dst[16] = {0x76,0x49,0xAB,0xAC,0x81,0x19,0xB2,0x46,0xCE,0xE9,0x8E,0x9B,0x12,0xE9,0x19,0x7D};
ret = xmedia_cipher_init();
if (XMEDIA_SUCCESS != ret) {
return ret;
}
ret = xmedia_cipher_create_handle(&chnid);
if (XMEDIA_SUCCESS != ret) {
xmedia_cipher_exit();
return ret;
}
printf("chnid 0x%d\n", chnid);
input_addr_vir = (xmedia_u8*)malloc(data_len);
if (input_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc input_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(input_addr_vir, 0, data_len);
output_addr_vir = (xmedia_u8*)malloc(data_len);
if (output_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc output_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(output_addr_vir, 0, data_len);
/* For encrypt */
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CBC,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
memset(input_addr_vir, 0x0, data_len);
memcpy(input_addr_vir, aes_src, data_len);
print_buffer("CBC-AES-128-ORI:", aes_src, sizeof(aes_src));
memset(output_addr_vir, 0x0, data_len);
ret = xmedia_cipher_encrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher encrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CBC-AES-128-ENC:", output_addr_vir, sizeof(aes_dst));
/* compare */
if (0 != memcmp(output_addr_vir, aes_dst, data_len)) {
ERR_CODE_CIPHER("cipher encrypt, memcmp failed!\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
/* For decrypt */
memcpy(input_addr_vir, aes_dst, data_len);
memset(output_addr_vir, 0x0, data_len);
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CBC,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
ret = xmedia_cipher_decrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher decrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CBC-AES-128-DEC:", output_addr_vir, data_len);
/* compare */
if (0 != memcmp(output_addr_vir, aes_src, data_len)) {
ERR_CODE_CIPHER("cipher decrypt, memcmp failed!\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
TEST_END_PASS();
__CIPHER_EXIT__:
if (XMEDIA_NULL != input_addr_vir) {
free(input_addr_vir);
input_addr_vir = XMEDIA_NULL;
}
if (XMEDIA_NULL != output_addr_vir) {
free(output_addr_vir);
output_addr_vir = XMEDIA_NULL;
}
xmedia_cipher_destroy_handle(chnid);
xmedia_cipher_exit();
return ret;
}
static xmedia_s32 cfb_aes128(xmedia_void)
{
xmedia_s32 ret = XMEDIA_SUCCESS;
xmedia_u32 data_len = 32;
xmedia_u8 *input_addr_vir = XMEDIA_NULL;
xmedia_u8 *output_addr_vir = XMEDIA_NULL;
xmedia_handle chnid = 0;
xmedia_u8 aes_key[16] = {"\x2B\x7E\x15\x16\x28\xAE\xD2\xA6\xAB\xF7\x15\x88\x09\xCF\x4F\x3C"};
xmedia_u8 aes_iv[16] = {"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F"};
xmedia_u8 aes_src[32] = {"\x6B\xC1\xBE\xE2\x2E\x40\x9F\x96\xE9\x3D\x7E\x11\x73\x93\x17\x2A\xAE\x2D\x8A\x57\x1E\x03\xAC\x9C\x9E\xB7\x6F\xAC\x45\xAF\x8E\x51"};
xmedia_u8 aes_dst[32] = {"\x3B\x3F\xD9\x2E\xB7\x2D\xAD\x20\x33\x34\x49\xF8\xE8\x3C\xFB\x4A\xC8\xA6\x45\x37\xA0\xB3\xA9\x3F\xCD\xE3\xCD\xAD\x9F\x1C\xE5\x8B"};
ret = xmedia_cipher_init();
if (XMEDIA_SUCCESS != ret) {
return ret;
}
ret = xmedia_cipher_create_handle(&chnid);
if (XMEDIA_SUCCESS != ret) {
xmedia_cipher_exit();
return ret;
}
input_addr_vir = (xmedia_u8*)malloc(data_len);
if (input_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc input_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(input_addr_vir, 0, data_len);
output_addr_vir = (xmedia_u8*)malloc(data_len);
if (output_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc output_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(output_addr_vir, 0, data_len);
/* For encrypt */
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CFB,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
memset(input_addr_vir, 0x0, data_len);
memcpy(input_addr_vir, aes_src, data_len);
print_buffer("CFB-AES-128-ORI:", aes_src, data_len);
memset(output_addr_vir, 0x0, data_len);
ret = xmedia_cipher_encrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher encrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CFB-AES-128-ENC:", output_addr_vir, data_len);
/* compare */
if (0 != memcmp(output_addr_vir, aes_dst, data_len)) {
ERR_CODE_CIPHER("Memcmp failed!\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
/* For decrypt */
memcpy(input_addr_vir, aes_dst, data_len);
memset(output_addr_vir, 0x0, data_len);
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CFB,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
ret = xmedia_cipher_decrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher decrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CFB-AES-128-DEC", output_addr_vir, data_len);
/* compare */
if (0 != memcmp(output_addr_vir, aes_src, data_len)) {
ERR_CODE_CIPHER("Memcmp failed!\n");
goto __CIPHER_EXIT__;
}
TEST_END_PASS();
__CIPHER_EXIT__:
if (XMEDIA_NULL != input_addr_vir) {
free(input_addr_vir);
input_addr_vir = XMEDIA_NULL;
}
if (XMEDIA_NULL != output_addr_vir) {
free(output_addr_vir);
output_addr_vir = XMEDIA_NULL;
}
xmedia_cipher_destroy_handle(chnid);
xmedia_cipher_exit();
return ret;
}
static xmedia_s32 ctr_aes128(xmedia_void)
{
xmedia_s32 ret = XMEDIA_SUCCESS;
xmedia_u32 data_len = 32;
xmedia_u8 *input_addr_vir = XMEDIA_NULL;
xmedia_u8 *output_addr_vir = XMEDIA_NULL;
xmedia_handle chnid = 0;
xmedia_u8 aes_key[16] = {"\x7E\x24\x06\x78\x17\xFA\xE0\xD7\x43\xD6\xCE\x1F\x32\x53\x91\x63"};
xmedia_u8 aes_iv[16] = {"\x00\x6C\xB6\xDB\xC0\x54\x3B\x59\xDA\x48\xD9\x0B\x00\x00\x00\x01"};
xmedia_u8 aes_src[32] = {"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1A\x1B\x1C\x1D\x1E\x1F"};
xmedia_u8 aes_dst[32] = {"\x51\x04\xA1\x06\x16\x8A\x72\xD9\x79\x0D\x41\xEE\x8E\xDA\xD3\x88\xEB\x2E\x1E\xFC\x46\xDA\x57\xC8\xFC\xE6\x30\xDF\x91\x41\xBE\x28"};
ret = xmedia_cipher_init();
if (XMEDIA_SUCCESS != ret) {
return ret;
}
ret = xmedia_cipher_create_handle(&chnid);
if (XMEDIA_SUCCESS != ret) {
xmedia_cipher_exit();
return ret;
}
input_addr_vir = (xmedia_u8*)malloc(data_len);
if (input_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc input_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(input_addr_vir, 0, data_len);
output_addr_vir = (xmedia_u8*)malloc(data_len);
if (output_addr_vir == XMEDIA_NULL) {
ERR_CODE_CIPHER("malloc output_addr_vir failed\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
memset(output_addr_vir, 0, data_len);
/* For encrypt */
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CTR,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
memcpy(input_addr_vir, aes_src, data_len);
print_buffer("CTR-AES-128-ORI:", aes_src, data_len);
ret = xmedia_cipher_encrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher encrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CTR-AES-128-ENC:", output_addr_vir, data_len);
/* compare */
if (0 != memcmp(output_addr_vir, aes_dst, data_len)) {
ERR_CODE_CIPHER("Memcmp failed!\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
/* For decrypt */
memcpy(input_addr_vir, aes_dst, data_len);
memset(output_addr_vir, 0x0, data_len);
ret = set_config_info(chnid,
XMEDIA_FALSE,
XMEDIA_CIPHER_ALG_AES,
XMEDIA_CIPHER_WORK_MODE_CTR,
XMEDIA_CIPHER_KEY_AES_128BIT,
aes_key,
aes_iv);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Set config info failed.\n");
goto __CIPHER_EXIT__;
}
ret = xmedia_cipher_decrypt_vir(chnid, input_addr_vir, output_addr_vir, data_len);
if (XMEDIA_SUCCESS != ret) {
ERR_CODE_CIPHER("Cipher decrypt failed.\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
print_buffer("CTR-AES-128-DEC", output_addr_vir, data_len);
/* compare */
if (0 != memcmp(output_addr_vir, aes_src, data_len)) {
ERR_CODE_CIPHER("Memcmp failed!\n");
ret = XMEDIA_FAILURE;
goto __CIPHER_EXIT__;
}
TEST_END_PASS();
__CIPHER_EXIT__:
if (XMEDIA_NULL != input_addr_vir) {
free(input_addr_vir);
input_addr_vir = XMEDIA_NULL;
}
if (XMEDIA_NULL != output_addr_vir) {
free(output_addr_vir);
output_addr_vir = XMEDIA_NULL;
}
xmedia_cipher_destroy_handle(chnid);
xmedia_cipher_exit();
return ret;
}
int sample_cipher_via(xmedia_void)
{
xmedia_s32 ret = XMEDIA_SUCCESS;
ret = cbc_aes128();
if (ret != XMEDIA_SUCCESS) {
return ret;
}
ret = cfb_aes128();
if (ret != XMEDIA_SUCCESS) {
return ret;
}
ret = ctr_aes128();
if (ret != XMEDIA_SUCCESS) {
return ret;
}
return XMEDIA_SUCCESS;
}
Reference in New Issue View Git Blame Copy Permalink