zlib inflateReset 导致内存泄漏（不是）

Question

我目前正在满足以下要求。 需求如下：在服务器端，一个大文件被分为 4000 字节的块（帧）。每个块依次被压缩（使用 zlib）并发送到客户端进程。例如，如果一个文件大小为 12000 字节，则它被分为 3 个块。

以上文件将有 3 个块 => Block-0、Block-1、Block-2

收到后，客户端解压缩每个块（或帧）并写入堆上分配的缓冲区。当客户端接收到与整个文件相对应的所有块时，所得文件的未压缩版本将写入磁盘。

我编写了一个例程

inflateData()

，它根据收到的块#执行以下操作：

当收到第一个区块时，

• 充气初始化
• 充气
• 充气重置 当收到中间块时，
• 充气
• 充气重置 当收到最后一个区块时，
• 充气
• 充气结束

通过上述例程，块的解压缩按预期发生。但我面临的问题是它会消耗大量内存，并且在某些时候整个系统会变慢。使用 valgrind 检查时，inflateInit2_ 会报告内存泄漏。这会导致系统资源耗尽。

==30359== 6 个块中的 57,312 字节可能在 67 个丢失记录 64 中丢失
==30359==在0x4A069EE：malloc（vg_replace_malloc.c：270）
==30359== 0x3E57808F1E：inflateInit2_（在/lib64/libz.so.1.2.3中）
==30359== 0x40C220：inflateData（productMaker.c：1668）

以下是

inflateData()

例程。

int inflateData(
    char*    const    inBuf,                  
    unsigned long     inLen,                  
    unsigned int      isFirstBlk,
    unsigned int      isLastBlk,
    const    char*    outBuf,                 
    unsigned long*    outLen)                 
{

  int have;
  int readsz;
  int bsize;
  static z_stream zstrm;
  int zerr;
  int flush;
  char out[CHUNK_SZ];
  char in[CHUNK_SZ];
  int ret,nwrite,idx = -1;
  int savedByteCntr=0;
  unsigned char *dstBuf;
  int  firstCall = 1;
  int totalBytesIn=0;
  int inflatedBytes=0;
  int decompByteCounter = 0;
  int num=0;


  ret = Z_OK;
  readsz = 0;
  bsize = CHUNK_SZ;

  dstBuf = (unsigned char *) outBuf;

  if(isFirstBlk){
        memset(&zstrm, '\0', sizeof(z_stream));
        zstrm.zalloc = Z_NULL;
        zstrm.zfree = Z_NULL;
        zstrm.opaque = Z_NULL;

        if ((zerr = inflateInit(&zstrm)) != Z_OK) {
                uerror("ERROR %d inflateInit (%s)",
                        zerr, decode_zlib_err(zerr));
                return -1;
        }
  }

    while(totalBytesIn < inLen ) {
      int compChunkSize = ((inLen - totalBytesIn) > 5120) ? 5120 :
                                                  (inLen - totalBytesIn);
      memcpy(in, inBuf + totalBytesIn, compChunkSize);

      zstrm.avail_in = inLen - totalBytesIn;
      zstrm.next_in = in ;

      zstrm.avail_out = CHUNK_SZ;
      zstrm.next_out = out;
      inflatedBytes = 0;

      while(ret != Z_STREAM_END) {
         ret  = inflate(&zstrm, Z_NO_FLUSH);
         if(ret < 0) {
          uerror(" Error %d inflate (%s)", ret, decode_zlib_err(ret));
          (void)inflateEnd(&zstrm);
           return ret;
         }
         inflatedBytes = CHUNK_SZ - zstrm.avail_out;

         if(inflatedBytes == 0) {
              unotice("\n Unable to decompress data - truncated");
              break;
         }
     totalBytesIn += zstrm.total_in;
     decompByteCounter += inflatedBytes;

     memcpy(dstBuf + savedByteCntr, out, inflatedBytes);
     savedByteCntr = decompByteCounter;
   }

   // Reset inflater for additional input
   ret = inflateReset(&zstrm);
   if(ret == Z_STREAM_ERROR){
      uerror(" Error %d inflateReset (%s)", ret, decode_zlib_err(ret));
      (void)inflateEnd(&zstrm);
       return ret;
   }
  }

if(isLastBlk){
 ret = inflateEnd(&zstrm);
 if(ret < 0) {
   uerror("Fail inflateEnd %d [%s] ", ret, decode_zlib_err(ret));
   return (ret);
 }
}

  *outLen = decompByteCounter;
   return 0;
}

Answer 1

您在使用

inflateData()

例程时犯了错误。

首先，以这种方式使用静态变量是一个可怕的想法。如果您使用

inflateData()

true 调用

isFirstBlk

两次，而没有使用

isLastBlk

true 进行中间调用，那么您将清除对第一组分配的引用，从而导致内存泄漏。

为了避免此类错误，您应该跟踪

zstrm

是否已初始化，并拒绝任何初始化已初始化流的尝试。更好的是甚至没有

isFirstBlk

，只需在第一次调用以及在

zstrm

true 调用后立即初始化

isLastBlk

即可。

因此，您要么执行上述操作，以

isFirstBlk

true 调用两次，要么以

isLastBlk

true 失败调用。