From 92f6b83b1685c5097266b3ae2f09b7ab728bf140 Mon Sep 17 00:00:00 2001
From: linuswck <linuswck@m-labs.hk>
Date: Mon, 17 Jun 2024 12:46:42 +0800
Subject: [PATCH] gui: Fix a bug which dfu cmd can cause connrst err - Kirdy
 may close and reset itself before host computer ends its session

---
 pykirdy/driver/kirdy_async.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pykirdy/driver/kirdy_async.py b/pykirdy/driver/kirdy_async.py
index fa0d859..c4ae76c 100644
--- a/pykirdy/driver/kirdy_async.py
+++ b/pykirdy/driver/kirdy_async.py
@@ -646,7 +646,11 @@ class Kirdy:
 
         # Reader needn't be closed
         self._writer.close()
-        await self._writer.wait_closed()
+        try:
+            await self._writer.wait_closed()
+        except ConnectionResetError:
+            # In Hard Reset/DFU cmd, Kirdy may close its socket first
+            pass
         self._reader = None
         self._writer = None