B
�/g^5d c @ s d Z dZddlZddlZddlZddlZddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd lm
Z
dd lmZ G d
d� de
je
j�ZG dd� dee
j�ZG dd� dee
j�ZG dd� de�ZG dd� deee
j�ZG dd� deee
j�ZG dd� dej�Z dS )z�Event loop using a proactor and related classes.
A proactor is a "notify-on-completion" multiplexer. Currently a
proactor is only implemented on Windows with IOCP.
)�BaseProactorEventLoop� N� )� base_events)� constants)�events)�futures)� protocols)�sslproto)�
transports)�loggerc s| e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z d dd�Z dd� Z
dd� Zdd� Z� ZS ) �_ProactorBasePipeTransportz*Base class for pipe and socket transports.Nc s� t � �||� | �|� || _| �|� || _d | _d | _d | _d| _ d| _
d| _ d| _ | jd k rl| j�
� | j�| jj| � |d k r�| j�tj|d � d S )Nr F)�super�__init__�
_set_extra�_sock� set_protocol�_server�_buffer� _read_fut�
_write_fut�_pending_write�
_conn_lost�_closing� _eof_writtenZ_attach�_loop� call_soon� _protocolZconnection_mader Z _set_result_unless_cancelled��self�loop�sock�protocol�waiter�extra�server�� __class__� �//usr/lib64/python3.7/asyncio/proactor_events.pyr
s$
z#_ProactorBasePipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� | jd k rP|�d| j�� � �� | jd k rl|�d| j��� | jd k r�|�d| j��� | jr�|�dt | j�� �� | j
r�|�d� d� d � |��S )
NZclosedZclosingzfd=zread=zwrite=zwrite_bufsize=z EOF writtenz<{}>z )
r% �__name__r Zappendr �filenor r r �lenr ZformatZjoin)r Zinfor& r& r' �__repr__2 s
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )NZpipe)�_extra�r r r& r& r' r D � z%_ProactorBasePipeTransport._set_extrac C s
|| _ d S �N�r )r r r& r& r' r G r. z'_ProactorBasePipeTransport.set_protocolc C � | j S r/ r0 �r r& r& r' � get_protocolJ r. z'_ProactorBasePipeTransport.get_protocolc C r1 r/ )r r2 r& r& r' �
is_closingM r. z%_ProactorBasePipeTransport.is_closingc C s\ | j r
d S d| _ | jd7 _| js>| jd kr>| j�| jd � | jd k rX| j�� d | _d S )NTr ) r r r r r r �_call_connection_lostr �cancelr2 r& r& r' �closeP s
z _ProactorBasePipeTransport.closec C s, | j d k r(tjd| ��t| d� | �� d S )Nzunclosed transport )Zsource)r �warningsZwarnZResourceWarningr7 r2 r& r& r' �__del__[ s
z"_ProactorBasePipeTransport.__del__� Fatal error on pipe transportc C sV zDt |t�r*| j�� rBtjd| |dd� n| j�||| | jd�� W d | �|� X d S )Nz%r: %sT�Zexc_info)�message� exceptionZ transportr ) �
isinstance�OSErrorr � get_debugr
�debug�call_exception_handlerr � _force_close)r �excr< r& r& r' � _fatal_errora s
z'_ProactorBasePipeTransport._fatal_errorc C s� | j d k r6| j �� s6|d kr*| j �d � n | j �|� | jr@d S d| _| jd7 _| jrj| j�� d | _| jr�| j�� d | _d| _ d | _
| j � | j
|� d S )NTr r )�
_empty_waiter�done�
set_resultZ
set_exceptionr r r r6 r r r r r r5 )r rD r& r& r' rC p s"
z'_ProactorBasePipeTransport._force_closec C s^ z| j �|� W d t| jd�r,| j�tj� | j�� d | _| j}|d k rX|� � d | _X d S )N�shutdown)
r Zconnection_lostZhasattrr rI �socketZ SHUT_RDWRr7 r Z_detach)r rD r# r& r& r' r5 � s
z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd k r |t| j�7 }|S r/ )r r r* )r Zsizer& r& r' �get_write_buffer_size� s
z0_ProactorBasePipeTransport.get_write_buffer_size�NNN)r: )r( �
__module__� __qualname__�__doc__r
r+ r r r3 r4 r7 r9 rE rC r5 rK �
__classcell__r& r& r$ r' r s
r c sT e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
� Z S )�_ProactorReadPipeTransportzTransport for read pipes.Nc s: d | _ d| _t� �||||||� | j�| j� d| _d S )NTF)�
_pending_data�_pausedr r
r r �
_loop_readingr r$ r& r' r
� s
z#_ProactorReadPipeTransport.__init__c C s | j o| j S r/ )rS r r2 r& r& r' �
is_reading� r. z%_ProactorReadPipeTransport.is_readingc C s0 | j s | jrd S d| _| j�� r,t�d| � d S )NTz%r pauses reading)r rS r r@ r
rA r2 r& r& r' �
pause_reading� s
z(_ProactorReadPipeTransport.pause_readingc C sn | j s | jsd S d| _| jd kr0| j�| jd � | j}d | _|d k rT| j�| j|� | j�� rjt �
d| � d S )NFz%r resumes reading) r rS r r r rT rR �_data_receivedr@ r
rA �r �datar& r& r' �resume_reading� s
z)_ProactorReadPipeTransport.resume_readingc
C sd | j �� rt�d| � y| j�� }W n. tk
rR } z| �|d� d S d }~X Y nX |s`| �� d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.) r r@ r
rA r Z eof_received� ExceptionrE r7 )r Z keep_openrD r& r& r' �
_eof_received� s
z(_ProactorReadPipeTransport._eof_receivedc
C s� | j r|| _d S |s | �� d S t| jtj�rryt�| j|� W q~ tk
rn } z| � |d� d S d }~X Y q~X n | j�
|� d S )Nz3Fatal error: protocol.buffer_updated() call failed.) rS rR r\ r> r r ZBufferedProtocolZ _feed_data_to_buffered_protor[ rE Z
data_received)r rY rD r& r& r' rW � s z)_ProactorReadPipeTransport._data_receivedc
C sd d }�zBy`|d k r2d | _ |�� r*|�� }n|�� | jr@d }d S |dkrLd S | jsf| jj�| j d�| _ W n� t
k
r� } z0| js�| � |d� n| j� � r�t
jddd� W d d }~X Y n� tk
r� } z| �|� W d d }~X Y nf tk
�r } z| � |d� W d d }~X Y n8 tjk
�r. | j�s*� Y nX | j�sF| j �| j� W d |d k �r^| �|� X d S )Ns i � z"Fatal read error on pipe transportz*Read error on pipe transport while closingTr; )r rG �resultr6 r rS r � _proactor�recvr ZConnectionAbortedErrorrE r@ r
rA �ConnectionResetErrorrC r? r �CancelledError�add_done_callbackrT rW )r �futrY rD r& r& r' rT � s>
z(_ProactorReadPipeTransport._loop_readingrL r/ ) r( rM rN rO r
rU rV rZ r\ rW rT rP r& r& r$ r' rQ � s rQ c s^ e Zd ZdZdZ� fdd�Zdd� Zddd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z � Z
S )� _ProactorBaseWritePipeTransportzTransport for write pipes.Tc s t � j||� d | _d S r/ )r r
rF �r ZargsZkwr$ r& r' r
. s z(_ProactorBaseWritePipeTransport.__init__c C s� t |tttf�s$tdt|�j� ���| jr2td��| j d k rDtd��|sLd S | j
rz| j
t j krht
�d� | j
d7 _
d S | jd kr�| jt|�d� n.| js�t|�| _| �� n| j�|� | �� d S )Nz/data argument must be a bytes-like object, not zwrite_eof() already calledz(unable to write; sendfile is in progressz socket.send() raised exception.r )rY )r> ZbytesZ bytearrayZ
memoryviewZ TypeErrorZtyper( r � RuntimeErrorrF r r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr
�warningr �
_loop_writingr �_maybe_pause_protocolZextendrX r& r& r' �write2 s*
z%_ProactorBaseWritePipeTransport.writeNc
C sT y�|d k r | j d kr | jr d S d | _ d| _|r6|�� |d krJ| j}d | _|s�| jrd| j�| jd � | jrx| j �
t j � | �
� nN| jj�| j |�| _ | j �� s�t|�| _| j �| j� | �� n| j �| j� | jd k r�| j d kr�| j�d � W n\ tk
�r } z| �|� W d d }~X Y n0 tk
�rN } z| �|d� W d d }~X Y nX d S )Nr z#Fatal write error on pipe transport)r r r r] r r r r5 r r rI rJ �SHUT_WRZ_maybe_resume_protocolr^ �sendrG r* rb rh ri rF rH r` rC r? rE )r �frY rD r& r& r' rh X s8
z-_ProactorBaseWritePipeTransport._loop_writingc C � dS �NTr& r2 r& r& r' �
can_write_eof� r. z-_ProactorBaseWritePipeTransport.can_write_eofc C s | � � d S r/ )r7 r2 r& r& r' � write_eof� r. z)_ProactorBaseWritePipeTransport.write_eofc C s | � d � d S r/ )rC r2 r& r& r' �abort� r. z%_ProactorBaseWritePipeTransport.abortc C s: | j d k rtd��| j�� | _ | jd kr4| j �d � | j S )NzEmpty waiter is already set)rF rf r Z
create_futurer rH r2 r& r& r' �_make_empty_waiter� s
z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s
d | _ d S r/ )rF r2 r& r& r' �_reset_empty_waiter� r. z3_ProactorBaseWritePipeTransport._reset_empty_waiter�NN)r( rM rN rO Z_start_tls_compatibler
rj rh rp rq rr rs rt rP r& r& r$ r' rd ( s &
)rd c s$ e Zd Z� fdd�Zdd� Z� ZS )�_ProactorWritePipeTransportc s4 t � j||� | jj�| jd�| _| j�| j� d S )Ni ) r r
r r^ r_ r r rb � _pipe_closedre r$ r& r' r
� s z$_ProactorWritePipeTransport.__init__c C s@ |� � r d S | jrd S d | _| jd k r4| �t� � n| �� d S r/ )Z cancelledr r r rC ZBrokenPipeErrorr7 )r rc r& r& r' rw � s
z(_ProactorWritePipeTransport._pipe_closed)r( rM rN r
rw rP r& r& r$ r' rv � s rv c @ s e Zd ZdZdd� Zdd� ZdS )� _ProactorDuplexPipeTransportzTransport for duplex pipes.c C rn )NFr& r2 r& r& r' rp � r. z*_ProactorDuplexPipeTransport.can_write_eofc C s t �d S r/ )ZNotImplementedErrorr2 r& r& r' rq � r. z&_ProactorDuplexPipeTransport.write_eofN)r( rM rN rO rp rq r& r& r& r' rx � s rx c sB e Zd ZdZejjZd � fdd� Zdd� Z dd� Z
d d
� Z � Z S ) �_ProactorSocketTransportz Transport for connected sockets.Nc s$ t � �||||||� t�|� d S r/ )r r
r Z _set_nodelayr r$ r& r' r
� s z!_ProactorSocketTransport.__init__c C s� || j d< y|�� | j d< W n4 tjtfk
rP | j�� rLtjd|dd� Y nX d| j kr�y|� � | j d< W n4 tjtfk
r� | j�� r�tjd|dd� Y nX d S )NrJ Zsocknamezgetsockname() failed on %rTr; �peernamezgetpeername() failed on %r)
r, Z getsocknamerJ Zerror�AttributeErrorr r@ r
rg Z getpeernamer- r& r& r' r � s
z#_ProactorSocketTransport._set_extrac C rn ro r& r2 r& r& r' rp � r. z&_ProactorSocketTransport.can_write_eofc C s2 | j s | jrd S d| _| jd kr.| j�tj� d S ro )r r r r rI rJ rk r2 r& r& r' rq � s
z"_ProactorSocketTransport.write_eofrL )
r( rM rN rO r Z
_SendfileModeZ
TRY_NATIVEZ_sendfile_compatibler
r rp rq rP r& r& r$ r' ry � s ry c s� e Zd Z� fdd�Zd1dd�Zd2dddddd�dd �Zd3d
d �Zd4d d
�Zd5dd�Z� fdd�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd d � Zd d � Zd d!� Zd"d#� Zd6d$d%�Zd&d'� Zd7d)d*�Zd+d,� Zd-d.� Zd/d0� Z� ZS )8r c sH t � �� t�d|jj� || _|| _d | _i | _ |�
| � | � � d S )NzUsing proactor: %s) r r
r
rA r% r( r^ � _selector�_self_reading_future�_accept_futuresZset_loop�_make_self_pipe)r Zproactorr$ r& r' r
� s
z BaseProactorEventLoop.__init__Nc C s t | |||||�S r/ )ry )r r r r! r" r# r& r& r' �_make_socket_transport� s
z,BaseProactorEventLoop._make_socket_transportF)� server_side�server_hostnamer" r# �ssl_handshake_timeoutc C s0 t j| |||||| d�}
t| ||
||d� |
jS )N)r� �r" r# )r Z SSLProtocolry Z_app_transport) r Zrawsockr �
sslcontextr! r� r� r" r# r� Z ssl_protocolr& r& r' �_make_ssl_transport� s
z)BaseProactorEventLoop._make_ssl_transportc C � t | ||||�S r/ )rx �r r r r! r" r& r& r' �_make_duplex_pipe_transport� s z1BaseProactorEventLoop._make_duplex_pipe_transportc C r� r/ )rQ r� r& r& r' �_make_read_pipe_transport � z/BaseProactorEventLoop._make_read_pipe_transportc C r� r/ )rv r� r& r& r' �_make_write_pipe_transport s z0BaseProactorEventLoop._make_write_pipe_transportc sP | � � rtd��| �� r d S | �� | �� | j�� d | _d | _t� �� d S )Nz!Cannot close a running event loop) Z
is_runningrf � is_closed�_stop_accept_futures�_close_self_piper^ r7 r| r r2 r$ r& r' r7 s
zBaseProactorEventLoop.closec � � | j �||�I d H S r/ )r^ r_ )r r Znr& r& r' � sock_recv r. z BaseProactorEventLoop.sock_recvc � r� r/ )r^ Z recv_into)r r Zbufr& r& r' �sock_recv_into# r. z$BaseProactorEventLoop.sock_recv_intoc � r� r/ )r^ rl )r r rY r& r& r' � sock_sendall& r. z"BaseProactorEventLoop.sock_sendallc � r� r/ )r^ Zconnect)r r Zaddressr& r& r' � sock_connect) r. z"BaseProactorEventLoop.sock_connectc � s | j �|�I d H S r/ )r^ �acceptr- r& r& r' � sock_accept, r. z!BaseProactorEventLoop.sock_acceptc
� s( y |� � }W n2 ttjfk
r> } zt�d��W d d }~X Y nX yt�|�j}W n, t k
r| } zt�d��W d d }~X Y nX |r�|n|}|s�dS t
|d�}|r�t
|| |�n|} t
||�}d}
zLxFt
| | |�}|dkr�|
S | j � ||||�I d H ||7 }|
|7 }
q�W W d |
dk�r"|�
|� X d S )Nznot a regular filer l �� )r) r{ �ioZUnsupportedOperationr ZSendfileNotAvailableError�osZfstatZst_sizer? Zminr^ ZsendfileZseek) r r �file�offset�countr) ZerrZfsizeZ blocksizeZend_posZ
total_sentr& r& r' �_sock_sendfile_native/ s2
z+BaseProactorEventLoop._sock_sendfile_nativec � sV |� � }|�� |�� I d H z | j|j|||dd�I d H S |�� |rP|�� X d S )NF)Zfallback)rU rV rs Z
sock_sendfiler rt rZ )r Ztranspr� r� r� rZ r& r& r' �_sendfile_nativeL s z&BaseProactorEventLoop._sendfile_nativec C sL | j d k r| j �� d | _ | j�� d | _| j�� d | _| jd8 _d S )Nr )r} r6 �_ssockr7 �_csock�
_internal_fdsr2 r& r& r' r� X s
z&BaseProactorEventLoop._close_self_pipec C sF t �� \| _| _| j�d� | j�d� | jd7 _| �| j� d S )NFr )rJ Z
socketpairr� r� Z setblockingr� r �_loop_self_readingr2 r& r& r' r b s
z%BaseProactorEventLoop._make_self_pipec
C s� y$|d k r|� � | j�| jd�}W nH tjk
r: d S tk
rl } z| �d|| d�� W d d }~X Y nX || _|� | j
� d S )Ni z.Error on reading from the event loop self pipe)r< r= r ) r] r^ r_ r� r ra r[ rB r} rb r� )r rm rD r& r& r' r� j s z(BaseProactorEventLoop._loop_self_readingc C s> y| j �d� W n( tk
r8 | jr4tjddd� Y nX d S )Ns z3Fail to write a null byte into the self-pipe socketTr; )r� rl r? �_debugr
rA r2 r& r& r' �_write_to_self| s z$BaseProactorEventLoop._write_to_self�d c s( d� ������fdd� � �� � � d S )Nc
s$ y�| d k rn| � � \}}�jr,t�d�||� �� }�d k rX�j||�dd|i��d� n�j||d|i�d� ��� rzd S �j���} W n~ t k
r� } zB��
� dkr� d|�d�� �� � n�jr�tjd �dd
� W d d }~X Y n8 t
jk
�r �� � Y nX | �j��
� < | �� � d S ) Nz#%r got a new connection from %r: %rTrz )r� r" r# r� r� i����zAccept failed on a socket)r< r= rJ zAccept failed on socket %rr; )r] r� r
rA r� r� r� r^ r� r? r) rB r7 r ra r~ rb )rm ZconnZaddrr rD �r �protocol_factoryr r# r r� r� r& r' r � s@
z2BaseProactorEventLoop._start_serving.<locals>.loopr/ )r )r r� r r� r# Zbacklogr� r& r� r' �_start_serving� s %z$BaseProactorEventLoop._start_servingc C s d S r/ r& )r Z
event_listr& r& r' �_process_events� r� z%BaseProactorEventLoop._process_eventsc C s* x| j �� D ] }|�� q W | j �� d S r/ )r~ Zvaluesr6 Zclear)r �futurer& r& r' r� � s z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j �|�� d �}|r |�� | j�|� |�� d S r/ )r~ Zpopr) r6 r^ �
_stop_servingr7 )r r r� r& r& r' r� � s
z#BaseProactorEventLoop._stop_servingrL r/ ru ru ru r/ )NNr� N)r( rM rN r
r� r� r� r� r� r7 r� r� r� r� r� r� r� r� r r� r� r� r� r� r� rP r& r& r$ r' r � s<
)r ) rO Z__all__r� r� rJ r8 Z r r r r r r r Zlogr
Z_FlowControlMixinZ
BaseTransportr Z
ReadTransportrQ ZWriteTransportrd rv Z Transportrx ry Z
BaseEventLoopr r& r& r& r' Z<module> s: m
'