File: //proc/self/root/lib/python3.9/site-packages/cockpit/__pycache__/transports.cpython-39.opt-1.pyc
a
���e�C � @ s* d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZmZmZmZmZmZmZmZ ddlmZmZ ej�d�Zdd� Z dZ e� e �Z d Z!G d
d � d ej"�Z#G d d
� d
ej$�Z%G dd� d�Z&G dd� de#ej'�Z'G dd� de#�Z(G dd� d�Z)dS )zKBi-directional asyncio.Transport implementations based on file descriptors.� N)�Any�ClassVar�Deque�Dict�List�Optional�Sequence�Tuple� )�
JsonObject�get_intz libc.so.6c G s t j| � dkrtd��d S )Nr zprctl() failed)�libc6�prctl�OSError)�args� r �6/usr/lib/python3.9/site-packages/cockpit/transports.pyr ' s r i c s e Zd ZU dZee ed< ejed< ej ed< e
e e ed< eed< eed< e
ed< e
ed < e
ed
< d Ze
ed < dCejej eee
eeef d�� fdd�
Zdd�dd�Ze
d�dd�Zdd�dd�Zdd�dd�Zdd�dd �Zdd�d d �ZdDe
e dd �d d!�Ze
d�d"d#�Zdd�d$d%�Z ed�d&d'�Z e eef d�d(d)�Z dEe
e e
e dd*�d+d,�Z dd�d-d.�Z!dd�d/d0�Z"dd�d1d2�Z#e dd3�d4d5�Z$e dd3�d6d7�Z%dd�d8d9�Z&ej'd�d:d;�Z(e
d�d<d=�Z)ej'dd>�d?d@�Z*dd�dAdB�Z+� Z,S )F�
_Transporti �
BLOCK_SIZE�_loop� _protocol�_queue�_in_fd�_out_fd�_closing� _is_reading�_eofF� _eio_is_eof���N)�loop�protocol�in_fd�out_fd�extrac s� t � �|� || _|| _t�d| |||� d | _d| _d| _d| _ || _
|| _ t �
|d� ||krnt �
|d� | j�| � | �� d S )Nz/Created transport %s for protocol %s, fds %d %dF)�super�__init__r r �logger�debugr r r r r r �os� set_blockingZconnection_made�resume_reading)�selfr r r! r" r# �� __class__r r r% B s z_Transport.__init__��returnc
C s� t �d| | j| j� zt�| jtj�}W nf ty< Y d S t y� } z<| j
rb|j t j krbd}n| �
|� W Y d }~d S W Y d }~n
d }~0 0 |dkr�t �dt|�� | j�|� n(t �d� | �� | j�� }|s�| �� d S )NzRead ready on %s %s %d� z read %d bytesz got EOF)r& r' r r r( �readr r �BlockingIOErrorr r �errnoZEIO�abort�lenZ
data_received�
_close_readerZ eof_received�close)r+ �data�excZ keep_openr r r � _read_ready] s$
$
z_Transport._read_readyc C s | j S �N)r �r+ r r r �
is_readingv s z_Transport.is_readingc C s | � � d| _d S �Nr )�
pause_readingr r< r r r r6 y s z_Transport._close_readerc C s | j r| j�| j� d| _ d S �NF)r r �
remove_readerr r< r r r r? } s z_Transport.pause_readingc C s, | j s(| jdkr(| j�| j| j� d| _ d S )Nr T)r r r �
add_readerr: r< r r r r* � s z_Transport.resume_readingc C s d S r; r r< r r r �_close� s z_Transport._close)r9 r/ c C s. d| _ | �� | �� | j�|� | �� d S �NT)r r6 �_remove_write_queuer Zconnection_lostrC )r+ r9 r r r r4 � s
z_Transport.abortc C s t �d S r; ��NotImplementedErrorr< r r r �
can_write_eof� s z_Transport.can_write_eofc C s d| _ | jd u r| �� d S rD )r r �_write_eof_nowr< r r r � write_eof� s
z_Transport.write_eofc C s" | j d u rdS tdd� | j D ��S )Nr c s s | ]}t |�V qd S r; )r5 )�.0�blockr r r � <genexpr>� r0 z3_Transport.get_write_buffer_size.<locals>.<genexpr>)r �sumr< r r r �get_write_buffer_size� s
z _Transport.get_write_buffer_sizec C s dS )N)r r r r< r r r �get_write_buffer_limits� s z"_Transport.get_write_buffer_limits)�high�lowr/ c C s d S r; r )r+ rQ rR r r r �set_write_buffer_limits� s z"_Transport.set_write_buffer_limitsc C s t �d S r; rF r< r r r rI � s z_Transport._write_eof_nowc
C s� zt �| j| j�}W nD ty* d}Y n0 tyX } z| �|� W Y d }~d S d }~0 0 |r�| j�� }t|�|kr�| j� ||d � � q�|t|�8 }qZ| js�| �
� | j r�| � � | j
r�| �� d S �Nr )r( �writevr r r2 r r4 �popleftr5 �
appendleftrE r rI r )r+ �n_bytesr9 rL r r r � _write_ready� s&
z_Transport._write_readyc C s, | j d ur(| j�� | j�| j� d | _ d S r; )r r Zresume_writingr Z
remove_writerr r< r r r rE � s
z _Transport._remove_write_queue)r8 r/ c C s. | j �| j| j� t�|f�| _| j�� d S r; ) r Z
add_writerr rY � collections�dequer r Z
pause_writing�r+ r8 r r r �_create_write_queue� s z _Transport._create_write_queuec
C s� | j d urJ| j �|� t| j �tkrFd�| j �}| j �� | j �|� d S zt�| j|�}W nD t yr d}Y n0 t
y� } z| � |� W Y d }~d S d }~0 0 |t|�kr�| � ||d � � d S )Nr0 r )
r �appendr5 �IOV_MAX�join�clearr( �writer r2 r r4 r] )r+ r8 Zall_datarX r9 r r r rb � s
z_Transport.writec C s2 | j r
d S d| _ | �� | jd ur&d S | �� d S rD )r r6 r r4 r< r r r r7 � s
z_Transport.closec C s | j S r; )r r< r r r � get_protocol� s z_Transport.get_protocolc C s | j S r; )r r< r r r �
is_closing� s z_Transport.is_closing)r r/ c C s t �d S r; rF )r+ r r r r � set_protocol� s z_Transport.set_protocolc C s | � � d S r; )rC r< r r r �__del__ s z_Transport.__del__)r r N)N)NN)-�__name__�
__module__� __qualname__r r �int�__annotations__�asyncio�AbstractEventLoop�Protocolr r �bytes�boolr r �str�objectr% r: r= r6 r? r* rC � Exceptionr4 rH rJ rO r rP rS rI rY rE r] rb r7 Z BaseProtocolrc rd re rf �
__classcell__r r r, r r 3 sP
��
r c @ s e Zd ZdZdd�dd�ZdS )�SubprocessProtocolzBAn extension to asyncio.Protocol for use with SubprocessTransport.Nr. c C s t �dS )z"Called when subprocess has exited.NrF r< r r r �process_exited s z!SubprocessProtocol.process_exited)rg rh ri �__doc__rv r r r r ru s ru c @ s e Zd Zed�dd�ZdS )�
WindowSize)�valuec C s t |d�| _t |d�| _d S )N�rows�cols)r rz r{ )r+ ry r r r r% s zWindowSize.__init__N)rg rh ri r r% r r r r rx
s rx c sn e Zd ZU dZdZee ed< dZee ed< dZ ed ed< ed ed< e
e j d �d
d ��Z
e
e je j d �d
d��Zdd�eed�dd�Zeedd�dd�Zddd�e jeee eee ed�� fdd�Zedd�d d �Zed �d d �Zdd �d d!�Zed �d"d#�Z ee d �d$d%�Z ee j d&�d'd(�Z e j!dd)�d*d+�Z"dd �d,d-�Z#dd �d.d/�Z$dd �d0d1�Z%� Z&S )2�SubprocessTransporta� A bi-directional transport speaking with stdin/out of a subprocess.
Note: this is not really a normal SubprocessTransport. Although it
implements the entire API of asyncio.SubprocessTransport, it is not
designed to be used with asyncio.SubprocessProtocol objects. Instead, it
pair with normal Protocol objects which also implement the
SubprocessProtocol defined in this module (which only has a
process_exited() method). Whatever the protocol writes is sent to stdin,
and whatever comes from stdout is given to the Protocol via the
.data_received() function.
If stderr is configured as a pipe, the transport will separately collect
data from it, making it available via the .get_stderr() method.
N� _returncode�_pty_fdzsubprocess.Popen[bytes]�_process�Spooler�_stderrr. c C s@ z t �t �t �� d�� t�� W S ttfy6 Y n0 t�� S rT ) r( r7 �
pidfd_open�getpidrl ZPidfdChildWatcher�AttributeErrorr ZSafeChildWatcherr r r r �_create_watcher) s
z#SubprocessTransport._create_watcher)r r/ c C s: d}t | |d �}|d u r6t�� }|�| � t| ||� |S )NZ!_cockpit_transports_child_watcher)�getattrr| r� Z attach_loop�setattr)r ZquarkZwatcherr r r � _get_watcher3 s
z SubprocessTransport._get_watcherF��reset�r� r/ c C s( | j d ur | j j|d�jdd�S dS d S )Nr� �replace)�errors� )r� �get�decode)r+ r� r r r �
get_stderr? s
z SubprocessTransport.get_stderr)�pid�coder/ c C s( || _ t�d| j � | js$| j�� d S )Nz Process exited with status %d)r} r& r' r r rv )r+ r� r� r r r �_exitedE s
zSubprocessTransport._exited)�pty�window)r r r r� r� �kwargsc s � fdd�}� rvt �� \| _}|d ur0| �|� ||d< tj|f|||dd�|��| _t �|� | j| j } }
d| _n8tj|ftj tj |d�|��| _| jj
� � } | jj � � }
| jj
d ur�t|| jj
� � �| _nd | _t� �||| |
� | �|��| jj| j� d S )Nc s$ t ttj� � r t�dtjd� d S rT )r �
SET_PDEATHSIG�signal�SIGTERM�fcntl�ioctl�termiosZ TIOCSCTTYr �r� r r �
preexec_fna s z0SubprocessTransport.__init__.<locals>.preexec_fn�stderrT)�stdin�stdoutr� Zstart_new_session)r� r� r� )r( �openptyr~ �set_window_size�
subprocess�Popenr r7 r �PIPEr� �filenor� r� r� r� r$ r% r� Zadd_child_handlerr� r� ) r+ r r r r� r� r� r� Z
session_fdr! r" r, r� r r% W s6
��
�� z SubprocessTransport.__init__)�sizer/ c C s$ t �| jtjt�d|j|j�� d S )NZ2H4x) r� r� r~ r� Z
TIOCSWINSZ�struct�packrz r{ )r+ r� r r r r� � s z#SubprocessTransport.set_window_sizec C s | j jd uS r; )r r� r< r r r rH � s z!SubprocessTransport.can_write_eofc C s | j j�� d| _d S r> )r r� r7 r r< r r r rI � s z"SubprocessTransport._write_eof_nowc C s | j jS r; )r r� r< r r r �get_pid� s zSubprocessTransport.get_pidc C s | j S r; )r} r< r r r �get_returncode� s z"SubprocessTransport.get_returncode)�fdr/ c C s t �d S r; rF )r+ r� r r r �get_pipe_transport� s z&SubprocessTransport.get_pipe_transport)�sigr/ c C sp | j d ur t�d|| jj� d S z&t�| jj|� t�d|| jj� W n$ tyj t�d|| jj� Y n0 d S )Nz3won't attempt %s to process %i. It exited already.zsent %s to process %iz3can't send %s to process %i. It's exited just now.)r} r& r' r r� r( �kill�ProcessLookupError)r+ r� r r r � send_signal� s
z SubprocessTransport.send_signalc C s | � tj� d S r; )r� r� r� r< r r r � terminate� s z SubprocessTransport.terminatec C s | � tj� d S r; )r� r� �SIGKILLr< r r r r� � s zSubprocessTransport.killc C sz | j d ur t�| j � d | _ | jd urv| jjd urF| jj�� d | j_z | �� W n" tyt t�d| jj � Y n0 d S )Nzcan't kill %i due to EPERM)
r~ r( r7 r r� r� �PermissionErrorr& r' r� r< r r r rC � s
zSubprocessTransport._close)'rg rh ri rw r} r rj rk r~ r � staticmethodrl ZAbstractChildWatcherr� rm r� rp rq r� r� ru r rx r r% r� rH rI r� r� � Transportr� r� ZSignalsr� r� r� rC rt r r r, r r| s>
��/r| c sN e Zd ZdZd
ejejeed�� fdd�
Ze d�dd �Z
d
d�d d �Z � Z S )�StdioTransportz�A bi-directional transport that corresponds to stdin/out.
Can talk to just about anything:
- files
- pipes
- character devices (including terminals)
- sockets
r r
)r r r� r� c s t � �||||� d S r; )r$ r% )r+ r r r� r� r, r r r% � s zStdioTransport.__init__r. c C s dS r@ r r< r r r rH � s z StdioTransport.can_write_eofNc C s t d��d S )NzCan't write EOF to stdout)� RuntimeErrorr< r r r rI � s z StdioTransport._write_eof_now)r r
)
rg rh ri rw rl rm rn rj r% rp rH rI rt r r r, r r� � s r� c @ s� e Zd ZU dZejed< eed< ee ed< ejed�dd�Z
dd �d
d �Z e d �d d
�Z
dd�e e d�dd�Zdd �dd�Zdd �dd�ZdS )r� z�Consumes data from an fd, storing it in a buffer.
This makes a copy of the fd, so you don't have to worry about holding it
open.
r �_fd� _contents)r r� c C s@ || _ d| _g | _t�|�| _t�| jd� |�| j| j� d S )Nr F)r r� r� r( �dupr) rB r: )r+ r r� r r r r% � s zSpooler.__init__Nr. c C s^ zt �| jd�}W n( ty& Y d S ty: d}Y n0 |dkrR| j�|� n| �� d S )Ni r0 )r( r1 r� r2 r r� r^ r7 r\ r r r r: � s
zSpooler._read_readyc C s, | j dkrdS t�| j gg g d�g g g fkS )Nr Fr )r� �selectr< r r r � _is_ready s
zSpooler._is_readyFr� r� c C s, | � � r| �� q d�| j�}|r(g | _|S )Nr0 )r� r: r` r� )r+ r� �resultr r r r� s
z Spooler.getc C s. | j dkr*| j�| j � t�| j � d| _ d S r> )r� r rA r( r7 r< r r r r7 s
z
Spooler.closec C s | � � d S r; )r7 r< r r r rf s zSpooler.__del__)rg rh ri rw rl rm rk rj r ro r% r: rp r� r� r7 rf r r r r r� � s
r� )*rw rl rZ Zctypesr3 r� Zloggingr( r� r� r� r� r� �typingr r r r r r r r Zjsonutilr r ZcdllZ LoadLibraryr
r r� Z getLoggerrg r& r_ r� r rn ru rx r| r� r� r r r r �<module> s6 (
T 9