o

�h��@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZzddlmZ
Wn
ey)


e	d
�
ZYn
wdZ
dZGdd�de�Z
dZdd	d
�
Zdd�Zd
d�Zdd�Zdd�Zdd�Zddd�
Zdd�Zd
S)�N�)
�NpipeSocket�
c
@seZ
dZd
S)�SocketErrorN)�__name__�
__module__�__qualname__�r	r	�F/usr/local/CyberCP/lib/python3.10/site-packages/docker/utils/socket.pyrs
r�m�c
Cs
tj
tjtjf}t|t�s-ttd
�st�|g
gg�
nt��}|�	|tj
tjB�
|��
z t|d�r9|�|
�
WSt|t
j�rE|�|
�
WSt�|��|
�WStyf
}
z
|j|v
r[�WYd}~dSd}~w
ty�
}
zt|t�ot|j�
dko|jdtk}|r�WYd}~dS�d}~w
w)z+
    Reads at most n bytes from socket
    �poll�recvNr
�)�errno�EINTR�EDEADLK�EWOULDBLOCK�
isinstancer�hasattr�selectr
�register�POLLIN�POLLPRIr�pysocket�SocketIO�read�os�fileno�OSError�	Exception�len�args�NPIPE_ENDED)�socket�
n�recoverable_errorsr
�
e�
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)r!rr)r$r%�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)r$r*�stream�actualr	r	r
�next_frame_headerLs


�
r3cCs|
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>es�zframes_iter.<locals>.<genexpr>)�frames_iter_tty�frames_iter_no_tty)r$�ttyr	r	r
�frames_iter\s
r;c
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)r3rr!)r$r1r%�result�data_lengthr	r	r
r9js �












��r9c
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!)r$r<r	r	r
r8s�


�r8FcCs�|
d
ur	d�|�
Sddg}|D]=}|dksJ�
|ddu
r5|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)NNr
r)�join�tuple)�frames�demux�outr6r	r	r
�consume_socket_output�s






rCcCs.|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)r4�STDERR�
ValueError)�	stream_idr*r	r	r
�
demux_adaptor�s



rG)
r)
F)rrrr$rr/�	transportr�ImportError�typer4rDr rr#rr,r3r;r9r8rCrGr	r	r	r
�<module>
s,






�

$

%