1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-23 23:08:14 +00:00

bootloader: more wip

This commit is contained in:
Pavol Rusnak 2017-02-17 15:49:43 +01:00
parent e8adff4683
commit d8efa32091
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
7 changed files with 84 additions and 80 deletions

View File

@ -81,7 +81,6 @@ SRC_C = \
bootloader/bootloader.c \
bootloader/crypto.c \
bootloader/sdcard.c \
bootloader/ui.c \
system_stm32.c \
$(wildcard boards/$(BOARD)/*.c)

View File

@ -1,20 +1,19 @@
#include STM32_HAL_H
#include "display.h"
// ### from main.c
void __attribute__((noreturn)) __fatal_error(const char *msg) {
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 (;;) {
__WFI();
}
}
void nlr_jump_fail(void *val) {
__fatal_error("FATAL: uncaught exception");
}
// ### from stm32_it.c
void SysTick_Handler(void) {

View File

@ -3,7 +3,6 @@
#include <string.h>
#include "crypto.h"
#include "ui.h"
#include "display.h"
#include "sdcard.h"
@ -11,6 +10,9 @@
#define STAGE2_START 0x08010000
#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 periph_init(void)
@ -31,26 +33,45 @@ void periph_init(void)
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();
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();
return;
return false;
}
uint8_t buf[SDCARD_BLOCK_SIZE] __attribute__((aligned(4)));
sdcard_read_blocks(buf, 0, 1);
if (!check_header(buf)) {
sdcard_power_off();
return;
sdcard_power_off();
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
HAL_FLASH_Unlock();
@ -64,55 +85,70 @@ void check_sdcard()
uint32_t SectorError = 0;
if (HAL_FLASHEx_Erase(&EraseInitStruct, &SectorError) != HAL_OK) {
HAL_FLASH_Lock();
sdcard_power_off();
return;
}
// copy stage 2 from SD card to Flash
uint32_t src;
int block = 0;
int offset = 256;
BOOTLOADER_PRINTLN("copying new stage 2 from SD card");
for (int i = 0; i < STAGE2_SIZE / 4; i++) {
memcpy(&src, buf + offset, 4);
if (HAL_FLASH_Program(TYPEPROGRAM_WORD, STAGE2_START + i * 4, src) != HAL_OK) {
break;
}
offset += 4;
if (offset == SDCARD_BLOCK_SIZE) {
offset = 0;
block++;
sdcard_read_blocks(buf, block, 1);
// copy stage 2 from SD card to Flash
uint32_t buf[SDCARD_BLOCK_SIZE / sizeof(uint32_t)];
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;
}
}
}
HAL_FLASH_Lock();
sdcard_power_off();
HAL_FLASH_Lock();
BOOTLOADER_PRINTLN("done");
}
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();
void halt(void)
{
for (;;) {
display_backlight(255);
HAL_Delay(950);
display_backlight(0);
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;
}

View File

@ -33,7 +33,7 @@ bool check_header(const uint8_t *data)
uint32_t codelen;
memcpy(&codelen, data + 12, 4);
if (codelen != 64 * 1024) return false;
if (codelen != 64 * 1024 - 256) return false;
uint32_t version;
memcpy(&version, data + 16, 4);
@ -46,8 +46,6 @@ bool check_header(const uint8_t *data)
uint8_t sig[64];
memcpy(sig, data + 0x00C0, 64);
// TODO: check signature
return true;
}

View File

@ -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);
}

View File

@ -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

View File

@ -42,7 +42,7 @@ class BootloaderImage:
self.reserved, self.sigidx, self.sig = header
assert self.magic == b'TRZB'
assert self.hdrlen == 256
assert self.codelen == 64*1024
assert self.codelen == 64*1024 - 256
assert self.reserved == 171 * b'\x00'
self.sigidx = bitmap_to_tuple(self.sigidx)
self.code = data[256:]