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))