2.addlZddlZddlZddlZddlZ ddlmZdZ dZ Gdde Z ddZdZd Zd Zd Zd Zdd ZdZy#e$r e dZY9wxYw)N) NpipeSocketc eZdZy) SocketErrorN)__name__ __module__ __qualname__5/usr/lib/python3/dist-packages/docker/utils/socket.pyrrsr rctjtjtjf}t |t st j |ggg t|dr|j|St |ttdr|j|Stj|j|S#t$r}|j|vrYd}~yd}~wwxYw)z+ Reads at most n bytes from socket recvSocketION)errnoEINTREDEADLK EWOULDBLOCK isinstancerselecthasattrrgetattrpysocketreadosfilenoOSError)socketnrecoverable_errorses r rrs  ++u}}e6G6GH fk * vhB' 66 ";;q> ! fgh ; <;;q> !wwv}}**  77, ,  -s$C6*C!#C C'C""C'ct}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 zUnexpected EOF)byteslenrr)rrdata next_datas r read_exactlyr'*sW 7D 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 )r*z>BxxxL)r'rstructunpack)rr%streamactuals r next_frame_headerr/8sHFA&]]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..QsEEEs)frames_iter_ttyframes_iter_no_tty)rttys r frames_iterr9Hs" 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)r/rr$)rr-rresult data_lengths r r7r7Vst '/  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$)rr;s r r6r6ks- f v;!   s!#c|durtj|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. FN)NNrr)r#jointuple)framesdemuxoutr4s r consume_socket_outputrDxs ~w||F## ,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)r2STDERR ValueError) stream_idr%s r demux_adaptorrIs; Fd| f d|I;&<=>>r )i)F)rrrrrr+ transportr ImportErrortyper2rF Exceptionrrr'r/r9r7r6rDrIr r r rNsx  '   ) *   *#* "J ?it*KsA AA