mitmproxy使用(二)-自定义脚本编写

时间：2025-03-19 06:58:42
1、脚本编写的两个条件
  • 1.1、编写一个 py 文件供 mitmproxy 加载，文件中定义了若干函数，这些函数实现了某些 mitmproxy 提供的事件，mitmproxy 会在某个事件发生时调用对应的函数
  • 1.2、编写一个 py 文件供 mitmproxy 加载，文件定义了【变量 addons】，addons 是个数组，每个元素是一个类实例，这些类有若干方法，这些方法实现了某些 mitmproxy 提供的事件，mitmproxy 会在某个事件发生时调用对应的方法。这些类，称为一个个 addon。
  • 基本模板为
from mitmproxy import http, ctx
import json

class xxx:
	def xxx:
	def xxx

addons = [
	xxx() //类名的加载，也可以定义多个类，然后以数组的形式添加，进行加载
]
2、事件
  • 这里引入事件的概念，是方便我们理解我们在某些事件中做出的处理，事件是针对生命周期而言的。
  • 针对不同的生命周期分为5类，不同的生命周期是指在哪一个层面看待事件（http生命周期、 Websocket 生命周期、TCP 生命周期、网络连接生命周期等。主要是http生命周期用的多些）。
2.1、针对http请求的生命周期
  • http请求–>htpp响应的过程
  • 来自客户端的 HTTP CONNECT 请求。这里的连接指的是客户端跟代理服务器建立的连接。CONNECT 请求不会触发request、response的其他http事件。
  • 客户端的 HTTP 请求的头部被成功读取。这个时候flow的request的body部分内容是空的
  • 客户端的 HTTP 请求被成功完整读取。
  • 服务端的 HTTP 响应的头部被成功读取。此时 flow 中的 response 的 body 是空的。
  • 服务端端的 HTTP 响应被成功完整读取。
3、客户端请求修改
3.1、修改请求针对域名，替换请求链接。
from mitmproxy import ctx, http
import json

class Modify:

 	    def request(self, flow):
 	    	#替换请求链接
        	if ("/api/ios/info?method=priceList"):
            #有分享
             = "/share?id=2653&useShare=1"
            ("修改链接")
        		
         
addons = [
	Modify()
]
3.2、修改cookie
from mitmproxy import ctx, http
import json

class Modify:

 	 def request(self, flow):
         #替换cookie，两种匹配请求链接的方式
        # if  == "":
        if ("/"):
            print()
            print()
            ["_testCookie1"] = "xxx-91"
            ["testCookie2"] = "123"

            req = ["_testCookie1"]
            (req)
         
addons = [
	Modify()
]
3.3、修改请求参数
from mitmproxy import ctx, http
import json

class Modify:
 	 def request(self, flow):
                if ("/customActivity/bookcode/doJoin"):
            ("modify request form")
            if .urlencoded_form:
                .urlencoded_form["code"] = "11111"
            else:
                .urlencoded_form = [
                    ("actId", "20727"),("nick","name")
                ]
         
addons = [
	Modify()
]
4、响应修改
4.1、修改响应的状态码
from mitmproxy import ctx, http
import json

class Modify:
	def response(self, flow):
		 if (""):
             = (404)
            ("modify status code")
 
 addons = [
	Modify()
]
4.2、修改响应的消息体-直接修改响应字段
from mitmproxy import ctx, http
import json

class Modify:
	def response(self, flow):
        if ("/activityInfo/getPrizeInfo=="):
       		 //获取响应的json字符串，转成python对象进行解析和修改
            response = (.get_text())
            response['limitCount'] = 1
            //修改完成后，奖python对象转成json字符串，set进请求的响应体重发送给客户端
            .set_text((response))
            ('modify limitCount')
 
 addons = [
	Modify()
]
4.3、修改响应的消息体-通过读取json文件的字符串返给客户端
from mitmproxy import ctx, http
import json

class Modify:
	def response(self, flow):
        if ("/activityInfo/getPrizeInfo=="):
        	//读取文件，在当前文件路径下执行脚本，否则需要写文件的绝对路径；不然会找不到该json文件
       		 with open('','rb') as f:
       		 	//从json文件中读取数据成python对象
                res = (f)
            //将读取的python对象转成json字符串发送给客户端
            .set_text((res))
            ("modify order status")
 
 addons = [
	Modify()
]

相关文章