This repository has been archived on 2024-09-10. You can view files and clone it, but cannot push or open issues or pull requests.
Douyin_Web_Live/proxy_script.py

26 lines
741 B
Python

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