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 |
Glenn L McGrath | c6992fe | 2004-04-25 05:11:19 +0000 | [diff] [blame] | 15 | * busybox functions by Glenn McGrath <bug1@iinet.net.au> |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 16 | * |
"Robert P. J. Day" | 801ab14 | 2006-07-12 07:56:04 +0000 | [diff] [blame] | 17 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
Glenn L McGrath | c2bf5ca | 2000-09-29 06:46:59 +0000 | [diff] [blame] | 18 | * |
| 19 | * gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 20 | * Copyright (C) 1992-1993 Jean-loup Gailly |
| 21 | * The unzip code was written and put in the public domain by Mark Adler. |
| 22 | * Portions of the lzw code are derived from the public domain 'compress' |
| 23 | * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies, |
| 24 | * Ken Turkowski, Dave Mack and Peter Jannesen. |
| 25 | * |
| 26 | * See the license_msg below and the file COPYING for the software license. |
| 27 | * See the file algorithm.doc for the compression algorithms and file formats. |
| 28 | */ |
| 29 | |
Glenn L McGrath | c2bf5ca | 2000-09-29 06:46:59 +0000 | [diff] [blame] | 30 | #include <stdlib.h> |
Glenn L McGrath | 7fd9294 | 2001-04-11 03:11:33 +0000 | [diff] [blame] | 31 | #include <string.h> |
| 32 | #include <unistd.h> |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 33 | #include <sys/types.h> |
| 34 | #include <sys/stat.h> |
| 35 | #include <fcntl.h> |
| 36 | |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 37 | #include "busybox.h" |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 38 | #include "unarchive.h" |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 39 | |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 40 | #define GUNZIP_OPT_STDOUT 1 |
| 41 | #define GUNZIP_OPT_FORCE 2 |
| 42 | #define GUNZIP_OPT_TEST 4 |
| 43 | #define GUNZIP_OPT_DECOMPRESS 8 |
Matt Kraai | a4a65e7 | 2002-04-29 15:32:32 +0000 | [diff] [blame] | 44 | |
Rob Landley | dfba741 | 2006-03-06 20:47:33 +0000 | [diff] [blame] | 45 | int gunzip_main(int argc, char **argv) |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 46 | { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 47 | char status = EXIT_SUCCESS; |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 48 | unsigned long opt; |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 49 | |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 50 | opt = bb_getopt_ulflags(argc, argv, "cftd"); |
Glenn L McGrath | 58e42d5 | 2001-03-28 05:38:24 +0000 | [diff] [blame] | 51 | /* if called as zcat */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 52 | if (strcmp(bb_applet_name, "zcat") == 0) { |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 53 | opt |= GUNZIP_OPT_STDOUT; |
Glenn L McGrath | bcfeb2a | 2001-04-18 13:34:09 +0000 | [diff] [blame] | 54 | } |
| 55 | |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 56 | do { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 57 | struct stat stat_buf; |
| 58 | const char *old_path = argv[optind]; |
| 59 | const char *delete_path = NULL; |
| 60 | char *new_path = NULL; |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 61 | int src_fd; |
| 62 | int dst_fd; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 63 | |
| 64 | optind++; |
| 65 | |
| 66 | if (old_path == NULL || strcmp(old_path, "-") == 0) { |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 67 | src_fd = STDIN_FILENO; |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 68 | opt |= GUNZIP_OPT_STDOUT; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 69 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 70 | src_fd = bb_xopen(old_path, O_RDONLY); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 71 | |
| 72 | /* Get the time stamp on the input file. */ |
Rob Landley | c5b1d4d | 2006-03-13 15:45:16 +0000 | [diff] [blame] | 73 | xstat(old_path, &stat_buf); |
Glenn L McGrath | 1ee52e8 | 2002-08-24 10:30:36 +0000 | [diff] [blame] | 74 | } |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 75 | |
| 76 | /* Check that the input is sane. */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 77 | if (isatty(src_fd) && ((opt & GUNZIP_OPT_FORCE) == 0)) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 78 | bb_error_msg_and_die |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 79 | ("compressed data not read from terminal. Use -f to force it."); |
| 80 | } |
| 81 | |
| 82 | /* Set output filename and number */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 83 | if (opt & GUNZIP_OPT_TEST) { |
"Vladimir N. Oleynik" | 6c35c7c | 2005-10-12 15:34:25 +0000 | [diff] [blame] | 84 | dst_fd = bb_xopen(bb_dev_null, O_WRONLY); /* why does test use filenum 2 ? */ |
Glenn L McGrath | ab77806 | 2004-01-05 12:35:05 +0000 | [diff] [blame] | 85 | } else if (opt & GUNZIP_OPT_STDOUT) { |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 86 | dst_fd = STDOUT_FILENO; |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 87 | } else { |
| 88 | char *extension; |
| 89 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 90 | new_path = bb_xstrdup(old_path); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 91 | |
| 92 | extension = strrchr(new_path, '.'); |
Glenn L McGrath | bf1cc8b | 2002-11-01 23:38:54 +0000 | [diff] [blame] | 93 | #ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS |
Glenn L McGrath | 9ef0944 | 2002-11-01 22:11:53 +0000 | [diff] [blame] | 94 | if (extension && (strcmp(extension, ".Z") == 0)) { |
| 95 | *extension = '\0'; |
| 96 | } else |
| 97 | #endif |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 98 | if (extension && (strcmp(extension, ".gz") == 0)) { |
| 99 | *extension = '\0'; |
| 100 | } else if (extension && (strcmp(extension, ".tgz") == 0)) { |
| 101 | extension[2] = 'a'; |
| 102 | extension[3] = 'r'; |
| 103 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 104 | bb_error_msg_and_die("Invalid extension"); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 105 | } |
| 106 | |
Rob Landley | 9a202c9 | 2006-06-13 14:54:42 +0000 | [diff] [blame] | 107 | /* Open output file (with correct permissions) */ |
| 108 | dst_fd = bb_xopen3(new_path, O_WRONLY | O_CREAT, stat_buf.st_mode); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 109 | |
| 110 | /* If unzip succeeds remove the old file */ |
| 111 | delete_path = old_path; |
| 112 | } |
| 113 | |
| 114 | /* do the decompression, and cleanup */ |
Rob Landley | 5343747 | 2006-07-16 08:14:35 +0000 | [diff] [blame] | 115 | if (xread_char(src_fd) == 0x1f) { |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 116 | unsigned char magic2; |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 117 | |
Rob Landley | 5343747 | 2006-07-16 08:14:35 +0000 | [diff] [blame] | 118 | magic2 = xread_char(src_fd); |
Glenn L McGrath | bf1cc8b | 2002-11-01 23:38:54 +0000 | [diff] [blame] | 119 | #ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 120 | if (magic2 == 0x9d) { |
Glenn L McGrath | 9ef0944 | 2002-11-01 22:11:53 +0000 | [diff] [blame] | 121 | status = uncompress(src_fd, dst_fd); |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 122 | } else |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 123 | #endif |
| 124 | if (magic2 == 0x8b) { |
| 125 | check_header_gzip(src_fd); |
Glenn L McGrath | 5699b85 | 2003-11-15 23:19:05 +0000 | [diff] [blame] | 126 | status = inflate_gunzip(src_fd, dst_fd); |
Glenn L McGrath | 5c99581 | 2002-09-30 05:30:29 +0000 | [diff] [blame] | 127 | if (status != 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 128 | bb_error_msg_and_die("Error inflating"); |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 129 | } |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 130 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 131 | bb_error_msg_and_die("Invalid magic"); |
Glenn L McGrath | 2e41d0c | 2002-09-27 06:46:02 +0000 | [diff] [blame] | 132 | } |
Glenn L McGrath | 563ac6e | 2002-11-01 21:40:52 +0000 | [diff] [blame] | 133 | } else { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 134 | bb_error_msg_and_die("Invalid magic"); |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 135 | } |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 136 | |
| 137 | if ((status != EXIT_SUCCESS) && (new_path)) { |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 138 | /* Unzip failed, remove new path instead of old path */ |
| 139 | delete_path = new_path; |
| 140 | } |
| 141 | |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 142 | if (dst_fd != STDOUT_FILENO) { |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 143 | close(dst_fd); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 144 | } |
Eric Andersen | 70060d2 | 2004-03-27 10:02:48 +0000 | [diff] [blame] | 145 | if (src_fd != STDIN_FILENO) { |
Glenn L McGrath | 7ca04f3 | 2002-09-25 02:47:48 +0000 | [diff] [blame] | 146 | close(src_fd); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 147 | } |
| 148 | |
| 149 | /* delete_path will be NULL if in test mode or from stdin */ |
| 150 | if (delete_path && (unlink(delete_path) == -1)) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 151 | bb_error_msg_and_die("Couldn't remove %s", delete_path); |
Glenn L McGrath | abac53b | 2002-08-24 14:32:17 +0000 | [diff] [blame] | 152 | } |
| 153 | |
| 154 | free(new_path); |
| 155 | |
| 156 | } while (optind < argc); |
| 157 | |
Matt Kraai | a4a65e7 | 2002-04-29 15:32:32 +0000 | [diff] [blame] | 158 | return status; |
Eric Andersen | b052b47 | 1999-11-18 00:21:59 +0000 | [diff] [blame] | 159 | } |