CasperSecurity
o
��!^� � @ sZ d dl Z d dlZd dlZd dlmZ e jd
dd��Ze jd dd��ZG dd � d e j�Z dS ) � N)�EOFc
# s� � | � � }|d ur |S | jjs0t� � � �| � t�� �� fdd�| j�E d H \}� � |f| j_n| jj\� }� �| � |�� z t� � j
|�E d H W S tj yf } z|� � | �
|�W Y d }~S d }~ww )Nc s � S �N� r ��pwr �0/usr/lib/python3/dist-packages/pexpect/_async.py�<lambda> s z expect_async.<locals>.<lambda>)�
existing_data�spawn�async_pw_transport�
PatternWaiter� set_expecter�asyncio�get_event_loop�connect_read_pipe�resume_reading�wait_for�fut� TimeoutError�
pause_reading�timeout)�expecterr �idx� transport�er r r � expect_async s* �
�
��r ���c c s� � g }| j �|d � |dd � D ]}| j|dd�E d H |�| j j� | j �|� q| j|dd�E d H }|dkrN| j �tj� | jddd�E d H td��d� || j jg �S )Nr � T)r �async_z1Continuation prompt found - input was incomplete:� )
�child�sendline�_expect_prompt�append�before�kill�signal�SIGINT�
ValueError�join)�repl�cmdlinesr �res�line�
prompt_idxr r r �repl_run_command_async s �r/ c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )r Nc C s || _ t�� | _d S r )r r �Futurer )�selfr r r r r
3 s zPatternWaiter.set_expecterc C �( | j �� s| j �|� | j�� d S d S r )r �done�
set_resultr r )r1 �resultr r r �found7 �
�zPatternWaiter.foundc C r2 r )r r3 �
set_exceptionr r �r1 �excr r r �error< r7 zPatternWaiter.errorc C s
|| _ d S r )r )r1 r r r r �connection_madeA s
z PatternWaiter.connection_madec
C s� | j j}|j�|�}|�|d� | j�� r#|j�|� |j �|� d S z| j �
|�}|d ur6| � |� W d S W d S t yU } z| j �
� | �|� W Y d }~d S d }~ww )N�read)r r
�_decoder�decode�_logr r3 �_before�write�_buffer�new_datar6 � Exception�erroredr; )r1 �datar
�s�indexr r r r �
data_receivedD s"
�
��zPatternWaiter.data_receivedc
C sX z d| j j_| j �� }W n ty$ } z | �|� W Y d }~d S d }~ww | �|� d S )NT)r r
�flag_eof�eofr r; r6 )r1 rI r r r r � eof_receivedW s
��zPatternWaiter.eof_receivedc C s< t |t�r|jtjkr| �� d S |d ur | �|� d S d S r )�
isinstance�OSError�errno�EIOrM r; r9 r r r �connection_lostb s
�z PatternWaiter.connection_lost) �__name__�
__module__� __qualname__r r
r6 r; r< rJ rM rR r r r r r 0 s r r )r )
r rP r& �pexpectr � coroutiner r/ �Protocolr r r r r �<module> s