B
    v9aTž  ã               @   s–  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	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Zejdkræedƒ‚dd„ ZG dd„ dejƒZG dd„ dejƒZG dd„ dejej ƒZ!G dd„ dej"ƒZ#G dd„ dƒZ$G dd„ de$ƒZ%G dd„ de%ƒZ&G d d!„ d!e%ƒZ'G d"d#„ d#ej(ƒZ)eZ*e)Z+dS )$é    Né   )Úbase_events)Úbase_subprocess)Ú	constants)Ú
coroutines)Úevents)Úfutures)Úselector_events)Útasks)Ú
transports)Úlogger)ÚSelectorEventLoopÚAbstractChildWatcherÚSafeChildWatcherÚFastChildWatcherÚDefaultEventLoopPolicyÚwin32z+Signals are not really supported on Windowsc             C   s   d S )N© )ÚsignumÚframer   r   ú!/usr/lib/python3.7/unix_events.pyÚ_sighandler_noop'   s    r   c                   sÆ   e Zd Zd(‡ 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d„Z
d*dd„Zd+dd„Zdd„ Zd,dddddœdd„Zd-ddddddœdd„Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Z‡  ZS ).Ú_UnixSelectorEventLoopNc                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(| jrZtjd| ›dt	| d | j 
¡  d S )NzClosing 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 ]}|sq|   |¡ qW d S )N)Ú_handle_signal)r   Údatar   r   r   r   Ú_process_self_dataD   s    
z)_UnixSelectorEventLoop._process_self_datac             G   sL  t  |¡st  |¡rtdƒ‚|  |¡ |  ¡  yt | j 	¡ ¡ W n2 t
tfk
rt } ztt|ƒƒ‚W d d }~X Y nX t ||| d ¡}|| j|< yt |t¡ t |d¡ W nš tk
rF } zz| j|= | jsyt d¡ W n4 t
tfk
r } zt d|¡ W d d }~X Y nX |jtjkr4td|› dƒ‚n‚ W d d }~X Y nX d S )Nz3coroutines cannot be used with add_signal_handler()Féÿÿÿÿzset_wakeup_fd(-1) failed: %szsig z cannot be caught)r   ZiscoroutineZiscoroutinefunctionÚ	TypeErrorÚ_check_signalZ_check_closedÚsignalÚset_wakeup_fdZ_csockÚfilenoÚ
ValueErrorÚOSErrorÚRuntimeErrorÚstrr   ZHandler   r   Úsiginterruptr   ÚinfoÚerrnoÚEINVAL)r   r)   ÚcallbackÚargsÚexcÚhandleZnexcr   r   r   Úadd_signal_handlerK   s0    



z)_UnixSelectorEventLoop.add_signal_handlerc             C   s8   | j  |¡}|d krd S |jr*|  |¡ n
|  |¡ d S )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 tk
r*   dS X |tjkr>tj}ntj}yt ||¡ W nB tk
r– } z$|jtj	kr„t
d|› dƒ‚n‚ W d d }~X Y nX | jsàyt d¡ W n2 ttfk
rÞ } zt d|¡ W d d }~X Y nX dS )NFzsig z cannot be caughtr-   zset_wakeup_fd(-1) failed: %sT)r/   r   ÚKeyErrorr0   ÚSIGINTÚdefault_int_handlerÚSIG_DFLr4   r9   r:   r5   r1   r3   r   r8   )r   r)   Zhandlerr=   r   r   r   r$   ‚   s(    

z,_UnixSelectorEventLoop.remove_signal_handlerc             C   sL   t |tƒstd|›ƒ‚d|  kr.tjk sHn td|› dtj› dƒ‚d S )Nzsig must be an int, not r   zsig z out of range(1, ú))Ú
isinstanceÚintr.   r0   ÚNSIGr3   )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   rJ   rK   rL   rM   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, tk
r„   | ¡  | 	¡ I d H  ‚ Y nX W d Q R X |S )N)rL   rM   )
r   Úget_child_watcherÚcreate_futureÚ_UnixSubprocessTransportÚadd_child_handlerZget_pidÚ_child_watcher_callbackÚ	Exceptionr    Z_wait)r   rK   r<   ÚshellÚstdinÚstdoutÚstderrÚbufsizerM   ÚkwargsÚwatcherrL   Útranspr   r   r   Ú_make_subprocess_transport¶   s    




