o gN%@s@dZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZd5ddZd6d d Zd7d d ZddZddZddZddZddZddZddZddZddZeadaead d!Z efd"d#Z!Gd$d%d%Z"e"Z#d&d'Z$d(d)Z%Gd*d+d+Z&d,d-Z'd.d/Z(d8d1d2Z)d8d3d4Z*dS)9z0 Useful functions used by the rest of paramiko. N)DEBUG zero_byte xffffffffmax_bytebyte_ordbyte_chr) SSHConfigFc Csd}d}|st|dkrt|ddkrd}t|dr.t}|r"t}|dt|d|}tdt|dD]}|d>td|||dd}q6|rW|ddt|>8}|S)zTturns a normalized byte string into a long-int (adapted from Crypto.Util.number)r >I)lenrrrrangestructunpack)salways_positiveoutnegativefillerir8/usr/local/lib/python3.10/dist-packages/paramiko/util.py inflate_long)s  &rTcCst}t|}|dkr%|dkr%td|t@|}|dL}|dkr%|dkst|D]}|dkr7|ddkr7n|dkrC|ddkrCn q)d}|dkrMt}nt}||dd}|ry|dkrit|dd krit|}|dkryt|dd kryt|}|S) zTturns a long-int into a normalized byte string (adapted from Crypto.Util.number)rr r r )rNr ) bytesintrpackr enumeraterrr)nadd_sign_paddingrrrrr deflate_long>s. r$cs~d}g}t||dkr%|t|||d|d7}t||dks |t|kr6|t||dfdd|DS)Nrcsg|]}|qSrr).0lineprefixrr esz!format_binary..)rappendformat_binary_line)datar*xrrr)r format_binary]s r0cCs4ddd|D}ddd|D}d||S)N cSsg|] }dt|qS)z{:02X}formatrr'crrrr+isz&format_binary_line..r%cSs*g|]}dt|t|ddqS)z.{:c}..?_r2r4rrrr+ks*z {:50s} {})joinr3)r.leftrightrrrr-hs   r-cCsRd}|D]"}t|}d|krdkrnn|t|7}q|td|7}q|S)Nr z%{:02X})rrbr3)rrr5rrrr safe_stringpsr>cCsrz|WSty8t|d}t|d}|dkrYdSt|d}|d@s4|dK}|d8}|d@r(|YSw)NFrr rr ) bit_lengthAttributeErrorr$rr)r"normhbytebitlenrrrr?{s     r?cCsdtjtdS)Nr% )r8 tracebackformat_exceptionsysexc_infosplitrrrr tb_stringssrJcCst}t}t|dkr|dd}|dkrK|}t|dkr$|||t||||}t|t|}||d|7}||8}|dks|S)a) Given a password, passphrase, or other human-source key, scramble it through a secure hash into some keyworthy bytes. This specific algorithm is used for encrypting/decrypting private key files. :param function hash_alg: A function which creates a new hash object, such as ``hashlib.sha256``. :param salt: data to salt the hash with. :type bytes salt: Hash salt bytes. :param str key: human-entered password or passphrase. :param int nbytes: number of bytes to generate. :return: Key data, as `bytes`. rNr)rrupdater=digestmin)hash_algsaltkeynbyteskeydatarLhash_objsizerrrgenerate_key_bytess       rUcCsddlm}||S)a Read a file of known SSH host keys, in the format used by openssh, and return a compound dict of ``hostname -> keytype ->`` `PKey `. The hostname may be an IP address or DNS name. The keytype will be either ``"ssh-rsa"`` or ``"ssh-dss"``. This type of file unfortunately doesn't exist on Windows, but on posix, it will usually be stored in ``os.path.expanduser("~/.ssh/known_hosts")``. Since 1.5.3, this is just a wrapper around `.HostKeys`. :param str filename: name of the file to read host keys from :return: nested dict of `.PKey` objects, indexed by hostname and then keytype r)HostKeys)paramiko.hostkeysrV)filenamerVrrrload_host_keyss rYcCst}|||S)z Provided only as a backward-compatible wrapper around `.SSHConfig`. .. deprecated:: 2.7 Use `SSHConfig.from_file` instead. )rparse)file_objconfigrrrparse_ssh_configs r]cCs ||S)zM Provided only as a backward-compatible wrapper around `.SSHConfig`. )lookup)hostnamer\rrrlookup_ssh_host_configs r`c Csdd|}}}dd|}}}|dkr7||}||||}}||||}}||||}}|dks|dkr?||7}|S)Nr rr) r/mu1u2u3v1v2v3qrrr mod_inversesric CsTztjWSty)ttd7att_Wdn1swYtjYSw)Nr )_g_thread_dataidr@_g_thread_lock_g_thread_counterrrrr get_thread_ids  rncCsftd}t|jdkrdS||t|d}t|}d}|d7}|t|d| |dS)zKsend paramiko logs to a logfile, if they're not already going somewhereparamikorNaz>%(levelname)-.3s [%(asctime)s.%(msecs)03d] thr=%(_threadid)-3dz %(name)s: %(message)sz%Y%m%d-%H:%M:%S) logging getLoggerrhandlerssetLevelopen StreamHandler setFormatter Formatter addHandler)rXlevelloggerfhandlerfrmrrr log_to_files    rc@seZdZddZdS)PFiltercCs t|_dS)NT)rn _threadid)selfrecordrrrfilterszPFilter.filterN)__name__ __module__ __qualname__rrrrrrs rcCst|}|t|SN)rqrr addFilter_pfilter)namer{rrr get_logger s  rcCsNt|t|kr dSd}tt|D]}|t||t||AO}q|dkS)NFr)rrr)rpr=resrrrrconstant_time_bytes_eqs rc@seZdZddZddZdS)ClosingContextManagercCs|Srr)rrrr __enter__szClosingContextManager.__enter__cCs |dSr)close)rtypevaluerErrr__exit__!s zClosingContextManager.__exit__N)rrrrrrrrrrs rcCst|t||Sr)maxrM)minimumvalmaximumrrr clamp_value%src CsDzt|WSty!z|WYSty |YYSww)z: Coerce to bytes if possible or return unchanged. )r= TypeErrorasbytesr@)rrrrr)s    rutf8cCs4t|tr|St|tr||Stdt|)zcast unicode or bytes to bytesExpected unicode or bytes, got ) isinstancerstrencoderrrencodingrrrr=?s   r=cCs4t|tr ||St|tr|Stdt|)z cast bytes or unicode to unicoder)rrdecoderrrrrrruJs   r)F)T)r%)r)+__doc__rGrrE threadingrqparamiko.commonrrrrrrparamiko.configrrr$r0r-r>r?rJrUrYr]r`rilocalrjrmLockrlrnrrrrrrrrr=rrrrrsD