U
ifT @ s* d Z ddlZddlZddlZddlZddlZddlZddlZz ddlZW n e k
r` dZY nX ddl
m Z ddl
m Z ddl
m
Z
ddl
mZ ddl
mZ dd l
mZ dd
l
mZ dd l
mZ dd lmZ d
d Zdd ZG dd de jZG dd dejejZG dd deZ G dd deZ dS ))BaseSelectorEventLoop N ) base_events) constants)events)futures) protocols)sslproto)
transports)trsock)loggerc C s8 z| |}W n tk
r$ Y dS X t|j|@ S d S NF)get_keyKeyErrorboolr )selectorfdZeventkey r d? Z!d@dA Z"dBdC Z#dDdE Z$dFdG Z%dHdI Z&dJdK Z'dLdM Z(dNdO Z)dPdQ Z* Z+S )Vr Nc sF t |d krt }td|jj || _| t
| _ d S )NzUsing selector: %s)
super__init__ selectorsZDefaultSelectorr debug __class____name__ _selector_make_self_pipeweakrefZWeakValueDictionary _transports)selfr r r r r 6 s
z BaseSelectorEventLoop.__init__extraserverc C s t | |||||S N)_SelectorSocketTransport)r& r protocolwaiterr) r* r r r _make_socket_transport@ s
z,BaseSelectorEventLoop._make_socket_transportF) server_sideserver_hostnamer) r* ssl_handshake_timeoutc C s0 t j| |||||| d}
t| ||
||d |
jS )N)r2 r( )r Z SSLProtocolr, Z_app_transport) r& Zrawsockr-
sslcontextr. r0 r1 r) r* r2 Z ssl_protocolr r r _make_ssl_transportE s z)BaseSelectorEventLoop._make_ssl_transportc C s t | |||||S r+ )_SelectorDatagramTransport)r& r r- addressr. r) r r r _make_datagram_transportR s
z.BaseSelectorEventLoop._make_datagram_transportc sL | rtd| r d S | t | jd k rH| j d | _d S )Nz!Cannot close a running event loop)Z
is_running RuntimeError is_closed_close_self_piper closer" r& r' r r r; W s
zBaseSelectorEventLoop.closec C sB | | j | j d | _| j d | _| jd8 _d S )Nr )_remove_reader_ssockfilenor; _csock
_internal_fdsr< r r r r: b s
z&BaseSelectorEventLoop._close_self_pipec C sN t \| _| _| jd | jd | jd7 _| | j | j d S )NFr ) socketZ
socketpairr> r@ setblockingrA _add_readerr? _read_from_selfr< r r r r# j s
z%BaseSelectorEventLoop._make_self_pipec C s d S r+ r r& datar r r _process_self_datar s z(BaseSelectorEventLoop._process_self_datac C sX z"| j d}|sW qT| | W q tk
r: Y q Y q tk
rP Y qTY q X q d S )Ni )r> recvrH InterruptedErrorBlockingIOErrorrF r r r rE u s z%BaseSelectorEventLoop._read_from_selfc C sN | j }|d krd S z|d W n( tk
rH | jrDtjddd Y nX d S )N z3Fail to write a null byte into the self-pipe socketTexc_info)r@ sendOSError_debugr r )r& Zcsockr r r _write_to_self s z$BaseSelectorEventLoop._write_to_selfd c
C s" | | | j|||||| d S r+ )rD r? _accept_connection)r& protocol_factoryr r3 r* backlogr2 r r r _start_serving s z$BaseSelectorEventLoop._start_servingc
C s t |D ]}z0| \}} | jr0td|| | |d W n tttfk
rZ Y d S t k
r }
zd|
j
t
j t
j t
j
t
jfkr| d|
t|d | | | tj| j|||||| n W 5 d }
~
X Y qX d| i} | ||| |||} | | qd S )Nz#%r got a new connection from %r: %rFz&socket.accept() out of system resource)message exceptionrB peername)rangeacceptrQ r r rC rK rJ ConnectionAbortedErrorrP errnoZEMFILEZENFILEZENOBUFSZENOMEMcall_exception_handlerr TransportSocketr= r? Z
call_laterr ZACCEPT_RETRY_DELAYrW _accept_connection2Z create_task)
r& rU r r3 r* rV r2 _connaddrexcr) r\ r r r rT sV z(BaseSelectorEventLoop._accept_connectionc
s d }d }zt| }| } |r8| j|||| d|||d}n| j||| ||d}z| I d H W n tk
rx | Y nX W nt ttfk
r Y n\ tk
r }
z>| jrd|
d} |d k r|| d< |d k r|| d< | | W 5 d }
~
X Y nX d S )NT)r. r0 r) r* r2 )r. r) r* z3Error on transport creation for incoming connection)rX rY r- transport)
create_futurer4 r/
BaseExceptionr;
SystemExitKeyboardInterruptrQ r_ ) r& rU rc r) r3 r* r2 r- rf r. re contextr r r ra sP z)BaseSelectorEventLoop._accept_connection2c
C s |}t |tsJzt| }W n* tttfk
rH td|d Y nX z| j| }W n tk
rl Y n X | st d|d|d S )NzInvalid file object: zFile descriptor z is used by transport )
r intr? AttributeErrorr
ValueErrorr% r
is_closingr8 )r& r r? rf r r r _ensure_fd_no_transport s
z-BaseSelectorEventLoop._ensure_fd_no_transportc G s | t||| d }z| j|}W n* tk
rR | j|tj|d f Y n>X |j|j }\}}| j
||tjB ||f |d k r| d S r+ )
_check_closedr Handler" r r registerr
EVENT_READrG modifycancel r& r callbackargsZhandler maskreaderwriterr r r rD s
z!BaseSelectorEventLoop._add_readerc C s | r dS z| j|}W n tk
r2 Y dS X |j|j }\}}|tj M }|sd| j| n| j ||d |f |d k r|
dS dS d S NFT) r9 r" r r r rG r rt
unregisterru rv r& r r rz r{ r| r r r r= s z$BaseSelectorEventLoop._remove_readerc G s | t||| d }z| j|}W n* tk
rR | j|tjd |f Y n>X |j|j }\}}| j
||tjB ||f |d k r| d S r+ ) rq r rr r" r r rs r EVENT_WRITErG ru rv rw r r r _add_writer% s
z!BaseSelectorEventLoop._add_writerc C s | r dS z| j|}W n tk
r2 Y dS X |j|j }\}}|tj M }|sd| j| n| j |||d f |d k r|
dS dS d S r} ) r9 r" r r r rG r r r~ ru rv r r r r _remove_writer4 s z$BaseSelectorEventLoop._remove_writerc G s | | | j||f| S r+ )rp rD r& r rx ry r r r
add_readerK s
z BaseSelectorEventLoop.add_readerc C s | | | |S r+ )rp r= r& r r r r
remove_readerP s
z#BaseSelectorEventLoop.remove_readerc G s | | | j||f| S r+ )rp r r r r r
add_writerU s
z BaseSelectorEventLoop.add_writerc C s | | | |S r+ )rp r r r r r
remove_writerZ s
z#BaseSelectorEventLoop.remove_writerc s t | | jr"| dkr"tdz ||W S ttfk
rF Y nX | }| }| || j
||| | t
| j| |I d H S Nr the socket must be non-blocking)r rQ
gettimeoutrn rI rK rJ rg r? r
_sock_recvadd_done_callback functoolspartial_sock_read_done)r& r nfutr r r r sock_recv_ s z BaseSelectorEventLoop.sock_recvc C s | | d S r+ )r r& r r r r r r t s z%BaseSelectorEventLoop._sock_read_donec
C s | r d S z||}W n\ ttfk
r4 Y d S ttfk
rL Y n6 tk
rv } z|| W 5 d }~X Y n X || d S r+ ) donerI rK rJ ri rj rh
set_exception
set_result)r& r r r rG re r r r r w s z BaseSelectorEventLoop._sock_recvc s t | | jr"| dkr"tdz ||W S ttfk
rF Y nX | }| }| || j
||| | t
| j| |I d H S r )r rQ r rn recv_intorK rJ rg r? r _sock_recv_intor r r r )r& r bufr r r r r sock_recv_into s z$BaseSelectorEventLoop.sock_recv_intoc
C s | r d S z||}W n\ ttfk
r4 Y d S ttfk
rL Y n6 tk
rv } z|| W 5 d }~X Y n X || d S r+ ) r r rK rJ ri rj rh r r )r& r r r nbytesre r r r r s z%BaseSelectorEventLoop._sock_recv_intoc s t | | jr"| dkr"tdz||}W n ttfk
rL d}Y nX |t|kr^d S | }| }|
t | j
| | || j||t||g |I d H S r )r rQ r rn rO rK rJ lenrg r? r r r _sock_write_doner
_sock_sendall
memoryview)r& r rG r r r r r r sock_sendall s&
z"BaseSelectorEventLoop.sock_sendallc
C s | r d S |d }z|||d }W nb ttfk
rD Y d S ttfk
r\ Y n2 tk
r } z|| W Y d S d }~X Y nX ||7 }|t|kr| d n||d< d S )Nr )
r rO rK rJ ri rj rh r r r )r& r r Zviewposstartr re r r r r s
z#BaseSelectorEventLoop._sock_sendallc s t | | jr"| dkr"tdttdr8|jtjkrf| j||j|j | dI d H }|d \}}}}}|
}| ||| |I d H S )Nr r AF_UNIX)familyprotoloop) r rQ r rn hasattrrB r r Z_ensure_resolvedr rg
_sock_connect)r& r r6 Zresolvedrb r r r r sock_connect s z"BaseSelectorEventLoop.sock_connectc
C s | }z|| W n ttfk
rV |t| j| | || j ||| Y nN t
t fk
rn Y n6 t k
r } z|
| W 5 d }~X Y n X |d d S r+ )r? ZconnectrK rJ r r r r r _sock_connect_cbri rj rh r r )r& r r r6 r re r r r r s z#BaseSelectorEventLoop._sock_connectc C s | | d S r+ )r r r r r r s z&BaseSelectorEventLoop._sock_write_donec
C s | r d S z,|tjtj}|dkr6t|d| W nZ ttfk
rP Y nN tt fk
rh Y n6 t
k
r } z| | W 5 d }~X Y n X | d d S )Nr zConnect call failed )
r Z
getsockoptrB Z
SOL_SOCKETZSO_ERRORrP rK rJ ri rj rh r r )r& r r r6 errre r r r r s z&BaseSelectorEventLoop._sock_connect_cbc sB t | | jr"| dkr"td| }| |d| |I d H S )Nr r F)r rQ r rn rg _sock_accept)r& r r r r r sock_accept s z!BaseSelectorEventLoop.sock_acceptc
C s | }|r| | | r"d S z| \}}|d W nn ttfk
rh | || j|d| Y nR t t
fk
r Y n: t k
r } z| | W 5 d }~X Y nX |
||f d S r} )r? r r r\ rC rK rJ r r ri rj rh r r )r& r Z
registeredr r rc r6 re r r r r * s
z"BaseSelectorEventLoop._sock_acceptc sp | j |j= | }| | I d H z | j|j|||ddI d H W S | |r^| || j |j< X d S )NF)Zfallback) r% _sock_fd
is_reading
pause_reading_make_empty_waiter_reset_empty_waiterresume_readingZ
sock_sendfile_sock)r& Ztranspfileoffsetcountr r r r _sendfile_native< s
z&BaseSelectorEventLoop._sendfile_nativec C s |D ]v\}}|j |j }\}}|tj@ rL|d k rL|jrB| | n
| | |tj@ r|d k r|jrp| | q| | qd S r+ ) fileobjrG r rt Z
_cancelledr= Z
_add_callbackr r )r& Z
event_listr rz r r{ r| r r r _process_eventsJ s
z%BaseSelectorEventLoop._process_eventsc C s | | | d S r+ )r= r? r; )r& r r r r
_stop_servingX s z#BaseSelectorEventLoop._stop_serving)N)N)N)NNN),r!
__module__ __qualname__r r/ r ZSSL_HANDSHAKE_TIMEOUTr4 r7 r; r: r# rH rE rR rW rT ra rp rD r= r r r r r r r r r r r r r r r r r r r r r r
__classcell__r r r' r r 0 s|
.
) r c s e Zd ZdZeZdZd fdd Zdd Zdd Z d d
Z
d d Z d
d Z dd Z
ejfddZd ddZdd Zdd Zdd Zd d Z ZS ) _SelectorTransporti Nc s t || t|| jd< z| | jd< W n tk
rN d | jd< Y nX d| jkrz| | jd< W n tj k
r d | jd< Y nX || _
| | _ d| _
| | || _| | _d| _d| _| jd k r| j | |j| j < d S )NrB ZsocknamerZ Fr )r r r r` _extraZ getsocknamerP Z getpeernamerB errorr r? r _protocol_connected set_protocol_server_buffer_factory_buffer
_conn_lost_closingZ_attachr% )r& r r r- r) r* r' r r r i s,
z_SelectorTransport.__init__c C s | j jg}| jd kr |d n| jr0|d |d| j | jd k r| j st| jj | jt
j }|rz|d n
|d t| jj | jt
j }|rd}nd}|
}|d| d | d
d d |S )
Nclosedclosingzfd=z read=pollingz read=idlepollingZidlezwrite=z<{}> )r r! r appendr r _loopr9 r r" r rt r get_write_buffer_sizeformatjoin)r& infor statebufsizer r r __repr__ s0
z_SelectorTransport.__repr__c C s | d d S r+ ) _force_closer< r r r abort s z_SelectorTransport.abortc C s || _ d| _d S NT) _protocolr r& r- r r r r s z _SelectorTransport.set_protocolc C s | j S r+ )r r< r r r get_protocol s z _SelectorTransport.get_protocolc C s | j S r+ )r r< r r r ro s z _SelectorTransport.is_closingc C sT | j r
d S d| _ | j| j | jsP| jd7 _| j| j | j| jd d S NTr ) r r r= r r r r call_soon_call_connection_lostr< r r r r; s z_SelectorTransport.closec C s, | j d k r(|d| t| d | j d S )Nzunclosed transport )source)r ResourceWarningr; )r& Z_warnr r r __del__ s
z_SelectorTransport.__del__Fatal error on transportc C sN t |tr(| j r@tjd| |dd n| j||| | jd | | d S )Nz%r: %sTrM )rX rY rf r- ) r rP r get_debugr r r_ r r )r& re rX r r r _fatal_error s
z _SelectorTransport._fatal_errorc C sd | j r
d S | jr(| j | j| j | jsBd| _| j| j | j d7 _ | j| j | d S r )
r r clearr r r r r= r r r& re r r r r s
z _SelectorTransport._force_closec C sV z| jr| j| W 5 | j d | _ d | _d | _| j}|d k rP| d | _X d S r+ )r r; r r r Z_detachr Zconnection_lost)r& re r* r r r r s
z(_SelectorTransport._call_connection_lostc C s
t | jS r+ )r r r< r r r r s z(_SelectorTransport.get_write_buffer_sizec G s" | j r
d S | jj||f| d S r+ )r r rD r r r r rD s z _SelectorTransport._add_reader)NN)r )r! r r max_size bytearrayr r r r r r r ro r; warningswarnr r r r r rD r r r r' r r ] s
r c s e Zd ZdZejjZd# fdd Z fddZ dd Z
d d
Z d d Z d
d Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Z fd d Zd d Zd!d" Z ZS )$r, TNc s~ d | _ t ||||| d| _d| _d | _t| j | j
| j j | | j
| j
| j| j |d k rz| j
tj|d d S r
)_read_ready_cbr r _eof_paused
_empty_waiterr Z _set_nodelayr r r r connection_maderD r _read_readyr _set_result_unless_cancelled)r& r r r- r. r) r* r' r r r s
z!_SelectorSocketTransport.__init__c s. t |tjr| j| _n| j| _t | d S r+ )r r ZBufferedProtocol_read_ready__get_bufferr _read_ready__data_receivedr r r r' r r r s
z%_SelectorSocketTransport.set_protocolc C s | j o| j S r+ )r r r< r r r r s z#_SelectorSocketTransport.is_readingc C s> | j s | jrd S d| _| j| j | j r:td| d S )NTz%r pauses reading)r r r r= r r r r r< r r r r s
z&_SelectorSocketTransport.pause_readingc C s@ | j s | jsd S d| _| | j| j | j r Y n4 tk
rp } z| |d W Y d S d }~X Y nX |r| j
| j n| d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.)
r r r r r Z eof_receivedri rj rh r r= r r; )r& Z keep_openre r r r r e s
z,_SelectorSocketTransport._read_ready__on_eofc
C s6 t |tttfs$tdt|j| jr2td| j d k rDtd|sLd S | j
rz| j
t j krht
d | j
d7 _
d S | js z| j|}W nb ttfk
r Y nb ttfk
r Y nJ tk
r } z| |d W Y d S d }~X Y nX ||d }|s d S | j| j| j | j | | d S )N/data argument must be a bytes-like object, not z%Cannot call write() after write_eof()z(unable to write; sendfile is in progress socket.send() raised exception.r %Fatal write error on socket transport) r bytesr r r typer! r r8 r r r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr warningr r rO rK rJ ri rj rh r r r r _write_readyextend_maybe_pause_protocol)r& rG r re r r r writez s:
z _SelectorSocketTransport.writec
C s | j r
d S z| j| j}W n ttfk
r4 Y n ttfk
rL Y n tk
r } z>| j
| j | j |
|d | jd k r| j| W 5 d }~X Y nnX |r| jd |= | | js| j
| j | jd k r| jd | jr| d n| jr| jtj d S )Nr )r r rO r rK rJ ri rj rh r r r r r r r _maybe_resume_protocolr r r r shutdownrB SHUT_WR)r& r re r r r r s2
z%_SelectorSocketTransport._write_readyc C s. | j s | jrd S d| _| js*| jtj d S r )r r r r r rB r r< r r r write_eof s
z"_SelectorSocketTransport.write_eofc C s dS r r r< r r r
can_write_eof s z&_SelectorSocketTransport.can_write_eofc s* t | | jd k r&| jtd d S )Nz Connection is closed by peer)r r r r ConnectionErrorr r' r r r s
z._SelectorSocketTransport._call_connection_lostc C s6 | j d k rtd| j | _ | js0| j d | j S )NzEmpty waiter is already set)r r8 r rg r r r< r r r r s
z+_SelectorSocketTransport._make_empty_waiterc C s
d | _ d S r+ )r r< r r r r s z,_SelectorSocketTransport._reset_empty_waiter)NNN)r! r r Z_start_tls_compatibler Z
_SendfileModeZ
TRY_NATIVEZ_sendfile_compatibler r r r r r r r r r r r r r r r r r r r' r r, s* %' r, c sF e Zd ZejZd fdd Zdd Zdd Zd
dd Z d
d Z
Z S )r5 Nc s^ t |||| || _| j| jj| | j| j| j| j |d k rZ| jt
j |d d S r+ ) r r _addressr r r r rD r r r r )r& r r r- r6 r. r) r' r r r s
z#_SelectorDatagramTransport.__init__c C s t dd | jD S )Nc s s | ]\}}t |V qd S r+ )r ).0rG rb r r r s zC_SelectorDatagramTransport.get_write_buffer_size..)sumr r< r r r r s z0_SelectorDatagramTransport.get_write_buffer_sizec
C s | j r
d S z| j| j\}}W n ttfk
r8 Y n tk
rd } z| j| W 5 d }~X Y nT t t
fk
r| Y n< t k
r } z| |d W 5 d }~X Y nX | j
|| d S )Nz&Fatal read error on datagram transport)r r Zrecvfromr rK rJ rP r error_receivedri rj rh r Zdatagram_receivedr& rG rd re r r r r s z&_SelectorDatagramTransport._read_readyc
C s t |tttfs$tdt|j|s,d S | jrV|d | jfkrPtd| j | j}| j r| jr| j t
j krxt
d | j d7 _ d S | jslz,| jd r| j| n| j|| W d S ttfk
r | j| j| j Y n tk
r } z| j| W Y d S d }~X Y nP t t fk
r6 Y n6 t k
rj } z| |d W Y d S d }~X Y nX | j t||f | ! d S )Nr z!Invalid address: must be None or r r rZ 'Fatal write error on datagram transport)"r r r r r r r! r
rn r r r r r r r r rO sendtorK rJ r r r
_sendto_readyrP r r ri rj rh r r r r r r r r sH
z!_SelectorDatagramTransport.sendtoc
C s | j r| j \}}z*| jd r.| j| n| j|| W q ttfk
rj | j ||f Y qY q t k
r } z| j
| W Y d S d }~X Y q t t
fk
r Y q tk
r } z| |d W Y d S d }~X Y q X q | | j s| j| j | jr| d d S )NrZ r )r popleftr r rO r rK rJ
appendleftrP r r ri rj rh r r r r r r r r r r r r * s2
z(_SelectorDatagramTransport._sendto_ready)NNN)N) r! r r collectionsdequer r r r r r r r r r' r r5 s
+r5 ) __all__r r^ r r rB r r$ r ImportError r r r r r r r
r logr r r Z
BaseEventLoopr Z_FlowControlMixinZ Transportr r, r5 r r r r sD
1 o