Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Gzip implementation for busybox |
| 4 | * |
Glenn L McGrath | c2bf5ca | 2000-09-29 06:46:59 +0000 | [diff] [blame] | 5 | * Based on GNU gzip v1.2.4 Copyright (C) 1992-1993 Jean-loup Gailly. |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 6 | * |
| 7 | * Originally adjusted for busybox by Sven Rudolph <sr1@inf.tu-dresden.de> |
| 8 | * based on gzip sources |
| 9 | * |
Eric Andersen | cb81e64 | 2003-07-14 21:21:08 +0000 | [diff] [blame] | 10 | * Adjusted further by Erik Andersen <andersen@codepoet.org> to support files as |
Eric Andersen | 50e4d66 | 2002-04-06 05:15:46 +0000 | [diff] [blame] | 11 | * well as stdin/stdout, and to generally behave itself wrt command line |
| 12 | * handling. |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 13 | * |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 14 | * General cleanup to better adhere to the style guide and make use of standard |
| 15 | * busybox functions by Glenn McGrath <bug1@optushome.com.au> |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 16 | * |
Erik Andersen | 61677fe | 2000-04-13 01:18:56 +0000 | [diff] [blame] | 17 | * This program is free software; you can redistribute it and/or modify |
| 18 | * it under the terms of the GNU General Public License as published by |
| 19 | * the Free Software Foundation; either version 2 of the License, or |
| 20 | * (at your option) any later version. |
| 21 | * |
| 22 | * This program is distributed in the hope that it will be useful, |
| 23 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 24 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 25 | * General Public License for more details. |
| 26 | * |
| 27 | * You should have received a copy of the GNU General Public License |
| 28 | * along with this program; if not, write to the Free Software |
| 29 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| 30 | * |
Glenn L McGrath | c2bf5ca | 2000-09-29 06:46:59 +0000 | [diff] [blame] | 31 | * |
| 32 | * gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 33 | * Copyright (C) 1992-1993 Jean-loup Gailly |
| 34 | * The unzip code was written and put in the public domain by Mark Adler. |
| 35 | * Portions of the lzw code are derived from the public domain 'compress' |
| 36 | * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies, |
| 37 | * Ken Turkowski, Dave Mack and Peter Jannesen. |
| 38 | * |
| 39 | * See the license_msg below and the file COPYING for the software license. |
| 40 | * See the file algorithm.doc for the compression algorithms and file formats. |
| 41 | */ |
| 42 | |
| 43 | #if 0 |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 44 | static char *license_msg[] = { |
| 45 | " Copyright (C) 1992-1993 Jean-loup Gailly", |
| 46 | " This program is free software; you can redistribute it and/or modify", |
| 47 | " it under the terms of the GNU General Public License as published by", |
| 48 | " the Free Software Foundation; either version 2, or (at your option)", |
| 49 | " any later version.", |
| 50 | "", |
| 51 | " This program is distributed in the hope that it will be useful,", |
| 52 | " but WITHOUT ANY WARRANTY; without even the implied warranty of", |
| 53 | " MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the", |
| 54 | " GNU General Public License for more details.", |
| 55 | "", |
| 56 | " You should have received a copy of the GNU General Public License", |
| 57 | " along with this program; if not, write to the Free Software", |
| 58 | " Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.", |
| 59 | 0 |
| 60 | }; |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 61 | #endif |
| 62 | |
Glenn L McGrath | c2bf5ca | 2000-09-29 06:46:59 +0000 | [diff] [blame] | 63 | #include <stdlib.h> |
Glenn L McGrath | 7fd9294 | 2001-04-11 03:11:33 +0000 | [diff] [blame] | 64 | #include <string.h> |
| 65 | #include <unistd.h> |
| 66 | #include <getopt.h> |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 67 | #include <sys/types.h> |
| 68 | #include <sys/stat.h> |
| 69 | #include <fcntl.h> |
| 70 | |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 71 | #include "busybox.h" |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 72 | #include "unarchive.h" |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 73 | |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 74 | #define GUNZIP_OPT_STDOUT 1 |
| 75 | #define GUNZIP_OPT_FORCE 2 |
| 76 | #define GUNZIP_OPT_TEST 4 |
| 77 | #define GUNZIP_OPT_DECOMPRESS 8 |
Matt Kraai | a4a65e7 | 2002-04-29 15:32:32 +0000 | [diff] [blame] | 78 | |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 79 | extern int gunzip_main(int argc, char **argv) |
| 80 | { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 81 | char status = EXIT_SUCCESS; |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 82 | unsigned long opt; |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 83 | |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 84 | opt = bb_getopt_ulflags(argc, argv, "cftd"); |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 85 | /* if called as zcat */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 86 | if (strcmp(bb_applet_name, "zcat") == 0) { |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 87 | opt |= GUNZIP_OPT_STDOUT; |
Glenn L McGrath | bcfeb2a | 2001-04-18 13:34:09 +0000 | [diff] [blame] | 88 | } |
| 89 | |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 90 | do { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 91 | struct stat stat_buf; |
| 92 | const char *old_path = argv[optind]; |
| 93 | const char *delete_path = NULL; |
| 94 | char *new_path = NULL; |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 95 | int src_fd; |
| 96 | int dst_fd; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 97 | |
| 98 | optind++; |
| 99 | |
| 100 | if (old_path == NULL || strcmp(old_path, "-") == 0) { |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 101 | src_fd = STDIN_FILENO; |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 102 | opt |= GUNZIP_OPT_STDOUT; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 103 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 104 | src_fd = bb_xopen(old_path, O_RDONLY); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 105 | |
| 106 | /* Get the time stamp on the input file. */ |
| 107 | if (stat(old_path, &stat_buf) < 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 108 | bb_error_msg_and_die("Couldn't stat file %s", old_path); |
Glenn L McGrath | 1ee52e8 | 2002-08-24 10:30:36 +0000 | [diff] [blame] | 109 | } |
| 110 | } |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 111 | |
| 112 | /* Check that the input is sane. */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 113 | if (isatty(src_fd) && ((opt & GUNZIP_OPT_FORCE) == 0)) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 114 | bb_error_msg_and_die |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 115 | ("compressed data not read from terminal. Use -f to force it."); |
| 116 | } |
| 117 | |
| 118 | /* Set output filename and number */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 119 | if (opt & GUNZIP_OPT_TEST) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 120 | dst_fd = bb_xopen("/dev/null", O_WRONLY); /* why does test use filenum 2 ? */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 121 | } else if (opt & GUNZIP_OPT_STDOUT) { |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 122 | dst_fd = STDOUT_FILENO; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 123 | } else { |
| 124 | char *extension; |
| 125 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 126 | new_path = bb_xstrdup(old_path); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 127 | |
| 128 | extension = strrchr(new_path, '.'); |
Glenn L McGrath | bf1cc8b | 2002-11-01 23:38:54 +0000 | [diff] [blame] | 129 | #ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS |
Glenn L McGrath | 9ef0944 | 2002-11-01 22:11:53 +0000 | [diff] [blame] | 130 | if (extension && (strcmp(extension, ".Z") == 0)) { |
| 131 | *extension = '\0'; |
| 132 | } else |
| 133 | #endif |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 134 | if (extension && (strcmp(extension, ".gz") == 0)) { |
| 135 | *extension = '\0'; |
| 136 | } else if (extension && (strcmp(extension, ".tgz") == 0)) { |
| 137 | extension[2] = 'a'; |
| 138 | extension[3] = 'r'; |
| 139 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 140 | bb_error_msg_and_die("Invalid extension"); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 141 | } |
| 142 | |
| 143 | /* Open output file */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 144 | dst_fd = bb_xopen(new_path, O_WRONLY | O_CREAT); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 145 | |
| 146 | /* Set permissions on the file */ |
Glenn L McGrath | c3b7f7d | 2002-08-26 17:17:27 +0000 | [diff] [blame] | 147 | chmod(new_path, stat_buf.st_mode); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 148 | |
| 149 | /* If unzip succeeds remove the old file */ |
| 150 | delete_path = old_path; |
| 151 | } |
| 152 | |
| 153 | /* do the decompression, and cleanup */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 154 | if (bb_xread_char(src_fd) == 0x1f) { |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 155 | unsigned char magic2; |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 156 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 157 | magic2 = bb_xread_char(src_fd); |
Glenn L McGrath | bf1cc8b | 2002-11-01 23:38:54 +0000 | [diff] [blame] | 158 | #ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 159 | if (magic2 == 0x9d) { |
Glenn L McGrath | 9ef0944 | 2002-11-01 22:11:53 +0000 | [diff] [blame] | 160 | status = uncompress(src_fd, dst_fd); |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 161 | } else |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 162 | #endif |
| 163 | if (magic2 == 0x8b) { |
| 164 | check_header_gzip(src_fd); |
Glenn L McGrath | 5699b85 | 2003-11-15 23:19:05 +0000 | [diff] [blame] | 165 | status = inflate_gunzip(src_fd, dst_fd); |
Glenn L McGrath | 5c99581 | 2002-09-30 05:30:29 +0000 | [diff] [blame] | 166 | if (status != 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 167 | bb_error_msg_and_die("Error inflating"); |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 168 | } |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 169 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 170 | bb_error_msg_and_die("Invalid magic"); |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 171 | } |
Glenn L McGrath | 563ac6e | 2002-11-01 21:40:52 +0000 | [diff] [blame] | 172 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 173 | bb_error_msg_and_die("Invalid magic"); |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 174 | } |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 175 | |
| 176 | if ((status != EXIT_SUCCESS) && (new_path)) { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 177 | /* Unzip failed, remove new path instead of old path */ |
| 178 | delete_path = new_path; |
| 179 | } |
| 180 | |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 181 | if (dst_fd != STDOUT_FILENO) { |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 182 | close(dst_fd); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 183 | } |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 184 | if (src_fd != STDIN_FILENO) { |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 185 | close(src_fd); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 186 | } |
| 187 | |
| 188 | /* delete_path will be NULL if in test mode or from stdin */ |
| 189 | if (delete_path && (unlink(delete_path) == -1)) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 190 | bb_error_msg_and_die("Couldn't remove %s", delete_path); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 191 | } |
| 192 | |
| 193 | free(new_path); |
| 194 | |
| 195 | } while (optind < argc); |
| 196 | |
Matt Kraai | a4a65e7 | 2002-04-29 15:32:32 +0000 | [diff] [blame] | 197 | return status; |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 198 | } |