source: MondoRescue/branches/2.2.5/mindi-busybox/archival/bbunzip.c @ 1765

Last change on this file since 1765 was 1765, checked in by Bruno Cornec, 12 years ago

Update to busybox 1.7.2

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