diff options
Diffstat (limited to 'libmproxy/proxy/server.py')
| -rw-r--r-- | libmproxy/proxy/server.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/libmproxy/proxy/server.py b/libmproxy/proxy/server.py index c77ab2a8..e0f058b5 100644 --- a/libmproxy/proxy/server.py +++ b/libmproxy/proxy/server.py @@ -3,7 +3,8 @@ from OpenSSL import SSL from netlib import tcp from .primitives import ProxyServerError, Log, ProxyError, ConnectionTypeChange, AddressPriority from .connection import ClientConnection, ServerConnection -from .. import version, protocol +from ..protocol.handle import handle_messages, handle_error +from .. import version class DummyServer: @@ -81,14 +82,14 @@ class ConnectionHandler: while not self.close: try: - protocol.handle_messages(self.conntype, self) + handle_messages(self.conntype, self) except ConnectionTypeChange: self.log("Connection Type Changed: %s" % self.conntype) continue # FIXME: Do we want to persist errors? except (ProxyError, tcp.NetLibError), e: - protocol.handle_error(self.conntype, self, e) + handle_error(self.conntype, self, e) except Exception, e: self.log(e.__class__) import traceback |