Home | History | Annotate | Line # | Download | only in gzip 
      1 /*	$NetBSD: unbzip2.c,v 1.14 2017/08/04 07:27:08 mrg Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2006 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Simon Burge.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 /* This file is #included by gzip.c */
     33 
     34 static off_t
     35 unbzip2(int in, int out, char *pre, size_t prelen, off_t *bytes_in)
     36 {
     37 	int		ret, end_of_file, cold = 0;
     38 	off_t		bytes_out = 0;
     39 	bz_stream	bzs;
     40 	static char	*inbuf, *outbuf;
     41 
     42 	if (inbuf == NULL)
     43 		inbuf = malloc(BUFLEN);
     44 	if (outbuf == NULL)
     45 		outbuf = malloc(BUFLEN);
     46 	if (inbuf == NULL || outbuf == NULL)
     47 	        maybe_err("malloc");
     48 
     49 	bzs.bzalloc = NULL;
     50 	bzs.bzfree = NULL;
     51 	bzs.opaque = NULL;
     52 
     53 	end_of_file = 0;
     54 	ret = BZ2_bzDecompressInit(&bzs, 0, 0);
     55 	if (ret != BZ_OK)
     56 	        maybe_errx("bzip2 init");
     57 
     58 	/* Prepend. */
     59 	bzs.avail_in = prelen;
     60 	bzs.next_in = pre;
     61 
     62 	if (bytes_in)
     63 		*bytes_in = prelen;
     64 
     65 	while (ret == BZ_OK) {
     66 		check_siginfo();
     67 	        if (bzs.avail_in == 0 && !end_of_file) {
     68 			ssize_t	n;
     69 
     70 	                n = read(in, inbuf, BUFLEN);
     71 	                if (n < 0)
     72 	                        maybe_err("read");
     73 	                if (n == 0)
     74 	                        end_of_file = 1;
     75 			infile_newdata(n);
     76 	                bzs.next_in = inbuf;
     77 	                bzs.avail_in = n;
     78 			if (bytes_in)
     79 				*bytes_in += n;
     80 	        }
     81 
     82 	        bzs.next_out = outbuf;
     83 	        bzs.avail_out = BUFLEN;
     84 	        ret = BZ2_bzDecompress(&bzs);
     85 
     86 	        switch (ret) {
     87 	        case BZ_STREAM_END:
     88 	        case BZ_OK:
     89 	                if (ret == BZ_OK && end_of_file) {
     90 				/*
     91 				 * If we hit this after a stream end, consider
     92 				 * it as the end of the whole file and don't
     93 				 * bail out.
     94 				 */
     95 				if (cold == 1)
     96 					ret = BZ_STREAM_END;
     97 				else
     98 					maybe_errx("truncated file");
     99 			}
    100 			cold = 0;
    101 	                if (!tflag && bzs.avail_out != BUFLEN) {
    102 				ssize_t	n;
    103 
    104 	                        n = write(out, outbuf, BUFLEN - bzs.avail_out);
    105 	                        if (n < 0)
    106 	                                maybe_err("write");
    107 	                	bytes_out += n;
    108 	                }
    109 			if (ret == BZ_STREAM_END && !end_of_file) {
    110 				if (BZ2_bzDecompressEnd(&bzs) != BZ_OK ||
    111 				    BZ2_bzDecompressInit(&bzs, 0, 0) != BZ_OK)
    112 					maybe_errx("bzip2 re-init");
    113 				cold = 1;
    114 				ret = BZ_OK;
    115 			}
    116 			break;
    117 
    118 	        case BZ_DATA_ERROR:
    119 	                maybe_warnx("bzip2 data integrity error");
    120 			break;
    121 
    122 	        case BZ_DATA_ERROR_MAGIC:
    123 	                maybe_warnx("bzip2 magic number error");
    124 			break;
    125 
    126 	        case BZ_MEM_ERROR:
    127 	                maybe_warnx("bzip2 out of memory");
    128 			break;
    129 
    130 		default:
    131 			maybe_warnx("unknown bzip2 error: %d", ret);
    132 			break;
    133 	        }
    134 	}
    135 
    136 	if (ret != BZ_STREAM_END || BZ2_bzDecompressEnd(&bzs) != BZ_OK)
    137 	        return (-1);
    138 
    139 	return (bytes_out);
    140 }
    141