jingrow/gunicorn
1
0
Fork 0
mirror of https://github.com/frappe/gunicorn.git synced 2026-07-03 19:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix WebSocket binary send when text key is None

Browse Source 
ASGI allows websocket.send messages to contain both text and bytes
keys where one is None. Check for truthy values instead of key existence.
This commit is contained in:
Benoit Chesneau 2026-04-03 23:53:03 +02:00
parent 746cc049d0
commit 725c03e8ff
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gunicorn/asgi/websocket.py
View File

@ -145,10 +145,13 @@ class WebSocketProtocol:
if self.closed:
raise RuntimeError("WebSocket closed")
if "text" in message:
await self._send_frame(OPCODE_TEXT, message["text"].encode("utf-8"))
elif "bytes" in message:
await self._send_frame(OPCODE_BINARY, message["bytes"])
# Check for truthy values since both keys may be present with None
text = message.get("text")
bytes_data = message.get("bytes")
if text is not None:
await self._send_frame(OPCODE_TEXT, text.encode("utf-8"))
elif bytes_data is not None:
await self._send_frame(OPCODE_BINARY, bytes_data)
elif msg_type == "websocket.close":
code = message.get("code", CLOSE_NORMAL)