fix: correctly handle WebSocket message fragmentation #1057

ThomasFarstrike · ThomasFarstrike · commit 49fa41a9ea1a · 2025-11-11T23:16:08.000+01:00
diff --git a/internal_filesystem/lib/aiohttp/aiohttp_ws.py b/internal_filesystem/lib/aiohttp/aiohttp_ws.py
@@ -166,7 +166,10 @@ async def handshake(self, uri, ssl, req):
 
     async def receive(self):
         while True:
-            opcode, payload = await self._read_frame()
+            opcode, payload, final = await self._read_frame()
+            while not final:
+                _, morepayload, final = await self._read_frame() # original opcode must be preserved
+                payload += morepayload
             send_opcode, data = self._process_websocket_frame(opcode, payload)
             if send_opcode:  # pragma: no cover
                 await self.send(data, send_opcode)
@@ -206,7 +209,7 @@ async def _read_frame(self):
         payload = await self.reader.readexactly(length)
         if has_mask:  # pragma: no cover
             payload = bytes(x ^ mask[i % 4] for i, x in enumerate(payload))
-        return opcode, payload
+        return opcode, payload, fin
 
 
 class ClientWebSocketResponse:
