ddlZddlZddlZddlZddlZ ddlmZdZ dZ Gdde Z dZddZd Zd Zd Zd Zd ZddZdZy#e$r e dZY;wxYw)N) NpipeSocketc eZdZy) SocketErrorN)__name__ __module__ __qualname__&lib/third_party/docker/utils/socket.pyrrsr rmcLtjtjtjf}t |t s~t tdstj|gggnUtj}|j|tjtjz|j t |dr|j|St |ttdr|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 pollrecvSocketIONr)errnoEINTREDEADLK EWOULDBLOCK isinstancerhasattrselectrregisterPOLLINPOLLPRIrgetattrpysocketreadosfilenoOSError Exceptionlenargs NPIPE_ENDED)socketnrecoverable_errorsre is_pipe_endeds r r r s:  ++u}}e6G6GH fk *vv& MM6(B +;;=D MM&&--&.."@ A IIK 66 ";;q> ! fgh ; <;;q> !wwv}}**  77, ,  - #FK82QVVq2k1   s7?D+*D+#D++ F#4E F#AFFF#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)bytesr%r r)r(r)data next_datas r read_exactlyr1>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 )r4z>BxxxL)r1rstructunpack)r(r/streamactuals r next_frame_headerr9LsHFA&]]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..esE-DE-Ds)frames_iter_ttyframes_iter_no_tty)r(ttys r frames_iterrC\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)r9r r%)r(r7r)result data_lengths r rArAjst '/  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)r r%)r(rEs r r@r@s- 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)framesdemuxoutr>s r consume_socket_outputrNs ~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)r<STDERR ValueError) stream_idr/s r demux_adaptorrSs; Fd| f d|I;&<=>>r )i)F)rr!rr(rr5 transportr ImportErrortyper<rPr$rr'r r1r9rCrAr@rNrSr r r rWs  '   )  !H   *#* "J ?Qt*KsA AA