blob: 327b3cf2cf1999c59a9a55e95b8f18f37daf3e5c [file] [log] [blame]
Denis Vlasenkoab9eef22007-03-07 22:02:23 +00001/* vi: set sw=4 ts=4: */
2/*
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +00003 * Common code for gunzip-like applets
Denis Vlasenkoab9eef22007-03-07 22:02:23 +00004 *
5 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
6 */
7
Denis Vlasenkob6adbf12007-05-26 19:00:18 +00008#include "libbb.h"
Denis Vlasenkoab9eef22007-03-07 22:02:23 +00009#include "unarchive.h"
10
11enum {
Denis Vlasenko75605782007-03-14 00:07:51 +000012 OPT_STDOUT = 0x1,
13 OPT_FORCE = 0x2,
Denis Vlasenko6fa3ab32007-10-21 18:59:58 +000014/* gunzip and bunzip2 only: */
Denis Vlasenko75605782007-03-14 00:07:51 +000015 OPT_VERBOSE = 0x4,
16 OPT_DECOMPRESS = 0x8,
17 OPT_TEST = 0x10,
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000018};
19
20static
21int open_to_or_warn(int to_fd, const char *filename, int flags, int mode)
22{
Denis Vlasenko50f7f442007-04-11 23:20:53 +000023 int fd = open3_or_warn(filename, flags, mode);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000024 if (fd < 0) {
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000025 return 1;
26 }
Denis Vlasenko50f7f442007-04-11 23:20:53 +000027 xmove_fd(fd, to_fd);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000028 return 0;
29}
30
Denis Vlasenko75605782007-03-14 00:07:51 +000031int bbunpack(char **argv,
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000032 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;
Denis Vlasenko6c939e02007-03-07 23:22:47 +000038 char *filename, *new_name;
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000039 smallint exitcode = 0;
40
41 do {
Denis Vlasenko6c939e02007-03-07 23:22:47 +000042 /* NB: new_name is *maybe* malloc'ed! */
43 new_name = NULL;
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000044 filename = *argv; /* can be NULL - 'streaming' bunzip2 */
Denis Vlasenko6c939e02007-03-07 23:22:47 +000045
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000046 if (filename && LONE_DASH(filename))
47 filename = NULL;
48
49 /* Open src */
50 if (filename) {
51 if (stat(filename, &stat_buf) != 0) {
Denis Vlasenko0c97c9d2007-10-01 11:58:38 +000052 bb_simple_perror_msg(filename);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000053 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
Denis Vlasenko6c939e02007-03-07 23:22:47 +000069 /* Open dst if we are going to unpack to file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000070 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 }
Denis Vlasenko6c939e02007-03-07 23:22:47 +000076 /* O_EXCL: "real" bunzip2 doesn't overwrite files */
Denis Vlasenko5dd8a032007-10-05 15:26:08 +000077 /* GNU gunzip does not bail out, but goes to next file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000078 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
Denis Vlasenko6c939e02007-03-07 23:22:47 +000083 /* Check that the input is sane */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000084 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? */
Denis Vlasenko6c939e02007-03-07 23:22:47 +000097 /* Delete _compressed_ file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000098 del = filename;
Denis Vlasenko6c939e02007-03-07 23:22:47 +000099 /* restore extension (unless tgz -> tar case) */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000100 if (new_name == filename)
101 filename[strlen(filename)] = '.';
102 }
Denis Vlasenko1bb552b2007-04-05 21:25:15 +0000103 xunlink(del);
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000104
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000105#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
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000112
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000113 free_name:
114 if (new_name != filename)
115 free(new_name);
116 }
117 } while (*argv && *++argv);
118
119 return exitcode;
120}
121
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000122#if ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNCOMPRESS
123
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000124static
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000125char* make_new_name_generic(char *filename, const char *expected_ext)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000126{
127 char *extension = strrchr(filename, '.');
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000128 if (!extension || strcmp(extension + 1, expected_ext) != 0) {
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000129 /* Mimic GNU gunzip - "real" bunzip2 tries to */
130 /* unpack file anyway, to file.out */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000131 return NULL;
132 }
133 *extension = '\0';
134 return filename;
135}
136
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000137#endif
138
139
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000140/*
Denis Vlasenko0beaff82007-09-21 13:16:32 +0000141 * Modified for busybox by Glenn McGrath
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000142 * 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
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000147#if ENABLE_BUNZIP2
148
149static
150char* make_new_name_bunzip2(char *filename)
151{
152 return make_new_name_generic(filename, "bz2");
153}
154
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000155static
156USE_DESKTOP(long long) int unpack_bunzip2(void)
157{
Denis Vlasenkoc14d39e2007-06-08 13:05:39 +0000158 return unpack_bz2_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000159}
160
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000161int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenko68404f12008-03-17 09:00:54 +0000162int bunzip2_main(int argc ATTRIBUTE_UNUSED, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000163{
Denis Vlasenko6fa3ab32007-10-21 18:59:58 +0000164 getopt32(argv, "cfvdt");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000165 argv += optind;
166 if (applet_name[2] == 'c')
167 option_mask32 |= OPT_STDOUT;
168
Denis Vlasenko75605782007-03-14 00:07:51 +0000169 return bbunpack(argv, make_new_name_bunzip2, unpack_bunzip2);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000170}
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
Denis Vlasenko0beaff82007-09-21 13:16:32 +0000188 * busybox functions by Glenn McGrath
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000189 *
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
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000213 extension++;
214 if (strcmp(extension, "tgz" + 1) == 0
Denis Vlasenkodbe6e662007-08-14 16:43:01 +0000215#if ENABLE_FEATURE_GUNZIP_UNCOMPRESS
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000216 || strcmp(extension, "Z") == 0
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000217#endif
218 ) {
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000219 extension[-1] = '\0';
Denis Vlasenko51742f42007-04-12 00:32:05 +0000220 } else if (strcmp(extension, "tgz") == 0) {
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000221 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) {
Denis Vlasenkoc14d39e2007-06-08 13:05:39 +0000244 status = unpack_gz_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000245 } else {
246 goto bad_magic;
247 }
248 if (status < 0) {
249 bb_error_msg("error inflating");
250 }
251 } else {
252 bad_magic:
253 bb_error_msg("invalid magic");
254 /* status is still == -1 */
255 }
256 return status;
257}
258
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000259int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenko68404f12008-03-17 09:00:54 +0000260int gunzip_main(int argc ATTRIBUTE_UNUSED, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000261{
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000262 getopt32(argv, "cfvdt");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000263 argv += optind;
264 /* if called as zcat */
265 if (applet_name[1] == 'c')
266 option_mask32 |= OPT_STDOUT;
267
Denis Vlasenko75605782007-03-14 00:07:51 +0000268 return bbunpack(argv, make_new_name_gunzip, unpack_gunzip);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000269}
270
271#endif
272
273
274/*
275 * Small lzma deflate implementation.
276 * Copyright (C) 2006 Aurelien Jacobs <aurel@gnuage.org>
277 *
278 * Based on bunzip.c from busybox
279 *
280 * Licensed under GPL v2, see file LICENSE in this tarball for details.
281 */
282
283#if ENABLE_UNLZMA
284
285static
286char* make_new_name_unlzma(char *filename)
287{
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000288 return make_new_name_generic(filename, "lzma");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000289}
290
291static
292USE_DESKTOP(long long) int unpack_unlzma(void)
293{
Denis Vlasenkoc14d39e2007-06-08 13:05:39 +0000294 return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000295}
296
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000297int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenko68404f12008-03-17 09:00:54 +0000298int unlzma_main(int argc ATTRIBUTE_UNUSED, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000299{
Denis Vlasenko211f7f82007-09-05 11:48:32 +0000300 getopt32(argv, "cf");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000301 argv += optind;
302 /* lzmacat? */
303 if (applet_name[4] == 'c')
304 option_mask32 |= OPT_STDOUT;
305
Denis Vlasenko75605782007-03-14 00:07:51 +0000306 return bbunpack(argv, make_new_name_unlzma, unpack_unlzma);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000307}
308
309#endif
310
311
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000312/*
313 * Uncompress applet for busybox (c) 2002 Glenn McGrath
314 *
315 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
316 */
317
318#if ENABLE_UNCOMPRESS
319
320static
321char* make_new_name_uncompress(char *filename)
322{
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000323 return make_new_name_generic(filename, "Z");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000324}
325
326static
327USE_DESKTOP(long long) int unpack_uncompress(void)
328{
329 USE_DESKTOP(long long) int status = -1;
330
331 if ((xread_char(STDIN_FILENO) != 0x1f) || (xread_char(STDIN_FILENO) != 0x9d)) {
332 bb_error_msg("invalid magic");
333 } else {
334 status = uncompress(STDIN_FILENO, STDOUT_FILENO);
335 }
336 return status;
337}
338
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000339int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenko68404f12008-03-17 09:00:54 +0000340int uncompress_main(int argc ATTRIBUTE_UNUSED, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000341{
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000342 getopt32(argv, "cf");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000343 argv += optind;
344
Denis Vlasenko75605782007-03-14 00:07:51 +0000345 return bbunpack(argv, make_new_name_uncompress, unpack_uncompress);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000346}
347
348#endif