source: MondoRescue/branches/2.2.9/mindi-busybox/archival/bbunzip.c@ 2725

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