From 9429f1bc7b33846c71d7c06f92506a4fc4c68049 Mon Sep 17 00:00:00 2001
From: smill <smill@cuckoo.sh>
Date: Wed, 14 Sep 2016 19:10:13 +0000
Subject: [PATCH] Prevent crash in case of a connection timeout.

---
 netlib/tcp.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/netlib/tcp.py b/netlib/tcp.py
index c3b8a4076..4e988ee3a 100644
--- a/netlib/tcp.py
+++ b/netlib/tcp.py
@@ -484,12 +484,14 @@ class _Connection(object):
         if not isinstance(self.connection, SSL.Connection):
             if not getattr(self.wfile, "closed", False):
                 try:
-                    self.wfile.flush()
-                    self.wfile.close()
+                    if self.wfile:
+                        self.wfile.flush()
+                        self.wfile.close()
                 except exceptions.TcpDisconnect:
                     pass
 
-            self.rfile.close()
+            if self.rfile:
+                self.rfile.close()
         else:
             try:
                 self.connection.shutdown()