python-reverse-proxy/proxy.py at master · MollardMichael/python-reverse-proxy · GitHub

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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#!/usr/bin/env python3
fromhttp.serverimportBaseHTTPRequestHandler,HTTPServer
importargparse, os, random, sys, requests

fromsocketserverimportThreadingMixIn
importthreading


hostname='en.wikipedia.org'

defmerge_two_dicts(x, y):
returnx|y

defset_header():
headers= {
'Host': hostname
 }

returnheaders

classProxyHTTPRequestHandler(BaseHTTPRequestHandler):
protocol_version='HTTP/1.0'
defdo_HEAD(self):
self.do_GET(body=False)
return

defdo_GET(self, body=True):
sent=False
try:
url='https://{}{}'.format(hostname, self.path)
req_header=self.parse_headers()

print(req_header)
print(url)
resp=requests.get(url, headers=merge_two_dicts(req_header, set_header()), verify=False)
sent=True

self.send_response(resp.status_code)
self.send_resp_headers(resp)
msg=resp.text
ifbody:
self.wfile.write(msg.encode(encoding='UTF-8',errors='strict'))
return
finally:
ifnotsent:
self.send_error(404, 'error trying to proxy')

defdo_POST(self, body=True):
sent=False
try:
url='https://{}{}'.format(hostname, self.path)
content_len=int(self.headers.getheader('content-length', 0))
post_body=self.rfile.read(content_len)
req_header=self.parse_headers()

resp=requests.post(url, data=post_body, headers=merge_two_dicts(req_header, set_header()), verify=False)
sent=True

self.send_response(resp.status_code)
self.send_resp_headers(resp)
ifbody:
self.wfile.write(resp.content)
return
finally:
ifnotsent:
self.send_error(404, 'error trying to proxy')

defparse_headers(self):
req_header= {}
forlineinself.headers:
line_parts= [o.strip() foroinline.split(':', 1)]
iflen(line_parts) ==2:
req_header[line_parts[0]] =line_parts[1]
returnreq_header

defsend_resp_headers(self, resp):
respheaders=resp.headers
print ('Response Header')
forkeyinrespheaders:
ifkeynotin ['Content-Encoding', 'Transfer-Encoding', 'content-encoding', 'transfer-encoding', 'content-length', 'Content-Length']:
print (key, respheaders[key])
self.send_header(key, respheaders[key])
self.send_header('Content-Length', len(resp.content))
self.end_headers()

defparse_args(argv=sys.argv[1:]):
parser=argparse.ArgumentParser(description='Proxy HTTP requests')
parser.add_argument('--port', dest='port', type=int, default=9999,
help='serve HTTP requests on specified port (default: random)')
parser.add_argument('--hostname', dest='hostname', type=str, default='en.wikipedia.org',
help='hostname to be processd (default: en.wikipedia.org)')
args=parser.parse_args(argv)
returnargs

classThreadedHTTPServer(ThreadingMixIn, HTTPServer):
"""Handle requests in a separate thread."""

defmain(argv=sys.argv[1:]):
globalhostname
args=parse_args(argv)
hostname=args.hostname
print('http server is starting on {} port {}...'.format(args.hostname, args.port))
server_address= ('127.0.0.1', args.port)
httpd=ThreadedHTTPServer(server_address, ProxyHTTPRequestHandler)
print('http server is running as reverse proxy')
httpd.serve_forever()

if__name__=='__main__':
main()