gzip.c revision 1.61 1 1.61 yamt /* $NetBSD: gzip.c,v 1.61 2004/07/12 10:21:11 yamt Exp $ */
2 1.1 mrg
3 1.1 mrg /*
4 1.37 mrg * Copyright (c) 1997, 1998, 2003, 2004 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.37 mrg __COPYRIGHT("@(#) Copyright (c) 1997, 1998, 2003, 2004 Matthew R. Green\n\
34 1.4 mrg All rights reserved.\n");
35 1.61 yamt __RCSID("$NetBSD: gzip.c,v 1.61 2004/07/12 10:21:11 yamt 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 * TODO:
42 1.6 mrg * - handle .taz/.tgz files?
43 1.10 mrg * - use mmap where possible
44 1.16 wiz * - handle some signals better (remove outfile?)
45 1.41 mrg * - make bzip2/compress -v/-t/-l support work as well as possible
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.10 mrg /* what type of file are we dealing with */
66 1.10 mrg enum filetype {
67 1.10 mrg FT_GZIP,
68 1.24 tsutsui #ifndef NO_BZIP2_SUPPORT
69 1.10 mrg FT_BZIP2,
70 1.24 tsutsui #endif
71 1.24 tsutsui #ifndef NO_COMPRESS_SUPPORT
72 1.22 mrg FT_Z,
73 1.24 tsutsui #endif
74 1.10 mrg FT_LAST,
75 1.10 mrg FT_UNKNOWN
76 1.10 mrg };
77 1.10 mrg
78 1.24 tsutsui #ifndef NO_BZIP2_SUPPORT
79 1.10 mrg #include <bzlib.h>
80 1.10 mrg
81 1.24 tsutsui #define BZ2_SUFFIX ".bz2"
82 1.24 tsutsui #define BZIP2_MAGIC "\102\132\150"
83 1.24 tsutsui #endif
84 1.24 tsutsui
85 1.24 tsutsui #ifndef NO_COMPRESS_SUPPORT
86 1.22 mrg #define Z_SUFFIX ".Z"
87 1.22 mrg #define Z_MAGIC "\037\235"
88 1.24 tsutsui #endif
89 1.10 mrg
90 1.10 mrg #define GZ_SUFFIX ".gz"
91 1.1 mrg
92 1.37 mrg #define BUFLEN (64 * 1024)
93 1.1 mrg
94 1.9 mrg #define GZIP_MAGIC0 0x1F
95 1.10 mrg #define GZIP_MAGIC1 0x8B
96 1.10 mrg #define GZIP_OMAGIC1 0x9E
97 1.9 mrg
98 1.48 mrg #define GZIP_TIMESTAMP (off_t)4
99 1.47 mrg #define GZIP_ORIGNAME (off_t)10
100 1.47 mrg
101 1.37 mrg #define HEAD_CRC 0x02
102 1.37 mrg #define EXTRA_FIELD 0x04
103 1.37 mrg #define ORIG_NAME 0x08
104 1.37 mrg #define COMMENT 0x10
105 1.1 mrg
106 1.37 mrg #define OS_CODE 3 /* Unix */
107 1.1 mrg
108 1.56 mrg #ifndef SMALL
109 1.56 mrg static char const *suffixes[] = {
110 1.56 mrg GZ_SUFFIX, ".z", ".taz", ".tgz", "-gz", "-z", "_z",
111 1.56 mrg #ifndef NO_BZIP2_SUPPORT
112 1.56 mrg BZ2_SUFFIX,
113 1.56 mrg #endif
114 1.56 mrg #ifndef NO_COMPRESS_SUPPORT
115 1.56 mrg Z_SUFFIX,
116 1.56 mrg #endif
117 1.56 mrg NULL
118 1.56 mrg };
119 1.56 mrg #endif /* SMALL */
120 1.56 mrg
121 1.53 mrg static const char gzip_version[] = "NetBSD gzip 20040711";
122 1.1 mrg
123 1.1 mrg static int cflag; /* stdout mode */
124 1.1 mrg static int dflag; /* decompress mode */
125 1.25 mrg static int lflag; /* list mode */
126 1.41 mrg static int numflag = 6; /* gzip -1..-9 value */
127 1.26 mrg
128 1.25 mrg #ifndef SMALL
129 1.1 mrg static int fflag; /* force mode */
130 1.1 mrg static int nflag; /* don't save name/timestamp */
131 1.1 mrg static int Nflag; /* don't restore name/timestamp */
132 1.1 mrg static int qflag; /* quiet mode */
133 1.1 mrg static int rflag; /* recursive mode */
134 1.26 mrg static int tflag; /* test */
135 1.26 mrg static char *Sflag;
136 1.26 mrg static int vflag; /* verbose mode */
137 1.25 mrg #else
138 1.25 mrg #define qflag 0
139 1.25 mrg #endif
140 1.26 mrg
141 1.46 mrg static int exit_value = 0; /* exit value */
142 1.46 mrg
143 1.45 agc static const char *suffix;
144 1.10 mrg #define suffix_len (strlen(suffix) + 1) /* len + nul */
145 1.1 mrg static char *infile; /* name of file coming in */
146 1.1 mrg
147 1.46 mrg static void maybe_err(const char *fmt, ...);
148 1.46 mrg static void maybe_errx(const char *fmt, ...);
149 1.1 mrg static void maybe_warn(const char *fmt, ...);
150 1.1 mrg static void maybe_warnx(const char *fmt, ...);
151 1.32 mrg static enum filetype file_gettype(u_char *);
152 1.37 mrg static off_t gz_compress(FILE *, int, off_t *, const char *, time_t);
153 1.46 mrg static off_t gz_uncompress(int, int, char *, size_t, off_t *, const char *);
154 1.45 agc static off_t file_compress(char *, char *, size_t);
155 1.45 agc static off_t file_uncompress(char *, char *, size_t);
156 1.51 mrg static off_t cat_fd(unsigned char *, size_t, off_t *, int fd);
157 1.1 mrg static void handle_pathname(char *);
158 1.1 mrg static void handle_file(char *, struct stat *);
159 1.1 mrg static void handle_stdin(void);
160 1.1 mrg static void handle_stdout(void);
161 1.6 mrg static void print_ratio(off_t, off_t, FILE *);
162 1.7 mrg static void print_list(int fd, off_t, const char *, time_t);
163 1.10 mrg static void usage(void);
164 1.10 mrg static void display_version(void);
165 1.26 mrg
166 1.26 mrg #ifndef SMALL
167 1.26 mrg static void prepend_gzip(char *, int *, char ***);
168 1.26 mrg static void handle_dir(char *, struct stat *);
169 1.31 mrg static void print_verbage(char *, char *, off_t, off_t);
170 1.45 agc static void print_test(const char *, int);
171 1.29 mrg static void copymodes(const char *, struct stat *);
172 1.56 mrg static int check_outfile(const char *outfile, struct stat *sb);
173 1.56 mrg static const char *check_suffix(char *);
174 1.26 mrg #endif
175 1.26 mrg
176 1.24 tsutsui #ifndef NO_BZIP2_SUPPORT
177 1.37 mrg static off_t unbzip2(int, int, char *, size_t, off_t *);
178 1.24 tsutsui #endif
179 1.26 mrg
180 1.24 tsutsui #ifndef NO_COMPRESS_SUPPORT
181 1.37 mrg static FILE *zopen(const char *, FILE *);
182 1.37 mrg static off_t zuncompress(FILE *, FILE *, char *, size_t, off_t *);
183 1.24 tsutsui #endif
184 1.1 mrg
185 1.1 mrg int main(int, char *p[]);
186 1.1 mrg
187 1.11 mrg #ifdef SMALL
188 1.11 mrg #define getopt_long(a,b,c,d,e) getopt(a,b,c)
189 1.11 mrg #else
190 1.3 jdolecek static const struct option longopts[] = {
191 1.1 mrg { "stdout", no_argument, 0, 'c' },
192 1.1 mrg { "to-stdout", no_argument, 0, 'c' },
193 1.1 mrg { "decompress", no_argument, 0, 'd' },
194 1.1 mrg { "uncompress", no_argument, 0, 'd' },
195 1.1 mrg { "force", no_argument, 0, 'f' },
196 1.1 mrg { "help", no_argument, 0, 'h' },
197 1.5 mrg { "list", no_argument, 0, 'l' },
198 1.1 mrg { "no-name", no_argument, 0, 'n' },
199 1.1 mrg { "name", no_argument, 0, 'N' },
200 1.1 mrg { "quiet", no_argument, 0, 'q' },
201 1.1 mrg { "recursive", no_argument, 0, 'r' },
202 1.1 mrg { "suffix", required_argument, 0, 'S' },
203 1.1 mrg { "test", no_argument, 0, 't' },
204 1.1 mrg { "verbose", no_argument, 0, 'v' },
205 1.1 mrg { "version", no_argument, 0, 'V' },
206 1.1 mrg { "fast", no_argument, 0, '1' },
207 1.1 mrg { "best", no_argument, 0, '9' },
208 1.1 mrg #if 0
209 1.1 mrg /*
210 1.10 mrg * This is what else GNU gzip implements. --ascii isn't useful
211 1.10 mrg * on NetBSD, and I don't care to have a --license.
212 1.1 mrg */
213 1.1 mrg { "ascii", no_argument, 0, 'a' },
214 1.1 mrg { "license", no_argument, 0, 'L' },
215 1.1 mrg #endif
216 1.12 mrg { NULL, no_argument, 0, 0 },
217 1.1 mrg };
218 1.11 mrg #endif
219 1.1 mrg
220 1.1 mrg int
221 1.1 mrg main(int argc, char **argv)
222 1.1 mrg {
223 1.28 mrg const char *progname = getprogname();
224 1.25 mrg #ifndef SMALL
225 1.15 mrg char *gzip;
226 1.25 mrg #endif
227 1.1 mrg int ch;
228 1.1 mrg
229 1.15 mrg /* XXX set up signals */
230 1.15 mrg
231 1.56 mrg suffix = GZ_SUFFIX;
232 1.10 mrg
233 1.25 mrg #ifndef SMALL
234 1.15 mrg if ((gzip = getenv("GZIP")) != NULL)
235 1.15 mrg prepend_gzip(gzip, &argc, &argv);
236 1.28 mrg #endif
237 1.15 mrg
238 1.1 mrg /*
239 1.1 mrg * XXX
240 1.1 mrg * handle being called `gunzip', `zcat' and `gzcat'
241 1.1 mrg */
242 1.1 mrg if (strcmp(progname, "gunzip") == 0)
243 1.1 mrg dflag = 1;
244 1.1 mrg else if (strcmp(progname, "zcat") == 0 ||
245 1.1 mrg strcmp(progname, "gzcat") == 0)
246 1.1 mrg dflag = cflag = 1;
247 1.25 mrg
248 1.25 mrg #ifdef SMALL
249 1.31 mrg #define OPT_LIST "cdhHltV123456789"
250 1.25 mrg #else
251 1.28 mrg #define OPT_LIST "cdfhHlnNqrS:tvV123456789"
252 1.25 mrg #endif
253 1.1 mrg
254 1.28 mrg while ((ch = getopt_long(argc, argv, OPT_LIST, longopts, NULL)) != -1)
255 1.1 mrg switch (ch) {
256 1.1 mrg case 'c':
257 1.1 mrg cflag = 1;
258 1.1 mrg break;
259 1.1 mrg case 'd':
260 1.1 mrg dflag = 1;
261 1.1 mrg break;
262 1.5 mrg case 'l':
263 1.5 mrg lflag = 1;
264 1.5 mrg dflag = 1;
265 1.5 mrg break;
266 1.26 mrg case 'V':
267 1.26 mrg display_version();
268 1.26 mrg /* NOTREACHED */
269 1.26 mrg case '1': case '2': case '3':
270 1.26 mrg case '4': case '5': case '6':
271 1.26 mrg case '7': case '8': case '9':
272 1.38 mrg numflag = ch - '0';
273 1.26 mrg break;
274 1.25 mrg #ifndef SMALL
275 1.25 mrg case 'f':
276 1.25 mrg fflag = 1;
277 1.25 mrg break;
278 1.1 mrg case 'n':
279 1.1 mrg nflag = 1;
280 1.1 mrg Nflag = 0;
281 1.1 mrg break;
282 1.1 mrg case 'N':
283 1.1 mrg nflag = 0;
284 1.1 mrg Nflag = 1;
285 1.1 mrg break;
286 1.1 mrg case 'q':
287 1.1 mrg qflag = 1;
288 1.1 mrg break;
289 1.1 mrg case 'r':
290 1.1 mrg rflag = 1;
291 1.1 mrg break;
292 1.1 mrg case 'S':
293 1.1 mrg Sflag = optarg;
294 1.1 mrg break;
295 1.1 mrg case 't':
296 1.1 mrg cflag = 1;
297 1.1 mrg tflag = 1;
298 1.1 mrg dflag = 1;
299 1.1 mrg break;
300 1.1 mrg case 'v':
301 1.1 mrg vflag = 1;
302 1.1 mrg break;
303 1.26 mrg #endif
304 1.12 mrg default:
305 1.12 mrg usage();
306 1.12 mrg /* NOTREACHED */
307 1.1 mrg }
308 1.1 mrg argv += optind;
309 1.1 mrg argc -= optind;
310 1.1 mrg
311 1.1 mrg if (argc == 0) {
312 1.1 mrg if (dflag) /* stdin mode */
313 1.1 mrg handle_stdin();
314 1.1 mrg else /* stdout mode */
315 1.1 mrg handle_stdout();
316 1.1 mrg } else {
317 1.1 mrg do {
318 1.1 mrg handle_pathname(argv[0]);
319 1.5 mrg } while (*++argv);
320 1.1 mrg }
321 1.29 mrg #ifndef SMALL
322 1.6 mrg if (qflag == 0 && lflag && argc > 1)
323 1.7 mrg print_list(-1, 0, "(totals)", 0);
324 1.29 mrg #endif
325 1.46 mrg exit(exit_value);
326 1.1 mrg }
327 1.1 mrg
328 1.1 mrg /* maybe print a warning */
329 1.1 mrg void
330 1.1 mrg maybe_warn(const char *fmt, ...)
331 1.1 mrg {
332 1.1 mrg va_list ap;
333 1.1 mrg
334 1.1 mrg if (qflag == 0) {
335 1.1 mrg va_start(ap, fmt);
336 1.1 mrg vwarn(fmt, ap);
337 1.1 mrg va_end(ap);
338 1.1 mrg }
339 1.46 mrg if (exit_value == 0)
340 1.46 mrg exit_value = 1;
341 1.1 mrg }
342 1.1 mrg
343 1.46 mrg /* ... without an errno. */
344 1.1 mrg void
345 1.1 mrg maybe_warnx(const char *fmt, ...)
346 1.1 mrg {
347 1.1 mrg va_list ap;
348 1.1 mrg
349 1.1 mrg if (qflag == 0) {
350 1.1 mrg va_start(ap, fmt);
351 1.1 mrg vwarnx(fmt, ap);
352 1.1 mrg va_end(ap);
353 1.1 mrg }
354 1.46 mrg if (exit_value == 0)
355 1.46 mrg exit_value = 1;
356 1.1 mrg }
357 1.1 mrg
358 1.46 mrg /* maybe print an error */
359 1.1 mrg void
360 1.46 mrg maybe_err(const char *fmt, ...)
361 1.1 mrg {
362 1.1 mrg va_list ap;
363 1.1 mrg
364 1.1 mrg if (qflag == 0) {
365 1.1 mrg va_start(ap, fmt);
366 1.1 mrg vwarn(fmt, ap);
367 1.1 mrg va_end(ap);
368 1.1 mrg }
369 1.46 mrg exit(2);
370 1.1 mrg }
371 1.1 mrg
372 1.46 mrg /* ... without an errno. */
373 1.9 mrg void
374 1.46 mrg maybe_errx(const char *fmt, ...)
375 1.9 mrg {
376 1.9 mrg va_list ap;
377 1.9 mrg
378 1.9 mrg if (qflag == 0) {
379 1.9 mrg va_start(ap, fmt);
380 1.9 mrg vwarnx(fmt, ap);
381 1.9 mrg va_end(ap);
382 1.9 mrg }
383 1.46 mrg exit(2);
384 1.9 mrg }
385 1.9 mrg
386 1.25 mrg #ifndef SMALL
387 1.15 mrg /* split up $GZIP and prepend it to the argument list */
388 1.15 mrg static void
389 1.15 mrg prepend_gzip(char *gzip, int *argc, char ***argv)
390 1.15 mrg {
391 1.15 mrg char *s, **nargv, **ac;
392 1.15 mrg int nenvarg = 0, i;
393 1.15 mrg
394 1.15 mrg /* scan how many arguments there are */
395 1.15 mrg for (s = gzip; *s; s++) {
396 1.15 mrg if (*s == ' ' || *s == '\t')
397 1.15 mrg continue;
398 1.15 mrg nenvarg++;
399 1.15 mrg for (; *s; s++)
400 1.15 mrg if (*s == ' ' || *s == '\t')
401 1.15 mrg break;
402 1.45 agc if (*s == 0x0)
403 1.18 mrg break;
404 1.15 mrg }
405 1.15 mrg /* punt early */
406 1.15 mrg if (nenvarg == 0)
407 1.15 mrg return;
408 1.15 mrg
409 1.15 mrg *argc += nenvarg;
410 1.15 mrg ac = *argv;
411 1.15 mrg
412 1.17 simonb nargv = (char **)malloc((*argc + 1) * sizeof(char *));
413 1.15 mrg if (nargv == NULL)
414 1.46 mrg maybe_err("malloc");
415 1.15 mrg
416 1.15 mrg /* stash this away */
417 1.15 mrg *argv = nargv;
418 1.15 mrg
419 1.15 mrg /* copy the program name first */
420 1.15 mrg i = 0;
421 1.15 mrg nargv[i++] = *(ac++);
422 1.15 mrg
423 1.15 mrg /* take a copy of $GZIP and add it to the array */
424 1.15 mrg s = strdup(gzip);
425 1.15 mrg if (s == NULL)
426 1.46 mrg maybe_err("strdup");
427 1.15 mrg for (; *s; s++) {
428 1.15 mrg if (*s == ' ' || *s == '\t')
429 1.15 mrg continue;
430 1.15 mrg nargv[i++] = s;
431 1.15 mrg for (; *s; s++)
432 1.15 mrg if (*s == ' ' || *s == '\t') {
433 1.15 mrg *s = 0;
434 1.15 mrg break;
435 1.15 mrg }
436 1.15 mrg }
437 1.15 mrg
438 1.15 mrg /* copy the original arguments and a NULL */
439 1.15 mrg while (*ac)
440 1.15 mrg nargv[i++] = *(ac++);
441 1.15 mrg nargv[i] = NULL;
442 1.15 mrg }
443 1.25 mrg #endif
444 1.15 mrg
445 1.1 mrg /* compress input to output then close both files */
446 1.37 mrg static off_t
447 1.37 mrg gz_compress(FILE *in, int out, off_t *gsizep, const char *origname, time_t mtime)
448 1.1 mrg {
449 1.37 mrg z_stream z;
450 1.37 mrg char inbuf[BUFLEN], outbuf[BUFLEN];
451 1.37 mrg off_t in_tot = 0, out_tot = 0;
452 1.37 mrg ssize_t in_size;
453 1.37 mrg char *str;
454 1.37 mrg int i, error;
455 1.37 mrg uLong crc;
456 1.37 mrg
457 1.37 mrg i = asprintf(&str, "%c%c%c%c%c%c%c%c%c%c%s",
458 1.37 mrg GZIP_MAGIC0, GZIP_MAGIC1,
459 1.37 mrg Z_DEFLATED, origname ? ORIG_NAME : 0,
460 1.37 mrg (int)mtime & 0xff,
461 1.37 mrg (int)(mtime >> 8) & 0xff,
462 1.37 mrg (int)(mtime >> 16) & 0xff,
463 1.37 mrg (int)(mtime >> 24) & 0xff,
464 1.37 mrg 0, OS_CODE, origname ? origname : "");
465 1.37 mrg if (i == -1)
466 1.46 mrg maybe_err("asprintf");
467 1.37 mrg if (origname)
468 1.37 mrg i++;
469 1.46 mrg if (write(out, str, i) != i) {
470 1.46 mrg maybe_warn("write");
471 1.46 mrg in_tot = -1;
472 1.46 mrg goto out;
473 1.46 mrg }
474 1.37 mrg free(str);
475 1.37 mrg
476 1.37 mrg memset(&z, 0, sizeof z);
477 1.37 mrg z.next_out = outbuf;
478 1.37 mrg z.avail_out = sizeof outbuf;
479 1.38 mrg z.zalloc = Z_NULL;
480 1.38 mrg z.zfree = Z_NULL;
481 1.37 mrg z.opaque = 0;
482 1.37 mrg
483 1.37 mrg error = deflateInit2(&z, numflag, Z_DEFLATED,
484 1.37 mrg -MAX_WBITS, 8, Z_DEFAULT_STRATEGY);
485 1.46 mrg if (error != Z_OK) {
486 1.46 mrg maybe_warnx("deflateInit2 failed");
487 1.46 mrg in_tot = -1;
488 1.46 mrg goto out;
489 1.46 mrg }
490 1.37 mrg
491 1.37 mrg crc = crc32(0L, Z_NULL, 0);
492 1.37 mrg for (;;) {
493 1.37 mrg if (z.avail_out == 0) {
494 1.46 mrg if (write(out, outbuf, sizeof outbuf) != sizeof outbuf) {
495 1.46 mrg maybe_warn("write");
496 1.46 mrg in_tot = -1;
497 1.46 mrg goto out;
498 1.46 mrg }
499 1.37 mrg
500 1.37 mrg out_tot += sizeof outbuf;
501 1.37 mrg z.next_out = outbuf;
502 1.37 mrg z.avail_out = sizeof outbuf;
503 1.37 mrg }
504 1.37 mrg
505 1.37 mrg if (z.avail_in == 0) {
506 1.37 mrg in_size = fread(inbuf, 1, sizeof inbuf, in);
507 1.46 mrg if (ferror(in)) {
508 1.46 mrg maybe_warn("fread");
509 1.46 mrg in_tot = -1;
510 1.46 mrg goto out;
511 1.46 mrg }
512 1.37 mrg if (in_size == 0)
513 1.37 mrg break;
514 1.37 mrg
515 1.37 mrg crc = crc32(crc, (const Bytef *)inbuf, (unsigned)in_size);
516 1.37 mrg in_tot += in_size;
517 1.37 mrg z.next_in = inbuf;
518 1.37 mrg z.avail_in = in_size;
519 1.37 mrg }
520 1.37 mrg
521 1.37 mrg error = deflate(&z, Z_NO_FLUSH);
522 1.46 mrg if (error != Z_OK && error != Z_STREAM_END) {
523 1.46 mrg maybe_warnx("deflate failed");
524 1.46 mrg in_tot = -1;
525 1.46 mrg goto out;
526 1.46 mrg }
527 1.37 mrg }
528 1.1 mrg
529 1.37 mrg /* clean up */
530 1.1 mrg for (;;) {
531 1.37 mrg size_t len;
532 1.37 mrg
533 1.37 mrg error = deflate(&z, Z_FINISH);
534 1.46 mrg if (error != Z_OK && error != Z_STREAM_END) {
535 1.46 mrg maybe_warnx("deflate failed");
536 1.46 mrg in_tot = -1;
537 1.46 mrg goto out;
538 1.46 mrg }
539 1.37 mrg
540 1.37 mrg len = sizeof outbuf - z.avail_out;
541 1.37 mrg
542 1.46 mrg if (write(out, outbuf, len) != len) {
543 1.46 mrg maybe_warn("write");
544 1.46 mrg out_tot = -1;
545 1.46 mrg goto out;
546 1.46 mrg }
547 1.37 mrg out_tot += len;
548 1.37 mrg z.next_out = outbuf;
549 1.37 mrg z.avail_out = sizeof outbuf;
550 1.37 mrg
551 1.37 mrg if (error == Z_STREAM_END)
552 1.1 mrg break;
553 1.37 mrg }
554 1.37 mrg
555 1.46 mrg if (deflateEnd(&z) != Z_OK) {
556 1.46 mrg maybe_warnx("deflateEnd failed");
557 1.46 mrg in_tot = -1;
558 1.46 mrg goto out;
559 1.46 mrg }
560 1.37 mrg
561 1.37 mrg i = asprintf(&str, "%c%c%c%c%c%c%c%c",
562 1.37 mrg (int)crc & 0xff,
563 1.37 mrg (int)(crc >> 8) & 0xff,
564 1.37 mrg (int)(crc >> 16) & 0xff,
565 1.37 mrg (int)(crc >> 24) & 0xff,
566 1.37 mrg (int)in_tot & 0xff,
567 1.37 mrg (int)(in_tot >> 8) & 0xff,
568 1.37 mrg (int)(in_tot >> 16) & 0xff,
569 1.37 mrg (int)(in_tot >> 24) & 0xff);
570 1.37 mrg if (i != 8)
571 1.46 mrg maybe_err("asprintf");
572 1.46 mrg if (write(out, str, i) != i) {
573 1.46 mrg maybe_warn("write");
574 1.46 mrg in_tot = -1;
575 1.46 mrg }
576 1.37 mrg free(str);
577 1.1 mrg
578 1.46 mrg out:
579 1.37 mrg if (gsizep)
580 1.37 mrg *gsizep = out_tot;
581 1.37 mrg return in_tot;
582 1.1 mrg }
583 1.1 mrg
584 1.37 mrg /*
585 1.37 mrg * uncompress input to output then close the input. return the
586 1.37 mrg * uncompressed size written, and put the compressed sized read
587 1.37 mrg * into `*gsizep'.
588 1.37 mrg */
589 1.1 mrg static off_t
590 1.46 mrg gz_uncompress(int in, int out, char *pre, size_t prelen, off_t *gsizep,
591 1.46 mrg const char *filename)
592 1.1 mrg {
593 1.37 mrg z_stream z;
594 1.37 mrg char outbuf[BUFLEN], inbuf[BUFLEN];
595 1.54 mrg off_t out_tot, out_sub_tot, in_tot;
596 1.37 mrg enum {
597 1.37 mrg GZSTATE_MAGIC0,
598 1.37 mrg GZSTATE_MAGIC1,
599 1.37 mrg GZSTATE_METHOD,
600 1.37 mrg GZSTATE_FLAGS,
601 1.37 mrg GZSTATE_SKIPPING,
602 1.37 mrg GZSTATE_EXTRA,
603 1.37 mrg GZSTATE_EXTRA2,
604 1.37 mrg GZSTATE_EXTRA3,
605 1.37 mrg GZSTATE_ORIGNAME,
606 1.37 mrg GZSTATE_COMMENT,
607 1.37 mrg GZSTATE_HEAD_CRC1,
608 1.37 mrg GZSTATE_HEAD_CRC2,
609 1.37 mrg GZSTATE_INIT,
610 1.55 mrg GZSTATE_READ,
611 1.55 mrg GZSTATE_CRC,
612 1.55 mrg GZSTATE_LEN,
613 1.37 mrg } state = GZSTATE_MAGIC0;
614 1.37 mrg int flags = 0, skip_count = 0;
615 1.37 mrg int error, done_reading = 0;
616 1.54 mrg uLong crc;
617 1.37 mrg
618 1.37 mrg #define ADVANCE() { z.next_in++; z.avail_in--; }
619 1.37 mrg
620 1.37 mrg memset(&z, 0, sizeof z);
621 1.37 mrg z.avail_in = prelen;
622 1.37 mrg z.next_in = pre;
623 1.37 mrg z.avail_out = sizeof outbuf;
624 1.37 mrg z.next_out = outbuf;
625 1.37 mrg z.zalloc = NULL;
626 1.37 mrg z.zfree = NULL;
627 1.37 mrg z.opaque = 0;
628 1.37 mrg
629 1.37 mrg in_tot = prelen;
630 1.37 mrg out_tot = 0;
631 1.1 mrg
632 1.37 mrg for (;;) {
633 1.37 mrg if (z.avail_in == 0 && done_reading == 0) {
634 1.37 mrg size_t in_size = read(in, inbuf, BUFLEN);
635 1.1 mrg
636 1.37 mrg if (in_size == -1) {
637 1.26 mrg #ifndef SMALL
638 1.46 mrg if (tflag && vflag)
639 1.46 mrg print_test(filename, 0);
640 1.37 mrg #endif
641 1.51 mrg maybe_warn("failed to read stdin");
642 1.46 mrg out_tot = -1;
643 1.46 mrg goto stop;
644 1.37 mrg } else if (in_size == 0)
645 1.37 mrg done_reading = 1;
646 1.37 mrg
647 1.37 mrg z.avail_in = in_size;
648 1.37 mrg z.next_in = inbuf;
649 1.37 mrg
650 1.37 mrg in_tot += in_size;
651 1.37 mrg }
652 1.54 mrg if (z.avail_in == 0) {
653 1.55 mrg if (done_reading && state != GZSTATE_MAGIC0)
654 1.55 mrg maybe_warnx("%s: unexpected end of file",
655 1.55 mrg filename);
656 1.54 mrg goto stop;
657 1.54 mrg }
658 1.37 mrg switch (state) {
659 1.37 mrg case GZSTATE_MAGIC0:
660 1.46 mrg if (*z.next_in != GZIP_MAGIC0) {
661 1.55 mrg maybe_warnx("input not gziped (MAGIC0)");
662 1.46 mrg out_tot = -1;
663 1.46 mrg goto stop;
664 1.46 mrg }
665 1.37 mrg ADVANCE();
666 1.37 mrg state++;
667 1.54 mrg out_sub_tot = 0;
668 1.54 mrg crc = crc32(0L, Z_NULL, 0);
669 1.37 mrg break;
670 1.37 mrg
671 1.37 mrg case GZSTATE_MAGIC1:
672 1.37 mrg if (*z.next_in != GZIP_MAGIC1 &&
673 1.46 mrg *z.next_in != GZIP_OMAGIC1) {
674 1.55 mrg maybe_warnx("input not gziped (MAGIC1)");
675 1.46 mrg out_tot = -1;
676 1.46 mrg goto stop;
677 1.46 mrg }
678 1.37 mrg ADVANCE();
679 1.37 mrg state++;
680 1.37 mrg break;
681 1.37 mrg
682 1.37 mrg case GZSTATE_METHOD:
683 1.46 mrg if (*z.next_in != Z_DEFLATED) {
684 1.51 mrg maybe_warnx("unknown compression method");
685 1.46 mrg out_tot = -1;
686 1.46 mrg goto stop;
687 1.46 mrg }
688 1.37 mrg ADVANCE();
689 1.37 mrg state++;
690 1.37 mrg break;
691 1.37 mrg
692 1.37 mrg case GZSTATE_FLAGS:
693 1.37 mrg flags = *z.next_in;
694 1.37 mrg ADVANCE();
695 1.37 mrg skip_count = 6;
696 1.37 mrg state++;
697 1.37 mrg break;
698 1.37 mrg
699 1.37 mrg case GZSTATE_SKIPPING:
700 1.37 mrg if (skip_count > 0) {
701 1.37 mrg skip_count--;
702 1.37 mrg ADVANCE();
703 1.37 mrg } else
704 1.37 mrg state++;
705 1.37 mrg break;
706 1.37 mrg
707 1.37 mrg case GZSTATE_EXTRA:
708 1.37 mrg if ((flags & EXTRA_FIELD) == 0) {
709 1.37 mrg state = GZSTATE_ORIGNAME;
710 1.37 mrg break;
711 1.37 mrg }
712 1.37 mrg skip_count = *z.next_in;
713 1.37 mrg ADVANCE();
714 1.37 mrg state++;
715 1.37 mrg break;
716 1.37 mrg
717 1.37 mrg case GZSTATE_EXTRA2:
718 1.37 mrg skip_count |= ((*z.next_in) << 8);
719 1.37 mrg ADVANCE();
720 1.37 mrg state++;
721 1.37 mrg break;
722 1.37 mrg
723 1.37 mrg case GZSTATE_EXTRA3:
724 1.37 mrg if (skip_count > 0) {
725 1.37 mrg skip_count--;
726 1.37 mrg ADVANCE();
727 1.1 mrg } else
728 1.37 mrg state++;
729 1.37 mrg break;
730 1.37 mrg
731 1.37 mrg case GZSTATE_ORIGNAME:
732 1.37 mrg if ((flags & ORIG_NAME) == 0) {
733 1.37 mrg state++;
734 1.37 mrg break;
735 1.37 mrg }
736 1.37 mrg if (*z.next_in == 0)
737 1.37 mrg state++;
738 1.37 mrg ADVANCE();
739 1.37 mrg break;
740 1.37 mrg
741 1.37 mrg case GZSTATE_COMMENT:
742 1.37 mrg if ((flags & COMMENT) == 0) {
743 1.37 mrg state++;
744 1.37 mrg break;
745 1.37 mrg }
746 1.37 mrg if (*z.next_in == 0)
747 1.37 mrg state++;
748 1.37 mrg ADVANCE();
749 1.37 mrg break;
750 1.37 mrg
751 1.37 mrg case GZSTATE_HEAD_CRC1:
752 1.37 mrg if (flags & HEAD_CRC)
753 1.37 mrg skip_count = 2;
754 1.37 mrg else
755 1.37 mrg skip_count = 0;
756 1.37 mrg state++;
757 1.37 mrg break;
758 1.37 mrg
759 1.37 mrg case GZSTATE_HEAD_CRC2:
760 1.37 mrg if (skip_count > 0) {
761 1.37 mrg skip_count--;
762 1.37 mrg ADVANCE();
763 1.37 mrg } else
764 1.37 mrg state++;
765 1.37 mrg break;
766 1.37 mrg
767 1.37 mrg case GZSTATE_INIT:
768 1.37 mrg if (inflateInit2(&z, -MAX_WBITS) != Z_OK) {
769 1.51 mrg maybe_warnx("failed to inflateInit");
770 1.46 mrg out_tot = -1;
771 1.37 mrg goto stop;
772 1.37 mrg }
773 1.37 mrg state++;
774 1.37 mrg break;
775 1.37 mrg
776 1.37 mrg case GZSTATE_READ:
777 1.37 mrg error = inflate(&z, Z_FINISH);
778 1.50 mrg /* Z_BUF_ERROR goes with Z_FINISH... */
779 1.37 mrg if (error == Z_STREAM_END || error == Z_BUF_ERROR) {
780 1.57 mrg ssize_t wr = BUFLEN - z.avail_out;
781 1.37 mrg
782 1.50 mrg /* Nothing left? */
783 1.50 mrg if (wr == 0)
784 1.46 mrg goto stop;
785 1.50 mrg
786 1.57 mrg crc = crc32(crc, (const Bytef *)outbuf, (unsigned)wr);
787 1.37 mrg if (
788 1.37 mrg #ifndef SMALL
789 1.37 mrg /* don't write anything with -t */
790 1.37 mrg tflag == 0 &&
791 1.37 mrg #endif
792 1.46 mrg write(out, outbuf, wr) != wr) {
793 1.52 yamt maybe_warn("error writing to output");
794 1.46 mrg out_tot = -1;
795 1.46 mrg goto stop;
796 1.46 mrg }
797 1.39 mrg
798 1.39 mrg out_tot += wr;
799 1.54 mrg out_sub_tot += wr;
800 1.37 mrg
801 1.53 mrg if (error == Z_STREAM_END) {
802 1.53 mrg inflateEnd(&z);
803 1.55 mrg state++;
804 1.53 mrg }
805 1.39 mrg
806 1.37 mrg z.next_out = outbuf;
807 1.37 mrg z.avail_out = BUFLEN;
808 1.37 mrg
809 1.37 mrg break;
810 1.37 mrg }
811 1.55 mrg case GZSTATE_CRC:
812 1.55 mrg {
813 1.55 mrg static int empty_buffer = 0;
814 1.55 mrg uLong origcrc;
815 1.55 mrg
816 1.55 mrg if (z.avail_in < 4) {
817 1.55 mrg if (!done_reading && empty_buffer++ < 4)
818 1.55 mrg continue;
819 1.55 mrg maybe_warnx("truncated input");
820 1.57 mrg out_tot = -1;
821 1.55 mrg goto stop;
822 1.55 mrg }
823 1.55 mrg empty_buffer = 0;
824 1.57 mrg origcrc = ((unsigned)z.next_in[0] & 0xff) |
825 1.57 mrg ((unsigned)z.next_in[1] & 0xff) << 8 |
826 1.57 mrg ((unsigned)z.next_in[2] & 0xff) << 16 |
827 1.57 mrg ((unsigned)z.next_in[3] & 0xff) << 24;
828 1.57 mrg if (origcrc != crc) {
829 1.55 mrg maybe_warnx("invalid compressed"
830 1.55 mrg " data--crc error");
831 1.57 mrg out_tot = -1;
832 1.57 mrg goto stop;
833 1.57 mrg }
834 1.55 mrg }
835 1.55 mrg
836 1.55 mrg z.avail_in -= 4;
837 1.55 mrg z.next_in += 4;
838 1.55 mrg
839 1.55 mrg if (!z.avail_in)
840 1.55 mrg goto stop;
841 1.55 mrg state++;
842 1.55 mrg break;
843 1.55 mrg case GZSTATE_LEN:
844 1.55 mrg {
845 1.55 mrg static int empty_buffer = 0;
846 1.55 mrg uLong origlen;
847 1.55 mrg
848 1.55 mrg if (z.avail_in < 4) {
849 1.55 mrg if (!done_reading && empty_buffer++ < 4)
850 1.55 mrg continue;
851 1.55 mrg maybe_warnx("truncated input");
852 1.57 mrg out_tot = -1;
853 1.55 mrg goto stop;
854 1.55 mrg }
855 1.55 mrg empty_buffer = 0;
856 1.57 mrg origlen = ((unsigned)z.next_in[0] & 0xff) |
857 1.57 mrg ((unsigned)z.next_in[1] & 0xff) << 8 |
858 1.57 mrg ((unsigned)z.next_in[2] & 0xff) << 16 |
859 1.57 mrg ((unsigned)z.next_in[3] & 0xff) << 24;
860 1.55 mrg
861 1.57 mrg if (origlen != out_sub_tot) {
862 1.55 mrg maybe_warnx("invalid compressed"
863 1.55 mrg " data--length error");
864 1.57 mrg out_tot = -1;
865 1.57 mrg goto stop;
866 1.57 mrg }
867 1.55 mrg }
868 1.55 mrg
869 1.55 mrg z.avail_in -= 4;
870 1.55 mrg z.next_in += 4;
871 1.55 mrg
872 1.37 mrg if (error < 0) {
873 1.51 mrg maybe_warnx("decompression error");
874 1.37 mrg out_tot = -1;
875 1.37 mrg goto stop;
876 1.37 mrg }
877 1.55 mrg state = GZSTATE_MAGIC0;
878 1.1 mrg break;
879 1.1 mrg }
880 1.37 mrg continue;
881 1.37 mrg stop:
882 1.37 mrg break;
883 1.37 mrg }
884 1.37 mrg if (state > GZSTATE_INIT)
885 1.37 mrg inflateEnd(&z);
886 1.1 mrg
887 1.26 mrg #ifndef SMALL
888 1.46 mrg if (tflag && vflag)
889 1.46 mrg print_test(filename, out_tot != -1);
890 1.26 mrg #endif
891 1.1 mrg
892 1.37 mrg if (gsizep)
893 1.37 mrg *gsizep = in_tot;
894 1.37 mrg return (out_tot);
895 1.1 mrg }
896 1.1 mrg
897 1.29 mrg #ifndef SMALL
898 1.1 mrg /*
899 1.1 mrg * set the owner, mode, flags & utimes for a file
900 1.1 mrg */
901 1.1 mrg static void
902 1.1 mrg copymodes(const char *file, struct stat *sbp)
903 1.1 mrg {
904 1.1 mrg struct timeval times[2];
905 1.1 mrg
906 1.1 mrg /*
907 1.1 mrg * If we have no info on the input, give this file some
908 1.1 mrg * default values and return..
909 1.1 mrg */
910 1.1 mrg if (sbp == NULL) {
911 1.1 mrg mode_t mask = umask(022);
912 1.1 mrg
913 1.1 mrg (void)chmod(file, DEFFILEMODE & ~mask);
914 1.1 mrg (void)umask(mask);
915 1.1 mrg return;
916 1.1 mrg }
917 1.1 mrg
918 1.1 mrg /* if the chown fails, remove set-id bits as-per compress(1) */
919 1.1 mrg if (chown(file, sbp->st_uid, sbp->st_gid) < 0) {
920 1.1 mrg if (errno != EPERM)
921 1.1 mrg maybe_warn("couldn't chown: %s", file);
922 1.1 mrg sbp->st_mode &= ~(S_ISUID|S_ISGID);
923 1.1 mrg }
924 1.1 mrg
925 1.1 mrg /* we only allow set-id and the 9 normal permission bits */
926 1.1 mrg sbp->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
927 1.1 mrg if (chmod(file, sbp->st_mode) < 0)
928 1.1 mrg maybe_warn("couldn't chmod: %s", file);
929 1.1 mrg
930 1.1 mrg /* only try flags if they exist already */
931 1.1 mrg if (sbp->st_flags != 0 && chflags(file, sbp->st_flags) < 0)
932 1.1 mrg maybe_warn("couldn't chflags: %s", file);
933 1.1 mrg
934 1.1 mrg TIMESPEC_TO_TIMEVAL(×[0], &sbp->st_atimespec);
935 1.1 mrg TIMESPEC_TO_TIMEVAL(×[1], &sbp->st_mtimespec);
936 1.1 mrg if (utimes(file, times) < 0)
937 1.1 mrg maybe_warn("couldn't utimes: %s", file);
938 1.1 mrg }
939 1.29 mrg #endif
940 1.1 mrg
941 1.32 mrg /* what sort of file is this? */
942 1.32 mrg static enum filetype
943 1.32 mrg file_gettype(u_char *buf)
944 1.32 mrg {
945 1.32 mrg
946 1.32 mrg if (buf[0] == GZIP_MAGIC0 &&
947 1.32 mrg (buf[1] == GZIP_MAGIC1 || buf[1] == GZIP_OMAGIC1))
948 1.32 mrg return FT_GZIP;
949 1.32 mrg else
950 1.32 mrg #ifndef NO_BZIP2_SUPPORT
951 1.32 mrg if (memcmp(buf, BZIP2_MAGIC, 3) == 0 &&
952 1.32 mrg buf[3] >= '0' && buf[3] <= '9')
953 1.32 mrg return FT_BZIP2;
954 1.32 mrg else
955 1.32 mrg #endif
956 1.32 mrg #ifndef NO_COMPRESS_SUPPORT
957 1.32 mrg if (memcmp(buf, Z_MAGIC, 2) == 0)
958 1.32 mrg return FT_Z;
959 1.32 mrg else
960 1.32 mrg #endif
961 1.32 mrg return FT_UNKNOWN;
962 1.32 mrg }
963 1.32 mrg
964 1.46 mrg #ifndef SMALL
965 1.46 mrg /* check the outfile is OK. */
966 1.46 mrg static int
967 1.46 mrg check_outfile(const char *outfile, struct stat *sb)
968 1.46 mrg {
969 1.46 mrg int ok = 1;
970 1.46 mrg
971 1.59 mrg if (lflag == 0 && stat(outfile, sb) == 0) {
972 1.59 mrg if (fflag)
973 1.59 mrg unlink(outfile);
974 1.59 mrg else if (isatty(STDIN_FILENO)) {
975 1.46 mrg char ans[10] = { 'n', '\0' }; /* default */
976 1.46 mrg
977 1.46 mrg fprintf(stderr, "%s already exists -- do you wish to "
978 1.46 mrg "overwrite (y or n)? " , outfile);
979 1.46 mrg (void)fgets(ans, sizeof(ans) - 1, stdin);
980 1.46 mrg if (ans[0] != 'y' && ans[0] != 'Y') {
981 1.46 mrg fprintf(stderr, "\tnot overwritting\n");
982 1.46 mrg ok = 0;
983 1.46 mrg } else
984 1.46 mrg unlink(outfile);
985 1.46 mrg } else {
986 1.46 mrg maybe_warnx("%s already exists -- skipping", outfile);
987 1.46 mrg ok = 0;
988 1.46 mrg }
989 1.46 mrg }
990 1.46 mrg return ok;
991 1.46 mrg }
992 1.56 mrg
993 1.56 mrg static const char *
994 1.56 mrg check_suffix(char *file)
995 1.56 mrg {
996 1.56 mrg char const **s;
997 1.56 mrg size_t slen, len = strlen(file);
998 1.56 mrg
999 1.56 mrg for (s = suffixes; *s; s++) {
1000 1.56 mrg slen = strlen(*s);
1001 1.57 mrg /* if it doesn't fit in "a.suf", don't bother */
1002 1.57 mrg if (slen + 1 > len)
1003 1.57 mrg continue;
1004 1.56 mrg if (strcmp(*s, file + len - slen) == 0)
1005 1.56 mrg return *s;
1006 1.56 mrg }
1007 1.56 mrg return NULL;
1008 1.56 mrg }
1009 1.46 mrg #endif
1010 1.46 mrg
1011 1.1 mrg /*
1012 1.1 mrg * compress the given file: create a corresponding .gz file and remove the
1013 1.1 mrg * original.
1014 1.1 mrg */
1015 1.31 mrg static off_t
1016 1.45 agc file_compress(char *file, char *outfile, size_t outsize)
1017 1.1 mrg {
1018 1.1 mrg FILE *in;
1019 1.37 mrg int out;
1020 1.1 mrg struct stat isb, osb;
1021 1.31 mrg off_t size;
1022 1.25 mrg #ifndef SMALL
1023 1.1 mrg u_int32_t mtime = 0;
1024 1.37 mrg char *savename;
1025 1.25 mrg #endif
1026 1.1 mrg
1027 1.1 mrg if (cflag == 0) {
1028 1.56 mrg #ifndef SMALL
1029 1.56 mrg const char *suff;
1030 1.56 mrg
1031 1.56 mrg if (fflag == 0 && (suff = check_suffix(file))) {
1032 1.56 mrg maybe_warnx("%s already has %s suffix -- unchanged",
1033 1.56 mrg file, suff);
1034 1.56 mrg goto lose;
1035 1.56 mrg }
1036 1.56 mrg
1037 1.56 mrg #endif
1038 1.56 mrg
1039 1.45 agc (void)strncpy(outfile, file, outsize - suffix_len);
1040 1.45 agc outfile[outsize - suffix_len] = '\0';
1041 1.45 agc (void)strlcat(outfile, suffix, outsize);
1042 1.1 mrg
1043 1.25 mrg #ifndef SMALL
1044 1.46 mrg if (check_outfile(outfile, &osb) == 0)
1045 1.46 mrg goto lose;
1046 1.46 mrg
1047 1.1 mrg if (stat(file, &isb) == 0) {
1048 1.33 tron if (isb.st_nlink > 1 && fflag == 0) {
1049 1.1 mrg maybe_warnx("%s has %d other link%s -- "
1050 1.33 tron "skipping", file, isb.st_nlink - 1,
1051 1.1 mrg isb.st_nlink == 1 ? "" : "s");
1052 1.1 mrg goto lose;
1053 1.1 mrg }
1054 1.1 mrg if (nflag == 0)
1055 1.1 mrg mtime = (u_int32_t)isb.st_mtime;
1056 1.34 mrg }
1057 1.25 mrg #endif
1058 1.1 mrg }
1059 1.1 mrg in = fopen(file, "r");
1060 1.46 mrg if (in == NULL) {
1061 1.46 mrg maybe_warn("can't fopen %s", file);
1062 1.46 mrg goto lose;
1063 1.46 mrg }
1064 1.1 mrg
1065 1.37 mrg #ifndef SMALL
1066 1.43 bouyer if (nflag == 0)
1067 1.43 bouyer savename = basename(file);
1068 1.43 bouyer else
1069 1.43 bouyer savename = NULL;
1070 1.1 mrg #endif
1071 1.43 bouyer if (cflag == 0) {
1072 1.37 mrg out = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
1073 1.39 mrg if (out == -1) {
1074 1.39 mrg maybe_warn("could not create output: %s", outfile);
1075 1.39 mrg goto lose;
1076 1.39 mrg }
1077 1.1 mrg } else
1078 1.37 mrg out = STDOUT_FILENO;
1079 1.1 mrg
1080 1.37 mrg #ifdef SMALL
1081 1.37 mrg gz_compress(in, out, NULL, NULL, 0);
1082 1.37 mrg #else
1083 1.37 mrg gz_compress(in, out, NULL, savename, mtime);
1084 1.37 mrg #endif
1085 1.1 mrg
1086 1.42 mrg (void)fclose(in);
1087 1.42 mrg
1088 1.1 mrg /*
1089 1.1 mrg * if we compressed to stdout, we don't know the size and
1090 1.1 mrg * we don't know the new file name, punt. if we can't stat
1091 1.1 mrg * the file, whine, otherwise set the size from the stat
1092 1.1 mrg * buffer. we only blow away the file if we can stat the
1093 1.1 mrg * output, just in case.
1094 1.1 mrg */
1095 1.1 mrg if (cflag == 0) {
1096 1.39 mrg if (close(out) == -1)
1097 1.39 mrg maybe_warn("couldn't close ouput");
1098 1.39 mrg
1099 1.1 mrg if (stat(outfile, &osb) < 0) {
1100 1.1 mrg maybe_warn("couldn't stat: %s", outfile);
1101 1.1 mrg maybe_warnx("leaving original %s", file);
1102 1.1 mrg size = 0;
1103 1.1 mrg } else {
1104 1.1 mrg unlink(file);
1105 1.1 mrg size = osb.st_size;
1106 1.1 mrg }
1107 1.29 mrg #ifndef SMALL
1108 1.1 mrg copymodes(outfile, &isb);
1109 1.29 mrg #endif
1110 1.1 mrg } else {
1111 1.1 mrg lose:
1112 1.46 mrg size = -1;
1113 1.1 mrg }
1114 1.1 mrg
1115 1.1 mrg return (size);
1116 1.1 mrg }
1117 1.1 mrg
1118 1.1 mrg /* uncompress the given file and remove the original */
1119 1.31 mrg static off_t
1120 1.45 agc file_uncompress(char *file, char *outfile, size_t outsize)
1121 1.1 mrg {
1122 1.1 mrg struct stat isb, osb;
1123 1.45 agc char *s;
1124 1.1 mrg off_t size;
1125 1.57 mrg ssize_t rbytes, len = strlen(file);
1126 1.37 mrg unsigned char header1[4], name[PATH_MAX + 1];
1127 1.10 mrg enum filetype method;
1128 1.48 mrg int fd, zfd;
1129 1.48 mrg #ifndef SMALL
1130 1.48 mrg time_t timestamp = 0;
1131 1.48 mrg #endif
1132 1.9 mrg
1133 1.9 mrg /* gather the old name info */
1134 1.1 mrg
1135 1.9 mrg fd = open(file, O_RDONLY);
1136 1.46 mrg if (fd < 0) {
1137 1.46 mrg maybe_warn("can't open %s", file);
1138 1.46 mrg goto lose;
1139 1.46 mrg }
1140 1.57 mrg rbytes = read(fd, header1, sizeof header1);
1141 1.57 mrg if (rbytes != sizeof header1) {
1142 1.10 mrg /* we don't want to fail here. */
1143 1.25 mrg #ifndef SMALL
1144 1.10 mrg if (fflag)
1145 1.42 mrg goto lose_close_it;
1146 1.25 mrg #endif
1147 1.57 mrg if (rbytes == -1)
1148 1.57 mrg maybe_warn("can't read %s", file);
1149 1.57 mrg else
1150 1.57 mrg maybe_warnx("%s: unexpected end of file", file);
1151 1.58 yamt goto lose_close_it;
1152 1.10 mrg }
1153 1.10 mrg
1154 1.32 mrg method = file_gettype(header1);
1155 1.29 mrg
1156 1.32 mrg #ifndef SMALL
1157 1.32 mrg if (Sflag == NULL) {
1158 1.32 mrg # ifndef NO_BZIP2_SUPPORT
1159 1.32 mrg if (method == FT_BZIP2)
1160 1.10 mrg suffix = BZ2_SUFFIX;
1161 1.32 mrg else
1162 1.26 mrg # endif
1163 1.32 mrg # ifndef NO_COMPRESS_SUPPORT
1164 1.32 mrg if (method == FT_Z)
1165 1.22 mrg suffix = Z_SUFFIX;
1166 1.26 mrg # endif
1167 1.32 mrg }
1168 1.9 mrg
1169 1.42 mrg if (fflag == 0 && method == FT_UNKNOWN) {
1170 1.42 mrg maybe_warnx("%s: not in gzip format", file);
1171 1.42 mrg goto lose_close_it;
1172 1.42 mrg }
1173 1.25 mrg #endif
1174 1.9 mrg
1175 1.10 mrg if (cflag == 0 || lflag) {
1176 1.57 mrg s = 0;
1177 1.57 mrg if (len - suffix_len + 1 > 0 &&
1178 1.57 mrg (s = &file[len - suffix_len + 1]) &&
1179 1.57 mrg strncmp(s, suffix, suffix_len) == 0) {
1180 1.10 mrg (void)strncpy(outfile, file, len - suffix_len + 1);
1181 1.10 mrg outfile[len - suffix_len + 1] = '\0';
1182 1.42 mrg } else if (lflag == 0) {
1183 1.57 mrg maybe_warnx("%s: unknown suffix -- ignored", file);
1184 1.42 mrg goto lose_close_it;
1185 1.42 mrg }
1186 1.10 mrg }
1187 1.10 mrg
1188 1.25 mrg #ifdef SMALL
1189 1.25 mrg if (method == FT_GZIP && lflag)
1190 1.25 mrg #else
1191 1.25 mrg if (method == FT_GZIP && (Nflag || lflag))
1192 1.25 mrg #endif
1193 1.25 mrg {
1194 1.48 mrg #ifndef SMALL
1195 1.48 mrg unsigned char header2[4]; /* timestamp */
1196 1.48 mrg
1197 1.48 mrg if (lseek(fd, GZIP_TIMESTAMP, SEEK_SET) == -1) {
1198 1.48 mrg maybe_warn("can't lseek %s", file);
1199 1.48 mrg goto close_header_read;
1200 1.48 mrg }
1201 1.48 mrg if (read(fd, header2, sizeof header2) != sizeof header2) {
1202 1.48 mrg if (fflag)
1203 1.48 mrg goto lose_close_it;
1204 1.48 mrg maybe_warn("can't read %s", file);
1205 1.58 yamt goto lose_close_it;
1206 1.48 mrg }
1207 1.48 mrg timestamp = ((time_t)header2[3] << 24)
1208 1.48 mrg + ((time_t)header2[2] << 16)
1209 1.48 mrg + ((time_t)header2[1] << 8)
1210 1.48 mrg + (time_t)header2[0];
1211 1.48 mrg #endif
1212 1.48 mrg
1213 1.9 mrg if (header1[3] & ORIG_NAME) {
1214 1.9 mrg int i;
1215 1.9 mrg
1216 1.47 mrg if (lseek(fd, GZIP_ORIGNAME, SEEK_SET) == -1) {
1217 1.47 mrg maybe_warn("can't lseek %s", file);
1218 1.47 mrg goto close_header_read;
1219 1.47 mrg }
1220 1.9 mrg rbytes = read(fd, name, PATH_MAX + 1);
1221 1.42 mrg if (rbytes < 0) {
1222 1.42 mrg maybe_warn("can't read %s", file);
1223 1.42 mrg goto lose_close_it;
1224 1.42 mrg }
1225 1.9 mrg for (i = 0; i < rbytes && name[i]; i++)
1226 1.9 mrg ;
1227 1.9 mrg if (i < rbytes) {
1228 1.9 mrg name[i] = 0;
1229 1.9 mrg /* now maybe merge old dirname */
1230 1.45 agc if (strchr(outfile, '/') == NULL)
1231 1.45 agc (void) strlcpy(outfile, name, outsize);
1232 1.9 mrg else {
1233 1.45 agc char newbuf[PATH_MAX + 1];
1234 1.45 agc
1235 1.45 agc (void) snprintf(newbuf, sizeof(newbuf),
1236 1.45 agc "%s/%s", dirname(outfile),
1237 1.45 agc name);
1238 1.45 agc (void) strlcpy(outfile, newbuf,
1239 1.45 agc outsize);
1240 1.1 mrg }
1241 1.1 mrg }
1242 1.1 mrg }
1243 1.9 mrg }
1244 1.47 mrg close_header_read:
1245 1.9 mrg close(fd);
1246 1.1 mrg
1247 1.31 mrg if (cflag == 0 || lflag) {
1248 1.25 mrg #ifndef SMALL
1249 1.46 mrg if (check_outfile(outfile, &osb) == 0)
1250 1.9 mrg goto lose;
1251 1.46 mrg
1252 1.31 mrg #endif
1253 1.9 mrg if (stat(file, &isb) == 0) {
1254 1.31 mrg #ifndef SMALL
1255 1.33 tron if (isb.st_nlink > 1 && lflag == 0 && fflag == 0) {
1256 1.9 mrg maybe_warnx("%s has %d other links -- skipping",
1257 1.9 mrg file, isb.st_nlink - 1);
1258 1.1 mrg goto lose;
1259 1.1 mrg }
1260 1.48 mrg if (nflag == 0 && timestamp)
1261 1.48 mrg isb.st_mtime = timestamp;
1262 1.31 mrg #endif
1263 1.9 mrg } else
1264 1.9 mrg goto lose;
1265 1.1 mrg }
1266 1.5 mrg
1267 1.48 mrg if (cflag == 0 && lflag == 0) {
1268 1.46 mrg zfd = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
1269 1.46 mrg if (zfd == -1) {
1270 1.46 mrg maybe_warn("can't open %s", outfile);
1271 1.46 mrg goto lose;
1272 1.46 mrg }
1273 1.46 mrg } else
1274 1.46 mrg zfd = STDOUT_FILENO;
1275 1.46 mrg
1276 1.24 tsutsui #ifndef NO_BZIP2_SUPPORT
1277 1.10 mrg if (method == FT_BZIP2) {
1278 1.46 mrg int in;
1279 1.5 mrg
1280 1.30 mrg /* XXX */
1281 1.46 mrg if (lflag) {
1282 1.46 mrg maybe_warnx("no -l with bzip2 files");
1283 1.46 mrg goto lose;
1284 1.46 mrg }
1285 1.26 mrg
1286 1.46 mrg if ((in = open(file, O_RDONLY)) == -1) {
1287 1.46 mrg maybe_warn("open for read: %s", file);
1288 1.42 mrg goto lose;
1289 1.42 mrg }
1290 1.5 mrg
1291 1.46 mrg size = unbzip2(in, zfd, NULL, 0, NULL);
1292 1.46 mrg if (size == -1) {
1293 1.36 mrg if (cflag == 0)
1294 1.36 mrg unlink(outfile);
1295 1.46 mrg maybe_warnx("%s: uncompress failed", file);
1296 1.10 mrg goto lose;
1297 1.10 mrg }
1298 1.42 mrg if (close(in) != 0)
1299 1.42 mrg maybe_warn("couldn't close input");
1300 1.46 mrg if (cflag == 0 && close(zfd) != 0)
1301 1.42 mrg maybe_warn("couldn't close output");
1302 1.24 tsutsui } else
1303 1.24 tsutsui #endif
1304 1.29 mrg
1305 1.24 tsutsui #ifndef NO_COMPRESS_SUPPORT
1306 1.24 tsutsui if (method == FT_Z) {
1307 1.22 mrg FILE *in, *out;
1308 1.22 mrg
1309 1.30 mrg /* XXX */
1310 1.46 mrg if (lflag) {
1311 1.46 mrg maybe_warnx("no -l with Lempel-Ziv files");
1312 1.46 mrg goto lose;
1313 1.46 mrg }
1314 1.26 mrg
1315 1.46 mrg if ((in = zopen(file, NULL)) == NULL) {
1316 1.46 mrg maybe_warn("open for read: %s", file);
1317 1.46 mrg goto lose;
1318 1.46 mrg }
1319 1.22 mrg
1320 1.46 mrg out = fdopen(zfd, "w");
1321 1.46 mrg if (out == NULL) {
1322 1.46 mrg maybe_warn("open for write: %s", outfile);
1323 1.46 mrg goto lose;
1324 1.22 mrg }
1325 1.22 mrg
1326 1.37 mrg size = zuncompress(in, out, NULL, 0, NULL);
1327 1.42 mrg if (ferror(in) || fclose(in) != 0) {
1328 1.42 mrg unlink(outfile);
1329 1.46 mrg (void)fclose(out);
1330 1.46 mrg maybe_warn("failed infile fclose");
1331 1.42 mrg }
1332 1.36 mrg if (cflag == 0) {
1333 1.46 mrg if (size == -1) {
1334 1.46 mrg maybe_warnx("%s: uncompress failed", file);
1335 1.46 mrg (void)fclose(out);
1336 1.36 mrg unlink(outfile);
1337 1.36 mrg goto lose;
1338 1.36 mrg }
1339 1.42 mrg if (fclose(out) != 0) {
1340 1.36 mrg unlink(outfile);
1341 1.46 mrg maybe_warn("failed outfile close");
1342 1.46 mrg goto lose;
1343 1.36 mrg }
1344 1.22 mrg }
1345 1.24 tsutsui } else
1346 1.24 tsutsui #endif
1347 1.51 mrg
1348 1.51 mrg #ifndef SMALL
1349 1.51 mrg if (method == FT_UNKNOWN) {
1350 1.51 mrg int in;
1351 1.51 mrg
1352 1.51 mrg in = open(file, O_RDONLY);
1353 1.51 mrg if (in == -1) {
1354 1.51 mrg maybe_warn("can't open %s", file);
1355 1.51 mrg goto lose;
1356 1.51 mrg }
1357 1.51 mrg size = cat_fd(NULL, 0, NULL, in);
1358 1.51 mrg } else
1359 1.51 mrg #endif
1360 1.24 tsutsui {
1361 1.46 mrg int in;
1362 1.37 mrg
1363 1.10 mrg if (lflag) {
1364 1.46 mrg if ((zfd = open(file, O_RDONLY)) == -1) {
1365 1.46 mrg maybe_warn("open: %s", file);
1366 1.46 mrg goto lose;
1367 1.46 mrg }
1368 1.45 agc print_list(zfd, isb.st_size, outfile, isb.st_mtime);
1369 1.61 yamt return -1; /* XXX */
1370 1.10 mrg }
1371 1.10 mrg
1372 1.37 mrg in = open(file, O_RDONLY);
1373 1.46 mrg if (in == -1) {
1374 1.46 mrg maybe_warn("can't open %s", file);
1375 1.46 mrg goto lose;
1376 1.46 mrg }
1377 1.1 mrg
1378 1.46 mrg size = gz_uncompress(in, zfd, NULL, 0, NULL, file);
1379 1.42 mrg (void)close(in);
1380 1.36 mrg if (cflag == 0) {
1381 1.45 agc if (close(zfd))
1382 1.42 mrg maybe_warn("failed close");
1383 1.37 mrg if (size == -1) {
1384 1.46 mrg maybe_warnx("%s: uncompress failed", file);
1385 1.36 mrg unlink(outfile);
1386 1.36 mrg goto lose;
1387 1.36 mrg }
1388 1.10 mrg }
1389 1.10 mrg }
1390 1.1 mrg
1391 1.1 mrg /* if testing, or we uncompressed to stdout, this is all we need */
1392 1.26 mrg #ifndef SMALL
1393 1.26 mrg if (tflag)
1394 1.26 mrg return (size);
1395 1.26 mrg #endif
1396 1.26 mrg if (cflag)
1397 1.1 mrg return (size);
1398 1.1 mrg
1399 1.1 mrg /*
1400 1.1 mrg * if we create a file...
1401 1.1 mrg */
1402 1.1 mrg if (cflag == 0) {
1403 1.1 mrg /*
1404 1.1 mrg * if we can't stat the file, or we are uncompressing to
1405 1.1 mrg * stdin, don't remove the file.
1406 1.1 mrg */
1407 1.1 mrg if (stat(outfile, &osb) < 0) {
1408 1.1 mrg maybe_warn("couldn't stat (leaving original): %s",
1409 1.1 mrg outfile);
1410 1.1 mrg goto lose;
1411 1.1 mrg }
1412 1.1 mrg if (osb.st_size != size) {
1413 1.1 mrg maybe_warn("stat gave different size: %llu != %llu "
1414 1.1 mrg "(leaving original)",
1415 1.1 mrg (unsigned long long)size,
1416 1.1 mrg (unsigned long long)osb.st_size);
1417 1.1 mrg goto lose;
1418 1.1 mrg }
1419 1.36 mrg if (cflag == 0)
1420 1.36 mrg unlink(file);
1421 1.1 mrg size = osb.st_size;
1422 1.29 mrg #ifndef SMALL
1423 1.1 mrg copymodes(outfile, &isb);
1424 1.29 mrg #endif
1425 1.1 mrg }
1426 1.1 mrg return (size);
1427 1.1 mrg
1428 1.42 mrg lose_close_it:
1429 1.42 mrg close(fd);
1430 1.1 mrg lose:
1431 1.46 mrg return -1;
1432 1.1 mrg }
1433 1.1 mrg
1434 1.37 mrg #ifndef SMALL
1435 1.37 mrg static off_t
1436 1.51 mrg cat_fd(unsigned char * prepend, size_t count, off_t *gsizep, int fd)
1437 1.37 mrg {
1438 1.37 mrg char buf[BUFLEN];
1439 1.37 mrg size_t rv;
1440 1.37 mrg off_t in_tot;
1441 1.37 mrg
1442 1.37 mrg in_tot = count;
1443 1.46 mrg if (write(STDOUT_FILENO, prepend, count) != count) {
1444 1.46 mrg maybe_warn("write to stdout");
1445 1.46 mrg return -1;
1446 1.46 mrg }
1447 1.37 mrg for (;;) {
1448 1.51 mrg rv = read(fd, buf, sizeof buf);
1449 1.51 mrg if (rv == 0)
1450 1.51 mrg break;
1451 1.51 mrg if (rv < 0) {
1452 1.51 mrg maybe_warn("read from fd %d", fd);
1453 1.51 mrg break;
1454 1.51 mrg }
1455 1.37 mrg
1456 1.46 mrg if (write(STDOUT_FILENO, buf, rv) != rv) {
1457 1.46 mrg maybe_warn("write to stdout");
1458 1.51 mrg break;
1459 1.46 mrg }
1460 1.37 mrg in_tot += rv;
1461 1.37 mrg }
1462 1.37 mrg
1463 1.37 mrg if (gsizep)
1464 1.37 mrg *gsizep = in_tot;
1465 1.37 mrg return (in_tot);
1466 1.37 mrg }
1467 1.37 mrg #endif
1468 1.37 mrg
1469 1.1 mrg static void
1470 1.1 mrg handle_stdin(void)
1471 1.1 mrg {
1472 1.37 mrg unsigned char header1[4];
1473 1.37 mrg off_t usize, gsize;
1474 1.37 mrg enum filetype method;
1475 1.37 mrg #ifndef NO_COMPRESS_SUPPORT
1476 1.37 mrg FILE *in;
1477 1.37 mrg #endif
1478 1.1 mrg
1479 1.25 mrg #ifndef SMALL
1480 1.5 mrg if (fflag == 0 && lflag == 0 && isatty(STDIN_FILENO)) {
1481 1.1 mrg maybe_warnx("standard input is a terminal -- ignoring");
1482 1.1 mrg return;
1483 1.1 mrg }
1484 1.25 mrg #endif
1485 1.5 mrg
1486 1.5 mrg if (lflag) {
1487 1.5 mrg struct stat isb;
1488 1.5 mrg
1489 1.37 mrg /* XXX could read the whole file, etc. */
1490 1.46 mrg if (fstat(STDIN_FILENO, &isb) < 0) {
1491 1.46 mrg maybe_warn("fstat");
1492 1.46 mrg return;
1493 1.46 mrg }
1494 1.7 mrg print_list(STDIN_FILENO, isb.st_size, "stdout", isb.st_mtime);
1495 1.5 mrg return;
1496 1.5 mrg }
1497 1.5 mrg
1498 1.46 mrg if (read(STDIN_FILENO, header1, sizeof header1) != sizeof header1) {
1499 1.46 mrg maybe_warn("can't read stdin");
1500 1.46 mrg return;
1501 1.46 mrg }
1502 1.37 mrg
1503 1.37 mrg method = file_gettype(header1);
1504 1.37 mrg switch (method) {
1505 1.37 mrg default:
1506 1.37 mrg #ifndef SMALL
1507 1.46 mrg if (fflag == 0) {
1508 1.46 mrg maybe_warnx("unknown compression format");
1509 1.46 mrg return;
1510 1.46 mrg }
1511 1.51 mrg usize = cat_fd(header1, sizeof header1, &gsize, STDIN_FILENO);
1512 1.37 mrg break;
1513 1.37 mrg #endif
1514 1.37 mrg case FT_GZIP:
1515 1.37 mrg usize = gz_uncompress(STDIN_FILENO, STDOUT_FILENO,
1516 1.46 mrg header1, sizeof header1, &gsize, "(stdin)");
1517 1.37 mrg break;
1518 1.37 mrg #ifndef NO_BZIP2_SUPPORT
1519 1.37 mrg case FT_BZIP2:
1520 1.37 mrg usize = unbzip2(STDIN_FILENO, STDOUT_FILENO,
1521 1.37 mrg header1, sizeof header1, &gsize);
1522 1.37 mrg break;
1523 1.37 mrg #endif
1524 1.37 mrg #ifndef NO_COMPRESS_SUPPORT
1525 1.37 mrg case FT_Z:
1526 1.46 mrg if ((in = zopen(NULL, stdin)) == NULL) {
1527 1.46 mrg maybe_warnx("zopen of stdin");
1528 1.46 mrg return;
1529 1.46 mrg }
1530 1.37 mrg
1531 1.37 mrg usize = zuncompress(in, stdout, header1, sizeof header1, &gsize);
1532 1.37 mrg break;
1533 1.37 mrg #endif
1534 1.37 mrg }
1535 1.37 mrg
1536 1.37 mrg #ifndef SMALL
1537 1.37 mrg if (vflag && !tflag && usize != -1 && gsize != -1)
1538 1.37 mrg print_verbage(NULL, 0, usize, gsize);
1539 1.37 mrg #endif
1540 1.37 mrg
1541 1.1 mrg }
1542 1.1 mrg
1543 1.1 mrg static void
1544 1.1 mrg handle_stdout(void)
1545 1.1 mrg {
1546 1.37 mrg off_t gsize, usize;
1547 1.1 mrg
1548 1.25 mrg #ifndef SMALL
1549 1.1 mrg if (fflag == 0 && isatty(STDOUT_FILENO)) {
1550 1.1 mrg maybe_warnx("standard output is a terminal -- ignoring");
1551 1.1 mrg return;
1552 1.1 mrg }
1553 1.25 mrg #endif
1554 1.37 mrg usize = gz_compress(stdin, STDOUT_FILENO, &gsize, NULL, 0);
1555 1.37 mrg
1556 1.37 mrg #ifndef SMALL
1557 1.37 mrg if (vflag && !tflag && usize != -1 && gsize != -1)
1558 1.37 mrg print_verbage(NULL, 0, usize, gsize);
1559 1.37 mrg #endif
1560 1.1 mrg }
1561 1.1 mrg
1562 1.1 mrg /* do what is asked for, for the path name */
1563 1.1 mrg static void
1564 1.1 mrg handle_pathname(char *path)
1565 1.1 mrg {
1566 1.44 agc char *opath = path, *s = NULL;
1567 1.1 mrg ssize_t len;
1568 1.1 mrg struct stat sb;
1569 1.1 mrg
1570 1.1 mrg /* check for stdout/stdin */
1571 1.1 mrg if (path[0] == '-' && path[1] == '\0') {
1572 1.1 mrg if (dflag)
1573 1.1 mrg handle_stdin();
1574 1.1 mrg else
1575 1.1 mrg handle_stdout();
1576 1.49 mrg return;
1577 1.1 mrg }
1578 1.1 mrg
1579 1.1 mrg retry:
1580 1.1 mrg if (stat(path, &sb) < 0) {
1581 1.1 mrg /* lets try <path>.gz if we're decompressing */
1582 1.44 agc if (dflag && s == NULL && errno == ENOENT) {
1583 1.1 mrg len = strlen(path);
1584 1.44 agc s = malloc(len + suffix_len + 1);
1585 1.44 agc if (s == NULL)
1586 1.46 mrg maybe_err("malloc");
1587 1.1 mrg memmove(s, path, len);
1588 1.10 mrg memmove(&s[len], suffix, suffix_len);
1589 1.44 agc s[len + suffix_len] = 0x0;
1590 1.1 mrg path = s;
1591 1.1 mrg goto retry;
1592 1.1 mrg }
1593 1.1 mrg maybe_warn("can't stat: %s", opath);
1594 1.1 mrg goto out;
1595 1.1 mrg }
1596 1.1 mrg
1597 1.1 mrg if (S_ISDIR(sb.st_mode)) {
1598 1.25 mrg #ifndef SMALL
1599 1.1 mrg if (rflag)
1600 1.1 mrg handle_dir(path, &sb);
1601 1.1 mrg else
1602 1.25 mrg #endif
1603 1.60 yamt maybe_warnx("%s is a directory", path);
1604 1.1 mrg goto out;
1605 1.1 mrg }
1606 1.1 mrg
1607 1.1 mrg if (S_ISREG(sb.st_mode))
1608 1.1 mrg handle_file(path, &sb);
1609 1.1 mrg
1610 1.1 mrg out:
1611 1.1 mrg if (s)
1612 1.1 mrg free(s);
1613 1.1 mrg }
1614 1.1 mrg
1615 1.1 mrg /* compress/decompress a file */
1616 1.1 mrg static void
1617 1.1 mrg handle_file(char *file, struct stat *sbp)
1618 1.1 mrg {
1619 1.31 mrg off_t usize, gsize;
1620 1.45 agc char outfile[PATH_MAX];
1621 1.1 mrg
1622 1.1 mrg infile = file;
1623 1.1 mrg if (dflag) {
1624 1.45 agc usize = file_uncompress(file, outfile, sizeof(outfile));
1625 1.46 mrg if (usize == -1)
1626 1.1 mrg return;
1627 1.1 mrg gsize = sbp->st_size;
1628 1.1 mrg } else {
1629 1.45 agc gsize = file_compress(file, outfile, sizeof(outfile));
1630 1.46 mrg if (gsize == -1)
1631 1.1 mrg return;
1632 1.1 mrg usize = sbp->st_size;
1633 1.1 mrg }
1634 1.1 mrg
1635 1.26 mrg
1636 1.26 mrg #ifndef SMALL
1637 1.1 mrg if (vflag && !tflag)
1638 1.45 agc print_verbage(file, (cflag) ? NULL : outfile, usize, gsize);
1639 1.26 mrg #endif
1640 1.1 mrg }
1641 1.1 mrg
1642 1.25 mrg #ifndef SMALL
1643 1.1 mrg /* this is used with -r to recursively decend directories */
1644 1.1 mrg static void
1645 1.1 mrg handle_dir(char *dir, struct stat *sbp)
1646 1.1 mrg {
1647 1.1 mrg char *path_argv[2];
1648 1.1 mrg FTS *fts;
1649 1.1 mrg FTSENT *entry;
1650 1.1 mrg
1651 1.1 mrg path_argv[0] = dir;
1652 1.1 mrg path_argv[1] = 0;
1653 1.1 mrg fts = fts_open(path_argv, FTS_PHYSICAL, NULL);
1654 1.1 mrg if (fts == NULL) {
1655 1.1 mrg warn("couldn't fts_open %s", dir);
1656 1.1 mrg return;
1657 1.1 mrg }
1658 1.1 mrg
1659 1.1 mrg while ((entry = fts_read(fts))) {
1660 1.1 mrg switch(entry->fts_info) {
1661 1.1 mrg case FTS_D:
1662 1.1 mrg case FTS_DP:
1663 1.1 mrg continue;
1664 1.1 mrg
1665 1.1 mrg case FTS_DNR:
1666 1.1 mrg case FTS_ERR:
1667 1.1 mrg case FTS_NS:
1668 1.1 mrg maybe_warn("%s", entry->fts_path);
1669 1.1 mrg continue;
1670 1.1 mrg case FTS_F:
1671 1.1 mrg handle_file(entry->fts_name, entry->fts_statp);
1672 1.1 mrg }
1673 1.1 mrg }
1674 1.1 mrg (void)fts_close(fts);
1675 1.1 mrg }
1676 1.25 mrg #endif
1677 1.1 mrg
1678 1.6 mrg /* print a ratio */
1679 1.6 mrg static void
1680 1.6 mrg print_ratio(off_t in, off_t out, FILE *where)
1681 1.6 mrg {
1682 1.35 mrg int64_t percent10; /* 10 * percent */
1683 1.35 mrg off_t diff = in - out;
1684 1.35 mrg char ch;
1685 1.6 mrg
1686 1.35 mrg if (in == 0)
1687 1.35 mrg percent10 = 0;
1688 1.35 mrg else if (diff > 0x400000) /* anything with 22 or more bits */
1689 1.35 mrg percent10 = diff / (in / 1000);
1690 1.35 mrg else
1691 1.35 mrg percent10 = (1000 * diff) / in;
1692 1.35 mrg
1693 1.35 mrg if (percent10 < 0) {
1694 1.35 mrg percent10 = -percent10;
1695 1.35 mrg ch = '-';
1696 1.35 mrg } else
1697 1.35 mrg ch = ' ';
1698 1.35 mrg
1699 1.35 mrg /*
1700 1.35 mrg * ugh. for negative percentages < 10, we need to avoid printing a
1701 1.35 mrg * a space between the "-" and the single number.
1702 1.35 mrg */
1703 1.35 mrg if (ch == '-' && percent10 / 10LL < 10)
1704 1.35 mrg fprintf(where, " -%1d.%1u%%", (unsigned)(percent10 / 10LL),
1705 1.35 mrg (unsigned)(percent10 % 10LL));
1706 1.6 mrg else
1707 1.35 mrg fprintf(where, "%c%2d.%1u%%", ch, (unsigned)(percent10 / 10LL),
1708 1.35 mrg (unsigned)(percent10 % 10LL));
1709 1.6 mrg }
1710 1.6 mrg
1711 1.26 mrg #ifndef SMALL
1712 1.1 mrg /* print compression statistics, and the new name (if there is one!) */
1713 1.1 mrg static void
1714 1.31 mrg print_verbage(char *file, char *nfile, off_t usize, off_t gsize)
1715 1.1 mrg {
1716 1.37 mrg if (file)
1717 1.37 mrg fprintf(stderr, "%s:%s ", file,
1718 1.37 mrg strlen(file) < 7 ? "\t\t" : "\t");
1719 1.31 mrg print_ratio((off_t)usize, (off_t)gsize, stderr);
1720 1.4 mrg if (nfile)
1721 1.4 mrg fprintf(stderr, " -- replaced with %s", nfile);
1722 1.1 mrg fprintf(stderr, "\n");
1723 1.1 mrg fflush(stderr);
1724 1.1 mrg }
1725 1.1 mrg
1726 1.1 mrg /* print test results */
1727 1.1 mrg static void
1728 1.45 agc print_test(const char *file, int ok)
1729 1.1 mrg {
1730 1.1 mrg
1731 1.46 mrg if (exit_value == 0 && ok == 0)
1732 1.46 mrg exit_value = 1;
1733 1.1 mrg fprintf(stderr, "%s:%s %s\n", file,
1734 1.1 mrg strlen(file) < 7 ? "\t\t" : "\t", ok ? "OK" : "NOT OK");
1735 1.1 mrg fflush(stderr);
1736 1.1 mrg }
1737 1.26 mrg #endif
1738 1.1 mrg
1739 1.5 mrg /* print a file's info ala --list */
1740 1.5 mrg /* eg:
1741 1.5 mrg compressed uncompressed ratio uncompressed_name
1742 1.5 mrg 354841 1679360 78.8% /usr/pkgsrc/distfiles/libglade-2.0.1.tar
1743 1.5 mrg */
1744 1.5 mrg static void
1745 1.35 mrg print_list(int fd, off_t out, const char *outfile, time_t ts)
1746 1.5 mrg {
1747 1.5 mrg static int first = 1;
1748 1.29 mrg #ifndef SMALL
1749 1.6 mrg static off_t in_tot, out_tot;
1750 1.29 mrg u_int32_t crc;
1751 1.29 mrg #endif
1752 1.35 mrg off_t in;
1753 1.5 mrg int rv;
1754 1.5 mrg
1755 1.7 mrg if (first) {
1756 1.26 mrg #ifndef SMALL
1757 1.7 mrg if (vflag)
1758 1.7 mrg printf("method crc date time ");
1759 1.26 mrg #endif
1760 1.7 mrg if (qflag == 0)
1761 1.7 mrg printf(" compressed uncompressed "
1762 1.7 mrg "ratio uncompressed_name\n");
1763 1.7 mrg }
1764 1.5 mrg first = 0;
1765 1.5 mrg
1766 1.6 mrg /* print totals? */
1767 1.29 mrg #ifndef SMALL
1768 1.6 mrg if (fd == -1) {
1769 1.6 mrg in = in_tot;
1770 1.6 mrg out = out_tot;
1771 1.29 mrg } else
1772 1.29 mrg #endif
1773 1.29 mrg {
1774 1.6 mrg /* read the last 4 bytes - this is the uncompressed size */
1775 1.7 mrg rv = lseek(fd, (off_t)(-8), SEEK_END);
1776 1.6 mrg if (rv != -1) {
1777 1.7 mrg unsigned char buf[8];
1778 1.6 mrg u_int32_t usize;
1779 1.6 mrg
1780 1.6 mrg if (read(fd, (char *)buf, sizeof(buf)) != sizeof(buf))
1781 1.37 mrg maybe_warn("read of uncompressed size");
1782 1.7 mrg usize = buf[4] | buf[5] << 8 | buf[6] << 16 | buf[7] << 24;
1783 1.35 mrg in = (off_t)usize;
1784 1.29 mrg #ifndef SMALL
1785 1.29 mrg crc = buf[0] | buf[1] << 8 | buf[2] << 16 | buf[3] << 24;
1786 1.29 mrg #endif
1787 1.6 mrg }
1788 1.6 mrg }
1789 1.5 mrg
1790 1.26 mrg #ifndef SMALL
1791 1.7 mrg if (vflag && fd == -1)
1792 1.7 mrg printf(" ");
1793 1.7 mrg else if (vflag) {
1794 1.7 mrg char *date = ctime(&ts);
1795 1.7 mrg
1796 1.7 mrg /* skip the day, 1/100th second, and year */
1797 1.7 mrg date += 4;
1798 1.7 mrg date[12] = 0;
1799 1.7 mrg printf("%5s %08x %11s ", "defla"/*XXX*/, crc, date);
1800 1.7 mrg }
1801 1.29 mrg in_tot += in;
1802 1.29 mrg out_tot += out;
1803 1.26 mrg #endif
1804 1.35 mrg printf("%12llu %12llu ", (unsigned long long)out, (unsigned long long)in);
1805 1.6 mrg print_ratio(in, out, stdout);
1806 1.6 mrg printf(" %s\n", outfile);
1807 1.5 mrg }
1808 1.5 mrg
1809 1.1 mrg /* display the usage of NetBSD gzip */
1810 1.1 mrg static void
1811 1.1 mrg usage(void)
1812 1.1 mrg {
1813 1.1 mrg
1814 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
1815 1.1 mrg fprintf(stderr,
1816 1.29 mrg "usage: %s [-" OPT_LIST "] [<file> [<file> ...]]\n"
1817 1.11 mrg #ifndef SMALL
1818 1.1 mrg " -c --stdout write to stdout, keep original files\n"
1819 1.1 mrg " --to-stdout\n"
1820 1.1 mrg " -d --decompress uncompress files\n"
1821 1.1 mrg " --uncompress\n"
1822 1.1 mrg " -f --force force overwriting & compress links\n"
1823 1.1 mrg " -h --help display this help\n"
1824 1.1 mrg " -n --no-name don't save original file name or time stamp\n"
1825 1.1 mrg " -N --name save or restore original file name and time stamp\n"
1826 1.1 mrg " -q --quiet output no warnings\n"
1827 1.1 mrg " -r --recursive recursively compress files in directories\n"
1828 1.1 mrg " -S .suf use suffix .suf instead of .gz\n"
1829 1.1 mrg " --suffix .suf\n"
1830 1.1 mrg " -t --test test compressed file\n"
1831 1.1 mrg " -v --verbose print extra statistics\n"
1832 1.1 mrg " -V --version display program version\n"
1833 1.1 mrg " -1 --fast fastest (worst) compression\n"
1834 1.1 mrg " -2 .. -8 set compression level\n"
1835 1.1 mrg " -9 --best best (slowest) compression\n",
1836 1.11 mrg #else
1837 1.11 mrg ,
1838 1.11 mrg #endif
1839 1.1 mrg getprogname());
1840 1.1 mrg exit(0);
1841 1.1 mrg }
1842 1.1 mrg
1843 1.1 mrg /* display the version of NetBSD gzip */
1844 1.1 mrg static void
1845 1.1 mrg display_version(void)
1846 1.1 mrg {
1847 1.1 mrg
1848 1.1 mrg fprintf(stderr, "%s\n", gzip_version);
1849 1.1 mrg exit(0);
1850 1.1 mrg }
1851 1.10 mrg
1852 1.24 tsutsui #ifndef NO_BZIP2_SUPPORT
1853 1.10 mrg #include "unbzip2.c"
1854 1.24 tsutsui #endif
1855 1.24 tsutsui #ifndef NO_COMPRESS_SUPPORT
1856 1.22 mrg #include "zuncompress.c"
1857 1.24 tsutsui #endif
1858