o

�2.a�@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZzddlmZ
Wn
ey)


e	d
�
ZYn
wdZ
dZGdd�de�Z
ddd	�
Zd
d�Zdd
�Zdd�Zdd�Zdd�Zddd�
Zdd�Zd
S)�N�)
�NpipeSocket�
c
@seZ
dZd
S)�SocketErrorN)�__name__�
__module__�__qualname__�r	r	�5/usr/lib/python3/dist-packages/docker/utils/socket.pyrs
r�c
Cs�tj
tjtjf}t|t�st�|g
gg�
z"t|d
�r!|�|
�
WSt|t	t
d��r/|�|
�
WSt�|�
�|
�WStyP
}
z
|j|v
rE�WYd}~dSd}~w
w)z+
    Reads at most n bytes from socket
    �recv�SocketION)�errno�EINTR�EDEADLK�EWOULDBLOCK�
isinstancer�select�hasattrr�getattr�pysocket�read�os�fileno�OSError)�socket�
n�recoverable_errors�
er	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)�bytes�lenrr)rr�data�	next_datar	r	r
�read_exactly*s




�r#c
Cs<zt|d
�}
Wn
t
y


YdSwt�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)r#r�struct�unpack)rr!�stream�actualr	r	r
�next_frame_header8s


�
r*cCs|
rd
d�t|�
D�
St
|�
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.
    c
ss�|]}
t|
fV
qdS)
N)
�STDOUT)�.0�framer	r	r
�	<genexpr>Qs�zframes_iter.<locals>.<genexpr>)�frames_iter_tty�frames_iter_no_tty)r�ttyr	r	r
�frames_iterHs
r2c
csh�	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.
    Tr
N)r*rr )rr(r�result�data_lengthr	r	r
r0Vs �












��r0c
cs$�	t|�
}
t
|
�
dkrdS|
V
q)z^
    Return a generator of data read from the socket when the tty setting is
    enabled.
    Tr
N)rr )rr3r	r	r
r/ks�


�r/FcCs�|
d
ur
t��
|�
Sddg}|D]=}|dksJ�
|ddu
r6|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)NNr
r)r�join�tuple)�frames�demux�outr-r	r	r
�consume_socket_outputxs





r:cCs.|tkr|
d
fS|t
krd
|
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!r	r	r
�
demux_adaptor�s



r>)
r)
F)rrrrrr&�	transportr�ImportError�typer+r;�	Exceptionrrr#r*r2r0r/r:r>r	r	r	r
�<module>
s*






�



%