shell bypass 403
��Yf�� �
@ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z d d l
m Z d d l
m
Z
d d l
m Z d d l
m Z d d l
m Z d d
l
m Z d d l
m Z d d l
m Z d d
l
m Z d d l m Z d d l m Z d d d d d g Z e j d k rbe d � � d d � Z y
e j Z Wn e k
r�d d � Z Yn XGd d � d e j � Z e! e d � r�d d � Z" n d d l# Z# d d � Z" Gd! d" � d" e j$ � Z% Gd# d$ � d$ e j& e j' � Z( e! e d% � r>e j) Z* n d d l# Z# d& d' � Z* Gd( d) � d) e j+ � Z, Gd* d � d � Z- Gd+ d, � d, e- � Z. Gd- d � d e. � Z/ Gd. d � d e. � Z0 Gd/ d0 � d0 e j1 � Z2 e Z3 e2 Z4 d S)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�DefaultEventLoopPolicy�win32z+Signals are not really supported on Windowsc C s d S)zDummy signal handler.N� )�signum�framer r �8/opt/alt/python35/lib64/python3.5/asyncio/unix_events.py�_sighandler_noop% s r c C s | S)Nr )�pathr r r �<lambda>. s r c
s e Z d Z d Z d � f d d � Z d d � Z � f d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d d d � Z d d d d � Z
e d d d � � Z d d � Z e d d d d d d d d � � Z e d d d d d! d d d" d# �� Z � S)$�_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 s8 t � j � x$ t | j � D] } | j | � q Wd S)N)r �close�listr �remove_signal_handler)r �sig)r" r r r% > s
z _UnixSelectorEventLoop.closec C s+ x$ | D] } | s q | j | � q Wd S)N)�_handle_signal)r �datar r r r �_process_self_dataC s
z)_UnixSelectorEventLoop._process_self_datac + G s� t j | � s t j | � r* t d � � | j | � | j � y t j | j j � � Wn= t
t f k
r� } z t t
| � � � WYd d } ~ Xn Xt j | | | � } | | j | <y$ t j | t � t j | d � Wn� t k
r�} z� | j | =| j sXy t j d � Wn; t
t f k
rW} z t j d | � WYd d } ~ Xn X| j t j k r�t d j | � � � n � WYd d } ~ Xn Xd S)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_handlerJ s0
%
#z)_UnixSelectorEventLoop.add_signal_handlerc C sL | j j | � } | d k r" d S| j r; | j | � n
| j | � d S)z2Internal helper that is the actual signal handler.N)r �getZ
_cancelledr' Z_add_callback_signalsafe)r r( r> r r r r) w s z%_UnixSelectorEventLoop._handle_signalc & C s | j | � y | j | =Wn t k
r3 d SYn X| t j k rO t j } n t j } y t j | | � WnR t k
r� } z2 | j t j k r� t
d j | � � � n � WYd d } ~ Xn X| j sy t j d � Wn; t
t f k
r} z t j d | � WYd d } ~ Xn Xd S)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: %sTr, )r. r �KeyErrorr/ �SIGINT�default_int_handler�SIG_DFLr3 r8 r9 r4 r: r0 r2 r r7 )r r( Zhandlerr= r r r r' � s(
#z,_UnixSelectorEventLoop.remove_signal_handlerc C sb t | t � s$ t d j | � � � d | k o> t j k n s^ t d j | t j � � � d S)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�intr- r: r/ �NSIGr2 )r r( r r r r. � 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 rI rJ rK rL r r r �_make_write_pipe_transport� s z1_UnixSelectorEventLoop._make_write_pipe_transportc k s� t j � �� }
| j � } t | | | | | | | | d | d | | �} |
j | j � | j | � y
| Ed HWn+ t k
r� }
z |
} WYd d }
~
Xn Xd } | d k r� | j � | j � Ed H| � Wd QRX| S)NrK rL )
r �get_child_watcherZ
create_future�_UnixSubprocessTransport�add_child_handlerZget_pid�_child_watcher_callback� Exceptionr% Z_wait)r rJ r< �shell�stdin�stdout�stderr�bufsizerL �kwargs�watcherrK �transpr= �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�
returncoder\ r r r rS � s z._UnixSelectorEventLoop._child_watcher_callback�ssl�sock�server_hostnamec c sB | r! | d k r9 t d � � n | d k r9 t d � � | d k r� | d k r] t d � � t j t j t j d � } y& | j d � | j | | � Ed HWq| j � � YqXn\ | d k r� t d � � | j t j k s� t j | � rt d j
| � � � | j d � | j | | | | � Ed H\ } } | | f S)Nz/you have to pass server_hostname when using sslz+server_hostname is only meaningful with sslz3path and sock can not be specified at the same timer Fz no path and sock were specifiedz2A UNIX Domain Stream Socket was expected, got {!r}) r2 r# �AF_UNIX� SOCK_STREAM� setblockingZ sock_connectr% �familyr �_is_stream_socketr: Z _create_connection_transport)r �protocol_factoryr ra rb rc � transportrJ r r r �create_unix_connection� s8
z-_UnixSelectorEventLoop.create_unix_connection�backlog�d c
! C s t | t � r t d � � | d k r�| d k r? t d � � t | � } t j t j t j � } | d d
k r� y, t j t
j | � j � r� t
j | � WnI t
k
r� Yn8 t k
r� } z t j d | | � WYd d } ~ Xn Xy | j | � Wq�t k
ri} zK | j � | j t j k rTd j | � } t t j | � d � n � WYd d } ~ Xq�| j � � Yq�XnO | d k r�t d � � | j t j k s�t j | � r�t d j | � � � t j | | 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 rn ) rE �boolr- r2 �_fspathr# rd re �stat�S_ISSOCK�os�st_mode�remove�FileNotFoundErrorr3 r �errorZbindr% r8 Z
EADDRINUSEr: rg r rh ZServerZlistenrf Z_start_serving)
r ri r rb rl ra r] r= �msgZserverr r r �create_unix_server� sP
&
z)_UnixSelectorEventLoop.create_unix_server)�__name__�
__module__� __qualname__�__doc__r r$ r% r+ r? r) r' r. rM rO r
r^ rS rk ry r r )r" r r 1 s, -
%r � set_blockingc C s t j | d � d S)NF)rs r~ )�fdr r r �_set_nonblocking9 s r� c C s< t j | t j � } | t j B} t j | t j | � d S)N)�fcntlZF_GETFLrs �
O_NONBLOCKZF_SETFL)r �flagsr r r r� > s
c s� e Z d Z d Z d d � f d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z e
j r� d d � Z d d d � Z d d � Z d d � Z � S) rH � i Nc s* t � j | � | | j d <| | _ | | _ | j � | _ | | _ d | _ t j
| j � j } t j
| � p� t j | � p� t j | � s� d | _ d | _ d | _ t d � � t | j � | j j | j j | � | j j | j j | j | j � | d k r&| j j t j | d � d S)NrI Fz)Pipe transport is for pipes/sockets only.)r r �_extra�_loop�_piper1 �_fileno� _protocol�_closingrs �fstatrt rq �S_ISFIFOrr �S_ISCHRr2 r� � call_soon�connection_made� _add_reader� _read_readyr � _set_result_unless_cancelled)r �looprI rJ rK rL �mode)r" r r r H s,
z _UnixReadPipeTransport.__init__c C s | j j g } | j d k r. | j d � n | j rD | j d � | j d | j � t | j d d � } | j d k r� | d k r� t j | | j t
j � } | r� | j d � q� | j d � n, | j d k r� | j d � n
| j d � d d j | � S)
N�closed�closingzfd=%s� _selector�polling�idle�openz<%s>� )
r" rz r� �appendr� r� �getattrr� r
�_test_selector_eventr Z
EVENT_READ�join)r r7 r! r� r r r �__repr__e s$
z _UnixReadPipeTransport.__repr__c C s� y t j | j | j � } WnL t t f k
r6 Yn� t k
rj } z | j | d � WYd d } ~ Xn� X| r� | j j | � ng | j
j � r� t j
d | � d | _ | j
j | j � | j
j | j j � | j
j | j d � d S)Nz"Fatal read error on pipe transportz%r was closed by peerT)rs �readr� �max_size�BlockingIOError�InterruptedErrorr3 � _fatal_errorr� Z
data_receivedr� � get_debugr r7 r� �_remove_readerr� Z eof_received�_call_connection_lost)r r* r= 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 rJ 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 | j d � d S)N)r� �_close)r r r r r% � s z _UnixReadPipeTransport.closec C s4 | j d k r0 t j d | t � | j j � d S)Nzunclosed transport %r)r� �warnings�warn�ResourceWarningr% )r r r r �__del__� s z _UnixReadPipeTransport.__del__z Fatal error on pipe transportc C s� t | t � rL | j t j k rL | j j � rw t j d | | d d �n+ | j j d | d | d | d | j i � | j
| � d S)Nz%r: %s�exc_infoT�message� exceptionrj rJ ) rE r3 r8 ZEIOr� r� r �debug�call_exception_handlerr� r� )r r= r� r r r r� � s ! z#_UnixReadPipeTransport._fatal_errorc C s6 d | _ | j j | j � | j j | j | � d S)NT)r� r� r� r� r� r� )r r= r r r r� � s z _UnixReadPipeTransport._closec C sD z | j j | � Wd | j j � d | _ d | _ d | _ Xd S)N)r� �connection_lostr� r% r� )r r= r r r r� � s
z,_UnixReadPipeTransport._call_connection_losti )rz r{ r| r� r r� r� r� r� r� r� r� r% r �PY34r� r� r� r� r r )r" r rH D s rH c s� e Z d Z d d � f d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z e j r� d d � Z d d � Z d d d � Z d d! d" � Z d# d$ � Z � S)%rN Nc
sj t � j | | � | | j d <| | _ | j � | _ | | _ t � | _ d | _ d | _
t j | j � j
} t j | � } t j | � } t j | � } | p� | p� | s� d | _ d | _ d | _ t d � � t | j � | j j | j j | � | s | rAt j j d � rA| j j | j j | j | j � | d k rf| j j t j | d � d S)NrI r Fz?Pipe transport is only for pipes, sockets and character devices�aix) r r r� r� r1 r� r� � bytearray�_buffer�
_conn_lostr� rs r� rt rq r� r� rr r2 r� r� r� r� �sys�platform�
startswithr� r� r r� )
r r� rI rJ rK rL r� Zis_charZis_fifoZ is_socket)r" r r r � s2
z _UnixWritePipeTransport.__init__c C s# | j j g } | j d k r. | j d � n | j rD | j d � | j d | j � t | j d d � } | j d k r� | d k r� t j | | j t
j � } | r� | j d � n
| j d � | j � } | j d | � n, | j d k r| j d � n
| j d � d d
j
| � S) Nr� r� zfd=%sr� r� r� z
bufsize=%sr� z<%s>r� )r" rz r� r� r� r� r� r� r
r� r Z EVENT_WRITE�get_write_buffer_sizer� )r r7 r! r� rY 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 sI | j j � r t j d | � | j r; | j t � � n
| j � d S)Nz%r was closed by peer)r� r� r r7 r� r� �BrokenPipeError)r r r r r�
s
z#_UnixWritePipeTransport._read_readyc C sa t | t � r t | � } | s% d S| j s7 | j ri | j t j k rV t j d � | j d 7_ d S| j sDy t
j | j | � } Wne t
t f k
r� d } YnH t k
r� } z( | j d 7_ | j | d � d SWYd d } ~ Xn X| t | � k r d S| d k r+t | � | d � } | j j | 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)rE r� �
memoryviewr� r� r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr �warningr� rs �writer� r� r� rT r� r� r� Z _add_writer� _write_readyZ_maybe_pause_protocol)r r* �nr= r r r r� s2
z _UnixWritePipeTransport.writec C s& y t j | j | j � } Wn{ t t f k
r6 Yn� t k
r� } zD | j j � | j d 7_ | j j
| j � | j | d � WYd d } ~ Xn� X| t | j � k r| j j � | j j
| j � | j
� | j r| j j | j � | j d � d S| d k r"| j d | � =d S)Nr z#Fatal write error on pipe transportr )rs r� r� r� r� r� rT �clearr� r� �_remove_writerr� r� Z_maybe_resume_protocolr� r� r� )r r� r= r r r r� 4 s&
#
z$_UnixWritePipeTransport._write_readyc C s d S)NTr )r r r r �
can_write_eofN s z%_UnixWritePipeTransport.can_write_eofc C sL | j r
d Sd | _ | j sH | j j | j � | j j | j d � d S)NT)r� r� r� r� r� r� r� )r r r r � write_eofQ s z!_UnixWritePipeTransport.write_eofc C s
| | _ d S)N)r� )r rJ r r r r� Z s z$_UnixWritePipeTransport.set_protocolc C s | j S)N)r� )r r r r r� ] s z$_UnixWritePipeTransport.get_protocolc C s | j S)N)r� )r r r r r� ` 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% c s z _UnixWritePipeTransport.closec C s4 | j d k r0 t j d | t � | j j � d S)Nzunclosed transport %r)r� r� r� r� r% )r r r r r� l s z _UnixWritePipeTransport.__del__c C s | j d � d S)N)r� )r r r r �abortq s z _UnixWritePipeTransport.abortz Fatal error on pipe transportc C sy t | t j � r= | j j � rh t j d | | d d �n+ | j j d | d | d | d | j i � | j | � d S)Nz%r: %sr� Tr� r� rj rJ )
rE r Z_FATAL_ERROR_IGNOREr� r� r r� r� r� r� )r r= r� r r r r� t s z$_UnixWritePipeTransport._fatal_errorc C s_ d | _ | j r% | j j | j � | j j � | j j | j � | j j | j | � d S)NT) r� r� r� r� r� r� r� r� r� )r r= r r r r� � s
z _UnixWritePipeTransport._closec C sD z | j j | � Wd | j j � d | _ d | _ d | _ Xd S)N)r� r� r� r% r� )r r= r r r r� � s
z-_UnixWritePipeTransport._call_connection_lost)rz 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 rN � s$ % ! rN �set_inheritablec C si t t d d � } t j | t j � } | sJ t j | t j | | B� n t j | t j | | @� 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 Z d Z d d � Z d S)rQ c K s� d } | t j k r= | j j � \ } } t | j � d � t j | d | d | d | d | d d d | | �| _ | d k r� | j � t | j
� d d | �| j _ d S)
NFrU rV rW rX Zuniversal_newlinesrY �wb� buffering) �
subprocess�PIPEr� r$ r� r1 �Popen�_procr% r� �detachrV ) r r< rU rV rW rX rY rZ Zstdin_wr r r �_start� s
z _UnixSubprocessTransport._startN)rz r{ r| r� r r r r rQ � s rQ c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)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 � � d S)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 r_ r; r< r r r rR � s z&AbstractChildWatcher.add_child_handlerc C s
t � � d S)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 r_ r r r �remove_child_handler� s z)AbstractChildWatcher.remove_child_handlerc C s
t � � d S)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 � � d S)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 � � d S)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 � � d S)zExit the watcher's contextN)r� )r �a�b�cr r r �__exit__� s z AbstractChildWatcher.__exit__N)
rz r{ r| r} rR r� r� r% r� r� r r r r r � s
c @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)�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� | j d k r4 | d k r4 | j r4 t j d t � | j d k rV | j j t j � | | _ | d k r� | j t j | j � | j
� d S)NzCA loop is being detached from a child watcher with pending handlers) r� r� r� r� �RuntimeWarningr' r/ �SIGCHLDr? � _sig_chldr� )r r� r r r r� s $ z BaseChildWatcher.attach_loopc C sV y | j � WnA t k
rQ } z! | j j d d d | i � WYd d } ~ Xn Xd S)Nr� z$Unknown exception in SIGCHLD handlerr� )r� rT r� r� )r r= r r r r� & s zBaseChildWatcher._sig_chldc C sA t j | � r t j | � St j | � r9 t j | � S| Sd S)N)rs � WIFSIGNALED�WTERMSIG� WIFEXITED� WEXITSTATUS)r �statusr r r �_compute_returncode2 s
z$BaseChildWatcher._compute_returncodeN)
rz r{ r| r r% r� r� r� r� r� r r r r r� s r� c sp e Z d Z d Z � f d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
� S)r ad 'Safe' child watcher implementation.
This implementation avoids disrupting other code spawning processes by
polling explicitly each process in the SIGCHLD handler instead of calling
os.waitpid(-1).
This is a safe solution but it has a significant overhead when handling a
big number of children (O(n) each time SIGCHLD is raised)
c s | j j � t � j � d S)N)r� r� r r% )r )r" r r r% K s
zSafeChildWatcher.closec C s | S)Nr )r r r r r� O s zSafeChildWatcher.__enter__c C s d S)Nr )r r� r� r� r r r r� R s zSafeChildWatcher.__exit__c G s? | j d k r t d � � | | f | j | <| j | � d S)NzICannot add child handler, the child watcher does not have a loop attached)r� r4 r� r� )r r_ r; r< r r r rR U s
z"SafeChildWatcher.add_child_handlerc C s/ y | j | =d SWn t k
r* d SYn Xd S)NTF)r� rA )r r_ r r r r� ` s
z%SafeChildWatcher.remove_child_handlerc C s+ x$ t | j � D] } | j | � q Wd S)N)r& r� r� )r r_ r r r r� g s z SafeChildWatcher._do_waitpid_allc C s� y t j | t j � \ } } Wn. t k
rO | } d } t j d | � YnB X| d k r` d S| j | � } | j j � r� t j d | | � y | j
j | � \ } } Wn7 t k
r� | j j � r� t j d | d d �Yn X| | | | � d S)N� z8Unknown child process pid %d, will report returncode 255r z$process %s exited with returncode %sz'Child watcher got an unexpected pid: %rr� T)
rs �waitpid�WNOHANG�ChildProcessErrorr r� r� r� r� r� r� �poprA )r r� r_ r� r` r; r< r r r r� l s*
z SafeChildWatcher._do_waitpid) rz r{ r| r} r% r� r� rR r� r� r� r r )r" r r @ s c sv e Z d Z d Z � f d d � Z � f d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
� S)r aW 'Fast' child watcher implementation.
This implementation reaps every terminated processes by calling
os.waitpid(-1) directly, possibly breaking other code spawning processes
and waiting for their termination.
There is no noticeable overhead when handling a big number of children
(O(1) each time a child terminates).
c s2 t � j � t j � | _ i | _ d | _ d S)Nr )r r � threadingZLock�_lock�_zombies�_forks)r )r" r r r � s
zFastChildWatcher.__init__c s+ | j j � | j j � t � j � d S)N)r� r� r� r r% )r )r" r r r% � s
zFastChildWatcher.closec
C s( | j � | j d 7_ | SWd QRXd S)Nr )r� r� )r r r r r� � s
zFastChildWatcher.__enter__c
C sg | j �G | j d 8_ | j s, | j r0 d St | j � } | j j � Wd QRXt j d | � d S)Nr z5Caught subprocesses termination from unknown pids: %s)r� r� r� r5 r� r r� )r r� r� r� Zcollateral_victimsr r r r� � s
zFastChildWatcher.__exit__c G s� | j d k r t d � � | j �G y | j j | � } Wn) t k
rf | | f | j | <d SYn XWd QRX| | | | � d S)NzICannot add child handler, the child watcher does not have a loop attached)r� r4 r� r� r� rA r� )r r_ r; r< r` r r r rR � s
z"FastChildWatcher.add_child_handlerc C s/ y | j | =d SWn t k
r* d SYn Xd S)NTF)r� rA )r r_ r r r r� � s
z%FastChildWatcher.remove_child_handlerc C s9 x2y t j d t j � \ } } Wn t k
r: d SYn X| d k rK d S| j | � } | j �� y | j j | � \ } } WnS t k
r� | j r� | | j
| <| j j � r� t
j d | | � w d } Yn# X| j j � r� t
j d | | � Wd QRX| d k r!t
j d | | � q | | | | � q Wd S)Nr r z,unknown process %s exited with returncode %sz$process %s exited with returncode %sz8Caught subprocess termination from unknown pid: %d -> %dr, )rs r� r� r� r� r� r� r� rA r� r� r� r� r r� r� )r r_ r� r` r; r< r r r r� � s6
z FastChildWatcher._do_waitpid_all) rz r{ r| r} r r% r� r� rR r� r� r r )r" r r � s c sd e Z d Z d Z e Z � f d d � Z d d � Z � f d d � Z d d � Z d
d � 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 s^ t j �N | j d k rS t � | _ t t j � t j � rS | j j | j j
� Wd QRXd S)N) r r� r� r rE r� �current_thread� _MainThreadr� �_localr� )r r r r �
_init_watcher s
z)_UnixDefaultEventLoopPolicy._init_watcherc sK t � j | � | j d k rG t t j � t j � rG | j j | � d S)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� rE r� r� r� r� )r r� )r" r r r�
s z*_UnixDefaultEventLoopPolicy.set_event_loopc C s | j d k r | 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 rP s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s) | j d k r | j j � | | _ d S)z$Set the watcher for child processes.N)r� r% )r r[ r r r �set_child_watcher% s
z-_UnixDefaultEventLoopPolicy.set_child_watcher) rz r{ r| r} r Z
_loop_factoryr r� r� rP r� r r )r" r r� � s
r� )5r} r8 rs r/ r# rq r� r� r� r� � r r r r r r r r
r r r
�logr �__all__r� � ImportErrorr Zfspathrp �AttributeErrorZBaseSelectorEventLoopr �hasattrr� r� Z
ReadTransportrH Z_FlowControlMixinZWriteTransportrN r� r� ZBaseSubprocessTransportrQ r r� r r ZBaseDefaultEventLoopPolicyr� r r r r r r �<module> sh
� � �
F=On2