Python 之 SSH实例(可接受大字节返回）

Posted 2021-02-11

服务器端代码

#Author Kang

import os
import subprocess
import socket

server = socket.socket()
server.bind((‘localhost‘,9999))
server.listen()

print("等待客户端连接:>>>>")

while True:
    conn,addr = server.accept()
    print("接受了一个新的链接！！")
    while True:
        data = conn.recv(1024)              
        cmd = subprocess.getstatusoutput(data.decode())
        print("执行命令:>>",cmd)
        res_size = str(len(cmd[1].encode("utf-8")))
        print(res_size)
        if len(cmd[1]) != 0:
            conn.send(res_size.encode("utf-8"))
            conn.send(cmd[1].encode("utf-8"))
        elif cmd[0] == 0 and len(cmd[1]) == 0:
            res = ‘命令执行成功，但没有返回值‘
            conn.send(res_size.encode("utf-8"))
            conn.send(res.encode("utf-8"))
        if len(data) == 0:
            break
server.close()

客户端代码

#Author Kang

import socket

client =  socket.socket()

client.connect((‘localhost‘,9999))

while True:
    msg = input("请输入你要传递的信息:>>>>>")
    if len(msg) == 0:
        continue
    client.send(msg.encode("utf-8"))              #发送命令给服务器端
    cmd_size = client.recv(1024)                  #接受服务器返回结果的长度
    print("接受结果的大小为:",cmd_size.decode())
    recv_size = 0
    res_data = ""
    while recv_size < int(cmd_size):        #判断当前接受的大小与服务器执行命令大小，如小于继续接受
        data = client.recv(1024)
        recv_size += len(data)
        res_data += data.decode()
    else:
        print(res_data)
client.close()