10.17小作业 基于TCP开发一款远程CMD程序

基于TCP开发一款远程CMD程序

客户端连接服务器后，可以向服务器发送命令

服务器收到命令后执行，无论执行是否成功，无论执行几遍，都将执行结果返回给客户端

注意: 执行系统指令使用subprocess模块完成.

本题目不考虑接收时的粘包问题

# 客户端
import socket

client = socket.socket()
client.connect(('127.0.0.1',8889))

while True:
    cmd = input('请发送命令：').strip()
    if len(cmd) == 0:
        continue

    if cmd == 'q':
        client.send(b'q')
        break

    cmd = cmd.encode('utf-8')
    client.send(cmd)

    byte_data = client.recv(1024)
    print(byte_data.decode('gbk'))

# 服务端

import socket
import subprocess

server = socket.socket()
server.bind(('127.0.0.1',8889))
server.listen(5)
while True:
    conn,addr = server.accept()

    while True:
        try:
            cmd = conn.recv(1024)
            if len(cmd) == 0:
                continue

            cmd = cmd.decode('utf-8')
            if cmd == 'q':break
            obj = subprocess.Popen(cmd,shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)  # 这里cmd必须是字符串类型，不是bytes类型的字符串
            print(obj)  # <subprocess.Popen object at 0x00000000029C6D30>   对象

            # print(obj.stdout.read())  # b'' 没赋值变量名返回的是空
            success = obj.stdout.read()  # 必须赋值一个变量才有值
            # print(success)  #  返回的是二进制，可以直接基于网络传输

            # print(obj.stderr.read())  # b'' 没赋值变量名返回的是空
            error = obj.stderr.read()
            # print(error)  # 返回不正确命令的报错信息

            if success:
                conn.send(success)
            else:
                conn.send(error)
        except Exception as e:  # 捕获客服端人为断开连接，不至于服务端报错率而断开，毕竟服务端是24h不间断服务的
            print(e)
            break
    conn.close()