diff options
author | Nikhil Soni <krsoninikhil@gmail.com> | 2017-02-23 16:19:18 +0530 |
---|---|---|
committer | Nikhil Soni <krsoninikhil@gmail.com> | 2017-03-03 12:34:36 +0530 |
commit | 317d183ba4eb78a16605da5e866726e8231a75fb (patch) | |
tree | 26bc814eff443a04e9c8afa070173afe2f4c8a18 | |
parent | 2e90373e4b484b94d0306b9764c7416b6156304f (diff) | |
download | mitmproxy-317d183ba4eb78a16605da5e866726e8231a75fb.tar.gz mitmproxy-317d183ba4eb78a16605da5e866726e8231a75fb.tar.bz2 mitmproxy-317d183ba4eb78a16605da5e866726e8231a75fb.zip |
Changes dns_spoofing example to use --keep-host-header
-rw-r--r-- | examples/complex/dns_spoofing.py | 11 | ||||
-rw-r--r-- | test/mitmproxy/test_examples.py | 2 |
2 files changed, 3 insertions, 10 deletions
diff --git a/examples/complex/dns_spoofing.py b/examples/complex/dns_spoofing.py index acda303d..2fd6b699 100644 --- a/examples/complex/dns_spoofing.py +++ b/examples/complex/dns_spoofing.py @@ -13,6 +13,8 @@ Usage: -s dns_spoofing.py # Used as the target location if neither SNI nor host header are present. -R http://example.com/ + # To avoid auto rewriting of host header by the reverse proxy target. + --keep-host-header mitmdump -p 80 -R http://localhost:443/ @@ -29,13 +31,6 @@ parse_host_header = re.compile(r"^(?P<host>[^:]+|\[.+\])(?::(?P<port>\d+))?$") class Rerouter: - def requestheaders(self, flow): - """ - The original host header is retrieved early - before flow.request is replaced by mitmproxy new outgoing request - """ - flow.metadata["original_host"] = flow.request.host_header - def request(self, flow): if flow.client_conn.ssl_established: flow.request.scheme = "https" @@ -46,7 +41,7 @@ class Rerouter: sni = None port = 80 - host_header = flow.metadata["original_host"] + host_header = flow.request.host_header m = parse_host_header.match(host_header) if m: host_header = m.group("host").strip("[]") diff --git a/test/mitmproxy/test_examples.py b/test/mitmproxy/test_examples.py index f3603fca..668d0d4a 100644 --- a/test/mitmproxy/test_examples.py +++ b/test/mitmproxy/test_examples.py @@ -114,13 +114,11 @@ class TestScripts(tservers.MasterTest): # Rewrite by reverse proxy mode f.request.scheme = "https" - f.request.host = "mitmproxy.org" f.request.port = 443 m.request(f) assert f.request.scheme == "http" - assert f.request.host == original_host assert f.request.port == 80 assert f.request.headers["Host"] == original_host |