python 模拟ssh登录并发

原创

mob649e81593bda 2023-07-27 08:17:05 ©著作权

文章标签 Python python 客户端 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e81593bda的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 模拟 SSH 登录并发

在网络通信中，SSH (Secure Shell) 是一种网络协议，用于在不安全的网络中安全地进行远程登录和执行命令。Python 提供了多个库来模拟 SSH 登录，使我们能够编写自动化脚本来执行远程操作。

本文将介绍如何使用 paramiko 库来模拟 SSH 登录并发，并提供代码示例。paramiko 是一个支持 SSH2 协议的 Python 库，提供了许多功能，包括远程执行命令、上传和下载文件等。

安装 paramiko

在开始之前，我们需要安装 paramiko 库。通过以下命令可以使用 pip 来安装：

pip install paramiko

连接远程主机

首先，我们需要建立与远程主机的连接。为了简化示例，我们将使用 localhost 作为远程主机。在实际应用中，您需要将主机名、用户名和密码替换为您自己的信息。

import paramiko

# 创建 SSH 客户端对象
client = paramiko.SSHClient()

# 允许连接到不在 known_hosts 文件中的主机
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接到远程主机
client.connect('localhost', username='your_username', password='your_password')

执行远程命令

一旦建立了连接，我们可以使用 exec_command 方法来执行远程命令。该方法返回一个三元组，包含了命令的标准输入、标准输出和标准错误输出。

# 执行远程命令
stdin, stdout, stderr = client.exec_command('ls')

# 读取命令的输出
output = stdout.read().decode('utf-8')

# 打印输出结果
print(output)

# 关闭连接
client.close()

并发执行命令

有时候，我们需要同时执行多个命令，并等待它们的结果。paramiko 提供了 Channel 类来实现并发执行命令。

import paramiko

# 创建 SSH 客户端对象
client = paramiko.SSHClient()

# 允许连接到不在 known_hosts 文件中的主机
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接到远程主机
client.connect('localhost', username='your_username', password='your_password')

# 打开一个 SSH 会话通道
channel = client.get_transport().open_session()

# 并发执行多个命令
commands = ['ls', 'pwd', 'whoami']
for command in commands:
    # 执行命令并等待结果
    channel.exec_command(command)
    exit_status = channel.recv_exit_status()

    # 读取命令的输出
    output = channel.recv(4096).decode('utf-8')

    # 打印输出结果
    print(f'Command: {command}')
    print(f'Exit status: {exit_status}')
    print(f'Output:\n{output}')

# 关闭连接
client.close()