GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: usr.bin/compress/main.c Lines: 94 425 22.1 %
Date: 2016-12-06 Branches: 63 431 14.6 %

Line Branch Exec Source
1
/*	$OpenBSD: main.c,v 1.91 2016/07/14 08:31:18 semarie Exp $	*/
2
3
/*
4
 * Copyright (c) 1992, 1993
5
 *	The Regents of the University of California.  All rights reserved.
6
 * Copyright (c) 1997-2002 Michael Shalayeff
7
 *
8
 * Redistribution and use in source and binary forms, with or without
9
 * modification, are permitted provided that the following conditions
10
 * are met:
11
 * 1. Redistributions of source code must retain the above copyright
12
 *    notice, this list of conditions and the following disclaimer.
13
 * 2. Redistributions in binary form must reproduce the above copyright
14
 *    notice, this list of conditions and the following disclaimer in the
15
 *    documentation and/or other materials provided with the distribution.
16
 * 3. Neither the name of the University nor the names of its contributors
17
 *    may be used to endorse or promote products derived from this software
18
 *    without specific prior written permission.
19
 *
20
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23
 * IN NO EVENT SHALL THE AUTHOR OR HIS RELATIVES BE LIABLE FOR ANY DIRECT,
24
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
25
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
26
 * SERVICES; LOSS OF MIND, USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
28
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
29
 * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
30
 * THE POSSIBILITY OF SUCH DAMAGE.
31
 */
32
33
#include <sys/time.h>
34
#include <sys/stat.h>
35
36
#include <getopt.h>
37
#include <err.h>
38
#include <errno.h>
39
#include <fts.h>
40
#include <libgen.h>
41
#include <stdio.h>
42
#include <stdlib.h>
43
#include <stdbool.h>
44
#include <string.h>
45
#include <unistd.h>
46
#include <limits.h>
47
#include <fcntl.h>
48
#include <paths.h>
49
#include "compress.h"
50
51
#define min(a,b) ((a) < (b)? (a) : (b))
52
53
int cat, decomp, pipin, force, verbose, testmode, list, recurse, storename;
54
extern char *__progname;
55
56
const struct compressor {
57
	const char *name;
58
	const char *suffix;
59
	const u_char *magic;
60
	const char *comp_opts;
61
	const char *decomp_opts;
62
	const char *cat_opts;
63
	void *(*open)(int, const char *, char *, int, u_int32_t, int);
64
	int (*read)(void *, char *, int);
65
	int (*write)(void *, const char *, int);
66
	int (*close)(void *, struct z_info *, const char *, struct stat *);
67
} c_table[] = {
68
#define M_DEFLATE (&c_table[0])
69
	{
70
		"deflate",
71
		".gz",
72
		"\037\213",
73
		"123456789ab:cdfhLlNnOo:qrS:tVv",
74
		"cfhLlNno:qrtVv",
75
		"fhqr",
76
		gz_open,
77
		gz_read,
78
		gz_write,
79
		gz_close
80
	},
81
#define M_COMPRESS (&c_table[1])
82
#ifndef SMALL
83
	{
84
		"compress",
85
		".Z",
86
		"\037\235",
87
		"123456789ab:cdfghlNnOo:qrS:tv",
88
		"cfhlNno:qrtv",
89
		"fghqr",
90
		z_open,
91
		zread,
92
		zwrite,
93
		z_close
94
	},
95
#endif /* SMALL */
96
#if 0
97
#define M_LZH (&c_table[2])
98
  { "lzh", ".lzh", "\037\240", lzh_open, lzh_read, lzh_write, lzh_close },
99
#define M_ZIP (&c_table[3])
100
  { "zip", ".zip", "PK", zip_open, zip_read, zip_write, zip_close },
101
#define M_PACK (&c_table[4])
102
  { "pack", ".pak", "\037\036", pak_open, pak_read, pak_write, pak_close },
103
#endif
104
  { NULL }
105
};
106
107
#ifndef SMALL
108
const struct compressor null_method = {
109
	"null",
110
	".nul",
111
	"XX",
112
	"123456789ab:cdfghlNnOo:qrS:tv",
113
	"cfhlNno:qrtv",
114
	"fghqr",
115
	null_open,
116
	null_read,
117
	null_write,
118
	null_close
119
};
120
#endif /* SMALL */
121
122
int permission(const char *);
123
__dead void usage(int);
124
int docompress(const char *, char *, const struct compressor *,
125
    int, struct stat *);
