[apple/file_cmds.git] / compress / compress.c

/*-
 * Copyright (c) 1992, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#ifndef lint
__used static const char copyright[] =
"@(#) Copyright (c) 1992, 1993\n\
	The Regents of the University of California.  All rights reserved.\n";
#endif

#if 0
#ifndef lint
static char sccsid[] = "@(#)compress.c	8.2 (Berkeley) 1/7/94";
#endif
#endif

#include <sys/cdefs.h>
__FBSDID("$FreeBSD: src/usr.bin/compress/compress.c,v 1.23 2010/12/11 08:32:16 joel Exp $");

#include <sys/param.h>
#include <sys/stat.h>
#include <sys/time.h>

#include <err.h>
#include <errno.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "zopen.h"

void	compress(const char *, const char *, int);
void	cwarn(const char *, ...) __printflike(1, 2);
void	cwarnx(const char *, ...) __printflike(1, 2);
void	decompress(const char *, const char *, int);
int	permission(const char *);
void	setfile(const char *, struct stat *);
void	usage(int);

int eval, force, verbose, cat;

int
main(int argc, char *argv[])
{
	enum {COMPRESS, DECOMPRESS} style;
	size_t len;
	int bits, ch;
	char *p, newname[MAXPATHLEN];

	if (argc < 1)
		usage(1);
	cat = 0;
	if ((p = rindex(argv[0], '/')) == NULL)
		p = argv[0];
	else
		++p;
	if (!strcmp(p, "uncompress"))
		style = DECOMPRESS;
	else if (!strcmp(p, "compress"))
		style = COMPRESS;
	else if (!strcmp(p, "zcat")) {
		cat = 1;
		style = DECOMPRESS;
	} else
		errx(1, "unknown program name");

	bits = 0;
	while ((ch = getopt(argc, argv, "b:cdfv")) != -1)
		switch(ch) {
		case 'b':
			bits = strtol(optarg, &p, 10);
			if (*p)
				errx(1, "illegal bit count -- %s", optarg);
			break;
		case 'c':
			cat = 1;
			break;
		case 'd':		/* Backward compatible. */
			style = DECOMPRESS;
			break;
		case 'f':
			force = 1;
			break;
		case 'v':
			verbose = 1;
			break;
		case '?':
		default:
			usage(style == COMPRESS);
		}
	argc -= optind;
	argv += optind;

	if (argc == 0) {
		cat = 1;
		switch(style) {
		case COMPRESS:
			(void)compress("/dev/stdin", "/dev/stdout", bits);
			break;
		case DECOMPRESS:
			(void)decompress("/dev/stdin", "/dev/stdout", bits);
			break;
		}
		exit (eval);
	}

	if (cat == 1 && argc > 1)
		errx(1, "the -c option permits only a single file argument");

	for (; *argv; ++argv)
		switch(style) {
		case COMPRESS:
			if (strcmp(*argv, "-") == 0) {
				cat = 1;
				compress("/dev/stdin", "/dev/stdout", bits);
				break;
			} else if (cat) {
				compress(*argv, "/dev/stdout", bits);
				break;
			}
			if ((p = rindex(*argv, '.')) != NULL &&
			    !strcmp(p, ".Z")) {
				cwarnx("%s: name already has trailing .Z",
				    *argv);
				break;
			}
			len = strlen(*argv);
			if (len > sizeof(newname) - 3) {
				cwarnx("%s: name too long", *argv);
				break;
			}
			memmove(newname, *argv, len);
			newname[len] = '.';
			newname[len + 1] = 'Z';
			newname[len + 2] = '\0';
			compress(*argv, newname, bits);
			break;
		case DECOMPRESS:
			if (strcmp(*argv, "-") == 0) {
				cat = 1;
				decompress("/dev/stdin", "/dev/stdout", bits);
				break;
			}
			len = strlen(*argv);
			if ((p = rindex(*argv, '.')) == NULL ||
			    strcmp(p, ".Z")) {
				if (len > sizeof(newname) - 3) {
					cwarnx("%s: name too long", *argv);
					break;
				}
				memmove(newname, *argv, len);
				newname[len] = '.';
				newname[len + 1] = 'Z';
				newname[len + 2] = '\0';
				decompress(newname,
				    cat ? "/dev/stdout" : *argv, bits);
			} else {
				if (len - 2 > sizeof(newname) - 1) {
					cwarnx("%s: name too long", *argv);
					break;
				}
				memmove(newname, *argv, len - 2);
				newname[len - 2] = '\0';
				decompress(*argv,
				    cat ? "/dev/stdout" : newname, bits);
			}
			break;
		}
	exit (eval);
}

