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