mgLwHdZddlZddlZddlZddlmZmZmZmZddlm Z m Z m Z m Z m Z ddlmZmZmZmZmZmZddlmZddlmZdd lmZmZmZdd lmZmZdd lmZm Z m!Z!dd lm"Z"m#Z#m$Z$dd l%m&Z&m'Z'ddl(m)Z)er ddl*m+Z+ddl,m-Z-erej\dZ/n ej\Z/e eee0fZ1e ee1efZ2e eejffZ4e eee/fZ5e eee/fZ6e ddgedfZ7ee7Z8e gdedfZ9ee9Z:e dge&fZ;ee;Z<e=Z>Gdde eZ?Gdde eZ@Gdde eZAGddeAee eZBGddeAee!eZCGddeAeZDGdd eDee"eZEGd!d"eDee#eZFGd#d$eDeGe$ZHy)%zSSH stream handlersN) TYPE_CHECKINGAnyAnyStr AsyncIterator)CallableDictGenericIterableList)OptionalPatternSetTupleUnioncast)EXTENDED_DATA_STDERR) SSHLogger) MaybeAwait BreakReceivedSignalReceived)SoftEOFReceivedTerminalSizeChanged)DataTypeSSHClientSessionSSHServerSession) SSHTCPSessionSSHUNIXSessionSSHTunTapSession) SFTPServerrun_sftp_server)run_scp_server) SSHChannel) SSHConnection SSHReader SSHWriter)r%r&r&SSHChannel[bytes]ceZdZdZ ddddddefdZd eefd Ze dd Z e d e fd Z dd e ded efdZded dfdZddZdded efdZd efdZ dded efdZded efdZd efdZd edeffdZy) r%zSSH read stream handlerNsessionSSHStreamSession[AnyStr]chanSSHChannel[AnyStr]datatypec.||_||_||_yN_session_chan _datatypeselfr)r+r-s /lib/python3.12/site-packages/asyncssh/stream.py__init__zSSHReader.__init__K4; +/ !returnczK|jj|j23d{}|7 6ywz'Allow SSHReader to be an async iteratorN)r1aiterr3)r5results r6 __aiter__zSSHReader.__aiter__Qs3!MM//? &L ?s&;979;9;c|jSz+The SSH channel associated with this streamr2r5s r6channelzSSHReader.channelWzzr9c.|jjSz*The SSH logger associated with this streamr2loggerrCs r6rIzSSHReader.logger]zz   r9namedefaultc:|jj||Sa3Return additional information about this stream This method returns extra information about the channel associated with this stream. See :meth:`get_extra_info() ` on :class:`SSHClientChannel` for additional information. r2get_extra_infor5rKrLs r6rPzSSHReader.get_extra_infoczz((w77r9datacP|jj||jy)aNFeed data to the associated session This method feeds data to the SSH session associated with this stream, providing compatibility with the :meth:`feed_data() ` method on :class:`asyncio.StreamReader`. This is mostly useful for testing. N)r1 data_receivedr3r5rSs r6 feed_datazSSHReader.feed_dataos ##D$..9r9c8|jjy)aaFeed EOF to the associated session This method feeds an end-of-file indication to the SSH session associated with this stream, providing compatibility with the :meth:`feed_eof() ` method on :class:`asyncio.StreamReader`. This is mostly useful for testing. N)r1 eof_receivedrCs r6feed_eofzSSHReader.feed_eof|s ""$r9ncnK|jj|j|dd{S7w)aRead data from the stream This method is a coroutine which reads up to `n` bytes or characters from the stream. If `n` is not provided or set to `-1`, it reads until EOF or a signal is received. If EOF is received and the receive buffer is empty, an empty `bytes` or `str` object is returned. If the next data in the stream is a signal, the signal is delivered as a raised exception. .. note:: Unlike traditional `asyncio` stream readers, the data will be delivered as either `bytes` or a `str` depending on whether an encoding was specified when the underlying channel was opened. FexactNr1readr3r5r[s r6r`zSSHReader.reads.(]]'''GGGG ,535chK|jj|jd{S7w)aRead one line from the stream This method is a coroutine which reads one line, ending in `'\n'`. If EOF is received before `'\n'` is found, the partial line is returned. If EOF is received and the receive buffer is empty, an empty `bytes` or `str` object is returned. If the next data in the stream is a signal, the signal is delivered as a raised exception. .. note:: In Python 3.5 and later, :class:`SSHReader` objects can also be used as async iterators, returning input data one line at a time. N)r1readliner3rCs r6rdzSSHReader.readlines'&]]++DNN;;;;s )202 separatorclK|jj||j|d{S7w)aRead data from the stream until `separator` is seen This method is a coroutine which reads from the stream until the requested separator is seen. If a match is found, the returned data will include the separator at the end. The `separator` argument can be a single `bytes` or `str` value, a sequence of multiple `bytes` or `str` values, or a compiled regex (`re.Pattern`) to match against, returning data as soon as a matching separator is found in the stream. When passing a regex pattern as the separator, the `max_separator_len` argument should be set to the maximum length of an expected separator match. This can greatly improve performance, by minimizing how far back into the stream must be searched for a match. When passing literal separators to match against, the max separator length will be set automatically. .. note:: For best results, a separator regex should both begin and end with data which is as unique as possible, and should not start or end with optional or repeated elements. Otherwise, you run the risk of failing to match parts of a separator when it is split across multiple reads. If EOF or a signal is received before a match occurs, an :exc:`IncompleteReadError ` is raised and its `partial` attribute will contain the data in the stream prior to the EOF or signal. If the next data in the stream is a signal, the signal is delivered as a raised exception. N)r1 readuntilr3)r5remax_separator_lens r6rgzSSHReader.readuntils9P]],,Y->@@ @@s +424cnK|jj|j|dd{S7w)aRead an exact amount of data from the stream This method is a coroutine which reads exactly n bytes or characters from the stream. If EOF or a signal is received in the stream before `n` bytes are read, an :exc:`IncompleteReadError ` is raised and its `partial` attribute will contain the data before the EOF or signal. If the next data in the stream is a signal, the signal is delivered as a raised exception. Tr]Nr_ras r6 readexactlyzSSHReader.readexactlys. ]]'''FFFFrbcL|jj|jS)zReturn whether the stream is at EOF This method returns `True` when EOF has been received and all data in the stream has been read. )r1at_eofr3rCs r6rlzSSHReader.at_eofs}}##DNN33r9c2|j|jfS)z6Get information needed to redirect from this SSHReaderr1r3rCs r6get_redirect_infozSSHReader.get_redirect_info}}dnn,,r9r/r:r,r:N)r)__name__ __module__ __qualname____doc__rr7rrr?propertyrDrrIstrrrPrWrZintr`rdobjectrgrjboolrlrror9r6r%r%Hs!CG" :"+"7?" v!6  ! !! 83 8 8 8 :f : : %HCHH,<<,-.)@)@28)@VG3G6G$44-5)CX)M#N-r9ceZdZdZ ddddddefdZedd Zed efd Z dd e d e d e fdZ d e fdZddZd e fdZddZddZded dfdZdeed dfdZddZd edeffdZy)r&zSSH write stream handlerNr)r*r+r,r-c.||_||_||_yr/r0r4s r6r7zSSHWriter.__init__r8r9r:c|jSrArBrCs r6rDzSSHWriter.channel rEr9c.|jjSrGrHrCs r6rIzSSHWriter.loggerrJr9rKrLc:|jj||SrNrOrQs r6rPzSSHWriter.get_extra_inforRr9c6|jjS)z4Return whether the stream supports :meth:`write_eof`)r2 can_write_eofrCs r6rzSSHWriter.can_write_eof"szz''))r9c6|jjS)zClose the channel .. note:: After this is called, no data can be read or written from any of the streams associated with this channel. )r2closerCs r6rzSSHWriter.close'szz!!r9c6|jjS)z,Return if the stream is closing or is closed)r2 is_closingrCs r6rzSSHWriter.is_closing1szz$$&&r9cTK|jjd{y7w)zWait until the stream is closed This should be called after :meth:`close` to wait until the underlying connection is closed. N)r2 wait_closedrCs r6rzSSHWriter.wait_closed6sjj$$&&&s (&(cjK|jj|jd{y7w)aWait until the write buffer on the channel is flushed This method is a coroutine which blocks the caller if the stream is currently paused for writing, returning when enough data has been sent on the channel to allow writing to resume. This can be used to avoid buffering an excessive amount of data in the channel's send buffer. N)r1drainr3rCs r6rzSSHWriter.drain@s$mm!!$..111s )313rScN|jj||jS)atWrite data to the stream This method writes bytes or characters to the stream. .. note:: Unlike traditional `asyncio` stream writers, the data must be supplied as either `bytes` or a `str` depending on whether an encoding was specified when the underlying channel was opened. )r2writer3rVs r6rzSSHWriter.writeMszzdnn55r9 list_of_datacN|jj||jS)z(Write a collection of data to the stream)r2 writelinesr3)r5rs r6rzSSHWriter.writelines[szz$$\4>>BBr9c6|jjS)aWrite EOF on the channel This method sends an end-of-file indication on the channel, after which no more data can be written. .. note:: On an :class:`SSHServerChannel` where multiple output streams are created, writing EOF on one stream signals EOF for all of them, since it applies to the channel as a whole. )r2 write_eofrCs r6rzSSHWriter.write_eof`szz##%%r9c2|j|jfS)z4Get information needed to redirect to this SSHWriterrnrCs r6rozSSHWriter.get_redirect_infoorpr9r/rqrr)rurvrwrxrr7ryrDrrIrzrrPr}rrrrrrrr rrrror~r9r6r&r&s"CG" :"+"7?"  ! !! 83 8 8 8*t* "'D' ' 2 6& 6T 6Cx'7CDC &-5)CX)M#N-r9c@eZdZdZddZdedeefdZdeddfdZ deddfdZ dede fd Z deddfd Z de fd Zde fd Zde fd ZddZdeeddfdZdededdfdZde fdZdede fdZddZddZdedede defdZdedefdZ d dededefdZdeddfdZy)!SSHStreamSessionzSSH stream session handlerr:Nc(d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ dgi|_ dtji|_ddi|_dt#i|_y)NstrictrF)r2_conn _encoding_errors_loop_limit _exception _eof_received_connection_lost _read_paused _write_paused _recv_buf_len _recv_bufasyncioLock _read_locks _read_waitersset_drain_waitersrCs r6r7zSSHStreamSession.__init__xs59 04 (, :>  /3" %!"/3Rj(,glln'=,0$<.2CE]r9r-cK|j|s/|j|d{|j|s.yy7wr<)rlrdr5r-s r6r=zSSHStreamSession.aiters7++h' h// /++h'/s&AAAAcK |jJ|jj}||j|<|d{d|j|<y7#d|j|<wxYww)z*Wait for more data to arrive on the streamN)r create_futurerr5r-waiters r6 _block_readzSSHStreamSession._block_readsj 0::) ))$(JJ$<$<$>F+1D  x (LL+/D  x ( +/D  x (s.A+|jxr|j| S)z;Return whether end of file has been received on the channel)rrrs r6rlzSSHStreamSession.at_eofs!!!B$..*B&BBr9cd|_y)z0Handle a request to pause writing on the channelTN)rrCs r6 pause_writingzSSHStreamSession.pause_writings "r9cVd|_|jD]}|j|y)z1Handle a request to resume writing on the channelFN)rrrrs r6resume_writingzSSHStreamSession.resume_writing s*#++H    ),r9r[r^c K|j|}g}d}|j|4d{ |r|dk7rt|dtr9|rd}nt t|j d}t|t rd}n|t|d}||cxkDrdkDr=nn:|j|dd||d|d|d<|xj|zc_ d}nP|jt t|j d|xj|zc_ ||z}|r|dk7r|jr |dk(s|dkDr|r|r|dkr|s|js|rn|j|d{Jdddd{t t|jrdndj|} |dkDr2|r0t!j"t t$| t| |z| S777o#1d{7swYxYww)zRead data from the channelFNTrr9)rr isinstance ExceptionrpoprrrrrrrrrjoinrIncompleteReadErrorbytes) r5r-r[r^recv_bufrS break_readrlr>s r6r`zSSHStreamSession.reads>>(+ ##H--16!(1+y9)-J!"&y(,,q/"BC)#?$% %&) HQK(A1yqy HQKO4&.qk!"o **a/*KKVX\\!_ =>&&!+&FA3166--/6a!eUQ8**j&&x000I.-NfDNNb<AA$G q5U--d5&.A.1&kAo? ? [.J1K.---s_'H G/H C:G6(A G62G23G69 H G4A+H 2G64H 6H<G? =HH cK |jt|d{S7#tj$r$}t t |j cYd}~Sd}~wwxYww)zRead one line from the streamN)rg_NEWLINErrrrpartial)r5r-rs r6rdzSSHStreamSession.readlineGsH -(;; ;;** - , , -s<A$"$A$AAAAAArecnK|s tdtt|jrdnd}|j|}|t ur6d}tt|jrdnd}t j|}nt|ttfrBt|}t jt jtt|}nt|tr|}ttt|}n}tt|jrdnd} tttt|} t!d | D}| j#d | D}t j|}d } d } |j$|4d { | t|krot|| t&r|r>g|d | |xj(| zc_t+j,tt|d |j/d } t| t0r|cd d d d {Stt&| tt|| }||z }|d k(rd nt!| dz|z d }|j3||}|rp|j5}g|d | ||d |d <|d |}|xj(|zc_|d s|j/d |j7|cd d d d {S| t|z } | dz } | t|kro|j8s |j:rNg|d | |xj(| zc_|j7t+j,tt|d |j=|d {77r77#1d {7swYy xYww) z4Read data from the channel until a separator is seenzSeparator cannot be emptyrr9r  ||c32K|]}t|ywr/)r.0seps r6 z-SSHStreamSession.readuntil..fs5WcSWsc3FK|]}tj|ywr/)reescapers r6rz-SSHStreamSession.readuntil..gs!DGS"))C.Gs!rN) ValueErrorrrrrrrcompilerrrzrrr listr maxrrrrrrrrsearchendrrrr)r5rer-rhbufrseplen separatorspatbarseplistcurbufbuflenrnewbufstartmatchidxs r6rgzSSHStreamSession.readuntilOs:89 962S9>>(+  Ffdnnd%HJ**Z(C  E3< 0^F**RYYtFI'>?@C  7 +&Fwv 2Cvdnns$?C4 0)<=G5W55F!DG!DDJ**Z(C##H--s8},!(6"2I>02HWf- ..&8.")"="= $UC 0$#88#+,,q/C)#?'*.--'+9c&: :!&(6*:;F6MC!'1A#fqj66I12MEJJsE2E#iik,.&)&)#$i !$3i**c1*'{$LLO224"C.--Fc&k)FaKFEs8},H$$(:(:(*HWf%&&&0&..0!55d5#6FMM&&x000W.--X1Y.--sFN5NN5BN  N5'N(N5-B3N  N5,N-N52!N A:N NN N5N5N5N  N2&N) 'N2.N5cK|j|r |jJ|jj}|j|j ||d{|j|j ||j|r|j r)|j}|s|jr t}|r|yy7j#|j|j wxYww)z-Wait for data written to the channel to drainN) rrrraddremoverrrBrokenPipeError)r5r-rrs r6rzSSHStreamSession.drains&&x0 =zz---(, (@(@(B##H-11&9 ##H-44V<&&x0  //C4--%'  ! ##H-44V&1<&*&8&8:G__W-?-?-Gy/7K % 0&9L % 0&9L$T%;%;%=w%0, NG((4 44++E66BG   w '::) ))& && JJ " "7ELL 9 (r9rcb|jdj||jdy)z+Handle an incoming exception on the channelN)rrr)r5rs r6exception_receivedz)SSHServerStreamSession.exception_received s) t##C( 4 r9msecc8|jt|y)z'Handle an incoming break on the channelT)r5r)r5r6s r6break_receivedz%SSHServerStreamSession.break_receiveds  d 34r9signalc8|jt|y)z(Handle an incoming signal on the channelN)r5r)r5r9s r6signal_receivedz&SSHServerStreamSession.signal_receiveds v 67r9c6|jty)z*Handle an incoming soft EOF on the channelN)r5rrCs r6soft_eof_receivedz(SSHServerStreamSession.soft_eof_receiveds  12r9widthheightpixwidth pixheightc>|jt||||y)z6Handle an incoming terminal size change on the channelN)r5r)r5r>r?r@rAs r6terminal_size_changedz,SSHServerStreamSession.terminal_size_changeds%  3E64, !r9r/rr)rurvrwrx_OptSocketSessionFactoryr7r3rFrGs@r6rIrI's'0(@0 ?r9rIceZdZdZy)SSHTCPStreamSessionzTCP stream session handlerNr r~r9r6rPrP?s$r9rPceZdZdZy)SSHUNIXStreamSessionzUNIX stream session handlerNr r~r9r6rRrRDs%r9rRc DeZdZdZdedeefdZ d dedede defdZ y) SSHTunTapStreamSessionzTUN/TAP stream session handlerr-r:cVK |j|d{}|r|ny#7wr<)r`)r5r-packets r6r=zSSHTunTapStreamSession.aiterKs/99X..F  .s )')r[r^c4K|j|}|jsw|rOtt|j d}|xj t |zc_|j|S|j|d{|jswy7w)z3Override read to preserve TUN/TAP packet boundariesrNr9) rrrrrrrrr)r5r-r[r^rrSs r6r`zSSHTunTapStreamSession.readVs>>(+$$E8<<?3""c$i/"**, &&x000$$1sBBBBBN)rsF) rurvrwrxrrrr=r{r}r`r~r9r6rTrTHsH( H u1E 79!&8+0r9rT)Irxrr)rtypingrrrrrrr r r r r rrrr constantsrloggingrmiscrrrrrr)rrrrrrr$r r!scpr"rDr# connectionr$Future _WaiterFuturer_RecvBuf _RecvBufMapr _ReadLocks _ReadWaiters _DrainWaitersSSHSocketSessionFactoryrNSSHServerSessionFactoryrDSFTPServerFactoryrEr|rr%r&rr r rIrPrRrrTr~r9r6rhs* <<::==+;;6AADD-#)NN4(MNNM fi'( )8Xf--. (GLL( ) Hh}556 Xs=112 "K#=$.t$4$56#$;<"$12? !23 8v-v-rq-q-hywvyx ,-f5-f5, nJ-f5-f5nJb?-f5?0%08'/% &1&9)&1&3E: