kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-01-12 22:48:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Store original host in flow metadata

Browse Source
This commit is contained in:
Ammonite
2017-01-29 14:33:53 +01:00
parent 0ca1916f1b
commit c2c6050df3
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/complex/dns_spoofing.py
View File

@@ -29,15 +29,12 @@ parse_host_header = re.compile(r"^(?P<host>[^:]+|\[.+\])(?::(?P<port>\d+))?$")
class Rerouter:
def __init__(self):
self.host_header = None
def requestheaders(self, flow):
"""
The original host header is retrieved early
before flow.request is replaced by mitmproxy new outgoing request
"""
self.host_header = flow.request.headers.get('host')
flow.metadata["original_host"] = flow.request.headers["Host"]
def request(self, flow):
if flow.client_conn.ssl_established:
@@ -49,13 +46,14 @@ class Rerouter:
sni = None
port = 80
m = parse_host_header.match(self.host_header)
host_header = flow.metadata["original_host"]
m = parse_host_header.match(host_header)
if m:
self.host_header = m.group("host").strip("[]")
host_header = m.group("host").strip("[]")
if m.group("port"):
port = int(m.group("port"))
flow.request.host = sni or self.host_header
flow.request.host = sni or host_header
flow.request.port = port