o
    gj                  
   @   s   d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl	m
Z
 ddlmZ d\ZZzddlZW n eyJ Z zeZW Y dZ[ndZ[ww ddlmZmZ d	ZG d
d dZdd ZG dd dZG dd deZdS )z2
Configuration file (aka ``ssh_config``) support.
    N)sha1)StringIO)partial)NN   )CouldNotCanonicalizeConfigParseError   c                   @   s   e Zd ZdZedZg ddgg dg dg dg dd	Zd
d Ze	dd Z
e	dd Ze	dd Zdd Zdd Zd.ddZdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- ZdS )/	SSHConfiga  
    Representation of config information as stored in the format used by
    OpenSSH. Queries can be made via `lookup`. The format is described in
    OpenSSH's ``ssh_config`` man page. This class is provided primarily as a
    convenience to posix users (since the OpenSSH format is a de-facto
    standard on posix) but should work fine on Windows too.

    .. versionadded:: 1.6
    z(\w+)(?:\s*=\s*|\s+)(.+))%C%h%l%L%n%p%r%ur   )r
   ~%dr   r   r   r   )r   r   r   r   )r   r   r   )	r
   r   r   r   r   r   r   r   r   )controlpathhostnameidentityfileproxycommand	proxyjump
match-execc                 C   s
   g | _ dS )a  
        Create a new OpenSSH config object.

        Note: the newer alternate constructors `from_path`, `from_file` and
        `from_text` are simpler to use, as they parse on instantiation. For
        example, instead of::

            config = SSHConfig()
            config.parse(open("some-path.config")

        you could::

            config = SSHConfig.from_file(open("some-path.config"))
            # Or more directly:
            config = SSHConfig.from_path("some-path.config")
            # Or if you have arbitrary ssh_config text from some other source:
            config = SSHConfig.from_text("Host foo\n\tUser bar")
        N)_config)self r   I/var/www/html/api-tag/env/lib/python3.10/site-packages/paramiko/config.py__init__H   s   
zSSHConfig.__init__c                 C   s   |  t|S )zg
        Create a new, parsed `SSHConfig` from ``text`` string.

        .. versionadded:: 2.7
        )	from_filer   )clstextr   r   r   	from_text]   s   zSSHConfig.from_textc                 C   s6   t |}| |W  d   S 1 sw   Y  dS )zr
        Create a new, parsed `SSHConfig` from the file found at ``path``.

        .. versionadded:: 2.7
        N)openr   )r    pathflor   r   r   	from_pathf   s   
$zSSHConfig.from_pathc                 C   s   |  }| | |S )zp
        Create a new, parsed `SSHConfig` from file-like object ``flo``.

        .. versionadded:: 2.7
        )parse)r    r%   objr   r   r   r   p   s   
zSSHConfig.from_filec                 C   sL  dgi d}|D ]}|  }|r|drqt| j|}|s&td||d }|d}|dv rT| j	
| di i}|d	krL| ||d	< q| ||d
< q|dkre| dkred|d |< q|dru|dru|dd }|dv r||d v r|d | 
| q|g|d |< q||d vr||d |< q| j	
| dS )z
        Read an OpenSSH config from the given file object.

        :param file_obj: a file-like object to read the config file from
        *)hostconfig#zUnparsable line {}r      )r*   matchr+   r*   matchesr   noneN")r   localforwardremoteforward)strip
