2 * SPDX-License-Identifier: BSD-3-Clause
4 * Copyright (c) 1992, 1993
5 * The Regents of the University of California. All rights reserved.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 static const char copyright[] =
34 "@(#) Copyright (c) 1992, 1993\n\
35 The Regents of the University of California. All rights reserved.\n";
40 static char sccsid[] = "@(#)compress.c 8.2 (Berkeley) 1/7/94";
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
47 #include <sys/param.h>
62 static void compress(const char *, const char *, int);
63 static void cwarn(const char *, ...) __printflike(1, 2);
64 static void cwarnx(const char *, ...) __printflike(1, 2);
65 static void decompress(const char *, const char *, int);
66 static int permission(const char *);
67 static void setfile(const char *, struct stat *);
68 static void usage(int);
70 static int eval, force, verbose;
73 main(int argc, char *argv[])
75 enum {COMPRESS, DECOMPRESS} style;
78 char *p, newname[MAXPATHLEN];
81 if ((p = strrchr(argv[0], '/')) == NULL)
85 if (!strcmp(p, "uncompress"))
87 else if (!strcmp(p, "compress"))
89 else if (!strcmp(p, "zcat")) {
93 errx(1, "unknown program name");
96 while ((ch = getopt(argc, argv, "b:cdfv")) != -1)
99 bits = strtol(optarg, &p, 10);
101 errx(1, "illegal bit count -- %s", optarg);
106 case 'd': /* Backward compatible. */
117 usage(style == COMPRESS);
125 (void)compress("/dev/stdin", "/dev/stdout", bits);
128 (void)decompress("/dev/stdin", "/dev/stdout", bits);
134 if (cat == 1 && style == COMPRESS && argc > 1)
135 errx(1, "the -c option permits only a single file argument");
137 for (; *argv; ++argv)
140 if (strcmp(*argv, "-") == 0) {
141 compress("/dev/stdin", "/dev/stdout", bits);
144 compress(*argv, "/dev/stdout", bits);
147 if ((p = strrchr(*argv, '.')) != NULL &&
149 cwarnx("%s: name already has trailing .Z",
154 if (len > sizeof(newname) - 3) {
155 cwarnx("%s: name too long", *argv);
158 memmove(newname, *argv, len);
160 newname[len + 1] = 'Z';
161 newname[len + 2] = '\0';
162 compress(*argv, newname, bits);
165 if (strcmp(*argv, "-") == 0) {
166 decompress("/dev/stdin", "/dev/stdout", bits);
170 if ((p = strrchr(*argv, '.')) == NULL ||
172 if (len > sizeof(newname) - 3) {
173 cwarnx("%s: name too long", *argv);
176 memmove(newname, *argv, len);
178 newname[len + 1] = 'Z';
179 newname[len + 2] = '\0';
181 cat ? "/dev/stdout" : *argv, bits);
183 if (len - 2 > sizeof(newname) - 1) {
184 cwarnx("%s: name too long", *argv);
187 memmove(newname, *argv, len - 2);
188 newname[len - 2] = '\0';
190 cat ? "/dev/stdout" : newname, bits);
198 compress(const char *in, const char *out, int bits)
203 int exists, isreg, oreg;
206 exists = !stat(out, &sb);
207 if (!force && exists && S_ISREG(sb.st_mode) && !permission(out))
209 isreg = oreg = !exists || S_ISREG(sb.st_mode);
212 if ((ifp = fopen(in, "r")) == NULL) {
216 if (stat(in, &isb)) { /* DON'T FSTAT! */
220 if (!S_ISREG(isb.st_mode))
223 if ((ofp = zopen(out, "w", bits)) == NULL) {
227 while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
228 if (fwrite(buf, 1, nr, ofp) != nr) {
233 if (ferror(ifp) || fclose(ifp)) {
246 if (stat(out, &sb)) {
251 if (!force && sb.st_size >= isb.st_size) {
253 (void)fprintf(stderr, "%s: file would grow; left unmodified\n",
267 (void)fprintf(stderr, "%s: ", out);
268 if (isb.st_size > sb.st_size)
269 (void)fprintf(stderr, "%.0f%% compression\n",
270 ((float)sb.st_size / isb.st_size) * 100.0);
272 (void)fprintf(stderr, "%.0f%% expansion\n",
273 ((float)isb.st_size / sb.st_size) * 100.0);
288 decompress(const char *in, const char *out, int bits)
293 int exists, isreg, oreg;
296 exists = !stat(out, &sb);
297 if (!force && exists && S_ISREG(sb.st_mode) && !permission(out))
299 isreg = oreg = !exists || S_ISREG(sb.st_mode);
302 if ((ifp = zopen(in, "r", bits)) == NULL) {
310 if (!S_ISREG(sb.st_mode))
314 * Try to read the first few uncompressed bytes from the input file
315 * before blindly truncating the output file.
317 if ((nr = fread(buf, 1, sizeof(buf), ifp)) == 0) {
322 if ((ofp = fopen(out, "w")) == NULL ||
323 (nr != 0 && fwrite(buf, 1, nr, ofp) != nr)) {
331 while ((nr = fread(buf, 1, sizeof(buf), ifp)) != 0)
332 if (fwrite(buf, 1, nr, ofp) != nr) {
337 if (ferror(ifp) || fclose(ifp)) {
366 setfile(const char *name, struct stat *fs)
368 static struct timespec tspec[2];
370 fs->st_mode &= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
372 tspec[0] = fs->st_atim;
373 tspec[1] = fs->st_mtim;
374 if (utimensat(AT_FDCWD, name, tspec, 0))
375 cwarn("utimensat: %s", name);
378 * Changing the ownership probably won't succeed, unless we're root
379 * or POSIX_CHOWN_RESTRICTED is not set. Set uid/gid before setting
380 * the mode; current BSD behavior is to remove all setuid bits on
381 * chown. If chown fails, lose setuid/setgid bits.
383 if (chown(name, fs->st_uid, fs->st_gid)) {
385 cwarn("chown: %s", name);
386 fs->st_mode &= ~(S_ISUID|S_ISGID);
388 if (chmod(name, fs->st_mode) && errno != EOPNOTSUPP)
389 cwarn("chmod: %s", name);
391 if (chflags(name, fs->st_flags) && errno != EOPNOTSUPP)
392 cwarn("chflags: %s", name);
396 permission(const char *fname)
400 if (!isatty(fileno(stderr)))
402 (void)fprintf(stderr, "overwrite %s? ", fname);
403 first = ch = getchar();
404 while (ch != '\n' && ch != EOF)
406 return (first == 'y');
410 usage(int iscompress)
413 (void)fprintf(stderr,
414 "usage: compress [-cfv] [-b bits] [file ...]\n");
416 (void)fprintf(stderr,
417 "usage: uncompress [-c] [-b bits] [file ...]\n");
422 cwarnx(const char *fmt, ...)
433 cwarn(const char *fmt, ...)