diff --git a/python/omnisocket_a_side/daemon.py b/python/omnisocket_a_side/daemon.py index e4fb251..453a965 100644 --- a/python/omnisocket_a_side/daemon.py +++ b/python/omnisocket_a_side/daemon.py @@ -971,8 +971,11 @@ class OmniDaemonHTTPHandler(BaseHTTPRequestHandler): self.send_header("Content-Length", str(len(payload))) self.send_header("Cache-Control", "no-store") self.send_header("Connection", "keep-alive") - self.end_headers() - self.wfile.write(payload) + try: + self.end_headers() + self.wfile.write(payload) + except (BrokenPipeError, ConnectionResetError): + return class ASideOmniDaemon: diff --git a/python/omnisocket_b_side/daemon.py b/python/omnisocket_b_side/daemon.py index 9ea9eba..c6fa064 100644 --- a/python/omnisocket_b_side/daemon.py +++ b/python/omnisocket_b_side/daemon.py @@ -1208,8 +1208,11 @@ class OmniDaemonHTTPHandler(BaseHTTPRequestHandler): self.send_header("Content-Length", str(len(body))) self.send_header("Cache-Control", "no-store") self.send_header("Connection", "keep-alive") - self.end_headers() - self.wfile.write(body) + try: + self.end_headers() + self.wfile.write(body) + except (BrokenPipeError, ConnectionResetError): + return class BSideOmniDaemon: