Fixed small issue!

Author: masterking32

dns_utils/ARQ.py

@@ -9,6 +9,8 @@
 
 
 class ARQStream:
+    _active_tasks = set()
+
     def __init__(
         self,
         stream_id,
@@ -53,6 +55,8 @@ def __init__(
         try:
             loop = asyncio.get_running_loop()
             self.io_task = loop.create_task(self._io_loop())
+            ARQStream._active_tasks.add(self.io_task)
+            self.io_task.add_done_callback(ARQStream._active_tasks.discard)
         except RuntimeError:
             self.io_task = None
 
@@ -93,9 +97,15 @@ async def _io_loop(self):
                         self.window_not_full.clear()
 
                 await self.enqueue_tx(3, self.stream_id, sn, raw_data)
+        except asyncio.CancelledError:
+            pass
+        except Exception as e:
+            if self.logger:
+                self.logger.debug(f"Stream {self.stream_id} IO loop error: {e}")
         finally:
             if not self.closed:
-                await self.close(reason="IO Loop Exit")
+                loop = asyncio.get_running_loop()
+                loop.create_task(self.close(reason="IO Loop Exit"))
 
     async def receive_data(self, sn, data):
         if self.closed:

server.py

@@ -522,9 +522,11 @@ async def handle_dns_requests(self) -> None:
                     continue
 
                 self.logger.error(f"Socket error: {e}. Exiting DNS request handler.")
+                await asyncio.sleep(0.1)
                 continue
             except Exception as e:
                 self.logger.exception(f"Unexpected error receiving DNS request: {e}")
+                await asyncio.sleep(0.1)
                 continue
             try:
                 task = self.loop.create_task(self._bounded_handle_request(data, addr))