CasperSecurity
o
��!^ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZz
ddlm Z m
Z
W n e y9 ddl m Z m
Z
Y nw ddl m
Z
mZ ddlmZ ddlmZ G dd � d e
�ZdS )
zJProvides an interface like pexpect.spawn interface using subprocess.Popen
� N)�Queue�Empty� )� SpawnBase�PY3)�EOF)� string_typesc sr e Zd Z d� fdd� ZdZdd � Zd
d � Zd d
� Zdd� Zdd� Z d dd�Z
dd� Z dd� Z dd� Z
� ZS ) �
PopenSpawn� �� N�strictc
s t t| �j|||||| d� |d u rtj�d�| _n| �tj�| _tdt j
t j t j
||
|d�} t j
dkrIt �� } | jt jO _| | d< t j| d< t|t�r]t j
dkr]tj|tjdkd �}t j|fi | ��| _| jj| _d
| _| �� | _t � | _ t j | j d �| _!| j!�"d � | j!�#� d S )
N)�timeout�maxread�searchwindowsize�logfile�encoding� codec_errors�asciir )�bufsize�stdin�stderr�stdout�cwd�
preexec_fn�env�win32� startupinfo�
creationflags�posix)r F)�targetT)$�superr �__init__�os�linesep�encode�crlf� string_type�dict�
subprocess�PIPE�STDOUT�sys�platform� STARTUPINFO�dwFlags�STARTF_USESHOWWINDOW�CREATE_NEW_PROCESS_GROUP�
isinstancer �shlex�split�name�Popen�proc�pid�closed�_bufr � _read_queue� threading�Thread�_read_incoming� _read_thread� setDaemon�start)
�self�cmdr
r r r r r r r r �kwargsr �� __class__� �5/usr/lib/python3/dist-packages/pexpect/popen_spawn.pyr! s4 � �
zPopenSpawn.__init__Fc C s | j }| jr |r||d � | _ |d |� S d| _td��|dkr$| j}n|d u r*d}t�� }t�� | |k rq|rqt|�|k rqz| j�� }W n t yN Y n#w |d u rWd| _n|| j
j |dd�7 }t�� | |k rq|rqt|�|k s>|d |� ||d � }| _ | � |d� |S )NTzEnd Of File (EOF).���g ��.AF��final�read)
r9 �_read_reached_eof�flag_eofr r
�time�lenr: �
get_nowaitr �_decoder�decode�_log)rA �sizer
�buf�t0�incoming�rrF rF rG �read_nonblockingA s4 � � zPopenSpawn.read_nonblockingc
C sv | j j�� } d}zt�|d�}W n ty) } z | �|d� W Y d}~nd}~ww |s4| j�d� dS | j�|� q)z6Run in a thread to move output from a pipe to a queue.r � i rK N) r6 r �filenor" rK �OSErrorrS r: �put)rA r[ rU �erF rF rG r= d s �� �zPopenSpawn._read_incomingc C s | � |� dS )zHThis is similar to send() except that there is no return value.
N��send)rA �srF rF rG �writeu s zPopenSpawn.writec C s |D ]}| � |� qdS )z�This calls write() for each element in the sequence.
The sequence can be any iterable object producing strings, typically a
list of strings. This does not add line separators. There is no return
value.
Nr_ )rA �sequencera rF rF rG �
writelinesz s �zPopenSpawn.writelinesc C sN | � |�}| �|d� | jj|dd�}tr | jj�|�S | jj�|� t|�S )zZSend data to the subprocess' stdin.
Returns the number of bytes written.
r` FrI ) �_coerce_send_stringrS �_encoderr$ r r6 r rb rO )rA ra �brF rF rG r` � s
zPopenSpawn.send� c C s | � |�}|| � | j� S )z�Wraps send(), sending string ``s`` to child process, with os.linesep
automatically appended. Returns number of bytes written. )r` r# )rA ra �nrF rF rG �sendline� s
zPopenSpawn.sendlinec C s8 | j �� }|dkr|| _d| _nd| _| | _d| _|S )zKWait for the subprocess to finish.
Returns the exit code.
r NT)r6 �wait�
exitstatus� signalstatus�
terminated)rA �statusrF rF rG rk � s
zPopenSpawn.waitc C sT t jdkr |tjtjfv rtj}n|tjtjfv r tj}ntj}t� | j
j |� dS )z}Sends a Unix signal to the subprocess.
Use constants from the :mod:`signal` module to specify which signal.
r N) r+ r, �signal�SIGINT� CTRL_C_EVENT�SIGBREAK�CTRL_BREAK_EVENT�SIGTERMr" �killr6 r7 )rA �sigrF rF rG rv � s
zPopenSpawn.killc C s | j j�� dS )z+Closes the stdin pipe from the writing end.N)r6 r �close)rA rF rF rG �sendeof� s zPopenSpawn.sendeof) r
r NNNNNr N)rh )�__name__�
__module__� __qualname__r! rL rY r= rb rd r` rj rk rv ry �
__classcell__rF rF rD rG r s �*#
r )�__doc__r" r; r( r+ rN rp r2 �queuer r � ImportError� spawnbaser r �
exceptionsr �utilsr r rF rF rF rG �<module> s" �