Method _read_frames

gunicorn/asgi/websocket.py:214–260

Read and process incoming WebSocket frames.

(self)

        self.transport.write("".join(headers).encode("latin-1"))

    async def _read_frames(self):
        """Read and process incoming WebSocket frames."""
        try:
            # Continue reading while not closed, or if we sent close but haven't
            # received client's close response yet (RFC 6455 close handshake)
            while not self.closed or (self._close_sent and not self._close_received):
                frame = await self._read_frame()
                if frame is None:
                    break

                opcode, payload = frame

                if opcode == OPCODE_CLOSE:
                    await self._handle_close(payload)
                    break

                if opcode == OPCODE_PING:
                    await self._send_frame(OPCODE_PONG, payload)
                elif opcode == OPCODE_PONG:
                    # Ignore pongs
                    pass
                elif opcode == OPCODE_TEXT:
                    await self._receive_queue.put({
                        "type": "websocket.receive",
                        "text": payload.decode("utf-8"),
                    })
                elif opcode == OPCODE_BINARY:
                    await self._receive_queue.put({
                        "type": "websocket.receive",
                        "bytes": payload,
                    })
                elif opcode == OPCODE_CONTINUATION:
                    # Handle fragmented messages
                    await self._handle_continuation(payload)

        except asyncio.CancelledError:
            raise
        except Exception as e:
            self.log.debug("WebSocket read error: %s", e)
        finally:
            # Signal disconnect
            if not self.closed:
                self.closed = True
                await self._receive_queue.put({
                    "type": "websocket.disconnect",
                    "code": self.close_code or CLOSE_ABNORMAL,
                })

    async def _read_frame(self):  # pylint: disable=too-many-return-statements
        """Read a single WebSocket frame.