void
compress(const char *in, const char *out, int bits)
{
	size_t nr;
	struct stat isb, sb;
	FILE *ifp = NULL, *ofp = NULL;
	int exists, isreg, oreg;
	u_char buf[1024];

	exists = !stat(out, &sb);
	if (!force && exists && S_ISREG(sb.st_mode) && !cat && !permission(out)) {
		cwarnx("%s already exists", out);
		return;
	}
	isreg = oreg = !exists || S_ISREG(sb.st_mode);

	if ((ifp = fopen(in, "r")) == NULL) {
		cwarn("%s", in);
		return;
	}
	if (stat(in, &isb)) {		/* DON'T FSTAT! */
		cwarn("%s", in);
		goto err;
	}
	if (!S_ISREG(isb.st_mode))
		isreg = 0;

	if ((ofp = zopen(out, "w", bits)) == NULL) {
		cwarn("%s", out);
		goto err;
	}
	while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
		if (fwrite(buf, 1, nr, ofp) != nr) {
			cwarn("%s", out);
			goto err;
		}

	if (ferror(ifp) || fclose(ifp)) {
		cwarn("%s", in);
		goto err;
	}
	ifp = NULL;

	if (fclose(ofp)) {
		cwarn("%s", out);
		goto err;
	}
	ofp = NULL;

	if (!cat && isreg) {
		if (stat(out, &sb)) {
			cwarn("%s", out);
			goto err;
		}

		if (!force && sb.st_size >= isb.st_size) {
			if (verbose)
		(void)fprintf(stderr, "%s: file would grow; left unmodified\n",
		    in);
			eval = 2;
			if (unlink(out))
				cwarn("%s", out);
			goto err;
		}

		setfile(out, &isb);

		if (unlink(in))
			cwarn("%s", in);

		if (verbose) {
			(void)fprintf(stderr, "%s: ", out);
			if (isb.st_size > sb.st_size)
				(void)fprintf(stderr, "%.0f%% compression\n",
				    ((float)sb.st_size / isb.st_size) * 100.0);
			else
				(void)fprintf(stderr, "%.0f%% expansion\n",
				    ((float)isb.st_size / sb.st_size) * 100.0);
		}
	}
	return;

err:	if (ofp) {
		if (!cat && oreg)
			(void)unlink(out);
		(void)fclose(ofp);
	}
	if (ifp)
		(void)fclose(ifp);
}

void
decompress(const char *in, const char *out, int bits)
{
	size_t nr;
	struct stat sb;
	FILE *ifp, *ofp;
	int exists, isreg, oreg;
	u_char buf[1024];

	exists = !stat(out, &sb);
	if (!force && exists && S_ISREG(sb.st_mode) && !cat && !permission(out)) {
		cwarnx("%s already exists", out);
		return;
	}
	isreg = oreg = !exists || S_ISREG(sb.st_mode);

	ofp = NULL;
	if ((ifp = zopen(in, "r", bits)) == NULL) {
		cwarn("%s", in);
		return;
	}
	if (stat(in, &sb)) {
		cwarn("%s", in);
		goto err;
	}
	if (!S_ISREG(sb.st_mode))
		isreg = 0;

	/*
	 * Try to read the first few uncompressed bytes from the input file
	 * before blindly truncating the output file.
	 */
	if ((nr = fread(buf, 1, sizeof(buf), ifp)) == 0) {
		cwarn("%s", in);
		(void)fclose(ifp);
		return;
	}
	if ((ofp = fopen(out, "w")) == NULL ||
	    (nr != 0 && fwrite(buf, 1, nr, ofp) != nr)) {
		cwarn("%s", out);
		(void)fclose(ifp);
		return;
	}

	while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
		if (fwrite(buf, 1, nr, ofp) != nr) {
			cwarn("%s", out);
			goto err;
		}

	if (ferror(ifp) || fclose(ifp)) {
		cwarn("%s", in);
		goto err;
	}
	ifp = NULL;

	if (fclose(ofp)) {
		cwarn("%s", out);
		goto err;
	}

	if (!cat && isreg) {
		setfile(out, &sb);

		if (unlink(in))
			cwarn("%s", in);
		if (verbose) {
			struct stat isb = sb;
			stat(out, &sb);
			(void)fprintf(stderr, "%s: ", out);
			if (isb.st_size > sb.st_size)
				(void)fprintf(stderr, "%.0f%% compression\n",
				    ((float)sb.st_size / isb.st_size) * 100.0);
			else
				(void)fprintf(stderr, "%.0f%% expansion\n",
				    ((float)isb.st_size / sb.st_size) * 100.0);
		}
	}
	return;

err:	if (ofp) {
		if (!cat && oreg)
			(void)unlink(out);
		(void)fclose(ofp);
	}
	if (ifp)
		(void)fclose(ifp);
}

