1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-25 17:09:44 +00:00

legacy: show webusb popup only in bootloader and only if firmware is not installed

This commit is contained in:
Pavol Rusnak 2020-02-12 20:56:42 +00:00
parent 07cd73ce57
commit 215020a592
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
4 changed files with 48 additions and 19 deletions

View File

@ -440,23 +440,38 @@ static void set_config(usbd_device *dev, uint16_t wValue) {
static usbd_device *usbd_dev;
static uint8_t usbd_control_buffer[256] __attribute__((aligned(2)));
static const struct usb_device_capability_descriptor *capabilities[] = {
static const struct usb_device_capability_descriptor *capabilities_landing[] = {
(const struct usb_device_capability_descriptor
*)&webusb_platform_capability_descriptor,
*)&webusb_platform_capability_descriptor_landing,
};
static const struct usb_bos_descriptor bos_descriptor = {
static const struct usb_device_capability_descriptor
*capabilities_no_landing[] = {
(const struct usb_device_capability_descriptor
*)&webusb_platform_capability_descriptor_no_landing,
};
static const struct usb_bos_descriptor bos_descriptor_landing = {
.bLength = USB_DT_BOS_SIZE,
.bDescriptorType = USB_DT_BOS,
.bNumDeviceCaps = sizeof(capabilities) / sizeof(capabilities[0]),
.capabilities = capabilities};
.bNumDeviceCaps =
sizeof(capabilities_landing) / sizeof(capabilities_landing[0]),
.capabilities = capabilities_landing};
static void usbInit(void) {
static const struct usb_bos_descriptor bos_descriptor_no_landing = {
.bLength = USB_DT_BOS_SIZE,
.bDescriptorType = USB_DT_BOS,
.bNumDeviceCaps =
sizeof(capabilities_no_landing) / sizeof(capabilities_no_landing[0]),
.capabilities = capabilities_no_landing};
static void usbInit(bool firmware_present) {
usbd_dev = usbd_init(&otgfs_usb_driver, &dev_descr, &config, usb_strings,
sizeof(usb_strings) / sizeof(const char *),
usbd_control_buffer, sizeof(usbd_control_buffer));
usbd_register_set_config_callback(usbd_dev, set_config);
usb21_setup(usbd_dev, &bos_descriptor);
usb21_setup(usbd_dev, firmware_present ? &bos_descriptor_no_landing
: &bos_descriptor_landing);
webusb_setup(usbd_dev, "trezor.io/start");
winusb_setup(usbd_dev, USB_INTERFACE_INDEX_MAIN);
}
@ -491,7 +506,7 @@ static void checkButtons(void) {
void usbLoop(void) {
bool firmware_present = firmware_present_new();
usbInit();
usbInit(firmware_present);
for (;;) {
usbd_poll(usbd_dev);
if (!firmware_present &&

View File

@ -373,7 +373,7 @@ static uint8_t usbd_control_buffer[256] __attribute__((aligned(2)));
static const struct usb_device_capability_descriptor *capabilities[] = {
(const struct usb_device_capability_descriptor
*)&webusb_platform_capability_descriptor,
*)&webusb_platform_capability_descriptor_no_landing,
};
static const struct usb_bos_descriptor bos_descriptor = {

View File

@ -23,8 +23,9 @@
#include "util.h"
#include "webusb.h"
const struct webusb_platform_descriptor webusb_platform_capability_descriptor =
{.bLength = WEBUSB_PLATFORM_DESCRIPTOR_SIZE,
const struct webusb_platform_descriptor
webusb_platform_capability_descriptor_landing = {
.bLength = WEBUSB_PLATFORM_DESCRIPTOR_SIZE,
.bDescriptorType = USB_DT_DEVICE_CAPABILITY,
.bDevCapabilityType = USB_DC_PLATFORM,
.bReserved = 0,
@ -33,6 +34,17 @@ const struct webusb_platform_descriptor webusb_platform_capability_descriptor =
.bVendorCode = WEBUSB_VENDOR_CODE,
.iLandingPage = 1};
const struct webusb_platform_descriptor
webusb_platform_capability_descriptor_no_landing = {
.bLength = WEBUSB_PLATFORM_DESCRIPTOR_SIZE,
.bDescriptorType = USB_DT_DEVICE_CAPABILITY,
.bDevCapabilityType = USB_DC_PLATFORM,
.bReserved = 0,
.platformCapabilityUUID = WEBUSB_UUID,
.bcdVersion = 0x0100,
.bVendorCode = WEBUSB_VENDOR_CODE,
.iLandingPage = 0};
static const char* webusb_https_url;
static enum usbd_request_return_codes webusb_control_vendor_request(

View File

@ -26,7 +26,9 @@
#define WEBUSB_VENDOR_CODE 0x01
extern const struct webusb_platform_descriptor
webusb_platform_capability_descriptor;
webusb_platform_capability_descriptor_landing;
extern const struct webusb_platform_descriptor
webusb_platform_capability_descriptor_no_landing;
extern void webusb_setup(usbd_device* usbd_dev, const char* https_url);