jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-29 08:07:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix ingress websocket forward not closing the websocket

Browse Source 
Becuase the async iterator was used, it would stop iteration as soon
as a close message was seen. This meant we would never send close to
the other side
This commit is contained in:
J. Nick Koston 2024-09-25 07:56:47 -05:00
parent 10b9e3b29c
commit 98601da3ea
No known key found for this signature in database
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
homeassistant/components/hassio/ingress.py
View File

@ -284,23 +284,32 @@ def _is_websocket(request: web.Request) -> bool:
)
_CLOSE_TYPES = {
aiohttp.WSMsgType.CLOSE,
aiohttp.WSMsgType.CLOSING,
aiohttp.WSMsgType.CLOSED,
}
async def _websocket_forward(
ws_from: web.WebSocketResponse | ClientWebSocketResponse,
ws_to: web.WebSocketResponse | ClientWebSocketResponse,
) -> None:
"""Handle websocket message directly."""
try:
async for msg in ws_from:
if msg.type is aiohttp.WSMsgType.TEXT:
while msg := await ws_from.receive():
msg_type = msg.type
if msg_type is aiohttp.WSMsgType.TEXT:
await ws_to.send_str(msg.data)
elif msg.type is aiohttp.WSMsgType.BINARY:
elif msg_type is aiohttp.WSMsgType.BINARY:
await ws_to.send_bytes(msg.data)
elif msg.type is aiohttp.WSMsgType.PING:
elif msg_type is aiohttp.WSMsgType.PING:
await ws_to.ping()
elif msg.type is aiohttp.WSMsgType.PONG:
elif msg_type is aiohttp.WSMsgType.PONG:
await ws_to.pong()
elif ws_to.closed:
elif msg_type in _CLOSE_TYPES:
await ws_to.close(code=ws_to.close_code, message=msg.extra) # type: ignore[arg-type]
break
except RuntimeError:
_LOGGER.debug("Ingress Websocket runtime error")
except ConnectionResetError: