mitmproxy_replace_host.py

mitmproxy_replace_host.py

import argparse
import sys
from mitmproxy import ctx

class Replacer:
    def __init__(self, src, dst):
        self.src, self.dst = src, dst

    def request(self, flow):
        if flow.request.host == self.src:
            flow.request.host = self.dst
            flow.request.headers["Host"] = self.dst

    def response(self, flow):
        headers = flow.response.headers
        loc_key = "Location"
        if loc_key in headers and self.dst in headers[loc_key]:
            headers[loc_key] = headers[loc_key].replace(self.dst, self.src)

def start():
    if len(sys.argv) < 3:
        ctx.log.info('Usage: mitmproxy -s "%s [src domain name] [dst domain name]"' % sys.argv[0])
        ctx.log.info('Example: mitmproxy -s "%s example-a.com example-b.com"' % sys.argv[0])
        sys.exit(1)

    parser = argparse.ArgumentParser()
    parser.add_argument("src", type=str)
    parser.add_argument("dst", type=str)
    args = parser.parse_args()
    return Replacer(args.src, args.dst)