3
\�O � @ s� d Z dgZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd
lm
Z
dd l m Z G d d
� d
e
j
e
j�ZG dd� dee
j�ZG dd� dee
j�ZG dd� de�ZG dd� deee
j�ZG dd� deee
j�ZG dd� dej�ZdS )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)�compat)� constants)�futures)�sslproto)�
transports)�loggerc s� e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e j rXdd� Z
d dd�Zdd� Zdd� Zdd� Z� ZS ) �_ProactorBasePipeTransportz*Base class for pipe and socket transports.Nc s� t � j||� | j|� || _|| _|| _d | _d | _d | _d| _ d| _
d| _ d| _ | jd k rh| jj
� | jj| jj| � |d k r�| jjtj|d � d S )Nr F)�super�__init__�
_set_extra�_sock� _protocol�_server�_buffer� _read_fut�
_write_fut�_pending_write�
_conn_lost�_closing� _eof_writtenZ_attach�_loop� call_soonZconnection_mader Z _set_result_unless_cancelled)�self�loop�sock�protocol�waiter�extra�server)� __class__� �//usr/lib64/python3.6/asyncio/proactor_events.pyr
s$
z#_ProactorBasePipeTransport.__init__c C s� | j jg}| jd kr |jd� n| jr0|jd� | jd k rN|jd| jj� � | jd k rh|jd| j � | jd k r�|jd| j � | jr�t | j�}|jd| � | j
r�|jd� dd j |� S )
N�closed�closingzfd=%szread=%szwrite=%rzwrite_bufsize=%sz EOF writtenz<%s>� ) r" �__name__r �appendr �filenor r r �lenr �join)r �info�bufsizer# r# r$ �__repr__/ s"
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )N�pipe)�_extra)r r r# r# r$ r B s z%_ProactorBasePipeTransport._set_extrac C s
|| _ d S )N)r )r r r# r# r$ � set_protocolE s z'_ProactorBasePipeTransport.set_protocolc C s | j S )N)r )r r# r# r$ � get_protocolH s z'_ProactorBasePipeTransport.get_protocolc C s | j S )N)r )r r# r# r$ �
is_closingK s z%_ProactorBasePipeTransport.is_closingc C s^ | j r
d S d| _ | jd7 _| j r@| jd kr@| jj| jd � | jd k rZ| jj� d | _d S )NTr ) r r r r r r �_call_connection_lostr �cancel)r r# r# r$ �closeN s
z _ProactorBasePipeTransport.closec C s* | j d k r&tjd| t| d� | j� d S )Nzunclosed transport %r)�source)r �warnings�warn�ResourceWarningr7 )r r# r# r$ �__del__] s
z"_ProactorBasePipeTransport.__del__� Fatal error on pipe transportc C sP t |tj�r*| jj� rBtjd| |dd� n| jj||| | jd�� | j |� d S )Nz%r: %sT)�exc_info)�message� exceptionZ transportr )
�
isinstancer Z_FATAL_ERROR_IGNOREr � get_debugr
�debug�call_exception_handlerr � _force_close)r �excr? r# r# r$ � _fatal_errorc s
z'_ProactorBasePipeTransport._fatal_errorc C sj | j r
d S d| _ | jd7 _| jr4| jj� d | _| jrJ| jj� d | _d| _d | _| jj| j |� d S )NTr r )
r r r r6 r r r r r r5 )r rF r# r# r$ rE p s
z'_ProactorBasePipeTransport._force_closec C s^ z| j j|� W d t| jd�r,| jjtj� | jj� d | _| j}|d k rX|j � d | _X d S )N�shutdown)
r Zconnection_lost�hasattrr rH �socketZ SHUT_RDWRr7 r Z_detach)r rF r! r# r# r$ r5 s
z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd k r |t| j�7 }|S )N)r r r+ )r �sizer# r# r$ �get_write_buffer_size� s
z0_ProactorBasePipeTransport.get_write_buffer_size)NNN)r= )r( �
__module__� __qualname__�__doc__r
r/ r r2 r3 r4 r7 r ZPY34r< rG rE r5 rL �
__classcell__r# r# )r" r$ r s
r c s< e Zd ZdZd � fdd� Zdd� Zdd� Zd d d
�Z� ZS )
�_ProactorReadPipeTransportzTransport for read pipes.Nc s4 t � j||||||� d| _d| _| jj| j� d S )NF)r r
�_paused�_reschedule_on_resumer r �
_loop_reading)r r r r r r r! )r" r# r$ r
� s z#_ProactorReadPipeTransport.__init__c C s0 | j s | jrd S d| _| jj� r,tjd| � d S )NTz%r pauses reading)r rR r rB r
rC )r r# r# r$ �
pause_reading� s
z(_ProactorReadPipeTransport.pause_readingc C sP | j s| j rd S d| _| jr6| jj| j| j� d| _| jj� rLtj d| � d S )NFz%r resumes reading)
r rR rS r r rT r rB r
rC )r r# r# r$ �resume_reading� s
z)_ProactorReadPipeTransport.resume_readingc C s� | j rd| _d S d }�z@yf|d k rN| j|ks@| jd kr<| js@t�d | _|j� }| jr\d }d S |dkrhd S | jjj| j d�| _W n� t
k
r� } z2| js�| j |d� n| jj � r�t
jddd� W Y d d }~X n� tk
r� } z| j|� W Y d d }~X n^ tk
�r$ } z| j |d� W Y d d }~X n0 tjk
�rD | j�s@� Y nX | jj| j� W d |�rl| jj|� n:|d k �r�| jj � �r�t
jd| � | jj� }|�s�| j� X d S )NT� i z"Fatal read error on pipe transportz*Read error on pipe transport while closing)r> z%r received EOF)rR rS r r �AssertionError�resultr � _proactor�recvr �ConnectionAbortedErrorrG rB r
rC �ConnectionResetErrorrE �OSErrorr �CancelledError�add_done_callbackrT r Z
data_receivedZ eof_receivedr7 )r �fut�datarF Z keep_openr# r# r$ rT � sL
z(_ProactorReadPipeTransport._loop_reading)NNN)N) r( rM rN rO r
rU rV rT rP r# r# )r" r$ rQ � s
rQ c @ s: e Zd ZdZdd� Zd
dd�Zdd� Zd d
� Zd d � ZdS )� _ProactorBaseWritePipeTransportzTransport for write pipes.c C s� t |tttf�s&dt|�j }t|��| jr4td��|s �peernamezgetpeername() failed on %r)
r1 Z getsocknamerJ �error�AttributeErrorr rB r
rj Z getpeername)r r r# r# r$ r f s
z#_ProactorSocketTransport._set_extrac C s dS )NTr# )r r# r# r$ rt v s z&_ProactorSocketTransport.can_write_eofc C s2 | j s | jrd S d| _| jd kr.| jjtj� d S )NT)r r r r rH rJ rp )r r# r# r$ ru y s
z"_ProactorSocketTransport.write_eof)NNN) r( rM rN rO r
r rt ru rP r# r# )r" r$ r \ s r c s� e Zd Z� fdd�Zd-dd�Zd.ddddd�dd �Zd/d
d �Zd0d d
�Zd1dd�Z� fdd�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd d � Zd d � Zd2d d!�Zd"d#� Zd3d%d&�Zd'd(� Zd)d*� Zd+d,� Z� ZS )4r c sH t � j� tjd|jj� || _|| _d | _i | _ |j
| � | j � d S )NzUsing proactor: %s) r r
r
rC r" r( rZ � _selector�_self_reading_future�_accept_futuresZset_loop�_make_self_pipe)r Zproactor)r" r# r$ r
� s
z BaseProactorEventLoop.__init__Nc C s t | |||||�S )N)r )r r r r r r! r# r# r$ �_make_socket_transport� s
z,BaseProactorEventLoop._make_socket_transportF)� server_side�server_hostnamer r! c
C s< t j� std��t j| |||||�} t| || ||d� | jS )NzOProactor event loop requires Python 3.5 or newer (ssl.MemoryBIO) to support SSL)r r! )r Z_is_sslproto_availabler~ Z SSLProtocolr Z_app_transport)
r Zrawsockr �
sslcontextr r� r� r r! Z ssl_protocolr# r# r$ �_make_ssl_transport� s
z)BaseProactorEventLoop._make_ssl_transportc C s t | ||||�S )N)r} )r r r r r r# r# r$ �_make_duplex_pipe_transport� s z1BaseProactorEventLoop._make_duplex_pipe_transportc C s t | ||||�S )N)rQ )r r r r r r# r# r$ �_make_read_pipe_transport� s z/BaseProactorEventLoop._make_read_pipe_transportc C s t | ||||�S )N)rw )r r r r r r# r# r$ �_make_write_pipe_transport� s z0BaseProactorEventLoop._make_write_pipe_transportc sP | j � rtd��| j� r d S | j� | j� | jj� d | _d | _t� j� d S )Nz!Cannot close a running event loop) Z
is_runningri � is_closed�_stop_accept_futures�_close_self_piperZ r7 r� r )r )r" r# r$ r7 � s
zBaseProactorEventLoop.closec C s | j j||�S )N)rZ r[ )r r �nr# r# r$ � sock_recv� s z BaseProactorEventLoop.sock_recvc C s | j j||�S )N)rZ rq )r r rb r# r# r$ � sock_sendall� s z"BaseProactorEventLoop.sock_sendallc C s | j j||�S )N)rZ Zconnect)r r Zaddressr# r# r$ � sock_connect� s z"BaseProactorEventLoop.sock_connectc C s | j j|�S )N)rZ �accept)r r r# r# r$ � sock_accept� s z!BaseProactorEventLoop.sock_acceptc C s t �d S )N)r~ )r r# r# r$ � _socketpair� s z!BaseProactorEventLoop._socketpairc C sL | j d k r| j j� d | _ | jj� d | _| jj� d | _| jd8 _d S )Nr )r� r6 �_ssockr7 �_csock�
_internal_fds)r r# r# r$ r� � s
z&BaseProactorEventLoop._close_self_pipec C sF | j � \| _| _| jjd� | jjd� | jd7 _| j| j� d S )NFr )r� r� r� Z setblockingr� r �_loop_self_reading)r r# r# r$ r� � s
z%BaseProactorEventLoop._make_self_pipec C s� y$|d k r|j � | jj| jd�}W nH tjk
r: d S tk
rl } z| jd|| d�� W Y d d }~X nX || _|j | j
� d S )Ni z.Error on reading from the event loop self pipe)r? r@ r ) rY rZ r[ r� r r_ � ExceptionrD r� r` r� )r rs rF r# r# r$ r� � s z(BaseProactorEventLoop._loop_self_readingc C s | j jd� d S )N� )r� rq )r r# r# r$ �_write_to_self� s z$BaseProactorEventLoop._write_to_self�d c s& d� �����fdd� � �j � � d S )Nc s" y�| d k rl| j � \}}�jr,tjd�||� �� }�d k rV�j||�dd|i�d� n�j||d|i�d� �j� rxd S �jj��} W n~ t k
r� } zD�j
� d kr��j d|�d�� �j � n�jr�tjd �dd
� W Y d d }~X n8 t
jk
�r �j � Y nX | �j�j
� < | j� � d S ) Nz#%r got a new connection from %r: %rTr� )r� r r! )r r! r zAccept failed on a socket)r? r@ rJ zAccept failed on socket %r)r> �����)rY Z_debugr
rC r� r� r� rZ r� r^ r* rD r7 r r_ r� r` )rs ZconnZaddrr rF )r �protocol_factoryr r! r r� r# r$ r � s>
z2BaseProactorEventLoop._start_serving..loop)N)r )r r� r r� r! Zbacklogr# )r r� r r! r r� r$ �_start_serving� s $z$BaseProactorEventLoop._start_servingc C s d S )Nr# )r Z
event_listr# r# r$ �_process_events s z%BaseProactorEventLoop._process_eventsc C s* x| j j� D ] }|j� q W | j j� d S )N)r� �valuesr6 �clear)r Zfuturer# r# r$ r� $ s z*BaseProactorEventLoop._stop_accept_futuresc C s | j � | jj|� |j� d S )N)r� rZ �
_stop_servingr7 )r r r# r# r$ r� ) s z#BaseProactorEventLoop._stop_serving)NNN)N)NN)NN)NN)N)NNr� )r( rM rN r
r� r� r� r� r� r7 r� r� r� r� r� r� r� r� r� r� r� r� r� rP r# r# )r" r$ r � s4
()rO �__all__rJ r9 � r r r r r r �logr
Z_FlowControlMixinZ
BaseTransportr Z
ReadTransportrQ ZWriteTransportrc rw Z Transportr} r Z
BaseEventLoopr r# r# r# r$ � s2 M T
#