使用 Python bz2 模块进行数据压缩与解压缩

import bz2
import struct

def bzip_blocks_decompress_all(data, offset):
    """
    解压缩包含多个 bzip2 压缩块的数据。

    Args:
        data: 包含压缩块的字节串。
        offset: 起始偏移量，指示第一个压缩块的起始位置。

    Returns:
        一个元组，包含：
            - 解压缩后的字节数组。
            - 每个压缩块在原始数据中的起始和结束位置列表。
            - 每个解压缩块在解压缩后的数据中的起始和结束位置列表。
    """
    frames = bytearray()
    places_to_bzip = []  # 压缩块在原始数据中的位置
    places_to_unbzip = [] # 解压缩块在解压缩数据中的位置
    while offset < len(data):
        # 读取压缩块的大小 (4 字节，大端序无符号整数)
        block_cmp_bytes = struct.unpack_from('>L', data, offset)[0]
        offset += 4

        # 记录解压块的起始位置
        start = len(frames)

        # 解压缩数据
        frames += bz2.decompress(data[offset:offset + block_cmp_bytes])

        # 记录解压块的结束位置
        end = len(frames)

        # 记录位置信息
        places_to_bzip.append([start, end])
        places_to_unbzip.append([offset, offset + block_cmp_bytes])

        offset += block_cmp_bytes

    return frames, places_to_bzip, places_to_unbzip

# 示例数据生成
def write_frame(f, data):
    bzdata = bz2.compress(data)
    # Write size of compressed data as big-endian 4-byte integer,
    # then the compressed data.
    f.write(struct.pack('>L', len(bzdata)) + bzdata)

with open('file.bin', 'wb') as f:
    f.write(b'A' * 24)  # header in the original data?
    write_frame(f, b'B' * 50)  # compressed frames
    write_frame(f, b'C' * 25)
    write_frame(f, b'D' * 30)
    write_frame(f, b'E' * 12)

# 读取数据
offset = 24
with open('file.bin','rb') as fobj:
    buffer = fobj.read()

# 解压缩数据
buffer_unbzip, places_to_bzip, places_to_unbzip = bzip_blocks_decompress_all(buffer, offset)

# 重新压缩并验证
for (bstart, bend), (unbstart, unbend) in zip(places_to_bzip, places_to_unbzip):
    a1 = buffer[unbstart:unbend]
    a2 = buffer_unbzip[bstart:bend]

    # 重新压缩
    a3 = bz2.compress(a2)
    print(a1 == a3, a2)