xref: /dragonfly/bin/sh/parser.c (revision 8a7bdfea)
1 /*-
2  * Copyright (c) 1991, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Kenneth Almquist.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *	This product includes software developed by the University of
19  *	California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  *
36  * @(#)parser.c	8.7 (Berkeley) 5/16/95
37  * $FreeBSD: src/bin/sh/parser.c,v 1.58 2006/11/05 18:36:05 stefanf Exp $
38  * $DragonFly: src/bin/sh/parser.c,v 1.12 2007/01/18 17:03:18 corecode Exp $
39  */
40 
41 #include <stdlib.h>
42 #include <unistd.h>
43 
44 #include "shell.h"
45 #include "parser.h"
46 #include "nodes.h"
47 #include "expand.h"	/* defines rmescapes() */
48 #include "syntax.h"
49 #include "options.h"
50 #include "input.h"
51 #include "output.h"
52 #include "var.h"
53 #include "error.h"
54 #include "memalloc.h"
55 #include "mystring.h"
56 #include "alias.h"
57 #include "show.h"
58 #include "eval.h"
59 #ifndef NO_HISTORY
60 #include "myhistedit.h"
61 #endif
62 
63 /*
64  * Shell command parser.
65  */
66 
67 #define	EOFMARKLEN	79
68 #define	PROMPTLEN	128
69 
70 /* values returned by readtoken */
71 #include "token.h"
72 
73 
74 
75 struct heredoc {
76 	struct heredoc *next;	/* next here document in list */
77 	union node *here;		/* redirection node */
78 	char *eofmark;		/* string indicating end of input */
79 	int striptabs;		/* if set, strip leading tabs */
80 };
81 
82 
83 
84 STATIC struct heredoc *heredoclist;	/* list of here documents to read */
85 STATIC int parsebackquote;	/* nonzero if we are inside backquotes */
86 STATIC int doprompt;		/* if set, prompt the user */
87 STATIC int needprompt;		/* true if interactive and at start of line */
88 STATIC int lasttoken;		/* last token read */
89 MKINIT int tokpushback;		/* last token pushed back */
90 STATIC char *wordtext;		/* text of last word returned by readtoken */
91 /*
92  * 1 == check for kwds
93  * 2 == also eat newlines
94  * 3 == check for TNOT
95  */
96 MKINIT int checkkwd;
97 STATIC struct nodelist *backquotelist;
98 STATIC union node *redirnode;
99 STATIC struct heredoc *heredoc;
100 STATIC int quoteflag;		/* set if (part of) last token was quoted */
101 STATIC int startlinno;		/* line # where last token started */
102 
103 /* XXX When 'noaliases' is set to one, no alias expansion takes place. */
104 static int noaliases = 0;
105 
106 
107 STATIC union node *list(int);
108 STATIC union node *andor(void);
109 STATIC union node *pipeline(void);
110 STATIC union node *command(void);
111 STATIC union node *simplecmd(union node **, union node *);
112 STATIC union node *makename(void);
113 STATIC void parsefname(void);
114 STATIC void parseheredoc(void);
115 STATIC int peektoken(void);
116 STATIC int readtoken(void);
117 STATIC int xxreadtoken(void);
118 STATIC int readtoken1(int, char const *, char *, int);
119 STATIC int noexpand(char *);
120 STATIC void synexpect(int);
121 STATIC void synerror(const char *);
122 STATIC void setprompt(int);
123 
124 
125 /*
126  * Read and parse a command.  Returns NEOF on end of file.  (NULL is a
127  * valid parse tree indicating a blank line.)
128  */
129 
130 union node *
131 parsecmd(int interact)
132 {
133 	int t;
134 
135 	tokpushback = 0;
136 	doprompt = interact;
137 	if (doprompt)
138 		setprompt(1);
139 	else
140 		setprompt(0);
141 	needprompt = 0;
142 	t = readtoken();
143 	if (t == TEOF)
144 		return NEOF;
145 	if (t == TNL)
146 		return NULL;
147 	tokpushback++;
148 	return list(1);
149 }
150 
151 
152 STATIC union node *
153 list(int nlflag)
154 {
155 	union node *n1, *n2, *n3;
156 	int tok;
157 
158 	checkkwd = 2;
159 	if (nlflag == 0 && tokendlist[peektoken()])
160 		return NULL;
161 	n1 = NULL;
162 	for (;;) {
163 		n2 = andor();
164 		tok = readtoken();
165 		if (tok == TBACKGND) {
166 			if (n2->type == NCMD || n2->type == NPIPE) {
167 				n2->ncmd.backgnd = 1;
168 			} else if (n2->type == NREDIR) {
169 				n2->type = NBACKGND;
170 			} else {
171 				n3 = (union node *)stalloc(sizeof (struct nredir));
172 				n3->type = NBACKGND;
173 				n3->nredir.n = n2;
174 				n3->nredir.redirect = NULL;
175 				n2 = n3;
176 			}
177 		}
178 		if (n1 == NULL) {
179 			n1 = n2;
180 		}
181 		else {
182 			n3 = (union node *)stalloc(sizeof (struct nbinary));
183 			n3->type = NSEMI;
184 			n3->nbinary.ch1 = n1;
185 			n3->nbinary.ch2 = n2;
186 			n1 = n3;
187 		}
188 		switch (tok) {
189 		case TBACKGND:
190 		case TSEMI:
191 			tok = readtoken();
192 			/* FALLTHROUGH */
193 		case TNL:
194 			if (tok == TNL) {
195 				parseheredoc();
196 				if (nlflag)
197 					return n1;
198 			} else {
199 				tokpushback++;
200 			}
201 			checkkwd = 2;
202 			if (tokendlist[peektoken()])
203 				return n1;
204 			break;
205 		case TEOF:
206 			if (heredoclist)
207 				parseheredoc();
208 			else
209 				pungetc();		/* push back EOF on input */
210 			return n1;
211 		default:
212 			if (nlflag)
213 				synexpect(-1);
214 			tokpushback++;
215 			return n1;
216 		}
217 	}
218 }
219 
220 
221 
222 STATIC union node *
223 andor(void)
224 {
225 	union node *n1, *n2, *n3;
226 	int t;
227 
228 	n1 = pipeline();
229 	for (;;) {
230 		if ((t = readtoken()) == TAND) {
231 			t = NAND;
232 		} else if (t == TOR) {
233 			t = NOR;
234 		} else {
235 			tokpushback++;
236 			return n1;
237 		}
238 		n2 = pipeline();
239 		n3 = (union node *)stalloc(sizeof (struct nbinary));
240 		n3->type = t;
241 		n3->nbinary.ch1 = n1;
242 		n3->nbinary.ch2 = n2;
243 		n1 = n3;
244 	}
245 }
246 
247 
248 
249 STATIC union node *
250 pipeline(void)
251 {
252 	union node *n1, *n2, *pipenode;
253 	struct nodelist *lp, *prev;
254 	int negate;
255 
256 	negate = 0;
257 	TRACE(("pipeline: entered\n"));
258 
259 	checkkwd = 3;
260 	while (readtoken() == TNOT)
261 		negate = !negate;
262 	tokpushback++;
263 
264 	n1 = command();
265 	if (readtoken() == TPIPE) {
266 		pipenode = (union node *)stalloc(sizeof (struct npipe));
267 		pipenode->type = NPIPE;
268 		pipenode->npipe.backgnd = 0;
269 		lp = (struct nodelist *)stalloc(sizeof (struct nodelist));
270 		pipenode->npipe.cmdlist = lp;
271 		lp->n = n1;
272 		do {
273 			int innernegate = 0;
274 
275 			checkkwd = 3;
276 			while (readtoken() == TNOT)
277 				innernegate = !innernegate;
278 			tokpushback++;
279 
280 			prev = lp;
281 			lp = (struct nodelist *)stalloc(sizeof (struct nodelist));
282 			lp->n = command();
283 			prev->next = lp;
284 
285 			if (innernegate) {
286 				n2 = (union node *)stalloc(sizeof (struct nnot));
287 				n2->type = NNOT;
288 				n2->nnot.com = lp->n;
289 				lp->n = n2;
290 			}
291 		} while (readtoken() == TPIPE);
292 		lp->next = NULL;
293 		n1 = pipenode;
294 	}
295 	tokpushback++;
296 	if (negate) {
297 		n2 = (union node *)stalloc(sizeof (struct nnot));
298 		n2->type = NNOT;
299 		n2->nnot.com = n1;
300 		return n2;
301 	} else
302 		return n1;
303 }
304 
305 
306 
307 STATIC union node *
308 command(void)
309 {
310 	union node *n1, *n2;
311 	union node *ap, **app;
312 	union node *cp, **cpp;
313 	union node *redir, **rpp;
314 	int t;
315 
316 	checkkwd = 2;
317 	redir = NULL;
318 	n1 = NULL;
319 	rpp = &redir;
320 
321 	/* Check for redirection which may precede command */
322 	while (readtoken() == TREDIR) {
323 		*rpp = n2 = redirnode;
324 		rpp = &n2->nfile.next;
325 		parsefname();
326 	}
327 	tokpushback++;
328 
329 	switch (readtoken()) {
330 	case TIF:
331 		n1 = (union node *)stalloc(sizeof (struct nif));
332 		n1->type = NIF;
333 		if ((n1->nif.test = list(0)) == NULL)
334 			synexpect(-1);
335 		if (readtoken() != TTHEN)
336 			synexpect(TTHEN);
337 		n1->nif.ifpart = list(0);
338 		n2 = n1;
339 		while (readtoken() == TELIF) {
340 			n2->nif.elsepart = (union node *)stalloc(sizeof (struct nif));
341 			n2 = n2->nif.elsepart;
342 			n2->type = NIF;
343 			if ((n2->nif.test = list(0)) == NULL)
344 				synexpect(-1);
345 			if (readtoken() != TTHEN)
346 				synexpect(TTHEN);
347 			n2->nif.ifpart = list(0);
348 		}
349 		if (lasttoken == TELSE)
350 			n2->nif.elsepart = list(0);
351 		else {
352 			n2->nif.elsepart = NULL;
353 			tokpushback++;
354 		}
355 		if (readtoken() != TFI)
356 			synexpect(TFI);
357 		checkkwd = 1;
358 		break;
359 	case TWHILE:
360 	case TUNTIL: {
361 		int got;
362 		n1 = (union node *)stalloc(sizeof (struct nbinary));
363 		n1->type = (lasttoken == TWHILE)? NWHILE : NUNTIL;
364 		if ((n1->nbinary.ch1 = list(0)) == NULL)
365 			synexpect(-1);
366 		if ((got=readtoken()) != TDO) {
367 TRACE(("expecting DO got %s %s\n", tokname[got], got == TWORD ? wordtext : ""));
368 			synexpect(TDO);
369 		}
370 		n1->nbinary.ch2 = list(0);
371 		if (readtoken() != TDONE)
372 			synexpect(TDONE);
373 		checkkwd = 1;
374 		break;
375 	}
376 	case TFOR:
377 		if (readtoken() != TWORD || quoteflag || ! goodname(wordtext))
378 			synerror("Bad for loop variable");
379 		n1 = (union node *)stalloc(sizeof (struct nfor));
380 		n1->type = NFOR;
381 		n1->nfor.var = wordtext;
382 		if (readtoken() == TWORD && ! quoteflag && equal(wordtext, "in")) {
383 			app = &ap;
384 			while (readtoken() == TWORD) {
385 				n2 = (union node *)stalloc(sizeof (struct narg));
386 				n2->type = NARG;
387 				n2->narg.text = wordtext;
388 				n2->narg.backquote = backquotelist;
389 				*app = n2;
390 				app = &n2->narg.next;
391 			}
392 			*app = NULL;
393 			n1->nfor.args = ap;
394 			if (lasttoken != TNL && lasttoken != TSEMI)
395 				synexpect(-1);
396 		} else {
397 			static char argvars[5] = {
398 				CTLVAR, VSNORMAL|VSQUOTE, '@', '=', '\0'
399 			};
400 			n2 = (union node *)stalloc(sizeof (struct narg));
401 			n2->type = NARG;
402 			n2->narg.text = argvars;
403 			n2->narg.backquote = NULL;
404 			n2->narg.next = NULL;
405 			n1->nfor.args = n2;
406 			/*
407 			 * Newline or semicolon here is optional (but note
408 			 * that the original Bourne shell only allowed NL).
409 			 */
410 			if (lasttoken != TNL && lasttoken != TSEMI)
411 				tokpushback++;
412 		}
413 		checkkwd = 2;
414 		if ((t = readtoken()) == TDO)
415 			t = TDONE;
416 		else if (t == TBEGIN)
417 			t = TEND;
418 		else
419 			synexpect(-1);
420 		n1->nfor.body = list(0);
421 		if (readtoken() != t)
422 			synexpect(t);
423 		checkkwd = 1;
424 		break;
425 	case TCASE:
426 		n1 = (union node *)stalloc(sizeof (struct ncase));
427 		n1->type = NCASE;
428 		if (readtoken() != TWORD)
429 			synexpect(TWORD);
430 		n1->ncase.expr = n2 = (union node *)stalloc(sizeof (struct narg));
431 		n2->type = NARG;
432 		n2->narg.text = wordtext;
433 		n2->narg.backquote = backquotelist;
434 		n2->narg.next = NULL;
435 		while (readtoken() == TNL);
436 		if (lasttoken != TWORD || ! equal(wordtext, "in"))
437 			synerror("expecting \"in\"");
438 		cpp = &n1->ncase.cases;
439 		noaliases = 1;	/* turn off alias expansion */
440 		checkkwd = 2, readtoken();
441 		while (lasttoken != TESAC) {
442 			*cpp = cp = (union node *)stalloc(sizeof (struct nclist));
443 			cp->type = NCLIST;
444 			app = &cp->nclist.pattern;
445 			if (lasttoken == TLP)
446 				readtoken();
447 			for (;;) {
448 				*app = ap = (union node *)stalloc(sizeof (struct narg));
449 				ap->type = NARG;
450 				ap->narg.text = wordtext;
451 				ap->narg.backquote = backquotelist;
452 				if (checkkwd = 2, readtoken() != TPIPE)
453 					break;
454 				app = &ap->narg.next;
455 				readtoken();
456 			}
457 			ap->narg.next = NULL;
458 			if (lasttoken != TRP)
459 				noaliases = 0, synexpect(TRP);
460 			cp->nclist.body = list(0);
461 
462 			checkkwd = 2;
463 			if ((t = readtoken()) != TESAC) {
464 				if (t != TENDCASE)
465 					noaliases = 0, synexpect(TENDCASE);
466 				else
467 					checkkwd = 2, readtoken();
468 			}
469 			cpp = &cp->nclist.next;
470 		}
471 		noaliases = 0;	/* reset alias expansion */
472 		*cpp = NULL;
473 		checkkwd = 1;
474 		break;
475 	case TLP:
476 		n1 = (union node *)stalloc(sizeof (struct nredir));
477 		n1->type = NSUBSHELL;
478 		n1->nredir.n = list(0);
479 		n1->nredir.redirect = NULL;
480 		if (readtoken() != TRP)
481 			synexpect(TRP);
482 		checkkwd = 1;
483 		break;
484 	case TBEGIN:
485 		n1 = list(0);
486 		if (readtoken() != TEND)
487 			synexpect(TEND);
488 		checkkwd = 1;
489 		break;
490 	/* Handle an empty command like other simple commands.  */
491 	case TSEMI:
492 	case TAND:
493 	case TOR:
494 		/*
495 		 * An empty command before a ; doesn't make much sense, and
496 		 * should certainly be disallowed in the case of `if ;'.
497 		 */
498 		if (!redir)
499 			synexpect(-1);
500 	case TNL:
501 	case TEOF:
502 	case TWORD:
503 	case TRP:
504 		tokpushback++;
505 		return simplecmd(rpp, redir);
506 	default:
507 		synexpect(-1);
508 	}
509 
510 	/* Now check for redirection which may follow command */
511 	while (readtoken() == TREDIR) {
512 		*rpp = n2 = redirnode;
513 		rpp = &n2->nfile.next;
514 		parsefname();
515 	}
516 	tokpushback++;
517 	*rpp = NULL;
518 	if (redir) {
519 		if (n1->type != NSUBSHELL) {
520 			n2 = (union node *)stalloc(sizeof (struct nredir));
521 			n2->type = NREDIR;
522 			n2->nredir.n = n1;
523 			n1 = n2;
524 		}
525 		n1->nredir.redirect = redir;
526 	}
527 	return n1;
528 }
529 
530 
531 STATIC union node *
532 simplecmd(union node **rpp, union node *redir)
533 {
534 	union node *args, **app;
535 	union node **orig_rpp = rpp;
536 	union node *n = NULL;
537 
538 	/* If we don't have any redirections already, then we must reset */
539 	/* rpp to be the address of the local redir variable.  */
540 	if (redir == 0)
541 		rpp = &redir;
542 
543 	args = NULL;
544 	app = &args;
545 	/*
546 	 * We save the incoming value, because we need this for shell
547 	 * functions.  There can not be a redirect or an argument between
548 	 * the function name and the open parenthesis.
549 	 */
550 	orig_rpp = rpp;
551 
552 	for (;;) {
553 		if (readtoken() == TWORD) {
554 			n = (union node *)stalloc(sizeof (struct narg));
555 			n->type = NARG;
556 			n->narg.text = wordtext;
557 			n->narg.backquote = backquotelist;
558 			*app = n;
559 			app = &n->narg.next;
560 		} else if (lasttoken == TREDIR) {
561 			*rpp = n = redirnode;
562 			rpp = &n->nfile.next;
563 			parsefname();	/* read name of redirection file */
564 		} else if (lasttoken == TLP && app == &args->narg.next
565 					    && rpp == orig_rpp) {
566 			/* We have a function */
567 			if (readtoken() != TRP)
568 				synexpect(TRP);
569 #ifdef notdef
570 			if (! goodname(n->narg.text))
571 				synerror("Bad function name");
572 #endif
573 			n->type = NDEFUN;
574 			n->narg.next = command();
575 			return n;
576 		} else {
577 			tokpushback++;
578 			break;
579 		}
580 	}
581 	*app = NULL;
582 	*rpp = NULL;
583 	n = (union node *)stalloc(sizeof (struct ncmd));
584 	n->type = NCMD;
585 	n->ncmd.backgnd = 0;
586 	n->ncmd.args = args;
587 	n->ncmd.redirect = redir;
588 	return n;
589 }
590 
591 STATIC union node *
592 makename(void)
593 {
594 	union node *n;
595 
596 	n = (union node *)stalloc(sizeof (struct narg));
597 	n->type = NARG;
598 	n->narg.next = NULL;
599 	n->narg.text = wordtext;
600 	n->narg.backquote = backquotelist;
601 	return n;
602 }
603 
604 void
605 fixredir(union node *n, const char *text, int err)
606 {
607 	TRACE(("Fix redir %s %d\n", text, err));
608 	if (!err)
609 		n->ndup.vname = NULL;
610 
611 	if (is_digit(text[0]) && text[1] == '\0')
612 		n->ndup.dupfd = digit_val(text[0]);
613 	else if (text[0] == '-' && text[1] == '\0')
614 		n->ndup.dupfd = -1;
615 	else {
616 
617 		if (err)
618 			synerror("Bad fd number");
619 		else
620 			n->ndup.vname = makename();
621 	}
622 }
623 
624 
625 STATIC void
626 parsefname(void)
627 {
628 	union node *n = redirnode;
629 
630 	if (readtoken() != TWORD)
631 		synexpect(-1);
632 	if (n->type == NHERE) {
633 		struct heredoc *here = heredoc;
634 		struct heredoc *p;
635 		int i;
636 
637 		if (quoteflag == 0)
638 			n->type = NXHERE;
639 		TRACE(("Here document %d\n", n->type));
640 		if (here->striptabs) {
641 			while (*wordtext == '\t')
642 				wordtext++;
643 		}
644 		if (! noexpand(wordtext) || (i = strlen(wordtext)) == 0 || i > EOFMARKLEN)
645 			synerror("Illegal eof marker for << redirection");
646 		rmescapes(wordtext);
647 		here->eofmark = wordtext;
648 		here->next = NULL;
649 		if (heredoclist == NULL)
650 			heredoclist = here;
651 		else {
652 			for (p = heredoclist ; p->next ; p = p->next);
653 			p->next = here;
654 		}
655 	} else if (n->type == NTOFD || n->type == NFROMFD) {
656 		fixredir(n, wordtext, 0);
657 	} else {
658 		n->nfile.fname = makename();
659 	}
660 }
661 
662 
663 /*
664  * Input any here documents.
665  */
666 
667 STATIC void
668 parseheredoc(void)
669 {
670 	struct heredoc *here;
671 	union node *n;
672 
673 	while (heredoclist) {
674 		here = heredoclist;
675 		heredoclist = here->next;
676 		if (needprompt) {
677 			setprompt(2);
678 			needprompt = 0;
679 		}
680 		readtoken1(pgetc(), here->here->type == NHERE? SQSYNTAX : DQSYNTAX,
681 				here->eofmark, here->striptabs);
682 		n = (union node *)stalloc(sizeof (struct narg));
683 		n->narg.type = NARG;
684 		n->narg.next = NULL;
685 		n->narg.text = wordtext;
686 		n->narg.backquote = backquotelist;
687 		here->here->nhere.doc = n;
688 	}
689 }
690 
691 STATIC int
692 peektoken(void)
693 {
694 	int t;
695 
696 	t = readtoken();
697 	tokpushback++;
698 	return (t);
699 }
700 
701 STATIC int
702 readtoken(void)
703 {
704 	int t;
705 	int savecheckkwd = checkkwd;
706 	struct alias *ap;
707 #ifdef DEBUG
708 	int alreadyseen = tokpushback;
709 #endif
710 
711 	top:
712 	t = xxreadtoken();
713 
714 	if (checkkwd) {
715 		/*
716 		 * eat newlines
717 		 */
718 		if (checkkwd > 1) {
719 			checkkwd = 0;
720 			while (t == TNL) {
721 				parseheredoc();
722 				t = xxreadtoken();
723 			}
724 		} else
725 			checkkwd = 0;
726 		/*
727 		 * check for keywords and aliases
728 		 */
729 		if (t == TWORD && !quoteflag)
730 		{
731 			const char * const *pp;
732 
733 			for (pp = parsekwd; *pp; pp++) {
734 				if (**pp == *wordtext && equal(*pp, wordtext))
735 				{
736 					lasttoken = t = pp - parsekwd + KWDOFFSET;
737 					TRACE(("keyword %s recognized\n", tokname[t]));
738 					goto out;
739 				}
740 			}
741 			if (noaliases == 0 &&
742 			    (ap = lookupalias(wordtext, 1)) != NULL) {
743 				pushstring(ap->val, strlen(ap->val), ap);
744 				checkkwd = savecheckkwd;
745 				goto top;
746 			}
747 		}
748 out:
749 		checkkwd = 0;
750 	}
751 #ifdef DEBUG
752 	if (!alreadyseen)
753 	    TRACE(("token %s %s\n", tokname[t], t == TWORD ? wordtext : ""));
754 	else
755 	    TRACE(("reread token %s %s\n", tokname[t], t == TWORD ? wordtext : ""));
756 #endif
757 	return (t);
758 }
759 
760 
761 /*
762  * Read the next input token.
763  * If the token is a word, we set backquotelist to the list of cmds in
764  *	backquotes.  We set quoteflag to true if any part of the word was
765  *	quoted.
766  * If the token is TREDIR, then we set redirnode to a structure containing
767  *	the redirection.
768  * In all cases, the variable startlinno is set to the number of the line
769  *	on which the token starts.
770  *
771  * [Change comment:  here documents and internal procedures]
772  * [Readtoken shouldn't have any arguments.  Perhaps we should make the
773  *  word parsing code into a separate routine.  In this case, readtoken
774  *  doesn't need to have any internal procedures, but parseword does.
775  *  We could also make parseoperator in essence the main routine, and
776  *  have parseword (readtoken1?) handle both words and redirection.]
777  */
778 
779 #define RETURN(token)	return lasttoken = token
780 
781 STATIC int
782 xxreadtoken(void)
783 {
784 	int c;
785 
786 	if (tokpushback) {
787 		tokpushback = 0;
788 		return lasttoken;
789 	}
790 	if (needprompt) {
791 		setprompt(2);
792 		needprompt = 0;
793 	}
794 	startlinno = plinno;
795 	for (;;) {	/* until token or start of word found */
796 		c = pgetc_macro();
797 		if (c == ' ' || c == '\t')
798 			continue;		/* quick check for white space first */
799 		switch (c) {
800 		case ' ': case '\t':
801 			continue;
802 		case '#':
803 			while ((c = pgetc()) != '\n' && c != PEOF);
804 			pungetc();
805 			continue;
806 		case '\\':
807 			if (pgetc() == '\n') {
808 				startlinno = ++plinno;
809 				if (doprompt)
810 					setprompt(2);
811 				else
812 					setprompt(0);
813 				continue;
814 			}
815 			pungetc();
816 			goto breakloop;
817 		case '\n':
818 			plinno++;
819 			needprompt = doprompt;
820 			RETURN(TNL);
821 		case PEOF:
822 			RETURN(TEOF);
823 		case '&':
824 			if (pgetc() == '&')
825 				RETURN(TAND);
826 			pungetc();
827 			RETURN(TBACKGND);
828 		case '|':
829 			if (pgetc() == '|')
830 				RETURN(TOR);
831 			pungetc();
832 			RETURN(TPIPE);
833 		case ';':
834 			if (pgetc() == ';')
835 				RETURN(TENDCASE);
836 			pungetc();
837 			RETURN(TSEMI);
838 		case '(':
839 			RETURN(TLP);
840 		case ')':
841 			RETURN(TRP);
842 		case '!':
843 			if (checkkwd == 3)
844 				RETURN(TNOT);
845 			/* else FALLTHROUGH */
846 		default:
847 			goto breakloop;
848 		}
849 	}
850 breakloop:
851 	return readtoken1(c, BASESYNTAX, (char *)NULL, 0);
852 #undef RETURN
853 }
854 
855 
856 
857 /*
858  * If eofmark is NULL, read a word or a redirection symbol.  If eofmark
859  * is not NULL, read a here document.  In the latter case, eofmark is the
860  * word which marks the end of the document and striptabs is true if
861  * leading tabs should be stripped from the document.  The argument firstc
862  * is the first character of the input token or document.
863  *
864  * Because C does not have internal subroutines, I have simulated them
865  * using goto's to implement the subroutine linkage.  The following macros
866  * will run code that appears at the end of readtoken1.
867  */
868 
869 #define CHECKEND()	{goto checkend; checkend_return:;}
870 #define PARSEREDIR()	{goto parseredir; parseredir_return:;}
871 #define PARSESUB()	{goto parsesub; parsesub_return:;}
872 #define PARSEBACKQOLD()	{oldstyle = 1; goto parsebackq; parsebackq_oldreturn:;}
873 #define PARSEBACKQNEW()	{oldstyle = 0; goto parsebackq; parsebackq_newreturn:;}
874 #define	PARSEARITH()	{goto parsearith; parsearith_return:;}
875 
876 STATIC int
877 readtoken1(int firstc, char const *syn, char *eofmark, int striptabs)
878 {
879 	int c = firstc;
880 	char const * volatile syntax = syn;
881 	char * volatile out;
882 	int len;
883 	char line[EOFMARKLEN + 1];
884 	struct nodelist *bqlist;
885 	volatile int quotef;
886 	volatile int dblquote;
887 	volatile int varnest;	/* levels of variables expansion */
888 	volatile int arinest;	/* levels of arithmetic expansion */
889 	volatile int parenlevel;	/* levels of parens in arithmetic */
890 	volatile int oldstyle;
891 	char const * volatile prevsyntax = NULL;	/* syntax before arithmetic */
892 	int synentry;
893 
894 	startlinno = plinno;
895 	dblquote = 0;
896 	if (syntax == DQSYNTAX)
897 		dblquote = 1;
898 	quotef = 0;
899 	bqlist = NULL;
900 	varnest = 0;
901 	arinest = 0;
902 	parenlevel = 0;
903 
904 	STARTSTACKSTR(out);
905 	loop: {	/* for each line, until end of word */
906 		CHECKEND();	/* set c to PEOF if at end of here document */
907 		for (;;) {	/* until end of line or end of word */
908 			CHECKSTRSPACE(3, out);	/* permit 3 calls to USTPUTC */
909 
910 			synentry = syntax[c];
911 
912 			switch(synentry) {
913 			case CNL:	/* '\n' */
914 				if (syntax == BASESYNTAX)
915 					goto endword;	/* exit outer loop */
916 				USTPUTC(c, out);
917 				plinno++;
918 				if (doprompt)
919 					setprompt(2);
920 				else
921 					setprompt(0);
922 				c = pgetc();
923 				goto loop;		/* continue outer loop */
924 			case CWORD:
925 				USTPUTC(c, out);
926 				break;
927 			case CCTL:
928 				if (eofmark == NULL || dblquote)
929 					USTPUTC(CTLESC, out);
930 				USTPUTC(c, out);
931 				break;
932 			case CBACK:	/* backslash */
933 				c = pgetc();
934 				if (c == PEOF) {
935 					USTPUTC('\\', out);
936 					pungetc();
937 				} else if (c == '\n') {
938 					plinno++;
939 					if (doprompt)
940 						setprompt(2);
941 					else
942 						setprompt(0);
943 				} else {
944 					if (dblquote && c != '\\' &&
945 					    c != '`' && c != '$' &&
946 					    (c != '"' || eofmark != NULL))
947 						USTPUTC('\\', out);
948 					if (SQSYNTAX[c] == CCTL)
949 						USTPUTC(CTLESC, out);
950 					else if (eofmark == NULL)
951 						USTPUTC(CTLQUOTEMARK, out);
952 					USTPUTC(c, out);
953 					quotef++;
954 				}
955 				break;
956 			case CSQUOTE:
957 				if (eofmark == NULL)
958 					USTPUTC(CTLQUOTEMARK, out);
959 				syntax = SQSYNTAX;
960 				break;
961 			case CDQUOTE:
962 				if (eofmark == NULL)
963 					USTPUTC(CTLQUOTEMARK, out);
964 				syntax = DQSYNTAX;
965 				dblquote = 1;
966 				break;
967 			case CENDQUOTE:
968 				if (eofmark != NULL && arinest == 0 &&
969 				    varnest == 0) {
970 					USTPUTC(c, out);
971 				} else {
972 					if (arinest) {
973 						syntax = ARISYNTAX;
974 						dblquote = 0;
975 					} else if (eofmark == NULL) {
976 						syntax = BASESYNTAX;
977 						dblquote = 0;
978 					}
979 					quotef++;
980 				}
981 				break;
982 			case CVAR:	/* '$' */
983 				PARSESUB();		/* parse substitution */
984 				break;
985 			case CENDVAR:	/* '}' */
986 				if (varnest > 0) {
987 					varnest--;
988 					USTPUTC(CTLENDVAR, out);
989 				} else {
990 					USTPUTC(c, out);
991 				}
992 				break;
993 			case CLP:	/* '(' in arithmetic */
994 				parenlevel++;
995 				USTPUTC(c, out);
996 				break;
997 			case CRP:	/* ')' in arithmetic */
998 				if (parenlevel > 0) {
999 					USTPUTC(c, out);
1000 					--parenlevel;
1001 				} else {
1002 					if (pgetc() == ')') {
1003 						if (--arinest == 0) {
1004 							USTPUTC(CTLENDARI, out);
1005 							syntax = prevsyntax;
1006 							if (syntax == DQSYNTAX)
1007 								dblquote = 1;
1008 							else
1009 								dblquote = 0;
1010 						} else
1011 							USTPUTC(')', out);
1012 					} else {
1013 						/*
1014 						 * unbalanced parens
1015 						 *  (don't 2nd guess - no error)
1016 						 */
1017 						pungetc();
1018 						USTPUTC(')', out);
1019 					}
1020 				}
1021 				break;
1022 			case CBQUOTE:	/* '`' */
1023 				PARSEBACKQOLD();
1024 				break;
1025 			case CEOF:
1026 				goto endword;		/* exit outer loop */
1027 			default:
1028 				if (varnest == 0)
1029 					goto endword;	/* exit outer loop */
1030 				USTPUTC(c, out);
1031 			}
1032 			c = pgetc_macro();
1033 		}
1034 	}
1035 endword:
1036 	if (syntax == ARISYNTAX)
1037 		synerror("Missing '))'");
1038 	if (syntax != BASESYNTAX && ! parsebackquote && eofmark == NULL)
1039 		synerror("Unterminated quoted string");
1040 	if (varnest != 0) {
1041 		startlinno = plinno;
1042 		synerror("Missing '}'");
1043 	}
1044 	USTPUTC('\0', out);
1045 	len = out - stackblock();
1046 	out = stackblock();
1047 	if (eofmark == NULL) {
1048 		if ((c == '>' || c == '<')
1049 		 && quotef == 0
1050 		 && len <= 2
1051 		 && (*out == '\0' || is_digit(*out))) {
1052 			PARSEREDIR();
1053 			return lasttoken = TREDIR;
1054 		} else {
1055 			pungetc();
1056 		}
1057 	}
1058 	quoteflag = quotef;
1059 	backquotelist = bqlist;
1060 	grabstackblock(len);
1061 	wordtext = out;
1062 	return lasttoken = TWORD;
1063 /* end of readtoken routine */
1064 
1065 
1066 
1067 /*
1068  * Check to see whether we are at the end of the here document.  When this
1069  * is called, c is set to the first character of the next input line.  If
1070  * we are at the end of the here document, this routine sets the c to PEOF.
1071  */
1072 
1073 checkend: {
1074 	if (eofmark) {
1075 		if (striptabs) {
1076 			while (c == '\t')
1077 				c = pgetc();
1078 		}
1079 		if (c == *eofmark) {
1080 			if (pfgets(line, sizeof line) != NULL) {
1081 				char *p, *q;
1082 
1083 				p = line;
1084 				for (q = eofmark + 1 ; *q && *p == *q ; p++, q++);
1085 				if (*p == '\n' && *q == '\0') {
1086 					c = PEOF;
1087 					plinno++;
1088 					needprompt = doprompt;
1089 				} else {
1090 					pushstring(line, strlen(line), NULL);
1091 				}
1092 			}
1093 		}
1094 	}
1095 	goto checkend_return;
1096 }
1097 
1098 
1099 /*
1100  * Parse a redirection operator.  The variable "out" points to a string
1101  * specifying the fd to be redirected.  The variable "c" contains the
1102  * first character of the redirection operator.
1103  */
1104 
1105 parseredir: {
1106 	char fd = *out;
1107 	union node *np;
1108 
1109 	np = (union node *)stalloc(sizeof (struct nfile));
1110 	if (c == '>') {
1111 		np->nfile.fd = 1;
1112 		c = pgetc();
1113 		if (c == '>')
1114 			np->type = NAPPEND;
1115 		else if (c == '&')
1116 			np->type = NTOFD;
1117 		else if (c == '|')
1118 			np->type = NCLOBBER;
1119 		else {
1120 			np->type = NTO;
1121 			pungetc();
1122 		}
1123 	} else {	/* c == '<' */
1124 		np->nfile.fd = 0;
1125 		c = pgetc();
1126 		if (c == '<') {
1127 			if (sizeof (struct nfile) != sizeof (struct nhere)) {
1128 				np = (union node *)stalloc(sizeof (struct nhere));
1129 				np->nfile.fd = 0;
1130 			}
1131 			np->type = NHERE;
1132 			heredoc = (struct heredoc *)stalloc(sizeof (struct heredoc));
1133 			heredoc->here = np;
1134 			if ((c = pgetc()) == '-') {
1135 				heredoc->striptabs = 1;
1136 			} else {
1137 				heredoc->striptabs = 0;
1138 				pungetc();
1139 			}
1140 		} else if (c == '&')
1141 			np->type = NFROMFD;
1142 		else if (c == '>')
1143 			np->type = NFROMTO;
1144 		else {
1145 			np->type = NFROM;
1146 			pungetc();
1147 		}
1148 	}
1149 	if (fd != '\0')
1150 		np->nfile.fd = digit_val(fd);
1151 	redirnode = np;
1152 	goto parseredir_return;
1153 }
1154 
1155 
1156 /*
1157  * Parse a substitution.  At this point, we have read the dollar sign
1158  * and nothing else.
1159  */
1160 
1161 parsesub: {
1162 	int subtype;
1163 	int typeloc;
1164 	int flags;
1165 	char *p;
1166 	static const char types[] = "}-+?=";
1167        int bracketed_name = 0; /* used to handle ${[0-9]*} variables */
1168 
1169 	c = pgetc();
1170 	if (c != '(' && c != '{' && (is_eof(c) || !is_name(c)) &&
1171 	    !is_special(c)) {
1172 		USTPUTC('$', out);
1173 		pungetc();
1174 	} else if (c == '(') {	/* $(command) or $((arith)) */
1175 		if (pgetc() == '(') {
1176 			PARSEARITH();
1177 		} else {
1178 			pungetc();
1179 			PARSEBACKQNEW();
1180 		}
1181 	} else {
1182 		USTPUTC(CTLVAR, out);
1183 		typeloc = out - stackblock();
1184 		USTPUTC(VSNORMAL, out);
1185 		subtype = VSNORMAL;
1186 		if (c == '{') {
1187 			bracketed_name = 1;
1188 			c = pgetc();
1189 			if (c == '#') {
1190 				if ((c = pgetc()) == '}')
1191 					c = '#';
1192 				else
1193 					subtype = VSLENGTH;
1194 			}
1195 			else
1196 				subtype = 0;
1197 		}
1198 		if (!is_eof(c) && is_name(c)) {
1199 			do {
1200 				STPUTC(c, out);
1201 				c = pgetc();
1202 			} while (!is_eof(c) && is_in_name(c));
1203 		} else if (is_digit(c)) {
1204 			if (bracketed_name) {
1205 				do {
1206 					STPUTC(c, out);
1207 					c = pgetc();
1208 				} while (is_digit(c));
1209 			} else {
1210 				STPUTC(c, out);
1211 				c = pgetc();
1212 			}
1213 		} else {
1214 			if (! is_special(c)) {
1215 				subtype = VSERROR;
1216 				if (c == '}')
1217 					pungetc();
1218 				else
1219 					USTPUTC(c, out);
1220 			} else {
1221 				USTPUTC(c, out);
1222 				c = pgetc();
1223 			}
1224 		}
1225 		flags = 0;
1226 		if (subtype == 0) {
1227 			switch (c) {
1228 			case ':':
1229 				flags = VSNUL;
1230 				c = pgetc();
1231 				/*FALLTHROUGH*/
1232 			default:
1233 				p = strchr(types, c);
1234 				if (p == NULL) {
1235 					if (flags == VSNUL)
1236 						STPUTC(':', out);
1237 					STPUTC(c, out);
1238 					subtype = VSERROR;
1239 				} else
1240 					subtype = p - types + VSNORMAL;
1241 				break;
1242 			case '%':
1243 			case '#':
1244 				{
1245 					int cc = c;
1246 					subtype = c == '#' ? VSTRIMLEFT :
1247 							     VSTRIMRIGHT;
1248 					c = pgetc();
1249 					if (c == cc)
1250 						subtype++;
1251 					else
1252 						pungetc();
1253 					break;
1254 				}
1255 			}
1256 		} else if (subtype != VSERROR) {
1257 			pungetc();
1258 		}
1259 		STPUTC('=', out);
1260 		if (subtype != VSLENGTH && (dblquote || arinest))
1261 			flags |= VSQUOTE;
1262 		*(stackblock() + typeloc) = subtype | flags;
1263 		if (subtype != VSNORMAL)
1264 			varnest++;
1265 	}
1266 	goto parsesub_return;
1267 }
1268 
1269 
1270 /*
1271  * Called to parse command substitutions.  Newstyle is set if the command
1272  * is enclosed inside $(...); nlpp is a pointer to the head of the linked
1273  * list of commands (passed by reference), and savelen is the number of
1274  * characters on the top of the stack which must be preserved.
1275  */
1276 
1277 parsebackq: {
1278 	struct nodelist **nlpp;
1279 	int savepbq;
1280 	union node *n;
1281 	char *volatile str;
1282 	struct jmploc jmploc;
1283 	struct jmploc *volatile savehandler;
1284 	int savelen;
1285 	volatile int saveprompt;
1286 
1287 	savepbq = parsebackquote;
1288 	if (setjmp(jmploc.loc)) {
1289 		if (str)
1290 			ckfree(str);
1291 		parsebackquote = 0;
1292 		handler = savehandler;
1293 		longjmp(handler->loc, 1);
1294 	}
1295 	INTOFF;
1296 	str = NULL;
1297 	savelen = out - stackblock();
1298 	if (savelen > 0) {
1299 		str = ckmalloc(savelen);
1300 		memcpy(str, stackblock(), savelen);
1301 	}
1302 	savehandler = handler;
1303 	handler = &jmploc;
1304 	INTON;
1305         if (oldstyle) {
1306                 /* We must read until the closing backquote, giving special
1307                    treatment to some slashes, and then push the string and
1308                    reread it as input, interpreting it normally.  */
1309                 char *pout;
1310                 int pc;
1311                 int psavelen;
1312                 char *pstr;
1313 
1314 
1315                 STARTSTACKSTR(pout);
1316 		for (;;) {
1317 			if (needprompt) {
1318 				setprompt(2);
1319 				needprompt = 0;
1320 			}
1321 			switch (pc = pgetc()) {
1322 			case '`':
1323 				goto done;
1324 
1325 			case '\\':
1326                                 if ((pc = pgetc()) == '\n') {
1327 					plinno++;
1328 					if (doprompt)
1329 						setprompt(2);
1330 					else
1331 						setprompt(0);
1332 					/*
1333 					 * If eating a newline, avoid putting
1334 					 * the newline into the new character
1335 					 * stream (via the STPUTC after the
1336 					 * switch).
1337 					 */
1338 					continue;
1339 				}
1340                                 if (pc != '\\' && pc != '`' && pc != '$'
1341                                     && (!dblquote || pc != '"'))
1342                                         STPUTC('\\', pout);
1343 				break;
1344 
1345 			case '\n':
1346 				plinno++;
1347 				needprompt = doprompt;
1348 				break;
1349 
1350 			case PEOF:
1351 			        startlinno = plinno;
1352 				synerror("EOF in backquote substitution");
1353  				break;
1354 
1355 			default:
1356 				break;
1357 			}
1358 			STPUTC(pc, pout);
1359                 }
1360 done:
1361                 STPUTC('\0', pout);
1362                 psavelen = pout - stackblock();
1363                 if (psavelen > 0) {
1364                         pstr = ckmalloc(psavelen);
1365                         memcpy(pstr, stackblock(), psavelen);
1366 			setinputstring(pstr, 1);
1367                 }
1368         }
1369 	nlpp = &bqlist;
1370 	while (*nlpp)
1371 		nlpp = &(*nlpp)->next;
1372 	*nlpp = (struct nodelist *)stalloc(sizeof (struct nodelist));
1373 	(*nlpp)->next = NULL;
1374 	parsebackquote = oldstyle;
1375 
1376 	if (oldstyle) {
1377 		saveprompt = doprompt;
1378 		doprompt = 0;
1379 	}
1380 
1381 	n = list(0);
1382 
1383 	if (oldstyle)
1384 		doprompt = saveprompt;
1385 	else {
1386 		if (readtoken() != TRP)
1387 			synexpect(TRP);
1388 	}
1389 
1390 	(*nlpp)->n = n;
1391         if (oldstyle) {
1392 		/*
1393 		 * Start reading from old file again, ignoring any pushed back
1394 		 * tokens left from the backquote parsing
1395 		 */
1396                 popfile();
1397 		tokpushback = 0;
1398 	}
1399 	while (stackblocksize() <= savelen)
1400 		growstackblock();
1401 	STARTSTACKSTR(out);
1402 	if (str) {
1403 		memcpy(out, str, savelen);
1404 		STADJUST(savelen, out);
1405 		INTOFF;
1406 		ckfree(str);
1407 		str = NULL;
1408 		INTON;
1409 	}
1410 	parsebackquote = savepbq;
1411 	handler = savehandler;
1412 	if (arinest || dblquote)
1413 		USTPUTC(CTLBACKQ | CTLQUOTE, out);
1414 	else
1415 		USTPUTC(CTLBACKQ, out);
1416 	if (oldstyle)
1417 		goto parsebackq_oldreturn;
1418 	else
1419 		goto parsebackq_newreturn;
1420 }
1421 
1422 /*
1423  * Parse an arithmetic expansion (indicate start of one and set state)
1424  */
1425 parsearith: {
1426 
1427 	if (++arinest == 1) {
1428 		prevsyntax = syntax;
1429 		syntax = ARISYNTAX;
1430 		USTPUTC(CTLARI, out);
1431 		if (dblquote)
1432 			USTPUTC('"',out);
1433 		else
1434 			USTPUTC(' ',out);
1435 	} else {
1436 		/*
1437 		 * we collapse embedded arithmetic expansion to
1438 		 * parenthesis, which should be equivalent
1439 		 */
1440 		USTPUTC('(', out);
1441 	}
1442 	goto parsearith_return;
1443 }
1444 
1445 } /* end of readtoken */
1446 
1447 
1448 
1449 #ifdef mkinit
1450 RESET {
1451 	tokpushback = 0;
1452 	checkkwd = 0;
1453 }
1454 #endif
1455 
1456 /*
1457  * Returns true if the text contains nothing to expand (no dollar signs
1458  * or backquotes).
1459  */
1460 
1461 STATIC int
1462 noexpand(char *text)
1463 {
1464 	char *p;
1465 	char c;
1466 
1467 	p = text;
1468 	while ((c = *p++) != '\0') {
1469 		if ( c == CTLQUOTEMARK)
1470 			continue;
1471 		if (c == CTLESC)
1472 			p++;
1473 		else if (BASESYNTAX[(int)c] == CCTL)
1474 			return 0;
1475 	}
1476 	return 1;
1477 }
1478 
1479 
1480 /*
1481  * Return true if the argument is a legal variable name (a letter or
1482  * underscore followed by zero or more letters, underscores, and digits).
1483  */
1484 
1485 int
1486 goodname(char *name)
1487 {
1488 	char *p;
1489 
1490 	p = name;
1491 	if (! is_name(*p))
1492 		return 0;
1493 	while (*++p) {
1494 		if (! is_in_name(*p))
1495 			return 0;
1496 	}
1497 	return 1;
1498 }
1499 
1500 
1501 /*
1502  * Called when an unexpected token is read during the parse.  The argument
1503  * is the token that is expected, or -1 if more than one type of token can
1504  * occur at this point.
1505  */
1506 
1507 STATIC void
1508 synexpect(int token)
1509 {
1510 	char msg[64];
1511 
1512 	if (token >= 0) {
1513 		fmtstr(msg, 64, "%s unexpected (expecting %s)",
1514 			tokname[lasttoken], tokname[token]);
1515 	} else {
1516 		fmtstr(msg, 64, "%s unexpected", tokname[lasttoken]);
1517 	}
1518 	synerror(msg);
1519 }
1520 
1521 
1522 STATIC void
1523 synerror(const char *msg)
1524 {
1525 	if (commandname)
1526 		outfmt(&errout, "%s: %d: ", commandname, startlinno);
1527 	outfmt(&errout, "Syntax error: %s\n", msg);
1528 	error((char *)NULL);
1529 }
1530 
1531 STATIC void
1532 setprompt(int which)
1533 {
1534 	whichprompt = which;
1535 
1536 #ifndef NO_HISTORY
1537 	if (!el)
1538 #endif
1539 		out2str(getprompt(NULL));
1540 }
1541 
1542 /*
1543  * called by editline -- any expansions to the prompt
1544  *    should be added here.
1545  */
1546 const char *
1547 getprompt(void *unused __unused)
1548 {
1549 	static char ps[PROMPTLEN];
1550 	const char *fmt;
1551 	int i, j, trim;
1552 
1553 	/*
1554 	 * Select prompt format.
1555 	 */
1556 	switch (whichprompt) {
1557 	case 0:
1558 		fmt = "";
1559 		break;
1560 	case 1:
1561 		fmt = ps1val();
1562 		break;
1563 	case 2:
1564 		fmt = ps2val();
1565 		break;
1566 	default:
1567 		return "<internal prompt error>";
1568 	}
1569 
1570 	/*
1571 	 * Format prompt string.
1572 	 */
1573 	for (i = 0; (i < 127) && (*fmt != '\0'); i++, fmt++)
1574 		if (*fmt == '\\')
1575 			switch (*++fmt) {
1576 
1577 				/*
1578 				 * Hostname.
1579 				 *
1580 				 * \h specifies just the local hostname,
1581 				 * \H specifies fully-qualified hostname.
1582 				 */
1583 			case 'h':
1584 			case 'H':
1585 				ps[i] = '\0';
1586 				gethostname(&ps[i], PROMPTLEN - i);
1587 				/* Skip to end of hostname. */
1588 				trim = (*fmt == 'h') ? '.' : '\0';
1589 				while ((ps[i+1] != '\0') && (ps[i+1] != trim))
1590 					i++;
1591 				break;
1592 
1593 				/*
1594 				 * Working directory.
1595 				 *
1596 				 * \W specifies just the final component,
1597 				 * \w specifies the entire path.
1598 				 */
1599 			case 'W':
1600 			case 'w':
1601 				ps[i] = '\0';
1602 				getcwd(&ps[i], PROMPTLEN - i);
1603 				if (*fmt == 'W') {
1604 					/* Final path component only. */
1605 					trim = 1;
1606 					for (j = i; ps[j] != '\0'; j++)
1607 					  if (ps[j] == '/')
1608 						trim = j + 1;
1609 					memmove(&ps[i], &ps[trim],
1610 					    j - trim + 1);
1611 				}
1612 				/* Skip to end of path. */
1613 				while (ps[i + 1] != '\0')
1614 					i++;
1615 				break;
1616 
1617 				/*
1618 				 * Superuser status.
1619 				 *
1620 				 * '$' for normal users, '#' for root.
1621 				 */
1622 			case '$':
1623 				ps[i] = (geteuid() != 0) ? '$' : '#';
1624 				break;
1625 
1626 				/*
1627 				 * A literal \.
1628 				 */
1629 			case '\\':
1630 				ps[i] = '\\';
1631 				break;
1632 
1633 				/*
1634 				 * Emit unrecognized formats verbatim.
1635 				 */
1636 			default:
1637 				ps[i++] = '\\';
1638 				ps[i] = *fmt;
1639 				break;
1640 			}
1641 		else
1642 			ps[i] = *fmt;
1643 	ps[i] = '\0';
1644 	return (ps);
1645 }
1646