With bootloader

This commit is contained in:
lulko 2025-03-17 15:43:34 +03:00
parent b5ff05bed6
commit 023026987c
8 changed files with 312 additions and 141 deletions

View file

@ -1,110 +1,146 @@
#include <Arduino.h>
#include <STM32_CAN.h>
//#include "CRC32.h"
#include "flash.h"
//CRC
// put function declarations here:
STM32_CAN Can(CAN2, DEF);
static CAN_message_t CAN_TX_msg;
static CAN_message_t CAN_inMsg;
void buff_data_can(size_t len,size_t len_frame) {
uint8_t buff[len] = {0};
uint32_t buff_32[len / 4] = {0};
uint32_t bias = 0;
while(Can.read(CAN_inMsg)) {
for(size_t i = 0; i < len_frame; i++,bias++) {
buff[bias] = CAN_inMsg.buf[i];
/* if data from len no end blink led 5 count and stop programm */
if(bias >= len){
RCC->AHB1ENR |= RCC_AHB1ENR_GPIOCEN;
GPIOC->MODER |= GPIO_MODER_MODE10_0;
for(int i = 0;i < 5;i++){
GPIOC->ODR ^= GPIO_ODR_OD10;
HAL_Delay(500);
}
return;
}
}
}
/* from 8byte to 32 byte */
for(size_t i = 0; i < len / 4; i++) {
buff_32[i] = (buff[i*4] << 24) |
(buff[i*4 + 1] << 16) |
(buff[i*4 + 2] << 8) |
buff[i*4 + 3];
}
/* CRC check */
/* Work with FLASH */
FLASH_UNLOCK;
uint32_t address = (uint32_t)APP_ADDR;
FLASH_32BYTE;
for(size_t i = 0; i < len; i++) {
if(i == 0) {
flash_erase_sector(1); // Erase sector 1 (APP_ADDR)
while(FLASH_BUSY);
}
flash_program_word(address + (i * 4), buff_32[i],0);
while(FLASH_BUSY);
}
flash_lock();
}
volatile bool fw_update = false;
volatile uint32_t fw_size = 0;
volatile uint32_t fw_crc = 0;
volatile uint32_t write_ptr = APP_ADDRESS;
static FLASH_RECORD flash_record = {0};
// Прототипы функций
void jump_to_app();
void process_can_message(const CAN_message_t &msg);
void erase_flash_pages();
bool verify_firmware();
void send_ack(uint8_t status);
void setup() {
Serial.setRx(HARDWARE_SERIAL_RX_PIN);
Serial.setTx(HARDWARE_SERIAL_TX_PIN);
Serial.begin(115200);
Can.begin();
Can.setBaudRate(1000000);
// put your setup code here, to run once:
RCC->AHB1ENR |= RCC_AHB1ENR_GPIOCEN;
GPIOC->MODER |= GPIO_MODER_MODE10_0 | GPIO_MODER_MODE11_0;;
GPIOC->ODR |= GPIO_ODR_OD11;
// Инициализация периферии
Serial.begin(115200);
Can.begin(1000000); // 1 Mbps
Can.setFilter(0,BOOT_CAN_ID,STD);
// Настройка GPIO
RCC->AHB1ENR |= RCC_AHB1ENR_GPIOCEN;
GPIOC->MODER |= GPIO_MODER_MODE10_0 | GPIO_MODER_MODE11_0;
GPIOC->ODR |= GPIO_ODR_OD11;
// Проверка флага обновления
/*erase_sector(6);
flash_program_word(FLAG_BOOT,0xDEADBEEF,0);
flash_record.data_id = addr_id;
flash_record.crc = 0x6933;
flash_record.value = 0x69;
flash_record.write_ptr_now = SECTOR_6;*/
flash_write(SECTOR_6, &flash_record);
flash_record = load_params();
/* Добавить проверку адреса, т.е во время отправки запроса прошивки по CAN
мы сохраняем как флаг, так и аддрес устройства к которому будет обращатлься во
время прошивки */
if(*(volatile uint32_t*)(FLAG_BOOT) == UPDATE_FLAG) {
fw_update = true;
GPIOC->ODR |= GPIO_ODR_OD10; // Индикация обновления
erase_flash_pages();
} else {
jump_to_app();
}
}
void loop() {
flash_program_word(FLAG_BOOT,1,1);
for(int i = 0; i < 4; i++ ) {
GPIOC->ODR ^= GPIO_ODR_OD11;
HAL_Delay(500);
}
if(FLAG_BOOT == 1){
while (Can.read(CAN_inMsg)){
// buff_data_can(32000,8);
}
/* Go to application */
volatile uint32_t* appjump = (volatile uint32_t*)APP_ADDR;
uint32_t msp_start = *(appjump);
uint32_t reset_handler = *(appjump + 1);
pFunction start = (pFunction)reset_handler;
__set_MSP(msp_start);
start();
}
// put your main code here, to run repeatedly:
if(fw_update) {
CAN_message_t msg;
if(Can.read(msg)) {
process_can_message(msg);
}
}
}
void process_can_message(const CAN_message_t &msg) {
switch(msg.id) {
case BOOT_CAN_ID:
if(msg.buf[0] == 0x01) { // Старт передачи
fw_size = *(uint32_t*)&msg.buf[1]; //размер прошивки тип 4 байта
fw_crc = *(uint32_t*)&msg.buf[5]; //crc
write_ptr = APP_ADDRESS;
send_ack(0x01);
}
break;
case DATA_CAN_ID: // Пакет данных
if(write_ptr < (APP_ADDRESS + fw_size)) {
write_flash_page((const uint8_t*)msg.buf, msg.len);
write_ptr += msg.len;
send_ack(0x02);
}
break;
case BOOT_CAN_END: // Завершение передачи
if(verify_firmware()) {
erase_sector(7); // Сброс флага
send_ack(0xAA);
NVIC_SystemReset();
} else {
send_ack(0x55);
}
break;
}
}
void jump_to_app() {
typedef void (*app_entry_t)(void);
auto app_entry = (app_entry_t)(*(volatile uint32_t*)(APP_ADDRESS + 4));
// SCB->VTOR = APP_ADDRESS;
__set_MSP(*(volatile uint32_t*)APP_ADDRESS);
app_entry();
}
void erase_flash_pages() {
FLASH_EraseInitTypeDef erase;
erase.TypeErase = FLASH_TYPEERASE_SECTORS;
erase.Sector = FLASH_SECTOR_1;
erase.NbSectors = 5;
erase.VoltageRange = FLASH_VOLTAGE_RANGE_3;
uint32_t error;
flash_unlock();
HAL_FLASHEx_Erase(&erase, &error);
flash_lock();
}
// CRC16 implementation for STM32
uint16_t CalculateCRC16(const uint8_t* data, uint32_t length) {
uint16_t crc = 0xFFFF;
while (length--) {
crc ^= (uint16_t)(*data++) << 8;
for(uint8_t i = 0; i < 8; i++) {
crc = crc & 0x8000 ? (crc << 1) ^ 0x8005 : crc << 1;
}
}
return crc;
}
bool verify_firmware() {
uint32_t calculated_crc = 0;
calculated_crc = CalculateCRC16((uint8_t*)fw_crc,fw_size);
if(calculated_crc != (uint16_t)fw_crc)
return false;
// Реализация проверки CRC
// ...
// return (calculated_crc == fw_crc);
return true;
}
void send_ack(uint8_t status) {
CAN_message_t ack;
ack.id = BOOT_CAN_ID + 2;
ack.len = 1;
ack.buf[0] = status;
Can.write(ack);
}