2016-09-18 136 views
1

我正在嘗試使用zlib進行解壓縮。我在zlib網站上查看教程,並通過代碼生成不同大小的輸出。zlib不同的解壓縮大小

int CZLib::Inflate() { 
    int ret; 
    unsigned int have; 
    z_stream zstream; 
    unsigned char in[CHUNK]; 
    unsigned char out[CHUNK]; 

    zstream.zalloc = Z_NULL; 
    zstream.zfree = Z_NULL; 
    zstream.opaque = Z_NULL; 
    zstream.avail_in = 0; 
    zstream.next_in = Z_NULL; 
    ret = inflateInit(&zstream); 
    if (ret != Z_OK) 
     return ret; 

    do { 
     zstream.avail_in = fread(in, 1, CHUNK, fin); 
     if (ferror(fin)) { 
      (void)inflateEnd(&zstream); 
      return Z_ERRNO; 
     } 
     if (zstream.avail_in == 0) break; 
     zstream.next_in = in; 

     do { 
      zstream.avail_out = CHUNK; 
      zstream.next_out = out; 
      ret = inflate(&zstream, Z_NO_FLUSH); 
      assert(ret != Z_STREAM_ERROR); 
      switch (ret) { 
       case Z_NEED_DICT: 
        ret = Z_DATA_ERROR; 
       case Z_DATA_ERROR: 
       case Z_MEM_ERROR: 
        (void)inflateEnd(&zstream); 
        return ret; 
      } 
      have = CHUNK - zstream.avail_out; 
      if (fwrite(out, 1, have, fout) != have || ferror(fout)) { 
       (void)inflateEnd(&zstream); 
       return Z_ERRNO; 
      } 

     } while (zstream.avail_out == 0); 
    } while (ret != Z_STREAM_END); 

    (void)inflateEnd(&zstream); 
    return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR; 
} 

int CZLib::Inflate(const std::string& src) { 
    std::vector<char> output; 

    z_stream zstream; 
    zstream.zalloc = Z_NULL; 
    zstream.zfree = Z_NULL; 
    zstream.opaque = Z_NULL; 
    zstream.avail_in = 0; 
    zstream.next_in = Z_NULL; 

    int ret = inflateInit(&zstream); 
    if (ret != Z_OK) 
     return ret; 

    unsigned char in[CHUNK]; 
    unsigned char out[CHUNK]; 

    int have = 0, nByte = CHUNK, off = 0, remaining = src.size(); 

    if (src.size() < CHUNK) nByte = src.size(); 
    do { 
     memcpy(in, &src[off], nByte); 
     off += nByte; 
     remaining -= nByte; 

     if (nByte > 0) zstream.avail_in = nByte; 

     if (remaining > CHUNK) { nByte = CHUNK; } 
     else { nByte = remaining; } 

     if (zstream.avail_in == 0) break; 
     zstream.next_in = in; 

     do { 
      zstream.avail_out = CHUNK; 
      zstream.next_out = out; 

      ret = inflate(&zstream, Z_NO_FLUSH); 
      have = CHUNK - zstream.avail_out; 
      output.insert(output.end(), out, out + have); 

     } while (zstream.avail_out == 0); 
    } while (ret != Z_STREAM_END); 

    CFile* file = new CFile("in.out", "wb"); 
    file->Write<char>(&output[0], output.size()); 
    delete file; 
    return ret; 
} 

它使用相同的數據。其中一個讀取磁盤上的文件,另一個使用內存(緩衝區方法)。 CHUNK大小16384.第一個代碼產生524288(0x80000)和其他524800(0x80200)字節。差異是512字節。爲什麼會發生?

回答

0

在第一個代碼示例,你這行

zstream.avail_in = fread(in, 1, CHUNK, fin); 

,然後你必須

if (zstream.avail_in == 0) break; 

停止循環。

在第二個代碼示例,您必須在同一行停止循環,但你也有這樣一行:

if (nByte > 0) zstream.avail_in = nByte; 
     ^^^^^^^^^ 
     So you only assign to zstream.avail_in when nByte > 0 

    .... 
    .... 

    if (zstream.avail_in == 0) break; 
     ^^^^^^^^^^^^^^^^ 
     Consequently this will not be true when nByte is zero and the 
     code will not exit 

試試這個:

zstream.avail_in = nByte; // Unconditional assignment 

    .... 

    if (zstream.avail_in <= 0) break; // Less or equal to zero 
+0

不,它並沒有解決問題。我並排檢查了兩個版本。所有變量值和循環次數都相同。我認爲問題是流或什麼..有趣的是,當我在函數中創建文件時,同樣的問題仍然存在。這太瘋狂了。 – streamer

+0

嗯 - 第二個例子中的邏輯是錯誤的,但是由於'inflate'的返回值,你可能會跳出外部循環。我不知道它是否重要,但我注意到你的第一個例子叫'inflateEnd',但你的第二個例子不是。 – 4386427

+0

你說得對第二個例子,謝謝。我解決了這個問題。該問題源於未能關閉文件流。我正在使用C風格的文件系統。當文件關閉正確的問題解決。當退出函數並調用析構函數時,文件寫入不完整(512字節) – streamer