1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-24 15:28:10 +00:00
trezor-firmware/src/main.py

111 lines
2.5 KiB
Python
Raw Normal View History

import boot # noqa: F401
2017-10-24 11:59:09 +00:00
2017-08-14 09:08:47 +00:00
from trezor import io
from trezor import log
from trezor import loop
from trezor import utils
from trezor import wire
from trezor import workflow
from apps.common.storage import get_device_id
log.level = log.DEBUG
2016-04-28 21:48:27 +00:00
2017-08-15 13:09:09 +00:00
# initialize the USB stack
2018-01-20 22:11:21 +00:00
2018-02-03 15:13:25 +00:00
usb_wire = io.WebUSB(
iface_num=0,
ep_in=0x81,
ep_out=0x01,
)
2018-01-20 22:11:21 +00:00
if __debug__:
2018-02-03 15:13:25 +00:00
usb_debug = io.WebUSB(
2018-02-09 15:45:36 +00:00
iface_num=1,
ep_in=0x82,
ep_out=0x02,
2018-02-03 15:13:25 +00:00
)
2018-01-20 22:11:21 +00:00
usb_vcp = io.VCP(
2018-02-09 15:45:36 +00:00
iface_num=2,
data_iface_num=3,
ep_in=0x83,
ep_out=0x03,
ep_cmd=0x84,
2018-01-20 22:11:21 +00:00
)
2018-02-09 15:45:36 +00:00
else:
usb_u2f = io.HID(
iface_num=1,
ep_in=0x82,
ep_out=0x02,
report_desc=bytes([
0x06, 0xd0, 0xf1, # USAGE_PAGE (FIDO Alliance)
0x09, 0x01, # USAGE (U2F HID Authenticator Device)
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
]),
)
2018-01-20 22:11:21 +00:00
2017-08-14 09:08:47 +00:00
usb = io.USB(
2017-04-21 14:06:22 +00:00
vendor_id=0x1209,
product_id=0x53C1,
release_num=0x0200,
2017-08-14 09:08:47 +00:00
manufacturer="SatoshiLabs",
product="TREZOR",
serial_number=get_device_id(),
interface="TREZOR Interface",
2017-08-14 09:08:47 +00:00
)
2018-01-20 22:11:21 +00:00
2017-08-14 09:08:47 +00:00
usb.add(usb_wire)
2017-10-30 16:34:21 +00:00
if __debug__:
usb.add(usb_debug)
2018-01-20 22:11:21 +00:00
usb.add(usb_vcp)
2018-02-09 15:45:36 +00:00
else:
usb.add(usb_u2f)
2017-08-15 13:09:09 +00:00
# load applications
from apps import homescreen
from apps import management
from apps import wallet
from apps import ethereum
2018-02-09 15:45:36 +00:00
if __debug__:
from apps import debug
else:
from apps import fido_u2f
2017-08-15 13:09:09 +00:00
# boot applications
homescreen.boot()
management.boot()
wallet.boot()
ethereum.boot()
2018-02-09 15:45:36 +00:00
if __debug__:
debug.boot()
else:
fido_u2f.boot(usb_u2f)
2017-08-15 13:09:09 +00:00
2017-08-21 11:22:44 +00:00
# initialize the wire codec and start the USB
2017-08-15 13:09:09 +00:00
wire.setup(usb_wire)
2017-10-30 16:34:21 +00:00
if __debug__:
wire.setup(usb_debug)
2017-08-21 11:22:44 +00:00
usb.open()
utils.set_mode_unprivileged()
2017-08-15 13:09:09 +00:00
# load default homescreen
2018-02-27 15:35:21 +00:00
from apps.homescreen.homescreen import homescreen
2016-04-28 21:48:27 +00:00
2017-08-15 13:09:09 +00:00
# run main even loop and specify which screen is default
2018-02-27 15:35:21 +00:00
workflow.startdefault(homescreen)
2017-08-21 11:22:44 +00:00
loop.run()