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