\ei& dZddlZddlZddlZddlZddlmZGddZ ddeded ed e d e d e f d Z ded e fdZ ded e fdZ ded e fdZded e fdZy)ah Dirty Arbiters Protocol Length-prefixed JSON message framing over Unix sockets. Provides both async (primary) and sync (for HTTP workers) APIs. Message Format: +----------------+------------------+ | 4-byte length | JSON payload | +----------------+------------------+ The length field is a 4-byte unsigned integer in network byte order (big-endian). N)DirtyProtocolErrorcveZdZdZdZej eZdZdZ dZ dZ dZ dZ ed ed efd Zed ed efd Zedej(d efdZedej,d ed dfdZedej0ded efdZedej0d efdZedej0d ed dfdZy) DirtyProtocolz0Length-prefixed JSON messages over Unix sockets.z!Iirequestresponseerrorchunkendmessagereturnc tj|jd}t|tj kDr)t dt|dtj dtjtjt|}||zS#ttf$r}t d|d}~wwxYw)a  Encode a message dict to length-prefixed bytes. Args: message: Dictionary to encode as JSON Returns: bytes: Length-prefixed encoded message Raises: DirtyProtocolError: If encoding fails utf-8Message too large: bytes (max: )zFailed to encode message: N) jsondumpsencodelenrMAX_MESSAGE_SIZErstructpack HEADER_FORMAT TypeError ValueError)r payloadlengthes M/var/www/ceduoda/envi/lib/python3.12/site-packages/gunicorn/dirty/protocol.pyrzDirtyProtocol.encode,s Gjj)009G7|m<<<()#g,8*;; > "..v66G##G,,AI** 199~"$)#aii.)9:)5568   .7** $*3qyy>*:;"8%  soF"CCCA*FD:*D8+D:/FCD5AD00D55F8D::F 0E==FFwriterNcKtj|}|j||jd{y7w)a Write a message to async stream. Args: writer: asyncio StreamWriter message: Dictionary to send Raises: DirtyProtocolError: If encoding fails ConnectionError: If write fails N)rrwritedrain)r5r r!s r write_message_asyncz!DirtyProtocol.write_message_asyncs3##G, Tllns:AAAsockncd}t||krh|j|t|z }|s5t|dk(r tdtdt|d||||z }t||krh|S)a Receive exactly n bytes from a socket. Args: sock: Socket to read from n: Number of bytes to read Returns: bytes: Received data Raises: DirtyProtocolError: If read fails or connection closed rzConnection closedzConnection closed after r+r#)rrecvr)r:r;r!r s r _recv_exactlyzDirtyProtocol._recv_exactlys$i!mIIa#d)m,Et9>,-@AA(.s4yk9J1#N! EMD$i!m r=cttj|tj}tjtj |d}|tj kDr td|dtj d|dk(r tdtj||}tj|S)z Read a complete message from socket (sync). Args: sock: Socket to read from Returns: dict: Decoded message Raises: DirtyProtocolError: If read fails or message is malformed rrrrr,) rr?r.rr2rrrr&)r:r3rrs r read_messagezDirtyProtocol.read_messages,,T=3L3LM}::FCAF M22 2$%fX.&778;  Q;$%=> > --dF;##G,,r=cPtj|}|j|y)z Write a message to socket (sync). Args: sock: Socket to write to message: Dictionary to send Raises: DirtyProtocolError: If encoding fails OSError: If write fails N)rrsendall)r:r r!s r write_messagezDirtyProtocol.write_messages ##G, Tr=)__name__ __module__ __qualname____doc__rrcalcsizer.rMSG_TYPE_REQUESTMSG_TYPE_RESPONSEMSG_TYPE_ERRORMSG_TYPE_CHUNK MSG_TYPE_END staticmethoddictbytesrr&r/ StreamReaderr4 StreamWriterr9socketintr?rArDr=r rrsa:M!&//-0K(!"NNLGGGG24U4t44.0-)=)=0-$0-0-d'*>*>+/48*FMMce6-6==-T--< FMM D T  r=r request_idapp_pathactionargskwargsr cRtj||||r t|ng|xsidS)a> Build a request message. Args: request_id: Unique request identifier app_path: Import path of the dirty app (e.g., 'myapp.ml:MLApp') action: Action to call on the app args: Positional arguments kwargs: Keyword arguments Returns: dict: Request message )typeidrXrYrZr[)rrJlist)rWrXrYrZr[s r make_requestr`s2 .."T ,B  r=c*tj||dS)z Build a success response message. Args: request_id: Request identifier this responds to result: Result value (must be JSON-serializable) Returns: dict: Response message )r]r^result)rrK)rWrbs r make_responserc s// r=cddlm}t||r|j}n5t|tr|}n"t |j t|id}tj||dS)z Build an error response message. Args: request_id: Request identifier this responds to error: DirtyError instance or dict with error info Returns: dict: Error response message r) DirtyError) error_typer details)r]r^r ) errorsre isinstanceto_dictrPr]rEstrrrL)rWr re error_dicts r make_error_responsermsf#%$]]_ E4  u+..5z ,, r=c*tj||dS)z Build a chunk message for streaming responses. Args: request_id: Request identifier this chunk belongs to data: Chunk data (must be JSON-serializable) Returns: dict: Chunk message )r]r^r!)rrM)rWr!s r make_chunk_messagero=s,, r=c(tj|dS)z Build an end-of-stream message. Args: request_id: Request identifier this ends Returns: dict: End message )r]r^)rrN)rWs r make_end_messagerqOs** r=)NN)rHr/rrrTrhrrrktuplerPr`rcrmrorqrVr=r rss  &UUt59SC-1=A2cd$C4<3$   r=