maint: port to platforms lacking SIGPIPE
[debian/gzip] / gzip.c
1 /* gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
2
3    Copyright (C) 1999, 2001-2002, 2006-2007, 2009-2012 Free Software
4    Foundation, Inc.
5    Copyright (C) 1992-1993 Jean-loup Gailly
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 3, or (at your option)
10    any later version.
11
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software Foundation,
19    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
20
21 /*
22  * The unzip code was written and put in the public domain by Mark Adler.
23  * Portions of the lzw code are derived from the public domain 'compress'
24  * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies,
25  * Ken Turkowski, Dave Mack and Peter Jannesen.
26  *
27  * See the license_msg below and the file COPYING for the software license.
28  * See the file algorithm.doc for the compression algorithms and file formats.
29  */
30
31 static char const *const license_msg[] = {
32 "Copyright (C) 2007, 2010, 2011 Free Software Foundation, Inc.",
33 "Copyright (C) 1993 Jean-loup Gailly.",
34 "This is free software.  You may redistribute copies of it under the terms of",
35 "the GNU General Public License <http://www.gnu.org/licenses/gpl.html>.",
36 "There is NO WARRANTY, to the extent permitted by law.",
37 0};
38
39 /* Compress files with zip algorithm and 'compress' interface.
40  * See help() function below for all options.
41  * Outputs:
42  *        file.gz:   compressed file with same mode, owner, and utimes
43  *     or stdout with -c option or if stdin used as input.
44  * If the output file name had to be truncated, the original name is kept
45  * in the compressed file.
46  * On MSDOS, file.tmp -> file.tmz. On VMS, file.tmp -> file.tmp-gz.
47  *
48  * Using gz on MSDOS would create too many file name conflicts. For
49  * example, foo.txt -> foo.tgz (.tgz must be reserved as shorthand for
50  * tar.gz). Similarly, foo.dir and foo.doc would both be mapped to foo.dgz.
51  * I also considered 12345678.txt -> 12345txt.gz but this truncates the name
52  * too heavily. There is no ideal solution given the MSDOS 8+3 limitation.
53  *
54  * For the meaning of all compilation flags, see comments in Makefile.in.
55  */
56
57 #include <config.h>
58 #include <ctype.h>
59 #include <sys/types.h>
60 #include <signal.h>
61 #include <stdbool.h>
62 #include <sys/stat.h>
63 #include <errno.h>
64
65 #include "closein.h"
66 #include "tailor.h"
67 #include "gzip.h"
68 #include "lzw.h"
69 #include "revision.h"
70 #include "timespec.h"
71
72 #include "fcntl-safer.h"
73 #include "getopt.h"
74 #include "ignore-value.h"
75 #include "stat-time.h"
76 #include "version.h"
77
78                 /* configuration */
79
80 #include <fcntl.h>
81 #include <limits.h>
82 #include <unistd.h>
83 #include <stdlib.h>
84 #include <errno.h>
85
86 #ifndef NO_DIR
87 # define NO_DIR 0
88 #endif
89 #if !NO_DIR
90 # include <dirent.h>
91 # include <savedir.h>
92 #endif
93
94 #ifndef NO_UTIME
95 #  include <utimens.h>
96 #endif
97
98 #define RW_USER (S_IRUSR | S_IWUSR)  /* creation mode for open() */
99
100 #ifndef MAX_PATH_LEN
101 #  define MAX_PATH_LEN   1024 /* max pathname length */
102 #endif
103
104 #ifndef SEEK_END
105 #  define SEEK_END 2
106 #endif
107
108 #ifndef CHAR_BIT
109 #  define CHAR_BIT 8
110 #endif
111
112 #ifdef off_t
113   off_t lseek (int fd, off_t offset, int whence);
114 #endif
115
116 #ifndef OFF_T_MIN
117 #define OFF_T_MIN (~ (off_t) 0 << (sizeof (off_t) * CHAR_BIT - 1))
118 #endif
119
120 #ifndef OFF_T_MAX
121 #define OFF_T_MAX (~ (off_t) 0 - OFF_T_MIN)
122 #endif
123
124 #ifndef SIGPIPE
125 # define SIGPIPE 0
126 #endif
127
128 /* Use SA_NOCLDSTOP as a proxy for whether the sigaction machinery is
129    present.  */
130 #ifndef SA_NOCLDSTOP
131 # define SA_NOCLDSTOP 0
132 # define sigprocmask(how, set, oset) /* empty */
133 # define sigset_t int
134 # if ! HAVE_SIGINTERRUPT
135 #  define siginterrupt(sig, flag) /* empty */
136 # endif
137 #endif
138
139 #ifndef HAVE_WORKING_O_NOFOLLOW
140 # define HAVE_WORKING_O_NOFOLLOW 0
141 #endif
142
143 /* Separator for file name parts (see shorten_name()) */
144 #ifdef NO_MULTIPLE_DOTS
145 #  define PART_SEP "-"
146 #else
147 #  define PART_SEP "."
148 #endif
149
150                 /* global buffers */
151
152 DECLARE(uch, inbuf,  INBUFSIZ +INBUF_EXTRA);
153 DECLARE(uch, outbuf, OUTBUFSIZ+OUTBUF_EXTRA);
154 DECLARE(ush, d_buf,  DIST_BUFSIZE);
155 DECLARE(uch, window, 2L*WSIZE);
156 #ifndef MAXSEG_64K
157     DECLARE(ush, tab_prefix, 1L<<BITS);
158 #else
159     DECLARE(ush, tab_prefix0, 1L<<(BITS-1));
160     DECLARE(ush, tab_prefix1, 1L<<(BITS-1));
161 #endif
162
163                 /* local variables */
164
165 /* If true, pretend that standard input is a tty.  This option
166    is deliberately not documented, and only for testing.  */
167 static bool presume_input_tty;
168
169 static int ascii = 0;        /* convert end-of-lines to local OS conventions */
170        int to_stdout = 0;    /* output to stdout (-c) */
171 static int decompress = 0;   /* decompress (-d) */
172 static int force = 0;        /* don't ask questions, compress links (-f) */
173 static int no_name = -1;     /* don't save or restore the original file name */
174 static int no_time = -1;     /* don't save or restore the original file time */
175 static int recursive = 0;    /* recurse through directories (-r) */
176 static int list = 0;         /* list the file contents (-l) */
177        int verbose = 0;      /* be verbose (-v) */
178        int quiet = 0;        /* be very quiet (-q) */
179 static int do_lzw = 0;       /* generate output compatible with old compress (-Z) */
180        int test = 0;         /* test .gz file integrity */
181 static int foreground = 0;   /* set if program run in foreground */
182        char *program_name;   /* program name */
183        int maxbits = BITS;   /* max bits per code for LZW */
184        int method = DEFLATED;/* compression method */
185        int level = 6;        /* compression level */
186        int exit_code = OK;   /* program exit code */
187        int save_orig_name;   /* set if original name must be saved */
188 static int last_member;      /* set for .zip and .Z files */
189 static int part_nb;          /* number of parts in .gz file */
190        struct timespec time_stamp; /* original time stamp (modification time) */
191        off_t ifile_size;      /* input file size, -1 for devices (debug only) */
192 static char *env;            /* contents of GZIP env variable */
193 static char **args = NULL;   /* argv pointer if GZIP env variable defined */
194 static char const *z_suffix; /* default suffix (can be set with --suffix) */
195 static size_t z_len;         /* strlen(z_suffix) */
196
197 /* The set of signals that are caught.  */
198 static sigset_t caught_signals;
199
200 /* If nonzero then exit with status WARNING, rather than with the usual
201    signal status, on receipt of a signal with this value.  This
202    suppresses a "Broken Pipe" message with some shells.  */
203 static int volatile exiting_signal;
204
205 /* If nonnegative, close this file descriptor and unlink ofname on error.  */
206 static int volatile remove_ofname_fd = -1;
207
208 off_t bytes_in;             /* number of input bytes */
209 off_t bytes_out;            /* number of output bytes */
210 static off_t total_in;      /* input bytes for all files */
211 static off_t total_out;     /* output bytes for all files */
212 char ifname[MAX_PATH_LEN]; /* input file name */
213 char ofname[MAX_PATH_LEN]; /* output file name */
214 static struct stat istat;         /* status for input file */
215 int  ifd;                  /* input file descriptor */
216 int  ofd;                  /* output file descriptor */
217 unsigned insize;           /* valid bytes in inbuf */
218 unsigned inptr;            /* index of next byte to be processed in inbuf */
219 unsigned outcnt;           /* bytes in output buffer */
220
221 static int handled_sig[] =
222   {
223     /* SIGINT must be first, as 'foreground' depends on it.  */
224     SIGINT
225
226 #ifdef SIGHUP
227     , SIGHUP
228 #endif
229 #if SIGPIPE
230     , SIGPIPE
231 #endif
232 #ifdef SIGTERM
233     , SIGTERM
234 #endif
235 #ifdef SIGXCPU
236     , SIGXCPU
237 #endif
238 #ifdef SIGXFSZ
239     , SIGXFSZ
240 #endif
241   };
242
243 /* For long options that have no equivalent short option, use a
244    non-character as a pseudo short option, starting with CHAR_MAX + 1.  */
245 enum
246 {
247   PRESUME_INPUT_TTY_OPTION = CHAR_MAX + 1
248 };
249
250 static const struct option longopts[] =
251 {
252  /* { name  has_arg  *flag  val } */
253     {"ascii",      0, 0, 'a'}, /* ascii text mode */
254     {"to-stdout",  0, 0, 'c'}, /* write output on standard output */
255     {"stdout",     0, 0, 'c'}, /* write output on standard output */
256     {"decompress", 0, 0, 'd'}, /* decompress */
257     {"uncompress", 0, 0, 'd'}, /* decompress */
258  /* {"encrypt",    0, 0, 'e'},    encrypt */
259     {"force",      0, 0, 'f'}, /* force overwrite of output file */
260     {"help",       0, 0, 'h'}, /* give help */
261  /* {"pkzip",      0, 0, 'k'},    force output in pkzip format */
262     {"list",       0, 0, 'l'}, /* list .gz file contents */
263     {"license",    0, 0, 'L'}, /* display software license */
264     {"no-name",    0, 0, 'n'}, /* don't save or restore original name & time */
265     {"name",       0, 0, 'N'}, /* save or restore original name & time */
266     {"-presume-input-tty", no_argument, NULL, PRESUME_INPUT_TTY_OPTION},
267     {"quiet",      0, 0, 'q'}, /* quiet mode */
268     {"silent",     0, 0, 'q'}, /* quiet mode */
269     {"recursive",  0, 0, 'r'}, /* recurse through directories */
270     {"suffix",     1, 0, 'S'}, /* use given suffix instead of .gz */
271     {"test",       0, 0, 't'}, /* test compressed file integrity */
272     {"no-time",    0, 0, 'T'}, /* don't save or restore the time stamp */
273     {"verbose",    0, 0, 'v'}, /* verbose mode */
274     {"version",    0, 0, 'V'}, /* display version number */
275     {"fast",       0, 0, '1'}, /* compress faster */
276     {"best",       0, 0, '9'}, /* compress better */
277     {"lzw",        0, 0, 'Z'}, /* make output compatible with old compress */
278     {"bits",       1, 0, 'b'}, /* max number of bits per code (implies -Z) */
279
280     { 0, 0, 0, 0 }
281 };
282
283 /* local functions */
284
285 local void try_help     (void) ATTRIBUTE_NORETURN;
286 local void help         (void);
287 local void license      (void);
288 local void version      (void);
289 local int input_eof     (void);
290 local void treat_stdin  (void);
291 local void treat_file   (char *iname);
292 local int create_outfile (void);
293 local char *get_suffix  (char *name);
294 local int  open_input_file (char *iname, struct stat *sbuf);
295 local void discard_input_bytes (size_t nbytes, unsigned int flags);
296 local int  make_ofname  (void);
297 local void shorten_name  (char *name);
298 local int  get_method   (int in);
299 local void do_list      (int ifd, int method);
300 local int  check_ofname (void);
301 local void copy_stat    (struct stat *ifstat);
302 local void install_signal_handlers (void);
303 local void remove_output_file (void);
304 local RETSIGTYPE abort_gzip_signal (int);
305 local void do_exit      (int exitcode) ATTRIBUTE_NORETURN;
306       int main          (int argc, char **argv);
307 static int (*work) (int infile, int outfile) = zip; /* function to call */
308
309 #if ! NO_DIR
310 local void treat_dir    (int fd, char *dir);
311 #endif
312
313 #define strequ(s1, s2) (strcmp((s1),(s2)) == 0)
314
315 static void
316 try_help ()
317 {
318   fprintf (stderr, "Try `%s --help' for more information.\n",
319            program_name);
320   do_exit (ERROR);
321 }
322
323 /* ======================================================================== */
324 local void help()
325 {
326     static char const* const help_msg[] = {
327  "Compress or uncompress FILEs (by default, compress FILES in-place).",
328  "",
329  "Mandatory arguments to long options are mandatory for short options too.",
330  "",
331 #if O_BINARY
332  "  -a, --ascii       ascii text; convert end-of-line using local conventions",
333 #endif
334  "  -c, --stdout      write on standard output, keep original files unchanged",
335  "  -d, --decompress  decompress",
336 /*  -e, --encrypt     encrypt */
337  "  -f, --force       force overwrite of output file and compress links",
338  "  -h, --help        give this help",
339 /*  -k, --pkzip       force output in pkzip format */
340  "  -l, --list        list compressed file contents",
341  "  -L, --license     display software license",
342 #ifdef UNDOCUMENTED
343  "  -m, --no-time     do not save or restore the original modification time",
344  "  -M, --time        save or restore the original modification time",
345 #endif
346  "  -n, --no-name     do not save or restore the original name and time stamp",
347  "  -N, --name        save or restore the original name and time stamp",
348  "  -q, --quiet       suppress all warnings",
349 #if ! NO_DIR
350  "  -r, --recursive   operate recursively on directories",
351 #endif
352  "  -S, --suffix=SUF  use suffix SUF on compressed files",
353  "  -t, --test        test compressed file integrity",
354  "  -v, --verbose     verbose mode",
355  "  -V, --version     display version number",
356  "  -1, --fast        compress faster",
357  "  -9, --best        compress better",
358 #ifdef LZW
359  "  -Z, --lzw         produce output compatible with old compress",
360  "  -b, --bits=BITS   max number of bits per code (implies -Z)",
361 #endif
362  "",
363  "With no FILE, or when FILE is -, read standard input.",
364  "",
365  "Report bugs to <bug-gzip@gnu.org>.",
366   0};
367     char const *const *p = help_msg;
368
369     printf ("Usage: %s [OPTION]... [FILE]...\n", program_name);
370     while (*p) printf ("%s\n", *p++);
371 }
372
373 /* ======================================================================== */
374 local void license()
375 {
376     char const *const *p = license_msg;
377
378     printf ("%s %s\n", program_name, Version);
379     while (*p) printf ("%s\n", *p++);
380 }
381
382 /* ======================================================================== */
383 local void version()
384 {
385     license ();
386     printf ("\n");
387     printf ("Written by Jean-loup Gailly.\n");
388 }
389
390 local void progerror (char const *string)
391 {
392     int e = errno;
393     fprintf (stderr, "%s: ", program_name);
394     errno = e;
395     perror(string);
396     exit_code = ERROR;
397 }
398
399 /* ======================================================================== */
400 int main (int argc, char **argv)
401 {
402     int file_count;     /* number of files to process */
403     size_t proglen;     /* length of program_name */
404     int optc;           /* current option */
405
406     EXPAND(argc, argv); /* wild card expansion if necessary */
407
408     program_name = gzip_base_name (argv[0]);
409     proglen = strlen (program_name);
410
411     atexit (close_stdin);
412
413     /* Suppress .exe for MSDOS, OS/2 and VMS: */
414     if (4 < proglen && strequ (program_name + proglen - 4, ".exe"))
415       program_name[proglen - 4] = '\0';
416
417     /* Add options in GZIP environment variable if there is one */
418     env = add_envopt(&argc, &argv, OPTIONS_VAR);
419     if (env != NULL) args = argv;
420
421 #ifndef GNU_STANDARD
422 # define GNU_STANDARD 1
423 #endif
424 #if !GNU_STANDARD
425     /* For compatibility with old compress, use program name as an option.
426      * Unless you compile with -DGNU_STANDARD=0, this program will behave as
427      * gzip even if it is invoked under the name gunzip or zcat.
428      *
429      * Systems which do not support links can still use -d or -dc.
430      * Ignore an .exe extension for MSDOS, OS/2 and VMS.
431      */
432     if (strncmp (program_name, "un",  2) == 0     /* ungzip, uncompress */
433         || strncmp (program_name, "gun", 3) == 0) /* gunzip */
434         decompress = 1;
435     else if (strequ (program_name + 1, "cat")     /* zcat, pcat, gcat */
436              || strequ (program_name, "gzcat"))   /* gzcat */
437         decompress = to_stdout = 1;
438 #endif
439
440     z_suffix = Z_SUFFIX;
441     z_len = strlen(z_suffix);
442
443     while ((optc = getopt_long (argc, argv, "ab:cdfhH?lLmMnNqrS:tvVZ123456789",
444                                 longopts, (int *)0)) != -1) {
445         switch (optc) {
446         case 'a':
447             ascii = 1; break;
448         case 'b':
449             maxbits = atoi(optarg);
450             for (; *optarg; optarg++)
451               if (! ('0' <= *optarg && *optarg <= '9'))
452                 {
453                   fprintf (stderr, "%s: -b operand is not an integer\n",
454                            program_name);
455                   try_help ();
456                 }
457             break;
458         case 'c':
459             to_stdout = 1; break;
460         case 'd':
461             decompress = 1; break;
462         case 'f':
463             force++; break;
464         case 'h': case 'H':
465             help(); do_exit(OK); break;
466         case 'l':
467             list = decompress = to_stdout = 1; break;
468         case 'L':
469             license(); do_exit(OK); break;
470         case 'm': /* undocumented, may change later */
471             no_time = 1; break;
472         case 'M': /* undocumented, may change later */
473             no_time = 0; break;
474         case 'n':
475             no_name = no_time = 1; break;
476         case 'N':
477             no_name = no_time = 0; break;
478         case PRESUME_INPUT_TTY_OPTION:
479             presume_input_tty = true; break;
480         case 'q':
481             quiet = 1; verbose = 0; break;
482         case 'r':
483 #if NO_DIR
484             fprintf (stderr, "%s: -r not supported on this system\n",
485                      program_name);
486             try_help ();
487 #else
488             recursive = 1;
489 #endif
490             break;
491         case 'S':
492 #ifdef NO_MULTIPLE_DOTS
493             if (*optarg == '.') optarg++;
494 #endif
495             z_len = strlen(optarg);
496             z_suffix = optarg;
497             break;
498         case 't':
499             test = decompress = to_stdout = 1;
500             break;
501         case 'v':
502             verbose++; quiet = 0; break;
503         case 'V':
504             version(); do_exit(OK); break;
505         case 'Z':
506 #ifdef LZW
507             do_lzw = 1; break;
508 #else
509             fprintf(stderr, "%s: -Z not supported in this version\n",
510                     program_name);
511             try_help ();
512             break;
513 #endif
514         case '1':  case '2':  case '3':  case '4':
515         case '5':  case '6':  case '7':  case '8':  case '9':
516             level = optc - '0';
517             break;
518         default:
519             /* Error message already emitted by getopt_long. */
520             try_help ();
521         }
522     } /* loop on all arguments */
523
524     /* By default, save name and timestamp on compression but do not
525      * restore them on decompression.
526      */
527     if (no_time < 0) no_time = decompress;
528     if (no_name < 0) no_name = decompress;
529
530     file_count = argc - optind;
531
532 #if O_BINARY
533 #else
534     if (ascii && !quiet) {
535         fprintf(stderr, "%s: option --ascii ignored on this system\n",
536                 program_name);
537     }
538 #endif
539     if (z_len == 0 || z_len > MAX_SUFFIX) {
540         fprintf(stderr, "%s: invalid suffix '%s'\n", program_name, z_suffix);
541         do_exit(ERROR);
542     }
543
544     if (do_lzw && !decompress) work = lzw;
545
546     /* Allocate all global buffers (for DYN_ALLOC option) */
547     ALLOC(uch, inbuf,  INBUFSIZ +INBUF_EXTRA);
548     ALLOC(uch, outbuf, OUTBUFSIZ+OUTBUF_EXTRA);
549     ALLOC(ush, d_buf,  DIST_BUFSIZE);
550     ALLOC(uch, window, 2L*WSIZE);
551 #ifndef MAXSEG_64K
552     ALLOC(ush, tab_prefix, 1L<<BITS);
553 #else
554     ALLOC(ush, tab_prefix0, 1L<<(BITS-1));
555     ALLOC(ush, tab_prefix1, 1L<<(BITS-1));
556 #endif
557
558     exiting_signal = quiet ? SIGPIPE : 0;
559     install_signal_handlers ();
560
561     /* And get to work */
562     if (file_count != 0) {
563         if (to_stdout && !test && !list && (!decompress || !ascii)) {
564             SET_BINARY_MODE(fileno(stdout));
565         }
566         while (optind < argc) {
567             treat_file(argv[optind++]);
568         }
569     } else {  /* Standard input */
570         treat_stdin();
571     }
572     if (list && !quiet && file_count > 1) {
573         do_list(-1, -1); /* print totals */
574     }
575     do_exit(exit_code);
576     return exit_code; /* just to avoid lint warning */
577 }
578
579 /* Return nonzero when at end of file on input.  */
580 local int
581 input_eof ()
582 {
583   if (!decompress || last_member)
584     return 1;
585
586   if (inptr == insize)
587     {
588       if (insize != INBUFSIZ || fill_inbuf (1) == EOF)
589         return 1;
590
591       /* Unget the char that fill_inbuf got.  */
592       inptr = 0;
593     }
594
595   return 0;
596 }
597
598 /* ========================================================================
599  * Compress or decompress stdin
600  */
601 local void treat_stdin()
602 {
603     if (!force && !list
604         && (presume_input_tty
605             || isatty(fileno((FILE *)(decompress ? stdin : stdout))))) {
606         /* Do not send compressed data to the terminal or read it from
607          * the terminal. We get here when user invoked the program
608          * without parameters, so be helpful. According to the GNU standards:
609          *
610          *   If there is one behavior you think is most useful when the output
611          *   is to a terminal, and another that you think is most useful when
612          *   the output is a file or a pipe, then it is usually best to make
613          *   the default behavior the one that is useful with output to a
614          *   terminal, and have an option for the other behavior.
615          *
616          * Here we use the --force option to get the other behavior.
617          */
618         if (! quiet)
619           fprintf (stderr,
620                    ("%s: compressed data not %s a terminal."
621                     " Use -f to force %scompression.\n"
622                     "For help, type: %s -h\n"),
623                    program_name,
624                    decompress ? "read from" : "written to",
625                    decompress ? "de" : "",
626                    program_name);
627         do_exit(ERROR);
628     }
629
630     if (decompress || !ascii) {
631         SET_BINARY_MODE(fileno(stdin));
632     }
633     if (!test && !list && (!decompress || !ascii)) {
634         SET_BINARY_MODE(fileno(stdout));
635     }
636     strcpy(ifname, "stdin");
637     strcpy(ofname, "stdout");
638
639     /* Get the file's time stamp and size.  */
640     if (fstat (fileno (stdin), &istat) != 0)
641       {
642         progerror ("standard input");
643         do_exit (ERROR);
644       }
645     ifile_size = S_ISREG (istat.st_mode) ? istat.st_size : -1;
646     time_stamp.tv_nsec = -1;
647     if (!no_time || list)
648       {
649         if (S_ISREG (istat.st_mode))
650           time_stamp = get_stat_mtime (&istat);
651         else
652           gettime (&time_stamp);
653       }
654
655     clear_bufs(); /* clear input and output buffers */
656     to_stdout = 1;
657     part_nb = 0;
658     ifd = fileno(stdin);
659
660     if (decompress) {
661         method = get_method(ifd);
662         if (method < 0) {
663             do_exit(exit_code); /* error message already emitted */
664         }
665     }
666     if (list) {
667         do_list(ifd, method);
668         return;
669     }
670
671     /* Actually do the compression/decompression. Loop over zipped members.
672      */
673     for (;;) {
674         if ((*work)(fileno(stdin), fileno(stdout)) != OK) return;
675
676         if (input_eof ())
677           break;
678
679         method = get_method(ifd);
680         if (method < 0) return; /* error message already emitted */
681         bytes_out = 0;            /* required for length check */
682     }
683
684     if (verbose) {
685         if (test) {
686             fprintf(stderr, " OK\n");
687
688         } else if (!decompress) {
689             display_ratio(bytes_in-(bytes_out-header_bytes), bytes_in, stderr);
690             fprintf(stderr, "\n");
691 #ifdef DISPLAY_STDIN_RATIO
692         } else {
693             display_ratio(bytes_out-(bytes_in-header_bytes), bytes_out,stderr);
694             fprintf(stderr, "\n");
695 #endif
696         }
697     }
698 }
699
700 /* ========================================================================
701  * Compress or decompress the given file
702  */
703 local void treat_file(iname)
704     char *iname;
705 {
706     /* Accept "-" as synonym for stdin */
707     if (strequ(iname, "-")) {
708         int cflag = to_stdout;
709         treat_stdin();
710         to_stdout = cflag;
711         return;
712     }
713
714     /* Check if the input file is present, set ifname and istat: */
715     ifd = open_input_file (iname, &istat);
716     if (ifd < 0)
717       return;
718
719     /* If the input name is that of a directory, recurse or ignore: */
720     if (S_ISDIR(istat.st_mode)) {
721 #if ! NO_DIR
722         if (recursive) {
723             treat_dir (ifd, iname);
724             /* Warning: ifname is now garbage */
725             return;
726         }
727 #endif
728         close (ifd);
729         WARN ((stderr, "%s: %s is a directory -- ignored\n",
730                program_name, ifname));
731         return;
732     }
733
734     if (! to_stdout)
735       {
736         if (! S_ISREG (istat.st_mode))
737           {
738             WARN ((stderr,
739                    "%s: %s is not a directory or a regular file - ignored\n",
740                    program_name, ifname));
741             close (ifd);
742             return;
743           }
744         if (istat.st_mode & S_ISUID)
745           {
746             WARN ((stderr, "%s: %s is set-user-ID on execution - ignored\n",
747                    program_name, ifname));
748             close (ifd);
749             return;
750           }
751         if (istat.st_mode & S_ISGID)
752           {
753             WARN ((stderr, "%s: %s is set-group-ID on execution - ignored\n",
754                    program_name, ifname));
755             close (ifd);
756             return;
757           }
758
759         if (! force)
760           {
761             if (istat.st_mode & S_ISVTX)
762               {
763                 WARN ((stderr,
764                        "%s: %s has the sticky bit set - file ignored\n",
765                        program_name, ifname));
766                 close (ifd);
767                 return;
768               }
769             if (2 <= istat.st_nlink)
770               {
771                 WARN ((stderr, "%s: %s has %lu other link%c -- unchanged\n",
772                        program_name, ifname,
773                        (unsigned long int) istat.st_nlink - 1,
774                        istat.st_nlink == 2 ? ' ' : 's'));
775                 close (ifd);
776                 return;
777               }
778           }
779       }
780
781     ifile_size = S_ISREG (istat.st_mode) ? istat.st_size : -1;
782     time_stamp.tv_nsec = -1;
783     if (!no_time || list)
784       time_stamp = get_stat_mtime (&istat);
785
786     /* Generate output file name. For -r and (-t or -l), skip files
787      * without a valid gzip suffix (check done in make_ofname).
788      */
789     if (to_stdout && !list && !test) {
790         strcpy(ofname, "stdout");
791
792     } else if (make_ofname() != OK) {
793         close (ifd);
794         return;
795     }
796
797     clear_bufs(); /* clear input and output buffers */
798     part_nb = 0;
799
800     if (decompress) {
801         method = get_method(ifd); /* updates ofname if original given */
802         if (method < 0) {
803             close(ifd);
804             return;               /* error message already emitted */
805         }
806     }
807     if (list) {
808         do_list(ifd, method);
809         if (close (ifd) != 0)
810           read_error ();
811         return;
812     }
813
814     /* If compressing to a file, check if ofname is not ambiguous
815      * because the operating system truncates names. Otherwise, generate
816      * a new ofname and save the original name in the compressed file.
817      */
818     if (to_stdout) {
819         ofd = fileno(stdout);
820         /* Keep remove_ofname_fd negative.  */
821     } else {
822         if (create_outfile() != OK) return;
823
824         if (!decompress && save_orig_name && !verbose && !quiet) {
825             fprintf(stderr, "%s: %s compressed to %s\n",
826                     program_name, ifname, ofname);
827         }
828     }
829     /* Keep the name even if not truncated except with --no-name: */
830     if (!save_orig_name) save_orig_name = !no_name;
831
832     if (verbose) {
833         fprintf(stderr, "%s:\t", ifname);
834     }
835
836     /* Actually do the compression/decompression. Loop over zipped members.
837      */
838     for (;;) {
839         if ((*work)(ifd, ofd) != OK) {
840             method = -1; /* force cleanup */
841             break;
842         }
843
844         if (input_eof ())
845           break;
846
847         method = get_method(ifd);
848         if (method < 0) break;    /* error message already emitted */
849         bytes_out = 0;            /* required for length check */
850     }
851
852     if (close (ifd) != 0)
853       read_error ();
854
855     if (!to_stdout)
856       {
857         sigset_t oldset;
858         int unlink_errno;
859
860         copy_stat (&istat);
861         if (close (ofd) != 0)
862           write_error ();
863
864         sigprocmask (SIG_BLOCK, &caught_signals, &oldset);
865         remove_ofname_fd = -1;
866         unlink_errno = xunlink (ifname) == 0 ? 0 : errno;
867         sigprocmask (SIG_SETMASK, &oldset, NULL);
868
869         if (unlink_errno)
870           {
871             WARN ((stderr, "%s: ", program_name));
872             if (!quiet)
873               {
874                 errno = unlink_errno;
875                 perror (ifname);
876               }
877           }
878       }
879
880     if (method == -1) {
881         if (!to_stdout)
882           remove_output_file ();
883         return;
884     }
885
886     /* Display statistics */
887     if(verbose) {
888         if (test) {
889             fprintf(stderr, " OK");
890         } else if (decompress) {
891             display_ratio(bytes_out-(bytes_in-header_bytes), bytes_out,stderr);
892         } else {
893             display_ratio(bytes_in-(bytes_out-header_bytes), bytes_in, stderr);
894         }
895         if (!test && !to_stdout) {
896             fprintf(stderr, " -- replaced with %s", ofname);
897         }
898         fprintf(stderr, "\n");
899     }
900 }
901
902 /* ========================================================================
903  * Create the output file. Return OK or ERROR.
904  * Try several times if necessary to avoid truncating the z_suffix. For
905  * example, do not create a compressed file of name "1234567890123."
906  * Sets save_orig_name to true if the file name has been truncated.
907  * IN assertions: the input file has already been open (ifd is set) and
908  *   ofname has already been updated if there was an original name.
909  * OUT assertions: ifd and ofd are closed in case of error.
910  */
911 local int create_outfile()
912 {
913   int name_shortened = 0;
914   int flags = (O_WRONLY | O_CREAT | O_EXCL
915                | (ascii && decompress ? 0 : O_BINARY));
916
917   for (;;)
918     {
919       int open_errno;
920       sigset_t oldset;
921
922       sigprocmask (SIG_BLOCK, &caught_signals, &oldset);
923       remove_ofname_fd = ofd = OPEN (ofname, flags, RW_USER);
924       open_errno = errno;
925       sigprocmask (SIG_SETMASK, &oldset, NULL);
926
927       if (0 <= ofd)
928         break;
929
930       switch (open_errno)
931         {
932 #ifdef ENAMETOOLONG
933         case ENAMETOOLONG:
934           shorten_name (ofname);
935           name_shortened = 1;
936           break;
937 #endif
938
939         case EEXIST:
940           if (check_ofname () != OK)
941             {
942               close (ifd);
943               return ERROR;
944             }
945           break;
946
947         default:
948           progerror (ofname);
949           close (ifd);
950           return ERROR;
951         }
952     }
953
954   if (name_shortened && decompress)
955     {
956       /* name might be too long if an original name was saved */
957       WARN ((stderr, "%s: %s: warning, name truncated\n",
958              program_name, ofname));
959     }
960
961   return OK;
962 }
963
964 /* ========================================================================
965  * Return a pointer to the 'z' suffix of a file name, or NULL. For all
966  * systems, ".gz", ".z", ".Z", ".taz", ".tgz", "-gz", "-z" and "_z" are
967  * accepted suffixes, in addition to the value of the --suffix option.
968  * ".tgz" is a useful convention for tar.z files on systems limited
969  * to 3 characters extensions. On such systems, ".?z" and ".??z" are
970  * also accepted suffixes. For Unix, we do not want to accept any
971  * .??z suffix as indicating a compressed file; some people use .xyz
972  * to denote volume data.
973  *   On systems allowing multiple versions of the same file (such as VMS),
974  * this function removes any version suffix in the given name.
975  */
976 local char *get_suffix(name)
977     char *name;
978 {
979     int nlen, slen;
980     char suffix[MAX_SUFFIX+3]; /* last chars of name, forced to lower case */
981     static char const *known_suffixes[] =
982        {NULL, ".gz", ".z", ".taz", ".tgz", "-gz", "-z", "_z",
983 #ifdef MAX_EXT_CHARS
984           "z",
985 #endif
986           NULL};
987     char const **suf = known_suffixes;
988
989     *suf = z_suffix;
990     if (strequ(z_suffix, "z")) suf++; /* check long suffixes first */
991
992 #ifdef SUFFIX_SEP
993     /* strip a version number from the file name */
994     {
995         char *v = strrchr(name, SUFFIX_SEP);
996         if (v != NULL) *v = '\0';
997     }
998 #endif
999     nlen = strlen(name);
1000     if (nlen <= MAX_SUFFIX+2) {
1001         strcpy(suffix, name);
1002     } else {
1003         strcpy(suffix, name+nlen-MAX_SUFFIX-2);
1004     }
1005     strlwr(suffix);
1006     slen = strlen(suffix);
1007     do {
1008        int s = strlen(*suf);
1009        if (slen > s && suffix[slen-s-1] != PATH_SEP
1010            && strequ(suffix + slen - s, *suf)) {
1011            return name+nlen-s;
1012        }
1013     } while (*++suf != NULL);
1014
1015     return NULL;
1016 }
1017
1018
1019 /* Open file NAME with the given flags and mode and store its status
1020    into *ST.  Return a file descriptor to the newly opened file, or -1
1021    (setting errno) on failure.  */
1022 static int
1023 open_and_stat (char *name, int flags, mode_t mode, struct stat *st)
1024 {
1025   int fd;
1026
1027   /* Refuse to follow symbolic links unless -c or -f.  */
1028   if (!to_stdout && !force)
1029     {
1030       if (HAVE_WORKING_O_NOFOLLOW)
1031         flags |= O_NOFOLLOW;
1032       else
1033         {
1034 #if HAVE_LSTAT || defined lstat
1035           if (lstat (name, st) != 0)
1036             return -1;
1037           else if (S_ISLNK (st->st_mode))
1038             {
1039               errno = ELOOP;
1040               return -1;
1041             }
1042 #endif
1043         }
1044     }
1045
1046   fd = OPEN (name, flags, mode);
1047   if (0 <= fd && fstat (fd, st) != 0)
1048     {
1049       int e = errno;
1050       close (fd);
1051       errno = e;
1052       return -1;
1053     }
1054   return fd;
1055 }
1056
1057
1058 /* ========================================================================
1059  * Set ifname to the input file name (with a suffix appended if necessary)
1060  * and istat to its stats. For decompression, if no file exists with the
1061  * original name, try adding successively z_suffix, .gz, .z, -z and .Z.
1062  * For MSDOS, we try only z_suffix and z.
1063  * Return an open file descriptor or -1.
1064  */
1065 static int
1066 open_input_file (iname, sbuf)
1067     char *iname;
1068     struct stat *sbuf;
1069 {
1070     int ilen;  /* strlen(ifname) */
1071     int z_suffix_errno = 0;
1072     static char const *suffixes[] = {NULL, ".gz", ".z", "-z", ".Z", NULL};
1073     char const **suf = suffixes;
1074     char const *s;
1075 #ifdef NO_MULTIPLE_DOTS
1076     char *dot; /* pointer to ifname extension, or NULL */
1077 #endif
1078     int fd;
1079     int open_flags = (O_RDONLY | O_NONBLOCK | O_NOCTTY
1080                       | (ascii && !decompress ? 0 : O_BINARY));
1081
1082     *suf = z_suffix;
1083
1084     if (sizeof ifname - 1 <= strlen (iname))
1085         goto name_too_long;
1086
1087     strcpy(ifname, iname);
1088
1089     /* If input file exists, return OK. */
1090     fd = open_and_stat (ifname, open_flags, RW_USER, sbuf);
1091     if (0 <= fd)
1092       return fd;
1093
1094     if (!decompress || errno != ENOENT) {
1095         progerror(ifname);
1096         return -1;
1097     }
1098     /* file.ext doesn't exist, try adding a suffix (after removing any
1099      * version number for VMS).
1100      */
1101     s = get_suffix(ifname);
1102     if (s != NULL) {
1103         progerror(ifname); /* ifname already has z suffix and does not exist */
1104         return -1;
1105     }
1106 #ifdef NO_MULTIPLE_DOTS
1107     dot = strrchr(ifname, '.');
1108     if (dot == NULL) {
1109         strcat(ifname, ".");
1110         dot = strrchr(ifname, '.');
1111     }
1112 #endif
1113     ilen = strlen(ifname);
1114     if (strequ(z_suffix, ".gz")) suf++;
1115
1116     /* Search for all suffixes */
1117     do {
1118         char const *s0 = s = *suf;
1119         strcpy (ifname, iname);
1120 #ifdef NO_MULTIPLE_DOTS
1121         if (*s == '.') s++;
1122         if (*dot == '\0') strcpy (dot, ".");
1123 #endif
1124 #ifdef MAX_EXT_CHARS
1125         if (MAX_EXT_CHARS < strlen (s) + strlen (dot + 1))
1126           dot[MAX_EXT_CHARS + 1 - strlen (s)] = '\0';
1127 #endif
1128         if (sizeof ifname <= ilen + strlen (s))
1129           goto name_too_long;
1130         strcat(ifname, s);
1131         fd = open_and_stat (ifname, open_flags, RW_USER, sbuf);
1132         if (0 <= fd)
1133           return fd;
1134         if (errno != ENOENT)
1135           {
1136             progerror (ifname);
1137             return -1;
1138           }
1139         if (strequ (s0, z_suffix))
1140           z_suffix_errno = errno;
1141     } while (*++suf != NULL);
1142
1143     /* No suffix found, complain using z_suffix: */
1144     strcpy(ifname, iname);
1145 #ifdef NO_MULTIPLE_DOTS
1146     if (*dot == '\0') strcpy(dot, ".");
1147 #endif
1148 #ifdef MAX_EXT_CHARS
1149     if (MAX_EXT_CHARS < z_len + strlen (dot + 1))
1150       dot[MAX_EXT_CHARS + 1 - z_len] = '\0';
1151 #endif
1152     strcat(ifname, z_suffix);
1153     errno = z_suffix_errno;
1154     progerror(ifname);
1155     return -1;
1156
1157  name_too_long:
1158     fprintf (stderr, "%s: %s: file name too long\n", program_name, iname);
1159     exit_code = ERROR;
1160     return -1;
1161 }
1162
1163 /* ========================================================================
1164  * Generate ofname given ifname. Return OK, or WARNING if file must be skipped.
1165  * Sets save_orig_name to true if the file name has been truncated.
1166  */
1167 local int make_ofname()
1168 {
1169     char *suff;            /* ofname z suffix */
1170
1171     strcpy(ofname, ifname);
1172     /* strip a version number if any and get the gzip suffix if present: */
1173     suff = get_suffix(ofname);
1174
1175     if (decompress) {
1176         if (suff == NULL) {
1177             /* With -t or -l, try all files (even without .gz suffix)
1178              * except with -r (behave as with just -dr).
1179              */
1180             if (!recursive && (list || test)) return OK;
1181
1182             /* Avoid annoying messages with -r */
1183             if (verbose || (!recursive && !quiet)) {
1184                 WARN((stderr,"%s: %s: unknown suffix -- ignored\n",
1185                       program_name, ifname));
1186             }
1187             return WARNING;
1188         }
1189         /* Make a special case for .tgz and .taz: */
1190         strlwr(suff);
1191         if (strequ(suff, ".tgz") || strequ(suff, ".taz")) {
1192             strcpy(suff, ".tar");
1193         } else {
1194             *suff = '\0'; /* strip the z suffix */
1195         }
1196         /* ofname might be changed later if infile contains an original name */
1197
1198     } else if (suff && ! force) {
1199         /* Avoid annoying messages with -r (see treat_dir()) */
1200         if (verbose || (!recursive && !quiet)) {
1201             /* Don't use WARN, as it affects exit status.  */
1202             fprintf (stderr, "%s: %s already has %s suffix -- unchanged\n",
1203                      program_name, ifname, suff);
1204         }
1205         return WARNING;
1206     } else {
1207         save_orig_name = 0;
1208
1209 #ifdef NO_MULTIPLE_DOTS
1210         suff = strrchr(ofname, '.');
1211         if (suff == NULL) {
1212             if (sizeof ofname <= strlen (ofname) + 1)
1213                 goto name_too_long;
1214             strcat(ofname, ".");
1215 #  ifdef MAX_EXT_CHARS
1216             if (strequ(z_suffix, "z")) {
1217                 if (sizeof ofname <= strlen (ofname) + 2)
1218                     goto name_too_long;
1219                 strcat(ofname, "gz"); /* enough room */
1220                 return OK;
1221             }
1222         /* On the Atari and some versions of MSDOS,
1223          * ENAMETOOLONG does not work correctly.  So we
1224          * must truncate here.
1225          */
1226         } else if (strlen(suff)-1 + z_len > MAX_SUFFIX) {
1227             suff[MAX_SUFFIX+1-z_len] = '\0';
1228             save_orig_name = 1;
1229 #  endif
1230         }
1231 #endif /* NO_MULTIPLE_DOTS */
1232         if (sizeof ofname <= strlen (ofname) + z_len)
1233             goto name_too_long;
1234         strcat(ofname, z_suffix);
1235
1236     } /* decompress ? */
1237     return OK;
1238
1239  name_too_long:
1240     WARN ((stderr, "%s: %s: file name too long\n", program_name, ifname));
1241     return WARNING;
1242 }
1243
1244 /* Discard NBYTES input bytes from the input, or up through the next
1245    zero byte if NBYTES == (size_t) -1.  If FLAGS say that the header
1246    CRC should be computed, update the CRC accordingly.  */
1247 static void
1248 discard_input_bytes (nbytes, flags)
1249     size_t nbytes;
1250     unsigned int flags;
1251 {
1252   while (nbytes != 0)
1253     {
1254       uch c = get_byte ();
1255       if (flags & HEADER_CRC)
1256         updcrc (&c, 1);
1257       if (nbytes != (size_t) -1)
1258         nbytes--;
1259       else if (! c)
1260         break;
1261     }
1262 }
1263
1264 /* ========================================================================
1265  * Check the magic number of the input file and update ofname if an
1266  * original name was given and to_stdout is not set.
1267  * Return the compression method, -1 for error, -2 for warning.
1268  * Set inptr to the offset of the next byte to be processed.
1269  * Updates time_stamp if there is one and --no-time is not used.
1270  * This function may be called repeatedly for an input file consisting
1271  * of several contiguous gzip'ed members.
1272  * IN assertions: there is at least one remaining compressed member.
1273  *   If the member is a zip file, it must be the only one.
1274  */
1275 local int get_method(in)
1276     int in;        /* input file descriptor */
1277 {
1278     uch flags;     /* compression flags */
1279     uch magic[10]; /* magic header */
1280     int imagic0;   /* first magic byte or EOF */
1281     int imagic1;   /* like magic[1], but can represent EOF */
1282     ulg stamp;     /* time stamp */
1283
1284     /* If --force and --stdout, zcat == cat, so do not complain about
1285      * premature end of file: use try_byte instead of get_byte.
1286      */
1287     if (force && to_stdout) {
1288         imagic0 = try_byte();
1289         magic[0] = imagic0;
1290         imagic1 = try_byte ();
1291         magic[1] = imagic1;
1292         /* If try_byte returned EOF, magic[1] == (char) EOF.  */
1293     } else {
1294         magic[0] = get_byte ();
1295         imagic0 = 0;
1296         if (magic[0]) {
1297             magic[1] = get_byte ();
1298             imagic1 = 0; /* avoid lint warning */
1299         } else {
1300             imagic1 = try_byte ();
1301             magic[1] = imagic1;
1302         }
1303     }
1304     method = -1;                 /* unknown yet */
1305     part_nb++;                   /* number of parts in gzip file */
1306     header_bytes = 0;
1307     last_member = RECORD_IO;
1308     /* assume multiple members in gzip file except for record oriented I/O */
1309
1310     if (memcmp(magic, GZIP_MAGIC, 2) == 0
1311         || memcmp(magic, OLD_GZIP_MAGIC, 2) == 0) {
1312
1313         method = (int)get_byte();
1314         if (method != DEFLATED) {
1315             fprintf(stderr,
1316                     "%s: %s: unknown method %d -- not supported\n",
1317                     program_name, ifname, method);
1318             exit_code = ERROR;
1319             return -1;
1320         }
1321         work = unzip;
1322         flags  = (uch)get_byte();
1323
1324         if ((flags & ENCRYPTED) != 0) {
1325             fprintf(stderr,
1326                     "%s: %s is encrypted -- not supported\n",
1327                     program_name, ifname);
1328             exit_code = ERROR;
1329             return -1;
1330         }
1331         if ((flags & RESERVED) != 0) {
1332             fprintf(stderr,
1333                     "%s: %s has flags 0x%x -- not supported\n",
1334                     program_name, ifname, flags);
1335             exit_code = ERROR;
1336             if (force <= 1) return -1;
1337         }
1338         stamp  = (ulg)get_byte();
1339         stamp |= ((ulg)get_byte()) << 8;
1340         stamp |= ((ulg)get_byte()) << 16;
1341         stamp |= ((ulg)get_byte()) << 24;
1342         if (stamp != 0 && !no_time)
1343           {
1344             time_stamp.tv_sec = stamp;
1345             time_stamp.tv_nsec = 0;
1346           }
1347
1348         magic[8] = get_byte ();  /* Ignore extra flags.  */
1349         magic[9] = get_byte ();  /* Ignore OS type.  */
1350
1351         if (flags & HEADER_CRC)
1352           {
1353             magic[2] = DEFLATED;
1354             magic[3] = flags;
1355             magic[4] = stamp & 0xff;
1356             magic[5] = (stamp >> 8) & 0xff;
1357             magic[6] = (stamp >> 16) & 0xff;
1358             magic[7] = stamp >> 24;
1359             updcrc (NULL, 0);
1360             updcrc (magic, 10);
1361           }
1362
1363         if ((flags & EXTRA_FIELD) != 0) {
1364             uch lenbuf[2];
1365             unsigned int len = lenbuf[0] = get_byte ();
1366             len |= (lenbuf[1] = get_byte ()) << 8;
1367             if (verbose) {
1368                 fprintf(stderr,"%s: %s: extra field of %u bytes ignored\n",
1369                         program_name, ifname, len);
1370             }
1371             if (flags & HEADER_CRC)
1372               updcrc (lenbuf, 2);
1373             discard_input_bytes (len, flags);
1374         }
1375
1376         /* Get original file name if it was truncated */
1377         if ((flags & ORIG_NAME) != 0) {
1378             if (no_name || (to_stdout && !list) || part_nb > 1) {
1379                 /* Discard the old name */
1380                 discard_input_bytes (-1, flags);
1381             } else {
1382                 /* Copy the base name. Keep a directory prefix intact. */
1383                 char *p = gzip_base_name (ofname);
1384                 char *base = p;
1385                 for (;;) {
1386                     *p = (char) get_byte ();
1387                     if (*p++ == '\0') break;
1388                     if (p >= ofname+sizeof(ofname)) {
1389                         gzip_error ("corrupted input -- file name too large");
1390                     }
1391                 }
1392                 if (flags & HEADER_CRC)
1393                   updcrc ((uch *) base, p - base);
1394                 p = gzip_base_name (base);
1395                 memmove (base, p, strlen (p) + 1);
1396                 /* If necessary, adapt the name to local OS conventions: */
1397                 if (!list) {
1398                    MAKE_LEGAL_NAME(base);
1399                    if (base) list=0; /* avoid warning about unused variable */
1400                 }
1401             } /* no_name || to_stdout */
1402         } /* ORIG_NAME */
1403
1404         /* Discard file comment if any */
1405         if ((flags & COMMENT) != 0) {
1406             discard_input_bytes (-1, flags);
1407         }
1408
1409         if (flags & HEADER_CRC)
1410           {
1411             unsigned int crc16 = updcrc (magic, 0) & 0xffff;
1412             unsigned int header16 = get_byte ();
1413             header16 |= ((unsigned int) get_byte ()) << 8;
1414             if (header16 != crc16)
1415               {
1416                 fprintf (stderr,
1417                          "%s: %s: header checksum 0x%04x != computed checksum 0x%04x\n",
1418                          program_name, ifname, header16, crc16);
1419                 exit_code = ERROR;
1420                 if (force <= 1)
1421                   return -1;
1422               }
1423           }
1424
1425         if (part_nb == 1) {
1426             header_bytes = inptr + 2*4; /* include crc and size */
1427         }
1428
1429     } else if (memcmp(magic, PKZIP_MAGIC, 2) == 0 && inptr == 2
1430             && memcmp((char*)inbuf, PKZIP_MAGIC, 4) == 0) {
1431         /* To simplify the code, we support a zip file when alone only.
1432          * We are thus guaranteed that the entire local header fits in inbuf.
1433          */
1434         inptr = 0;
1435         work = unzip;
1436         if (check_zipfile(in) != OK) return -1;
1437         /* check_zipfile may get ofname from the local header */
1438         last_member = 1;
1439
1440     } else if (memcmp(magic, PACK_MAGIC, 2) == 0) {
1441         work = unpack;
1442         method = PACKED;
1443
1444     } else if (memcmp(magic, LZW_MAGIC, 2) == 0) {
1445         work = unlzw;
1446         method = COMPRESSED;
1447         last_member = 1;
1448
1449     } else if (memcmp(magic, LZH_MAGIC, 2) == 0) {
1450         work = unlzh;
1451         method = LZHED;
1452         last_member = 1;
1453
1454     } else if (force && to_stdout && !list) { /* pass input unchanged */
1455         method = STORED;
1456         work = copy;
1457         if (imagic1 != EOF)
1458             inptr--;
1459         last_member = 1;
1460         if (imagic0 != EOF) {
1461             write_buf(fileno(stdout), magic, 1);
1462             bytes_out++;
1463         }
1464     }
1465     if (method >= 0) return method;
1466
1467     if (part_nb == 1) {
1468         fprintf (stderr, "\n%s: %s: not in gzip format\n",
1469                  program_name, ifname);
1470         exit_code = ERROR;
1471         return -1;
1472     } else {
1473         if (magic[0] == 0)
1474           {
1475             int inbyte;
1476             for (inbyte = imagic1;  inbyte == 0;  inbyte = try_byte ())
1477               continue;
1478             if (inbyte == EOF)
1479               {
1480                 if (verbose)
1481                   WARN ((stderr, "\n%s: %s: decompression OK, trailing zero bytes ignored\n",
1482                          program_name, ifname));
1483                 return -3;
1484               }
1485           }
1486
1487         WARN((stderr, "\n%s: %s: decompression OK, trailing garbage ignored\n",
1488               program_name, ifname));
1489         return -2;
1490     }
1491 }
1492
1493 /* ========================================================================
1494  * Display the characteristics of the compressed file.
1495  * If the given method is < 0, display the accumulated totals.
1496  * IN assertions: time_stamp, header_bytes and ifile_size are initialized.
1497  */
1498 local void do_list(ifd, method)
1499     int ifd;     /* input file descriptor */
1500     int method;  /* compression method */
1501 {
1502     ulg crc;  /* original crc */
1503     static int first_time = 1;
1504     static char const *const methods[MAX_METHODS] = {
1505         "store",  /* 0 */
1506         "compr",  /* 1 */
1507         "pack ",  /* 2 */
1508         "lzh  ",  /* 3 */
1509         "", "", "", "", /* 4 to 7 reserved */
1510         "defla"}; /* 8 */
1511     int positive_off_t_width = 1;
1512     off_t o;
1513
1514     for (o = OFF_T_MAX;  9 < o;  o /= 10) {
1515         positive_off_t_width++;
1516     }
1517
1518     if (first_time && method >= 0) {
1519         first_time = 0;
1520         if (verbose)  {
1521             printf("method  crc     date  time  ");
1522         }
1523         if (!quiet) {
1524             printf("%*.*s %*.*s  ratio uncompressed_name\n",
1525                    positive_off_t_width, positive_off_t_width, "compressed",
1526                    positive_off_t_width, positive_off_t_width, "uncompressed");
1527         }
1528     } else if (method < 0) {
1529         if (total_in <= 0 || total_out <= 0) return;
1530         if (verbose) {
1531             printf("                            ");
1532         }
1533         if (verbose || !quiet) {
1534             fprint_off(stdout, total_in, positive_off_t_width);
1535             printf(" ");
1536             fprint_off(stdout, total_out, positive_off_t_width);
1537             printf(" ");
1538         }
1539         display_ratio(total_out-(total_in-header_bytes), total_out, stdout);
1540         /* header_bytes is not meaningful but used to ensure the same
1541          * ratio if there is a single file.
1542          */
1543         printf(" (totals)\n");
1544         return;
1545     }
1546     crc = (ulg)~0; /* unknown */
1547     bytes_out = -1L;
1548     bytes_in = ifile_size;
1549
1550     if (!RECORD_IO && method == DEFLATED && !last_member) {
1551         /* Get the crc and uncompressed size for gzip'ed (not zip'ed) files.
1552          * If the lseek fails, we could use read() to get to the end, but
1553          * --list is used to get quick results.
1554          * Use "gunzip < foo.gz | wc -c" to get the uncompressed size if
1555          * you are not concerned about speed.
1556          */
1557         bytes_in = lseek(ifd, (off_t)(-8), SEEK_END);
1558         if (bytes_in != -1L) {
1559             uch buf[8];
1560             bytes_in += 8L;
1561             if (read(ifd, (char*)buf, sizeof(buf)) != sizeof(buf)) {
1562                 read_error();
1563             }
1564             crc       = LG(buf);
1565             bytes_out = LG(buf+4);
1566         }
1567     }
1568
1569     if (verbose)
1570       {
1571         struct tm *tm = localtime (&time_stamp.tv_sec);
1572         printf ("%5s %08lx ", methods[method], crc);
1573         if (tm)
1574           printf ("%s%3d %02d:%02d ",
1575                   ("Jan\0Feb\0Mar\0Apr\0May\0Jun\0Jul\0Aug\0Sep\0Oct\0Nov\0Dec"
1576                    + 4 * tm->tm_mon),
1577                   tm->tm_mday, tm->tm_hour, tm->tm_min);
1578         else
1579           printf ("??? ?? ??:?? ");
1580       }
1581     fprint_off(stdout, bytes_in, positive_off_t_width);
1582     printf(" ");
1583     fprint_off(stdout, bytes_out, positive_off_t_width);
1584     printf(" ");
1585     if (bytes_in  == -1L) {
1586         total_in = -1L;
1587         bytes_in = bytes_out = header_bytes = 0;
1588     } else if (total_in >= 0) {
1589         total_in  += bytes_in;
1590     }
1591     if (bytes_out == -1L) {
1592         total_out = -1L;
1593         bytes_in = bytes_out = header_bytes = 0;
1594     } else if (total_out >= 0) {
1595         total_out += bytes_out;
1596     }
1597     display_ratio(bytes_out-(bytes_in-header_bytes), bytes_out, stdout);
1598     printf(" %s\n", ofname);
1599 }
1600
1601 /* ========================================================================
1602  * Shorten the given name by one character, or replace a .tar extension
1603  * with .tgz. Truncate the last part of the name which is longer than
1604  * MIN_PART characters: 1234.678.012.gz -> 123.678.012.gz. If the name
1605  * has only parts shorter than MIN_PART truncate the longest part.
1606  * For decompression, just remove the last character of the name.
1607  *
1608  * IN assertion: for compression, the suffix of the given name is z_suffix.
1609  */
1610 local void shorten_name(name)
1611     char *name;
1612 {
1613     int len;                 /* length of name without z_suffix */
1614     char *trunc = NULL;      /* character to be truncated */
1615     int plen;                /* current part length */
1616     int min_part = MIN_PART; /* current minimum part length */
1617     char *p;
1618
1619     len = strlen(name);
1620     if (decompress) {
1621         if (len <= 1)
1622           gzip_error ("name too short");
1623         name[len-1] = '\0';
1624         return;
1625     }
1626     p = get_suffix(name);
1627     if (! p)
1628       gzip_error ("can't recover suffix\n");
1629     *p = '\0';
1630     save_orig_name = 1;
1631
1632     /* compress 1234567890.tar to 1234567890.tgz */
1633     if (len > 4 && strequ(p-4, ".tar")) {
1634         strcpy(p-4, ".tgz");
1635         return;
1636     }
1637     /* Try keeping short extensions intact:
1638      * 1234.678.012.gz -> 123.678.012.gz
1639      */
1640     do {
1641         p = strrchr(name, PATH_SEP);
1642         p = p ? p+1 : name;
1643         while (*p) {
1644             plen = strcspn(p, PART_SEP);
1645             p += plen;
1646             if (plen > min_part) trunc = p-1;
1647             if (*p) p++;
1648         }
1649     } while (trunc == NULL && --min_part != 0);
1650
1651     if (trunc != NULL) {
1652         do {
1653             trunc[0] = trunc[1];
1654         } while (*trunc++);
1655         trunc--;
1656     } else {
1657         trunc = strrchr(name, PART_SEP[0]);
1658         if (!trunc)
1659           gzip_error ("internal error in shorten_name");
1660         if (trunc[1] == '\0') trunc--; /* force truncation */
1661     }
1662     strcpy(trunc, z_suffix);
1663 }
1664
1665 /* ========================================================================
1666  * The compressed file already exists, so ask for confirmation.
1667  * Return ERROR if the file must be skipped.
1668  */
1669 local int check_ofname()
1670 {
1671     /* Ask permission to overwrite the existing file */
1672     if (!force) {
1673         int ok = 0;
1674         fprintf (stderr, "%s: %s already exists;", program_name, ofname);
1675         if (foreground && (presume_input_tty || isatty(fileno(stdin)))) {
1676             fprintf(stderr, " do you wish to overwrite (y or n)? ");
1677             fflush(stderr);
1678             ok = yesno();
1679         }
1680         if (!ok) {
1681             fprintf(stderr, "\tnot overwritten\n");
1682             if (exit_code == OK) exit_code = WARNING;
1683             return ERROR;
1684         }
1685     }
1686     if (xunlink (ofname)) {
1687         progerror(ofname);
1688         return ERROR;
1689     }
1690     return OK;
1691 }
1692
1693
1694 /* ========================================================================
1695  * Copy modes, times, ownership from input file to output file.
1696  * IN assertion: to_stdout is false.
1697  */
1698 local void copy_stat(ifstat)
1699     struct stat *ifstat;
1700 {
1701     mode_t mode = ifstat->st_mode & S_IRWXUGO;
1702     int r;
1703
1704 #ifndef NO_UTIME
1705     struct timespec timespec[2];
1706     timespec[0] = get_stat_atime (ifstat);
1707     timespec[1] = get_stat_mtime (ifstat);
1708
1709     if (decompress && 0 <= time_stamp.tv_nsec
1710         && ! (timespec[1].tv_sec == time_stamp.tv_sec
1711               && timespec[1].tv_nsec == time_stamp.tv_nsec))
1712       {
1713         timespec[1] = time_stamp;
1714         if (verbose > 1) {
1715             fprintf(stderr, "%s: time stamp restored\n", ofname);
1716         }
1717       }
1718
1719     if (fdutimens (ofd, ofname, timespec) != 0)
1720       {
1721         int e = errno;
1722         WARN ((stderr, "%s: ", program_name));
1723         if (!quiet)
1724           {
1725             errno = e;
1726             perror (ofname);
1727           }
1728       }
1729 #endif
1730
1731 #ifndef NO_CHOWN
1732     /* Copy ownership */
1733 # if HAVE_FCHOWN
1734     ignore_value (fchown (ofd, ifstat->st_uid, ifstat->st_gid));
1735 # elif HAVE_CHOWN
1736     ignore_value (chown (ofname, ifstat->st_uid, ifstat->st_gid));
1737 # endif
1738 #endif
1739
1740     /* Copy the protection modes */
1741 #if HAVE_FCHMOD
1742     r = fchmod (ofd, mode);
1743 #else
1744     r = chmod (ofname, mode);
1745 #endif
1746     if (r != 0) {
1747         int e = errno;
1748         WARN ((stderr, "%s: ", program_name));
1749         if (!quiet) {
1750             errno = e;
1751             perror(ofname);
1752         }
1753     }
1754 }
1755
1756 #if ! NO_DIR
1757
1758 /* ========================================================================
1759  * Recurse through the given directory.
1760  */
1761 local void treat_dir (fd, dir)
1762     int fd;
1763     char *dir;
1764 {
1765     DIR      *dirp;
1766     char     nbuf[MAX_PATH_LEN];
1767     char *entries;
1768     char const *entry;
1769     size_t entrylen;
1770
1771     dirp = fdopendir (fd);
1772
1773     if (dirp == NULL) {
1774         progerror(dir);
1775         close (fd);
1776         return ;
1777     }
1778
1779     entries = streamsavedir (dirp);
1780     if (! entries)
1781       progerror (dir);
1782     if (closedir (dirp) != 0)
1783       progerror (dir);
1784     if (! entries)
1785       return;
1786
1787     for (entry = entries; *entry; entry += entrylen + 1) {
1788         size_t len = strlen (dir);
1789         entrylen = strlen (entry);
1790         if (strequ (entry, ".") || strequ (entry, ".."))
1791           continue;
1792         if (len + entrylen < MAX_PATH_LEN - 2) {
1793             strcpy(nbuf,dir);
1794             if (len != 0 /* dir = "" means current dir on Amiga */
1795 #ifdef PATH_SEP2
1796                 && dir[len-1] != PATH_SEP2
1797 #endif
1798 #ifdef PATH_SEP3
1799                 && dir[len-1] != PATH_SEP3
1800 #endif
1801             ) {
1802                 nbuf[len++] = PATH_SEP;
1803             }
1804             strcpy (nbuf + len, entry);
1805             treat_file(nbuf);
1806         } else {
1807             fprintf(stderr,"%s: %s/%s: pathname too long\n",
1808                     program_name, dir, entry);
1809             exit_code = ERROR;
1810         }
1811     }
1812     free (entries);
1813 }
1814 #endif /* ! NO_DIR */
1815
1816 /* Make sure signals get handled properly.  */
1817
1818 static void
1819 install_signal_handlers ()
1820 {
1821   int nsigs = sizeof handled_sig / sizeof handled_sig[0];
1822   int i;
1823
1824 #if SA_NOCLDSTOP
1825   struct sigaction act;
1826
1827   sigemptyset (&caught_signals);
1828   for (i = 0; i < nsigs; i++)
1829     {
1830       sigaction (handled_sig[i], NULL, &act);
1831       if (act.sa_handler != SIG_IGN)
1832         sigaddset (&caught_signals, handled_sig[i]);
1833     }
1834
1835   act.sa_handler = abort_gzip_signal;
1836   act.sa_mask = caught_signals;
1837   act.sa_flags = 0;
1838
1839   for (i = 0; i < nsigs; i++)
1840     if (sigismember (&caught_signals, handled_sig[i]))
1841       {
1842         if (i == 0)
1843           foreground = 1;
1844         sigaction (handled_sig[i], &act, NULL);
1845       }
1846 #else
1847   for (i = 0; i < nsigs; i++)
1848     if (signal (handled_sig[i], SIG_IGN) != SIG_IGN)
1849       {
1850         if (i == 0)
1851           foreground = 1;
1852         signal (handled_sig[i], abort_gzip_signal);
1853         siginterrupt (handled_sig[i], 1);
1854       }
1855 #endif
1856 }
1857
1858 /* ========================================================================
1859  * Free all dynamically allocated variables and exit with the given code.
1860  */
1861 local void do_exit(exitcode)
1862     int exitcode;
1863 {
1864     static int in_exit = 0;
1865
1866     if (in_exit) exit(exitcode);
1867     in_exit = 1;
1868     free(env);
1869     env  = NULL;
1870     free(args);
1871     args = NULL;
1872     FREE(inbuf);
1873     FREE(outbuf);
1874     FREE(d_buf);
1875     FREE(window);
1876 #ifndef MAXSEG_64K
1877     FREE(tab_prefix);
1878 #else
1879     FREE(tab_prefix0);
1880     FREE(tab_prefix1);
1881 #endif
1882     exit(exitcode);
1883 }
1884
1885 /* ========================================================================
1886  * Close and unlink the output file.
1887  */
1888 static void
1889 remove_output_file ()
1890 {
1891   int fd;
1892   sigset_t oldset;
1893
1894   sigprocmask (SIG_BLOCK, &caught_signals, &oldset);
1895   fd = remove_ofname_fd;
1896   if (0 <= fd)
1897     {
1898       remove_ofname_fd = -1;
1899       close (fd);
1900       xunlink (ofname);
1901     }
1902   sigprocmask (SIG_SETMASK, &oldset, NULL);
1903 }
1904
1905 /* ========================================================================
1906  * Error handler.
1907  */
1908 void
1909 abort_gzip ()
1910 {
1911    remove_output_file ();
1912    do_exit(ERROR);
1913 }
1914
1915 /* ========================================================================
1916  * Signal handler.
1917  */
1918 static RETSIGTYPE
1919 abort_gzip_signal (sig)
1920      int sig;
1921 {
1922   if (! SA_NOCLDSTOP)
1923     signal (sig, SIG_IGN);
1924    remove_output_file ();
1925    if (sig == exiting_signal)
1926      _exit (WARNING);
1927    signal (sig, SIG_DFL);
1928    raise (sig);
1929 }