o Nrf@sddlZddlZddlZddlZddlZzddlmZWn ey)e dZYnwdZ dZ Gddde Z dZdd d Zd d Zd dZddZddZddZdddZddZdS)N) NpipeSocketc@s eZdZdS) SocketErrorN)__name__ __module__ __qualname__r r \/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/docker/utils/socket.pyrsrmc Cstjtjtjf}t|ts-ttdst|gggnt}| |tj tj B|z t|dr9| |WSt|t jrE||WSt||WStyf}z |j|vr[WYd}~dSd}~wty}zt|tot|jdko|jdtk}|rWYd}~dSd}~ww)z+ Reads at most n bytes from socket pollrecvNr)errnoZEINTRZEDEADLK EWOULDBLOCK isinstancerhasattrselectr registerPOLLINPOLLPRIrpysocketSocketIOreadosfilenoOSError Exceptionlenargs NPIPE_ENDED)socketnZrecoverable_errorsr eZ is_pipe_endedr r r rs:          rcCsFd}t||kr!t||t|}|std||7}t||ks|S)z] Reads exactly n bytes from socket Raises SocketError if there isn't enough data zUnexpected EOF)rrr)r"r#dataZ next_datar r r read_exactly>s  r'cCs<zt|d}Wn tyYdSwtd|\}}||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)r'rstructunpack)r"r&streamactualr r r next_frame_headerLs r.cCs|r ddt|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. css|]}t|fVqdS)N)STDOUT).0framer r r eszframes_iter..)frames_iter_ttyframes_iter_no_tty)r"ttyr r r frames_iter\sr6ccsh t|\}}|dkrdS|dkr3t||}|durqt|}|dkr&dS||8}||fV|dksq)zc Returns a generator of data read from the socket when the tty setting is not enabled. TrN)r.rr)r"r,r#resultZ data_lengthr r r r4js    r4ccs$ t|}t|dkrdS|Vq)z^ Return a generator of data read from the socket when the tty setting is enabled. TrN)rr)r"r7r r r r3s r3FcCs|dur d|Sddg}|D]=}|dksJ|ddur5|ddur*|d|d<q|d|d7<q|ddurB|d|d<q|d|d7<qt|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)framesZdemuxoutr1r r r consume_socket_outputs     r<cCs.|tkr|dfS|tkrd|fSt|d)z[ Utility to demultiplex stdout and stderr when reading frames from the socket. Nz is not a valid stream)r/STDERR ValueError)Z stream_idr&r r r demux_adaptors r?)r )F)rrrr"rr* transportr ImportErrortyper/r=rrr!rr'r.r6r4r3r<r?r r r r s,   $ %