126
int dodecompress(const char *, char *, const struct compressor *,
127
    int, struct stat *);
128
const struct compressor *check_method(int);
129
const char *check_suffix(const char *);
130
char *set_outfile(const char *, char *, size_t);
131
void list_stats(const char *, const struct compressor *, struct z_info *);
132
void verbose_info(const char *, off_t, off_t, u_int32_t);
133
134
const struct option longopts[] = {
135
#ifndef SMALL
136
	{ "ascii",	no_argument,		0, 'a' },
137
	{ "stdout",	no_argument,		0, 'c' },
138
	{ "to-stdout",	no_argument,		0, 'c' },
139
	{ "decompress",	no_argument,		0, 'd' },
140
	{ "uncompress",	no_argument,		0, 'd' },
141
	{ "force",	no_argument,		0, 'f' },
142
	{ "help",	no_argument,		0, 'h' },
143
	{ "list",	no_argument,		0, 'l' },
144
	{ "license",	no_argument,		0, 'L' },
145
	{ "no-name",	no_argument,		0, 'n' },
146
	{ "name",	no_argument,		0, 'N' },
147
	{ "quiet",	no_argument,		0, 'q' },
148
	{ "recursive",	no_argument,		0, 'r' },
149
	{ "suffix",	required_argument,	0, 'S' },
150
	{ "test",	no_argument,		0, 't' },
151
	{ "verbose",	no_argument,		0, 'v' },
152
	{ "version",	no_argument,		0, 'V' },
153
	{ "fast",	no_argument,		0, '1' },
154
	{ "best",	no_argument,		0, '9' },
155
#endif /* SMALL */
156
	{ NULL }
157
};
158
159
int
160
main(int argc, char *argv[])
161
3
{
162
	FTS *ftsp;
163
	FTSENT *entry;
164
	const struct compressor *method;
165
	const char *optstr, *s;
166
	char *p, *infile;
167
	char outfile[PATH_MAX], _infile[PATH_MAX], suffix[16];
168
	int bits, ch, error, rc, cflag, oflag;
169
170
3
	if (pledge("stdio rpath wpath cpath fattr chown", NULL) == -1)
171
		err(1, "pledge");
172
173
3
	bits = cflag = oflag = 0;
174
3
	storename = -1;
175
3
	p = __progname;
176
3
	if (p[0] == 'g') {
177
3
		method = M_DEFLATE;
178
3
		bits = 6;
179
3
		p++;
180
	} else {
181
#ifdef SMALL
182
		method = M_DEFLATE;
183
#else
184
		method = M_COMPRESS;
185
#endif /* SMALL */
186
	}
187
3
	optstr = method->comp_opts;
188
189
3
	decomp = 0;
190
3
	pmode = MODE_COMP;
191
3
	if (!strcmp(p, "zcat")) {
192
		decomp++;
193
		cflag = 1;
194
		pmode = MODE_CAT;
195
	} else {
196

3
		if (p[0] == 'u' && p[1] == 'n') {
197
			p += 2;
198
			decomp++;
199
			pmode = MODE_DECOMP;
200
		}
201
202

3
		if (strcmp(p, "zip") &&
203
		    strcmp(p, "compress"))
204
			errx(1, "unknown program name");
205
	}
206
207
3
	strlcpy(suffix, method->suffix, sizeof(suffix));
208
209

3
	if (method == M_DEFLATE && (p = getenv("GZIP")) != NULL) {
210
		char *evbuf, *last, **nargv = NULL;
211
		int argc_extra = 0, nargc = 0;
212
213
		if ((evbuf = strdup(p)) == NULL)
214
			err(1, NULL);
215
		for ((p = strtok_r(evbuf, " ", &last)); p != NULL;
216
		    (p = strtok_r(NULL, " ", &last))) {
217
			if (nargc + 1 >= argc_extra) {
218
				argc_extra += 1024;
219
				nargv = reallocarray(nargv,
220
				    argc + argc_extra + 1, sizeof(char *));
221
				if (nargv == NULL)
222
					err(1, NULL);
223
			}
224
			nargv[++nargc] = p;
225
		}
226
		if (nargv != NULL) {
227
			nargv[0] = *argv++;
228
			while ((nargv[++nargc] = *argv++))
229
				;
230
			argv = nargv;
231
			argc = nargc;
232
		}
233
	}
234
235
3
	optstr += pmode;
236
12
	while ((ch = getopt_long(argc, argv, optstr, longopts, NULL)) != -1)
237





6
		switch (ch) {
238
		case '1':
239
		case '2':
240
		case '3':
241
		case '4':
242
		case '5':
243
		case '6':
244
		case '7':
245
		case '8':
246
		case '9':
247
			method = M_DEFLATE;
248
			strlcpy(suffix, method->suffix, sizeof(suffix));
249
			bits = ch - '0';
250
			break;
251
		case 'a':
252
			warnx("option -a is ignored on this system");
253
			break;
254
		case 'b':
255
			bits = strtol(optarg, &p, 10);
256
			/*
257
			 * POSIX 1002.3 says 9 <= bits <= 14 for portable
258
			 * apps, but says the implementation may allow
259
			 * greater.
260
			 */
261
			if (*p)
262
				errx(1, "illegal bit count -- %s", optarg);
263
			break;
264
		case 'c':
265
3
			cflag = 1;
266
3
			break;
267
		case 'd':		/* Backward compatible. */
268
3
			decomp++;
269
3
			break;
270
		case 'f':
271
			force++;
272
			break;
273
		case 'g':
274
			method = M_DEFLATE;
275
			strlcpy(suffix, method->suffix, sizeof(suffix));
276
			bits = 6;
277
			break;
278
		case 'l':
279
			list++;
280
			testmode = 1;
281
			decomp++;
282
			break;
283
		case 'n':
284
			storename = 0;
285
			break;
286
		case 'N':
287
			storename = 1;
288
			break;
289
#ifndef SMALL
290
		case 'O':
291
			method = M_COMPRESS;
292
			strlcpy(suffix, method->suffix, sizeof(suffix));
293
			break;
294
#endif /* SMALL */
295
		case 'o':
296
			if (strlcpy(outfile, optarg,
297
			    sizeof(outfile)) >= sizeof(outfile))
298
				errx(1, "-o argument is too long");
299
			oflag = 1;
300
			break;
301
		case 'q':
302
			verbose = -1;
303
			break;
304
		case 'S':
305
			p = suffix;
306
			if (optarg[0] != '.')
307
				*p++ = '.';
308
			strlcpy(p, optarg, sizeof(suffix) - (p - suffix));
309
			p = optarg;
310
			break;
311
		case 't':
312
			testmode = 1;
313
			decomp++;
314
			break;
315
		case 'V':
316
			exit (0);
317
		case 'v':
318
			verbose++;
319
			break;
320
		case 'L':
321
			exit (0);
322
		case 'r':
323
			recurse++;
324
			break;
325
326
		case 'h':
327
			usage(0);
328
			break;
329
		default:
330
			usage(1);
331
		}
332
3
	argc -= optind;
333
3
	argv += optind;
334
335

3
	if (cflag || testmode || (!oflag && argc == 0))
336
3
		if (pledge("stdio rpath wpath cpath", NULL) == -1)
337
			err(1, "pledge");
338
339
3
	if (argc == 0) {
340
3
		argv = calloc(2, sizeof(char *));
341
3
		if (argv == NULL)
342
			err(1, NULL);
343
3
		argv[0] = "-";
344
3
		argc = 1;
345
	}
346

3
	if (oflag && (recurse || argc > 1))
347
		errx(1, "-o option may only be used with a single input file");
348
349
3
	if ((cat && argc) + testmode + oflag > 1)
350
		errx(1, "may not mix -o, -c, or -t options");
351
	/*
352
	 * By default, when compressing store the original name and timestamp
353
	 * in the header.  Do not restore these when decompressing unless
354
	 * the -N option is given.
355
	 */
356
3
	if (storename == -1)
357
3
		storename = !decomp;
358
359
3
	if ((ftsp = fts_open(argv, FTS_PHYSICAL|FTS_NOCHDIR, 0)) == NULL)
360
		err(1, NULL);
361
9
	for (rc = SUCCESS; (entry = fts_read(ftsp)) != NULL;) {
362
3
		cat = cflag;
363
3
		pipin = 0;
364
3
		infile = entry->fts_path;
365

3
		if (infile[0] == '-' && infile[1] == '\0') {
366
3
			infile = "stdin";
367
3
			pipin++;
368
3
			if (!oflag)
369
3
				cat = 1;
370
		}
371
		else
372
			switch (entry->fts_info) {
373
			case FTS_D:
374
				if (!recurse) {
375
					warnx("%s is a directory: ignored",
376
					    infile);
377
					fts_set(ftsp, entry, FTS_SKIP);
378
				}
379
				continue;
380
			case FTS_DP:
381
				continue;
382
			case FTS_NS:
383
				/*
384
				 * If file does not exist and has no suffix,
385
				 * tack on the default suffix and try that.
386
				 */
387
				if (entry->fts_errno == ENOENT) {
388
					p = strrchr(entry->fts_accpath, '.');
389
					if ((p == NULL ||
390
					    strcmp(p, suffix) != 0) &&
391
					    snprintf(_infile, sizeof(_infile),
392
					    "%s%s", infile, suffix) <
393
					    sizeof(_infile) &&
394
					    stat(_infile, entry->fts_statp) ==
395
					    0 &&
396
					    S_ISREG(entry->fts_statp->st_mode)) {
397
						infile = _infile;
398
						break;
399
					}
400
				}
401
			case FTS_ERR:
402
			case FTS_DNR:
403
				warnx("%s: %s", infile,
404
				    strerror(entry->fts_errno));
405
				rc = rc ? rc : WARNING;
406
				continue;
407
			default:
408
				if (!S_ISREG(entry->fts_statp->st_mode) &&
409
				    !(S_ISLNK(entry->fts_statp->st_mode) &&
410
				    cat)) {
411
					warnx("%s not a regular file%s",
412
					    infile, cat ? "" : ": unchanged");
413
					rc = rc ? rc : WARNING;
414
					continue;
415
				}
416
				break;
417
			}
418
419

3
		if (!decomp && !pipin && (s = check_suffix(infile)) != NULL) {
420
			warnx("%s already has %s suffix -- unchanged",
421
			    infile, s);
422
			rc = rc ? rc : WARNING;
423
			continue;
424
		}
425
426
3
		if (!oflag) {
427
3
			if (cat)
428
3
				strlcpy(outfile, "stdout", sizeof(outfile));
429
			else if (decomp) {
430
				if (set_outfile(infile, outfile,
431
				    sizeof outfile) == NULL) {
432
					if (!recurse) {
433
						warnx("%s: unknown suffix: "
434
						    "ignored", infile);
435
						rc = rc ? rc : WARNING;
436
					}
437
					continue;
438
				}
439
			} else {
440
				if (snprintf(outfile, sizeof(outfile),
441
				    "%s%s", infile, suffix) >= sizeof(outfile)) {
442
					warnx("%s%s: name too long",
443
					    infile, suffix);
444
					rc = rc ? rc : WARNING;
445
					continue;
446
				}
447
			}
448
		}
449
450

3
		if (verbose > 0 && !pipin && !list)
451
			fprintf(stderr, "%s:\t", infile);
452
453
3
		error = (decomp ? dodecompress : docompress)
454
		    (infile, outfile, method, bits, entry->fts_statp);
455
456
3
		switch (error) {
457
		case SUCCESS:
458

3
			if (!cat && !testmode) {
459
				if (!pipin && unlink(infile) && verbose >= 0)
460
					warn("input: %s", infile);
461
			}
462
			break;
463
		case WARNING:
464
			rc = rc ? rc : WARNING;
465
			break;
466
		default:
467
			rc = FAILURE;
468
			break;
469
		}
470
	}
471
3
	if (list)
472
		list_stats(NULL, NULL, NULL);
473
3
	fts_close(ftsp);
474
3
	exit(rc);
475
}
476
477
int
478
docompress(const char *in, char *out, const struct compressor *method,
479
    int bits, struct stat *sb)
