Home | History | Annotate | Line # | Download | only in gzip
unbzip2.c revision 1.8 
      1 /*	$NetBSD: unbzip2.c,v 1.8 2005/09/15 09:11:30 mrg Exp $	*/
      2 
      3 /* This file is #included by gzip.c */
      4 
      5 static off_t
      6 unbzip2(int in, int out, char *pre, size_t prelen, off_t *bytes_in)
      7 {
      8 	int		ret, end_of_file;
      9 	off_t		bytes_out = 0;
     10 	bz_stream	bzs;
     11 	static char	*inbuf, *outbuf;
     12 
     13 	if (inbuf == NULL)
     14 		inbuf = malloc(BUFLEN);
     15 	if (outbuf == NULL)
     16 		outbuf = malloc(BUFLEN);
     17 	if (inbuf == NULL || outbuf == NULL)
     18 	        maybe_err("malloc");
     19 
     20 	bzs.bzalloc = NULL;
     21 	bzs.bzfree = NULL;
     22 	bzs.opaque = NULL;
     23 
     24 	end_of_file = 0;
     25 	ret = BZ2_bzDecompressInit(&bzs, 0, 0);
     26 	if (ret != BZ_OK)
     27 	        maybe_errx("bzip2 init");
     28 
     29 	/* Prepend. */
     30 	bzs.avail_in = prelen;
     31 	bzs.next_in = pre;
     32 
     33 	if (bytes_in)
     34 		*bytes_in = prelen;
     35 
     36 	while (ret != BZ_STREAM_END) {
     37 	        if (bzs.avail_in == 0 && !end_of_file) {
     38 			ssize_t	n;
     39 
     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 				ssize_t	n;
     62 
     63 	                        n = write(out, outbuf, BUFLEN - bzs.avail_out);
     64 	                        if (n < 0)
     65 	                                maybe_err("write");
     66 	                	bytes_out += n;
     67 	                }
     68 	                break;
     69 
     70 	        case BZ_DATA_ERROR:
     71 	                maybe_warnx("bzip2 data integrity error");
     72 			break;
     73 
     74 	        case BZ_DATA_ERROR_MAGIC:
     75 	                maybe_warnx("bzip2 magic number error");
     76 			break;
     77 
     78 	        case BZ_MEM_ERROR:
     79 	                maybe_warnx("bzip2 out of memory");
     80 			break;
     81 
     82 	        }
     83 	}
     84 
     85 	if (ret != BZ_STREAM_END || BZ2_bzDecompressEnd(&bzs) != BZ_OK)
     86 	        return (-1);
     87 
     88 	return (bytes_out);
     89 }
     90