B
‡Ž€Æ›Õ0œã @ sš d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddl mZ ddl mZ ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dZe jdkrêedƒ‚dd„ ZG dd„ dej ƒZ G dd„ dej ƒZ G dd„ dej ej!ƒZ"G dd„ dej#ƒZ$G dd „ d ƒZ%G d d „ d e%ƒZ&G d d „ d e&ƒZ'G d!d"„ d"e&ƒZ(G d#d$„ d$ej)ƒZ*e Z+e*Z,dS )%z2Selector event loop for Unix with signal handling.é Né )Ú base_events)Úbase_subprocess)Ú constants)Ú
coroutines)Úevents)Úfutures)Úselector_events)Útasks)Ú
transports)Úlogger)ÚSelectorEventLoopÚAbstractChildWatcherÚSafeChildWatcherÚFastChildWatcherÚDefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s dS )zDummy signal handler.N© )ÚsignumÚframer r ú8/opt/alt/python37/lib64/python3.7/asyncio/unix_events.pyÚ_sighandler_noop' s r c sÊ e Zd ZdZd)‡ fdd„ Z‡ fdd„Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
d*dd„Z d+dd„Z d,dd„Z
dd„ Zd-dddddœdd„Zd.dd ddd d œd d „Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Z‡ ZS )/Ú_UnixSelectorEventLoopzdUnix event loop.
Adds signal handling and UNIX Domain Socket support to SelectorEventLoop.
Nc s t ƒ |¡ i | _d S )N)ÚsuperÚ__init__Ú_signal_handlers)ÚselfÚselector)Ú __class__r r r 2 s z _UnixSelectorEventLoop.__init__c s^ t ƒ ¡ t ¡ s2xFt| jƒD ]}| |¡ q W n(| jrZtjd| ›dt | d | j
¡ 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 6 s
z _UnixSelectorEventLoop.closec C s" x |D ]}|sq| |¡ qW d S )N)Ú_handle_signal)r Údatar r r r Ú_process_self_dataD s
z)_UnixSelectorEventLoop._process_self_datac
G sL t |¡st |¡r tdƒ‚| |¡ | ¡ yt | j ¡ ¡ W n2 t
t fk
rt } zt t
|ƒƒ‚W dd}~X Y nX t ||| d¡}|| j|< y t |t¡ t |d¡ W nš t k
rF } zz| j|= | jsyt d¡ W n4 t
t fk
r } zt d|¡ W dd}~X Y nX |jtjkr4t d|› dƒ‚n‚ W dd}~X Y 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()NFéÿÿÿÿz set_wakeup_fd(-1) failed: %szsig z 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)r r( ÚcallbackÚargsÚexcÚhandleZnexcr r r Úadd_signal_handlerK s0
z)_UnixSelectorEventLoop.add_signal_handlerc C s8 | j |¡}|dkrdS |jr*| |¡ n
| |¡ dS )z2Internal helper that is the actual signal handler.N)r ÚgetZ
_cancelledr# Z_add_callback_signalsafe)r r( r= r r r r) x s z%_UnixSelectorEventLoop._handle_signalc
C sä | |¡ y | j|= W n tk
r* dS X |tjkr>tj}ntj}yt ||¡ W nB tk
r– } z$|jtj kr„t
d|› dƒ‚n‚ W dd}~X Y nX | jsàyt d¡ W n2 t tfk
rÞ } zt
d|¡ W dd}~X Y nX dS )zwRemove a handler for a signal. UNIX only.
Return True if a signal handler was removed, False if not.
Fzsig z cannot be caughtNr, z set_wakeup_fd(-1) failed: %sT)r. r ÚKeyErrorr/ ÚSIGINTÚdefault_int_handlerÚSIG_DFLr3 r8 r9 r4 r0 r2 r r7 )r r( Zhandlerr< r r r r# ‚ s(
z,_UnixSelectorEventLoop.remove_signal_handlerc C sL t |tƒstd|›ƒ‚d| kr.tjk sHn td|› dtj› dƒ‚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.
zsig must be an int, not r zsig z out of range(1, ú)N)Ú
isinstanceÚintr- 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
Ë s” t ¡ ‚}
| ¡ } t| |||||||f| |dœ| —Ž} |
| ¡ | j| ¡ y| I d H W n, tk
r„ | ¡ | ¡ I d H ‚ Y nX W d Q R X | S )N)rK rL )
r Úget_child_watcherÚ
create_futureÚ_UnixSubprocessTransportÚadd_child_handlerZget_pidÚ_child_watcher_callbackÚ Exceptionr Z_wait)
r rJ r; ÚshellÚstdinÚstdoutÚstderrÚbufsizerL ÚkwargsÚwatcherrK Útranspr r r Ú_make_subprocess_transport¶ s
z1_UnixSelectorEventLoop._make_subprocess_transportc C s | |j|¡ d S )N)Zcall_soon_threadsafeZ_process_exited)r ÚpidÚ
returncoder] r r r rT Ë s z._UnixSelectorEventLoop._child_watcher_callback)ÚsslÚsockÚserver_hostnameÚssl_handshake_timeoutc à s |d kst |tƒst‚|r,|d krLtdƒ‚n |d k r s(
z,_UnixSelectorEventLoop._sock_sendfile_nativec C s | ¡ } |d k r| |¡ | ¡ r4| |||¡ d S |rd|| }|dkrd| |||¡ | |¡ d S yt | |||¡}
W n* ttfk
rÆ |d kr¢| ||¡ |
| | j || ||||||¡
Y nH t k
rj } z†|d k r| j
t
jkrt| ƒtk rtdt
jƒ} | | _| } |dkrBt d¡}
| |||¡ | |
¡ n| |||¡ | | ¡ W d d } ~ X Y n¤ tk
r¤ } z | |||¡ | | ¡ W d d } ~ X Y njX |
dkrÊ| |||¡ | |¡ nD||
7 }||
7 }|d krð| ||¡ |
| | j || ||||||¡
d S )Nr zsocket is not connectedzos.sendfile call failed)r1 Ú
remove_writerÚ cancelledÚ _sock_sendfile_update_fileposZ
set_resultrf r ÚBlockingIOErrorÚInterruptedErrorÚ _sock_add_cancellation_callbackZ
add_writerrˆ r3 r8 ZENOTCONNrm ÚConnectionErrorÚ __cause__r rƒ Z
set_exceptionrU )r r Z
registered_fdrb r1 rŠ r‹ rŒ Ú
total_sentÚfdZsentr< Únew_excr~ r r r rˆ U s`
z1_UnixSelectorEventLoop._sock_sendfile_native_implc C s |dkrt ||t j¡ d S )Nr )rf ÚlseekÚSEEK_SET)r r1 rŠ r— r r r r‘ ™ s z4_UnixSelectorEventLoop._sock_sendfile_update_fileposc s ‡ ‡fdd„}| |¡ d S )Nc s& | ¡ r"ˆ ¡ }|dkr"ˆ |¡ d S )Nr, )r r1 r )r r˜ )r rb r r Úcbž s zB_UnixSelectorEventLoop._sock_add_cancellation_callback..cb)Zadd_done_callback)r r rb rœ r )r rb r r” s z6_UnixSelectorEventLoop._sock_add_cancellation_callback)N)NN)NN)N)N)N)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r+ r> r) r# r. rM rO r^ rT rq r€ rŽ rˆ r‘ r” Ú
__classcell__r r )r r r , s4 -
*?Dr c s„ e Zd ZdZd ‡ fdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
d dd„Zdd„ Zd d „ Z‡ ZS ) rH i Nc sÚ t ƒ |¡ || jd< || _|| _| ¡ | _|| _d| _d| _ t
| j¡j }t
|¡s„t
|¡s„t
|¡s„d | _d | _d | _tdƒ‚t
| jd¡ | j | jj| ¡ | j | jj| j| j¡ |d k rÖ| j tj|d ¡ d S )NrI Fz)Pipe transport is for pipes/sockets only.)r r Ú_extraÚ_loopÚ_piper1 Ú_filenoÚ _protocolÚ_closingÚ_pausedrf r† rz rx ÚS_ISFIFOry ÚS_ISCHRr2 Ú set_blockingÚ call_soonÚconnection_madeÚ _add_readerÚ _read_readyr Ú _set_result_unless_cancelled)r rv rI rJ rK rL Úmode)r r r r ª s.
z _UnixReadPipeTransport.__init__c C sÀ | j jg}| jd kr | d¡ n| jr0| d¡ | d| j› ¡ t| jdd ƒ}| jd k r|d k rt || jt
j ¡}|r„| d¡ q°| d¡ n | jd k r¦| d¡ n
| d¡ d d
|¡¡S )
NÚclosedÚclosingzfd=Ú _selectorÚpollingÚidleÚopenz<{}>ú )r r r¤ Úappendr§ r¥ Úgetattrr£ r Ú_test_selector_eventÚ selectorsZ
EVENT_READÚformatÚjoin)r r7 r rµ r r r Ú__repr__È s"
z _UnixReadPipeTransport.__repr__c
C sº yt | j| j¡}W nD ttfk
r, Y nŠ tk
rX } z| |d¡ W d d }~X Y n^X |rl| j |¡ nJ| j
¡ r‚t
d| ¡ d| _| j
| j¡ | j
| jj¡ | j
| jd ¡ d S )Nz"Fatal read error on pipe transportz%r was closed by peerT)rf Úreadr¥ Úmax_sizer’ r“ r3 Ú _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 s | jrd S d| _| j | j¡ | j ¡ r:t d| ¡ d S )NTz%r pauses reading)r§ r¨ r£ rÄ r¥ rà r Údebug)r r r r Ú
pause_readingï s
z$_UnixReadPipeTransport.pause_readingc C sB | j s | jsd S d| _| j | j| j¡ | j ¡ r>t d| ¡ d S )NFz%r resumes reading) r§ r¨ r£ r® r¥ 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| d ¡ d S )N)r§ Ú_close)r r r r r s z _UnixReadPipeTransport.closec C s. | j d k r*tjd| ›t| d | j ¡ d S )Nzunclosed transport )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| j ¡ rLtjd| |dd n| j ||| | j dœ¡ |
|¡ d S )Nz%r: %sT)Úexc_info)ÚmessageÚ exceptionrp rJ ) rE r3 r8 ZEIOr£ rà r rÆ Úcall_exception_handlerr¦ rÌ )r r< rÐ r r r r s
z#_UnixReadPipeTransport._fatal_errorc C s( d| _ | j | j¡ | j | j|¡ d S )NT)r§ r£ rÄ r¥ r¬ rÅ )r r< r r r rÌ s z _UnixReadPipeTransport._closec C s4 z| j |¡ W d | j ¡ d | _d | _ d | _X d S )N)r¦ Úconnection_lostr¤ r r£ )r r< r r r rÅ % s
z,_UnixReadPipeTransport._call_connection_lost)NN)rÎ )r rž rŸ rÁ r r¿ r¯ rÇ rÈ rÉ rÊ rË r rÍ r rÌ rÅ r¡ r r )r r rH ¦ s
rH c s¢ e Zd Zd%‡ fdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Zdd„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zd d „ Zd&d d „Zd'd!d"„Zd#d$„ Z‡ ZS )(rN Nc
sþ t ƒ ||¡ || jd< || _| ¡ | _|| _tƒ | _d| _ d| _
t | j¡j
}t |¡}t |¡}t |¡} |s”|s”| s”d | _d | _d | _tdƒ‚t | jd¡ | j | jj| ¡ | sÈ|ràtj d¡sà| j | jj| j| j¡ |d k rú| j t j |d ¡ d S )NrI r Fz?Pipe transport is only for pipes, sockets and character devicesZaix) r r r¢ r¤ r1 r¥ r¦ Ú bytearrayÚ_bufferÚ
_conn_lostr§ rf r† rz rx rª r© ry r2 r« r£ r¬ r r ÚplatformÚ
startswithr® r¯ r r° )
r rv rI rJ rK rL r± Zis_charZis_fifoZ is_socket)r r r r 2 s2
z _UnixWritePipeTransport.__init__c C sØ | j jg}| jd kr | d¡ n| jr0| d¡ | d| j› ¡ t| jdd ƒ}| jd k r¨|d k r¨t || jt
j ¡}|r„| d¡ n
| d¡ | ¡ }| d|› ¡ n | jd k r¾| d¡ n
| d¡ d
d
|¡¡S ) Nr² r³ zfd=r´ rµ r¶ zbufsize=r· z<{}>r¸ )r r r¤ r¹ r§ r¥ rº r£ r r» r¼ Z EVENT_WRITEÚget_write_buffer_sizer½ r¾ )r r7 r rµ rZ r r r r¿ W s&
z _UnixWritePipeTransport.__repr__c C s
t | jƒS )N)ÚlenrÕ )r r r r rÙ o s z-_UnixWritePipeTransport.get_write_buffer_sizec C s6 | j ¡ rt d| ¡ | jr*| tƒ ¡ n| ¡ d S )Nz%r was closed by peer)r£ rà r r7 rÕ rÌ ÚBrokenPipeError)r r r r r¯ r s
z#_UnixWritePipeTransport._read_readyc
C s2 t |tttfƒs tt|ƒƒ‚t |tƒr.t|ƒ}|s6d S | jsB| jrj| jtj krXt
d¡ | jd7 _d S | j syt
| j|¡}W nV ttfk
r d}Y n< tk
rÚ } z | jd7 _| |d¡ d S d }~X Y nX |t|ƒkrìd S |dkrt|ƒ|d … }| j | j| j¡ | j |7 _ | ¡ d S )Nz=pipe closed by peer or os.write(pipe, data) raised exception.r r z#Fatal write error on pipe transport)rE ÚbytesrÔ Ú
memoryviewre ÚreprrÖ r§ r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr ÚwarningrÕ rf Úwriter¥ r’ r“ rU r rÚ r£ Z _add_writerÚ _write_readyZ_maybe_pause_protocol)r r* Únr< r r r rà { s4
z _UnixWritePipeTransport.writec
C sö | j stdƒ‚yt | j| j ¡}W nj ttfk
r: Y n¸ tk
rŒ } z6| j ¡ | j d7 _ | j
| j¡ | |d¡ W d d }~X Y nfX |t
| j ƒkrÞ| j ¡ | j
| j¡ | ¡ | jrÚ| j
| j¡ | d ¡ d S |dkrò| j d |…= d S )NzData should not be emptyr z#Fatal write error on pipe transportr )rÕ re rf rà r¥ r’ r“ rU r' rÖ r£ Ú_remove_writerr rÚ Z_maybe_resume_protocolr§ rÄ rÅ )r râ r< r r r rá œ s(
z$_UnixWritePipeTransport._write_readyc C s dS )NTr )r r r r Ú
can_write_eof¶ s z%_UnixWritePipeTransport.can_write_eofc C sB | j r
d S | jst‚d| _ | js>| j | j¡ | j | jd ¡ d S )NT) r§ r¤ re rÕ r£ rÄ r¥ r¬ rÅ )r r r r Ú write_eof¹ s
z!_UnixWritePipeTransport.write_eofc C s
|| _ d S )N)r¦ )r rJ r r r rÉ Â 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| js| ¡ d S )N)r¤ r§ rå )r r r r r Ë s z _UnixWritePipeTransport.closec C s. | j d k r*tjd| ›t| d | j ¡ d S )Nzunclosed transport )r )r¤ r$ r% r& r )r r r r rÍ Ð s
z _UnixWritePipeTransport.__del__c C s | d ¡ d S )N)rÌ )r r r r ÚabortÖ s z _UnixWritePipeTransport.abortú Fatal error on pipe transportc C sN t |tƒr(| j ¡ r@tjd| |dd n| j ||| | jdœ¡ | |¡ d S )Nz%r: %sT)rÏ )rÐ rÑ rp rJ ) rE r3 r£ rà r rÆ rÒ r¦ rÌ )r r< rÐ r r r r ٠s
z$_UnixWritePipeTransport._fatal_errorc C sF d| _ | jr| 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 s4 z| j |¡ W d | j ¡ d | _d | _ d | _X d S )N)r¦ rÓ r¤ r r£ )r r< 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 rN / s" % !
rN c @ s e Zd Zdd„ ZdS )rR c K sŠ d }|t jkrt ¡ \}}zPt j|f||||d|dœ|—Ž| _|d k rh| ¡ t| ¡ d|d| j_ d }W d |d k r„| ¡ | ¡ X d S )NF)rV rW rX rY Zuniversal_newlinesrZ Úwb)Ú buffering)
Ú
subprocessÚPIPErh Z
socketpairÚPopenÚ_procr r· ÚdetachrW ) r r; rV rW rX rY rZ r[ Zstdin_wr r r Ú_startû s
z _UnixSubprocessTransport._startN)r rž rŸ rï r r r r rR ù s rR 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 r_ r: r; r r r rS ) 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 r_ r r r Úremove_child_handler4 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 rv 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 F 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__M s z AbstractChildWatcher.__enter__c C s
t ƒ ‚dS )zExit the watcher's contextN)rð )r ÚaÚbÚcr r r Ú__exit__S s z AbstractChildWatcher.__exit__N)
r rž rŸ r rS rñ rò r ró r÷ r r r r r s
r 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 Z s zBaseChildWatcher.__init__c C s | 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_waitpida s z BaseChildWatcher._do_waitpidc C s
t ƒ ‚d S )N)rð )r r r r Ú_do_waitpid_alld s z BaseChildWatcher._do_waitpid_allc C s~ |d kst |tjƒst‚| jd k r<|d kr<| jr Ú _sig_chldrü )r rv r r r rò g s
z BaseChildWatcher.attach_loopc
C sF y | ¡ W n4 tk
r@ } z| j d|dœ¡ W d d }~X Y nX d S )Nz$Unknown exception in SIGCHLD handler)rÐ rÑ )rü rU r£ rÒ )r r< r r r rÿ { s zBaseChildWatcher._sig_chldc C s2 t |¡rt |¡ S t |¡r*t |¡S |S d S )N)rf Ú WIFSIGNALEDÚWTERMSIGÚ WIFEXITEDÚ WEXITSTATUS)r Ústatusr r r Ú_compute_returncode‡ s
z$BaseChildWatcher._compute_returncodeN)
r rž rŸ r r rû rü rò rÿ r r r r r rø X s rø c sP e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
‡ 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 ¡ tƒ ¡ d S )N)rù r' r r )r )r r r r s
zSafeChildWatcher.closec C s | S )Nr )r r r r ró ¤ s zSafeChildWatcher.__enter__c C s d S )Nr )r rô rõ rö r r r r÷ § s zSafeChildWatcher.__exit__c G s. | j d krtdƒ‚||f| 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 rS ª s
z"SafeChildWatcher.add_child_handlerc C s& y | j |= dS tk
r dS X d S )NTF)rù r@ )r r_ r r r rñ µ s
z%SafeChildWatcher.remove_child_handlerc C s" x t | jƒD ]}| |¡ q W d S )N)r" rù rû )r r_ r r r rü ¼ s z SafeChildWatcher._do_waitpid_allc C sÒ |dks t ‚yt |tj¡\}}W n( tk
rJ |}d}t d|¡ Y n0X |dkrXd S | |¡}| j ¡ rzt
d||¡ y| j |¡\}}W n. t
k
r¼ | j ¡ r¸tjd|dd Y nX |||f|žŽ d S )Nr éÿ z8Unknown child process pid %d, will report returncode 255z$process %s exited with returncode %sz'Child watcher got an unexpected pid: %rT)rÏ )re rf ÚwaitpidÚWNOHANGÚChildProcessErrorr rß r r£ rà rÆ rù Úpopr@ )r rú r_ r r` r: r; r r r rû Á s,
z SafeChildWatcher._do_waitpid) r rž rŸ r r ró r÷ rS rñ rü rû r¡ r r )r r r • s r c sT e Zd ZdZ‡ fdd„Z‡ fdd„Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
‡ 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 s$ t ƒ ¡ t ¡ | _i | _d| _d S )Nr )r r Ú threadingZLockÚ_lockÚ_zombiesÚ_forks)r )r r r r î s
zFastChildWatcher.__init__c s" | j ¡ | j ¡ tƒ ¡ d S )N)rù r' r
r r )r )r r r r ô s
zFastChildWatcher.closec C s$ | j | jd7 _| S Q R X d S )Nr )r r )r r r r ró ù s zFastChildWatcher.__enter__c C sT | j 8 | jd8 _| js"| js&d S t| jƒ}| j ¡ W d Q R X t 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 sz | j stdƒ‚| jd kr tdƒ‚| j: y| j |¡}W n" tk
rZ ||f| j|< d S X W d Q R X |||f|žŽ d S )Nz Must use the context managerzICannot add child handler, the child watcher does not have a loop attached) r re r£ r4 r r
r
r@ rù )r r_ r: r; r` r r r rS
s
z"FastChildWatcher.add_child_handlerc C s& y | j |= dS tk
r dS X d S )NTF)rù r@ )r r_ r r r rñ s
z%FastChildWatcher.remove_child_handlerc C sö xðyt dt j¡\}}W n tk
r, d S X |dkr:d S | |¡}| jv y| j |¡\}}W nB tk
r¢ | j rš|| j
|< | j ¡ r˜t
d||¡ wd }Y nX | j ¡ r¼t
d||¡ W d Q R X |d krÞt
d||¡ q|||f|žŽ qW d S )Nr, r z,unknown process %s exited with returncode %sz$process %s exited with returncode %sz8Caught subprocess termination from unknown pid: %d -> %d)rf 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ü ' s6
z FastChildWatcher._do_waitpid_all) r rž rŸ r r r ró r÷ rS rñ rü r¡ r r )r r r ä s r 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 ƒ ¡ d | _d S )N)r r Ú_watcher)r )r r r r V s
z$_UnixDefaultEventLoopPolicy.__init__c C sH t j8 | jd kr:tƒ | _tt ¡ tjƒr:| j | j j
¡ W d Q R X d S )N) r r r r rE r Úcurrent_threadÚ _MainThreadrò Ú_localr£ )r r r r Ú
_init_watcherZ s
z)_UnixDefaultEventLoopPolicy._init_watcherc s6 t ƒ |¡ | jdk r2tt ¡ tjƒr2| j |¡ 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 rE r r r rò )r rv )r r r r b s
z*_UnixDefaultEventLoopPolicy.set_event_loopc C s | j dkr| ¡ | 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 p s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s4 |dkst |tƒst‚| jdk r*| j ¡ || _dS )z$Set the watcher for child processes.N)rE r re r r )r r\ r r r Úset_child_watcherz s
z-_UnixDefaultEventLoopPolicy.set_child_watcher) r rž rŸ r r Z
_loop_factoryr r r rP r r¡ r r )r r r R s
r )-r r8 r„ rf r¼ r/ rh rx rê r r r$ Ú r r r r r r r r
r Úlogr Ú__all__r× Ú ImportErrorr ZBaseSelectorEventLoopr Z
ReadTransportrH Z_FlowControlMixinZWriteTransportrN ZBaseSubprocessTransportrR r rø r r ZBaseDefaultEventLoopPolicyr r
r r r r r Ú sR
|
JF=On3