git.gag.com Git - debian/gzip/blob - gzip.c

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