usr.bin/gzip/unbzip2.c

   1 /*      $NetBSD: unbzip2.c,v 1.5 2004/05/25 04:34:40 mrg Exp $  */
   2 /*      $DragonFly: src/usr.bin/gzip/unbzip2.c,v 1.1 2004/10/26 11:19:31 joerg Exp $ */
   3
   4 /* This file is #included by gzip.c */
   5
   6 static off_t
   7 unbzip2(int in, int out, char *pre, size_t prelen, off_t *bytes_in)
   8 {
   9         int             ret, end_of_file;
  10         ssize_t         n;
  11         off_t           bytes_out = 0;
  12         bz_stream       bzs;
  13         static char     *inbuf, *outbuf;
  14
  15         if (inbuf == NULL)
  16                 inbuf = malloc(BUFLEN);
  17         if (outbuf == NULL)
  18                 outbuf = malloc(BUFLEN);
  19         if (inbuf == NULL || outbuf == NULL)
  20                 maybe_err("malloc");
  21
  22         bzs.bzalloc = NULL;
  23         bzs.bzfree = NULL;
  24         bzs.opaque = NULL;
  25
  26         end_of_file = 0;
  27         ret = BZ2_bzDecompressInit(&bzs, 0, 0);
  28         if (ret != BZ_OK)
  29                 maybe_errx("bzip2 init");
  30
  31         /* Prepend. */
  32         bzs.avail_in = prelen;
  33         bzs.next_in = pre;
  34
  35         if (bytes_in)
  36                 *bytes_in = prelen;
  37
  38         while (ret != BZ_STREAM_END) {
  39                 if (bzs.avail_in == 0 && !end_of_file) {
  40                         n = read(in, inbuf, BUFLEN);
  41                         if (n < 0)
  42                                 maybe_err("read");
  43                         if (n == 0)
  44                                 end_of_file = 1;
  45                         bzs.next_in = inbuf;
  46                         bzs.avail_in = n;
  47                         if (bytes_in)
  48                                 *bytes_in += n;
  49                 }
  50
  51                 bzs.next_out = outbuf;
  52                 bzs.avail_out = BUFLEN;
  53                 ret = BZ2_bzDecompress(&bzs);
  54
  55                 switch (ret) {
  56                 case BZ_STREAM_END:
  57                 case BZ_OK:
  58                         if (ret == BZ_OK && end_of_file)
  59                                 maybe_err("read");
  60                         if (!tflag) {
  61                                 n = write(out, outbuf, BUFLEN - bzs.avail_out);
  62                                 if (n < 0)
  63                                         maybe_err("write");
  64                         }
  65                         bytes_out += n;
  66                         break;
  67
  68                 case BZ_DATA_ERROR:
  69                         maybe_warnx("bzip2 data integrity error");
  70                         break;
  71
  72                 case BZ_DATA_ERROR_MAGIC:
  73                         maybe_warnx("bzip2 magic number error");
  74                         break;
  75
  76                 case BZ_MEM_ERROR:
  77                         maybe_warnx("bzip2 out of memory");
  78                         break;
  79
  80                 }
  81         }
  82
  83         if (ret != BZ_STREAM_END || BZ2_bzDecompressEnd(&bzs) != BZ_OK)
  84                 return (-1);
  85
  86         return (bytes_out);
  87 }