U
iÛf h ã @ s& d Z ddlZddlZddlZddlZeedƒr6e d7 Z ddlmZ ddlmZ ddlm Z dd lm
Z
dd
lm Z dd l m
Z
dd lmZ d
Zd dedœdd„Zd dedœdd„Zeedƒràd!dedœdd„Zd"dedœdd„ZG dd„ de jƒZG dd„ dee jƒZG dd „ d ƒZG d d „ d ƒZdS )#)Ú StreamReaderÚ StreamWriterÚStreamReaderProtocolÚopen_connectionÚ start_serveré NZAF_UNIX)Úopen_unix_connectionÚstart_unix_serveré )Ú
coroutines)Úevents)Ú
exceptions)Úformat_helpers)Ú protocols)Úlogger)Úsleepi )ÚloopÚlimitc ‹ st |d krt ¡ }ntjdtdd t||d}t||d‰ |j‡ fdd„| |f|ŽI d H \}}t|ˆ ||ƒ}||fS )Nú[The loop argument is deprecated since Python 3.8, and scheduled for removal in Python 3.10.é ©Ú
stacklevel©r r ©r c s ˆ S ©N© r ©Úprotocolr ú4/opt/alt/python38/lib64/python3.8/asyncio/streams.pyÚ5 ó z!open_connection..) r Úget_event_loopÚwarningsÚwarnÚDeprecationWarningr r Zcreate_connectionr ) ÚhostÚportr r ÚkwdsÚreaderÚ transportÚ_Úwriterr r r r s"
þ
ÿÿr c ‹ sJ ˆd krt ¡ ‰ntjdtdd ‡ ‡‡fdd„}ˆj|||f|ŽI d H S )Nr r r c s t ˆˆd} t| ˆ ˆd}|S ©Nr r ©r r ©r' r ©Úclient_connected_cbr r r r ÚfactoryX s
ÿz start_server..factory)r r r! r" r# Z
create_server)r/ r$ r% r r r& r0 r r. r r : s
þr c ‹ sr |d krt ¡ }ntjdtdd t||d}t||d‰ |j‡ fdd„| f|ŽI d H \}}t|ˆ ||ƒ}||fS )Nr r r r r c s ˆ S r r r r r r r p r z&open_unix_connection..) r r r! r" r# r r Zcreate_unix_connectionr )Úpathr r r& r' r( r) r* r r r r d s
þ
ÿÿr c ‹ sH ˆd krt ¡ ‰ntjdtdd ‡ ‡‡fdd„}ˆj||f|ŽI d H S )Nr r r c s t ˆˆd} t| ˆ ˆd}|S r+ r, r- r. r r r0 ~ s
ÿz"start_unix_server..factory)r r r! r" r# Zcreate_unix_server)r/ r1 r r r& r0 r r. r r t s
þr c @ s> e Zd Zddd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ ZdS )ÚFlowControlMixinNc C s0 |d krt ¡ | _n|| _d| _d | _d| _d S ©NF)r r Ú_loopÚ_pausedÚ
_drain_waiterÚ_connection_lost)Úselfr r r r Ú__init__‘ s zFlowControlMixin.__init__c C s d| _ | j ¡ r t d| ¡ d S )NTz%r pauses writing)r5 r4 Ú get_debugr Údebug©r8 r r r Ú
pause_writingš s
z FlowControlMixin.pause_writingc C sF d| _ | j ¡ r t d| ¡ | j}|d k rBd | _| ¡ sB| d ¡ d S )NFz%r resumes writing)r5 r4 r: r r; r6 ÚdoneÚ
set_result©r8 Úwaiterr r r Úresume_writing s
z FlowControlMixin.resume_writingc C sV d| _ | jsd S | j}|d kr"d S d | _| ¡ r4d S |d krH| d ¡ n
| |¡ d S ©NT)r7 r5 r6 r> r? Ú
set_exception©r8 ÚexcrA r r r Úconnection_lost¬ s z FlowControlMixin.connection_lostc à s< | j rtdƒ‚| jsd S | j}| j ¡ }|| _|I d H d S )NzConnection lost)r7 ÚConnectionResetErrorr5 r6 r4 Ú
create_futurer@ r r r Ú
_drain_helper¼ s
z FlowControlMixin._drain_helperc C s t ‚d S r )ÚNotImplementedError©r8 Ústreamr r r Ú_get_close_waiterÇ s z"FlowControlMixin._get_close_waiter)N) Ú__name__Ú
__module__Ú __qualname__r9 r= rB rG rJ rN r r r r r2 ‡ s
r2 c sb e Zd ZdZd‡ fdd„ Zedd„ ƒZdd„ Z‡ fdd „Zd
d „ Z d d
„ Z
dd„ Z dd„ Z ‡ Z
S )r Nc sn t ƒ j|d |d k r,t |¡| _|j| _nd | _|d k r@|| _d| _d | _d | _ || _
d| _ | j
¡ | _d S )Nr F)Úsuperr9 ÚweakrefÚrefÚ_stream_reader_wrÚ_source_tracebackÚ_strong_readerÚ_reject_connectionÚ_stream_writerÚ
_transportÚ_client_connected_cbÚ _over_sslr4 rI Ú_closed)r8 Z
stream_readerr/ r ©Ú __class__r r r9 Ö s
z StreamReaderProtocol.__init__c C s | j d krd S | ¡ S r )rU r< r r r Ú_stream_readeré s
z#StreamReaderProtocol._stream_readerc C s® | j r6ddi}| jr | j|d< | j |¡ | ¡ d S || _| j}|d k rT| |¡ | d¡d k | _ | j
d k rªt || || jƒ| _ |
|| j ¡}t
|¡r¤| j |¡ d | _d S )NÚmessagezpAn open stream was garbage collected prior to establishing network connection; call "stream.close()" explicitly.Zsource_tracebackZ
sslcontext)rX rV r4 Zcall_exception_handlerÚabortrZ r` Ú
set_transportÚget_extra_infor\ r[ r rY r
Z iscoroutineZ create_taskrW )r8 r( Úcontextr' Úresr r r Úconnection_madeï s2 ÿ
þÿ
z$StreamReaderProtocol.connection_madec sx | j }|d k r*|d kr | ¡ n
| |¡ | j ¡ sV|d krJ| j d ¡ n | j |¡ tƒ |¡ d | _d | _ d | _
d S r ) r` Úfeed_eofrD r] r> r? rR rG rU rY rZ )r8 rF r' r^ r r rG
s
z$StreamReaderProtocol.connection_lostc C s | j }|d k r| |¡ d S r )r` Ú feed_data)r8 Údatar' r r r Ú
data_received s z"StreamReaderProtocol.data_receivedc C s$ | j }|d k r| ¡ | jr dS dS )NFT)r` rh r\ )r8 r' r r r Ú eof_received s z!StreamReaderProtocol.eof_receivedc C s | j S r )r] rL r r r rN + s z&StreamReaderProtocol._get_close_waiterc C s" | j }| ¡ r | ¡ s | ¡ d S r )r] r> Ú cancelledÚ exception)r8 Úclosedr r r Ú__del__. s z StreamReaderProtocol.__del__)NN)rO rP rQ rV r9 Úpropertyr` rg rG rk rl rN rp Ú
__classcell__r r r^ r r Ë s
r c @ sr e Zd Zdd„ Zdd„ Zedd„ ƒZdd„ Zd d
„ Zd d „ Z d
d„ Z
dd„ Z dd„ Z dd„ Z
ddd„Zdd„ ZdS )r c C s4 || _ || _|| _|| _| j ¡ | _| j d ¡ d S r )rZ Ú _protocolÚ_readerr4 rI Z
_complete_futr? )r8 r( r r' r r r r r9 @ s zStreamWriter.__init__c C s@ | j jd| j›g}| jd k r0| d| j›¡ d d |¡¡S )Nú
transport=zreader=ú<{}>ú )r_ rO rZ rt ÚappendÚformatÚjoin©r8 Úinfor r r Ú__repr__J s
zStreamWriter.__repr__c C s | j S r ©rZ r< r r r r( P s zStreamWriter.transportc C s | j |¡ d S r )rZ Úwrite©r8 rj r r r r T s zStreamWriter.writec C s | j |¡ d S r )rZ Ú
writelinesr€ r r r r W s zStreamWriter.writelinesc C s
| j ¡ S r )rZ Ú write_eofr< r r r r‚ Z s zStreamWriter.write_eofc C s
| j ¡ S r )rZ Ú
can_write_eofr< r r r rƒ ] s zStreamWriter.can_write_eofc C s
| j ¡ S r )rZ Úcloser< r r r r„ ` s zStreamWriter.closec C s
| j ¡ S r )rZ Ú
is_closingr< r r r r… c s zStreamWriter.is_closingc à s | j | ¡I d H d S r )rs rN r< r r r Ú wait_closedf s zStreamWriter.wait_closedNc C s | j ||¡S r )rZ rd )r8 ÚnameÚdefaultr r r rd i s zStreamWriter.get_extra_infoc à sL | j d k r | j ¡ }|d k r |‚| j ¡ r8tdƒI d H | j ¡ I d H d S )Nr )rt rn rZ r… r rs rJ )r8 rF r r r Údrainl s
zStreamWriter.drain)N)rO rP rQ r9 r} rq r( r r r‚ rƒ r„ r… r† rd r‰ r r r r r 6 s
r c @ s¢ e Zd ZdZedfdd„Zdd„ 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 sv |dkrt dƒ‚|| _|d kr*t ¡ | _n|| _tƒ | _d| _d | _d | _ d | _
d| _ | j ¡ rrt
t d¡¡| _d S )Nr zLimit cannot be <= 0Fr )Ú
ValueErrorÚ_limitr r r4 Ú bytearrayÚ_bufferÚ_eofÚ_waiterÚ
_exceptionrZ r5 r: r
Ú
extract_stackÚsysÚ _getframerV )r8 r r r r r r9 Š s
ÿzStreamReader.__init__c C s¶ dg}| j r"| t| j ƒ› d¡ | jr2| d¡ | jtkrN| d| j› ¡ | jrf| d| j›¡ | jr~| d| j›¡ | jr–| d| j›¡ | j r¦| d¡ d
d
|¡¡S ) Nr z bytesÚeofzlimit=zwaiter=z
exception=ru Zpausedrv rw ) r rx ÚlenrŽ r‹ Ú_DEFAULT_LIMITr r rZ r5 ry rz r{ r r r r} s
zStreamReader.__repr__c C s | j S r )r r< r r r rn ² s zStreamReader.exceptionc C s0 || _ | j}|d k r,d | _| ¡ s,| |¡ d S r )r r rm rD rE r r r rD µ s zStreamReader.set_exceptionc C s* | j }|d k r&d | _ | ¡ s&| d ¡ d S r )r rm r? r@ r r r Ú_wakeup_waiter¾ s
zStreamReader._wakeup_waiterc C s
|| _ d S r r~ )r8 r( r r r rc Æ s zStreamReader.set_transportc C s* | j r&t| jƒ| jkr&d| _ | j ¡ d S r3 )r5 r• r r‹ rZ Ú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 rh Ï s zStreamReader.feed_eofc C s | j o | j S r )rŽ r r< r r r Úat_eofÓ s zStreamReader.at_eofc C st |sd S | j |¡ | ¡ | jd k rp| jspt| j ƒd| j krpz| j ¡ W n tk
rh d | _Y nX d| _d S )Nr T) r Úextendr— rZ r5 r• r‹ Z
pause_readingrK r€ r r r ri × s
ÿþ zStreamReader.feed_datac à sX | j d k rt|› dƒ‚| jr.d| _| j ¡ | j ¡ | _ z| j I d H W 5 d | _ X d S )NzF() called while another coroutine is already waiting for incoming dataF)r Ú RuntimeErrorr5 rZ r˜ r4 rI )r8 Z func_namer r r Ú_wait_for_dataí s
ÿ
zStreamReader._wait_for_datac
à sº d}t |ƒ}z| |¡I d H }W n” tjk
rN } z|j W Y ¢S d }~X Y nh tjk
r´ } zH| j ||j¡r„| jd |j| …= n
| j ¡ |
¡ t |j d ƒ‚W 5 d }~X Y nX |S )Nó
r )
r• Ú readuntilr ÚIncompleteReadErrorÚpartialÚLimitOverrunErrorr Ú
startswithÚconsumedÚclearr™ rŠ Úargs)r8 ÚsepÚseplenÚlineÚer r r Úreadline s
zStreamReader.readlinerž c à sü t |ƒ}|dkrtdƒ‚| jd k r(| j‚d}t | jƒ}|| |kr|| j ||¡}|dkrZq´|d | }|| jkr|t d|¡‚| jr¢t | jƒ}| j
¡ t |d ¡‚| d¡I d H q,|| jkrÊt d|¡‚| jd || … }| jd || …= |
¡ t |ƒS )Nr z,Separator should be at least one-byte stringéÿÿÿÿr 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¨ ÚoffsetZbuflenZisepÚchunkr r r rŸ ( s>
þ
ÿzStreamReader.readuntilr¬ c à sœ | j d k r| j ‚|dkr dS |dk rVg }| | j¡I d H }|s@qL| |¡ q(d |¡S | jsr| jsr| d¡I d H t| jd |… ƒ}| jd |…= | ¡ |S )Nr r Úread)
r r± r‹ rx rz r rŽ r r® r™ )r8 ÚnZblocksÚblockrj r r r r± ƒ s"
zStreamReader.readc à sÀ |dk rt dƒ‚| jd k r | j‚|dkr,dS t| jƒ|k rr| jr`t| jƒ}| j ¡ t ||¡‚| d¡I d H q,t| jƒ|kr–t| jƒ}| j ¡ n t| 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
incompleterj r r r r´ µ s&
zStreamReader.readexactlyc C s | S r r r< r r r Ú __aiter__Þ s zStreamReader.__aiter__c à s | ¡ I d H }|dkrt‚|S )Nr )r« ÚStopAsyncIteration)r8 Úvalr r r Ú __anext__á s zStreamReader.__anext__)rž )r¬ )rO rP rQ rV r– r9 r} rn rD r— rc r™ rh rš ri r r« rŸ r± r´ rµ r¸ r r r r r † s$
[
2)r )NN)NN)N)N)Ú__all__Zsocketr’ r! rS ÚhasattrÚ r
r r r
r Úlogr Ztasksr r– r r r r ZProtocolr2 r r r r r r r Ú sF
ÿ ! ÿ '
ÿ ÿ DkP