biddlZddlZddlZddlZddlZ ddlmZdZ dZ Gdde Z dZddZd Zd Zd Zd Zd ZddZdZy#e$r e dZY;wxYw)N) NpipeSocketc eZdZy) SocketErrorN)__name__ __module__ __qualname__N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/docker/utils/socket.pyrrsr rmcLtjtjtjf}t |t s~t tdstj|gggnUtj}|j|tjtjz|j t |dr|j|St |tjr|j|St!j|j#|S#t$$r}|j|vrYd}~yd}~wt&$rO}t |t xr0t)|j*dkDxr|j*dt,k(}|rYd}~yd}~wwxYw)z+ Reads at most n bytes from socket pollrecvNr)errnoEINTREDEADLK EWOULDBLOCK isinstancerhasattrselectrregisterPOLLINPOLLPRIrpysocketSocketIOreadosfilenoOSError Exceptionlenargs NPIPE_ENDED)socketnrecoverable_errorsre is_pipe_endeds r rrs9  ++u}}e6G6GH fk *vv& MM6(B +;;=D MM&&--&.."@ A IIK 66 ";;q> ! fh// 0;;q> !wwv}}**  77, ,  - #FK82QVVq2k1   s7?D+*D+#D++ F#4E F#AFFF#cd}t||kr9t||t|z }|s td||z }t||kr9|S)z] Reads exactly n bytes from socket Raises SocketError if there isn't enough data r zUnexpected EOF)r$rr)r'r(data next_datas r read_exactlyr/>sU D d)a-SY/ ./ /  d)a- Kr ct t|d}tjd|\}}||fS#t$rYywxYw)z Returns the stream and size of the next frame of data waiting to be read from socket, according to the protocol defined here: https://docs.docker.com/engine/api/v1.24/#attach-to-a-container )r2z>BxxxL)r/rstructunpack)r'r-streamactuals r next_frame_headerr7LsHFA&]]8T2NFF F  s + 77c@|rdt|DSt|S)z Return a generator of frames read from socket. A frame is a tuple where the first item is the stream number and the second item is a chunk of data. If the tty setting is enabled, the streams are multiplexed into the stdout stream. c3,K|] }t|fyw)N)STDOUT).0frames r zframes_iter..esEEEs)frames_iter_ttyframes_iter_no_tty)r'ttys r frames_iterrA\s" E_V-DEE!&))r c#K t|\}}|dkry|dkDr1t||}|t|}|dk(ry||z}||f|dkDr1Kw)zc Returns a generator of data read from the socket when the tty setting is not enabled. rN)r7rr$)r'r5r(result data_lengths r r?r?jst '/  q5 !e&!_F~f+Ka  A6" "!e s A AAc#JK t|}t|dk(ry|w)z^ Return a generator of data read from the socket when the tty setting is enabled. rN)rr$)r'rCs r r>r>s- f v;!   s!#c|durdj|Sddg}|D]K}|dk7sJ|d|d |d|d<|dxx|dz cc<.|d |d|d<<|dxx|dz cc<Mt|S)a Iterate through frames read from the socket and return the result. Args: demux (bool): If False, stdout and stderr are multiplexed, and the result is the concatenation of all the frames. If True, the streams are demultiplexed, and the result is a 2-tuple where each item is the concatenation of frames belonging to the same stream. Fr N)NNrr)jointuple)framesdemuxoutr<s r consume_socket_outputrLs ~xx ,C # $$$ 8 1v~qAA%("1v~qAA%(" # :r cR|tk(r|dfS|tk(rd|fSt|d)z[ Utility to demultiplex stdout and stderr when reading frames from the socket. Nz is not a valid stream)r:STDERR ValueError) stream_idr-s r demux_adaptorrQs; Fd| f d|I;&<=>>r )i)F)rr rr'rr3 transportr ImportErrortyper:rNr#rr&rr/r7rAr?r>rLrQr r r rUs  '   )  !H   *#* "J ?Qt*KsA AA