Python使用struct模块解决TCP黏包问题详解
本文深入解析TCP协议中黏包现象的成因,并详细介绍如何使用Python的struct模块定制通信协议。通过示例代码展示服务端与客户端如何利用struct.pack和unpack将数据长度转换为固定长度的bytes,从而在传输前告知接收端数据大小,有效解决黏包问题,确保网络通信的准确性与稳定性。
TCP黏包现象成因分析
为什么会出现黏包现象
黏包现象仅存在于TCP协议中。由于TCP是面向流的传输协议,其内部包含缓存机制以防止数据丢失,这导致在连续发送小数据包或接收数据大小不匹配时,容易引发黏包。其根本原因在于接收端无法预先知晓发送端数据的实际长度。
基于struct模块的解决方案
解决黏包问题的思路
在传输大量数据之前,发送端应首先告知接收端即将发送的数据大小。为了更优雅地解决这一问题,可以利用Python的模块来定制通信协议。
struct模块核心功能解析
关于模块
该模块的主要功能是将一个特定类型的数据(例如整数)转换为固定长度的字节序列。
import struct
ret = struct.pack('i',456872783) #'i'代表int,就是即将要把一共数字转换成固定长度(4个字节)的bystes类型
print(ret)
num = struct.unpack('i',ret) #转换回来,返回一个元组
print(num[0]) #提前元组中的值得到4096
利用解决黏包问题
服务端实现逻辑如下
服务端与客户端代码实现
import struct
import socket
sk = socket.socket()
sk.bind(('127.0.0.1',8080))
sk.listen()
conn,addr = sk.accept()
while True:
cmd = input('>>>')
if cmd == 'q': #当输入‘q'时,结束,并向客户端发送一个'q'。
conn.send(b'q')
break
conn.send(cmd.encode('gbk')) #将输入的cmd命令发送给客户端
num = conn.recv(4) #接收字节信息(返回的消息长度信息)。
num = struct.unpack('i',num)[0] #将接收的字节码转化为原来的类型并放在一个元组里面,后面加[0]是提前出元组中的值。
res = conn.recv(int(num)).decode('gbk') #接收长度为num 的消息。
print(res) #打印
conn.close()
sk.close()
客户端实现逻辑如下
import struct
import socket
import subprocess
sk = socket.socket()
sk.connect(('127.0.0.1',8080))
while True:
cmd = sk.recv(1024).decode('gbk') #接收服务端发送来的cmd命令
if cmd == 'q': #当接收到‘q'时,结束。
break
# 在客户端执行接收到的cmd命令。并将正确的消息和错误的消息分别放入stdout和stderr管道。
res = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
std_out = res.stdout.read() #读取管道内正确的消息
std_err = res.stderr.read() #读取管道内错误的消息
len_num = len(std_out)+len(std_err) #计算正确和错误消息的总长度
num_by = struct.pack('i',len_num) #将消息总长度转换成长度为4的字节码
sk.send(num_by) #发送消息长度信息
sk.send(std_out) #发送正确消息
sk.send(std_err) #发送错误消息
sk.close()
以上即为使用Python的模块解决TCP黏包问题的详细内容,更多关于Python 模块的技术资料,请参阅本站其他相关文章。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









