git.maemo.org Git - busybox4maemo/blob - 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 and bunzip2 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_simple_perror_msg(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 does 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
 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) MAIN_EXTERNALLY_VISIBLE;
 162 int bunzip2_main(int argc ATTRIBUTE_UNUSED, char **argv)
 163 {
 164         getopt32(argv, "cfvdt");
 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
 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                         status = unpack_gz_stream(STDIN_FILENO, STDOUT_FILENO);
 245                 } 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
 259 int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 260 int gunzip_main(int argc ATTRIBUTE_UNUSED, char **argv)
 261 {
 262         getopt32(argv, "cfvdt");
 263         argv += optind;
 264         /* if called as zcat */
 265         if (applet_name[1] == 'c')
 266                 option_mask32 |= OPT_STDOUT;
 267
 268         return bbunpack(argv, make_new_name_gunzip, unpack_gunzip);
 269 }
 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
 285 static
 286 char* make_new_name_unlzma(char *filename)
 287 {
 288         return make_new_name_generic(filename, "lzma");
 289 }
 290
 291 static
 292 USE_DESKTOP(long long) int unpack_unlzma(void)
 293 {
 294         return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO);
 295 }
 296
 297 int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 298 int unlzma_main(int argc ATTRIBUTE_UNUSED, char **argv)
 299 {
 300         getopt32(argv, "cf");
 301         argv += optind;
 302         /* lzmacat? */
 303         if (applet_name[4] == 'c')
 304                 option_mask32 |= OPT_STDOUT;
 305
 306         return bbunpack(argv, make_new_name_unlzma, unpack_unlzma);
 307 }
 308
 309 #endif
 310
 311
 312 /*
 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
 320 static
 321 char* make_new_name_uncompress(char *filename)
 322 {
 323         return make_new_name_generic(filename, "Z");
 324 }
 325
 326 static
 327 USE_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
 339 int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 340 int uncompress_main(int argc ATTRIBUTE_UNUSED, char **argv)
 341 {
 342         getopt32(argv, "cf");
 343         argv += optind;
 344
 345         return bbunpack(argv, make_new_name_uncompress, unpack_uncompress);
 346 }
 347
 348 #endif