o
0i @ s$ d Z dZddlZddlZddlZddlZddlZddlZddlZzddl Z W n e
y1 dZ Y nw ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dd ZG dd de jZG dd dejejZG dd deZ G dd deZ dS )zEvent loop using a selector and related classes.
A selector is a "notify-when-ready" multiplexer. For a subclass which
also includes support for signal handling, see the unix_events sub-module.
)BaseSelectorEventLoop N ) base_events) constants)events)futures) protocols)sslproto)
transports)trsock)loggerc C s2 z| |}W n
ty Y dS w t|j|@ S NF)get_keyKeyErrorboolr )selectorfdZeventkey r >/opt/alt/python310/lib64/python3.10/asyncio/selector_events.py_test_selector_event s r c s e Zd ZdZdS fdd ZdSdddddZ dSddddejd d
d Z dTd d
Z fddZ
dd Z dd Z dd Z
dd Zdd Zdddejfdd Zdddejfd d Zddejfd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Z dSd5d6Z d7d8 Z d9d: Z d;d< Z d=d> Z!d?d@ Z"dAdB Z#dCdD Z$dSdEdFZ%dGdH Z&dIdJ Z'dKdL Z(dMdN Z)dOdP Z*dQdR Z+ Z,S )Ur zJSelector event loop.
See events.EventLoop for API specification.
Nc sF t |d u r
t }td|jj || _| t
| _ d S )NzUsing selector: %s)
super__init__ selectorsZDefaultSelectorr debug __class____name__ _selector_make_self_pipeweakrefZWeakValueDictionary _transports)selfr r r r r 1 s
z BaseSelectorEventLoop.__init__extraserverc C t | |||||S N)_SelectorSocketTransport)r! sockprotocolwaiterr$ 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)r/ r# )r Z SSLProtocolr( Z_app_transport) r! Zrawsockr*
sslcontextr+ r- r. r$ r% r/ Z ssl_protocolr r r _make_ssl_transport@ s z)BaseSelectorEventLoop._make_ssl_transportc C r& r' )_SelectorDatagramTransport)r! r) r* addressr+ r$ r r r _make_datagram_transportM s z.BaseSelectorEventLoop._make_datagram_transportc sP | rtd| rd S | t | jd ur&| j d | _d S d S )Nz!Cannot close a running event loop)Z
is_running RuntimeError is_closed_close_self_piper closer r! r" r r r8 R s
zBaseSelectorEventLoop.closec C sB | | j | j d | _| j d | _| jd8 _d S )Nr )_remove_reader_ssockfilenor8 _csock
_internal_fdsr9 r r r r7 ] s
z&BaseSelectorEventLoop._close_self_pipec C sN t \| _| _| jd | jd | jd7 _| | j | j d S )NFr ) socketZ
socketpairr; r= setblockingr> _add_readerr< _read_from_selfr9 r r r r e s
z%BaseSelectorEventLoop._make_self_pipec C s d S r' r r! datar r r _process_self_datam z(BaseSelectorEventLoop._process_self_datac C sN z| j d}|s
W d S | | W n ty Y q ty% Y d S w q)NTi )r; recvrE InterruptedErrorBlockingIOErrorrC r r r rB p s z%BaseSelectorEventLoop._read_from_selfc C sR | j }|d u r d S z|d W d S ty( | jr%tjddd Y d S Y d S w )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' )rA r< _accept_connection)r! protocol_factoryr) r0 r% backlogr/ r r r _start_serving s
z$BaseSelectorEventLoop._start_servingc
C s t |D ]}}z| \}} | jrtd|| | |d W nO tttfy, Y d S t ym }
z6|
j
t
j t
j t
j
t
jfv rb| d|
t|d | | | tj| j|||||| n W Y d }
~
qd }
~
ww d| i} | ||| |||} | | qd S )Nz#%r got a new connection from %r: %rFz&socket.accept() out of system resource)message exceptionr? peername)rangeacceptrO r r r@ rI rH ConnectionAbortedErrorrN errnoZEMFILEZENFILEZENOBUFSZENOMEMcall_exception_handlerr TransportSocketr: r< Z
call_laterr ZACCEPT_RETRY_DELAYrU _accept_connection2Z create_task)
r! rS r) r0 r% rT r/ _connaddrexcr$ rZ r r r rR sH
z(BaseSelectorEventLoop._accept_connectionc
s d }d }z7| }| } |r | j|||| d|||d}n
| j||| ||d}z | I d H W W d S ty< | w ttfyF ty| }
z+| jrqd|
d} |d ur]|| d< |d ure|| d< | | W Y d }
~
d S W Y d }
~
d S d }
~
ww )NT)r+ r- r$ r% r/ )r+ r$ r% z3Error on transport creation for incoming connection)rV rW r* transport)
create_futurer1 r,
BaseExceptionr8
SystemExitKeyboardInterruptrO r] ) r! rS ra r$ r0 r% r/ r* rd r+ rc contextr r r r_ sJ z)BaseSelectorEventLoop._accept_connection2c
C s |}t |ts"zt| }W n tttfy! td|d w z| j| }W n
ty3 Y d S w | sBt d|d|d S )NzInvalid file object: zFile descriptor z is used by transport )
isinstanceintr< AttributeError TypeError
ValueErrorr r
is_closingr5 )r! r r< rd r r r _ensure_fd_no_transport s&
z-BaseSelectorEventLoop._ensure_fd_no_transportc G s | t||| d }z| j|}W n ty) | j|tj|d f Y |S w |j|j }\}}| j
||tjB ||f |d urH| |S r' )
_check_closedr Handler r r registerr
EVENT_READrD modifycancel r! r callbackargshandler maskreaderwriterr r r rA " z!BaseSelectorEventLoop._add_readerc C s | rdS z| j|}W n
ty Y dS w |j|j}\}}|tj M }|s1| j| n
| j ||d |f |d urE|
dS dS )NFT) r6 r r r r rD r rt
unregisterru rv r! r r r{ r| r} r r r r: s z$BaseSelectorEventLoop._remove_readerc G s | t||| d }z| j|}W n ty) | j|tjd |f Y |S w |j|j }\}}| j
||tjB ||f |d urH| |S r' ) rq r rr r r r rs r EVENT_WRITErD ru rv rw r r r _add_writer! r~ z!BaseSelectorEventLoop._add_writerc C s | rdS z| j|}W n
ty Y dS w |j|j}\}}|tj M }|s1| j| n
| j |||df |durE|
dS dS )Remove a writer callback.FNT) r6 r r r r rD r r r ru rv r r r r _remove_writer1 s z$BaseSelectorEventLoop._remove_writerc G " | | | j||g|R dS )zAdd a reader callback.N)rp rA r! r rx ry r r r
add_readerH
z BaseSelectorEventLoop.add_readerc C | | | |S )zRemove a reader callback.)rp r: r! r r r r
remove_readerM
z#BaseSelectorEventLoop.remove_readerc G r )zAdd a writer callback..N)rp r r r r r
add_writerR r z BaseSelectorEventLoop.add_writerc C r )r )rp r r r r r
remove_writerW r z#BaseSelectorEventLoop.remove_writerc t | | jr| dkrtdz||W S ttfy$ Y nw | }| }|
| | || j |||}|
tj| j||d |I dH S )zReceive data from the socket.
The return value is a bytes object representing the data received.
The maximum amount of data to be received at once is specified by
nbytes.
r the socket must be non-blockingrz N)r _check_ssl_socketrO
gettimeoutrn rG rI rH re r< rp rA
_sock_recvadd_done_callback functoolspartial_sock_read_done)r! r) nfutr rz r r r sock_recv\ s"
z BaseSelectorEventLoop.sock_recvc C " |d u s| s| | d S d S r' ) cancelledr r! r r rz r r r r r z%BaseSelectorEventLoop._sock_read_donec
C | rd S z||}W n, ttfy Y d S ttfy" ty9 } z || W Y d }~d S d }~ww || d S r' ) donerG rI rH rg rh rf
set_exception
set_result)r! r r) r rD rc r r r r v s z BaseSelectorEventLoop._sock_recvc r )zReceive data from the socket.
The received data is written into *buf* (a writable buffer).
The return value is the number of bytes written.
r r r N)r r rO r rn recv_intorI rH re r< rp rA _sock_recv_intor r r r )r! r) bufr r rz r r r sock_recv_into s"
z$BaseSelectorEventLoop.sock_recv_intoc
C r r' ) r r rI rH rg rh rf r r )r! r r) r nbytesrc r r r r s z%BaseSelectorEventLoop._sock_recv_intoc s t | | jr| dkrtdz||}W n
ttfy' d}Y nw |t|kr0dS | }|
}| | | || j
||t||g}|tj| j||d |I dH S )a Send data to the socket.
The socket must be connected to a remote socket. This method continues
to send data from data until either all data has been sent or an
error occurs. None is returned on success. On error, an exception is
raised, and there is no way to determine how much data, if any, was
successfully processed by the receiving end of the connection.
r r Nr )r r rO r rn rM rI rH lenre r< rp r
_sock_sendall
memoryviewr r r _sock_write_done)r! r) rD r r r rz r r r sock_sendall s*
z"BaseSelectorEventLoop.sock_sendallc
C s | rd S |d }z |||d }W n, ttfy! Y d S ttfy* tyA } z || W Y d }~d S d }~ww ||7 }|t|krS| d d S ||d< d S )Nr )
r rM rI rH rg rh rf r r r )r! r r) Zviewposstartr rc r r r r s$
z#BaseSelectorEventLoop._sock_sendallc s t | | jr| dkrtd|jtjks"t jr;|jtj kr;| j
||j|j |j | dI dH }|d \}}}}}|
}| ||| z|I dH W d}S d}w )zTConnect to a remote socket at address.
This method is a coroutine.
r r )familytypeprotoloopN)r r rO r rn r r? ZAF_INETZ _HAS_IPv6ZAF_INET6Z_ensure_resolvedr r re
_sock_connect)r! r) r3 Zresolvedr` r r r r sock_connect s$
z"BaseSelectorEventLoop.sock_connectc
C s | }zaz|| W nE ttfy2 | | | || j|||}|tj | j
||d Y n* t t fy; t
yQ } z || W Y d }~nd }~ww |d W d }d S W d }d S W d }d S d }w )Nr )r< ZconnectrI rH rp r _sock_connect_cbr r r r rg rh rf r r )r! r r) r3 r rz rc r r r r s.
z#BaseSelectorEventLoop._sock_connectc C r r' )r r r r r r r r z&BaseSelectorEventLoop._sock_write_donec
C s | rd S zUz|tjtj}|dkr t|d| W n* ttfy( Y n* tt fy1 t
yG } z | | W Y d }~nd }~ww | d W d }d S W d }d S W d }d S d }w )Nr zConnect call failed )
r Z
getsockoptr? Z
SOL_SOCKETZSO_ERRORrN rI rH rg rh rf r r )r! r r) r3 errrc r r r r s* z&BaseSelectorEventLoop._sock_connect_cbc sD t | | jr| dkrtd| }| || |I dH S )aW Accept a connection.
The socket must be bound to an address and listening for connections.
The return value is a pair (conn, address) where conn is a new socket
object usable to send and receive data on the connection, and address
is the address bound to the socket on the other end of the connection.
r r N)r r rO r rn re _sock_accept)r! r) r r r r sock_accept* s
z!BaseSelectorEventLoop.sock_acceptc
C s | }z
| \}}|d W nF ttfy7 | | | || j||}|t j
| j ||d Y d S t t
fy@ tyW } z || W Y d }~d S d }~ww |||f d S )NFr )r< rZ r@ rI rH rp rA r r r r r rg rh rf r r )r! r r) r ra r3 rz rc r r r r 9 s"
z"BaseSelectorEventLoop._sock_acceptc s | j |j= | }| | I d H z | j|j|||ddI d H W | |r.| || j |j< S | |r?| || j |j< w )NF)Zfallback) r _sock_fd
is_reading
pause_reading_make_empty_waiterZ
sock_sendfile_sock_reset_empty_waiterresume_reading)r! Ztranspfileoffsetcountr r r r _sendfile_nativeJ s"
z&BaseSelectorEventLoop._sendfile_nativec C s |D ];\}}|j |j}\}}|tj@ r&|d ur&|jr!| | n| | |tj@ r=|d ur=|jr8| | q| | qd S r' ) fileobjrD r rt Z
_cancelledr: Z
_add_callbackr r )r! Z
event_listr r{ r r| r} r r r _process_eventsX s
z%BaseSelectorEventLoop._process_eventsc C s | | | d S r' )r: r< r8 )r! r) r r r
_stop_servingf s z#BaseSelectorEventLoop._stop_servingr' NNN)-r
__module__ __qualname____doc__r r, r ZSSL_HANDSHAKE_TIMEOUTr1 r4 r8 r7 r rE rB rP rU rR r_ rp rA 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 + sl
.
)
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 ty& d | jd< Y nw d| jvrEz | | jd< W n tj yD d | jd< Y nw || _
| | _ d| _
| | || _| | _d| _d| _| jd urm| j | |j| j < d S )Nr? ZsocknamerX Fr )r r r r^ _extraZ getsocknamerN Z getpeernamer? 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 w s0
z_SelectorTransport.__init__c C s | j jg}| jd u r|d n| jr|d |d| j | jd urc| j sct| jj | jt
j }|r=|d n|d t| jj | jt
j }|rQd}nd}|
}|d| d | d
d d |S )
Nclosedclosingzfd=z read=pollingz read=idlepollingZidlezwrite=z<{}> )r r r appendr r _loopr6 r r r rt r get_write_buffer_sizeformatjoin)r! infor statebufsizer r r __repr__ s.
z_SelectorTransport.__repr__c C s | d d S r' ) _force_closer9 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 | j S r' )r r9 r r r get_protocol z _SelectorTransport.get_protocolc C r r' )r r9 r r r ro r z _SelectorTransport.is_closingc C sX | j rd S d| _ | j| j | js*| jd7 _| j| j | j| jd d S d S NTr ) r r r: r r r r call_soon_call_connection_lostr9 r r r r8 s z_SelectorTransport.closec C s0 | j d ur|d| t| d | j d S d S )Nzunclosed transport )source)r ResourceWarningr8 )r! Z_warnr r r __del__ s
z_SelectorTransport.__del__Fatal error on transportc C sN t |tr| j rtjd| |dd n | j||| | jd | | d S )Nz%r: %sTrK )rV rW rd r* ) rj rN r get_debugr r r] r r )r! rc rV r r r _fatal_error s
z _SelectorTransport._fatal_errorc C sd | j rd S | jr| j | j| j | js!d| _| j| j | j d7 _ | j| j | d S r )
r r clearr r r r r: r r r! rc r r r r s
z _SelectorTransport._force_closec C s z*| j r
| j| W | j d | _d | _d | _| j}|d ur)| d | _d S d S | j d | _d | _d | _| j}|d urG| d | _w r' )r r Zconnection_lostr r8 r r Z_detach)r! rc r% r r r r s*
z(_SelectorTransport._call_connection_lostc C s
t | jS r' )r r r9 r r r r
z(_SelectorTransport.get_write_buffer_sizec G s$ | j rd S | jj||g|R d S r' )r r rA r r r r rA s z _SelectorTransport._add_reader)NN)r )r r r max_size bytearrayr r r r r r r ro r8 warningswarnr r r r r rA r r r r" r r k s"
r c s e Zd ZdZejjZ d# 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 ur?| j
tj|d d S d S r
)_read_ready_cbr r _eof_paused
_empty_waiterr Z _set_nodelayr r r r connection_maderA 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' )rj 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 r9 r r r r s z#_SelectorSocketTransport.is_readingc C sB | j s| jrd S d| _| j| j | j r td| d S d S )NTz%r pauses reading)r r r r: r r r r r9 r r r r " s
z&_SelectorSocketTransport.pause_readingc C sD | j s| jsd S d| _| | j| j | j r td| d S d S )NFz%r resumes reading) r r rA r r r r r r r9 r r r r * s
z'_SelectorSocketTransport.resume_readingc C s | d S r' )r r9 r r r r 2 s z$_SelectorSocketTransport._read_readyc
C sD | j rd S z| jd}t|stdW n" ttfy ty7 } z
| |d W Y d }~d S d }~ww z| j
|}W n- t t fyL Y d S ttfyU tym } z
| |d W Y d }~d S d }~ww |sv|
d S z | j| W d S ttfy ty } z
| |d W Y d }~d S d }~ww )Nz%get_buffer() returned an empty bufferz/Fatal error: protocol.get_buffer() call failed.$Fatal read error on socket transportz3Fatal error: protocol.buffer_updated() call failed.)r r Z
get_bufferr r5 rg rh rf r r r rI rH _read_ready__on_eofZbuffer_updated)r! r rc r r r r r 5 sP z0_SelectorSocketTransport._read_ready__get_bufferc
C s | j rd S z | j| j}W n- ttfy Y d S ttfy# ty; } z
| |d W Y d }~d S d }~ww |sD|
d S z | j | W d S ttfyW tyo } z
| |d W Y d }~d S d }~ww )Nr z2Fatal error: protocol.data_received() call failed.)
r r rG r rI rH rg rh rf r r r Z
data_received)r! rD rc r r r r Z s4 z3_SelectorSocketTransport._read_ready__data_receivedc
C s | j r td| z| j }W n" ttfy ty4 } z
| |d W Y d }~d S d }~ww |r@| j
| j d S | d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.)
r r r r r Z eof_receivedrg rh rf r r: r r8 )r! Z keep_openrc r r r r s s
z,_SelectorSocketTransport._read_ready__on_eofc
C s( t |tttfstdt|j| jrtd| j d ur"td|s&d S | j
r=| j
t j kr4t
d | j
d7 _
d S | jsz| j|}W n, ttfyS Y n, ttfy\ tyt } z
| |d W Y d }~d S d }~ww ||d }|sd 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) rj bytesr r rm r r r r5 r r r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr warningr r rM rI rH rg rh rf r r r r _write_readyextend_maybe_pause_protocol)r! rD r rc r r r write sB
z _SelectorSocketTransport.writec
C s8 | j sJ d| jr d S z | j| j }W nK ttfy! Y d S ttfy* ty` } z+| j
| j | j |
|d | jd urU| j| W Y d }~d S W Y d }~d S d }~ww |ri| j d |= | | j s| j
| j | jd ur| jd | jr| d d S | jr| jtj d S d S d S )NzData should not be emptyr )r r r rM rI rH rg rh rf r r r r r r r _maybe_resume_protocolr r r r shutdownr? SHUT_WR)r! r rc r r r r s>
z%_SelectorSocketTransport._write_readyc C s2 | j s| jrd S d| _| js| jtj d S d S r )r r r r r r? r r9 r r r write_eof s z"_SelectorSocketTransport.write_eofc C s dS r r r9 r r r
can_write_eof rF z&_SelectorSocketTransport.can_write_eofc s. t | | jd ur| jtd d S 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 ur td| j | _ | js| j d | j S )NzEmpty waiter is already set)r r5 r re r r r9 r r r r s
z+_SelectorSocketTransport._make_empty_waiterc C s
d | _ d S r' )r r9 r r r r r z,_SelectorSocketTransport._reset_empty_waiterr )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 sJ e Zd ZejZ d fdd Zdd Zdd Zd
dd Z d
d Z
Z S )r2 Nc sb t |||| || _| j| jj| | j| j| j| j |d ur/| jt
j |d d S d S r' ) r r _addressr r r r rA r r r r )r! r r) r* r3 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 ).0rD r` r r r s zC_SelectorDatagramTransport.get_write_buffer_size..)sumr r9 r r r r s z0_SelectorDatagramTransport.get_write_buffer_sizec
C s | j rd S z | j| j\}}W nE ttfy Y d S ty4 } z
| j| W Y d }~d S d }~w t t
fy= t yU } z
| |d W Y d }~d S d }~ww | j
|| d S )Nz&Fatal read error on datagram transport)r r Zrecvfromr rI rH rN r error_receivedrg rh rf r Zdatagram_receivedr! rD rb rc r r r r s z&_SelectorDatagramTransport._read_readyc
C s| t |tttfstdt|j|sd S | jr+|d | jfvr(td| j | j}| j rE| jrE| j t
j kr
z(_SelectorDatagramTransport._sendto_readyr r' ) r r r collectionsdequer r r r r r! r r r r" r r2 s
+r2 ) r __all__r$ r\ r r r? r r Zssl ImportError r r r r r r r
r logr r Z
BaseEventLoopr Z_FlowControlMixinZ Transportr r( r2 r r r r sH D o