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