o
0id @ s& d Z ddlZddlZddlZddlZddlZeedr e d7 Z ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd l
mZ dd lmZ d
Zd edddZd edddZeedrpd edddZd edddZG dd de jZG dd dee jZG dd d ZG d d d ZdS )!) StreamReader StreamWriterStreamReaderProtocolopen_connection start_server NZAF_UNIX)open_unix_connectionstart_unix_server )
coroutines)events)
exceptions)format_helpers) protocols)logger)sleepi )limitc sb t }t||d}t||d |j fdd| |fi |I d H \}}t| ||}||fS )Nr loopr c S N r protocolr 6/opt/alt/python310/lib64/python3.10/asyncio/streams.py1 z!open_connection..)r get_running_loopr r Zcreate_connectionr ) hostportr kwdsr reader transport_writerr r r r s r c s8 t fdd}j|||fi |I d H S )Nc t d} t| d}|S Nr r r r r! r client_connected_cbr r r r factoryO
z start_server..factory)r r Z
create_server)r* r r r r r+ r r) r r 6 s
r c s` t }t||d}t||d |j fdd| fi |I d H \}}t| ||}||fS )Nr r c r r r r r r r r c r z&open_unix_connection..)r r r r Zcreate_unix_connectionr )pathr r r r! r" r# r$ r r r r [ s r c s6 t fdd}j||fi |I d H S )Nc r% r& r' r( r) r r r+ l r, z"start_unix_server..factory)r r Zcreate_unix_server)r* r- r r r+ r r) r r g s
r c @ s@ e Zd Z dddZdd Zdd Zdd Zd
d Zd d
ZdS )FlowControlMixinNc C s8 |d u r t jdd| _n|| _d| _t | _d| _d S )N )
stacklevelF)r _get_event_loop_loop_paused collectionsdeque_drain_waiters_connection_lost)selfr r r r __init__ s
zFlowControlMixin.__init__c C s$ d| _ | j rtd| d S d S )NTz%r pauses writing)r3 r2 get_debugr debugr8 r r r
pause_writing s
z FlowControlMixin.pause_writingc C s> d| _ | j rtd| | jD ] }| s |d qd S )NFz%r resumes writing)r3 r2 r: r r; r6 done
set_resultr8 waiterr r r resume_writing s
z FlowControlMixin.resume_writingc C sF d| _ | jsd S | jD ]}| s |d u r|d q || q d S NT)r7 r3 r6 r> r?
set_exceptionr8 excrA r r r connection_lost s
z FlowControlMixin.connection_lostc s\ | j rtd| js
d S | j }| j| z|I d H W | j| d S | j| w )NzConnection lost)r7 ConnectionResetErrorr3 r2
create_futurer6 appendremover@ r r r
_drain_helper s
z FlowControlMixin._drain_helperc C s t r )NotImplementedErrorr8 streamr r r _get_close_waiter z"FlowControlMixin._get_close_waiterr ) __name__
__module__ __qualname__r9 r= rB rG rL rP r r r r r. u s
r. c sd e Zd Z dZd fdd Zedd Zdd Z fdd Zd
d Z d d
Z
dd Z dd Z Z
S )r Nc st t j|d |d urt|| _|j| _nd | _|d ur || _d| _d | _d | _ d | _
|| _ d| _ | j
| _d S )Nr F)superr9 weakrefref_stream_reader_wr_source_traceback_strong_reader_reject_connection_stream_writer_task
_transport_client_connected_cb _over_sslr2 rI _closed)r8 Z
stream_readerr* r __class__r r r9 s
z StreamReaderProtocol.__init__c C s | j d u rd S | S r )rX r< r r r _stream_reader s
z#StreamReaderProtocol._stream_readerc C s | j rddi}| jr| j|d< | j| | d S || _| j}|d ur*|| |dd u| _ | j
d urXt || || j| _ |
|| j }t
|rS| j|| _d | _d S d S )NmessagezpAn open stream was garbage collected prior to establishing network connection; call "stream.close()" explicitly.Zsource_tracebackZ
sslcontext)r[ rY r2 Zcall_exception_handlerabortr^ rd
set_transportget_extra_infor` r_ r r\ r
Z iscoroutineZ create_taskr] rZ )r8 r" contextr! resr r r connection_made s2
z$StreamReaderProtocol.connection_madec s~ | j }|d ur|d u r| n|| | j s+|d u r%| jd n| j| t | d | _d | _ d | _
d | _ d S r ) rd feed_eofrD ra r> r? rU rG rX r\ r] r^ )r8 rF r! rb r r rG s
z$StreamReaderProtocol.connection_lostc C s | j }|d ur|| d S d S r )rd feed_data)r8 datar! r r r
data_received s z"StreamReaderProtocol.data_receivedc C s$ | j }|d ur | | jrdS dS )NFT)rd rl r` )r8 r! r r r eof_received s z!StreamReaderProtocol.eof_receivedc C | j S r )ra rN r r r rP z&StreamReaderProtocol._get_close_waiterc C sD z| j }W n
ty Y d S w | r | s | d S d S d S r )ra AttributeErrorr> cancelled exception)r8 closedr r r __del__ s
z StreamReaderProtocol.__del__NN)rR rS rT rY r9 propertyrd rk rG ro rp rP rw
__classcell__r r rb r r s
r c @ st e Zd Z dd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dddZdd ZdS )r c C s4 || _ || _|| _|| _| j | _| jd d S r )r^ _protocol_readerr2 rI Z
_complete_futr? )r8 r" r r! r r r r r9 0 s zStreamWriter.__init__c C s@ | j jd| jg}| jd ur|d| j dd|S )N
transport=zreader=<{}> )rc rR r^ r| rJ formatjoinr8 infor r r __repr__: s
zStreamWriter.__repr__c C rq r r^ r< r r r r" @ s zStreamWriter.transportc C | j | d S r )r^ writer8 rn r r r r D zStreamWriter.writec C r r )r^
writelinesr r r r r G r zStreamWriter.writelinesc C
| j S r )r^ write_eofr< r r r r J
zStreamWriter.write_eofc C r r )r^
can_write_eofr< r r r r M r zStreamWriter.can_write_eofc C r r )r^ closer< r r r r P r zStreamWriter.closec C r r )r^
is_closingr< r r r r S r zStreamWriter.is_closingc s | j | I d H d S r )r{ rP r< r r r wait_closedV s zStreamWriter.wait_closedNc C s | j ||S r )r^ rh )r8 namedefaultr r r rh Y s zStreamWriter.get_extra_infoc sP | j d ur| j }|d ur|| j r tdI d H | j I d H d S )Nr )r| ru r^ r r r{ rL )r8 rF r r r drain\ s
zStreamWriter.drainr )rR rS rT r9 r ry r" r r r r r r r rh r r r r r r & s
r c @ s e Zd ZdZedfddZdd Zdd Zdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd Zdd Zd&dd Zd'd d Zd d! Zd"d# Zd$d% ZdS )(r Nc C sz |dkrt d|| _|d u rt | _n|| _t | _d| _d | _d | _ d | _
d| _ | j r;t
td| _d S d S )Nr zLimit cannot be <= 0Fr )
ValueError_limitr r1 r2 bytearray_buffer_eof_waiter
_exceptionr^ r3 r: r
extract_stacksys _getframerY )r8 r r r r r r9 z s"
zStreamReader.__init__c C s dg}| j r|t| j d | jr|d | jtkr'|d| j | jr3|d| j | jr?|d| j | jrK|d| j | j rS|d d
d
|S ) Nr z byteseofzlimit=zwaiter=z
exception=r} Zpausedr~ r ) r rJ lenr r _DEFAULT_LIMITr r r^ r3 r r r r r r r s
zStreamReader.__repr__c C rq r )r r< r r r ru rr zStreamReader.exceptionc C s8 || _ | j}|d urd | _| s|| d S d S d S r )r r rt rD rE r r r rD s zStreamReader.set_exceptionc C s4 | j }|d urd | _ | s|d d S d S d S r )r rt r? r@ r r r _wakeup_waiter s zStreamReader._wakeup_waiterc C s
|| _ d S r r )r8 r" r r r rg s
zStreamReader.set_transportc C s2 | j rt| j| jkrd| _ | j d S d S d S )NF)r3 r r r r^ resume_readingr< r r r _maybe_resume_transport s z$StreamReader._maybe_resume_transportc C s d| _ | d S rC )r r r< r r r rl s zStreamReader.feed_eofc C s | j o| j S r )r r r< r r r at_eof s zStreamReader.at_eofc C s |sd S | j | | | jd ur:| jsz| j W n
ty4 d | _Y d S w d| _d S d S d S d S )N T) r extendr r^ r3 r r Z
pause_readingrM r r r r rm s
zStreamReader.feed_datac s` | j d urt| d| jrd| _| j | j | _ z | j I d H W d | _ d S d | _ w )NzF() called while another coroutine is already waiting for incoming dataF)r RuntimeErrorr3 r^ r r2 rI )r8 Z func_namer r r _wait_for_data s
zStreamReader._wait_for_datac
s d}t |}z | |I d H }W |S tjy) } z |jW Y d }~S d }~w tjyW } z"| j||jrC| jd |j| = n| j |
t |j d d }~ww )N
r )
r readuntilr IncompleteReadErrorpartialLimitOverrunErrorr
startswithconsumedclearr r args)r8 sepseplenlineer r r readline s$
zStreamReader.readliner c s t |}|dkrtd| jd ur| jd} t | j}|| |krA| j||}|dkr0n-|d | }|| jkrAtd|| jrTt | j}| j
t |d | dI d H q|| jkrhtd|| jd || }| jd || = |
t |S ) Nr z,Separator should be at least one-byte stringTr z2Separator is not found, and chunk exceed the limitr z2Separator is found, but chunk is longer than limit)r r r r findr r r r bytesr r r r )r8 Z separatorr offsetZbuflenZisepchunkr r r r sD
zStreamReader.readuntilr c s | j d ur
| j |dkrdS |dk r.g } | | jI d H }|s#n|| qd|S | js<| js<| dI d H t| jd | }| jd |= | |S )Nr Tread)
r r r rJ r r r r r r )r8 nZblocksblockrn r r r r s s*
zStreamReader.readc s |dk r
t d| jd ur| j|dkrdS t| j|k rA| jr2t| j}| j t||| dI d H t| j|k s t| j|krSt| j}| j nt| jd | }| jd |= |
|S )Nr z*readexactly size can not be less than zeror readexactly) r r r r r r r r r r r )r8 r Z
incompletern r r r r s,
zStreamReader.readexactlyc C s | S r r r< r r r __aiter__ rQ zStreamReader.__aiter__c s | I d H }|dkrt|S )Nr )r StopAsyncIteration)r8 valr r r __anext__ s
zStreamReader.__anext__)r )r )rR rS rT rY r r9 r ru rD r rg r rl r rm r r r r r r r r r r r r v s&
[3) r rx r )__all__r4 Zsocketr warningsrV hasattr r
r r r
r logr Ztasksr r r r r r ZProtocolr. r r r r r r r sB
" @qP