U
vf3 @ s d dl mZmZ d dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z
mZ d dlmZmZmZmZ d dlmZ G d d
d
eZG d d d eZG d
d dZdS ) ) encodebytes decodebytesN)MutableMapping)sha1)HMAC)PKeyUnknownKeyType)
get_loggerconstant_time_bytes_eqbu) SSHExceptionc @ s e Zd ZdZd#ddZdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zed$d!d"ZdS )%HostKeysaZ
Representation of an OpenSSH-style "known hosts" file. Host keys can be
read from one or more files, and then individual hosts can be looked up to
verify server keys during SSH negotiation.
A `.HostKeys` object can be treated like a dict; any dict lookup is
equivalent to calling `lookup`.
.. versionadded:: 1.5.3
Nc C s g | _ |dk r| | dS )z
Create a new HostKeys object, optionally loading keys from an OpenSSH
style host-key file.
:param str filename: filename to load host keys from, or ``None``
N)_entriesload)selffilename r B/opt/alt/python38/lib/python3.8/site-packages/paramiko/hostkeys.py__init__/ s zHostKeys.__init__c C sH | j D ](}||jkr|j |kr||_ dS q| j t|g| dS )a.
Add a host key entry to the table. Any existing entry for a
``(hostname, keytype)`` pair will be replaced.
:param str hostname: the hostname (or IP) to add
:param str keytype: key type (``"ssh-rsa"`` or ``"ssh-dss"``)
:param .PKey key: the key to add
N)r hostnameskeyget_nameappend HostKeyEntry)r hostnameZkeytyper er r r add; s
z HostKeys.addc
C s t |d}t|dD ]\}}| }t|dks|d dkr@qzt||}W n tk
rh Y qY nX |dk r|j}|D ] }| ||j r||j
| q|t|jr| j | qW 5 Q R X dS )a]
Read a file of known SSH host keys, in the format used by OpenSSH.
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")``.
If this method is called multiple times, the host keys are merged,
not cleared. So multiple calls to `load` will just call `add`,
replacing any existing entries and adding new ones.
:param str filename: name of the file to read host keys from
:raises: ``IOError`` -- if there was an error reading the file
r r #N)
open enumeratestriplenr from_liner
r checkr remover r )r r flinenolineentryZ
_hostnameshr r r r J s
z
HostKeys.loadc C s< t |d(}| jD ]}| }|r|| qW 5 Q R X dS )a
Save host keys into a file, in the format used by OpenSSH. The order
of keys in the file will be preserved when possible (if these keys were
loaded from a file originally). The single exception is that combined
lines will be split into individual key lines, which is arguably a bug.
:param str filename: name of the file to write
:raises: ``IOError`` -- if there was an error writing the file
.. versionadded:: 1.6.1
wN)r! r to_linewrite)r r r( r r* r r r savej s
z
HostKeys.savec sV G fdddt }g }| jD ]}| |r || q t|dkrJdS | || S )a
Find a hostkey entry for a given hostname or IP. If no entry is found,
``None`` is returned. Otherwise a dictionary of keytype to key is
returned. The keytype will be either ``"ssh-rsa"`` or ``"ssh-dss"``.
:param str hostname: the hostname (or IP) to lookup
:return: dict of `str` -> `.PKey` keys associated with this host
(or ``None``)
c sH e Zd Zdd Zdd Zdd Zdd Zd d
Z fd d Zd
d Z dS )z HostKeys.lookup..SubDictc S s || _ || _|| _d S N)Z _hostnamer _hostkeys)r r entriesZhostkeysr r r r s z)HostKeys.lookup..SubDict.__init__c s s | D ]
}|V qd S r1 keysr kr r r __iter__ s z)HostKeys.lookup..SubDict.__iter__c S s t | S r1 r$ r5 r r r r __len__ s z(HostKeys.lookup..SubDict.__len__c S s: t | jD ]"}|j |kr
| j| q6q
t|d S r1 )listr r r r' KeyErrorr r r r r r __delitem__ s
z,HostKeys.lookup..SubDict.__delitem__c S s0 | j D ] }|j |kr|j S qt|d S r1 )r r r r= r> r r r __getitem__ s
z,HostKeys.lookup..SubDict.__getitem__c sZ | j D ](}|jd krq|j |kr||_ qVqt g|}| j | | jj | d S r1 )r r r r r r2 )r r valr r r r __setitem__ s
z,HostKeys.lookup..SubDict.__setitem__c S s dd | j D S )Nc S s g | ]}|j d k r|j qS r1 )r r ).0r r r r
s
z9HostKeys.lookup..SubDict.keys..r r: r r r r5 s z%HostKeys.lookup..SubDict.keysN)
__name__
__module__ __qualname__r r8 r; r? r@ rC r5 r rB r r SubDict s rJ r N)r r _hostname_matchesr r$ )r r rJ r3 r r rB r lookup} s 0
zHostKeys.lookupc C sD |j D ]8}||ks8|dr|dst| |||r dS qdS )zl
Tests whether ``hostname`` string matches given SubDict ``entry``.
:returns bool:
|1|TF)r
startswithr
hash_host)r r r+ r, r r r rK s
zHostKeys._hostname_matchesc C sB | |}|dkrdS || d}|dkr2dS | | kS )a=
Return True if the given key is associated with the given hostname
in this dictionary.
:param str hostname: hostname (or IP) of the SSH server
:param .PKey key: the key to check
:return:
``True`` if the key is associated with the hostname; else ``False``
NF)rL getr Zasbytes)r r r r7 Zhost_keyr r r r& s
zHostKeys.checkc C s
g | _ dS )z;
Remove all host keys from the dictionary.
NrF r: r r r clear s zHostKeys.clearc c s | D ]
}|V qd S r1 r4 r6 r r r r8 s zHostKeys.__iter__c C s t | S r1 r9 r: r r r r; s zHostKeys.__len__c C s | |}|d krt||S r1 )rL r= )r r retr r r r@ s
zHostKeys.__getitem__c C sL d }t | jD ] \}}| ||r|} q,q|d kr)rc r r r: r r r __repr__ s zHostKeyEntry.__repr__)NN)N) rG rH rI rg r classmethodr% r. rz r r r r r : s
,r )base64r r rq r] ro collections.abcr hashlibr rf r Z
paramiko.pkeyr r Z
paramiko.utilr r
r r Zparamiko.ssh_exceptionr
r Exceptionri r r r r r s