U
e5dۿ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z ddl m
Z
ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dZe jdkr�ed��dd� Z G dd� dej �Z G dd� dej �Z G dd� dej!ej"�Z#G dd� dej$�Z%G dd � d �Z&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&�Z+G d'd(� d(e&�Z,G d)d*� d*ej-�Z.e Z/e.Z0dS )+� N� )� base_events)�base_subprocess)� constants)�
coroutines)�events)�
exceptions)�futures)�selector_events)�tasks)�
transports)�logger)�SelectorEventLoop�AbstractChildWatcher�SafeChildWatcher�FastChildWatcher�MultiLoopChildWatcher�ThreadedChildWatcher�DefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s d S �N� )�signum�framer r �+/usr/lib64/python3.8/asyncio/unix_events.py�_sighandler_noop* s r c s� e Zd Zd(� fdd� Z� fdd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z d)dd�Z
d*dd�Z d+dd�Z dd� Z
d,ddddd�dd�Zd-ddddd d �d d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Z� ZS ).�_UnixSelectorEventLoopNc s t � �|� i | _d S r )�super�__init__�_signal_handlers)�self�selector�� __class__r r r 5 s z _UnixSelectorEventLoop.__init__c sZ t � �� t�� s.t| j�D ]}| �|� q n(| jrVtjd| �d�t | d� | j�
� 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 �sigr! r r r% 9 s
�z _UnixSelectorEventLoop.closec C s |D ]}|sq| � |� qd S r )�_handle_signal)r �datar r r r �_process_self_dataG s z)_UnixSelectorEventLoop._process_self_datac
G sL t �|�st �|�r td��| �|� | �� zt�| j� � � W n2 t
t fk
rt } zt t
|���W 5 d }~X Y nX t�||| d �}|| j|< z t�|t� t�|d� W n� t k
�rF } zz| j|= | j�szt�d� W n4 t
t fk
�r } zt�d|� W 5 d }~X Y nX |jtjk�r4t d|� d���n� W 5 d }~X Y nX d S )Nz3coroutines cannot be used with add_signal_handler()F������ set_wakeup_fd(-1) failed: %s�sig � 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)r r/ �callback�args�exc�handleZnexcr r r �add_signal_handlerN s2
�
z)_UnixSelectorEventLoop.add_signal_handlerc C s8 | j �|�}|d krd S |jr*| �|� n
| �|� d S r )r �getZ
_cancelledr) Z_add_callback_signalsafe)r r/ rG r r r r0 { s z%_UnixSelectorEventLoop._handle_signalc
C s� | � |� z | j|= W n tk
r, Y dS X |tjkr@tj}ntj}zt�||� W nB tk
r� } z$|jtj kr�t
d|� d���n� W 5 d }~X Y nX | js�zt� d� W n2 t tfk
r� } zt
�d|� W 5 d }~X Y nX dS )NFr5 r6 r3 r4 T)r8 r �KeyErrorr9 �SIGINT�default_int_handler�SIG_DFLr= rB rC r> r: r< r
rA )r r/ �handlerrF r r r r) � s(
z,_UnixSelectorEventLoop.remove_signal_handlerc C s6 t |t�std|����|t�� kr2td|� ���d S )Nzsig must be an int, not zinvalid signal number )�
isinstance�intr7 r9 �
valid_signalsr< r. r r r r8 � s
z$_UnixSelectorEventLoop._check_signalc C s t | ||||�S r )�_UnixReadPipeTransport�r �pipe�protocol�waiter�extrar r r �_make_read_pipe_transport� s z0_UnixSelectorEventLoop._make_read_pipe_transportc C s t | ||||�S r )�_UnixWritePipeTransportrS r r r �_make_write_pipe_transport� s z1_UnixSelectorEventLoop._make_write_pipe_transportc
� s� t �� ��}
|
�� std��| �� } t| |||||||f| |d�| ��} |
�| �� | j| � z| I d H W nD t t
fk
r� � Y n, t k
r� | � � | �
� I d H � Y nX W 5 Q R X | S )NzRasyncio.get_child_watcher() is not activated, subprocess support is not installed.)rV rW )r �get_child_watcher� is_activer> �
create_future�_UnixSubprocessTransport�add_child_handlerZget_pid�_child_watcher_callback�
SystemExit�KeyboardInterrupt�
BaseExceptionr% Z_wait)
r rU rE �shell�stdin�stdout�stderr�bufsizerW �kwargs�watcherrV �transpr r r �_make_subprocess_transport� s8
� ��
�z1_UnixSelectorEventLoop._make_subprocess_transportc C s | � |j|� d S r )�call_soon_threadsafeZ_process_exited)r �pid�
returncoderk r r r r` � s z._UnixSelectorEventLoop._child_watcher_callback)�ssl�sock�server_hostname�ssl_handshake_timeoutc � s
|r|d kr6t d��n |d k r&t d��|d k r6t d��|d k r�|d k rNt d��t�|�}t�tjtjd�}z |�d� | �||�I d H W q� |�� � Y q�X n@|d kr�t d��|j tjks�|j
tjkr�t d|����|�d� | j |||||d �I d H \}}||fS )
Nz/you have to pass server_hostname when using sslz+server_hostname is only meaningful with ssl�1ssl_handshake_timeout is only meaningful with ssl�3path and sock can not be specified at the same timer Fz no path and sock were specified�.A UNIX Domain Stream Socket was expected, got )rs ) r< �os�fspath�socket�AF_UNIX� SOCK_STREAM� setblockingZ sock_connectr% �family�typeZ _create_connection_transport) r �protocol_factory�pathrp rq rr rs � transportrU r r r �create_unix_connection� sR ���
��
�z-_UnixSelectorEventLoop.create_unix_connection�d T)rq �backlogrp rs �
start_servingc
� s� t |t�rtd��|d k r&|s&td��|d k �rH|d k r@td��t�|�}t�tjtj�}|d dkr�z t �
t� |�j �r�t� |� W nB t
k
r� Y n0 tk
r� } zt�d||� W 5 d }~X Y nX z|�|� W nl tk
�r0 } z8|�� | jtjk�r d|�d�}
ttj|
�d �n� W 5 d } ~ X Y n |�� � Y nX n<|d k�rZtd ��|jtjk�sv|jtjk�r�td
|����|�d � t�| |g||||�} |�r�| �� tj d| d �I d H | S )
Nz*ssl argument must be an SSLContext or Nonert ru r )r � z2Unable to check or remove stale UNIX socket %r: %rzAddress z is already in usez-path was not specified, and no sock specifiedrv F)�loop) rO �boolr7 r< rw rx ry rz r{ �stat�S_ISSOCK�st_mode�remove�FileNotFoundErrorr= r
�errorZbindr% rB Z
EADDRINUSEr} r~ r| r ZServerZ_start_servingr �sleep) r r r� rq r� rp rs r� �errrF �msgZserverr r r �create_unix_server sn
�
�
�
�
��
�z)_UnixSelectorEventLoop.create_unix_serverc
� s� z
t j W n, tk
r6 } zt�d��W 5 d }~X Y nX z |�� }W n2 ttjfk
rv } zt�d��W 5 d }~X Y nX zt �|�j }W n, t
k
r� } zt�d��W 5 d }~X Y nX |r�|n|} | s�dS | � � }
| � |
d ||||| d� |
I d H S )Nz os.sendfile() is not availableznot a regular filer )
rw �sendfile�AttributeErrorr �SendfileNotAvailableErrorr; �io�UnsupportedOperation�fstat�st_sizer= r] �_sock_sendfile_native_impl) r rq �file�offset�countrF r; r� Zfsize� blocksize�futr r r �_sock_sendfile_nativeJ s2
� �z,_UnixSelectorEventLoop._sock_sendfile_nativec C s, |� � } |d k r| �|� |�� r4| �|||� d S |rd|| }|dkrd| �|||� |�|� d S zt�| |||�}
W �nD ttfk
r� |d kr�| � ||� | �
| | j || ||||||�
Y �nb t k
�rj } z�|d k �r| j
t
jk�rt| �tk �rtdt
j�} | | _| } |dk�rBt�d�}
| �|||� |�|
� n| �|||� |�| � W 5 d } ~ X Y n� ttfk
�r� � Y n� tk
�r� } z | �|||� |�| � W 5 d } ~ X Y njX |
dk�r�| �|||� |�|� nD||
7 }||
7 }|d k�r
| � ||� | �
| | j || ||||||�
d S )Nr zsocket is not connectedzos.sendfile call failed)r; �
remove_writer� cancelled� _sock_sendfile_update_fileposZ
set_resultrw r� �BlockingIOError�InterruptedError� _sock_add_cancellation_callbackZ
add_writerr� r= rB ZENOTCONNr~ �ConnectionError� __cause__r r� Z
set_exceptionra rb rc )r r� Z
registered_fdrq r; r� r� r� �
total_sent�fdZsentrF �new_excr� r r r r� a s�
�
�
� �
�
�z1_UnixSelectorEventLoop._sock_sendfile_native_implc C s |dkrt �||t j� d S �Nr )rw �lseek�SEEK_SET)r r; r� r� r r r r� � s z4_UnixSelectorEventLoop._sock_sendfile_update_fileposc s � �fdd�}|� |� d S )Nc s& | � � r"��� }|dkr"� �|� d S )Nr3 )r� r; r� )r� r� �r rq r r �cb� s zB_UnixSelectorEventLoop._sock_add_cancellation_callback..cb)Zadd_done_callback)r r� rq r� r r� r r� � s z6_UnixSelectorEventLoop._sock_add_cancellation_callback)N)NN)NN)N)N)N)�__name__�
__module__� __qualname__r r% r2 rH r0 r) r8 rX rZ rl r` r� r� r� r� r� r� �
__classcell__r r r! r r / sF -
�
�
�
� � . � � CFr c s� e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
jfdd�Zd dd�Zdd� Zd d � Z� ZS ) rR i Nc s� t � �|� || jd< || _|| _|�� | _|| _d| _d| _ t
� | j�j }t
�|�s�t
�|�s�t
�|�s�d | _d | _d | _td��t
�| jd� | j�| jj| � | j�| jj| j| j� |d k r�| j�tj|d � d S )NrT Fz)Pipe transport is for pipes/sockets only.)r r �_extra�_loop�_piper; �_fileno� _protocol�_closing�_pausedrw r� r� r� �S_ISFIFOr� �S_ISCHRr< � set_blocking� call_soon�connection_made� _add_reader� _read_readyr � _set_result_unless_cancelled)r r� rT rU rV rW �moder! r r r � s:
�� �
�z _UnixReadPipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� |�d| j� �� t| jdd �}| jd k r�|d k r�t� || jt
j �}|r�|�d� q�|�d� n | jd k r�|�d� n
|�d� d� d �
|��S )
N�closed�closing�fd=� _selector�polling�idle�open�<{}>� )r" r� r� �appendr� r� �getattrr� r
�_test_selector_event� selectorsZ
EVENT_READ�format�join)r rA r r� r r r �__repr__� s(
�
z _UnixReadPipeTransport.__repr__c
C s� zt �| j| j�}W nD ttfk
r, Y n� tk
rX } z| �|d� W 5 d }~X Y n^X |rl| j� |� nJ| j
� � r�t �
d| � d| _| j
�| j� | j
�| jj� | j
�| jd � d S )Nz"Fatal read error on pipe transport�%r was closed by peerT)rw �readr� �max_sizer� r� r= � _fatal_errorr� Z
data_receivedr� � get_debugr
rA r� �_remove_readerr� Z eof_received�_call_connection_lost)r r1 rF r r r r� � s
z"_UnixReadPipeTransport._read_readyc C s> | j s | jrd S d| _| j�| j� | j�� r:t�d| � d S )NTz%r pauses reading)r� r� r� r� r� r� r
�debug�r r r r �
pause_reading� s
z$_UnixReadPipeTransport.pause_readingc C sB | j s | jsd S d| _| j�| j| j� | j�� r>t�d| � d S )NFz%r resumes reading) r� r� r� r� r� r� r� r
r� r� r r r �resume_reading s
z%_UnixReadPipeTransport.resume_readingc C s
|| _ d S r �r� �r rU r r r � set_protocol
s z#_UnixReadPipeTransport.set_protocolc C s | j S r r� r� r r r � get_protocol s z#_UnixReadPipeTransport.get_protocolc C s | j S r �r� r� r r r �
is_closing s z!_UnixReadPipeTransport.is_closingc C s | j s| �d � d S r )r� �_closer� r r r r% s z _UnixReadPipeTransport.closec C s, | j d k r(|d| ��t| d� | j �� d S �Nzunclosed transport r# �r� r, r% �r �_warnr r r �__del__ s
z _UnixReadPipeTransport.__del__� Fatal error on pipe transportc C sZ t |t�r4|jtjkr4| j�� rLtjd| |dd� n| j�||| | j d�� | �
|� d S �Nz%r: %sT��exc_info)�message� exceptionr� rU ) rO r= rB ZEIOr� r� r
r� �call_exception_handlerr� r� �r rF 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 rF r r r r� - s z _UnixReadPipeTransport._closec C s4 z| j�|� W 5 | j �� d | _ d | _d | _X d S r �r� r% r� r� Zconnection_lostr� r r r r� 2 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 r! r rR � s
rR c s� e Zd Zd%� fdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zejfdd�Zd d � Zd&d d �Zd'd!d"�Zd#d$� Z� ZS )(rY Nc
s� t � �||� || jd< || _|�� | _|| _t� | _d| _ d| _
t � | j�j
}t�|�}t�|�}t�|�} |s�|s�| s�d | _d | _d | _td��t �| jd� | j�| jj| � | s�|r�tj�d�s�| j�| jj| j| j� |d k r�| j�t j |d � d S )NrT r Fz?Pipe transport is only for pipes, sockets and character devicesZaix) r r r� r� r; r� r� � bytearray�_buffer�
_conn_lostr� rw r� r� r� r� r� r� r< r� r� r� r� r& �platform�
startswithr� r� r r� )
r r� rT rU rV rW r� Zis_charZis_fifoZ is_socketr! r r r ? s:
�
�z _UnixWritePipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� |�d| j� �� t| jdd �}| jd k r�|d k r�t� || jt
j �}|r�|�d� n
|�d� | � � }|�d|� �� n | jd k r�|�d� n
|�d� d �
d
�|��S ) Nr� r� r� r� r� r� zbufsize=r� r� r� )r" r� r� r� r� r� r� r� r
r� r� Z EVENT_WRITE�get_write_buffer_sizer� r� )r rA r r� rh r r r r� d s,
�
z _UnixWritePipeTransport.__repr__c C s
t | j�S r )�lenr� r� r r r r� | s z-_UnixWritePipeTransport.get_write_buffer_sizec C s6 | j �� rt�d| � | jr*| �t� � n| �� d S )Nr� )r� r� r
rA r� r� �BrokenPipeErrorr� r r r r� s
z#_UnixWritePipeTransport._read_readyc
C s4 t |t�rt|�}|sd S | js&| jrN| jtjkr |}d}t�d|� Y n.X |dkrLd S t|�}| j�� rlt� d||� z| j
� |�\}}W n. t k
r� | j�� r�tjd|dd� Y nX |||f|�� d S )N�� �8Unknown child process pid %d, will report returncode 255r �$process %s exited with returncode %s�'Child watcher got an unexpected pid: %rTr� )
rw �waitpid�WNOHANG�ChildProcessErrorr
r r# r� r� r� r% �poprJ )r r& rn r" ro rD rE r r r r' � s4 �
�
� z SafeChildWatcher._do_waitpid) r� r� r� r% r r r_ r r( r' r� r r r! r r � s r c sP e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )r c s$ t � �� t�� | _i | _d| _d S r� )r r � threadingZLock�_lock�_zombies�_forksr� r! r r r s
zFastChildWatcher.__init__c s" | j �� | j�� t� �� d S r )r% r- r: r r% r� r! r r r% s
zFastChildWatcher.closec
C s0 | j � | jd7 _| W 5 Q R � S Q R X d S )Nr )r9 r; r� r r r r s zFastChildWatcher.__enter__c C s^ | j �B | jd8 _| js"| js0W 5 Q R � d S t| j�}| j�� W 5 Q R X t�d|� d S )Nr z5Caught subprocesses termination from unknown pids: %s)r9 r; r: r? r- r
r )r r r r Zcollateral_victimsr r r r s
�zFastChildWatcher.__exit__c G sf | j �F z| j�|�}W n. tk
rF ||f| j|< Y W 5 Q R � d S X W 5 Q R X |||f|�� d S r )r9 r: r7 rJ r% )r rn rD rE ro r r r r_ ' s z"FastChildWatcher.add_child_handlerc C s* z| j |= W dS tk
r$ Y dS X d S r- r. r r r r r 5 s
z%FastChildWatcher.remove_child_handlerc C s� zt �dt j�\}}W n tk
r, Y d S X |dkr:d S t|�}| j�� z| j�|�\}}W nN tk
r� | j r�|| j
|< | j � � r�t
�d||� Y W 5 Q R � q d }Y nX | j � � r�t
�d||� W 5 Q R X |d kr�t
�d||� q |||f|�� q d S )Nr3 r z,unknown process %s exited with returncode %sr2 z8Caught subprocess termination from unknown pid: %d -> %d)rw r4 r5 r6 r# r9 r% r7 rJ r; r: r� r� r
r� r )r rn r" ro rD rE r r r r( < s@
�
� �z FastChildWatcher._do_waitpid_all) r� r� r� r r% r r r_ r r( r� r r r! r r � s
r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )r c C s i | _ d | _d S r )r% �_saved_sighandlerr� r r r r z s z MultiLoopChildWatcher.__init__c C s
| j d k S r )r< r� r r r r\ ~ s z MultiLoopChildWatcher.is_activec C sT | j �� | jd krd S t�tj�}|| jkr:t�d� nt�tj| j� d | _d S )Nz+SIGCHLD handler was changed by outside code) r% r- r< r9 � getsignalr* r+ r
r )r rN r r r r% � s
zMultiLoopChildWatcher.closec C s | S r r r� r r r r � s z MultiLoopChildWatcher.__enter__c C s d S r r �r �exc_typeZexc_valZexc_tbr r r r � s z MultiLoopChildWatcher.__exit__c G s& t �� }|||f| j|< | �|� d S r )r �get_running_loopr% r' )r rn rD rE r� r r r r_ � s z'MultiLoopChildWatcher.add_child_handlerc C s* z| j |= W dS tk
r$ Y dS X d S r- r. r r r r r � s
z*MultiLoopChildWatcher.remove_child_handlerc C sN | j d k rd S t�tj| j�| _ | j d kr� s �z6ThreadedChildWatcher._join_threads..)r( rE �valuesr� )r �threadsrK r r r rF � s z"ThreadedChildWatcher._join_threadsc C s | S r r r� r r r r s z ThreadedChildWatcher.__enter__c C s d S r r r> r r r r s z ThreadedChildWatcher.__exit__c C s6 dd� t | j�� �D �}|r2|| j� d�t| d� d S )Nc S s g | ]}|� � r|�qS r )rG rI r r r rL s �z0ThreadedChildWatcher.__del__..z0 has registered but not finished child processesr# )r( rE rM r" r, )r r� rN r r r r� s �z ThreadedChildWatcher.__del__c G sF t �� }tj| jdt| j�� �||||fdd�}|| j|< |�� d S )Nzwaitpid-T)�target�namerE rH ) r r@ r8 ZThreadr' �nextrD rE �start)r rn rD rE r� rK r r r r_ s
�
z&ThreadedChildWatcher.add_child_handlerc C s dS r� r r r r r r s z)ThreadedChildWatcher.remove_child_handlerc C s d S r r r r r r r s z ThreadedChildWatcher.attach_loopc C s� zt �|d�\}}W n( tk
r< |}d}t�d|� Y n X t|�}|�� r\t�d||� |�� rtt�d||� n|j |||f|�� | j
� |� d S )Nr r0 r1 r2 rA ) rw r4 r6 r
r r# r� r� rB rm rE r7 )r r� r& rD rE rn r" ro r r r r' " s& �
�z ThreadedChildWatcher._do_waitpidN)r� r� r� r r\ r% rF r r r* r+ r� r_ r r r' r r r r r � s
r c sD e Zd ZeZ� fdd�Zdd� Z� fdd�Zdd� Zd d
� Z � Z
S ) �_UnixDefaultEventLoopPolicyc s t � �� d | _d S r )r r �_watcherr� r! r r r A s
z$_UnixDefaultEventLoopPolicy.__init__c C sH t j�8 | jd kr:t� | _tt�� tj�r:| j�| j j
� W 5 Q R X d S r ) r r9 rT r rO r8 �current_thread� _MainThreadr �_localr� r� r r r �
_init_watcherE s
�z)_UnixDefaultEventLoopPolicy._init_watcherc s6 t � �|� | jd k r2tt�� tj�r2| j�|� d S r )r �set_event_looprT rO r8 rU rV r r r! r r rY M s
�z*_UnixDefaultEventLoopPolicy.set_event_loopc C s | j d kr| �� | j S r )rT rX r� r r r r[ [ s
z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s | j d k r| j �� || _ d S r )rT r% )r rj r r r �set_child_watchere s
z-_UnixDefaultEventLoopPolicy.set_child_watcher) r� r� r� r Z
_loop_factoryr rX rY r[ rZ r� r r r! r rS = s
rS )1rB r� rC rw r� r9 ry r� r r& r8 r* � r r r r r r r r
r r �logr
�__all__r� � ImportErrorr ZBaseSelectorEventLoopr Z
ReadTransportrR Z_FlowControlMixinZWriteTransportrY ZBaseSubprocessTransportr^ r r# r$ r r r r ZBaseDefaultEventLoopPolicyrS r r r r r r � s^
� NO5Ji}Y3