Python3之基于tcp模拟ssh远程执行命令

服务端

import socket
import subprocess

phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
phone.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
phone.bind(('127.0.0.1', 8080))

phone.listen(5)

#print('starting...')
while 1:

    conn, addr = phone.accept()
    #print(conn, addr)
    while 1:
        try:
            data = conn.recv(1024)
            if not data: break
            res = subprocess.Popen(data.decode('utf-8'), shell=True,
                                   stdout=subprocess.PIPE,
                                   stderr=subprocess.PIPE)

            stdout = res.stdout.read()
            stderr = res.stderr.read()

            conn.send(stdout + stderr)
        except Exception as e:
            break
    conn.close()

phone.close()

客户端

import socket

phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

phone.connect(('127.0.0.1', 8080))

while 1:
    msg = input('请输入:').strip()
    if not msg: continue

    phone.send(msg.encode('utf-8'))

    data = phone.recv(1024)
    print(data.decode('gbk'))

phone.close()