3
\�� �
@ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddlmZ ddlmZ dddddgZejdkr�ed��dd� Zy
ej Z W n e k
�r, dd� Z Y nX G dd � d ej �Z e!ed ��rVd d � Z"nddl#Z#d d � Z"G d!d"� d"ej$�Z%G d#d$� d$ej&ej'�Z(e!ed%��r�ej)Z*nddl#Z#d&d'� Z*G d(d)� d)e j+�Z,G d*d� d�Z-G d+d,� d,e-�Z.G d-d� de.�Z/G d.d� de.�Z0G d/d0� d0ej1�Z2e Z3e2Z4dS )1z2Selector event loop for Unix with signal handling.� N� )� base_events)�base_subprocess)�compat)� constants)�
coroutines)�events)�futures)�selector_events)� selectors)�
transports)� coroutine)�logger�SelectorEventLoop�AbstractChildWatcher�SafeChildWatcher�FastChildWatcher�DefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s dS )zDummy signal handler.N� )�signum�framer r �+/usr/lib64/python3.6/asyncio/unix_events.py�_sighandler_noop% s r c C s | S )Nr )�pathr r r �. s r c s� e Zd ZdZd"� fdd� Zdd� Z� fdd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z d#dd�Z d$dd�Z
ed%dd��Zdd� Zedddd�d d ��Zed&dd dd �d d!��Z� ZS )'�_UnixSelectorEventLoopzdUnix event loop.
Adds signal handling and UNIX Domain Socket support to SelectorEventLoop.
Nc s t � j|� i | _d S )N)�super�__init__�_signal_handlers)�self�selector)� __class__r r r 7 s z _UnixSelectorEventLoop.__init__c C s t j� S )N)�socketZ
socketpair)r r r r � _socketpair; s z"_UnixSelectorEventLoop._socketpairc s^ t � j� tj� s2xFt| j�D ]}| j|� q W n(| jrZtjd| �d�t | d� | jj
� d S )NzClosing the loop z@ on interpreter shutdown stage, skipping signal handlers removal)�source) r �close�sys�
is_finalizing�listr �remove_signal_handler�warnings�warn�ResourceWarning�clear)r �sig)r! r r r% > s
z _UnixSelectorEventLoop.closec C s" x |D ]}|sq| j |� qW d S )N)�_handle_signal)r �datar r r r �_process_self_dataL s
z)_UnixSelectorEventLoop._process_self_datac + G sH t j|�st j|�r td��| j|� | j� ytj| jj � � W n2 t
t fk
rt } zt t
|���W Y dd}~X nX tj||| �}|| j|< y tj|t� tj|d� W n� t k
�rB } zz| j|= | j�sytjd� W n4 t
t fk
�r } ztjd|� W Y dd}~X nX |jtjk�r0t dj|���n� W Y dd}~X nX dS )z�Add a handler for a signal. UNIX only.
Raise ValueError if the signal number is invalid or uncatchable.
Raise RuntimeError if there is a problem setting up the handler.
z3coroutines cannot be used with add_signal_handler()NFr z set_wakeup_fd(-1) failed: %szsig {} cannot be caught�����)r Z iscoroutineZiscoroutinefunction� TypeError�
_check_signalZ
_check_closed�signal�
set_wakeup_fdZ_csock�fileno�
ValueError�OSError� RuntimeError�strr ZHandler r � siginterruptr �info�errno�EINVAL�format)r r. �callback�args�exc�handleZnexcr r r �add_signal_handlerS s0
z)_UnixSelectorEventLoop.add_signal_handlerc C s8 | j j|�}|dkrdS |jr*| j|� n
| j|� dS )z2Internal helper that is the actual signal handler.N)r �getZ
_cancelledr) Z_add_callback_signalsafe)r r. rD r r r r/ � s z%_UnixSelectorEventLoop._handle_signalc & C s� | j |� y | j|= W n tk
r* dS X |tjkr>tj}ntj}ytj||� W n@ tk
r� } z$|jtj kr�t
dj |���n� W Y dd}~X nX | js�ytj d� W n2 t
tfk
r� } ztjd|� W Y dd}~X nX dS )zwRemove a handler for a signal. UNIX only.
Return True if a signal handler was removed, False if not.
Fzsig {} cannot be caughtNr z set_wakeup_fd(-1) failed: %sTr2 )r4 r �KeyErrorr5 �SIGINT�default_int_handler�SIG_DFLr9 r> r? r: r@ r6 r8 r r= )r r. ZhandlerrC r r r r) � s(
z,_UnixSelectorEventLoop.remove_signal_handlerc C sH t |t�stdj|���d| ko,tjk n sDtdj|tj���dS )z�Internal helper to validate a signal.
Raise ValueError if the signal number is invalid or uncatchable.
Raise RuntimeError if there is a problem setting up the handler.
z sig must be an int, not {!r}r zsig {} out of range(1, {})N)�
isinstance�intr3 r@ r5 �NSIGr8 )r r. r r r r4 � s
z$_UnixSelectorEventLoop._check_signalc C s t | ||||�S )N)�_UnixReadPipeTransport)r �pipe�protocol�waiter�extrar r r �_make_read_pipe_transport� s z0_UnixSelectorEventLoop._make_read_pipe_transportc C s t | ||||�S )N)�_UnixWritePipeTransport)r rO rP rQ rR r r r �_make_write_pipe_transport� s z1_UnixSelectorEventLoop._make_write_pipe_transportc k s� t j� ��}
| j� } t| |||||||f| |d�| ��} |
j| j� | j| � y| E d H W n& tk
r~ }
z
|
}W Y d d }
~
X nX d }|d k r�| j� | j � E d H |�W d Q R X | S )N)rQ rR )
r �get_child_watcherZ
create_future�_UnixSubprocessTransport�add_child_handlerZget_pid�_child_watcher_callback� Exceptionr% Z_wait)r rP rB �shell�stdin�stdout�stderr�bufsizerR �kwargs�watcherrQ �transprC �errr r r �_make_subprocess_transport� s$
z1_UnixSelectorEventLoop._make_subprocess_transportc C s | j |j|� d S )N)Zcall_soon_threadsafeZ_process_exited)r �pid�
returncoderb r r r rY � s z._UnixSelectorEventLoop._child_watcher_callback)�ssl�sock�server_hostnamec c s |d kst |t�st�|r,|d kr|d k�rBtd��|jtjk�s`tj|j� �rntdj|���tj| |g�} |j|� |j d � | j |||| � | S ) Nz*ssl argument must be an SSLContext or Nonez3path and sock can not be specified at the same timer � z2Unable to check or remove stale UNIX socket %r: %rz Address {!r} is already in usez-path was not specified, and no sock specifiedz2A UNIX Domain Stream Socket was expected, got {!r}F)r rv ) rK �boolr3 r8 �_fspathr" rk rl �stat�S_ISSOCK�os�st_mode�remove�FileNotFoundErrorr9 r �errorZbindr% r> Z
EADDRINUSEr@ rn r ro rp ZServerZlistenrm Z_start_serving)
r rq r rh ru rg rc rC �msgZserverr r r �create_unix_server sP
z)_UnixSelectorEventLoop.create_unix_server)N)NN)NN)N)N)�__name__�
__module__� __qualname__�__doc__r r# r% r1 rE r/ r) r4 rS rU r
rd rY rs r� �
__classcell__r r )r! r r 1 s, -
%r � set_blockingc C s t j| d� d S )NF)r{ r� )�fdr r r �_set_nonblockingB s r� c C s, t j | t j�}|tjB }t j | t j|� d S )N)�fcntlZF_GETFLr{ �
O_NONBLOCKZF_SETFL)r� �flagsr r r r� G s
c s� e Zd Zd Zd � fdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z e
jrhdd� Zd!dd�Zdd � Zd d � Z� ZS )"rN � i Nc s� t � j|� || jd< || _|| _|j� | _|| _d| _t j
| j�j }t j
|�pbt j|�pbt j|�s~d | _d | _d | _td��t| j� | jj| jj| � | jj| jj| j| j� |d k r�| jjtj|d � d S )NrO Fz)Pipe transport is for pipes/sockets only.)r r �_extra�_loop�_piper7 �_fileno� _protocol�_closingr{ �fstatr| ry �S_ISFIFOrz �S_ISCHRr8 r� � call_soon�connection_made� _add_reader� _read_readyr � _set_result_unless_cancelled)r �looprO rP rQ rR �mode)r! r r r Q s,
z _UnixReadPipeTransport.__init__c C s� | j jg}| jd kr |jd� n| jr0|jd� |jd| j � t| jdd �}| jd k r�|d k r�tj || jt
j �}|r�|jd� q�|jd� n | jd k r�|jd� n
|jd� dd j |� S )
N�closed�closingzfd=%s� _selector�polling�idle�openz<%s>� )
r! r� r� �appendr� r� �getattrr� r
�_test_selector_eventr Z
EVENT_READ�join)r r= r r� r r r �__repr__n s$
z _UnixReadPipeTransport.__repr__c C s� yt j| j| j�}W nD ttfk
r, Y n� tk
rX } z| j|d� W Y d d }~X n^X |rl| jj |� nJ| j
j � r�t j
d| � d| _| j
j| j� | j
j| jj� | j
j| jd � d S )Nz"Fatal read error on pipe transportz%r was closed by peerT)r{ �readr� �max_size�BlockingIOError�InterruptedErrorr9 � _fatal_errorr� Z
data_receivedr� � get_debugr r= r� �_remove_readerr� Z eof_received�_call_connection_lost)r r0 rC r r r r� � s
z"_UnixReadPipeTransport._read_readyc C s | j j| j� d S )N)r� r� r� )r r r r �
pause_reading� s z$_UnixReadPipeTransport.pause_readingc C s | j j| j| j� d S )N)r� r� r� r� )r r r r �resume_reading� s z%_UnixReadPipeTransport.resume_readingc C s
|| _ d S )N)r� )r rP r r r � set_protocol� s z#_UnixReadPipeTransport.set_protocolc C s | j S )N)r� )r r r r � get_protocol� s z#_UnixReadPipeTransport.get_protocolc C s | j S )N)r� )r r r r �
is_closing� s z!_UnixReadPipeTransport.is_closingc C s | j s| jd � d S )N)r� �_close)r r r r r% � s z _UnixReadPipeTransport.closec C s, | j d k r(tjd| t| d� | j j� d S )Nzunclosed transport %r)r$ )r� r* r+ r, r% )r r r r �__del__� s
z _UnixReadPipeTransport.__del__� Fatal error on pipe transportc C sZ t |t�r4|jtjkr4| jj� rLtjd| |dd� n| jj||| | j d�� | j
|� d S )Nz%r: %sT)�exc_info)�message� exceptionrr rP ) rK r9 r> ZEIOr� r� r �debug�call_exception_handlerr� r� )r rC r� r r r r� � s
z#_UnixReadPipeTransport._fatal_errorc C s( d| _ | jj| j� | jj| j|� d S )NT)r� r� r� r� r� r� )r rC r r r r� � s z _UnixReadPipeTransport._closec C s4 z| j j|� W d | jj� d | _d | _ d | _X d S )N)r� �connection_lostr� r% r� )r rC r r r r� � s
z,_UnixReadPipeTransport._call_connection_losti )NN)r� )r� r� r� r� r r� r� r� r� r� r� r� r% r �PY34r� r� r� r� r� r r )r! r rN M s
rN c s� e Zd Zd%� fdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zejr|dd� Zd d � Zd&d d �Zd'd!d"�Zd#d$� Z� ZS )(rT Nc
s� t � j||� || jd< || _|j� | _|| _t� | _d| _ d| _
t j | j�j
}tj|�}tj|�}tj|�} |px|px| s�d | _d | _d | _td��t| j� | jj| jj| � | s�|r�tjjd� r�| jj| jj| j| j� |d k r�| jjt j |d � d S )NrO r Fz?Pipe transport is only for pipes, sockets and character devices�aix) r r r� r� r7 r� r� � bytearray�_buffer�
_conn_lostr� r{ r� r| ry r� r� rz r8 r� r� r� r� r& �platform�
startswithr� r� r r� )
r r� rO rP rQ rR r� Zis_charZis_fifoZ is_socket)r! r r r � s2
z _UnixWritePipeTransport.__init__c C s� | j jg}| jd kr |jd� n| jr0|jd� |jd| j � t| jdd �}| jd k r�|d k r�tj || jt
j �}|r�|jd� n
|jd� | j � }|jd| � n | jd k r�|jd� n
|jd� d d
j
|� S ) Nr� r� zfd=%sr� r� r� z
bufsize=%sr� z<%s>r� )r! r� r� r� r� r� r� r� r
r� r Z EVENT_WRITE�get_write_buffer_sizer� )r r= r r� r_ r r r r� � s(
z _UnixWritePipeTransport.__repr__c C s
t | j�S )N)�lenr� )r r r r r� s z-_UnixWritePipeTransport.get_write_buffer_sizec C s6 | j j� rtjd| � | jr*| jt� � n| j� d S )Nz%r was closed by peer)r� r� r r= r� r� �BrokenPipeError)r r r r r� s
z#_UnixWritePipeTransport._read_readyc C s0 t |tttf�s tt|���t |t�r.t|�}|s6d S | jsB| jrj| jtj krXt
j d� | jd7 _d S | j �syt
j| j|�}W nT ttfk
r� d}Y n: tk
r� } z | jd7 _| j|d� d S d }~X nX |t|�kr�d S |dk�rt|�|d � }| jj| j| j� | j |7 _ | j� d S )Nz=pipe closed by peer or os.write(pipe, data) raised exception.r r z#Fatal write error on pipe transport)rK �bytesr� �
memoryviewrj �reprr� r� r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr �warningr� r{ �writer� r� r� rZ r� r� r� Z _add_writer� _write_readyZ_maybe_pause_protocol)r r0 �nrC r r r r� s4
z _UnixWritePipeTransport.writec C s� | j std��ytj| j| j �}W nj ttfk
r: Y n� tk
r� } z8| j j� | j d7 _ | j
j | j� | j |d� W Y d d }~X nfX |t
| j �kr�| j j� | j
j | j� | j� | jr�| j
j| j� | jd � d S |dkr�| j d |�= d S )NzData should not be emptyr z#Fatal write error on pipe transportr )r� rj r{ r� r� r� r� rZ r- r� r� �_remove_writerr� r� Z_maybe_resume_protocolr� r� r� )r r� rC r r r r� > s(
z$_UnixWritePipeTransport._write_readyc C s dS )NTr )r r r r �
can_write_eofX s z%_UnixWritePipeTransport.can_write_eofc C sB | j r
d S | jst�d| _ | js>| jj| j� | jj| jd � d S )NT) r� r� rj r� r� r� r� r� r� )r r r r � write_eof[ s
z!_UnixWritePipeTransport.write_eofc C s
|| _ d S )N)r� )r rP r r r r� d s z$_UnixWritePipeTransport.set_protocolc C s | j S )N)r� )r r r r r� g s z$_UnixWritePipeTransport.get_protocolc C s | j S )N)r� )r r r r r� j s z"_UnixWritePipeTransport.is_closingc C s | j d k r| j r| j� d S )N)r� r� r� )r r r r r% m s z _UnixWritePipeTransport.closec C s, | j d k r(tjd| t| d� | j j� d S )Nzunclosed transport %r)r$ )r� r* r+ r, r% )r r r r r� v s
z _UnixWritePipeTransport.__del__c C s | j d � d S )N)r� )r r r r �abort| s z _UnixWritePipeTransport.abort� Fatal error on pipe transportc C sP t |tj�r*| jj� rBtjd| |dd� n| jj||| | jd�� | j |� d S )Nz%r: %sT)r� )r� r� rr rP )
rK r Z_FATAL_ERROR_IGNOREr� r� r r� r� r� r� )r rC r� r r r r� s
z$_UnixWritePipeTransport._fatal_errorc C sF d| _ | jr| jj| j� | jj� | jj| j� | jj| j|� d S )NT) r� r� r� r� r� r- r� r� r� )r rC r r r r� � s
z _UnixWritePipeTransport._closec C s4 z| j j|� W d | jj� d | _d | _ d | _X d S )N)r� r� r� r% r� )r rC r r r r� � s
z-_UnixWritePipeTransport._call_connection_lost)NN)r� )N)r� r� r� r r� r� r� r� r� r� r� r� r� r� r% r r� r� r� r� r� r� r� r r )r! r rT � s$ % !
rT �set_inheritablec C sN t tdd�}tj| tj�}|s4tj| tj||B � ntj| tj|| @ � d S )NZ
FD_CLOEXECr )r� r� ZF_GETFDZF_SETFD)r� Z inheritableZ cloexec_flag�oldr r r �_set_inheritable� s
r� c @ s e Zd Zdd� ZdS )rW c K sv d }|t jkr*| jj� \}}t|j� d� t j|f||||d|d�|��| _|d k rr|j� t |j
� d|d�| j_ d S )NF)r[ r\ r] r^ Zuniversal_newlinesr_ �wb)� buffering) �
subprocess�PIPEr� r# r� r7 �Popen�_procr% r� �detachr\ ) r rB r[ r\ r] r^ r_ r` Zstdin_wr r r �_start� s
z _UnixSubprocessTransport._startN)r� r� r� r� r r r r rW � s rW c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )r aH Abstract base class for monitoring child processes.
Objects derived from this class monitor a collection of subprocesses and
report their termination or interruption by a signal.
New callbacks are registered with .add_child_handler(). Starting a new
process must be done within a 'with' block to allow the watcher to suspend
its activity until the new process if fully registered (this is needed to
prevent a race condition in some implementations).
Example:
with watcher:
proc = subprocess.Popen("sleep 1")
watcher.add_child_handler(proc.pid, callback)
Notes:
Implementations of this class must be thread-safe.
Since child watcher objects may catch the SIGCHLD signal and call
waitpid(-1), there should be only one active object per process.
c G s
t � �dS )a Register a new child handler.
Arrange for callback(pid, returncode, *args) to be called when
process 'pid' terminates. Specifying another callback for the same
process replaces the previous handler.
Note: callback() must be thread-safe.
N)�NotImplementedError)r re rA rB r r r rX � s z&AbstractChildWatcher.add_child_handlerc C s
t � �dS )z�Removes the handler for process 'pid'.
The function returns True if the handler was successfully removed,
False if there was nothing to remove.N)r� )r re r r r �remove_child_handler� s z)AbstractChildWatcher.remove_child_handlerc C s
t � �dS )z�Attach the watcher to an event loop.
If the watcher was previously attached to an event loop, then it is
first detached before attaching to the new loop.
Note: loop may be None.
N)r� )r r� r r r � attach_loop� s z AbstractChildWatcher.attach_loopc C s
t � �dS )zlClose the watcher.
This must be called to make sure that any underlying resource is freed.
N)r� )r r r r r% � s zAbstractChildWatcher.closec C s
t � �dS )zdEnter the watcher's context and allow starting new processes
This function must return selfN)r� )r r r r � __enter__ s z AbstractChildWatcher.__enter__c C s
t � �dS )zExit the watcher's contextN)r� )r �a�b�cr r r �__exit__ s z AbstractChildWatcher.__exit__N)
r� r� r� r� rX r� r� r% r� r� r r r r r � s
c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�BaseChildWatcherc C s d | _ i | _d S )N)r� �
_callbacks)r r r r r s zBaseChildWatcher.__init__c C s | j d � d S )N)r� )r r r r r% s zBaseChildWatcher.closec C s
t � �d S )N)r� )r � expected_pidr r r � _do_waitpid s z BaseChildWatcher._do_waitpidc C s
t � �d S )N)r� )r r r r �_do_waitpid_all s z BaseChildWatcher._do_waitpid_allc C s~ |d kst |tj�st�| jd k r<|d kr<| jr %dr2 )r{ r� r� r� r� r� r� r� rG r� r� r� r� r r� r� )r re r� rf rA rB r r r r� � s6
z FastChildWatcher._do_waitpid_all) r� r� r� r� r r% r� r� rX r� r� r� r r )r! r r � s c sH e Zd ZdZeZ� fdd�Zdd� Z� fdd�Zdd � Z d
d � Z
� Z S ) �_UnixDefaultEventLoopPolicyz:UNIX event loop policy with a watcher for child processes.c s t � j� d | _d S )N)r r �_watcher)r )r! r r r s
z$_UnixDefaultEventLoopPolicy.__init__c
C sH t j�8 | jd kr:t� | _ttj� tj�r:| jj| j j
� W d Q R X d S )N) r r� r r rK r� �current_thread� _MainThreadr� �_localr� )r r r r �
_init_watcher s
z)_UnixDefaultEventLoopPolicy._init_watcherc s6 t � j|� | jdk r2ttj� tj�r2| jj|� dS )z�Set the event loop.
As a side effect, if a child watcher was set before, then calling
.set_event_loop() from the main thread will call .attach_loop(loop) on
the child watcher.
N)r �set_event_loopr rK r� r r r� )r r� )r! r r r s
z*_UnixDefaultEventLoopPolicy.set_event_loopc C s | j dkr| j� | j S )zzGet the watcher for child processes.
If not yet set, a SafeChildWatcher object is automatically created.
N)r r )r r r r rV & s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s4 |dkst |t�st�| jdk r*| jj� || _dS )z$Set the watcher for child processes.N)rK r rj r r% )r ra r r r �set_child_watcher0 s
z-_UnixDefaultEventLoopPolicy.set_child_watcher) r� r� r� r� r Z
_loop_factoryr r r rV r r� r r )r! r r s
r )5r� r> r{ r5 r" ry r� r& r� r* � r r r r r r r r
r r r
�logr �__all__r� � ImportErrorr �fspathrx �AttributeErrorZBaseSelectorEventLoopr �hasattrr� r� Z
ReadTransportrN Z_FlowControlMixinZWriteTransportrT r� r� ZBaseSubprocessTransportrW r r� r r ZBaseDefaultEventLoopPolicyr r r r r r r � sn
O
F=On2