void
setfile(const char *name, struct stat *fs)
{
	static struct timeval tv[2];

	fs->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;

	TIMESPEC_TO_TIMEVAL(&tv[0], &fs->st_atimespec);
	TIMESPEC_TO_TIMEVAL(&tv[1], &fs->st_mtimespec);
	if (utimes(name, tv))
		cwarn("utimes: %s", name);

	/*
	 * Changing the ownership probably won't succeed, unless we're root
	 * or POSIX_CHOWN_RESTRICTED is not set.  Set uid/gid before setting
	 * the mode; current BSD behavior is to remove all setuid bits on
	 * chown.  If chown fails, lose setuid/setgid bits.
	 */
	if (chown(name, fs->st_uid, fs->st_gid)) {
		if (errno != EPERM)
			cwarn("chown: %s", name);
		fs->st_mode &= ~(S_ISUID|S_ISGID);
	}
	if (chmod(name, fs->st_mode) && errno != ENOTSUP)
		cwarn("chmod: %s", name);

	if (chflags(name, fs->st_flags) && errno != ENOTSUP)
		cwarn("chflags: %s", name);
}

int
permission(const char *fname)
{
	int ch, first;

	if (!isatty(fileno(stderr)))
		return (0);
	(void)fprintf(stderr, "overwrite %s? ", fname);
	first = ch = getchar();
	while (ch != '\n' && ch != EOF)
		ch = getchar();
	return (first == 'y');
}

void
usage(int iscompress)
{
	if (iscompress)
		(void)fprintf(stderr,
		    "usage: compress [-cfv] [-b bits] [file ...]\n");
	else
		(void)fprintf(stderr,
		    "usage: uncompress [-cfv] [-b bits] [file ...]\n");
	exit(1);
}

void
cwarnx(const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	vwarnx(fmt, ap);
	va_end(ap);
	eval = 1;
}

void
cwarn(const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	vwarn(fmt, ap);
	va_end(ap);
	eval = 1;
}
Commit	Line	Data
44a7a5ab A	1	/*-
	2	* Copyright (c) 1992, 1993
	3	* The Regents of the University of California. All rights reserved.
	4	*
	5	* Redistribution and use in source and binary forms, with or without
	6	* modification, are permitted provided that the following conditions
	7	* are met:
	8	* 1. Redistributions of source code must retain the above copyright
	9	* notice, this list of conditions and the following disclaimer.
	10	* 2. Redistributions in binary form must reproduce the above copyright
	11	* notice, this list of conditions and the following disclaimer in the
	12	* documentation and/or other materials provided with the distribution.
44a7a5ab A	13	* 4. Neither the name of the University nor the names of its contributors
	14	* may be used to endorse or promote products derived from this software
	15	* without specific prior written permission.
	16	*
	17	* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
	18	* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	19	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	20	* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
	21	* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	22	* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
	23	* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
	24	* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
	25	* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
	26	* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
	27	* SUCH DAMAGE.
	28	*/
	29
