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