]>
git.saurik.com Git - apple/file_cmds.git/blob - compress/compress.c
2 * Copyright (c) 1992, 1993
3 * The Regents of the University of California. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by the University of
16 * California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 static const char copyright
[] =
36 "@(#) Copyright (c) 1992, 1993\n\
37 The Regents of the University of California. All rights reserved.\n";
42 static char sccsid
[] = "@(#)compress.c 8.2 (Berkeley) 1/7/94";
46 #include <sys/cdefs.h>
47 __FBSDID("$FreeBSD: src/usr.bin/compress/compress.c,v 1.21 2003/06/14 13:41:31 trhodes Exp $");
49 #include <sys/param.h>
63 void compress(const char *, const char *, int);
64 void cwarn(const char *, ...) __printflike(1, 2);
65 void cwarnx(const char *, ...) __printflike(1, 2);
66 void decompress(const char *, const char *, int);
67 int permission(const char *);
68 void setfile(const char *, struct stat
*);
71 int eval
, force
, verbose
, cat
;
74 main(int argc
, char *argv
[])
76 enum {COMPRESS
, DECOMPRESS
} style
;
79 char *p
, newname
[MAXPATHLEN
];
84 if ((p
= rindex(argv
[0], '/')) == NULL
)
88 if (!strcmp(p
, "uncompress"))
90 else if (!strcmp(p
, "compress"))
92 else if (!strcmp(p
, "zcat")) {
96 errx(1, "unknown program name");
99 while ((ch
= getopt(argc
, argv
, "b:cdfv")) != -1)
102 bits
= strtol(optarg
, &p
, 10);
104 errx(1, "illegal bit count -- %s", optarg
);
109 case 'd': /* Backward compatible. */
120 usage(style
== COMPRESS
);
129 (void)compress("/dev/stdin", "/dev/stdout", bits
);
132 (void)decompress("/dev/stdin", "/dev/stdout", bits
);
138 if (cat
== 1 && argc
> 1)
139 errx(1, "the -c option permits only a single file argument");
141 for (; *argv
; ++argv
)
144 if (strcmp(*argv
, "-") == 0) {
146 compress("/dev/stdin", "/dev/stdout", bits
);
149 compress(*argv
, "/dev/stdout", bits
);
152 if ((p
= rindex(*argv
, '.')) != NULL
&&
154 cwarnx("%s: name already has trailing .Z",
159 if (len
> sizeof(newname
) - 3) {
160 cwarnx("%s: name too long", *argv
);
163 memmove(newname
, *argv
, len
);
165 newname
[len
+ 1] = 'Z';
166 newname
[len
+ 2] = '\0';
167 compress(*argv
, newname
, bits
);
170 if (strcmp(*argv
, "-") == 0) {
172 decompress("/dev/stdin", "/dev/stdout", bits
);
176 if ((p
= rindex(*argv
, '.')) == NULL
||
178 if (len
> sizeof(newname
) - 3) {
179 cwarnx("%s: name too long", *argv
);
182 memmove(newname
, *argv
, len
);
184 newname
[len
+ 1] = 'Z';
185 newname
[len
+ 2] = '\0';
187 cat
? "/dev/stdout" : *argv
, bits
);
189 if (len
- 2 > sizeof(newname
) - 1) {
190 cwarnx("%s: name too long", *argv
);
193 memmove(newname
, *argv
, len
- 2);
194 newname
[len
- 2] = '\0';
196 cat
? "/dev/stdout" : newname
, bits
);
204 compress(const char *in
, const char *out
, int bits
)
209 int exists
, isreg
, oreg
;
212 exists
= !stat(out
, &sb
);
213 if (!force
&& exists
&& S_ISREG(sb
.st_mode
) && !cat
&& !permission(out
)) {
214 cwarnx("%s already exists", out
);
217 isreg
= oreg
= !exists
|| S_ISREG(sb
.st_mode
);
220 if ((ifp
= fopen(in
, "r")) == NULL
) {
224 if (stat(in
, &isb
)) { /* DON'T FSTAT! */
228 if (!S_ISREG(isb
.st_mode
))
231 if ((ofp
= zopen(out
, "w", bits
)) == NULL
) {
235 while ((nr
= fread(buf
, 1, sizeof(buf
), ifp
)) != 0)
236 if (fwrite(buf
, 1, nr
, ofp
) != nr
) {
241 if (ferror(ifp
) || fclose(ifp
)) {
254 if (stat(out
, &sb
)) {
259 if (!force
&& sb
.st_size
>= isb
.st_size
) {
261 (void)fprintf(stderr
, "%s: file would grow; left unmodified\n",
275 (void)fprintf(stderr
, "%s: ", out
);
276 if (isb
.st_size
> sb
.st_size
)
277 (void)fprintf(stderr
, "%.0f%% compression\n",
278 ((float)sb
.st_size
/ isb
.st_size
) * 100.0);
280 (void)fprintf(stderr
, "%.0f%% expansion\n",
281 ((float)isb
.st_size
/ sb
.st_size
) * 100.0);
296 decompress(const char *in
, const char *out
, int bits
)
301 int exists
, isreg
, oreg
;
304 exists
= !stat(out
, &sb
);
305 if (!force
&& exists
&& S_ISREG(sb
.st_mode
) && !cat
&& !permission(out
)) {
306 cwarnx("%s already exists", out
);
309 isreg
= oreg
= !exists
|| S_ISREG(sb
.st_mode
);
312 if ((ifp
= zopen(in
, "r", bits
)) == NULL
) {
320 if (!S_ISREG(sb
.st_mode
))
324 * Try to read the first few uncompressed bytes from the input file
325 * before blindly truncating the output file.
327 if ((nr
= fread(buf
, 1, sizeof(buf
), ifp
)) == 0) {
332 if ((ofp
= fopen(out
, "w")) == NULL
||
333 (nr
!= 0 && fwrite(buf
, 1, nr
, ofp
) != nr
)) {
339 while ((nr
= fread(buf
, 1, sizeof(buf
), ifp
)) != 0)
340 if (fwrite(buf
, 1, nr
, ofp
) != nr
) {
345 if (ferror(ifp
) || fclose(ifp
)) {
362 struct stat isb
= sb
;
364 (void)fprintf(stderr
, "%s: ", out
);
365 if (isb
.st_size
> sb
.st_size
)
366 (void)fprintf(stderr
, "%.0f%% compression\n",
367 ((float)sb
.st_size
/ isb
.st_size
) * 100.0);
369 (void)fprintf(stderr
, "%.0f%% expansion\n",
370 ((float)isb
.st_size
/ sb
.st_size
) * 100.0);
385 setfile(const char *name
, struct stat
*fs
)
387 static struct timeval tv
[2];
389 fs
->st_mode
&= S_ISUID
|S_ISGID
|S_IRWXU
|S_IRWXG
|S_IRWXO
;
391 TIMESPEC_TO_TIMEVAL(&tv
[0], &fs
->st_atimespec
);
392 TIMESPEC_TO_TIMEVAL(&tv
[1], &fs
->st_mtimespec
);
393 if (utimes(name
, tv
))
394 cwarn("utimes: %s", name
);
397 * Changing the ownership probably won't succeed, unless we're root
398 * or POSIX_CHOWN_RESTRICTED is not set. Set uid/gid before setting
399 * the mode; current BSD behavior is to remove all setuid bits on
400 * chown. If chown fails, lose setuid/setgid bits.
402 if (chown(name
, fs
->st_uid
, fs
->st_gid
)) {
404 cwarn("chown: %s", name
);
405 fs
->st_mode
&= ~(S_ISUID
|S_ISGID
);
407 if (chmod(name
, fs
->st_mode
) && errno
!= ENOTSUP
)
408 cwarn("chmod: %s", name
);
410 if (chflags(name
, fs
->st_flags
) && errno
!= ENOTSUP
)
411 cwarn("chflags: %s", name
);
415 permission(const char *fname
)
419 if (!isatty(fileno(stderr
)))
421 (void)fprintf(stderr
, "overwrite %s? ", fname
);
422 first
= ch
= getchar();
423 while (ch
!= '\n' && ch
!= EOF
)
425 return (first
== 'y');
429 usage(int iscompress
)
432 (void)fprintf(stderr
,
433 "usage: compress [-cfv] [-b bits] [file ...]\n");
435 (void)fprintf(stderr
,
436 "usage: uncompress [-cfv] [-b bits] [file ...]\n");
441 cwarnx(const char *fmt
, ...)
452 cwarn(const char *fmt
, ...)