diff options
Diffstat (limited to 'libmproxy/protocol/rawtcp.py')
| -rw-r--r-- | libmproxy/protocol/rawtcp.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/libmproxy/protocol/rawtcp.py b/libmproxy/protocol/rawtcp.py index 24c19523..5f08fd17 100644 --- a/libmproxy/protocol/rawtcp.py +++ b/libmproxy/protocol/rawtcp.py @@ -5,8 +5,9 @@ import six import sys from OpenSSL import SSL +from netlib.exceptions import TcpException -from netlib.tcp import NetLibError, ssl_read_select +from netlib.tcp import ssl_read_select from netlib.utils import clean_bin from ..exceptions import ProtocolException from .base import Layer @@ -64,7 +65,7 @@ class RawTCPLayer(Layer): "info" ) - except (socket.error, NetLibError, SSL.Error) as e: + except (socket.error, TcpException, SSL.Error) as e: six.reraise( ProtocolException, ProtocolException("TCP connection closed unexpectedly: {}".format(repr(e))), |