• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*	$OpenBSD: tree.c,v 1.20 2012/06/27 07:17:19 otto Exp $	*/
2 
3 /*-
4  * Copyright (c) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010,
5  *		 2011, 2012, 2013, 2015
6  *	Thorsten Glaser <tg@mirbsd.org>
7  *
8  * Provided that these terms and disclaimer and all copyright notices
9  * are retained or reproduced in an accompanying document, permission
10  * is granted to deal in this work without restriction, including un-
11  * limited rights to use, publicly perform, distribute, sell, modify,
12  * merge, give away, or sublicence.
13  *
14  * This work is provided "AS IS" and WITHOUT WARRANTY of any kind, to
15  * the utmost extent permitted by applicable law, neither express nor
16  * implied; without malicious intent or gross negligence. In no event
17  * may a licensor, author or contributor be held liable for indirect,
18  * direct, other damage, loss, or other issues arising in any way out
19  * of dealing in the work, even if advised of the possibility of such
20  * damage or existence of a defect, except proven that it results out
21  * of said person's immediate fault when using the work as intended.
22  */
23 
24 #include "sh.h"
25 
26 __RCSID("$MirOS: src/bin/mksh/tree.c,v 1.72.2.1 2015/04/12 22:32:35 tg Exp $");
27 
28 #define INDENT	8
29 
30 static void ptree(struct op *, int, struct shf *);
31 static void pioact(struct shf *, struct ioword *);
32 static const char *wdvarput(struct shf *, const char *, int, int);
33 static void vfptreef(struct shf *, int, const char *, va_list);
34 static struct ioword **iocopy(struct ioword **, Area *);
35 static void iofree(struct ioword **, Area *);
36 
37 /* "foo& ; bar" and "foo |& ; bar" are invalid */
38 static bool prevent_semicolon;
39 
40 static const char Telif_pT[] = "elif %T";
41 
42 /*
43  * print a command tree
44  */
45 static void
ptree(struct op * t,int indent,struct shf * shf)46 ptree(struct op *t, int indent, struct shf *shf)
47 {
48 	const char **w;
49 	struct ioword **ioact;
50 	struct op *t1;
51 	int i;
52 
53  Chain:
54 	if (t == NULL)
55 		return;
56 	switch (t->type) {
57 	case TCOM:
58 		prevent_semicolon = false;
59 		/*
60 		 * special-case 'var=<<EOF' (rough; see
61 		 * exec.c:execute() for full code)
62 		 */
63 		if (
64 		    /* we have zero arguments, i.e. no programme to run */
65 		    t->args[0] == NULL &&
66 		    /* we have exactly one variable assignment */
67 		    t->vars[0] != NULL && t->vars[1] == NULL &&
68 		    /* we have exactly one I/O redirection */
69 		    t->ioact != NULL && t->ioact[0] != NULL &&
70 		    t->ioact[1] == NULL &&
71 		    /* of type "here document" (or "here string") */
72 		    (t->ioact[0]->ioflag & IOTYPE) == IOHERE) {
73 			fptreef(shf, indent, "%S", t->vars[0]);
74 			break;
75 		}
76 
77 		if (t->vars) {
78 			w = (const char **)t->vars;
79 			while (*w)
80 				fptreef(shf, indent, "%S ", *w++);
81 		} else
82 			shf_puts("#no-vars# ", shf);
83 		if (t->args) {
84 			w = t->args;
85 			while (*w)
86 				fptreef(shf, indent, "%S ", *w++);
87 		} else
88 			shf_puts("#no-args# ", shf);
89 		break;
90 	case TEXEC:
91 		t = t->left;
92 		goto Chain;
93 	case TPAREN:
94 		fptreef(shf, indent + 2, "( %T) ", t->left);
95 		break;
96 	case TPIPE:
97 		fptreef(shf, indent, "%T| ", t->left);
98 		t = t->right;
99 		goto Chain;
100 	case TLIST:
101 		fptreef(shf, indent, "%T%;", t->left);
102 		t = t->right;
103 		goto Chain;
104 	case TOR:
105 	case TAND:
106 		fptreef(shf, indent, "%T%s %T",
107 		    t->left, (t->type == TOR) ? "||" : "&&", t->right);
108 		break;
109 	case TBANG:
110 		shf_puts("! ", shf);
111 		prevent_semicolon = false;
112 		t = t->right;
113 		goto Chain;
114 	case TDBRACKET:
115 		w = t->args;
116 		shf_puts("[[", shf);
117 		while (*w)
118 			fptreef(shf, indent, " %S", *w++);
119 		shf_puts(" ]] ", shf);
120 		break;
121 	case TSELECT:
122 	case TFOR:
123 		fptreef(shf, indent, "%s %s ",
124 		    (t->type == TFOR) ? "for" : Tselect, t->str);
125 		if (t->vars != NULL) {
126 			shf_puts("in ", shf);
127 			w = (const char **)t->vars;
128 			while (*w)
129 				fptreef(shf, indent, "%S ", *w++);
130 			fptreef(shf, indent, "%;");
131 		}
132 		fptreef(shf, indent + INDENT, "do%N%T", t->left);
133 		fptreef(shf, indent, "%;done ");
134 		break;
135 	case TCASE:
136 		fptreef(shf, indent, "case %S in", t->str);
137 		for (t1 = t->left; t1 != NULL; t1 = t1->right) {
138 			fptreef(shf, indent, "%N(");
139 			w = (const char **)t1->vars;
140 			while (*w) {
141 				fptreef(shf, indent, "%S%c", *w,
142 				    (w[1] != NULL) ? '|' : ')');
143 				++w;
144 			}
145 			fptreef(shf, indent + INDENT, "%N%T%N;%c", t1->left,
146 			    t1->u.charflag);
147 		}
148 		fptreef(shf, indent, "%Nesac ");
149 		break;
150 #ifdef DEBUG
151 	case TELIF:
152 		internal_errorf("TELIF in tree.c:ptree() unexpected");
153 		/* FALLTHROUGH */
154 #endif
155 	case TIF:
156 		i = 2;
157 		t1 = t;
158 		goto process_TIF;
159 		do {
160 			t1 = t1->right;
161 			i = 0;
162 			fptreef(shf, indent, "%;");
163  process_TIF:
164 			/* 5 == strlen("elif ") */
165 			fptreef(shf, indent + 5 - i, Telif_pT + i, t1->left);
166 			t1 = t1->right;
167 			if (t1->left != NULL) {
168 				fptreef(shf, indent, "%;");
169 				fptreef(shf, indent + INDENT, "%s%N%T",
170 				    "then", t1->left);
171 			}
172 		} while (t1->right && t1->right->type == TELIF);
173 		if (t1->right != NULL) {
174 			fptreef(shf, indent, "%;");
175 			fptreef(shf, indent + INDENT, "%s%N%T",
176 			    "else", t1->right);
177 		}
178 		fptreef(shf, indent, "%;fi ");
179 		break;
180 	case TWHILE:
181 	case TUNTIL:
182 		/* 6 == strlen("while "/"until ") */
183 		fptreef(shf, indent + 6, "%s %T",
184 		    (t->type == TWHILE) ? "while" : "until",
185 		    t->left);
186 		fptreef(shf, indent, "%;");
187 		fptreef(shf, indent + INDENT, "do%N%T", t->right);
188 		fptreef(shf, indent, "%;done ");
189 		break;
190 	case TBRACE:
191 		fptreef(shf, indent + INDENT, "{%N%T", t->left);
192 		fptreef(shf, indent, "%;} ");
193 		break;
194 	case TCOPROC:
195 		fptreef(shf, indent, "%T|& ", t->left);
196 		prevent_semicolon = true;
197 		break;
198 	case TASYNC:
199 		fptreef(shf, indent, "%T& ", t->left);
200 		prevent_semicolon = true;
201 		break;
202 	case TFUNCT:
203 		fpFUNCTf(shf, indent, tobool(t->u.ksh_func), t->str, t->left);
204 		break;
205 	case TTIME:
206 		fptreef(shf, indent, "%s %T", "time", t->left);
207 		break;
208 	default:
209 		shf_puts("<botch>", shf);
210 		prevent_semicolon = false;
211 		break;
212 	}
213 	if ((ioact = t->ioact) != NULL) {
214 		bool need_nl = false;
215 
216 		while (*ioact != NULL)
217 			pioact(shf, *ioact++);
218 		/* Print here documents after everything else... */
219 		ioact = t->ioact;
220 		while (*ioact != NULL) {
221 			struct ioword *iop = *ioact++;
222 
223 			/* heredoc is NULL when tracing (set -x) */
224 			if ((iop->ioflag & (IOTYPE | IOHERESTR)) == IOHERE &&
225 			    iop->heredoc) {
226 				shf_putc('\n', shf);
227 				shf_puts(iop->heredoc, shf);
228 				fptreef(shf, indent, "%s",
229 				    iop->ioflag & IONDELIM ? "<<" :
230 				    evalstr(iop->delim, 0));
231 				need_nl = true;
232 			}
233 		}
234 		/*
235 		 * Last delimiter must be followed by a newline (this
236 		 * often leads to an extra blank line, but it's not
237 		 * worth worrying about)
238 		 */
239 		if (need_nl) {
240 			shf_putc('\n', shf);
241 			prevent_semicolon = true;
242 		}
243 	}
244 }
245 
246 static void
pioact(struct shf * shf,struct ioword * iop)247 pioact(struct shf *shf, struct ioword *iop)
248 {
249 	unsigned short flag = iop->ioflag;
250 	unsigned short type = flag & IOTYPE;
251 	short expected;
252 
253 	expected = (type == IOREAD || type == IORDWR || type == IOHERE) ? 0 :
254 	    (type == IOCAT || type == IOWRITE) ? 1 :
255 	    (type == IODUP && (iop->unit == !(flag & IORDUP))) ? iop->unit :
256 	    iop->unit + 1;
257 	if (iop->unit != expected)
258 		shf_fprintf(shf, "%d", (int)iop->unit);
259 
260 	switch (type) {
261 	case IOREAD:
262 		shf_putc('<', shf);
263 		break;
264 	case IOHERE:
265 		shf_puts("<<", shf);
266 		if (flag & IOSKIP)
267 			shf_putc('-', shf);
268 		break;
269 	case IOCAT:
270 		shf_puts(">>", shf);
271 		break;
272 	case IOWRITE:
273 		shf_putc('>', shf);
274 		if (flag & IOCLOB)
275 			shf_putc('|', shf);
276 		break;
277 	case IORDWR:
278 		shf_puts("<>", shf);
279 		break;
280 	case IODUP:
281 		shf_puts(flag & IORDUP ? "<&" : ">&", shf);
282 		break;
283 	}
284 	/* name/delim are NULL when printing syntax errors */
285 	if (type == IOHERE) {
286 		if (iop->delim)
287 			wdvarput(shf, iop->delim, 0, WDS_TPUTS);
288 		if (flag & IOHERESTR)
289 			shf_putc(' ', shf);
290 	} else if (iop->name) {
291 		if (flag & IONAMEXP)
292 			print_value_quoted(shf, iop->name);
293 		else
294 			wdvarput(shf, iop->name, 0, WDS_TPUTS);
295 		shf_putc(' ', shf);
296 	}
297 	prevent_semicolon = false;
298 }
299 
300 /* variant of fputs for ptreef and wdstrip */
301 static const char *
wdvarput(struct shf * shf,const char * wp,int quotelevel,int opmode)302 wdvarput(struct shf *shf, const char *wp, int quotelevel, int opmode)
303 {
304 	int c;
305 	const char *cs;
306 
307 	/*-
308 	 * problems:
309 	 *	`...` -> $(...)
310 	 *	'foo' -> "foo"
311 	 *	x${foo:-"hi"} -> x${foo:-hi} unless WDS_TPUTS
312 	 *	x${foo:-'hi'} -> x${foo:-hi} unless WDS_KEEPQ
313 	 * could change encoding to:
314 	 *	OQUOTE ["'] ... CQUOTE ["']
315 	 *	COMSUB [(`] ...\0	(handle $ ` \ and maybe " in `...` case)
316 	 */
317 	while (/* CONSTCOND */ 1)
318 		switch (*wp++) {
319 		case EOS:
320 			return (--wp);
321 		case ADELIM:
322 		case CHAR:
323 			c = *wp++;
324 			if ((opmode & WDS_MAGIC) &&
325 			    (ISMAGIC(c) || c == '[' || c == '!' ||
326 			    c == '-' || c == ']' || c == '*' || c == '?'))
327 				shf_putc(MAGIC, shf);
328 			shf_putc(c, shf);
329 			break;
330 		case QCHAR: {
331 			bool doq;
332 
333 			c = *wp++;
334 			doq = (c == '"' || c == '`' || c == '$' || c == '\\');
335 			if (opmode & WDS_TPUTS) {
336 				if (quotelevel == 0)
337 					doq = true;
338 			} else {
339 				if (!(opmode & WDS_KEEPQ))
340 					doq = false;
341 			}
342 			if (doq)
343 				shf_putc('\\', shf);
344 			shf_putc(c, shf);
345 			break;
346 		}
347 		case COMSUB:
348 			shf_puts("$(", shf);
349 			cs = ")";
350  pSUB:
351 			while ((c = *wp++) != 0)
352 				shf_putc(c, shf);
353 			shf_puts(cs, shf);
354 			break;
355 		case FUNSUB:
356 			c = ' ';
357 			if (0)
358 				/* FALLTHROUGH */
359 		case VALSUB:
360 			  c = '|';
361 			shf_putc('$', shf);
362 			shf_putc('{', shf);
363 			shf_putc(c, shf);
364 			cs = ";}";
365 			goto pSUB;
366 		case EXPRSUB:
367 			shf_puts("$((", shf);
368 			cs = "))";
369 			goto pSUB;
370 		case OQUOTE:
371 			if (opmode & WDS_TPUTS) {
372 				quotelevel++;
373 				shf_putc('"', shf);
374 			}
375 			break;
376 		case CQUOTE:
377 			if (opmode & WDS_TPUTS) {
378 				if (quotelevel)
379 					quotelevel--;
380 				shf_putc('"', shf);
381 			}
382 			break;
383 		case OSUBST:
384 			shf_putc('$', shf);
385 			if (*wp++ == '{')
386 				shf_putc('{', shf);
387 			while ((c = *wp++) != 0)
388 				shf_putc(c, shf);
389 			wp = wdvarput(shf, wp, 0, opmode);
390 			break;
391 		case CSUBST:
392 			if (*wp++ == '}')
393 				shf_putc('}', shf);
394 			return (wp);
395 		case OPAT:
396 			if (opmode & WDS_MAGIC) {
397 				shf_putc(MAGIC, shf);
398 				shf_putchar(*wp++ | 0x80, shf);
399 			} else {
400 				shf_putchar(*wp++, shf);
401 				shf_putc('(', shf);
402 			}
403 			break;
404 		case SPAT:
405 			c = '|';
406 			if (0)
407 		case CPAT:
408 				c = /*(*/ ')';
409 			if (opmode & WDS_MAGIC)
410 				shf_putc(MAGIC, shf);
411 			shf_putc(c, shf);
412 			break;
413 		}
414 }
415 
416 /*
417  * this is the _only_ way to reliably handle
418  * variable args with an ANSI compiler
419  */
420 /* VARARGS */
421 void
fptreef(struct shf * shf,int indent,const char * fmt,...)422 fptreef(struct shf *shf, int indent, const char *fmt, ...)
423 {
424 	va_list va;
425 
426 	va_start(va, fmt);
427 	vfptreef(shf, indent, fmt, va);
428 	va_end(va);
429 }
430 
431 /* VARARGS */
432 char *
snptreef(char * s,ssize_t n,const char * fmt,...)433 snptreef(char *s, ssize_t n, const char *fmt, ...)
434 {
435 	va_list va;
436 	struct shf shf;
437 
438 	shf_sopen(s, n, SHF_WR | (s ? 0 : SHF_DYNAMIC), &shf);
439 
440 	va_start(va, fmt);
441 	vfptreef(&shf, 0, fmt, va);
442 	va_end(va);
443 
444 	/* shf_sclose NUL terminates */
445 	return (shf_sclose(&shf));
446 }
447 
448 static void
vfptreef(struct shf * shf,int indent,const char * fmt,va_list va)449 vfptreef(struct shf *shf, int indent, const char *fmt, va_list va)
450 {
451 	int c;
452 
453 	while ((c = *fmt++)) {
454 		if (c == '%') {
455 			switch ((c = *fmt++)) {
456 			case 'c':
457 				/* character (octet, probably) */
458 				shf_putchar(va_arg(va, int), shf);
459 				break;
460 			case 's':
461 				/* string */
462 				shf_puts(va_arg(va, char *), shf);
463 				break;
464 			case 'S':
465 				/* word */
466 				wdvarput(shf, va_arg(va, char *), 0, WDS_TPUTS);
467 				break;
468 			case 'd':
469 				/* signed decimal */
470 				shf_fprintf(shf, "%d", va_arg(va, int));
471 				break;
472 			case 'u':
473 				/* unsigned decimal */
474 				shf_fprintf(shf, "%u", va_arg(va, unsigned int));
475 				break;
476 			case 'T':
477 				/* format tree */
478 				ptree(va_arg(va, struct op *), indent, shf);
479 				goto dont_trash_prevent_semicolon;
480 			case ';':
481 				/* newline or ; */
482 			case 'N':
483 				/* newline or space */
484 				if (shf->flags & SHF_STRING) {
485 					if (c == ';' && !prevent_semicolon)
486 						shf_putc(';', shf);
487 					shf_putc(' ', shf);
488 				} else {
489 					int i;
490 
491 					shf_putc('\n', shf);
492 					i = indent;
493 					while (i >= 8) {
494 						shf_putc('\t', shf);
495 						i -= 8;
496 					}
497 					while (i--)
498 						shf_putc(' ', shf);
499 				}
500 				break;
501 			case 'R':
502 				/* I/O redirection */
503 				pioact(shf, va_arg(va, struct ioword *));
504 				break;
505 			default:
506 				shf_putc(c, shf);
507 				break;
508 			}
509 		} else
510 			shf_putc(c, shf);
511 		prevent_semicolon = false;
512  dont_trash_prevent_semicolon:
513 		;
514 	}
515 }
516 
517 /*
518  * copy tree (for function definition)
519  */
520 struct op *
tcopy(struct op * t,Area * ap)521 tcopy(struct op *t, Area *ap)
522 {
523 	struct op *r;
524 	const char **tw;
525 	char **rw;
526 
527 	if (t == NULL)
528 		return (NULL);
529 
530 	r = alloc(sizeof(struct op), ap);
531 
532 	r->type = t->type;
533 	r->u.evalflags = t->u.evalflags;
534 
535 	if (t->type == TCASE)
536 		r->str = wdcopy(t->str, ap);
537 	else
538 		strdupx(r->str, t->str, ap);
539 
540 	if (t->vars == NULL)
541 		r->vars = NULL;
542 	else {
543 		tw = (const char **)t->vars;
544 		while (*tw)
545 			++tw;
546 		rw = r->vars = alloc2(tw - (const char **)t->vars + 1,
547 		    sizeof(*tw), ap);
548 		tw = (const char **)t->vars;
549 		while (*tw)
550 			*rw++ = wdcopy(*tw++, ap);
551 		*rw = NULL;
552 	}
553 
554 	if (t->args == NULL)
555 		r->args = NULL;
556 	else {
557 		tw = t->args;
558 		while (*tw)
559 			++tw;
560 		r->args = (const char **)(rw = alloc2(tw - t->args + 1,
561 		    sizeof(*tw), ap));
562 		tw = t->args;
563 		while (*tw)
564 			*rw++ = wdcopy(*tw++, ap);
565 		*rw = NULL;
566 	}
567 
568 	r->ioact = (t->ioact == NULL) ? NULL : iocopy(t->ioact, ap);
569 
570 	r->left = tcopy(t->left, ap);
571 	r->right = tcopy(t->right, ap);
572 	r->lineno = t->lineno;
573 
574 	return (r);
575 }
576 
577 char *
wdcopy(const char * wp,Area * ap)578 wdcopy(const char *wp, Area *ap)
579 {
580 	size_t len;
581 
582 	len = wdscan(wp, EOS) - wp;
583 	return (memcpy(alloc(len, ap), wp, len));
584 }
585 
586 /* return the position of prefix c in wp plus 1 */
587 const char *
wdscan(const char * wp,int c)588 wdscan(const char *wp, int c)
589 {
590 	int nest = 0;
591 
592 	while (/* CONSTCOND */ 1)
593 		switch (*wp++) {
594 		case EOS:
595 			return (wp);
596 		case ADELIM:
597 			if (c == ADELIM)
598 				return (wp + 1);
599 			/* FALLTHROUGH */
600 		case CHAR:
601 		case QCHAR:
602 			wp++;
603 			break;
604 		case COMSUB:
605 		case FUNSUB:
606 		case VALSUB:
607 		case EXPRSUB:
608 			while (*wp++ != 0)
609 				;
610 			break;
611 		case OQUOTE:
612 		case CQUOTE:
613 			break;
614 		case OSUBST:
615 			nest++;
616 			while (*wp++ != '\0')
617 				;
618 			break;
619 		case CSUBST:
620 			wp++;
621 			if (c == CSUBST && nest == 0)
622 				return (wp);
623 			nest--;
624 			break;
625 		case OPAT:
626 			nest++;
627 			wp++;
628 			break;
629 		case SPAT:
630 		case CPAT:
631 			if (c == wp[-1] && nest == 0)
632 				return (wp);
633 			if (wp[-1] == CPAT)
634 				nest--;
635 			break;
636 		default:
637 			internal_warningf(
638 			    "wdscan: unknown char 0x%x (carrying on)",
639 			    wp[-1]);
640 		}
641 }
642 
643 /*
644  * return a copy of wp without any of the mark up characters and with
645  * quote characters (" ' \) stripped. (string is allocated from ATEMP)
646  */
647 char *
wdstrip(const char * wp,int opmode)648 wdstrip(const char *wp, int opmode)
649 {
650 	struct shf shf;
651 
652 	shf_sopen(NULL, 32, SHF_WR | SHF_DYNAMIC, &shf);
653 	wdvarput(&shf, wp, 0, opmode);
654 	/* shf_sclose NUL terminates */
655 	return (shf_sclose(&shf));
656 }
657 
658 static struct ioword **
iocopy(struct ioword ** iow,Area * ap)659 iocopy(struct ioword **iow, Area *ap)
660 {
661 	struct ioword **ior;
662 	int i;
663 
664 	ior = iow;
665 	while (*ior)
666 		++ior;
667 	ior = alloc2(ior - iow + 1, sizeof(struct ioword *), ap);
668 
669 	for (i = 0; iow[i] != NULL; i++) {
670 		struct ioword *p, *q;
671 
672 		p = iow[i];
673 		q = alloc(sizeof(struct ioword), ap);
674 		ior[i] = q;
675 		*q = *p;
676 		if (p->name != NULL)
677 			q->name = wdcopy(p->name, ap);
678 		if (p->delim != NULL)
679 			q->delim = wdcopy(p->delim, ap);
680 		if (p->heredoc != NULL)
681 			strdupx(q->heredoc, p->heredoc, ap);
682 	}
683 	ior[i] = NULL;
684 
685 	return (ior);
686 }
687 
688 /*
689  * free tree (for function definition)
690  */
691 void
tfree(struct op * t,Area * ap)692 tfree(struct op *t, Area *ap)
693 {
694 	char **w;
695 
696 	if (t == NULL)
697 		return;
698 
699 	if (t->str != NULL)
700 		afree(t->str, ap);
701 
702 	if (t->vars != NULL) {
703 		for (w = t->vars; *w != NULL; w++)
704 			afree(*w, ap);
705 		afree(t->vars, ap);
706 	}
707 
708 	if (t->args != NULL) {
709 		/*XXX we assume the caller is right */
710 		union mksh_ccphack cw;
711 
712 		cw.ro = t->args;
713 		for (w = cw.rw; *w != NULL; w++)
714 			afree(*w, ap);
715 		afree(t->args, ap);
716 	}
717 
718 	if (t->ioact != NULL)
719 		iofree(t->ioact, ap);
720 
721 	tfree(t->left, ap);
722 	tfree(t->right, ap);
723 
724 	afree(t, ap);
725 }
726 
727 static void
iofree(struct ioword ** iow,Area * ap)728 iofree(struct ioword **iow, Area *ap)
729 {
730 	struct ioword **iop;
731 	struct ioword *p;
732 
733 	iop = iow;
734 	while ((p = *iop++) != NULL) {
735 		if (p->name != NULL)
736 			afree(p->name, ap);
737 		if (p->delim != NULL)
738 			afree(p->delim, ap);
739 		if (p->heredoc != NULL)
740 			afree(p->heredoc, ap);
741 		afree(p, ap);
742 	}
743 	afree(iow, ap);
744 }
745 
746 void
fpFUNCTf(struct shf * shf,int i,bool isksh,const char * k,struct op * v)747 fpFUNCTf(struct shf *shf, int i, bool isksh, const char *k, struct op *v)
748 {
749 	if (isksh)
750 		fptreef(shf, i, "%s %s %T", Tfunction, k, v);
751 	else
752 		fptreef(shf, i, "%s() %T", k, v);
753 }
754 
755 
756 /* for jobs.c */
757 void
vistree(char * dst,size_t sz,struct op * t)758 vistree(char *dst, size_t sz, struct op *t)
759 {
760 	unsigned int c;
761 	char *cp, *buf;
762 	size_t n;
763 
764 	buf = alloc(sz + 16, ATEMP);
765 	snptreef(buf, sz + 16, "%T", t);
766 	cp = buf;
767  vist_loop:
768 	if (UTFMODE && (n = utf_mbtowc(&c, cp)) != (size_t)-1) {
769 		if (c == 0 || n >= sz)
770 			/* NUL or not enough free space */
771 			goto vist_out;
772 		/* copy multibyte char */
773 		sz -= n;
774 		while (n--)
775 			*dst++ = *cp++;
776 		goto vist_loop;
777 	}
778 	if (--sz == 0 || (c = (unsigned char)(*cp++)) == 0)
779 		/* NUL or not enough free space */
780 		goto vist_out;
781 	if (ISCTRL(c & 0x7F)) {
782 		/* C0 or C1 control character or DEL */
783 		if (--sz == 0)
784 			/* not enough free space for two chars */
785 			goto vist_out;
786 		*dst++ = (c & 0x80) ? '$' : '^';
787 		c = UNCTRL(c & 0x7F);
788 	} else if (UTFMODE && c > 0x7F) {
789 		/* better not try to display broken multibyte chars */
790 		/* also go easy on the Unicode: no U+FFFD here */
791 		c = '?';
792 	}
793 	*dst++ = c;
794 	goto vist_loop;
795 
796  vist_out:
797 	*dst = '\0';
798 	afree(buf, ATEMP);
799 }
800 
801 #ifdef DEBUG
802 void
dumpchar(struct shf * shf,int c)803 dumpchar(struct shf *shf, int c)
804 {
805 	if (ISCTRL(c & 0x7F)) {
806 		/* C0 or C1 control character or DEL */
807 		shf_putc((c & 0x80) ? '$' : '^', shf);
808 		c = UNCTRL(c & 0x7F);
809 	}
810 	shf_putc(c, shf);
811 }
812 
813 /* see: wdvarput */
814 static const char *
dumpwdvar_i(struct shf * shf,const char * wp,int quotelevel)815 dumpwdvar_i(struct shf *shf, const char *wp, int quotelevel)
816 {
817 	int c;
818 
819 	while (/* CONSTCOND */ 1) {
820 		switch(*wp++) {
821 		case EOS:
822 			shf_puts("EOS", shf);
823 			return (--wp);
824 		case ADELIM:
825 			shf_puts("ADELIM=", shf);
826 			if (0)
827 		case CHAR:
828 				shf_puts("CHAR=", shf);
829 			dumpchar(shf, *wp++);
830 			break;
831 		case QCHAR:
832 			shf_puts("QCHAR<", shf);
833 			c = *wp++;
834 			if (quotelevel == 0 ||
835 			    (c == '"' || c == '`' || c == '$' || c == '\\'))
836 				shf_putc('\\', shf);
837 			dumpchar(shf, c);
838 			goto closeandout;
839 		case COMSUB:
840 			shf_puts("COMSUB<", shf);
841  dumpsub:
842 			while ((c = *wp++) != 0)
843 				dumpchar(shf, c);
844  closeandout:
845 			shf_putc('>', shf);
846 			break;
847 		case FUNSUB:
848 			shf_puts("FUNSUB<", shf);
849 			goto dumpsub;
850 		case VALSUB:
851 			shf_puts("VALSUB<", shf);
852 			goto dumpsub;
853 		case EXPRSUB:
854 			shf_puts("EXPRSUB<", shf);
855 			goto dumpsub;
856 		case OQUOTE:
857 			shf_fprintf(shf, "OQUOTE{%d", ++quotelevel);
858 			break;
859 		case CQUOTE:
860 			shf_fprintf(shf, "%d}CQUOTE", quotelevel);
861 			if (quotelevel)
862 				quotelevel--;
863 			else
864 				shf_puts("(err)", shf);
865 			break;
866 		case OSUBST:
867 			shf_puts("OSUBST(", shf);
868 			dumpchar(shf, *wp++);
869 			shf_puts(")[", shf);
870 			while ((c = *wp++) != 0)
871 				dumpchar(shf, c);
872 			shf_putc('|', shf);
873 			wp = dumpwdvar_i(shf, wp, 0);
874 			break;
875 		case CSUBST:
876 			shf_puts("]CSUBST(", shf);
877 			dumpchar(shf, *wp++);
878 			shf_putc(')', shf);
879 			return (wp);
880 		case OPAT:
881 			shf_puts("OPAT=", shf);
882 			dumpchar(shf, *wp++);
883 			break;
884 		case SPAT:
885 			shf_puts("SPAT", shf);
886 			break;
887 		case CPAT:
888 			shf_puts("CPAT", shf);
889 			break;
890 		default:
891 			shf_fprintf(shf, "INVAL<%u>", (uint8_t)wp[-1]);
892 			break;
893 		}
894 		shf_putc(' ', shf);
895 	}
896 }
897 void
dumpwdvar(struct shf * shf,const char * wp)898 dumpwdvar(struct shf *shf, const char *wp)
899 {
900 	dumpwdvar_i(shf, wp, 0);
901 }
902 
903 void
dumpioact(struct shf * shf,struct op * t)904 dumpioact(struct shf *shf, struct op *t)
905 {
906 	struct ioword **ioact, *iop;
907 
908 	if ((ioact = t->ioact) == NULL)
909 		return;
910 
911 	shf_puts("{IOACT", shf);
912 	while ((iop = *ioact++) != NULL) {
913 		unsigned short type = iop->ioflag & IOTYPE;
914 #define DT(x) case x: shf_puts(#x, shf); break;
915 #define DB(x) if (iop->ioflag & x) shf_puts("|" #x, shf);
916 
917 		shf_putc(';', shf);
918 		switch (type) {
919 		DT(IOREAD)
920 		DT(IOWRITE)
921 		DT(IORDWR)
922 		DT(IOHERE)
923 		DT(IOCAT)
924 		DT(IODUP)
925 		default:
926 			shf_fprintf(shf, "unk%d", type);
927 		}
928 		DB(IOEVAL)
929 		DB(IOSKIP)
930 		DB(IOCLOB)
931 		DB(IORDUP)
932 		DB(IONAMEXP)
933 		DB(IOBASH)
934 		DB(IOHERESTR)
935 		DB(IONDELIM)
936 		shf_fprintf(shf, ",unit=%d", (int)iop->unit);
937 		if (iop->delim) {
938 			shf_puts(",delim<", shf);
939 			dumpwdvar(shf, iop->delim);
940 			shf_putc('>', shf);
941 		}
942 		if (iop->name) {
943 			if (iop->ioflag & IONAMEXP) {
944 				shf_puts(",name=", shf);
945 				print_value_quoted(shf, iop->name);
946 			} else {
947 				shf_puts(",name<", shf);
948 				dumpwdvar(shf, iop->name);
949 				shf_putc('>', shf);
950 			}
951 		}
952 		if (iop->heredoc) {
953 			shf_puts(",heredoc=", shf);
954 			print_value_quoted(shf, iop->heredoc);
955 		}
956 #undef DT
957 #undef DB
958 	}
959 	shf_putc('}', shf);
960 }
961 
962 void
dumptree(struct shf * shf,struct op * t)963 dumptree(struct shf *shf, struct op *t)
964 {
965 	int i, j;
966 	const char **w, *name;
967 	struct op *t1;
968 	static int nesting;
969 
970 	for (i = 0; i < nesting; ++i)
971 		shf_putc('\t', shf);
972 	++nesting;
973 	shf_puts("{tree:" /*}*/, shf);
974 	if (t == NULL) {
975 		name = "(null)";
976 		goto out;
977 	}
978 	dumpioact(shf, t);
979 	switch (t->type) {
980 #define OPEN(x) case x: name = #x; shf_puts(" {" #x ":", shf); /*}*/
981 
982 	OPEN(TCOM)
983 		if (t->vars) {
984 			i = 0;
985 			w = (const char **)t->vars;
986 			while (*w) {
987 				shf_putc('\n', shf);
988 				for (j = 0; j < nesting; ++j)
989 					shf_putc('\t', shf);
990 				shf_fprintf(shf, " var%d<", i++);
991 				dumpwdvar(shf, *w++);
992 				shf_putc('>', shf);
993 			}
994 		} else
995 			shf_puts(" #no-vars#", shf);
996 		if (t->args) {
997 			i = 0;
998 			w = t->args;
999 			while (*w) {
1000 				shf_putc('\n', shf);
1001 				for (j = 0; j < nesting; ++j)
1002 					shf_putc('\t', shf);
1003 				shf_fprintf(shf, " arg%d<", i++);
1004 				dumpwdvar(shf, *w++);
1005 				shf_putc('>', shf);
1006 			}
1007 		} else
1008 			shf_puts(" #no-args#", shf);
1009 		break;
1010 	OPEN(TEXEC)
1011  dumpleftandout:
1012 		t = t->left;
1013  dumpandout:
1014 		shf_putc('\n', shf);
1015 		dumptree(shf, t);
1016 		break;
1017 	OPEN(TPAREN)
1018 		goto dumpleftandout;
1019 	OPEN(TPIPE)
1020  dumpleftmidrightandout:
1021 		shf_putc('\n', shf);
1022 		dumptree(shf, t->left);
1023 /* middumprightandout: (unused) */
1024 		shf_fprintf(shf, "/%s:", name);
1025  dumprightandout:
1026 		t = t->right;
1027 		goto dumpandout;
1028 	OPEN(TLIST)
1029 		goto dumpleftmidrightandout;
1030 	OPEN(TOR)
1031 		goto dumpleftmidrightandout;
1032 	OPEN(TAND)
1033 		goto dumpleftmidrightandout;
1034 	OPEN(TBANG)
1035 		goto dumprightandout;
1036 	OPEN(TDBRACKET)
1037 		i = 0;
1038 		w = t->args;
1039 		while (*w) {
1040 			shf_putc('\n', shf);
1041 			for (j = 0; j < nesting; ++j)
1042 				shf_putc('\t', shf);
1043 			shf_fprintf(shf, " arg%d<", i++);
1044 			dumpwdvar(shf, *w++);
1045 			shf_putc('>', shf);
1046 		}
1047 		break;
1048 	OPEN(TFOR)
1049  dumpfor:
1050 		shf_fprintf(shf, " str<%s>", t->str);
1051 		if (t->vars != NULL) {
1052 			i = 0;
1053 			w = (const char **)t->vars;
1054 			while (*w) {
1055 				shf_putc('\n', shf);
1056 				for (j = 0; j < nesting; ++j)
1057 					shf_putc('\t', shf);
1058 				shf_fprintf(shf, " var%d<", i++);
1059 				dumpwdvar(shf, *w++);
1060 				shf_putc('>', shf);
1061 			}
1062 		}
1063 		goto dumpleftandout;
1064 	OPEN(TSELECT)
1065 		goto dumpfor;
1066 	OPEN(TCASE)
1067 		shf_fprintf(shf, " str<%s>", t->str);
1068 		i = 0;
1069 		for (t1 = t->left; t1 != NULL; t1 = t1->right) {
1070 			shf_putc('\n', shf);
1071 			for (j = 0; j < nesting; ++j)
1072 				shf_putc('\t', shf);
1073 			shf_fprintf(shf, " sub%d[(", i);
1074 			w = (const char **)t1->vars;
1075 			while (*w) {
1076 				dumpwdvar(shf, *w);
1077 				if (w[1] != NULL)
1078 					shf_putc('|', shf);
1079 				++w;
1080 			}
1081 			shf_putc(')', shf);
1082 			dumpioact(shf, t);
1083 			shf_putc('\n', shf);
1084 			dumptree(shf, t1->left);
1085 			shf_fprintf(shf, " ;%c/%d]", t1->u.charflag, i++);
1086 		}
1087 		break;
1088 	OPEN(TWHILE)
1089 		goto dumpleftmidrightandout;
1090 	OPEN(TUNTIL)
1091 		goto dumpleftmidrightandout;
1092 	OPEN(TBRACE)
1093 		goto dumpleftandout;
1094 	OPEN(TCOPROC)
1095 		goto dumpleftandout;
1096 	OPEN(TASYNC)
1097 		goto dumpleftandout;
1098 	OPEN(TFUNCT)
1099 		shf_fprintf(shf, " str<%s> ksh<%s>", t->str,
1100 		    t->u.ksh_func ? "yes" : "no");
1101 		goto dumpleftandout;
1102 	OPEN(TTIME)
1103 		goto dumpleftandout;
1104 	OPEN(TIF)
1105  dumpif:
1106 		shf_putc('\n', shf);
1107 		dumptree(shf, t->left);
1108 		t = t->right;
1109 		dumpioact(shf, t);
1110 		if (t->left != NULL) {
1111 			shf_puts(" /TTHEN:\n", shf);
1112 			dumptree(shf, t->left);
1113 		}
1114 		if (t->right && t->right->type == TELIF) {
1115 			shf_puts(" /TELIF:", shf);
1116 			t = t->right;
1117 			dumpioact(shf, t);
1118 			goto dumpif;
1119 		}
1120 		if (t->right != NULL) {
1121 			shf_puts(" /TELSE:\n", shf);
1122 			dumptree(shf, t->right);
1123 		}
1124 		break;
1125 	OPEN(TEOF)
1126  dumpunexpected:
1127 		shf_puts("unexpected", shf);
1128 		break;
1129 	OPEN(TELIF)
1130 		goto dumpunexpected;
1131 	OPEN(TPAT)
1132 		goto dumpunexpected;
1133 	default:
1134 		name = "TINVALID";
1135 		shf_fprintf(shf, "{T<%d>:" /*}*/, t->type);
1136 		goto dumpunexpected;
1137 
1138 #undef OPEN
1139 	}
1140  out:
1141 	shf_fprintf(shf, /*{*/ " /%s}\n", name);
1142 	--nesting;
1143 }
1144 #endif
1145