zlib中调用inflatePrime()是否需先设置stream.next_in压缩缓冲？

问题本质 #

你遇到的invalid code lengths set错误，源于重启解压时比特偏移处理逻辑错误，以及对inflatePrime()工作机制的误解——该函数完全不依赖stream.next_in，它直接操作zlib内部的比特缓冲区。

错误点拆解 #

1. inflatePrime使用逻辑错误

   • 读取aByte后未回退文件指针，导致后续ZREAD64跳过了该字节的剩余比特部分
   • 比特位提取计算错误，aByte >> (8 - streamBits)没有保留正确的高位比特

2. 状态保存的比特偏移提取错误
   stream.data_type的低7位表示未处理的比特数，你用stream.data_type & (~0x1C0)提取是错误的，正确的提取方式是stream.data_type & 0x7F。

修正后的完整流程 #

第一次启动：正确保存解压状态 #

if ((stream.data_type & 0x0080) && some_16MB_wrote_to_output) {
    inflateGetDictionary(&stream, dictionary_buf, &dictLength);
    // 提取未处理的比特数（data_type低7位）
    streamBits = stream.data_type & 0x7F;
    // 保存未读取的压缩数据起始位置
    streamCurrentPos = ZTELL64(z_filefunc, filestream) - stream.avail_in;
}

第二次启动：正确恢复解压状态 #

// 定位到保存的压缩数据起始位置
ZSEEK64(z_filefunc, filestream, streamCurrentPos, ZLIB_FILEFUNC_SEEK_SET);

if (streamBits > 0) {
    int aByte = 0;
    // 读取包含剩余比特的字节
    unz64local_getByte(z_filefunc, filestream, &aByte);
    // 向zlib内部插入指定数量的高位比特
    inflatePrime(&stream, streamBits, aByte);
    // 回退文件指针，确保后续读取包含该字节的完整数据
    ZSEEK64(z_filefunc, filestream, -1, ZLIB_FILEFUNC_SEEK_CUR);
}

// 恢复字典
inflateSetDictionary(&stream, dictionary_buf, dictLength);

// 读取压缩数据到缓冲区
ZREAD64(z_filefunc, filestream, read_buffer, someSize);
stream.next_in = read_buffer;
stream.avail_in = someSize;

// 重启解压
int ret = inflate(&stream, Z_BLOCK);

关键说明 #

• inflatePrime的工作机制：该函数直接将传入字节的前N个高位比特（N为第二个参数）插入到zlib的内部比特流中，无需依赖stream.next_in的缓冲。
• 比特偏移恢复：保存的streamBits是上次解压剩余的未处理比特数，这些比特位于当前压缩字节的高位，因此读取该字节插入比特后，必须回退文件指针，避免后续读取跳过该字节。
• 字典恢复顺序：必须在插入比特后、设置next_in前调用inflateSetDictionary，确保zlib重启时使用正确的历史字典。

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