startswithrer.   SETTINGS_REGEXr   formatgrouplowerr   append
_get_hosts_get_matchesendswith)r   file_objcontextliner.   keyvaluer   r   r   r'   {   s8   
zSSHConfig.parsec                 C   s   | j |d}d|vr||d< |dddv }t|dd}|rC|d|krC|d	  }| |||}||d< | j ||d
d
d}|S | j ||dd
d}|S )a  
        Return a dict (`SSHConfigDict`) of config options for a given hostname.

        The host-matching rules of OpenSSH's ``ssh_config`` man page are used:
        For each parameter, the first obtained value will be used.  The
        configuration files contain sections separated by ``Host`` and/or
        ``Match`` specifications, and that section is only applied for hosts
        which match the given patterns or keywords

        Since the first obtained value for each parameter is used, more host-
        specific declarations should be given near the beginning of the file,
        and general defaults at the end.

        The keys in the returned dict are all normalized to lowercase (look for
        ``"port"``, not ``"Port"``. The values are processed according to the
        rules for substitution variable expansion in ``ssh_config``.

        Finally, please see the docs for `SSHConfigDict` for deeper info on
        features such as optional type conversion methods, e.g.::

            conf = my_config.lookup('myhost')
            assert conf['passwordauthentication'] == 'yes'
            assert conf.as_bool('passwordauthentication') is True

        .. note::
            If there is no explicitly configured ``HostName`` value, it will be
            set to the being-looked-up hostname, which is as close as we can
            get to OpenSSH's behavior around that particular option.

        :param str hostname: the hostname to lookup

        .. versionchanged:: 2.5
            Returns `SSHConfigDict` objects instead of dict literals.
        .. versionchanged:: 2.7
            Added canonicalization support.
        .. versionchanged:: 2.7
            Added ``Match`` support.
        .. versionchanged:: 3.3
            Added ``Match final`` support.
        )r   r   canonicalizehostnameN)yesalwayscanonicalizemaxdotsr   .canonicaldomainsT)	canonicalfinalF)_lookupgetintcountsplitcanonicalize)r   r   optionscanonmaxdotsdomainsr   r   r   lookup   s"   *zSSHConfig.lookupNFc                    s   d u rt  | jD ]K}| |dg |s$| |dg |||s$q
|d  D ]*\ } vrA|d ur<|d d  n| < q* dkrT   fdd|D  q*q
|r^| |S )Nr*   r/   r+   r   c                 3   s     | ]}|  vr|V  qd S Nr   .0xrC   rS   r   r   	<genexpr>  s    z$SSHConfig._lookup.<locals>.<genexpr>)SSHConfigDictr   _pattern_matchesrN   _does_matchitemsextend_expand_variables)r   r   rS   rK   rL   rA   rD   r   r\   r   rM      s4   


zSSHConfig._lookupc              	   C   s   d}|D ].}d ||}t||}|dur|d }nzt|}W n
 tjy+   Y nw |r2|  S q|dddkr=|S t|)ag  
        Return canonicalized version of ``hostname``.

        :param str hostname: Target hostname.
        :param options: An `SSHConfigDict` from a previous lookup pass.
        :param domains: List of domains (e.g. ``["paramiko.org"]``).

        :returns: A canonicalized hostname if one was found, else ``None``.

        .. versionadded:: 2.7
        Fz{}.{}Nr   canonicalizefallbacklocalrF   )r9   _addressfamily_host_lookupsocketgethostbynamegaierrorrN   r   )r   r   rS   rV   founddomain	candidatefamily_specificr   r   r   rR     s"   

zSSHConfig.canonicalizec                 C   s$   t  }| jD ]	}||d  q|S )z
        Return the set of literal hostnames defined in the SSH config (both
        explicit hostnames and wildcard entries).
        r*   )setr   update)r   hostsentryr   r   r   get_hostnamesE  s   
zSSHConfig.get_hostnamesc                 C   sZ   t |dr
|d}d}|D ]}|dr"t||dd  r" dS t||r*d}q|S )NrQ   ,F!r   T)hasattrrQ   r6   fnmatch)r   patternstargetr.   patternr   r   r   r_   O  s   

