python通过socket实现多个连接并实现ssh功能详解

首先我们来详细讲解如何通过 Python 的 socket 模块实现多个连接并实现 SSH 功能。

什么是 socket

Socket，顾名思义就是套接字，可以理解为一种通信机制，它是计算机之间进行通信的一种约定或一种方式。在计算机网络中，Socket 起着非常重要的作用，它可以用于实现客户端和服务器之间的通信。

socket 的基本用法

在 Python 中，使用 socket 模块可以实现 Socket 的各种操作，例如建立连接、传输数据、关闭连接等。下面，我们来看一下 socket 的基本用法。

1. 建立套接字

在 Python 中，使用 socket.socket() 方法来创建一个 Socket 对象，代码如下：

import socket

# 创建 Socket 对象
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

接下来，我们需要设置服务器的 IP 地址以及端口号：

# 设置 IP 地址和端口号
host = 'localhost'
port = 9999
# 连接服务器
s.connect((host, port))

1. 发送数据

在建立了连接之后，我们就可以开始发送数据了。使用 Socket 对象的 send() 方法，将数据发送给服务器，代码如下：

# 发送数据
msg = 'Hello, World!'
s.send(msg.encode('utf-8'))

1. 接收数据

接收服务器返回的数据，我们要使用 Socket 对象的 recv() 方法，代码如下：

# 接收服务器返回的数据
data = s.recv(1024)
print(data.decode('utf-8'))

这里，我们将服务器返回的数据使用 decode() 方法解码，变成我们能看懂的字符串。

1. 关闭连接

完成数据通信后，需要关闭连接。使用 Socket 对象的 close() 方法即可关闭连接，代码如下：

# 关闭连接
s.close()

Python 通过 socket 实现多个连接并实现 SSH 功能

了解了 socket 的基本用法之后，我们就可以开始实现多个连接并实现 SSH 功能了。具体实现可以借助 paramiko 模块。

1. 安装 paramiko 模块

paramiko 是一个实现 SSH 的第三方模块，可以方便的实现 Python 连接远程机器，交互式的执行命令。我们可以使用 pip 工具来安装 paramiko 模块，命令如下：

pip install paramiko

1. 建立 SSH 连接

import paramiko

# 创建 SSH 连接
client = paramiko.SSHClient()

# 自动添加 ssh key
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接远程机器
client.connect(hostname='localhost', port=22, username='username', password='password')

在上面的代码中，我们使用 paramiko.SSHClient() 方法来创建一个 SSHClient 对象。该对象可以负责与远程机器建立 SSH 连接，使用 set_missing_host_key_policy 方法来设置当远程机器的 host key 不在本地 known_hosts 文件中时，自动添加 host key。 connect 方法可用来连接远程机器，并传入需要连接的远程机器的 hostname、port、username 和 password。

1. 执行 Shell 命令

我们可以使用 invoke_shell 方法来执行 Shell 命令，代码如下：

# 执行 Shell 命令
shell = client.invoke_shell()
shell.send("echo 'hello world'")

以上代码将 "echo 'hello world'" 发送给远程机器，执行后会返回一个结果。

1. 关闭连接

完成后，我们需要手动关闭 SSH 连接，代码如下：

# 关闭连接
client.close()

示例代码

下面是一个完整的示例代码，实现了通过 socket 实现多个连接并实现 SSH 功能。

import socket
import paramiko

# 创建 Socket 对象
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 设置 IP 地址和端口号，绑定 Socket
host = 'localhost'
port = 9999
s.bind((host, port))

# 设置最大连接数
s.listen(5)

# 循环接受客户端连接
while True:
    # 接受客户端连接
    conn, addr = s.accept()
    print('Connected by', addr)

    # 创建 SSH 连接
    client = paramiko.SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(hostname='localhost', port=22, username='username', password='password')

    # 执行 Shell 命令
    shell = client.invoke_shell()
    while True:
        data = conn.recv(1024)
        if not data:
            break
        shell.send(data)
        output = shell.recv(1024)
        conn.send(output)

    # 关闭连接
    client.close()
    conn.close()

在上面的代码中，我们建立了一个 Socket 连接，并通过 bind 方法绑定 IP 地址和端口号。在接收到客户端请求连接之后，建立了一个 SSH 连接，然后通过 while 循环接收客户端发送的数据，并将数据发送到远程机器执行，并将结果返回给客户端。完成后，我们手动关闭了 SSH 连接和客户端连接。