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