python:實現簡單的ssh

標籤：continue   nbsp   簡單的   class   ddr   size   tin   enc   用戶端   

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()

 

python:實現簡單的ssh