mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-22 15:37:45 +00:00
22 lines
682 B
Python
22 lines
682 B
Python
"""Process individual messages from a WebSocket connection."""
|
|
import re
|
|
from mitmproxy import ctx, http
|
|
|
|
|
|
def websocket_message(flow: http.HTTPFlow):
|
|
# get the latest message
|
|
message = flow.websocket.messages[-1]
|
|
|
|
# was the message sent from the client or server?
|
|
if message.from_client:
|
|
ctx.log.info(f"Client sent a message: {message.content}")
|
|
else:
|
|
ctx.log.info(f"Server sent a message: {message.content}")
|
|
|
|
# manipulate the message content
|
|
message.content = re.sub(r'^Hello', 'HAPPY', message.content)
|
|
|
|
if 'FOOBAR' in message.content:
|
|
# kill the message and not send it to the other endpoint
|
|
message.content = ""
|