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