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