mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-24 23:38:09 +00:00
bootloader: more wip
This commit is contained in:
parent
e8adff4683
commit
d8efa32091
@ -81,7 +81,6 @@ SRC_C = \
|
|||||||
bootloader/bootloader.c \
|
bootloader/bootloader.c \
|
||||||
bootloader/crypto.c \
|
bootloader/crypto.c \
|
||||||
bootloader/sdcard.c \
|
bootloader/sdcard.c \
|
||||||
bootloader/ui.c \
|
|
||||||
system_stm32.c \
|
system_stm32.c \
|
||||||
$(wildcard boards/$(BOARD)/*.c)
|
$(wildcard boards/$(BOARD)/*.c)
|
||||||
|
|
||||||
|
@ -1,20 +1,19 @@
|
|||||||
#include STM32_HAL_H
|
#include STM32_HAL_H
|
||||||
|
#include "display.h"
|
||||||
|
|
||||||
// ### from main.c
|
// ### from main.c
|
||||||
|
|
||||||
void __attribute__((noreturn)) __fatal_error(const char *msg) {
|
void __attribute__((noreturn)) __fatal_error(const char *msg) {
|
||||||
for (volatile uint32_t delay = 0; delay < 10000000; delay++) {
|
for (volatile uint32_t delay = 0; delay < 10000000; delay++) {
|
||||||
}
|
}
|
||||||
// TODO: printf("FATAL ERROR: %s\n", msg);
|
display_print("FATAL ERROR:\n", -1);
|
||||||
|
display_print(msg, -1);
|
||||||
|
display_print("\n", -1);
|
||||||
for (;;) {
|
for (;;) {
|
||||||
__WFI();
|
__WFI();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void nlr_jump_fail(void *val) {
|
|
||||||
__fatal_error("FATAL: uncaught exception");
|
|
||||||
}
|
|
||||||
|
|
||||||
// ### from stm32_it.c
|
// ### from stm32_it.c
|
||||||
|
|
||||||
void SysTick_Handler(void) {
|
void SysTick_Handler(void) {
|
||||||
|
@ -3,7 +3,6 @@
|
|||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "crypto.h"
|
#include "crypto.h"
|
||||||
#include "ui.h"
|
|
||||||
#include "display.h"
|
#include "display.h"
|
||||||
#include "sdcard.h"
|
#include "sdcard.h"
|
||||||
|
|
||||||
@ -11,6 +10,9 @@
|
|||||||
#define STAGE2_START 0x08010000
|
#define STAGE2_START 0x08010000
|
||||||
#define STAGE2_SIZE (64 * 1024)
|
#define STAGE2_SIZE (64 * 1024)
|
||||||
|
|
||||||
|
#define BOOTLOADER_PRINT(X) display_print(X, -1)
|
||||||
|
#define BOOTLOADER_PRINTLN(X) display_print(X "\n", -1)
|
||||||
|
|
||||||
void SystemClock_Config(void);
|
void SystemClock_Config(void);
|
||||||
|
|
||||||
void periph_init(void)
|
void periph_init(void)
|
||||||
@ -31,26 +33,45 @@ void periph_init(void)
|
|||||||
display_backlight(255);
|
display_backlight(255);
|
||||||
}
|
}
|
||||||
|
|
||||||
void check_sdcard()
|
bool check_sdcard(void)
|
||||||
{
|
{
|
||||||
if (!sdcard_is_present()) return;
|
BOOTLOADER_PRINTLN("checking for SD card");
|
||||||
|
|
||||||
|
if (!sdcard_is_present()) {
|
||||||
|
BOOTLOADER_PRINTLN("no SD card found");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("SD card found");
|
||||||
|
|
||||||
sdcard_power_on();
|
sdcard_power_on();
|
||||||
|
|
||||||
uint64_t cap = sdcard_get_capacity_in_bytes();
|
uint64_t cap = sdcard_get_capacity_in_bytes();
|
||||||
if (cap < SDCARD_BLOCK_SIZE + STAGE2_SIZE) {
|
if (cap < STAGE2_SIZE) {
|
||||||
|
BOOTLOADER_PRINTLN("SD card too small");
|
||||||
sdcard_power_off();
|
sdcard_power_off();
|
||||||
return;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t buf[SDCARD_BLOCK_SIZE] __attribute__((aligned(4)));
|
uint8_t buf[SDCARD_BLOCK_SIZE] __attribute__((aligned(4)));
|
||||||
|
|
||||||
sdcard_read_blocks(buf, 0, 1);
|
sdcard_read_blocks(buf, 0, 1);
|
||||||
|
|
||||||
if (!check_header(buf)) {
|
|
||||||
sdcard_power_off();
|
sdcard_power_off();
|
||||||
return;
|
|
||||||
|
if (check_header(buf)) {
|
||||||
|
BOOTLOADER_PRINTLN("SD card header is valid");
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
BOOTLOADER_PRINTLN("SD card header is invalid");
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void copy_sdcard(void)
|
||||||
|
{
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("erasing old stage 2");
|
||||||
|
|
||||||
// erase STAGE2_SECTOR
|
// erase STAGE2_SECTOR
|
||||||
HAL_FLASH_Unlock();
|
HAL_FLASH_Unlock();
|
||||||
@ -64,55 +85,70 @@ void check_sdcard()
|
|||||||
uint32_t SectorError = 0;
|
uint32_t SectorError = 0;
|
||||||
if (HAL_FLASHEx_Erase(&EraseInitStruct, &SectorError) != HAL_OK) {
|
if (HAL_FLASHEx_Erase(&EraseInitStruct, &SectorError) != HAL_OK) {
|
||||||
HAL_FLASH_Lock();
|
HAL_FLASH_Lock();
|
||||||
sdcard_power_off();
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// copy stage 2 from SD card to Flash
|
BOOTLOADER_PRINTLN("copying new stage 2 from SD card");
|
||||||
uint32_t src;
|
|
||||||
int block = 0;
|
|
||||||
int offset = 256;
|
|
||||||
|
|
||||||
for (int i = 0; i < STAGE2_SIZE / 4; i++) {
|
// copy stage 2 from SD card to Flash
|
||||||
memcpy(&src, buf + offset, 4);
|
uint32_t buf[SDCARD_BLOCK_SIZE / sizeof(uint32_t)];
|
||||||
if (HAL_FLASH_Program(TYPEPROGRAM_WORD, STAGE2_START + i * 4, src) != HAL_OK) {
|
|
||||||
|
sdcard_power_on();
|
||||||
|
for (int i = 0; i < STAGE2_SIZE / SDCARD_BLOCK_SIZE; i++) {
|
||||||
|
sdcard_read_blocks((uint8_t *)buf, i, 1);
|
||||||
|
for (int j = 0; j < SDCARD_BLOCK_SIZE / sizeof(uint32_t); j++) {
|
||||||
|
if (HAL_FLASH_Program(TYPEPROGRAM_WORD, STAGE2_START + i * SDCARD_BLOCK_SIZE + j * sizeof(uint32_t), buf[j]) != HAL_OK) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
offset += 4;
|
|
||||||
if (offset == SDCARD_BLOCK_SIZE) {
|
|
||||||
offset = 0;
|
|
||||||
block++;
|
|
||||||
sdcard_read_blocks(buf, block, 1);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
HAL_FLASH_Lock();
|
|
||||||
sdcard_power_off();
|
sdcard_power_off();
|
||||||
|
|
||||||
|
HAL_FLASH_Lock();
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("done");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void halt(void)
|
||||||
int main(void) {
|
{
|
||||||
|
|
||||||
periph_init();
|
|
||||||
|
|
||||||
screen_stage1();
|
|
||||||
|
|
||||||
check_sdcard();
|
|
||||||
|
|
||||||
check_signature();
|
|
||||||
|
|
||||||
if (check_header((const uint8_t *)STAGE2_START)) {
|
|
||||||
screen_stage2_jump();
|
|
||||||
// TODO: jump to second stage
|
|
||||||
}
|
|
||||||
|
|
||||||
screen_stage2_invalid();
|
|
||||||
|
|
||||||
for (;;) {
|
for (;;) {
|
||||||
display_backlight(255);
|
display_backlight(255);
|
||||||
HAL_Delay(950);
|
HAL_Delay(950);
|
||||||
display_backlight(0);
|
display_backlight(0);
|
||||||
HAL_Delay(50);
|
HAL_Delay(50);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(void)
|
||||||
|
{
|
||||||
|
|
||||||
|
periph_init();
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("TREZOR Bootloader");
|
||||||
|
BOOTLOADER_PRINTLN("=================");
|
||||||
|
BOOTLOADER_PRINTLN("starting stage 1");
|
||||||
|
|
||||||
|
if (check_sdcard()) {
|
||||||
|
copy_sdcard();
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("checking stage 2");
|
||||||
|
if (check_header((const uint8_t *)STAGE2_START)) {
|
||||||
|
BOOTLOADER_PRINTLN("valid stage 2 header");
|
||||||
|
if (check_signature()) {
|
||||||
|
BOOTLOADER_PRINTLN("valid stage 2 signature");
|
||||||
|
BOOTLOADER_PRINTLN("JUMP!");
|
||||||
|
// TODO: jump to second stage
|
||||||
|
halt();
|
||||||
|
} else {
|
||||||
|
BOOTLOADER_PRINTLN("invalid stage 2 signature");
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
BOOTLOADER_PRINTLN("invalid stage 2 header");
|
||||||
|
}
|
||||||
|
|
||||||
|
BOOTLOADER_PRINTLN("HALT!");
|
||||||
|
halt();
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -33,7 +33,7 @@ bool check_header(const uint8_t *data)
|
|||||||
|
|
||||||
uint32_t codelen;
|
uint32_t codelen;
|
||||||
memcpy(&codelen, data + 12, 4);
|
memcpy(&codelen, data + 12, 4);
|
||||||
if (codelen != 64 * 1024) return false;
|
if (codelen != 64 * 1024 - 256) return false;
|
||||||
|
|
||||||
uint32_t version;
|
uint32_t version;
|
||||||
memcpy(&version, data + 16, 4);
|
memcpy(&version, data + 16, 4);
|
||||||
@ -46,8 +46,6 @@ bool check_header(const uint8_t *data)
|
|||||||
uint8_t sig[64];
|
uint8_t sig[64];
|
||||||
memcpy(sig, data + 0x00C0, 64);
|
memcpy(sig, data + 0x00C0, 64);
|
||||||
|
|
||||||
// TODO: check signature
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,17 +0,0 @@
|
|||||||
#include "ui.h"
|
|
||||||
#include "display.h"
|
|
||||||
|
|
||||||
void screen_stage1(void)
|
|
||||||
{
|
|
||||||
display_print("BL stage 1\n", -1);
|
|
||||||
}
|
|
||||||
|
|
||||||
void screen_stage2_jump(void)
|
|
||||||
{
|
|
||||||
display_print("BL stage 2 jump\n", -1);
|
|
||||||
}
|
|
||||||
|
|
||||||
void screen_stage2_invalid(void)
|
|
||||||
{
|
|
||||||
display_print("BL stage 2 invalid\n", -1);
|
|
||||||
}
|
|
@ -1,11 +0,0 @@
|
|||||||
#ifndef __BOOTLOADER_UI_H__
|
|
||||||
#define __BOOTLOADER_UI_H__
|
|
||||||
|
|
||||||
#define ui_WHITE 0xFFFF
|
|
||||||
#define ui_BLACK 0x0000
|
|
||||||
|
|
||||||
void screen_stage1(void);
|
|
||||||
void screen_stage2_jump(void);
|
|
||||||
void screen_stage2_invalid(void);
|
|
||||||
|
|
||||||
#endif
|
|
@ -42,7 +42,7 @@ class BootloaderImage:
|
|||||||
self.reserved, self.sigidx, self.sig = header
|
self.reserved, self.sigidx, self.sig = header
|
||||||
assert self.magic == b'TRZB'
|
assert self.magic == b'TRZB'
|
||||||
assert self.hdrlen == 256
|
assert self.hdrlen == 256
|
||||||
assert self.codelen == 64*1024
|
assert self.codelen == 64*1024 - 256
|
||||||
assert self.reserved == 171 * b'\x00'
|
assert self.reserved == 171 * b'\x00'
|
||||||
self.sigidx = bitmap_to_tuple(self.sigidx)
|
self.sigidx = bitmap_to_tuple(self.sigidx)
|
||||||
self.code = data[256:]
|
self.code = data[256:]
|
||||||
|
Loading…
Reference in New Issue
Block a user