2017-10-24 11:59:09 +00:00
|
|
|
import boot
|
|
|
|
|
2017-08-14 09:08:47 +00:00
|
|
|
from trezor import io
|
2017-08-16 13:02:03 +00:00
|
|
|
from trezor import log
|
|
|
|
from trezor import loop
|
2016-09-21 12:21:18 +00:00
|
|
|
from trezor import wire
|
2017-08-16 13:02:03 +00:00
|
|
|
from trezor import workflow
|
|
|
|
|
|
|
|
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
|
|
|
|
|
|
|
if True:
|
|
|
|
usb_wire = io.HID(
|
|
|
|
iface_num=0,
|
|
|
|
ep_in=0x81,
|
|
|
|
ep_out=0x01,
|
2017-10-30 16:34:21 +00:00
|
|
|
report_desc=bytes([
|
2018-01-20 22:11:21 +00:00
|
|
|
0x06, 0x00, 0xff, # USAGE_PAGE (Vendor Defined)
|
2017-10-30 16:34:21 +00:00
|
|
|
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
|
|
|
|
]),
|
|
|
|
)
|
2018-01-20 22:11:21 +00:00
|
|
|
|
2017-08-14 09:08:47 +00:00
|
|
|
usb_u2f = io.HID(
|
2018-01-20 22:11:21 +00:00
|
|
|
iface_num=3,
|
2017-07-12 14:04:03 +00:00
|
|
|
ep_in=0x84,
|
|
|
|
ep_out=0x03,
|
2017-05-09 14:12:02 +00:00
|
|
|
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
|
|
|
|
|
|
|
if __debug__:
|
|
|
|
if True:
|
|
|
|
usb_debug = io.HID(
|
|
|
|
iface_num=4,
|
|
|
|
ep_in=0x85,
|
|
|
|
ep_out=0x04,
|
|
|
|
report_desc=bytes([
|
|
|
|
0x06, 0x01, 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
|
|
|
|
]),
|
|
|
|
)
|
|
|
|
usb_vcp = io.VCP(
|
|
|
|
iface_num=1,
|
|
|
|
data_iface_num=2,
|
|
|
|
ep_in=0x82,
|
|
|
|
ep_out=0x02,
|
|
|
|
ep_cmd=0x83,
|
|
|
|
)
|
|
|
|
|
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=0x0002,
|
2017-08-14 09:08:47 +00:00
|
|
|
manufacturer="SatoshiLabs",
|
|
|
|
product="TREZOR",
|
|
|
|
serial_number="000000000000000000000000",
|
|
|
|
)
|
2018-01-20 22:11:21 +00:00
|
|
|
|
2017-08-14 09:08:47 +00:00
|
|
|
usb.add(usb_wire)
|
2018-01-20 22:11:21 +00:00
|
|
|
usb.add(usb_u2f)
|
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)
|
2016-10-11 10:33:02 +00:00
|
|
|
|
2017-08-15 13:09:09 +00:00
|
|
|
# load applications
|
|
|
|
if __debug__:
|
|
|
|
from apps import debug
|
|
|
|
from apps import homescreen
|
|
|
|
from apps import management
|
|
|
|
from apps import wallet
|
|
|
|
from apps import ethereum
|
|
|
|
from apps import fido_u2f
|
|
|
|
|
|
|
|
# boot applications
|
|
|
|
if __debug__:
|
|
|
|
debug.boot()
|
|
|
|
homescreen.boot()
|
|
|
|
management.boot()
|
|
|
|
wallet.boot()
|
|
|
|
ethereum.boot()
|
|
|
|
fido_u2f.boot(usb_u2f)
|
|
|
|
|
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()
|
2016-09-21 12:21:18 +00:00
|
|
|
|
2017-08-15 13:09:09 +00:00
|
|
|
# load default homescreen
|
2016-12-08 15:18:58 +00:00
|
|
|
from apps.homescreen.homescreen import layout_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
|
2017-08-16 13:02:03 +00:00
|
|
|
workflow.startdefault(layout_homescreen)
|
2017-08-21 11:22:44 +00:00
|
|
|
loop.run()
|