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