3
fO @ s d Z dgZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd
lm
Z
dd l m Z G d d
d
e
j
e
jZG dd dee
jZG dd dee
jZG dd deZG dd deee
jZG dd deee
jZG dd dejZdS )zEvent 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 ddZdd 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)selfloopsockprotocolwaiterextraserver) __class__ ) r" __name__r appendr filenor r r lenr join)r infobufsizer# r# r$ __repr__/ s"
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )Npipe)_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 warningswarnResourceWarningr7 )r r# r# r$ __del__] s
z"_ProactorBasePipeTransport.__del__ Fatal error on pipe transportc C sP t |tjr*| jj rBtjd| |dd n| jj||| | jd | j | d S )Nz%r: %sT)exc_info)message exceptionZ transportr )
isinstancer Z_FATAL_ERROR_IGNOREr get_debugr
debugcall_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| jdr,| jjtj | jj d | _| j}|d k rX|j d | _X d S )Nshutdown)
r Zconnection_losthasattrr 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| j7 }|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@td | _|j }| jr\d }d S |dkrhd S | jjj| j d| _W n t
k
r } z2| js| j |d n| jj rt
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 | js@ Y nX | jj| j W d |rl| jj| n:|d k r| jj rt
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 AssertionErrorresultr _proactorrecvr ConnectionAbortedErrorrG rB r
rC ConnectionResetErrorrE OSErrorr CancelledErroradd_done_callbackrT r Z
data_receivedZ eof_receivedr7 )r futdatarF 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
ddZdd Zd d
Zd d ZdS ) _ProactorBaseWritePipeTransportzTransport for write pipes.c C s t |tttfs&dt|j }t|| jr4td|s peernamezgetpeername() failed on %r)
r1 Z getsocknamerJ errorAttributeErrorr 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 fddZd-ddZd.ddddddd Zd/d
d Zd0d d
Zd1ddZ fddZ 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_sideserver_hostnamer r! c
C s< t j stdt 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_selfd c s& d fdd j d S )Nc s" y| d k rl| j \}}jr,tjd|| }d k rVj||dd|id nj||d|id j rxd S jj} W n~ t k
r } zDj
d krj d|d j njrtjd dd
W Y d d }~X n8 t
jk
r j Y nX | jj
< | 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
#