利用struct模块解决int类型粘包问题

Posted 2021-06-30 J哥.

大家好我是J哥

基于tcp远程协议实现执行命令时如果你打开的数据量很大就会产生粘包问题，其中的原因是因为tcp协议中所运用的算法，而udp协议则不会出现粘包问题

粘包问题出现的原因：

 1. 因为tcp是流式协议（数据像水流一样黏在一起，没有任何边界） （Bytes>1024）
 2. 收数据没收干净，有残留 就会和下一次命令结果混在一起

解决方法：

解决的核心法门 ：每次都收干净，不要有任何残留

客户端：

# -*- coding: utf-8 -*-
"""
版权所有：J哥
微信号：wws_0904
"""
# import socket
import struct
from socket import *

client = socket(AF_INET, SOCK_STREAM)  # TCP协议

client.connect(('127.0.0.1', 8080))

# client.listen(5)

while True:
    msg = input('输入命令>>：').strip()
    if len(msg) == 0: continue
    client.send(msg.encode('utf-8'))

    # 解决粘包问题思路：
    # 1.拿到数据的总大小 total_size
    # 2.recv_size=0,循环接收，每接收一次，recv_size+=接收的长度
    # 3.直到recv_size=total_size 结束循环

    header = client.recv(4)

    total_size = struct.unpack('i', header)[0]

    recv_size = 0
    cmd_res = b''
    while recv_size < total_size:
        cmd_data = client.recv(1024)  # 本次接收最大接收1024个字节（Bytes）
        recv_size += len(cmd_data)
        print(cmd_data.decode('gbk'), end='')  # win 解码 用gbk
        # cmd_res += cmd_data
        # print(cmd_res.decode('utf-8'))  # mac 解码
    else:
        print()
client.close()

# 粘包问题出现的原因
# 1. 因为tcp是流式协议（数据像水流一样黏在一起，没有任何边界） （Bytes>1024）
# 2. 收数据没收干净，有残留 就会和下一次命令结果混在一起

# 解决的核心法门 ：每次都收干净，不要有任何残留

服务端：

通过struct模块 让其长度——>固定的Bytes

该模块可以把一个类型，如数字，转成固定长度的bytes

>>> struct.pack('i',1111111111111)

struct.error: 'i' format requires -2147483648 <= number <= 2147483647 #这个是范围

正常用'i'模式就够用了，如果不行就用'Q'模式

# -*- coding: utf-8 -*-
"""
版权所有：J哥
微信号：wws_0904
"""
# import socket
from socket import *
import subprocess
import struct  # 解析出int-->固定长度的bytes

# ('127.0.0.1', 8080)
server = socket(AF_INET, SOCK_STREAM)

server.bind(('127.0.0.1', 8080))
server.listen(5)

# 服务端做的两件事
# 1.循环的从半连接池中取出链接请求与其建立双向链接，拿到链接对象
while True:
    conn, client_addr = server.accept()  # conn双向链接

    # 2，拿到链接对象与其进行通信循环
    while True:
        try:
            cmd = conn.recv(1024)
            if len(cmd) == 0: break
            # conn.send(cmd.upper())
            obj = subprocess.Popen(cmd.decode('utf-8'), shell=True,
                                   stdout=subprocess.PIPE, stderr=subprocess.PIPE)
            stdout_res = obj.stdout.read()  # Bytes类型
            stderr_res = obj.stderr.read()
            total_size = len(stdout_res) + len(stderr_res)

            # 先发数据的头信息（固定长度的信息（Bytes））,对数据的描述信息
            # int-->固定长度的bytes  用到模块 struct
            header = struct.pack('i', total_size)
            conn.send(header)

            # 再发真实的数据
            conn.send(stdout_res + stderr_res)
        except Exception as e:
            break
    conn.close()

看到这里留下你的痕迹吧！算是对J哥的最大的鼓励。