480
{
481
#ifndef SMALL
482
	u_char buf[Z_BUFSIZE];
483
	char *name;
484
	int error, ifd, ofd, flags, oreg;
485
	void *cookie;
486
	ssize_t nr;
487
	u_int32_t mtime;
488
	struct z_info info;
489
	struct stat osb;
490
491
	mtime = 0;
492
	flags = oreg = 0;
493
	error = SUCCESS;
494
	name = NULL;
495
	cookie  = NULL;
496
497
	if (pipin)
498
		ifd = dup(STDIN_FILENO);
499
	else
500
		ifd = open(in, O_RDONLY);
501
	if (ifd < 0) {
502
		if (verbose >= 0)
503
			warn("%s", in);
504
		return (FAILURE);
505
	}
506
507
	if (cat)
508
		ofd = dup(STDOUT_FILENO);
509
	else {
510
		if (stat(out, &osb) == 0) {
511
			oreg = S_ISREG(osb.st_mode);
512
			if (!force && oreg && !permission(out)) {
513
				(void) close(ifd);
514
				return (WARNING);
515
			}
516
		}
517
		ofd = open(out, O_WRONLY|O_CREAT|O_TRUNC, S_IWUSR);
518
	}
519
	if (ofd < 0) {
520
		if (verbose >= 0)
521
			warn("%s", out);
522
		(void) close(ifd);
523
		return (FAILURE);
524
	}
525
526
	if (method != M_COMPRESS && !force && isatty(ofd)) {
527
		if (verbose >= 0)
528
			warnx("%s: won't write compressed data to terminal",
529
			    out);
530
		(void) close(ofd);
531
		(void) close(ifd);
532
		return (FAILURE);
533
	}
534
535
	if (!pipin && storename) {
536
		name = basename(in);
537
		mtime = (u_int32_t)sb->st_mtime;
538
	}
539
	if ((cookie = (*method->open)(ofd, "w", name, bits, mtime, flags)) == NULL) {
540
		if (verbose >= 0)
541
			warn("%s", out);
542
		if (oreg)
543
			(void) unlink(out);
544
		(void) close(ofd);
545
		(void) close(ifd);
546
		return (FAILURE);
547
	}
548
549
	while ((nr = read(ifd, buf, sizeof(buf))) > 0)
550
		if ((method->write)(cookie, buf, nr) != nr) {
551
			if (verbose >= 0)
552
				warn("%s", out);
553
			error = FAILURE;
554
			break;
555
		}
556
557
	if (!error && nr < 0) {
558
		if (verbose >= 0)
559
			warn("%s", in);
560
		error = FAILURE;
561
	}
562
563
	if ((method->close)(cookie, &info, out, sb)) {
564
		if (!error && verbose >= 0)
565
			warn("%s", out);
566
		error = FAILURE;
567
	}
568
569
	if (close(ifd)) {
570
		if (!error && verbose >= 0)
571
			warn("%s", in);
572
		error = FAILURE;
573
	}
574
575
	if (!force && !cat && info.total_out >= info.total_in) {
576
		if (verbose > 0)
577
			fprintf(stderr, "file would grow; left unmodified\n");
578
		(void) unlink(out);
579
		error = WARNING;
580
	}
581
582
	if (error) {
583
		if (oreg)
584
			(void) unlink(out);
585
	} else if (verbose > 0)
586
		verbose_info(out, info.total_out, info.total_in, info.hlen);
587
588
	return (error);
589
#else
590
	warnx("compression not supported");
591
	return (FAILURE);
592
#endif
593
}
594
595
const struct compressor *
596
check_method(int fd)
597
3
{
598
	const struct compressor *method;
599
	u_char magic[2];
600
601
3
	if (read(fd, magic, sizeof(magic)) != 2)
602
		return (NULL);
603
3
	for (method = &c_table[0]; method->name != NULL; method++) {
604

3
		if (magic[0] == method->magic[0] &&
605
		    magic[1] == method->magic[1])
606
3
			return (method);
607
	}
608
#ifndef SMALL
609
	if (force && cat) {
610
		null_magic[0] = magic[0];
611
		null_magic[1] = magic[1];
612
		return (&null_method);
613
	}
614
#endif /* SMALL */
615
	return (NULL);
616
}
617
618
int
619
dodecompress(const char *in, char *out, const struct compressor *method,
620
    int bits, struct stat *sb)
