CasperSecurity
o
`�wi� � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ej Z ej
Z
ej Z G d d
� d
ej ej
�ZG d d � d �Zdddejfd
d�Zdddejd�dd�ZdS ))�create_subprocess_exec�create_subprocess_shell� N� )�events)� protocols)�streams)�tasks)�loggerc sX e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�SubprocessStreamProtocolz0Like StreamReaderProtocol, but for a subprocess.c sH t � j|d� || _d | _ | _| _d | _d| _g | _| j �
� | _ d S )N)�loopF) �super�__init__�_limit�stdin�stdout�stderr�
_transport�_process_exited� _pipe_fds�_loop�
create_future�
_stdin_closed)�self�limitr �� __class__� �)/usr/lib/python3.10/asyncio/subprocess.pyr
s z!SubprocessStreamProtocol.__init__c C sn | j jg}| jd ur|�d| j��� | jd ur!|�d| j��� | jd ur/|�d| j��� d�d�|��S )Nzstdin=zstdout=zstderr=z<{}>� )r �__name__r �appendr r �format�join)r �infor r r �__repr__ s
z!SubprocessStreamProtocol.__repr__c C s� || _ |�d�}|d ur"tj| j| jd�| _| j�|� | j� d� |�d�}|d urAtj| j| jd�| _
| j
�|� | j� d� |�d�}|d urWtj || d | jd�| _ d S d S )Nr �r r � r )�protocol�readerr )
r �get_pipe_transportr � StreamReaderr r r �
set_transportr r r � StreamWriterr )r � transport�stdout_transport�stderr_transport�stdin_transportr r r �connection_made( s.
�
�
��z(SubprocessStreamProtocol.connection_madec C s> |dkr| j }n
|dkr| j}nd }|d ur |�|� d S d S �Nr r& )r r � feed_data)r �fd�datar( r r r �pipe_data_received@ s �z+SubprocessStreamProtocol.pipe_data_receivedc C s� |dkr(| j }|d ur|�� | �|� |d u r | j�d � d S | j�|� d S |dkr0| j}n
|dkr8| j}nd }|d urL|d u rG|�� n|�|� || j v rW| j �
|� | � � d S )Nr r r& ) r �close�connection_lostr �
set_result�
set_exceptionr r �feed_eofr �remove�_maybe_close_transport)r r4 �exc�piper( r r r �pipe_connection_lostJ s,
�
z-SubprocessStreamProtocol.pipe_connection_lostc C s d| _ | �� d S )NT)r r= �r r r r �process_exitede s z'SubprocessStreamProtocol.process_exitedc C s0 t | j�dkr| jr| j�� d | _d S d S d S )Nr )�lenr r r r7 rA r r r r= i s
�z/SubprocessStreamProtocol._maybe_close_transportc C s || j u r| jS d S �N)r r )r �streamr r r �_get_close_waitern s
�z*SubprocessStreamProtocol._get_close_waiter)
r �
__module__� __qualname__�__doc__r
r$ r1 r6 r@ rB r= rF �
__classcell__r r r r r
s
r
c @ sj e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
ddd�ZdS )�Processc C s8 || _ || _|| _|j| _|j| _|j| _|�� | _d S rD )r � _protocolr r r r �get_pid�pid)r r- r' r r r r r
t s zProcess.__init__c C s d| j j� d| j� d�S )N�<r �>)r r rN rA r r r r$ } s zProcess.__repr__c C s
| j �� S rD )r �get_returncoderA r r r �
returncode� s
zProcess.returncodec � s �| j �� I dH S )z?Wait until the process exit and return the process return code.N)r �_waitrA r r r �wait� s �z Process.waitc C s | j �|� d S rD )r � send_signal)r �signalr r r rU � s zProcess.send_signalc C � | j �� d S rD )r � terminaterA r r r rX � � zProcess.terminatec C rW rD )r �killrA r r r rZ � rY z Process.killc
� s� �| j �� }| j�|� |rt�d| t|�� z
| j�� I d H W n tt fy> } z|r4t�d| |� W Y d }~nd }~ww |rGt�d| � | j�
� d S )Nz%%r communicate: feed stdin (%s bytes)z %r communicate: stdin got %rz%r communicate: close stdin) r � get_debugr �writer �debugrC �drain�BrokenPipeError�ConnectionResetErrorr7 )r �inputr] r> r r r � _feed_stdin� s" �
��� zProcess._feed_stdinc � s �d S rD r rA r r r �_noop� s �z
Process._noopc � s� �| j �|�}|dkr| j}n |dksJ �| j}| j�� r,|dkr#dnd}t�d| |� |�� I d H }| j�� rG|dkr>dnd}t�d| |� |� � |S )Nr& r r r z%r communicate: read %sz%r communicate: close %s)
r r) r r r r[ r r] �readr7 )r r4 r- rE �name�outputr r r � _read_stream� s �
zProcess._read_streamNc � s� �|d ur | � |�}n| �� }| jd ur| �d�}n| �� }| jd ur)| �d�}n| �� }t�|||�I d H \}}}| �� I d H ||fS r2 )rb rc r rg r r �gatherrT )r ra r r r r r r � communicate� s �
zProcess.communicaterD )r rG rH r
r$ �propertyrR rT rU rX rZ rb rc rg ri r r r r rK s s
rK c � sJ �t �� �� �fdd�}�j|| f|||d�|��I d H \}}t||��S )Nc � t � �d�S �Nr% �r
r r% r r �<lambda>� � �z)create_subprocess_shell.<locals>.<lambda>�r r r )r �get_running_loop�subprocess_shellrK ) �cmdr r r r �kwds�protocol_factoryr- r' r r% r r � s ���� r )r r r r c
� sP �t �� �� �fdd�}�j|| g|�R |||d�|��I d H \}} t|| ��S )Nc rk rl rm r r% r r rn � ro z(create_subprocess_exec.<locals>.<lambda>rp )r rq �subprocess_execrK )
�programr r r r �argsrt ru r- r' r r% r r � s ����� r )�__all__�
subprocess� r r r r �logr �PIPE�STDOUT�DEVNULL�FlowControlMixin�SubprocessProtocolr
rK �_DEFAULT_LIMITr r r r r r �<module> s( �bU
� �