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