python:实现简单的ssh

socket_server_ssh.py

#!usr/bin/env python
# -*- coding:utf-8 -*-

__author__ = "Samson"

import socket
import os

server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.bind(("localhost",6969))
server.listen()
while True:
    conn,addr = server.accept()
    print("new conn:",addr)
    while True:
        data = conn.recv(1024)
        if not data:
            print("客户端已断开!")
            break
        print("执行指令:",data)
        cmd_res = os.popen(data.decode()).read()
        if len(cmd_res) == 0:
            print("Command has no output")
        conn.send(str(len(cmd_res.encode())).encode("utf-8"))#连续两条send可能会出现粘包，建议send后立即recv一下
        conn.send(cmd_res.encode("utf-8"))
server.close()

socket_server_client.py

#!usr/bin/env python
# -*- coding:utf-8 -*-

__author__ = "Samson"
import socket
client = socket.socket()
client.connect(("localhost",6969))
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)
    received_size = 0
    while received_size < int(cmd_res_size.decode()):
        data = client.recv(1024)
        received_size += len(data)#接收到的数据可能小于1024
        print(data.decode())
    else:
        print("cmd res receive done!")

client.close()