Python - pysftp / paramiko - 使用指纹验证主机密钥

问题描述：

这段代码抛出了异常。如何在不将 SSH 指纹存储在文件中的情况下验证它？我相信下面的代码是为公钥设计的。但是，连接 SFTP 服务器的客户端验证了指纹，却没有获取到公钥。

import os
import shutil

import pysftp
import paramiko

connection_info = {
    'server': "example.com",
    'user': "user",
    'passwd': "password",
    'target_dir': "out/prod",
    'hostkey': "ssh-rsa 2048 d8:4e:f1:f1:f1:f1:f1:f1:21:31:41:14:13:12:11:aa",
}

def move_files_from_server_to_local(server, localpath):
    target_dir = server['target_dir']
    keydata = "d8:4e:f1:f1:f1:f1:f1:f1:21:31:41:14:13:12:11:aa"
    key = paramiko.RSAKey(data=decodebytes(keydata))
    options = pysftp.CnOpts()
    options.hostkeys.add('example.com', 'ssh-rsa', key)
    with pysftp.Connection(
                    server['server'],
                    username=server['user'],
                    password=server['passwd'],
                    cnopts=options) as conn:
        conn.get_d(target_dir, localpath)
        delete_files_from_dir(conn, target_dir)

move_files_from_server_to_local(connection_info, "/")

该代码基于使用 pysftp 验证主机密钥。

解决方案 1：

根据您的需要，您可以使用以下两种方法之一：

如果您只需要验证一个特定的主机密钥

使用ssh-keyscan（或类似方法）检索主机公钥：

ssh-keyscan example.com > tmp.pub

其tmp.pub格式如下（known_hosts文件格式）：

example.com ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEA0hVqZOvZ7yWgie9OHdTORJVI5fJJoH1yEGamAd5G3werH0z7e9ybtq1mGUeRkJtea7bzru0ISR0EZ9HIONoGYrDmI7S+BiwpDBUKjva4mAsvzzvsy6Ogy/apkxm6Kbcml8u4wjxaOw3NKzKqeBvR3pc+nQVA+SJUZq8D2XBRd4EDUFXeLzwqwen9G7gSLGB1hJkSuRtGRfOHbLUuCKNR8RV82i3JvlSnAwb3MwN0m3WGdlJA8J+5YAg4e6JgSKrsCObZK7W1R6iuyuH1zA+dtAHyDyYVHB4FnYZPL0hgz2PSb9c+iDEiFcT/lT4/dQ+kRW6DYn66lS8peS8zCJ9CSQ==

现在，您可以使用以下方法计算该公钥的指纹ssh-keygen：

ssh-keygen -l -f tmp.pub -E md5

（-E md5仅使用支持多种指纹算法且默认为 SHA256 的较新版本的 OpenSSH）

您将获得类似以下内容的信息：

2048 MD5:c4:26:18:cf:a0:15:9a:5f:f3:bf:96:d8:3b:19:ef:7b example.com (RSA)

如果指纹与您拥有的指纹匹配，那么您现在可以安全地假设这tmp.pub是一个合法的公钥并在代码中使用它：

from base64 import decodebytes
# ...

keydata = b"""AAAAB3NzaC1yc2EAAAABIwAAAQEA0hV..."""
key = paramiko.RSAKey(data=decodebytes(keydata))
cnopts = pysftp.CnOpts()
cnopts.hostkeys.add('example.com', 'ssh-rsa', key)

with pysftp.Connection(host, username, password, cnopts=cnopts) as sftp:

（基于使用 pysftp 验证主机密钥）

如果您需要根据指纹自动验证主机密钥

例如因为指纹来自外部配置。

我不确定 pysftp 有限的 API 是否允许这样做。你可能需要跳过 pysftp，直接使用Paramiko 库（pysftp 内部使用 Paramiko）。

使用 Paramiko，您可以巧妙地实现MissingHostKeyPolicy接口。

从如何AutoAddPolicy实现开始：

class AutoAddPolicy (MissingHostKeyPolicy):
    """
    Policy for automatically adding the hostname and new host key to the
    local `.HostKeys` object, and saving it.  This is used by `.SSHClient`.
    """

    def missing_host_key(self, client, hostname, key):
        client._host_keys.add(hostname, key.get_name(), key)
        if client._host_keys_filename is not None:
            client.save_host_keys(client._host_keys_filename)
        client._log(DEBUG, 'Adding %s host key for %s: %s' %
                    (key.get_name(), hostname, hexlify(key.get_fingerprint())))

请注意，代码中指纹可用hexlify(key.get_fingerprint())。只需将该值与您现有的指纹进行比较即可。如果匹配，则返回。否则，抛出异常，就像RejectPolicy所做的那样。

另一个解决方案（即使使用 pysftp 也能工作）是实现PKey仅保存指纹的功能。并实现其__eq__方法（或__cmp__Paramiko 3.0 之前版本）以仅比较指纹。然后可以将这样的 实例PKey添加到cnopts.hostkeys.add。

OP 在他的回答中发布了这种方法的实现。据称对于 Python 3，需要更复杂的实现，如使用 pysftp 和 Python 3 仅使用服务器指纹连接到 SFTP 服务器中所示。

解决方案 2：

根据 Martin Prikryl 的回答，以下是我的解决方案。

def trim_fingerprint(fingerprint):
    if fingerprint.startswith('ssh-rsa 2048 '):
        return fingerprint[len('ssh-rsa 2048 '):]
    return fingerprint
def clean_fingerprint(fingerprint):
    return trim_fingerprint(fingerprint).replace(':', '')

class FingerprintKey:
    def __init__(self, fingerprint):
        self.fingerprint = clean_fingerprint(fingerprint)
    def compare(self, other):
        if callable(getattr(other, "get_fingerprint", None)):
            return other.get_fingerprint() == self.fingerprint
        elif clean_fingerprint(other) == self.get_fingerprint():
            return True
        elif md5(other).digest().encode('hex') == self.fingerprint:
            return True
        else:
            return False
    def __cmp__(self, other):
        return self.compare(other)
    def __contains__(self, other):
        return self.compare(other)
    def __eq__(self, other):
        return self.compare(other)
    def __ne__(self, other):
        return not self.compare(other)
    def get_fingerprint(self):
        return self.fingerprint
    def get_name(self):
        return u'ssh-rsa'
    def asbytes(self):
         # Note: This returns itself.
         #   That way when comparisons are done to asbytes return value,
         #   this class can handle the comparison.
        return self

用法：

options = pysftp.CnOpts()
options.hostkeys.clear()
options.hostkeys.add('www.example.com', u'ssh-rsa', FingerprintKey("ssh-rsa 2048 d8:4e:f1:f1:f1:f1:f1:f1:21:31:41:14:13:12:11:aa"))
    with pysftp.Connection(
                    'www.example.com',
                    username='user',
                    password='password',
                    cnopts=options) as conn:
        conn.get_d('remote/filedir', 'c:/local/output')