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