如何使用C从bzip2存档中提取所有数据？

Question

我有一个由一些bzip2档案组成的连接文件.我也知道该bzip2文件中各个块的大小.

我想bzip2从单个bzip2数据块解压缩流,并将输出写入标准输出.

首先,我使用fseek将文件光标移动到所需的存档字节,然后将文件的"size"-chunk读入BZ2_bzRead调用:

int headerSize = 1234;
int firstChunkSize = 123456;
FILE *fp = fopen("pathToConcatenatedFile", "r+b");
char *bzBuf = malloc(sizeof(char) * firstChunkSize);
int bzError, bzNBuf;
BZFILE *bzFp = BZ2_bzReadOpen(&bzError, *fp, 0, 0, NULL, 0);

# move cursor past header of known size, to the first bzip2 "chunk"
fseek(*fp, headerSize, SEEK_SET); 

while (bzError != BZ_STREAM_END) {
    # read the first chunk of known size, decompress it
    bzNBuf = BZ2_bzRead(&bzError, bzFp, bzBuf, firstChunkSize);
    fprintf(stdout, bzBuf);
}

BZ2_bzReadClose(&bzError, bzFp);
free(bzBuf);
fclose(fp);

问题是,当我将fprintf语句的输出与bzip2在命令行上运行的输出进行比较时,我得到两个不同的答案.

具体来说,我从此代码获得的输出少于bzip2在命令行上运行的输出.

更具体地说,此代码的输出是命令行进程输出的较小子集,我错过了bzip2感兴趣块的尾端.

我已经通过另一种技术验证了命令行bzip2提供了正确的答案,因此,我的C代码的一些问题导致在块的末尾输出丢失.我只是不知道那是什么问题.

如果您熟悉bzip2或者libbzip2,您能否就上述代码示例中的错误提供任何建议？感谢您的意见.

Answer 1

这是我的源代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <bzlib.h>

int
bunzip_one(FILE *f) {
  int bzError;
  BZFILE *bzf;
  char buf[4096];

  bzf = BZ2_bzReadOpen(&bzError, f, 0, 0, NULL, 0);
  if (bzError != BZ_OK) {
    fprintf(stderr, "E: BZ2_bzReadOpen: %d\n", bzError);
    return -1;
  }

  while (bzError == BZ_OK) {
    int nread = BZ2_bzRead(&bzError, bzf, buf, sizeof buf);
    if (bzError == BZ_OK || bzError == BZ_STREAM_END) {
      size_t nwritten = fwrite(buf, 1, nread, stdout);
      if (nwritten != (size_t) nread) {
        fprintf(stderr, "E: short write\n");
        return -1;
      }
    }
  }

  if (bzError != BZ_STREAM_END) {
    fprintf(stderr, "E: bzip error after read: %d\n", bzError);
    return -1;
  }

  BZ2_bzReadClose(&bzError, bzf);
  return 0;
}

int
bunzip_many(const char *fname) {
  FILE *f;

  f = fopen(fname, "rb");
  if (f == NULL) {
    perror(fname);
    return -1;
  }

  fseek(f, 0, SEEK_SET);
  if (bunzip_one(f) == -1)
    return -1;

  fseek(f, 42, SEEK_SET); /* hello.bz2 is 42 bytes long in my case */
  if (bunzip_one(f) == -1)
    return -1;

  fclose(f);
  return 0;
}

int
main(int argc, char **argv) {
  if (argc < 2) {
    fprintf(stderr, "usage: bunz <fname>\n");
    return EXIT_FAILURE;
  }
  return bunzip_many(argv[1]) != 0 ? EXIT_FAILURE : EXIT_SUCCESS;
}

• 我非常关心正确的错误检查.例如,我确定在尝试访问缓冲区bzError时BZ_OK或BZ_STREAM_END之前.文档明确指出,对于bzError返回数字的其他值未定义.
• 大约50%的代码与错误处理有关,不应该吓到你.应该是这样的.期待到处都是错误.
• 代码仍有一些错误.如果出现错误,则不会正确释放资源(f,bzf).

这些是我用于测试的命令:

$ echo hello > hello
$ echo world > world
$ bzip2 hello
$ bzip2 world
$ cat hello.bz2 world.bz2 > helloworld.bz2
$ gcc -W -Wall -Os -o bunz bunz.c -lbz2
$ ls -l *.bz2
-rw-r--r-- 1 roland None 42 Oct 12 09:26 hello.bz2
-rw-r--r-- 1 roland None 86 Oct 12 09:36 helloworld.bz2
-rw-r--r-- 1 roland None 44 Oct 12 09:26 world.bz2
$ ./bunz.exe helloworld.bz2 
hello
world