a
XC?h�} � @ sT d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
dd lmZ dd
lmZ dd lmZ dd lmZ d
d� ZG dd� dejej�ZG dd� deej�ZG dd� deej�ZG dd� de�ZG dd� deej �Z G dd� deeej �Z G dd � d eeej �Z G d d � d e j!�Z"dS ) )�BaseProactorEventLoop� N� )� base_events)� constants)�futures)�
exceptions)� protocols)�sslproto)�
transports)�trsock)�loggerc C s� t �|�| jd< z|�� | jd< W n. tjyP | j�� rLtj d|dd� Y n0 d| jvr�z|�
� | jd< W n tjy� d | jd< Y n0 d S )N�socketZsocknamezgetsockname() failed on %rT��exc_info�peername) r �TransportSocket�_extraZ getsocknamer
�error�_loop� get_debugr �warningZ getpeername)� transport�sock� r �� )
r4 �__name__r �appendr( �filenor$ r% r# �lenr) �format�join)r- �infor r r �__repr__H s
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )N�pipe)r �r- r r r r r Z s z%_ProactorBasePipeTransport._set_extrac C s
|| _ d S �N�r+ )r- r/ r r r r! ] s z'_ProactorBasePipeTransport.set_protocolc C s | j S rB rC �r- r r r � get_protocol` s z'_ProactorBasePipeTransport.get_protocolc C s | j S rB )r( rD r r r �
is_closingc s z%_ProactorBasePipeTransport.is_closingc C s\ | j r
d S d| _ | jd7 _| js>| jd u r>| j�| jd � | jd urX| j�� d | _d S )NTr ) r( r' r# r% r r* �_call_connection_lostr$ �cancelrD r r r �closef s
z _ProactorBasePipeTransport.closec C s* | j d ur&|d| ��t| d� | �� d S )Nzunclosed transport )�source)r �ResourceWarningrI )r- Z_warnr r r �__del__q s
z"_ProactorBasePipeTransport.__del__� Fatal error on pipe transportc C s` zNt |t�r*| j�� rBtjd| |dd� n| j�||| | jd�� W | �|� n | �|� 0 d S )Nz%r: %sTr )�message� exceptionr r/ ) �
isinstance�OSErrorr r r �debug�call_exception_handlerr+ � _force_close)r- �excrN r r r � _fatal_errorv s
�
z'_ProactorBasePipeTransport._fatal_errorc C s� | j d ur6| j �� s6|d u r*| 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% rH r$ r& r# r r* rG )r- rU r r r rT � s"
z'_ProactorBasePipeTransport._force_closec C s� zd| j �|� W t| jd�r8| j�� dkr8| j�tj� | j�� d | _| j }|d ur�|�
� d | _ nVt| jd�r�| j�� dkr�| j�tj� | j�� d | _| j }|d ur�|�
� d | _ 0 d S )N�shutdown�����) r+ Zconnection_lost�hasattrr r: rZ r
Z SHUT_RDWRrI r" Z_detach)r- rU r2 r r r rG � s$
�
z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd ur |t| j�7 }|S rB )r& r# r; )r- �sizer r r �get_write_buffer_size� s
z0_ProactorBasePipeTransport.get_write_buffer_size)NNN)rM )r8 �
__module__� __qualname__r r? r r! rE rF rI �warnings�warnrL rV rT rG r^ �
__classcell__r r r3 r r . s �
r c sP e Zd Zd� fdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zddd�Z � Z
S )�_ProactorReadPipeTransportNc s: d | _ d| _t� �||||||� | j�| j� d| _d S )NTF)�
_pending_data�_pausedr r r r* �
_loop_readingr, r3 r r r � s
z#_ProactorReadPipeTransport.__init__c C s | j o| j S rB )rf r( rD r r r �
is_reading� s z%_ProactorReadPipeTransport.is_readingc C s0 | j s | jrd S d| _| j�� r,t�d| � d S )NTz%r pauses reading)r( rf r r r rR rD r r r �
pause_reading� s
z(_ProactorReadPipeTransport.pause_readingc C sn | j s | jsd S d| _| jd u r0| j�| jd � | j}d | _|d urT| j�| j|� | j�� rjt �
d| � d S )NFz%r resumes reading) r( rf r$ r r* rg re �_data_receivedr r rR �r- �datar r r �resume_reading� s
z)_ProactorReadPipeTransport.resume_readingc
C s~ | j �� rt�d| � z| j�� }W nH ttfy< � Y n2 tyl } z| � |d� W Y d }~d S d }~0 0 |sz| �
� d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.) r r r rR r+ Z eof_received�
SystemExit�KeyboardInterrupt�
BaseExceptionrV rI )r- Z keep_openrU r r r �
_eof_received� s
�z(_ProactorReadPipeTransport._eof_receivedc
C s� | j r|| _d S |s | �� d S t| jtj�r�zt�| j|� W q� tt fyX � Y q� t
y� } z| � |d� W Y d }~d S d }~0 0 n | j� |� d S )Nz3Fatal error: protocol.buffer_updated() call failed.)
rf re rq rP r+ r ZBufferedProtocolZ _feed_data_to_buffered_protorn ro rp rV Z
data_received)r- rl rU r r r rj � s" �z)_ProactorReadPipeTransport._data_receivedc
C s� d }�z�z�|d ur2d | _ |�� r*|�� }n|�� | jrVd }W W |d urR| �|� d S |dkrxW W |d urt| �|� d S | js�| jj� | j
d�| _ W n� t y� } z8| js�| � |d� n| j�
� r�tjddd� W Y d }~n�d }~0 t�y } z| �|� W Y d }~nnd }~0 t�yB } z| � |d� W Y d }~n>d }~0 tj�y` | j�s\� Y n0 | j�sx| j �| j� W |d u�r�| �|� n|d u�r�| �|� 0 d S )N� i � z"Fatal read error on pipe transportz*Read error on pipe transport while closingTr )r$ rX �resultrH r( rj rf r � _proactor�recvr �ConnectionAbortedErrorrV r r rR �ConnectionResetErrorrT rQ r �CancelledError�add_done_callbackrg )r- �futrl rU r r r rg sP
�
�
�
�
z(_ProactorReadPipeTransport._loop_reading)NNN)N) r8 r_ r` r rh ri rm rq rj rg rc r r r3 r rd � s � rd c sZ e Zd ZdZ� fdd�Zdd� Zddd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )� _ProactorBaseWritePipeTransportTc s t � j|i |�� d | _d S rB )r r rW �r- �args�kwr3 r r r G s z(_ProactorBaseWritePipeTransport.__init__c C s� t |tttf�s$tdt|�j� ���| jr2td��| j d urDtd��|sLd S | j
rz| j
t j krht
�d� | j
d7 _
d S | jd u r�| 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 )rl )rP �bytes� bytearray�
memoryview� TypeError�typer8 r) � RuntimeErrorrW r' r �!LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r% �
_loop_writingr# �_maybe_pause_protocol�extendrk r r r �writeK s0 ��
z%_ProactorBaseWritePipeTransport.writeNc
C sZ z�|d ur | j d u r | jr W d S d | _ d| _|r8|�� |d u rL| j}d | _|s�| jrf| j�| jd � | jrz| j �
t j � | �
� nN| jj�| j |�| _ | j �� s�t|�| _| j �| j� | �� n| j �| j� | jd ur�| j d u r�| j�d � W n` t�y$ } z| �|� W Y d }~n:d }~0 t�yT } z| �|d� W Y d }~n
d }~0 0 d S )Nr z#Fatal write error on pipe transport)r% r( r& rs r# r r* rG r) r rZ r
�SHUT_WR�_maybe_resume_protocolrt �sendrX r; ry r� r� rW rY rw rT rQ rV )r- �frl rU r r r r� q s8
z-_ProactorBaseWritePipeTransport._loop_writingc C s dS �NTr rD r r r �
can_write_eof� s z-_ProactorBaseWritePipeTransport.can_write_eofc C s | � � d S rB )rI rD r r r � write_eof� s z)_ProactorBaseWritePipeTransport.write_eofc C s | � d � d S rB �rT rD r r r �abort� s z%_ProactorBaseWritePipeTransport.abortc C s: | j d urtd��| j�� | _ | jd u r4| j �d � | j S )NzEmpty waiter is already set)rW r� r Z
create_futurer% rY rD r r r �_make_empty_waiter� s
z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s
d | _ d S rB )rW rD r r r �_reset_empty_waiter� s z3_ProactorBaseWritePipeTransport._reset_empty_waiter)NN)
r8 r_ r` Z_start_tls_compatibler r� r� r� r� r� r� r� rc r r r3 r r{ A s &
)r{ c s$ e Zd Z� fdd�Zdd� Z� ZS )�_ProactorWritePipeTransportc s8 t � j|i |�� | jj�| jd�| _| j�| j� d S )N� ) r r r rt ru r r$ ry � _pipe_closedr| r3 r r r � s z$_ProactorWritePipeTransport.__init__c C s@ |� � r d S | jrd S d | _| jd ur4| �t� � n| �� d S rB )Z cancelledr( r$ r% rT �BrokenPipeErrorrI )r- rz r r r r� � s
z(_ProactorWritePipeTransport._pipe_closed)r8 r_ r` r r� rc r r r3 r r� � s r� c sX e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zdd d �Zdd
d�Z ddd�Z
� Z S )�_ProactorDatagramTransporti Nc s> || _ d | _t� j|||||d� t�� | _| j�| j � d S )N)r0 r1 )
�_addressrW r r � collections�dequer# r r* rg )r- r. r r/ �addressr0 r1 r3 r r r � s
z#_ProactorDatagramTransport.__init__c C s t | |� d S rB �r rA r r r r � s z%_ProactorDatagramTransport._set_extrac C s t dd� | jD ��S )Nc s s | ]\}}t |�V qd S rB )r; )�.0rl �_r r r � � rr zC_ProactorDatagramTransport.get_write_buffer_size..)�sumr# rD r r r r^ � s z0_ProactorDatagramTransport.get_write_buffer_sizec C s | � d � d S rB r� rD r r r r� � s z _ProactorDatagramTransport.abortc C s� t |tttf�s tdt|���|s&d S | jd urN|d | jfvrNtd| j� ���| jr�| jr�| jt j
krpt � d� | jd7 _d S | j
�t|�|f� | jd u r�| �� | �� d S )Nz,data argument must be bytes-like object (%r)z!Invalid address: must be None or z!socket.sendto() raised exception.r )rP r r� r� r� r� r� �
ValueErrorr' r r� r r r# r9 r% r� r� )r- rl �addrr r r �sendto� s& �
�
z!_ProactorDatagramTransport.sendtoc
C s z�| j rW d S d | _|r |�� | jr2| j rN| jrN| jrH| j�| jd � W d S | j� � \}}| jd ur|| jj
� | j |�| _n| jj
j
| j ||d�| _W n^ ty� } z| j�|� W Y d }~nNd }~0 ty� } z| �|d� W Y d }~n d }~0 0 | j�| j� | �� d S )N)r� z'Fatal write error on datagram transport)r' r% rs r# r� r( r r* rG �popleftrt r� r r� rQ r+ �error_received� ExceptionrV ry r� r� )r- rz rl r� rU r r r r� � s2
� � "z(_ProactorDatagramTransport._loop_writingc
C s| d }�zZz�| j r*W W |r&| j�||� d S d | _|d ur�|�� }| jrdd }W W |r`| j�||� d S | jd ur||| j }}n|\}}| j r�W W |r�| j�||� d S | jd ur�| jj� | j
| j �| _n| jj� | j
| j �| _W nP t
�y } z| j�|� W Y d }~nBd }~0 tj�y. | j�s*� Y n 0 | jd u�rJ| j�| j� W |�rx| j�||� n|�rv| j�||� 0 d S rB )r' r+ Zdatagram_receivedr$ rs r( r� r rt ru r �max_sizeZrecvfromrQ r� r rx ry rg )r- rz rl r� �resrU r r r rg sT $��
�
� �
�z(_ProactorDatagramTransport._loop_reading)NNN)N)N)N) r8 r_ r` r� r r r^ r� r� r� rg rc r r r3 r r� � s �
!r� c @ s e Zd Zdd� Zdd� ZdS )� _ProactorDuplexPipeTransportc C s dS )NFr rD r r r r� K s z*_ProactorDuplexPipeTransport.can_write_eofc C s t �d S rB )�NotImplementedErrorrD r r r r� N s z&_ProactorDuplexPipeTransport.write_eofN)r8 r_ r` r� r� r r r r r� F s r� c s> e Zd ZejjZd
� fdd� Zdd� Zdd� Z dd � Z
� Z S ) �_ProactorSocketTransportNc s$ t � �||||||� t�|� d S rB )r r r Z _set_nodelayr, r3 r r r Y s z!_ProactorSocketTransport.__init__c C s t | |� d S rB r� rA r r r r ^ s z#_ProactorSocketTransport._set_extrac C s dS r� r rD r r r r� a s z&_ProactorSocketTransport.can_write_eofc C s2 | j s | jrd S d| _| jd u r.| j�tj� d S r� )r( r) r% r rZ r
r� rD r r r r� d s
z"_ProactorSocketTransport.write_eof)NNN) r8 r_ r` r Z
_SendfileModeZ
TRY_NATIVEZ_sendfile_compatibler r r� r� rc r r r3 r r� R s �r� c s� e Zd Z� fdd�Zd3dd�Zd4dddddd�dd �Zd5d
d �Zd6d d
�Zd7dd�Zd8dd�Z � fdd�Z
dd� Z dd� Z dd� Z
dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd9d&d'�Zd(d)� Zd:d+d,�Zd-d.� Zd/d0� Zd1d2� Z� ZS );r c sh t � �� t�d|jj� || _|| _d | _i | _ |�
| � | � � t �
� t �� u rdt�| j�� � d S )NzUsing proactor: %s)r r r rR r4 r8 rt � _selector�_self_reading_future�_accept_futuresZset_loop�_make_self_pipe� threading�current_thread� main_thread�signal�
set_wakeup_fd�_csockr: )r- Zproactorr3 r r r n s
z BaseProactorEventLoop.__init__Nc C s t | |||||�S rB )r� )r- r r/ r0 r1 r2 r r r �_make_socket_transport{ s
�z,BaseProactorEventLoop._make_socket_transportF)� server_side�server_hostnamer1 r2 �ssl_handshake_timeoutc C s0 t j| |||||| d�}
t| ||
||d� |
jS )N)r� �r1 r2 )r Z SSLProtocolr� Z_app_transport) r- Zrawsockr/ �
sslcontextr0 r� r� r1 r2 r� Z ssl_protocolr r r �_make_ssl_transport� s ��z)BaseProactorEventLoop._make_ssl_transportc C s t | |||||�S rB )r� )r- r r/ r� r0 r1 r r r �_make_datagram_transport� s
�z.BaseProactorEventLoop._make_datagram_transportc C s t | ||||�S rB )r� �r- r r/ r0 r1 r r r �_make_duplex_pipe_transport� s �z1BaseProactorEventLoop._make_duplex_pipe_transportc C s t | ||||�S rB )rd r� r r r �_make_read_pipe_transport� s z/BaseProactorEventLoop._make_read_pipe_transportc C s t | ||||�S rB )r� r� r r r �_make_write_pipe_transport� s �z0BaseProactorEventLoop._make_write_pipe_transportc sj | � � rtd��| �� r d S t�� t�� u r6t�d� | �� | � � | j
� � d | _
d | _ t
� � � d S )Nz!Cannot close a running event loopr[ )Z
is_runningr� � is_closedr� r� r� r� r� �_stop_accept_futures�_close_self_pipert rI r� r rD r3 r r rI � s
zBaseProactorEventLoop.closec � s | j �||�I d H S rB )rt ru )r- r �nr r r � sock_recv� s z BaseProactorEventLoop.sock_recvc � s | j �||�I d H S rB )rt Z recv_into)r- r Zbufr r r �sock_recv_into� s z$BaseProactorEventLoop.sock_recv_intoc � s | j �||�I d H S rB )rt r� )r- r rl r r r � sock_sendall� s z"BaseProactorEventLoop.sock_sendallc � s | j �||�I d H S rB )rt Zconnect)r- r r� r r r � sock_connect� s z"BaseProactorEventLoop.sock_connectc � s | j �|�I d H S rB )rt �acceptrA r r r � sock_accept� s z!BaseProactorEventLoop.sock_acceptc
� s> z |� � }W n4 ttjfy@ } zt�d��W Y d }~n
d }~0 0 zt�|�j}W n t yn t�d��Y n0 |rx|n|}|s�dS t
|d�}|r�t
|| |�n|} t
||�}d}
zpt
| | |�}|dkr�|
W |
dkr�|� |� S | j �
||||�I d H ||7 }|
|7 }
q�W |
dk�r:|� |� n|
dk�r8|� |� 0 d S )Nznot a regular filer l �� )r: �AttributeError�io�UnsupportedOperationr ZSendfileNotAvailableError�os�fstat�st_sizerQ �min�seekrt �sendfile) r- r �file�offset�countr: �errZfsizeZ blocksizeZend_posZ
total_sentr r r �_sock_sendfile_native� s:
�
�
z+BaseProactorEventLoop._sock_sendfile_nativec � sn |� � }|�� |�� I d H z4| j|j|||dd�I d H W |�� |rP|�� S n|�� |rh|�� 0 d S )NF)Zfallback)rh ri r� Z
sock_sendfiler r� rm )r- Ztranspr� r� r� rm r r r �_sendfile_native� s" � � �z&BaseProactorEventLoop._sendfile_nativec C sL | j d ur| j �� d | _ | j�� d | _| j�� d | _| jd8 _d S )Nr )r� rH �_ssockrI r� �
_internal_fdsrD r r r r� � s
z&BaseProactorEventLoop._close_self_pipec C s: t �� \| _| _| j�d� | j�d� | jd7 _d S )NFr )r
Z
socketpairr� r� Z setblockingr� rD r r r r� � s z%BaseProactorEventLoop._make_self_pipec
C s� z4|d ur|� � | j|ur"W d S | j�| jd�}W n` tjyJ Y d S ttfy` � Y nH t y� } z | �
d|| d�� W Y d }~n d }~0 0 || _|� | j � d S )Ni z.Error on reading from the event loop self pipe)rN rO r. )
rs r� rt ru r� r rx rn ro rp rS ry �_loop_self_reading)r- r� rU r r r r� � s$
� z(BaseProactorEventLoop._loop_self_readingc C sL | j }|d u rd S z|�d� W n& tyF | jrBtjddd� Y n0 d S )N� z3Fail to write a null byte into the self-pipe socketTr )r� r� rQ �_debugr rR )r- Zcsockr r r �_write_to_self s �z$BaseProactorEventLoop._write_to_self�d c s( d� ������fdd� � �� � � d S )Nc
s, z�| d urn| � � \}}�jr,t�d�||� �� }�d urX�j||�dd|i��d� n�j||d|i�d� ��� r|W d S �j���} W n� t y� } zP��
� dkrȈ� d|t �
��d�� ��� n�jr�tjd �dd
� W Y d }~n>d }~0 tj�y ��� Y n0 | �j��
� < | �� � d S ) Nz#%r got a new connection from %r: %rTr )r� r1 r2 r� r� r[ zAccept failed on a socket)rN rO r
zAccept failed on socket %rr )rs r� r rR r� r� r� rt r� rQ r: rS r r rI r rx r� ry )r� Zconnr� r/ rU �r. �protocol_factoryr- r2 r r� r� r r r. 0 sJ ��� �
�z2BaseProactorEventLoop._start_serving..loop)N)r* )r- r� r r� r2 Zbacklogr� r r� r �_start_serving, s %z$BaseProactorEventLoop._start_servingc C s d S rB r )r- Z
event_listr r r �_process_eventsW s z%BaseProactorEventLoop._process_eventsc C s& | j �� D ] }|�� q
| j �� d S rB )r� �valuesrH �clear)r- �futurer r r r� [ s
z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j �|�� d �}|r |�� | j�|� |�� d S rB )r� �popr: rH rt �
_stop_servingrI )r- r r� r r r r� ` s
z#BaseProactorEventLoop._stop_serving)NNN)N)NNN)NN)NN)NN)N)NNr� N)r8 r_ r` r r� r� r� r� r� r� rI r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rc r r r3 r r l sN
�
�� �
�
�
�
�
+r )#�__all__r� r� r
ra r� r� r� � r r r r r r r
r �logr r Z_FlowControlMixinZ
BaseTransportr Z
ReadTransportrd ZWriteTransportr{ r� ZDatagramTransportr� Z Transportr� r� Z
BaseEventLoopr r r r r � sT � � �n � � �