00337e45	30	#include <sys/cdefs.h>
44a7a5ab	31	#ifndef lint
00337e45	32	__used static const char copyright[] =
c59d3020 A	33	"@(#) Copyright (c) 1992, 1993\n\
	34	The Regents of the University of California. All rights reserved.\n";
	35	#endif
44a7a5ab	36
44a7a5ab	37	#if 0
c59d3020	38	#ifndef lint
44a7a5ab	39	static char sccsid[] = "@(#)compress.c 8.2 (Berkeley) 1/7/94";
44a7a5ab	40	#endif
c59d3020 A	41	#endif
	42
	43	#include <sys/cdefs.h>
e19e38b2	44	__FBSDID("$FreeBSD: src/usr.bin/compress/compress.c,v 1.23 2010/12/11 08:32:16 joel Exp $");
44a7a5ab A	45
44a7a5ab A	46	#include <sys/param.h>
44a7a5ab	47	#include <sys/stat.h>
c59d3020	48	#include <sys/time.h>
44a7a5ab A	49
	50	#include <err.h>
	51	#include <errno.h>
c59d3020	52	#include <stdarg.h>
44a7a5ab A	53	#include <stdio.h>
	54	#include <stdlib.h>
	55	#include <string.h>
	56	#include <unistd.h>
	57
c59d3020	58	#include "zopen.h"
44a7a5ab	59
c59d3020 A	60	void compress(const char , const char , int);
	61	void cwarn(const char *, ...) __printflike(1, 2);
	62	void cwarnx(const char *, ...) __printflike(1, 2);
	63	void decompress(const char , const char , int);
	64	int permission(const char *);
	65	void setfile(const char , struct stat );
	66	void usage(int);
