From a129b072d347b2b511bff87318e9ccd18d3012fb Mon Sep 17 00:00:00 2001 From: Jan Pochyla Date: Thu, 14 Jul 2016 15:57:05 +0200 Subject: [PATCH] ignore socket.recv timeout in udp transport --- trezorlib/transport_udp.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/trezorlib/transport_udp.py b/trezorlib/transport_udp.py index 27e95da..f521873 100644 --- a/trezorlib/transport_udp.py +++ b/trezorlib/transport_udp.py @@ -2,7 +2,6 @@ import socket from select import select -import time from .transport import TransportV2, ConnectionError class UdpTransport(TransportV2): @@ -40,7 +39,12 @@ class UdpTransport(TransportV2): self.socket.sendall(chunk) def _read_chunk(self): - data = self.socket.recv(64) + while True: + try: + data = self.socket.recv(64) + break + except socket.timeout: + continue if len(data) != 64: raise Exception("Unexpected chunk size: %d" % len(data))