z1_UnixSelectorEventLoop._make_subprocess_transportc             C   s   |   |j|¡ d S )N)Zcall_soon_threadsafeZ_process_exited)r   ÚpidÚ
returncoder^   r   r   r   rU   Ë   s    z._UnixSelectorEventLoop._child_watcher_callback)ÚsslÚsockÚserver_hostnameÚssl_handshake_timeoutc      	      Ã   s
  |r|d kr6t dƒ‚n |d k	r&t dƒ‚|d k	r6t dƒ‚|d k	r¤|d k	rNt dƒ‚t |¡}t tjtjd¡}y | d¡ |  ||¡I d H  W qä   | ¡  ‚ Y qäX n@|d kr´t dƒ‚|j	tjksÌ|j
tjkrÚt d|›ƒ‚| d¡ | j|||||d	I d H \}}||fS )
Nz/you have to pass server_hostname when using sslz+server_hostname is only meaningful with sslz1ssl_handshake_timeout is only meaningful with sslz3path and sock can not be specified at the same timer   Fzno path and sock were specifiedz.A UNIX Domain Stream Socket was expected, got )re   )r3   ÚosÚfspathÚsocketÚAF_UNIXÚSOCK_STREAMÚsetblockingZsock_connectr    ÚfamilyÚtypeZ_create_connection_transport)	r   Úprotocol_factoryÚpathrb   rc   rd   re   Ú	transportrK   r   r   r   Úcreate_unix_connectionÎ   s@    



z-_UnixSelectorEventLoop.create_unix_connectionéd   T)rc   Úbacklogrb   re   Ústart_servingc         
   Ã   sÊ  t |tƒrtdƒ‚|d k	r&|s&tdƒ‚|d k	rH|d k	r@tdƒ‚t |¡}t tjtj¡}|d dkrÊy t	 
t 	|¡j¡r„t |¡ W nB tk
rš   Y n0 tk
rÈ } zt d||¡ W d d }~X Y nX y| |¡ W nl tk
r0 }	 z8| ¡  |	jtjkrd|›d}
ttj|
ƒd ‚n‚ W d d }	~	X Y n   | ¡  ‚ Y nX n<|d krZtd	ƒ‚|jtjksv|jtjkr„td
|›ƒ‚| d¡ t | |g||||¡}|rÆ| ¡  tjd| dI d H  |S )Nz*ssl argument must be an SSLContext or Nonez1ssl_handshake_timeout is only meaningful with sslz3path and sock can not be specified at the same timer   )r   ú z2Unable to check or remove stale UNIX socket %r: %rzAddress z is already in usez-path was not specified, and no sock specifiedz.A UNIX Domain Stream Socket was expected, got F)Úloop)rF   Úboolr.   r3   rf   rg   rh   ri   rj   ÚstatÚS_ISSOCKÚst_modeÚremoveÚFileNotFoundErrorr4   r   ÚerrorZbindr    r9   Z
EADDRINUSErl   rm   rk   r   ZServerZ_start_servingr
   Zsleep)r   rn   ro   rc   rs   rb   re   rt   Úerrr=   ÚmsgZserverr   r   r   Úcreate_unix_serverû   sZ    






z)_UnixSelectorEventLoop.create_unix_serverc          
   Ã   sô   y
