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 
449 		/* squash NULs */
450 		s = memchr(buf+npart, 0, n);
451 		if(s){
452 			for(t=s; s<buf+npart+n; s++)
453 				if(*t = *s)	/* assign = */
454 					t++;
455 			n = t-(buf+npart);
456 		}
457 
458 		n += npart;
459 
460 		/* hold on to final partial rune */
461 		npart = 0;
462 		while(n>0 && (buf[n-1]&0xC0)){
463 			--n;
464 			npart++;
465 			if((buf[n]&0xC0)!=0x80){
466 				if(fullrune(buf+n, npart)){
467 					w = chartorune(&r, buf+n);
468 					n += w;
469 					npart -= w;
470 				}
471 				break;
472 			}
473 		}
474 		if(n > 0){
475 			memmove(hold, buf+n, npart);
476 			buf[n] = 0;
477 			n = label(buf, n);
478 			buf[n] = 0;
479 			qlock(&q.lk);
480 			m = sprint(x, "#%d", q.p);
481 			if(fswrite(afd, x, m) != m)
482 				error("stdout writing address");
483 			if(fswrite(dfd, buf, n) != n)
484 				error("stdout writing body");
485 			q.p += nrunes(buf, n);
486 			qunlock(&q.lk);
487 			memmove(buf, hold, npart);
488 		}
489 	}
490 }
491 
492 char wdir[256];
493 int
494 label(char *sr, int n)
495 {
496 	char *sl, *el, *er, *r;
497 
498 	er = sr+n;
499 	for(r=er-1; r>=sr; r--)
500 		if(*r == '\007')
501 			break;
502 	if(r < sr)
503 		return n;
504 
505 	el = r+1;
506 	if(el-sr > sizeof wdir)
507 		sr = el - sizeof wdir;
508 	for(sl=el-3; sl>=sr; sl--)
509 		if(sl[0]=='\033' && sl[1]==']' && sl[2]==';')
510 			break;
511 	if(sl < sr)
512 		return n;
513 
514 	*r = 0;
515 	snprint(wdir, sizeof wdir, "name %s/-%s\n0\n", sl+3, name);
516 	fswrite(ctlfd, wdir, strlen(wdir));
517 
518 	memmove(sl, el, er-el);
519 	n -= (el-sl);
520 	return n;
521 }
522 
523 int
524 delete(Event *e)
525 {
526 	uint q0, q1;
527 	int deltap;
528 
529 	q0 = e->q0;
530 	q1 = e->q1;
531 	if(q1 <= q.p)
532 		return e->q1-e->q0;
533 	if(q0 >= q.p+ntyper)
534 		return 0;
535 	deltap = 0;
536 	if(q0 < q.p){
537 		deltap = q.p-q0;
538 		q0 = 0;
539 	}else
540 		q0 -= q.p;
541 	if(q1 > q.p+ntyper)
542 		q1 = ntyper;
543 	else
544 		q1 -= q.p;
545 	deltype(q0, q1);
546 	return deltap;
547 }
548 
549 void
550 addtype(int c, uint p0, char *b, int nb, int nr)
551 {
552 	int i, w;
553 	Rune r;
554 	uint p;
555 	char *b0;
556 
557 	for(i=0; i<nb; i+=w){
558 		w = chartorune(&r, b+i);
559 		if((r==0x7F||r==3) && c=='K'){
560 			write(rcfd, "\x7F", 1);
561 			/* toss all typing */
562 			q.p += ntyper+nr;
563 			ntypebreak = 0;
564 			ntypeb = 0;
565 			ntyper = 0;
566 			/* buglet:  more than one delete ignored */
567 			return;
568 		}
569 		if(r=='\n' || r==0x04)
570 			ntypebreak++;
571 	}
572 	typing = realloc(typing, ntypeb+nb);
573 	if(typing == nil)
574 		error("realloc");
575 	if(p0 == ntyper)
576 		memmove(typing+ntypeb, b, nb);
577 	else{
578 		b0 = typing;
579 		for(p=0; p<p0 && b0<typing+ntypeb; p++){
580 			w = chartorune(&r, b0+i);
581 			b0 += w;
582 		}
583 		if(p != p0)
584 			error("typing: findrune");
585 		memmove(b0+nb, b0, (typing+ntypeb)-b0);
586 		memmove(b0, b, nb);
587 	}
588 	ntypeb += nb;
589 	ntyper += nr;
590 }
591 
592 void
593 sendtype(int fd0)
594 {
595 	int i, n, nr;
596 
597 	while(ntypebreak){
598 		for(i=0; i<ntypeb; i++)
599 			if(typing[i]=='\n' || typing[i]==0x04){
600 				n = i+1;
601 				i++;
602 				if(write(fd0, typing, n) != n)
603 					error("sending to program");
604 				nr = nrunes(typing, i);
605 				q.p += nr;
606 				ntyper -= nr;
607 				ntypeb -= i;
608 				memmove(typing, typing+i, ntypeb);
609 				ntypebreak--;
610 				goto cont2;
611 			}
612 		print("no breakchar\n");
613 		ntypebreak = 0;
614 cont2:;
615 	}
616 }
617 
618 void
619 deltype(uint p0, uint p1)
620 {
621 	int w;
622 	uint p, b0, b1;
623 	Rune r;
624 
625 	/* advance to p0 */
626 	b0 = 0;
627 	for(p=0; p<p0 && b0<ntypeb; p++){
628 		w = chartorune(&r, typing+b0);
629 		b0 += w;
630 	}
631 	if(p != p0)
632 		error("deltype 1");
633 	/* advance to p1 */
634 	b1 = b0;
635 	for(; p<p1 && b1<ntypeb; p++){
636 		w = chartorune(&r, typing+b1);
637 		b1 += w;
638 		if(r=='\n' || r==0x04)
639 			ntypebreak--;
640 	}
641 	if(p != p1)
642 		error("deltype 2");
643 	memmove(typing+b0, typing+b1, ntypeb-b1);
644 	ntypeb -= b1-b0;
645 	ntyper -= p1-p0;
646 }
647 
648 void
649 type(Event *e, int fd0, CFid *afd, CFid *dfd)
650 {
651 	int m, n, nr;
652 	char buf[128];
653 
654 	if(e->nr > 0)
655 		addtype(e->c1, e->q0-q.p, e->b, e->nb, e->nr);
656 	else{
657 		m = e->q0;
658 		while(m < e->q1){
659 			n = sprint(buf, "#%d", m);
660 			fswrite(afd, buf, n);
661 			n = fsread(dfd, buf, sizeof buf);
662 			nr = nrunes(buf, n);
663 			while(m+nr > e->q1){
664 				do; while(n>0 && (buf[--n]&0xC0)==0x80);
665 				--nr;
666 			}
667 			if(n == 0)
668 				break;
669 			addtype(e->c1, m-q.p, buf, n, nr);
670 			m += nr;
671 		}
672 	}
673 	sendtype(fd0);
674 }
675 
676 void
677 sende(Event *e, int fd0, CFid *cfd, CFid *afd, CFid *dfd, int donl)
678 {
679 	int l, m, n, nr, lastc, end;
680 	char abuf[16], buf[128];
681 
682 	end = q.p+ntyper;
683 	l = sprint(abuf, "#%d", end);
684 	fswrite(afd, abuf, l);
685 	if(e->nr > 0){
686 		fswrite(dfd, e->b, e->nb);
687 		addtype(e->c1, ntyper, e->b, e->nb, e->nr);
688 		lastc = e->r[e->nr-1];
689 	}else{
690 		m = e->q0;
691 		lastc = 0;
692 		while(m < e->q1){
693 			n = sprint(buf, "#%d", m);
694 			fswrite(afd, buf, n);
695 			n = fsread(dfd, buf, sizeof buf);
696 			nr = nrunes(buf, n);
697 			while(m+nr > e->q1){
698 				do; while(n>0 && (buf[--n]&0xC0)==0x80);
699 				--nr;
700 			}
701 			if(n == 0)
702 				break;
703 			l = sprint(abuf, "#%d", end);
704 			fswrite(afd, abuf, l);
705 			fswrite(dfd, buf, n);
706 			addtype(e->c1, ntyper, buf, n, nr);
707 			lastc = buf[n-1];
708 			m += nr;
709 			end += nr;
710 		}
711 	}
712 	if(donl && lastc!='\n'){
713 		fswrite(dfd, "\n", 1);
714 		addtype(e->c1, ntyper, "\n", 1, 1);
715 	}
716 	fswrite(cfd, "dot=addr", 8);
717 	sendtype(fd0);
718 }