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