450 lines
14 KiB
C
450 lines
14 KiB
C
/*
|
|
* Copyright (c) 2006-2021, RT-Thread Development Team
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Change Logs:
|
|
* Date Author Notes
|
|
* 2020-06-27 thread-liu first version
|
|
*/
|
|
|
|
#include <board.h>
|
|
|
|
#include "drv_crypto.h"
|
|
#include <hwcrypto.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#define __is_print(ch) ((unsigned int)((ch) - ' ') < 127u - ' ')
|
|
static void dump_hex(const rt_uint8_t *ptr, rt_size_t buflen)
|
|
{
|
|
unsigned char *buf = (unsigned char *)ptr;
|
|
int i, j;
|
|
|
|
for (i = 0; i < buflen; i += 16)
|
|
{
|
|
rt_kprintf("%08X: ", i);
|
|
|
|
for (j = 0; j < 16; j++)
|
|
{
|
|
if (i + j < buflen)
|
|
{
|
|
rt_kprintf("%02X ", buf[i + j]);
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf(" ");
|
|
}
|
|
}
|
|
rt_kprintf(" ");
|
|
|
|
for (j = 0; j < 16; j++)
|
|
{
|
|
if (i + j < buflen)
|
|
{
|
|
rt_kprintf("%c", __is_print(buf[i + j]) ? buf[i + j] : '.');
|
|
}
|
|
}
|
|
rt_kprintf("\n");
|
|
}
|
|
}
|
|
|
|
#if defined(BSP_USING_RNG)
|
|
static rt_err_t hw_rng_sample(int random_num)
|
|
{
|
|
rt_err_t result = RT_EOK;
|
|
int i = 0, num0 = 0, num1 = 0;
|
|
|
|
if (random_num == 0)
|
|
{
|
|
return -RT_ERROR;
|
|
}
|
|
|
|
for (i = 0; i< random_num; i++)
|
|
{
|
|
result = rt_hwcrypto_rng_update();
|
|
rt_kprintf("%d ", result);
|
|
result%2 ? num1++ : num0++;
|
|
}
|
|
rt_kprintf("\neven numbers : %d, odd numbers: %d\n",num1, num0);
|
|
|
|
return RT_EOK;
|
|
}
|
|
#endif
|
|
|
|
#if defined(BSP_USING_CRC)
|
|
static void hw_crc_sample(uint8_t *temp, int size)
|
|
{
|
|
struct rt_hwcrypto_ctx *ctx;
|
|
rt_uint32_t result = 0;
|
|
|
|
struct hwcrypto_crc_cfg cfg =
|
|
{
|
|
.last_val = 0xFFFFFFFF,
|
|
.poly = 0x04C11DB7,
|
|
.width = 32,
|
|
.xorout = 0x00000000,
|
|
.flags = 0,
|
|
};
|
|
|
|
ctx = rt_hwcrypto_crc_create(rt_hwcrypto_dev_default(), HWCRYPTO_CRC_CRC32);
|
|
rt_hwcrypto_crc_cfg(ctx, &cfg);
|
|
|
|
result = rt_hwcrypto_crc_update(ctx, temp, size);
|
|
|
|
rt_kprintf("crc result: %x \n", result);
|
|
|
|
rt_hwcrypto_crc_destroy(ctx);
|
|
}
|
|
#endif
|
|
|
|
#if defined(BSP_USING_HASH)
|
|
static void hw_hash_sample()
|
|
{
|
|
struct rt_hwcrypto_ctx *ctx = RT_NULL;
|
|
const uint8_t hash_input[] = "RT-Thread was born in 2006, it is an open source, neutral, and community-based real-time operating system (RTOS).";
|
|
|
|
static uint8_t sha1_output[20];
|
|
static uint8_t sha1_except[20] = {0xff, 0x3c, 0x95, 0x54, 0x95, 0xf0, 0xad,
|
|
0x02, 0x1b, 0xa8, 0xbc, 0xa2, 0x2e, 0xa5,
|
|
0xb0, 0x62, 0x1b, 0xdf, 0x7f, 0xec};
|
|
|
|
static uint8_t md5_output[16];
|
|
static uint8_t md5_except[16] = {0x40, 0x86, 0x03, 0x80, 0x0d, 0x8c, 0xb9,
|
|
0x4c, 0xd6, 0x7d, 0x28, 0xfc, 0xf6, 0xc3,
|
|
0xac, 0x8b};
|
|
|
|
static uint8_t sha224_output[28];
|
|
static uint8_t sha224_except[28] = {0x6f, 0x62, 0x52, 0x7d, 0x80, 0xe6,
|
|
0x9f, 0x82, 0x78, 0x7a, 0x46, 0x91,
|
|
0xb0, 0xe9, 0x64, 0x89, 0xe6, 0xc3,
|
|
0x6b, 0x7e, 0xcf, 0xca, 0x11, 0x42,
|
|
0xc8, 0x77, 0x13, 0x79};
|
|
static uint8_t sha256_output[32];
|
|
static uint8_t sha256_except[32] = {0x74, 0x19, 0xb9, 0x0e, 0xd1, 0x46,
|
|
0x37, 0x0a, 0x55, 0x18, 0x26, 0x6c,
|
|
0x50, 0xd8, 0x71, 0x34, 0xfa, 0x1f,
|
|
0x5f, 0x5f, 0xe4, 0x9a, 0xe9, 0x40,
|
|
0x0a, 0x7d, 0xa0, 0x26, 0x1b, 0x86,
|
|
0x67, 0x45};
|
|
rt_kprintf("======================== Hash Test start ========================\n");
|
|
rt_kprintf("Hash Test string: \n");
|
|
dump_hex(hash_input, sizeof(hash_input));
|
|
|
|
/* sh1 test*/
|
|
rt_kprintf("\n============ SHA1 Test Start ============\n");
|
|
ctx = rt_hwcrypto_hash_create(rt_hwcrypto_dev_default(), HWCRYPTO_TYPE_SHA1);
|
|
if (ctx == RT_NULL)
|
|
{
|
|
rt_kprintf("create hash[%08x] context err!\n", HWCRYPTO_TYPE_SHA1);
|
|
return ;
|
|
}
|
|
rt_kprintf("Create sha1 type success!\n");
|
|
rt_kprintf("Except sha1 result:\n");
|
|
dump_hex(sha1_except, sizeof(sha1_except));
|
|
|
|
/* start sha1 */
|
|
rt_hwcrypto_hash_update(ctx, hash_input, rt_strlen((char const *)hash_input));
|
|
/* get sha1 result */
|
|
rt_hwcrypto_hash_finish(ctx, sha1_output, rt_strlen((char const *)sha1_output));
|
|
|
|
rt_kprintf("Actual sha1 result:\n");
|
|
dump_hex(sha1_output, sizeof(sha1_output));
|
|
|
|
if(rt_memcmp(sha1_output, sha1_except, sizeof(sha1_except)/sizeof(sha1_except[0])) != 0)
|
|
{
|
|
rt_kprintf("Hash type sha1 Test error, The actual result is not equal to the except result\n");
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Hash type sha1 Test success, The actual result is equal to the except result\n");
|
|
}
|
|
/* deinit hash*/
|
|
rt_hwcrypto_hash_destroy(ctx);
|
|
rt_kprintf("============ SHA1 Test Over ============\n");
|
|
|
|
/* md5 test*/
|
|
rt_kprintf("\n============ MD5 Test Start ============\n");
|
|
ctx = rt_hwcrypto_hash_create(rt_hwcrypto_dev_default(), HWCRYPTO_TYPE_MD5);
|
|
if (ctx == RT_NULL)
|
|
{
|
|
rt_kprintf("create hash[%08x] context err!\n", HWCRYPTO_TYPE_MD5);
|
|
return ;
|
|
}
|
|
rt_kprintf("Create md5 type success!\n");
|
|
rt_kprintf("Except md5 result:\n");
|
|
dump_hex(md5_except, sizeof(md5_except));
|
|
|
|
/* start md5 */
|
|
rt_hwcrypto_hash_update(ctx, hash_input, rt_strlen((char const *)hash_input));
|
|
/* get md5 result */
|
|
rt_hwcrypto_hash_finish(ctx, md5_output, rt_strlen((char const *)md5_output));
|
|
|
|
rt_kprintf("Actual md5 result:\n");
|
|
dump_hex(md5_output, sizeof(md5_output));
|
|
|
|
if(rt_memcmp(md5_output, md5_except, sizeof(md5_except)/sizeof(md5_except[0])) != 0)
|
|
{
|
|
rt_kprintf("Hash type md5 Test error, The actual result is not equal to the except result\n");
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Hash type md5 Test success, The actual result is equal to the except result\n");
|
|
}
|
|
/* deinit hash*/
|
|
rt_hwcrypto_hash_destroy(ctx);
|
|
rt_kprintf("============ MD5 Test Over ============\n");
|
|
|
|
/* sha224 test */
|
|
rt_kprintf("\n============ SHA224 Test Start ============\n");
|
|
ctx = rt_hwcrypto_hash_create(rt_hwcrypto_dev_default(), HWCRYPTO_TYPE_SHA224);
|
|
if (ctx == RT_NULL)
|
|
{
|
|
rt_kprintf("create hash[%08x] context err!\n", HWCRYPTO_TYPE_SHA224);
|
|
return ;
|
|
}
|
|
rt_kprintf("Create sha224 type success!\n");
|
|
rt_kprintf("Except sha224 result:\n");
|
|
dump_hex(sha224_except, sizeof(sha224_except));
|
|
|
|
/* start sha224 */
|
|
rt_hwcrypto_hash_update(ctx, hash_input, rt_strlen((char const *)hash_input));
|
|
/* get sha224 result */
|
|
rt_hwcrypto_hash_finish(ctx, sha224_output, rt_strlen((char const *)sha224_output));
|
|
|
|
rt_kprintf("Actual sha224 result:\n");
|
|
dump_hex(sha224_output, sizeof(sha224_output));
|
|
|
|
if(rt_memcmp(sha224_output, sha224_except, sizeof(sha224_except)/sizeof(sha224_except[0])) != 0)
|
|
{
|
|
rt_kprintf("Hash type sha224 Test error, The actual result is not equal to the except result\n");
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Hash type sha224 Test success, The actual result is equal to the except result\n");
|
|
}
|
|
rt_hwcrypto_hash_destroy(ctx);
|
|
rt_kprintf("============ SHA224 Test Over ============\n");
|
|
|
|
/* sha256 test*/
|
|
rt_kprintf("\n============ SHA256 Test Start ============\n");
|
|
ctx = rt_hwcrypto_hash_create(rt_hwcrypto_dev_default(), HWCRYPTO_TYPE_SHA256);
|
|
if (ctx == RT_NULL)
|
|
{
|
|
rt_kprintf("create hash[%08x] context err!\n", HWCRYPTO_TYPE_SHA256);
|
|
return ;
|
|
}
|
|
|
|
rt_kprintf("Create sha256 type success!\n");
|
|
rt_kprintf("Except sha256 result:\n");
|
|
dump_hex(sha256_except, sizeof(sha256_except));
|
|
|
|
/* start sha256 */
|
|
rt_hwcrypto_hash_update(ctx, hash_input, rt_strlen((char const *)hash_input));
|
|
/* get sha256 result */
|
|
rt_hwcrypto_hash_finish(ctx, sha256_output, rt_strlen((char const *)sha256_output));
|
|
|
|
rt_kprintf("Actual sha256 result\n");
|
|
dump_hex(sha256_output, sizeof(sha256_output));
|
|
|
|
if(rt_memcmp(sha256_output, sha256_except, sizeof(sha256_except)/sizeof(sha256_except[0])) != 0)
|
|
{
|
|
rt_kprintf("Hash type sha256 Test error, The actual result is not equal to the except result\n");
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Hash type sha256 Test success, The actual result is equal to the except result\n");
|
|
}
|
|
/* destory */
|
|
rt_hwcrypto_hash_destroy(ctx);
|
|
rt_kprintf("============ SHA256 Test Over ============\n");
|
|
rt_kprintf("======================== Hash Test over! ========================\n");
|
|
}
|
|
#endif
|
|
|
|
#if defined(BSP_USING_CRYP)
|
|
/* key*/
|
|
static const rt_uint8_t cryp_key[16] = {0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xA, 0xB, 0xC, 0xD, 0xE, 0xF};
|
|
|
|
static void hw_aes_cbc(const rt_uint8_t in[32], rt_uint8_t out[32], hwcrypto_mode mode)
|
|
{
|
|
struct rt_hwcrypto_ctx *ctx;
|
|
|
|
ctx = rt_hwcrypto_symmetric_create(rt_hwcrypto_dev_default(), HWCRYPTO_TYPE_AES_CBC);
|
|
if (ctx == RT_NULL)
|
|
{
|
|
rt_kprintf("create AES-CBC context err!");
|
|
return;
|
|
}
|
|
rt_hwcrypto_symmetric_setkey(ctx, cryp_key, 128);
|
|
rt_hwcrypto_symmetric_crypt(ctx, mode, 32, in, out);
|
|
rt_hwcrypto_symmetric_destroy(ctx);
|
|
}
|
|
|
|
static void hw_cryp_sample()
|
|
{
|
|
rt_uint8_t buf_in[32];
|
|
rt_uint8_t buf_out[32];
|
|
int i;
|
|
|
|
/* Populating test data */
|
|
for (i = 0; i < sizeof(buf_in); i++)
|
|
{
|
|
buf_in[i] = i;
|
|
}
|
|
|
|
/* dump primitive data */
|
|
rt_kprintf("key : \n");
|
|
dump_hex(cryp_key, sizeof(cryp_key));
|
|
rt_kprintf("primitive data : \n");
|
|
dump_hex(buf_in, sizeof(buf_in));
|
|
|
|
rt_memset(buf_out, 0, sizeof(buf_out));
|
|
|
|
/* encrypt */
|
|
hw_aes_cbc(buf_in, buf_out, HWCRYPTO_MODE_ENCRYPT);
|
|
/* dump encrypt data */
|
|
rt_kprintf("AES-enc : \n");
|
|
dump_hex(buf_out, sizeof(buf_out));
|
|
|
|
rt_memset(buf_in, 0, sizeof(buf_in));
|
|
|
|
/* decrypt */
|
|
hw_aes_cbc(buf_out, buf_in, HWCRYPTO_MODE_DECRYPT);
|
|
|
|
/* dump decrypt data */
|
|
rt_kprintf("AES-dec : \n");
|
|
dump_hex(buf_in, sizeof(buf_in));
|
|
}
|
|
#endif
|
|
|
|
static int crypto(int argc, char **argv)
|
|
{
|
|
int result = RT_EOK;
|
|
static rt_device_t device = RT_NULL;
|
|
char *result_str;
|
|
|
|
if (argc > 1)
|
|
{
|
|
if (!strcmp(argv[1], "probe"))
|
|
{
|
|
if (argc == 3)
|
|
{
|
|
char *dev_name = argv[2];
|
|
device = rt_device_find(dev_name);
|
|
result_str = (device == RT_NULL) ? "failure" : "success";
|
|
rt_kprintf("probe %s %s \n", argv[2], result_str);
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("crypto probe <crypto_name> - probe crypto by name\n");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (device == RT_NULL)
|
|
{
|
|
rt_kprintf("Please using 'crypto probe <crypto_name>' first\n");
|
|
return -RT_ERROR;
|
|
}
|
|
if (!strcmp(argv[1], "rng"))
|
|
{
|
|
#if defined (BSP_USING_RNG)
|
|
if (argc == 3)
|
|
{
|
|
result = hw_rng_sample(atoi(argv[2]));
|
|
if(result != RT_EOK)
|
|
{
|
|
rt_kprintf("please input a legal number, not <%d>\n", atoi(argv[2]));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("rng <number> - generate <number> digital\n");
|
|
}
|
|
|
|
#else
|
|
rt_kprintf("please enable RNG first!\n");
|
|
#endif
|
|
}
|
|
else if (!strcmp(argv[1], "crc"))
|
|
{
|
|
#if defined (BSP_USING_CRC)
|
|
int size = 0, i = 0;
|
|
if (argc > 3)
|
|
{
|
|
size = argc - 2;
|
|
uint8_t *data = rt_malloc(size);
|
|
if (data)
|
|
{
|
|
for (i = 0; i < size; i++)
|
|
{
|
|
data[i] = strtol(argv[2 + i], NULL, 0);
|
|
}
|
|
hw_crc_sample(data, size);
|
|
rt_free(data);
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Low memory!\n");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("crypto crc data1 ... dataN - calculate data1 ... dataN crc\n");
|
|
}
|
|
#else
|
|
rt_kprintf("please enable CRC first!\n");
|
|
#endif
|
|
}
|
|
else if (!strcmp(argv[1], "hash"))
|
|
{
|
|
#if defined (BSP_USING_HASH)
|
|
if (argc == 3)
|
|
{
|
|
hw_hash_sample();
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("crypto hash sample - hash use sample\n");
|
|
}
|
|
#else
|
|
rt_kprintf("please enable CRC first!\n");
|
|
#endif
|
|
}
|
|
else if (!strcmp(argv[1], "cryp"))
|
|
{
|
|
#if defined (BSP_USING_CRYP)
|
|
if (argc == 3)
|
|
{
|
|
hw_cryp_sample();
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("crypto cryp sample - encrypt and decrypt data sample\n");
|
|
}
|
|
#else
|
|
rt_kprintf("please enable CRYP first!\n");
|
|
#endif
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Unknown command. Please enter 'crypto' for help\n");
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_kprintf("Usage: \n");
|
|
rt_kprintf("crypto probe <crypto_name> - probe crypto by name\n");
|
|
rt_kprintf("crypto rng number - generate numbers digital\n");
|
|
rt_kprintf("crypto crc data1 ... dataN - calculate data1 ... dataN crc\n");
|
|
rt_kprintf("crypto hash sample - hash use sample\n");
|
|
rt_kprintf("crypto cryp sample - encrypt and decrypt data\n");
|
|
result = -RT_ERROR;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
MSH_CMD_EXPORT(crypto, crypto function);
|