2017-04-01 10:58:58 -07:00
|
|
|
#include <string.h>
|
2017-04-12 09:31:23 -07:00
|
|
|
#include <sys/types.h>
|
2017-04-01 10:58:58 -07:00
|
|
|
|
2017-03-20 17:41:49 -07:00
|
|
|
#include "common.h"
|
2017-03-20 07:41:21 -07:00
|
|
|
#include "display.h"
|
2017-04-01 03:57:14 -07:00
|
|
|
#include "image.h"
|
2017-03-28 04:04:54 -07:00
|
|
|
#include "flash.h"
|
2017-10-11 12:15:22 -07:00
|
|
|
#include "rng.h"
|
2017-04-05 10:33:50 -07:00
|
|
|
#include "touch.h"
|
2017-03-28 04:04:54 -07:00
|
|
|
#include "usb.h"
|
2017-04-06 08:21:03 -07:00
|
|
|
#include "version.h"
|
2017-04-28 10:50:51 -07:00
|
|
|
#include "mini_printf.h"
|
2017-03-20 07:41:21 -07:00
|
|
|
|
2017-04-12 09:31:23 -07:00
|
|
|
#include "messages.h"
|
|
|
|
|
2017-04-05 08:41:10 -07:00
|
|
|
#define IMAGE_MAGIC 0x465A5254 // TRZF
|
|
|
|
#define IMAGE_MAXSIZE (7 * 128 * 1024)
|
|
|
|
|
2017-04-05 06:10:33 -07:00
|
|
|
void display_vendor(const uint8_t *vimg, const char *vstr, uint32_t vstr_len, uint32_t fw_version)
|
2017-04-01 10:58:58 -07:00
|
|
|
{
|
2017-04-01 11:12:48 -07:00
|
|
|
display_clear();
|
2017-04-01 10:58:58 -07:00
|
|
|
if (memcmp(vimg, "TOIf", 4) != 0) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
uint16_t w = *(uint16_t *)(vimg + 4);
|
|
|
|
uint16_t h = *(uint16_t *)(vimg + 6);
|
2017-04-01 11:12:48 -07:00
|
|
|
if (w != 120 || h != 120) {
|
|
|
|
return;
|
|
|
|
}
|
2017-04-01 10:58:58 -07:00
|
|
|
uint32_t datalen = *(uint32_t *)(vimg + 8);
|
2017-04-05 08:41:10 -07:00
|
|
|
display_image(60, 32, w, h, vimg + 12, datalen);
|
2017-10-01 11:29:23 -07:00
|
|
|
display_text_center(120, 192, vstr, vstr_len, FONT_BOLD, COLOR_WHITE, COLOR_BLACK);
|
2017-04-28 10:50:51 -07:00
|
|
|
char ver_str[32];
|
|
|
|
mini_snprintf(ver_str, sizeof(ver_str), "%d.%d.%d.%d",
|
2017-05-02 08:35:39 -07:00
|
|
|
(int)(fw_version & 0xFF),
|
|
|
|
(int)((fw_version >> 8) & 0xFF),
|
|
|
|
(int)((fw_version >> 16) & 0xFF),
|
|
|
|
(int)((fw_version >> 24) & 0xFF)
|
2017-04-28 10:50:51 -07:00
|
|
|
);
|
2017-10-01 11:29:23 -07:00
|
|
|
display_text_center(120, 215, ver_str, -1, FONT_BOLD, COLOR_GRAY128, COLOR_BLACK);
|
2017-04-01 11:12:48 -07:00
|
|
|
display_refresh();
|
2017-04-01 10:58:58 -07:00
|
|
|
}
|
|
|
|
|
2017-10-01 08:46:58 -07:00
|
|
|
const uint8_t BOOTLOADER_KEY_M = 2;
|
2017-09-29 08:31:59 -07:00
|
|
|
const uint8_t BOOTLOADER_KEY_N = 3;
|
|
|
|
static const uint8_t * const BOOTLOADER_KEYS[] = {
|
2017-10-12 03:17:04 -07:00
|
|
|
#if PRODUCTION
|
2017-10-04 11:20:01 -07:00
|
|
|
(const uint8_t *)"\xc2\xc8\x7a\x49\xc5\xa3\x46\x09\x77\xfb\xb2\xec\x9d\xfe\x60\xf0\x6b\xd6\x94\xdb\x82\x44\xbd\x49\x81\xfe\x3b\x7a\x26\x30\x7f\x3f",
|
2017-10-04 06:55:33 -07:00
|
|
|
(const uint8_t *)"\x80\xd0\x36\xb0\x87\x39\xb8\x46\xf4\xcb\x77\x59\x30\x78\xde\xb2\x5d\xc9\x48\x7a\xed\xcf\x52\xe3\x0b\x4f\xb7\xcd\x70\x24\x17\x8a",
|
|
|
|
(const uint8_t *)"\xb8\x30\x7a\x71\xf5\x52\xc6\x0a\x4c\xbb\x31\x7f\xf4\x8b\x82\xcd\xbf\x6b\x6b\xb5\xf0\x4c\x92\x0f\xec\x7b\xad\xf0\x17\x88\x37\x51",
|
|
|
|
#else
|
2017-09-29 08:31:59 -07:00
|
|
|
(const uint8_t *)"\xd7\x59\x79\x3b\xbc\x13\xa2\x81\x9a\x82\x7c\x76\xad\xb6\xfb\xa8\xa4\x9a\xee\x00\x7f\x49\xf2\xd0\x99\x2d\x99\xb8\x25\xad\x2c\x48",
|
|
|
|
(const uint8_t *)"\x63\x55\x69\x1c\x17\x8a\x8f\xf9\x10\x07\xa7\x47\x8a\xfb\x95\x5e\xf7\x35\x2c\x63\xe7\xb2\x57\x03\x98\x4c\xf7\x8b\x26\xe2\x1a\x56",
|
|
|
|
(const uint8_t *)"\xee\x93\xa4\xf6\x6f\x8d\x16\xb8\x19\xbb\x9b\xeb\x9f\xfc\xcd\xfc\xdc\x14\x12\xe8\x7f\xee\x6a\x32\x4c\x2a\x99\xa1\xe0\xe6\x71\x48",
|
2017-10-04 06:55:33 -07:00
|
|
|
#endif
|
2017-09-29 08:31:59 -07:00
|
|
|
};
|
|
|
|
|
2017-03-31 17:32:05 -07:00
|
|
|
void check_and_jump(void)
|
|
|
|
{
|
2017-04-01 10:24:41 -07:00
|
|
|
vendor_header vhdr;
|
2017-10-05 11:18:35 -07:00
|
|
|
if (!vendor_parse_header((const uint8_t *)FIRMWARE_START, &vhdr)) {
|
2017-04-28 10:50:51 -07:00
|
|
|
display_printf("invalid vendor header\n");
|
2017-04-01 10:24:41 -07:00
|
|
|
return;
|
|
|
|
}
|
2017-04-05 07:24:43 -07:00
|
|
|
|
2017-10-05 11:18:35 -07:00
|
|
|
if (!vendor_check_signature((const uint8_t *)FIRMWARE_START, &vhdr, BOOTLOADER_KEY_M, BOOTLOADER_KEY_N, BOOTLOADER_KEYS)) {
|
2017-04-28 10:50:51 -07:00
|
|
|
display_printf("invalid vendor header signature\n");
|
2017-04-01 10:24:41 -07:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2017-04-05 06:10:33 -07:00
|
|
|
image_header hdr;
|
2017-10-05 11:18:35 -07:00
|
|
|
if (!image_parse_header((const uint8_t *)(FIRMWARE_START + vhdr.hdrlen), IMAGE_MAGIC, IMAGE_MAXSIZE, &hdr)) {
|
2017-04-28 10:50:51 -07:00
|
|
|
display_printf("invalid firmware header\n");
|
2017-04-05 06:10:33 -07:00
|
|
|
return;
|
|
|
|
}
|
2017-04-05 07:24:43 -07:00
|
|
|
|
2017-09-29 08:31:59 -07:00
|
|
|
if (image_check_signature((const uint8_t *)(FIRMWARE_START + vhdr.hdrlen), &hdr, vhdr.vsig_m, vhdr.vsig_n, vhdr.vpub)) {
|
2017-04-05 06:10:33 -07:00
|
|
|
display_vendor(vhdr.vimg, (const char *)vhdr.vstr, vhdr.vstr_len, hdr.version);
|
2017-10-05 08:31:05 -07:00
|
|
|
if (vhdr.vtrust < 50) {
|
|
|
|
touch_click();
|
|
|
|
} else {
|
|
|
|
hal_delay(1000);
|
|
|
|
}
|
2017-04-01 10:24:41 -07:00
|
|
|
jump_to(FIRMWARE_START + vhdr.hdrlen + HEADER_SIZE);
|
2017-03-31 17:32:05 -07:00
|
|
|
} else {
|
2017-04-28 10:50:51 -07:00
|
|
|
display_printf("invalid firmware signature\n");
|
2017-03-31 17:32:05 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-28 04:04:54 -07:00
|
|
|
int usb_init_all(void) {
|
|
|
|
static const usb_dev_info_t dev_info = {
|
2017-08-14 02:08:47 -07:00
|
|
|
.vendor_id = 0x1209,
|
|
|
|
.product_id = 0x53C0,
|
|
|
|
.release_num = 0x0002,
|
|
|
|
.manufacturer = (const uint8_t *)"SatoshiLabs",
|
|
|
|
.product = (const uint8_t *)"TREZOR Bootloader",
|
|
|
|
.serial_number = (const uint8_t *)"",
|
2017-03-28 04:04:54 -07:00
|
|
|
};
|
2017-06-14 11:40:31 -07:00
|
|
|
static uint8_t hid_rx_buffer[USB_PACKET_SIZE];
|
2017-04-06 07:03:04 -07:00
|
|
|
static const uint8_t hid_report_desc[] = {
|
|
|
|
0x06, 0x00, 0xff, // USAGE_PAGE (Vendor Defined)
|
|
|
|
0x09, 0x01, // USAGE (1)
|
|
|
|
0xa1, 0x01, // COLLECTION (Application)
|
|
|
|
0x09, 0x20, // USAGE (Input Report Data)
|
|
|
|
0x15, 0x00, // LOGICAL_MINIMUM (0)
|
|
|
|
0x26, 0xff, 0x00, // LOGICAL_MAXIMUM (255)
|
|
|
|
0x75, 0x08, // REPORT_SIZE (8)
|
|
|
|
0x95, 0x40, // REPORT_COUNT (64)
|
|
|
|
0x81, 0x02, // INPUT (Data,Var,Abs)
|
|
|
|
0x09, 0x21, // USAGE (Output Report Data)
|
|
|
|
0x15, 0x00, // LOGICAL_MINIMUM (0)
|
|
|
|
0x26, 0xff, 0x00, // LOGICAL_MAXIMUM (255)
|
|
|
|
0x75, 0x08, // REPORT_SIZE (8)
|
|
|
|
0x95, 0x40, // REPORT_COUNT (64)
|
|
|
|
0x91, 0x02, // OUTPUT (Data,Var,Abs)
|
|
|
|
0xc0 // END_COLLECTION
|
|
|
|
};
|
2017-03-28 04:04:54 -07:00
|
|
|
static const usb_hid_info_t hid_info = {
|
2017-06-14 11:40:31 -07:00
|
|
|
.iface_num = USB_IFACE_NUM,
|
2017-03-28 04:04:54 -07:00
|
|
|
.ep_in = USB_EP_DIR_IN | 0x01,
|
|
|
|
.ep_out = USB_EP_DIR_OUT | 0x01,
|
|
|
|
.subclass = 0,
|
|
|
|
.protocol = 0,
|
|
|
|
.max_packet_len = sizeof(hid_rx_buffer),
|
2017-04-06 07:03:04 -07:00
|
|
|
.rx_buffer = hid_rx_buffer,
|
2017-03-28 04:04:54 -07:00
|
|
|
.polling_interval = 1,
|
2017-04-06 07:03:04 -07:00
|
|
|
.report_desc_len = sizeof(hid_report_desc),
|
|
|
|
.report_desc = hid_report_desc,
|
2017-03-28 04:04:54 -07:00
|
|
|
};
|
|
|
|
|
2017-10-11 15:32:46 -07:00
|
|
|
trassert(0 == usb_init(&dev_info), NULL);
|
|
|
|
trassert(0 == usb_hid_add(&hid_info), NULL);
|
|
|
|
trassert(0 == usb_start(), NULL);
|
2017-03-28 04:04:54 -07:00
|
|
|
|
|
|
|
return 0;
|
2017-04-05 10:33:50 -07:00
|
|
|
}
|
|
|
|
|
2017-04-06 06:53:47 -07:00
|
|
|
void mainloop(void)
|
|
|
|
{
|
2017-10-11 15:32:46 -07:00
|
|
|
trassert(0 == flash_init(), NULL);
|
|
|
|
trassert(0 == usb_init_all(), NULL);
|
2017-04-06 17:34:53 -07:00
|
|
|
|
2017-06-20 10:53:24 -07:00
|
|
|
display_clear();
|
|
|
|
|
2017-04-27 10:18:57 -07:00
|
|
|
uint8_t buf[USB_PACKET_SIZE];
|
2017-04-06 17:34:53 -07:00
|
|
|
|
2017-04-12 09:31:23 -07:00
|
|
|
for (;;) {
|
2017-04-27 10:18:57 -07:00
|
|
|
int r = usb_hid_read_blocking(USB_IFACE_NUM, buf, USB_PACKET_SIZE, 100);
|
2017-06-16 06:40:27 -07:00
|
|
|
if (r != USB_PACKET_SIZE) {
|
2017-04-12 09:31:23 -07:00
|
|
|
continue;
|
|
|
|
}
|
2017-10-11 15:32:46 -07:00
|
|
|
trassert(r == USB_PACKET_SIZE, NULL);
|
2017-04-27 06:32:25 -07:00
|
|
|
uint16_t msg_id;
|
|
|
|
uint32_t msg_size;
|
2017-06-14 11:40:31 -07:00
|
|
|
if (!msg_parse_header(buf, &msg_id, &msg_size)) {
|
|
|
|
// invalid header -> discard
|
2017-04-12 09:31:23 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
switch (msg_id) {
|
|
|
|
case 0: // Initialize
|
2017-06-16 06:40:27 -07:00
|
|
|
process_msg_Initialize(USB_IFACE_NUM, msg_size, buf);
|
2017-04-12 09:31:23 -07:00
|
|
|
break;
|
|
|
|
case 1: // Ping
|
2017-06-16 06:40:27 -07:00
|
|
|
process_msg_Ping(USB_IFACE_NUM, msg_size, buf);
|
2017-04-12 09:31:23 -07:00
|
|
|
break;
|
|
|
|
case 6: // FirmwareErase
|
2017-06-16 06:40:27 -07:00
|
|
|
process_msg_FirmwareErase(USB_IFACE_NUM, msg_size, buf);
|
2017-04-12 09:31:23 -07:00
|
|
|
break;
|
|
|
|
case 7: // FirmwareUpload
|
2017-06-16 06:40:27 -07:00
|
|
|
process_msg_FirmwareUpload(USB_IFACE_NUM, msg_size, buf);
|
2017-04-12 09:31:23 -07:00
|
|
|
break;
|
|
|
|
default:
|
2017-06-16 06:40:27 -07:00
|
|
|
process_msg_unknown(USB_IFACE_NUM, msg_size, buf);
|
2017-04-12 09:31:23 -07:00
|
|
|
break;
|
2017-04-06 17:34:53 -07:00
|
|
|
}
|
|
|
|
}
|
2017-04-06 06:53:47 -07:00
|
|
|
}
|
|
|
|
|
2017-03-23 08:22:58 -07:00
|
|
|
int main(void)
|
2017-03-20 07:41:21 -07:00
|
|
|
{
|
2017-10-11 12:15:22 -07:00
|
|
|
__stack_chk_guard = rng_get();
|
|
|
|
|
2017-03-23 08:22:58 -07:00
|
|
|
periph_init();
|
2017-03-20 07:41:21 -07:00
|
|
|
|
2017-09-07 18:04:23 -07:00
|
|
|
display_pwm_init();
|
|
|
|
display_orientation(0);
|
|
|
|
display_backlight(255);
|
2017-03-28 04:04:54 -07:00
|
|
|
|
2017-10-11 15:32:46 -07:00
|
|
|
trassert(0 == touch_init(), NULL);
|
2017-03-28 04:04:54 -07:00
|
|
|
|
2017-06-14 11:40:31 -07:00
|
|
|
uint32_t touched = 0;
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
|
|
touched |= touch_read();
|
|
|
|
}
|
|
|
|
|
|
|
|
if (touched != 0) {
|
2017-04-05 10:33:50 -07:00
|
|
|
mainloop();
|
|
|
|
} else {
|
|
|
|
check_and_jump();
|
|
|
|
}
|
2017-03-29 11:50:45 -07:00
|
|
|
|
2017-10-11 15:32:46 -07:00
|
|
|
trassert(0, "halt");
|
2017-03-20 07:41:21 -07:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|