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

服务端

1 1 from socket import *
2 2 import subprocess,json,struct
3 3 
4 4 server= socket(AF_INET,SOCK_STREAM)
5 5 server.bind(('127.0.0.1',8081))
6 6 server.listen(5)
7 7 
8 8 while True:
9 9     client,client_addr=server.accept()
1010     while True:
1111         try:
1212             cmd=client.recv(1024).decode('utf-8')
1313             if not cmd:break
1414             # 执行命令
1515             obj=subprocess.Popen(cmd,
1616                                  shell=True,
1717                                  stdout=subprocess.PIPE,
1818                                  stderr=subprocess.PIPE
1919                                  )
2020             # 显示正确和错误信息
2121             stdout=obj.stdout.read()
2222             stderr=obj.stderr.read()
2323             print('执行结果长为%s' %(len(stdout)+len(stderr)))
2424             # 制定报头
2525             head_dic={'filename':'a.txt',
2626                       'md5':'rdgdgfdgfdgd',
2727                       'size':len(stdout)+len(stderr)
2828                       }
2929             head_json=json.dumps(head_dic)
3030             head_bytes=head_json.encode('utf-8')
3131             # 发送报头长度
3232             head_len=len(head_bytes)
3333             head_len_bytes = struct.pack('i', head_len)
3434             client.send(head_len_bytes)
3535             # 发送报头
3636             client.send(head_bytes)
3737             # 发送真实数据
3838             client.send(stdout)
3939             client.send(stderr)
4040         except ConnectionResetError:
4141             print('客户端异常关闭')
4242             break
4343     client.close()
4444 
4545 server.close()

客户端

1 1 from socket import *
2 2 import json,struct
3 3 
4 4 client= socket(AF_INET,SOCK_STREAM)
5 5 client.connect(('127.0.0.1',8081))
6 6 
7 7 while True:
8 8     cmd=input('>>>').strip()
9 9     if not cmd:continue
1010     client.send(cmd.encode('utf-8'))
1111     # 先接受报头长度
1212     head_len=struct.unpack('i',client.recv(4))[0]
1313     # 接收报头
1414     head_bytes=client.recv(head_len)
1515     # 解析报头
1616     head_json=head_bytes.decode('utf-8')
1717     head_dic=json.loads(head_json)
1818     print(head_json)
1919     total_size=head_dic['size']
2020     # 取出真实数据
2121     recv_size=0
2222     finnally_data=b''
2323     while total_size>recv_size:
2424         recv_data=client.recv(1024)
2525         finnally_data+=recv_data
2626         recv_size+=len(recv_data)
2727     print(finnally_data.decode('gbk'))
2828 
2929 client.close()