TCP 远程执行CMD (解决粘包问题) 代码

服务端

from socket import *
import subprocess,json,struct

server= socket(AF_INET,SOCK_STREAM)
server.bind(('127.0.0.1',8081))
server.listen(5)

while True:
    client,client_addr=server.accept()
    while True:
        try:
            cmd=client.recv(1024).decode('utf-8')
            if not cmd:break
            # 执行命令
            obj=subprocess.Popen(cmd,
                                 shell=True,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE
                                 )
            # 显示正确和错误信息
            stdout=obj.stdout.read()
            stderr=obj.stderr.read()
            print('执行结果长为%s' %(len(stdout)+len(stderr)))
            # 制定报头
            head_dic={'filename':'a.txt',
                      'md5':'rdgdgfdgfdgd',
                      'size':len(stdout)+len(stderr)
                      }
            head_json=json.dumps(head_dic)
            head_bytes=head_json.encode('utf-8')
            # 发送报头长度
            head_len=len(head_bytes)
            head_len_bytes = struct.pack('i', head_len)
            client.send(head_len_bytes)
            # 发送报头
            client.send(head_bytes)
            # 发送真实数据
            client.send(stdout)
            client.send(stderr)
        except ConnectionResetError:
            print('客户端异常关闭')
            break
    client.close()

server.close()

客户端

from socket import *
import json,struct

client= socket(AF_INET,SOCK_STREAM)
client.connect(('127.0.0.1',8081))

while True:
    cmd=input('>>>').strip()
    if not cmd:continue
    client.send(cmd.encode('utf-8'))
    # 先接受报头长度
    head_len=struct.unpack('i',client.recv(4))[0]
    # 接收报头
    head_bytes=client.recv(head_len)
    # 解析报头
    head_json=head_bytes.decode('utf-8')
    head_dic=json.loads(head_json)
    print(head_json)
    total_size=head_dic['size']
    # 取出真实数据
    recv_size=0
    finnally_data=b''
    while total_size>recv_size:
        recv_data=client.recv(1024)
        finnally_data+=recv_data
        recv_size+=len(recv_data)
    print(finnally_data.decode('gbk'))

client.close()