44a7a5ab	67
c59d3020	68	int eval, force, verbose, cat;
44a7a5ab A	69
44a7a5ab A	70	int
c59d3020	71	main(int argc, char *argv[])
44a7a5ab	72	{
c59d3020	73	enum {COMPRESS, DECOMPRESS} style;
44a7a5ab	74	size_t len;
c59d3020	75	int bits, ch;
44a7a5ab A	76	char *p, newname[MAXPATHLEN];
44a7a5ab A	77
864a4b6e A	78	if (argc < 1)
864a4b6e A	79	usage(1);
c59d3020 A	80	cat = 0;
c59d3020 A	81	if ((p = rindex(argv[0], '/')) == NULL)
44a7a5ab A	82	p = argv[0];
	83	else
	84	++p;
	85	if (!strcmp(p, "uncompress"))
	86	style = DECOMPRESS;
c59d3020 A	87	else if (!strcmp(p, "compress"))
	88	style = COMPRESS;
	89	else if (!strcmp(p, "zcat")) {
	90	cat = 1;
	91	style = DECOMPRESS;
	92	} else
44a7a5ab A	93	errx(1, "unknown program name");
44a7a5ab A	94
c59d3020	95	bits = 0;
44a7a5ab A	96	while ((ch = getopt(argc, argv, "b:cdfv")) != -1)
	97	switch(ch) {
	98	case 'b':
	99	bits = strtol(optarg, &p, 10);
	100	if (*p)
	101	errx(1, "illegal bit count -- %s", optarg);
	102	break;
	103	case 'c':
	104	cat = 1;
	105	break;
	106	case 'd': /* Backward compatible. */
	107	style = DECOMPRESS;
	108	break;
	109	case 'f':
	110	force = 1;
	111	break;
	112	case 'v':
	113	verbose = 1;
	114	break;
	115	case '?':
	116	default:
	117	usage(style == COMPRESS);
	118	}
	119	argc -= optind;
	120	argv += optind;
	121
	122	if (argc == 0) {
c59d3020	123	cat = 1;
44a7a5ab A	124	switch(style) {
44a7a5ab A	125	case COMPRESS:
44a7a5ab A	126	(void)compress("/dev/stdin", "/dev/stdout", bits);
	127	break;
	128	case DECOMPRESS:
44a7a5ab A	129	(void)decompress("/dev/stdin", "/dev/stdout", bits);
	130	break;
	131	}
	132	exit (eval);
	133	}
	134
	135	if (cat == 1 && argc > 1)
	136	errx(1, "the -c option permits only a single file argument");
	137
c59d3020	138	for (; *argv; ++argv)
44a7a5ab A	139	switch(style) {
44a7a5ab A	140	case COMPRESS:
c59d3020 A	141	if (strcmp(*argv, "-") == 0) {
	142	cat = 1;
	143	compress("/dev/stdin", "/dev/stdout", bits);
	144	break;
	145	} else if (cat) {
44a7a5ab A	146	compress(*argv, "/dev/stdout", bits);
	147	break;
	148	}
c59d3020	149	if ((p = rindex(*argv, '.')) != NULL &&
44a7a5ab A	150	!strcmp(p, ".Z")) {
	151	cwarnx("%s: name already has trailing .Z",
	152	*argv);
	153	break;
	154	}
	155	len = strlen(*argv);
	156	if (len > sizeof(newname) - 3) {
	157	cwarnx("%s: name too long", *argv);
	158	break;
	159	}
	160	memmove(newname, *argv, len);
	161	newname[len] = '.';
	162	newname[len + 1] = 'Z';
	163	newname[len + 2] = '\0';
	164	compress(*argv, newname, bits);
	165	break;
	166	case DECOMPRESS:
c59d3020 A	167	if (strcmp(*argv, "-") == 0) {
	168	cat = 1;
	169	decompress("/dev/stdin", "/dev/stdout", bits);
	170	break;
	171	}
44a7a5ab	172	len = strlen(*argv);
c59d3020	173	if ((p = rindex(*argv, '.')) == NULL \|\|
44a7a5ab A	174	strcmp(p, ".Z")) {
	175	if (len > sizeof(newname) - 3) {
	176	cwarnx("%s: name too long", *argv);
	177	break;
	178	}
	179	memmove(newname, *argv, len);
	180	newname[len] = '.';
	181	newname[len + 1] = 'Z';
	182	newname[len + 2] = '\0';
	183	decompress(newname,
	184	cat ? "/dev/stdout" : *argv, bits);
44a7a5ab A	185	} else {
	186	if (len - 2 > sizeof(newname) - 1) {
	187	cwarnx("%s: name too long", *argv);
	188	break;
	189	}
	190	memmove(newname, *argv, len - 2);
	191	newname[len - 2] = '\0';
	192	decompress(*argv,
	193	cat ? "/dev/stdout" : newname, bits);
44a7a5ab A	194	}
	195	break;
	196	}
44a7a5ab A	197	exit (eval);
	198	}
	199
	200	void