621
3
{
622
	u_char buf[Z_BUFSIZE];
623
	char oldname[PATH_MAX];
624
	int error, oreg, ifd, ofd;
625
	void *cookie;
626
	ssize_t nr;
627
	struct z_info info;
628
	struct stat osb;
629
630
3
	oreg = 0;
631
3
	error = SUCCESS;
632
3
	cookie = NULL;
633
634
3
	if (pipin)
635
3
		ifd = dup(STDIN_FILENO);
636
	else
637
		ifd = open(in, O_RDONLY);
638
3
	if (ifd < 0) {
639
		if (verbose >= 0)
640
			warn("%s", in);
641
		return -1;
642
	}
643
644

3
	if (!force && isatty(ifd)) {
645
		if (verbose >= 0)
646
			warnx("%s: won't read compressed data from terminal",
647
			    in);
648
		close (ifd);
649
		return -1;
650
	}
651
652
3
	if ((method = check_method(ifd)) == NULL) {
653
		if (verbose >= 0)
654
			warnx("%s: unrecognized file format", in);
655
		close (ifd);
656
		return -1;
657
	}
658
659
	/* XXX - open constrains outfile to MAXPATHLEN so this is safe */
660
3
	oldname[0] = '\0';
661
3
	if ((cookie = (*method->open)(ifd, "r", oldname, bits, 0, 1)) == NULL) {
662
		if (verbose >= 0)
663
			warn("%s", in);
664
		close (ifd);
665
		return (FAILURE);
666
	}
667

3
	if (storename && oldname[0] != '\0') {
668
		char *cp = strrchr(out, '/');
669
		if (cp != NULL) {
670
			*(cp + 1) = '\0';
671
			strlcat(out, oldname, PATH_MAX);
672
		} else
673
			strlcpy(out, oldname, PATH_MAX);
674
		cat = 0;			/* XXX should -c override? */
675
	}
676
677
3
	if (testmode)
678
		ofd = -1;
679
	else {
680
3
		if (cat)
681
3
			ofd = dup(STDOUT_FILENO);
682
		else {
683
			if (stat(out, &osb) == 0) {
684
				oreg = S_ISREG(osb.st_mode);
685
				if (!force && oreg && !permission(out)) {
686
					(void) close(ifd);
687
					return (WARNING);
688
				}
689
			}
690
			ofd = open(out, O_WRONLY|O_CREAT|O_TRUNC, S_IWUSR);
691
		}
692
3
		if (ofd < 0) {
693
			if (verbose >= 0)
694
				warn("%s", in);
695
			(method->close)(cookie, NULL, NULL, NULL);
696
			return (FAILURE);
697
		}
698
	}
699
700
7
	while ((nr = (method->read)(cookie, buf, sizeof(buf))) > 0) {
701

4
		if (ofd != -1 && write(ofd, buf, nr) != nr) {
702
			if (verbose >= 0)
703
				warn("%s", out);
704
			error = FAILURE;
705
			break;
706
		}
707
	}
708
709
3
	if (!error && nr < 0) {
710
		if (verbose >= 0)
711
			warnx("%s: %s", in,
712
			    errno == EINVAL ? "crc error" : strerror(errno));
713
		error = errno == EINVAL ? WARNING : FAILURE;
714
	}
715
716
3
	if ((method->close)(cookie, &info, NULL, NULL)) {
717
		if (!error && verbose >= 0)
718
			warnx("%s", in);
719
		error = FAILURE;
720
	}
721

3
	if (storename && !cat) {
722
		if (info.mtime != 0) {
723
			sb->st_mtimespec.tv_sec =
724
			    sb->st_atimespec.tv_sec = info.mtime;
725
			sb->st_mtimespec.tv_nsec =
726
			    sb->st_atimespec.tv_nsec = 0;
727
		} else
728
			storename = 0;		/* no timestamp to restore */
729
	}
730
3
	if (error == SUCCESS)
731
3
		setfile(out, ofd, sb);
732
733

3
	if (ofd != -1 && close(ofd)) {
734
		if (!error && verbose >= 0)
735
			warn("%s", out);
736
		error = FAILURE;
737
	}
738
739
3
	if (!error) {
740
3
		if (list) {
741
			if (info.mtime == 0)
742
				info.mtime = (u_int32_t)sb->st_mtime;
743
			list_stats(out, method, &info);
744
3
		} else if (verbose > 0) {
745
			verbose_info(out, info.total_in, info.total_out,
746
			    info.hlen);
747
		}
748
	}
749
750
	/* On error, clean up the file we created but preserve errno. */
751
3
	if (error && oreg)
752
		unlink(out);
753
754
3
	return (error);
755
}
756
757
void
758
setfile(const char *name, int fd, struct stat *fs)
759
6
{
760
	struct timespec ts[2];
761
762

6
	if (name == NULL || cat || testmode)
763
		return;
764
765
	/*
766
	 * If input was a pipe we don't have any info to restore but we
767
	 * must set the mode since the current mode on the file is 0200.
768
	 */
769
	if (pipin) {
770
		mode_t mask = umask(022);
771
		fchmod(fd, DEFFILEMODE & ~mask);
772
		umask(mask);
773
		return;
774
	}
775
776
	/*
777
	 * Changing the ownership probably won't succeed, unless we're root
778
	 * or POSIX_CHOWN_RESTRICTED is not set.  Set uid/gid bits are not
779
	 * allowed.
780
	 */
781
	fs->st_mode &= ACCESSPERMS;
782
	if (fchown(fd, fs->st_uid, fs->st_gid)) {
783
		if (errno != EPERM)
784
			warn("fchown: %s", name);
785
		fs->st_mode &= ~(S_ISUID|S_ISGID);
786
	}
787
	if (fchmod(fd, fs->st_mode))
788
		warn("fchmod: %s", name);
789
790
	if (fs->st_flags && fchflags(fd, fs->st_flags))
791
		warn("fchflags: %s", name);
792
793
	ts[0] = fs->st_atim;
794
	ts[1] = fs->st_mtim;
795
	if (futimens(fd, ts))
796
		warn("futimens: %s", name);
797
}
798
799
int
800
permission(const char *fname)
801
{
802
	int ch, first;
803
804
	if (!isatty(fileno(stderr)))
805
		return (0);
806
	(void)fprintf(stderr, "overwrite %s? ", fname);
807
	first = ch = getchar();
808
	while (ch != '\n' && ch != EOF)
809
		ch = getchar();
810
	return (first == 'y');
811
}
812
813
/*
814
 * Check infile for a known suffix and return the suffix portion or NULL.
815
 */
