This repository has been archived on 2023-07-17. You can view files and clone it, but cannot push or open issues or pull requests.
bl_mcu_sdk/examples/boot2_iap/blsp_media_boot.c
2021-06-20 12:25:46 +08:00

397 lines
12 KiB
C

/**
******************************************************************************
* @file blsp_media_boot.c
* @version V1.2
* @date
* @brief This file is the peripheral case c file
******************************************************************************
* @attention
*
* <h2><center>&copy; COPYRIGHT(c) 2018 Bouffalo Lab</center></h2>
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* 3. Neither the name of Bouffalo Lab nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
******************************************************************************
*/
#include "stdint.h"
#include "string.h"
#include "blsp_port.h"
#include "bflb_platform.h"
#include "blsp_bootinfo.h"
#include "blsp_common.h"
#include "blsp_boot_parser.h"
#include "blsp_media_boot.h"
#include "softcrc.h"
#include "bflb_eflash_loader_uart.h"
/** @addtogroup BL606_BLSP_Boot2
* @{
*/
/** @addtogroup BLSP_MEDIA_BOOT
* @{
*/
/** @defgroup BLSP_MEDIA_BOOT_Private_Macros
* @{
*/
/*@} end of group BLSP_MEDIA_BOOT_Private_Macros */
/** @defgroup BLSP_MEDIA_BOOT_Private_Types
* @{
*/
/*@} end of group BLSP_MEDIA_BOOT_Private_Types */
/** @defgroup BLSP_MEDIA_BOOT_Private_Variables
* @{
*/
/*@} end of group BLSP_MEDIA_BOOT_Private_Variables */
/** @defgroup BLSP_MEDIA_BOOT_Global_Variables
* @{
*/
extern SEC_Eng_SHA256_Ctx g_sha_ctx;
/*@} end of group BLSP_MEDIA_BOOT_Global_Variables */
/** @defgroup BLSP_MEDIA_BOOT_Private_Fun_Declaration
* @{
*/
/*@} end of group BLSP_MEDIA_BOOT_Private_Fun_Declaration */
/** @defgroup BLSP_MEDIA_BOOT_Private_Functions_User_Define
* @{
*/
/*@} end of group BLSP_MEDIA_BOOT_Private_Functions_User_Define */
/** @defgroup BLSP_MEDIA_BOOT_Private_Functions
* @{
*/
/****************************************************************************/ /**
* @brief Media boot calculate hash
*
* @param startAddr: Start address to calculate
* @param totalLen: Data length to calculate
*
* @return BL_Err_Type
*
*******************************************************************************/
static int32_t blsp_mediaboot_cal_hash(uint32_t start_addr, uint32_t total_len)
{
int32_t deal_len = 0;
int32_t read_len = 0;
uint32_t addr = start_addr;
int32_t ret;
uint32_t dump_cnt = 0;
while (deal_len < total_len) {
read_len = total_len - deal_len;
if (read_len > BFLB_BOOT2_READBUF_SIZE) {
read_len = BFLB_BOOT2_READBUF_SIZE;
}
ret = blsp_mediaboot_read(addr, g_boot2_read_buf, read_len);
if (ret != BFLB_BOOT2_SUCCESS) {
return ret;
}
if (dump_cnt == 0) {
if (blsp_boot2_dump_critical_flag()) {
blsp_dump_data(g_boot2_read_buf, read_len);
}
dump_cnt = 1;
}
/* Update hash*/
Sec_Eng_SHA256_Update(&g_sha_ctx, SEC_ENG_SHA_ID0, (uint8_t *)g_boot2_read_buf, read_len);
//blsp_dump_data((uint8_t *)g_boot2_read_buf,readLen);
addr += read_len;
deal_len += read_len;
}
return BFLB_BOOT2_SUCCESS;
}
/****************************************************************************/ /**
* @brief Media boot read signature
*
* @param addr: Start address to read signature
* @param len: Signature length pointer
*
* @return BL_Err_Type
*
*******************************************************************************/
#if 0
static int32_t BLSP_MediaBoot_Read_Signaure(uint32_t addr, uint32_t *len)
{
int32_t ret = BFLB_BOOT2_SUCCESS;
uint32_t sig_len = 0;
uint8_t *ptmp;
/* Read signature*/
ret = blsp_mediaboot_read(addr, (uint8_t *)&sig_len, sizeof(sig_len));
if(ret != BFLB_BOOT2_SUCCESS)
{
return ret;
}
addr += sizeof(sig_len);
if(sig_len > BFLB_BOOT2_SIGN_MAXSIZE)
{
return BFLB_BOOT2_IMG_SIGNATURE_LEN_ERROR;
}
/* Tail 4 bytes for crc */
ptmp = ((uint8_t *)g_boot2_read_buf);
ret = blsp_mediaboot_read(addr, (uint8_t *)(ptmp + sizeof(sig_len)), sig_len + 4);
if(ret != BFLB_BOOT2_SUCCESS)
{
return ret;
}
ARCH_MemCpy_Fast(g_boot2_read_buf, &sig_len, sizeof(sig_len));
addr += (sig_len + 4);
*len = sig_len;
return ret;
}
#endif
/****************************************************************************/ /**
* @brief Media boot parse one firmware
*
* @param g_boot_img_cfg: Boot image config pointer to hold parsed information
* @param bootheaderAddr: Bootheader start address
* @param imgAddr: Image start address
*
* @return BL_Err_Type
*
*******************************************************************************/
static int32_t blsp_mediaboot_parse_one_fw(boot_image_config *boot_img_cfg, uint32_t boot_header_addr, uint32_t img_addr)
{
uint32_t addr = boot_header_addr;
int32_t ret;
// uint32_t sig_len=0;
/* Read boot header*/
MSG_DBG("R header from %08x\r\n", addr);
ret = blsp_mediaboot_read(addr, g_boot2_read_buf, sizeof(boot_header_config));
if (ret != BFLB_BOOT2_SUCCESS) {
return ret;
}
if (blsp_boot2_dump_critical_flag()) {
blsp_dump_data(g_boot2_read_buf, sizeof(boot_header_config));
}
addr += sizeof(boot_header_config);
ret = blsp_boot_parse_bootheader(boot_img_cfg, (uint8_t *)g_boot2_read_buf);
if (ret != BFLB_BOOT2_SUCCESS) {
return ret;
}
/* Due to OTA, the flash_offset is changed, so copy from partition info */
boot_img_cfg->img_start.flash_offset = img_addr;
if (boot_img_cfg->no_segment) {
/* Flash image */
if (!boot_img_cfg->hash_ignore) {
MSG_DBG("Cal hash\r\n");
ret = blsp_mediaboot_cal_hash(img_addr,
boot_img_cfg->img_segment_info.img_len);
if (ret != BFLB_BOOT2_SUCCESS) {
return ret;
}
ret = blsp_boot_parser_check_hash(boot_img_cfg);
if (ret != BFLB_BOOT2_SUCCESS) {
return ret;
}
}
// ret=blsp_boot_parser_check_signature(g_boot_img_cfg);
// if(ret!=BFLB_BOOT2_SUCCESS){
// return ret;
// }
boot_img_cfg->img_valid = 1;
} else {
boot_img_cfg->img_valid = 0;
}
return ret;
}
/*@} end of group BLSP_MEDIA_BOOT_Private_Functions */
/** @defgroup BLSP_MEDIA_BOOT_Public_Functions
* @{
*/
/****************************************************************************/ /**
* @brief Media boot read data
*
* @param addr: Start address to read
* @param data: Data buffer to hold read data
* @param len: Data length to read
*
* @return BL_Err_Type
*
*******************************************************************************/
int32_t ATTR_TCM_SECTION blsp_mediaboot_read(uint32_t addr, uint8_t *data, uint32_t len)
{
XIP_SFlash_Read_Via_Cache_Need_Lock(BLSP_BOOT2_XIP_BASE + addr, data, len);
return BFLB_BOOT2_SUCCESS;
}
/****************************************************************************/ /**
* @brief Media boot main process
*
* @param cpuBootheaderAddr[BFLB_BOOT2_CPU_MAX]: CPU bootheader address list
* @param cpuRollBack[BFLB_BOOT2_CPU_MAX]: CPU need roll back flag hold list
* @param rollBack: 1 for rollback when imge error occurs, 0 for not rollback when imge error occurs
*
* @return BL_Err_Type
*
*******************************************************************************/
int32_t blsp_mediaboot_main(uint32_t cpu_boot_header_addr[BFLB_BOOT2_CPU_MAX], uint8_t cpu_roll_back[BFLB_BOOT2_CPU_MAX], uint8_t roll_back)
{
int32_t ret;
uint32_t i = 0;
uint32_t valid_img_found = 0;
uint32_t boot_header_addr[BFLB_BOOT2_CPU_MAX];
MSG_DBG("Media boot main\r\n");
/* Reset some parameters*/
for (i = 0; i < BFLB_BOOT2_CPU_MAX; i++) {
memset(&g_boot_img_cfg[i], 0, sizeof(g_boot_img_cfg[i]));
boot_header_addr[i] = cpu_boot_header_addr[i];
cpu_roll_back[i] = 0;
}
g_boot_img_cfg[0].halt_cpu1 = 0;
/* Try to boot from flash */
for (i = 0; i < g_cpu_count; i++) {
if (boot_header_addr[i] == 0) {
MSG_ERR("CPU %d not boot\r\n", i);
continue;
}
ret = blsp_mediaboot_parse_one_fw(&g_boot_img_cfg[i], boot_header_addr[i],
boot_header_addr[i] + BFLB_FW_IMG_OFFSET_AFTER_HEADER);
if (ret != BFLB_BOOT2_SUCCESS) {
MSG_ERR("CPU %d boot fail\r\n", i);
cpu_roll_back[i] = 1;
} else {
valid_img_found++;
}
}
if (valid_img_found != g_cpu_count && 1 == roll_back) {
/* For CP and DP, found CPU0 image is taken as correct when the other not found, others as wrong and try to rollback */
if (boot_header_addr[1] == 0 && valid_img_found == 1) {
MSG_DBG("Found One img Only\r\n");
} else {
MSG_ERR("Image roll back\r\n");
return BFLB_BOOT2_IMG_Roll_Back;
}
}
if (valid_img_found == 0) {
MSG_ERR("no valid img found\r\n");
return BFLB_BOOT2_IMG_ALL_INVALID_ERROR;
}
/* Get msp and pc value */
for (i = 0; i < g_cpu_count; i++) {
if (g_boot_img_cfg[i].img_valid) {
if (g_boot_img_cfg[i].entry_point == 0) {
#ifdef ARCH_ARM
blsp_mediaboot_read(g_boot_img_cfg[i].img_start.flash_offset,
(uint8_t *)&g_boot_img_cfg[i].msp_val, 4);
blsp_mediaboot_read(g_boot_img_cfg[i].img_start.flash_offset + 4,
(uint8_t *)&g_boot_img_cfg[i].entry_point, 4);
#endif
#ifdef ARCH_RISCV
g_boot_img_cfg[i].entry_point = g_boot_cpu_cfg[i].default_xip_addr;
#endif
}
}
}
if (blsp_boot2_get_feature_flag() == BLSP_BOOT2_CP_FLAG) {
/*co-processor*/
g_boot_img_cfg[1].img_start.flash_offset = g_boot_img_cfg[0].img_start.flash_offset;
g_boot_img_cfg[1].msp_val = g_boot_img_cfg[0].msp_val;
g_boot_img_cfg[1].entry_point = g_boot_img_cfg[0].entry_point;
g_boot_img_cfg[1].cache_enable = g_boot_img_cfg[0].cache_enable;
g_boot_img_cfg[1].img_valid = 1;
g_boot_img_cfg[1].cache_way_disable = 0xf;
}
MSG_DBG("%08x,%08x\r\n", g_boot_img_cfg[0].msp_val, g_boot_img_cfg[0].entry_point);
MSG_DBG("%08x,%08x\r\n", g_boot_img_cfg[1].msp_val, g_boot_img_cfg[1].entry_point);
MSG_DBG("%08x,%08x\r\n", g_boot_img_cfg[0].img_start.flash_offset, g_boot_img_cfg[0].cache_way_disable);
MSG_DBG("%08x,%08x\r\n", g_boot_img_cfg[1].img_start.flash_offset, g_boot_img_cfg[1].cache_way_disable);
MSG_DBG("CPU Count %d,%d\r\n", g_cpu_count, g_boot_img_cfg[0].halt_cpu1);
blsp_boot2_show_timer();
/* Fix invalid pc and msp */
blsp_fix_invalid_msp_pc();
if (BFLB_EFLASH_LOADER_HANDSHAKE_SUSS == bflb_eflash_loader_uart_handshake_poll()) {
bflb_eflash_loader_main();
}
/* Prepare jump to entry*/
blsp_mediaboot_pre_jump();
/* We should never get here unless something is wrong */
return BFLB_BOOT2_FAIL;
}
/*@} end of group BLSP_MEDIA_BOOT_Public_Functions */
/*@} end of group BLSP_MEDIA_BOOT */
/*@} end of group BL606_BLSP_Boot2 */