Merge pull request #18532 from andresriancho/fix/EOF-error

EOF error
andresriancho · Apr 29, 2020 · fd00d3a · fd00d3a
2 parents 0ead996 + 8e82e19
commit fd00d3a
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 4 deletions.
diff --git a/w3af/core/data/url/handlers/keepalive/connections.py b/w3af/core/data/url/handlers/keepalive/connections.py
@@ -33,7 +33,7 @@
 from .utils import debug
 
 from w3af.core.controllers.exceptions import HTTPRequestException
-from w3af.core.data.url.openssl.ssl_wrapper import wrap_socket
+from w3af.core.data.url.openssl_wrapper.ssl_wrapper import wrap_socket
 
 
 class UniqueID(object):

diff --git a/w3af/core/data/url/openssl/__init__.py → ...core/data/url/openssl_wrapper/__init__.py b/w3af/core/data/url/openssl/__init__.py → ...core/data/url/openssl_wrapper/__init__.py
diff --git a/w3af/core/data/url/openssl/ssl_wrapper.py → ...e/data/url/openssl_wrapper/ssl_wrapper.py b/w3af/core/data/url/openssl/ssl_wrapper.py → ...e/data/url/openssl_wrapper/ssl_wrapper.py
@@ -18,6 +18,7 @@
 import socket
 import select
 import OpenSSL
+from OpenSSL.SSL import SysCallError
 
 from ndg.httpsclient.subj_alt_name import SubjectAltName
 from pyasn1.codec.der.decoder import decode as der_decoder
@@ -122,7 +123,7 @@ def close(self):
     def recv(self, *args, **kwargs):
         try:
             data = self.ssl_conn.recv(*args, **kwargs)
-        except OpenSSL.SSL.ZeroReturnError:
+        except (OpenSSL.SSL.ZeroReturnError, SysCallError):
             # empty string signalling that the other side has closed the
             # connection or that some kind of error happen and no more reads
             # should be done on this socket

diff --git a/w3af/core/data/url/openssl/tests/__init__.py → ...ata/url/openssl_wrapper/tests/__init__.py b/w3af/core/data/url/openssl/tests/__init__.py → ...ata/url/openssl_wrapper/tests/__init__.py
diff --git a/...ata/url/openssl/tests/test_ssl_wrapper.py → ...openssl_wrapper/tests/test_ssl_wrapper.py b/...ata/url/openssl/tests/test_ssl_wrapper.py → ...openssl_wrapper/tests/test_ssl_wrapper.py
@@ -23,7 +23,7 @@
 import OpenSSL
 import ssl
 
-from w3af.core.data.url.openssl.ssl_wrapper import OpenSSLReformattedError
+from w3af.core.data.url.openssl_wrapper.ssl_wrapper import OpenSSLReformattedError
 
 
 class TestSSLError(unittest.TestCase):

diff --git a/w3af/plugins/audit/ssl_certificate.py b/w3af/plugins/audit/ssl_certificate.py
@@ -37,7 +37,7 @@
 from w3af.core.data.options.option_types import INPUT_FILE
 from w3af.core.data.options.option_list import OptionList
 from w3af.core.data.parsers.doc.url import URL
-from w3af.core.data.url.openssl.ssl_wrapper import wrap_socket
+from w3af.core.data.url.openssl_wrapper.ssl_wrapper import wrap_socket
 from w3af.core.data.kb.info import Info
 from w3af.core.data.kb.vuln import Vuln