q792602257/Douyin_Web_Live
Archived
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

自己发出的websocket信息不再发送请求了（因为会造成报错），现处理峰值为1w/min消息

Browse Source
This commit is contained in:
Jerry Yan 2022-06-03 23:56:33 +08:00
parent 09da646b99
commit 1cd43be0c5
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/mitmproxy.py
View File

@ -4,7 +4,6 @@ import time
from mitmproxy import http from mitmproxy import http
import re import re
import requests import requests
import base64
session = requests.session() session = requests.session()
@ -13,9 +12,12 @@ class Writer:
def websocket_message(self, flow: http.HTTPFlow): def websocket_message(self, flow: http.HTTPFlow):
re_c = re.search('webcast\d-ws-web-.*\.douyin\.com', flow.request.host) re_c = re.search('webcast\d-ws-web-.*\.douyin\.com', flow.request.host)
if re_c: if re_c:
message = flow.websocket.messages[-1].content message = flow.websocket.messages[-1]
session.post("http://127.0.0.1:5000/message", data=message, headers={ if message.from_client:
"X-MITM_TS": str(time.time()), 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 "X_REFERER": flow.request.host
}, timeout=(1, 1)) }, timeout=(1, 1))