用 Python 的 mitmproxy 库做请求拦截、修改时遇到'NoneType' object has no attribute 'text'

class MyAdd:
    def request(self, flow):
        # do something in request
        if "pan.baidu.com" in flow.request.pretty_url:            
            flow.kill()
            
        if "test_kkk.cn" = flow.request.host:
            with open('./init') as f_init:
                flow.response.text = f_init.read()
                flow.response.status_code = 200
                
    def response(self, flow):
        if "baidu.com" not in flow.request.pretty_url:
            print(flow.response.text)

报错

    flow.response.text = f_init.read()
AttributeError: 'NoneType' object has no attribute 'text'

而如果把 if "test_kkk.cn" = flow.request.host 放到 response 函数下则能正确修改，但不能满足“直接返回给目标程序，不再到达服务器”的要求。该怎么实现呢

6 条回复 • 2020-09-25 13:24:17 +08:00

ljhaoboy

2020-09-25 11:24:14 +08:00

https://github.com/mitmproxy/mitmproxy/blob/v0.9.2/examples/redirect_requests.py
康康这个，好像是符合你的要求的

xiaolinjia

2020-09-25 11:27:08 +08:00

虽然没用过这个库，不过看报错就是因为 flow.response=None 。
于是推测原因是，在 request 方法里，这库还没有给 flow.response 对象实例化。
然后问题又来了，不到达服务器，又怎么有 response 响应呢？

xiaolinjia

2020-09-25 11:31:48 +08:00

@xiaolinjia 看了下楼上，自己在 request 方法里实例化个 flow.Response 对象应该是可行的。

ikamu

2020-09-25 12:44:45 +08:00

@ljhaoboy 看了下这个库，有点老以至于很多类的位置不再与目录文件对应。

@xiaolinjia 出现问题的原因明白了，得在某处实例个 flow.Response 对象，这个依旧有点不知该在哪添。

ikamu

2020-09-25 12:46:48 +08:00

结局：顺着一楼的链接中找到了新版的，https 。。。github 。com/mitmproxy/mitmproxy/blob/v5.x/examples/addons/http-reply-from-proxy.py 。虽然和实例个 flow.Response 对象，然后用 flow.response.text 进行赋值的预期目标有区别，但还算是符合。
```
def start():
Addon = MyAdd()
options = Options(listen_host='127.0.0.1', listen_port=8080)
config = ProxyConfig(options)
m = DumpMaster (options)
m.server = ProxyServer(config)
m.addons.add(Addon)
m.run()
```
调用方式，期待其他答案。

ikamu

2020-09-25 13:24:17 +08:00

方法找到了：添加 flow.response = http.HTTPResponse.make()
```
from mitmproxy import http

flow.response = http.HTTPResponse.make()
if "test_kkk.cn" = flow.request.host:
with open('./init') as f_init:
flow.response.text = f_init.read()
flow.response.status_code = 200
```