Как извлечь все данные из архива bzip2 с помощью C?

Question

У меня есть объединенный файл, состоящий из некоторого количества bzip2 архивов.Я также знаю размеры отдельных bzip2 кусков в этом файле.

Я хотел бы распаковать поток bzip2 из отдельного блока данных bzip2 и записать вывод в стандартный вывод.

Сначала я использую fseek, чтобы переместить курсор файла кжелаемый байт архива, а затем считайте «size» -часть файла в вызове 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 возвращаемое число равно undefined .
• Вас не должно пугать, что около 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