mitmproxy/examples/complex/change_upstream_proxy.py

25 lines
908 B
Python
Raw Normal View History

2014-11-10 14:51:26 +00:00
# This scripts demonstrates how mitmproxy can switch to a second/different upstream proxy
# in upstream proxy mode.
#
# Usage: mitmdump -U http://default-upstream-proxy.local:8080/ -s change_upstream_proxy.py
#
# If you want to change the target server, you should modify flow.request.host and flow.request.port
2015-05-30 00:03:28 +00:00
def proxy_address(flow):
# Poor man's loadbalancing: route every second domain through the alternative proxy.
if hash(flow.request.host) % 2 == 1:
return ("localhost", 8082)
else:
return ("localhost", 8081)
2016-07-08 01:37:33 +00:00
def request(flow):
if flow.request.method == "CONNECT":
# If the decision is done by domain, one could also modify the server address here.
# We do it after CONNECT here to have the request data available as well.
return
address = proxy_address(flow)
if flow.live:
2016-05-29 08:23:39 +00:00
flow.live.change_upstream_proxy_server(address)