mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
edbb3d6791
* update ws injection example, fixes #4751 * Update websocket-inject-message.py * Revert "Update websocket-inject-message.py" This reverts commit 86b4cda2f4d6d694db7d0d8b30ec04c3ef123181. Co-authored-by: Maximilian Hils <github@maximilianhils.com> Co-authored-by: Maximilian Hils <git@maximilianhils.com>
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
"""
|
|
Inject a WebSocket message into a running connection.
|
|
|
|
This example shows how to inject a WebSocket message into a running connection.
|
|
"""
|
|
import asyncio
|
|
|
|
from mitmproxy import ctx, http
|
|
|
|
|
|
# Simple example: Inject a message as a response to an event
|
|
|
|
def websocket_message(flow: http.HTTPFlow):
|
|
assert flow.websocket is not None # make type checker happy
|
|
last_message = flow.websocket.messages[-1]
|
|
if last_message.is_text and "secret" in last_message.text:
|
|
last_message.drop()
|
|
ctx.master.commands.call("inject.websocket", flow, last_message.from_client, "ssssssh".encode())
|
|
|
|
|
|
# Complex example: Schedule a periodic timer
|
|
|
|
async def inject_async(flow: http.HTTPFlow):
|
|
msg = "hello from mitmproxy! "
|
|
assert flow.websocket is not None # make type checker happy
|
|
while flow.websocket.timestamp_end is None:
|
|
ctx.master.commands.call("inject.websocket", flow, True, msg.encode())
|
|
await asyncio.sleep(1)
|
|
msg = msg[1:] + msg[:1]
|
|
|
|
|
|
def websocket_start(flow: http.HTTPFlow):
|
|
asyncio.create_task(inject_async(flow))
|