zSSHConfig._pattern_matchesc                 C   sN  g }|d d  }t  }|r|d}	d }
|dd }|dd }|	d |	d }}|dkr6| ||	r6dS |dkr=|}
nU|d	krCd
S |dkrR|pJ|}| ||}
n@|dkr]| ||}
n5|dkrl|pd|}| ||}
n&|dkrw| ||}
n|dkr| ||d|}td u rttj	|dd
dj
}
|
d ur| |
|	rdS ||	 |s|S )Nr   r   usertypeparamrK   FrL   allTr*   originalhost	localuserexecr   stdout)hidewarn)getpassgetuserpoprN   _should_failr_   	_tokenizeinvokeinvoke_import_errorrunokr<   )r   
match_listtarget_hostnamerK   rL   rS   matched
candidateslocal_usernamerk   passedconfigured_hostconfigured_usertype_r{   hostvalry   exec_cmdr   r   r   r`   `  sL   

0zSSHConfig._does_matchc                 C   s   |d r|S | S )Nnegater   )r   
would_passrk   r   r   r   r     s   zSSHConfig._should_failc                 C   s   |  |}|s	|S |}|dkr|d|}d|v r|d }nt}t }d|v r-|d }	n|}	t dd }
t||
}t	j
d}|
| t| |	 }t|  |||
||||	||d
}|}| D ]\}}||vroqf||t|}qf|S )a  
        Tokenize a string based on current config/hostname data.

        :param config: Current config data.
        :param target_hostname: Original target connection hostname.
        :param key: Config key being tokenized (used to filter token list).
        :param value: Config value being tokenized.

        :returns: The tokenized version of the input ``value`` string.
        r   portry   rI   r   r   )
r
   r   r   r   r   r   r   r   r   r   )_allowed_tokensrN   SSH_PORTr   r   rf   gethostnamerQ   LazyFqdnosr$   
expanduserreprr   encode	hexdigestra   replacestr)r   r+   r   rC   rD   allowed_tokensconfigured_hostnamer   ry   
remoteuserlocal_hostname
local_fqdnhomedirtohashreplacements	tokenizedfindr   r   r   r   r     sD   



zSSHConfig._tokenizec                 C   s   | j |g S )aJ  
        Given config ``key``, return list of token strings to tokenize.

        .. note::
            This feels like it wants to eventually go away, but is used to
            preserve as-strict-as-possible compatibility with OpenSSH, which
            for whatever reason only applies some tokens to some config keys.
        )TOKENS_BY_CONFIG_KEYrN   r   rC   r   r   r   r     s   	zSSHConfig._allowed_tokensc                 C   sr   |D ]4}|| du rqt | j|||}t|| tr.t|| D ]\}}|||| |< q q||| ||< q|S )aA  
        Return a dict of config options with expanded substitutions
        for a given original & current target hostname.

        Please refer to :doc:`/api/config` for details.

        :param dict config: the currently parsed config
        :param str hostname: the hostname whose config is being looked up
        N)r   r   
isinstancelist	enumerate)r   r+   r   k	tokenizerirD   r   r   r   rc     s   
zSSHConfig._expand_variablesc                 C   s*   zt |W S  ty   td|w )z>
        Return a list of host_names from host value.
        zUnparsable host {})shlexrQ   
ValueErrorr   r9   )r   r*   r   r   r   r=     s
   zSSHConfig._get_hostsc           	         s  g }t |}|rHdddd}|d}|dr#d|d< |dd }||d	< |d
v r1|| q|s:td||d|d< || |s	dd |D }d|v rd tt fdd|tt fdd|}}d}t	|rud}nd|v r|
d|
dkrd}|durt||S )z
        Parse a specific Match config line into a list-of-dicts for its values.

        Performs some parse-time validation as well.
        NF)rz   r{   r   r   rs   Tr   r   rz   )r|   rK   rL   z'Missing parameter to Match '{}' keywordr{   c                 S   s   g | ]}|d  qS )rz   r   rY   r   r   r   
<listcomp>  s    z*SSHConfig._get_matches.<locals>.<listcomp>r|   )r|   rK   c                    s   |  v S rX   r   r[   	allowabler   r   <lambda>"      z(SSHConfig._get_matches.<locals>.<lambda>c                    s   |  vS rX   r   r   r   r   r   r   #  r   z>Match does not allow 'all' mixed with anything but 'canonical'rK   z-Match does not allow 'all' before 'canonical')r   rQ   r   r6   r<   r   r9   r   filteranyindex)	r   r.   r/   tokensr   keywordsr   baderrr   r   r   r>     sB   




zSSHConfig._get_matches)NFF)__name__
__module____qualname____doc__r7   compiler8   r   r   classmethodr"   r&   r   r'   rW   rM   rR   rq   r_   r`   r   r   r   rc   r=   r>   r   r   r   r   r	   .   s<    



	

