a4e525f2e96de6b9acc5c7141a7ec10163308d06
[people/mcb30/busybox.git] / archival / bbunzip.c
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 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 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
124 static
125 char* 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
149 static
150 char* make_new_name_bunzip2(char *filename)
151 {
152         return make_new_name_generic(filename, "bz2");
153 }
154
155 static
156 USE_DESKTOP(long long) int unpack_bunzip2(void)
157 {
158         return unpack_bz2_stream(STDIN_FILENO, STDOUT_FILENO);
159 }
160
161 int bunzip2_main(int argc, char **argv);
162 int 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
205 static
206 char* 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
231 static
232 USE_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
260 int gunzip_main(int argc, char **argv);
261 int 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
286 static
287 char* make_new_name_unlzma(char *filename)
288 {
289         return make_new_name_generic(filename, "lzma");
290 }
291
292 static
293 USE_DESKTOP(long long) int unpack_unlzma(void)
294 {
295         return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
296 }
297
298 int unlzma_main(int argc, char **argv);
299 int 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
321 static
322 char* make_new_name_uncompress(char *filename)
323 {
324         return make_new_name_generic(filename, "Z");
325 }
326
327 static
328 USE_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
340 int uncompress_main(int argc, char **argv);
341 int 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