dc32-retro-tech-addon/firmware/ch32v003_bootloader/user/iap.c

263 lines
6.6 KiB
C

/********************************** (C) COPYRIGHT *******************************
* File Name : iap.c
* Author : WCH
* Version : V1.0.0
* Date : 2023/12/21
* Description : IAP
*******************************************************************************
* Copyright (c) 2021 Nanjing Qinheng Microelectronics Co., Ltd.
* Attention: This software (modified or not) and binary are used for
* microcontroller manufactured by Nanjing Qinheng Microelectronics.
*******************************************************************************/
#include "iap.h"
#include "string.h"
#include "flash.h"
#include "core_riscv.h"
/******************************************************************************/
#define FLASH_Base 0x08000000
#define USBD_DATA_SIZE 64
u32 Program_addr = FLASH_Base;
u32 Verity_addr = FLASH_Base;
u8 Verity_Star_flag = 0;
u8 Fast_Program_Buf[128];
u16 CodeLen = 0;
u8 End_Flag = 0;
u8 EP2_Rx_Buffer[USBD_DATA_SIZE];
#define isp_cmd_t ((isp_cmd *)EP2_Rx_Buffer)
/*********************************************************************
* @fn USART1_CFG
*
* @brief GPIOD-USART1 init
*
* @return
*/
void USART1_CFG(u32 baudrate)
{
uint32_t w;
// configure GPIO pins for USART mode
GPIOD->BCR = (((uint32_t)0x01) << 6); // pull down RX line
w = GPIOD->CFGLR & ~(0x0ff00000);
GPIOD->CFGLR = w | 0x08a00000; // Set GPIOD Mode, Speed
// configure USART
USART1->CTLR2 = USART_StopBits_1;
USART1->CTLR1 = USART_Parity_No | USART_Mode_Rx | USART_Mode_Tx; // Set USART mode, Parity
USART1->CTLR3 = USART_HardwareFlowControl_None;
USART1->BRR = (13 << 4); // 115200; was originally 0x34 for 460800 supposedly?
USART1->CTLR1 |= ((uint16_t)0x2000); // enable USART
}
/*********************************************************************
* @fn RecData_Deal
*
* @brief UART-USB
*
* @return ERR_ERROR - ERROR
* ERR_SCUESS - SCUESS
* ERR_End - End
*/
u8 RecData_Deal(void)
{
u8 i, s, Lenth;
Lenth = isp_cmd_t->Len;
switch ( isp_cmd_t->Cmd) {
case CMD_IAP_ERASE:
FLASH_Unlock_Fast();
FLASH_EraseAllPages();
s = ERR_SCUESS;
break;
case CMD_IAP_PROM:
for (i = 0; i < Lenth; i++) {
Fast_Program_Buf[CodeLen + i] = isp_cmd_t->data[i];
}
CodeLen += Lenth;
if (CodeLen >= 64) {
CH32_IAP_Program(Program_addr, (u32*) Fast_Program_Buf);
CodeLen -= 64;
for (i = 0; i < CodeLen; i++) {
Fast_Program_Buf[i] = Fast_Program_Buf[64 + i];
}
Program_addr += 0x40;
}
s = ERR_SCUESS;
break;
case CMD_IAP_VERIFY:
if (Verity_Star_flag == 0) {
Verity_Star_flag = 1;
for (i = 0; i < (64 - CodeLen); i++) {
Fast_Program_Buf[CodeLen + i] = 0xFF;
}
CH32_IAP_Program(Program_addr, (u32*) Fast_Program_Buf);
CodeLen = 0;
}
s = ERR_SCUESS;
for (i = 0; i < Lenth; i++) {
if (isp_cmd_t->data[i] != *(u8*) (Verity_addr + i)) {
s = ERR_ERROR;
break;
}
}
Verity_addr += Lenth;
break;
case CMD_IAP_END:
Verity_Star_flag = 0;
End_Flag = 1;
Program_addr = FLASH_Base;
Verity_addr = FLASH_Base;
s = ERR_End;
FLASH->CTLR |= ((uint32_t)0x00008000);
FLASH->CTLR |= ((uint32_t)0x00000080);
break;
default:
s = ERR_ERROR;
break;
}
return s;
}
/*********************************************************************
* @fn GPIO_Cfg_init
*
* @brief GPIOC init
*
* @return none
*/
void GPIO_Cfg_init(void)
{
}
/*********************************************************************
* @fn UART3_SendMultiyData
*
* @brief Deal device Endpoint 3 OUT.
*
* @param l: Data length.
*
* @return none
*/
void UART1_SendMultiyData(u8* pbuf, u8 num)
{
u8 i = 0;
while(i<num)
{
while(USART_GetFlagStatus(USART1, USART_FLAG_TC) == RESET);
USART_SendData(USART1, pbuf[i]);
i++;
}
}
/*********************************************************************
* @fn UART1_SendMultiyData
*
* @brief USART3 send data
*
* @param pbuf - Packet to be sent
* num - Number of data sent
*
* @return none
*/
void UART1_SendData(u8 data)
{
while(USART_GetFlagStatus(USART1, USART_FLAG_TC) == RESET);
USART_SendData(USART1, data);
}
/*********************************************************************
* @fn Uart1_Rx
*
* @brief Uart3 receive date
*
* @return none
*/
u8 Uart1_Rx(void)
{
// todo: rewrite the whole receive routine to not depend
// on this easily breakable code.
while(USART_GetFlagStatus(USART1, USART_FLAG_RXNE) == RESET);
return USART_ReceiveData(USART1);
}
/*********************************************************************
* @fn UART_Rx_Deal
*
* @brief UART Rx data deal
*
* @return none
*/
void UART_Rx_Deal(void)
{
u8 i, s;
u8 Data_add = 0;
if (Uart1_Rx() == Uart_Sync_Head1)
{
if (Uart1_Rx() == Uart_Sync_Head2)
{
isp_cmd_t->Cmd = Uart1_Rx();
Data_add += isp_cmd_t->Cmd;
isp_cmd_t->Len = Uart1_Rx();
Data_add += isp_cmd_t->Len;
isp_cmd_t->Rev[0] = Uart1_Rx();
Data_add += isp_cmd_t->Rev[0];
isp_cmd_t->Rev[1] = Uart1_Rx();
Data_add += isp_cmd_t->Rev[1];
if ((isp_cmd_t->Cmd == CMD_IAP_PROM)
|| (isp_cmd_t->Cmd == CMD_IAP_VERIFY))
{
for (i = 0; i < isp_cmd_t->Len; i++) {
isp_cmd_t->data[i] = Uart1_Rx();
Data_add += isp_cmd_t->data[i];
}
}
if (Uart1_Rx() == Data_add)
{
s = RecData_Deal();
if (s != ERR_End)
{
UART1_SendData(0x00);
if (s == ERR_ERROR)
{
UART1_SendData(0x01);
}
else
{
UART1_SendData(0x00);
}
}
}
}
}
}