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 <ctype.h>
  5 5993a8f2 2003-11-23 devnull #include "diff.h"
  6 5993a8f2 2003-11-23 devnull 
  7 5993a8f2 2003-11-23 devnull struct line {
  8 5993a8f2 2003-11-23 devnull 	int	serial;
  9 5993a8f2 2003-11-23 devnull 	int	value;
 10 5993a8f2 2003-11-23 devnull };
 11 5993a8f2 2003-11-23 devnull extern struct line *file[2];
 12 5993a8f2 2003-11-23 devnull extern int len[2];
 13 5993a8f2 2003-11-23 devnull extern long *ixold, *ixnew;
 14 5993a8f2 2003-11-23 devnull extern int *J;
 15 5993a8f2 2003-11-23 devnull 
 16 5993a8f2 2003-11-23 devnull static Biobuf *input[2];
 17 5993a8f2 2003-11-23 devnull static char *file1, *file2;
 18 5993a8f2 2003-11-23 devnull static int firstchange;
 19 5993a8f2 2003-11-23 devnull 
 20 5993a8f2 2003-11-23 devnull #define MAXLINELEN	4096
 21 5993a8f2 2003-11-23 devnull #define MIN(x, y)	((x) < (y) ? (x): (y))
 22 5993a8f2 2003-11-23 devnull 
 23 5993a8f2 2003-11-23 devnull static int
 24 5993a8f2 2003-11-23 devnull readline(Biobuf *bp, char *buf)
 25 5993a8f2 2003-11-23 devnull {
 26 5993a8f2 2003-11-23 devnull 	int c;
 27 5993a8f2 2003-11-23 devnull 	char *p, *e;
 28 5993a8f2 2003-11-23 devnull 
 29 5993a8f2 2003-11-23 devnull 	p = buf;
 30 5993a8f2 2003-11-23 devnull 	e = p + MAXLINELEN-1;
 31 5993a8f2 2003-11-23 devnull 	do {
 32 5993a8f2 2003-11-23 devnull 		c = Bgetc(bp);
 33 5993a8f2 2003-11-23 devnull 		if (c < 0) {
 34 5993a8f2 2003-11-23 devnull 			if (p == buf)
 35 5993a8f2 2003-11-23 devnull 				return -1;
 36 5993a8f2 2003-11-23 devnull 			break;
 37 5993a8f2 2003-11-23 devnull 		}
 38 5993a8f2 2003-11-23 devnull 		if (c == '\n')
 39 5993a8f2 2003-11-23 devnull 			break;
 40 5993a8f2 2003-11-23 devnull 		*p++ = c;
 41 5993a8f2 2003-11-23 devnull 	} while (p < e);
 42 5993a8f2 2003-11-23 devnull 	*p = 0;
 43 5993a8f2 2003-11-23 devnull 	if (c != '\n' && c >= 0) {
 44 5993a8f2 2003-11-23 devnull 		do c = Bgetc(bp);
 45 5993a8f2 2003-11-23 devnull 		while (c >= 0 && c != '\n');
 46 5993a8f2 2003-11-23 devnull 	}
 47 5993a8f2 2003-11-23 devnull 	return p - buf;
 48 5993a8f2 2003-11-23 devnull }
 49 5993a8f2 2003-11-23 devnull 
 50 5993a8f2 2003-11-23 devnull #define HALFLONG 16
 51 5993a8f2 2003-11-23 devnull #define low(x)	(x&((1L<<HALFLONG)-1))
 52 5993a8f2 2003-11-23 devnull #define high(x)	(x>>HALFLONG)
 53 5993a8f2 2003-11-23 devnull 
 54 5993a8f2 2003-11-23 devnull /*
 55 5993a8f2 2003-11-23 devnull  * hashing has the effect of
 56 5993a8f2 2003-11-23 devnull  * arranging line in 7-bit bytes and then
 57 5993a8f2 2003-11-23 devnull  * summing 1-s complement in 16-bit hunks 
 58 5993a8f2 2003-11-23 devnull  */
 59 5993a8f2 2003-11-23 devnull static int
 60 5993a8f2 2003-11-23 devnull readhash(Biobuf *bp, char *buf)
 61 5993a8f2 2003-11-23 devnull {
 62 5993a8f2 2003-11-23 devnull 	long sum;
 63 5993a8f2 2003-11-23 devnull 	unsigned shift;
 64 5993a8f2 2003-11-23 devnull 	char *p;
 65 5993a8f2 2003-11-23 devnull 	int len, space;
 66 5993a8f2 2003-11-23 devnull 
 67 5993a8f2 2003-11-23 devnull 	sum = 1;
 68 5993a8f2 2003-11-23 devnull 	shift = 0;
 69 5993a8f2 2003-11-23 devnull 	if ((len = readline(bp, buf)) == -1)
 70 5993a8f2 2003-11-23 devnull 		return 0;
 71 5993a8f2 2003-11-23 devnull 	p = buf;
 72 5993a8f2 2003-11-23 devnull 	switch(bflag)	/* various types of white space handling */
 73 5993a8f2 2003-11-23 devnull 	{
 74 5993a8f2 2003-11-23 devnull 	case 0:
 75 5993a8f2 2003-11-23 devnull 		while (len--) {
 76 5993a8f2 2003-11-23 devnull 			sum += (long)*p++ << (shift &= (HALFLONG-1));
 77 5993a8f2 2003-11-23 devnull 			shift += 7;
 78 5993a8f2 2003-11-23 devnull 		}
 79 5993a8f2 2003-11-23 devnull 		break;
 80 5993a8f2 2003-11-23 devnull 	case 1:
 81 5993a8f2 2003-11-23 devnull 		/*
 82 5993a8f2 2003-11-23 devnull 		 * coalesce multiple white-space
 83 5993a8f2 2003-11-23 devnull 		 */
 84 5993a8f2 2003-11-23 devnull 		for (space = 0; len--; p++) {
 85 5993a8f2 2003-11-23 devnull 			if (isspace(*p)) {
 86 5993a8f2 2003-11-23 devnull 				space++;
 87 5993a8f2 2003-11-23 devnull 				continue;
 88 5993a8f2 2003-11-23 devnull 			}
 89 5993a8f2 2003-11-23 devnull 			if (space) {
 90 5993a8f2 2003-11-23 devnull 				shift += 7;
 91 5993a8f2 2003-11-23 devnull 				space = 0;
 92 5993a8f2 2003-11-23 devnull 			}
 93 5993a8f2 2003-11-23 devnull 			sum += (long)*p << (shift &= (HALFLONG-1));
 94 5993a8f2 2003-11-23 devnull 			shift += 7;
 95 5993a8f2 2003-11-23 devnull 		}
 96 5993a8f2 2003-11-23 devnull 		break;
 97 5993a8f2 2003-11-23 devnull 	default:
 98 5993a8f2 2003-11-23 devnull 		/*
 99 5993a8f2 2003-11-23 devnull 		 * strip all white-space
100 5993a8f2 2003-11-23 devnull 		 */
101 5993a8f2 2003-11-23 devnull 		while (len--) {
102 5993a8f2 2003-11-23 devnull 			if (isspace(*p)) {
103 5993a8f2 2003-11-23 devnull 				p++;
104 5993a8f2 2003-11-23 devnull 				continue;
105 5993a8f2 2003-11-23 devnull 			}
106 5993a8f2 2003-11-23 devnull 			sum += (long)*p++ << (shift &= (HALFLONG-1));
107 5993a8f2 2003-11-23 devnull 			shift += 7;
108 5993a8f2 2003-11-23 devnull 		}
109 5993a8f2 2003-11-23 devnull 		break;
110 5993a8f2 2003-11-23 devnull 	}
111 5993a8f2 2003-11-23 devnull 	sum = low(sum) + high(sum);
112 5993a8f2 2003-11-23 devnull 	return ((short)low(sum) + (short)high(sum));
113 5993a8f2 2003-11-23 devnull }
114 5993a8f2 2003-11-23 devnull 
115 5993a8f2 2003-11-23 devnull Biobuf *
116 5993a8f2 2003-11-23 devnull prepare(int i, char *arg)
117 5993a8f2 2003-11-23 devnull {
118 5993a8f2 2003-11-23 devnull 	struct line *p;
119 5993a8f2 2003-11-23 devnull 	int j, h;
120 5993a8f2 2003-11-23 devnull 	Biobuf *bp;
121 5993a8f2 2003-11-23 devnull 	char *cp, buf[MAXLINELEN];
122 5993a8f2 2003-11-23 devnull 	int nbytes;
123 5993a8f2 2003-11-23 devnull 	Rune r;
124 5993a8f2 2003-11-23 devnull 
125 5993a8f2 2003-11-23 devnull 	bp = Bopen(arg, OREAD);
126 5993a8f2 2003-11-23 devnull 	if (!bp) {
127 5993a8f2 2003-11-23 devnull 		panic(mflag ? 0: 2, "cannot open %s: %r\n", arg);
128 5993a8f2 2003-11-23 devnull 		return 0;
129 5993a8f2 2003-11-23 devnull 	}
130 5993a8f2 2003-11-23 devnull 	if (binary)
131 5993a8f2 2003-11-23 devnull 		return bp;
132 5993a8f2 2003-11-23 devnull 	nbytes = Bread(bp, buf, MIN(1024, MAXLINELEN));
133 5993a8f2 2003-11-23 devnull 	if (nbytes > 0) {
134 5993a8f2 2003-11-23 devnull 		cp = buf;
135 5993a8f2 2003-11-23 devnull 		while (cp < buf+nbytes-UTFmax) {
136 5993a8f2 2003-11-23 devnull 			/*
137 5993a8f2 2003-11-23 devnull 			 * heuristic for a binary file in the
138 5993a8f2 2003-11-23 devnull 			 * brave new UNICODE world
139 5993a8f2 2003-11-23 devnull 			 */
140 5993a8f2 2003-11-23 devnull 			cp += chartorune(&r, cp);
141 5993a8f2 2003-11-23 devnull 			if (r == 0 || (r > 0x7f && r <= 0xa0)) {
142 5993a8f2 2003-11-23 devnull 				binary++;
143 5993a8f2 2003-11-23 devnull 				return bp;
144 5993a8f2 2003-11-23 devnull 			}
145 5993a8f2 2003-11-23 devnull 		}
146 5993a8f2 2003-11-23 devnull 		Bseek(bp, 0, 0);
147 5993a8f2 2003-11-23 devnull 	}
148 5993a8f2 2003-11-23 devnull 	p = MALLOC(struct line, 3);
149 5993a8f2 2003-11-23 devnull 	for (j = 0; h = readhash(bp, buf); p[j].value = h)
150 5993a8f2 2003-11-23 devnull 		p = REALLOC(p, struct line, (++j+3));
151 5993a8f2 2003-11-23 devnull 	len[i] = j;
152 5993a8f2 2003-11-23 devnull 	file[i] = p;
153 5993a8f2 2003-11-23 devnull 	input[i] = bp;			/*fix*/
154 5993a8f2 2003-11-23 devnull 	if (i == 0) {			/*fix*/
155 5993a8f2 2003-11-23 devnull 		file1 = arg;
156 5993a8f2 2003-11-23 devnull 		firstchange = 0;
157 5993a8f2 2003-11-23 devnull 	}
158 5993a8f2 2003-11-23 devnull 	else
159 5993a8f2 2003-11-23 devnull 		file2 = arg;
160 5993a8f2 2003-11-23 devnull 	return bp;
161 5993a8f2 2003-11-23 devnull }
162 5993a8f2 2003-11-23 devnull 
163 5993a8f2 2003-11-23 devnull static int
164 5993a8f2 2003-11-23 devnull squishspace(char *buf)
165 5993a8f2 2003-11-23 devnull {
166 5993a8f2 2003-11-23 devnull 	char *p, *q;
167 5993a8f2 2003-11-23 devnull 	int space;
168 5993a8f2 2003-11-23 devnull 
169 5993a8f2 2003-11-23 devnull 	for (space = 0, q = p = buf; *q; q++) {
170 5993a8f2 2003-11-23 devnull 		if (isspace(*q)) {
171 5993a8f2 2003-11-23 devnull 			space++;
172 5993a8f2 2003-11-23 devnull 			continue;
173 5993a8f2 2003-11-23 devnull 		}
174 5993a8f2 2003-11-23 devnull 		if (space && bflag == 1) {
175 5993a8f2 2003-11-23 devnull 			*p++ = ' ';
176 5993a8f2 2003-11-23 devnull 			space = 0;
177 5993a8f2 2003-11-23 devnull 		}
178 5993a8f2 2003-11-23 devnull 		*p++ = *q;
179 5993a8f2 2003-11-23 devnull 	}
180 5993a8f2 2003-11-23 devnull 	*p = 0;
181 5993a8f2 2003-11-23 devnull 	return p - buf;
182 5993a8f2 2003-11-23 devnull }
183 5993a8f2 2003-11-23 devnull 
184 5993a8f2 2003-11-23 devnull /*
185 5993a8f2 2003-11-23 devnull  * need to fix up for unexpected EOF's
186 5993a8f2 2003-11-23 devnull  */
187 5993a8f2 2003-11-23 devnull void
188 5993a8f2 2003-11-23 devnull check(Biobuf *bf, Biobuf *bt)
189 5993a8f2 2003-11-23 devnull {
190 5993a8f2 2003-11-23 devnull 	int f, t, flen, tlen;
191 5993a8f2 2003-11-23 devnull 	char fbuf[MAXLINELEN], tbuf[MAXLINELEN];
192 5993a8f2 2003-11-23 devnull 
193 5993a8f2 2003-11-23 devnull 	ixold[0] = ixnew[0] = 0;
194 5993a8f2 2003-11-23 devnull 	for (f = t = 1; f < len[0]; f++) {
195 5993a8f2 2003-11-23 devnull 		flen = readline(bf, fbuf);
196 5993a8f2 2003-11-23 devnull 		ixold[f] = ixold[f-1] + flen + 1;		/* ftell(bf) */
197 5993a8f2 2003-11-23 devnull 		if (J[f] == 0)
198 5993a8f2 2003-11-23 devnull 			continue;
199 5993a8f2 2003-11-23 devnull 		do {
200 5993a8f2 2003-11-23 devnull 			tlen = readline(bt, tbuf);
201 5993a8f2 2003-11-23 devnull 			ixnew[t] = ixnew[t-1] + tlen + 1;	/* ftell(bt) */
202 5993a8f2 2003-11-23 devnull 		} while (t++ < J[f]);
203 5993a8f2 2003-11-23 devnull 		if (bflag) {
204 5993a8f2 2003-11-23 devnull 			flen = squishspace(fbuf);
205 5993a8f2 2003-11-23 devnull 			tlen = squishspace(tbuf);
206 5993a8f2 2003-11-23 devnull 		}
207 5993a8f2 2003-11-23 devnull 		if (flen != tlen || strcmp(fbuf, tbuf))
208 5993a8f2 2003-11-23 devnull 			J[f] = 0;
209 5993a8f2 2003-11-23 devnull 	}
210 5993a8f2 2003-11-23 devnull 	while (t < len[1]) {
211 5993a8f2 2003-11-23 devnull 		tlen = readline(bt, tbuf);
212 5993a8f2 2003-11-23 devnull 		ixnew[t] = ixnew[t-1] + tlen + 1;	/* fseek(bt) */
213 5993a8f2 2003-11-23 devnull 		t++;
214 5993a8f2 2003-11-23 devnull 	}
215 5993a8f2 2003-11-23 devnull }
216 5993a8f2 2003-11-23 devnull 
217 5993a8f2 2003-11-23 devnull static void
218 5993a8f2 2003-11-23 devnull range(int a, int b, char *separator)
219 5993a8f2 2003-11-23 devnull {
220 5993a8f2 2003-11-23 devnull 	Bprint(&stdout, "%d", a > b ? b: a);
221 5993a8f2 2003-11-23 devnull 	if (a < b)
222 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, "%s%d", separator, b);
223 5993a8f2 2003-11-23 devnull }
224 5993a8f2 2003-11-23 devnull 
225 5993a8f2 2003-11-23 devnull static void
226 5993a8f2 2003-11-23 devnull fetch(long *f, int a, int b, Biobuf *bp, char *s)
227 5993a8f2 2003-11-23 devnull {
228 5993a8f2 2003-11-23 devnull 	char buf[MAXLINELEN];
229 5993a8f2 2003-11-23 devnull 
230 5993a8f2 2003-11-23 devnull 	Bseek(bp, f[a-1], 0);
231 5993a8f2 2003-11-23 devnull 	while (a++ <= b) {
232 5993a8f2 2003-11-23 devnull 		readline(bp, buf);
233 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, "%s%s\n", s, buf);
234 5993a8f2 2003-11-23 devnull 	}
235 5993a8f2 2003-11-23 devnull }
236 5993a8f2 2003-11-23 devnull 
237 5993a8f2 2003-11-23 devnull void
238 5993a8f2 2003-11-23 devnull change(int a, int b, int c, int d)
239 5993a8f2 2003-11-23 devnull {
240 5993a8f2 2003-11-23 devnull 	char verb;
241 5993a8f2 2003-11-23 devnull 	char buf[4];
242 5993a8f2 2003-11-23 devnull 
243 5993a8f2 2003-11-23 devnull 	if (a > b && c > d)
244 5993a8f2 2003-11-23 devnull 		return;
245 5993a8f2 2003-11-23 devnull 	anychange = 1;
246 5993a8f2 2003-11-23 devnull 	if (mflag && firstchange == 0) {
247 5993a8f2 2003-11-23 devnull 		if(mode) {
248 5993a8f2 2003-11-23 devnull 			buf[0] = '-';
249 5993a8f2 2003-11-23 devnull 			buf[1] = mode;
250 5993a8f2 2003-11-23 devnull 			buf[2] = ' ';
251 5993a8f2 2003-11-23 devnull 			buf[3] = '\0';
252 5993a8f2 2003-11-23 devnull 		} else {
253 5993a8f2 2003-11-23 devnull 			buf[0] = '\0';
254 5993a8f2 2003-11-23 devnull 		}
255 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, "diff %s%s %s\n", buf, file1, file2);
256 5993a8f2 2003-11-23 devnull 		firstchange = 1;
257 5993a8f2 2003-11-23 devnull 	}
258 5993a8f2 2003-11-23 devnull 	verb = a > b ? 'a': c > d ? 'd': 'c';
259 5993a8f2 2003-11-23 devnull 	switch(mode) {
260 5993a8f2 2003-11-23 devnull 	case 'e':
261 5993a8f2 2003-11-23 devnull 		range(a, b, ",");
262 5993a8f2 2003-11-23 devnull 		Bputc(&stdout, verb);
263 5993a8f2 2003-11-23 devnull 		break;
264 5993a8f2 2003-11-23 devnull 	case 0:
265 5993a8f2 2003-11-23 devnull 		range(a, b, ",");
266 5993a8f2 2003-11-23 devnull 		Bputc(&stdout, verb);
267 5993a8f2 2003-11-23 devnull 		range(c, d, ",");
268 5993a8f2 2003-11-23 devnull 		break;
269 5993a8f2 2003-11-23 devnull 	case 'n':
270 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, "%s:", file1);
271 5993a8f2 2003-11-23 devnull 		range(a, b, ",");
272 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, " %c ", verb);
273 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, "%s:", file2);
274 5993a8f2 2003-11-23 devnull 		range(c, d, ",");
275 5993a8f2 2003-11-23 devnull 		break;
276 5993a8f2 2003-11-23 devnull 	case 'f':
277 5993a8f2 2003-11-23 devnull 		Bputc(&stdout, verb);
278 5993a8f2 2003-11-23 devnull 		range(a, b, " ");
279 5993a8f2 2003-11-23 devnull 		break;
280 5993a8f2 2003-11-23 devnull 	}
281 5993a8f2 2003-11-23 devnull 	Bputc(&stdout, '\n');
282 5993a8f2 2003-11-23 devnull 	if (mode == 0 || mode == 'n') {
283 5993a8f2 2003-11-23 devnull 		fetch(ixold, a, b, input[0], "< ");
284 5993a8f2 2003-11-23 devnull 		if (a <= b && c <= d)
285 5993a8f2 2003-11-23 devnull 			Bprint(&stdout, "---\n");
286 5993a8f2 2003-11-23 devnull 	}
287 5993a8f2 2003-11-23 devnull 	fetch(ixnew, c, d, input[1], mode == 0 || mode == 'n' ? "> ": "");
288 5993a8f2 2003-11-23 devnull 	if (mode != 0 && mode != 'n' && c <= d)
289 5993a8f2 2003-11-23 devnull 		Bprint(&stdout, ".\n");
290 5993a8f2 2003-11-23 devnull }
291 5993a8f2 2003-11-23 devnull