3
��KYsx � @ sv d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z y d dl Z W n e
k
r� d dlZ Y nX d dlmZmZ ddlmZ e jj� Zejd�p�ejd�Zer�dZddlmZ nd Ze jd d
kZ e r�d d � Z n(d
d � Z G dd� de �Z G dd� de �Z d \a!a"dd� Z#G dd� de$�Z%dd� Z&dd� Z'G dd� de(�Z)G d d � d e)�Z*dS ) � N)� STDIN_FILENO�CHILD� )�which�solaris�sunosF)� _fork_ptyT� c C s
t | g�S )N)�bytes)�i� r � /usr/lib/python3.6/ptyprocess.py�_byte( s r c C s t | �S )N)�chr)r r r r
r + s c @ s e Zd ZdS )�FileNotFoundErrorN)�__name__�
__module__� __qualname__r r r r
r . s r c @ s e Zd ZdS )� TimeoutErrorN)r r r r r r r
r / s r c C s� t dk rtdk rdS ynddlm} m} ytjj� }W n tk
rR tj j� }Y nX t
tj |�d | �}t
tj |�d | �}W n\ t t
tttjfk
r� y ddlm}m} || }}W n t k
r� d\}}Y nX Y nX t|�at|�a dS ) zkSet constants _EOF and _INTR.
This avoids doing potentially costly operations on module load.
Nr )�VEOF�VINTR� )�CEOF�CINTRr � )r r )�_EOF�_INTR�termiosr r �sys� __stdin__�fileno�
ValueError�
__stdout__�ord� tcgetattr� ImportError�OSError�IOError�errorr r r )r r �fdZintr�eofr r r r r
�_make_eof_intr3 s$ r+ c @ s e Zd ZdZdS )�PtyProcessErrorz%Generic error class for this package.N)r r r �__doc__r r r r
r, T s r, c C s d}yt j| �}W nT t jk
rf } z6|jd tjkrTt|jd d|jd |f ��� W Y d d }~X nX |r�|d t jB |d< n|d t j @ |d< yt j| t j |� W nR tk
r� } z6|jd tjkr�t|jd d|jd |f ��� W Y d d }~X nX d S )Nz,setecho() may not be called on this platformr z%s: %s.r r )
r r$ r( �args�errno�EINVALr' �ECHOZ tcsetattrZTCSANOW)r) �state�errmsg�attr�errr r r
�_setechoZ s r6 c C s0 t tdd�}tjd||dd�}tj| ||� d S )N�
TIOCSWINSZi����HHHHr igt�)�getattrr �struct�pack�fcntl�ioctl)r) �rows�colsr7 �sr r r
� _setwinsizer s rA c @ sP e Zd ZdZeZer8ejj d�Zdj d�Z
e dd� �Z nejZdZ
e
jjZ dZdZdZdZdd� ZedCd d
��Zdd� Ze dd� �Ze dd� �Zdd� Zdd� ZdDdd�Z dd� Z d d � Z dEd d �Z d d!� Z d"d#� Z!dFd%d&�Z"d'd(� Z#dGd)d*�Z$dHd+d,�Zd-d.� Z%d/d0� Z&d1d2� Z'd3d4� Z(dId6d7�Z)d8d9� Z*d:d;� Z+d<d=� Z,d>d?� Z-d@dA� Z.dS )J�
PtyProcessz�This class represents a process running in a pseudoterminal.
The main constructor is the :meth:`spawn` classmethod.
�asciiz
c C s8 yt jjj| �S tk
r2 t jj| jdd��S X d S )NrC �replace)r �stdout�buffer�write�AttributeError�decode)�br r r
�write_to_stdout� s zPtyProcess.write_to_stdoutNc C sv t � || _|| _tj|ddd�}tj|dddd�}tj||�| _d| _d| _d | _ d | _
d | _ d| _ d| _
d| _d S )N�rbr )� buffering�wbF)rM �closefdg�������?)r+ �pidr) �io�open�BufferedRWPair�fileobj�
terminated�closed�
exitstatus� signalstatus�status�flag_eof�delayafterclose�delayafterterminate)�selfrP r) ZreadfZwritefr r r
�__init__� s zPtyProcess.__init__T� �P c N C s� t |ttf�std| ��|dd� }|d }t|�}|dkrNtdd| ��|}||d< tj� \} }
trxt j
� \} } n t j � \} } | t
k�r�yttf|�� W n: tk
r� }
z |
jd tjtjfkrʂ W Y dd}
~
X nX |�s6yttd� W nD ttjfk
�r4 }
z |
jd tjtjfk�r$� W Y dd}
~
X nX tj| � tj|
tjtj� t j t j �d }tj d|
� tj |
d |� |dk �r�tj |� |dk �ry
|� W nn t!k
�r } zPt"|�j#}d j$|t%|��}t&�r�|j'd
�}tj(|
|� tj|
� tj)d� W Y dd}~X nX y*|dk�r0tj*||� ntj+|||� W nh t,k
�r� }
zJd j$|
jt%|
��}t&�rv|j'd
�}tj(|
|� tj|
� tj)tj-� W Y dd}
~
X nX | | | �}||_.|dk �r�||_/|dk �r�||_0tj|
� tj1| d �}tj| � t2|�dk�rryL|j3d
d�\}}}t4t5|j6d�t!�}||j6d
d��}|t,k�rRt7|�|_W n t!d| ��Y nX |�y|j8|� W nB tk
�r� }
z$|
jd tjtjtj9fk�r�� W Y dd}
~
X nX |S )a% Start the given command in a child process in a pseudo terminal.
This does all the fork/exec type of stuff for a pty, and returns an
instance of PtyProcess.
If preexec_fn is supplied, it will be called with no arguments in the
child process before exec-ing the specified command.
It may, for instance, set signal handlers to SIG_DFL or SIG_IGN.
Dimensions of the psuedoterminal used for the subprocess can be
specified as a tuple (rows, cols), or the default (24, 80) will be used.
z)Expected a list or tuple for argv, got %rNr z%The command was not found or was not zexecutable: %s.Fr r z{}:0:{}zutf-8z
OSError:{}:{}i � :� rC rD z)Subprocess failed, got bad error data: %r):�
isinstance�list�tuple� TypeErrorr r �os�pipe�use_native_pty_fork�pty�forkr Zfork_ptyr rA r r' r. r/ r0 ZENOTTYr6 r r( �closer< ZF_SETFDZ
FD_CLOEXEC�resourceZ getrlimitZ
RLIMIT_NOFILE�
closerange�chdir� Exception�typer �format�str�PY3�encoderG �_exit�execv�execvper& �EX_OSERR�argv�env�
launch_dir�read�len�splitr9 �builtinsrI �int�
setwinsizeZENXIO)�clsrz �cwdr{ �echoZ
preexec_fnZ
dimensionsZcommandZcommand_with_pathZexec_err_pipe_readZexec_err_pipe_writerP r) r5 Zmax_fd�eZenameZtosendZinstZ
exec_err_dataZerrclassZerrno_sr3 �exctypeZ exceptionr r r
�spawn� s�
zPtyProcess.spawnc C s| t | �j}| jd k rft| j�g}| jd k r:|jd| j � | jd k rT|jd| j � dj|dj|��S dj|| j | j
�S d S )Nzenv=%rzcwd=%rz {}.spawn({})z, z{}(pid={}, fd={})) rq r rz �reprr{ �appendr| rr �joinrP r) )r] Zclsnamer. r r r
�__repr__M s
zPtyProcess.__repr__c C s t | t�s| jd�S | S )Nzutf-8)rc r
ru )r@ r r r
�_coerce_send_string[ s
z PtyProcess._coerce_send_stringc C s | S )Nr )r@ r r r
�_coerce_read_stringa s z PtyProcess._coerce_read_stringc C s$ | j s y | j� W n Y nX dS )aL This makes sure that no system resources are left open. Python only
garbage collects Python objects. OS file descriptors are not Python
objects, so they must be handled explicitly. If the child file
descriptor was opened outside of this class (passed to the constructor)
then this does not close it. N)rV rl )r] r r r
�__del__e s
zPtyProcess.__del__c C s | j S )zCThis returns the file descriptor of the pty for the child.
)r) )r] r r r
r w s zPtyProcess.filenoc C sN | j sJ| j� | jj� tj| j� | j� r>| j|�s>t d��d| _
d| _ dS )a? This closes the connection with the child application. Note that
calling close() more than once is valid. This emulates standard Python
behavior with files. Set force to True if you want to make sure that
the child is terminated (SIGKILL is sent if the child ignores SIGHUP
and SIGINT). z Could not terminate the child.r TN���) rV �flushrT rl �time�sleepr[ �isalive� terminater, r) )r] �forcer r r
rl | s
zPtyProcess.closec C s dS )zWThis does nothing. It is here to support the interface for a
File-like object. Nr )r] r r r
r� � s zPtyProcess.flushc C s t j| j�S )a^ This returns True if the file descriptor is open and connected to a
tty(-like) device, else False.
On SVR4-style platforms implementing streams, such as SunOS and HP-UX,
the child pty may not appear as a terminal device. This means
methods such as setecho(), setwinsize(), getwinsize() may raise an
IOError. )rg �isattyr) )r] r r r
r� � s zPtyProcess.isattyc C s\ |dk rt j � | }xB| j� s"dS |dk r6|dk r6dS |dk rJ|t j � }t jd� qW dS )a� This waits until the terminal ECHO flag is set False. This returns
True if the echo mode is off. This returns False if the ECHO flag was
not set False before the timeout. This can be used to detect when the
child is waiting for a password. Usually a child application will turn
off echo mode when it is waiting for the user to enter a password. For
example, instead of expecting the "password:" prompt you can wait for
the child to set ECHO off::
p = pexpect.spawn('ssh user@example.com')
p.waitnoecho()
p.sendline(mypassword)
If timeout==None then this method to block until ECHO flag is False.
NTr Fg�������?)r� �getechor� )r] ZtimeoutZend_timer r r
�
waitnoecho� s zPtyProcess.waitnoechoc C s� yt j| j�}W nX t jk
rh } z:d}|jd tjkrVt|jd d|jd |f ��� W Y dd}~X nX t|d t j @ �| _
| j
S )a This returns the terminal echo mode. This returns True if echo is
on or False if echo is off. Child applications that are expecting you
to enter a password often set ECHO False. See waitnoecho().
Not supported on platforms where ``isatty()`` returns False. z,getecho() may not be called on this platformr z%s: %s.r Nr ) r r$ r) r( r. r/ r0 r' �boolr1 r� )r] r4 r5 r3 r r r
r� � s zPtyProcess.getechoc C s t | j|� || _dS )aZ This sets the terminal echo mode on or off. Note that anything the
child sent before the echo will be lost, so you should be sure that
your input buffer is empty before you call setecho(). For example, the
following will work as expected::
p = pexpect.spawn('cat') # Echo is on by default.
p.sendline('1234') # We expect see this twice from the child...
p.expect(['1234']) # ... once from the tty echo...
p.expect(['1234']) # ... and again from cat itself.
p.setecho(False) # Turn off tty echo
p.sendline('abcd') # We will set this only once (echoed by cat).
p.sendline('wxyz') # We will set this only once (echoed by cat)
p.expect(['abcd'])
p.expect(['wxyz'])
The following WILL NOT WORK because the lines sent before the setecho
will be lost::
p = pexpect.spawn('cat')
p.sendline('1234')
p.setecho(False) # Turn off tty echo
p.sendline('abcd') # We will set this only once (echoed by cat).
p.sendline('wxyz') # We will set this only once (echoed by cat)
p.expect(['1234'])
p.expect(['1234'])
p.expect(['abcd'])
p.expect(['wxyz'])
Not supported on platforms where ``isatty()`` returns False.
N)r6 r) r� )r] r2 r r r
�setecho� s zPtyProcess.setecho� c C sr y| j j|�}W nF ttfk
rV } z&|jd tjkrDd| _td��� W Y dd}~X nX |dkrnd| _td��|S )a Read and return at most ``size`` bytes from the pty.
Can block if there is nothing to read. Raises :exc:`EOFError` if the
terminal was closed.
Unlike Pexpect's ``read_nonblocking`` method, this doesn't try to deal
with the vagaries of EOF on platforms that do strange things, like IRIX
or older Solaris systems. It handles the errno=EIO pattern used on
Linux, and the empty-string return used on BSD platforms and (seemingly)
on recent Solaris.
r Tz,End Of File (EOF). Exception style platform.N� z/End Of File (EOF). Empty string style platform.) rT �read1r& r' r. r/ �EIOrZ �EOFError)r] �sizer@ r5 r r r
r} � s zPtyProcess.readc C sp y| j j� }W nF ttfk
rT } z&|jd tjkrBd| _td��� W Y dd}~X nX |dkrld| _td��|S )z�Read one line from the pseudoterminal, and return it as unicode.
Can block if there is nothing to read. Raises :exc:`EOFError` if the
terminal was closed.
r Tz,End Of File (EOF). Exception style platform.Nr� z/End Of File (EOF). Empty string style platform.) rT �readliner& r' r. r/ r� rZ r� )r] r@ r5 r r r
r� s zPtyProcess.readlinec C s | j j|�}|r| j j� |S )N)rT rG r� )r] rJ r� �nr r r
�_writeb s
zPtyProcess._writebc C s | j ||d�S )zaWrite bytes to the pseudoterminal.
Returns the number of bytes written.
)r� )r� )r] r@ r� r r r
rG & s zPtyProcess.writec
C s� |j � }t|�}d| ko"dkn rN|td� d }t|�}| j|�|fS ddddddddd d d
d d � }||krxdS t|| �}| j|�|fS )a Helper method that wraps send() with mnemonic access for sending control
character to the child (such as Ctrl-C or Ctrl-D). For example, to send
Ctrl-G (ASCII 7, bell, '')::
child.sendcontrol('g')
See also, sendintr() and sendeof().
�a �z �ar r � � � � � � ) �@�`�[�{�\�|�]�}�^�~�_�?r� )r r� )�lowerr# r r� )r] �charr� Zbyte�dr r r
� sendcontrol- s" zPtyProcess.sendcontrolc C s | j t�tfS )a1 This sends an EOF to the child. This sends a character which causes
the pending parent output buffer to be sent to the waiting child
program without waiting for end-of-line. If it is the first character
of the line, the read() in the user program returns 0, which signifies
end-of-file. This means to work as expected a sendeof() has to be
called at the beginning of a line. This method does not send a newline.
It is the responsibility of the caller to ensure the eof is sent at the
beginning of a line. )r� r )r] r r r
�sendeofI s
zPtyProcess.sendeofc C s | j t�tfS )znThis sends a SIGINT to the child. It does not require
the SIGINT to be the first character on a line. )r� r )r] r r r
�sendintrU s zPtyProcess.sendintrc C s | j S )z@This returns True if the EOF exception was ever raised.
)rZ )r] r r r
r* [ s zPtyProcess.eofFc C s� | j � s dS y�| jtj� tj| j� | j � s2dS | jtj� tj| j� | j � sVdS | jtj� tj| j� | j � szdS |r�| jtj � tj| j� | j � s�dS dS dS t
k
r� tj| j� | j � s�dS dS Y nX dS )z�This forces a child process to terminate. It starts nicely with
SIGHUP and SIGINT. If "force" is True then moves onto SIGKILL. This
returns True if the child was terminated. This returns False if the
child could not be terminated. TFN) r� �kill�signal�SIGHUPr� r� r\ �SIGCONT�SIGINT�SIGKILLr& )r] r� r r r
r� a s6 zPtyProcess.terminatec C s� | j � r tj| jd�\}}n| jS tj|�| _tj|�rX|| _tj|�| _d| _d| _ n<tj
|�r�|| _d| _tj |�| _d| _ ntj |�r�t
d��| jS ) am This waits until the child exits. This is a blocking call. This will
not read any data from the child, so this will block forever if the
child has unread output and has terminated. In other words, the child
may have printed output then called exit(), but, the child is
technically still alive until its output is read by the parent. r NTz!Called wait() on a stopped child z.process. This is not supported. Is some other z2process attempting job control with our child pid?zOCalled wait() on a stopped child process. This is not supported. Is some other z�Called wait() on a stopped child process. This is not supported. Is some other process attempting job control with our child pid?)r� rg �waitpidrP rW � WEXITSTATUS� WIFEXITEDrY rX rU � WIFSIGNALED�WTERMSIG�
WIFSTOPPEDr, )r] rP rY r r r
�wait� s&
zPtyProcess.waitc C sP | j r
dS | jrd}ntj}ytj| j|�\}}W n: tk
rl } z |jtjkrZt d��n� W Y dd}~X nX |dkr�ytj| j|�\}}W n: tk
r� } z |jtjkr�t d��n� W Y dd}~X nX |dkr�d S |dkr�d S tj
|��r || _ tj |�| _
d| _d | _ n@tj|��r8|| _ d| _
tj|�| _d | _ ntj|��rLt d��dS )aZ This tests if the child process is running or not. This is
non-blocking. If the child was terminated then this will read the
exitstatus or signalstatus of the child. This returns True if the child
process appears to be running or False if not. It can take literally
SECONDS for Solaris to return the right status. Fr z isalive() encountered condition z0where "terminated" is 0, but there was no child z)process. Did someone else call waitpid() zon our process?Nz-that should never happen. There was no child Tz,where child process is stopped. This is not z,supported. Is some other process attempting z job control with our child pid?zPisalive() encountered condition where "terminated" is 0, but there was no child zyisalive() encountered condition where "terminated" is 0, but there was no child process. Did someone else call waitpid() z�isalive() encountered condition where "terminated" is 0, but there was no child process. Did someone else call waitpid() on our process?zMisalive() encountered condition that should never happen. There was no child zvisalive() encountered condition that should never happen. There was no child process. Did someone else call waitpid() z�isalive() encountered condition that should never happen. There was no child process. Did someone else call waitpid() on our process?zLisalive() encountered condition where child process is stopped. This is not zxisalive() encountered condition where child process is stopped. This is not supported. Is some other process attempting z�isalive() encountered condition where child process is stopped. This is not supported. Is some other process attempting job control with our child pid?)rU rZ rg �WNOHANGr� rP r& r/ ZECHILDr, r� rY r� rW rX r� r� r� )r] Zwaitpid_optionsrP rY r� r r r
r� � sR zPtyProcess.isalivec C s | j � rtj| j|� dS )a Send the given signal to the child application.
In keeping with UNIX tradition it has a misleading name. It does not
necessarily kill the child unless you send the right signal. See the
:mod:`signal` module for constants representing signal numbers.
N)r� rg r� rP )r] Zsigr r r
r� � s zPtyProcess.killc C sB t tdd�}tjddddd�}tj| j||�}tjd|�dd� S )zNReturn the window size of the pseudoterminal as a tuple (rows, cols).
�
TIOCGWINSZiht@r8 r rb )r9 r r: r; r< r= r) �unpack)r] r� r@ �xr r r
�
getwinsize� s zPtyProcess.getwinsizec C s t | j||�S )a@ Set the terminal window size of the child tty.
This will cause a SIGWINCH signal to be sent to the child. This does not
change the physical window size. It changes the size reported to
TTY-aware applications like vi or curses -- applications that respond to
the SIGWINCH signal.
)rA r) )r] r> r? r r r
r� s zPtyProcess.setwinsize�r_ r` )NNTNr� )T)N)r� )T)T)F)/r r r r- r
� string_typert rg �linesepru Zcrlf� staticmethodrK r rE rG �encodingrz r{ r| r^ � classmethodr� r� r� r� r� r rl r� r� r� r� r� r} r� r� r� r� r� r* r� r� r� r� r� r� r r r r
rB | sV
$
( M rB c sV e Zd ZdZereZneZd
� fdd� Zd� fdd� Z � fd d
�Z
� fd d �Z � Z S )�PtyProcessUnicodez�Unicode wrapper around a process running in a pseudoterminal.
This class exposes a similar interface to :class:`PtyProcess`, but its read
methods return unicode, and its :meth:`write` accepts unicode.
�utf-8�strictc s4 t t| �j||� || _|| _tj|�|d�| _d S )N)�errors)�superr� r^ r� � codec_errors�codecs�getincrementaldecoder�decoder)r] rP r) r� r� )� __class__r r
r^ s zPtyProcessUnicode.__init__� c s t t| �j|�}| jj|dd�S )a Read at most ``size`` bytes from the pty, return them as unicode.
Can block if there is nothing to read. Raises :exc:`EOFError` if the
terminal was closed.
The size argument still refers to bytes, not unicode code points.
F)�final)r� r� r} r� rI )r] r� rJ )r� r r
r} # s zPtyProcessUnicode.readc s t t| �j� }| jj|dd�S )z�Read one line from the pseudoterminal, and return it as unicode.
Can block if there is nothing to read. Raises :exc:`EOFError` if the
terminal was closed.
F)r� )r� r� r� r� rI )r] rJ )r� r r
r� . s zPtyProcessUnicode.readlinec s |j | j�}tt| �j|�S )zlWrite the unicode string ``s`` to the pseudoterminal.
Returns the number of bytes written.
)ru r� r� r� rG )r] r@ rJ )r� r r
rG 7 s zPtyProcessUnicode.write)r� r� )r� )
r r r r- rt rs r� Zunicoder^ r} r� rG �
__classcell__r r )r� r
r� s r� )NN)+r� r/ r< rQ rg rj rm r� r: r r r� r� r% Z __builtin__r r �utilr �platformr� Z _platform�
startswithZ _is_solarisri � r � version_infort r r& r r r r r+ rp r, r6 rA �objectrB r� r r r r
�<module> sT
!