o 2.a@sddlZddlZddlZddlZddlZzddlmZWn ey)e dZYnwdZ dZ Gddde Z ddd Zd d Zd d ZddZddZddZdddZddZdS)N) NpipeSocketc@s eZdZdS) SocketErrorN)__name__ __module__ __qualname__r r 5/usr/lib/python3/dist-packages/docker/utils/socket.pyrsrc Cstjtjtjf}t|tst|gggz"t|dr!||WSt|t t dr/| |WSt | |WStyP}z |j|vrEWYd}~dSd}~ww)z+ Reads at most n bytes from socket recvZSocketION)errnoZEINTRZEDEADLKZ EWOULDBLOCK isinstancerselecthasattrr getattrpysocketreadosfilenoOSError)socketnZrecoverable_errorser r r rs     rcCsHt}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)byteslenrr)rrdataZ next_datar r r read_exactly*s  rcCs<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 )rz>BxxxL)rrstructunpack)rrstreamZactualr r r next_frame_header8s 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 Qszframes_iter..)frames_iter_ttyframes_iter_no_tty)rZttyr r r frames_iterHsr*ccsh 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)rr"rresultZ data_lengthr r r r)Vs    r)ccs$ t|}t|dkrdS|Vq)z^ Return a generator of data read from the socket when the tty setting is enabled. TrN)rr)rr+r r r r(ks r(FcCs|dur t|Sddg}|D]=}|dksJ|ddur6|ddur+|d|d<q|d|d7<q|ddurC|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. FN)NNrr)rjointuple)ZframesZdemuxoutr&r r r consume_socket_outputxs     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_idrr r r demux_adaptors r2)r )F)r rrrrr Z transportr ImportErrortyper$r0 Exceptionrrrr#r*r)r(r/r2r r r r s*    %