arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-26 09:28:13 +00:00
Code Issues Releases Wiki Activity

transport_hid: more strict interface detection

Browse Source
This commit is contained in:
Jan Pochyla 2017-09-04 17:30:07 +02:00
parent 2a3f613242
commit b60ab51f9b
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
trezorlib/transport_hid.py
View File

@ -53,11 +53,18 @@ class HidTransport(Transport):
@staticmethod
def enumerate(debug=False):
return [
HidTransport(dev) for dev in hid.enumerate(0, 0)
if ((is_trezor1(dev) or is_trezor2(dev) or is_trezor2_bl(dev)) and
(is_debug(dev) == debug))
]
devices = []
for dev in hid.enumerate(0, 0):
if not (is_trezor1(dev) or is_trezor2(dev) or is_trezor2_bl(dev)):
continue
if debug:
if not is_debuglink(dev):
continue
else:
if not is_wirelink(dev):
continue
devices.append(HidTransport(dev))
return devices
@staticmethod
def find_by_path(path=None):
@ -148,5 +155,9 @@ def is_trezor2_bl(dev):
return (dev['vendor_id'], dev['product_id']) == DEV_TREZOR2_BL
def is_debug(dev):
return (dev['usage_page'] == 0xFF01 or dev['interface_number'] == 1)
def is_wirelink(dev):
return (dev['usage_page'] == 0xFF00 or dev['interface_number'] == 0)
def is_debuglink(dev):
return (dev['usage_page'] == 0xFF01 or dev['interface_number'] == 1)