1 /* gzip.c - gzip/gunzip/zcat
2 *
3 * Copyright 2017 The Android Open Source Project
4 *
5 * GZIP RFC: http://www.ietf.org/rfc/rfc1952.txt
6 *
7 * todo: qtv --rsyncable
8
9 // gzip.net version allows all options for all commands.
10 USE_GZIP(NEWTOY(gzip, "cdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
11 USE_GUNZIP(NEWTOY(gunzip, "cdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
12 USE_ZCAT(NEWTOY(zcat, "cdfk123456789[-123456789]", TOYFLAG_USR|TOYFLAG_BIN))
13
14 config GZIP
15 bool "gzip"
16 default y
17 depends on TOYBOX_LIBZ
18 help
19 usage: gzip [-19cdfk] [FILE...]
20
21 Compress files. With no files, compresses stdin to stdout.
22 On success, the input files are removed and replaced by new
23 files with the .gz suffix.
24
25 -c Output to stdout
26 -d Decompress (act as gunzip)
27 -f Force: allow overwrite of output file
28 -k Keep input files (default is to remove)
29 -# Compression level 1-9 (1:fastest, 6:default, 9:best)
30
31 config GUNZIP
32 bool "gunzip"
33 default y
34 depends on TOYBOX_LIBZ
35 help
36 usage: gunzip [-cfk] [FILE...]
37
38 Decompress files. With no files, decompresses stdin to stdout.
39 On success, the input files are removed and replaced by new
40 files without the .gz suffix.
41
42 -c Output to stdout (act as zcat)
43 -f Force: allow read from tty
44 -k Keep input files (default is to remove)
45
46 config ZCAT
47 bool "zcat"
48 default y
49 depends on TOYBOX_LIBZ
50 help
51 usage: zcat [FILE...]
52
53 Decompress files to stdout. Like `gzip -dc`.
54
55 -f Force: allow read from tty
56 */
57
58 #define FORCE_FLAGS
59 #define FOR_gzip
60 #include "toys.h"
61
62 #include <zlib.h>
63
GLOBALS(int level;)64 GLOBALS(
65 int level;
66 )
67
68 static void fix_time(const char *path, struct stat *sb)
69 {
70 struct timespec times[] = { sb->st_atim, sb->st_mtim };
71
72 if (utimensat(AT_FDCWD, path, times, 0)) perror_exit("utimensat");
73 }
74
do_zlib(int in_fd,int out_fd)75 static int do_zlib(int in_fd, int out_fd)
76 {
77 int len, err = 0, dd = toys.optflags&FLAG_d;
78 char *b = "r";
79 gzFile gz;
80
81 if (!dd) {
82 sprintf(b = toybuf, "w%d", TT.level);
83 if (out_fd == 1) out_fd = xdup(out_fd);
84 }
85 if (!(gz = gzdopen(dd ? in_fd : out_fd, b))) perror_exit("gzdopen");
86 if (dd) {
87 while ((len = gzread(gz, toybuf, sizeof(toybuf))) > 0)
88 if (len != writeall(out_fd, toybuf, len)) break;
89 } else {
90 while ((len = read(in_fd, toybuf, sizeof(toybuf))) > 0)
91 if (len != gzwrite(gz, toybuf, len)) break;
92 }
93
94 err = !!len;
95 if (len>0 || err == Z_ERRNO) perror_msg(dd ? "write" : "read");
96 if (len<0)
97 error_msg("%s%s: %s", "gz", dd ? "read" : "write", gzerror(gz, &len));
98
99 if (gzclose(gz) != Z_OK) perror_msg("gzclose"), err++;
100
101 return err;
102 }
103
do_gzip(int in_fd,char * arg)104 static void do_gzip(int in_fd, char *arg)
105 {
106 struct stat sb;
107 int len, out_fd = 0;
108 char *out_name = 0;
109
110 // Are we writing to stdout?
111 if (!in_fd || (toys.optflags&FLAG_c)) out_fd = 1;
112 if (isatty(in_fd)) {
113 if (!(toys.optflags&FLAG_f)) {
114 error_msg("%s:need -f to read TTY"+3*!!in_fd, arg);
115 return;
116 } else out_fd = 1;
117 }
118
119 // Are we reading file.gz to write to file?
120 if (!out_fd) {
121 if (fstat(in_fd, &sb)) {
122 perror_msg("%s", arg);
123 return;
124 }
125
126 if (!(toys.optflags&FLAG_d)) out_name = xmprintf("%s%s", arg, ".gz");
127 else {
128 // "gunzip x.gz" will decompress "x.gz" to "x".
129 if ((len = strlen(arg))<4 || strcmp(arg+len-3, ".gz")) {
130 error_msg("no .gz: %s", arg);
131 return;
132 }
133 out_name = xstrdup(arg);
134 out_name[len-3] = 0;
135 }
136
137 out_fd = xcreate(out_name,
138 O_CREAT|O_WRONLY|WARN_ONLY|(O_EXCL*!(toys.optflags&FLAG_f)), sb.st_mode);
139 if (out_fd == -1) return;
140 }
141
142 // if (CFG_TOYBOX_LIBZ)
143 if (do_zlib(in_fd, out_fd) && out_name) arg = out_name;
144 if (out_fd != 1) close(out_fd);
145
146 if (out_name) {
147 fix_time(out_name, &sb);
148 if (!(toys.optflags&FLAG_k)) if (unlink(arg)) perror_msg("unlink %s", arg);
149 free(out_name);
150 }
151 }
152
gzip_main(void)153 void gzip_main(void)
154 {
155 for (TT.level = 0; TT.level<9; TT.level++)
156 if ((toys.optflags>>TT.level)&1) break;
157 if (!(TT.level = 9-TT.level)) TT.level = 6;
158
159 loopfiles(toys.optargs, do_gzip);
160 }
161
gunzip_main(void)162 void gunzip_main(void)
163 {
164 toys.optflags |= FLAG_d;
165 gzip_main();
166 }
167
zcat_main(void)168 void zcat_main(void)
169 {
170 toys.optflags |= (FLAG_c|FLAG_d);
171 gzip_main();
172 }
173