j f؆ @ 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 ree d n d d Z Gd d d e j Z e e d rd 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# re 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+ Z- Gd- d. d. e j. Z/ e Z0 e/ Z1 d S)/z2Selector event loop for Unix with signal handling. N ) base_events)base_subprocess)compat) constants)
coroutines)events)futures)selector_events) selectors)
transports) coroutine)loggerSelectorEventLoopAbstractChildWatcherSafeChildWatcherFastChildWatcherDefaultEventLoopPolicywin32z+Signals are not really supported on Windowsc C s d S)zDummy signal handler.N )signumZframer r 8/opt/alt/python34/lib64/python3.4/asyncio/unix_events.py_sighandler_noop% 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)selfZselector) __class__r r r 0 s z _UnixSelectorEventLoop.__init__c C s
t j S)N)socketZ
socketpair)r r r r _socketpair4 s z"_UnixSelectorEventLoop._socketpairc s8 t j x$ t | j D] } | j | q Wd S)N)r closelistr remove_signal_handler)r sig)r r r r! 7 s
z _UnixSelectorEventLoop.closec C s. x' | D] } | s q n | j | q Wd S)N)_handle_signal)r datar r r r _process_self_data< s
z)_UnixSelectorEventLoop._process_self_datac + G s t j | s t j | r- t d n | j | | j y t j | j j Wn= t
t f k
r } z t t
| WYd d } ~ Xn Xt j | | | } | | j | )
r rm r{ appendr r| r
_test_selector_eventrz _selectorr Z
EVENT_READjoin)r r3 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 | nj | j
j r t j
d | n 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)rr readr| max_sizeBlockingIOErrorInterruptedErrorr/ _fatal_errorr Z
data_receivedrz get_debugr r3 r
remove_readerr Z eof_received_call_connection_lost)r r& r9 r r r r X s # z"_UnixReadPipeTransport._read_readyc C s | j j | j d S)N)rz r r| )r r r r
pause_readingj s z$_UnixReadPipeTransport.pause_readingc C s | j j | j | j d S)N)rz r r| r )r r r r resume_readingm s z%_UnixReadPipeTransport.resume_readingc C s | j S)N)r )r r r r
is_closingp s z!_UnixReadPipeTransport.is_closingc C s | j s | j d n d S)N)r _close)r r r r r! s s z _UnixReadPipeTransport.closec C s7 | j d k r3 t j d | t | j j n d S)Nzunclosed transport %r)r{ warningswarnResourceWarningr! )r r r r __del__{ s z _UnixReadPipeTransport.__del__z Fatal error on pipe transportc C s t | t rO | j t j k rO | j j r~ t j d | | d d q~ n/ | j j i | d 6| d 6| d 6| j d 6 | j
| d S)Nz%r: %sexc_infoTmessage exceptionrf rF ) rA r/ r4 ZEIOrz r r debugcall_exception_handlerr r )r r9 r r r r r s ! z#_UnixReadPipeTransport._fatal_errorc C s6 d | _ | j j | j | j j | j | d S)NT)r rz r r| r r )r r9 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! rz )r r9 r r r r s
z,_UnixReadPipeTransport._call_connection_losti )rm rn ro r r r r r r r r! r PY34r r r r r r )r r rD * s rD 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 e
j r d d Z n d d Z d d d Z d d d Z d d Z S)!rJ Nc sC t j | | | | j d <| | _ | j | _ t j | j j } t j
| } | p} t j | p} t j | s t
d n t | j | | _ g | _ d | _ d | _ | j j | j j | | s t j j d r| j j | j j | j | j n | d k r?| j j t j | d n d S)NrE z?Pipe transport is only for pipes, sockets and character devicesr Faix) r r ry r{ r- r| rr r} r~ r r r r r. rt r _buffer
_conn_lostr rz r r sysplatform
startswithr r r r )r rL rE rF rG rH r Z is_socket)r r r r s,
z _UnixWritePipeTransport.__init__c C s | j j g } | j d k r. | j d n | j rG | j d n | j d | j | j d k r t j | j j | j t
j } | r | j d n
| j d | j } | j d | n
| j d d d j
| S) Nr r zfd=%sr r z
bufsize=%sz<%s>r )r rm r{ r r r| r
r rz r r Z EVENT_WRITEget_write_buffer_sizer )r r3 r rV r r r r s"
z _UnixWritePipeTransport.__repr__c C s t d d | j D S)Nc s s | ] } t | Vq d S)N)len).0r& r r r s z@_UnixWritePipeTransport.get_write_buffer_size..)sumr )r r r r r s z-_UnixWritePipeTransport.get_write_buffer_sizec C sL | j j r" t j d | n | j r> | j t n
| j d S)Nz%r was closed by peer)rz r r r3 r r BrokenPipeError)r r r r r s
z#_UnixWritePipeTransport._read_readyc C sh t | t r t | } n | s( d S| j s: | j ro | j t j k r\ t j d n | j d 7_ d S| j sJy 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 rd S| d k r.| | d } n | j j | j | j n | j j | | j d S)Nz=pipe closed by peer or os.write(pipe, data) raised exception.r r z#Fatal write error on pipe transport)rA bytearray
memoryviewr r r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr warningr rr writer| r r rQ r r rz Z
add_writer _write_readyr Z_maybe_pause_protocol)r r& nr9 r r r r s2
z _UnixWritePipeTransport.writec C sU d j | j } | j j y t j | j | } Wn~ t t f k
rb | j j | Yn t k
r } z7 | j
d 7_
| j j | j | j
| d WYd d } ~ Xn X| t | k r"| j j | j | j | j r | j r | j j | j | j d n d S| d k rA| | d } n | j j | d S)N r z#Fatal write error on pipe transportr )r r clearrr r r| r r r rQ r rz
remove_writerr r Z_maybe_resume_protocolr r r )r r& r r9 r r r r s(
#
z$_UnixWritePipeTransport._write_readyc C s d S)NTr )r r r r
can_write_eof s z%_UnixWritePipeTransport.can_write_eofc C sO | j r
d Sd | _ | j sK | j j | j | j j | j d n d S)NT)r r rz r r| r r )r r r r write_eof# s z!_UnixWritePipeTransport.write_eofc 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 n d S)N)r{ r r )r r r r r! / s z _UnixWritePipeTransport.closec C s7 | j d k r3 t j d | t | j j n d S)Nzunclosed transport %r)r{ r r r r! )r r r r r 8 s z _UnixWritePipeTransport.__del__c C s | j d d S)N)r )r r r r abort= s z _UnixWritePipeTransport.abortz Fatal error on pipe transportc C s t | t t f rC | j j rr t j d | | d d qr n/ | j j i | d 6| d 6| d 6| j d 6 | j | d S)Nz%r: %sr Tr r rf rF )
rA r ConnectionResetErrorrz r r r r r r )r r9 r r r r r @ s z$_UnixWritePipeTransport._fatal_errorc C sb d | _ | j r( | j j | j n | j j | j j | j | j j | j | d S)NT) r r rz r r| r r r r )r r9 r r r r N s
z _UnixWritePipeTransport._closec C sD z | j j | Wd | j j d | _ d | _ d | _ Xd S)N)r r r{ r! rz )r r9 r r r r V s
z-_UnixWritePipeTransport._call_connection_lost)rm rn ro r r r r r r r r r r! r r r r r r r r r )r r rJ s ! ! rJ 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 )getattrru ZF_GETFDZF_SETFD)rs Z inheritableZ cloexec_flagoldr r r _set_inheritablef s
r c @ s e Z d Z d d Z d S)rN c K s d } | t j k r@ | j j \ } } t | j d n t j | d | d | d | d | d d d | | | _ | d k r | j t | j
d d | | j _ n d S)
NFrR rS rT rU Zuniversal_newlinesrV wb buffering)
subprocessPIPErz r r r- Popen_procr! opendetachrS ) r r8 rR rS rT rU rV rW Zstdin_wr r r _startr s
z _UnixSubprocessTransport._startN)rm rn ro r r r r r rN p s rN 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\ r7 r8 r r r rO s z&AbstractChildWatcher.add_child_handlerc C s
t d S)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 r\ r r r remove_child_handler s z)AbstractChildWatcher.remove_child_handlerc C s
t d S)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 rL 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 abcr r r __exit__ s z AbstractChildWatcher.__exit__N)
rm rn ro rp rO 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 | _ d S)N)rz )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 sa | j d k r% | j j t j n | | _ | d k r] | j t j | j | j n d S)N)rz r# r+ SIGCHLDr; _sig_chldr )r rL r r r r s z BaseChildWatcher.attach_loopc C sX y | j WnC t k
rS } z# | j j i d d 6| d 6 WYd d } ~ Xn Xd S)Nz$Unknown exception in SIGCHLD handlerr r )r rQ rz r )r r9 r r r r s zBaseChildWatcher._sig_chldc C sA t j | r t j | St j | r9 t j | S| Sd S)N)rr WIFSIGNALEDWTERMSIG WIFEXITED WEXITSTATUS)r statusr r r _compute_returncode s
z$BaseChildWatcher._compute_returncodeN)
rm rn ro r r! r r r r r r r r r r s r c s 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
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 t j i | _ d S)N)r r
_callbacks)r )r r r r s
zSafeChildWatcher.__init__c s | j j t j 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$ | | f | j | <| j | d S)N)r r )r r\ r7 r8 r r r rO s z"SafeChildWatcher.add_child_handlerc C s/ y | j | =d SWn t k
r* d SYn Xd S)NTF)r r= )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 + s z SafeChildWatcher._do_waitpid_allc C s y t j | t j \ } } Wn. t k
rO | } d } t j d | YnE X| d k r` d S| j | } | j j r t j d | | n y | j
j | \ } } Wn: t k
r | j j r t j d | d d n 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)
rr waitpidWNOHANGChildProcessErrorr r r rz r r r popr= )r r r\ r r] r7 r8 r r r r 0 s*
z SafeChildWatcher._do_waitpid) rm rn ro rp r r! r r rO 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 s; t j i | _ t j | _ i | _ d | _ d S)Nr )r 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! d s
zFastChildWatcher.closec
C s' | j | j d 7_ | SWd QXd S)Nr )r r )r r r r r i s
zFastChildWatcher.__enter__c
C sf | j G | j d 8_ | j s, | j r0 d St | j } | j j Wd QXt j d | d S)Nr z5Caught subprocesses termination from unknown pids: %s)r r r r1 r r r )r r r r Zcollateral_victimsr r r r o s
zFastChildWatcher.__exit__c G sf | j G y | j j | } Wn) t k
rK | | f | j | %dr( )rr r r r r r r r r= r r rz r r r r )r r\ r r] r7 r8 r r r r s6
z FastChildWatcher._do_waitpid_all) rm rn ro rp r r! r r rO r r r r )r r r S 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 sc t j T | j d k rY t | _ t t j t j rY | j j | j j
qY n Wd QXd S)N) r r r r rA r current_thread _MainThreadr _localrz )r r r r
_init_watcher s
z)_UnixDefaultEventLoopPolicy._init_watcherc sN t j | | j d k rJ t t j t j rJ | j j | n d S)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 rA r r r r )r rL )r r r r s z*_UnixDefaultEventLoopPolicy.set_event_loopc C s# | j d k r | j n | 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 rM s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s, | j d k r | j j n | | _ d S)z$Set the watcher for child processes.N)r r! )r rX r r r set_child_watcher s z-_UnixDefaultEventLoopPolicy.set_child_watcher) rm rn ro rp r Z
_loop_factoryr r r rM r r r )r r r s
r )2rp r4 rr r+ r r r r r r r r r r r r r r
r r r
logr __all__r ImportErrorr ZBaseSelectorEventLoopr hasattrrt ru Z
ReadTransportrD Z_FlowControlMixinZWriteTransportrJ r r ZBaseSubprocessTransportrN r r r r ZBaseDefaultEventLoopPolicyr r r r r r r s^ s
F6Ni2