1 5993a8f2 2003-11-23 devnull #include <u.h>
  2 5993a8f2 2003-11-23 devnull #include <libc.h>
  3 5993a8f2 2003-11-23 devnull #include <bio.h>
  4 5993a8f2 2003-11-23 devnull #include "diff.h"
  5 5993a8f2 2003-11-23 devnull 
  6 5993a8f2 2003-11-23 devnull #define	DIRECTORY(s)		((s)->qid.type&QTDIR)
  7 5993a8f2 2003-11-23 devnull #define	REGULAR_FILE(s)		((s)->type == 'M' && !DIRECTORY(s))
  8 5993a8f2 2003-11-23 devnull 
  9 5993a8f2 2003-11-23 devnull Biobuf	stdout;
 10 5993a8f2 2003-11-23 devnull 
 11 5993a8f2 2003-11-23 devnull static char *tmp[] = {"/tmp/diff1", "/tmp/diff2"};
 12 5993a8f2 2003-11-23 devnull static int whichtmp;
 13 5993a8f2 2003-11-23 devnull static char *progname;
 14 5993a8f2 2003-11-23 devnull static char usage[] = "diff [ -efmnbwr ] file1 ... file2\n";
 15 5993a8f2 2003-11-23 devnull 
 16 5993a8f2 2003-11-23 devnull static void
 17 5993a8f2 2003-11-23 devnull rmtmpfiles(void)
 18 5993a8f2 2003-11-23 devnull {
 19 5993a8f2 2003-11-23 devnull 	while (whichtmp > 0) {
 20 5993a8f2 2003-11-23 devnull 		whichtmp--;
 21 5993a8f2 2003-11-23 devnull 		remove(tmp[whichtmp]);
 22 5993a8f2 2003-11-23 devnull 	}
 23 5993a8f2 2003-11-23 devnull }
 24 5993a8f2 2003-11-23 devnull 
 25 5993a8f2 2003-11-23 devnull void	
 26 5993a8f2 2003-11-23 devnull done(int status)
 27 5993a8f2 2003-11-23 devnull {
 28 5993a8f2 2003-11-23 devnull 	rmtmpfiles();
 29 8ad51794 2004-03-25 devnull Bflush(&stdout);
 30 8ad51794 2004-03-25 devnull Bterm(&stdout);
 31 5993a8f2 2003-11-23 devnull 	switch(status)
 32 5993a8f2 2003-11-23 devnull 	{
 33 5993a8f2 2003-11-23 devnull 	case 0:
 34 5993a8f2 2003-11-23 devnull 		exits("");
 35 5993a8f2 2003-11-23 devnull 	case 1:
 36 5993a8f2 2003-11-23 devnull 		exits("some");
 37 5993a8f2 2003-11-23 devnull 	default:
 38 5993a8f2 2003-11-23 devnull 		exits("error");
 39 5993a8f2 2003-11-23 devnull 	}
 40 5993a8f2 2003-11-23 devnull 	/*NOTREACHED*/
 41 5993a8f2 2003-11-23 devnull }
 42 5993a8f2 2003-11-23 devnull 
 43 5993a8f2 2003-11-23 devnull void
 44 5993a8f2 2003-11-23 devnull panic(int status, char *fmt, ...)
 45 5993a8f2 2003-11-23 devnull {
 46 5993a8f2 2003-11-23 devnull 	va_list arg;
 47 5993a8f2 2003-11-23 devnull 
 48 5993a8f2 2003-11-23 devnull 	Bflush(&stdout);
 49 5993a8f2 2003-11-23 devnull 
 50 5993a8f2 2003-11-23 devnull 	fprint(2, "%s: ", progname);
 51 5993a8f2 2003-11-23 devnull 	va_start(arg, fmt);
 52 5993a8f2 2003-11-23 devnull 	vfprint(2, fmt, arg);
 53 5993a8f2 2003-11-23 devnull 	va_end(arg);
 54 5993a8f2 2003-11-23 devnull 	if (status)
 55 5993a8f2 2003-11-23 devnull 		done(status);
 56 5993a8f2 2003-11-23 devnull 		/*NOTREACHED*/
 57 5993a8f2 2003-11-23 devnull }
 58 5993a8f2 2003-11-23 devnull 
 59 5993a8f2 2003-11-23 devnull static int
 60 5993a8f2 2003-11-23 devnull catch(void *a, char *msg)
 61 5993a8f2 2003-11-23 devnull {
 62 5993a8f2 2003-11-23 devnull 	USED(a);
 63 5993a8f2 2003-11-23 devnull 	panic(2, msg);
 64 5993a8f2 2003-11-23 devnull 	return 1;
 65 5993a8f2 2003-11-23 devnull }
 66 5993a8f2 2003-11-23 devnull 
 67 5993a8f2 2003-11-23 devnull int
 68 5993a8f2 2003-11-23 devnull mkpathname(char *pathname, char *path, char *name)
 69 5993a8f2 2003-11-23 devnull {
 70 5993a8f2 2003-11-23 devnull 	if (strlen(path) + strlen(name) > MAXPATHLEN) {
 71 5993a8f2 2003-11-23 devnull 		panic(0, "pathname %s/%s too long\n", path, name);
 72 5993a8f2 2003-11-23 devnull 		return 1;
 73 5993a8f2 2003-11-23 devnull 	}
 74 5993a8f2 2003-11-23 devnull 	sprint(pathname, "%s/%s", path, name);
 75 5993a8f2 2003-11-23 devnull 	return 0;
 76 5993a8f2 2003-11-23 devnull }
 77 5993a8f2 2003-11-23 devnull 	
 78 5993a8f2 2003-11-23 devnull static char *
 79 5993a8f2 2003-11-23 devnull mktmpfile(int input, Dir **sb)
 80 5993a8f2 2003-11-23 devnull {
 81 5993a8f2 2003-11-23 devnull 	int fd, i;
 82 5993a8f2 2003-11-23 devnull 	char *p;
 83 5993a8f2 2003-11-23 devnull 	char buf[8192];
 84 5993a8f2 2003-11-23 devnull 
 85 5993a8f2 2003-11-23 devnull 	atnotify(catch, 1);
 86 5993a8f2 2003-11-23 devnull 	p = tmp[whichtmp++];
 87 5993a8f2 2003-11-23 devnull 	fd = create(p, OWRITE, 0600);
 88 5993a8f2 2003-11-23 devnull 	if (fd < 0) {
 89 5993a8f2 2003-11-23 devnull 		panic(mflag ? 0: 2, "cannot create %s: %r\n", p);
 90 5993a8f2 2003-11-23 devnull 		return 0;
 91 5993a8f2 2003-11-23 devnull 	}
 92 5993a8f2 2003-11-23 devnull 	while ((i = read(input, buf, sizeof(buf))) > 0) {
 93 5993a8f2 2003-11-23 devnull 		if ((i = write(fd, buf, i)) < 0)
 94 5993a8f2 2003-11-23 devnull 			break;
 95 5993a8f2 2003-11-23 devnull 	}
 96 5993a8f2 2003-11-23 devnull 	*sb = dirfstat(fd);
 97 5993a8f2 2003-11-23 devnull 	close(fd);
 98 5993a8f2 2003-11-23 devnull 	if (i < 0) {
 99 5993a8f2 2003-11-23 devnull 		panic(mflag ? 0: 2, "cannot read/write %s: %r\n", p);
100 5993a8f2 2003-11-23 devnull 		return 0;
101 5993a8f2 2003-11-23 devnull 	}
102 5993a8f2 2003-11-23 devnull 	return p;
103 5993a8f2 2003-11-23 devnull }
104 5993a8f2 2003-11-23 devnull 
105 5993a8f2 2003-11-23 devnull static char *
106 5993a8f2 2003-11-23 devnull statfile(char *file, Dir **sb)
107 5993a8f2 2003-11-23 devnull {
108 5993a8f2 2003-11-23 devnull 	Dir *dir;
109 5993a8f2 2003-11-23 devnull 	int input;
110 5993a8f2 2003-11-23 devnull 
111 5993a8f2 2003-11-23 devnull 	dir = dirstat(file);
112 5993a8f2 2003-11-23 devnull 	if(dir == nil) {
113 5993a8f2 2003-11-23 devnull 		if (strcmp(file, "-") || (dir = dirfstat(0)) == nil) {
114 5993a8f2 2003-11-23 devnull 			panic(mflag ? 0: 2, "cannot stat %s: %r\n", file);
115 5993a8f2 2003-11-23 devnull 			return 0;
116 5993a8f2 2003-11-23 devnull 		}
117 5993a8f2 2003-11-23 devnull 		free(dir);
118 5993a8f2 2003-11-23 devnull 		return mktmpfile(0, sb);
119 5993a8f2 2003-11-23 devnull 	}
120 5993a8f2 2003-11-23 devnull 	else if (!REGULAR_FILE(dir) && !DIRECTORY(dir)) {
121 5993a8f2 2003-11-23 devnull 		free(dir);
122 5993a8f2 2003-11-23 devnull 		if ((input = open(file, OREAD)) == -1) {
123 5993a8f2 2003-11-23 devnull 			panic(mflag ? 0: 2, "cannot open %s: %r\n", file);
124 5993a8f2 2003-11-23 devnull 			return 0;
125 5993a8f2 2003-11-23 devnull 		}
126 5993a8f2 2003-11-23 devnull 		file = mktmpfile(input, sb);
127 5993a8f2 2003-11-23 devnull 		close(input);
128 5993a8f2 2003-11-23 devnull 	}
129 5993a8f2 2003-11-23 devnull 	else
130 5993a8f2 2003-11-23 devnull 		*sb = dir;
131 5993a8f2 2003-11-23 devnull 	return file;
132 5993a8f2 2003-11-23 devnull }
133 5993a8f2 2003-11-23 devnull 
134 5993a8f2 2003-11-23 devnull void
135 5993a8f2 2003-11-23 devnull diff(char *f, char *t, int level)
136 5993a8f2 2003-11-23 devnull {
137 5993a8f2 2003-11-23 devnull 	char *fp, *tp, *p, fb[MAXPATHLEN+1], tb[MAXPATHLEN+1];
138 5993a8f2 2003-11-23 devnull 	Dir *fsb, *tsb;
139 5993a8f2 2003-11-23 devnull 
140 5993a8f2 2003-11-23 devnull 	if ((fp = statfile(f, &fsb)) == 0)
141 5993a8f2 2003-11-23 devnull 		return;
142 5993a8f2 2003-11-23 devnull 	if ((tp = statfile(t, &tsb)) == 0){
143 5993a8f2 2003-11-23 devnull 		free(fsb);
144 5993a8f2 2003-11-23 devnull 		return;
145 5993a8f2 2003-11-23 devnull 	}
146 5993a8f2 2003-11-23 devnull 	if (DIRECTORY(fsb) && DIRECTORY(tsb)) {
147 5993a8f2 2003-11-23 devnull 		if (rflag || level == 0)
148 5993a8f2 2003-11-23 devnull 			diffdir(fp, tp, level);
149 5993a8f2 2003-11-23 devnull 		else
150 5993a8f2 2003-11-23 devnull 			Bprint(&stdout, "Common subdirectories: %s and %s\n",
151 5993a8f2 2003-11-23 devnull 				fp, tp);
152 5993a8f2 2003-11-23 devnull 	}
153 5993a8f2 2003-11-23 devnull 	else if (REGULAR_FILE(fsb) && REGULAR_FILE(tsb))
154 5993a8f2 2003-11-23 devnull 		diffreg(fp, tp);
155 5993a8f2 2003-11-23 devnull 	else {
156 5993a8f2 2003-11-23 devnull 		if (REGULAR_FILE(fsb)) {
157 5993a8f2 2003-11-23 devnull 			if ((p = utfrrune(f, '/')) == 0)
158 5993a8f2 2003-11-23 devnull 				p = f;
159 5993a8f2 2003-11-23 devnull 			else
160 5993a8f2 2003-11-23 devnull 				p++;
161 5993a8f2 2003-11-23 devnull 			if (mkpathname(tb, tp, p) == 0)
162 5993a8f2 2003-11-23 devnull 				diffreg(fp, tb);
163 5993a8f2 2003-11-23 devnull 		}
164 5993a8f2 2003-11-23 devnull 		else {
165 5993a8f2 2003-11-23 devnull 			if ((p = utfrrune(t, '/')) == 0)
166 5993a8f2 2003-11-23 devnull 				p = t;
167 5993a8f2 2003-11-23 devnull 			else
168 5993a8f2 2003-11-23 devnull 				p++;
169 5993a8f2 2003-11-23 devnull 			if (mkpathname(fb, fp, p) == 0)
170 5993a8f2 2003-11-23 devnull 				diffreg(fb, tp);
171 5993a8f2 2003-11-23 devnull 		}
172 5993a8f2 2003-11-23 devnull 	}
173 5993a8f2 2003-11-23 devnull 	free(fsb);
174 5993a8f2 2003-11-23 devnull 	free(tsb);
175 5993a8f2 2003-11-23 devnull }
176 5993a8f2 2003-11-23 devnull 
177 5993a8f2 2003-11-23 devnull void
178 5993a8f2 2003-11-23 devnull main(int argc, char *argv[])
179 5993a8f2 2003-11-23 devnull {
180 5993a8f2 2003-11-23 devnull 	char *p;
181 5993a8f2 2003-11-23 devnull 	int i;
182 5993a8f2 2003-11-23 devnull 	Dir *fsb, *tsb;
183 5993a8f2 2003-11-23 devnull 
184 5993a8f2 2003-11-23 devnull 	Binit(&stdout, 1, OWRITE);
185 5993a8f2 2003-11-23 devnull 	progname = *argv;
186 5993a8f2 2003-11-23 devnull 	while (--argc && (*++argv)[0] == '-' && (*argv)[1]) {
187 5993a8f2 2003-11-23 devnull 		for (p = *argv+1; *p; p++) {
188 5993a8f2 2003-11-23 devnull 			switch (*p) {
189 5993a8f2 2003-11-23 devnull 
190 5993a8f2 2003-11-23 devnull 			case 'e':
191 5993a8f2 2003-11-23 devnull 			case 'f':
192 5993a8f2 2003-11-23 devnull 			case 'n':
193 5993a8f2 2003-11-23 devnull 				mode = *p;
194 5993a8f2 2003-11-23 devnull 				break;
195 5993a8f2 2003-11-23 devnull 
196 5993a8f2 2003-11-23 devnull 			case 'w':
197 5993a8f2 2003-11-23 devnull 				bflag = 2;
198 5993a8f2 2003-11-23 devnull 				break;
199 5993a8f2 2003-11-23 devnull 
200 5993a8f2 2003-11-23 devnull 			case 'b':
201 5993a8f2 2003-11-23 devnull 				bflag = 1;
202 5993a8f2 2003-11-23 devnull 				break;
203 5993a8f2 2003-11-23 devnull 
204 5993a8f2 2003-11-23 devnull 			case 'r':
205 5993a8f2 2003-11-23 devnull 				rflag = 1;
206 5993a8f2 2003-11-23 devnull 				break;
207 5993a8f2 2003-11-23 devnull 
208 5993a8f2 2003-11-23 devnull 			case 'm':
209 5993a8f2 2003-11-23 devnull 				mflag = 1;	
210 5993a8f2 2003-11-23 devnull 				break;
211 5993a8f2 2003-11-23 devnull 
212 5993a8f2 2003-11-23 devnull 			case 'h':
213 5993a8f2 2003-11-23 devnull 			default:
214 5993a8f2 2003-11-23 devnull 				progname = "Usage";
215 5993a8f2 2003-11-23 devnull 				panic(2, usage);
216 5993a8f2 2003-11-23 devnull 			}
217 5993a8f2 2003-11-23 devnull 		}
218 5993a8f2 2003-11-23 devnull 	}
219 5993a8f2 2003-11-23 devnull 	if (argc < 2)
220 5993a8f2 2003-11-23 devnull 		panic(2, usage, progname);
221 5993a8f2 2003-11-23 devnull 	if ((tsb = dirstat(argv[argc-1])) == nil)
222 5993a8f2 2003-11-23 devnull 		panic(2, "can't stat %s\n", argv[argc-1]);
223 5993a8f2 2003-11-23 devnull 	if (argc > 2) {
224 5993a8f2 2003-11-23 devnull 		if (!DIRECTORY(tsb))
225 5993a8f2 2003-11-23 devnull 			panic(2, usage, progname);
226 5993a8f2 2003-11-23 devnull 		mflag = 1;
227 5993a8f2 2003-11-23 devnull 	}
228 5993a8f2 2003-11-23 devnull 	else {
229 5993a8f2 2003-11-23 devnull 		if ((fsb = dirstat(argv[0])) == nil)
230 5993a8f2 2003-11-23 devnull 			panic(2, "can't stat %s\n", argv[0]);
231 5993a8f2 2003-11-23 devnull 		if (DIRECTORY(fsb) && DIRECTORY(tsb))
232 5993a8f2 2003-11-23 devnull 			mflag = 1;
233 5993a8f2 2003-11-23 devnull 		free(fsb);
234 5993a8f2 2003-11-23 devnull 	}
235 5993a8f2 2003-11-23 devnull 	free(tsb);
236 5993a8f2 2003-11-23 devnull 	for (i = 0; i < argc-1; i++) {
237 5993a8f2 2003-11-23 devnull 		diff(argv[i], argv[argc-1], 0);
238 5993a8f2 2003-11-23 devnull 		rmtmpfiles();
239 5993a8f2 2003-11-23 devnull 	}
240 5993a8f2 2003-11-23 devnull 	done(anychange);
241 5993a8f2 2003-11-23 devnull 	/*NOTREACHED*/
242 5993a8f2 2003-11-23 devnull }
243 5993a8f2 2003-11-23 devnull 
244 5993a8f2 2003-11-23 devnull static char noroom[] = "out of memory - try diff -h\n";
245 5993a8f2 2003-11-23 devnull 
246 5993a8f2 2003-11-23 devnull void *
247 5993a8f2 2003-11-23 devnull emalloc(unsigned n)
248 5993a8f2 2003-11-23 devnull {
249 5993a8f2 2003-11-23 devnull 	register void *p;
250 5993a8f2 2003-11-23 devnull 
251 5993a8f2 2003-11-23 devnull 	if ((p = malloc(n)) == 0)
252 5993a8f2 2003-11-23 devnull 		panic(2, noroom);
253 5993a8f2 2003-11-23 devnull 	return p;
254 5993a8f2 2003-11-23 devnull }
255 5993a8f2 2003-11-23 devnull 
256 5993a8f2 2003-11-23 devnull void *
257 5993a8f2 2003-11-23 devnull erealloc(void *p, unsigned n)
258 5993a8f2 2003-11-23 devnull {
259 5993a8f2 2003-11-23 devnull 	register void *rp;
260 5993a8f2 2003-11-23 devnull 
261 5993a8f2 2003-11-23 devnull 	if ((rp = realloc(p, n)) == 0)
262 5993a8f2 2003-11-23 devnull 		panic(2, noroom);
263 5993a8f2 2003-11-23 devnull 	return rp;
264 5993a8f2 2003-11-23 devnull }