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