arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-08 14:31:06 +00:00
Code Issues Releases Wiki Activity

Fixes reading of message headers

Browse Source
This commit is contained in:
slush 2013-03-10 17:52:04 +01:00
parent 41b706d56a
commit c0d267e7d8
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bitkeylib/transport_hid.py
View File

@ -9,6 +9,14 @@ DEVICE_IDS = [
(0x08f7, 0x0002), # EasyTemp
]
class FakeRead(object):
# Let's pretend we have a file-like interface
def __init__(self, func):
self.func = func
def read(self, size):
return self.func(size)
class HidTransport(Transport):
def __init__(self, device, *args, **kwargs):
self.hid = None
@ -51,7 +59,7 @@ class HidTransport(Transport):
msg = msg[to_send:]
def _read(self):
(msg_type, datalen) = self._read_headers(self._raw_read)
(msg_type, datalen) = self._read_headers(FakeRead(self._raw_read))
return (msg_type, self._raw_read(datalen))
def _raw_read(self, length):