busybox/archival/bbunzip.c

/* vi: set sw=4 ts=4: */
/*
 *  Common code for gunzip-like applets
 *
 *  Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

#include "libbb.h"
#include "unarchive.h"

enum {
	OPT_STDOUT = 0x1,
	OPT_FORCE = 0x2,
/* gunzip and bunzip2 only: */
	OPT_VERBOSE = 0x4,
	OPT_DECOMPRESS = 0x8,
	OPT_TEST = 0x10,
};

static
int open_to_or_warn(int to_fd, const char *filename, int flags, int mode)
{
	int fd = open3_or_warn(filename, flags, mode);
	if (fd < 0) {
		return 1;
	}
	xmove_fd(fd, to_fd);
	return 0;
}

int FAST_FUNC bbunpack(char **argv,
	char* (*make_new_name)(char *filename),
	IF_DESKTOP(long long) int (*unpacker)(unpack_info_t *info)
)
{
	struct stat stat_buf;
	IF_DESKTOP(long long) int status;
	char *filename, *new_name;
	smallint exitcode = 0;
	unpack_info_t info;

	do {
		/* NB: new_name is *maybe* malloc'ed! */
		new_name = NULL;
		filename = *argv; /* can be NULL - 'streaming' bunzip2 */

		if (filename && LONE_DASH(filename))
			filename = NULL;

		/* Open src */
		if (filename) {
			if (stat(filename, &stat_buf) != 0) {
				bb_simple_perror_msg(filename);
 err:
				exitcode = 1;
				goto free_name;
			}
			if (open_to_or_warn(STDIN_FILENO, filename, O_RDONLY, 0))
				goto err;
		}

		/* Special cases: test, stdout */
		if (option_mask32 & (OPT_STDOUT|OPT_TEST)) {
			if (option_mask32 & OPT_TEST)
				if (open_to_or_warn(STDOUT_FILENO, bb_dev_null, O_WRONLY, 0))
					goto err;
			filename = NULL;
		}

		/* Open dst if we are going to unpack to file */
		if (filename) {
			new_name = make_new_name(filename);
			if (!new_name) {
				bb_error_msg("%s: unknown suffix - ignored", filename);
				goto err;
			}

			/* -f: overwrite existing output files */
			if (option_mask32 & OPT_FORCE) {
				unlink(new_name);
			}

			/* O_EXCL: "real" bunzip2 doesn't overwrite files */
			/* GNU gunzip does not bail out, but goes to next file */
			if (open_to_or_warn(STDOUT_FILENO, new_name, O_WRONLY | O_CREAT | O_EXCL,
					stat_buf.st_mode))
				goto err;
		}

		/* Check that the input is sane */
		if (isatty(STDIN_FILENO) && (option_mask32 & OPT_FORCE) == 0) {
			bb_error_msg_and_die("compressed data not read from terminal, "
					"use -f to force it");
		}

		/* memset(&info, 0, sizeof(info)); */
		info.mtime = 0; /* so far it has one member only */
		status = unpacker(&info);
		if (status < 0)
			exitcode = 1;
		xclose(STDOUT_FILENO); /* with error check! */

		if (filename) {
			char *del = new_name;
			if (status >= 0) {
				/* TODO: restore other things? */
				if (info.mtime) {
					struct timeval times[2];

					times[1].tv_sec = times[0].tv_sec = info.mtime;
					times[1].tv_usec = times[0].tv_usec = 0;
					/* Note: we closed it first.
					 * On some systems calling utimes
					 * then closing resets the mtime
					 * back to current time. */
					utimes(new_name, times); /* ignoring errors */
				}

				/* Delete _compressed_ file */
				del = filename;
				/* restore extension (unless tgz -> tar case) */
				if (new_name == filename)
					filename[strlen(filename)] = '.';
			}
			xunlink(del);

#if 0 /* Currently buggy - wrong name: "a.gz: 261% - replaced with a.gz" */
			/* Extreme bloat for gunzip compat */
			if (ENABLE_DESKTOP && (option_mask32 & OPT_VERBOSE) && status >= 0) {
				fprintf(stderr, "%s: %u%% - replaced with %s\n",
					filename, (unsigned)(stat_buf.st_size*100 / (status+1)), new_name);
			}
#endif

 free_name:
			if (new_name != filename)
				free(new_name);
		}
	} while (*argv && *++argv);

	return exitcode;
}

#if ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNCOMPRESS

static
char* make_new_name_generic(char *filename, const char *expected_ext)
{
	char *extension = strrchr(filename, '.');
	if (!extension || strcmp(extension + 1, expected_ext) != 0) {
		/* Mimic GNU gunzip - "real" bunzip2 tries to */
		/* unpack file anyway, to file.out */
		return NULL;
	}
	*extension = '\0';
	return filename;
}