816
const char *
817
check_suffix(const char *infile)
818
{
819
	int i;
820
	char *suf, *sep, *separators = ".-_";
821
	static char *suffixes[] = { "Z", "gz", "z", "tgz", "taz", NULL };
822
823
	for (sep = separators; *sep != '\0'; sep++) {
824
		if ((suf = strrchr(infile, *sep)) == NULL)
825
			continue;
826
		suf++;
827
828
		for (i = 0; suffixes[i] != NULL; i++) {
829
			if (strcmp(suf, suffixes[i]) == 0)
830
				return (suf - 1);
831
		}
832
	}
833
	return (NULL);
834
}
835
836
/*
837
 * Set outfile based on the suffix.  In most cases we just strip
838
 * off the suffix but things like .tgz and .taz are special.
839
 */
840
char *
841
set_outfile(const char *infile, char *outfile, size_t osize)
842
{
843
	const char *s;
844
	char *cp;
845
846
	if ((s = check_suffix(infile)) == NULL)
847
		return (NULL);
848
849
	(void)strlcpy(outfile, infile, osize);
850
	cp = outfile + (s - infile) + 1;
851
	/*
852
	 * Convert tgz and taz -> tar, else drop the suffix.
853
	 */
854
	if (strcmp(cp, "tgz") == 0) {
855
		cp[1] = 'a';
856
		cp[2] = 'r';
857
	} else if (strcmp(cp, "taz") == 0)
858
		cp[2] = 'r';
859
	else
860
		cp[-1] = '\0';
861
	return (outfile);
862
}
863
864
/*
865
 * Print output for the -l option.
866
 */
