toaruos/apps/gunzip.c

99 lines
2.0 KiB
C

/* vim: tabstop=4 shiftwidth=4 noexpandtab
* This file is part of ToaruOS and is released under the terms
* of the NCSA / University of Illinois License - see LICENSE.md
* Copyright (C) 2020 K. Lange
*
* gunzip - decompress gzip-compressed payloads
*/
#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <toaru/inflate.h>
static uint8_t _get(struct inflate_context * ctx) {
return fgetc(ctx->input_priv);
}
static void _write(struct inflate_context * ctx, unsigned int sym) {
fputc(sym, ctx->output_priv);
}
static int usage(int argc, char * argv[]) {
fprintf(stderr,
"gunzip - decompress gzip-compressed payloads\n"
"\n"
"usage: %s [-c] name...\n"
"\n"
" -c \033[3mwrite to stdout; implies -k\033[0m\n"
" -k \033[3mkeep original files unchanged\033[0m\n"
"\n", argv[0]);
return 1;
}
int main(int argc, char * argv[]) {
int to_stdout = 0;
int keep = 0;
int opt;
while ((opt = getopt(argc, argv, "ck")) != -1) {
switch (opt) {
case 'c':
to_stdout = 1;
keep = 1;
break;
case 'k':
keep = 1;
break;
default:
fprintf(stderr, "%s: unrecognized option '%c'\n", argv[0], opt);
break;
}
}
if (optind >= argc) {
return usage(argc, argv);
}
FILE * f = fopen(argv[optind],"r");
if (!f) {
fprintf(stderr, "%s: %s: %s\n", argv[0], argv[optind], strerror(errno));
return 1;
}
if (!to_stdout && strcmp(&argv[optind][strlen(argv[optind])-3],".gz")) {
fprintf(stderr, "%s: %s: Don't know how to interpret file name\n", argv[0], argv[optind]);
return 1;
}
struct inflate_context ctx;
ctx.input_priv = f;
if (to_stdout) {
ctx.output_priv = stdout;
} else {
char * tmp = strdup(argv[optind]);
tmp[strlen(argv[optind])-3] = '\0';
ctx.output_priv = fopen(tmp,"w");
free(tmp);
}
ctx.get_input = _get;
ctx.write_output = _write;
ctx.ring = NULL; /* Use the global one */
gzip_decompress(&ctx);
if (!to_stdout) {
fclose(ctx.output_priv);
}
if (!keep) {
unlink(argv[optind]);
}
return 0;
}