gzip.c revision 1.6 1 1.6 mrg /* $NetBSD: gzip.c,v 1.6 2003/12/26 14:49:37 mrg Exp $ */
2 1.1 mrg
3 1.1 mrg /*
4 1.1 mrg * Copyright (c) 1997, 1998, 2003 Matthew R. Green
5 1.1 mrg * All rights reserved.
6 1.1 mrg *
7 1.1 mrg * Redistribution and use in source and binary forms, with or without
8 1.1 mrg * modification, are permitted provided that the following conditions
9 1.1 mrg * are met:
10 1.1 mrg * 1. Redistributions of source code must retain the above copyright
11 1.1 mrg * notice, this list of conditions and the following disclaimer.
12 1.1 mrg * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 mrg * notice, this list of conditions and the following disclaimer in the
14 1.1 mrg * documentation and/or other materials provided with the distribution.
15 1.1 mrg * 3. The name of the author may not be used to endorse or promote products
16 1.1 mrg * derived from this software without specific prior written permission.
17 1.1 mrg *
18 1.1 mrg * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 1.1 mrg * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 1.1 mrg * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 1.1 mrg * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 1.1 mrg * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
23 1.1 mrg * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
24 1.1 mrg * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
25 1.1 mrg * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
26 1.1 mrg * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 1.1 mrg * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 1.1 mrg * SUCH DAMAGE.
29 1.1 mrg */
30 1.1 mrg
31 1.4 mrg #include <sys/cdefs.h>
32 1.4 mrg #ifndef lint
33 1.4 mrg __COPYRIGHT("@(#) Copyright (c) 1997, 1998, 2003 Matthew R. Green\n\
34 1.4 mrg All rights reserved.\n");
35 1.6 mrg __RCSID("$NetBSD: gzip.c,v 1.6 2003/12/26 14:49:37 mrg Exp $");
36 1.4 mrg #endif /* not lint */
37 1.4 mrg
38 1.1 mrg /*
39 1.1 mrg * gzip.c -- GPL free gzip using zlib.
40 1.1 mrg *
41 1.6 mrg * very minor portions of this code are (very loosely) derived from
42 1.1 mrg * the minigzip.c in the zlib distribution.
43 1.6 mrg *
44 1.6 mrg * TODO:
45 1.6 mrg * - handle .taz/.tgz files?
46 1.1 mrg */
47 1.1 mrg
48 1.1 mrg #include <sys/param.h>
49 1.1 mrg #include <sys/stat.h>
50 1.1 mrg #include <sys/time.h>
51 1.1 mrg
52 1.1 mrg #include <unistd.h>
53 1.1 mrg #include <stdio.h>
54 1.1 mrg #include <string.h>
55 1.1 mrg #include <stdlib.h>
56 1.1 mrg #include <err.h>
57 1.1 mrg #include <errno.h>
58 1.1 mrg #include <fcntl.h>
59 1.1 mrg #include <zlib.h>
60 1.1 mrg #include <fts.h>
61 1.1 mrg #include <libgen.h>
62 1.1 mrg #include <stdarg.h>
63 1.1 mrg #include <getopt.h>
64 1.1 mrg
65 1.1 mrg #ifndef GZ_SUFFIX
66 1.1 mrg # define GZ_SUFFIX ".gz"
67 1.1 mrg #endif
68 1.1 mrg
69 1.1 mrg #define BUFLEN 4096
70 1.1 mrg
71 1.1 mrg #define ORIG_NAME 0x08
72 1.1 mrg
73 1.1 mrg /* Define this if you have the NetBSD gzopenfull(3) extension to zlib(3) */
74 1.1 mrg #define HAVE_ZLIB_GZOPENFULL 0
75 1.1 mrg
76 1.3 jdolecek static const char gzip_version[] = "NetBSD gzip 2.0";
77 1.1 mrg
78 1.3 jdolecek static char gzipflags[3]; /* `w' or `r', possible with [1-9] */
79 1.1 mrg static int cflag; /* stdout mode */
80 1.1 mrg static int dflag; /* decompress mode */
81 1.1 mrg static int fflag; /* force mode */
82 1.5 mrg static int lflag; /* list mode */
83 1.1 mrg static int nflag; /* don't save name/timestamp */
84 1.1 mrg static int Nflag; /* don't restore name/timestamp */
85 1.1 mrg static int qflag; /* quiet mode */
86 1.1 mrg static int rflag; /* recursive mode */
87 1.1 mrg static int tflag; /* test */
88 1.1 mrg static int vflag; /* verbose mode */
89 1.3 jdolecek static const char *Sflag = GZ_SUFFIX; /* suffix (.gz) */
90 1.1 mrg
91 1.4 mrg static int suffix_len; /* length of suffix; includes nul */
92 1.1 mrg static char *newfile; /* name of newly created file */
93 1.1 mrg static char *infile; /* name of file coming in */
94 1.1 mrg
95 1.1 mrg static void maybe_err(int rv, const char *fmt, ...);
96 1.1 mrg static void maybe_warn(const char *fmt, ...);
97 1.1 mrg static void maybe_warnx(const char *fmt, ...);
98 1.1 mrg static void usage(void);
99 1.1 mrg static void display_version(void);
100 1.1 mrg static void gz_compress(FILE *, gzFile);
101 1.1 mrg static off_t gz_uncompress(gzFile, FILE *);
102 1.1 mrg static void copymodes(const char *, struct stat *);
103 1.1 mrg static ssize_t file_compress(char *);
104 1.1 mrg static ssize_t file_uncompress(char *);
105 1.1 mrg static void handle_pathname(char *);
106 1.1 mrg static void handle_file(char *, struct stat *);
107 1.1 mrg static void handle_dir(char *, struct stat *);
108 1.1 mrg static void handle_stdin(void);
109 1.1 mrg static void handle_stdout(void);
110 1.6 mrg static void print_ratio(off_t, off_t, FILE *);
111 1.1 mrg static void print_verbage(char *, char *, ssize_t, ssize_t);
112 1.1 mrg static void print_test(char *, int);
113 1.6 mrg static void print_list(int fd, off_t, const char *);
114 1.1 mrg
115 1.1 mrg int main(int, char *p[]);
116 1.1 mrg
117 1.3 jdolecek static const struct option longopts[] = {
118 1.1 mrg { "stdout", no_argument, 0, 'c' },
119 1.1 mrg { "to-stdout", no_argument, 0, 'c' },
120 1.1 mrg { "decompress", no_argument, 0, 'd' },
121 1.1 mrg { "uncompress", no_argument, 0, 'd' },
122 1.1 mrg { "force", no_argument, 0, 'f' },
123 1.1 mrg { "help", no_argument, 0, 'h' },
124 1.5 mrg { "list", no_argument, 0, 'l' },
125 1.1 mrg { "no-name", no_argument, 0, 'n' },
126 1.1 mrg { "name", no_argument, 0, 'N' },
127 1.1 mrg { "quiet", no_argument, 0, 'q' },
128 1.1 mrg { "recursive", no_argument, 0, 'r' },
129 1.1 mrg { "suffix", required_argument, 0, 'S' },
130 1.1 mrg { "test", no_argument, 0, 't' },
131 1.1 mrg { "verbose", no_argument, 0, 'v' },
132 1.1 mrg { "version", no_argument, 0, 'V' },
133 1.1 mrg { "fast", no_argument, 0, '1' },
134 1.1 mrg { "best", no_argument, 0, '9' },
135 1.1 mrg #if 0
136 1.1 mrg /*
137 1.1 mrg * This is what else GNU gzip implements. Maybe --list is
138 1.1 mrg * useful, but --ascii isn't useful on NetBSD, and I don't
139 1.1 mrg * care to have a --license.
140 1.1 mrg */
141 1.1 mrg { "ascii", no_argument, 0, 'a' },
142 1.1 mrg { "license", no_argument, 0, 'L' },
143 1.1 mrg #endif
144 1.1 mrg };
145 1.1 mrg
146 1.1 mrg int
147 1.1 mrg main(int argc, char **argv)
148 1.1 mrg {
149 1.1 mrg const char *progname = getprogname();
150 1.1 mrg int ch;
151 1.1 mrg
152 1.1 mrg gzipflags[0] = 'w';
153 1.1 mrg gzipflags[1] = '\0';
154 1.1 mrg
155 1.1 mrg /*
156 1.1 mrg * XXX
157 1.1 mrg * handle being called `gunzip', `zcat' and `gzcat'
158 1.1 mrg */
159 1.1 mrg if (strcmp(progname, "gunzip") == 0)
160 1.1 mrg dflag = 1;
161 1.1 mrg else if (strcmp(progname, "zcat") == 0 ||
162 1.1 mrg strcmp(progname, "gzcat") == 0)
163 1.1 mrg dflag = cflag = 1;
164 1.1 mrg
165 1.1 mrg while ((ch = getopt_long(argc, argv, "cdfhHlnNqrS:tvV123456789",
166 1.1 mrg longopts, NULL)) != -1)
167 1.1 mrg switch (ch) {
168 1.1 mrg case 'c':
169 1.1 mrg cflag = 1;
170 1.1 mrg break;
171 1.1 mrg case 'd':
172 1.1 mrg dflag = 1;
173 1.1 mrg break;
174 1.1 mrg case 'f':
175 1.1 mrg fflag = 1;
176 1.1 mrg break;
177 1.1 mrg case 'h':
178 1.1 mrg case 'H':
179 1.1 mrg usage();
180 1.1 mrg /* NOTREACHED */
181 1.5 mrg case 'l':
182 1.5 mrg lflag = 1;
183 1.5 mrg tflag = 1;
184 1.5 mrg dflag = 1;
185 1.5 mrg break;
186 1.1 mrg case 'n':
187 1.1 mrg nflag = 1;
188 1.1 mrg Nflag = 0;
189 1.1 mrg break;
190 1.1 mrg case 'N':
191 1.1 mrg nflag = 0;
192 1.1 mrg Nflag = 1;
193 1.1 mrg break;
194 1.1 mrg case 'q':
195 1.1 mrg qflag = 1;
196 1.1 mrg break;
197 1.1 mrg case 'r':
198 1.1 mrg rflag = 1;
199 1.1 mrg break;
200 1.1 mrg case 'S':
201 1.1 mrg Sflag = optarg;
202 1.1 mrg break;
203 1.1 mrg case 't':
204 1.1 mrg cflag = 1;
205 1.1 mrg tflag = 1;
206 1.1 mrg dflag = 1;
207 1.1 mrg break;
208 1.1 mrg case 'v':
209 1.1 mrg vflag = 1;
210 1.1 mrg break;
211 1.1 mrg case 'V':
212 1.1 mrg display_version();
213 1.1 mrg /* NOTREACHED */
214 1.1 mrg case '1': case '2': case '3':
215 1.1 mrg case '4': case '5': case '6':
216 1.1 mrg case '7': case '8': case '9':
217 1.1 mrg gzipflags[1] = (char)ch;
218 1.1 mrg gzipflags[2] = '\0';
219 1.1 mrg break;
220 1.1 mrg }
221 1.1 mrg argv += optind;
222 1.1 mrg argc -= optind;
223 1.1 mrg if (dflag)
224 1.1 mrg gzipflags[0] = 'r';
225 1.1 mrg
226 1.5 mrg if (lflag && vflag)
227 1.5 mrg errx(1, "does not support --list and --verbose together yet");
228 1.5 mrg
229 1.4 mrg suffix_len = strlen(Sflag) + 1;
230 1.4 mrg
231 1.1 mrg if (argc == 0) {
232 1.1 mrg if (dflag) /* stdin mode */
233 1.1 mrg handle_stdin();
234 1.1 mrg else /* stdout mode */
235 1.1 mrg handle_stdout();
236 1.1 mrg } else {
237 1.1 mrg do {
238 1.1 mrg handle_pathname(argv[0]);
239 1.5 mrg } while (*++argv);
240 1.1 mrg }
241 1.6 mrg if (qflag == 0 && lflag && argc > 1)
242 1.6 mrg print_list(-1, 0, "(totals)");
243 1.1 mrg exit(0);
244 1.1 mrg }
245 1.1 mrg
246 1.1 mrg /* maybe print a warning */
247 1.1 mrg void
248 1.1 mrg maybe_warn(const char *fmt, ...)
249 1.1 mrg {
250 1.1 mrg va_list ap;
251 1.1 mrg
252 1.1 mrg if (qflag == 0) {
253 1.1 mrg va_start(ap, fmt);
254 1.1 mrg vwarn(fmt, ap);
255 1.1 mrg va_end(ap);
256 1.1 mrg }
257 1.1 mrg }
258 1.1 mrg
259 1.1 mrg void
260 1.1 mrg maybe_warnx(const char *fmt, ...)
261 1.1 mrg {
262 1.1 mrg va_list ap;
263 1.1 mrg
264 1.1 mrg if (qflag == 0) {
265 1.1 mrg va_start(ap, fmt);
266 1.1 mrg vwarnx(fmt, ap);
267 1.1 mrg va_end(ap);
268 1.1 mrg }
269 1.1 mrg }
270 1.1 mrg
271 1.1 mrg /* maybe print a warning */
272 1.1 mrg void
273 1.1 mrg maybe_err(int rv, const char *fmt, ...)
274 1.1 mrg {
275 1.1 mrg va_list ap;
276 1.1 mrg
277 1.1 mrg if (qflag == 0) {
278 1.1 mrg va_start(ap, fmt);
279 1.1 mrg vwarn(fmt, ap);
280 1.1 mrg va_end(ap);
281 1.1 mrg }
282 1.1 mrg exit(rv);
283 1.1 mrg }
284 1.1 mrg
285 1.1 mrg /* compress input to output then close both files */
286 1.1 mrg static void
287 1.1 mrg gz_compress(FILE *in, gzFile out)
288 1.1 mrg {
289 1.1 mrg char buf[BUFLEN];
290 1.1 mrg ssize_t len;
291 1.1 mrg int i;
292 1.1 mrg
293 1.1 mrg for (;;) {
294 1.1 mrg len = fread(buf, 1, sizeof(buf), in);
295 1.1 mrg if (ferror(in))
296 1.1 mrg maybe_err(1, "fread");
297 1.1 mrg if (len == 0)
298 1.1 mrg break;
299 1.1 mrg
300 1.1 mrg if ((ssize_t)gzwrite(out, buf, len) != len)
301 1.1 mrg maybe_err(1, gzerror(out, &i));
302 1.1 mrg }
303 1.1 mrg if (fclose(in) < 0)
304 1.1 mrg maybe_err(1, "failed fclose");
305 1.1 mrg if (gzclose(out) != Z_OK)
306 1.1 mrg maybe_err(1, "failed gzclose");
307 1.1 mrg }
308 1.1 mrg
309 1.1 mrg /* uncompress input to output then close the input */
310 1.1 mrg static off_t
311 1.1 mrg gz_uncompress(gzFile in, FILE *out)
312 1.1 mrg {
313 1.1 mrg char buf[BUFLEN];
314 1.1 mrg off_t size;
315 1.1 mrg ssize_t len;
316 1.1 mrg int i;
317 1.1 mrg
318 1.1 mrg for (size = 0;;) {
319 1.1 mrg len = gzread(in, buf, sizeof(buf));
320 1.1 mrg
321 1.1 mrg if (len < 0) {
322 1.1 mrg if (tflag) {
323 1.1 mrg print_test(infile, 0);
324 1.1 mrg return (0);
325 1.1 mrg } else
326 1.1 mrg maybe_err(1, gzerror(in, &i));
327 1.1 mrg } else if (len == 0) {
328 1.1 mrg if (tflag)
329 1.1 mrg print_test(infile, 1);
330 1.1 mrg break;
331 1.1 mrg }
332 1.1 mrg
333 1.1 mrg size += len;
334 1.1 mrg
335 1.1 mrg /* don't write anything with -t */
336 1.1 mrg if (tflag)
337 1.1 mrg continue;
338 1.1 mrg
339 1.1 mrg if (fwrite(buf, 1, (unsigned)len, out) != (ssize_t)len)
340 1.1 mrg maybe_err(1, "failed fwrite");
341 1.1 mrg }
342 1.1 mrg if (gzclose(in) != Z_OK)
343 1.1 mrg maybe_err(1, "failed gzclose");
344 1.1 mrg
345 1.1 mrg return (size);
346 1.1 mrg }
347 1.1 mrg
348 1.1 mrg /*
349 1.1 mrg * set the owner, mode, flags & utimes for a file
350 1.1 mrg */
351 1.1 mrg static void
352 1.1 mrg copymodes(const char *file, struct stat *sbp)
353 1.1 mrg {
354 1.1 mrg struct timeval times[2];
355 1.1 mrg
356 1.1 mrg /*
357 1.1 mrg * If we have no info on the input, give this file some
358 1.1 mrg * default values and return..
359 1.1 mrg */
360 1.1 mrg if (sbp == NULL) {
361 1.1 mrg mode_t mask = umask(022);
362 1.1 mrg
363 1.1 mrg (void)chmod(file, DEFFILEMODE & ~mask);
364 1.1 mrg (void)umask(mask);
365 1.1 mrg return;
366 1.1 mrg }
367 1.1 mrg
368 1.1 mrg /* if the chown fails, remove set-id bits as-per compress(1) */
369 1.1 mrg if (chown(file, sbp->st_uid, sbp->st_gid) < 0) {
370 1.1 mrg if (errno != EPERM)
371 1.1 mrg maybe_warn("couldn't chown: %s", file);
372 1.1 mrg sbp->st_mode &= ~(S_ISUID|S_ISGID);
373 1.1 mrg }
374 1.1 mrg
375 1.1 mrg /* we only allow set-id and the 9 normal permission bits */
376 1.1 mrg sbp->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
377 1.1 mrg if (chmod(file, sbp->st_mode) < 0)
378 1.1 mrg maybe_warn("couldn't chmod: %s", file);
379 1.1 mrg
380 1.1 mrg /* only try flags if they exist already */
381 1.1 mrg if (sbp->st_flags != 0 && chflags(file, sbp->st_flags) < 0)
382 1.1 mrg maybe_warn("couldn't chflags: %s", file);
383 1.1 mrg
384 1.1 mrg TIMESPEC_TO_TIMEVAL(×[0], &sbp->st_atimespec);
385 1.1 mrg TIMESPEC_TO_TIMEVAL(×[1], &sbp->st_mtimespec);
386 1.1 mrg if (utimes(file, times) < 0)
387 1.1 mrg maybe_warn("couldn't utimes: %s", file);
388 1.1 mrg }
389 1.1 mrg
390 1.1 mrg /*
391 1.1 mrg * compress the given file: create a corresponding .gz file and remove the
392 1.1 mrg * original.
393 1.1 mrg */
394 1.1 mrg static ssize_t
395 1.1 mrg file_compress(char *file)
396 1.1 mrg {
397 1.1 mrg FILE *in;
398 1.1 mrg gzFile out;
399 1.1 mrg struct stat isb, osb;
400 1.5 mrg char outfile[MAXPATHLEN];
401 1.1 mrg ssize_t size;
402 1.1 mrg u_int32_t mtime = 0;
403 1.1 mrg
404 1.1 mrg if (cflag == 0) {
405 1.1 mrg (void)strncpy(outfile, file, MAXPATHLEN - suffix_len);
406 1.1 mrg outfile[MAXPATHLEN - suffix_len] = '\0';
407 1.1 mrg (void)strlcat(outfile, Sflag, sizeof(outfile));
408 1.1 mrg
409 1.1 mrg if (fflag == 0) {
410 1.1 mrg if (stat(outfile, &osb) == 0) {
411 1.1 mrg maybe_warnx("%s already exists -- skipping",
412 1.1 mrg outfile);
413 1.1 mrg goto lose;
414 1.1 mrg }
415 1.1 mrg }
416 1.1 mrg if (stat(file, &isb) == 0) {
417 1.1 mrg if (isb.st_nlink > 1) {
418 1.1 mrg maybe_warnx("%s has %d other link%s -- "
419 1.1 mrg "skipping", file, isb.st_nlink-1,
420 1.1 mrg isb.st_nlink == 1 ? "" : "s");
421 1.1 mrg goto lose;
422 1.1 mrg }
423 1.1 mrg if (nflag == 0)
424 1.1 mrg mtime = (u_int32_t)isb.st_mtime;
425 1.1 mrg }
426 1.1 mrg }
427 1.1 mrg in = fopen(file, "r");
428 1.1 mrg if (in == 0)
429 1.1 mrg maybe_err(1, "can't fopen %s", file);
430 1.1 mrg
431 1.1 mrg if (cflag == 0) {
432 1.1 mrg #if HAVE_ZLIB_GZOPENFULL
433 1.1 mrg char *savename;
434 1.1 mrg
435 1.1 mrg if (nflag == 0)
436 1.1 mrg savename = basename(file);
437 1.1 mrg else
438 1.1 mrg savename = NULL;
439 1.1 mrg out = gzopenfull(outfile, gzipflags, savename, mtime);
440 1.1 mrg #else
441 1.1 mrg out = gzopen(outfile, gzipflags);
442 1.1 mrg #endif
443 1.1 mrg } else
444 1.1 mrg out = gzdopen(STDOUT_FILENO, gzipflags);
445 1.1 mrg
446 1.1 mrg if (out == 0)
447 1.1 mrg maybe_err(1, "can't gz%sopen %s",
448 1.1 mrg cflag ? "d" : "",
449 1.1 mrg cflag ? "stdout" : outfile);
450 1.1 mrg
451 1.1 mrg gz_compress(in, out);
452 1.1 mrg
453 1.1 mrg /*
454 1.1 mrg * if we compressed to stdout, we don't know the size and
455 1.1 mrg * we don't know the new file name, punt. if we can't stat
456 1.1 mrg * the file, whine, otherwise set the size from the stat
457 1.1 mrg * buffer. we only blow away the file if we can stat the
458 1.1 mrg * output, just in case.
459 1.1 mrg */
460 1.1 mrg if (cflag == 0) {
461 1.1 mrg if (stat(outfile, &osb) < 0) {
462 1.1 mrg maybe_warn("couldn't stat: %s", outfile);
463 1.1 mrg maybe_warnx("leaving original %s", file);
464 1.1 mrg size = 0;
465 1.1 mrg } else {
466 1.1 mrg unlink(file);
467 1.1 mrg size = osb.st_size;
468 1.1 mrg }
469 1.1 mrg newfile = outfile;
470 1.1 mrg copymodes(outfile, &isb);
471 1.1 mrg } else {
472 1.1 mrg lose:
473 1.1 mrg size = 0;
474 1.1 mrg newfile = 0;
475 1.1 mrg }
476 1.1 mrg
477 1.1 mrg return (size);
478 1.1 mrg }
479 1.1 mrg
480 1.1 mrg /* uncompress the given file and remove the original */
481 1.1 mrg static ssize_t
482 1.1 mrg file_uncompress(char *file)
483 1.1 mrg {
484 1.1 mrg struct stat isb, osb;
485 1.5 mrg char buf[PATH_MAX];
486 1.1 mrg char *outfile = buf, *s;
487 1.1 mrg FILE *out;
488 1.1 mrg gzFile in;
489 1.1 mrg off_t size;
490 1.1 mrg ssize_t len = strlen(file);
491 1.1 mrg
492 1.5 mrg if (cflag == 0 || lflag) {
493 1.1 mrg s = &file[len - suffix_len + 1];
494 1.1 mrg if (strncmp(s, Sflag, suffix_len) == 0) {
495 1.1 mrg (void)strncpy(outfile, file, len - suffix_len + 1);
496 1.1 mrg outfile[len - suffix_len + 1] = '\0';
497 1.1 mrg } else
498 1.1 mrg maybe_err(1, "unknown suffix %s", s);
499 1.1 mrg
500 1.1 mrg /* gather the old name info */
501 1.5 mrg if (Nflag || lflag) {
502 1.1 mrg int fd;
503 1.1 mrg char header1[10], name[PATH_MAX + 1];
504 1.1 mrg
505 1.1 mrg fd = open(file, O_RDONLY);
506 1.1 mrg if (fd < 0)
507 1.1 mrg maybe_err(1, "can't open %s", file);
508 1.1 mrg if (read(fd, header1, 10) != 10)
509 1.1 mrg maybe_err(1, "can't read %s", file);
510 1.1 mrg
511 1.1 mrg if (header1[3] & ORIG_NAME) {
512 1.1 mrg size_t rbytes;
513 1.1 mrg int i;
514 1.1 mrg
515 1.1 mrg rbytes = read(fd, name, PATH_MAX + 1);
516 1.1 mrg if (rbytes < 0)
517 1.1 mrg maybe_err(1, "can't read %s", file);
518 1.1 mrg for (i = 0; i < rbytes && name[i]; i++)
519 1.1 mrg ;
520 1.1 mrg if (i < rbytes) {
521 1.1 mrg name[i] = 0;
522 1.1 mrg /* now maybe merge old dirname */
523 1.1 mrg if (strchr(outfile, '/') == 0)
524 1.1 mrg outfile = name;
525 1.1 mrg else {
526 1.1 mrg char *dir = dirname(outfile);
527 1.1 mrg if (asprintf(&outfile, "%s/%s",
528 1.1 mrg dir, name) == -1)
529 1.1 mrg maybe_err(1, "malloc");
530 1.1 mrg }
531 1.1 mrg }
532 1.1 mrg }
533 1.1 mrg close(fd);
534 1.1 mrg }
535 1.1 mrg
536 1.1 mrg if (fflag == 0) {
537 1.5 mrg if (lflag == 0 && stat(outfile, &osb) == 0) {
538 1.1 mrg maybe_warnx("%s already exists -- skipping",
539 1.1 mrg outfile);
540 1.1 mrg goto lose;
541 1.1 mrg }
542 1.5 mrg if (stat(file, &isb) == 0) {
543 1.5 mrg if (isb.st_nlink > 1 && lflag == 0) {
544 1.5 mrg maybe_warnx("%s has %d other link%s -- "
545 1.1 mrg "skipping", file, isb.st_nlink-1,
546 1.1 mrg isb.st_nlink == 1 ? "" : "s");
547 1.5 mrg goto lose;
548 1.5 mrg }
549 1.5 mrg } else
550 1.1 mrg goto lose;
551 1.1 mrg }
552 1.1 mrg }
553 1.5 mrg
554 1.5 mrg if (lflag) {
555 1.5 mrg int fd;
556 1.5 mrg
557 1.5 mrg if ((fd = open(file, O_RDONLY)) == -1)
558 1.5 mrg maybe_err(1, "open");
559 1.6 mrg print_list(fd, isb.st_size, outfile);
560 1.5 mrg return 0; /* XXX */
561 1.5 mrg }
562 1.5 mrg
563 1.1 mrg in = gzopen(file, gzipflags);
564 1.1 mrg if (in == NULL)
565 1.1 mrg maybe_err(1, "can't gzopen %s", file);
566 1.1 mrg
567 1.1 mrg if (cflag == 0) {
568 1.1 mrg int fd;
569 1.1 mrg
570 1.1 mrg /* Use open(2) directly to get a safe file. */
571 1.1 mrg fd = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
572 1.1 mrg if (fd < 0)
573 1.1 mrg maybe_err(1, "can't open %s", outfile);
574 1.1 mrg out = fdopen(fd, "w");
575 1.1 mrg if (out == NULL)
576 1.1 mrg maybe_err(1, "can't fdopen %s", outfile);
577 1.1 mrg } else
578 1.1 mrg out = stdout;
579 1.1 mrg
580 1.1 mrg if ((size = gz_uncompress(in, out)) == 0)
581 1.1 mrg goto lose;
582 1.1 mrg
583 1.1 mrg /* if testing, or we uncompressed to stdout, this is all we need */
584 1.1 mrg if (tflag || cflag)
585 1.1 mrg return (size);
586 1.1 mrg
587 1.1 mrg /*
588 1.1 mrg * if we create a file...
589 1.1 mrg */
590 1.1 mrg if (cflag == 0) {
591 1.1 mrg /* close the file */
592 1.1 mrg if (fclose(out))
593 1.1 mrg maybe_err(1, "failed fclose");
594 1.1 mrg
595 1.1 mrg /*
596 1.1 mrg * if we can't stat the file, or we are uncompressing to
597 1.1 mrg * stdin, don't remove the file.
598 1.1 mrg */
599 1.1 mrg if (stat(outfile, &osb) < 0) {
600 1.1 mrg maybe_warn("couldn't stat (leaving original): %s",
601 1.1 mrg outfile);
602 1.1 mrg goto lose;
603 1.1 mrg }
604 1.1 mrg if (osb.st_size != size) {
605 1.1 mrg maybe_warn("stat gave different size: %llu != %llu "
606 1.1 mrg "(leaving original)",
607 1.1 mrg (unsigned long long)size,
608 1.1 mrg (unsigned long long)osb.st_size);
609 1.1 mrg goto lose;
610 1.1 mrg }
611 1.1 mrg newfile = outfile;
612 1.1 mrg unlink(file);
613 1.1 mrg size = osb.st_size;
614 1.1 mrg copymodes(outfile, &isb);
615 1.1 mrg }
616 1.1 mrg return (size);
617 1.1 mrg
618 1.1 mrg lose:
619 1.1 mrg newfile = 0;
620 1.1 mrg return (0);
621 1.1 mrg }
622 1.1 mrg
623 1.1 mrg static void
624 1.1 mrg handle_stdin(void)
625 1.1 mrg {
626 1.1 mrg gzFile *file;
627 1.1 mrg
628 1.5 mrg if (fflag == 0 && lflag == 0 && isatty(STDIN_FILENO)) {
629 1.1 mrg maybe_warnx("standard input is a terminal -- ignoring");
630 1.1 mrg return;
631 1.1 mrg }
632 1.5 mrg
633 1.5 mrg if (lflag) {
634 1.5 mrg struct stat isb;
635 1.5 mrg
636 1.5 mrg if (fstat(STDIN_FILENO, &isb) < 0)
637 1.5 mrg maybe_err(1, "fstat");
638 1.6 mrg print_list(STDIN_FILENO, isb.st_size, "stdout");
639 1.5 mrg return;
640 1.5 mrg }
641 1.5 mrg
642 1.1 mrg file = gzdopen(STDIN_FILENO, gzipflags);
643 1.1 mrg if (file == NULL)
644 1.1 mrg maybe_err(1, "can't gzdopen stdin");
645 1.1 mrg gz_uncompress(file, stdout);
646 1.1 mrg }
647 1.1 mrg
648 1.1 mrg static void
649 1.1 mrg handle_stdout(void)
650 1.1 mrg {
651 1.1 mrg gzFile *file;
652 1.1 mrg
653 1.1 mrg if (fflag == 0 && isatty(STDOUT_FILENO)) {
654 1.1 mrg maybe_warnx("standard output is a terminal -- ignoring");
655 1.1 mrg return;
656 1.1 mrg }
657 1.1 mrg file = gzdopen(STDOUT_FILENO, gzipflags);
658 1.1 mrg if (file == NULL)
659 1.1 mrg maybe_err(1, "can't gzdopen stdout");
660 1.1 mrg gz_compress(stdin, file);
661 1.1 mrg }
662 1.1 mrg
663 1.1 mrg /* do what is asked for, for the path name */
664 1.1 mrg static void
665 1.1 mrg handle_pathname(char *path)
666 1.1 mrg {
667 1.1 mrg char *opath = path, *s = 0;
668 1.1 mrg ssize_t len;
669 1.1 mrg struct stat sb;
670 1.1 mrg
671 1.1 mrg /* check for stdout/stdin */
672 1.1 mrg if (path[0] == '-' && path[1] == '\0') {
673 1.1 mrg if (dflag)
674 1.1 mrg handle_stdin();
675 1.1 mrg else
676 1.1 mrg handle_stdout();
677 1.1 mrg }
678 1.1 mrg
679 1.1 mrg retry:
680 1.1 mrg if (stat(path, &sb) < 0) {
681 1.1 mrg /* lets try <path>.gz if we're decompressing */
682 1.1 mrg if (dflag && s == 0 && errno == ENOENT) {
683 1.1 mrg len = strlen(path);
684 1.1 mrg s = malloc(len + suffix_len);
685 1.1 mrg if (s == 0)
686 1.1 mrg maybe_err(1, "malloc");
687 1.1 mrg memmove(s, path, len);
688 1.1 mrg memmove(&s[len], Sflag, suffix_len);
689 1.1 mrg path = s;
690 1.1 mrg goto retry;
691 1.1 mrg }
692 1.1 mrg maybe_warn("can't stat: %s", opath);
693 1.1 mrg goto out;
694 1.1 mrg }
695 1.1 mrg
696 1.1 mrg if (S_ISDIR(sb.st_mode)) {
697 1.1 mrg if (rflag)
698 1.1 mrg handle_dir(path, &sb);
699 1.1 mrg else
700 1.1 mrg maybe_warn("%s is a directory", path);
701 1.1 mrg goto out;
702 1.1 mrg }
703 1.1 mrg
704 1.1 mrg if (S_ISREG(sb.st_mode))
705 1.1 mrg handle_file(path, &sb);
706 1.1 mrg
707 1.1 mrg out:
708 1.1 mrg if (s)
709 1.1 mrg free(s);
710 1.1 mrg return;
711 1.1 mrg }
712 1.1 mrg
713 1.1 mrg /* compress/decompress a file */
714 1.1 mrg static void
715 1.1 mrg handle_file(char *file, struct stat *sbp)
716 1.1 mrg {
717 1.1 mrg ssize_t usize, gsize;
718 1.1 mrg
719 1.1 mrg infile = file;
720 1.1 mrg if (dflag) {
721 1.1 mrg usize = file_uncompress(file);
722 1.1 mrg if (usize == 0)
723 1.1 mrg return;
724 1.1 mrg gsize = sbp->st_size;
725 1.1 mrg } else {
726 1.1 mrg gsize = file_compress(file);
727 1.1 mrg if (gsize == 0)
728 1.1 mrg return;
729 1.1 mrg usize = sbp->st_size;
730 1.1 mrg }
731 1.1 mrg
732 1.1 mrg if (vflag && !tflag)
733 1.1 mrg print_verbage(file, cflag == 0 ? newfile : 0, usize, gsize);
734 1.1 mrg }
735 1.1 mrg
736 1.1 mrg /* this is used with -r to recursively decend directories */
737 1.1 mrg static void
738 1.1 mrg handle_dir(char *dir, struct stat *sbp)
739 1.1 mrg {
740 1.1 mrg char *path_argv[2];
741 1.1 mrg FTS *fts;
742 1.1 mrg FTSENT *entry;
743 1.1 mrg
744 1.1 mrg path_argv[0] = dir;
745 1.1 mrg path_argv[1] = 0;
746 1.1 mrg fts = fts_open(path_argv, FTS_PHYSICAL, NULL);
747 1.1 mrg if (fts == NULL) {
748 1.1 mrg warn("couldn't fts_open %s", dir);
749 1.1 mrg return;
750 1.1 mrg }
751 1.1 mrg
752 1.1 mrg while ((entry = fts_read(fts))) {
753 1.1 mrg switch(entry->fts_info) {
754 1.1 mrg case FTS_D:
755 1.1 mrg case FTS_DP:
756 1.1 mrg continue;
757 1.1 mrg
758 1.1 mrg case FTS_DNR:
759 1.1 mrg case FTS_ERR:
760 1.1 mrg case FTS_NS:
761 1.1 mrg maybe_warn("%s", entry->fts_path);
762 1.1 mrg continue;
763 1.1 mrg case FTS_F:
764 1.1 mrg handle_file(entry->fts_name, entry->fts_statp);
765 1.1 mrg }
766 1.1 mrg }
767 1.1 mrg (void)fts_close(fts);
768 1.1 mrg }
769 1.1 mrg
770 1.6 mrg /* print a ratio */
771 1.6 mrg static void
772 1.6 mrg print_ratio(off_t in, off_t out, FILE *where)
773 1.6 mrg {
774 1.6 mrg u_int64_t percent;
775 1.6 mrg
776 1.6 mrg if (out == 0)
777 1.6 mrg percent = 0;
778 1.6 mrg else if (out < 1000 * 1000)
779 1.6 mrg percent = 999 - (in * 1000) / out;
780 1.6 mrg else
781 1.6 mrg percent = 999 - in / (out / 1000);
782 1.6 mrg fprintf(where, "%3lu.%1lu%%", (unsigned long)percent / 10UL,
783 1.6 mrg (unsigned long)percent % 10);
784 1.6 mrg }
785 1.6 mrg
786 1.1 mrg /* print compression statistics, and the new name (if there is one!) */
787 1.1 mrg static void
788 1.4 mrg print_verbage(char *file, char *nfile, ssize_t usize, ssize_t gsize)
789 1.1 mrg {
790 1.6 mrg fprintf(stderr, "%s:%s ", file,
791 1.6 mrg strlen(file) < 7 ? "\t\t" : "\t");
792 1.6 mrg print_ratio(usize, gsize, stderr);
793 1.4 mrg if (nfile)
794 1.4 mrg fprintf(stderr, " -- replaced with %s", nfile);
795 1.1 mrg fprintf(stderr, "\n");
796 1.1 mrg fflush(stderr);
797 1.1 mrg }
798 1.1 mrg
799 1.1 mrg /* print test results */
800 1.1 mrg static void
801 1.1 mrg print_test(char *file, int ok)
802 1.1 mrg {
803 1.1 mrg
804 1.1 mrg fprintf(stderr, "%s:%s %s\n", file,
805 1.1 mrg strlen(file) < 7 ? "\t\t" : "\t", ok ? "OK" : "NOT OK");
806 1.1 mrg fflush(stderr);
807 1.1 mrg }
808 1.1 mrg
809 1.5 mrg /* print a file's info ala --list */
810 1.5 mrg /* eg:
811 1.5 mrg compressed uncompressed ratio uncompressed_name
812 1.5 mrg 354841 1679360 78.8% /usr/pkgsrc/distfiles/libglade-2.0.1.tar
813 1.5 mrg */
814 1.5 mrg static void
815 1.6 mrg print_list(int fd, off_t in, const char *outfile)
816 1.5 mrg {
817 1.5 mrg static int first = 1;
818 1.6 mrg static off_t in_tot, out_tot;
819 1.6 mrg off_t out;
820 1.5 mrg int rv;
821 1.5 mrg
822 1.5 mrg if (qflag == 0 && first)
823 1.5 mrg printf(" compressed uncompressed ratio uncompressed_name\n");
824 1.5 mrg first = 0;
825 1.5 mrg
826 1.6 mrg /* print totals? */
827 1.6 mrg if (fd == -1) {
828 1.6 mrg in = in_tot;
829 1.6 mrg out = out_tot;
830 1.6 mrg } else {
831 1.6 mrg /* read the last 4 bytes - this is the uncompressed size */
832 1.6 mrg rv = lseek(fd, (off_t)(-4), SEEK_END);
833 1.6 mrg if (rv != -1) {
834 1.6 mrg unsigned char buf[4];
835 1.6 mrg u_int32_t usize;
836 1.6 mrg
837 1.6 mrg if (read(fd, (char *)buf, sizeof(buf)) != sizeof(buf))
838 1.6 mrg maybe_err(1, "read of uncompressed size");
839 1.6 mrg usize = buf[0] | buf[1] << 8 | buf[2] << 16 | buf[3] << 24;
840 1.6 mrg out = (off_t)usize;
841 1.6 mrg }
842 1.6 mrg }
843 1.5 mrg
844 1.6 mrg printf("%12llu %12llu ", (unsigned long long)in, (unsigned long long)out);
845 1.6 mrg print_ratio(in, out, stdout);
846 1.6 mrg printf(" %s\n", outfile);
847 1.6 mrg in_tot += in;
848 1.6 mrg out_tot += out;
849 1.5 mrg }
850 1.5 mrg
851 1.1 mrg /* display the usage of NetBSD gzip */
852 1.1 mrg static void
853 1.1 mrg usage(void)
854 1.1 mrg {
855 1.1 mrg
856 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
857 1.1 mrg fprintf(stderr,
858 1.1 mrg "Usage: %s [-cdfhnNqrStvV123456789] [<file> [<file> ...]]\n"
859 1.1 mrg " -c --stdout write to stdout, keep original files\n"
860 1.1 mrg " --to-stdout\n"
861 1.1 mrg " -d --decompress uncompress files\n"
862 1.1 mrg " --uncompress\n"
863 1.1 mrg " -f --force force overwriting & compress links\n"
864 1.1 mrg " -h --help display this help\n"
865 1.1 mrg " -n --no-name don't save original file name or time stamp\n"
866 1.1 mrg " -N --name save or restore original file name and time stamp\n"
867 1.1 mrg " -q --quiet output no warnings\n"
868 1.1 mrg " -r --recursive recursively compress files in directories\n"
869 1.1 mrg " -S .suf use suffix .suf instead of .gz\n"
870 1.1 mrg " --suffix .suf\n"
871 1.1 mrg " -t --test test compressed file\n"
872 1.1 mrg " -v --verbose print extra statistics\n"
873 1.1 mrg " -V --version display program version\n"
874 1.1 mrg " -1 --fast fastest (worst) compression\n"
875 1.1 mrg " -2 .. -8 set compression level\n"
876 1.1 mrg " -9 --best best (slowest) compression\n",
877 1.1 mrg getprogname());
878 1.1 mrg fflush(stderr);
879 1.1 mrg exit(0);
880 1.1 mrg }
881 1.1 mrg
882 1.1 mrg /* display the version of NetBSD gzip */
883 1.1 mrg static void
884 1.1 mrg display_version(void)
885 1.1 mrg {
886 1.1 mrg
887 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
888 1.1 mrg fflush(stderr);
889 1.1 mrg exit(0);
890 1.1 mrg }
891