o
0i @ sX d Z dZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl m Z ddl m
Z
dd l mZ dd
l mZ dd l mZ dd l mZ dd
lmZ dd ZG dd dejejZG dd deejZG dd deejZG dd deZ G dd deej Z G dd deeej Z G d d d eeej Z!G d d d e
j"Z#dS ) 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) constants)futures)
exceptions) protocols)sslproto)
transports)trsock)loggerc C s t || jd< z | | jd< W n tjy( | j r&tj d|dd Y nw d| jvrIz
|
| jd< W d S tjyH d | jd< Y d S w d S )NsocketZsocknamezgetsockname() failed on %rTexc_infopeername) r TransportSocket_extraZ getsocknamer
error_loop get_debugr warningZ getpeername) transportsock r >/opt/alt/python310/lib64/python3.10/asyncio/proactor_events.py_set_socket_extra s"
r c s e Zd ZdZ d fdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
e j fddZ
d ddZdd Zdd Zdd Z ZS ) _ProactorBasePipeTransportz*Base class for pipe and socket transports.Nc s t || | | || _| | || _d | _d | _d | _d| _ d| _
d| _ d| _ d| _
| jd ur9| j | j| jj| |d urQ| jtj|d d S d S )Nr F)super__init__
_set_extra_sock set_protocol_server_buffer _read_fut
_write_fut_pending_write
_conn_lost_closing_called_connection_lost _eof_writtenZ_attachr call_soon _protocolZconnection_mader Z _set_result_unless_cancelledselfloopr protocolwaiterextraserver __class__r r r 2 s*
z#_ProactorBasePipeTransport.__init__c C s | j jg}| jd u r|d n| jr|d | jd ur(|d| j | jd ur6|d| j | jd urD|d| j | jrR|dt | j | j
rZ|d d d |S )
Nclosedclosingzfd=zread=zwrite=zwrite_bufsize=z EOF writtenz<{}> )
r5 __name__r appendr( filenor$ r% r# lenr* formatjoin)r. infor r r __repr__I s
z#_ProactorBasePipeTransport.__repr__c C s || j d< d S )Npipe)r r. r r r r r [ z%_ProactorBasePipeTransport._set_extrac C s
|| _ d S Nr, )r. r0 r r r r! ^
z'_ProactorBasePipeTransport.set_protocolc C | j S rD rE r. r r r get_protocola z'_ProactorBasePipeTransport.get_protocolc C rG rD )r( rH r r r
is_closingd rJ z%_ProactorBasePipeTransport.is_closingc C s` | j rd S d| _ | jd7 _| js | jd u r | j| jd | jd ur.| j d | _d S d S )NTr ) r( r' r# r% r r+ _call_connection_lostr$ cancelrH r r r closeg s
z _ProactorBasePipeTransport.closec C s0 | j d ur|d| t| d | j d S d S )Nzunclosed transport )source)r ResourceWarningrN )r. Z_warnr r r __del__r s
z"_ProactorBasePipeTransport.__del__ Fatal error on pipe transportc C sn z0t |tr| j rtjd| |dd n| j||| | jd W | | d S W | | d S | | w )Nz%r: %sTr )message exceptionr r0 )
isinstanceOSErrorr r r debugcall_exception_handlerr, _force_close)r. excrS r r r _fatal_errorw s
z'_ProactorBasePipeTransport._fatal_errorc C s | j d ur| j s|d u r| j d n| j | | jr#| jr#d S d| _| jd7 _| jr8| j d | _| j rC| j d | _ d| _
d | _ | j
| j| d S )NTr r )
_empty_waiterdone
set_resultZ
set_exceptionr( r) r' r% rM r$ r& r# r r+ rL )r. rZ r r r rY s"
z'_ProactorBasePipeTransport._force_closec C s | j rd S z6| j| W t| jdr!| j dkr!| jtj | j d | _| j
}|d ur7| d | _
d| _ d S t| jdrP| j dkrP| jtj | j d | _| j
}|d urf| d | _
d| _ w )NshutdownT) r) r, Zconnection_losthasattrr r; r_ r
Z SHUT_RDWRrN r" Z_detach)r. rZ r3 r r r rL s,
z0_ProactorBasePipeTransport._call_connection_lostc C s" | j }| jd ur|t| j7 }|S rD )r& r# r< )r. sizer r r get_write_buffer_size s
z0_ProactorBasePipeTransport.get_write_buffer_sizeNNN)rR )r9
__module__ __qualname____doc__r r@ r r! rI rK rN warningswarnrQ r[ rY rL rc
__classcell__r r r4 r r . s
r c sX e Zd ZdZ d fdd Zdd Zdd Zd
d Zd d
Zdd Z dddZ
Z S )_ProactorReadPipeTransportzTransport for read pipes.N c sD d| _ d| _t |||||| t|| _| j| j d| _d S )Nr` TF) _pending_data_length_pausedr r bytearray_datar r+
_loop_reading)r. r/ r r0 r1 r2 r3 Z buffer_sizer4 r r r s
z#_ProactorReadPipeTransport.__init__c C s | j o| j S rD )rn r( rH r r r
is_reading s z%_ProactorReadPipeTransport.is_readingc C s4 | j s| jrd S d| _| j rtd| d S d S )NTz%r pauses reading)r( rn r r r rW rH r r r
pause_reading s
z(_ProactorReadPipeTransport.pause_readingc C s~ | j s| jsd S d| _| jd u r| j| jd | j}d| _|dkr0| j| j| jd | | | j r=t
d| d S d S )NFr` z%r resumes reading) r( rn r$ r r+ rq rm _data_receivedrp r r rW )r. lengthr r r resume_reading s
z)_ProactorReadPipeTransport.resume_readingc
C s~ | j r td| z| j }W n" ttfy ty4 } z
| |d W Y d }~d S d }~ww |s=|
d S d S )Nz%r received EOFz1Fatal error: protocol.eof_received() call failed.) r r r rW r, Z eof_received
SystemExitKeyboardInterrupt
BaseExceptionr[ rN )r. Z keep_openrZ r r r
_eof_received s
z(_ProactorReadPipeTransport._eof_receivedc
C s | j r|| _d S |dkr| d S t| jtjrFz
t| j| W d S tt fy- t
yE } z
| |d W Y d }~d S d }~ww | j | d S )Nr z3Fatal error: protocol.buffer_updated() call failed.)
rn rm rz rU r, r ZBufferedProtocolZ _feed_data_to_buffered_protorw rx ry r[ Z
data_received)r. dataru rZ r r r rt s& z)_ProactorReadPipeTransport._data_receivedc
C s d}d }zzR|d ur5d | _ | r1| }|dkr)W W |dkr'| || d S d S | jd | }n| | jrHW W |dkrF| || d S d S | jsV| jj
| j | j| _ W nb t y } z | jsi|
|d n | j rutjddd W Y d }~nId }~w ty } z | | W Y d }~n3d }~w ty } z |
|d W Y d }~n d }~w tjy | js Y n w | js| j | j W |dkr| || d S d S |dkr| || w w )Nr` r z"Fatal read error on pipe transportz*Read error on pipe transport while closingTr )r$ r] resultrt rp rM r( rn r _proactor recv_intor ConnectionAbortedErrorr[ r r rW ConnectionResetErrorrY rV r CancelledErroradd_done_callbackrq )r. futru r{ rZ r r r rq sb "
z(_ProactorReadPipeTransport._loop_reading)NNNrl rD ) r9 re rf rg r rr rs rv rz rt rq rj r r r4 r rk s
rk c s^ e Zd ZdZdZ fddZdd Zddd Zd
d Zd d
Z dd Z
dd Z dd Z Z
S ) _ProactorBaseWritePipeTransportzTransport for write pipes.Tc s t j|i | d | _d S rD )r r r\ r. argskwr4 r r r M s
z(_ProactorBaseWritePipeTransport.__init__c C s t |tttfstdt|j | jrtd| j d ur"td|s&d S | j
r=| j
t j kr4t
d | j
d7 _
d S | jd u rL| jt|d d S | jsZt|| _| d S | 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 )r{ )rU bytesro
memoryview TypeErrortyper9 r* RuntimeErrorr\ r' r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r%
_loop_writingr# _maybe_pause_protocolextend)r. r{ r r r writeQ s0
z%_ProactorBaseWritePipeTransport.writeNc
C sd z|d ur| j d u r| jrW d S d | _ d| _|r | |d u r&| j}d | _|sB| jr3| j| jd | jr=| j
t j |
n'| jj| j || _ | j sbt|| _| j | j | n| j | j | jd ur|| j d u r| jd W d S W d S W d S ty } z | | W Y d }~d S d }~w ty } z
| |d W Y d }~d S d }~ww )Nr z#Fatal write error on pipe transport)r% r( r& r| r# r r+ rL r* r r_ r
SHUT_WR_maybe_resume_protocolr} sendr] r< r r r r\ r^ r rY rV r[ )r. fr{ rZ r r r r w s@
z-_ProactorBaseWritePipeTransport._loop_writingc C dS NTr rH r r r
can_write_eof z-_ProactorBaseWritePipeTransport.can_write_eofc C s | d S rD )rN rH r r r write_eof s z)_ProactorBaseWritePipeTransport.write_eofc C | d d S rD rY rH r r r abort rC z%_ProactorBaseWritePipeTransport.abortc C s: | j d ur td| j | _ | jd u r| j d | j S )NzEmpty waiter is already set)r\ r r Z
create_futurer% r^ rH r r r _make_empty_waiter s
z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s
d | _ d S rD )r\ rH r r r _reset_empty_waiter rF z3_ProactorBaseWritePipeTransport._reset_empty_waiterNN)r9 re rf rg Z_start_tls_compatibler r r r r r r r rj r r r4 r r G 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 r} recvr r$ r _pipe_closedr r4 r r r s z$_ProactorWritePipeTransport.__init__c C sB | rd S | jr d S d | _| jd ur| t d S | d S rD )Z cancelledr( r$ r% rY BrokenPipeErrorrN )r. r r r r r s
z(_ProactorWritePipeTransport._pipe_closed)r9 re rf r r rj r r r4 r r s r c s\ e Zd ZdZ d 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)r1 r2 )
_addressr\ r r collectionsdequer# r r+ rq )r. r/ r r0 addressr1 r2 r4 r r r s
z#_ProactorDatagramTransport.__init__c C t | | d S rD r rB r r r r rC z%_ProactorDatagramTransport._set_extrac C s t dd | jD S )Nc s s | ] \}}t |V qd S rD )r< ).0r{ _r r r s zC_ProactorDatagramTransport.get_write_buffer_size..)sumr# rH r r r rc s z0_ProactorDatagramTransport.get_write_buffer_sizec C r rD r rH r r r r rC z _ProactorDatagramTransport.abortc C s t |tttfstdt||sd S | jd ur'|d | jfvr'td| j | jrA| jrA| jt j
kr8t d | jd7 _d S | j
t||f | jd u rT| | d S )Nz,data argument must be bytes-like object (%r)z!Invalid address: must be None or z!socket.sendto() raised exception.r )rU r ro r r r r
ValueErrorr' r r r r r# r: r% r r )r. r{ addrr r r sendto s&
z!_ProactorDatagramTransport.sendtoc
C s zK| j rW d S d | _|r| | jr| j r'| jr'| jr$| j| jd W d S | j \}}| jd ur>| jj
| j || _n | jj
j
| j ||d| _W n1 tyd } z
| j| W Y d }~d S d }~w ty| } z
| |d W Y d }~d S d }~ww | j| j | d S )N)r z'Fatal write error on datagram transport)r' r% r| r# r r( r r+ rL popleftr} r r r rV r, error_received Exceptionr[ r r r )r. r r{ r rZ r r r r s:
z(_ProactorDatagramTransport._loop_writingc
C s~ d }zzq| j rW W |r| j|| d S d S d | _|d urE| }| jr5d }W W |r3| j|| d S d S | jd urA|| j}}n|\}}| j rWW W |rU| j|| d S d S | jd urh| jj | j
| j | _n | jj | j
| j | _W n% t
y } z | j| W Y d }~n d }~w tjy | js Y n
w | jd ur| j| j W |r| j|| d S d S |r| j|| w w rD )r' r, Zdatagram_receivedr$ r| r( r r r} r r max_sizeZrecvfromrV r r r r rq )r. r r{ r resrZ r r r rq s` $
z(_ProactorDatagramTransport._loop_readingrd rD ) r9 re rf r r r rc r r r rq rj r r r4 r r s
!r c @ s e Zd ZdZdd Zdd ZdS ) _ProactorDuplexPipeTransportzTransport for duplex pipes.c C r )NFr rH r r r r Q r z*_ProactorDuplexPipeTransport.can_write_eofc C s t rD )NotImplementedErrorrH r r r r T r z&_ProactorDuplexPipeTransport.write_eofN)r9 re rf rg r r r r r r r L s r c sF e Zd ZdZejjZ d fdd Zdd Z dd Z
d d
Z Z S ) _ProactorSocketTransportz Transport for connected sockets.Nc s$ t |||||| t| d S rD )r r r Z _set_nodelayr- r4 r r r _ s z!_ProactorSocketTransport.__init__c C r rD r rB r r r r d rC z#_ProactorSocketTransport._set_extrac C r r r rH r r r r g r z&_ProactorSocketTransport.can_write_eofc C s6 | j s| jrd S d| _| jd u r| jtj d S d S r )r( r* r% r r_ r
r rH r r r r j s
z"_ProactorSocketTransport.write_eofrd )
r9 re rf rg r Z
_SendfileModeZ
TRY_NATIVEZ_sendfile_compatibler r r r rj r r r4 r r X s r c s e Zd Z fddZ d3ddZ d4dddddddd Z d3d
d Z d5d d
Z d5ddZ d5ddZ fddZ
dd Z dd Z dd Z
dd Zd d Zd d Zd d! Zd"d# Zd$d% Zd4d&d'Zd(d) Z * d6d+d,Zd-d. Zd/d0 Zd1d2 Z ZS )7r c sl t td|jj || _|| _d | _i | _ |
| | t
t u r4t| j d S d S )NzUsing proactor: %s)r r r rW r5 r9 r} _selector_self_reading_future_accept_futuresZset_loop_make_self_pipe threadingcurrent_thread main_threadsignal
set_wakeup_fd_csockr; )r. Zproactorr4 r r r t s
z BaseProactorEventLoop.__init__Nc C t | |||||S rD )r )r. r r0 r1 r2 r3 r r r _make_socket_transport
z,BaseProactorEventLoop._make_socket_transportF) server_sideserver_hostnamer2 r3 ssl_handshake_timeoutc C s0 t j| |||||| d}
t| ||
||d |
jS )N)r r2 r3 )r Z SSLProtocolr Z_app_transport) r. Zrawsockr0
sslcontextr1 r r r2 r3 r Z ssl_protocolr r r _make_ssl_transport s z)BaseProactorEventLoop._make_ssl_transportc C r rD )r )r. r r0 r r1 r2 r r r _make_datagram_transport r z.BaseProactorEventLoop._make_datagram_transportc C t | ||||S rD )r r. r r0 r1 r2 r r r _make_duplex_pipe_transport s z1BaseProactorEventLoop._make_duplex_pipe_transportc C r rD )rk r r r r _make_read_pipe_transport s z/BaseProactorEventLoop._make_read_pipe_transportc C r rD )r r r r r _make_write_pipe_transport s z0BaseProactorEventLoop._make_write_pipe_transportc sj | rtd| rd S t t u rtd | | | 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_piper} rN r r rH r4 r r rN s
zBaseProactorEventLoop.closec | j ||I d H S rD )r} r )r. r nr r r sock_recv z BaseProactorEventLoop.sock_recvc r rD )r} r~ )r. r Zbufr r r sock_recv_into r z$BaseProactorEventLoop.sock_recv_intoc r rD )r} r )r. r r{ r r r sock_sendall r z"BaseProactorEventLoop.sock_sendallc r rD )r} Zconnect)r. r r r r r sock_connect r z"BaseProactorEventLoop.sock_connectc s | j |I d H S rD )r} acceptrB r r r sock_accept s z!BaseProactorEventLoop.sock_acceptc
s z| }W n ttjfy } ztdd }~ww zt|j}W n t y0 tdw |r5|n|}|s;dS t
|d}|rIt
|| |n|} t
||}d}
z. t
| | |}|dkrl|
W |
dkrk| | S S | j
||||I d H ||7 }|
|7 }
qT|
dkr| | w w )Nznot a regular filer l )r; AttributeErrorioUnsupportedOperationr ZSendfileNotAvailableErrorosfstatst_sizerV minseekr} sendfile) r. r fileoffsetcountr; errZfsizeZ blocksizeZend_posZ
total_sentr r r _sock_sendfile_native sD
z+BaseProactorEventLoop._sock_sendfile_nativec sn | }| | I d H z| j|j|||ddI d H W | |r*| S S | |r6| w w )NF)Zfallback)rr rs r Z
sock_sendfiler r rv )r. Ztranspr r r rv 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 rM _ssockrN r
_internal_fdsrH 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 rH r r r r s z%BaseProactorEventLoop._make_self_pipec
C s z|d ur | | j|urW d S | j| jd}W n/ tjy% Y d S ttfy. t yI } z|
d|| d W Y d }~d S d }~ww || _| | j d S )Ni z.Error on reading from the event loop self pipe)rS rT r/ )
r| r r} r r r r rw rx ry rX r _loop_self_reading)r. r rZ r r r r s(
z(BaseProactorEventLoop._loop_self_readingc 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 socketTr )r r rV _debugr rW )r. Zcsockr r r _write_to_self s z$BaseProactorEventLoop._write_to_selfd c s( d fdd d S )Nc
sJ zE| d ur7| \}}jrtd|| }d ur,j||dd|id n j||d|id r>W d S j} W nQ t y } z7
dkrd d|t
d njrvtjd dd
W Y d }~d S W Y d }~d S W Y d }~d S d }~w tjy Y d S w | j
< | d S ) Nz#%r got a new connection from %r: %rTr )r r2 r3 r r r` zAccept failed on a socket)rS rT r
zAccept failed on socket %rr )r| r r rW r r r r} r rV r; rX r r rN r r r r )r Zconnr r0 rZ r/ protocol_factoryr. r3 r r r r r r/ 6 sR
z2BaseProactorEventLoop._start_serving..looprD )r+ )r. r r r r3 Zbacklogr r r r _start_serving2 s %z$BaseProactorEventLoop._start_servingc C s d S rD r )r. Z
event_listr r r _process_events] s z%BaseProactorEventLoop._process_eventsc C s& | j D ]}| q| j d S rD )r valuesrM clear)r. futurer r r r a s
z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j | d }|r| | j| | d S rD )r popr; rM r}
_stop_servingrN )r. r r r r r r f s
z#BaseProactorEventLoop._stop_servingrd rD r )NNr N)r9 re rf r r r r r r r rN r r r r r r r r r r r r r r r rj r r r4 r r r sP
+r )$rg __all__r r r
rh r r r r r r r r r r
r logr r Z_FlowControlMixinZ
BaseTransportr Z
ReadTransportrk ZWriteTransportr r ZDatagramTransportr Z Transportr r Z
BaseEventLoopr r r r r sX n