1 #include <u.h>
2 #include <libc.h>
3 #include <thread.h>
4 #include <fcall.h>
5 #include <9pclient.h>
6 #include "term.h"
7 
8 int noecho = 1;
9 
10 #define	EVENTSIZE	256
11 #define	STACK	32768
12 
13 typedef struct Event Event;
14 typedef struct Q Q;
15 
16 struct Event
17 {
18 	int	c1;
19 	int	c2;
20 	int	q0;
21 	int	q1;
22 	int	flag;
23 	int	nb;
24 	int	nr;
25 	char	b[EVENTSIZE*UTFmax+1];
26 	Rune	r[EVENTSIZE+1];
27 };
28 
29 Event blank = {
30 	'M',
31 	'X',
32 	0, 0, 0, 1, 1,
33 	{ ' ', 0 },
34 	{ ' ', 0 },
35 };
36 
37 struct Q
38 {
39 	QLock	lk;
40 	int		p;
41 	int		k;
42 };
43 
44 Q	q;
45 
46 CFid *eventfd;
47 CFid *addrfd;
48 CFid *datafd;
49 CFid *ctlfd;
50 // int bodyfd;
51 
52 char	*typing;
53 int	ntypeb;
54 int	ntyper;
55 int	ntypebreak;
56 int	debug;
57 int	rcfd;
58 
59 char *name;
60 
61 char **prog;
62 Channel *cwait;
63 int pid = -1;
64 
65 int	label(char*, int);
66 void	error(char*);
67 void	stdinproc(void*);
68 void	stdoutproc(void*);
69 void	type(Event*, int, CFid*, CFid*);
70 void	sende(Event*, int, CFid*, CFid*, CFid*, int);
71 char	*onestring(int, char**);
72 int	delete(Event*);
73 void	deltype(uint, uint);
74 void	runproc(void*);
75 
76 int
77 fsfidprint(CFid *fid, char *fmt, ...)
78 {
79 	char buf[256];
80 	va_list arg;
81 	int n;
82 
83 	va_start(arg, fmt);
84 	n = vsnprint(buf, sizeof buf, fmt, arg);
85 	va_end(arg);
86 	return fswrite(fid, buf, n);
87 }
88 
89 void
90 usage(void)
91 {
92 	fprint(2, "usage: win cmd args...\n");
93 	threadexitsall("usage");
94 }
95 
96 int
97 nopipes(void *v, char *msg)
98 {
99 	USED(v);
100 	if(strcmp(msg, "sys: write on closed pipe") == 0)
101 		return 1;
102 	return 0;
103 }
104 
105 void
106 waitthread(void *v)
107 {
108 	recvp(cwait);
109 	threadexitsall(nil);
110 }
111 
112 void
113 hangupnote(void *a, char *msg)
114 {
115 	if(strcmp(msg, "hangup") == 0 && pid != 0){
116 		postnote(PNGROUP, pid, "hangup");
117 		noted(NDFLT);
118 	}
119 	if(strstr(msg, "child")){
120 		/* bug: do better */
121 		threadexitsall(0);
122 	}
123 	noted(NDFLT);
124 }
125 
126 void
127 threadmain(int argc, char **argv)
128 {
129 	int fd, id;
130 	char buf[256];
131 	char buf1[128];
132 	CFsys *fs;
133 
134 	ARGBEGIN{
135 	case 'd':
136 		debug = 1;
137 		break;
138 	case 'n':
139 		name = EARGF(usage());
140 		break;
141 	default:
142 		usage();
143 	}ARGEND
144 
145 	prog = argv;
146 
147 	if(name == nil){
148 		if(argc > 0)
149 			name = argv[0];
150 		else{
151 			name = sysname();
152 			if(name == nil)
153 				name = "gnot";
154 		}
155 	}
156 
157 	notedisable("sys: write on closed pipe");
158 	noteenable("sys: child");
159 	notify(hangupnote);
160 
161 	if((fs = nsmount("acme", "")) == 0)
162 		sysfatal("nsmount acme: %r");
163 	ctlfd = fsopen(fs, "new/ctl", ORDWR|OCEXEC);
164 	if(ctlfd == 0 || fsread(ctlfd, buf, 12) != 12)
165 		sysfatal("ctl: %r");
166 	id = atoi(buf);
167 	snprint(buf, sizeof buf, "%d", id);
168 	putenv("winid", buf);
169 	sprint(buf, "%d/tag", id);
170 	fd = fsopenfd(fs, buf, OWRITE|OCEXEC);
171 	write(fd, " Send Delete", 12);
172 	close(fd);
173 	sprint(buf, "%d/event", id);
174 	eventfd = fsopen(fs, buf, ORDWR|OCEXEC);
175 	sprint(buf, "%d/addr", id);
176 	addrfd = fsopen(fs, buf, ORDWR|OCEXEC);
177 	sprint(buf, "%d/data", id);
178 	datafd = fsopen(fs, buf, ORDWR|OCEXEC);
179 	sprint(buf, "%d/body", id);
180 /*	bodyfd = fsopenfd(fs, buf, ORDWR|OCEXEC); */
181 	if(eventfd==nil || addrfd==nil || datafd==nil)
182 		sysfatal("data files: %r");
183 /*
184 	if(eventfd<0 || addrfd<0 || datafd<0 || bodyfd<0)
185 		sysfatal("data files: %r");
186 */
187 	fsunmount(fs);
188 
189 	cwait = threadwaitchan();
190 	threadcreate(waitthread, nil, STACK);
191 	pid = rcstart(argc, argv, &rcfd, nil);
192 	if(pid == -1)
193 		sysfatal("exec failed");
194 
195 	getwd(buf1, sizeof buf1);
196 	sprint(buf, "name %s/-%s\n0\n", buf1, name);
197 	fswrite(ctlfd, buf, strlen(buf));
198 	sprint(buf, "dumpdir %s/\n", buf1);
199 	fswrite(ctlfd, buf, strlen(buf));
200 	sprint(buf, "dump %s\n", onestring(argc, argv));
201 	fswrite(ctlfd, buf, strlen(buf));
202 	
203 	updatewinsize(25, 80, 0, 0);
204 	proccreate(stdoutproc, nil, STACK);
205 	stdinproc(nil);
206 }
207 
208 void
209 error(char *s)
210 {
211 	if(s)
212 		fprint(2, "win: %s: %r\n", s);
213 	else
214 		s = "kill";
215 	if(pid != -1)
216 		postnote(PNGROUP, pid, "hangup");
217 	threadexitsall(s);
218 }
219 
220 char*
221 onestring(int argc, char **argv)
222 {
223 	char *p;
224 	int i, n;
225 	static char buf[1024];
226 
227 	if(argc == 0)
228 		return "";
229 	p = buf;
230 	for(i=0; i<argc; i++){
231 		n = strlen(argv[i]);
232 		if(p+n+1 >= buf+sizeof buf)
233 			break;
234 		memmove(p, argv[i], n);
235 		p += n;
236 		*p++ = ' ';
237 	}
238 	p[-1] = 0;
239 	return buf;
240 }
241 
242 int
243 getec(CFid *efd)
244 {
245 	static char buf[8192];
246 	static char *bufp;
247 	static int nbuf;
248 
249 	if(nbuf == 0){
250 		nbuf = fsread(efd, buf, sizeof buf);
251 		if(nbuf <= 0)
252 			error(nil);
253 		bufp = buf;
254 	}
255 	--nbuf;
256 	return *bufp++;
257 }
258 
259 int
260 geten(CFid *efd)
261 {
262 	int n, c;
263 
264 	n = 0;
265 	while('0'<=(c=getec(efd)) && c<='9')
266 		n = n*10+(c-'0');
267 	if(c != ' ')
268 		error("event number syntax");
269 	return n;
270 }
271 
272 int
273 geter(CFid *efd, char *buf, int *nb)
274 {
275 	Rune r;
276 	int n;
277 
278 	r = getec(efd);
279 	buf[0] = r;
280 	n = 1;
281 	if(r < Runeself)
282 		goto Return;
283 	while(!fullrune(buf, n))
284 		buf[n++] = getec(efd);
285 	chartorune(&r, buf);
286     Return:
287 	*nb = n;
288 	return r;
289 }
290 
291 void
292 gete(CFid *efd, Event *e)
293 {
294 	int i, nb;
295 
296 	e->c1 = getec(efd);
297 	e->c2 = getec(efd);
298 	e->q0 = geten(efd);
299 	e->q1 = geten(efd);
300 	e->flag = geten(efd);
301 	e->nr = geten(efd);
302 	if(e->nr > EVENTSIZE)
303 		error("event string too long");
304 	e->nb = 0;
305 	for(i=0; i<e->nr; i++){
306 		e->r[i] = geter(efd, e->b+e->nb, &nb);
307 		e->nb += nb;
308 	}
309 	e->r[e->nr] = 0;
310 	e->b[e->nb] = 0;
311 	if(getec(efd) != '\n')
312 		error("event syntax 2");
313 }
314 
315 int
316 nrunes(char *s, int nb)
317 {
318 	int i, n;
319 	Rune r;
320 
321 	n = 0;
322 	for(i=0; i<nb; n++)
323 		i += chartorune(&r, s+i);
324 	return n;
325 }
326 
327 void
328 stdinproc(void *v)
329 {
330 	CFid *cfd = ctlfd;
331 	CFid *efd = eventfd;
332 	CFid *dfd = datafd;
333 	CFid *afd = addrfd;
334 	int fd0 = rcfd;
335 	Event e, e2, e3, e4;
336 
337 	USED(v);
338 
339 	for(;;){
340 		if(debug)
341 			fprint(2, "typing[%d,%d)\n", q.p, q.p+ntyper);
342 		gete(efd, &e);
343 		if(debug)
344 			fprint(2, "msg %c%c q[%d,%d)... ", e.c1, e.c2, e.q0, e.q1);
345 		qlock(&q.lk);
346 		switch(e.c1){
347 		default:
348 		Unknown:
349 			print("unknown message %c%c\n", e.c1, e.c2);
350 			break;
351 
352 		case 'E':	/* write to body; can't affect us */
353 			if(debug)
354 				fprint(2, "shift typing %d... ", e.q1-e.q0);
355 			q.p += e.q1-e.q0;
356 			break;
357 
358 		case 'F':	/* generated by our actions; ignore */
359 			break;
360 
361 		case 'K':
362 		case 'M':
363 			switch(e.c2){
364 			case 'I':
365 				if(e.q0 < q.p){
366 					if(debug)
367 						fprint(2, "shift typing %d... ", e.q1-e.q0);
368 					q.p += e.q1-e.q0;
369 				}
370 				else if(e.q0 <= q.p+ntyper){
371 					if(debug)
372 						fprint(2, "type... ");
373 					type(&e, fd0, afd, dfd);
374 				}
375 				break;
376 
377 			case 'D':
378 				q.p -= delete(&e);
379 				break;
380 
381 			case 'x':
382 			case 'X':
383 				if(e.flag & 2)
384 					gete(efd, &e2);
385 				if(e.flag & 8){
386 					gete(efd, &e3);
387 					gete(efd, &e4);
388 				}
389 				if(e.flag&1 || (e.c2=='x' && e.nr==0 && e2.nr==0)){
390 					/* send it straight back */
391 					fsfidprint(efd, "%c%c%d %d\n", e.c1, e.c2, e.q0, e.q1);
392 					break;
393 				}
394 				if(e.q0==e.q1 && (e.flag&2)){
395 					e2.flag = e.flag;
396 					e = e2;
397 				}
398 				if(e.flag & 8){
399 					if(e.q1 != e.q0){
400 						sende(&e, fd0, cfd, afd, dfd, 0);
401 						sende(&blank, fd0, cfd, afd, dfd, 0);
402 					}
403 					sende(&e3, fd0, cfd, afd, dfd, 1);
404 				}else	 if(e.q1 != e.q0)
405 					sende(&e, fd0, cfd, afd, dfd, 1);
406 				break;
407 
408 			case 'l':
409 			case 'L':
410 				/* just send it back */
411 				if(e.flag & 2)
412 					gete(efd, &e2);
413 				fsfidprint(efd, "%c%c%d %d\n", e.c1, e.c2, e.q0, e.q1);
414 				break;
415 
416 			case 'd':
417 			case 'i':
418 				break;
419 
420 			default:
421 				goto Unknown;
422 			}
423 		}
424 		qunlock(&q.lk);
425 	}
426 }
427 
428 void
429 stdoutproc(void *v)
430 {
431 	int fd1 = rcfd;
432 	CFid *afd = addrfd;
433 	CFid *dfd = datafd;
434 	int n, m, w, npart;
435 	char *buf, *s, *t;
436 	Rune r;
437 	char x[16], hold[UTFmax];
438 
439 	USED(v);
440 	buf = malloc(8192+UTFmax+1);
441 	npart = 0;
442 	for(;;){
443 		/* Let typing have a go -- maybe there's a rubout waiting. */
444 		yield();
445 		n = read(fd1, buf+npart, 8192);
446 		if(n < 0)
447 			error(nil);
448 		if(n == 0)
449 			continue;
450 
451 		/* squash NULs */
452 		s = memchr(buf+npart, 0, n);
453 		if(s){
454 			for(t=s; s<buf+npart+n; s++)
455 				if(*t = *s)	/* assign = */
456 					t++;
457 			n = t-(buf+npart);
458 		}
459 
460 		n += npart;
461 
462 		/* hold on to final partial rune */
463 		npart = 0;
464 		while(n>0 && (buf[n-1]&0xC0)){
465 			--n;
466 			npart++;
467 			if((buf[n]&0xC0)!=0x80){
468 				if(fullrune(buf+n, npart)){
469 					w = chartorune(&r, buf+n);
470 					n += w;
471 					npart -= w;
472 				}
473 				break;
474 			}
475 		}
476 		if(n > 0){
477 			memmove(hold, buf+n, npart);
478 			buf[n] = 0;
479 			n = label(buf, n);
480 			buf[n] = 0;
481 			qlock(&q.lk);
482 			m = sprint(x, "#%d", q.p);
483 			if(fswrite(afd, x, m) != m)
484 				error("stdout writing address");
485 			if(fswrite(dfd, buf, n) != n)
486 				error("stdout writing body");
487 			q.p += nrunes(buf, n);
488 			qunlock(&q.lk);
489 			memmove(buf, hold, npart);
490 		}
491 	}
492 }
493 
494 char wdir[256];
495 int
496 label(char *sr, int n)
497 {
498 	char *sl, *el, *er, *r;
499 
500 	er = sr+n;
501 	for(r=er-1; r>=sr; r--)
502 		if(*r == '\007')
503 			break;
504 	if(r < sr)
505 		return n;
506 
507 	el = r+1;
508 	if(el-sr > sizeof wdir)
509 		sr = el - sizeof wdir;
510 	for(sl=el-3; sl>=sr; sl--)
511 		if(sl[0]=='\033' && sl[1]==']' && sl[2]==';')
512 			break;
513 	if(sl < sr)
514 		return n;
515 
516 	*r = 0;
517 	snprint(wdir, sizeof wdir, "name %s/-%s\n0\n", sl+3, name);
518 	fswrite(ctlfd, wdir, strlen(wdir));
519 
520 	memmove(sl, el, er-el);
521 	n -= (el-sl);
522 	return n;
523 }
524 
525 int
526 delete(Event *e)
527 {
528 	uint q0, q1;
529 	int deltap;
530 
531 	q0 = e->q0;
532 	q1 = e->q1;
533 	if(q1 <= q.p)
534 		return e->q1-e->q0;
535 	if(q0 >= q.p+ntyper)
536 		return 0;
537 	deltap = 0;
538 	if(q0 < q.p){
539 		deltap = q.p-q0;
540 		q0 = 0;
541 	}else
542 		q0 -= q.p;
543 	if(q1 > q.p+ntyper)
544 		q1 = ntyper;
545 	else
546 		q1 -= q.p;
547 	deltype(q0, q1);
548 	return deltap;
549 }
550 
551 void
552 addtype(int c, uint p0, char *b, int nb, int nr)
553 {
554 	int i, w;
555 	Rune r;
556 	uint p;
557 	char *b0;
558 
559 	for(i=0; i<nb; i+=w){
560 		w = chartorune(&r, b+i);
561 		if((r==0x7F||r==3) && c=='K'){
562 			write(rcfd, "\x7F", 1);
563 			/* toss all typing */
564 			q.p += ntyper+nr;
565 			ntypebreak = 0;
566 			ntypeb = 0;
567 			ntyper = 0;
568 			/* buglet:  more than one delete ignored */
569 			return;
570 		}
571 		if(r=='\n' || r==0x04)
572 			ntypebreak++;
573 	}
574 	typing = realloc(typing, ntypeb+nb);
575 	if(typing == nil)
576 		error("realloc");
577 	if(p0 == ntyper)
578 		memmove(typing+ntypeb, b, nb);
579 	else{
580 		b0 = typing;
581 		for(p=0; p<p0 && b0<typing+ntypeb; p++){
582 			w = chartorune(&r, b0+i);
583 			b0 += w;
584 		}
585 		if(p != p0)
586 			error("typing: findrune");
587 		memmove(b0+nb, b0, (typing+ntypeb)-b0);
588 		memmove(b0, b, nb);
589 	}
590 	ntypeb += nb;
591 	ntyper += nr;
592 }
593 
594 void
595 sendtype(int fd0)
596 {
597 	int i, n, nr;
598 
599 	while(ntypebreak){
600 		for(i=0; i<ntypeb; i++)
601 			if(typing[i]=='\n' || typing[i]==0x04){
602 				n = i+1;
603 				i++;
604 				if(write(fd0, typing, n) != n)
605 					error("sending to program");
606 				nr = nrunes(typing, i);
607 				q.p += nr;
608 				ntyper -= nr;
609 				ntypeb -= i;
610 				memmove(typing, typing+i, ntypeb);
611 				ntypebreak--;
612 				goto cont2;
613 			}
614 		print("no breakchar\n");
615 		ntypebreak = 0;
616 cont2:;
617 	}
618 }
619 
620 void
621 deltype(uint p0, uint p1)
622 {
623 	int w;
624 	uint p, b0, b1;
625 	Rune r;
626 
627 	/* advance to p0 */
628 	b0 = 0;
629 	for(p=0; p<p0 && b0<ntypeb; p++){
630 		w = chartorune(&r, typing+b0);
631 		b0 += w;
632 	}
633 	if(p != p0)
634 		error("deltype 1");
635 	/* advance to p1 */
636 	b1 = b0;
637 	for(; p<p1 && b1<ntypeb; p++){
638 		w = chartorune(&r, typing+b1);
639 		b1 += w;
640 		if(r=='\n' || r==0x04)
641 			ntypebreak--;
642 	}
643 	if(p != p1)
644 		error("deltype 2");
645 	memmove(typing+b0, typing+b1, ntypeb-b1);
646 	ntypeb -= b1-b0;
647 	ntyper -= p1-p0;
648 }
649 
650 void
651 type(Event *e, int fd0, CFid *afd, CFid *dfd)
652 {
653 	int m, n, nr;
654 	char buf[128];
655 
656 	if(e->nr > 0)
657 		addtype(e->c1, e->q0-q.p, e->b, e->nb, e->nr);
658 	else{
659 		m = e->q0;
660 		while(m < e->q1){
661 			n = sprint(buf, "#%d", m);
662 			fswrite(afd, buf, n);
663 			n = fsread(dfd, buf, sizeof buf);
664 			nr = nrunes(buf, n);
665 			while(m+nr > e->q1){
666 				do; while(n>0 && (buf[--n]&0xC0)==0x80);
667 				--nr;
668 			}
669 			if(n == 0)
670 				break;
671 			addtype(e->c1, m-q.p, buf, n, nr);
672 			m += nr;
673 		}
674 	}
675 	sendtype(fd0);
676 }
677 
678 void
679 sende(Event *e, int fd0, CFid *cfd, CFid *afd, CFid *dfd, int donl)
680 {
681 	int l, m, n, nr, lastc, end;
682 	char abuf[16], buf[128];
683 
684 	end = q.p+ntyper;
685 	l = sprint(abuf, "#%d", end);
686 	fswrite(afd, abuf, l);
687 	if(e->nr > 0){
688 		fswrite(dfd, e->b, e->nb);
689 		addtype(e->c1, ntyper, e->b, e->nb, e->nr);
690 		lastc = e->r[e->nr-1];
691 	}else{
692 		m = e->q0;
693 		lastc = 0;
694 		while(m < e->q1){
695 			n = sprint(buf, "#%d", m);
696 			fswrite(afd, buf, n);
697 			n = fsread(dfd, buf, sizeof buf);
698 			nr = nrunes(buf, n);
699 			while(m+nr > e->q1){
700 				do; while(n>0 && (buf[--n]&0xC0)==0x80);
701 				--nr;
702 			}
703 			if(n == 0)
704 				break;
705 			l = sprint(abuf, "#%d", end);
706 			fswrite(afd, abuf, l);
707 			fswrite(dfd, buf, n);
708 			addtype(e->c1, ntyper, buf, n, nr);
709 			lastc = buf[n-1];
710 			m += nr;
711 			end += nr;
712 		}
713 	}
714 	if(donl && lastc!='\n'){
715 		fswrite(dfd, "\n", 1);
716 		addtype(e->c1, ntyper, "\n", 1, 1);
717 	}
718 	fswrite(cfd, "dot=addr", 8);
719 	sendtype(fd0);
720 }