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