summaryrefslogtreecommitdiff
path: root/archival/gunzip.c
diff options
context:
space:
mode:
authorMatt Kraai2002-04-29 15:32:32 +0000
committerMatt Kraai2002-04-29 15:32:32 +0000
commita4a65e7f993b0ce70f235c76c4e5955581065a3a (patch)
tree213b101164633343c6ccf22bc40b8a5f3459b869 /archival/gunzip.c
parent051e4c15db08aa4e224c152a1406b3603bbeee5a (diff)
downloadbusybox-a4a65e7f993b0ce70f235c76c4e5955581065a3a.zip
busybox-a4a65e7f993b0ce70f235c76c4e5955581065a3a.tar.gz
* archival/gunzip.c (gunzip_file): New.
(gunzip_main): Handle no arguments correctly.
Diffstat (limited to 'archival/gunzip.c')
-rw-r--r--archival/gunzip.c198
1 files changed, 99 insertions, 99 deletions
diff --git a/archival/gunzip.c b/archival/gunzip.c
index e40982c..e47dd3f 100644
--- a/archival/gunzip.c
+++ b/archival/gunzip.c
@@ -66,18 +66,100 @@ static char *license_msg[] = {
#include <getopt.h>
#include "busybox.h"
+const int gunzip_to_stdout = 1;
+const int gunzip_force = 2;
+const int gunzip_test = 4;
+const int gunzip_verbose = 8;
+
+static int gunzip_file (const char *path, int flags)
+{
+ FILE *in_file, *out_file;
+ struct stat stat_buf;
+ const char *delete_path;
+ char *out_path = NULL;
+
+ if (path == NULL || strcmp (path, "-") == 0) {
+ in_file = stdin;
+ flags |= gunzip_to_stdout;
+ } else {
+ if ((in_file = wfopen(path, "r")) == NULL)
+ return -1;
+
+ if (flags & gunzip_verbose) {
+ fprintf(stderr, "%s:\t", path);
+ }
+
+ /* set the buffer size */
+ setvbuf(in_file, NULL, _IOFBF, 0x8000);
+
+ /* Get the time stamp on the input file. */
+ if (stat(path, &stat_buf) < 0) {
+ error_msg_and_die("Couldn't stat file %s", path);
+ }
+ }
+
+ /* Check that the input is sane. */
+ if (isatty(fileno(in_file)) && (flags & gunzip_force) == 0)
+ 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;
+ } else {
+ char *extension;
+ int length = strlen(path);
+
+ extension = strrchr(path, '.');
+ if (extension && strcmp(extension, ".gz") == 0) {
+ length -= 3;
+ } else if (extension && strcmp(extension, ".tgz") == 0) {
+ length -= 4;
+ } else {
+ error_msg_and_die("Invalid extension");
+ }
+ out_path = (char *) xcalloc(sizeof(char), length + 1);
+ strncpy(out_path, path, length);
+
+ /* Open output file */
+ out_file = xfopen(out_path, "w");
+
+ /* Set permissions on the file */
+ chmod(out_path, stat_buf.st_mode);
+ }
+
+ /* do the decompression, and cleanup */
+ if (unzip(in_file, out_file) == 0) {
+ /* Success, remove .gz file */
+ delete_path = path;
+ if (flags & gunzip_verbose) {
+ fprintf(stderr, "OK\n");
+ }
+ } else {
+ /* remove failed attempt */
+ delete_path = out_path;
+ }
+
+ fclose(out_file);
+ fclose(in_file);
+
+ if (delete_path && !(flags & gunzip_test)) {
+ if (unlink(delete_path) < 0) {
+ error_msg_and_die("Couldn't remove %s", delete_path);
+ }
+ }
+
+ free(out_path);
+
+ return 0;
+}
+
extern int gunzip_main(int argc, char **argv)
{
int flags = 0;
- int opt = 0;
- int delete_old_file, file_count;
- struct stat stat_buf;
- FILE *in_file, *out_file;
- char *if_name, *of_name, *delete_file_name;
- const int gunzip_to_stdout = 1;
- const int gunzip_force = 2;
- const int gunzip_test = 4;
- const int gunzip_verbose = 8;
+ int i, opt;
+ int status = EXIT_SUCCESS;
/* if called as zcat */
if (strcmp(applet_name, "zcat") == 0)
@@ -108,95 +190,13 @@ extern int gunzip_main(int argc, char **argv)
}
}
- file_count = argc - optind;
- while (file_count==0 || optind < argc) {
- in_file = stdin;
- out_file = NULL;
- if_name = NULL;
- of_name = NULL;
- delete_file_name = NULL;
- delete_old_file = FALSE;
-
- /* Set input filename and number */
- if (argv[optind] == NULL || strcmp(argv[optind], "-") == 0) {
- flags |= gunzip_to_stdout;
- /* Skip to the end */
- optind = argc;
- } else {
- if_name = strdup(argv[optind]);
- /* Open input file */
- in_file = xfopen(if_name, "r");
-
- if (flags & gunzip_verbose) {
- fprintf(stderr, "%s:\t", if_name);
- }
-
- /* set the buffer size */
- setvbuf(in_file, NULL, _IOFBF, 0x8000);
-
- /* Get the time stamp on the input file. */
- if (stat(if_name, &stat_buf) < 0) {
- error_msg_and_die("Couldn't stat file %s", if_name);
- }
- }
+ if (optind == argc) {
+ if (gunzip_file (NULL, flags) < 0)
+ status = EXIT_FAILURE;
+ } else
+ for (i = optind; i < argc; i++)
+ if (gunzip_file (argv[i], flags) < 0)
+ status = EXIT_FAILURE;
- /* Check that the input is sane. */
- if (isatty(fileno(in_file)) && (flags & gunzip_force) == 0)
- 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;
- } else {
- char *extension;
- int length = strlen(if_name);
-
- delete_old_file = TRUE;
- extension = strrchr(if_name, '.');
- if (extension && strcmp(extension, ".gz") == 0) {
- length -= 3;
- } else if (extension && strcmp(extension, ".tgz") == 0) {
- length -= 4;
- } else {
- error_msg_and_die("Invalid extension");
- }
- of_name = (char *) xcalloc(sizeof(char), length + 1);
- strncpy(of_name, if_name, length);
-
- /* Open output file */
- out_file = xfopen(of_name, "w");
-
- /* Set permissions on the file */
- chmod(of_name, stat_buf.st_mode);
- }
-
- /* do the decompression, and cleanup */
- if (unzip(in_file, out_file) == 0) {
- /* Success, remove .gz file */
- delete_file_name = if_name;
- if (flags & gunzip_verbose) {
- fprintf(stderr, "OK\n");
- }
- } else {
- /* remove failed attempt */
- delete_file_name = of_name;
- }
-
- fclose(out_file);
- fclose(in_file);
-
- if (delete_old_file == TRUE && !(flags & gunzip_test)) {
- if (unlink(delete_file_name) < 0) {
- error_msg_and_die("Couldnt remove %s", delete_file_name);
- }
- }
-
- free(of_name);
- optind++;
- } /* while () */
-
- return(EXIT_SUCCESS);
+ return status;
}
-