gzip.c revision 1.7 1 1.7 mrg /* $NetBSD: gzip.c,v 1.7 2003/12/26 15:06:16 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.7 mrg __RCSID("$NetBSD: gzip.c,v 1.7 2003/12/26 15:06:16 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.7 mrg static void print_list(int fd, off_t, const char *, time_t);
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.4 mrg suffix_len = strlen(Sflag) + 1;
227 1.4 mrg
228 1.1 mrg if (argc == 0) {
229 1.1 mrg if (dflag) /* stdin mode */
230 1.1 mrg handle_stdin();
231 1.1 mrg else /* stdout mode */
232 1.1 mrg handle_stdout();
233 1.1 mrg } else {
234 1.1 mrg do {
235 1.1 mrg handle_pathname(argv[0]);
236 1.5 mrg } while (*++argv);
237 1.1 mrg }
238 1.6 mrg if (qflag == 0 && lflag && argc > 1)
239 1.7 mrg print_list(-1, 0, "(totals)", 0);
240 1.1 mrg exit(0);
241 1.1 mrg }
242 1.1 mrg
243 1.1 mrg /* maybe print a warning */
244 1.1 mrg void
245 1.1 mrg maybe_warn(const char *fmt, ...)
246 1.1 mrg {
247 1.1 mrg va_list ap;
248 1.1 mrg
249 1.1 mrg if (qflag == 0) {
250 1.1 mrg va_start(ap, fmt);
251 1.1 mrg vwarn(fmt, ap);
252 1.1 mrg va_end(ap);
253 1.1 mrg }
254 1.1 mrg }
255 1.1 mrg
256 1.1 mrg void
257 1.1 mrg maybe_warnx(const char *fmt, ...)
258 1.1 mrg {
259 1.1 mrg va_list ap;
260 1.1 mrg
261 1.1 mrg if (qflag == 0) {
262 1.1 mrg va_start(ap, fmt);
263 1.1 mrg vwarnx(fmt, ap);
264 1.1 mrg va_end(ap);
265 1.1 mrg }
266 1.1 mrg }
267 1.1 mrg
268 1.1 mrg /* maybe print a warning */
269 1.1 mrg void
270 1.1 mrg maybe_err(int rv, const char *fmt, ...)
271 1.1 mrg {
272 1.1 mrg va_list ap;
273 1.1 mrg
274 1.1 mrg if (qflag == 0) {
275 1.1 mrg va_start(ap, fmt);
276 1.1 mrg vwarn(fmt, ap);
277 1.1 mrg va_end(ap);
278 1.1 mrg }
279 1.1 mrg exit(rv);
280 1.1 mrg }
281 1.1 mrg
282 1.1 mrg /* compress input to output then close both files */
283 1.1 mrg static void
284 1.1 mrg gz_compress(FILE *in, gzFile out)
285 1.1 mrg {
286 1.1 mrg char buf[BUFLEN];
287 1.1 mrg ssize_t len;
288 1.1 mrg int i;
289 1.1 mrg
290 1.1 mrg for (;;) {
291 1.1 mrg len = fread(buf, 1, sizeof(buf), in);
292 1.1 mrg if (ferror(in))
293 1.1 mrg maybe_err(1, "fread");
294 1.1 mrg if (len == 0)
295 1.1 mrg break;
296 1.1 mrg
297 1.1 mrg if ((ssize_t)gzwrite(out, buf, len) != len)
298 1.1 mrg maybe_err(1, gzerror(out, &i));
299 1.1 mrg }
300 1.1 mrg if (fclose(in) < 0)
301 1.1 mrg maybe_err(1, "failed fclose");
302 1.1 mrg if (gzclose(out) != Z_OK)
303 1.1 mrg maybe_err(1, "failed gzclose");
304 1.1 mrg }
305 1.1 mrg
306 1.1 mrg /* uncompress input to output then close the input */
307 1.1 mrg static off_t
308 1.1 mrg gz_uncompress(gzFile in, FILE *out)
309 1.1 mrg {
310 1.1 mrg char buf[BUFLEN];
311 1.1 mrg off_t size;
312 1.1 mrg ssize_t len;
313 1.1 mrg int i;
314 1.1 mrg
315 1.1 mrg for (size = 0;;) {
316 1.1 mrg len = gzread(in, buf, sizeof(buf));
317 1.1 mrg
318 1.1 mrg if (len < 0) {
319 1.1 mrg if (tflag) {
320 1.1 mrg print_test(infile, 0);
321 1.1 mrg return (0);
322 1.1 mrg } else
323 1.1 mrg maybe_err(1, gzerror(in, &i));
324 1.1 mrg } else if (len == 0) {
325 1.1 mrg if (tflag)
326 1.1 mrg print_test(infile, 1);
327 1.1 mrg break;
328 1.1 mrg }
329 1.1 mrg
330 1.1 mrg size += len;
331 1.1 mrg
332 1.1 mrg /* don't write anything with -t */
333 1.1 mrg if (tflag)
334 1.1 mrg continue;
335 1.1 mrg
336 1.1 mrg if (fwrite(buf, 1, (unsigned)len, out) != (ssize_t)len)
337 1.1 mrg maybe_err(1, "failed fwrite");
338 1.1 mrg }
339 1.1 mrg if (gzclose(in) != Z_OK)
340 1.1 mrg maybe_err(1, "failed gzclose");
341 1.1 mrg
342 1.1 mrg return (size);
343 1.1 mrg }
344 1.1 mrg
345 1.1 mrg /*
346 1.1 mrg * set the owner, mode, flags & utimes for a file
347 1.1 mrg */
348 1.1 mrg static void
349 1.1 mrg copymodes(const char *file, struct stat *sbp)
350 1.1 mrg {
351 1.1 mrg struct timeval times[2];
352 1.1 mrg
353 1.1 mrg /*
354 1.1 mrg * If we have no info on the input, give this file some
355 1.1 mrg * default values and return..
356 1.1 mrg */
357 1.1 mrg if (sbp == NULL) {
358 1.1 mrg mode_t mask = umask(022);
359 1.1 mrg
360 1.1 mrg (void)chmod(file, DEFFILEMODE & ~mask);
361 1.1 mrg (void)umask(mask);
362 1.1 mrg return;
363 1.1 mrg }
364 1.1 mrg
365 1.1 mrg /* if the chown fails, remove set-id bits as-per compress(1) */
366 1.1 mrg if (chown(file, sbp->st_uid, sbp->st_gid) < 0) {
367 1.1 mrg if (errno != EPERM)
368 1.1 mrg maybe_warn("couldn't chown: %s", file);
369 1.1 mrg sbp->st_mode &= ~(S_ISUID|S_ISGID);
370 1.1 mrg }
371 1.1 mrg
372 1.1 mrg /* we only allow set-id and the 9 normal permission bits */
373 1.1 mrg sbp->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
374 1.1 mrg if (chmod(file, sbp->st_mode) < 0)
375 1.1 mrg maybe_warn("couldn't chmod: %s", file);
376 1.1 mrg
377 1.1 mrg /* only try flags if they exist already */
378 1.1 mrg if (sbp->st_flags != 0 && chflags(file, sbp->st_flags) < 0)
379 1.1 mrg maybe_warn("couldn't chflags: %s", file);
380 1.1 mrg
381 1.1 mrg TIMESPEC_TO_TIMEVAL(×[0], &sbp->st_atimespec);
382 1.1 mrg TIMESPEC_TO_TIMEVAL(×[1], &sbp->st_mtimespec);
383 1.1 mrg if (utimes(file, times) < 0)
384 1.1 mrg maybe_warn("couldn't utimes: %s", file);
385 1.1 mrg }
386 1.1 mrg
387 1.1 mrg /*
388 1.1 mrg * compress the given file: create a corresponding .gz file and remove the
389 1.1 mrg * original.
390 1.1 mrg */
391 1.1 mrg static ssize_t
392 1.1 mrg file_compress(char *file)
393 1.1 mrg {
394 1.1 mrg FILE *in;
395 1.1 mrg gzFile out;
396 1.1 mrg struct stat isb, osb;
397 1.5 mrg char outfile[MAXPATHLEN];
398 1.1 mrg ssize_t size;
399 1.1 mrg u_int32_t mtime = 0;
400 1.1 mrg
401 1.1 mrg if (cflag == 0) {
402 1.1 mrg (void)strncpy(outfile, file, MAXPATHLEN - suffix_len);
403 1.1 mrg outfile[MAXPATHLEN - suffix_len] = '\0';
404 1.1 mrg (void)strlcat(outfile, Sflag, sizeof(outfile));
405 1.1 mrg
406 1.1 mrg if (fflag == 0) {
407 1.1 mrg if (stat(outfile, &osb) == 0) {
408 1.1 mrg maybe_warnx("%s already exists -- skipping",
409 1.1 mrg outfile);
410 1.1 mrg goto lose;
411 1.1 mrg }
412 1.1 mrg }
413 1.1 mrg if (stat(file, &isb) == 0) {
414 1.1 mrg if (isb.st_nlink > 1) {
415 1.1 mrg maybe_warnx("%s has %d other link%s -- "
416 1.1 mrg "skipping", file, isb.st_nlink-1,
417 1.1 mrg isb.st_nlink == 1 ? "" : "s");
418 1.1 mrg goto lose;
419 1.1 mrg }
420 1.1 mrg if (nflag == 0)
421 1.1 mrg mtime = (u_int32_t)isb.st_mtime;
422 1.1 mrg }
423 1.1 mrg }
424 1.1 mrg in = fopen(file, "r");
425 1.1 mrg if (in == 0)
426 1.1 mrg maybe_err(1, "can't fopen %s", file);
427 1.1 mrg
428 1.1 mrg if (cflag == 0) {
429 1.1 mrg #if HAVE_ZLIB_GZOPENFULL
430 1.1 mrg char *savename;
431 1.1 mrg
432 1.1 mrg if (nflag == 0)
433 1.1 mrg savename = basename(file);
434 1.1 mrg else
435 1.1 mrg savename = NULL;
436 1.1 mrg out = gzopenfull(outfile, gzipflags, savename, mtime);
437 1.1 mrg #else
438 1.1 mrg out = gzopen(outfile, gzipflags);
439 1.1 mrg #endif
440 1.1 mrg } else
441 1.1 mrg out = gzdopen(STDOUT_FILENO, gzipflags);
442 1.1 mrg
443 1.1 mrg if (out == 0)
444 1.1 mrg maybe_err(1, "can't gz%sopen %s",
445 1.1 mrg cflag ? "d" : "",
446 1.1 mrg cflag ? "stdout" : outfile);
447 1.1 mrg
448 1.1 mrg gz_compress(in, out);
449 1.1 mrg
450 1.1 mrg /*
451 1.1 mrg * if we compressed to stdout, we don't know the size and
452 1.1 mrg * we don't know the new file name, punt. if we can't stat
453 1.1 mrg * the file, whine, otherwise set the size from the stat
454 1.1 mrg * buffer. we only blow away the file if we can stat the
455 1.1 mrg * output, just in case.
456 1.1 mrg */
457 1.1 mrg if (cflag == 0) {
458 1.1 mrg if (stat(outfile, &osb) < 0) {
459 1.1 mrg maybe_warn("couldn't stat: %s", outfile);
460 1.1 mrg maybe_warnx("leaving original %s", file);
461 1.1 mrg size = 0;
462 1.1 mrg } else {
463 1.1 mrg unlink(file);
464 1.1 mrg size = osb.st_size;
465 1.1 mrg }
466 1.1 mrg newfile = outfile;
467 1.1 mrg copymodes(outfile, &isb);
468 1.1 mrg } else {
469 1.1 mrg lose:
470 1.1 mrg size = 0;
471 1.1 mrg newfile = 0;
472 1.1 mrg }
473 1.1 mrg
474 1.1 mrg return (size);
475 1.1 mrg }
476 1.1 mrg
477 1.1 mrg /* uncompress the given file and remove the original */
478 1.1 mrg static ssize_t
479 1.1 mrg file_uncompress(char *file)
480 1.1 mrg {
481 1.1 mrg struct stat isb, osb;
482 1.5 mrg char buf[PATH_MAX];
483 1.1 mrg char *outfile = buf, *s;
484 1.1 mrg FILE *out;
485 1.1 mrg gzFile in;
486 1.1 mrg off_t size;
487 1.1 mrg ssize_t len = strlen(file);
488 1.1 mrg
489 1.5 mrg if (cflag == 0 || lflag) {
490 1.1 mrg s = &file[len - suffix_len + 1];
491 1.1 mrg if (strncmp(s, Sflag, suffix_len) == 0) {
492 1.1 mrg (void)strncpy(outfile, file, len - suffix_len + 1);
493 1.1 mrg outfile[len - suffix_len + 1] = '\0';
494 1.1 mrg } else
495 1.1 mrg maybe_err(1, "unknown suffix %s", s);
496 1.1 mrg
497 1.1 mrg /* gather the old name info */
498 1.5 mrg if (Nflag || lflag) {
499 1.1 mrg int fd;
500 1.1 mrg char header1[10], name[PATH_MAX + 1];
501 1.1 mrg
502 1.1 mrg fd = open(file, O_RDONLY);
503 1.1 mrg if (fd < 0)
504 1.1 mrg maybe_err(1, "can't open %s", file);
505 1.1 mrg if (read(fd, header1, 10) != 10)
506 1.1 mrg maybe_err(1, "can't read %s", file);
507 1.1 mrg
508 1.1 mrg if (header1[3] & ORIG_NAME) {
509 1.1 mrg size_t rbytes;
510 1.1 mrg int i;
511 1.1 mrg
512 1.1 mrg rbytes = read(fd, name, PATH_MAX + 1);
513 1.1 mrg if (rbytes < 0)
514 1.1 mrg maybe_err(1, "can't read %s", file);
515 1.1 mrg for (i = 0; i < rbytes && name[i]; i++)
516 1.1 mrg ;
517 1.1 mrg if (i < rbytes) {
518 1.1 mrg name[i] = 0;
519 1.1 mrg /* now maybe merge old dirname */
520 1.1 mrg if (strchr(outfile, '/') == 0)
521 1.1 mrg outfile = name;
522 1.1 mrg else {
523 1.1 mrg char *dir = dirname(outfile);
524 1.1 mrg if (asprintf(&outfile, "%s/%s",
525 1.1 mrg dir, name) == -1)
526 1.1 mrg maybe_err(1, "malloc");
527 1.1 mrg }
528 1.1 mrg }
529 1.1 mrg }
530 1.1 mrg close(fd);
531 1.1 mrg }
532 1.1 mrg
533 1.1 mrg if (fflag == 0) {
534 1.5 mrg if (lflag == 0 && stat(outfile, &osb) == 0) {
535 1.1 mrg maybe_warnx("%s already exists -- skipping",
536 1.1 mrg outfile);
537 1.1 mrg goto lose;
538 1.1 mrg }
539 1.5 mrg if (stat(file, &isb) == 0) {
540 1.5 mrg if (isb.st_nlink > 1 && lflag == 0) {
541 1.5 mrg maybe_warnx("%s has %d other link%s -- "
542 1.1 mrg "skipping", file, isb.st_nlink-1,
543 1.1 mrg isb.st_nlink == 1 ? "" : "s");
544 1.5 mrg goto lose;
545 1.5 mrg }
546 1.5 mrg } else
547 1.1 mrg goto lose;
548 1.1 mrg }
549 1.1 mrg }
550 1.5 mrg
551 1.5 mrg if (lflag) {
552 1.5 mrg int fd;
553 1.5 mrg
554 1.5 mrg if ((fd = open(file, O_RDONLY)) == -1)
555 1.5 mrg maybe_err(1, "open");
556 1.7 mrg print_list(fd, isb.st_size, outfile, isb.st_mtime);
557 1.5 mrg return 0; /* XXX */
558 1.5 mrg }
559 1.5 mrg
560 1.1 mrg in = gzopen(file, gzipflags);
561 1.1 mrg if (in == NULL)
562 1.1 mrg maybe_err(1, "can't gzopen %s", file);
563 1.1 mrg
564 1.1 mrg if (cflag == 0) {
565 1.1 mrg int fd;
566 1.1 mrg
567 1.1 mrg /* Use open(2) directly to get a safe file. */
568 1.1 mrg fd = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
569 1.1 mrg if (fd < 0)
570 1.1 mrg maybe_err(1, "can't open %s", outfile);
571 1.1 mrg out = fdopen(fd, "w");
572 1.1 mrg if (out == NULL)
573 1.1 mrg maybe_err(1, "can't fdopen %s", outfile);
574 1.1 mrg } else
575 1.1 mrg out = stdout;
576 1.1 mrg
577 1.1 mrg if ((size = gz_uncompress(in, out)) == 0)
578 1.1 mrg goto lose;
579 1.1 mrg
580 1.1 mrg /* if testing, or we uncompressed to stdout, this is all we need */
581 1.1 mrg if (tflag || cflag)
582 1.1 mrg return (size);
583 1.1 mrg
584 1.1 mrg /*
585 1.1 mrg * if we create a file...
586 1.1 mrg */
587 1.1 mrg if (cflag == 0) {
588 1.1 mrg /* close the file */
589 1.1 mrg if (fclose(out))
590 1.1 mrg maybe_err(1, "failed fclose");
591 1.1 mrg
592 1.1 mrg /*
593 1.1 mrg * if we can't stat the file, or we are uncompressing to
594 1.1 mrg * stdin, don't remove the file.
595 1.1 mrg */
596 1.1 mrg if (stat(outfile, &osb) < 0) {
597 1.1 mrg maybe_warn("couldn't stat (leaving original): %s",
598 1.1 mrg outfile);
599 1.1 mrg goto lose;
600 1.1 mrg }
601 1.1 mrg if (osb.st_size != size) {
602 1.1 mrg maybe_warn("stat gave different size: %llu != %llu "
603 1.1 mrg "(leaving original)",
604 1.1 mrg (unsigned long long)size,
605 1.1 mrg (unsigned long long)osb.st_size);
606 1.1 mrg goto lose;
607 1.1 mrg }
608 1.1 mrg newfile = outfile;
609 1.1 mrg unlink(file);
610 1.1 mrg size = osb.st_size;
611 1.1 mrg copymodes(outfile, &isb);
612 1.1 mrg }
613 1.1 mrg return (size);
614 1.1 mrg
615 1.1 mrg lose:
616 1.1 mrg newfile = 0;
617 1.1 mrg return (0);
618 1.1 mrg }
619 1.1 mrg
620 1.1 mrg static void
621 1.1 mrg handle_stdin(void)
622 1.1 mrg {
623 1.1 mrg gzFile *file;
624 1.1 mrg
625 1.5 mrg if (fflag == 0 && lflag == 0 && isatty(STDIN_FILENO)) {
626 1.1 mrg maybe_warnx("standard input is a terminal -- ignoring");
627 1.1 mrg return;
628 1.1 mrg }
629 1.5 mrg
630 1.5 mrg if (lflag) {
631 1.5 mrg struct stat isb;
632 1.5 mrg
633 1.5 mrg if (fstat(STDIN_FILENO, &isb) < 0)
634 1.5 mrg maybe_err(1, "fstat");
635 1.7 mrg print_list(STDIN_FILENO, isb.st_size, "stdout", isb.st_mtime);
636 1.5 mrg return;
637 1.5 mrg }
638 1.5 mrg
639 1.1 mrg file = gzdopen(STDIN_FILENO, gzipflags);
640 1.1 mrg if (file == NULL)
641 1.1 mrg maybe_err(1, "can't gzdopen stdin");
642 1.1 mrg gz_uncompress(file, stdout);
643 1.1 mrg }
644 1.1 mrg
645 1.1 mrg static void
646 1.1 mrg handle_stdout(void)
647 1.1 mrg {
648 1.1 mrg gzFile *file;
649 1.1 mrg
650 1.1 mrg if (fflag == 0 && isatty(STDOUT_FILENO)) {
651 1.1 mrg maybe_warnx("standard output is a terminal -- ignoring");
652 1.1 mrg return;
653 1.1 mrg }
654 1.1 mrg file = gzdopen(STDOUT_FILENO, gzipflags);
655 1.1 mrg if (file == NULL)
656 1.1 mrg maybe_err(1, "can't gzdopen stdout");
657 1.1 mrg gz_compress(stdin, file);
658 1.1 mrg }
659 1.1 mrg
660 1.1 mrg /* do what is asked for, for the path name */
661 1.1 mrg static void
662 1.1 mrg handle_pathname(char *path)
663 1.1 mrg {
664 1.1 mrg char *opath = path, *s = 0;
665 1.1 mrg ssize_t len;
666 1.1 mrg struct stat sb;
667 1.1 mrg
668 1.1 mrg /* check for stdout/stdin */
669 1.1 mrg if (path[0] == '-' && path[1] == '\0') {
670 1.1 mrg if (dflag)
671 1.1 mrg handle_stdin();
672 1.1 mrg else
673 1.1 mrg handle_stdout();
674 1.1 mrg }
675 1.1 mrg
676 1.1 mrg retry:
677 1.1 mrg if (stat(path, &sb) < 0) {
678 1.1 mrg /* lets try <path>.gz if we're decompressing */
679 1.1 mrg if (dflag && s == 0 && errno == ENOENT) {
680 1.1 mrg len = strlen(path);
681 1.1 mrg s = malloc(len + suffix_len);
682 1.1 mrg if (s == 0)
683 1.1 mrg maybe_err(1, "malloc");
684 1.1 mrg memmove(s, path, len);
685 1.1 mrg memmove(&s[len], Sflag, suffix_len);
686 1.1 mrg path = s;
687 1.1 mrg goto retry;
688 1.1 mrg }
689 1.1 mrg maybe_warn("can't stat: %s", opath);
690 1.1 mrg goto out;
691 1.1 mrg }
692 1.1 mrg
693 1.1 mrg if (S_ISDIR(sb.st_mode)) {
694 1.1 mrg if (rflag)
695 1.1 mrg handle_dir(path, &sb);
696 1.1 mrg else
697 1.1 mrg maybe_warn("%s is a directory", path);
698 1.1 mrg goto out;
699 1.1 mrg }
700 1.1 mrg
701 1.1 mrg if (S_ISREG(sb.st_mode))
702 1.1 mrg handle_file(path, &sb);
703 1.1 mrg
704 1.1 mrg out:
705 1.1 mrg if (s)
706 1.1 mrg free(s);
707 1.1 mrg return;
708 1.1 mrg }
709 1.1 mrg
710 1.1 mrg /* compress/decompress a file */
711 1.1 mrg static void
712 1.1 mrg handle_file(char *file, struct stat *sbp)
713 1.1 mrg {
714 1.1 mrg ssize_t usize, gsize;
715 1.1 mrg
716 1.1 mrg infile = file;
717 1.1 mrg if (dflag) {
718 1.1 mrg usize = file_uncompress(file);
719 1.1 mrg if (usize == 0)
720 1.1 mrg return;
721 1.1 mrg gsize = sbp->st_size;
722 1.1 mrg } else {
723 1.1 mrg gsize = file_compress(file);
724 1.1 mrg if (gsize == 0)
725 1.1 mrg return;
726 1.1 mrg usize = sbp->st_size;
727 1.1 mrg }
728 1.1 mrg
729 1.1 mrg if (vflag && !tflag)
730 1.1 mrg print_verbage(file, cflag == 0 ? newfile : 0, usize, gsize);
731 1.1 mrg }
732 1.1 mrg
733 1.1 mrg /* this is used with -r to recursively decend directories */
734 1.1 mrg static void
735 1.1 mrg handle_dir(char *dir, struct stat *sbp)
736 1.1 mrg {
737 1.1 mrg char *path_argv[2];
738 1.1 mrg FTS *fts;
739 1.1 mrg FTSENT *entry;
740 1.1 mrg
741 1.1 mrg path_argv[0] = dir;
742 1.1 mrg path_argv[1] = 0;
743 1.1 mrg fts = fts_open(path_argv, FTS_PHYSICAL, NULL);
744 1.1 mrg if (fts == NULL) {
745 1.1 mrg warn("couldn't fts_open %s", dir);
746 1.1 mrg return;
747 1.1 mrg }
748 1.1 mrg
749 1.1 mrg while ((entry = fts_read(fts))) {
750 1.1 mrg switch(entry->fts_info) {
751 1.1 mrg case FTS_D:
752 1.1 mrg case FTS_DP:
753 1.1 mrg continue;
754 1.1 mrg
755 1.1 mrg case FTS_DNR:
756 1.1 mrg case FTS_ERR:
757 1.1 mrg case FTS_NS:
758 1.1 mrg maybe_warn("%s", entry->fts_path);
759 1.1 mrg continue;
760 1.1 mrg case FTS_F:
761 1.1 mrg handle_file(entry->fts_name, entry->fts_statp);
762 1.1 mrg }
763 1.1 mrg }
764 1.1 mrg (void)fts_close(fts);
765 1.1 mrg }
766 1.1 mrg
767 1.6 mrg /* print a ratio */
768 1.6 mrg static void
769 1.6 mrg print_ratio(off_t in, off_t out, FILE *where)
770 1.6 mrg {
771 1.6 mrg u_int64_t percent;
772 1.6 mrg
773 1.6 mrg if (out == 0)
774 1.6 mrg percent = 0;
775 1.6 mrg else if (out < 1000 * 1000)
776 1.6 mrg percent = 999 - (in * 1000) / out;
777 1.6 mrg else
778 1.6 mrg percent = 999 - in / (out / 1000);
779 1.6 mrg fprintf(where, "%3lu.%1lu%%", (unsigned long)percent / 10UL,
780 1.6 mrg (unsigned long)percent % 10);
781 1.6 mrg }
782 1.6 mrg
783 1.1 mrg /* print compression statistics, and the new name (if there is one!) */
784 1.1 mrg static void
785 1.4 mrg print_verbage(char *file, char *nfile, ssize_t usize, ssize_t gsize)
786 1.1 mrg {
787 1.6 mrg fprintf(stderr, "%s:%s ", file,
788 1.6 mrg strlen(file) < 7 ? "\t\t" : "\t");
789 1.6 mrg print_ratio(usize, gsize, stderr);
790 1.4 mrg if (nfile)
791 1.4 mrg fprintf(stderr, " -- replaced with %s", nfile);
792 1.1 mrg fprintf(stderr, "\n");
793 1.1 mrg fflush(stderr);
794 1.1 mrg }
795 1.1 mrg
796 1.1 mrg /* print test results */
797 1.1 mrg static void
798 1.1 mrg print_test(char *file, int ok)
799 1.1 mrg {
800 1.1 mrg
801 1.1 mrg fprintf(stderr, "%s:%s %s\n", file,
802 1.1 mrg strlen(file) < 7 ? "\t\t" : "\t", ok ? "OK" : "NOT OK");
803 1.1 mrg fflush(stderr);
804 1.1 mrg }
805 1.1 mrg
806 1.5 mrg /* print a file's info ala --list */
807 1.5 mrg /* eg:
808 1.5 mrg compressed uncompressed ratio uncompressed_name
809 1.5 mrg 354841 1679360 78.8% /usr/pkgsrc/distfiles/libglade-2.0.1.tar
810 1.5 mrg */
811 1.5 mrg static void
812 1.7 mrg print_list(int fd, off_t in, const char *outfile, time_t ts)
813 1.5 mrg {
814 1.5 mrg static int first = 1;
815 1.6 mrg static off_t in_tot, out_tot;
816 1.6 mrg off_t out;
817 1.7 mrg u_int32_t crc;
818 1.5 mrg int rv;
819 1.5 mrg
820 1.7 mrg if (first) {
821 1.7 mrg if (vflag)
822 1.7 mrg printf("method crc date time ");
823 1.7 mrg if (qflag == 0)
824 1.7 mrg printf(" compressed uncompressed "
825 1.7 mrg "ratio uncompressed_name\n");
826 1.7 mrg }
827 1.5 mrg first = 0;
828 1.5 mrg
829 1.6 mrg /* print totals? */
830 1.6 mrg if (fd == -1) {
831 1.6 mrg in = in_tot;
832 1.6 mrg out = out_tot;
833 1.6 mrg } else {
834 1.6 mrg /* read the last 4 bytes - this is the uncompressed size */
835 1.7 mrg rv = lseek(fd, (off_t)(-8), SEEK_END);
836 1.6 mrg if (rv != -1) {
837 1.7 mrg unsigned char buf[8];
838 1.6 mrg u_int32_t usize;
839 1.6 mrg
840 1.6 mrg if (read(fd, (char *)buf, sizeof(buf)) != sizeof(buf))
841 1.6 mrg maybe_err(1, "read of uncompressed size");
842 1.7 mrg crc = buf[0] | buf[1] << 8 | buf[2] << 16 | buf[3] << 24;
843 1.7 mrg usize = buf[4] | buf[5] << 8 | buf[6] << 16 | buf[7] << 24;
844 1.6 mrg out = (off_t)usize;
845 1.6 mrg }
846 1.6 mrg }
847 1.5 mrg
848 1.7 mrg if (vflag && fd == -1)
849 1.7 mrg printf(" ");
850 1.7 mrg else if (vflag) {
851 1.7 mrg char *date = ctime(&ts);
852 1.7 mrg
853 1.7 mrg /* skip the day, 1/100th second, and year */
854 1.7 mrg date += 4;
855 1.7 mrg date[12] = 0;
856 1.7 mrg printf("%5s %08x %11s ", "defla"/*XXX*/, crc, date);
857 1.7 mrg }
858 1.6 mrg printf("%12llu %12llu ", (unsigned long long)in, (unsigned long long)out);
859 1.6 mrg print_ratio(in, out, stdout);
860 1.6 mrg printf(" %s\n", outfile);
861 1.6 mrg in_tot += in;
862 1.6 mrg out_tot += out;
863 1.5 mrg }
864 1.5 mrg
865 1.1 mrg /* display the usage of NetBSD gzip */
866 1.1 mrg static void
867 1.1 mrg usage(void)
868 1.1 mrg {
869 1.1 mrg
870 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
871 1.1 mrg fprintf(stderr,
872 1.1 mrg "Usage: %s [-cdfhnNqrStvV123456789] [<file> [<file> ...]]\n"
873 1.1 mrg " -c --stdout write to stdout, keep original files\n"
874 1.1 mrg " --to-stdout\n"
875 1.1 mrg " -d --decompress uncompress files\n"
876 1.1 mrg " --uncompress\n"
877 1.1 mrg " -f --force force overwriting & compress links\n"
878 1.1 mrg " -h --help display this help\n"
879 1.1 mrg " -n --no-name don't save original file name or time stamp\n"
880 1.1 mrg " -N --name save or restore original file name and time stamp\n"
881 1.1 mrg " -q --quiet output no warnings\n"
882 1.1 mrg " -r --recursive recursively compress files in directories\n"
883 1.1 mrg " -S .suf use suffix .suf instead of .gz\n"
884 1.1 mrg " --suffix .suf\n"
885 1.1 mrg " -t --test test compressed file\n"
886 1.1 mrg " -v --verbose print extra statistics\n"
887 1.1 mrg " -V --version display program version\n"
888 1.1 mrg " -1 --fast fastest (worst) compression\n"
889 1.1 mrg " -2 .. -8 set compression level\n"
890 1.1 mrg " -9 --best best (slowest) compression\n",
891 1.1 mrg getprogname());
892 1.1 mrg fflush(stderr);
893 1.1 mrg exit(0);
894 1.1 mrg }
895 1.1 mrg
896 1.1 mrg /* display the version of NetBSD gzip */
897 1.1 mrg static void
898 1.1 mrg display_version(void)
899 1.1 mrg {
900 1.1 mrg
901 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
902 1.1 mrg fflush(stderr);
903 1.1 mrg exit(0);
904 1.1 mrg }
905