867
void
868
list_stats(const char *name, const struct compressor *method,
869
    struct z_info *info)
870
{
871
	static off_t compressed_total, uncompressed_total, header_total;
872
	static u_int nruns;
873
	char *timestr;
874
875
	if (nruns == 0) {
876
		if (verbose >= 0) {
877
			if (verbose > 0)
878
				fputs("method  crc      date   time  ", stdout);
879
			puts("compressed  uncompressed  ratio  uncompressed_name");
880
		}
881
	}
882
	nruns++;
883
884
	if (name != NULL) {
885
		if (verbose > 0) {
886
			time_t t = info->mtime;		/* XXX 32 bit mtime */
887
888
			timestr = ctime(&t) + 4;
889
			timestr[12] = '\0';
890
			if (timestr[4] == ' ')
891
				timestr[4] = '0';
892
			printf("%-7.7s %08x %s ", method->name, info->crc,
893
			    timestr);
894
		}
895
		printf("%10lld    %10lld  %4.1f%%  %s\n",
896
		    (long long)(info->total_in + info->hlen),
897
		    (long long)info->total_out,
898
		    ((long long)info->total_out - (long long)info->total_in) *
899
		    100.0 / info->total_out, name);
900
		compressed_total += info->total_in;
901
		uncompressed_total += info->total_out;
902
		header_total += info->hlen;
903
	} else if (verbose >= 0) {
904
		if (nruns < 3)		/* only do totals for > 1 files */
905
			return;
906
		if (verbose > 0)
907
			fputs("                              ", stdout);
908
		printf("%10lld    %10lld  %4.1f%%  (totals)\n",
909
		    (long long)(compressed_total + header_total),
910
		    (long long)uncompressed_total,
911
		    (uncompressed_total - compressed_total) *
912
		    100.0 / uncompressed_total);
913
	}
914
}
915
916
void
917
verbose_info(const char *file, off_t compressed, off_t uncompressed,
918
    u_int32_t hlen)
