diff --git a/trezorlib/transport.py b/trezorlib/transport.py index c7e806074..f2aaeb065 100644 --- a/trezorlib/transport.py +++ b/trezorlib/transport.py @@ -81,11 +81,13 @@ class Transport(object): # Align cursor to the beginning of the header ("##") c = read_f.read(1) + i = 0 while c != '#': - if c == '': + i += 1 + if i >= 64: # timeout raise Exception("Timed out while waiting for the magic character") - print "Warning: Aligning to magic characters" + #print "Aligning to magic characters" c = read_f.read(1) if read_f.read(1) != "#": diff --git a/trezorlib/transport_hid.py b/trezorlib/transport_hid.py index 19186bad6..885bfae7e 100644 --- a/trezorlib/transport_hid.py +++ b/trezorlib/transport_hid.py @@ -1,6 +1,7 @@ '''USB HID implementation of Transport.''' import hid +import time from transport import Transport, NotImplementedException DEVICE_IDS = [ @@ -40,6 +41,7 @@ class HidTransport(Transport): self.buffer = '' path = self.device.split(':') self.hid = hid.device(int(path[0], 16), int(path[1], 16))#, path[2]) + self.hid.set_nonblocking(True) self.hid.send_feature_report([0x41, 0x01]) # enable UART self.hid.send_feature_report([0x43, 0x03]) # purge TX/RX FIFOs @@ -54,8 +56,8 @@ class HidTransport(Transport): def _write(self, msg): msg = bytearray(msg) while len(msg): - # Report ID, data padded to 62 bytes - self.hid.write([63,] + list(msg[:63]) + [0]*(63-len(msg[0:63]))) + # Report ID, data padded to 63 bytes + self.hid.write([63, ] + list(msg[:63]) + [0] * (63 - len(msg[:63]))) msg = msg[63:] def _read(self): @@ -65,6 +67,9 @@ class HidTransport(Transport): def _raw_read(self, length): while len(self.buffer) < length: data = self.hid.read(64) + if not len(data): + time.sleep(0.05) + continue report_id = data[0] @@ -73,7 +78,7 @@ class HidTransport(Transport): raise Exception("Not implemented") # Payload received, skip the report ID - self.buffer += str(bytearray(data[1:64])) + self.buffer += str(bytearray(data[1:])) ret = self.buffer[:length] self.buffer = self.buffer[length:]