\[\z@sSdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd lmZdd lmZd ddddgZejdkr)ednddZGdddejZeedrlddZnddlZddZGdddejZGdddej ej!Z"Gd d!d!e j#Z$Gd"ddZ%Gd#d$d$e%Z&Gd%dde&Z'Gd&dde&Z(Gd'd(d(e j)Z*eZ+e*Z,dS))z2Selector event loop for Unix with signal handling.N) base_events)base_subprocess) constants)events)selector_events) selectors) transports) coroutine)loggerSelectorEventLoopAbstractChildWatcherSafeChildWatcherFastChildWatcherDefaultEventLoopPolicyZwin32z+Signals are not really supported on WindowscCsdS)zDummy signal handler.N)signumZframerr)/usr/lib/python3.4/asyncio/unix_events.py_sighandler_noop!src seZdZdZdfddZddZfddZd d Zd d Zd dZ ddZ ddZ ddddZ ddddZ edddZddZeddddddddZedddd d!ddd"d#ZS)$_UnixSelectorEventLoopzdUnix event loop. Adds signal handling and UNIX Domain Socket support to SelectorEventLoop. Ncstj|i|_dS)N)super__init___signal_handlers)selfZselector) __class__rrr,sz_UnixSelectorEventLoop.__init__cCs tjS)N)socketZ socketpair)rrrr _socketpair0sz"_UnixSelectorEventLoop._socketpaircs8tjx$t|jD]}|j|qWdS)N)rcloselistrremove_signal_handler)rsig)rrrr3s z_UnixSelectorEventLoop.closecCs.x'|D]}|sqn|j|qWdS)N)_handle_signal)rdatarrrr_process_self_data8s z)_UnixSelectorEventLoop._process_self_datac+Gsk|j|ytj|jjWn=ttfk rf}ztt|WYdd}~XnXt j |||}||j |r)rr[r\Zbindr*rr/Z EADDRINUSEr1ZfamilyrZServerZlistenr]Z_start_serving) rr^r_rXrbrWr4msgZserverrrrcreate_unix_servers:          z)_UnixSelectorEventLoop.create_unix_server)__name__ __module__ __qualname____doc__rrrr#r6r!rr%rErGr rTrKrarfrr)rrr&s,    (    r set_blockingcCstj|ddS)NF)osrk)fdrrr_set_nonblockingsrncCs<tj|tj}|tjB}tj|tj|dS)N)fcntlZF_GETFLrl O_NONBLOCKZF_SETFL)rmflagsrrrrn s cseZdZdZddfddZddZdd Zd d Zd d ZddZ dddZ ddZ ddZ S)r@iNcs tj|||jd<||_||_|j|_tj|jj }t j |p}t j |p}t j |stdnt|j||_d|_|jj|j|j|jj|jj||dk r|jj|jdndS)NrAz)Pipe transport is for pipes/sockets only.F)rr_extra_loop_piper(_filenorlfstatst_modestatS_ISFIFOS_ISSOCKS_ISCHRr)rn _protocol_closing add_reader _read_ready call_soonconnection_made_set_result_unless_cancelled)rlooprArBrCrDmode)rrrrs"       z_UnixReadPipeTransport.__init__cCs|jjd|jg}|jdk rotj|jj|jtj }|r_|j dq||j dn |j dddj |S)Nzfd=%spollingidleclosedz<%s> ) rrgrvrur_test_selector_eventrt _selectorrZ EVENT_READappendjoin)rr.rrrr__repr__)s  z_UnixReadPipeTransport.__repr__cCsytj|j|j}WnLttfk r6Yntk rj}z|j|dWYdd}~XnX|r|jj |nj|j j rt j d|nd|_|j j|j|j j|jj|j j|jddS)Nz"Fatal read error on pipe transportz%r was closed by peerT)rlreadrvmax_sizeBlockingIOErrorInterruptedErrorr* _fatal_errorr}Z data_receivedrt get_debugr r.r~ remove_readerrZ eof_received_call_connection_lost)rr"r4rrrr7s# z"_UnixReadPipeTransport._read_readycCs|jj|jdS)N)rtrrv)rrrr pause_readingIsz$_UnixReadPipeTransport.pause_readingcCs|jj|j|jdS)N)rtrrvr)rrrrresume_readingLsz%_UnixReadPipeTransport.resume_readingcCs|js|jdndS)N)r~_close)rrrrrOs z_UnixReadPipeTransport.closezFatal error on pipe transportcCst|trO|jtjkrO|jjr~tjd||ddq~n/|jji|d6|d6|d6|j d6|j |dS)Nz%r: %sexc_infoTmessage exceptionr`rB) r<r*r/ZEIOrtrr debugcall_exception_handlerr}r)rr4rrrrrSs! z#_UnixReadPipeTransport._fatal_errorcCs6d|_|jj|j|jj|j|dS)NT)r~rtrrvrr)rr4rrrras z_UnixReadPipeTransport._closec CsDz|jj|Wd|jjd|_d|_d|_XdS)N)r}connection_lostrurrt)rr4rrrrfs    z,_UnixReadPipeTransport._call_connection_losti) rgrhrirrrrrrrrrrrr)rrr@s       r@cseZdZddfddZddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ dddZ dddZddZS)rFNcs=tj|||jd<||_||_|j|_tj|jj }t j |}|pt j |pt j |stdnt|j||_g|_d|_d|_|stjjd r|jj|j|jn|jj|jj||dk r9|jj|jdndS)NrAz?Pipe transport is only for pipes, sockets and character devicesrFZaix)rrrsrtrur(rvrlrwrxryr{rzr|r)rnr}_buffer _conn_lostr~sysplatform startswithrrrrr)rrrArBrCrDrZ is_socket)rrrrss*         z _UnixWritePipeTransport.__init__cCs|jjd|jg}|jdk rtj|jj|jtj }|r_|j dn |j d|j }|j d|n |j dddj |S)Nzfd=%srrz bufsize=%srz<%s>r) rrgrvrurrrtrrZ EVENT_WRITErget_write_buffer_sizer)rr.rrPrrrrs    z _UnixWritePipeTransport.__repr__cCstdd|jDS)Ncss|]}t|VqdS)N)len).0r"rrr sz@_UnixWritePipeTransport.get_write_buffer_size..)sumr)rrrrrsz-_UnixWritePipeTransport.get_write_buffer_sizecCsL|jjr"tjd|n|jr>|jtn |jdS)Nz%r was closed by peer)rtrr r.rrBrokenPipeError)rrrrrs  z#_UnixWritePipeTransport._read_readycCst|tttfs*tt|t|trHt|}n|sRdS|jsd|jr|jtj krt j dn|jd7_dS|j styt j|j|}Wnettfk rd}YnHtk r"}z(|jd7_|j|ddSWYdd}~XnX|t|kr9dS|dkrX||d}n|jj|j|jn|j j||jdS)Nz=pipe closed by peer or os.write(pipe, data) raised exception.rrz#Fatal write error on pipe transport)r<bytes bytearray memoryviewrZreprrr~rZ!LOG_THRESHOLD_FOR_CONNLOST_WRITESr warningrrlwritervrr ExceptionrrrtZ add_writer _write_readyrZ_maybe_pause_protocol)rr"nr4rrrrs4*   z_UnixWritePipeTransport.writecCsgdj|j}|s$td|jjytj|j|}Wn~ttfk rt|jj |Ynt k r}z7|j d7_ |j j |j|j|dWYdd}~XnX|t|kr4|j j |j|j|j r0|jr0|j j|j|jdndS|dkrS||d}n|jj |dS)NzData should not be emptyrz#Fatal write error on pipe transportr)rrrZclearrlrrvrrrrrrt remove_writerrrZ_maybe_resume_protocolr~rr)rr"rr4rrrrs* #  z$_UnixWritePipeTransport._write_readycCsdS)NTr)rrrr can_write_eofsz%_UnixWritePipeTransport.can_write_eofcCs^|jr dS|jstd|_|jsZ|jj|j|jj|jdndS)NT) r~rurZrrtrrvrr)rrrr write_eofs   z!_UnixWritePipeTransport.write_eofcCs|js|jndS)N)r~r)rrrrrs z_UnixWritePipeTransport.closecCs|jddS)N)r)rrrrabortsz_UnixWritePipeTransport.abortzFatal error on pipe transportcCst|ttfrC|jjrrtjd||ddqrn/|jji|d6|d6|d6|jd6|j |dS)Nz%r: %srTrrr`rB) r<rConnectionResetErrorrtrr rrr}r)rr4rrrrrs z$_UnixWritePipeTransport._fatal_errorcCsbd|_|jr(|jj|jn|jj|jj|j|jj|j|dS)NT) r~rrtrrvrrrr)rr4rrrrs    z_UnixWritePipeTransport._closec CsDz|jj|Wd|jjd|_d|_d|_XdS)N)r}rrurrt)rr4rrrrs    z-_UnixWritePipeTransport._call_connection_lost)rgrhrirrrrrrrrrrrrrrr)rrrFps    !    rFc@seZdZddZdS)rIc Ksd}|tjkr-|jj\}}ntj|d|d|d|d|ddd|||_|dk r|jt|jdd ||j_ ndS) NrLrMrNrOZuniversal_newlinesFrPwb buffering) subprocessPIPErtrPopen_procropendetachrM) rr3rLrMrNrOrPrQZstdin_wrrr_start$s  z_UnixSubprocessTransport._startN)rgrhrirrrrrrI"s rIc@s^eZdZdZddZddZddZdd Zd d Zd d Z dS)r aHAbstract 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. cGs tdS)aRegister 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)rrUr2r3rrrrJLs z&AbstractChildWatcher.add_child_handlercCs tdS)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)rrUrrrremove_child_handlerWsz)AbstractChildWatcher.remove_child_handlercCs tdS)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)rrrrr attach_loop_sz AbstractChildWatcher.attach_loopcCs tdS)zlClose the watcher. This must be called to make sure that any underlying resource is freed. N)r)rrrrriszAbstractChildWatcher.closecCs tdS)zdEnter the watcher's context and allow starting new processes This function must return selfN)r)rrrr __enter__pszAbstractChildWatcher.__enter__cCs tdS)zExit the watcher's contextN)r)rabcrrr__exit__vszAbstractChildWatcher.__exit__N) rgrhrirjrJrrrrrrrrrr 5s    c@sdeZdZddZddZddZddZd d Zd d Zd dZ dS)BaseChildWatchercCs d|_dS)N)rt)rrrrr}szBaseChildWatcher.__init__cCs|jddS)N)r)rrrrrszBaseChildWatcher.closecCs tdS)N)r)r expected_pidrrr _do_waitpidszBaseChildWatcher._do_waitpidcCs tdS)N)r)rrrr_do_waitpid_allsz BaseChildWatcher._do_waitpid_allcCs|dks$t|tjs$t|jdk rI|jjtjn||_|dk r|jtj|j |j ndS)N) r<rZAbstractEventLooprZrtrr&SIGCHLDr6 _sig_chldr)rrrrrrs$  zBaseChildWatcher.attach_loopcCsXy|jWnCtk rS}z#|jjidd6|d6WYdd}~XnXdS)Nz$Unknown exception in SIGCHLD handlerrr)rrrtr)rr4rrrrs  zBaseChildWatcher._sig_chldcCsAtj|rtj| Stj|r9tj|S|SdS)N)rl WIFSIGNALEDWTERMSIG WIFEXITED WEXITSTATUS)rstatusrrr_compute_returncodes  z$BaseChildWatcher._compute_returncodeN) rgrhrirrrrrrrrrrrr{s       rcseZdZdZfddZfddZddZdd Zd d Zd d Z ddZ ddZ S)rad'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) cstji|_dS)N)rr _callbacks)r)rrrrs zSafeChildWatcher.__init__cs|jjtjdS)N)rrrr)r)rrrrs zSafeChildWatcher.closecCs|S)Nr)rrrrrszSafeChildWatcher.__enter__cCsdS)Nr)rrrrrrrrszSafeChildWatcher.__exit__cGs$||f|j|<|j|dS)N)rr)rrUr2r3rrrrJsz"SafeChildWatcher.add_child_handlerc Cs/y|j|=dSWntk r*dSYnXdS)NTF)rr8)rrUrrrrs   z%SafeChildWatcher.remove_child_handlercCs+x$t|jD]}|j|qWdS)N)rrr)rrUrrrrsz SafeChildWatcher._do_waitpid_allcCs|dkstytj|tj\}}Wn.tk ra|}d}tjd|YnEX|dkrrdS|j|}|jj rtj d||ny|j j |\}}Wn:t k r|jj rtjd|ddnYnX||||dS)Nrz8Unknown child process pid %d, will report returncode 255z$process %s exited with returncode %sz'Child watcher got an unexpected pid: %rrT)rZrlwaitpidWNOHANGChildProcessErrorr rrrtrrrpopr8)rrrUrrVr2r3rrrrs,       zSafeChildWatcher._do_waitpid) rgrhrirjrrrrrJrrrrr)rrrs      csveZdZdZfddZfddZddZdd Zd d Zd d Z ddZ S)raW'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). cs;tji|_tj|_i|_d|_dS)Nr)rrr threadingZLock_lock_zombies_forks)r)rrrr s    zFastChildWatcher.__init__cs+|jj|jjtjdS)N)rrrrr)r)rrrrs  zFastChildWatcher.closec Cs'|j|jd7_|SWdQXdS)Nr)rr)rrrrrs zFastChildWatcher.__enter__c Csf|jG|jd8_|js,|j r0dSt|j}|jjWdQXtjd|dS)Nrz5Caught subprocesses termination from unknown pids: %s)rrrr,rr r)rrrrZcollateral_victimsrrrrs zFastChildWatcher.__exit__cGs{|jstd|jGy|jj|}Wn)tk r`||f|j| %dr$)rlrrrrrrrr8rrrtrr rr)rrUrrVr2r3rrrr=s6             z FastChildWatcher._do_waitpid_all) rgrhrirjrrrrrJrrrr)rrrs    csdeZdZdZeZfddZddZfddZdd Z d d Z S) _UnixDefaultEventLoopPolicyZXXXcstjd|_dS)N)rr_watcher)r)rrrrls z$_UnixDefaultEventLoopPolicy.__init__c CsctjT|jdkrYt|_ttjtjrY|jj|j j qYnWdQXdS)N) rrrrr<rcurrent_thread _MainThreadr_localrt)rrrr _init_watcherps     z)_UnixDefaultEventLoopPolicy._init_watchercsNtj||jdk rJttjtjrJ|jj|ndS)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)rset_event_looprr<rrrr)rr)rrrrxsz*_UnixDefaultEventLoopPolicy.set_event_loopcCs#|jdkr|jn|jS)zzGet the watcher for child processes. If not yet set, a SafeChildWatcher object is automatically created. N)rr)rrrrrHs z-_UnixDefaultEventLoopPolicy.get_child_watchercCsM|dks!t|ts!t|jdk r@|jjn||_dS)z$Set the watcher for child processes.N)r<r rZrr)rrRrrrset_child_watchers!z-_UnixDefaultEventLoopPolicy.set_child_watcher) rgrhrirjrZ _loop_factoryrrrrHrrr)rrrhs   r)-rjr/rlr&rryrrrrrrrrrr Z coroutinesr logr __all__r ImportErrorrZBaseSelectorEventLooprhasattrrnroZ ReadTransportr@Z_FlowControlMixinZWriteTransportrFZBaseSubprocessTransportrIr rrrZBaseDefaultEventLoopPolicyrr rrrrrsN            _ F6Ni2