919
{
920
	if (testmode) {
921
		fputs("OK\n", stderr);
922
		return;
923
	}
924
	if (!pipin) {
925
		fprintf(stderr, "\t%4.1f%% -- replaced with %s\n",
926
		    (uncompressed - compressed) * 100.0 / uncompressed, file);
927
	}
928
	compressed += hlen;
929
	fprintf(stderr, "%lld bytes in, %lld bytes out\n",
930
	    (long long)(decomp ? compressed : uncompressed),
931
	    (long long)(decomp ? uncompressed : compressed));
932
}
933
934
__dead void
935
usage(int status)
936
{
937
	const bool gzip = (__progname[0] == 'g');
938
939
	switch (pmode) {
940
	case MODE_COMP:
941
		fprintf(stderr, "usage: %s [-123456789cdf%sh%slNnOqrt%sv] "
942
		    "[-b bits] [-o filename] [-S suffix]\n"
943
		    "       %*s [file ...]\n", __progname,
944
		    !gzip ? "g" : "", gzip ? "L" : "", gzip ? "V" : "",
945
		    (int)strlen(__progname), "");
946
		break;
947
	case MODE_DECOMP:
948
		fprintf(stderr, "usage: %s [-cfh%slNnqrt%sv] [-o filename] "
949
		    "[file ...]\n", __progname,
950
		    gzip ? "L" : "", gzip ? "V" : "");
951
		break;
952
	case MODE_CAT:
953
		fprintf(stderr, "usage: %s [-f%shqr] [file ...]\n",
954
		    __progname, gzip ? "" : "g");
955
		break;
956
	}
957
	exit(status);
958
}