ڊjI dZddlZddlZGddeZGddZddeded ed e fd Z dded e d e fd Z deded e fdZ y)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"} NceZdZdZy) ProtocolErrorzProtocol-level error.N)__name__ __module__ __qualname____doc__Z/var/www/io.vulcan-creative.com/venv/lib/python3.12/site-packages/gunicorn/ctl/protocol.pyrrsr rceZdZdZdZededefdZededefdZ edefdZ edefdZ edefd Z edefd Z y ) ControlProtocolz Protocol implementation for control socket communication. Uses 4-byte big-endian length prefix followed by JSON payload. idatareturnctj|jd}tjdt |}||zS)z Encode a message for transmission. Args: data: Dictionary to encode Returns: Length-prefixed JSON bytes utf-8>I)jsondumpsencodestructpacklen)rpayloadlengths r encode_messagezControlProtocol.encode_message)s<**T"))'2T3w<0r ct|dkr tdtjd|ddd}t|d|zkr td|dd|z}t j |j dS)z Decode a message from bytes. Args: data: Raw bytes (length prefix + JSON payload) Returns: Decoded dictionary zMessage too shortrNrzIncomplete messager)rrrunpackrloadsdecode)rrrs r decode_messagezControlProtocol.decode_message8sz t9q= 34 4tT"1X.q1 t9q6z ! 45 5qV$zz'..122r cxd}t|dkrK|jdt|z }|s|s tdtd||z }t|dkrKt j d|d}|t jkDrtd|d}t||krH|jt|t|z d}|s td ||z }t||krH tj|jd S#tj$r}td |d }~wwxYw) 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 r rzConnection closedzIncomplete length prefixrrMessage too large: izIncomplete payloadrInvalid JSON: N) rrecvConnectionErrorrrrr MAX_MESSAGE_SIZEminrrr JSONDecodeError)sock length_datachunkr payload_dataes r read_messagezControlProtocol.read_messageMs?  +"IIa#k"223E")*=>>#$>?? 5 K +"t[1!4 O44 4"5fX >? ? ,&(IIc&3|+<"? ?$//77  6::l11':; ;28## 6. 45 5 6sFCBAC*B+C0#BCCB>+B99B>>CcKtj|}|j||jd{y7w)z Write one message to an async writer. Args: writer: asyncio StreamWriter data: Message dictionary to send N)r rwritedrain)writerrr2s r write_message_asyncz#ControlProtocol.write_message_asyncs3"006 Wllns:AAAN)rrrrr' staticmethoddictbytesrr!r/r3r7r<r r r r r s(  T  e    3U3t33((6d(6(6T $  6D662   r r request_idcommandargsrc|||xsgdS)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 )idrArBr )r@rArBs r make_requestrEs  r rc|d|xsidS)z Create a success response message. Args: request_id: Request identifier being responded to data: Response data Returns: Response dictionary ok)rDstatusrr )r@rs r make_responserIs  r errorc|d|dS)z Create an error response message. Args: request_id: Request identifier being responded to error: Error message Returns: Error response dictionary rJ)rDrHrJr )r@rJs r make_error_responserLs r )N) rrr Exceptionrr intstrlistr>rErIrLr r r rQs $  I JJZS3dd&c$Cr