Home | History | Annotate | Line # | Download | only in gzip
gzip.c revision 1.61
      1  1.61      yamt /*	$NetBSD: gzip.c,v 1.61 2004/07/12 10:21:11 yamt Exp $	*/
      2   1.1       mrg 
      3   1.1       mrg /*
      4  1.37       mrg  * Copyright (c) 1997, 1998, 2003, 2004 Matthew R. Green
      5   1.1       mrg  * All rights reserved.
      6   1.1       mrg  *
      7   1.1       mrg  * Redistribution and use in source and binary forms, with or without
      8   1.1       mrg  * modification, are permitted provided that the following conditions
      9   1.1       mrg  * are met:
     10   1.1       mrg  * 1. Redistributions of source code must retain the above copyright
     11   1.1       mrg  *    notice, this list of conditions and the following disclaimer.
     12   1.1       mrg  * 2. Redistributions in binary form must reproduce the above copyright
     13   1.1       mrg  *    notice, this list of conditions and the following disclaimer in the
     14   1.1       mrg  *    documentation and/or other materials provided with the distribution.
     15   1.1       mrg  * 3. The name of the author may not be used to endorse or promote products
     16   1.1       mrg  *    derived from this software without specific prior written permission.
     17   1.1       mrg  *
     18   1.1       mrg  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19   1.1       mrg  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20   1.1       mrg  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21   1.1       mrg  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22   1.1       mrg  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     23   1.1       mrg  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     24   1.1       mrg  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     25   1.1       mrg  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     26   1.1       mrg  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     27   1.1       mrg  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     28   1.1       mrg  * SUCH DAMAGE.
     29   1.1       mrg  */
     30   1.1       mrg 
     31   1.4       mrg #include <sys/cdefs.h>
     32   1.4       mrg #ifndef lint
     33  1.37       mrg __COPYRIGHT("@(#) Copyright (c) 1997, 1998, 2003, 2004 Matthew R. Green\n\
     34   1.4       mrg      All rights reserved.\n");
     35  1.61      yamt __RCSID("$NetBSD: gzip.c,v 1.61 2004/07/12 10:21:11 yamt Exp $");
     36   1.4       mrg #endif /* not lint */
     37   1.4       mrg 
     38   1.1       mrg /*
     39   1.1       mrg  * gzip.c -- GPL free gzip using zlib.
     40   1.1       mrg  *
     41   1.6       mrg  * TODO:
     42   1.6       mrg  *	- handle .taz/.tgz files?
     43  1.10       mrg  *	- use mmap where possible
     44  1.16       wiz  *	- handle some signals better (remove outfile?)
     45  1.41       mrg  *	- make bzip2/compress -v/-t/-l support work as well as possible
     46   1.1       mrg  */
     47   1.1       mrg 
     48   1.1       mrg #include <sys/param.h>
     49   1.1       mrg #include <sys/stat.h>
     50   1.1       mrg #include <sys/time.h>
     51   1.1       mrg 
     52   1.1       mrg #include <unistd.h>
     53   1.1       mrg #include <stdio.h>
     54   1.1       mrg #include <string.h>
     55   1.1       mrg #include <stdlib.h>
     56   1.1       mrg #include <err.h>
     57   1.1       mrg #include <errno.h>
     58   1.1       mrg #include <fcntl.h>
     59   1.1       mrg #include <zlib.h>
     60   1.1       mrg #include <fts.h>
     61   1.1       mrg #include <libgen.h>
     62   1.1       mrg #include <stdarg.h>
     63   1.1       mrg #include <getopt.h>
     64   1.1       mrg 
     65  1.10       mrg /* what type of file are we dealing with */
     66  1.10       mrg enum filetype {
     67  1.10       mrg 	FT_GZIP,
     68  1.24   tsutsui #ifndef NO_BZIP2_SUPPORT
     69  1.10       mrg 	FT_BZIP2,
     70  1.24   tsutsui #endif
     71  1.24   tsutsui #ifndef NO_COMPRESS_SUPPORT
     72  1.22       mrg 	FT_Z,
     73  1.24   tsutsui #endif
     74  1.10       mrg 	FT_LAST,
     75  1.10       mrg 	FT_UNKNOWN
     76  1.10       mrg };
     77  1.10       mrg 
     78  1.24   tsutsui #ifndef NO_BZIP2_SUPPORT
     79  1.10       mrg #include <bzlib.h>
     80  1.10       mrg 
     81  1.24   tsutsui #define BZ2_SUFFIX	".bz2"
     82  1.24   tsutsui #define BZIP2_MAGIC	"\102\132\150"
     83  1.24   tsutsui #endif
     84  1.24   tsutsui 
     85  1.24   tsutsui #ifndef NO_COMPRESS_SUPPORT
     86  1.22       mrg #define Z_SUFFIX	".Z"
     87  1.22       mrg #define Z_MAGIC		"\037\235"
     88  1.24   tsutsui #endif
     89  1.10       mrg 
     90  1.10       mrg #define GZ_SUFFIX	".gz"
     91   1.1       mrg 
     92  1.37       mrg #define BUFLEN		(64 * 1024)
     93   1.1       mrg 
     94   1.9       mrg #define GZIP_MAGIC0	0x1F
     95  1.10       mrg #define GZIP_MAGIC1	0x8B
     96  1.10       mrg #define GZIP_OMAGIC1	0x9E
     97   1.9       mrg 
     98  1.48       mrg #define GZIP_TIMESTAMP	(off_t)4
     99  1.47       mrg #define GZIP_ORIGNAME	(off_t)10
    100  1.47       mrg 
    101  1.37       mrg #define HEAD_CRC	0x02
    102  1.37       mrg #define EXTRA_FIELD	0x04
    103  1.37       mrg #define ORIG_NAME	0x08
    104  1.37       mrg #define COMMENT		0x10
    105   1.1       mrg 
    106  1.37       mrg #define OS_CODE		3	/* Unix */
    107   1.1       mrg 
    108  1.56       mrg #ifndef SMALL
    109  1.56       mrg static	char	const *suffixes[] = {
    110  1.56       mrg 	GZ_SUFFIX, ".z", ".taz", ".tgz", "-gz", "-z", "_z",
    111  1.56       mrg #ifndef NO_BZIP2_SUPPORT
    112  1.56       mrg 	BZ2_SUFFIX,
    113  1.56       mrg #endif
    114  1.56       mrg #ifndef NO_COMPRESS_SUPPORT
    115  1.56       mrg 	Z_SUFFIX,
    116  1.56       mrg #endif
    117  1.56       mrg 	NULL
    118  1.56       mrg };
    119  1.56       mrg #endif /* SMALL */
    120  1.56       mrg 
    121  1.53       mrg static	const char	gzip_version[] = "NetBSD gzip 20040711";
    122   1.1       mrg 
    123   1.1       mrg static	int	cflag;			/* stdout mode */
    124   1.1       mrg static	int	dflag;			/* decompress mode */
    125  1.25       mrg static	int	lflag;			/* list mode */
    126  1.41       mrg static	int	numflag = 6;		/* gzip -1..-9 value */
    127  1.26       mrg 
    128  1.25       mrg #ifndef SMALL
    129   1.1       mrg static	int	fflag;			/* force mode */
    130   1.1       mrg static	int	nflag;			/* don't save name/timestamp */
    131   1.1       mrg static	int	Nflag;			/* don't restore name/timestamp */
    132   1.1       mrg static	int	qflag;			/* quiet mode */
    133   1.1       mrg static	int	rflag;			/* recursive mode */
    134  1.26       mrg static	int	tflag;			/* test */
    135  1.26       mrg static	char	*Sflag;
    136  1.26       mrg static	int	vflag;			/* verbose mode */
    137  1.25       mrg #else
    138  1.25       mrg #define		qflag	0
    139  1.25       mrg #endif
    140  1.26       mrg 
    141  1.46       mrg static	int	exit_value = 0;		/* exit value */
    142  1.46       mrg 
    143  1.45       agc static	const char	*suffix;
    144  1.10       mrg #define suffix_len	(strlen(suffix) + 1)	/* len + nul */
    145   1.1       mrg static	char	*infile;		/* name of file coming in */
    146   1.1       mrg 
    147  1.46       mrg static	void	maybe_err(const char *fmt, ...);
    148  1.46       mrg static	void	maybe_errx(const char *fmt, ...);
    149   1.1       mrg static	void	maybe_warn(const char *fmt, ...);
    150   1.1       mrg static	void	maybe_warnx(const char *fmt, ...);
    151  1.32       mrg static	enum filetype file_gettype(u_char *);
    152  1.37       mrg static	off_t	gz_compress(FILE *, int, off_t *, const char *, time_t);
    153  1.46       mrg static	off_t	gz_uncompress(int, int, char *, size_t, off_t *, const char *);
    154  1.45       agc static	off_t	file_compress(char *, char *, size_t);
    155  1.45       agc static	off_t	file_uncompress(char *, char *, size_t);
    156  1.51       mrg static	off_t	cat_fd(unsigned char *, size_t, off_t *, int fd);
    157   1.1       mrg static	void	handle_pathname(char *);
    158   1.1       mrg static	void	handle_file(char *, struct stat *);
    159   1.1       mrg static	void	handle_stdin(void);
    160   1.1       mrg static	void	handle_stdout(void);
    161   1.6       mrg static	void	print_ratio(off_t, off_t, FILE *);
    162   1.7       mrg static	void	print_list(int fd, off_t, const char *, time_t);
    163  1.10       mrg static	void	usage(void);
    164  1.10       mrg static	void	display_version(void);
    165  1.26       mrg 
    166  1.26       mrg #ifndef SMALL
    167  1.26       mrg static	void	prepend_gzip(char *, int *, char ***);
    168  1.26       mrg static	void	handle_dir(char *, struct stat *);
    169  1.31       mrg static	void	print_verbage(char *, char *, off_t, off_t);
    170  1.45       agc static	void	print_test(const char *, int);
    171  1.29       mrg static	void	copymodes(const char *, struct stat *);
    172  1.56       mrg static	int	check_outfile(const char *outfile, struct stat *sb);
    173  1.56       mrg static	const char *check_suffix(char *);
    174  1.26       mrg #endif
    175  1.26       mrg 
    176  1.24   tsutsui #ifndef NO_BZIP2_SUPPORT
    177  1.37       mrg static	off_t	unbzip2(int, int, char *, size_t, off_t *);
    178  1.24   tsutsui #endif
    179  1.26       mrg 
    180  1.24   tsutsui #ifndef NO_COMPRESS_SUPPORT
    181  1.37       mrg static	FILE 	*zopen(const char *, FILE *);
    182  1.37       mrg static	off_t	zuncompress(FILE *, FILE *, char *, size_t, off_t *);
    183  1.24   tsutsui #endif
    184   1.1       mrg 
    185   1.1       mrg int main(int, char *p[]);
    186   1.1       mrg 
    187  1.11       mrg #ifdef SMALL
    188  1.11       mrg #define getopt_long(a,b,c,d,e) getopt(a,b,c)
    189  1.11       mrg #else
    190   1.3  jdolecek static const struct option longopts[] = {
    191   1.1       mrg 	{ "stdout",		no_argument,		0,	'c' },
    192   1.1       mrg 	{ "to-stdout",		no_argument,		0,	'c' },
    193   1.1       mrg 	{ "decompress",		no_argument,		0,	'd' },
    194   1.1       mrg 	{ "uncompress",		no_argument,		0,	'd' },
    195   1.1       mrg 	{ "force",		no_argument,		0,	'f' },
    196   1.1       mrg 	{ "help",		no_argument,		0,	'h' },
    197   1.5       mrg 	{ "list",		no_argument,		0,	'l' },
    198   1.1       mrg 	{ "no-name",		no_argument,		0,	'n' },
    199   1.1       mrg 	{ "name",		no_argument,		0,	'N' },
    200   1.1       mrg 	{ "quiet",		no_argument,		0,	'q' },
    201   1.1       mrg 	{ "recursive",		no_argument,		0,	'r' },
    202   1.1       mrg 	{ "suffix",		required_argument,	0,	'S' },
    203   1.1       mrg 	{ "test",		no_argument,		0,	't' },
    204   1.1       mrg 	{ "verbose",		no_argument,		0,	'v' },
    205   1.1       mrg 	{ "version",		no_argument,		0,	'V' },
    206   1.1       mrg 	{ "fast",		no_argument,		0,	'1' },
    207   1.1       mrg 	{ "best",		no_argument,		0,	'9' },
    208   1.1       mrg #if 0
    209   1.1       mrg 	/*
    210  1.10       mrg 	 * This is what else GNU gzip implements.  --ascii isn't useful
    211  1.10       mrg 	 * on NetBSD, and I don't care to have a --license.
    212   1.1       mrg 	 */
    213   1.1       mrg 	{ "ascii",		no_argument,		0,	'a' },
    214   1.1       mrg 	{ "license",		no_argument,		0,	'L' },
    215   1.1       mrg #endif
    216  1.12       mrg 	{ NULL,			no_argument,		0,	0 },
    217   1.1       mrg };
    218  1.11       mrg #endif
    219   1.1       mrg 
    220   1.1       mrg int
    221   1.1       mrg main(int argc, char **argv)
    222   1.1       mrg {
    223  1.28       mrg 	const char *progname = getprogname();
    224  1.25       mrg #ifndef SMALL
    225  1.15       mrg 	char *gzip;
    226  1.25       mrg #endif
    227   1.1       mrg 	int ch;
    228   1.1       mrg 
    229  1.15       mrg 	/* XXX set up signals */
    230  1.15       mrg 
    231  1.56       mrg 	suffix = GZ_SUFFIX;
    232  1.10       mrg 
    233  1.25       mrg #ifndef SMALL
    234  1.15       mrg 	if ((gzip = getenv("GZIP")) != NULL)
    235  1.15       mrg 		prepend_gzip(gzip, &argc, &argv);
    236  1.28       mrg #endif
    237  1.15       mrg 
    238   1.1       mrg 	/*
    239   1.1       mrg 	 * XXX
    240   1.1       mrg 	 * handle being called `gunzip', `zcat' and `gzcat'
    241   1.1       mrg 	 */
    242   1.1       mrg 	if (strcmp(progname, "gunzip") == 0)
    243   1.1       mrg 		dflag = 1;
    244   1.1       mrg 	else if (strcmp(progname, "zcat") == 0 ||
    245   1.1       mrg 		 strcmp(progname, "gzcat") == 0)
    246   1.1       mrg 		dflag = cflag = 1;
    247  1.25       mrg 
    248  1.25       mrg #ifdef SMALL
    249  1.31       mrg #define OPT_LIST "cdhHltV123456789"
    250  1.25       mrg #else
    251  1.28       mrg #define OPT_LIST "cdfhHlnNqrS:tvV123456789"
    252  1.25       mrg #endif
    253   1.1       mrg 
    254  1.28       mrg 	while ((ch = getopt_long(argc, argv, OPT_LIST, longopts, NULL)) != -1)
    255   1.1       mrg 		switch (ch) {
    256   1.1       mrg 		case 'c':
    257   1.1       mrg 			cflag = 1;
    258   1.1       mrg 			break;
    259   1.1       mrg 		case 'd':
    260   1.1       mrg 			dflag = 1;
    261   1.1       mrg 			break;
    262   1.5       mrg 		case 'l':
    263   1.5       mrg 			lflag = 1;
    264   1.5       mrg 			dflag = 1;
    265   1.5       mrg 			break;
    266  1.26       mrg 		case 'V':
    267  1.26       mrg 			display_version();
    268  1.26       mrg 			/* NOTREACHED */
    269  1.26       mrg 		case '1': case '2': case '3':
    270  1.26       mrg 		case '4': case '5': case '6':
    271  1.26       mrg 		case '7': case '8': case '9':
    272  1.38       mrg 			numflag = ch - '0';
    273  1.26       mrg 			break;
    274  1.25       mrg #ifndef SMALL
    275  1.25       mrg 		case 'f':
    276  1.25       mrg 			fflag = 1;
    277  1.25       mrg 			break;
    278   1.1       mrg 		case 'n':
    279   1.1       mrg 			nflag = 1;
    280   1.1       mrg 			Nflag = 0;
    281   1.1       mrg 			break;
    282   1.1       mrg 		case 'N':
    283   1.1       mrg 			nflag = 0;
    284   1.1       mrg 			Nflag = 1;
    285   1.1       mrg 			break;
    286   1.1       mrg 		case 'q':
    287   1.1       mrg 			qflag = 1;
    288   1.1       mrg 			break;
    289   1.1       mrg 		case 'r':
    290   1.1       mrg 			rflag = 1;
    291   1.1       mrg 			break;
    292   1.1       mrg 		case 'S':
    293   1.1       mrg 			Sflag = optarg;
    294   1.1       mrg 			break;
    295   1.1       mrg 		case 't':
    296   1.1       mrg 			cflag = 1;
    297   1.1       mrg 			tflag = 1;
    298   1.1       mrg 			dflag = 1;
    299   1.1       mrg 			break;
    300   1.1       mrg 		case 'v':
    301   1.1       mrg 			vflag = 1;
    302   1.1       mrg 			break;
    303  1.26       mrg #endif
    304  1.12       mrg 		default:
    305  1.12       mrg 			usage();
    306  1.12       mrg 			/* NOTREACHED */
    307   1.1       mrg 		}
    308   1.1       mrg 	argv += optind;
    309   1.1       mrg 	argc -= optind;
    310   1.1       mrg 
    311   1.1       mrg 	if (argc == 0) {
    312   1.1       mrg 		if (dflag)	/* stdin mode */
    313   1.1       mrg 			handle_stdin();
    314   1.1       mrg 		else		/* stdout mode */
    315   1.1       mrg 			handle_stdout();
    316   1.1       mrg 	} else {
    317   1.1       mrg 		do {
    318   1.1       mrg 			handle_pathname(argv[0]);
    319   1.5       mrg 		} while (*++argv);
    320   1.1       mrg 	}
    321  1.29       mrg #ifndef SMALL
    322   1.6       mrg 	if (qflag == 0 && lflag && argc > 1)
    323   1.7       mrg 		print_list(-1, 0, "(totals)", 0);
    324  1.29       mrg #endif
    325  1.46       mrg 	exit(exit_value);
    326   1.1       mrg }
    327   1.1       mrg 
    328   1.1       mrg /* maybe print a warning */
    329   1.1       mrg void
    330   1.1       mrg maybe_warn(const char *fmt, ...)
    331   1.1       mrg {
    332   1.1       mrg 	va_list ap;
    333   1.1       mrg 
    334   1.1       mrg 	if (qflag == 0) {
    335   1.1       mrg 		va_start(ap, fmt);
    336   1.1       mrg 		vwarn(fmt, ap);
    337   1.1       mrg 		va_end(ap);
    338   1.1       mrg 	}
    339  1.46       mrg 	if (exit_value == 0)
    340  1.46       mrg 		exit_value = 1;
    341   1.1       mrg }
    342   1.1       mrg 
    343  1.46       mrg /* ... without an errno. */
    344   1.1       mrg void
    345   1.1       mrg maybe_warnx(const char *fmt, ...)
    346   1.1       mrg {
    347   1.1       mrg 	va_list ap;
    348   1.1       mrg 
    349   1.1       mrg 	if (qflag == 0) {
    350   1.1       mrg 		va_start(ap, fmt);
    351   1.1       mrg 		vwarnx(fmt, ap);
    352   1.1       mrg 		va_end(ap);
    353   1.1       mrg 	}
    354  1.46       mrg 	if (exit_value == 0)
    355  1.46       mrg 		exit_value = 1;
    356   1.1       mrg }
    357   1.1       mrg 
    358  1.46       mrg /* maybe print an error */
    359   1.1       mrg void
    360  1.46       mrg maybe_err(const char *fmt, ...)
    361   1.1       mrg {
    362   1.1       mrg 	va_list ap;
    363   1.1       mrg 
    364   1.1       mrg 	if (qflag == 0) {
    365   1.1       mrg 		va_start(ap, fmt);
    366   1.1       mrg 		vwarn(fmt, ap);
    367   1.1       mrg 		va_end(ap);
    368   1.1       mrg 	}
    369  1.46       mrg 	exit(2);
    370   1.1       mrg }
    371   1.1       mrg 
    372  1.46       mrg /* ... without an errno. */
    373   1.9       mrg void
    374  1.46       mrg maybe_errx(const char *fmt, ...)
    375   1.9       mrg {
    376   1.9       mrg 	va_list ap;
    377   1.9       mrg 
    378   1.9       mrg 	if (qflag == 0) {
    379   1.9       mrg 		va_start(ap, fmt);
    380   1.9       mrg 		vwarnx(fmt, ap);
    381   1.9       mrg 		va_end(ap);
    382   1.9       mrg 	}
    383  1.46       mrg 	exit(2);
    384   1.9       mrg }
    385   1.9       mrg 
    386  1.25       mrg #ifndef SMALL
    387  1.15       mrg /* split up $GZIP and prepend it to the argument list */
    388  1.15       mrg static void
    389  1.15       mrg prepend_gzip(char *gzip, int *argc, char ***argv)
    390  1.15       mrg {
    391  1.15       mrg 	char *s, **nargv, **ac;
    392  1.15       mrg 	int nenvarg = 0, i;
    393  1.15       mrg 
    394  1.15       mrg 	/* scan how many arguments there are */
    395  1.15       mrg 	for (s = gzip; *s; s++) {
    396  1.15       mrg 		if (*s == ' ' || *s == '\t')
    397  1.15       mrg 			continue;
    398  1.15       mrg 		nenvarg++;
    399  1.15       mrg 		for (; *s; s++)
    400  1.15       mrg 			if (*s == ' ' || *s == '\t')
    401  1.15       mrg 				break;
    402  1.45       agc 		if (*s == 0x0)
    403  1.18       mrg 			break;
    404  1.15       mrg 	}
    405  1.15       mrg 	/* punt early */
    406  1.15       mrg 	if (nenvarg == 0)
    407  1.15       mrg 		return;
    408  1.15       mrg 
    409  1.15       mrg 	*argc += nenvarg;
    410  1.15       mrg 	ac = *argv;
    411  1.15       mrg 
    412  1.17    simonb 	nargv = (char **)malloc((*argc + 1) * sizeof(char *));
    413  1.15       mrg 	if (nargv == NULL)
    414  1.46       mrg 		maybe_err("malloc");
    415  1.15       mrg 
    416  1.15       mrg 	/* stash this away */
    417  1.15       mrg 	*argv = nargv;
    418  1.15       mrg 
    419  1.15       mrg 	/* copy the program name first */
    420  1.15       mrg 	i = 0;
    421  1.15       mrg 	nargv[i++] = *(ac++);
    422  1.15       mrg 
    423  1.15       mrg 	/* take a copy of $GZIP and add it to the array */
    424  1.15       mrg 	s = strdup(gzip);
    425  1.15       mrg 	if (s == NULL)
    426  1.46       mrg 		maybe_err("strdup");
    427  1.15       mrg 	for (; *s; s++) {
    428  1.15       mrg 		if (*s == ' ' || *s == '\t')
    429  1.15       mrg 			continue;
    430  1.15       mrg 		nargv[i++] = s;
    431  1.15       mrg 		for (; *s; s++)
    432  1.15       mrg 			if (*s == ' ' || *s == '\t') {
    433  1.15       mrg 				*s = 0;
    434  1.15       mrg 				break;
    435  1.15       mrg 			}
    436  1.15       mrg 	}
    437  1.15       mrg 
    438  1.15       mrg 	/* copy the original arguments and a NULL */
    439  1.15       mrg 	while (*ac)
    440  1.15       mrg 		nargv[i++] = *(ac++);
    441  1.15       mrg 	nargv[i] = NULL;
    442  1.15       mrg }
    443  1.25       mrg #endif
    444  1.15       mrg 
    445   1.1       mrg /* compress input to output then close both files */
    446  1.37       mrg static off_t
    447  1.37       mrg gz_compress(FILE *in, int out, off_t *gsizep, const char *origname, time_t mtime)
    448   1.1       mrg {
    449  1.37       mrg 	z_stream z;
    450  1.37       mrg 	char inbuf[BUFLEN], outbuf[BUFLEN];
    451  1.37       mrg 	off_t in_tot = 0, out_tot = 0;
    452  1.37       mrg 	ssize_t in_size;
    453  1.37       mrg 	char *str;
    454  1.37       mrg 	int i, error;
    455  1.37       mrg 	uLong crc;
    456  1.37       mrg 
    457  1.37       mrg 	i = asprintf(&str, "%c%c%c%c%c%c%c%c%c%c%s",
    458  1.37       mrg 		     GZIP_MAGIC0, GZIP_MAGIC1,
    459  1.37       mrg 		     Z_DEFLATED, origname ? ORIG_NAME : 0,
    460  1.37       mrg 		     (int)mtime & 0xff,
    461  1.37       mrg 		     (int)(mtime >> 8) & 0xff,
    462  1.37       mrg 		     (int)(mtime >> 16) & 0xff,
    463  1.37       mrg 		     (int)(mtime >> 24) & 0xff,
    464  1.37       mrg 		     0, OS_CODE, origname ? origname : "");
    465  1.37       mrg 	if (i == -1)
    466  1.46       mrg 		maybe_err("asprintf");
    467  1.37       mrg 	if (origname)
    468  1.37       mrg 		i++;
    469  1.46       mrg 	if (write(out, str, i) != i) {
    470  1.46       mrg 		maybe_warn("write");
    471  1.46       mrg 		in_tot = -1;
    472  1.46       mrg 		goto out;
    473  1.46       mrg 	}
    474  1.37       mrg 	free(str);
    475  1.37       mrg 
    476  1.37       mrg 	memset(&z, 0, sizeof z);
    477  1.37       mrg 	z.next_out = outbuf;
    478  1.37       mrg 	z.avail_out = sizeof outbuf;
    479  1.38       mrg 	z.zalloc = Z_NULL;
    480  1.38       mrg 	z.zfree = Z_NULL;
    481  1.37       mrg 	z.opaque = 0;
    482  1.37       mrg 
    483  1.37       mrg 	error = deflateInit2(&z, numflag, Z_DEFLATED,
    484  1.37       mrg 			     -MAX_WBITS, 8, Z_DEFAULT_STRATEGY);
    485  1.46       mrg 	if (error != Z_OK) {
    486  1.46       mrg 		maybe_warnx("deflateInit2 failed");
    487  1.46       mrg 		in_tot = -1;
    488  1.46       mrg 		goto out;
    489  1.46       mrg 	}
    490  1.37       mrg 
    491  1.37       mrg 	crc = crc32(0L, Z_NULL, 0);
    492  1.37       mrg 	for (;;) {
    493  1.37       mrg 		if (z.avail_out == 0) {
    494  1.46       mrg 			if (write(out, outbuf, sizeof outbuf) != sizeof outbuf) {
    495  1.46       mrg 				maybe_warn("write");
    496  1.46       mrg 				in_tot = -1;
    497  1.46       mrg 				goto out;
    498  1.46       mrg 			}
    499  1.37       mrg 
    500  1.37       mrg 			out_tot += sizeof outbuf;
    501  1.37       mrg 			z.next_out = outbuf;
    502  1.37       mrg 			z.avail_out = sizeof outbuf;
    503  1.37       mrg 		}
    504  1.37       mrg 
    505  1.37       mrg 		if (z.avail_in == 0) {
    506  1.37       mrg 			in_size = fread(inbuf, 1, sizeof inbuf, in);
    507  1.46       mrg 			if (ferror(in)) {
    508  1.46       mrg 				maybe_warn("fread");
    509  1.46       mrg 				in_tot = -1;
    510  1.46       mrg 				goto out;
    511  1.46       mrg 			}
    512  1.37       mrg 			if (in_size == 0)
    513  1.37       mrg 				break;
    514  1.37       mrg 
    515  1.37       mrg 			crc = crc32(crc, (const Bytef *)inbuf, (unsigned)in_size);
    516  1.37       mrg 			in_tot += in_size;
    517  1.37       mrg 			z.next_in = inbuf;
    518  1.37       mrg 			z.avail_in = in_size;
    519  1.37       mrg 		}
    520  1.37       mrg 
    521  1.37       mrg 		error = deflate(&z, Z_NO_FLUSH);
    522  1.46       mrg 		if (error != Z_OK && error != Z_STREAM_END) {
    523  1.46       mrg 			maybe_warnx("deflate failed");
    524  1.46       mrg 			in_tot = -1;
    525  1.46       mrg 			goto out;
    526  1.46       mrg 		}
    527  1.37       mrg 	}
    528   1.1       mrg 
    529  1.37       mrg 	/* clean up */
    530   1.1       mrg 	for (;;) {
    531  1.37       mrg 		size_t len;
    532  1.37       mrg 
    533  1.37       mrg 		error = deflate(&z, Z_FINISH);
    534  1.46       mrg 		if (error != Z_OK && error != Z_STREAM_END) {
    535  1.46       mrg 			maybe_warnx("deflate failed");
    536  1.46       mrg 			in_tot = -1;
    537  1.46       mrg 			goto out;
    538  1.46       mrg 		}
    539  1.37       mrg 
    540  1.37       mrg 		len = sizeof outbuf - z.avail_out;
    541  1.37       mrg 
    542  1.46       mrg 		if (write(out, outbuf, len) != len) {
    543  1.46       mrg 			maybe_warn("write");
    544  1.46       mrg 			out_tot = -1;
    545  1.46       mrg 			goto out;
    546  1.46       mrg 		}
    547  1.37       mrg 		out_tot += len;
    548  1.37       mrg 		z.next_out = outbuf;
    549  1.37       mrg 		z.avail_out = sizeof outbuf;
    550  1.37       mrg 
    551  1.37       mrg 		if (error == Z_STREAM_END)
    552   1.1       mrg 			break;
    553  1.37       mrg 	}
    554  1.37       mrg 
    555  1.46       mrg 	if (deflateEnd(&z) != Z_OK) {
    556  1.46       mrg 		maybe_warnx("deflateEnd failed");
    557  1.46       mrg 		in_tot = -1;
    558  1.46       mrg 		goto out;
    559  1.46       mrg 	}
    560  1.37       mrg 
    561  1.37       mrg 	i = asprintf(&str, "%c%c%c%c%c%c%c%c",
    562  1.37       mrg 		 (int)crc & 0xff,
    563  1.37       mrg 		 (int)(crc >> 8) & 0xff,
    564  1.37       mrg 		 (int)(crc >> 16) & 0xff,
    565  1.37       mrg 		 (int)(crc >> 24) & 0xff,
    566  1.37       mrg 		 (int)in_tot & 0xff,
    567  1.37       mrg 		 (int)(in_tot >> 8) & 0xff,
    568  1.37       mrg 		 (int)(in_tot >> 16) & 0xff,
    569  1.37       mrg 		 (int)(in_tot >> 24) & 0xff);
    570  1.37       mrg 	if (i != 8)
    571  1.46       mrg 		maybe_err("asprintf");
    572  1.46       mrg 	if (write(out, str, i) != i) {
    573  1.46       mrg 		maybe_warn("write");
    574  1.46       mrg 		in_tot = -1;
    575  1.46       mrg 	}
    576  1.37       mrg 	free(str);
    577   1.1       mrg 
    578  1.46       mrg out:
    579  1.37       mrg 	if (gsizep)
    580  1.37       mrg 		*gsizep = out_tot;
    581  1.37       mrg 	return in_tot;
    582   1.1       mrg }
    583   1.1       mrg 
    584  1.37       mrg /*
    585  1.37       mrg  * uncompress input to output then close the input.  return the
    586  1.37       mrg  * uncompressed size written, and put the compressed sized read
    587  1.37       mrg  * into `*gsizep'.
    588  1.37       mrg  */
    589   1.1       mrg static off_t
    590  1.46       mrg gz_uncompress(int in, int out, char *pre, size_t prelen, off_t *gsizep,
    591  1.46       mrg 	      const char *filename)
    592   1.1       mrg {
    593  1.37       mrg 	z_stream z;
    594  1.37       mrg 	char outbuf[BUFLEN], inbuf[BUFLEN];
    595  1.54       mrg 	off_t out_tot, out_sub_tot, in_tot;
    596  1.37       mrg 	enum {
    597  1.37       mrg 		GZSTATE_MAGIC0,
    598  1.37       mrg 		GZSTATE_MAGIC1,
    599  1.37       mrg 		GZSTATE_METHOD,
    600  1.37       mrg 		GZSTATE_FLAGS,
    601  1.37       mrg 		GZSTATE_SKIPPING,
    602  1.37       mrg 		GZSTATE_EXTRA,
    603  1.37       mrg 		GZSTATE_EXTRA2,
    604  1.37       mrg 		GZSTATE_EXTRA3,
    605  1.37       mrg 		GZSTATE_ORIGNAME,
    606  1.37       mrg 		GZSTATE_COMMENT,
    607  1.37       mrg 		GZSTATE_HEAD_CRC1,
    608  1.37       mrg 		GZSTATE_HEAD_CRC2,
    609  1.37       mrg 		GZSTATE_INIT,
    610  1.55       mrg 		GZSTATE_READ,
    611  1.55       mrg 		GZSTATE_CRC,
    612  1.55       mrg 		GZSTATE_LEN,
    613  1.37       mrg 	} state = GZSTATE_MAGIC0;
    614  1.37       mrg 	int flags = 0, skip_count = 0;
    615  1.37       mrg 	int error, done_reading = 0;
    616  1.54       mrg 	uLong crc;
    617  1.37       mrg 
    618  1.37       mrg #define ADVANCE()       { z.next_in++; z.avail_in--; }
    619  1.37       mrg 
    620  1.37       mrg 	memset(&z, 0, sizeof z);
    621  1.37       mrg 	z.avail_in = prelen;
    622  1.37       mrg 	z.next_in = pre;
    623  1.37       mrg 	z.avail_out = sizeof outbuf;
    624  1.37       mrg 	z.next_out = outbuf;
    625  1.37       mrg 	z.zalloc = NULL;
    626  1.37       mrg 	z.zfree = NULL;
    627  1.37       mrg 	z.opaque = 0;
    628  1.37       mrg 
    629  1.37       mrg 	in_tot = prelen;
    630  1.37       mrg 	out_tot = 0;
    631   1.1       mrg 
    632  1.37       mrg 	for (;;) {
    633  1.37       mrg 		if (z.avail_in == 0 && done_reading == 0) {
    634  1.37       mrg 			size_t in_size = read(in, inbuf, BUFLEN);
    635   1.1       mrg 
    636  1.37       mrg 			if (in_size == -1) {
    637  1.26       mrg #ifndef SMALL
    638  1.46       mrg 				if (tflag && vflag)
    639  1.46       mrg 					print_test(filename, 0);
    640  1.37       mrg #endif
    641  1.51       mrg 				maybe_warn("failed to read stdin");
    642  1.46       mrg 				out_tot = -1;
    643  1.46       mrg 				goto stop;
    644  1.37       mrg 			} else if (in_size == 0)
    645  1.37       mrg 				done_reading = 1;
    646  1.37       mrg 
    647  1.37       mrg 			z.avail_in = in_size;
    648  1.37       mrg 			z.next_in = inbuf;
    649  1.37       mrg 
    650  1.37       mrg 			in_tot += in_size;
    651  1.37       mrg 		}
    652  1.54       mrg 		if (z.avail_in == 0) {
    653  1.55       mrg 			if (done_reading && state != GZSTATE_MAGIC0)
    654  1.55       mrg 				maybe_warnx("%s: unexpected end of file",
    655  1.55       mrg 					    filename);
    656  1.54       mrg 			goto stop;
    657  1.54       mrg 		}
    658  1.37       mrg 		switch (state) {
    659  1.37       mrg 		case GZSTATE_MAGIC0:
    660  1.46       mrg 			if (*z.next_in != GZIP_MAGIC0) {
    661  1.55       mrg 				maybe_warnx("input not gziped (MAGIC0)");
    662  1.46       mrg 				out_tot = -1;
    663  1.46       mrg 				goto stop;
    664  1.46       mrg 			}
    665  1.37       mrg 			ADVANCE();
    666  1.37       mrg 			state++;
    667  1.54       mrg 			out_sub_tot = 0;
    668  1.54       mrg 			crc = crc32(0L, Z_NULL, 0);
    669  1.37       mrg 			break;
    670  1.37       mrg 
    671  1.37       mrg 		case GZSTATE_MAGIC1:
    672  1.37       mrg 			if (*z.next_in != GZIP_MAGIC1 &&
    673  1.46       mrg 			    *z.next_in != GZIP_OMAGIC1) {
    674  1.55       mrg 				maybe_warnx("input not gziped (MAGIC1)");
    675  1.46       mrg 				out_tot = -1;
    676  1.46       mrg 				goto stop;
    677  1.46       mrg 			}
    678  1.37       mrg 			ADVANCE();
    679  1.37       mrg 			state++;
    680  1.37       mrg 			break;
    681  1.37       mrg 
    682  1.37       mrg 		case GZSTATE_METHOD:
    683  1.46       mrg 			if (*z.next_in != Z_DEFLATED) {
    684  1.51       mrg 				maybe_warnx("unknown compression method");
    685  1.46       mrg 				out_tot = -1;
    686  1.46       mrg 				goto stop;
    687  1.46       mrg 			}
    688  1.37       mrg 			ADVANCE();
    689  1.37       mrg 			state++;
    690  1.37       mrg 			break;
    691  1.37       mrg 
    692  1.37       mrg 		case GZSTATE_FLAGS:
    693  1.37       mrg 			flags = *z.next_in;
    694  1.37       mrg 			ADVANCE();
    695  1.37       mrg 			skip_count = 6;
    696  1.37       mrg 			state++;
    697  1.37       mrg 			break;
    698  1.37       mrg 
    699  1.37       mrg 		case GZSTATE_SKIPPING:
    700  1.37       mrg 			if (skip_count > 0) {
    701  1.37       mrg 				skip_count--;
    702  1.37       mrg 				ADVANCE();
    703  1.37       mrg 			} else
    704  1.37       mrg 				state++;
    705  1.37       mrg 			break;
    706  1.37       mrg 
    707  1.37       mrg 		case GZSTATE_EXTRA:
    708  1.37       mrg 			if ((flags & EXTRA_FIELD) == 0) {
    709  1.37       mrg 				state = GZSTATE_ORIGNAME;
    710  1.37       mrg 				break;
    711  1.37       mrg 			}
    712  1.37       mrg 			skip_count = *z.next_in;
    713  1.37       mrg 			ADVANCE();
    714  1.37       mrg 			state++;
    715  1.37       mrg 			break;
    716  1.37       mrg 
    717  1.37       mrg 		case GZSTATE_EXTRA2:
    718  1.37       mrg 			skip_count |= ((*z.next_in) << 8);
    719  1.37       mrg 			ADVANCE();
    720  1.37       mrg 			state++;
    721  1.37       mrg 			break;
    722  1.37       mrg 
    723  1.37       mrg 		case GZSTATE_EXTRA3:
    724  1.37       mrg 			if (skip_count > 0) {
    725  1.37       mrg 				skip_count--;
    726  1.37       mrg 				ADVANCE();
    727   1.1       mrg 			} else
    728  1.37       mrg 				state++;
    729  1.37       mrg 			break;
    730  1.37       mrg 
    731  1.37       mrg 		case GZSTATE_ORIGNAME:
    732  1.37       mrg 			if ((flags & ORIG_NAME) == 0) {
    733  1.37       mrg 				state++;
    734  1.37       mrg 				break;
    735  1.37       mrg 			}
    736  1.37       mrg 			if (*z.next_in == 0)
    737  1.37       mrg 				state++;
    738  1.37       mrg 			ADVANCE();
    739  1.37       mrg 			break;
    740  1.37       mrg 
    741  1.37       mrg 		case GZSTATE_COMMENT:
    742  1.37       mrg 			if ((flags & COMMENT) == 0) {
    743  1.37       mrg 				state++;
    744  1.37       mrg 				break;
    745  1.37       mrg 			}
    746  1.37       mrg 			if (*z.next_in == 0)
    747  1.37       mrg 				state++;
    748  1.37       mrg 			ADVANCE();
    749  1.37       mrg 			break;
    750  1.37       mrg 
    751  1.37       mrg 		case GZSTATE_HEAD_CRC1:
    752  1.37       mrg 			if (flags & HEAD_CRC)
    753  1.37       mrg 				skip_count = 2;
    754  1.37       mrg 			else
    755  1.37       mrg 				skip_count = 0;
    756  1.37       mrg 			state++;
    757  1.37       mrg 			break;
    758  1.37       mrg 
    759  1.37       mrg 		case GZSTATE_HEAD_CRC2:
    760  1.37       mrg 			if (skip_count > 0) {
    761  1.37       mrg 				skip_count--;
    762  1.37       mrg 				ADVANCE();
    763  1.37       mrg 			} else
    764  1.37       mrg 				state++;
    765  1.37       mrg 			break;
    766  1.37       mrg 
    767  1.37       mrg 		case GZSTATE_INIT:
    768  1.37       mrg 			if (inflateInit2(&z, -MAX_WBITS) != Z_OK) {
    769  1.51       mrg 				maybe_warnx("failed to inflateInit");
    770  1.46       mrg 				out_tot = -1;
    771  1.37       mrg 				goto stop;
    772  1.37       mrg 			}
    773  1.37       mrg 			state++;
    774  1.37       mrg 			break;
    775  1.37       mrg 
    776  1.37       mrg 		case GZSTATE_READ:
    777  1.37       mrg 			error = inflate(&z, Z_FINISH);
    778  1.50       mrg 			/* Z_BUF_ERROR goes with Z_FINISH... */
    779  1.37       mrg 			if (error == Z_STREAM_END || error == Z_BUF_ERROR) {
    780  1.57       mrg 				ssize_t wr = BUFLEN - z.avail_out;
    781  1.37       mrg 
    782  1.50       mrg 				/* Nothing left? */
    783  1.50       mrg 				if (wr == 0)
    784  1.46       mrg 					goto stop;
    785  1.50       mrg 
    786  1.57       mrg 				crc = crc32(crc, (const Bytef *)outbuf, (unsigned)wr);
    787  1.37       mrg 				if (
    788  1.37       mrg #ifndef SMALL
    789  1.37       mrg 				    /* don't write anything with -t */
    790  1.37       mrg 				    tflag == 0 &&
    791  1.37       mrg #endif
    792  1.46       mrg 				    write(out, outbuf, wr) != wr) {
    793  1.52      yamt 					maybe_warn("error writing to output");
    794  1.46       mrg 					out_tot = -1;
    795  1.46       mrg 					goto stop;
    796  1.46       mrg 				}
    797  1.39       mrg 
    798  1.39       mrg 				out_tot += wr;
    799  1.54       mrg 				out_sub_tot += wr;
    800  1.37       mrg 
    801  1.53       mrg 				if (error == Z_STREAM_END) {
    802  1.53       mrg 					inflateEnd(&z);
    803  1.55       mrg 					state++;
    804  1.53       mrg 				}
    805  1.39       mrg 
    806  1.37       mrg 				z.next_out = outbuf;
    807  1.37       mrg 				z.avail_out = BUFLEN;
    808  1.37       mrg 
    809  1.37       mrg 				break;
    810  1.37       mrg 			}
    811  1.55       mrg 		case GZSTATE_CRC:
    812  1.55       mrg 			{
    813  1.55       mrg 				static int empty_buffer = 0;
    814  1.55       mrg 				uLong origcrc;
    815  1.55       mrg 
    816  1.55       mrg 				if (z.avail_in < 4) {
    817  1.55       mrg 					if (!done_reading && empty_buffer++ < 4)
    818  1.55       mrg 						continue;
    819  1.55       mrg 					maybe_warnx("truncated input");
    820  1.57       mrg 					out_tot = -1;
    821  1.55       mrg 					goto stop;
    822  1.55       mrg 				}
    823  1.55       mrg 				empty_buffer = 0;
    824  1.57       mrg 				origcrc = ((unsigned)z.next_in[0] & 0xff) |
    825  1.57       mrg 					((unsigned)z.next_in[1] & 0xff) << 8 |
    826  1.57       mrg 					((unsigned)z.next_in[2] & 0xff) << 16 |
    827  1.57       mrg 					((unsigned)z.next_in[3] & 0xff) << 24;
    828  1.57       mrg 				if (origcrc != crc) {
    829  1.55       mrg 					maybe_warnx("invalid compressed"
    830  1.55       mrg 					     " data--crc error");
    831  1.57       mrg 					out_tot = -1;
    832  1.57       mrg 					goto stop;
    833  1.57       mrg 				}
    834  1.55       mrg 			}
    835  1.55       mrg 
    836  1.55       mrg 			z.avail_in -= 4;
    837  1.55       mrg 			z.next_in += 4;
    838  1.55       mrg 
    839  1.55       mrg 			if (!z.avail_in)
    840  1.55       mrg 				goto stop;
    841  1.55       mrg 			state++;
    842  1.55       mrg 			break;
    843  1.55       mrg 		case GZSTATE_LEN:
    844  1.55       mrg 			{
    845  1.55       mrg 				static int empty_buffer = 0;
    846  1.55       mrg 				uLong origlen;
    847  1.55       mrg 
    848  1.55       mrg 				if (z.avail_in < 4) {
    849  1.55       mrg 					if (!done_reading && empty_buffer++ < 4)
    850  1.55       mrg 						continue;
    851  1.55       mrg 					maybe_warnx("truncated input");
    852  1.57       mrg 					out_tot = -1;
    853  1.55       mrg 					goto stop;
    854  1.55       mrg 				}
    855  1.55       mrg 				empty_buffer = 0;
    856  1.57       mrg 				origlen = ((unsigned)z.next_in[0] & 0xff) |
    857  1.57       mrg 					((unsigned)z.next_in[1] & 0xff) << 8 |
    858  1.57       mrg 					((unsigned)z.next_in[2] & 0xff) << 16 |
    859  1.57       mrg 					((unsigned)z.next_in[3] & 0xff) << 24;
    860  1.55       mrg 
    861  1.57       mrg 				if (origlen != out_sub_tot) {
    862  1.55       mrg 					maybe_warnx("invalid compressed"
    863  1.55       mrg 					     " data--length error");
    864  1.57       mrg 					out_tot = -1;
    865  1.57       mrg 					goto stop;
    866  1.57       mrg 				}
    867  1.55       mrg 			}
    868  1.55       mrg 
    869  1.55       mrg 			z.avail_in -= 4;
    870  1.55       mrg 			z.next_in += 4;
    871  1.55       mrg 
    872  1.37       mrg 			if (error < 0) {
    873  1.51       mrg 				maybe_warnx("decompression error");
    874  1.37       mrg 				out_tot = -1;
    875  1.37       mrg 				goto stop;
    876  1.37       mrg 			}
    877  1.55       mrg 			state = GZSTATE_MAGIC0;
    878   1.1       mrg 			break;
    879   1.1       mrg 		}
    880  1.37       mrg 		continue;
    881  1.37       mrg stop:
    882  1.37       mrg 		break;
    883  1.37       mrg 	}
    884  1.37       mrg 	if (state > GZSTATE_INIT)
    885  1.37       mrg 		inflateEnd(&z);
    886   1.1       mrg 
    887  1.26       mrg #ifndef SMALL
    888  1.46       mrg 	if (tflag && vflag)
    889  1.46       mrg 		print_test(filename, out_tot != -1);
    890  1.26       mrg #endif
    891   1.1       mrg 
    892  1.37       mrg 	if (gsizep)
    893  1.37       mrg 		*gsizep = in_tot;
    894  1.37       mrg 	return (out_tot);
    895   1.1       mrg }
    896   1.1       mrg 
    897  1.29       mrg #ifndef SMALL
    898   1.1       mrg /*
    899   1.1       mrg  * set the owner, mode, flags & utimes for a file
    900   1.1       mrg  */
    901   1.1       mrg static void
    902   1.1       mrg copymodes(const char *file, struct stat *sbp)
    903   1.1       mrg {
    904   1.1       mrg 	struct timeval times[2];
    905   1.1       mrg 
    906   1.1       mrg 	/*
    907   1.1       mrg 	 * If we have no info on the input, give this file some
    908   1.1       mrg 	 * default values and return..
    909   1.1       mrg 	 */
    910   1.1       mrg 	if (sbp == NULL) {
    911   1.1       mrg 		mode_t mask = umask(022);
    912   1.1       mrg 
    913   1.1       mrg 		(void)chmod(file, DEFFILEMODE & ~mask);
    914   1.1       mrg 		(void)umask(mask);
    915   1.1       mrg 		return;
    916   1.1       mrg 	}
    917   1.1       mrg 
    918   1.1       mrg 	/* if the chown fails, remove set-id bits as-per compress(1) */
    919   1.1       mrg 	if (chown(file, sbp->st_uid, sbp->st_gid) < 0) {
    920   1.1       mrg 		if (errno != EPERM)
    921   1.1       mrg 			maybe_warn("couldn't chown: %s", file);
    922   1.1       mrg 		sbp->st_mode &= ~(S_ISUID|S_ISGID);
    923   1.1       mrg 	}
    924   1.1       mrg 
    925   1.1       mrg 	/* we only allow set-id and the 9 normal permission bits */
    926   1.1       mrg 	sbp->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
    927   1.1       mrg 	if (chmod(file, sbp->st_mode) < 0)
    928   1.1       mrg 		maybe_warn("couldn't chmod: %s", file);
    929   1.1       mrg 
    930   1.1       mrg 	/* only try flags if they exist already */
    931   1.1       mrg         if (sbp->st_flags != 0 && chflags(file, sbp->st_flags) < 0)
    932   1.1       mrg 		maybe_warn("couldn't chflags: %s", file);
    933   1.1       mrg 
    934   1.1       mrg 	TIMESPEC_TO_TIMEVAL(&times[0], &sbp->st_atimespec);
    935   1.1       mrg 	TIMESPEC_TO_TIMEVAL(&times[1], &sbp->st_mtimespec);
    936   1.1       mrg 	if (utimes(file, times) < 0)
    937   1.1       mrg 		maybe_warn("couldn't utimes: %s", file);
    938   1.1       mrg }
    939  1.29       mrg #endif
    940   1.1       mrg 
    941  1.32       mrg /* what sort of file is this? */
    942  1.32       mrg static enum filetype
    943  1.32       mrg file_gettype(u_char *buf)
    944  1.32       mrg {
    945  1.32       mrg 
    946  1.32       mrg 	if (buf[0] == GZIP_MAGIC0 &&
    947  1.32       mrg 	    (buf[1] == GZIP_MAGIC1 || buf[1] == GZIP_OMAGIC1))
    948  1.32       mrg 		return FT_GZIP;
    949  1.32       mrg 	else
    950  1.32       mrg #ifndef NO_BZIP2_SUPPORT
    951  1.32       mrg 	if (memcmp(buf, BZIP2_MAGIC, 3) == 0 &&
    952  1.32       mrg 	    buf[3] >= '0' && buf[3] <= '9')
    953  1.32       mrg 		return FT_BZIP2;
    954  1.32       mrg 	else
    955  1.32       mrg #endif
    956  1.32       mrg #ifndef NO_COMPRESS_SUPPORT
    957  1.32       mrg 	if (memcmp(buf, Z_MAGIC, 2) == 0)
    958  1.32       mrg 		return FT_Z;
    959  1.32       mrg 	else
    960  1.32       mrg #endif
    961  1.32       mrg 		return FT_UNKNOWN;
    962  1.32       mrg }
    963  1.32       mrg 
    964  1.46       mrg #ifndef SMALL
    965  1.46       mrg /* check the outfile is OK. */
    966  1.46       mrg static int
    967  1.46       mrg check_outfile(const char *outfile, struct stat *sb)
    968  1.46       mrg {
    969  1.46       mrg 	int ok = 1;
    970  1.46       mrg 
    971  1.59       mrg 	if (lflag == 0 && stat(outfile, sb) == 0) {
    972  1.59       mrg 		if (fflag)
    973  1.59       mrg 			unlink(outfile);
    974  1.59       mrg 		else if (isatty(STDIN_FILENO)) {
    975  1.46       mrg 			char ans[10] = { 'n', '\0' };	/* default */
    976  1.46       mrg 
    977  1.46       mrg 			fprintf(stderr, "%s already exists -- do you wish to "
    978  1.46       mrg 					"overwrite (y or n)? " , outfile);
    979  1.46       mrg 			(void)fgets(ans, sizeof(ans) - 1, stdin);
    980  1.46       mrg 			if (ans[0] != 'y' && ans[0] != 'Y') {
    981  1.46       mrg 				fprintf(stderr, "\tnot overwritting\n");
    982  1.46       mrg 				ok = 0;
    983  1.46       mrg 			} else
    984  1.46       mrg 				unlink(outfile);
    985  1.46       mrg 		} else {
    986  1.46       mrg 			maybe_warnx("%s already exists -- skipping", outfile);
    987  1.46       mrg 			ok = 0;
    988  1.46       mrg 		}
    989  1.46       mrg 	}
    990  1.46       mrg 	return ok;
    991  1.46       mrg }
    992  1.56       mrg 
    993  1.56       mrg static const char *
    994  1.56       mrg check_suffix(char *file)
    995  1.56       mrg {
    996  1.56       mrg 	char const **s;
    997  1.56       mrg 	size_t slen, len = strlen(file);
    998  1.56       mrg 
    999  1.56       mrg 	for (s = suffixes; *s; s++) {
   1000  1.56       mrg 		slen = strlen(*s);
   1001  1.57       mrg 		/* if it doesn't fit in "a.suf", don't bother */
   1002  1.57       mrg 		if (slen + 1 > len)
   1003  1.57       mrg 			continue;
   1004  1.56       mrg 		if (strcmp(*s, file + len - slen) == 0)
   1005  1.56       mrg 			return *s;
   1006  1.56       mrg 	}
   1007  1.56       mrg 	return NULL;
   1008  1.56       mrg }
   1009  1.46       mrg #endif
   1010  1.46       mrg 
   1011   1.1       mrg /*
   1012   1.1       mrg  * compress the given file: create a corresponding .gz file and remove the
   1013   1.1       mrg  * original.
   1014   1.1       mrg  */
   1015  1.31       mrg static off_t
   1016  1.45       agc file_compress(char *file, char *outfile, size_t outsize)
   1017   1.1       mrg {
   1018   1.1       mrg 	FILE *in;
   1019  1.37       mrg 	int out;
   1020   1.1       mrg 	struct stat isb, osb;
   1021  1.31       mrg 	off_t size;
   1022  1.25       mrg #ifndef SMALL
   1023   1.1       mrg 	u_int32_t mtime = 0;
   1024  1.37       mrg 	char *savename;
   1025  1.25       mrg #endif
   1026   1.1       mrg 
   1027   1.1       mrg 	if (cflag == 0) {
   1028  1.56       mrg #ifndef SMALL
   1029  1.56       mrg 		const char *suff;
   1030  1.56       mrg 
   1031  1.56       mrg 		if (fflag == 0 && (suff = check_suffix(file))) {
   1032  1.56       mrg 			maybe_warnx("%s already has %s suffix -- unchanged",
   1033  1.56       mrg 				    file, suff);
   1034  1.56       mrg 			goto lose;
   1035  1.56       mrg 		}
   1036  1.56       mrg 
   1037  1.56       mrg #endif
   1038  1.56       mrg 
   1039  1.45       agc 		(void)strncpy(outfile, file, outsize - suffix_len);
   1040  1.45       agc 		outfile[outsize - suffix_len] = '\0';
   1041  1.45       agc 		(void)strlcat(outfile, suffix, outsize);
   1042   1.1       mrg 
   1043  1.25       mrg #ifndef SMALL
   1044  1.46       mrg 		if (check_outfile(outfile, &osb) == 0)
   1045  1.46       mrg 			goto lose;
   1046  1.46       mrg 
   1047   1.1       mrg 		if (stat(file, &isb) == 0) {
   1048  1.33      tron 			if (isb.st_nlink > 1 && fflag == 0) {
   1049   1.1       mrg 				maybe_warnx("%s has %d other link%s -- "
   1050  1.33      tron 					    "skipping", file, isb.st_nlink - 1,
   1051   1.1       mrg 					    isb.st_nlink == 1 ? "" : "s");
   1052   1.1       mrg 				goto lose;
   1053   1.1       mrg 			}
   1054   1.1       mrg 			if (nflag == 0)
   1055   1.1       mrg 				mtime = (u_int32_t)isb.st_mtime;
   1056  1.34       mrg 		}
   1057  1.25       mrg #endif
   1058   1.1       mrg 	}
   1059   1.1       mrg 	in = fopen(file, "r");
   1060  1.46       mrg 	if (in == NULL) {
   1061  1.46       mrg 		maybe_warn("can't fopen %s", file);
   1062  1.46       mrg 		goto lose;
   1063  1.46       mrg 	}
   1064   1.1       mrg 
   1065  1.37       mrg #ifndef SMALL
   1066  1.43    bouyer 	if (nflag == 0)
   1067  1.43    bouyer 		savename = basename(file);
   1068  1.43    bouyer 	else
   1069  1.43    bouyer 		savename = NULL;
   1070   1.1       mrg #endif
   1071  1.43    bouyer 	if (cflag == 0) {
   1072  1.37       mrg 		out = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
   1073  1.39       mrg 		if (out == -1) {
   1074  1.39       mrg 			maybe_warn("could not create output: %s", outfile);
   1075  1.39       mrg 			goto lose;
   1076  1.39       mrg 		}
   1077   1.1       mrg 	} else
   1078  1.37       mrg 		out = STDOUT_FILENO;
   1079   1.1       mrg 
   1080  1.37       mrg #ifdef SMALL
   1081  1.37       mrg 	gz_compress(in, out, NULL, NULL, 0);
   1082  1.37       mrg #else
   1083  1.37       mrg 	gz_compress(in, out, NULL, savename, mtime);
   1084  1.37       mrg #endif
   1085   1.1       mrg 
   1086  1.42       mrg 	(void)fclose(in);
   1087  1.42       mrg 
   1088   1.1       mrg 	/*
   1089   1.1       mrg 	 * if we compressed to stdout, we don't know the size and
   1090   1.1       mrg 	 * we don't know the new file name, punt.  if we can't stat
   1091   1.1       mrg 	 * the file, whine, otherwise set the size from the stat
   1092   1.1       mrg 	 * buffer.  we only blow away the file if we can stat the
   1093   1.1       mrg 	 * output, just in case.
   1094   1.1       mrg 	 */
   1095   1.1       mrg 	if (cflag == 0) {
   1096  1.39       mrg 		if (close(out) == -1)
   1097  1.39       mrg 			maybe_warn("couldn't close ouput");
   1098  1.39       mrg 
   1099   1.1       mrg 		if (stat(outfile, &osb) < 0) {
   1100   1.1       mrg 			maybe_warn("couldn't stat: %s", outfile);
   1101   1.1       mrg 			maybe_warnx("leaving original %s", file);
   1102   1.1       mrg 			size = 0;
   1103   1.1       mrg 		} else {
   1104   1.1       mrg 			unlink(file);
   1105   1.1       mrg 			size = osb.st_size;
   1106   1.1       mrg 		}
   1107  1.29       mrg #ifndef SMALL
   1108   1.1       mrg 		copymodes(outfile, &isb);
   1109  1.29       mrg #endif
   1110   1.1       mrg 	} else {
   1111   1.1       mrg lose:
   1112  1.46       mrg 		size = -1;
   1113   1.1       mrg 	}
   1114   1.1       mrg 
   1115   1.1       mrg 	return (size);
   1116   1.1       mrg }
   1117   1.1       mrg 
   1118   1.1       mrg /* uncompress the given file and remove the original */
   1119  1.31       mrg static off_t
   1120  1.45       agc file_uncompress(char *file, char *outfile, size_t outsize)
   1121   1.1       mrg {
   1122   1.1       mrg 	struct stat isb, osb;
   1123  1.45       agc 	char *s;
   1124   1.1       mrg 	off_t size;
   1125  1.57       mrg 	ssize_t rbytes, len = strlen(file);
   1126  1.37       mrg 	unsigned char header1[4], name[PATH_MAX + 1];
   1127  1.10       mrg 	enum filetype method;
   1128  1.48       mrg 	int fd, zfd;
   1129  1.48       mrg #ifndef SMALL
   1130  1.48       mrg 	time_t timestamp = 0;
   1131  1.48       mrg #endif
   1132   1.9       mrg 
   1133   1.9       mrg 	/* gather the old name info */
   1134   1.1       mrg 
   1135   1.9       mrg 	fd = open(file, O_RDONLY);
   1136  1.46       mrg 	if (fd < 0) {
   1137  1.46       mrg 		maybe_warn("can't open %s", file);
   1138  1.46       mrg 		goto lose;
   1139  1.46       mrg 	}
   1140  1.57       mrg 	rbytes = read(fd, header1, sizeof header1);
   1141  1.57       mrg 	if (rbytes != sizeof header1) {
   1142  1.10       mrg 		/* we don't want to fail here. */
   1143  1.25       mrg #ifndef SMALL
   1144  1.10       mrg 		if (fflag)
   1145  1.42       mrg 			goto lose_close_it;
   1146  1.25       mrg #endif
   1147  1.57       mrg 		if (rbytes == -1)
   1148  1.57       mrg 			maybe_warn("can't read %s", file);
   1149  1.57       mrg 		else
   1150  1.57       mrg 			maybe_warnx("%s: unexpected end of file", file);
   1151  1.58      yamt 		goto lose_close_it;
   1152  1.10       mrg 	}
   1153  1.10       mrg 
   1154  1.32       mrg 	method = file_gettype(header1);
   1155  1.29       mrg 
   1156  1.32       mrg #ifndef SMALL
   1157  1.32       mrg 	if (Sflag == NULL) {
   1158  1.32       mrg # ifndef NO_BZIP2_SUPPORT
   1159  1.32       mrg 		if (method == FT_BZIP2)
   1160  1.10       mrg 			suffix = BZ2_SUFFIX;
   1161  1.32       mrg 		else
   1162  1.26       mrg # endif
   1163  1.32       mrg # ifndef NO_COMPRESS_SUPPORT
   1164  1.32       mrg 		if (method == FT_Z)
   1165  1.22       mrg 			suffix = Z_SUFFIX;
   1166  1.26       mrg # endif
   1167  1.32       mrg 	}
   1168   1.9       mrg 
   1169  1.42       mrg 	if (fflag == 0 && method == FT_UNKNOWN) {
   1170  1.42       mrg 		maybe_warnx("%s: not in gzip format", file);
   1171  1.42       mrg 		goto lose_close_it;
   1172  1.42       mrg 	}
   1173  1.25       mrg #endif
   1174   1.9       mrg 
   1175  1.10       mrg 	if (cflag == 0 || lflag) {
   1176  1.57       mrg 		s = 0;
   1177  1.57       mrg 		if (len - suffix_len + 1 > 0 &&
   1178  1.57       mrg 		    (s = &file[len - suffix_len + 1]) &&
   1179  1.57       mrg 		    strncmp(s, suffix, suffix_len) == 0) {
   1180  1.10       mrg 			(void)strncpy(outfile, file, len - suffix_len + 1);
   1181  1.10       mrg 			outfile[len - suffix_len + 1] = '\0';
   1182  1.42       mrg 		} else if (lflag == 0) {
   1183  1.57       mrg 			maybe_warnx("%s: unknown suffix -- ignored", file);
   1184  1.42       mrg 			goto lose_close_it;
   1185  1.42       mrg 		}
   1186  1.10       mrg 	}
   1187  1.10       mrg 
   1188  1.25       mrg #ifdef SMALL
   1189  1.25       mrg 	if (method == FT_GZIP && lflag)
   1190  1.25       mrg #else
   1191  1.25       mrg 	if (method == FT_GZIP && (Nflag || lflag))
   1192  1.25       mrg #endif
   1193  1.25       mrg 	{
   1194  1.48       mrg #ifndef SMALL
   1195  1.48       mrg 		unsigned char header2[4];	/* timestamp */
   1196  1.48       mrg 
   1197  1.48       mrg 		if (lseek(fd, GZIP_TIMESTAMP, SEEK_SET) == -1) {
   1198  1.48       mrg 			maybe_warn("can't lseek %s", file);
   1199  1.48       mrg 			goto close_header_read;
   1200  1.48       mrg 		}
   1201  1.48       mrg 		if (read(fd, header2, sizeof header2) != sizeof header2) {
   1202  1.48       mrg 			if (fflag)
   1203  1.48       mrg 				goto lose_close_it;
   1204  1.48       mrg 			maybe_warn("can't read %s", file);
   1205  1.58      yamt 			goto lose_close_it;
   1206  1.48       mrg                }
   1207  1.48       mrg                timestamp = ((time_t)header2[3] << 24)
   1208  1.48       mrg                          + ((time_t)header2[2] << 16)
   1209  1.48       mrg                          + ((time_t)header2[1] << 8)
   1210  1.48       mrg                          +  (time_t)header2[0];
   1211  1.48       mrg #endif
   1212  1.48       mrg 
   1213   1.9       mrg 		if (header1[3] & ORIG_NAME) {
   1214   1.9       mrg 			int i;
   1215   1.9       mrg 
   1216  1.47       mrg 			if (lseek(fd, GZIP_ORIGNAME, SEEK_SET) == -1) {
   1217  1.47       mrg 				maybe_warn("can't lseek %s", file);
   1218  1.47       mrg 				goto close_header_read;
   1219  1.47       mrg 			}
   1220   1.9       mrg 			rbytes = read(fd, name, PATH_MAX + 1);
   1221  1.42       mrg 			if (rbytes < 0) {
   1222  1.42       mrg 				maybe_warn("can't read %s", file);
   1223  1.42       mrg 				goto lose_close_it;
   1224  1.42       mrg 			}
   1225   1.9       mrg 			for (i = 0; i < rbytes && name[i]; i++)
   1226   1.9       mrg 				;
   1227   1.9       mrg 			if (i < rbytes) {
   1228   1.9       mrg 				name[i] = 0;
   1229   1.9       mrg 				/* now maybe merge old dirname */
   1230  1.45       agc 				if (strchr(outfile, '/') == NULL)
   1231  1.45       agc 					(void) strlcpy(outfile, name, outsize);
   1232   1.9       mrg 				else {
   1233  1.45       agc 					char	newbuf[PATH_MAX + 1];
   1234  1.45       agc 
   1235  1.45       agc 					(void) snprintf(newbuf, sizeof(newbuf),
   1236  1.45       agc 						"%s/%s", dirname(outfile),
   1237  1.45       agc 						name);
   1238  1.45       agc 					(void) strlcpy(outfile, newbuf,
   1239  1.45       agc 						outsize);
   1240   1.1       mrg 				}
   1241   1.1       mrg 			}
   1242   1.1       mrg 		}
   1243   1.9       mrg 	}
   1244  1.47       mrg close_header_read:
   1245   1.9       mrg 	close(fd);
   1246   1.1       mrg 
   1247  1.31       mrg 	if (cflag == 0 || lflag) {
   1248  1.25       mrg #ifndef SMALL
   1249  1.46       mrg 		if (check_outfile(outfile, &osb) == 0)
   1250   1.9       mrg 			goto lose;
   1251  1.46       mrg 
   1252  1.31       mrg #endif
   1253   1.9       mrg 		if (stat(file, &isb) == 0) {
   1254  1.31       mrg #ifndef SMALL
   1255  1.33      tron 			if (isb.st_nlink > 1 && lflag == 0 && fflag == 0) {
   1256   1.9       mrg 				maybe_warnx("%s has %d other links -- skipping",
   1257   1.9       mrg 				    file, isb.st_nlink - 1);
   1258   1.1       mrg 				goto lose;
   1259   1.1       mrg 			}
   1260  1.48       mrg 			if (nflag == 0 && timestamp)
   1261  1.48       mrg 				isb.st_mtime = timestamp;
   1262  1.31       mrg #endif
   1263   1.9       mrg 		} else
   1264   1.9       mrg 			goto lose;
   1265   1.1       mrg 	}
   1266   1.5       mrg 
   1267  1.48       mrg 	if (cflag == 0 && lflag == 0) {
   1268  1.46       mrg 		zfd = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0600);
   1269  1.46       mrg 		if (zfd == -1) {
   1270  1.46       mrg 			maybe_warn("can't open %s", outfile);
   1271  1.46       mrg 			goto lose;
   1272  1.46       mrg 		}
   1273  1.46       mrg 	} else
   1274  1.46       mrg 		zfd = STDOUT_FILENO;
   1275  1.46       mrg 
   1276  1.24   tsutsui #ifndef NO_BZIP2_SUPPORT
   1277  1.10       mrg 	if (method == FT_BZIP2) {
   1278  1.46       mrg 		int in;
   1279   1.5       mrg 
   1280  1.30       mrg 		/* XXX */
   1281  1.46       mrg 		if (lflag) {
   1282  1.46       mrg 			maybe_warnx("no -l with bzip2 files");
   1283  1.46       mrg 			goto lose;
   1284  1.46       mrg 		}
   1285  1.26       mrg 
   1286  1.46       mrg 		if ((in = open(file, O_RDONLY)) == -1) {
   1287  1.46       mrg 			maybe_warn("open for read: %s", file);
   1288  1.42       mrg 			goto lose;
   1289  1.42       mrg 		}
   1290   1.5       mrg 
   1291  1.46       mrg 		size = unbzip2(in, zfd, NULL, 0, NULL);
   1292  1.46       mrg 		if (size == -1) {
   1293  1.36       mrg 			if (cflag == 0)
   1294  1.36       mrg 				unlink(outfile);
   1295  1.46       mrg 			maybe_warnx("%s: uncompress failed", file);
   1296  1.10       mrg 			goto lose;
   1297  1.10       mrg 		}
   1298  1.42       mrg 		if (close(in) != 0)
   1299  1.42       mrg 			maybe_warn("couldn't close input");
   1300  1.46       mrg 		if (cflag == 0 && close(zfd) != 0)
   1301  1.42       mrg 			maybe_warn("couldn't close output");
   1302  1.24   tsutsui 	} else
   1303  1.24   tsutsui #endif
   1304  1.29       mrg 
   1305  1.24   tsutsui #ifndef NO_COMPRESS_SUPPORT
   1306  1.24   tsutsui 	if (method == FT_Z) {
   1307  1.22       mrg 		FILE *in, *out;
   1308  1.22       mrg 
   1309  1.30       mrg 		/* XXX */
   1310  1.46       mrg 		if (lflag) {
   1311  1.46       mrg 			maybe_warnx("no -l with Lempel-Ziv files");
   1312  1.46       mrg 			goto lose;
   1313  1.46       mrg 		}
   1314  1.26       mrg 
   1315  1.46       mrg 		if ((in = zopen(file, NULL)) == NULL) {
   1316  1.46       mrg 			maybe_warn("open for read: %s", file);
   1317  1.46       mrg 			goto lose;
   1318  1.46       mrg 		}
   1319  1.22       mrg 
   1320  1.46       mrg 		out = fdopen(zfd, "w");
   1321  1.46       mrg 		if (out == NULL) {
   1322  1.46       mrg 			maybe_warn("open for write: %s", outfile);
   1323  1.46       mrg 			goto lose;
   1324  1.22       mrg 		}
   1325  1.22       mrg 
   1326  1.37       mrg 		size = zuncompress(in, out, NULL, 0, NULL);
   1327  1.42       mrg 		if (ferror(in) || fclose(in) != 0) {
   1328  1.42       mrg 			unlink(outfile);
   1329  1.46       mrg 			(void)fclose(out);
   1330  1.46       mrg 			maybe_warn("failed infile fclose");
   1331  1.42       mrg 		}
   1332  1.36       mrg 		if (cflag == 0) {
   1333  1.46       mrg 			if (size == -1) {
   1334  1.46       mrg 				maybe_warnx("%s: uncompress failed", file);
   1335  1.46       mrg 				(void)fclose(out);
   1336  1.36       mrg 				unlink(outfile);
   1337  1.36       mrg 				goto lose;
   1338  1.36       mrg 			}
   1339  1.42       mrg 			if (fclose(out) != 0) {
   1340  1.36       mrg 				unlink(outfile);
   1341  1.46       mrg 				maybe_warn("failed outfile close");
   1342  1.46       mrg 				goto lose;
   1343  1.36       mrg 			}
   1344  1.22       mrg 		}
   1345  1.24   tsutsui 	} else
   1346  1.24   tsutsui #endif
   1347  1.51       mrg 
   1348  1.51       mrg #ifndef SMALL
   1349  1.51       mrg 	if (method == FT_UNKNOWN) {
   1350  1.51       mrg 		int in;
   1351  1.51       mrg 
   1352  1.51       mrg 		in = open(file, O_RDONLY);
   1353  1.51       mrg 		if (in == -1) {
   1354  1.51       mrg 			maybe_warn("can't open %s", file);
   1355  1.51       mrg 			goto lose;
   1356  1.51       mrg 		}
   1357  1.51       mrg 		size = cat_fd(NULL, 0, NULL, in);
   1358  1.51       mrg 	} else
   1359  1.51       mrg #endif
   1360  1.24   tsutsui 	{
   1361  1.46       mrg 		int in;
   1362  1.37       mrg 
   1363  1.10       mrg 		if (lflag) {
   1364  1.46       mrg 			if ((zfd = open(file, O_RDONLY)) == -1) {
   1365  1.46       mrg 				maybe_warn("open: %s", file);
   1366  1.46       mrg 				goto lose;
   1367  1.46       mrg 			}
   1368  1.45       agc 			print_list(zfd, isb.st_size, outfile, isb.st_mtime);
   1369  1.61      yamt 			return -1;	/* XXX */
   1370  1.10       mrg 		}
   1371  1.10       mrg 
   1372  1.37       mrg 		in = open(file, O_RDONLY);
   1373  1.46       mrg 		if (in == -1) {
   1374  1.46       mrg 			maybe_warn("can't open %s", file);
   1375  1.46       mrg 			goto lose;
   1376  1.46       mrg 		}
   1377   1.1       mrg 
   1378  1.46       mrg 		size = gz_uncompress(in, zfd, NULL, 0, NULL, file);
   1379  1.42       mrg 		(void)close(in);
   1380  1.36       mrg 		if (cflag == 0) {
   1381  1.45       agc 			if (close(zfd))
   1382  1.42       mrg 				maybe_warn("failed close");
   1383  1.37       mrg 			if (size == -1) {
   1384  1.46       mrg 				maybe_warnx("%s: uncompress failed", file);
   1385  1.36       mrg 				unlink(outfile);
   1386  1.36       mrg 				goto lose;
   1387  1.36       mrg 			}
   1388  1.10       mrg 		}
   1389  1.10       mrg 	}
   1390   1.1       mrg 
   1391   1.1       mrg 	/* if testing, or we uncompressed to stdout, this is all we need */
   1392  1.26       mrg #ifndef SMALL
   1393  1.26       mrg 	if (tflag)
   1394  1.26       mrg 		return (size);
   1395  1.26       mrg #endif
   1396  1.26       mrg 	if (cflag)
   1397   1.1       mrg 		return (size);
   1398   1.1       mrg 
   1399   1.1       mrg 	/*
   1400   1.1       mrg 	 * if we create a file...
   1401   1.1       mrg 	 */
   1402   1.1       mrg 	if (cflag == 0) {
   1403   1.1       mrg 		/*
   1404   1.1       mrg 		 * if we can't stat the file, or we are uncompressing to
   1405   1.1       mrg 		 * stdin, don't remove the file.
   1406   1.1       mrg 		 */
   1407   1.1       mrg 		if (stat(outfile, &osb) < 0) {
   1408   1.1       mrg 			maybe_warn("couldn't stat (leaving original): %s",
   1409   1.1       mrg 				   outfile);
   1410   1.1       mrg 			goto lose;
   1411   1.1       mrg 		}
   1412   1.1       mrg 		if (osb.st_size != size) {
   1413   1.1       mrg 			maybe_warn("stat gave different size: %llu != %llu "
   1414   1.1       mrg 			    "(leaving original)",
   1415   1.1       mrg 			    (unsigned long long)size,
   1416   1.1       mrg 			    (unsigned long long)osb.st_size);
   1417   1.1       mrg 			goto lose;
   1418   1.1       mrg 		}
   1419  1.36       mrg 		if (cflag == 0)
   1420  1.36       mrg 			unlink(file);
   1421   1.1       mrg 		size = osb.st_size;
   1422  1.29       mrg #ifndef SMALL
   1423   1.1       mrg 		copymodes(outfile, &isb);
   1424  1.29       mrg #endif
   1425   1.1       mrg 	}
   1426   1.1       mrg 	return (size);
   1427   1.1       mrg 
   1428  1.42       mrg lose_close_it:
   1429  1.42       mrg 	close(fd);
   1430   1.1       mrg lose:
   1431  1.46       mrg 	return -1;
   1432   1.1       mrg }
   1433   1.1       mrg 
   1434  1.37       mrg #ifndef SMALL
   1435  1.37       mrg static off_t
   1436  1.51       mrg cat_fd(unsigned char * prepend, size_t count, off_t *gsizep, int fd)
   1437  1.37       mrg {
   1438  1.37       mrg 	char buf[BUFLEN];
   1439  1.37       mrg 	size_t rv;
   1440  1.37       mrg 	off_t in_tot;
   1441  1.37       mrg 
   1442  1.37       mrg 	in_tot = count;
   1443  1.46       mrg 	if (write(STDOUT_FILENO, prepend, count) != count) {
   1444  1.46       mrg 		maybe_warn("write to stdout");
   1445  1.46       mrg 		return -1;
   1446  1.46       mrg 	}
   1447  1.37       mrg 	for (;;) {
   1448  1.51       mrg 		rv = read(fd, buf, sizeof buf);
   1449  1.51       mrg 		if (rv == 0)
   1450  1.51       mrg 			break;
   1451  1.51       mrg 		if (rv < 0) {
   1452  1.51       mrg 			maybe_warn("read from fd %d", fd);
   1453  1.51       mrg 			break;
   1454  1.51       mrg 		}
   1455  1.37       mrg 
   1456  1.46       mrg 		if (write(STDOUT_FILENO, buf, rv) != rv) {
   1457  1.46       mrg 			maybe_warn("write to stdout");
   1458  1.51       mrg 			break;
   1459  1.46       mrg 		}
   1460  1.37       mrg 		in_tot += rv;
   1461  1.37       mrg 	}
   1462  1.37       mrg 
   1463  1.37       mrg 	if (gsizep)
   1464  1.37       mrg 		*gsizep = in_tot;
   1465  1.37       mrg 	return (in_tot);
   1466  1.37       mrg }
   1467  1.37       mrg #endif
   1468  1.37       mrg 
   1469   1.1       mrg static void
   1470   1.1       mrg handle_stdin(void)
   1471   1.1       mrg {
   1472  1.37       mrg 	unsigned char header1[4];
   1473  1.37       mrg 	off_t usize, gsize;
   1474  1.37       mrg 	enum filetype method;
   1475  1.37       mrg #ifndef NO_COMPRESS_SUPPORT
   1476  1.37       mrg 	FILE *in;
   1477  1.37       mrg #endif
   1478   1.1       mrg 
   1479  1.25       mrg #ifndef SMALL
   1480   1.5       mrg 	if (fflag == 0 && lflag == 0 && isatty(STDIN_FILENO)) {
   1481   1.1       mrg 		maybe_warnx("standard input is a terminal -- ignoring");
   1482   1.1       mrg 		return;
   1483   1.1       mrg 	}
   1484  1.25       mrg #endif
   1485   1.5       mrg 
   1486   1.5       mrg 	if (lflag) {
   1487   1.5       mrg 		struct stat isb;
   1488   1.5       mrg 
   1489  1.37       mrg 		/* XXX could read the whole file, etc. */
   1490  1.46       mrg 		if (fstat(STDIN_FILENO, &isb) < 0) {
   1491  1.46       mrg 			maybe_warn("fstat");
   1492  1.46       mrg 			return;
   1493  1.46       mrg 		}
   1494   1.7       mrg 		print_list(STDIN_FILENO, isb.st_size, "stdout", isb.st_mtime);
   1495   1.5       mrg 		return;
   1496   1.5       mrg 	}
   1497   1.5       mrg 
   1498  1.46       mrg 	if (read(STDIN_FILENO, header1, sizeof header1) != sizeof header1) {
   1499  1.46       mrg 		maybe_warn("can't read stdin");
   1500  1.46       mrg 		return;
   1501  1.46       mrg 	}
   1502  1.37       mrg 
   1503  1.37       mrg 	method = file_gettype(header1);
   1504  1.37       mrg 	switch (method) {
   1505  1.37       mrg 	default:
   1506  1.37       mrg #ifndef SMALL
   1507  1.46       mrg 		if (fflag == 0) {
   1508  1.46       mrg 			maybe_warnx("unknown compression format");
   1509  1.46       mrg 			return;
   1510  1.46       mrg 		}
   1511  1.51       mrg 		usize = cat_fd(header1, sizeof header1, &gsize, STDIN_FILENO);
   1512  1.37       mrg 		break;
   1513  1.37       mrg #endif
   1514  1.37       mrg 	case FT_GZIP:
   1515  1.37       mrg 		usize = gz_uncompress(STDIN_FILENO, STDOUT_FILENO,
   1516  1.46       mrg 			      header1, sizeof header1, &gsize, "(stdin)");
   1517  1.37       mrg 		break;
   1518  1.37       mrg #ifndef NO_BZIP2_SUPPORT
   1519  1.37       mrg 	case FT_BZIP2:
   1520  1.37       mrg 		usize = unbzip2(STDIN_FILENO, STDOUT_FILENO,
   1521  1.37       mrg 				header1, sizeof header1, &gsize);
   1522  1.37       mrg 		break;
   1523  1.37       mrg #endif
   1524  1.37       mrg #ifndef NO_COMPRESS_SUPPORT
   1525  1.37       mrg 	case FT_Z:
   1526  1.46       mrg 		if ((in = zopen(NULL, stdin)) == NULL) {
   1527  1.46       mrg 			maybe_warnx("zopen of stdin");
   1528  1.46       mrg 			return;
   1529  1.46       mrg 		}
   1530  1.37       mrg 
   1531  1.37       mrg 		usize = zuncompress(in, stdout, header1, sizeof header1, &gsize);
   1532  1.37       mrg 		break;
   1533  1.37       mrg #endif
   1534  1.37       mrg 	}
   1535  1.37       mrg 
   1536  1.37       mrg #ifndef SMALL
   1537  1.37       mrg         if (vflag && !tflag && usize != -1 && gsize != -1)
   1538  1.37       mrg 		print_verbage(NULL, 0, usize, gsize);
   1539  1.37       mrg #endif
   1540  1.37       mrg 
   1541   1.1       mrg }
   1542   1.1       mrg 
   1543   1.1       mrg static void
   1544   1.1       mrg handle_stdout(void)
   1545   1.1       mrg {
   1546  1.37       mrg 	off_t gsize, usize;
   1547   1.1       mrg 
   1548  1.25       mrg #ifndef SMALL
   1549   1.1       mrg 	if (fflag == 0 && isatty(STDOUT_FILENO)) {
   1550   1.1       mrg 		maybe_warnx("standard output is a terminal -- ignoring");
   1551   1.1       mrg 		return;
   1552   1.1       mrg 	}
   1553  1.25       mrg #endif
   1554  1.37       mrg 	usize = gz_compress(stdin, STDOUT_FILENO, &gsize, NULL, 0);
   1555  1.37       mrg 
   1556  1.37       mrg #ifndef SMALL
   1557  1.37       mrg         if (vflag && !tflag && usize != -1 && gsize != -1)
   1558  1.37       mrg 		print_verbage(NULL, 0, usize, gsize);
   1559  1.37       mrg #endif
   1560   1.1       mrg }
   1561   1.1       mrg 
   1562   1.1       mrg /* do what is asked for, for the path name */
   1563   1.1       mrg static void
   1564   1.1       mrg handle_pathname(char *path)
   1565   1.1       mrg {
   1566  1.44       agc 	char *opath = path, *s = NULL;
   1567   1.1       mrg 	ssize_t len;
   1568   1.1       mrg 	struct stat sb;
   1569   1.1       mrg 
   1570   1.1       mrg 	/* check for stdout/stdin */
   1571   1.1       mrg 	if (path[0] == '-' && path[1] == '\0') {
   1572   1.1       mrg 		if (dflag)
   1573   1.1       mrg 			handle_stdin();
   1574   1.1       mrg 		else
   1575   1.1       mrg 			handle_stdout();
   1576  1.49       mrg 		return;
   1577   1.1       mrg 	}
   1578   1.1       mrg 
   1579   1.1       mrg retry:
   1580   1.1       mrg 	if (stat(path, &sb) < 0) {
   1581   1.1       mrg 		/* lets try <path>.gz if we're decompressing */
   1582  1.44       agc 		if (dflag && s == NULL && errno == ENOENT) {
   1583   1.1       mrg 			len = strlen(path);
   1584  1.44       agc 			s = malloc(len + suffix_len + 1);
   1585  1.44       agc 			if (s == NULL)
   1586  1.46       mrg 				maybe_err("malloc");
   1587   1.1       mrg 			memmove(s, path, len);
   1588  1.10       mrg 			memmove(&s[len], suffix, suffix_len);
   1589  1.44       agc 			s[len + suffix_len] = 0x0;
   1590   1.1       mrg 			path = s;
   1591   1.1       mrg 			goto retry;
   1592   1.1       mrg 		}
   1593   1.1       mrg 		maybe_warn("can't stat: %s", opath);
   1594   1.1       mrg 		goto out;
   1595   1.1       mrg 	}
   1596   1.1       mrg 
   1597   1.1       mrg 	if (S_ISDIR(sb.st_mode)) {
   1598  1.25       mrg #ifndef SMALL
   1599   1.1       mrg 		if (rflag)
   1600   1.1       mrg 			handle_dir(path, &sb);
   1601   1.1       mrg 		else
   1602  1.25       mrg #endif
   1603  1.60      yamt 			maybe_warnx("%s is a directory", path);
   1604   1.1       mrg 		goto out;
   1605   1.1       mrg 	}
   1606   1.1       mrg 
   1607   1.1       mrg 	if (S_ISREG(sb.st_mode))
   1608   1.1       mrg 		handle_file(path, &sb);
   1609   1.1       mrg 
   1610   1.1       mrg out:
   1611   1.1       mrg 	if (s)
   1612   1.1       mrg 		free(s);
   1613   1.1       mrg }
   1614   1.1       mrg 
   1615   1.1       mrg /* compress/decompress a file */
   1616   1.1       mrg static void
   1617   1.1       mrg handle_file(char *file, struct stat *sbp)
   1618   1.1       mrg {
   1619  1.31       mrg 	off_t usize, gsize;
   1620  1.45       agc 	char	outfile[PATH_MAX];
   1621   1.1       mrg 
   1622   1.1       mrg 	infile = file;
   1623   1.1       mrg 	if (dflag) {
   1624  1.45       agc 		usize = file_uncompress(file, outfile, sizeof(outfile));
   1625  1.46       mrg 		if (usize == -1)
   1626   1.1       mrg 			return;
   1627   1.1       mrg 		gsize = sbp->st_size;
   1628   1.1       mrg 	} else {
   1629  1.45       agc 		gsize = file_compress(file, outfile, sizeof(outfile));
   1630  1.46       mrg 		if (gsize == -1)
   1631   1.1       mrg 			return;
   1632   1.1       mrg 		usize = sbp->st_size;
   1633   1.1       mrg 	}
   1634   1.1       mrg 
   1635  1.26       mrg 
   1636  1.26       mrg #ifndef SMALL
   1637   1.1       mrg 	if (vflag && !tflag)
   1638  1.45       agc 		print_verbage(file, (cflag) ? NULL : outfile, usize, gsize);
   1639  1.26       mrg #endif
   1640   1.1       mrg }
   1641   1.1       mrg 
   1642  1.25       mrg #ifndef SMALL
   1643   1.1       mrg /* this is used with -r to recursively decend directories */
   1644   1.1       mrg static void
   1645   1.1       mrg handle_dir(char *dir, struct stat *sbp)
   1646   1.1       mrg {
   1647   1.1       mrg 	char *path_argv[2];
   1648   1.1       mrg 	FTS *fts;
   1649   1.1       mrg 	FTSENT *entry;
   1650   1.1       mrg 
   1651   1.1       mrg 	path_argv[0] = dir;
   1652   1.1       mrg 	path_argv[1] = 0;
   1653   1.1       mrg 	fts = fts_open(path_argv, FTS_PHYSICAL, NULL);
   1654   1.1       mrg 	if (fts == NULL) {
   1655   1.1       mrg 		warn("couldn't fts_open %s", dir);
   1656   1.1       mrg 		return;
   1657   1.1       mrg 	}
   1658   1.1       mrg 
   1659   1.1       mrg 	while ((entry = fts_read(fts))) {
   1660   1.1       mrg 		switch(entry->fts_info) {
   1661   1.1       mrg 		case FTS_D:
   1662   1.1       mrg 		case FTS_DP:
   1663   1.1       mrg 			continue;
   1664   1.1       mrg 
   1665   1.1       mrg 		case FTS_DNR:
   1666   1.1       mrg 		case FTS_ERR:
   1667   1.1       mrg 		case FTS_NS:
   1668   1.1       mrg 			maybe_warn("%s", entry->fts_path);
   1669   1.1       mrg 			continue;
   1670   1.1       mrg 		case FTS_F:
   1671   1.1       mrg 			handle_file(entry->fts_name, entry->fts_statp);
   1672   1.1       mrg 		}
   1673   1.1       mrg 	}
   1674   1.1       mrg 	(void)fts_close(fts);
   1675   1.1       mrg }
   1676  1.25       mrg #endif
   1677   1.1       mrg 
   1678   1.6       mrg /* print a ratio */
   1679   1.6       mrg static void
   1680   1.6       mrg print_ratio(off_t in, off_t out, FILE *where)
   1681   1.6       mrg {
   1682  1.35       mrg 	int64_t percent10;	/* 10 * percent */
   1683  1.35       mrg 	off_t diff = in - out;
   1684  1.35       mrg 	char ch;
   1685   1.6       mrg 
   1686  1.35       mrg 	if (in == 0)
   1687  1.35       mrg 		percent10 = 0;
   1688  1.35       mrg 	else if (diff > 0x400000) 	/* anything with 22 or more bits */
   1689  1.35       mrg 		percent10 = diff / (in / 1000);
   1690  1.35       mrg 	else
   1691  1.35       mrg 		percent10 = (1000 * diff) / in;
   1692  1.35       mrg 
   1693  1.35       mrg 	if (percent10 < 0) {
   1694  1.35       mrg 		percent10 = -percent10;
   1695  1.35       mrg 		ch = '-';
   1696  1.35       mrg 	} else
   1697  1.35       mrg 		ch = ' ';
   1698  1.35       mrg 
   1699  1.35       mrg 	/*
   1700  1.35       mrg 	 * ugh.  for negative percentages < 10, we need to avoid printing a
   1701  1.35       mrg 	 * a space between the "-" and the single number.
   1702  1.35       mrg 	 */
   1703  1.35       mrg 	if (ch == '-' && percent10 / 10LL < 10)
   1704  1.35       mrg 		fprintf(where, " -%1d.%1u%%", (unsigned)(percent10 / 10LL),
   1705  1.35       mrg 					      (unsigned)(percent10 % 10LL));
   1706   1.6       mrg 	else
   1707  1.35       mrg 		fprintf(where, "%c%2d.%1u%%", ch, (unsigned)(percent10 / 10LL),
   1708  1.35       mrg 					          (unsigned)(percent10 % 10LL));
   1709   1.6       mrg }
   1710   1.6       mrg 
   1711  1.26       mrg #ifndef SMALL
   1712   1.1       mrg /* print compression statistics, and the new name (if there is one!) */
   1713   1.1       mrg static void
   1714  1.31       mrg print_verbage(char *file, char *nfile, off_t usize, off_t gsize)
   1715   1.1       mrg {
   1716  1.37       mrg 	if (file)
   1717  1.37       mrg 		fprintf(stderr, "%s:%s  ", file,
   1718  1.37       mrg 		    strlen(file) < 7 ? "\t\t" : "\t");
   1719  1.31       mrg 	print_ratio((off_t)usize, (off_t)gsize, stderr);
   1720   1.4       mrg 	if (nfile)
   1721   1.4       mrg 		fprintf(stderr, " -- replaced with %s", nfile);
   1722   1.1       mrg 	fprintf(stderr, "\n");
   1723   1.1       mrg 	fflush(stderr);
   1724   1.1       mrg }
   1725   1.1       mrg 
   1726   1.1       mrg /* print test results */
   1727   1.1       mrg static void
   1728  1.45       agc print_test(const char *file, int ok)
   1729   1.1       mrg {
   1730   1.1       mrg 
   1731  1.46       mrg 	if (exit_value == 0 && ok == 0)
   1732  1.46       mrg 		exit_value = 1;
   1733   1.1       mrg 	fprintf(stderr, "%s:%s  %s\n", file,
   1734   1.1       mrg 	    strlen(file) < 7 ? "\t\t" : "\t", ok ? "OK" : "NOT OK");
   1735   1.1       mrg 	fflush(stderr);
   1736   1.1       mrg }
   1737  1.26       mrg #endif
   1738   1.1       mrg 
   1739   1.5       mrg /* print a file's info ala --list */
   1740   1.5       mrg /* eg:
   1741   1.5       mrg   compressed uncompressed  ratio uncompressed_name
   1742   1.5       mrg       354841      1679360  78.8% /usr/pkgsrc/distfiles/libglade-2.0.1.tar
   1743   1.5       mrg */
   1744   1.5       mrg static void
   1745  1.35       mrg print_list(int fd, off_t out, const char *outfile, time_t ts)
   1746   1.5       mrg {
   1747   1.5       mrg 	static int first = 1;
   1748  1.29       mrg #ifndef SMALL
   1749   1.6       mrg 	static off_t in_tot, out_tot;
   1750  1.29       mrg 	u_int32_t crc;
   1751  1.29       mrg #endif
   1752  1.35       mrg 	off_t in;
   1753   1.5       mrg 	int rv;
   1754   1.5       mrg 
   1755   1.7       mrg 	if (first) {
   1756  1.26       mrg #ifndef SMALL
   1757   1.7       mrg 		if (vflag)
   1758   1.7       mrg 			printf("method  crc     date  time  ");
   1759  1.26       mrg #endif
   1760   1.7       mrg 		if (qflag == 0)
   1761   1.7       mrg 			printf("  compressed uncompressed  "
   1762   1.7       mrg 			       "ratio uncompressed_name\n");
   1763   1.7       mrg 	}
   1764   1.5       mrg 	first = 0;
   1765   1.5       mrg 
   1766   1.6       mrg 	/* print totals? */
   1767  1.29       mrg #ifndef SMALL
   1768   1.6       mrg 	if (fd == -1) {
   1769   1.6       mrg 		in = in_tot;
   1770   1.6       mrg 		out = out_tot;
   1771  1.29       mrg 	} else
   1772  1.29       mrg #endif
   1773  1.29       mrg 	{
   1774   1.6       mrg 		/* read the last 4 bytes - this is the uncompressed size */
   1775   1.7       mrg 		rv = lseek(fd, (off_t)(-8), SEEK_END);
   1776   1.6       mrg 		if (rv != -1) {
   1777   1.7       mrg 			unsigned char buf[8];
   1778   1.6       mrg 			u_int32_t usize;
   1779   1.6       mrg 
   1780   1.6       mrg 			if (read(fd, (char *)buf, sizeof(buf)) != sizeof(buf))
   1781  1.37       mrg 				maybe_warn("read of uncompressed size");
   1782   1.7       mrg 			usize = buf[4] | buf[5] << 8 | buf[6] << 16 | buf[7] << 24;
   1783  1.35       mrg 			in = (off_t)usize;
   1784  1.29       mrg #ifndef SMALL
   1785  1.29       mrg 			crc = buf[0] | buf[1] << 8 | buf[2] << 16 | buf[3] << 24;
   1786  1.29       mrg #endif
   1787   1.6       mrg 		}
   1788   1.6       mrg 	}
   1789   1.5       mrg 
   1790  1.26       mrg #ifndef SMALL
   1791   1.7       mrg 	if (vflag && fd == -1)
   1792   1.7       mrg 		printf("                            ");
   1793   1.7       mrg 	else if (vflag) {
   1794   1.7       mrg 		char *date = ctime(&ts);
   1795   1.7       mrg 
   1796   1.7       mrg 		/* skip the day, 1/100th second, and year */
   1797   1.7       mrg 		date += 4;
   1798   1.7       mrg 		date[12] = 0;
   1799   1.7       mrg 		printf("%5s %08x %11s ", "defla"/*XXX*/, crc, date);
   1800   1.7       mrg 	}
   1801  1.29       mrg 	in_tot += in;
   1802  1.29       mrg 	out_tot += out;
   1803  1.26       mrg #endif
   1804  1.35       mrg 	printf("%12llu %12llu ", (unsigned long long)out, (unsigned long long)in);
   1805   1.6       mrg 	print_ratio(in, out, stdout);
   1806   1.6       mrg 	printf(" %s\n", outfile);
   1807   1.5       mrg }
   1808   1.5       mrg 
   1809   1.1       mrg /* display the usage of NetBSD gzip */
   1810   1.1       mrg static void
   1811   1.1       mrg usage(void)
   1812   1.1       mrg {
   1813   1.1       mrg 
   1814   1.1       mrg 	fprintf(stderr, "%s\n", gzip_version);
   1815   1.1       mrg 	fprintf(stderr,
   1816  1.29       mrg     "usage: %s [-" OPT_LIST "] [<file> [<file> ...]]\n"
   1817  1.11       mrg #ifndef SMALL
   1818   1.1       mrg     " -c --stdout          write to stdout, keep original files\n"
   1819   1.1       mrg     "    --to-stdout\n"
   1820   1.1       mrg     " -d --decompress      uncompress files\n"
   1821   1.1       mrg     "    --uncompress\n"
   1822   1.1       mrg     " -f --force           force overwriting & compress links\n"
   1823   1.1       mrg     " -h --help            display this help\n"
   1824   1.1       mrg     " -n --no-name         don't save original file name or time stamp\n"
   1825   1.1       mrg     " -N --name            save or restore original file name and time stamp\n"
   1826   1.1       mrg     " -q --quiet           output no warnings\n"
   1827   1.1       mrg     " -r --recursive       recursively compress files in directories\n"
   1828   1.1       mrg     " -S .suf              use suffix .suf instead of .gz\n"
   1829   1.1       mrg     "    --suffix .suf\n"
   1830   1.1       mrg     " -t --test            test compressed file\n"
   1831   1.1       mrg     " -v --verbose         print extra statistics\n"
   1832   1.1       mrg     " -V --version         display program version\n"
   1833   1.1       mrg     " -1 --fast            fastest (worst) compression\n"
   1834   1.1       mrg     " -2 .. -8             set compression level\n"
   1835   1.1       mrg     " -9 --best            best (slowest) compression\n",
   1836  1.11       mrg #else
   1837  1.11       mrg     ,
   1838  1.11       mrg #endif
   1839   1.1       mrg 	    getprogname());
   1840   1.1       mrg 	exit(0);
   1841   1.1       mrg }
   1842   1.1       mrg 
   1843   1.1       mrg /* display the version of NetBSD gzip */
   1844   1.1       mrg static void
   1845   1.1       mrg display_version(void)
   1846   1.1       mrg {
   1847   1.1       mrg 
   1848   1.1       mrg 	fprintf(stderr, "%s\n", gzip_version);
   1849   1.1       mrg 	exit(0);
   1850   1.1       mrg }
   1851  1.10       mrg 
   1852  1.24   tsutsui #ifndef NO_BZIP2_SUPPORT
   1853  1.10       mrg #include "unbzip2.c"
   1854  1.24   tsutsui #endif
   1855  1.24   tsutsui #ifndef NO_COMPRESS_SUPPORT
   1856  1.22       mrg #include "zuncompress.c"
   1857  1.24   tsutsui #endif
   1858