aboutsummaryrefslogtreecommitdiff
path: root/toys/lsb
diff options
context:
space:
mode:
authorRob Landley <rob@landley.net>2019-04-21 19:47:16 -0500
committerRob Landley <rob@landley.net>2019-04-21 19:47:16 -0500
commit106d81abd8dacfc7fdbc2c94cbaafb0d18950649 (patch)
tree8f09f6da325abefdf98184421c3de49ef4754842 /toys/lsb
parent7964e1f78b58d9c365361cc36b0422d9d56cd204 (diff)
downloadtoybox-106d81abd8dacfc7fdbc2c94cbaafb0d18950649.tar.gz
Promote gzip/zcat but keep gunzip default n until lib/deflate.c finished.
Diffstat (limited to 'toys/lsb')
-rw-r--r--toys/lsb/gzip.c172
1 files changed, 172 insertions, 0 deletions
diff --git a/toys/lsb/gzip.c b/toys/lsb/gzip.c
new file mode 100644
index 00000000..7bd8d381
--- /dev/null
+++ b/toys/lsb/gzip.c
@@ -0,0 +1,172 @@
+/* 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, "cdfk123456789[-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 y
+ 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 n
+ 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) {
+ 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(out, ".gz"))>in) out = xstrndup(out, 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 (futimens(ofd, times)) perror_exit("utimensat");
+ 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();
+}