网络编程——粘包解决

粘包：让发送端在发送数据之前,先发送数据的长度,让接收端知道有要接收多少数据,然后使用循环接收所有数据

粘包问题：接收端(client)不清楚发送端(server)会发多长的数据

模拟粘包：

服务端

import socket
import subprocess
# 创建服务器对象
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定ip &port
server.bind(('127.0.0.1', 8989))
# 设置监听 设置连接数
server.listen(128)
# 获取客户端连接
conn, addr = server.accept()
while 1:
    try:
        # 获取客户端 数据
        data = conn.recv(1024)
        if not data:
            raise ConnectionResetError
        print(f"客户端发送的数据------------{data.decode('gbk')}")
        # 调用subprocess 模块 获取 cmd 的结果
        sub = subprocess.Popen(
            data.decode('gbk'),
            shell=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE
        )
        # 获取对应的结果
        out_data = sub.stdout.read()
        err_data = sub.stderr.read()
        # 发送数据 给客户端
        conn.send(out_data+err_data)
    except ConnectionResetError:
        conn.close()
        break
server.close()

客户端

import socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 连接服务器
client.connect(('127.0.0.1', 8989))
while 1:
    # 发送数据给服务器
    msg = input("请输入cmd 指令： q 退出").strip()
    if not msg:
        continue
    elif msg == 'q':
        client.close()
        break
    # 给 服务器发送 数据
    client.send(msg.encode('gbk'))
    # 接收 服务器数据
    data = client.recv(1024)
    print(f"服务器发送的数据----------{data.decode('gbk')}")

结果：

从以上运行结果可以看出，第二次输入dir2指令本来该显示报错指令，但因为粘包问题，显示的仍然是上一次指令的结果。

粘包解决:

粘包的主要原因：

- 发送方每次写入数据 < 套接字（Socket）缓冲区大小；

- 接收方读取套接字（Socket）缓冲区数据不够及时。

解决问题：让发送端（server） 发送数据之前，先返回一个数据长度，再发送数据

代码：

服务端：

import socket
import subprocess
import struct
# 创建服务器对象
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定ip &port
server.bind(('127.0.0.1', 8989))
# 设置监听 设置连接数
server.listen(128)
print("服务器启动》》》》》》》》》》》")
# 获取客户端连接
conn, addr = server.accept()
while 1:
    try:
        # 获取客户端 数据
        data = conn.recv(1024)
        if not data:
            raise ConnectionResetError
        print(f"客户端发送的数据------------{data.decode('gbk')}")
        # 调用subprocess 模块 获取 cmd 的结果
        sub = subprocess.Popen(
            data.decode('gbk'),
            shell=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE
        )
        # 获取对应的结果
        out_data = sub.stdout.read()
        err_data = sub.stderr.read()
        # 发送数据包长度
        # 打包 数据长度
        data_head = struct.pack('i', len(out_data + err_data))
        conn.send(data_head)
        # 发送数据 给客户端
        conn.send(out_data+err_data)
    except ConnectionResetError:
        conn.close()
        break
server.close()

客户端：

import socket
import struct
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 连接服务器
client.connect(('127.0.0.1', 8989))
while 1:
    # 发送数据给服务器
    msg = input("请输入cmd 指令(q 退出)：").strip()
    if not msg:
        continue
    elif msg == 'q':
        client.close()
        break
    # 给 服务器发送 数据
    client.send(msg.encode('gbk'))
    # 1. 接收数据包 长度
    data_len = client.recv(4)
    rest_data_len = struct.unpack('i', data_len)
    print(rest_data_len, '数据长度')
    # 准备一个空字节数据
    rest_data = b''
    # 准备数据的初始长度
    num_len = 0
    while num_len < rest_data_len[0]:
        # 接收 服务器数据
        data = client.recv(1024)
        # 拼接数据
        rest_data +=data
        # 加上 获取的字节长度
        num_len += len(data)
    print(f"服务器发送的数据----------{rest_data.decode('gbk')}")

运行结果：

客户端

服务端