t j W n, tk
r6 } zt d¡‚W d d }~X Y nX y| ¡ }W n2 ttjfk
rv } zt d¡‚W d d }~X Y nX yt  |¡j	}W n, t
k
r´ } zt d¡‚W d d }~X Y nX |r¾|n|}	|	sÊdS |  ¡ }
|  |
d |||||	d¡ |
I d H S )Nzos.sendfile() is not availableznot a regular filer   )rf   ÚsendfileÚAttributeErrorr   ÚSendfileNotAvailableErrorr2   ÚioÚUnsupportedOperationÚfstatÚst_sizer4   rR   Ú_sock_sendfile_native_impl)r   rc   ÚfileÚoffsetÚcountr=   r2   r~   ZfsizeÚ	blocksizeÚfutr   r   r   Ú_sock_sendfile_native>  s(    
z,_UnixSelectorEventLoop._sock_sendfile_nativec	             C   s  |  ¡ }	|d k	r|  |¡ | ¡ r4|  |||¡ d S |rd|| }|dkrd|  |||¡ | |¡ d S yt |	|||¡}
W n* ttfk
rÆ   |d kr¢|  	||¡ |  
|	| j||	||||||¡
 Y nH tk
rj } z†|d k	r|jtjkrt|ƒtk	rtdtjƒ}||_|}|dkrBt d¡}|  |||¡ | |¡ n|  |||¡ | |¡ W d d }~X Y n¤ tk
r¤ } z|  |||¡ | |¡ W d d }~X Y njX |
dkrÊ|  |||¡ | |¡ nD||
7 }||
7 }|d krð|  	||¡ |  
|	| j||	||||||¡
 d S )Nr   zsocket is not connectedzos.sendfile call failed)r2   Úremove_writerÚ	cancelledÚ_sock_sendfile_update_fileposZ
