修复Minilibx中加载特定PNG时出现的Zlib Inflate Error (-5)(Z_BUF_ERROR)问题

看起来你遇到了个挺棘手的特定PNG兼容问题——大部分图都正常加载，唯独这张触发了Zlib的Z_BUF_ERROR，而且已经排查了常见的点，我来帮你分析下核心问题和修复方向：

核心问题分析 #

从你的代码和日志来看，问题出在Zlib返回值的错误判断逻辑和IDAT块处理的收尾逻辑上，这两个点刚好在这张特殊PNG的场景下触发了错误：

1. Z_BUF_ERROR被误判为致命错误 #

Zlib的文档明确说明：Z_BUF_ERROR不一定是错误，它只是表示「当前没有输入数据，或者输出缓冲区已满」，属于正常的流程信号。但你的代码里直接把它归为致命错误：

if (z_ret != Z_OK && z_ret != Z_STREAM_END) {
    inflateEnd(&z_strm);
    free(buffer);
    return (ERR_ZLIB);
}

当最后一个IDAT块的剩余数据喂给Zlib后，Zlib可能还需要更多输入（或者内部还有未输出的数据），这时候调用inflate会返回Z_BUF_ERROR，但你的代码直接抛出错误退出了。

2. 未处理Zlib内部缓冲区的剩余数据 #

当所有IDAT块都喂完后，Zlib的内部缓冲区可能还残留着未输出的解压数据，你的代码直接跳过了这部分，导致最终的b_pos和预期长度不匹配，或者在收尾时触发错误。

具体修复步骤 #

步骤1：修正Zlib返回值的错误判断 #

允许Z_BUF_ERROR作为非致命情况，只有真正的错误（如数据损坏、内存问题）才触发退出：

z_ret = inflate(&z_strm, Z_NO_FLUSH);
// 只在致命错误时退出
if (z_ret != Z_OK && z_ret != Z_STREAM_END && z_ret != Z_BUF_ERROR) {
    inflateEnd(&z_strm);
    free(buffer);
    return (ERR_ZLIB);
}
// 处理Z_BUF_ERROR的正常情况：如果没有输入数据，退出当前IDAT块的内部循环
if (z_ret == Z_BUF_ERROR && z_strm.avail_in == 0) {
    break;
}

步骤2：添加IDAT块处理后的收尾解压 #

所有IDAT块遍历完成后，继续调用inflate处理Zlib内部的剩余数据，直到返回Z_STREAM_END：

// 原代码的外部IDAT循环结束后，添加这段
z_strm.avail_in = 0; // 明确告知Zlib没有更多输入数据
z_ret = Z_OK;
while (z_ret != Z_STREAM_END) {
    z_strm.avail_out = Z_CHUNK;
    z_strm.next_out = z_out;
    // 用Z_FINISH触发剩余数据的解压
    z_ret = inflate(&z_strm, Z_FINISH);
    if (z_ret != Z_OK && z_ret != Z_STREAM_END) {
        inflateEnd(&z_strm);
        free(buffer);
        return (ERR_ZLIB);
    }
    // 复制剩余解压数据到buffer
    size_t output_len = Z_CHUNK - z_strm.avail_out;
    if (b_pos + output_len > (size_t)pi->width * pi->height * pi->bpp + pi->height) {
        inflateEnd(&z_strm);
        free(buffer);
        return (ERR_DATA_MISMATCH);
    }
    memcpy(buffer + b_pos, z_out, output_len);
    b_pos += output_len;
}

步骤3：优化IDAT块的内部循环逻辑 #

避免在没有输入数据时重复调用inflate，修改内部循环的条件，确保只在有输入数据或输出空间时继续：

// 替换原有的内部while循环
do {
    z_strm.avail_out = Z_CHUNK;
    z_strm.next_out = z_out;
    printf("len=%u avail_in=%u avail_out=%u\n", len, z_strm.avail_in, Z_CHUNK);
    if (z_strm.avail_in == 0) {
        printf("Chunk @ %p | len=%u | type=%.4s\n", dat, len, dat + 4);
        break;
    }
    z_ret = inflate(&z_strm, Z_NO_FLUSH);
    // 错误判断（用步骤1里的修正版）
    if (z_ret != Z_OK && z_ret != Z_STREAM_END && z_ret != Z_BUF_ERROR) {
        inflateEnd(&z_strm);
        free(buffer);
        return (ERR_ZLIB);
    }
    // 复制输出数据
    size_t output_len = Z_CHUNK - z_strm.avail_out;
    if (b_pos + output_len > (size_t)pi->width * pi->height * pi->bpp + pi->height) {
        inflateEnd(&z_strm);
        free(buffer);
        return (ERR_DATA_MISMATCH);
    }
    memcpy(buffer + b_pos, z_out, output_len);
    b_pos += output_len;
// 只有当输出缓冲区被填满（需要继续解压）时，才循环
} while (z_strm.avail_out == 0 && z_ret == Z_OK);

额外验证点 #

1. 确认mipng_is_type函数能正确识别连续的IDAT块，不会漏读或误判
2. 检查ntohl的使用是否正确（确保IDAT块的长度是按网络字节序解析的）
3. 确认pi->bpp、pi->width、pi->height这些参数的解析是正确的（比如PNG的位深度、颜色类型转换是否正确）

这些修改应该能解决你遇到的Z_BUF_ERROR问题，因为它修正了Zlib处理的核心逻辑，兼容了那些需要分多次收尾解压的PNG文件。

内容来源于stack exchange