mitmproxy实践

前2篇已经说明了mitmproxy的基本使用方法，这篇主要讲讲在实际测试中的应用

首先附上github地址:https://github.com/mitmproxy/mitmproxy，上面的内容永远是最新的

1. 作为一名测试穿戴设备相关app的工程师，与数据打交道是常事，那么，如果想要获取app上传给服务器的文件，怎么获取？

上传文件，http请求头header的Content-Type字段multipart/form-data
注意：如果要修改请求参数，必须得了解Content_Type的类型，Content_Type主要告知请求的消息主要用了何种编码，常见3中类型：

• application/x-www-form-urlencoded
  form形式，提交的数据按照 key1=val1&key2=val2 的方式进行编码
• application/json
  JSON编码方式，消息主体是序列化后的 JSON 字符串
• multipart/form-data
  主要是传输二进制流等，boundary区分普通文件内容和二进制流内容

def request(flow: http.HTTPFlow) -> None:
    target=["/course/data/save"]
    url_path = flow.request.path
    print("完整收到请求(包括body部分)={}".format(url_path))
    if url_path.split("?")[0] in target:
        ctx.log.info("#"*50)
        ctx.log.info("课程接口")
        ctx.log.info(flow.request.url)
        fp=open("courseData.dat","wb")
        for key,value in flow.request.multipart_form.items():
            print(key)
            if key==b"courseData":#文件名
                fp.write(value)
        fp.close()

2. 接口返回字段变更，为了兼容，暂时未去掉，未来去掉，这种情况下，app需要做好兼容性测试，那么需要通过模拟删除相应字段

def response(self,flow: http.HTTPFlow) -> None:
        if flowfilter.match(self.filter, flow):
            url_path = flow.request.path
            if "***由于隐私该处省略***" in url_path:
                 ctx.log.warn("#" * 60)
                 res=json.loads(flow.response.text)
                 for i in res.get("data"):
                       i.pop('happenDate')#删除happenDate字段，这个地方得看返回内容的结构
                 flow.response.set_text(json.dumps(res))

3. 笔者测试的app在header中需要带上一段自定义的字段，当中途新增接口，有时候后开发会忘记添加，那么，测试人员在测试迭代需求的时候，需要检查一遍新增接口请求，阅读完接口后可编写脚本，待发布app时直接执行

def request(self,flow: http.HTTPFlow) -> None:
      if url_path.split("?")[0] in "***特定接口***":
                  ctx.log.warn("#" * 60)
                  dict_obj = json.loads(flow.request.headers.get("selfHeader","{}")) # 字典字符串,headers为字典对象

4.自定义options

默认options：
https://docs.mitmproxy.org/stable/concepts-options/

#自定义filter
def load(self, loader):
        """
        脚本载入时执行，对应--options参数，默认有多个可供选择的项，如flow_detail，同时可在载入脚本时自定义选项
        option格式：（name,typespec,defult,help）
        """
        loader.add_option("filter", str, "", "过滤规则")
  
def configure(self,updated):
        """脚本载入，读取配置"""
        self.filter =flowfilter.parse(ctx.options.filter) #得到一个过滤器

 def request(self,flow: http.HTTPFlow) -> None:
        if flowfilter.match(self.filter, flow):
            print("filter")

结语

在使用fitler这个过滤功能时，发现源代码有一处错误，如

class FDomain(_Rex):
    code = "d"
    help = "Domain"
    flags = re.IGNORECASE
    is_binary = False

    @only(http.HTTPFlow, websocket.WebSocketFlow)
    def __call__(self, f):
        if isinstance(f, websocket.WebSocketFlow):
            f = f.handshake_flow
        return bool(
            self.re.search(f.request.host) or
            self.re.search(f.request.pretty_host) #这里应该添加str，否则一直报错，其他地方类似
        )