BTCPrivate
/
python-trezor
mirror of https://github.com/BTCPrivate/python-trezor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Added padding to HID messages

This commit is contained in:
slush 2013-09-09 15:37:39 +02:00
parent 813fb233a1
commit 682578f1ad
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
bitkeylib/transport_hid.py
View File

@ -54,12 +54,9 @@ class HidTransport(Transport):
def _write(self, msg):
msg = bytearray(msg)
while len(msg):
to_send = min(63, len(msg))
#print [to_send,] + list(msg[:to_send])
self.hid.write([to_send,] + list(msg[:to_send]))
msg = msg[to_send:]
# Report ID, data padded to 62 bytes
self.hid.write([63,] + list(msg[:63]) + [0]*(63-len(msg[0:63])))
msg = msg[63:]
def _read(self):
(msg_type, datalen) = self._read_headers(FakeRead(self._raw_read))
@ -68,14 +65,15 @@ class HidTransport(Transport):
def _raw_read(self, length):
while len(self.buffer) < length:
data = self.hid.read(64)
report_id = data[0]
if report_id > 63:
# Command report
raise Exception("Not implemented")
# Payload received
self.buffer += str(bytearray(data[1:report_id+1]))
# Payload received, skip the report ID
self.buffer += str(bytearray(data[1:64]))
ret = self.buffer[:length]
self.buffer = self.buffer[length:]