Fix #35

LedgerHQ · Jun 30, 2022 · a7f821b · a7f821b
1 parent 75bf879
commit a7f821b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/ledgerwallet/transport/device.py b/ledgerwallet/transport/device.py
@@ -17,11 +17,11 @@ def write(self, data: bytes):
         raise NotImplementedError
 
     @abstractmethod
-    def read(self) -> bytes:
+    def read(self, timeout: int = 0) -> bytes:
         raise NotImplementedError
 
     @abstractmethod
-    def exchange(self, data: bytes) -> bytes:
+    def exchange(self, data: bytes, timeout: int = 0) -> bytes:
         raise NotImplementedError
 
     @abstractmethod

diff --git a/ledgerwallet/transport/hid.py b/ledgerwallet/transport/hid.py
@@ -68,7 +68,7 @@ def read(self, timeout: int = 1000) -> bytes:
 
     def exchange(self, data: bytes, timeout: int = 1000):
         self.write(data)
-        return self.read(timeout)
+        return self.read(timeout=timeout)
 
     def close(self):
         if self.opened:

diff --git a/ledgerwallet/transport/tcp.py b/ledgerwallet/transport/tcp.py
@@ -36,11 +36,11 @@ def write(self, data: bytes):
         data_to_send = int.to_bytes(len(data), 4, "big") + data
         self.socket.send(data_to_send)
 
-    def read(self) -> bytes:
+    def read(self, timeout: int = 0) -> bytes:
         packet_len = int.from_bytes(self.socket.recv(4), "big")
         return self.socket.recv(packet_len + 2)
 
-    def exchange(self, data: bytes):
+    def exchange(self, data: bytes, timeout: int = 0):
         self.write(data)
         return self.read()