X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;ds=sidebyside;f=archival%2Fgunzip.c;h=7b939290b4b13e80234aca26a6eddbacc9449689;hb=44e216f2642ff4ea53026792e80df95df71e9905;hp=ac2fb17eda119eda766b053f84f4eba5d536f195;hpb=c3b7f7d6e710ad896349cd5a2e913d3efc79275d;p=oweals%2Fbusybox.git diff --git a/archival/gunzip.c b/archival/gunzip.c index ac2fb17ed..7b939290b 100644 --- a/archival/gunzip.c +++ b/archival/gunzip.c @@ -7,13 +7,13 @@ * Originally adjusted for busybox by Sven Rudolph * based on gzip sources * - * Adjusted further by Erik Andersen to support files as + * Adjusted further by Erik Andersen to support files as * well as stdin/stdout, and to generally behave itself wrt command line * handling. * * General cleanup to better adhere to the style guide and make use of standard - * busybox functions by Glenn McGrath - * + * busybox functions by Glenn McGrath + * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or @@ -63,95 +63,82 @@ static char *license_msg[] = { #include #include #include -#include +#include +#include +#include + #include "busybox.h" +#include "unarchive.h" -const char gunzip_to_stdout = 1; -const char gunzip_force = 2; -const char gunzip_test = 4; +#define GUNZIP_OPT_STDOUT 1 +#define GUNZIP_OPT_FORCE 2 +#define GUNZIP_OPT_TEST 4 +#define GUNZIP_OPT_DECOMPRESS 8 -extern int gunzip_main(int argc, char **argv) +int gunzip_main(int argc, char **argv) { char status = EXIT_SUCCESS; - char flags = 0; - int opt; + unsigned long opt; + opt = bb_getopt_ulflags(argc, argv, "cftd"); /* if called as zcat */ - if (strcmp(applet_name, "zcat") == 0) { - flags |= gunzip_to_stdout; - } - - while ((opt = getopt(argc, argv, "ctfhd")) != -1) { - switch (opt) { - case 'c': - flags |= gunzip_to_stdout; - break; - case 'f': - flags |= gunzip_force; - break; - case 't': - flags |= gunzip_test; - break; - case 'd': /* Used to convert gzip to gunzip. */ - break; - default: - show_usage(); /* exit's inside usage */ - } + if (strcmp(bb_applet_name, "zcat") == 0) { + opt |= GUNZIP_OPT_STDOUT; } do { - FILE *in_file, *out_file; struct stat stat_buf; const char *old_path = argv[optind]; const char *delete_path = NULL; char *new_path = NULL; + int src_fd; + int dst_fd; optind++; if (old_path == NULL || strcmp(old_path, "-") == 0) { - in_file = stdin; - flags |= gunzip_to_stdout; + src_fd = STDIN_FILENO; + opt |= GUNZIP_OPT_STDOUT; } else { - in_file = wfopen(old_path, "r"); - if (in_file == NULL) { - status = EXIT_FAILURE; - break; - } + src_fd = bb_xopen(old_path, O_RDONLY); /* Get the time stamp on the input file. */ - if (stat(old_path, &stat_buf) < 0) { - error_msg_and_die("Couldn't stat file %s", old_path); - } + xstat(old_path, &stat_buf); } /* Check that the input is sane. */ - if (isatty(fileno(in_file)) && ((flags & gunzip_force) == 0)) { - error_msg_and_die + if (isatty(src_fd) && ((opt & GUNZIP_OPT_FORCE) == 0)) { + bb_error_msg_and_die ("compressed data not read from terminal. Use -f to force it."); } /* Set output filename and number */ - if (flags & gunzip_test) { - out_file = xfopen("/dev/null", "w"); /* why does test use filenum 2 ? */ - } else if (flags & gunzip_to_stdout) { - out_file = stdout; + if (opt & GUNZIP_OPT_TEST) { + dst_fd = bb_xopen(bb_dev_null, O_WRONLY); /* why does test use filenum 2 ? */ + } else if (opt & GUNZIP_OPT_STDOUT) { + dst_fd = STDOUT_FILENO; } else { char *extension; - new_path = xstrdup(old_path); + new_path = bb_xstrdup(old_path); extension = strrchr(new_path, '.'); +#ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS + if (extension && (strcmp(extension, ".Z") == 0)) { + *extension = '\0'; + } else +#endif if (extension && (strcmp(extension, ".gz") == 0)) { *extension = '\0'; } else if (extension && (strcmp(extension, ".tgz") == 0)) { extension[2] = 'a'; extension[3] = 'r'; } else { - error_msg_and_die("Invalid extension"); + bb_error_msg_and_die("Invalid extension"); } /* Open output file */ - out_file = xfopen(new_path, "w"); + dst_fd = bb_xopen(new_path, O_WRONLY | O_CREAT); /* Set permissions on the file */ chmod(new_path, stat_buf.st_mode); @@ -161,21 +148,43 @@ extern int gunzip_main(int argc, char **argv) } /* do the decompression, and cleanup */ - if ((unzip(in_file, out_file) != 0) && (new_path)) { + if (bb_xread_char(src_fd) == 0x1f) { + unsigned char magic2; + + magic2 = bb_xread_char(src_fd); +#ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS + if (magic2 == 0x9d) { + status = uncompress(src_fd, dst_fd); + } else +#endif + if (magic2 == 0x8b) { + check_header_gzip(src_fd); + status = inflate_gunzip(src_fd, dst_fd); + if (status != 0) { + bb_error_msg_and_die("Error inflating"); + } + } else { + bb_error_msg_and_die("Invalid magic"); + } + } else { + bb_error_msg_and_die("Invalid magic"); + } + + if ((status != EXIT_SUCCESS) && (new_path)) { /* Unzip failed, remove new path instead of old path */ delete_path = new_path; } - if (out_file != stdout) { - fclose(out_file); + if (dst_fd != STDOUT_FILENO) { + close(dst_fd); } - if (in_file != stdin) { - fclose(in_file); + if (src_fd != STDIN_FILENO) { + close(src_fd); } /* delete_path will be NULL if in test mode or from stdin */ if (delete_path && (unlink(delete_path) == -1)) { - error_msg_and_die("Couldn't remove %s", delete_path); + bb_error_msg_and_die("Couldn't remove %s", delete_path); } free(new_path);