mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 16:17:49 +00:00
95 lines
3.3 KiB
Python
95 lines
3.3 KiB
Python
from libmproxy.proxy import ProxyError, ConnectionHandler
|
|
from netlib import http
|
|
|
|
|
|
def handle_messages(conntype, connection_handler):
|
|
handler = None
|
|
if conntype == "http":
|
|
handler = HTTPHandler(connection_handler)
|
|
else:
|
|
raise NotImplementedError
|
|
|
|
return handler.handle_messages()
|
|
|
|
|
|
class ConnectionTypeChange(Exception):
|
|
pass
|
|
|
|
|
|
class ProtocolHandler(object):
|
|
def __init__(self, c):
|
|
self.c = c
|
|
|
|
|
|
class HTTPHandler(ProtocolHandler):
|
|
|
|
def handle_messages(self):
|
|
while self.handle_request():
|
|
pass
|
|
self.c.close = True
|
|
|
|
def handle_request(self):
|
|
request = self.read_request()
|
|
if request is None:
|
|
return
|
|
raise NotImplementedError
|
|
|
|
def read_request(self):
|
|
self.c.client_conn.rfile.reset_timestamps()
|
|
|
|
request_line = self.get_line(self.c.client_conn.rfile)
|
|
method, path, httpversion = http.parse_init(request_line)
|
|
headers = self.read_headers(authenticate=True)
|
|
|
|
if self.c.mode == "regular":
|
|
if method == "CONNECT":
|
|
r = http.parse_init_connect(request_line)
|
|
if not r:
|
|
raise ProxyError(400, "Bad HTTP request line: %s"%repr(request_line))
|
|
host, port, _ = r
|
|
if self.c.config.forward_proxy:
|
|
#FIXME: Treat as request, no custom handling
|
|
self.c.server_conn.wfile.write(request_line)
|
|
for key, value in headers.items():
|
|
self.c.server_conn.wfile.write("%s: %s\r\n"%(key, value))
|
|
self.c.server_conn.wfile.write("\r\n")
|
|
else:
|
|
self.c.server_address = (host, port)
|
|
self.c.establish_server_connection()
|
|
|
|
self.c.handle_ssl()
|
|
self.c.determine_conntype("transparent", host, port)
|
|
raise ConnectionTypeChange
|
|
else:
|
|
r = http.parse_init_proxy(request_line)
|
|
if not r:
|
|
raise ProxyError(400, "Bad HTTP request line: %s"%repr(request_line))
|
|
method, scheme, host, port, path, httpversion = r
|
|
if not self.c.config.forward_proxy:
|
|
if (not self.c.server_conn) or (self.c.server_address != (host, port)):
|
|
self.c.server_address = (host, port)
|
|
self.c.establish_server_connection()
|
|
|
|
def get_line(self, fp):
|
|
"""
|
|
Get a line, possibly preceded by a blank.
|
|
"""
|
|
line = fp.readline()
|
|
if line == "\r\n" or line == "\n": # Possible leftover from previous message
|
|
line = fp.readline()
|
|
return line
|
|
|
|
def read_headers(self, authenticate=False):
|
|
headers = http.read_headers(self.c.client_conn.rfile)
|
|
if headers is None:
|
|
raise ProxyError(400, "Invalid headers")
|
|
if authenticate and self.c.config.authenticator:
|
|
if self.c.config.authenticator.authenticate(headers):
|
|
self.c.config.authenticator.clean(headers)
|
|
else:
|
|
raise ProxyError(
|
|
407,
|
|
"Proxy Authentication Required",
|
|
self.c.config.authenticator.auth_challenge_headers()
|
|
)
|
|
return headers |