jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Check is_missing_host_keys before adding them

pull/12/head
Sheng 2018-04-10 08:26:32 +08:00
parent 838d453336
commit 53ed3aef7c
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@ -40,8 +40,19 @@ class AutoAddPolicy(paramiko.client.MissingHostKeyPolicy):
""" """
lock = threading.Lock() lock = threading.Lock()
def is_missing_host_keys(self, client, hostname, key):
k = client._host_keys.lookup(hostname)
if k is None:
return True
host_key = k.get(key.get_name(), None)
if host_key is None:
return True
if host_key != key:
raise paramiko.BadHostKeyException(hostname, key, host_key)
def missing_host_key(self, client, hostname, key): def missing_host_key(self, client, hostname, key):
with self.lock: with self.lock:
if self.is_missing_host_keys(client, hostname, key):
keytype = key.get_name() keytype = key.get_name()
logging.info( logging.info(
'Adding {} host key for {}'.format(keytype, hostname) 'Adding {} host key for {}'.format(keytype, hostname)