blob: bd6047e135d9e60b6e3d398ecbe8535f26eac63d [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Erik Andersen61677fe2000-04-13 01:18:56 +00002/*
3 * Gzip implementation for busybox
4 *
Glenn L McGrathc2bf5ca2000-09-29 06:46:59 +00005 * Based on GNU gzip v1.2.4 Copyright (C) 1992-1993 Jean-loup Gailly.
Erik Andersen61677fe2000-04-13 01:18:56 +00006 *
7 * Originally adjusted for busybox by Sven Rudolph <sr1@inf.tu-dresden.de>
8 * based on gzip sources
9 *
Eric Andersencb81e642003-07-14 21:21:08 +000010 * Adjusted further by Erik Andersen <andersen@codepoet.org> to support files as
Eric Andersen50e4d662002-04-06 05:15:46 +000011 * well as stdin/stdout, and to generally behave itself wrt command line
12 * handling.
Erik Andersen61677fe2000-04-13 01:18:56 +000013 *
Glenn L McGrath58e42d52001-03-28 05:38:24 +000014 * General cleanup to better adhere to the style guide and make use of standard
Glenn L McGrathc6992fe2004-04-25 05:11:19 +000015 * busybox functions by Glenn McGrath <bug1@iinet.net.au>
Eric Andersenc7bda1c2004-03-15 08:29:22 +000016 *
"Robert P. J. Day"801ab142006-07-12 07:56:04 +000017 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
Glenn L McGrathc2bf5ca2000-09-29 06:46:59 +000018 *
19 * gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
Eric Andersenb052b471999-11-18 00:21:59 +000020 * 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 McGrathc2bf5ca2000-09-29 06:46:59 +000030#include <stdlib.h>
Glenn L McGrath7fd92942001-04-11 03:11:33 +000031#include <string.h>
32#include <unistd.h>
Glenn L McGrath7ca04f32002-09-25 02:47:48 +000033#include <sys/types.h>
34#include <sys/stat.h>
35#include <fcntl.h>
36
Eric Andersencbe31da2001-02-20 06:14:08 +000037#include "busybox.h"
Glenn L McGrath7ca04f32002-09-25 02:47:48 +000038#include "unarchive.h"
Eric Andersenb052b471999-11-18 00:21:59 +000039
Glenn L McGrathab778062004-01-05 12:35:05 +000040#define GUNZIP_OPT_STDOUT 1
41#define GUNZIP_OPT_FORCE 2
42#define GUNZIP_OPT_TEST 4
43#define GUNZIP_OPT_DECOMPRESS 8
Matt Kraaia4a65e72002-04-29 15:32:32 +000044
Rob Landleydfba7412006-03-06 20:47:33 +000045int gunzip_main(int argc, char **argv)
Glenn L McGrath58e42d52001-03-28 05:38:24 +000046{
Glenn L McGrathabac53b2002-08-24 14:32:17 +000047 char status = EXIT_SUCCESS;
Glenn L McGrathab778062004-01-05 12:35:05 +000048 unsigned long opt;
Glenn L McGrath58e42d52001-03-28 05:38:24 +000049
Glenn L McGrathab778062004-01-05 12:35:05 +000050 opt = bb_getopt_ulflags(argc, argv, "cftd");
Glenn L McGrath58e42d52001-03-28 05:38:24 +000051 /* if called as zcat */
Manuel Novoa III cad53642003-03-19 09:13:01 +000052 if (strcmp(bb_applet_name, "zcat") == 0) {
Glenn L McGrathab778062004-01-05 12:35:05 +000053 opt |= GUNZIP_OPT_STDOUT;
Glenn L McGrathbcfeb2a2001-04-18 13:34:09 +000054 }
55
Glenn L McGrathabac53b2002-08-24 14:32:17 +000056 do {
Glenn L McGrathabac53b2002-08-24 14:32:17 +000057 struct stat stat_buf;
58 const char *old_path = argv[optind];
59 const char *delete_path = NULL;
60 char *new_path = NULL;
Glenn L McGrath7ca04f32002-09-25 02:47:48 +000061 int src_fd;
62 int dst_fd;
Glenn L McGrathabac53b2002-08-24 14:32:17 +000063
64 optind++;
65
66 if (old_path == NULL || strcmp(old_path, "-") == 0) {
Eric Andersen70060d22004-03-27 10:02:48 +000067 src_fd = STDIN_FILENO;
Glenn L McGrathab778062004-01-05 12:35:05 +000068 opt |= GUNZIP_OPT_STDOUT;
Glenn L McGrathabac53b2002-08-24 14:32:17 +000069 } else {
Manuel Novoa III cad53642003-03-19 09:13:01 +000070 src_fd = bb_xopen(old_path, O_RDONLY);
Glenn L McGrathabac53b2002-08-24 14:32:17 +000071
72 /* Get the time stamp on the input file. */
Rob Landleyc5b1d4d2006-03-13 15:45:16 +000073 xstat(old_path, &stat_buf);
Glenn L McGrath1ee52e82002-08-24 10:30:36 +000074 }
Glenn L McGrathabac53b2002-08-24 14:32:17 +000075
76 /* Check that the input is sane. */
Glenn L McGrathab778062004-01-05 12:35:05 +000077 if (isatty(src_fd) && ((opt & GUNZIP_OPT_FORCE) == 0)) {
Manuel Novoa III cad53642003-03-19 09:13:01 +000078 bb_error_msg_and_die
Glenn L McGrathabac53b2002-08-24 14:32:17 +000079 ("compressed data not read from terminal. Use -f to force it.");
80 }
81
82 /* Set output filename and number */
Glenn L McGrathab778062004-01-05 12:35:05 +000083 if (opt & GUNZIP_OPT_TEST) {
"Vladimir N. Oleynik"6c35c7c2005-10-12 15:34:25 +000084 dst_fd = bb_xopen(bb_dev_null, O_WRONLY); /* why does test use filenum 2 ? */
Glenn L McGrathab778062004-01-05 12:35:05 +000085 } else if (opt & GUNZIP_OPT_STDOUT) {
Eric Andersen70060d22004-03-27 10:02:48 +000086 dst_fd = STDOUT_FILENO;
Glenn L McGrathabac53b2002-08-24 14:32:17 +000087 } else {
88 char *extension;
89
Manuel Novoa III cad53642003-03-19 09:13:01 +000090 new_path = bb_xstrdup(old_path);
Glenn L McGrathabac53b2002-08-24 14:32:17 +000091
92 extension = strrchr(new_path, '.');
Glenn L McGrathbf1cc8b2002-11-01 23:38:54 +000093#ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS
Glenn L McGrath9ef09442002-11-01 22:11:53 +000094 if (extension && (strcmp(extension, ".Z") == 0)) {
95 *extension = '\0';
96 } else
97#endif
Glenn L McGrathabac53b2002-08-24 14:32:17 +000098 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 cad53642003-03-19 09:13:01 +0000104 bb_error_msg_and_die("Invalid extension");
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000105 }
106
Rob Landley9a202c92006-06-13 14:54:42 +0000107 /* Open output file (with correct permissions) */
108 dst_fd = bb_xopen3(new_path, O_WRONLY | O_CREAT, stat_buf.st_mode);
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000109
110 /* If unzip succeeds remove the old file */
111 delete_path = old_path;
112 }
113
114 /* do the decompression, and cleanup */
Rob Landley53437472006-07-16 08:14:35 +0000115 if (xread_char(src_fd) == 0x1f) {
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000116 unsigned char magic2;
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000117
Rob Landley53437472006-07-16 08:14:35 +0000118 magic2 = xread_char(src_fd);
Glenn L McGrathbf1cc8b2002-11-01 23:38:54 +0000119#ifdef CONFIG_FEATURE_GUNZIP_UNCOMPRESS
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000120 if (magic2 == 0x9d) {
Glenn L McGrath9ef09442002-11-01 22:11:53 +0000121 status = uncompress(src_fd, dst_fd);
Eric Andersenc7bda1c2004-03-15 08:29:22 +0000122 } else
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000123#endif
124 if (magic2 == 0x8b) {
125 check_header_gzip(src_fd);
Glenn L McGrath5699b852003-11-15 23:19:05 +0000126 status = inflate_gunzip(src_fd, dst_fd);
Glenn L McGrath5c995812002-09-30 05:30:29 +0000127 if (status != 0) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000128 bb_error_msg_and_die("Error inflating");
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000129 }
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000130 } else {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000131 bb_error_msg_and_die("Invalid magic");
Glenn L McGrath2e41d0c2002-09-27 06:46:02 +0000132 }
Glenn L McGrath563ac6e2002-11-01 21:40:52 +0000133 } else {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000134 bb_error_msg_and_die("Invalid magic");
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000135 }
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000136
137 if ((status != EXIT_SUCCESS) && (new_path)) {
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000138 /* Unzip failed, remove new path instead of old path */
139 delete_path = new_path;
140 }
141
Eric Andersen70060d22004-03-27 10:02:48 +0000142 if (dst_fd != STDOUT_FILENO) {
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000143 close(dst_fd);
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000144 }
Eric Andersen70060d22004-03-27 10:02:48 +0000145 if (src_fd != STDIN_FILENO) {
Glenn L McGrath7ca04f32002-09-25 02:47:48 +0000146 close(src_fd);
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000147 }
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 cad53642003-03-19 09:13:01 +0000151 bb_error_msg_and_die("Couldn't remove %s", delete_path);
Glenn L McGrathabac53b2002-08-24 14:32:17 +0000152 }
153
154 free(new_path);
155
156 } while (optind < argc);
157
Matt Kraaia4a65e72002-04-29 15:32:32 +0000158 return status;
Eric Andersenb052b471999-11-18 00:21:59 +0000159}