set_resultrf   r   ÚBlockingIOErrorÚInterruptedErrorÚ_sock_add_cancellation_callbackZ
add_writerrˆ   r4   r9   ZENOTCONNrm   ÚConnectionErrorÚ	__cause__r   rƒ   Zset_exceptionrV   )r   r   Zregistered_fdrc   r2   rŠ   r‹   rŒ   Ú
total_sentÚfdZsentr=   Únew_excr~   r   r   r   rˆ   U  s`    






z1_UnixSelectorEventLoop._sock_sendfile_native_implc             C   s   |dkrt  ||t j¡ d S )Nr   )rf   ÚlseekÚSEEK_SET)r   r2   rŠ   r—   r   r   r   r‘   ™  s    z4_UnixSelectorEventLoop._sock_sendfile_update_fileposc                s   ‡ ‡fdd„}|  |¡ d S )Nc                s&   |   ¡ r"ˆ ¡ }|dkr"ˆ  |¡ d S )Nr-   )r   r2   r   )r   r˜   )r   rc   r   r   Úcbž  s    zB_UnixSelectorEventLoop._sock_add_cancellation_callback.<locals>.cb)Zadd_done_callback)r   r   rc   rœ   r   )r   rc   r   r”     s    z6_UnixSelectorEventLoop._sock_add_cancellation_callback)N)NN)NN)N)N)N)Ú__name__Ú
__module__Ú__qualname__r   r    r,   r?   r*   r$   r/   rN   rP   r_   rU   rq   r€   rŽ   rˆ   r‘   r”   Ú__classcell__r   r   )r   r   r   ,   s2   -
  
 

*?Dr   c                   s„   e Zd ZdZ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ddd„Zdd„ Zdd„ Z‡  ZS ) rI   i   Nc                sÔ   t ƒ  |¡ || jd< || _|| _| ¡ | _|| _d| _t	 
| j¡j}t |¡s~t |¡s~t |¡s~d | _d | _d | _tdƒ‚t	 | jd¡ | j | jj| ¡ | j | jj| j| j¡ |d k	rÐ| j tj|d ¡ d S )NrJ   Fz)Pipe transport is for pipes/sockets only.)r   r   Ú_extraÚ_loopÚ_piper2   Ú_filenoÚ	_protocolÚ_closingrf   r†   rz   rx   ÚS_ISFIFOry   ÚS_ISCHRr3   Úset_blockingÚ	call_soonÚconnection_madeÚ_add_readerÚ_read_readyr   Ú_set_result_unless_cancelled)r   rv   rJ   rK   rL   rM   Úmode)r   r   r   r   ª  s,    





z_UnixReadPipeTransport.__init__c             C   sÀ   | j jg}| jd kr | d¡ n| jr0| d¡ | d| j› ¡ t| jdd ƒ}| jd k	r|d k	rt 	|| jt
j¡}|r„| d¡ q°| d¡ n | jd k	r¦| d¡ n
| d¡ d d	 |¡¡S )
NÚclosedÚclosingzfd=Ú	_selectorÚpollingÚidleÚopenz<{}>ú )r   r   r£   Úappendr¦   r¤   Úgetattrr¢   r	   Ú_test_selector_eventÚ	selectorsZ
EVENT_READÚformatÚjoin)r   r8   r   r³   r   r   r   Ú__repr__Ç  s"    




z_UnixReadPipeTransport.__repr__c          
   C   sº   yt  | j| j¡}W nD ttfk
r,   Y nŠ tk
rX } z|  |d¡ W d d }~X Y n^X |rl| j 	|¡ nJ| j
 ¡ r‚t d| ¡ d| _| j
 | j¡ | j
 | jj¡ | j
 | jd ¡ d S )Nz"Fatal read error on pipe transportz%r was closed by peerT)rf   Úreadr¤   Úmax_sizer’   r“   r4   Ú_fatal_errorr¥   Zdata_receivedr¢   Ú	get_debugr   r8   r¦   Ú_remove_readerrª   Zeof_receivedÚ_call_connection_lost)r   r+   r=   r   r   r   r­   Ü  s    
z"_UnixReadPipeTransport._read_readyc             C   s   | j  | j¡ d S )N)r¢   rÂ   r¤   )r   r   r   r   Úpause_readingî  s    z$_UnixReadPipeTransport.pause_readingc             C   s   | 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   rK   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*tjd| ›t| d | j  ¡  d S )Nzunclosed 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|jtjkr4| j ¡ rLtjd| |dd n| j ||| | j	dœ¡ |  
|¡ d S )Nz%r: %sT)Úexc_info)ÚmessageÚ	exceptionrp   rK   )rF   r4   r9   ZEIOr¢   rÁ   r   ÚdebugÚ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   rI   ¦  s   
rI   c                   s¢   e Zd Z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dd„ Zdd„ Zdd„ Zdd„ Zd&dd „Zd'd!d"„Zd#d$„ Z‡  ZS )(rO   Nc       
         sþ   t ƒ  ||¡ || jd< || _| ¡ | _|| _tƒ | _d| _	d| _
t | j¡j}t |¡}t |¡}t |¡}	|s”|s”|	s”d | _d | _d | _tdƒ‚t | jd¡ | j | jj| ¡ |	sÈ|ràtj d¡sà| j | jj| j| j¡ |d k	rú| j tj|d ¡ d S )NrJ   r   Fz?Pipe transport is only for pipes, sockets and character devicesZaix)r   r   r¡   r£   r2   r¤   r¥   Ú	bytearrayÚ_bufferÚ
_conn_lostr¦   rf   r†   rz   rx   r¨   r§   ry   r3   r©   r¢   rª   r«   r!   ÚplatformÚ
startswithr¬   r­   r   r®   )
r   rv   rJ   rK   rL   rM   r¯   Zis_charZis_fifoZ	is_socket)r   r   r   r   '  s2    





z _UnixWritePipeTransport.__init__c             C   sØ   | j jg}| jd kr | d¡ n| jr0| d¡ | d| j› ¡ t| jdd ƒ}| jd k	r¨|d k	r¨t 	|| jt
j¡}|r„| d¡ n
| d¡ |  ¡ }| d|› ¡ n | jd k	r¾| d¡ n
| d¡ d	 d
 |¡¡S )Nr°   r±   zfd=r²   r³   r´   zbufsize=rµ   z<{}>r¶   )r   r   r£   r·   r¦   r¤   r¸   r¢   r	   r¹   rº   ZEVENT_WRITEÚget_write_buffer_sizer»   r¼   )r   r8   r   r³   r[   r   r   r   r½   L  s&    





z _UnixWritePipeTransport.__repr__c             C   s
   t | jƒS )N)ÚlenrÓ   )r   r   r   r   r×   d  s    z-_UnixWritePipeTransport.get_write_buffer_sizec             C   s6   | j  ¡ rt d| ¡ | jr*|  tƒ ¡ n|  ¡  d S )Nz%r was closed by peer)r¢   rÁ   r   r8   rÓ   rÉ   ÚBrokenPipeError)r   r   r   r   r­   g  s
    
z#_UnixWritePipeTransport._read_readyc          
   C   s  t |tƒrt|ƒ}|sd S | js&| jrN| jtjkr<t d¡ |  jd7  _d S | j	søyt
 | j|¡}W nV ttfk
r‚   d}Y n< tk
r¼ } z|  jd7  _|  |d¡ d S d }~X Y nX |t|ƒkrÎd S |dkræt|ƒ|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)rF   rÒ   Ú
memoryviewrÔ   r¦   r   Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr   ÚwarningrÓ   rf   Úwriter¤   r’   r“   rV   rÀ   rØ   r¢   Z_add_writerÚ_write_readyZ_maybe_pause_protocol)r   r+   Únr=   r   r   r   rÜ   p  s2    

z_UnixWritePipeTransport.writec          
   C   sè   yt  | j| j¡}W nj ttfk
r,   Y n¸ tk
r~ } z6| j ¡  |  jd7  _| j	 
| j¡ |  |d¡ W d d }~X Y nfX |t| jƒkrÐ| j ¡  | j	 
| j¡ |  ¡  | jrÌ| j	 | j¡ |  d ¡ d S |dkrä| jd |…= d S )Nr   z#Fatal write error on pipe transportr   )rf   rÜ   r¤   rÓ   r’   r“   rV   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   dS )NTr   )r   r   r   r   Úcan_write_eof«  s    z%_UnixWritePipeTransport.can_write_eofc             C   s8   | j r
d S d| _ | js4| j | j¡ | j | jd ¡ 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   rK   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| js|  ¡  d S )N)r£   r¦   rá   )r   r   r   r   r    À  s    z_UnixWritePipeTransport.closec             C   s.   | j d k	r*tjd| ›t| d | j  ¡  d S )Nzunclosed 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@tjd| |dd n| j ||| | jdœ¡ |  |¡ d S )Nz%r: %sT)rÌ   )rÍ   rÎ   rp   rK   )	rF   r4   r¢   rÁ   r   rÏ   rÐ   r¥   rÉ   )r   r=   rÍ   r   r   r   rÀ   Î  s    

z$_UnixWritePipeTransport._fatal_errorc             C   sF   d| _ | jr| 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   rO   $  s"   %	!	

rO   c               @   s   e Zd Zdd„ ZdS )rS   c       	   	   K   sŠ   d }|t jkrt ¡ \}}zPt j|f||||d|dœ|—Ž| _|d k	rh| ¡  t| ¡ d|d| j_	d }W d |d k	r„| ¡  | ¡  X d S )NF)rW   rX   rY   rZ   Zuniversal_newlinesr[   Úwb)Ú	buffering)
Ú
subprocessÚPIPErh   Z
socketpairÚPopenÚ_procr    rµ   ÚdetachrX   )	r   r<   rW   rX   rY   rZ   r[   r\   Zstdin_wr   r   r   Ú_startð  s    
z_UnixSubprocessTransport._startN)r   rž   rŸ   rë   r   r   r   r   rS   î  s   rS   c               @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )r   c             G   s
   t ƒ ‚d S )N)ÚNotImplementedError)r   r`   r;   r<   r   r   r   rT     s    	z&AbstractChildWatcher.add_child_handlerc             C   s
   t ƒ ‚d S )N)rì   )r   r`   r   r   r   Úremove_child_handler)  s    z)AbstractChildWatcher.remove_child_handlerc             C   s
   t ƒ ‚d S )N)rì   )r   rv   r   r   r   Úattach_loop1  s    z AbstractChildWatcher.attach_loopc             C   s
   t ƒ ‚d S )N)rì   )r   r   r   r   r    ;  s    zAbstractChildWatcher.closec             C   s
   t ƒ ‚d S )N)rì   )r   r   r   r   Ú	__enter__B  s    zAbstractChildWatcher.__enter__c             C   s
   t ƒ ‚d S )N)rì   )r   ÚaÚbÚcr   r   r   Ú__exit__H  s    zAbstractChildWatcher.__exit__N)	r   rž   rŸ   rT   rí   rî   r    rï   ró   r   r   r   r   r     s   
r   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 | _ i | _d S )N)r¢   Ú
_callbacks)r   r   r   r   r   O  s    zBaseChildWatcher.__init__c             C   s   |   d ¡ d S )N)rî   )r   r   r   r   r    S  s    zBaseChildWatcher.closec             C   s
   t ƒ ‚d S )N)rì   )r   Úexpected_pidr   r   r   Ú_do_waitpidV  s    zBaseChildWatcher._do_waitpidc             C   s
   t ƒ ‚d S )N)rì   )r   r   r   r   Ú_do_waitpid_allY  s    z BaseChildWatcher._do_waitpid_allc             C   sf   | j d k	r$|d kr$| jr$t dt¡ | j d k	r<| j  tj¡ || _ |d k	rb| tj| j	¡ |  
¡  d S )NzCA loop is being detached from a child watcher with pending handlers)r¢   rõ   r%   r&   ÚRuntimeWarningr$   r0   ÚSIGCHLDr?   Ú	_sig_chldrø   )r   rv   r   r   r   rî   \  s    
zBaseChildWatcher.attach_loopc          
   C   sF   y|   ¡  W n4 tk
r@ } z| j d|dœ¡ W d d }~X Y nX d S )Nz$Unknown exception in SIGCHLD handler)rÍ   rÎ   )rø   rV   r¢   rÐ   )r   r=   r   r   r   rû   p  s    zBaseChildWatcher._sig_chldc             C   s2   t  |¡rt  |¡ 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ô   M  s   rô   c                   sL   e Zd Z‡ fdd„Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	‡  Z
S )r   c                s   | j  ¡  tƒ  ¡  d S )N)rõ   r(   r   r    )r   )r   r   r   r    •  s    
zSafeChildWatcher.closec             C   s   | S )Nr   )r   r   r   r   rï   ™  s    zSafeChildWatcher.__enter__c             C   s   d S )Nr   )r   rð   rñ   rò   r   r   r   ró   œ  s    zSafeChildWatcher.__exit__c             G   s.   | j d krtdƒ‚||f| j|< |  |¡ d S )NzICannot add child handler, the child watcher does not have a loop attached)r¢   r5   rõ   r÷   )r   r`   r;   r<   r   r   r   rT   Ÿ  s
    
