mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 08:11:00 +00:00
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
KILL = 0 # const for killed requests
|
|
|
|
|
|
class ConnectionTypeChange(Exception):
|
|
"""
|
|
Gets raised if the connetion type has been changed (e.g. after HTTP/1.1 101 Switching Protocols).
|
|
It's up to the raising ProtocolHandler to specify the new conntype before raising the exception.
|
|
"""
|
|
pass
|
|
|
|
|
|
class ProtocolHandler(object):
|
|
def __init__(self, c):
|
|
self.c = c
|
|
"""@type : libmproxy.proxy.ConnectionHandler"""
|
|
|
|
def handle_messages(self):
|
|
"""
|
|
This method gets called if a client connection has been made. Depending on the proxy settings,
|
|
a server connection might already exist as well.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
def handle_error(self, error):
|
|
"""
|
|
This method gets called should there be an uncaught exception during the connection.
|
|
This might happen outside of handle_messages, e.g. if the initial SSL handshake fails in transparent mode.
|
|
"""
|
|
raise error
|
|
|
|
from . import http, tcp
|
|
|
|
protocols = dict(
|
|
http = dict(handler=http.HTTPHandler, flow=http.HTTPFlow),
|
|
tcp = dict(handler=tcp.TCPHandler),
|
|
)
|
|
|
|
|
|
def _handler(conntype, connection_handler):
|
|
if conntype in protocols:
|
|
return protocols[conntype]["handler"](connection_handler)
|
|
|
|
raise NotImplementedError
|
|
|
|
|
|
def handle_messages(conntype, connection_handler):
|
|
return _handler(conntype, connection_handler).handle_messages()
|
|
|
|
|
|
def handle_error(conntype, connection_handler, error):
|
|
return _handler(conntype, connection_handler).handle_error(error) |