1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
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
|
