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