From 796ef08c2aaa8357f73372de1324697f972cf5d0 Mon Sep 17 00:00:00 2001
From: Kyle Altendorf <sda@fstab.net>
Date: Mon, 11 Nov 2024 20:44:19 -0500
Subject: [PATCH] check network errors and their `.__cause__` for expected
 error types

---
 chia/server/ws_connection.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/chia/server/ws_connection.py b/chia/server/ws_connection.py
index 17d59924a05d..330105fe3da8 100644
--- a/chia/server/ws_connection.py
+++ b/chia/server/ws_connection.py
@@ -378,8 +378,9 @@ async def outbound_handler(self) -> None:
         except asyncio.CancelledError:
             pass
         except Exception as e:
+            expected_types = (BrokenPipeError, ConnectionResetError, TimeoutError)
             expected = False
-            if isinstance(e, (BrokenPipeError, ConnectionResetError, TimeoutError)):
+            if isinstance(e, expected_types) or isinstance(e.__cause__, expected_types):
                 expected = True
             elif isinstance(e, OSError):
                 if e.errno in {113}: