o
0i @ s" d Z ddlZddlZddlZddlZddlZddlZddlZzddlZW n e y0 dZY nw 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 ZG dd de jZG dd dejejZG dd deZG dd deZ dS ))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 Z dR fdd ZdRdddddZ dRddddejdd d
Z dSd d Z fd
dZ dd Z
dd Z dd Z dd Z
dd ZdddejfddZdddejfd d Zddejfd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 ZdRd4d5Z d6d7 Z d8d9 Z d:d; Z dd? Z!d@dA Z"dBdC Z#dRdDdEZ$dFdG Z%dHdI Z&dJdK Z'dLdM Z(dNdO Z)dPdQ Z* Z+S )Tr 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 }|s2| j| n
| j |||d f |d urF|
dS dS r ) 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 d S r' )rp rA r! r rx ry r r r
add_readerH
z BaseSelectorEventLoop.add_readerc C | | | |S r' )rp r: r! r r r r
remove_readerM
z#BaseSelectorEventLoop.remove_readerc G r r' )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 d H S Nr the socket must be non-blockingrz )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 r )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|kr1d S | }|
}| | | || j
||t||g}|tj| j||d |I d H S r )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 d H }|d \}}}}}|
}| ||| z|I d H W d }S d }w )Nr r )familytypeprotoloop)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 sF t | | jr| dkrtd| }| || |I d H S )Nr r )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__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 s* | j rd S z | j| j}W nK ttfy Y d S ttfy# tyY } z+| j
| j | j |
|d | jd urN| j| W Y d }~d S W Y d }~d S d }~ww |rb| jd |= | | js| j
| j | jd ur{| jd | jr| d d S | jr| jtj d S d S d S )Nr )r r rM r 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 ) __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