From a595bcf731af78ef0b8ad3cf1a5e81e5f8b0e6d3 Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Sun, 5 Jun 2022 09:38:24 +0800
Subject: [PATCH] =?UTF-8?q?mitm=20script=E4=BD=8D=E7=BD=AE=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 proxy_script.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 proxy_script.py

diff --git a/proxy_script.py b/proxy_script.py
new file mode 100644
index 0000000..8ee8c68
--- /dev/null
+++ b/proxy_script.py
@@ -0,0 +1,25 @@
+# ! IMPORT ! make sure you ran mitmproxy with this script, 
+# eg: `/path/to/mitmproxy -s mitmproxy.py`
+import time
+from mitmproxy import http
+import re
+import requests
+
+session = requests.session()
+
+
+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]
+            if message.from_client:
+                return
+            content = message.content
+            session.post("http://127.0.0.1:5000/message", headers={
+                "X-MITM-TS": str(time.time()),
+                "X_REFERER": flow.request.host
+            }, data=content, timeout=(1, 1))
+
+
+addons = [Writer()]