python - 在 python 脚本中使用 mitmproxy

Question

我是 mitmproxy 的新手。但我不知道如何在 python 脚本中使用它。

我想将 mitmproxy 放入我的 python 脚本中，就像一个库一样，还具体化了端口或主机等所有内容，并在我的 python 脚本中使用 Request 或 Response 进行了一些修改。所以当我像这样开始我的脚本时

python sample.py

一切都将自动运行，无需像这样从命令行运行 mitmproxy

mitmproxy -s sample.py

感谢阅读。

Answer 1

你可以使用这样的东西。此代码取自 mithproxy github 上发布的一个问题，发现 here

from mitmproxy import proxy, options
from mitmproxy.tools.dump import DumpMaster
from mitmproxy.addons import core


class AddHeader:
    def __init__(self):
        self.num = 0

    def response(self, flow):
        self.num = self.num + 1
        print(self.num)
        flow.response.headers["count"] = str(self.num)


addons = [
    AddHeader()
]

opts = options.Options(listen_host='127.0.0.1', listen_port=8080)
pconf = proxy.config.ProxyConfig(opts)

m = DumpMaster(None)
m.server = proxy.server.ProxyServer(pconf)
# print(m.addons)
m.addons.add(addons)
print(m.addons)
# m.addons.add(core.Core())

try:
    m.run()
except KeyboardInterrupt:
    m.shutdown()