mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
9af8f4bb31
This commit is largely based on work by Thiago Arrais (@thiagoarrais) and Shane Bradfield (@l33tLumberjack). I wasn't really able to get their PR reasonably merged onto the latest master, so I reapplied their changes manually here and did some further improvements on that.
25 lines
908 B
Python
25 lines
908 B
Python
# 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
|
|
|
|
|
|
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)
|
|
|
|
|
|
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:
|
|
flow.live.change_upstream_proxy_server(address)
|