mg`dZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZmZmZmZmZddlmZmZmZmZmZmZddlmZmZmZmZmZmZdd l m!Z!m"Z"d d l#m$Z$d d l#m%Z%m&Z&d d l#m'Z'm(Z(d dl#m)Z)m*Z*d dl#m+Z+d dl#m,Z,m-Z-d dl#m.Z.ej^dk7r ddl0Z0ddl1Z1ddl2Z2ee3e3fZ4ee4Z5ee3e3e3e3fZ6edee3e3fe6fZ7Gdde"Z8Gdde"Z9Gdde"Z:edZ;eede;fZe>e fZ?ee?edfZ@eeAeBfZCeeCeCfZDeeBe fZEeeBe3fZFeejejfZIeejejfZLeeeBe3feeBe3e3e3ffZMe ZNeNjZOejdZQd dddd ZRd d d!d"d#d$d%ZSdd&eTd'eBd(eBfd)ZUdd*e3d+eBd,eBd(eBfd-ZVd.eeTd(eWfd/ZX dd0eeBe3fd'eBd1e3d(ee3eBffd2ZYd3ZZd4eBd(eBfd5Z[d6eBd(eBfd7Z\d6eBd(eIfd8Z]d6eBd(eLfd9Z^dd:eEd;eBdZ`ed:eEd;e!d?d(eAfd@Z`ed:eEd;e!dAd(eBfdBZ`ddCZ`dd:eEdDeAd;eBd(e3fdEZad&eBdFeeBe3fd+eBd(ebfdGZcd&eBd(e3fdHZdd&eBd(ebfdIZedJeBd(eeBfdKZfedLedMNZgGdOdPeegZhedQeegfZiedQehegfZjdReiegd(ejegfdSZkddTZldUedVe3dWe3dXe3dYe3d(df dZZmGd[d\ZnGd]d^eoZpGd_d`epaZqGdbdcerZsGdddeesZtGdfdgetZuGdhdietZvGdjdketZwGdldmetZxGdndoetZyGdpdqetZzGdrdsetZ{GdtduetZ|GdvdwetZ}GdxdyetZ~Gdzd{esZGd|d}erZGd~derZGdderZGdderZGdderZGdderZe,e|e)eye*eze%eue.e~e-e}e'ewe&eve+e{e(exi Zde3deBdeBd(etfdZy)z+Miscellaneous utility classes and functionsN)PathPurePath) SystemRandom) TracebackType)AnyAsyncContextManager AwaitableCallableDict) GeneratorGenericIOMappingOptionalSequence)TupleTypeTypeVarUnioncastoverload)LiteralProtocol) DEFAULT_LANG)DISC_COMPRESSION_ERRORDISC_CONNECTION_LOST)DISC_HOST_KEY_NOT_VERIFIABLEDISC_ILLEGAL_USER_NAME)DISC_KEY_EXCHANGE_FAILEDDISC_MAC_ERROR)#DISC_NO_MORE_AUTH_METHODS_AVAILABLE)DISC_PROTOCOL_ERROR#DISC_PROTOCOL_VERSION_NOT_SUPPORTED)DISC_SERVICE_NOT_AVAILABLEwin32czeZdZdZedefdZedefdZedefdZ de fdZ defdZ de dd fd Z y ) _HashzProtocol for hashing datareturncy)zReturn the hash digest sizeNselfs -lib/python3.12/site-packages/asyncssh/misc.py digest_sizez_Hash.digest_size<cy)zReturn the hash block sizeNr+r,s r. block_sizez_Hash.block_size@r0r1cy)zReturn the hash nameNr+r,s r.namez _Hash.nameDr0r1cy)z)Return the digest value as a bytes objectNr+r,s r.digestz _Hash.digestHr0r1cy)z9Return the digest value as a string of hexadecimal digitsNr+r,s r. hexdigestz_Hash.hexdigestKr0r1 _Hash__dataNcy)z7Update this hash object's state with the provided bytesNr+)r-r:s r.updatez _Hash.updateNr0r1)__name__ __module__ __qualname____doc__propertyintr/r3strr5bytesr7r9r<r+r1r.r(r(9s# *S**)C))#c##88H3HFUFtFr1r(c"eZdZdZddedefdZy)HashTypez2Protocol for returning the type of a hash function_HashType__datar)cy)zCreate a new hash objectNr+)r-rGs r.__call__zHashType.__call__Ur0r1N).)r=r>r?r@rDr(rIr+r1r.rFrFRs<'u'u'r1rFceZdZdZddZy)_SupportsWaitClosedz'A class that supports async wait_closedNc Kyw)zWait for transport to closeNr+r,s r. wait_closedz_SupportsWaitClosed.wait_closed\ssr)N)r=r>r?r@rMr+r1r.rKrKYs 1*r1rK_Tr+)NNNz ([A-Za-z])iii@)kmg<iiQi: )rPsrRhdwvalueprefixr)c*t|}|r||zSdS)z:Return a string with optional prefix if value is non-emptyrPrC)rYrZs r. hide_emptyr]{s JE"6E>**r1lengthlabelsuffixc&d|||dk7r|fzSdfzS)z-Return a label with an optional plural suffixz%d %s%srrPr+)r^r_r`s r.pluralrbs' 1vE EE"E EEr1seqc&td|DS)z*Return if a sequence contains all integersc3<K|]}t|tywN) isinstancerB).0is r. zall_ints..s/3az!S!3s)all)rcs r.all_intsrls /3/ //r1symbols strip_leadingc~|jDcic]\}}|j|r|||dc}}Scc}}w)z8Return a mapping from values to symbol names for loggingN)items startswith)rmrZrnr5rYs r.get_symbol_namesrrsJ>DDc>tjt|S)z@Wrapper for ipaddress.ip_address which supports scoped addresses)rrr)rs r.rrs    4T : ;;r1c|jd}|dk\r |d|||d}}nd}tjt||zS)z@Wrapper for ipaddress.ip_network which supports scoped addresses/rNrP)rr ip_networkr)rrmasks r.rrsN ))C.C ax$3Zcdd    4T :T A BBr1filenamemode bufferingcLtt|j||S)z)Open a file with home directory expansion)r)openr expanduser)rrrs r. open_filers  X))+TY GGr1cyz5Read from a binary file with home directory expansionNr+)rs r. read_filerr0r1rbcyrr+rrs r.rrr0r1rcy)z3Read from a text file with home directory expansionNr+rs r.rrr0r1cft||5}|jcdddS#1swYyxYw)z.Read from a file with home directory expansionN)rread)rrfs r.rrs% 8T "avvx # " "s'0datacht||5}|j|cdddS#1swYyxYw)z7Write or append to a file with home directory expansionN)rwrite)rrrrs r. write_filers' 8T "awwt} # " "s(1suffixesc  tj|drjdnj t fdt dt dDS#t$rtd|zdwxYw)z4Parse a series of integers followed by unit suffixesrPc3nK|],}t||dzjz.yw)rN)floatlower)rhrimatchesrs r.rjz_parse_units..s=75q$x! 0B0B0D'EE5s25rInvalid N) _unit_patternsplitappendpopsumrangelenKeyError ValueError)rYrr_rs ` @r. _parse_unitsrs~!!%(Gr{r 77!!S\1577 7 7e+,$67s (A))Bc6tt|tdS)z4Parse a byte count with optional k, m, or g suffixesz byte count)rBr _byte_unitsrYs r.parse_byte_countrs |E; = >>r1c$t|tdS)z=Parse a time interval with optional s, m, h, d, or w suffixesz time interval)r _time_unitsrs r.parse_time_intervalr s {O <&AAAAexc_type exc_value tracebackcK|jJ|jj|||d{}d|_|S7 wrf)r __aexit__)r-rrr exit_results r.rz_ACMWrapper.__aexit__/sQ  ,,, --77 i,, ! ,s/A?A)r=r>r?r@r rrr rrrrr BaseExceptionrboolrr+r1r.rr st'1Yt_1&9S$_5&4$4 m1D(E #+M#: #+M#: ?G~ r1r.rc^tjdttffd }|S)aDecorator for functions returning asynchronous context managers This decorator can be used on functions which return objects intended to be async context managers. The object returned by the function should implement __aenter__ and __aexit__ methods to run when the async context is entered and exited. This wrapper also allows the use of "await" on the function being decorated, to return the context manager without entering it. r)c&t|i|S)z:Return an async context manager wrapper for this coroutine)r)argskwargsrs r.context_wrapperz.async_context_manager..context_wrapperLs40011r1) functoolswrapsrr)rrs` r.async_context_managerr?s4__T2K,=22 r1c`K |jd{y7#t$rYywxYww)aWait for a StreamWriter to close, if Python version supports it Python 3.8 triggers a false error report about garbage collecting an open stream if a close is in progress when a StreamWriter is garbage collected. This can be avoided by calling wait_closed(), but that method is not available in Python releases prior to 3.7. This function wraps this call, ignoring the error if the method is not available. N)rMAttributeError)writers r.maybe_wait_closedrUs.   """    s(.. +.+.ttywidthheightpixwidth pixheightc |tj|tjt j d||||y)zSet the terminal size of a TTYhhhhN)fcntlioctltermios TIOCSWINSZstructpack)rrrrrs r.set_terminal_sizergs/ KKW'' FFE8YGIr1cheZdZUdZeeefed<d deddefdZ deddfdZ deeefddfdZ y) Optionsz#Container for configuration optionsrNoptionsc \|rlt|t|s7tdt|jdt|j|jj |_ni|_|jj ||jdi|jy)Nrz, got r+)rgr TypeErrorr=rcopyr<prepare)r-r rs r.rzOptions.__init__ts gtDz2!%d!4!4d7m6L6L!NOO"..--/DKDK 6" #t{{#r1r)c y)z!Pre-process configuration optionsNr+r-rs r.rzOptions.preparer0r1cr|jj||jdi|jy)z4Update options based on keyword parameters passed inNr+)rr<rrs r.r<zOptions.updates+ 6" #t{{#r1rf) r=r>r?r@r rCobject__annotations__rrrr<r+r1r.r r os[- f  $ 3 $f $0040$T#v+.$4$r1r c ReZdZdZdeddedeedfdeee fddf fd Z xZ S) _RecordMetaz)Metaclass for general-purpose record typemcsr5bases.nsr)c r|dk7rttttf|jdij }|Dcic]}||j|}}|j Dcic] \}}||vs ||}}}||d<tt t|!||||Scc}wcc}}w)NRecordr __slots__) rrrCgetkeysrprsuper__new__) rr5rrfieldsrQdefaultsv __class__s r.r z_RecordMeta.__new__s 8 '#s(+&&!2B799= .45f266!9 fH5#%88:A:41a&!Q$:BA&B{OKdE2!FGG 6AsB.4 B3B3) r=r>r?r@rrCrrr rr  __classcell__r$s@r.rrsS3 HT-( H HE$).s%%>.<12743C%D..2nn%>>? ?r1c^fdjD}djd|DS)Nc 3ZK|]"}|j|t|f$ywrf)_formatr0r1s r.rjz!Record.__str__..s0+)adll1gdA&678)s(+r-c36K|]\}}| |d|yw)Nz: r+)rhrQr#s r.rjz!Record.__str__..sOvtq!Q*vs  )rr3)r-valuess` r.__str__zRecord.__str__s-+>>+yyOvOOOr1rQr#ct|S)zFormat a field as a stringr\)r-rQr#s r.r7zRecord._formats 1v r1N)r=r>r?r@rrrCrrrr4r:rr7r+r1r.rrsc&(IwsF{#( f  ?#? PP HSMr1r) metaclassc4eZdZdZefdededeffd ZxZS)ErrorzGeneral SSH errorcodereasonlangcNt||||_||_||_yrf)rrr?r@rA)r-r?r@rAr$s r.rzError.__init__s%     r1) r=r>r?r@rrBrCrr%r&s@r.r>r>s(;GS#Sr1r>ceZdZdZy)DisconnectErroraSSH disconnect error This exception is raised when a serious error occurs which causes the SSH connection to be disconnected. Exception codes should be taken from :ref:`disconnect reason codes `. See below for exception subclasses tied to specific disconnect reasons if you want to customize your handling by reason. :param code: Disconnect reason, taken from :ref:`disconnect reason codes ` :param reason: A human-readable reason for the disconnect :param lang: (optional) The language the reason is in :type code: `int` :type reason: `str` :type lang: `str` Nr=r>r?r@r+r1r.rDrDsr1rDc0eZdZdZefdedeffd ZxZS)CompressionErrora^SSH compression error This exception is raised when an error occurs while compressing or decompressing data sent on the SSH connection. :param reason: Details about the compression error :param lang: (optional) The language the reason is in :type reason: `str` :type lang: `str` r@rAc0t|t||yrf)rrrr-r@rAr$s r.rzCompressionError.__init__ />r1r=r>r?r@rrCrr%r&s@r.rGrG% 1=?s?#??r1rGc0eZdZdZefdedeffd ZxZS)ConnectionLostaSSH connection lost This exception is raised when the SSH connection to the remote system is unexpectedly lost. It can also occur as a result of the remote system failing to respond to keepalive messages or as a result of a login timeout, when those features are enabled. :param reason: Details about the connection failure :param lang: (optional) The language the reason is in :type reason: `str` :type lang: `str` r@rAc0t|t||yrf)rrrrIs r.rzConnectionLost.__init__s -vt` :param reason: A human-readable reason for the channel open failure :param lang: The language the reason is in :type code: `int` :type reason: `str` :type lang: `str` NrEr+r1r.rlrlsr1rlceZdZdZy)ChannelListenErrorzSSH channel listen error This exception is raised to report failures in setting up remote SSH connection listeners. :param details: Details of the listen failure :type details: `str` NrEr+r1r.rnrns r1rnc0eZdZdZefdedeffd ZxZS)PasswordChangeRequiredaSSH password change required This exception is raised during password validation on the server to indicate that a password change is required. It should be raised when the password provided is valid but expired, to trigger the client to provide a new password. :param prompt: The prompt requesting that the user enter a new password :param lang: The language that the prompt is in :type prompt: `str` :type lang: `str` promptrAcFt|d|z||_||_y)NzPassword change required: %s)rrrqrA)r-rqrAr$s r.rzPasswordChangeRequired.__init__s$ 7&@A  r1rKr&s@r.rprps% 1=s#r1rpc(eZdZdZdeffd ZxZS) BreakReceivedzSSH break request received This exception is raised on an SSH server stdin stream when the client sends a break on the channel. :param msec: The duration of the break in milliseconds :type msec: `int` msecc8t|d|z||_y)NzBreak for %s msec)rrru)r-rur$s r.rzBreakReceived.__init__s ,t34 r1)r=r>r?r@rBrr%r&s@r.rtrts Sr1rtc(eZdZdZdeffd ZxZS)SignalReceivedzSSH signal request received This exception is raised on an SSH server stdin stream when the client sends a signal on the channel. :param signal: The name of the signal sent by the client :type signal: `str` signalc8t|d|z||_y)Nz Signal: %s)rrry)r-ryr$s r.rzSignalReceived.__init__s ./ r1)r=r>r?r@rCrr%r&s@r.rxrxs sr1rxc$eZdZdZdfd ZxZS)SoftEOFReceivedzSSH soft EOF request received This exception is raised on an SSH server stdin stream when the client sends an EOF from within the line editor on the channel. c$t|dy)NzSoft EOF)rr)r-r$s r.rzSoftEOFReceived.__init__s $r1rN)r=r>r?r@rr%r&s@r.r|r|s%%r1r|cJeZdZdZdedededeffd ZedefdZxZ S) TerminalSizeChangeda"SSH terminal size change notification received This exception is raised on an SSH server stdin stream when the client sends a terminal size change on the channel. :param width: The new terminal width :param height: The new terminal height :param pixwidth: The new terminal width in pixels :param pixheight: The new terminal height in pixels :type width: `int` :type height: `int` :type pixwidth: `int` :type pixheight: `int` rrrrc vt|d|d|d|d|d ||_||_||_||_y)NzTerminal size change: (r-r2)rrrrrr)r-rrrrr$s r.rzTerminalSizeChanged.__init__s: 9> ?    "r1r)c^|j|j|j|jfS)z-Return terminal size as a tuple of 4 integers)rrrrr,s r. term_sizezTerminalSizeChanged.term_size s#zz4;; t~~EEr1) r=r>r?r@rBrrATermSizerr%r&s@r.rrsG(#c#3#####F8FFr1rr?r@rAcd t|||S#t$rt|d||fz|cYSwxYw)zBMap disconnect error code to appropriate DisconnectError exceptionz %s (error %d))_disc_error_maprrD)r?r@rAs r.construct_disc_errorr5sCMt$VT22 Mt_~%EtLLMs //)r-)rU)r)r)r)wb)rrKr)N)r@rrrerrrpathlibrrrandomrtypesrtypingrrr r r r r rrrrrrrrrrtyping_extensionsrr constantsrrrrrr r!r"r#r$r%rrrrrB TermModes TermModesArgr TermSizeArgr(rFrKrODefTuple MaybeAwaitrExcInfo OptExcInforDrC BytesOrStrBytesOrStrDictFilePathHostPort IPv4Address IPv6Address IPAddress IPv4Network IPv6Network IPNetworkSockAddr_random randrangecompilerrrrr]rbrrlrrrzr~rrrrrrrrrrrrr_ACMCoro_ACMWrapperFuncrrrr rrr Exceptionr>rDrGrNrQrTrWrZr]r`rfrirlrnrprtrxr|rrrr+r1r.rsI*2 "FFFF>>/#CK?:O1<<7 CH   " c3# $D%S/834 FHF2'x'*(* T] rB  2y}$ % ]#]MA B 7E"233 4 5#:  j*,- h  c? )'')>)>> ? )'')>)>> ? sCx%S#s(:";; < .     =) 4inE 12E:/ +f+c+S+F3FsFCF#F 0(6"0t0+,(gc3h/(($'(07S0A(C33ssB> >> >cS77wsCx'8777"?C?C? =s=u=   v0DA'$-8 C4( )3 D 112?43H, $I2IcI3I #I03I8<I$$8 H$ H {BIe.??&=_=*E?E&?o?&AA$77(LL&