#endif


/*
 *  Modified for busybox by Glenn McGrath
 *  Added support output to stdout by Thomas Lundquist <thomasez@zelow.no>
 *
 *  Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

#if ENABLE_BUNZIP2

static
char* make_new_name_bunzip2(char *filename)
{
	return make_new_name_generic(filename, "bz2");
}

static
IF_DESKTOP(long long) int unpack_bunzip2(unpack_info_t *info UNUSED_PARAM)
{
	return unpack_bz2_stream_prime(STDIN_FILENO, STDOUT_FILENO);
}

int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int bunzip2_main(int argc UNUSED_PARAM, char **argv)
{
	getopt32(argv, "cfvdt");
	argv += optind;
	if (applet_name[2] == 'c')
		option_mask32 |= OPT_STDOUT;

	return bbunpack(argv, make_new_name_bunzip2, unpack_bunzip2);
}

#endif


/*
 * Gzip implementation for busybox
 *
 * Based on GNU gzip v1.2.4 Copyright (C) 1992-1993 Jean-loup Gailly.
 *
 * Originally adjusted for busybox by Sven Rudolph <sr1@inf.tu-dresden.de>
 * based on gzip sources
 *
 * Adjusted further by Erik Andersen <andersen@codepoet.org> to support files as
 * well as stdin/stdout, and to generally behave itself wrt command line
 * handling.
 *
 * General cleanup to better adhere to the style guide and make use of standard
 * busybox functions by Glenn McGrath
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 *
 * gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
 * Copyright (C) 1992-1993 Jean-loup Gailly
 * The unzip code was written and put in the public domain by Mark Adler.
 * Portions of the lzw code are derived from the public domain 'compress'
 * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies,
 * Ken Turkowski, Dave Mack and Peter Jannesen.
 *
 * See the license_msg below and the file COPYING for the software license.
 * See the file algorithm.doc for the compression algorithms and file formats.
 */

#if ENABLE_GUNZIP

static
char* make_new_name_gunzip(char *filename)
{
	char *extension = strrchr(filename, '.');

	if (!extension)
		return NULL;

	extension++;
	if (strcmp(extension, "tgz" + 1) == 0
#if ENABLE_FEATURE_SEAMLESS_Z
	 || (extension[0] == 'Z' && extension[1] == '\0')
#endif
	) {
		extension[-1] = '\0';
	} else if (strcmp(extension, "tgz") == 0) {
		filename = xstrdup(filename);
		extension = strrchr(filename, '.');
		extension[2] = 'a';
		extension[3] = 'r';
	} else {
		return NULL;
	}
	return filename;
}

static
IF_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info)
{
	IF_DESKTOP(long long) int status = -1;

	/* do the decompression, and cleanup */
	if (xread_char(STDIN_FILENO) == 0x1f) {
		unsigned char magic2;

		magic2 = xread_char(STDIN_FILENO);
		if (ENABLE_FEATURE_SEAMLESS_Z && magic2 == 0x9d) {
			status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
		} else if (magic2 == 0x8b) {
			status = unpack_gz_stream_with_info(STDIN_FILENO, STDOUT_FILENO, info);
		} else {
			goto bad_magic;
		}
		if (status < 0) {
			bb_error_msg("error inflating");
		}
	} else {
 bad_magic:
		bb_error_msg("invalid magic");
		/* status is still == -1 */
	}
	return status;
}

/*
 * Linux kernel build uses gzip -d -n. We accept and ignore it.
 * Man page says:
 * -n --no-name
 * gzip: do not save the original file name and time stamp.
 * (The original name is always saved if the name had to be truncated.)
 * gunzip: do not restore the original file name/time even if present
 * (remove only the gzip suffix from the compressed file name).
 * This option is the default when decompressing.
 * -N --name
 * gzip: always save the original file name and time stamp (this is the default)
 * gunzip: restore the original file name and time stamp if present.
 */

int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int gunzip_main(int argc UNUSED_PARAM, char **argv)
{
	getopt32(argv, "cfvdtn");
	argv += optind;
	/* if called as zcat */
	if (applet_name[1] == 'c')
		option_mask32 |= OPT_STDOUT;

	return bbunpack(argv, make_new_name_gunzip, unpack_gunzip);
}

#endif


/*
 * Small lzma deflate implementation.
 * Copyright (C) 2006  Aurelien Jacobs <aurel@gnuage.org>
 *
 * Based on bunzip.c from busybox
 *
 * Licensed under GPL v2, see file LICENSE in this tarball for details.
 */