z"SafeChildWatcher.add_child_handlerc             C   s&   y| j |= dS  tk
r    dS X d S )NTF)rõ   rA   )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Æ   yt  |t j¡\}}W n( tk
r>   |}d}t d|¡ Y n0X |dkrLd S |  |¡}| j ¡ rnt 	d||¡ y| j
 |¡\}}W n. tk
r°   | j ¡ r¬tjd|dd Y nX |||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Ì   )rf   ÚwaitpidÚWNOHANGÚChildProcessErrorr   rÛ   r  r¢   rÁ   rÏ   rõ   ÚpoprA   )r   rö   r`   r   ra   r;   r<   r   r   r   r÷   ¶  s*    


zSafeChildWatcher._do_waitpid)r   rž   rŸ   r    rï   ró   rT   rí   rø   r÷   r    r   r   )r   r   r   Š  s   r   c                   sP   e 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	‡  Z
S )r   c                s$   t ƒ  ¡  t ¡ | _i | _d| _d S )Nr   )r   r   Ú	threadingZLockÚ_lockÚ_zombiesÚ_forks)r   )r   r   r   r   ã  s    

zFastChildWatcher.__init__c                s"   | j  ¡  | j ¡  tƒ  ¡  d S )N)rõ   r(   r	  r   r    )r   )r   r   r   r    é  s    

zFastChildWatcher.closec          	   C   s$   | j  |  jd7  _| S Q R X d S )Nr   )r  r
  )r   r   r   r   rï   î  s    zFastChildWatcher.__enter__c          	   C   sT   | j 8 |  jd8  _| js"| js&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	  r6   r(   r   rÛ   )r   rð   rñ   rò   Zcollateral_victimsr   r   r   ró   ô  s    
zFastChildWatcher.__exit__c          	   G   sl   | j d krtdƒ‚| j: y| j |¡}W n" tk
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¢   r5   r  r	  r  rA   rõ   )r   r`   r;   r<   ra   r   r   r   rT     s    
z"FastChildWatcher.add_child_handlerc             C   s&   y| j |= dS  tk
r    dS X d S )NTF)rõ   rA   )r   r`   r   r   r   rí     s
    z%FastChildWatcher.remove_child_handlerc          	   C   sö   xðyt  dt j¡\}}W n tk
r,   d S X |dkr:d S |  |¡}| jv y| j |¡\}}W nB tk
r¢   | j	rš|| j
|< | j ¡ r˜t d||¡ wd }Y nX | j ¡ r¼t d||¡ W d Q R X |d krÞt d||¡ q|||f|žŽ  qW 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  rA   r
  r	  r¢   rÁ   r   rÏ   rÛ   )r   r`   r   ra   r;   r<   r   r   r   rø     s6    





