123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174 |
- /* gzip.c - gzip/gunzip/zcat
- *
- * Copyright 2017 The Android Open Source Project
- *
- * See http://refspecs.linuxfoundation.org/LSB_4.1.0/LSB-Core-generic/LSB-Core-generic/gzip.html
- * GZIP RFC: http://www.ietf.org/rfc/rfc1952.txt
- *
- * todo: qtv --rsyncable
- // gzip.net version allows all options for all commands.
- USE_GZIP(NEWTOY(gzip, "ncdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
- USE_GUNZIP(NEWTOY(gunzip, "cdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
- USE_ZCAT(NEWTOY(zcat, "cdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
- config GZIP
- bool "gzip"
- default n
- help
- usage: gzip [-19cdfk] [FILE...]
- Compress files. With no files, compresses stdin to stdout.
- On success, the input files are removed and replaced by new
- files with the .gz suffix.
- -c Output to stdout
- -d Decompress (act as gunzip)
- -f Force: allow overwrite of output file
- -k Keep input files (default is to remove)
- -# Compression level 1-9 (1:fastest, 6:default, 9:best)
- config GUNZIP
- bool "gunzip"
- default y
- help
- usage: gunzip [-cfk] [FILE...]
- Decompress files. With no files, decompresses stdin to stdout.
- On success, the input files are removed and replaced by new
- files without the .gz suffix.
- -c Output to stdout (act as zcat)
- -f Force: allow read from tty
- -k Keep input files (default is to remove)
- config ZCAT
- bool "zcat"
- default y
- help
- usage: zcat [FILE...]
- Decompress files to stdout. Like `gzip -dc`.
- -f Force: allow read from tty
- */
- #define FORCE_FLAGS
- #define FOR_gzip
- #include "toys.h"
- GLOBALS(
- int level;
- )
- // Use assembly optimized zlib code?
- #if CFG_TOYBOX_LIBZ
- #include <zlib.h>
- // Read from in_fd, write to out_fd, decompress if dd else compress
- static int do_deflate(int in_fd, int out_fd, int dd, int level)
- {
- int len, err = 0;
- char *b = "r";
- gzFile gz;
- if (!dd) {
- sprintf(b = toybuf, "w%d", level);
- if (out_fd == 1) out_fd = xdup(out_fd);
- }
- if (!(gz = gzdopen(dd ? in_fd : out_fd, b))) perror_exit("gzdopen");
- if (dd) {
- if (gzdirect(gz)) error_exit("not gzip");
- while ((len = gzread(gz, toybuf, sizeof(toybuf))) > 0)
- if (len != writeall(out_fd, toybuf, len)) break;
- } else {
- while ((len = read(in_fd, toybuf, sizeof(toybuf))) > 0)
- if (len != gzwrite(gz, toybuf, len)) break;
- }
- err = !!len;
- if (len>0 || err == Z_ERRNO) perror_msg(dd ? "write" : "read");
- if (len<0)
- error_msg("%s%s: %s", "gz", dd ? "read" : "write", gzerror(gz, &len));
- if (gzclose(gz) != Z_OK) perror_msg("gzclose"), err++;
- return err;
- }
- // Use toybox's builtin lib/deflate.c
- #else
- // Read from in_fd, write to out_fd, decompress if dd else compress
- static int do_deflate(int in_fd, int out_fd, int dd, int level)
- {
- int x;
- if (dd) WOULD_EXIT(x, gunzip_fd(in_fd, out_fd));
- else WOULD_EXIT(x, gzip_fd(in_fd, out_fd));
- return x;
- }
- #endif
- static void do_gzip(int ifd, char *in)
- {
- struct stat sb;
- char *out = 0;
- int ofd = 0;
- // Are we writing to stdout?
- if (!ifd || FLAG(c)) ofd = 1;
- if (isatty(ifd)) {
- if (!FLAG(f)) return error_msg("%s:need -f to read TTY"+3*!!ifd, in);
- else ofd = 1;
- }
- // Are we reading file.gz to write to file?
- if (!ofd) {
- if (fstat(ifd, &sb)) return perror_msg("%s", in);
- // Add or remove .gz suffix as necessary
- if (!FLAG(d)) out = xmprintf("%s%s", in, ".gz");
- else if ((out = strend(in, ".gz"))>in) out = xstrndup(in, out-in);
- else return error_msg("no .gz: %s", in);
- ofd = xcreate(out, O_CREAT|O_WRONLY|WARN_ONLY|(O_EXCL*!FLAG(f)),sb.st_mode);
- if (ofd == -1) return;
- }
- if (do_deflate(ifd, ofd, FLAG(d), TT.level)) in = out;
- if (out) {
- struct timespec times[] = {sb.st_atim, sb.st_mtim};
- if (utimensat(AT_FDCWD, out, times, 0)) perror_exit("utimensat");
- if (chmod(out, sb.st_mode)) perror_exit("chmod");
- close(ofd);
- if (!FLAG(k) && in && unlink(in)) perror_msg("unlink %s", in);
- free(out);
- }
- }
- void gzip_main(void)
- {
- // This depends on 1-9 being at the end of the option list
- for (TT.level = 0; TT.level<9; TT.level++)
- if ((toys.optflags>>TT.level)&1) break;
- if (!(TT.level = 9-TT.level)) TT.level = 6;
- loopfiles(toys.optargs, do_gzip);
- }
- void gunzip_main(void)
- {
- toys.optflags |= FLAG_d;
- gzip_main();
- }
- void zcat_main(void)
- {
- toys.optflags |= (FLAG_c|FLAG_d);
- gzip_main();
- }
|