c59d3020	201	compress(const char in, const char out, int bits)
44a7a5ab	202	{
c59d3020	203	size_t nr;
44a7a5ab	204	struct stat isb, sb;
4d0bb651	205	FILE ifp = NULL, ofp = NULL;
44a7a5ab	206	int exists, isreg, oreg;
c59d3020	207	u_char buf[1024];
44a7a5ab	208
c59d3020 A	209	exists = !stat(out, &sb);
	210	if (!force && exists && S_ISREG(sb.st_mode) && !cat && !permission(out)) {
	211	cwarnx("%s already exists", out);
	212	return;
	213	}
	214	isreg = oreg = !exists \|\| S_ISREG(sb.st_mode);
44a7a5ab	215
44a7a5ab A	216	if ((ifp = fopen(in, "r")) == NULL) {
	217	cwarn("%s", in);
	218	return;
	219	}
c59d3020 A	220	if (stat(in, &isb)) { /* DON'T FSTAT! */
	221	cwarn("%s", in);
	222	goto err;
	223	}
	224	if (!S_ISREG(isb.st_mode))
44a7a5ab A	225	isreg = 0;
	226
	227	if ((ofp = zopen(out, "w", bits)) == NULL) {
	228	cwarn("%s", out);
	229	goto err;
	230	}
	231	while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
	232	if (fwrite(buf, 1, nr, ofp) != nr) {
	233	cwarn("%s", out);
	234	goto err;
	235	}
	236
	237	if (ferror(ifp) \|\| fclose(ifp)) {
	238	cwarn("%s", in);
	239	goto err;
	240	}
	241	ifp = NULL;
	242
	243	if (fclose(ofp)) {
	244	cwarn("%s", out);
	245	goto err;
	246	}
	247	ofp = NULL;
	248
c59d3020	249	if (!cat && isreg) {
44a7a5ab A	250	if (stat(out, &sb)) {
	251	cwarn("%s", out);
	252	goto err;
	253	}
	254
	255	if (!force && sb.st_size >= isb.st_size) {
	256	if (verbose)
c59d3020 A	257	(void)fprintf(stderr, "%s: file would grow; left unmodified\n",
	258	in);
	259	eval = 2;
44a7a5ab A	260	if (unlink(out))
	261	cwarn("%s", out);
	262	goto err;
	263	}
	264
	265	setfile(out, &isb);
	266
	267	if (unlink(in))
	268	cwarn("%s", in);
	269
	270	if (verbose) {
c59d3020	271	(void)fprintf(stderr, "%s: ", out);
44a7a5ab	272	if (isb.st_size > sb.st_size)
c59d3020 A	273	(void)fprintf(stderr, "%.0f%% compression\n",
c59d3020 A	274	((float)sb.st_size / isb.st_size) * 100.0);
44a7a5ab	275	else
c59d3020 A	276	(void)fprintf(stderr, "%.0f%% expansion\n",
c59d3020 A	277	((float)isb.st_size / sb.st_size) * 100.0);
44a7a5ab A	278	}
	279	}
	280	return;
	281
	282	err: if (ofp) {
c59d3020	283	if (!cat && oreg)
44a7a5ab A	284	(void)unlink(out);
	285	(void)fclose(ofp);
	286	}
	287	if (ifp)
	288	(void)fclose(ifp);
	289	}
	290
	291	void
c59d3020	292	decompress(const char in, const char out, int bits)
44a7a5ab	293	{
c59d3020	294	size_t nr;
44a7a5ab A	295	struct stat sb;
	296	FILE ifp, ofp;
	297	int exists, isreg, oreg;
c59d3020	298	u_char buf[1024];
44a7a5ab	299
c59d3020 A	300	exists = !stat(out, &sb);
	301	if (!force && exists && S_ISREG(sb.st_mode) && !cat && !permission(out)) {
	302	cwarnx("%s already exists", out);
44a7a5ab A	303	return;
44a7a5ab A	304	}
c59d3020	305	isreg = oreg = !exists \|\| S_ISREG(sb.st_mode);
44a7a5ab	306
4d0bb651	307	ofp = NULL;
44a7a5ab	308	if ((ifp = zopen(in, "r", bits)) == NULL) {
c59d3020 A	309	cwarn("%s", in);
	310	return;
	311	}
	312	if (stat(in, &sb)) {
44a7a5ab A	313	cwarn("%s", in);
	314	goto err;
	315	}
c59d3020	316	if (!S_ISREG(sb.st_mode))
44a7a5ab A	317	isreg = 0;
44a7a5ab A	318
c59d3020 A	319	/*
	320	* Try to read the first few uncompressed bytes from the input file
	321	* before blindly truncating the output file.
	322	*/
	323	if ((nr = fread(buf, 1, sizeof(buf), ifp)) == 0) {
	324	cwarn("%s", in);
	325	(void)fclose(ifp);
	326	return;
	327	}
	328	if ((ofp = fopen(out, "w")) == NULL \|\|
	329	(nr != 0 && fwrite(buf, 1, nr, ofp) != nr)) {
	330	cwarn("%s", out);
	331	(void)fclose(ifp);
	332	return;
	333	}
	334
44a7a5ab A	335	while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
	336	if (fwrite(buf, 1, nr, ofp) != nr) {
	337	cwarn("%s", out);
	338	goto err;
	339	}
	340
	341	if (ferror(ifp) \|\| fclose(ifp)) {
	342	cwarn("%s", in);
	343	goto err;
	344	}
	345	ifp = NULL;
	346
	347	if (fclose(ofp)) {
	348	cwarn("%s", out);
	349	goto err;
	350	}
	351
c59d3020	352	if (!cat && isreg) {
44a7a5ab A	353	setfile(out, &sb);
	354
	355	if (unlink(in))
	356	cwarn("%s", in);
c59d3020 A	357	if (verbose) {
	358	struct stat isb = sb;
	359	stat(out, &sb);
	360	(void)fprintf(stderr, "%s: ", out);
	361	if (isb.st_size > sb.st_size)
	362	(void)fprintf(stderr, "%.0f%% compression\n",
	363	((float)sb.st_size / isb.st_size) * 100.0);
	364	else
	365	(void)fprintf(stderr, "%.0f%% expansion\n",
	366	((float)isb.st_size / sb.st_size) * 100.0);
	367	}
44a7a5ab A	368	}
	369	return;
	370
	371	err: if (ofp) {
c59d3020	372	if (!cat && oreg)
44a7a5ab A	373	(void)unlink(out);
	374	(void)fclose(ofp);
	375	}
	376	if (ifp)
	377	(void)fclose(ifp);
	378	}
	379
	380	void