z FastChildWatcher._do_waitpid_all)r   rž   rŸ   r   r    rï   ró   rT   rí   rø   r    r   r   )r   r   r   Ù  s   
r   c                   sD   e Zd ZeZ‡ fdd„Zdd„ Z‡ fdd„Zdd„ Zd	d
„ Z	‡  Z
S )Ú_UnixDefaultEventLoopPolicyc                s   t ƒ  ¡  d | _d S )N)r   r   Ú_watcher)r   )r   r   r   r   K  s    
z$_UnixDefaultEventLoopPolicy.__init__c          	   C   sH   t j8 | jd kr:tƒ | _tt ¡ tjƒr:| j | j	j
¡ W d Q R X d S )N)r   r  r  r   rF   r  Úcurrent_threadÚ_MainThreadrî   Ú_localr¢   )r   r   r   r   Ú_init_watcherO  s    
z)_UnixDefaultEventLoopPolicy._init_watcherc                s6   t ƒ  |¡ | jd k	r2tt ¡ tjƒr2| j |¡ d S )N)r   Úset_event_loopr  rF   r  r  r  rî   )r   rv   )r   r   r   r  W  s    
z*_UnixDefaultEventLoopPolicy.set_event_loopc             C   s   | j d kr|  ¡  | j S )N)r  r  )r   r   r   r   rQ   e  s    
z-_UnixDefaultEventLoopPolicy.get_child_watcherc             C   s   | j d k	r| j  ¡  || _ d S )N)r  r    )r   r]   r   r   r   Úset_child_watchero  s    

z-_UnixDefaultEventLoopPolicy.set_child_watcher)r   rž   rŸ   r   Z_loop_factoryr   r  r  rQ   r  r    r   r   )r   r   r  G  s   
r  ),r9   r„   rf   rº   r0   rh   rx   ræ   r!   r  r%   Ú r   r   r   r   r   r   r	   r
   r   Úlogr   Ú__all__rÕ   ÚImportErrorr   ZBaseSelectorEventLoopr   ZReadTransportrI   Z_FlowControlMixinZWriteTransportrO   ZBaseSubprocessTransportrS   r   rô   r   r   ZBaseDefaultEventLoopPolicyr  r   r   r   r   r   r   Ú<module>   sN   
  |~ JF=On3