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