Method read_message

gunicorn/ctl/protocol.py:78–118

Read one message from a socket. Args: sock: Socket to read from Returns: Decoded message dictionary Raises: ProtocolError: If message is malformed ConnectionError: If connection is closed

(sock)

    @staticmethod
    def read_message(sock) -> dict:
        """
        Read one message from a socket.

        Args:
            sock: Socket to read from

        Returns:
            Decoded message dictionary

        Raises:
            ProtocolError: If message is malformed
            ConnectionError: If connection is closed
        """
        # Read length prefix
        length_data = b''
        while len(length_data) < 4:
            chunk = sock.recv(4 - len(length_data))
            if not chunk:
                if not length_data:
                    raise ConnectionError("Connection closed")
                raise ProtocolError("Incomplete length prefix")
            length_data += chunk

        length = struct.unpack('>I', length_data)[0]

        if length > ControlProtocol.MAX_MESSAGE_SIZE:
            raise ProtocolError(f"Message too large: {length}")

        # Read payload
        payload_data = b''
        while len(payload_data) < length:
            chunk = sock.recv(min(length - len(payload_data), 65536))
            if not chunk:
                raise ProtocolError("Incomplete payload")
            payload_data += chunk

        try:
            return json.loads(payload_data.decode('utf-8'))
        except json.JSONDecodeError as e:
            raise ProtocolError(f"Invalid JSON: {e}")

    @staticmethod
    def write_message(sock, data: dict):