1 #include <u.h>
2 #include <libc.h>
3 #include <bio.h>
4 #include "bzlib.h"
5 
6 static	int	bzipf(char*, int);
7 static	int	bzip(char*, long, int, Biobuf*);
8 
9 static	Biobuf	bout;
10 static	int	level;
11 static	int	debug;
12 static	int	verbose;
13 
14 
15 static void
16 usage(void)
17 {
18 	fprint(2, "usage: bzip2 [-vcD] [-1-9] [file ...]\n");
19 	exits("usage");
20 }
21 
22 void
23 main(int argc, char **argv)
24 {
25 	int i, ok, stdout;
26 
27 	level = 6;
28 	stdout = 0;
29 	ARGBEGIN{
30 	case 'D':
31 		debug++;
32 		break;
33 	case 'v':
34 		verbose++;
35 		break;
36 	case 'c':
37 		stdout++;
38 		break;
39 	case '1': case '2': case '3': case '4':
40 	case '5': case '6': case '7': case '8': case '9':
41 		level = ARGC() - '0';
42 		break;
43 	default:
44 		usage();
45 		break;
46 	}ARGEND
47 
48 	if(argc == 0){
49 		Binit(&bout, 1, OWRITE);
50 		ok = bzip(nil, time(0), 0, &bout);
51 		Bterm(&bout);
52 	}else{
53 		ok = 1;
54 		for(i = 0; i < argc; i++)
55 			ok &= bzipf(argv[i], stdout);
56 	}
57 	exits(ok ? nil: "errors");
58 }
59 
60 static int
61 bzipf(char *file, int stdout)
62 {
63 	Dir *dir;
64 	char ofile[128], *f, *s;
65 	int ifd, ofd, ok;
66 
67 	ifd = open(file, OREAD);
68 	if(ifd < 0){
69 		fprint(2, "bzip2: can't open %s: %r\n", file);
70 		return 0;
71 	}
72 	dir = dirfstat(ifd);
73 	if(dir == nil){
74 		fprint(2, "bzip2: can't stat %s: %r\n", file);
75 		close(ifd);
76 		return 0;
77 	}
78 	if(dir->mode & DMDIR){
79 		fprint(2, "bzip2: can't compress a directory\n");
80 		close(ifd);
81 		free(dir);
82 		return 0;
83 	}
84 
85 	if(stdout){
86 		ofd = 1;
87 		strcpy(ofile, "<stdout>");
88 	}else{
89 		f = strrchr(file, '/');
90 		if(f != nil)
91 			f++;
92 		else
93 			f = file;
94 		s = strrchr(f, '.');
95 		if(s != nil && s != ofile && strcmp(s, ".tar") == 0){
96 			*s = '\0';
97 			snprint(ofile, sizeof(ofile), "%s.tbz", f);
98 		}else
99 			snprint(ofile, sizeof(ofile), "%s.bz2", f);
100 		ofd = create(ofile, OWRITE, 0666);
101 		if(ofd < 0){
102 			fprint(2, "bzip2: can't open %s: %r\n", ofile);
103 			free(dir);
104 			close(ifd);
105 			return 0;
106 		}
107 	}
108 
109 	if(verbose)
110 		fprint(2, "compressing %s to %s\n", file, ofile);
111 
112 	Binit(&bout, ofd, OWRITE);
113 	ok = bzip(file, dir->mtime, ifd, &bout);
114 	if(!ok || Bflush(&bout) < 0){
115 		fprint(2, "bzip2: error writing %s: %r\n", ofile);
116 		if(!stdout)
117 			remove(ofile);
118 	}
119 	Bterm(&bout);
120 	free(dir);
121 	close(ifd);
122 	close(ofd);
123 	return ok;
124 }
125 
126 static int
127 bzip(char *file, long mtime, int ifd, Biobuf *bout)
128 {
129 	int e, n, done, onemore;
130 	char buf[8192];
131 	char obuf[8192];
132 	Biobuf bin;
133 	bz_stream strm;
134 
135 	USED(file);
136 	USED(mtime);
137 
138 	memset(&strm, 0, sizeof strm);
139 	BZ2_bzCompressInit(&strm, level, verbose, 0);
140 
141 	strm.next_in = buf;
142 	strm.avail_in = 0;
143 	strm.next_out = obuf;
144 	strm.avail_out = sizeof obuf;
145 
146 	done = 0;
147 	Binit(&bin, ifd, OREAD);
148 
149 	/*
150 	 * onemore is a crummy hack to go 'round the loop
151 	 * once after we finish, to flush the output buffer.
152 	 */
153 	onemore = 1;
154 	SET(e);
155 	do {
156 		if(!done && strm.avail_in < sizeof buf) {
157 			if(strm.avail_in)
158 				memmove(buf, strm.next_in, strm.avail_in);
159 			
160 			n = Bread(&bin, buf+strm.avail_in, sizeof(buf)-strm.avail_in);
161 			if(n <= 0)
162 				done = 1;
163 			else
164 				strm.avail_in += n;
165 			strm.next_in = buf;
166 		}
167 		if(strm.avail_out < sizeof obuf) {
168 			Bwrite(bout, obuf, sizeof(obuf)-strm.avail_out);
169 			strm.next_out = obuf;
170 			strm.avail_out = sizeof obuf;
171 		}
172 
173 		if(onemore == 0)
174 			break;
175 	} while((e=BZ2_bzCompress(&strm, done ? BZ_FINISH : BZ_RUN)) == BZ_RUN_OK || e == BZ_FINISH_OK || onemore--);
176 
177 	if(e != BZ_STREAM_END) {
178 		fprint(2, "bzip2: compress failed\n");
179 		return 0;
180 	}
181 
182 	if(BZ2_bzCompressEnd(&strm) != BZ_OK) {
183 		fprint(2, "bzip2: compress end failed (can't happen)\n");
184 		return 0;
185 	}
186 
187 	Bterm(&bin);
188 
189 	return 1;
190 }