tunneld: ignore ConnectionAbortedError raised from CoreDeviceProxy

doronz88 · Sep 11, 2024 · 26856a0 · 26856a0
1 parent f755482
commit 26856a0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pymobiledevice3/tunneld.py b/pymobiledevice3/tunneld.py
@@ -140,7 +140,8 @@ async def monitor_usbmux_task(self) -> None:
                             continue
                         try:
                             service = CoreDeviceTunnelProxy(create_using_usbmux(mux_device.serial))
-                        except (MuxException, InvalidServiceError, GetProhibitedError, construct.core.StreamError):
+                        except (MuxException, InvalidServiceError, GetProhibitedError, construct.core.StreamError,
+                                ConnectionAbortedError):
                             continue
                         self.tunnel_tasks[task_identifier] = TunnelTask(
                             udid=mux_device.serial,
@@ -380,7 +381,8 @@ async def clear_tunnels() -> fastapi.Response:
         @self._app.get('/cancel')
         async def cancel_tunnel(udid: str) -> fastapi.Response:
             self._tunneld_core.cancel(udid=udid)
-            data = json.dumps({'operation': 'cancel', 'udid': udid, 'data': True, 'message': f'tunnel {udid} Canceled ...'})
+            data = json.dumps(
+                {'operation': 'cancel', 'udid': udid, 'data': True, 'message': f'tunnel {udid} Canceled ...'})
             return generate_http_response(data)
 
         @self._app.get('/hello')