只有tcp才会出现粘包问题
解决粘包问题的核心思想:每次都会读完收到的内容
自定义报头:
固定报头长度,报头内容为数据长度
:
该模块可以把一个类型,如数字,转成固定长度的bytes,这样就可以把数据长度转换成固定长度的bytes
用法:struct.pack('i',1111111111111)
但是之前我们了解过报头的内容并不止这一个,所以,我们可以用字典的形式作为报头,字典内容比如是文件名,md5值,数据长度,然后用json来序列化这个字典给他发送出去
这样又会产生一个新的问题,,,,接收方因为无法确定字典的长度而又会粘包
所以就是:
首先通过struct发送转换成bytes的字典的长度,,然后发送报头字典,最后发送数据
练习
让我们基于tcp先制作一个远程执行命令的程序(1:执行错误命令 2:执行dir 3:执行tasklist(查看进程列表))
res.stdout.read()读出的就是GBK编码的,在接收端需要用
服务端
from socket import *
import json, struct, subprocess
server = socket()
server.bind(('127.0.0.1', 8080))
server.listen(5)
while True:
conn, addr = server.accept()
while True:
try:
cmd = conn.recv(1024)
res = subprocess.Popen(cmd.decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout = res.stdout.read()
stderr = res.stderr.read()
# 这是报头字典
hand_dic={
'file_name':'a.txt',
'md5':'4f56ds4f65ds4f',
'data_size':len(stdout)+len(stderr)
}
#字典序列化后转化为二进制
hand_json = json.dumps(hand_dic)
hand_bytes = hand_json.encode('utf-8')
#发报头字典长度
hand_size=struct.pack('i',len(hand_bytes))
conn.send(hand_size)
#发报头字典
conn.send(hand_bytes)
#发数据
conn.send(stdout + stderr)
except ConnectionResetError:
break
except ConnectionAbortedError:
break
conn.close()
server.close()
客户端
from socket import *
import json, struct
client=socket()
client.connect(('127.0.0.1',8080))
while True:
cmd=input('::>').strip()
client.send(cmd.encode('utf-8'))
#接收报头字典长度
hand_size=struct.unpack('i',client.recv(4))[0]
#接收报头
hand_bytes=client.recv(hand_size)#收到bytes格式的json
hand_json=hand_bytes.decode('utf-8')#还原成json格式
hand=json.loads(hand_json)#反序列化
#数据长度
total_size=hand['data_size']
# 循环取值
data_size=0#已经接收的长度
res=b''#用来存放每次取出的数据
while data_size<total_size:
data=client.recv(1024) #每次取1024字节
res+=data #取完后存于res
data_size+=len(data)#更新已经取出的长度
print(res.decode('gbk'))#打印取出的数据,注意!!!Windows默认gbk编码的,所以这里要用gbk解码
client.close()