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