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
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	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