Python--网络编程-----socket编程示例--模拟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', 8081))  # 0-65535:0-1024给操作系统使用
phone.listen(5)

print('starting......')
while True:  # 链接循环
    conn, client_addr = phone.accept()
    print(client_addr)

    while True:  # 通信循环
        try:
            # 1、收命令
            cmd = conn.recv(1024)
            if not cmd:  # 适用于linux操作系统
                break
            print('客户端的数据', cmd)

            # 2、执行命令，拿到结果
            obj = subprocess.Popen(cmd.decode('utf-8'), shell=True,
                                   stdout=subprocess.PIPE,
                                   stderr=subprocess.PIPE)

            stdout = obj.stdout.read()
            stderr = obj.stderr.read()
            # 3、把命令的结果返回给客户端
            conn.send(stdout+stderr)

        except ConnectionResetError:  # 适用于Windows操作系统
            break

    conn.close()

phone.close()

二、客户端

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', 8081))  # 0-65535:0-1024给操作系统使用
phone.listen(5)

print('starting......')
while True:  # 链接循环
    conn, client_addr = phone.accept()
    print(client_addr)

    while True:  # 通信循环
        try:
            # 1、收命令
            cmd = conn.recv(1024)
            if not cmd:  # 适用于linux操作系统
                break
            print('客户端的数据', cmd)

            # 2、执行命令，拿到结果
            obj = subprocess.Popen(cmd.decode('utf-8'), shell=True,
                                   stdout=subprocess.PIPE,
                                   stderr=subprocess.PIPE)

            stdout = obj.stdout.read()
            stderr = obj.stderr.read()
            # 3、把命令的结果返回给客户端
            conn.send(stdout+stderr)

        except ConnectionResetError:  # 适用于Windows操作系统
            break

    conn.close()

phone.close()

在客户端运行dir，结果为：

C:Usersxu516PycharmProjectsPython全栈开发venvScriptspython.exe "C:/Users/xu516/PycharmProjects/Python全栈开发/第三模块/网络编程/04 模拟ssh远程执行命令/客户端.py"
>>:dir
 驱动器 C 中的卷是 OS
 卷的序列号是 7849-BAF9

 C:Usersxu516PycharmProjectsPython全栈开发第三模块网络编程4 模拟ssh远程执行命令 的目录

2018/04/06  13:07    <DIR>          .
2018/04/06  13:07    <DIR>          ..
2018/04/05  21:18                 0 __init__.py
2018/04/06  13:07               376 客户端.py
2018/04/06  13:06             1,146 服务端.py
2018/04/06  11:56               722 补充.py
               4 个文件          2,244 字节
               2 个目录 36,491,399,168 可用字节

>>:

服务端运行结果：

starting......
('127.0.0.1', 62806)
客户端的数据 b'dir'