jI dZddlZddlZGddeZGddZddeded ed e fd Z dded e d e fd Z deded e fdZ 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"} NceZdZdZdS) ProtocolErrorzProtocol-level error.N)__name__ __module__ __qualname____doc__b/var/www/finuniver-perm.ru/html/student/venv/lib/python3.11/site-packages/gunicorn/ctl/protocol.pyrrsr rceZdZdZdZededefdZededefdZ edefdZ edefdZ edefd Z edefd Z d S) ControlProtocolz Protocol implementation for control socket communication. Uses 4-byte big-endian length prefix followed by JSON payload. idatareturnctj|d}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)sC*T""))'22T3w<<00r c>t|dkrtdtjd|ddd}t|d|zkrtd|dd|z}t 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_message8s t99q== 344 4tT"1"X..q1 t99q6z ! ! 455 5qV|$z'..11222r cd}t|dkr_|dt|z }|s |stdtd||z }t|dk_t jd|d}|t jkrtd|d}t||kr\|t|t|z d}|std ||z }t||k\ tj | d 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"2"2233E @"?)*=>>>#$>??? 5 K +""t[11!4 O4 4 4 >f > >?? ? ,&((IIc&3|+<+<"f > >?? ?$//77777777  6:l11'::;; ;# 6 6 6 4 4 455 5 6s7&BC-B??CcKt|}|||d{VdS)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_asyncsS"0066 Wllnnr N)rrrrr' staticmethoddictbytesrr!r/r3r7r<r r r r r s(  T  e    \  3U3t333\3((6d(6(6(6\(6T $   \ 6D666\62    \   r r request_idcommandargsrc|||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 )idrArBr )r@rArBs r make_requestrEs    r rc|d|pidS)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   JJJJJJJJZS3dd&c$Cr