add timeout for send_recv_packet

LeeBoNorth · Apr 19, 2022 · 1dcc9e2 · 1dcc9e2
1 parent f985f26
commit 1dcc9e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tidevice/_imagemounter.py b/tidevice/_imagemounter.py
@@ -95,7 +95,6 @@ def prepare(self):
         """
         Note: LookupImage might stuck and no response
         """
-        self.psock.get_socket().settimeout(10.0)
         return super().prepare()
 
     def lookup(self, image_type="Developer") -> List[bytes]:

diff --git a/tidevice/_safe_socket.py b/tidevice/_safe_socket.py
@@ -198,9 +198,10 @@ def recv_packet(self, header_size=None) -> dict:
             logger.debug("RECV(%d): %s", self.id, payload)
         return payload
 
-    def send_recv_packet(self, payload: dict) -> dict:
-        self.send_packet(payload)
-        return self.recv_packet()
+    def send_recv_packet(self, payload: dict, timeout: float = 10.0) -> dict:
+        with set_socket_timeout(self.get_socket(), timeout):
+            self.send_packet(payload)
+            return self.recv_packet()
 
 
 class PlistSocketProperty: