a
i} @ 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_extra getsocknamer
error_loop get_debugr warning getpeername) transportsock r //usr/lib64/python3.9/asyncio/proactor_events.py_set_socket_extra s
r c s~ e Zd Zd fdd Zdd Zdd Zdd Zd
d Zd d
Zdd Z e
j fddZ d ddZ
dd Zdd Zdd Z ZS ) _ProactorBasePipeTransportNc s t || | | || _| | || _d | _d | _d | _d| _ d| _
d| _ d| _ | jd url| j
| j| jj| |d ur| jtj|d d S )Nr F)super__init__
_set_extra_sock set_protocol_server_buffer _read_fut
_write_fut_pending_write
_conn_lost_closing _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| jr0|d | jd urP|d| j | jd url|d| j | jd ur|d| j | jr|dt | j | j
r|d d d |S )
Nclosedclosingzfd=zread=zwrite=zwrite_bufsize=z EOF writtenz<{}> )
r6 __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/ r1 r r r r# ] s z'_ProactorBasePipeTransport.set_protocolc C s | j S rD rE r/ r r r get_protocol` s z'_ProactorBasePipeTransport.get_protocolc C s | j S rD )r* rF 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& cancelrF r r r closef s
z _ProactorBasePipeTransport.closec C s* | j d ur&|d| t| d | d S )Nzunclosed transport )source)r" ResourceWarningrK )r/ _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 r1 )
isinstanceOSErrorr r r debugcall_exception_handlerr- _force_close)r/ excrQ 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_result
set_exceptionr* r) r' rJ r& r( r% r r, rI )r/ rX r r r rW 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< r^ r
SHUT_RDWRrK r$ Z_detach)r/ rX r4 r r r rI 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_size)NNN)rP )r:
__module__ __qualname__r rA r! r# rG rH rK warningswarnrO rY rW rI rc
__classcell__r r r5 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. r5 r r r s
z#_ProactorReadPipeTransport.__init__c C s | j o| j S rD )rk r* rF 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* rk r r r rU rF 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* rk r& r r, rl rj _data_receivedr r rU 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 rU r- Z eof_received
SystemExitKeyboardInterrupt
BaseExceptionrY rK )r/ Z keep_openrX 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.)
rk rj rv rS r- r ZBufferedProtocolZ _feed_data_to_buffered_protors rt ru rY Z
data_received)r/ rq rX r r r ro 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& r[ resultrJ r* ro rk r _proactorrecvr" ConnectionAbortedErrorrY r r rU ConnectionResetErrorrW rT r CancelledErroradd_done_callbackrl )r/ futrq rX r r r rl sP
z(_ProactorReadPipeTransport._loop_reading)NNN)N) r: rd re r rm rn rr rv ro rl rh r r r5 r ri s ri 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 rD )r r rZ r/ argskwr5 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 )rq )rS bytes bytearray
memoryview TypeErrortyper: r+ RuntimeErrorrZ r) r !LOG_THRESHOLD_FOR_CONNLOST_WRITESr r r'
_loop_writingr% _maybe_pause_protocolextendrp 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( rx r% r r, rI r+ r" r^ r
SHUT_WR_maybe_resume_protocolry sendr[ r= r~ r r rZ r\ r| rW rT rY )r/ frq rX r r r r q s8
z-_ProactorBaseWritePipeTransport._loop_writingc C s dS NTr rF r r r
can_write_eof s z-_ProactorBaseWritePipeTransport.can_write_eofc C s | d S rD )rK rF r r r write_eof s z)_ProactorBaseWritePipeTransport.write_eofc C s | d d S rD rW rF 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)rZ r r Z
create_futurer' r\ rF r r r _make_empty_waiter s
z2_ProactorBaseWritePipeTransport._make_empty_waiterc C s
d | _ d S rD )rZ rF r r r _reset_empty_waiter s z3_ProactorBaseWritePipeTransport._reset_empty_waiter)NN)
r: rd re Z_start_tls_compatibler r r r r r r r rh r r r5 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 ry rz r" r& r~ _pipe_closedr r5 r r r s z$_ProactorWritePipeTransport.__init__c C s@ | r d S | jrd S d | _| jd ur4| t n| d S rD ) cancelledr* r& r' rW BrokenPipeErrorrK )r/ r r r r r s
z(_ProactorWritePipeTransport._pipe_closed)r: rd re r r rh r r r5 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)r2 r3 )
_addressrZ r r collectionsdequer% r r, rl )r/ r0 r r1 addressr2 r3 r5 r r r s
z#_ProactorDatagramTransport.__init__c C s t | | d S rD r rC r r r r! s z%_ProactorDatagramTransport._set_extrac C s t dd | jD S )Nc s s | ]\}}t |V qd S rD )r= ).0rq _r r r rw zC_ProactorDatagramTransport.get_write_buffer_size..)sumr% rF r r r rc s z0_ProactorDatagramTransport.get_write_buffer_sizec C s | d d S rD r rF 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 )rS r r r r r r
ValueErrorr) r r r r r% r; r' r r )r/ rq 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' rx r% r r* r r, rI popleftry r r" r rT r- error_received ExceptionrY r~ r r )r/ r rq r rX 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 rD )r) r- Zdatagram_receivedr& rx r* r r ry rz r" max_sizerecvfromrT r r r} r~ rl )r/ r rq r resrX r r r rl sT $
z(_ProactorDatagramTransport._loop_reading)NNN)N)N)N) r: rd re r r r! rc r r r rl rh r r r5 r r s
!r c @ s e Zd Zdd Zdd ZdS ) _ProactorDuplexPipeTransportc C s dS )NFr rF r r r r K s z*_ProactorDuplexPipeTransport.can_write_eofc C s t d S rD )NotImplementedErrorrF r r r r N s z&_ProactorDuplexPipeTransport.write_eofN)r: rd re 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 rD )r r r Z _set_nodelayr. r5 r r r Y s z!_ProactorSocketTransport.__init__c C s t | | d S rD r rC r r r r! ^ s z#_ProactorSocketTransport._set_extrac C s dS r r rF 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" r^ r
r rF r r r r d s
z"_ProactorSocketTransport.write_eof)NNN) r: rd re r Z
_SendfileModeZ
TRY_NATIVEZ_sendfile_compatibler r! r r rh r r r5 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 rU r6 r: ry _selector_self_reading_future_accept_futuresZset_loop_make_self_pipe threadingcurrent_thread main_threadsignal
set_wakeup_fd_csockr< )r/ Zproactorr5 r r r n s
z BaseProactorEventLoop.__init__Nc C s t | |||||S rD )r )r/ r r1 r2 r3 r4 r r r _make_socket_transport{ s
z,BaseProactorEventLoop._make_socket_transportF) server_sideserver_hostnamer3 r4 ssl_handshake_timeoutc C s0 t j| |||||| d}
t| ||
||d |
jS )N)r r3 r4 )r Z SSLProtocolr Z_app_transport) r/ Zrawsockr1
sslcontextr2 r r r3 r4 r Z ssl_protocolr r r _make_ssl_transport s z)BaseProactorEventLoop._make_ssl_transportc C s t | |||||S rD )r )r/ r r1 r r2 r3 r r r _make_datagram_transport s
z.BaseProactorEventLoop._make_datagram_transportc C s t | ||||S rD )r r/ r r1 r2 r3 r r r _make_duplex_pipe_transport s z1BaseProactorEventLoop._make_duplex_pipe_transportc C s t | ||||S rD )ri r r r r _make_read_pipe_transport s z/BaseProactorEventLoop._make_read_pipe_transportc C s t | ||||S rD )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_pipery rK r r rF r5 r r rK s
zBaseProactorEventLoop.closec s | j ||I d H S rD )ry rz )r/ r nr r r sock_recv s z BaseProactorEventLoop.sock_recvc s | j ||I d H S rD )ry recv_into)r/ r bufr r r sock_recv_into s z$BaseProactorEventLoop.sock_recv_intoc s | j ||I d H S rD )ry r )r/ r rq r r r sock_sendall s z"BaseProactorEventLoop.sock_sendallc s | j ||I d H S rD )ry connect)r/ r r r r r sock_connect s z"BaseProactorEventLoop.sock_connectc s | j |I d H S rD )ry acceptrC 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_sizerT minseekry sendfile) r/ r fileoffsetcountr< errfsize blocksizeZend_pos
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)fallback)rm rn r Z
sock_sendfiler" r rr )r/ Ztranspr r r rr 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 rJ _ssockrK r
_internal_fdsrF r r r r s
z&BaseProactorEventLoop._close_self_pipec C s: t \| _| _| jd | jd | jd7 _d S )NFr )r
socketpairr r setblockingr rF 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)rQ rR r0 )
rx r ry rz r r r} rs rt ru rV r~ _loop_self_reading)r/ r rX 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 rT _debugr rU )r/ csockr 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 r3 r4 r r r_ zAccept failed on a socket)rQ rR r
zAccept failed on socket %rr )rx r r rU r r r ry r rT r< rV r r rK r r} r r~ )r connr r1 rX r0 protocol_factoryr/ r4 r r r r r r0 0 sJ
z2BaseProactorEventLoop._start_serving..loop)N)r, )r/ r r r r4 backlogr r r r _start_serving, s %z$BaseProactorEventLoop._start_servingc C s d S rD r )r/ Z
event_listr r r _process_eventsW s z%BaseProactorEventLoop._process_eventsc C s& | j D ] }| q
| j d S rD )r valuesrJ clear)r/ futurer r r r [ s
z*BaseProactorEventLoop._stop_accept_futuresc C s6 | j | d }|r | | j| | d S rD )r popr< rJ ry
_stop_servingrK )r/ r r r r r r ` s
z#BaseProactorEventLoop._stop_serving)NNN)N)NNN)NN)NN)NN)N)NNr N)r: rd re r r r r r r r rK r r r r r r r r r r r r r r r rh r r r5 r r l sN
+r )#__all__r r r
rf r r r r r r r r r r
r logr r Z_FlowControlMixinZ
BaseTransportr Z
ReadTransportri ZWriteTransportr r ZDatagramTransportr Z Transportr r Z
BaseEventLoopr r r r r sT n