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 dddddgZejdkr�ed��dd� 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$��r�ej(Z)nd dl"Z"d%d&� Z)G d'd(� d(e j*�Z+G d)d� d�Z,G d*d+� d+e,�Z-G d,d� de-�Z.G d-d� de-�Z/G d.d/� d/ej0�Z1e Z2e1Z3dS )0� N� )� base_events)�base_subprocess)�compat)� constants)�
coroutines)�events)�futures)�selector_events)� selectors)�
transports)� coroutine)�logger�SelectorEventLoop�AbstractChildWatcher�SafeChildWatcher�FastChildWatcher�DefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s d S )N� )�signum�framer 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� fdd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
d"dd�Z d#dd�Z e
d$dd��Zdd� Ze
dddd�dd ��Ze
d%dd dd �d d ��Z� ZS )&�_UnixSelectorEventLoopNc s t � j|� i | _d S )N)�super�__init__�_signal_handlers)�self�selector)� __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| j�D ]}| j|� q W n(| jrZtjd| �d�t | d� | jj
� d S )NzClosing 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% > 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|= | j�sytjd� W n4 t
t fk
�r } ztjd|� W Y d d }~X nX |jtjk�r0t 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_closed�signal�
set_wakeup_fdZ_csock�fileno�
ValueError�OSError� RuntimeError�strr ZHandler r � siginterruptr �info�errno�EINVAL�format)r r. �callback�args�exc�handleZnexcr 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 kr�t
dj |���n� W Y d d }~X nX | js�ytj 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 �SIGINT�default_int_handler�SIG_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 |t�stdj|���d| ko,tjk n sDtdj|tj���d S )Nz sig must be an int, not {!r}r zsig {} out of range(1, {}))�
isinstance�intr3 r@ r5 �NSIGr8 )r r. r r r r4 � 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 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�_UnixSubprocessTransport�add_child_handlerZget_pid�_child_watcher_callback� Exceptionr% Z_wait)r rP rB �shell�stdin�stdout�stderr�bufsizerR �kwargs�watcherrQ �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)�ssl�sock�server_hostnamec c s� |r|d kr&t d��n|d k r&t d��|d k r�|d k r>t d��tjtjtjd�}y |jd� | j||�E d H W q� |j� � Y q�X nB|d kr�t d��|jtjks�tj |j
� r�t 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_socket�typer@ Z _create_connection_transport)r �protocol_factoryr rg rh ri � transportrP r r r �create_unix_connection� s8
z-_UnixSelectorEventLoop.create_unix_connection�d )rh �backlogrg c
! C s� t |t�rtd��|d k �r0|d k r,td��t|�}tjtjtj�}|d d
kr�y 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� |jtjk�rdj|�}ttj|�d �n� W Y d d }~X n |j� � Y nX n>|d k�rBtd��|jtjk�s`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 �stat�S_ISSOCK�os�st_mode�remove�FileNotFoundErrorr9 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!dd�Zdd � Zd d � Z� ZS )"rN � i Nc s� t � j|� || jd< || _|| _|j� | _|| _d| _t j
| j�j }t j
|�pbt j|�pbt j|�s~d | _d | _d | _td��t| 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_soon�connection_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 r�tj || jt
j �}|r�|jd� q�|jd� n | jd k r�|jd� n
|jd� dd j |� S )
N�closed�closingzfd=%s� _selector�polling�idle�openz<%s>� )
r! r� r� �appendr� r� �getattrr� r
�_test_selector_eventr Z
EVENT_READ�join)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 � r�t 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_size�BlockingIOError�InterruptedErrorr9 � _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 |t�r4|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 �debug�call_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 | j�j
}tj|�}tj|�}tj|�} |px|px| s�d | _d | _d | _td��t| j� | jj| jj| � | s�|r�tjjd� 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 devices�aix) 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 r�tj || 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_WRITE�get_write_buffer_sizer� )r r= r r� r_ r r r r� � s(
z _UnixWritePipeTransport.__repr__c C s
t | j�S )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 |t�rt|�}|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 |tj�r*| 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_flag�oldr 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) �
subprocess�PIPEr� 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 �a�b�cr 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 � 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� s r� c sL e Zd Z� fdd�Zdd� 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 | j�D ]}| 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� r�tjd|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 �waitpid�WNOHANG�ChildProcessErrorr 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� fdd�Z� fdd�Zdd� 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� x�yt jdt j�\}}W n tk
r, d S X |dkr:d S | j|�}| j�v y| jj|�\}}W nB tk
r� | j r�|| j
|< | j j � r�t
jd||� wd }Y nX | j j � r�t
jd||� W d Q R X |d kr�t
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� fdd�Zdd� Z� fdd�Zdd� 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 j�8 | jd kr:t� | _ttj� tj�r:| 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� tj�r2| 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