diff options
Diffstat (limited to 'libmproxy/protocol2/layer.py')
-rw-r--r-- | libmproxy/protocol2/layer.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/libmproxy/protocol2/layer.py b/libmproxy/protocol2/layer.py index e9f5c667..2775845e 100644 --- a/libmproxy/protocol2/layer.py +++ b/libmproxy/protocol2/layer.py @@ -35,7 +35,7 @@ import threading from netlib import tcp from ..proxy import Log from ..proxy.connection import ServerConnection -from .messages import Connect, Reconnect, ChangeServer +from .messages import Connect, Reconnect, ChangeServer, Kill from ..exceptions import ProtocolException @@ -116,6 +116,9 @@ class ServerConnectionMixin(object): return True elif message == ChangeServer: raise NotImplementedError + elif message == Kill: + self._disconnect() + return False @property |