147 lines
4.0 KiB
C
147 lines
4.0 KiB
C
/* *
|
|
* Copyright (c) CompanyNameMagicTag 2018-2019. All rights reserved.
|
|
* Description: Upgrade base funciton - tool.
|
|
* Author: CompanyName
|
|
* Create: 2018-10-31
|
|
*/
|
|
|
|
#include "upg_base_tool.h"
|
|
#include <soc_mdm_types.h>
|
|
#include <soc_mdm_flash.h>
|
|
#include <soc_mdm_crc.h>
|
|
#include <soc_stdlib.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#define EXT_BLOCK_SIZE 0x100
|
|
td_u8 g_upg_tool_block[EXT_BLOCK_SIZE];
|
|
|
|
EXT_PRV td_u32 uapi_hash_flash_content_extend(const upg_rsa_verify_param *param, td_u8 *hash_result,
|
|
td_u32 hash_result_size);
|
|
|
|
td_u32 uapi_crc32_flash_content(td_u32 crc_start, td_u32 flash_addr, td_u32 total_size, td_u32 *crc_result)
|
|
{
|
|
td_u8 *block = g_upg_tool_block;
|
|
td_u32 offset;
|
|
td_u32 this_size;
|
|
td_u32 ret;
|
|
if (crc_result == TD_NULL) {
|
|
return EXT_ERR_INVALID_PARAMETER;
|
|
}
|
|
|
|
for (offset = 0; offset < total_size;) {
|
|
this_size = (total_size - offset > EXT_BLOCK_SIZE) ? EXT_BLOCK_SIZE : (total_size - offset);
|
|
|
|
ret = uapi_flash_read(flash_addr + offset, this_size, block);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
crc_start = uapi_crc32(crc_start, block, this_size);
|
|
offset += this_size;
|
|
}
|
|
|
|
*crc_result = crc_start;
|
|
|
|
return EXT_ERR_SUCCESS;
|
|
}
|
|
|
|
td_u32 uapi_hash_flash_content(td_u32 flash_addr, td_u32 total_size, td_u8 *hash_result, td_u32 hash_result_size)
|
|
{
|
|
td_u8 *block = g_upg_tool_block;
|
|
td_u32 offset;
|
|
td_u32 this_size;
|
|
td_u32 ret;
|
|
td_u16 hash_handle = 0;
|
|
|
|
ext_unref_param(hash_result_size);
|
|
|
|
ret = uapi_hash_start(&hash_handle);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
for (offset = 0; offset < total_size;) {
|
|
this_size = (total_size - offset > EXT_BLOCK_SIZE) ? EXT_BLOCK_SIZE : (total_size - offset);
|
|
|
|
ret = uapi_flash_read(flash_addr + offset, this_size, block);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
ret = uapi_hash_update(hash_handle, block, this_size);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
offset += this_size;
|
|
}
|
|
|
|
return uapi_hash_final(hash_handle, hash_result);
|
|
}
|
|
|
|
EXT_PRV td_u32 uapi_hash_flash_content_extend(const upg_rsa_verify_param *param, td_u8 *hash_result,
|
|
td_u32 hash_result_size)
|
|
{
|
|
td_u8 *block = g_upg_tool_block;
|
|
uintptr_t offset;
|
|
uintptr_t this_size;
|
|
td_u32 ret;
|
|
td_u32 i;
|
|
td_u16 hash_handle = 0;
|
|
|
|
ext_unref_param(hash_result_size);
|
|
|
|
if (param == TD_NULL) {
|
|
return EXT_ERR_FAILURE;
|
|
}
|
|
|
|
ret = uapi_hash_start(&hash_handle);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
for (i = 0; i < param->buffer_count; i++) {
|
|
for (offset = 0; offset < param->total_size[i];) {
|
|
this_size =
|
|
(param->total_size[i] - offset > EXT_BLOCK_SIZE) ? EXT_BLOCK_SIZE : (param->total_size[i] - offset);
|
|
|
|
if (param->flash_flag[i]) {
|
|
uapi_flash_read(param->flash_addr[i] + offset, this_size, block);
|
|
} else {
|
|
ret = (td_u32)memcpy_s(block, EXT_BLOCK_SIZE, (char *)(param->flash_addr[i] + offset), this_size);
|
|
ext_unref_param(ret);
|
|
}
|
|
|
|
ret = uapi_hash_update(hash_handle, block, this_size);
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
offset += this_size;
|
|
}
|
|
}
|
|
|
|
return uapi_hash_final(hash_handle, hash_result);
|
|
}
|
|
|
|
td_u32 uapi_rsa_verify_flash_content(const upg_rsa_verify_param *param)
|
|
{
|
|
td_u32 ret;
|
|
td_u8 hash_result[EXT_SHA256_SIZE] = { 0 };
|
|
|
|
/* calculate SHA256 */
|
|
ret = uapi_hash_flash_content_extend(param, hash_result, sizeof(hash_result));
|
|
if (ret != EXT_ERR_SUCCESS) {
|
|
return ret;
|
|
}
|
|
|
|
/* Verify signature. */
|
|
return uapi_rsa_verify(param->key_n, param->key_e, param->key_len, param->pad_mode, hash_result, param->sign);
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|