blob: 75489f2a5a96519299c8c0321e5f2f3346547844 [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 Vlasenkodefc1ea2008-06-27 02:52:20 +000031int FAST_FUNC bbunpack(char **argv,
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000032 char* (*make_new_name)(char *filename),
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +000033 USE_DESKTOP(long long) int (*unpacker)(unpack_info_t *info)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000034)
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;
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +000040 unpack_info_t info;
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000041
42 do {
Denis Vlasenko6c939e02007-03-07 23:22:47 +000043 /* NB: new_name is *maybe* malloc'ed! */
44 new_name = NULL;
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000045 filename = *argv; /* can be NULL - 'streaming' bunzip2 */
Denis Vlasenko6c939e02007-03-07 23:22:47 +000046
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000047 if (filename && LONE_DASH(filename))
48 filename = NULL;
49
50 /* Open src */
51 if (filename) {
52 if (stat(filename, &stat_buf) != 0) {
Denis Vlasenko0c97c9d2007-10-01 11:58:38 +000053 bb_simple_perror_msg(filename);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000054 err:
55 exitcode = 1;
56 goto free_name;
57 }
58 if (open_to_or_warn(STDIN_FILENO, filename, O_RDONLY, 0))
59 goto err;
60 }
61
62 /* Special cases: test, stdout */
63 if (option_mask32 & (OPT_STDOUT|OPT_TEST)) {
64 if (option_mask32 & OPT_TEST)
65 if (open_to_or_warn(STDOUT_FILENO, bb_dev_null, O_WRONLY, 0))
66 goto err;
67 filename = NULL;
68 }
69
Denis Vlasenko6c939e02007-03-07 23:22:47 +000070 /* Open dst if we are going to unpack to file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000071 if (filename) {
72 new_name = make_new_name(filename);
73 if (!new_name) {
74 bb_error_msg("%s: unknown suffix - ignored", filename);
75 goto err;
76 }
Denis Vlasenkoa04cc472008-06-25 20:54:45 +000077
78 /* -f: overwrite existing output files */
79 if (option_mask32 & OPT_FORCE) {
80 unlink(new_name);
81 }
82
Denis Vlasenko6c939e02007-03-07 23:22:47 +000083 /* O_EXCL: "real" bunzip2 doesn't overwrite files */
Denis Vlasenko5dd8a032007-10-05 15:26:08 +000084 /* GNU gunzip does not bail out, but goes to next file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000085 if (open_to_or_warn(STDOUT_FILENO, new_name, O_WRONLY | O_CREAT | O_EXCL,
86 stat_buf.st_mode))
87 goto err;
88 }
89
Denis Vlasenko6c939e02007-03-07 23:22:47 +000090 /* Check that the input is sane */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000091 if (isatty(STDIN_FILENO) && (option_mask32 & OPT_FORCE) == 0) {
92 bb_error_msg_and_die("compressed data not read from terminal, "
93 "use -f to force it");
94 }
95
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +000096 /* memset(&info, 0, sizeof(info)); */
97 info.mtime = 0; /* so far it has one member only */
98 status = unpacker(&info);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +000099 if (status < 0)
100 exitcode = 1;
101
102 if (filename) {
103 char *del = new_name;
104 if (status >= 0) {
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000105 /* TODO: restore other things? */
106 if (info.mtime) {
107 struct utimbuf times;
108
109 times.actime = info.mtime;
110 times.modtime = info.mtime;
111 /* Close first.
112 * On some systems calling utime
113 * then closing resets the mtime. */
114 close(STDOUT_FILENO);
115 /* Ignoring errors */
116 utime(new_name, &times);
117 }
118
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000119 /* Delete _compressed_ file */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000120 del = filename;
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000121 /* restore extension (unless tgz -> tar case) */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000122 if (new_name == filename)
123 filename[strlen(filename)] = '.';
124 }
Denis Vlasenko1bb552b2007-04-05 21:25:15 +0000125 xunlink(del);
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000126
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000127#if 0 /* Currently buggy - wrong name: "a.gz: 261% - replaced with a.gz" */
128 /* Extreme bloat for gunzip compat */
129 if (ENABLE_DESKTOP && (option_mask32 & OPT_VERBOSE) && status >= 0) {
130 fprintf(stderr, "%s: %u%% - replaced with %s\n",
131 filename, (unsigned)(stat_buf.st_size*100 / (status+1)), new_name);
132 }
133#endif
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000134
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000135 free_name:
136 if (new_name != filename)
137 free(new_name);
138 }
139 } while (*argv && *++argv);
140
141 return exitcode;
142}
143
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000144#if ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNCOMPRESS
145
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000146static
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000147char* make_new_name_generic(char *filename, const char *expected_ext)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000148{
149 char *extension = strrchr(filename, '.');
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000150 if (!extension || strcmp(extension + 1, expected_ext) != 0) {
Denis Vlasenko6c939e02007-03-07 23:22:47 +0000151 /* Mimic GNU gunzip - "real" bunzip2 tries to */
152 /* unpack file anyway, to file.out */
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000153 return NULL;
154 }
155 *extension = '\0';
156 return filename;
157}
158
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000159#endif
160
161
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000162/*
Denis Vlasenko0beaff82007-09-21 13:16:32 +0000163 * Modified for busybox by Glenn McGrath
Denis Vlasenkoa4688bf2007-03-11 10:56:37 +0000164 * Added support output to stdout by Thomas Lundquist <thomasez@zelow.no>
165 *
166 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
167 */
168
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000169#if ENABLE_BUNZIP2
170
171static
172char* make_new_name_bunzip2(char *filename)
173{
174 return make_new_name_generic(filename, "bz2");
175}
176
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000177static
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000178USE_DESKTOP(long long) int unpack_bunzip2(unpack_info_t *info UNUSED_PARAM)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000179{
Denis Vlasenkoe9ad84d2008-08-05 13:10:34 +0000180 return unpack_bz2_stream_prime(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000181}
182
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000183int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000184int bunzip2_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000185{
Denis Vlasenko6fa3ab32007-10-21 18:59:58 +0000186 getopt32(argv, "cfvdt");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000187 argv += optind;
188 if (applet_name[2] == 'c')
189 option_mask32 |= OPT_STDOUT;
190
Denis Vlasenko75605782007-03-14 00:07:51 +0000191 return bbunpack(argv, make_new_name_bunzip2, unpack_bunzip2);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000192}
193
194#endif
195
196
197/*
198 * Gzip implementation for busybox
199 *
200 * Based on GNU gzip v1.2.4 Copyright (C) 1992-1993 Jean-loup Gailly.
201 *
202 * Originally adjusted for busybox by Sven Rudolph <sr1@inf.tu-dresden.de>
203 * based on gzip sources
204 *
205 * Adjusted further by Erik Andersen <andersen@codepoet.org> to support files as
206 * well as stdin/stdout, and to generally behave itself wrt command line
207 * handling.
208 *
209 * General cleanup to better adhere to the style guide and make use of standard
Denis Vlasenko0beaff82007-09-21 13:16:32 +0000210 * busybox functions by Glenn McGrath
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000211 *
212 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
213 *
214 * gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
215 * Copyright (C) 1992-1993 Jean-loup Gailly
216 * The unzip code was written and put in the public domain by Mark Adler.
217 * Portions of the lzw code are derived from the public domain 'compress'
218 * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies,
219 * Ken Turkowski, Dave Mack and Peter Jannesen.
220 *
221 * See the license_msg below and the file COPYING for the software license.
222 * See the file algorithm.doc for the compression algorithms and file formats.
223 */
224
225#if ENABLE_GUNZIP
226
227static
228char* make_new_name_gunzip(char *filename)
229{
230 char *extension = strrchr(filename, '.');
231
232 if (!extension)
233 return NULL;
234
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000235 extension++;
236 if (strcmp(extension, "tgz" + 1) == 0
Denis Vlasenkoe9ad84d2008-08-05 13:10:34 +0000237#if ENABLE_FEATURE_SEAMLESS_Z
238 || (extension[0] == 'Z' && extension[1] == '\0')
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000239#endif
240 ) {
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000241 extension[-1] = '\0';
Denis Vlasenko51742f42007-04-12 00:32:05 +0000242 } else if (strcmp(extension, "tgz") == 0) {
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000243 filename = xstrdup(filename);
244 extension = strrchr(filename, '.');
245 extension[2] = 'a';
246 extension[3] = 'r';
247 } else {
248 return NULL;
249 }
250 return filename;
251}
252
253static
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000254USE_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000255{
256 USE_DESKTOP(long long) int status = -1;
257
258 /* do the decompression, and cleanup */
259 if (xread_char(STDIN_FILENO) == 0x1f) {
260 unsigned char magic2;
261
262 magic2 = xread_char(STDIN_FILENO);
Denis Vlasenkoe9ad84d2008-08-05 13:10:34 +0000263 if (ENABLE_FEATURE_SEAMLESS_Z && magic2 == 0x9d) {
264 status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000265 } else if (magic2 == 0x8b) {
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000266 status = unpack_gz_stream_with_info(STDIN_FILENO, STDOUT_FILENO, info);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000267 } else {
268 goto bad_magic;
269 }
270 if (status < 0) {
271 bb_error_msg("error inflating");
272 }
273 } else {
274 bad_magic:
275 bb_error_msg("invalid magic");
276 /* status is still == -1 */
277 }
278 return status;
279}
280
Denis Vlasenko52978092008-03-30 13:11:47 +0000281/*
282 * Linux kernel build uses gzip -d -n. We accept and ignore it.
283 * Man page says:
284 * -n --no-name
285 * gzip: do not save the original file name and time stamp.
286 * (The original name is always saved if the name had to be truncated.)
287 * gunzip: do not restore the original file name/time even if present
288 * (remove only the gzip suffix from the compressed file name).
289 * This option is the default when decompressing.
290 * -N --name
291 * gzip: always save the original file name and time stamp (this is the default)
292 * gunzip: restore the original file name and time stamp if present.
293 */
294
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000295int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000296int gunzip_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000297{
Denis Vlasenko52978092008-03-30 13:11:47 +0000298 getopt32(argv, "cfvdtn");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000299 argv += optind;
300 /* if called as zcat */
301 if (applet_name[1] == 'c')
302 option_mask32 |= OPT_STDOUT;
303
Denis Vlasenko75605782007-03-14 00:07:51 +0000304 return bbunpack(argv, make_new_name_gunzip, unpack_gunzip);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000305}
306
307#endif
308
309
310/*
311 * Small lzma deflate implementation.
312 * Copyright (C) 2006 Aurelien Jacobs <aurel@gnuage.org>
313 *
314 * Based on bunzip.c from busybox
315 *
316 * Licensed under GPL v2, see file LICENSE in this tarball for details.
317 */
318
319#if ENABLE_UNLZMA
320
321static
322char* make_new_name_unlzma(char *filename)
323{
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000324 return make_new_name_generic(filename, "lzma");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000325}
326
327static
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000328USE_DESKTOP(long long) int unpack_unlzma(unpack_info_t *info UNUSED_PARAM)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000329{
Denis Vlasenkoc14d39e2007-06-08 13:05:39 +0000330 return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000331}
332
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000333int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000334int unlzma_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000335{
Denis Vlasenko211f7f82007-09-05 11:48:32 +0000336 getopt32(argv, "cf");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000337 argv += optind;
338 /* lzmacat? */
339 if (applet_name[4] == 'c')
340 option_mask32 |= OPT_STDOUT;
341
Denis Vlasenko75605782007-03-14 00:07:51 +0000342 return bbunpack(argv, make_new_name_unlzma, unpack_unlzma);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000343}
344
345#endif
346
347
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000348/*
349 * Uncompress applet for busybox (c) 2002 Glenn McGrath
350 *
351 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
352 */
353
354#if ENABLE_UNCOMPRESS
355
356static
357char* make_new_name_uncompress(char *filename)
358{
Denis Vlasenkobebbd8c2007-03-09 20:49:55 +0000359 return make_new_name_generic(filename, "Z");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000360}
361
362static
Denis Vlasenko1a9e9bd2008-11-01 12:54:56 +0000363USE_DESKTOP(long long) int unpack_uncompress(unpack_info_t *info UNUSED_PARAM)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000364{
365 USE_DESKTOP(long long) int status = -1;
366
367 if ((xread_char(STDIN_FILENO) != 0x1f) || (xread_char(STDIN_FILENO) != 0x9d)) {
368 bb_error_msg("invalid magic");
369 } else {
Denis Vlasenkoe9ad84d2008-08-05 13:10:34 +0000370 status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000371 }
372 return status;
373}
374
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000375int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000376int uncompress_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000377{
Denis Vlasenkofe7cd642007-08-18 15:32:12 +0000378 getopt32(argv, "cf");
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000379 argv += optind;
380
Denis Vlasenko75605782007-03-14 00:07:51 +0000381 return bbunpack(argv, make_new_name_uncompress, unpack_uncompress);
Denis Vlasenkoab9eef22007-03-07 22:02:23 +0000382}
383
384#endif