From 1cd43be0c5edacc0cd0b50bc6561245cbedf4cf1 Mon Sep 17 00:00:00 2001 From: Jerry Yan <792602257@qq.com> Date: Fri, 3 Jun 2022 23:56:33 +0800 Subject: [PATCH] =?UTF-8?q?=E8=87=AA=E5=B7=B1=E5=8F=91=E5=87=BA=E7=9A=84we?= =?UTF-8?q?bsocket=E4=BF=A1=E6=81=AF=E4=B8=8D=E5=86=8D=E5=8F=91=E9=80=81?= =?UTF-8?q?=E8=AF=B7=E6=B1=82=E4=BA=86=EF=BC=88=E5=9B=A0=E4=B8=BA=E4=BC=9A?= =?UTF-8?q?=E9=80=A0=E6=88=90=E6=8A=A5=E9=94=99=EF=BC=89=EF=BC=8C=E7=8E=B0?= =?UTF-8?q?=E5=A4=84=E7=90=86=E5=B3=B0=E5=80=BC=E4=B8=BA1w/min=E6=B6=88?= =?UTF-8?q?=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/mitmproxy.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/scripts/mitmproxy.py b/scripts/mitmproxy.py index e11b25c..8169ec3 100644 --- a/scripts/mitmproxy.py +++ b/scripts/mitmproxy.py @@ -4,7 +4,6 @@ import time from mitmproxy import http import re import requests -import base64 session = requests.session() @@ -13,9 +12,12 @@ class Writer: def websocket_message(self, flow: http.HTTPFlow): re_c = re.search('webcast\d-ws-web-.*\.douyin\.com', flow.request.host) if re_c: - message = flow.websocket.messages[-1].content - session.post("http://127.0.0.1:5000/message", data=message, headers={ - "X-MITM_TS": str(time.time()), + message = flow.websocket.messages[-1] + if message.from_client: + return + content = message.content + session.post("http://127.0.0.1:5000/message", data=content, headers={ + "X-MITM-TS": str(time.time()), "X_REFERER": flow.request.host }, timeout=(1, 1))