service_connection: ignore IncompleteReadError in aio_recvall()

doronz88 · Oct 20, 2024 · 9188102 · 9188102
1 parent b013644
commit 9188102
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pymobiledevice3/service_connection.py b/pymobiledevice3/service_connection.py
@@ -157,7 +157,13 @@ def recv_prefixed(self, endianity='>') -> bytes:
 
     async def aio_recvall(self, size: int) -> bytes:
         """ receive a payload """
-        return await self.reader.readexactly(size)
+        data = b''
+        while len(data) < size:
+            try:
+                data += await self.reader.readexactly(size)
+            except asyncio.IncompleteReadError:
+                pass
+        return data
 
     async def aio_recv_prefixed(self, endianity='>') -> bytes:
         """ receive a data block prefixed with a u32 length field """