1 ff3adf60 2004-04-14 devnull #include <u.h>
2 ff3adf60 2004-04-14 devnull #include <libc.h>
3 ff3adf60 2004-04-14 devnull #include <bio.h>
4 ff3adf60 2004-04-14 devnull #include <flate.h>
5 ff3adf60 2004-04-14 devnull #include "gzip.h"
7 ff3adf60 2004-04-14 devnull static int gzipf(char*, int);
8 ff3adf60 2004-04-14 devnull static int gzip(char*, long, int, Biobuf*);
9 ff3adf60 2004-04-14 devnull static int crcread(void *fd, void *buf, int n);
10 ff3adf60 2004-04-14 devnull static int gzwrite(void *bout, void *buf, int n);
12 ff3adf60 2004-04-14 devnull static Biobuf bout;
13 ff3adf60 2004-04-14 devnull static ulong crc;
14 ff3adf60 2004-04-14 devnull static ulong *crctab;
15 ff3adf60 2004-04-14 devnull static int debug;
16 ff3adf60 2004-04-14 devnull static int eof;
17 ff3adf60 2004-04-14 devnull static int level;
18 ff3adf60 2004-04-14 devnull static ulong totr;
19 ff3adf60 2004-04-14 devnull static int verbose;
22 ff3adf60 2004-04-14 devnull usage(void)
24 ff3adf60 2004-04-14 devnull fprint(2, "usage: gzip [-vcD] [-1-9] [file ...]\n");
25 ff3adf60 2004-04-14 devnull exits("usage");
29 ff3adf60 2004-04-14 devnull main(int argc, char *argv[])
31 ff3adf60 2004-04-14 devnull int i, ok, stdout;
32 892de798 2004-04-19 devnull char **oargv;
34 892de798 2004-04-19 devnull oargv = argv;
35 ff3adf60 2004-04-14 devnull level = 6;
36 ff3adf60 2004-04-14 devnull stdout = 0;
37 ff3adf60 2004-04-14 devnull ARGBEGIN{
38 ff3adf60 2004-04-14 devnull case 'D':
41 892de798 2004-04-19 devnull case 'd':
43 892de798 2004-04-19 devnull * gnu tar expects gzip -d to decompress
44 892de798 2004-04-19 devnull * humor it. ugh.
46 892de798 2004-04-19 devnull /* remove -d from command line - magic! */
47 892de798 2004-04-19 devnull if(strcmp(argv[0], "-d") == 0){
48 892de798 2004-04-19 devnull while(*argv++)
49 892de798 2004-04-19 devnull *(argv-1) = *argv;
51 892de798 2004-04-19 devnull memmove(_args-1, _args, strlen(_args)+1);
52 892de798 2004-04-19 devnull exec("gunzip", oargv);
53 892de798 2004-04-19 devnull sysfatal("exec gunzip failed");
55 b1455f33 2004-04-30 devnull case 'f':
56 b1455f33 2004-04-30 devnull /* force */
58 ff3adf60 2004-04-14 devnull case 'v':
59 ff3adf60 2004-04-14 devnull verbose++;
61 ff3adf60 2004-04-14 devnull case 'c':
62 ff3adf60 2004-04-14 devnull stdout = 1;
64 ff3adf60 2004-04-14 devnull case '1': case '2': case '3': case '4':
65 ff3adf60 2004-04-14 devnull case '5': case '6': case '7': case '8': case '9':
66 ff3adf60 2004-04-14 devnull level = ARGC() - '0';
73 ff3adf60 2004-04-14 devnull crctab = mkcrctab(GZCRCPOLY);
74 ff3adf60 2004-04-14 devnull ok = deflateinit();
75 ff3adf60 2004-04-14 devnull if(ok != FlateOk)
76 ff3adf60 2004-04-14 devnull sysfatal("deflateinit failed: %s\n", flateerr(ok));
78 ff3adf60 2004-04-14 devnull if(argc == 0){
79 ff3adf60 2004-04-14 devnull Binit(&bout, 1, OWRITE);
80 ff3adf60 2004-04-14 devnull ok = gzip(nil, time(0), 0, &bout);
81 ff3adf60 2004-04-14 devnull Bterm(&bout);
84 ff3adf60 2004-04-14 devnull for(i = 0; i < argc; i++)
85 ff3adf60 2004-04-14 devnull ok &= gzipf(argv[i], stdout);
87 ff3adf60 2004-04-14 devnull exits(ok ? nil: "errors");
90 ff3adf60 2004-04-14 devnull static int
91 ff3adf60 2004-04-14 devnull gzipf(char *file, int stdout)
93 ff3adf60 2004-04-14 devnull Dir *dir;
94 ff3adf60 2004-04-14 devnull char ofile[256], *f, *s;
95 ff3adf60 2004-04-14 devnull int ifd, ofd, ok;
97 ff3adf60 2004-04-14 devnull ifd = open(file, OREAD);
98 ff3adf60 2004-04-14 devnull if(ifd < 0){
99 ff3adf60 2004-04-14 devnull fprint(2, "gzip: can't open %s: %r\n", file);
100 ff3adf60 2004-04-14 devnull return 0;
102 ff3adf60 2004-04-14 devnull dir = dirfstat(ifd);
103 ff3adf60 2004-04-14 devnull if(dir == nil){
104 ff3adf60 2004-04-14 devnull fprint(2, "gzip: can't stat %s: %r\n", file);
105 ff3adf60 2004-04-14 devnull close(ifd);
106 ff3adf60 2004-04-14 devnull return 0;
108 ff3adf60 2004-04-14 devnull if(dir->mode & DMDIR){
109 ff3adf60 2004-04-14 devnull fprint(2, "gzip: can't compress a directory\n");
110 ff3adf60 2004-04-14 devnull close(ifd);
111 ff3adf60 2004-04-14 devnull free(dir);
112 ff3adf60 2004-04-14 devnull return 0;
115 ff3adf60 2004-04-14 devnull if(stdout){
116 ff3adf60 2004-04-14 devnull ofd = 1;
117 ff3adf60 2004-04-14 devnull strcpy(ofile, "<stdout>");
119 ff3adf60 2004-04-14 devnull f = strrchr(file, '/');
120 ff3adf60 2004-04-14 devnull if(f != nil)
123 ff3adf60 2004-04-14 devnull f = file;
124 ff3adf60 2004-04-14 devnull s = strrchr(f, '.');
125 ff3adf60 2004-04-14 devnull if(s != nil && s != ofile && strcmp(s, ".tar") == 0){
126 ff3adf60 2004-04-14 devnull *s = '\0';
127 ff3adf60 2004-04-14 devnull snprint(ofile, sizeof(ofile), "%s.tgz", f);
129 ff3adf60 2004-04-14 devnull snprint(ofile, sizeof(ofile), "%s.gz", f);
130 ff3adf60 2004-04-14 devnull ofd = create(ofile, OWRITE, 0666);
131 ff3adf60 2004-04-14 devnull if(ofd < 0){
132 ff3adf60 2004-04-14 devnull fprint(2, "gzip: can't open %s: %r\n", ofile);
133 ff3adf60 2004-04-14 devnull close(ifd);
134 ff3adf60 2004-04-14 devnull return 0;
138 ff3adf60 2004-04-14 devnull if(verbose)
139 ff3adf60 2004-04-14 devnull fprint(2, "compressing %s to %s\n", file, ofile);
141 ff3adf60 2004-04-14 devnull Binit(&bout, ofd, OWRITE);
142 ff3adf60 2004-04-14 devnull ok = gzip(file, dir->mtime, ifd, &bout);
143 ff3adf60 2004-04-14 devnull if(!ok || Bflush(&bout) < 0){
144 ff3adf60 2004-04-14 devnull fprint(2, "gzip: error writing %s: %r\n", ofile);
145 ff3adf60 2004-04-14 devnull if(!stdout)
146 ff3adf60 2004-04-14 devnull remove(ofile);
148 ff3adf60 2004-04-14 devnull Bterm(&bout);
149 ff3adf60 2004-04-14 devnull free(dir);
150 ff3adf60 2004-04-14 devnull close(ifd);
151 ff3adf60 2004-04-14 devnull close(ofd);
152 ff3adf60 2004-04-14 devnull return ok;
155 ff3adf60 2004-04-14 devnull static int
156 ff3adf60 2004-04-14 devnull gzip(char *file, long mtime, int ifd, Biobuf *bout)
158 ff3adf60 2004-04-14 devnull int flags, err;
160 ff3adf60 2004-04-14 devnull flags = 0;
161 ff3adf60 2004-04-14 devnull Bputc(bout, GZMAGIC1);
162 ff3adf60 2004-04-14 devnull Bputc(bout, GZMAGIC2);
163 ff3adf60 2004-04-14 devnull Bputc(bout, GZDEFLATE);
165 ff3adf60 2004-04-14 devnull if(file != nil)
166 ff3adf60 2004-04-14 devnull flags |= GZFNAME;
167 ff3adf60 2004-04-14 devnull Bputc(bout, flags);
169 ff3adf60 2004-04-14 devnull Bputc(bout, mtime);
170 ff3adf60 2004-04-14 devnull Bputc(bout, mtime>>8);
171 ff3adf60 2004-04-14 devnull Bputc(bout, mtime>>16);
172 ff3adf60 2004-04-14 devnull Bputc(bout, mtime>>24);
174 ff3adf60 2004-04-14 devnull Bputc(bout, 0);
175 ff3adf60 2004-04-14 devnull Bputc(bout, GZOSINFERNO);
177 ff3adf60 2004-04-14 devnull if(flags & GZFNAME)
178 ff3adf60 2004-04-14 devnull Bwrite(bout, file, strlen(file)+1);
180 ff3adf60 2004-04-14 devnull crc = 0;
181 ff3adf60 2004-04-14 devnull eof = 0;
182 ff3adf60 2004-04-14 devnull totr = 0;
183 ff3adf60 2004-04-14 devnull err = deflate(bout, gzwrite, (void*)ifd, crcread, level, debug);
184 ff3adf60 2004-04-14 devnull if(err != FlateOk){
185 ff3adf60 2004-04-14 devnull fprint(2, "gzip: deflate failed: %s\n", flateerr(err));
186 ff3adf60 2004-04-14 devnull return 0;
189 ff3adf60 2004-04-14 devnull Bputc(bout, crc);
190 ff3adf60 2004-04-14 devnull Bputc(bout, crc>>8);
191 ff3adf60 2004-04-14 devnull Bputc(bout, crc>>16);
192 ff3adf60 2004-04-14 devnull Bputc(bout, crc>>24);
194 ff3adf60 2004-04-14 devnull Bputc(bout, totr);
195 ff3adf60 2004-04-14 devnull Bputc(bout, totr>>8);
196 ff3adf60 2004-04-14 devnull Bputc(bout, totr>>16);
197 ff3adf60 2004-04-14 devnull Bputc(bout, totr>>24);
199 ff3adf60 2004-04-14 devnull return 1;
202 ff3adf60 2004-04-14 devnull static int
203 ff3adf60 2004-04-14 devnull crcread(void *fd, void *buf, int n)
205 ff3adf60 2004-04-14 devnull int nr, m;
208 ff3adf60 2004-04-14 devnull for(; !eof && n > 0; n -= m){
209 ff3adf60 2004-04-14 devnull m = read((int)fd, (char*)buf+nr, n);
210 ff3adf60 2004-04-14 devnull if(m <= 0){
211 ff3adf60 2004-04-14 devnull eof = 1;
212 ff3adf60 2004-04-14 devnull if(m < 0)
213 ff3adf60 2004-04-14 devnull return -1;
216 ff3adf60 2004-04-14 devnull nr += m;
218 ff3adf60 2004-04-14 devnull crc = blockcrc(crctab, crc, buf, nr);
219 ff3adf60 2004-04-14 devnull totr += nr;
220 ff3adf60 2004-04-14 devnull return nr;
223 ff3adf60 2004-04-14 devnull static int
224 ff3adf60 2004-04-14 devnull gzwrite(void *bout, void *buf, int n)
226 ff3adf60 2004-04-14 devnull if(n != Bwrite(bout, buf, n)){
227 ff3adf60 2004-04-14 devnull eof = 1;
228 ff3adf60 2004-04-14 devnull return -1;
230 ff3adf60 2004-04-14 devnull return n;