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 dejejZG dd deejZG dd deejZG dd deZG 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| jvrz|
| jd< W n tjy d | jd< Y n0 d S )NsocketZsocknamezgetsockname() failed on %rTexc_infopeername) r TransportSocket_extraZ getsocknamer
error_loop get_debugr warningZ getpeername) transportsock r )
r4 __name__r appendr( filenor$ r% r# lenr) formatjoin)r- infor r r __repr__H s
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )Npipe)r r- r r r r r Z s z%_ProactorBasePipeTransport._set_extrac C s
|| _ d S Nr+ )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 |tr*| j rBtjd| |dd n| j||| | jd W | | n | | 0 d S )Nz%r: %sTr )message exceptionr r/ )
isinstanceOSErrorr r r debugcall_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_waiterdone
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| jdr8| j dkr8| jtj | j d | _| j }|d ur|
d | _ nVt| jdr| j dkr| jtj | j d | _| j }|d ur|
d | _ 0 d S )Nshutdown) r+ Zconnection_losthasattrr 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| j7 }|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 warningswarnrL 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
ZdddZ 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,td| 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 rtd| 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
SystemExitKeyboardInterrupt
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| jtjrzt| 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 }zz|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
rtjddd W Y d }~nd }~0 ty } z| | W Y d }~nnd }~0 tyB } z| |d W Y d }~n>d }~0 tjy` | js\ Y n0 | jsx| j | j W |d ur| | n|d ur| | 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 _proactorrecvr ConnectionAbortedErrorrV r r rR ConnectionResetErrorrT rQ r CancelledErroradd_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 fddZdd ZdddZd 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- argskwr3 r r r G s z(_ProactorBaseWritePipeTransport.__init__c C s t |tttfs$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.| jst|| _| 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 TypeErrortyper8 r) RuntimeErrorrW r' r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r%
_loop_writingr# _maybe_pause_protocolextendrk 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 st|| _| j | j | n| j | j | jd ur| j d u r| jd W n` ty$ } z| | W Y d }~n:d }~0 tyT } 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 fddZdd 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
dZ dddZ
Z S )_ProactorDatagramTransporti Nc s> || _ d | _t j|||||d t | _| j| j d S )N)r0 r1 )
_addressrW r r collectionsdequer# 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 |tttfs 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 rW 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 tjy. | js* Y n 0 | jd urJ| 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.| jtj 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 fddZd3ddZd4dddddddd Zd5d
d Zd6d d
Zd7ddZd8ddZ fddZ
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 td|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 threadingcurrent_thread main_threadsignal
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_sideserver_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 r6td | | | 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@ } ztdW Y d }~n
d }~0 0 zt|j}W n t yn tdY n0 |rx|n|}|sdS t
|d}|rt
|| |n|} t
||}d}
zpt
| | |}|dkr|
W |
dkr| | S | j
||||I d H ||7 }|
|7 }
qW |
dkr:| | n|
dkr8| | 0 d S )Nznot a regular filer l )r: AttributeErrorioUnsupportedOperationr ZSendfileNotAvailableErrorosfstatst_sizerQ minseekrt sendfile) r- r fileoffsetcountr: errZfsizeZ blocksizeZend_posZ
total_sentr r r _sock_sendfile_native s:
z+BaseProactorEventLoop._sock_sendfile_nativec sn | }| | I d H z4| j|j|||ddI 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 \| _| _| jd | jd | 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_selfd c s( d fdd d S )Nc
s, z| d urn| \}}jr,td|| }d urXj||dd|id nj||d|id r|W d S j} W n t y } zP
dkrȈ d|t
d njrtjd dd
W Y d }~n>d }~0 tjy 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