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 dddddgZejdkreddd 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%rej)Z*nddl#Z#d&d' Z*G d(d) d)e j+Z,G d*d dZ-G d+d, d,e-Z.G d-d de.Z/G d.d de.Z0G d/d0 d0ej1Z2e Z3e2Z4dS )1z2Selector event loop for Unix with signal handling. N ) base_events)base_subprocess)compat) constants)
coroutines)events)futures)selector_events) selectors)
transports) coroutine)loggerSelectorEventLoopAbstractChildWatcherSafeChildWatcherFastChildWatcherDefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s dS )zDummy signal handler.N )signumframer 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 fddZd d
Zd d Zd
d Z dd Z
dd Z d#ddZ d$ddZ
ed%ddZdd Zeddddd 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)selfselector) __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| jD ]}| j| q W n(| jrZtjd| dt | d | jj
d S )NzClosing the loop z@ on interpreter shutdown stage, skipping signal handlers removal)source) r closesys
is_finalizinglistr remove_signal_handlerwarningswarnResourceWarningclear)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|= | jsytjd W n4 t
t fk
r } ztjd| W Y dd}~X nX |jtjkr0t dj|n W Y dd}~X nX dS )zAdd 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_closedsignal
set_wakeup_fdZ_csockfileno
ValueErrorOSError RuntimeErrorstrr ZHandler r siginterruptr infoerrnoEINVALformat)r r. callbackargsexchandleZnexcr 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 krt
dj |n W Y dd}~X nX | jsytj 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 SIGINTdefault_int_handlerSIG_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 |tstdj|d| ko,tjk n sDtdj|tjdS )zInternal 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)
isinstanceintr3 r@ r5 NSIGr8 )r r. r r r r4 s
z$_UnixSelectorEventLoop._check_signalc C s t | ||||S )N)_UnixReadPipeTransport)r pipeprotocolwaiterextrar 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_UnixSubprocessTransportadd_child_handlerZget_pid_child_watcher_callback Exceptionr% Z_wait)r rP rB shellstdinstdoutstderrbufsizerR kwargswatcherrQ 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)sslsockserver_hostnamec c s |d kst |tst|r,|d kr|d krBtd|jtjks`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 statS_ISSOCKosst_moderemoveFileNotFoundErrorr9 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!ddZdd Zd d Z ZS )"rN i Nc s t j| || jd< || _|| _|j | _|| _d| _t j
| jj }t j
|pbt j|pbt j|s~d | _d | _d | _tdt| 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_soonconnection_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 rtj || jt
j }|r|jd q|jd n | jd k r|jd n
|jd dd j | S )
Nclosedclosingzfd=%s _selectorpollingidleopenz<%s> )
r! r r appendr r getattrr r
_test_selector_eventr Z
EVENT_READjoin)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 rt 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_sizeBlockingIOErrorInterruptedErrorr9 _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 |tr4|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 debugcall_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 | jj
}tj|}tj|}tj|} |px|px| sd | _d | _d | _tdt| j | jj| jj| | s|rtjjd 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 devicesaix) 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 rtj || 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_WRITEget_write_buffer_sizer )r r= r r r_ r r r r s(
z _UnixWritePipeTransport.__repr__c C s
t | jS )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 |tttfs tt|t |tr.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|krd S |dkrt||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 stdytj| 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 | jstd| _ | 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 |tjr*| 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_flagoldr 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)
subprocessPIPEr 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 )zRemoves 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 )zAttach 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 abcr 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 |tjst| 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 fddZdd Z fddZdd 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 j8 | jd kr:t | _ttj tjr:| 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 tjr2| jj| dS )zSet 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 |tst| 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