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