<
B#)
8>	r	   c                 C   sd   | dd }|dkrdS ztj}|dkrtj}t| d|tjtjtjW S  tj	y1   Y dS w )a  
    Try looking up ``hostname`` in an IPv4 or IPv6 specific manner.

    This is an odd duck due to needing use in two divergent use cases. It looks
    up ``AddressFamily`` in ``options`` and if it is ``inet`` or ``inet6``,
    this function uses `socket.getaddrinfo` to perform a family-specific
    lookup, returning the result if successful.

    In any other situation -- lookup failure, or ``AddressFamily`` being
    unspecified or ``any`` -- ``None`` is returned instead and the caller is
    expected to do something situation-appropriate like calling
    `socket.gethostbyname`.

    :param str hostname: Hostname to look up.
    :param options: `SSHConfigDict` instance w/ parsed options.
    :returns: ``getaddrinfo``-style tuples, or ``None``, depending.
    addressfamilyr   Ninet)
rN   r;   rf   AF_INET6AF_INETgetaddrinfo
SOCK_DGRAM
IPPROTO_IPAI_CANONNAMErh   )r   rS   address_familyfamilyr   r   r   re   /  s$   re   c                   @   s"   e Zd ZdZdddZdd ZdS )r   z7
    Returns the host's fqdn on request as string.
    Nc                 C   s   d | _ || _|| _d S rX   )fqdnr+   r*   )r   r+   r*   r   r   r   r   Y  s   
zLazyFqdn.__init__c           	      C   sl   | j d u r3d }t| j| j}|d ur(|D ]}|\}}}}}|r'd|v r'|} nq|d u r0t }|| _ | j S )NrI   )r   re   r*   r+   rf   getfqdn)	r   r   resultsresafsocktypeproto	canonnamesar   r   r   __str__^  s   
zLazyFqdn.__str__rX   )r   r   r   r   r   r   r   r   r   r   r   T  s    
r   c                   @   s    e Zd ZdZdd Zdd ZdS )r^   a  
    A dictionary wrapper/subclass for per-host configuration structures.

    This class introduces some usage niceties for consumers of `SSHConfig`,
    specifically around the issue of variable type conversions: normal value
    access yields strings, but there are now methods such as `as_bool` and
    `as_int` that yield casted values instead.

    For example, given the following ``ssh_config`` file snippet::

        Host foo.example.com
            PasswordAuthentication no
            Compression yes
            ServerAliveInterval 60

    the following code highlights how you can access the raw strings as well as
    usefully Python type-casted versions (recalling that keys are all
    normalized to lowercase first)::

        my_config = SSHConfig()
        my_config.parse(open('~/.ssh/config'))
        conf = my_config.lookup('foo.example.com')

        assert conf['passwordauthentication'] == 'no'
        assert conf.as_bool('passwordauthentication') is False
        assert conf['compression'] == 'yes'
        assert conf.as_bool('compression') is True
        assert conf['serveraliveinterval'] == '60'
        assert conf.as_int('serveraliveinterval') == 60

    .. versionadded:: 2.5
    c                 C   s"   | | }t |tr|S | dkS )a  
        Express given key's value as a boolean type.

        Typically, this is used for ``ssh_config``'s pseudo-boolean values
        which are either ``"yes"`` or ``"no"``. In such cases, ``"yes"`` yields
        ``True`` and any other value becomes ``False``.

        .. note::
            If (for whatever reason) the stored value is already boolean in
            nature, it's simply returned.

        .. versionadded:: 2.5
        rF   )r   boolr;   )r   rC   valr   r   r   as_bool  s   
zSSHConfigDict.as_boolc                 C   s   t | | S )z
        Express given key's value as an integer, if possible.

        This method will raise ``ValueError`` or similar if the value is not
        int-appropriate, same as the builtin `int` type.

        .. versionadded:: 2.5
        )rO   r   r   r   r   as_int  s   	zSSHConfigDict.as_intN)r   r   r   r   r   r   r   r   r   r   r^   z  s    !r^   )r   ru   r   r   r7   r   rf   hashlibr   ior   	functoolsr   r   r   ImportErroressh_exceptionr   r   r   r	   re   r   dictr^   r   r   r   r   <module>   s6       %&