用socket来实现ssh功能远程执行命令

用socket来实现ssh功能远程执行命令，并获取返回值

socket_server.py

# -*- coding: utf-8 -*-
'''
@Time    : 2021/9/17 9:50
@Author  : HenryHE
@FileName: socket_server.py
@Software: PyCharm
'''
import socket
import subprocess
import os
server=socket.socket()
server.bind(("localhost",9999))
server.listen()
print("开始等待新的连接")
while True:
    conn,addr=server.accept()
    print("进来了一个新链接",addr)
    while True:
        data=conn.recv(1024)
        if not data:
            print("客户端%s连接已断开"%addr)
            break
        print("执行指令%s"%data.decode("utf-8"))
        #cmd_res = os.popen(data.decode()).read()
        # if len(cmd_res) ==0 :
        #     cmd_res="命令没有输出"
        #conn.send(cmd_res.encode("utf-8"))
        res=subprocess.Popen(data.decode("utf-8"),shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
        a=res.stdout.read()
        b=res.stderr.read()
        if a:
            print("发送的数据大小",len(a))
            conn.send(str(len(a)).encode("utf-8"))
            data=conn.recv(1024)
            conn.send(a)
        else:
            print("发送的数据大小",len(b))
            conn.send(str(len(b)).encode("utf-8"))
            data=conn.recv(1024)
            conn.send(b)
server.close()

socket_client.py

# -*- coding: utf-8 -*-

'''
@Time    : 2021/9/17 9:50
@Author  : HenryHE
@FileName: socket_client.py
@Software: PyCharm
'''
import socket
client=socket.socket()
client.connect(('localhost',9999))

while True:
    cmd = input("请输入需要执行的指令：").strip()
    if len(cmd) ==0:continue
    client.send(cmd.encode("utf-8"))
    cmd_res_size = client.recv(1024)
    print("命令结果大小：",cmd_res_size.decode(encoding='gbk'))
    received_size =0
    client.send("我跟你确认了".encode())
    while received_size != int(cmd_res_size.decode()):
        data = client.recv(1024)
        received_size+=len(data)#每次收到都有可能小于1024
        print(data.decode(encoding="gbk"))
    else:
        print("命令输出结果接收完毕")
client.close()