c59d3020	381	setfile(const char name, struct stat fs)
44a7a5ab A	382	{
	383	static struct timeval tv[2];
	384
	385	fs->st_mode &= S_ISUID\|S_ISGID\|S_IRWXU\|S_IRWXG\|S_IRWXO;
	386
	387	TIMESPEC_TO_TIMEVAL(&tv[0], &fs->st_atimespec);
	388	TIMESPEC_TO_TIMEVAL(&tv[1], &fs->st_mtimespec);
	389	if (utimes(name, tv))
	390	cwarn("utimes: %s", name);
	391
	392	/*
	393	* Changing the ownership probably won't succeed, unless we're root
	394	* or POSIX_CHOWN_RESTRICTED is not set. Set uid/gid before setting
	395	* the mode; current BSD behavior is to remove all setuid bits on
	396	* chown. If chown fails, lose setuid/setgid bits.
	397	*/
	398	if (chown(name, fs->st_uid, fs->st_gid)) {
	399	if (errno != EPERM)
	400	cwarn("chown: %s", name);
	401	fs->st_mode &= ~(S_ISUID\|S_ISGID);
	402	}
864a4b6e	403	if (chmod(name, fs->st_mode) && errno != ENOTSUP)
c59d3020	404	cwarn("chmod: %s", name);
44a7a5ab	405
864a4b6e	406	if (chflags(name, fs->st_flags) && errno != ENOTSUP)
44a7a5ab A	407	cwarn("chflags: %s", name);
	408	}
	409
	410	int
c59d3020	411	permission(const char *fname)
44a7a5ab A	412	{
	413	int ch, first;
	414
	415	if (!isatty(fileno(stderr)))
	416	return (0);
	417	(void)fprintf(stderr, "overwrite %s? ", fname);
	418	first = ch = getchar();
	419	while (ch != '\n' && ch != EOF)
	420	ch = getchar();
	421	return (first == 'y');
	422	}
	423
	424	void
c59d3020	425	usage(int iscompress)
44a7a5ab A	426	{
	427	if (iscompress)
	428	(void)fprintf(stderr,
	429	"usage: compress [-cfv] [-b bits] [file ...]\n");
	430	else
	431	(void)fprintf(stderr,
c59d3020	432	"usage: uncompress [-cfv] [-b bits] [file ...]\n");
44a7a5ab A	433	exit(1);
	434	}
	435
	436	void
44a7a5ab	437	cwarnx(const char *fmt, ...)
44a7a5ab A	438	{
44a7a5ab A	439	va_list ap;
c59d3020	440
44a7a5ab	441	va_start(ap, fmt);
44a7a5ab A	442	vwarnx(fmt, ap);
	443	va_end(ap);
	444	eval = 1;
	445	}
	446
	447	void
44a7a5ab	448	cwarn(const char *fmt, ...)
44a7a5ab A	449	{
44a7a5ab A	450	va_list ap;
c59d3020	451
44a7a5ab	452	va_start(ap, fmt);
44a7a5ab A	453	vwarn(fmt, ap);
	454	va_end(ap);
	455	eval = 1;
	456	}