a ‹»ìæ¨i4Iã@s¼ddlZddlZddlZddlZddlZddlZzddlmZWne yZe dƒZYn0dZ 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 ú7/usr/lib/python3.9/site-packages/docker/utils/socket.pyrsréc Cs°tjtjtjf}tjr0t|tƒs0t |ggg¡zJt |dƒrH|  |¡WStjrjt|t t dƒƒrj|  |¡WSt  | ¡|¡WStyª}z|j|vr–‚WYd}~n d}~00dS)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|ƒ|krBxxxL)rrÚstructÚunpack)rrÚstreamÚactualr 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 Ú Sózframes_iter..N)Úframes_iter_ttyÚframes_iter_no_tty)rZttyr r r Ú frames_iterJsr-ccs`t|ƒ\}}|dkrq\|dkrt||ƒ}|dur2qt|ƒ}|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¦|durt ¡ |¡Sddg}|D]z}|dks2J‚|ddurn|ddurX|d|d<qœ|d|d7<q"|ddurˆ|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Útuple)ZframesZdemuxÚoutr(r r r Úconsume_socket_outputzs     r2cCs2|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 r6)r )F)r rrrrr!rZ transportrÚ ImportErrorÚtyper&r3Ú Exceptionrrrr%r-r,r+r2r6r r r r Ús(   %