Fix AttributeError instead of EOFError

Gallopsled · Jul 8, 2019 · dce8fde · dce8fde
1 parent b0e1ce3
commit dce8fde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pwnlib/tubes/sock.py b/pwnlib/tubes/sock.py
@@ -65,7 +65,7 @@ def send_raw(self, data):
             self.sock.sendall(data)
         except IOError as e:
             eof_numbers = [errno.EPIPE, errno.ECONNRESET, errno.ECONNREFUSED]
-            if e.message == 'Socket is closed' or e.errno in eof_numbers:
+            if e.errno in eof_numbers or 'Socket is closed' in e.args:
                 self.shutdown("send")
                 raise EOFError
             else: