1 #include <u.h>
2 #include <libc.h>
3 #include <draw.h>
4 #include <thread.h>
5 #include <cursor.h>
6 #include <mouse.h>
7 #include <keyboard.h>
8 #include <frame.h>
9 #include <fcall.h>
10 #include <plumb.h>
11 #include "dat.h"
12 #include "edit.h"
13 #include "fns.h"
14 
15 int	Glooping;
16 int	nest;
17 char	Enoname[] = "no file name given";
18 
19 Address	addr;
20 File	*menu;
21 Rangeset	sel;
22 extern	Text*	curtext;
23 Rune	*collection;
24 int	ncollection;
25 
26 int	append(File*, Cmd*, long);
27 int	pdisplay(File*);
28 void	pfilename(File*);
29 void	looper(File*, Cmd*, int);
30 void	filelooper(Cmd*, int);
31 void	linelooper(File*, Cmd*);
32 Address	lineaddr(long, Address, int);
33 int	filematch(File*, String*);
34 File	*tofile(String*);
35 Rune*	cmdname(File *f, String *s, int);
36 void	runpipe(Text*, int, Rune*, int, int);
37 
38 void
39 clearcollection(void)
40 {
41 	free(collection);
42 	collection = nil;
43 	ncollection = 0;
44 }
45 
46 void
47 resetxec(void)
48 {
49 	Glooping = nest = 0;
50 	clearcollection();
51 }
52 
53 void
54 mkaddr(Address *a, File *f)
55 {
56 	a->r.q0 = f->curtext->q0;
57 	a->r.q1 = f->curtext->q1;
58 	a->f = f;
59 }
60 
61 int
62 cmdexec(Text *t, Cmd *cp)
63 {
64 	int i;
65 	Addr *ap;
66 	File *f;
67 	Window *w;
68 	Address dot;
69 
70 	if(t == nil)
71 		w = nil;
72 	else
73 		w = t->w;
74 	if(w==nil && (cp->addr==0 || cp->addr->type!='"') &&
75 	    !utfrune("bBnqUXY!", cp->cmdc) &&
76 	    !(cp->cmdc=='D' && cp->u.text))
77 		editerror("no current window");
78 	i = cmdlookup(cp->cmdc);	/* will be -1 for '{' */
79 	f = nil;
80 	if(t && t->w){
81 		t = &t->w->body;
82 		f = t->file;
83 		f->curtext = t;
84 	}
85 	if(i>=0 && cmdtab[i].defaddr != aNo){
86 		if((ap=cp->addr)==0 && cp->cmdc!='\n'){
87 			cp->addr = ap = newaddr();
88 			ap->type = '.';
89 			if(cmdtab[i].defaddr == aAll)
90 				ap->type = '*';
91 		}else if(ap && ap->type=='"' && ap->next==0 && cp->cmdc!='\n'){
92 			ap->next = newaddr();
93 			ap->next->type = '.';
94 			if(cmdtab[i].defaddr == aAll)
95 				ap->next->type = '*';
96 		}
97 		if(cp->addr){	/* may be false for '\n' (only) */
98 			static Address none = {0,0,nil};
99 			if(f){
100 				mkaddr(&dot, f);
101 				addr = cmdaddress(ap, dot, 0);
102 			}else	/* a " */
103 				addr = cmdaddress(ap, none, 0);
104 			f = addr.f;
105 			t = f->curtext;
106 		}
107 	}
108 	switch(cp->cmdc){
109 	case '{':
110 		mkaddr(&dot, f);
111 		if(cp->addr != nil)
112 			dot = cmdaddress(cp->addr, dot, 0);
113 		for(cp = cp->u.cmd; cp; cp = cp->next){
114 			if(dot.r.q1 > t->file->b.nc)
115 				editerror("dot extends past end of buffer during { command");
116 			t->q0 = dot.r.q0;
117 			t->q1 = dot.r.q1;
118 			cmdexec(t, cp);
119 		}
120 		break;
121 	default:
122 		if(i < 0)
123 			editerror("unknown command %c in cmdexec", cp->cmdc);
124 		i = (*cmdtab[i].fn)(t, cp);
125 		return i;
126 	}
127 	return 1;
128 }
129 
130 char*
131 edittext(Window *w, int q, Rune *r, int nr)
132 {
133 	File *f;
134 
135 	f = w->body.file;
136 	switch(editing){
137 	case Inactive:
138 		return "permission denied";
139 	case Inserting:
140 		eloginsert(f, q, r, nr);
141 		return nil;
142 	case Collecting:
143 		collection = runerealloc(collection, ncollection+nr+1);
144 		runemove(collection+ncollection, r, nr);
145 		ncollection += nr;
146 		collection[ncollection] = '\0';
147 		return nil;
148 	default:
149 		return "unknown state in edittext";
150 	}
151 }
152 
153 /* string is known to be NUL-terminated */
154 Rune*
155 filelist(Text *t, Rune *r, int nr)
156 {
157 	if(nr == 0)
158 		return nil;
159 	r = skipbl(r, nr, &nr);
160 	if(r[0] != '<')
161 		return runestrdup(r);
162 	/* use < command to collect text */
163 	clearcollection();
164 	runpipe(t, '<', r+1, nr-1, Collecting);
165 	return collection;
166 }
167 
168 int
169 a_cmd(Text *t, Cmd *cp)
170 {
171 	return append(t->file, cp, addr.r.q1);
172 }
173 
174 int
175 b_cmd(Text *t, Cmd *cp)
176 {
177 	File *f;
178 
179 	USED(t);
180 	f = tofile(cp->u.text);
181 	if(nest == 0)
182 		pfilename(f);
183 	curtext = f->curtext;
184 	return TRUE;
185 }
186 
187 int
188 B_cmd(Text *t, Cmd *cp)
189 {
190 	Rune *list, *r, *s;
191 	int nr;
192 
193 	list = filelist(t, cp->u.text->r, cp->u.text->n);
194 	if(list == nil)
195 		editerror(Enoname);
196 	r = list;
197 	nr = runestrlen(r);
198 	r = skipbl(r, nr, &nr);
199 	if(nr == 0)
200 		new(t, t, nil, 0, 0, r, 0);
201 	else while(nr > 0){
202 		s = findbl(r, nr, &nr);
203 		*s = '\0';
204 		new(t, t, nil, 0, 0, r, runestrlen(r));
205 		if(nr > 0)
206 			r = skipbl(s+1, nr-1, &nr);
207 	}
208 	clearcollection();
209 	return TRUE;
210 }
211 
212 int
213 c_cmd(Text *t, Cmd *cp)
214 {
215 	elogreplace(t->file, addr.r.q0, addr.r.q1, cp->u.text->r, cp->u.text->n);
216 	t->q0 = addr.r.q0;
217 	t->q1 = addr.r.q1;
218 	return TRUE;
219 }
220 
221 int
222 d_cmd(Text *t, Cmd *cp)
223 {
224 	USED(cp);
225 	if(addr.r.q1 > addr.r.q0)
226 		elogdelete(t->file, addr.r.q0, addr.r.q1);
227 	t->q0 = addr.r.q0;
228 	t->q1 = addr.r.q0;
229 	return TRUE;
230 }
231 
232 void
233 D1(Text *t)
234 {
235 	if(t->w->body.file->ntext>1 || winclean(t->w, FALSE))
236 		colclose(t->col, t->w, TRUE);
237 }
238 
239 int
240 D_cmd(Text *t, Cmd *cp)
241 {
242 	Rune *list, *r, *s, *n;
243 	int nr, nn;
244 	Window *w;
245 	Runestr dir, rs;
246 	char buf[128];
247 
248 	list = filelist(t, cp->u.text->r, cp->u.text->n);
249 	if(list == nil){
250 		D1(t);
251 		return TRUE;
252 	}
253 	dir = dirname(t, nil, 0);
254 	r = list;
255 	nr = runestrlen(r);
256 	r = skipbl(r, nr, &nr);
257 	do{
258 		s = findbl(r, nr, &nr);
259 		*s = '\0';
260 		/* first time through, could be empty string, meaning delete file empty name */
261 		nn = runestrlen(r);
262 		if(r[0]=='/' || nn==0 || dir.nr==0){
263 			rs.r = runestrdup(r);
264 			rs.nr = nn;
265 		}else{
266 			n = runemalloc(dir.nr+1+nn);
267 			runemove(n, dir.r, dir.nr);
268 			n[dir.nr] = '/';
269 			runemove(n+dir.nr+1, r, nn);
270 			rs = cleanrname(runestr(n, dir.nr+1+nn));
271 		}
272 		w = lookfile(rs.r, rs.nr);
273 		if(w == nil){
274 			snprint(buf, sizeof buf, "no such file %.*S", rs.nr, rs.r);
275 			free(rs.r);
276 			editerror(buf);
277 		}
278 		free(rs.r);
279 		D1(&w->body);
280 		if(nr > 0)
281 			r = skipbl(s+1, nr-1, &nr);
282 	}while(nr > 0);
283 	clearcollection();
284 	free(dir.r);
285 	return TRUE;
286 }
287 
288 static int
289 readloader(void *v, uint q0, Rune *r, int nr)
290 {
291 	if(nr > 0)
292 		eloginsert(v, q0, r, nr);
293 	return 0;
294 }
295 
296 int
297 e_cmd(Text *t, Cmd *cp)
298 {
299 	Rune *name;
300 	File *f;
301 	int i, isdir, q0, q1, fd, nulls, samename, allreplaced;
302 	char *s, tmp[128];
303 	Dir *d;
304 
305 	f = t->file;
306 	q0 = addr.r.q0;
307 	q1 = addr.r.q1;
308 	if(cp->cmdc == 'e'){
309 		if(winclean(t->w, TRUE)==FALSE)
310 			editerror("");	/* winclean generated message already */
311 		q0 = 0;
312 		q1 = f->b.nc;
313 	}
314 	allreplaced = (q0==0 && q1==f->b.nc);
315 	name = cmdname(f, cp->u.text, cp->cmdc=='e');
316 	if(name == nil)
317 		editerror(Enoname);
318 	i = runestrlen(name);
319 	samename = runeeq(name, i, t->file->name, t->file->nname);
320 	s = runetobyte(name, i);
321 	free(name);
322 	fd = open(s, OREAD);
323 	if(fd < 0){
324 		snprint(tmp, sizeof tmp, "can't open %s: %r", s);
325 		free(s);
326 		editerror(tmp);
327 	}
328 	d = dirfstat(fd);
329 	isdir = (d!=nil && (d->qid.type&QTDIR));
330 	free(d);
331 	if(isdir){
332 		close(fd);
333 		snprint(tmp, sizeof tmp, "%s is a directory", s);
334 		free(s);
335 		editerror(tmp);
336 	}
337 	elogdelete(f, q0, q1);
338 	nulls = 0;
339 	loadfile(fd, q1, &nulls, readloader, f);
340 	free(s);
341 	close(fd);
342 	if(nulls)
343 		warning(nil, "%s: NUL bytes elided\n", s);
344 	else if(allreplaced && samename)
345 		f->editclean = TRUE;
346 	return TRUE;
347 }
348 
349 static Rune Lempty[] = { 0 };
350 int
351 f_cmd(Text *t, Cmd *cp)
352 {
353 	Rune *name;
354 	String *str;
355 	String empty;
356 
357 	if(cp->u.text == nil){
358 		empty.n = 0;
359 		empty.r = Lempty;
360 		str = &empty;
361 	}else
362 		str = cp->u.text;
363 	name = cmdname(t->file, str, TRUE);
364 	free(name);
365 	pfilename(t->file);
366 	return TRUE;
367 }
368 
369 int
370 g_cmd(Text *t, Cmd *cp)
371 {
372 	if(t->file != addr.f){
373 		warning(nil, "internal error: g_cmd f!=addr.f\n");
374 		return FALSE;
375 	}
376 	if(rxcompile(cp->re->r) == FALSE)
377 		editerror("bad regexp in g command");
378 	if(rxexecute(t, nil, addr.r.q0, addr.r.q1, &sel) ^ cp->cmdc=='v'){
379 		t->q0 = addr.r.q0;
380 		t->q1 = addr.r.q1;
381 		return cmdexec(t, cp->u.cmd);
382 	}
383 	return TRUE;
384 }
385 
386 int
387 i_cmd(Text *t, Cmd *cp)
388 {
389 	return append(t->file, cp, addr.r.q0);
390 }
391 
392 void
393 copy(File *f, Address addr2)
394 {
395 	long p;
396 	int ni;
397 	Rune *buf;
398 
399 	buf = fbufalloc();
400 	for(p=addr.r.q0; p<addr.r.q1; p+=ni){
401 		ni = addr.r.q1-p;
402 		if(ni > RBUFSIZE)
403 			ni = RBUFSIZE;
404 		bufread(&f->b, p, buf, ni);
405 		eloginsert(addr2.f, addr2.r.q1, buf, ni);
406 	}
407 	fbuffree(buf);
408 }
409 
410 void
411 move(File *f, Address addr2)
412 {
413 	if(addr.f!=addr2.f || addr.r.q1<=addr2.r.q0){
414 		elogdelete(f, addr.r.q0, addr.r.q1);
415 		copy(f, addr2);
416 	}else if(addr.r.q0 >= addr2.r.q1){
417 		copy(f, addr2);
418 		elogdelete(f, addr.r.q0, addr.r.q1);
419 	}else
420 		error("move overlaps itself");
421 }
422 
423 int
424 m_cmd(Text *t, Cmd *cp)
425 {
426 	Address dot, addr2;
427 
428 	mkaddr(&dot, t->file);
429 	addr2 = cmdaddress(cp->u.mtaddr, dot, 0);
430 	if(cp->cmdc == 'm')
431 		move(t->file, addr2);
432 	else
433 		copy(t->file, addr2);
434 	return TRUE;
435 }
436 
437 int
438 p_cmd(Text *t, Cmd *cp)
439 {
440 	USED(cp);
441 	return pdisplay(t->file);
442 }
443 
444 int
445 s_cmd(Text *t, Cmd *cp)
446 {
447 	int i, j, k, c, m, n, nrp, didsub;
448 	long p1, op, delta;
449 	String *buf;
450 	Rangeset *rp;
451 	char *err;
452 	Rune *rbuf;
453 
454 	n = cp->num;
455 	op= -1;
456 	if(rxcompile(cp->re->r) == FALSE)
457 		editerror("bad regexp in s command");
458 	nrp = 0;
459 	rp = nil;
460 	delta = 0;
461 	didsub = FALSE;
462 	for(p1 = addr.r.q0; p1<=addr.r.q1 && rxexecute(t, nil, p1, addr.r.q1, &sel); ){
463 		if(sel.r[0].q0 == sel.r[0].q1){	/* empty match? */
464 			if(sel.r[0].q0 == op){
465 				p1++;
466 				continue;
467 			}
468 			p1 = sel.r[0].q1+1;
469 		}else
470 			p1 = sel.r[0].q1;
471 		op = sel.r[0].q1;
472 		if(--n>0)
473 			continue;
474 		nrp++;
475 		rp = erealloc(rp, nrp*sizeof(Rangeset));
476 		rp[nrp-1] = sel;
477 	}
478 	rbuf = fbufalloc();
479 	buf = allocstring(0);
480 	for(m=0; m<nrp; m++){
481 		buf->n = 0;
482 		buf->r[0] = '\0';
483 		sel = rp[m];
484 		for(i = 0; i<cp->u.text->n; i++)
485 			if((c = cp->u.text->r[i])=='\\' && i<cp->u.text->n-1){
486 				c = cp->u.text->r[++i];
487 				if('1'<=c && c<='9') {
488 					j = c-'0';
489 					if(sel.r[j].q1-sel.r[j].q0>RBUFSIZE){
490 						err = "replacement string too long";
491 						goto Err;
492 					}
493 					bufread(&t->file->b, sel.r[j].q0, rbuf, sel.r[j].q1-sel.r[j].q0);
494 					for(k=0; k<sel.r[j].q1-sel.r[j].q0; k++)
495 						Straddc(buf, rbuf[k]);
496 				}else
497 				 	Straddc(buf, c);
498 			}else if(c!='&')
499 				Straddc(buf, c);
500 			else{
501 				if(sel.r[0].q1-sel.r[0].q0>RBUFSIZE){
502 					err = "right hand side too long in substitution";
503 					goto Err;
504 				}
505 				bufread(&t->file->b, sel.r[0].q0, rbuf, sel.r[0].q1-sel.r[0].q0);
506 				for(k=0; k<sel.r[0].q1-sel.r[0].q0; k++)
507 					Straddc(buf, rbuf[k]);
508 			}
509 		elogreplace(t->file, sel.r[0].q0, sel.r[0].q1,  buf->r, buf->n);
510 		delta -= sel.r[0].q1-sel.r[0].q0;
511 		delta += buf->n;
512 		didsub = 1;
513 		if(!cp->flag)
514 			break;
515 	}
516 	free(rp);
517 	freestring(buf);
518 	fbuffree(rbuf);
519 	if(!didsub && nest==0)
520 		editerror("no substitution");
521 	t->q0 = addr.r.q0;
522 	t->q1 = addr.r.q1;
523 	return TRUE;
524 
525 Err:
526 	free(rp);
527 	freestring(buf);
528 	fbuffree(rbuf);
529 	editerror(err);
530 	return FALSE;
531 }
532 
533 int
534 u_cmd(Text *t, Cmd *cp)
535 {
536 	int n, oseq, flag;
537 
538 	n = cp->num;
539 	flag = TRUE;
540 	if(n < 0){
541 		n = -n;
542 		flag = FALSE;
543 	}
544 	oseq = -1;
545 	while(n-->0 && t->file->seq!=0 && t->file->seq!=oseq){
546 		oseq = t->file->seq;
547 		undo(t, nil, nil, flag, 0, nil, 0);
548 	}
549 	return TRUE;
550 }
551 
552 int
553 w_cmd(Text *t, Cmd *cp)
554 {
555 	Rune *r;
556 	File *f;
557 
558 	f = t->file;
559 	if(f->seq == seq)
560 		editerror("can't write file with pending modifications");
561 	r = cmdname(f, cp->u.text, FALSE);
562 	if(r == nil)
563 		editerror("no name specified for 'w' command");
564 	putfile(f, addr.r.q0, addr.r.q1, r, runestrlen(r));
565 	/* r is freed by putfile */
566 	return TRUE;
567 }
568 
569 int
570 x_cmd(Text *t, Cmd *cp)
571 {
572 	if(cp->re)
573 		looper(t->file, cp, cp->cmdc=='x');
574 	else
575 		linelooper(t->file, cp);
576 	return TRUE;
577 }
578 
579 int
580 X_cmd(Text *t, Cmd *cp)
581 {
582 	USED(t);
583 
584 	filelooper(cp, cp->cmdc=='X');
585 	return TRUE;
586 }
587 
588 void
589 runpipe(Text *t, int cmd, Rune *cr, int ncr, int state)
590 {
591 	Rune *r, *s;
592 	int n;
593 	Runestr dir;
594 	Window *w;
595 	QLock *q;
596 
597 	r = skipbl(cr, ncr, &n);
598 	if(n == 0)
599 		editerror("no command specified for %c", cmd);
600 	w = nil;
601 	if(state == Inserting){
602 		w = t->w;
603 		t->q0 = addr.r.q0;
604 		t->q1 = addr.r.q1;
605 		if(cmd == '<' || cmd=='|')
606 			elogdelete(t->file, t->q0, t->q1);
607 	}
608 	s = runemalloc(n+2);
609 	s[0] = cmd;
610 	runemove(s+1, r, n);
611 	n++;
612 	dir.r = nil;
613 	dir.nr = 0;
614 	if(t != nil)
615 		dir = dirname(t, nil, 0);
616 	if(dir.nr==1 && dir.r[0]=='.'){	/* sigh */
617 		free(dir.r);
618 		dir.r = nil;
619 		dir.nr = 0;
620 	}
621 	editing = state;
622 	if(t!=nil && t->w!=nil)
623 		incref(&t->w->ref);	/* run will decref */
624 	run(w, runetobyte(s, n), dir.r, dir.nr, TRUE, nil, nil, TRUE);
625 	free(s);
626 	if(t!=nil && t->w!=nil)
627 		winunlock(t->w);
628 	qunlock(&row.lk);
629 	recvul(cedit);
630 	/*
631 	 * The editoutlk exists only so that we can tell when
632 	 * the editout file has been closed.  It can get closed *after*
633 	 * the process exits because, since the process cannot be 
634 	 * connected directly to editout (no 9P kernel support), 
635 	 * the process is actually connected to a pipe to another
636 	 * process (arranged via 9pserve) that reads from the pipe
637 	 * and then writes the data in the pipe to editout using
638 	 * 9P transactions.  This process might still have a couple
639 	 * writes left to copy after the original process has exited.
640 	 */
641 	if(w)
642 		q = &w->editoutlk;
643 	else
644 		q = &editoutlk;
645 	qlock(q);	/* wait for file to close */
646 	qunlock(q);
647 	qlock(&row.lk);
648 	editing = Inactive;
649 	if(t!=nil && t->w!=nil)
650 		winlock(t->w, 'M');
651 }
652 
653 int
654 pipe_cmd(Text *t, Cmd *cp)
655 {
656 	runpipe(t, cp->cmdc, cp->u.text->r, cp->u.text->n, Inserting);
657 	return TRUE;
658 }
659 
660 long
661 nlcount(Text *t, long q0, long q1)
662 {
663 	long nl;
664 	Rune *buf;
665 	int i, nbuf;
666 
667 	buf = fbufalloc();
668 	nbuf = 0;
669 	i = nl = 0;
670 	while(q0 < q1){
671 		if(i == nbuf){
672 			nbuf = q1-q0;
673 			if(nbuf > RBUFSIZE)
674 				nbuf = RBUFSIZE;
675 			bufread(&t->file->b, q0, buf, nbuf);
676 			i = 0;
677 		}
678 		if(buf[i++] == '\n')
679 			nl++;
680 		q0++;
681 	}
682 	fbuffree(buf);
683 	return nl;
684 }
685 
686 void
687 printposn(Text *t, int charsonly)
688 {
689 	long l1, l2;
690 
691 	if (t != nil && t->file != nil && t->file->name != nil)
692 		warning(nil, "%.*S:", t->file->nname, t->file->name);
693 	if(!charsonly){
694 		l1 = 1+nlcount(t, 0, addr.r.q0);
695 		l2 = l1+nlcount(t, addr.r.q0, addr.r.q1);
696 		/* check if addr ends with '\n' */
697 		if(addr.r.q1>0 && addr.r.q1>addr.r.q0 && textreadc(t, addr.r.q1-1)=='\n')
698 			--l2;
699 		warning(nil, "%lud", l1);
700 		if(l2 != l1)
701 			warning(nil, ",%lud", l2);
702 		warning(nil, "\n");
703 		return;
704 	}
705 	warning(nil, "#%d", addr.r.q0);
706 	if(addr.r.q1 != addr.r.q0)
707 		warning(nil, ",#%d", addr.r.q1);
708 	warning(nil, "\n");
709 }
710 
711 int
712 eq_cmd(Text *t, Cmd *cp)
713 {
714 	int charsonly;
715 
716 	switch(cp->u.text->n){
717 	case 0:
718 		charsonly = FALSE;
719 		break;
720 	case 1:
721 		if(cp->u.text->r[0] == '#'){
722 			charsonly = TRUE;
723 			break;
724 		}
725 	default:
726 		SET(charsonly);
727 		editerror("newline expected");
728 	}
729 	printposn(t, charsonly);
730 	return TRUE;
731 }
732 
733 int
734 nl_cmd(Text *t, Cmd *cp)
735 {
736 	Address a;
737 	File *f;
738 
739 	f = t->file;
740 	if(cp->addr == 0){
741 		/* First put it on newline boundaries */
742 		mkaddr(&a, f);
743 		addr = lineaddr(0, a, -1);
744 		a = lineaddr(0, a, 1);
745 		addr.r.q1 = a.r.q1;
746 		if(addr.r.q0==t->q0 && addr.r.q1==t->q1){
747 			mkaddr(&a, f);
748 			addr = lineaddr(1, a, 1);
749 		}
750 	}
751 	textshow(t, addr.r.q0, addr.r.q1, 1);
752 	return TRUE;
753 }
754 
755 int
756 append(File *f, Cmd *cp, long p)
757 {
758 	if(cp->u.text->n > 0)
759 		eloginsert(f, p, cp->u.text->r, cp->u.text->n);
760 	f->curtext->q0 = p;
761 	f->curtext->q1 = p;
762 	return TRUE;
763 }
764 
765 int
766 pdisplay(File *f)
767 {
768 	long p1, p2;
769 	int np;
770 	Rune *buf;
771 
772 	p1 = addr.r.q0;
773 	p2 = addr.r.q1;
774 	if(p2 > f->b.nc)
775 		p2 = f->b.nc;
776 	buf = fbufalloc();
777 	while(p1 < p2){
778 		np = p2-p1;
779 		if(np>RBUFSIZE-1)
780 			np = RBUFSIZE-1;
781 		bufread(&f->b, p1, buf, np);
782 		buf[np] = '\0';
783 		warning(nil, "%S", buf);
784 		p1 += np;
785 	}
786 	fbuffree(buf);
787 	f->curtext->q0 = addr.r.q0;
788 	f->curtext->q1 = addr.r.q1;
789 	return TRUE;
790 }
791 
792 void
793 pfilename(File *f)
794 {
795 	int dirty;
796 	Window *w;
797 
798 	w = f->curtext->w;
799 	/* same check for dirty as in settag, but we know ncache==0 */
800 	dirty = !w->isdir && !w->isscratch && f->mod;
801 	warning(nil, "%c%c%c %.*S\n", " '"[dirty],
802 		'+', " ."[curtext!=nil && curtext->file==f], f->nname, f->name);
803 }
804 
805 void
806 loopcmd(File *f, Cmd *cp, Range *rp, long nrp)
807 {
808 	long i;
809 
810 	for(i=0; i<nrp; i++){
811 		f->curtext->q0 = rp[i].q0;
812 		f->curtext->q1 = rp[i].q1;
813 		cmdexec(f->curtext, cp);
814 	}
815 }
816 
817 void
818 looper(File *f, Cmd *cp, int xy)
819 {
820 	long p, op, nrp;
821 	Range r, tr;
822 	Range *rp;
823 
824 	r = addr.r;
825 	op= xy? -1 : r.q0;
826 	nest++;
827 	if(rxcompile(cp->re->r) == FALSE)
828 		editerror("bad regexp in %c command", cp->cmdc);
829 	nrp = 0;
830 	rp = nil;
831 	for(p = r.q0; p<=r.q1; ){
832 		if(!rxexecute(f->curtext, nil, p, r.q1, &sel)){ /* no match, but y should still run */
833 			if(xy || op>r.q1)
834 				break;
835 			tr.q0 = op, tr.q1 = r.q1;
836 			p = r.q1+1;	/* exit next loop */
837 		}else{
838 			if(sel.r[0].q0==sel.r[0].q1){	/* empty match? */
839 				if(sel.r[0].q0==op){
840 					p++;
841 					continue;
842 				}
843 				p = sel.r[0].q1+1;
844 			}else
845 				p = sel.r[0].q1;
846 			if(xy)
847 				tr = sel.r[0];
848 			else
849 				tr.q0 = op, tr.q1 = sel.r[0].q0;
850 		}
851 		op = sel.r[0].q1;
852 		nrp++;
853 		rp = erealloc(rp, nrp*sizeof(Range));
854 		rp[nrp-1] = tr;
855 	}
856 	loopcmd(f, cp->u.cmd, rp, nrp);
857 	free(rp);
858 	--nest;
859 }
860 
861 void
862 linelooper(File *f, Cmd *cp)
863 {
864 	long nrp, p;
865 	Range r, linesel;
866 	Address a, a3;
867 	Range *rp;
868 
869 	nest++;
870 	nrp = 0;
871 	rp = nil;
872 	r = addr.r;
873 	a3.f = f;
874 	a3.r.q0 = a3.r.q1 = r.q0;
875 	a = lineaddr(0, a3, 1);
876 	linesel = a.r;
877 	for(p = r.q0; p<r.q1; p = a3.r.q1){
878 		a3.r.q0 = a3.r.q1;
879 		if(p!=r.q0 || linesel.q1==p){
880 			a = lineaddr(1, a3, 1);
881 			linesel = a.r;
882 		}
883 		if(linesel.q0 >= r.q1)
884 			break;
885 		if(linesel.q1 >= r.q1)
886 			linesel.q1 = r.q1;
887 		if(linesel.q1 > linesel.q0)
888 			if(linesel.q0>=a3.r.q1 && linesel.q1>a3.r.q1){
889 				a3.r = linesel;
890 				nrp++;
891 				rp = erealloc(rp, nrp*sizeof(Range));
892 				rp[nrp-1] = linesel;
893 				continue;
894 			}
895 		break;
896 	}
897 	loopcmd(f, cp->u.cmd, rp, nrp);
898 	free(rp);
899 	--nest;
900 }
901 
902 struct Looper
903 {
904 	Cmd *cp;
905 	int	XY;
906 	Window	**w;
907 	int	nw;
908 } loopstruct;	/* only one; X and Y can't nest */
909 
910 void
911 alllooper(Window *w, void *v)
912 {
913 	Text *t;
914 	struct Looper *lp;
915 	Cmd *cp;
916 
917 	lp = v;
918 	cp = lp->cp;
919 /*	if(w->isscratch || w->isdir) */
920 /*		return; */
921 	t = &w->body;
922 	/* only use this window if it's the current window for the file */
923 	if(t->file->curtext != t)
924 		return;
925 /*	if(w->nopen[QWevent] > 0) */
926 /*		return; */
927 	/* no auto-execute on files without names */
928 	if(cp->re==nil && t->file->nname==0)
929 		return;
930 	if(cp->re==nil || filematch(t->file, cp->re)==lp->XY){
931 		lp->w = erealloc(lp->w, (lp->nw+1)*sizeof(Window*));
932 		lp->w[lp->nw++] = w;
933 	}
934 }
935 
936 void
937 alllocker(Window *w, void *v)
938 {
939 	if(v)
940 		incref(&w->ref);
941 	else
942 		winclose(w);
943 }
944 
945 void
946 filelooper(Cmd *cp, int XY)
947 {
948 	int i;
949 
950 	if(Glooping++)
951 		editerror("can't nest %c command", "YX"[XY]);
952 	nest++;
953 
954 	loopstruct.cp = cp;
955 	loopstruct.XY = XY;
956 	if(loopstruct.w)	/* error'ed out last time */
957 		free(loopstruct.w);
958 	loopstruct.w = nil;
959 	loopstruct.nw = 0;
960 	allwindows(alllooper, &loopstruct);
961 	/*
962 	 * add a ref to all windows to keep safe windows accessed by X
963 	 * that would not otherwise have a ref to hold them up during
964 	 * the shenanigans.  note this with globalincref so that any
965 	 * newly created windows start with an extra reference.
966 	 */
967 	allwindows(alllocker, (void*)1);
968 	globalincref = 1;
969 	for(i=0; i<loopstruct.nw; i++)
970 		cmdexec(&loopstruct.w[i]->body, cp->u.cmd);
971 	allwindows(alllocker, (void*)0);
972 	globalincref = 0;
973 	free(loopstruct.w);
974 	loopstruct.w = nil;
975 
976 	--Glooping;
977 	--nest;
978 }
979 
980 void
981 nextmatch(File *f, String *r, long p, int sign)
982 {
983 	if(rxcompile(r->r) == FALSE)
984 		editerror("bad regexp in command address");
985 	if(sign >= 0){
986 		if(!rxexecute(f->curtext, nil, p, 0x7FFFFFFFL, &sel))
987 			editerror("no match for regexp");
988 		if(sel.r[0].q0==sel.r[0].q1 && sel.r[0].q0==p){
989 			if(++p>f->b.nc)
990 				p = 0;
991 			if(!rxexecute(f->curtext, nil, p, 0x7FFFFFFFL, &sel))
992 				editerror("address");
993 		}
994 	}else{
995 		if(!rxbexecute(f->curtext, p, &sel))
996 			editerror("no match for regexp");
997 		if(sel.r[0].q0==sel.r[0].q1 && sel.r[0].q1==p){
998 			if(--p<0)
999 				p = f->b.nc;
1000 			if(!rxbexecute(f->curtext, p, &sel))
1001 				editerror("address");
1002 		}
1003 	}
1004 }
1005 
1006 File	*matchfile(String*);
1007 Address	charaddr(long, Address, int);
1008 Address	lineaddr(long, Address, int);
1009 
1010 Address
1011 cmdaddress(Addr *ap, Address a, int sign)
1012 {
1013 	File *f = a.f;
1014 	Address a1, a2;
1015 
1016 	do{
1017 		switch(ap->type){
1018 		case 'l':
1019 		case '#':
1020 			a = (*(ap->type=='#'?charaddr:lineaddr))(ap->num, a, sign);
1021 			break;
1022 
1023 		case '.':
1024 			mkaddr(&a, f);
1025 			break;
1026 
1027 		case '$':
1028 			a.r.q0 = a.r.q1 = f->b.nc;
1029 			break;
1030 
1031 		case '\'':
1032 editerror("can't handle '");
1033 /*			a.r = f->mark; */
1034 			break;
1035 
1036 		case '?':
1037 			sign = -sign;
1038 			if(sign == 0)
1039 				sign = -1;
1040 			/* fall through */
1041 		case '/':
1042 			nextmatch(f, ap->u.re, sign>=0? a.r.q1 : a.r.q0, sign);
1043 			a.r = sel.r[0];
1044 			break;
1045 
1046 		case '"':
1047 			f = matchfile(ap->u.re);
1048 			mkaddr(&a, f);
1049 			break;
1050 
1051 		case '*':
1052 			a.r.q0 = 0, a.r.q1 = f->b.nc;
1053 			return a;
1054 
1055 		case ',':
1056 		case ';':
1057 			if(ap->u.left)
1058 				a1 = cmdaddress(ap->u.left, a, 0);
1059 			else
1060 				a1.f = a.f, a1.r.q0 = a1.r.q1 = 0;
1061 			if(ap->type == ';'){
1062 				f = a1.f;
1063 				a = a1;
1064 				f->curtext->q0 = a1.r.q0;
1065 				f->curtext->q1 = a1.r.q1;
1066 			}
1067 			if(ap->next)
1068 				a2 = cmdaddress(ap->next, a, 0);
1069 			else
1070 				a2.f = a.f, a2.r.q0 = a2.r.q1 = f->b.nc;
1071 			if(a1.f != a2.f)
1072 				editerror("addresses in different files");
1073 			a.f = a1.f, a.r.q0 = a1.r.q0, a.r.q1 = a2.r.q1;
1074 			if(a.r.q1 < a.r.q0)
1075 				editerror("addresses out of order");
1076 			return a;
1077 
1078 		case '+':
1079 		case '-':
1080 			sign = 1;
1081 			if(ap->type == '-')
1082 				sign = -1;
1083 			if(ap->next==0 || ap->next->type=='+' || ap->next->type=='-')
1084 				a = lineaddr(1L, a, sign);
1085 			break;
1086 		default:
1087 			error("cmdaddress");
1088 			return a;
1089 		}
1090 	}while(ap = ap->next);	/* assign = */
1091 	return a;
1092 }
1093 
1094 struct Tofile{
1095 	File		*f;
1096 	String	*r;
1097 };
1098 
1099 void
1100 alltofile(Window *w, void *v)
1101 {
1102 	Text *t;
1103 	struct Tofile *tp;
1104 
1105 	tp = v;
1106 	if(tp->f != nil)
1107 		return;
1108 	if(w->isscratch || w->isdir)
1109 		return;
1110 	t = &w->body;
1111 	/* only use this window if it's the current window for the file */
1112 	if(t->file->curtext != t)
1113 		return;
1114 /*	if(w->nopen[QWevent] > 0) */
1115 /*		return; */
1116 	if(runeeq(tp->r->r, tp->r->n, t->file->name, t->file->nname))
1117 		tp->f = t->file;
1118 }
1119 
1120 File*
1121 tofile(String *r)
1122 {
1123 	struct Tofile t;
1124 	String rr;
1125 
1126 	rr.r = skipbl(r->r, r->n, &rr.n);
1127 	t.f = nil;
1128 	t.r = &rr;
1129 	allwindows(alltofile, &t);
1130 	if(t.f == nil)
1131 		editerror("no such file\"%S\"", rr.r);
1132 	return t.f;
1133 }
1134 
1135 void
1136 allmatchfile(Window *w, void *v)
1137 {
1138 	struct Tofile *tp;
1139 	Text *t;
1140 
1141 	tp = v;
1142 	if(w->isscratch || w->isdir)
1143 		return;
1144 	t = &w->body;
1145 	/* only use this window if it's the current window for the file */
1146 	if(t->file->curtext != t)
1147 		return;
1148 /*	if(w->nopen[QWevent] > 0) */
1149 /*		return; */
1150 	if(filematch(w->body.file, tp->r)){
1151 		if(tp->f != nil)
1152 			editerror("too many files match \"%S\"", tp->r->r);
1153 		tp->f = w->body.file;
1154 	}
1155 }
1156 
1157 File*
1158 matchfile(String *r)
1159 {
1160 	struct Tofile tf;
1161 
1162 	tf.f = nil;
1163 	tf.r = r;
1164 	allwindows(allmatchfile, &tf);
1165 
1166 	if(tf.f == nil)
1167 		editerror("no file matches \"%S\"", r->r);
1168 	return tf.f;
1169 }
1170 
1171 int
1172 filematch(File *f, String *r)
1173 {
1174 	char *buf;
1175 	Rune *rbuf;
1176 	Window *w;
1177 	int match, i, dirty;
1178 	Rangeset s;
1179 
1180 	/* compile expr first so if we get an error, we haven't allocated anything */
1181 	if(rxcompile(r->r) == FALSE)
1182 		editerror("bad regexp in file match");
1183 	buf = fbufalloc();
1184 	w = f->curtext->w;
1185 	/* same check for dirty as in settag, but we know ncache==0 */
1186 	dirty = !w->isdir && !w->isscratch && f->mod;
1187 	snprint(buf, BUFSIZE, "%c%c%c %.*S\n", " '"[dirty],
1188 		'+', " ."[curtext!=nil && curtext->file==f], f->nname, f->name);
1189 	rbuf = bytetorune(buf, &i);
1190 	fbuffree(buf);
1191 	match = rxexecute(nil, rbuf, 0, i, &s);
1192 	free(rbuf);
1193 	return match;
1194 }
1195 
1196 Address
1197 charaddr(long l, Address addr, int sign)
1198 {
1199 	if(sign == 0)
1200 		addr.r.q0 = addr.r.q1 = l;
1201 	else if(sign < 0)
1202 		addr.r.q1 = addr.r.q0 -= l;
1203 	else if(sign > 0)
1204 		addr.r.q0 = addr.r.q1 += l;
1205 	if(addr.r.q0<0 || addr.r.q1>addr.f->b.nc)
1206 		editerror("address out of range");
1207 	return addr;
1208 }
1209 
1210 Address
1211 lineaddr(long l, Address addr, int sign)
1212 {
1213 	int n;
1214 	int c;
1215 	File *f = addr.f;
1216 	Address a;
1217 	long p;
1218 
1219 	a.f = f;
1220 	if(sign >= 0){
1221 		if(l == 0){
1222 			if(sign==0 || addr.r.q1==0){
1223 				a.r.q0 = a.r.q1 = 0;
1224 				return a;
1225 			}
1226 			a.r.q0 = addr.r.q1;
1227 			p = addr.r.q1-1;
1228 		}else{
1229 			if(sign==0 || addr.r.q1==0){
1230 				p = 0;
1231 				n = 1;
1232 			}else{
1233 				p = addr.r.q1-1;
1234 				n = textreadc(f->curtext, p++)=='\n';
1235 			}
1236 			while(n < l){
1237 				if(p >= f->b.nc)
1238 					editerror("address out of range");
1239 				if(textreadc(f->curtext, p++) == '\n')
1240 					n++;
1241 			}
1242 			a.r.q0 = p;
1243 		}
1244 		while(p < f->b.nc && textreadc(f->curtext, p++)!='\n')
1245 			;
1246 		a.r.q1 = p;
1247 	}else{
1248 		p = addr.r.q0;
1249 		if(l == 0)
1250 			a.r.q1 = addr.r.q0;
1251 		else{
1252 			for(n = 0; n<l; ){	/* always runs once */
1253 				if(p == 0){
1254 					if(++n != l)
1255 						editerror("address out of range");
1256 				}else{
1257 					c = textreadc(f->curtext, p-1);
1258 					if(c != '\n' || ++n != l)
1259 						p--;
1260 				}
1261 			}
1262 			a.r.q1 = p;
1263 			if(p > 0)
1264 				p--;
1265 		}
1266 		while(p > 0 && textreadc(f->curtext, p-1)!='\n')	/* lines start after a newline */
1267 			p--;
1268 		a.r.q0 = p;
1269 	}
1270 	return a;
1271 }
1272 
1273 struct Filecheck
1274 {
1275 	File	*f;
1276 	Rune	*r;
1277 	int nr;
1278 };
1279 
1280 void
1281 allfilecheck(Window *w, void *v)
1282 {
1283 	struct Filecheck *fp;
1284 	File *f;
1285 
1286 	fp = v;
1287 	f = w->body.file;
1288 	if(w->body.file == fp->f)
1289 		return;
1290 	if(runeeq(fp->r, fp->nr, f->name, f->nname))
1291 		warning(nil, "warning: duplicate file name \"%.*S\"\n", fp->nr, fp->r);
1292 }
1293 
1294 Rune*
1295 cmdname(File *f, String *str, int set)
1296 {
1297 	Rune *r, *s;
1298 	int n;
1299 	struct Filecheck fc;
1300 	Runestr newname;
1301 
1302 	r = nil;
1303 	n = str->n;
1304 	s = str->r;
1305 	if(n == 0){
1306 		/* no name; use existing */
1307 		if(f->nname == 0)
1308 			return nil;
1309 		r = runemalloc(f->nname+1);
1310 		runemove(r, f->name, f->nname);
1311 		return r;
1312 	}
1313 	s = skipbl(s, n, &n);
1314 	if(n == 0)
1315 		goto Return;
1316 
1317 	if(s[0] == '/'){
1318 		r = runemalloc(n+1);
1319 		runemove(r, s, n);
1320 	}else{
1321 		newname = dirname(f->curtext, runestrdup(s), n);
1322 		n = newname.nr;
1323 		r = runemalloc(n+1);
1324 		runemove(r, newname.r, n);
1325 		free(newname.r);
1326 	}
1327 	fc.f = f;
1328 	fc.r = r;
1329 	fc.nr = n;
1330 	allwindows(allfilecheck, &fc);
1331 	if(f->nname == 0)
1332 		set = TRUE;
1333 
1334     Return:
1335 	if(set && !runeeq(r, n, f->name, f->nname)){
1336 		filemark(f);
1337 		f->mod = TRUE;
1338 		f->curtext->w->dirty = TRUE;
1339 		winsetname(f->curtext->w, r, n);
1340 	}
1341 	return r;
1342 }