.flush can raise NetlibDisconnect. This fixes a traceback found in fuzzing.

2024-11-23 08:11:00 +00:00 · 2014-11-07 16:01:41 +13:00 · 2014-11-07 16:01:41 +13:00 · 0811a9ebde
commit 0811a9ebde
parent 9ce2f473f6
1 changed files with 4 additions and 1 deletions
--- a/netlib/tcp.py
+++ b/netlib/tcp.py
@ -253,7 +253,10 @@ class _Connection(object):
        # Closing the socket is not our task, therefore we don't call close then.
        if type(self.connection) != SSL.Connection:
            if not getattr(self.wfile, "closed", False):
-                self.wfile.flush()
+                try:
+                    self.wfile.flush()
+                except NetLibDisconnect:
+                    pass

            self.wfile.close()
            self.rfile.close()