U ™K7\\ã@s¾ddlZddlZddlZddlZddlZddlZzddlmZWne k r\e dƒZYnXdZ dZ Gdd„de ƒZddd „Zd d „Zd d „Zdd„Zdd„Zdd„Zddd„Zdd„ZdS)éNé)Ú NpipeSocketéc@s eZdZdS)Ú SocketErrorN)Ú__name__Ú __module__Ú __qualname__©r r ú5/usr/lib/python3/dist-packages/docker/utils/socket.pyrsréc Cs®tjtjtjf}tjr0t|tƒs0t |ggg¡zJt |dƒrH|  |¡WStjrjt|t t dƒƒrj|  |¡WSt  | ¡|¡WStk r¨}z|j|kr˜‚W5d}~XYnXdS)z+ Reads at most n bytes from socket ÚrecvZSocketION)ÚerrnoZEINTRZEDEADLKZ EWOULDBLOCKÚsixZPY3Ú isinstancerÚselectÚhasattrr ÚgetattrÚpysocketÚreadÚosÚfilenoÚEnvironmentError)ÚsocketÚnZrecoverable_errorsÚer r r rs    rcCs@t ¡}t|ƒ|krzt|dƒ}Wntk r$YdSXt d|¡\}}||fS)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)rrÚstructZunpack)rrÚstreamZactualr r r Únext_frame_header:s r#cCs"|rdd„t|ƒDƒSt|ƒSdS)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. css|]}t|fVqdS)N)ÚSTDOUT)Ú.0Úframer r r Ú Sszframes_iter..N)Úframes_iter_ttyÚframes_iter_no_tty)rZttyr r r Ú frames_iterJsr*ccs`t|ƒ\}}|dkrq\|dkrt||ƒ}|dkr2qt|ƒ}|dkrFdS||8}||fVqqdS)zc Returns a generator of data read from the socket when the tty setting is not enabled. rN)r#rr)rr"rÚresultZ data_lengthr r r r)Xs  r)ccs$t|ƒ}t|ƒdkrdS|VqdS)z^ Return a generator of data read from the socket when the tty setting is enabled. rN)rr)rr+r r r r(ms r(FcCs¦|dkrt ¡ |¡Sddg}|D]z}|dks2t‚|ddk rn|ddkrX|d|d<qœ|d|d7<q"|ddkrˆ|d|d<q"|d|d7<q"t|ƒ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)rrÚjoinÚAssertionErrorÚtuple)ZframesZdemuxÚoutr&r r r Úconsume_socket_outputzs     r0cCs2|tkr|dfS|tkr d|fStd |¡ƒ‚dS)z[ Utility to demultiplex stdout and stderr when reading frames from the socket. Nz{0} is not a valid stream)r$ÚSTDERRÚ ValueErrorÚformat)Z stream_idrr r r Ú demux_adaptorŸs r4)r )F)r rrrrr!rZ transportrÚ ImportErrorÚtyper$r1Ú Exceptionrrrr#r*r)r(r0r4r r r r Ús(  %