mitmproxy开发笔记–干货分享

0x01 工具介绍

mitmproxy是一个支持HTTP和HTTPS的抓包程序,有类似Fiddler、Charles的功能。并且支持Python Api、WEB接口,除了命令行形式的控制台,mitmproxy还有两个关联组件:mitmdump和mitmweb。

0x02 事件

针对不同的生命周期分为5类,不同的生命周期是指在哪一个层面看待事件(http生命周期、 Websocket 生命周期、TCP 生命周期、网络连接生命周期等。主要是http生命周期用的多些)。(这里只介绍HTTP)。

def http_connect(self, flow: mitmproxy.http.HTTPFlow): 收到了来自客户端的 HTTP CONNECT 请求。
def requestheaders(self, flow: mitmproxy.http.HTTPFlow):来自客户端的 HTTP 请求的头部被成功读取。
def request(self, flow: mitmproxy.http.HTTPFlow):来自客户端的 HTTP 请求被成功完整读取。
def responseheaders(self, flow: mitmproxy.http.HTTPFlow):来自服务端的 HTTP 响应的头部被成功读取。
def response(self, flow: mitmproxy.http.HTTPFlow):来自服务端端的 HTTP 响应被成功完整读取。
def error(self, flow: mitmproxy.http.HTTPFlow):发生了一个 HTTP 错误。

0x03 客户端请求修改

request常见属性:url、headers、method、cookies、host、method
例子:改一个返回属性
def response(flow):
    flow.response.text=flow.response.text.replace('ChromeDriver','')

修改请求针对域名,替换请求链接

from mitmproxy import ctx, http
import json

class Modify:

        def request(self, flow):
            #替换请求链接
            if flow.request.url.startswith("http://spay1.shuqireader.com/api/ios/info?method=priceList"):
            #有分享
            flow.request.url = "http://activity.x.xxx.xx.cn/share?id=2653&useShare=1"
            ctx.log.info("修改链接")

addons = [
    Modify()
]

修改cookie

from mitmproxy import ctx, http
import json

class Modify:

     def request(self, flow):
         #替换cookie,两种匹配请求链接的方式
        # if flow.request.host == "xxx.x.xxx.com.cn":
        if flow.request.url.startswith("https://xxx.x.xxx.com.cn/"):
            print(flow.request.url)
            print(flow.request.cookies)
            flow.request.cookies["_testCookie1"] = "xxx-91"
            flow.request.cookies["testCookie2"] = "123"

            req = flow.request.cookies["_testCookie1"]
            ctx.log.info(req)

addons = [
    Modify()
]

修改请求参数

from mitmproxy import ctx, http
import json

class Modify:
     def request(self, flow):
                if flow.request.url.startswith("http://xxx.x.xxx.com.cn/customActivity/bookcode/doJoin"):
            ctx.log.info("modify request form")
            if flow.request.urlencoded_form:
                flow.request.urlencoded_form["code"] = "11111"
            else:
                flow.request.urlencoded_form = [
                    ("actId", "20727"),("nick","name")
                ]

addons = [
    Modify()
]

0x04 响应修改

修改响应的状态码

from mitmproxy import ctx, http
import json

class Modify:
    def response(self, flow):
         if flow.request.url.startswith("http://baidu.com.cn"):
            flow.response = http.HTTPResponse.make(404)
            ctx.log.info("modify status code")

 addons = [
    Modify()
]

修改响应的消息体-直接修改响应字段

from mitmproxy import ctx, http
import json

class Modify:
    def response(self, flow):
        if flow.request.url.startswith("https://xxx.x.xxx.com.cn/activityInfo/getPrizeInfo=="):
             //获取响应的json字符串,转成python对象进行解析和修改
            response = json.loads(flow.response.get_text())
            response['limitCount'] = 1
            //修改完成后,奖python对象转成json字符串,set进请求的响应体重发送给客户端
            flow.response.set_text(json.dumps(response))
            ctx.log.info('modify limitCount')

 addons = [
    Modify()
]

修改响应的消息体-通过读取json文件的字符串返给客户端

from mitmproxy import ctx, http
import json

class Modify:
    def response(self, flow):
        if flow.request.url.startswith("https://xxx.x.xxx.com.cn/activityInfo/getPrizeInfo=="):
            //读取文件,在当前文件路径下执行脚本,否则需要写文件的绝对路径;不然会找不到该json文件
             with open('getStatus.json','rb') as f:
                //从json文件中读取数据成python对象
                res = json.load(f)
            //将读取的python对象转成json字符串发送给客户端
            flow.response.set_text(json.dumps(res))
            ctx.log.info("modify order status")

 addons = [
    Modify()
]

0x05 简单的例子

** 替****换****请求****内容**

import re
from mitmproxy import ctx

def request(flow):
    remote_host = flow.request.pretty_host
    method = flow.request.method

    ctx.log.info('remote_host %s'%remote_host)
    ctx.log.info('request.host %s' % flow.request.host)
    ctx.log.info('request.path %s' % flow.request.path)
    ctx.log.info('request.pretty_url %s' % flow.request.pretty_url)
    ctx.log.info('request.query %s' % flow.request.query)
    ctx.log.info('request.method %s' % flow.request.method)
    ctx.log.info('client host %s' % flow.client_conn.address.host)

    flow.request.replace('Mozilla', 'Google', re.M)

其中:

  • flow.request.pretty_host 请求的域名。如果希望对不同的网站采取不同的策略,可使用该值。
  • flow.request.host 请求的 IP 地址。
  • flow.request.path 请求的路径。如果希望对不同的请求路径采用不同的策略,可使用该值。
  • flow.client_conn.address.host 客户端 IP。如果希望对不同的客户端采取不同的策略,可使用该值。

ctx.log.info 可以将内容打印在 mitmproxy 的 Event log 窗口中(采用快捷键 e 打开) ,下面是访问网站 https://zengrong.net 的 log 内容:

remote_host zengrong.net
request.host 121.40.23.108
request.path /
request.pretty_url https://zengrong.net/
request.query MultiDictView[]
request.method GET
client host 192.168.43.196

上面代码中最后还有一句 replace ,是将请求中的所有 Mozilla 字符串替换成 Google 。这里的“所有”包含下面的内容:header、path、body。

动态修改代理

有 5 个客户端,我希望这 5 个客户端在访问同一个网站的时候,使用不同的代理服务器。该如何处理?

def request(flow):
    client_ip = flow.client_conn.address.host
    if client_ip == '192.168.0.10':
        proxy = ('1.2.3.4', 8080)
    else:
        proxy = ('3.4.5.6', 8080)
    if flow.live:
        flow.live.change_upstream_proxy_server(proxy)

0x06 参考链接

https://blog.csdn.net/qianwenjun_19930314/article/details/88227335
https://blog.csdn.net/rookie_is_me/article/details/86571757
https://blog.zengrong.net/post/2653.html
官网例子:https://github.com/mitmproxy/mitmproxy

你可能感兴趣的:(mitmproxy开发笔记–干货分享)