#if ENABLE_UNLZMA

static
char* make_new_name_unlzma(char *filename)
{
	return make_new_name_generic(filename, "lzma");
}

static
IF_DESKTOP(long long) int unpack_unlzma(unpack_info_t *info UNUSED_PARAM)
{
	return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
}

int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int unlzma_main(int argc UNUSED_PARAM, char **argv)
{
	getopt32(argv, "cf");
	argv += optind;
	/* lzmacat? */
	if (applet_name[4] == 'c')
		option_mask32 |= OPT_STDOUT;

	return bbunpack(argv, make_new_name_unlzma, unpack_unlzma);
}

#endif


/*
 *	Uncompress applet for busybox (c) 2002 Glenn McGrath
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

#if ENABLE_UNCOMPRESS

static
char* make_new_name_uncompress(char *filename)
{
	return make_new_name_generic(filename, "Z");
}

static
IF_DESKTOP(long long) int unpack_uncompress(unpack_info_t *info UNUSED_PARAM)
{
	IF_DESKTOP(long long) int status = -1;

	if ((xread_char(STDIN_FILENO) != 0x1f) || (xread_char(STDIN_FILENO) != 0x9d)) {
		bb_error_msg("invalid magic");
	} else {
		status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
	}
	return status;
}

int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int uncompress_main(int argc UNUSED_PARAM, char **argv)
{
	getopt32(argv, "cf");
	argv += optind;

	return bbunpack(argv, make_new_name_uncompress, unpack_uncompress);
}

#endif
1	/* vi: set sw=4 ts=4: */
2	/*
3	* Common code for gunzip-like applets
4	*
5	* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
6	*/
7
8	#include "libbb.h"
9	#include "unarchive.h"
10
11	enum {
12	OPT_STDOUT = 0x1,
13	OPT_FORCE = 0x2,
14	/* gunzip and bunzip2 only: */
15	OPT_VERBOSE = 0x4,
16	OPT_DECOMPRESS = 0x8,
17	OPT_TEST = 0x10,
18	};
19
20	static
21	int open_to_or_warn(int to_fd, const char *filename, int flags, int mode)
22	{
23	int fd = open3_or_warn(filename, flags, mode);
24	if (fd < 0) {
25	return 1;
26	}
27	xmove_fd(fd, to_fd);
28	return 0;
29	}
30
31	int FAST_FUNC bbunpack(char **argv,
32	char* (make_new_name)(char filename),
33	IF_DESKTOP(long long) int (unpacker)(unpack_info_t info)
34	)
35	{
36	struct stat stat_buf;
37	IF_DESKTOP(long long) int status;
38	char filename, new_name;
39	smallint exitcode = 0;
40	unpack_info_t info;
41
42	do {
43	/* NB: new_name is maybe malloc'ed! */
44	new_name = NULL;
45	filename = argv; / can be NULL - 'streaming' bunzip2 */
46
47	if (filename && LONE_DASH(filename))
48	filename = NULL;
49
50	/* Open src */
51	if (filename) {
52	if (stat(filename, &stat_buf) != 0) {
53	bb_simple_perror_msg(filename);
54	err:
55	exitcode = 1;
56	goto free_name;
57	}
58	if (open_to_or_warn(STDIN_FILENO, filename, O_RDONLY, 0))
59	goto err;
60	}
61
62	/* Special cases: test, stdout */
63	if (option_mask32 & (OPT_STDOUT\|OPT_TEST)) {
64	if (option_mask32 & OPT_TEST)
65	if (open_to_or_warn(STDOUT_FILENO, bb_dev_null, O_WRONLY, 0))
66	goto err;
67	filename = NULL;
68	}
69
70	/* Open dst if we are going to unpack to file */
71	if (filename) {
72	new_name = make_new_name(filename);
73	if (!new_name) {
74	bb_error_msg("%s: unknown suffix - ignored", filename);
75	goto err;
76	}
77
78	/* -f: overwrite existing output files */
79	if (option_mask32 & OPT_FORCE) {
80	unlink(new_name);
81	}
82
83	/* O_EXCL: "real" bunzip2 doesn't overwrite files */
84	/* GNU gunzip does not bail out, but goes to next file */
85	if (open_to_or_warn(STDOUT_FILENO, new_name, O_WRONLY \| O_CREAT \| O_EXCL,
86	stat_buf.st_mode))
87	goto err;
88	}
89
90	/* Check that the input is sane */
91	if (isatty(STDIN_FILENO) && (option_mask32 & OPT_FORCE) == 0) {
92	bb_error_msg_and_die("compressed data not read from terminal, "
93	"use -f to force it");
94	}
95
96	/* memset(&info, 0, sizeof(info)); */
97	info.mtime = 0; /* so far it has one member only */
98	status = unpacker(&info);
99	if (status < 0)
100	exitcode = 1;
101	xclose(STDOUT_FILENO); /* with error check! */
102
103	if (filename) {
104	char *del = new_name;
105	if (status >= 0) {
106	/* TODO: restore other things? */
107	if (info.mtime) {
108	struct timeval times[2];
109
110	times[1].tv_sec = times[0].tv_sec = info.mtime;
111	times[1].tv_usec = times[0].tv_usec = 0;
112	/* Note: we closed it first.
113	* On some systems calling utimes
114	* then closing resets the mtime
115	* back to current time. */
116	utimes(new_name, times); /* ignoring errors */
117	}
118
119	/* Delete _compressed_ file */
120	del = filename;
121	/* restore extension (unless tgz -> tar case) */
122	if (new_name == filename)
123	filename[strlen(filename)] = '.';
124	}
125	xunlink(del);
126
127	#if 0 /* Currently buggy - wrong name: "a.gz: 261% - replaced with a.gz" */
128	/* Extreme bloat for gunzip compat */
129	if (ENABLE_DESKTOP && (option_mask32 & OPT_VERBOSE) && status >= 0) {
130	fprintf(stderr, "%s: %u%% - replaced with %s\n",
131	filename, (unsigned)(stat_buf.st_size*100 / (status+1)), new_name);
132	}
133	#endif
134
135	free_name:
136	if (new_name != filename)
137	free(new_name);
138	}
139	} while (argv && ++argv);
140
141	return exitcode;
142	}
143
144	#if ENABLE_BUNZIP2 \|\| ENABLE_UNLZMA \|\| ENABLE_UNCOMPRESS
145
146	static
147	char* make_new_name_generic(char filename, const char expected_ext)
148	{
149	char *extension = strrchr(filename, '.');
150	if (!extension \|\| strcmp(extension + 1, expected_ext) != 0) {
151	/* Mimic GNU gunzip - "real" bunzip2 tries to */
152	/* unpack file anyway, to file.out */
153	return NULL;
154	}
155	*extension = '\0';
156	return filename;
157	}
158
159	#endif
160
161
162	/*
163	* Modified for busybox by Glenn McGrath
164	* Added support output to stdout by Thomas Lundquist <thomasez@zelow.no>
165	*
166	* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
167	*/
168
169	#if ENABLE_BUNZIP2
170
171	static
172	char* make_new_name_bunzip2(char *filename)
173	{
174	return make_new_name_generic(filename, "bz2");
175	}
176
177	static
178	IF_DESKTOP(long long) int unpack_bunzip2(unpack_info_t *info UNUSED_PARAM)
179	{
180	return unpack_bz2_stream_prime(STDIN_FILENO, STDOUT_FILENO);
181	}
182
183	int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
184	int bunzip2_main(int argc UNUSED_PARAM, char **argv)
185	{
186	getopt32(argv, "cfvdt");
187	argv += optind;
188	if (applet_name[2] == 'c')
189	option_mask32 \|= OPT_STDOUT;
190
191	return bbunpack(argv, make_new_name_bunzip2, unpack_bunzip2);
192	}
193
194	#endif
195
196
197	/*
198	* Gzip implementation for busybox
199	*
200	* Based on GNU gzip v1.2.4 Copyright (C) 1992-1993 Jean-loup Gailly.
201	*
202	* Originally adjusted for busybox by Sven Rudolph <sr1@inf.tu-dresden.de>
203	* based on gzip sources
204	*
205	* Adjusted further by Erik Andersen <andersen@codepoet.org> to support files as
206	* well as stdin/stdout, and to generally behave itself wrt command line
207	* handling.
208	*
209	* General cleanup to better adhere to the style guide and make use of standard
210	* busybox functions by Glenn McGrath
211	*
212	* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
213	*
214	* gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
215	* Copyright (C) 1992-1993 Jean-loup Gailly
216	* The unzip code was written and put in the public domain by Mark Adler.
217	* Portions of the lzw code are derived from the public domain 'compress'
218	* written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies,
219	* Ken Turkowski, Dave Mack and Peter Jannesen.
220	*
221	* See the license_msg below and the file COPYING for the software license.
222	* See the file algorithm.doc for the compression algorithms and file formats.
223	*/
224
225	#if ENABLE_GUNZIP
226
227	static
228	char* make_new_name_gunzip(char *filename)
229	{
230	char *extension = strrchr(filename, '.');
231
232	if (!extension)
233	return NULL;
234
235	extension++;
236	if (strcmp(extension, "tgz" + 1) == 0
237	#if ENABLE_FEATURE_SEAMLESS_Z
238	\|\| (extension[0] == 'Z' && extension[1] == '\0')
239	#endif
240	) {
241	extension[-1] = '\0';
242	} else if (strcmp(extension, "tgz") == 0) {
243	filename = xstrdup(filename);
244	extension = strrchr(filename, '.');
245	extension[2] = 'a';
246	extension[3] = 'r';
247	} else {
248	return NULL;
249	}
250	return filename;
251	}
252
253	static
254	IF_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info)
255	{
256	IF_DESKTOP(long long) int status = -1;
257
258	/* do the decompression, and cleanup */
259	if (xread_char(STDIN_FILENO) == 0x1f) {
260	unsigned char magic2;
261
262	magic2 = xread_char(STDIN_FILENO);
263	if (ENABLE_FEATURE_SEAMLESS_Z && magic2 == 0x9d) {
264	status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
265	} else if (magic2 == 0x8b) {
266	status = unpack_gz_stream_with_info(STDIN_FILENO, STDOUT_FILENO, info);
267	} else {
268	goto bad_magic;
269	}
270	if (status < 0) {
271	bb_error_msg("error inflating");
272	}
273	} else {
274	bad_magic:
275	bb_error_msg("invalid magic");
276	/* status is still == -1 */
277	}
278	return status;
279	}
280
281	/*
282	* Linux kernel build uses gzip -d -n. We accept and ignore it.
283	* Man page says:
284	* -n --no-name
285	* gzip: do not save the original file name and time stamp.
286	* (The original name is always saved if the name had to be truncated.)
287	* gunzip: do not restore the original file name/time even if present
288	* (remove only the gzip suffix from the compressed file name).
289	* This option is the default when decompressing.
290	* -N --name
291	* gzip: always save the original file name and time stamp (this is the default)
292	* gunzip: restore the original file name and time stamp if present.
293	*/
294
295	int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
296	int gunzip_main(int argc UNUSED_PARAM, char **argv)
297	{
298	getopt32(argv, "cfvdtn");
299	argv += optind;
300	/* if called as zcat */
301	if (applet_name[1] == 'c')
302	option_mask32 \|= OPT_STDOUT;
303
304	return bbunpack(argv, make_new_name_gunzip, unpack_gunzip);
305	}
306
307	#endif
308
309
310	/*
311	* Small lzma deflate implementation.
312	* Copyright (C) 2006 Aurelien Jacobs <aurel@gnuage.org>
313	*
314	* Based on bunzip.c from busybox
315	*
316	* Licensed under GPL v2, see file LICENSE in this tarball for details.
317	*/
318
319	#if ENABLE_UNLZMA
320
321	static
322	char* make_new_name_unlzma(char *filename)
323	{
324	return make_new_name_generic(filename, "lzma");
325	}
326
327	static
328	IF_DESKTOP(long long) int unpack_unlzma(unpack_info_t *info UNUSED_PARAM)
329	{
330	return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
331	}
332
333	int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
334	int unlzma_main(int argc UNUSED_PARAM, char **argv)
335	{
336	getopt32(argv, "cf");
337	argv += optind;
338	/* lzmacat? */
339	if (applet_name[4] == 'c')
340	option_mask32 \|= OPT_STDOUT;
341
342	return bbunpack(argv, make_new_name_unlzma, unpack_unlzma);
343	}
344
345	#endif
346
347
348	/*
349	* Uncompress applet for busybox (c) 2002 Glenn McGrath
350	*
351	* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
352	*/
353
354	#if ENABLE_UNCOMPRESS
355
356	static
357	char* make_new_name_uncompress(char *filename)
358	{
359	return make_new_name_generic(filename, "Z");
360	}
361
362	static
363	IF_DESKTOP(long long) int unpack_uncompress(unpack_info_t *info UNUSED_PARAM)
364	{
365	IF_DESKTOP(long long) int status = -1;
366
367	if ((xread_char(STDIN_FILENO) != 0x1f) \|\| (xread_char(STDIN_FILENO) != 0x9d)) {
368	bb_error_msg("invalid magic");
369	} else {
370	status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
371	}
372	return status;
373	}
374
375	int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
376	int uncompress_main(int argc UNUSED_PARAM, char **argv)
377	{
378	getopt32(argv, "cf");
379	argv += optind;
380
381	return bbunpack(argv, make_new_name_uncompress, unpack_uncompress);
382	}
383
384	#endif