111 lines
2.6 KiB
Python
111 lines
2.6 KiB
Python
import boot # noqa: F401
|
|
|
|
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
|
|
|
|
# initialize the USB stack
|
|
|
|
usb_wire = io.WebUSB(
|
|
iface_num=0,
|
|
ep_in=0x81,
|
|
ep_out=0x01,
|
|
)
|
|
|
|
if __debug__:
|
|
usb_debug = io.WebUSB(
|
|
iface_num=1,
|
|
ep_in=0x82,
|
|
ep_out=0x02,
|
|
)
|
|
usb_vcp = io.VCP(
|
|
iface_num=2,
|
|
data_iface_num=3,
|
|
ep_in=0x83,
|
|
ep_out=0x03,
|
|
ep_cmd=0x84,
|
|
)
|
|
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
|
|
]),
|
|
)
|
|
|
|
usb = io.USB(
|
|
vendor_id=0x1209,
|
|
product_id=0x53C1,
|
|
release_num=0x0200,
|
|
manufacturer="SatoshiLabs",
|
|
product="TREZOR",
|
|
serial_number=get_device_id(),
|
|
interface="TREZOR Interface",
|
|
)
|
|
|
|
usb.add(usb_wire)
|
|
if __debug__:
|
|
usb.add(usb_debug)
|
|
usb.add(usb_vcp)
|
|
else:
|
|
usb.add(usb_u2f)
|
|
|
|
# load applications
|
|
from apps import homescreen
|
|
from apps import management
|
|
from apps import wallet
|
|
from apps import ethereum
|
|
if __debug__:
|
|
from apps import debug
|
|
else:
|
|
from apps import fido_u2f
|
|
|
|
# boot applications
|
|
homescreen.boot()
|
|
management.boot()
|
|
wallet.boot()
|
|
ethereum.boot()
|
|
if __debug__:
|
|
debug.boot()
|
|
else:
|
|
fido_u2f.boot(usb_u2f)
|
|
|
|
# initialize the wire codec and start the USB
|
|
wire.setup(usb_wire)
|
|
if __debug__:
|
|
wire.setup(usb_debug)
|
|
usb.open()
|
|
|
|
utils.set_mode_unprivileged()
|
|
|
|
# load default homescreen
|
|
from apps.homescreen.homescreen import layout_homescreen
|
|
|
|
# run main even loop and specify which screen is default
|
|
workflow.startdefault(layout_homescreen)
|
|
loop.run()
|