Reorganise, make it just one function, remove -v option it didnt work properly anyway, dont setvbuf it doesnt make any difference in performance.

This commit is contained in:
Glenn L McGrath 2002-08-24 14:32:17 +00:00
parent 1ee52e8b14
commit abac53b33c

View File

@ -66,40 +66,66 @@ 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;
const char gunzip_to_stdout = 1;
const char gunzip_force = 2;
const char gunzip_test = 4;
static int gunzip_file(const char *path, int flags)
extern int gunzip_main(int argc, char **argv)
{
char status = EXIT_SUCCESS;
char flags = 0;
int opt;
/* 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 */
}
}
do {
FILE *in_file, *out_file;
struct stat stat_buf;
const char *old_path = argv[optind];
const char *delete_path = NULL;
char *out_path = NULL;
char *new_path = NULL;
if (path == NULL || strcmp(path, "-") == 0) {
optind++;
if (old_path == NULL || strcmp(old_path, "-") == 0) {
in_file = stdin;
flags |= gunzip_to_stdout;
} else {
if ((in_file = wfopen(path, "r")) == NULL) {
return -1;
in_file = wfopen(old_path, "r");
if (in_file == NULL) {
status = EXIT_FAILURE;
break;
}
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);
if (stat(old_path, &stat_buf) < 0) {
error_msg_and_die("Couldn't stat file %s", old_path);
}
}
/* Check that the input is sane. */
if (isatty(fileno(in_file)) && (flags & gunzip_force) == 0) {
if (isatty(fileno(in_file)) && ((flags & gunzip_force) == 0)) {
error_msg_and_die
("compressed data not read from terminal. Use -f to force it.");
}
@ -111,37 +137,33 @@ static int gunzip_file(const char *path, int flags)
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;
new_path = xstrdup(old_path);
extension = strrchr(new_path, '.');
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");
}
out_path = xstrndup(path, length);
/* Open output file */
out_file = xfopen(out_path, "w");
out_file = xfopen(new_path, "w");
/* Set permissions on the file */
chmod(out_path, stat_buf.st_mode);
chmod(old_path, stat_buf.st_mode);
/* If unzip succeeds remove the old file */
delete_path = old_path;
}
/* do the decompression, and cleanup */
if (unzip(in_file, out_file) == 0) {
/* Success, remove .gz file */
if (!(flags & gunzip_to_stdout)) {
delete_path = path;
}
if (flags & gunzip_verbose) {
fprintf(stderr, "OK\n");
}
} else {
/* remove failed attempt */
delete_path = out_path;
if ((unzip(in_file, out_file) != 0) && (new_path)) {
/* Unzip failed, remove new path instead of old path */
delete_path = new_path;
}
if (out_file != stdout) {
@ -151,63 +173,14 @@ static int gunzip_file(const char *path, int flags)
fclose(in_file);
}
if (delete_path && !(flags & gunzip_test)) {
if (unlink(delete_path) < 0) {
/* 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);
}
}
free(out_path);
free(new_path);
return 0;
}
} while (optind < argc);
extern int gunzip_main(int argc, char **argv)
{
int flags = 0;
int i, opt;
int status = EXIT_SUCCESS;
/* if called as zcat */
if (strcmp(applet_name, "zcat") == 0) {
flags |= gunzip_to_stdout;
}
while ((opt = getopt(argc, argv, "ctfhdqv")) != -1) {
switch (opt) {
case 'c':
flags |= gunzip_to_stdout;
break;
case 'f':
flags |= gunzip_force;
break;
case 't':
flags |= gunzip_test;
break;
case 'v':
flags |= gunzip_verbose;
break;
case 'd': /* Used to convert gzip to gunzip. */
break;
case 'q':
error_msg("-q option not supported, ignored");
break;
case 'h':
default:
show_usage(); /* exit's inside usage */
}
}
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;
}
}
}
return status;
}