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