通用技术 在 Python 程序中启动 mitmproxy 并控制 addon

黑水 · 2020年11月15日 · 最后由 ballballW5 回复于 2022年03月02日 · 6634 次阅读

适用于 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')
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
共收到 3 条回复 时间 点赞
黑水 【请教】pytest+mitmproxy 怎么联合使用? 中提及了此贴 06月23日 15:24

请教下 ,如果屏蔽 Web server listening

huaxiaotian 回复

什么意思?

谢谢,对我很有帮助

需要 登录 后方可回复, 如果你还没有账号请点击这里 注册