Python3 zlib.uncompress能否单独防范解压炸弹？需捕获MemoryError吗？

好问题！只靠捕获zlib.error绝对不足以防范解压炸弹，咱们来好好掰扯清楚：

为什么zlib.error防不住？ #

zlib.error只会在压缩数据格式非法、校验失败（比如CRC错误）这类场景下抛出，但解压炸弹本质是格式完全合法的压缩文件——它用极高的压缩比（比如把几GB的全零数据压缩成几KB），让解压过程疯狂占用内存。这种情况下zlib会正常工作，根本不会触发zlib.error，直到系统内存被耗尽。

那要不要捕获MemoryError？ #

可以捕获，但不能把它当作主要的防御手段：

• 当MemoryError抛出时，程序的运行状态已经很危险了——系统内存可能已经被榨干，甚至连处理异常的资源都不够，严重的话会导致整个进程崩溃，甚至拖垮系统（比如开启了swap的情况下，磁盘会被疯狂读写，系统卡顿到无响应）。
• 有些情况下，系统可能不会立刻抛出MemoryError，而是靠swap硬撑，这时候你的程序可能已经失去响应，根本没机会处理异常。

正确的防范姿势：主动限制解压大小 #

最可靠的方法是分块解压并监控已解压数据的体积，一旦超过你设定的安全阈值就主动终止操作。推荐使用zlib.decompressobj()来实现，而不是直接用zlib.uncompress()（后者会一次性把所有数据解压到内存里）。

举个简单的实现例子：

import zlib

def safe_zlib_uncompress(compressed_data, max_uncompressed_size=10*1024*1024):
    """安全解压zlib数据，限制最大解压后大小为10MB"""
    decompressor = zlib.decompressobj()
    uncompressed = b""
    chunk_size = 4096  # 分块处理的大小
    
    for i in range(0, len(compressed_data), chunk_size):
        chunk = compressed_data[i:i+chunk_size]
        uncompressed_chunk = decompressor.decompress(chunk)
        # 检查累计大小是否超过阈值
        if len(uncompressed) + len(uncompressed_chunk) > max_uncompressed_size:
            raise ValueError("解压后数据超过最大允许大小，疑似解压炸弹")
        uncompressed += uncompressed_chunk
    
    # 处理剩余的压缩数据
    uncompressed += decompressor.flush()
    if len(uncompressed) > max_uncompressed_size:
        raise ValueError("解压后数据超过最大允许大小，疑似解压炸弹")
    
    return uncompressed

总结 #

• 只捕获zlib.error完全无法应对解压炸弹，因为这类炸弹是合法的压缩数据；
• MemoryError可以作为最后一道兜底防线，但依赖它风险极高；
• 最有效的方式是主动限制解压后的最大数据体积，用分块解压的方式实时监控，从根源上避免内存被耗尽。

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