Magellan Linux

Annotation of /trunk/mkinitrd-magellan/busybox/archival/bbunzip.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1123 - (hide annotations) (download)
Wed Aug 18 21:56:57 2010 UTC (13 years, 9 months ago) by niro
File MIME type: text/plain
File size: 11209 byte(s)
-updated to busybox-1.17.1
1 niro 816 /* 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     #include "libbb.h"
8     #include "unarchive.h"
9    
10     enum {
11 niro 1123 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 niro 816 };
18    
19     static
20     int 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    
30 niro 1123 char* FAST_FUNC append_ext(char *filename, const char *expected_ext)
31     {
32     return xasprintf("%s.%s", filename, expected_ext);
33     }
34    
35 niro 816 int FAST_FUNC bbunpack(char **argv,
36 niro 1123 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 niro 816 )
40     {
41     struct stat stat_buf;
42 niro 984 IF_DESKTOP(long long) int status;
43 niro 816 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 niro 1123 new_name = make_new_name(filename, expected_ext);
78 niro 816 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 niro 984 xclose(STDOUT_FILENO); /* with error check! */
107 niro 816
108     if (filename) {
109     char *del = new_name;
110     if (status >= 0) {
111     /* TODO: restore other things? */
112     if (info.mtime) {
113 niro 984 struct timeval times[2];
114 niro 816
115 niro 984 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 niro 816 }
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 niro 1123 #if ENABLE_UNCOMPRESS || ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNXZ
150 niro 816 static
151 niro 1123 char* FAST_FUNC make_new_name_generic(char *filename, const char *expected_ext)
152 niro 816 {
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 niro 1123 * Uncompress applet for busybox (c) 2002 Glenn McGrath
167 niro 816 *
168 niro 1123 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
169 niro 816 */
170 niro 1123 #if ENABLE_UNCOMPRESS
171 niro 816 static
172 niro 1123 IF_DESKTOP(long long) int FAST_FUNC unpack_uncompress(unpack_info_t *info UNUSED_PARAM)
173 niro 816 {
174 niro 1123 IF_DESKTOP(long long) int status = -1;
175 niro 816
176 niro 1123 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 niro 816 }
183 niro 1123 int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
184     int uncompress_main(int argc UNUSED_PARAM, char **argv)
185 niro 816 {
186 niro 1123 getopt32(argv, "cf");
187 niro 816 argv += optind;
188    
189 niro 1123 return bbunpack(argv, unpack_uncompress, make_new_name_generic, "Z");
190 niro 816 }
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 tarball for details.
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
222     static
223 niro 1123 char* FAST_FUNC make_new_name_gunzip(char *filename, const char *expected_ext UNUSED_PARAM)
224 niro 816 {
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     }
247     static
248 niro 1123 IF_DESKTOP(long long) int FAST_FUNC unpack_gunzip(unpack_info_t *info)
249 niro 816 {
250 niro 984 IF_DESKTOP(long long) int status = -1;
251 niro 816
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     */
287     int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
288     int 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 niro 1123 return bbunpack(argv, unpack_gunzip, make_new_name_gunzip, /*unused:*/ NULL);
297 niro 816 }
298 niro 1123 #endif
299 niro 816
300 niro 1123
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 tarball for details.
306     */
307     //usage:#define bunzip2_trivial_usage
308     //usage: "[OPTIONS] [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
321     static
322     IF_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     }
326     int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
327     int 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 niro 816 #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 GPL v2, see file LICENSE in this tarball for details.
346     */
347     #if ENABLE_UNLZMA
348     static
349 niro 1123 IF_DESKTOP(long long) int FAST_FUNC unpack_unlzma(unpack_info_t *info UNUSED_PARAM)
350 niro 816 {
351     return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
352     }
353     int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
354     int unlzma_main(int argc UNUSED_PARAM, char **argv)
355     {
356 niro 1123 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 niro 816 option_mask32 |= OPT_STDOUT;
365    
366 niro 1123 argv += optind;
367     return bbunpack(argv, unpack_unlzma, make_new_name_generic, "lzma");
368 niro 816 }
369     #endif
370    
371    
372 niro 1123 #if ENABLE_UNXZ
373 niro 816 static
374 niro 1123 IF_DESKTOP(long long) int FAST_FUNC unpack_unxz(unpack_info_t *info UNUSED_PARAM)
375 niro 816 {
376 niro 1123 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 niro 816 bb_error_msg("invalid magic");
383 niro 1123 return -1;
384 niro 816 }
385 niro 1123 return unpack_xz_stream(STDIN_FILENO, STDOUT_FILENO);
386 niro 816 }
387 niro 1123 int unxz_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
388     int 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 niro 816
400     argv += optind;
401 niro 1123 return bbunpack(argv, unpack_unxz, make_new_name_generic, "xz");
402 niro 816 }
403     #endif