Python网络编程基础 ❸ struct模块 解决黏包问题

struct模块

　　1、demo代码

import struct

# 将一个数字打包，转为bytes
len = struct.pack('i', 123434)  # 第一参数i，表示整数，即第二个传入的数据类型。i最常用
print(len)

# 解开包，是一个两位的元组
re = struct.unpack('i', len)
print(re[0])

　　2、上传和下载文件演示

　　　　1)服务端

import socket,json,struct

sk = socket.socket()
sk.bind(('127.0.0.1', 8888))
sk.listen(5)

request = sk.accept()[0]

# 接收数据长度
msg_len = request.recv(4)
data_len = struct.unpack('i', msg_len)

# 接收数据
data = request.recv(int(data_len[0])).decode('utf-8') #
print(data)
dic = json.loads(data)

if dic['operate'] == 'upload':
    with open(dic['filename'], 'wb') as f:
        f.write(dic['content'].encode('utf-8'))
elif dic['operate'] == 'download':
    pass

request.close()
sk.close()

　　　　2）客户端（上传）

import socket,json,struct
from os import path

# 客户端进行上传和下载

sk = socket.socket()
sk.connect(('127.0.0.1', 8888))


def get_filename(file_path):
    '''
    获取路径中的文件名
    :param file_path:
    :return:
    '''
    return path.basename(file_path)

operate = ['upload', 'download']
for i, oper in enumerate(operate, 1):
    print('%d  %s'%(i, oper))
num = input('请输入操作对应的序号：>>>').strip()

if num == '1':
    '''上传操作'''
    file_path = r'C:\Users\台风\Documents\bar.job'  # 文件名不能是中文！
    # 告诉对方要传数据的大小
    file_size = struct.pack('i', 4096)
    sk.send(file_size)
    # 告诉对方要上传的文件的名字
    filename = get_filename(file_path)
    # 准备数据
    with open(file_path, encoding='utf-8') as f:
        data = f.read()
    # 用字典将所有报头和数据封装在一起
    dic = {'operate': 'upload', 'filename': filename, 'content': data}
    # 将字典使用json序列化，用于传输
    dic_json = json.dumps(dic)
    print(dic_json)
    sk.send(dic_json.encode('utf-8'))
elif num == '2':
    '''下载操作'''
    pass

sk.close()

 

解决黏包问题

　　1、初步解决，发送数据前先将其大小发过去

　　　　1）server端

import socket

sk = socket.socket()
sk.bind(('127.0.0.1', 8888))
sk.listen()

conn, addr = sk.accept()
msg = conn.recv(1024).decode('utf-8')  # 先接收数据长度数据

data = conn.recv(int(msg))   # 根据长度设定缓存大小
print(data.decode('utf-8'))

data2 = conn.recv(int(msg))
print(data2.decode('utf-8'))

conn.close()
sk.close()

　　　　2）客户端　　　　

import socket, time
# 黏包的原因是没有给定要传信息的长度，如果事先知道就不会出现了
# 所以在每次传递前，发送长度消息
sk = socket.socket()
ip_port = ('127.0.0.1', 8888)
sk.connect(ip_port)

sk.send(b'4')
time.sleep(0.00001)  # 发送数据长度，与后面的发送数据之间，要隔开一段时间，否者会跟后面的数据黏包；
# 但后面的数据就不会了，如果发送的数据大于4，会丢弃！

sk.send(b'hero')
sk.send(b'goodbye')

sk.close()

　　2、使用struct解决远程执行dos命令时的黏包问题

　　　　1）服务器端

import socket,struct

sk = socket.socket()
sk.bind(('127.0.0.1', 8080))
sk.listen()

conn, addr = sk.accept()

while 1:
    ipt = input('>>>').encode('gbk')
    conn.send(ipt)
    msg = conn.recv(4)
    data_length = struct.unpack('i', msg)[0]
    data = conn.recv(data_length).decode('gbk')
    print(data)

conn.close()

sk.close()

　　　　2）客户端

import socket, subprocess, struct

sk = socket.socket()
sk.connect(('127.0.0.1', 8080))

while 1:
    msg = sk.recv(1024).decode('utf-8')
    print(msg)
    sp = subprocess.Popen(msg, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    # 得到要发送数据的长度
    std_err = sp.stderr.read()
    std_out = sp.stdout.read()
    data_length = len(std_err+std_out)

    # 使用struct
    length = struct.pack('i', data_length)
    # 发送数据长度
    sk.send(length)

    sk.send(std_err)
    sk.send(std_out)

sk.close()