适用于 mitmproxy v5.3
mitmproxy 本身是个 Python 包,比如mitmweb
命令实际是如下代码
#!/Users/niconi/venv/bin/python3
# -*- coding: utf-8 -*-
import re
import sys
from mitmproxy.tools.main import mitmweb
if __name__ == '__main__':
sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
sys.exit(mitmweb())
mitmproxy 提供了 addon 的方式编写代理规则,比如如下 addon 给每个请求加上一个请求头 mitmproxy:default
:
# addon.py
from mitmproxy import contentviews, http
class Rewrite:
flag = 'default'
@classmethod
def set_flag(self, value):
self.flag = value
def request(self, flow):
flow.request.headers['mitmproxy'] = self.flag
addons = [
Rewrite()
]
如果只是想在一个 Python 程序中启动 mitmproxy,可以用 subprocess
,但因为是另一个内存空间,很难做到在程序运行时控制 addon,用线程可以达到这个效果:
# mitmweb_wraper.py
from mitmproxy import proxy, options
from mitmproxy.tools.web.master import WebMaster
import threading,asyncio
import addon
class MitmWebWraper(object):
mitmweb = None
thread = None
def loop_in_thread(self, loop, mitmweb):
asyncio.set_event_loop(loop)
mitmweb.run()
def start(self):
opts = options.Options(listen_host='0.0.0.0', listen_port=8080,
confdir='~/.mitmproxy', ssl_insecure=True)
pconf = proxy.config.ProxyConfig(opts)
self.mitmweb = WebMaster(opts)
self.mitmweb.server = proxy.server.ProxyServer(pconf)
self.mitmweb.addons.add(addon)
loop = asyncio.get_event_loop()
self.thread = threading.Thread(
target=self.loop_in_thread, args=(loop, self.mitmweb))
try:
self.thread.start()
except KeyboardInterrupt:
self.mitmweb.shutdown()
if __name__ == "__main__":
mitmweb_wraper = MitmWebWraper()
mitmweb_wraper.start()
在另一个 Python 程序中,通过 mitmweb_wraper 启动 mitmweb,并且 import addon,对 addon 的修改会立即生效,新增的请求头会变成 mitmproxy:changed
:
import mitmweb_wraper, addon
mitmweb_wraper = mitmweb_wraper.MitmWebWraper()
mitmweb_wraper.start()
addon.Rewrite().set_flag('changed')