iOS Swift下实现与后端兼容的zlib对称压缩函数问题

问题核心 #

现有解压函数需设置headerSize=2才能正常解析后端数据，说明后端返回的压缩数据格式与本地默认压缩输出不一致——大概率是后端压缩数据未包含zlib默认的2字节头部，或者解压逻辑需要跳过前2字节进行解析。我们需要调整压缩逻辑，生成与后端格式完全匹配的压缩数据。

实现方案（Swift 5.8，iOS 16+） #

使用系统Compression框架实现适配逻辑，以下两种方案对应不同的后端格式场景：

方案1：生成无zlib头部的原始Deflate数据

如果解压函数的headerSize=2是指跳过zlib默认头部，直接生成原始deflate压缩数据（不带zlib头）即可适配：

import Foundation
import Compression

func compress(data: Data) -> Data? {
    let bufferSize = 4096
    var compressedData = Data()
    
    guard let stream = compression_stream_init(
        allocator: nil,
        operation: COMPRESSION_STREAM_ENCODE,
        algorithm: COMPRESSION_ALGORITHM_ZLIB,
        flags: COMPRESSION_STREAM_ZLIB_NO_HEADER
    ) else {
        return nil
    }
    defer {
        compression_stream_destroy(stream)
    }
    
    var inputBuffer = [UInt8](data)
    var outputBuffer = [UInt8](repeating: 0, count: bufferSize)
    
    stream?.src_ptr = inputBuffer
    stream?.src_size = inputBuffer.count
    stream?.dst_ptr = outputBuffer
    stream?.dst_size = outputBuffer.count
    
    var status = compression_stream_process(stream, flags: COMPRESSION_STREAM_FINALIZE)
    while status == COMPRESSION_STATUS_OK {
        compressedData.append(outputBuffer, count: outputBuffer.count - stream!.dst_size)
        
        stream?.dst_ptr = outputBuffer
        stream?.dst_size = outputBuffer.count
        
        status = compression_stream_process(stream, flags: COMPRESSION_STREAM_FINALIZE)
    }
    
    guard status == COMPRESSION_STATUS_END else {
        return nil
    }
    
    compressedData.append(outputBuffer, count: outputBuffer.count - stream!.dst_size)
    return compressedData
}

方案2：手动添加2字节自定义头部

如果后端数据格式是2字节头部 + zlib压缩数据，且解压函数需要跳过这2字节解析，可在标准zlib压缩数据前添加后端约定的2字节头部：

import Foundation
import Compression

func compressWithCustomHeader(data: Data) -> Data? {
    guard let zlibCompressed = data.compressed(using: .zlib) else {
        return nil
    }
    
    // 替换为后端实际使用的2字节头部值，示例用0x0001
    var customHeader: UInt16 = 0x0001
    let headerData = Data(bytes: &customHeader, count: MemoryLayout<UInt16>.size)
    
    return headerData + zlibCompressed
}

验证步骤 #

1. 用上述压缩函数生成数据后，传入现有decompress函数并设置headerSize=2，检查是否能正常还原原始数据。
2. 测试后端返回的压缩数据，确认现有解压函数仍能正常解析，确保兼容性。

内容的提问来源于stack exchange，提问作者Darrarski