o iI @sdZddlZddlZGdddeZGdddZddeded ed e fd d Z dded e d e fddZ deded e fddZ dS)a Control Socket Protocol JSON-based protocol with length-prefixed framing for the control interface. Message Format: +----------------+------------------+ | Length (4B BE) | JSON Payload | +----------------+------------------+ Request Format: {"id": 1, "command": "show", "args": ["workers"]} Response Format: {"id": 1, "status": "ok", "data": {...}} {"id": 1, "status": "error", "error": "message"} Nc@seZdZdZdS) ProtocolErrorzProtocol-level error.N)__name__ __module__ __qualname____doc__rrI/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/ctl/protocol.pyrsrc@seZdZdZdZededefddZededefddZ edefd d Z edefd d Z edefd dZ edefddZ dS)ControlProtocolz Protocol implementation for control socket communication. Uses 4-byte big-endian length prefix followed by JSON payload. idatareturncCs(t|d}tdt|}||S)z Encode a message for transmission. Args: data: Dictionary to encode Returns: Length-prefixed JSON bytes utf-8>I)jsondumpsencodestructpacklen)r payloadlengthrrrencode_message)s zControlProtocol.encode_messagecCsdt|dkr tdtd|ddd}t|d|kr"td|dd|}t|dS)z Decode a message from bytes. Args: data: Raw bytes (length prefix + JSON payload) Returns: Decoded dictionary zMessage too shortr NrzIncomplete messager )rrrunpackrloadsdecode)r rrrrrdecode_message8s zControlProtocol.decode_messagec Csd}t|dkr'|dt|}|s|stdtd||7}t|dkstd|d}|tjkr;td|d}t||kr_|t|t|d}|sUtd ||7}t||ksCz t | d WSt j y|}ztd |d }~ww) a 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 rzConnection closedzIncomplete length prefixr rMessage too large: izIncomplete payloadr Invalid JSON: N) rrecvConnectionErrorrrrr MAX_MESSAGE_SIZEminrrrJSONDecodeError)sock length_datachunkr payload_dataerrr read_messageMs2     zControlProtocol.read_messagecCst|}||dS)z Write one message to a socket. Args: sock: Socket to write to data: Message dictionary to send N)r rsendall)r$r messagerrr write_messagexs zControlProtocol.write_messagec s|dIdH}td|d}|tjkrtd|||IdH}z t|dWStj yB}ztd|d}~ww)z Read one message from an async reader. Args: reader: asyncio StreamReader Returns: Decoded message dictionary rNr rrr r) readexactlyrrr r!rrrrr#)readerr%rr'r(rrrread_message_asyncs  z"ControlProtocol.read_message_asynccs(t|}|||IdHdS)z Write one message to an async writer. Args: writer: asyncio StreamWriter data: Message dictionary to send N)r rwritedrain)writerr r+rrrwrite_message_asyncs z#ControlProtocol.write_message_asyncN)rrrrr! staticmethoddictbytesrrr)r,r/r3rrrrr s* r request_idcommandargsr cCs|||pgdS)z Create a request message. Args: request_id: Unique request identifier command: Command name (e.g., "show workers") args: Optional list of arguments Returns: Request dictionary )idr8r9r)r7r8r9rrr make_requests r;r cCs|d|pidS)z Create a success response message. Args: request_id: Request identifier being responded to data: Response data Returns: Response dictionary ok)r:statusr r)r7r rrr make_responses r>errorcCs |d|dS)z Create an error response message. Args: request_id: Request identifier being responded to error: Error message Returns: Error response dictionary r?)r:r=r?r)r7r?rrrmake_error_responses r@)N) rrr Exceptionrr intstrlistr5r;r>r@rrrrs