3
\
@ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
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
lmZ ddlmZ dddddgZejdkreddd Zy
ejZ W n e k
r( dd Z Y nX G dd dej Z e ed rRd d Z!nd dl"Z"d d Z!G d d! d!ej#Z$G d"d# d#ej%ej&Z'e ed$rej(Z)nd dl"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-Z/G d.d/ d/ej0Z1e Z2e1Z3dS )0 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 d S )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 Zd! fdd Zdd Z fddZdd Zd
d Zd d
Zdd Z dd Z
d"ddZ d#ddZ e
d$ddZdd Ze
dddddd Ze
d%dd dd d d Z ZS )&_UnixSelectorEventLoopNc 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 d d }~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 d d }~X nX |jtjkr0t dj|n W Y d d }~X nX d S )Nz3coroutines cannot be used with add_signal_handler()Fr 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|}|d krd S |jr*| j| n
| j| d S )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 d d }~X nX | jsytj d W n2 t
tfk
r } ztjd| W Y d d }~X nX dS )NFzsig {} cannot be caughtr 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|tjd S )Nz sig must be an int, not {!r}r zsig {} out of range(1, {}))
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 |r|d kr&t dn|d k r&t d|d k r|d k r>t dtjtjtjd}y |jd | j||E d H W q |j Y qX nB|d krt d|jtjkstj |j
rt dj ||jd | j ||||E d H \}}||fS )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})
r8 r" AF_UNIX SOCK_STREAM setblockingZ sock_connectr% familyr _is_stream_sockettyper@ Z _create_connection_transport)r protocol_factoryr rg rh ri transportrP r r r create_unix_connection s8
z-_UnixSelectorEventLoop.create_unix_connectiond )rh backlogrg c
! C s t |trtd|d k r0|d k r,tdt|}tjtjtj}|d d
kry tj t
j|j rnt
j | W nB t
k
r Y n0 tk
r } ztjd|| W Y d d }~X nX y|j| W nj tk
r } z8|j |jtjkrdj|}ttj|d n W Y d d }~X n |j Y nX n>|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 ru ) rK boolr3 r8 _fspathr" rj rk statS_ISSOCKosst_moderemoveFileNotFoundErrorr9 r errorZbindr% r> Z
EADDRINUSEr@ rm r rn ro ZServerZlistenrl Z_start_serving)
r rp r rh rt rg rc rC msgZserverr r r create_unix_server sP
z)_UnixSelectorEventLoop.create_unix_server)N)NN)NN)N)N)__name__
__module__ __qualname__r r# r% r1 rE r/ r) r4 rS rU r
rd rY rr r
__classcell__r r )r! r r 1 s* -
%r set_blockingc C s t j| d d S )NF)rz r )fdr r r _set_nonblockingB s r c C s, t j | t j}|tjB }t j | t j| d S )N)fcntlZF_GETFLrz
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_closingrz fstatr{ rx S_ISFIFOry 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)rz 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 exceptionrq 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 rz r r{ rx r r ry 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 s t |trt|}|sd S | js&| jrN| jtjkr s&
z$_UnixWritePipeTransport._write_readyc C s dS )NTr )r r r r
can_write_eofX s z%_UnixWritePipeTransport.can_write_eofc C s8 | j r
d S d| _ | js4| jj| j | jj| jd d S )NT)r 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 rq 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 Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )r c G s
t d S )N)NotImplementedError)r re rA rB r r r rX s z&AbstractChildWatcher.add_child_handlerc C s
t d S )N)r )r re r r r remove_child_handler s z)AbstractChildWatcher.remove_child_handlerc C s
t d S )N)r )r r r r r attach_loop s z AbstractChildWatcher.attach_loopc C s
t d S )N)r )r r r r r% s zAbstractChildWatcher.closec C s
t d S )N)r )r r r r __enter__ s z AbstractChildWatcher.__enter__c C s
t d S )N)r )r abcr r r __exit__ s z AbstractChildWatcher.__exit__N) 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 sf | j d k r$|d kr$| jr$tjdt | j d k r<| j jtj || _ |d k rb|jtj| j | j
d S )NzCA loop is being detached from a child watcher with pending handlers) r r r* r+ RuntimeWarningr) r5 SIGCHLDrE _sig_chldr )r r r r r r s
z BaseChildWatcher.attach_loopc C sF y | j W n4 tk
r@ } z| jjd|d W Y d d }~X nX d S )Nz$Unknown exception in SIGCHLD handler)r r )r rZ r r )r rC r r r r 1 s zBaseChildWatcher._sig_chldc C s2 t j|rt j| S t j|r*t j|S |S d S )N)rz WIFSIGNALEDWTERMSIG 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 s r c sL e Zd Z fddZdd Zdd Zdd Zd d
Zd d Zd
d Z Z
S )r c s | j j t j d S )N)r r- r r% )r )r! r r r% V s
zSafeChildWatcher.closec C s | S )Nr )r r r r r Z 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|< | j| d S )NzICannot add child handler, the child watcher does not have a loop attached)r r: r r )r re rA rB r r r rX ` s
z"SafeChildWatcher.add_child_handlerc C s& y | j |= dS tk
r dS X d S )NTF)r rG )r re r r r r k s
z%SafeChildWatcher.remove_child_handlerc C s" x t | jD ]}| j| q W d S )N)r( r r )r re r r r r r s z SafeChildWatcher._do_waitpid_allc C s yt j|t j\}}W n( tk
r> |}d}tjd| Y n0X |dkrLd S | j|}| jj rntj d|| y| j
j |\}}W n. t k
r | jj rtjd|dd Y nX |||f| 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: %rT)r )
rz waitpidWNOHANGChildProcessErrorr r r r r r r poprG )r r re r rf rA rB r r r r w s*
z SafeChildWatcher._do_waitpid) r r r r% r r rX r r r r r r )r! r r K s c sP e Zd Z fddZ fddZdd Zdd Zd d
Zd d Zd
d Z Z
S )r c s$ t j tj | _i | _d| _d S )Nr )r r threadingZLock_lock_zombies_forks)r )r! r r r s
zFastChildWatcher.__init__c s" | j j | jj t j 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 sV | j : | jd8 _| js$| j r(d S t| j}| jj W d Q R X tjd| d S )Nr z5Caught subprocesses termination from unknown pids: %s)r r r r; r- r r )r r r r Zcollateral_victimsr r r r s
zFastChildWatcher.__exit__c G sl | j d krtd| j: y| jj|}W n" tk
rL ||f| j|< d S X W d Q R X |||f| d S )NzICannot add child handler, the child watcher does not have a loop attached)r r: r r r rG r )r re rA rB rf r r r rX s
z"FastChildWatcher.add_child_handlerc C s& y | j |= dS tk
r dS X d S )NTF)r rG )r re r r r r s
z%FastChildWatcher.remove_child_handlerc C s xyt jdt j\}}W n tk
r, d S X |dkr:d S | j|}| jv y| jj|\}}W nB tk
r | j r|| j
|< | j j rt
jd|| wd }Y nX | j j rt
jd|| W d Q R X |d krt
jd|| 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 -> %dr2 )rz 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 rX r r r r r )r! r r s
c sD e Zd ZeZ fddZdd Z fddZdd Zd d
Z Z
S ) _UnixDefaultEventLoopPolicyc 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| | jd k r2ttj tjr2| jj| d S )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 d kr| j | j S )N)r r )r r r r rV & s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s | j d k r| j j || _ d S )N)r r% )r ra r r r set_child_watcher0 s
z-_UnixDefaultEventLoopPolicy.set_child_watcher) r r r r Z
_loop_factoryr r r rV r r r r )r! r r s
r )4r> rz r5 r" rx r r& r r* r r r r r r r r
r r r
logr __all__r ImportErrorr fspathrw AttributeErrorZBaseSelectorEventLoopr hasattrr r Z
ReadTransportrN Z_FlowControlMixinZWriteTransportrT r r ZBaseSubprocessTransportrW r r r r ZBaseDefaultEventLoopPolicyr r r r r r r sl
O
F=On2