xref: /plan9/sys/src/ape/cmd/pdksh/expr.c (revision 7dd7cddf99dd7472612f1413b4da293630e6b1bc)
1*7dd7cddfSDavid du Colombier /*
2*7dd7cddfSDavid du Colombier  * Korn expression evaluation
3*7dd7cddfSDavid du Colombier  */
4*7dd7cddfSDavid du Colombier /*
5*7dd7cddfSDavid du Colombier  * todo: better error handling: if in builtin, should be builtin error, etc.
6*7dd7cddfSDavid du Colombier  */
7*7dd7cddfSDavid du Colombier 
8*7dd7cddfSDavid du Colombier #include "sh.h"
9*7dd7cddfSDavid du Colombier #include <ctype.h>
10*7dd7cddfSDavid du Colombier 
11*7dd7cddfSDavid du Colombier 
12*7dd7cddfSDavid du Colombier /* The order of these enums is constrained by the order of opinfo[] */
13*7dd7cddfSDavid du Colombier enum token {
14*7dd7cddfSDavid du Colombier 	/* some (long) unary operators */
15*7dd7cddfSDavid du Colombier 	O_PLUSPLUS = 0, O_MINUSMINUS,
16*7dd7cddfSDavid du Colombier 	/* binary operators */
17*7dd7cddfSDavid du Colombier 	O_EQ, O_NE,
18*7dd7cddfSDavid du Colombier 	/* assignments are assumed to be in range O_ASN .. O_BORASN */
19*7dd7cddfSDavid du Colombier 	O_ASN, O_TIMESASN, O_DIVASN, O_MODASN, O_PLUSASN, O_MINUSASN,
20*7dd7cddfSDavid du Colombier 	       O_LSHIFTASN, O_RSHIFTASN, O_BANDASN, O_BXORASN, O_BORASN,
21*7dd7cddfSDavid du Colombier 	O_LSHIFT, O_RSHIFT,
22*7dd7cddfSDavid du Colombier 	O_LE, O_GE, O_LT, O_GT,
23*7dd7cddfSDavid du Colombier 	O_LAND,
24*7dd7cddfSDavid du Colombier 	O_LOR,
25*7dd7cddfSDavid du Colombier 	O_TIMES, O_DIV, O_MOD,
26*7dd7cddfSDavid du Colombier 	O_PLUS, O_MINUS,
27*7dd7cddfSDavid du Colombier 	O_BAND,
28*7dd7cddfSDavid du Colombier 	O_BXOR,
29*7dd7cddfSDavid du Colombier 	O_BOR,
30*7dd7cddfSDavid du Colombier 	O_TERN,
31*7dd7cddfSDavid du Colombier 	O_COMMA,
32*7dd7cddfSDavid du Colombier 	/* things after this aren't used as binary operators */
33*7dd7cddfSDavid du Colombier 	/* unary that are not also binaries */
34*7dd7cddfSDavid du Colombier 	O_BNOT, O_LNOT,
35*7dd7cddfSDavid du Colombier 	/* misc */
36*7dd7cddfSDavid du Colombier 	OPEN_PAREN, CLOSE_PAREN, CTERN,
37*7dd7cddfSDavid du Colombier 	/* things that don't appear in the opinfo[] table */
38*7dd7cddfSDavid du Colombier 	VAR, LIT, END, BAD
39*7dd7cddfSDavid du Colombier     };
40*7dd7cddfSDavid du Colombier #define IS_BINOP(op) (((int)op) >= (int)O_EQ && ((int)op) <= (int)O_COMMA)
41*7dd7cddfSDavid du Colombier #define IS_ASSIGNOP(op)	((int)(op) >= (int)O_ASN && (int)(op) <= (int)O_BORASN)
42*7dd7cddfSDavid du Colombier 
43*7dd7cddfSDavid du Colombier enum prec {
44*7dd7cddfSDavid du Colombier 	P_PRIMARY = 0,		/* VAR, LIT, (), ~ ! - + */
45*7dd7cddfSDavid du Colombier 	P_MULT,			/* * / % */
46*7dd7cddfSDavid du Colombier 	P_ADD,			/* + - */
47*7dd7cddfSDavid du Colombier 	P_SHIFT,		/* << >> */
48*7dd7cddfSDavid du Colombier 	P_RELATION,		/* < <= > >= */
49*7dd7cddfSDavid du Colombier 	P_EQUALITY,		/* == != */
50*7dd7cddfSDavid du Colombier 	P_BAND,			/* & */
51*7dd7cddfSDavid du Colombier 	P_BXOR,			/* ^ */
52*7dd7cddfSDavid du Colombier 	P_BOR,			/* | */
53*7dd7cddfSDavid du Colombier 	P_LAND,			/* && */
54*7dd7cddfSDavid du Colombier 	P_LOR,			/* || */
55*7dd7cddfSDavid du Colombier 	P_TERN,			/* ?: */
56*7dd7cddfSDavid du Colombier 	P_ASSIGN,		/* = *= /= %= += -= <<= >>= &= ^= |= */
57*7dd7cddfSDavid du Colombier 	P_COMMA			/* , */
58*7dd7cddfSDavid du Colombier     };
59*7dd7cddfSDavid du Colombier #define MAX_PREC	P_COMMA
60*7dd7cddfSDavid du Colombier 
61*7dd7cddfSDavid du Colombier struct opinfo {
62*7dd7cddfSDavid du Colombier 	char		name[4];
63*7dd7cddfSDavid du Colombier 	int		len;	/* name length */
64*7dd7cddfSDavid du Colombier 	enum prec	prec;	/* precidence: lower is higher */
65*7dd7cddfSDavid du Colombier };
66*7dd7cddfSDavid du Colombier 
67*7dd7cddfSDavid du Colombier /* Tokens in this table must be ordered so the longest are first
68*7dd7cddfSDavid du Colombier  * (eg, += before +).  If you change something, change the order
69*7dd7cddfSDavid du Colombier  * of enum token too.
70*7dd7cddfSDavid du Colombier  */
71*7dd7cddfSDavid du Colombier static const struct opinfo opinfo[] = {
72*7dd7cddfSDavid du Colombier 		{ "++",	 2, P_PRIMARY },	/* before + */
73*7dd7cddfSDavid du Colombier 		{ "--",	 2, P_PRIMARY },	/* before - */
74*7dd7cddfSDavid du Colombier 		{ "==",	 2, P_EQUALITY },	/* before = */
75*7dd7cddfSDavid du Colombier 		{ "!=",	 2, P_EQUALITY },	/* before ! */
76*7dd7cddfSDavid du Colombier 		{ "=",	 1, P_ASSIGN },		/* keep assigns in a block */
77*7dd7cddfSDavid du Colombier 		{ "*=",	 2, P_ASSIGN },
78*7dd7cddfSDavid du Colombier 		{ "/=",	 2, P_ASSIGN },
79*7dd7cddfSDavid du Colombier 		{ "%=",	 2, P_ASSIGN },
80*7dd7cddfSDavid du Colombier 		{ "+=",	 2, P_ASSIGN },
81*7dd7cddfSDavid du Colombier 		{ "-=",	 2, P_ASSIGN },
82*7dd7cddfSDavid du Colombier 		{ "<<=", 3, P_ASSIGN },
83*7dd7cddfSDavid du Colombier 		{ ">>=", 3, P_ASSIGN },
84*7dd7cddfSDavid du Colombier 		{ "&=",	 2, P_ASSIGN },
85*7dd7cddfSDavid du Colombier 		{ "^=",	 2, P_ASSIGN },
86*7dd7cddfSDavid du Colombier 		{ "|=",	 2, P_ASSIGN },
87*7dd7cddfSDavid du Colombier 		{ "<<",	 2, P_SHIFT },
88*7dd7cddfSDavid du Colombier 		{ ">>",	 2, P_SHIFT },
89*7dd7cddfSDavid du Colombier 		{ "<=",	 2, P_RELATION },
90*7dd7cddfSDavid du Colombier 		{ ">=",	 2, P_RELATION },
91*7dd7cddfSDavid du Colombier 		{ "<",	 1, P_RELATION },
92*7dd7cddfSDavid du Colombier 		{ ">",	 1, P_RELATION },
93*7dd7cddfSDavid du Colombier 		{ "&&",	 2, P_LAND },
94*7dd7cddfSDavid du Colombier 		{ "||",	 2, P_LOR },
95*7dd7cddfSDavid du Colombier 		{ "*",	 1, P_MULT },
96*7dd7cddfSDavid du Colombier 		{ "/",	 1, P_MULT },
97*7dd7cddfSDavid du Colombier 		{ "%",	 1, P_MULT },
98*7dd7cddfSDavid du Colombier 		{ "+",	 1, P_ADD },
99*7dd7cddfSDavid du Colombier 		{ "-",	 1, P_ADD },
100*7dd7cddfSDavid du Colombier 		{ "&",	 1, P_BAND },
101*7dd7cddfSDavid du Colombier 		{ "^",	 1, P_BXOR },
102*7dd7cddfSDavid du Colombier 		{ "|",	 1, P_BOR },
103*7dd7cddfSDavid du Colombier 		{ "?",	 1, P_TERN },
104*7dd7cddfSDavid du Colombier 		{ ",",	 1, P_COMMA },
105*7dd7cddfSDavid du Colombier 		{ "~",	 1, P_PRIMARY },
106*7dd7cddfSDavid du Colombier 		{ "!",	 1, P_PRIMARY },
107*7dd7cddfSDavid du Colombier 		{ "(",	 1, P_PRIMARY },
108*7dd7cddfSDavid du Colombier 		{ ")",	 1, P_PRIMARY },
109*7dd7cddfSDavid du Colombier 		{ ":",	 1, P_PRIMARY },
110*7dd7cddfSDavid du Colombier 		{ "",	 0, P_PRIMARY } /* end of table */
111*7dd7cddfSDavid du Colombier 	    };
112*7dd7cddfSDavid du Colombier 
113*7dd7cddfSDavid du Colombier 
114*7dd7cddfSDavid du Colombier typedef struct expr_state Expr_state;
115*7dd7cddfSDavid du Colombier struct expr_state {
116*7dd7cddfSDavid du Colombier 	const char *expression;		/* expression being evaluated */
117*7dd7cddfSDavid du Colombier 	const char *tokp;		/* lexical position */
118*7dd7cddfSDavid du Colombier 	enum token  tok;		/* token from token() */
119*7dd7cddfSDavid du Colombier 	int	    noassign;		/* don't do assigns (for ?:,&&,||) */
120*7dd7cddfSDavid du Colombier 	struct tbl *val;		/* value from token() */
121*7dd7cddfSDavid du Colombier 	struct tbl *evaling;		/* variable that is being recursively
122*7dd7cddfSDavid du Colombier 					 * expanded (EXPRINEVAL flag set)
123*7dd7cddfSDavid du Colombier 					 */
124*7dd7cddfSDavid du Colombier };
125*7dd7cddfSDavid du Colombier 
126*7dd7cddfSDavid du Colombier enum error_type { ET_UNEXPECTED, ET_BADLIT, ET_RECURSIVE,
127*7dd7cddfSDavid du Colombier 		  ET_LVALUE, ET_RDONLY, ET_STR };
128*7dd7cddfSDavid du Colombier 
129*7dd7cddfSDavid du Colombier static void        evalerr  ARGS((Expr_state *es, enum error_type type,
130*7dd7cddfSDavid du Colombier 				  const char *str)) GCC_FUNC_ATTR(noreturn);
131*7dd7cddfSDavid du Colombier static struct tbl *evalexpr ARGS((Expr_state *es, enum prec prec));
132*7dd7cddfSDavid du Colombier static void        token    ARGS((Expr_state *es));
133*7dd7cddfSDavid du Colombier static struct tbl *do_ppmm  ARGS((Expr_state *es, enum token op,
134*7dd7cddfSDavid du Colombier 				  struct tbl *vasn, bool_t is_prefix));
135*7dd7cddfSDavid du Colombier static void	   assign_check ARGS((Expr_state *es, enum token op,
136*7dd7cddfSDavid du Colombier 				      struct tbl *vasn));
137*7dd7cddfSDavid du Colombier static struct tbl *tempvar  ARGS((void));
138*7dd7cddfSDavid du Colombier static struct tbl *intvar   ARGS((Expr_state *es, struct tbl *vp));
139*7dd7cddfSDavid du Colombier 
140*7dd7cddfSDavid du Colombier /*
141*7dd7cddfSDavid du Colombier  * parse and evalute expression
142*7dd7cddfSDavid du Colombier  */
143*7dd7cddfSDavid du Colombier int
evaluate(expr,rval,error_ok)144*7dd7cddfSDavid du Colombier evaluate(expr, rval, error_ok)
145*7dd7cddfSDavid du Colombier 	const char *expr;
146*7dd7cddfSDavid du Colombier 	long *rval;
147*7dd7cddfSDavid du Colombier 	int error_ok;
148*7dd7cddfSDavid du Colombier {
149*7dd7cddfSDavid du Colombier 	struct tbl v;
150*7dd7cddfSDavid du Colombier 	int ret;
151*7dd7cddfSDavid du Colombier 
152*7dd7cddfSDavid du Colombier 	v.flag = DEFINED|INTEGER;
153*7dd7cddfSDavid du Colombier 	v.type = 0;
154*7dd7cddfSDavid du Colombier 	ret = v_evaluate(&v, expr, error_ok);
155*7dd7cddfSDavid du Colombier 	*rval = v.val.i;
156*7dd7cddfSDavid du Colombier 	return ret;
157*7dd7cddfSDavid du Colombier }
158*7dd7cddfSDavid du Colombier 
159*7dd7cddfSDavid du Colombier /*
160*7dd7cddfSDavid du Colombier  * parse and evalute expression, storing result in vp.
161*7dd7cddfSDavid du Colombier  */
162*7dd7cddfSDavid du Colombier int
v_evaluate(vp,expr,error_ok)163*7dd7cddfSDavid du Colombier v_evaluate(vp, expr, error_ok)
164*7dd7cddfSDavid du Colombier 	struct tbl *vp;
165*7dd7cddfSDavid du Colombier 	const char *expr;
166*7dd7cddfSDavid du Colombier 	volatile int error_ok;
167*7dd7cddfSDavid du Colombier {
168*7dd7cddfSDavid du Colombier 	struct tbl *v;
169*7dd7cddfSDavid du Colombier 	Expr_state curstate;
170*7dd7cddfSDavid du Colombier 	Expr_state * const es = &curstate;
171*7dd7cddfSDavid du Colombier 	int i;
172*7dd7cddfSDavid du Colombier 
173*7dd7cddfSDavid du Colombier 	/* save state to allow recursive calls */
174*7dd7cddfSDavid du Colombier 	curstate.expression = curstate.tokp = expr;
175*7dd7cddfSDavid du Colombier 	curstate.noassign = 0;
176*7dd7cddfSDavid du Colombier 	curstate.evaling = (struct tbl *) 0;
177*7dd7cddfSDavid du Colombier 
178*7dd7cddfSDavid du Colombier 	newenv(E_ERRH);
179*7dd7cddfSDavid du Colombier 	i = ksh_sigsetjmp(e->jbuf, 0);
180*7dd7cddfSDavid du Colombier 	if (i) {
181*7dd7cddfSDavid du Colombier 		/* Clear EXPRINEVAL in of any variables we were playing with */
182*7dd7cddfSDavid du Colombier 		if (curstate.evaling)
183*7dd7cddfSDavid du Colombier 			curstate.evaling->flag &= ~EXPRINEVAL;
184*7dd7cddfSDavid du Colombier 		quitenv();
185*7dd7cddfSDavid du Colombier 		if (i == LAEXPR) {
186*7dd7cddfSDavid du Colombier 			if (error_ok == KSH_RETURN_ERROR)
187*7dd7cddfSDavid du Colombier 				return 0;
188*7dd7cddfSDavid du Colombier 			errorf(null);
189*7dd7cddfSDavid du Colombier 		}
190*7dd7cddfSDavid du Colombier 		unwind(i);
191*7dd7cddfSDavid du Colombier 		/*NOTREACHED*/
192*7dd7cddfSDavid du Colombier 	}
193*7dd7cddfSDavid du Colombier 
194*7dd7cddfSDavid du Colombier 	token(es);
195*7dd7cddfSDavid du Colombier #if 1 /* ifdef-out to disallow empty expressions to be treated as 0 */
196*7dd7cddfSDavid du Colombier 	if (es->tok == END) {
197*7dd7cddfSDavid du Colombier 		es->tok = LIT;
198*7dd7cddfSDavid du Colombier 		es->val = tempvar();
199*7dd7cddfSDavid du Colombier 	}
200*7dd7cddfSDavid du Colombier #endif /* 0 */
201*7dd7cddfSDavid du Colombier 	v = intvar(es, evalexpr(es, MAX_PREC));
202*7dd7cddfSDavid du Colombier 
203*7dd7cddfSDavid du Colombier 	if (es->tok != END)
204*7dd7cddfSDavid du Colombier 		evalerr(es, ET_UNEXPECTED, (char *) 0);
205*7dd7cddfSDavid du Colombier 
206*7dd7cddfSDavid du Colombier 	if (vp->flag & INTEGER)
207*7dd7cddfSDavid du Colombier 		setint_v(vp, v);
208*7dd7cddfSDavid du Colombier 	else
209*7dd7cddfSDavid du Colombier 		/* can fail if readony */
210*7dd7cddfSDavid du Colombier 		setstr(vp, str_val(v), error_ok);
211*7dd7cddfSDavid du Colombier 
212*7dd7cddfSDavid du Colombier 	quitenv();
213*7dd7cddfSDavid du Colombier 
214*7dd7cddfSDavid du Colombier 	return 1;
215*7dd7cddfSDavid du Colombier }
216*7dd7cddfSDavid du Colombier 
217*7dd7cddfSDavid du Colombier static void
evalerr(es,type,str)218*7dd7cddfSDavid du Colombier evalerr(es, type, str)
219*7dd7cddfSDavid du Colombier 	Expr_state *es;
220*7dd7cddfSDavid du Colombier 	enum error_type type;
221*7dd7cddfSDavid du Colombier 	const char *str;
222*7dd7cddfSDavid du Colombier {
223*7dd7cddfSDavid du Colombier 	char tbuf[2];
224*7dd7cddfSDavid du Colombier 	const char *s;
225*7dd7cddfSDavid du Colombier 
226*7dd7cddfSDavid du Colombier 	switch (type) {
227*7dd7cddfSDavid du Colombier 	case ET_UNEXPECTED:
228*7dd7cddfSDavid du Colombier 		switch (es->tok) {
229*7dd7cddfSDavid du Colombier 		case VAR:
230*7dd7cddfSDavid du Colombier 			s = es->val->name;
231*7dd7cddfSDavid du Colombier 			break;
232*7dd7cddfSDavid du Colombier 		case LIT:
233*7dd7cddfSDavid du Colombier 			s = str_val(es->val);
234*7dd7cddfSDavid du Colombier 			break;
235*7dd7cddfSDavid du Colombier 		case END:
236*7dd7cddfSDavid du Colombier 			s = "end of expression";
237*7dd7cddfSDavid du Colombier 			break;
238*7dd7cddfSDavid du Colombier 		case BAD:
239*7dd7cddfSDavid du Colombier 			tbuf[0] = *es->tokp;
240*7dd7cddfSDavid du Colombier 			tbuf[1] = '\0';
241*7dd7cddfSDavid du Colombier 			s = tbuf;
242*7dd7cddfSDavid du Colombier 			break;
243*7dd7cddfSDavid du Colombier 		default:
244*7dd7cddfSDavid du Colombier 			s = opinfo[(int)es->tok].name;
245*7dd7cddfSDavid du Colombier 		}
246*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: unexpected `%s'", es->expression, s);
247*7dd7cddfSDavid du Colombier 		break;
248*7dd7cddfSDavid du Colombier 
249*7dd7cddfSDavid du Colombier 	case ET_BADLIT:
250*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: bad number `%s'", es->expression, str);
251*7dd7cddfSDavid du Colombier 		break;
252*7dd7cddfSDavid du Colombier 
253*7dd7cddfSDavid du Colombier 	case ET_RECURSIVE:
254*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: expression recurses on parameter `%s'",
255*7dd7cddfSDavid du Colombier 			es->expression, str);
256*7dd7cddfSDavid du Colombier 		break;
257*7dd7cddfSDavid du Colombier 
258*7dd7cddfSDavid du Colombier 	case ET_LVALUE:
259*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: %s requires lvalue",
260*7dd7cddfSDavid du Colombier 			es->expression, str);
261*7dd7cddfSDavid du Colombier 		break;
262*7dd7cddfSDavid du Colombier 
263*7dd7cddfSDavid du Colombier 	case ET_RDONLY:
264*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: %s applied to read only variable",
265*7dd7cddfSDavid du Colombier 			es->expression, str);
266*7dd7cddfSDavid du Colombier 		break;
267*7dd7cddfSDavid du Colombier 
268*7dd7cddfSDavid du Colombier 	default: /* keep gcc happy */
269*7dd7cddfSDavid du Colombier 	case ET_STR:
270*7dd7cddfSDavid du Colombier 		warningf(TRUE, "%s: %s", es->expression, str);
271*7dd7cddfSDavid du Colombier 		break;
272*7dd7cddfSDavid du Colombier 	}
273*7dd7cddfSDavid du Colombier 	unwind(LAEXPR);
274*7dd7cddfSDavid du Colombier }
275*7dd7cddfSDavid du Colombier 
276*7dd7cddfSDavid du Colombier static struct tbl *
evalexpr(es,prec)277*7dd7cddfSDavid du Colombier evalexpr(es, prec)
278*7dd7cddfSDavid du Colombier 	Expr_state *es;
279*7dd7cddfSDavid du Colombier 	enum prec prec;
280*7dd7cddfSDavid du Colombier {
281*7dd7cddfSDavid du Colombier 	struct tbl *vl, UNINITIALIZED(*vr), *vasn;
282*7dd7cddfSDavid du Colombier 	enum token op;
283*7dd7cddfSDavid du Colombier 	long UNINITIALIZED(res);
284*7dd7cddfSDavid du Colombier 
285*7dd7cddfSDavid du Colombier 	if (prec == P_PRIMARY) {
286*7dd7cddfSDavid du Colombier 		op = es->tok;
287*7dd7cddfSDavid du Colombier 		if (op == O_BNOT || op == O_LNOT || op == O_MINUS
288*7dd7cddfSDavid du Colombier 		    || op == O_PLUS)
289*7dd7cddfSDavid du Colombier 		{
290*7dd7cddfSDavid du Colombier 			token(es);
291*7dd7cddfSDavid du Colombier 			vl = intvar(es, evalexpr(es, P_PRIMARY));
292*7dd7cddfSDavid du Colombier 			if (op == O_BNOT)
293*7dd7cddfSDavid du Colombier 				vl->val.i = ~vl->val.i;
294*7dd7cddfSDavid du Colombier 			else if (op == O_LNOT)
295*7dd7cddfSDavid du Colombier 				vl->val.i = !vl->val.i;
296*7dd7cddfSDavid du Colombier 			else if (op == O_MINUS)
297*7dd7cddfSDavid du Colombier 				vl->val.i = -vl->val.i;
298*7dd7cddfSDavid du Colombier 			/* op == O_PLUS is a no-op */
299*7dd7cddfSDavid du Colombier 		} else if (op == OPEN_PAREN) {
300*7dd7cddfSDavid du Colombier 			token(es);
301*7dd7cddfSDavid du Colombier 			vl = evalexpr(es, MAX_PREC);
302*7dd7cddfSDavid du Colombier 			if (es->tok != CLOSE_PAREN)
303*7dd7cddfSDavid du Colombier 				evalerr(es, ET_STR, "missing )");
304*7dd7cddfSDavid du Colombier 			token(es);
305*7dd7cddfSDavid du Colombier 		} else if (op == O_PLUSPLUS || op == O_MINUSMINUS) {
306*7dd7cddfSDavid du Colombier 			token(es);
307*7dd7cddfSDavid du Colombier 			vl = do_ppmm(es, op, es->val, TRUE);
308*7dd7cddfSDavid du Colombier 			token(es);
309*7dd7cddfSDavid du Colombier 		} else if (op == VAR || op == LIT) {
310*7dd7cddfSDavid du Colombier 			vl = es->val;
311*7dd7cddfSDavid du Colombier 			token(es);
312*7dd7cddfSDavid du Colombier 		} else {
313*7dd7cddfSDavid du Colombier 			evalerr(es, ET_UNEXPECTED, (char *) 0);
314*7dd7cddfSDavid du Colombier 			/*NOTREACHED*/
315*7dd7cddfSDavid du Colombier 		}
316*7dd7cddfSDavid du Colombier 		if (es->tok == O_PLUSPLUS || es->tok == O_MINUSMINUS) {
317*7dd7cddfSDavid du Colombier 			vl = do_ppmm(es, es->tok, vl, FALSE);
318*7dd7cddfSDavid du Colombier 			token(es);
319*7dd7cddfSDavid du Colombier 		}
320*7dd7cddfSDavid du Colombier 		return vl;
321*7dd7cddfSDavid du Colombier 	}
322*7dd7cddfSDavid du Colombier 	vl = evalexpr(es, ((int) prec) - 1);
323*7dd7cddfSDavid du Colombier 	for (op = es->tok; IS_BINOP(op) && opinfo[(int) op].prec == prec;
324*7dd7cddfSDavid du Colombier 		op = es->tok)
325*7dd7cddfSDavid du Colombier 	{
326*7dd7cddfSDavid du Colombier 		token(es);
327*7dd7cddfSDavid du Colombier 		vasn = vl;
328*7dd7cddfSDavid du Colombier 		if (op != O_ASN) /* vl may not have a value yet */
329*7dd7cddfSDavid du Colombier 			vl = intvar(es, vl);
330*7dd7cddfSDavid du Colombier 		if (IS_ASSIGNOP(op)) {
331*7dd7cddfSDavid du Colombier 			assign_check(es, op, vasn);
332*7dd7cddfSDavid du Colombier 			vr = intvar(es, evalexpr(es, P_ASSIGN));
333*7dd7cddfSDavid du Colombier 		} else if (op != O_TERN && op != O_LAND && op != O_LOR)
334*7dd7cddfSDavid du Colombier 			vr = intvar(es, evalexpr(es, ((int) prec) - 1));
335*7dd7cddfSDavid du Colombier 		if ((op == O_DIV || op == O_MOD || op == O_DIVASN
336*7dd7cddfSDavid du Colombier 		     || op == O_MODASN) && vr->val.i == 0)
337*7dd7cddfSDavid du Colombier 		{
338*7dd7cddfSDavid du Colombier 			if (es->noassign)
339*7dd7cddfSDavid du Colombier 				vr->val.i = 1;
340*7dd7cddfSDavid du Colombier 			else
341*7dd7cddfSDavid du Colombier 				evalerr(es, ET_STR, "zero divisor");
342*7dd7cddfSDavid du Colombier 		}
343*7dd7cddfSDavid du Colombier 		switch ((int) op) {
344*7dd7cddfSDavid du Colombier 		case O_TIMES:
345*7dd7cddfSDavid du Colombier 		case O_TIMESASN:
346*7dd7cddfSDavid du Colombier 			res = vl->val.i * vr->val.i;
347*7dd7cddfSDavid du Colombier 			break;
348*7dd7cddfSDavid du Colombier 		case O_DIV:
349*7dd7cddfSDavid du Colombier 		case O_DIVASN:
350*7dd7cddfSDavid du Colombier 			res = vl->val.i / vr->val.i;
351*7dd7cddfSDavid du Colombier 			break;
352*7dd7cddfSDavid du Colombier 		case O_MOD:
353*7dd7cddfSDavid du Colombier 		case O_MODASN:
354*7dd7cddfSDavid du Colombier 			res = vl->val.i % vr->val.i;
355*7dd7cddfSDavid du Colombier 			break;
356*7dd7cddfSDavid du Colombier 		case O_PLUS:
357*7dd7cddfSDavid du Colombier 		case O_PLUSASN:
358*7dd7cddfSDavid du Colombier 			res = vl->val.i + vr->val.i;
359*7dd7cddfSDavid du Colombier 			break;
360*7dd7cddfSDavid du Colombier 		case O_MINUS:
361*7dd7cddfSDavid du Colombier 		case O_MINUSASN:
362*7dd7cddfSDavid du Colombier 			res = vl->val.i - vr->val.i;
363*7dd7cddfSDavid du Colombier 			break;
364*7dd7cddfSDavid du Colombier 		case O_LSHIFT:
365*7dd7cddfSDavid du Colombier 		case O_LSHIFTASN:
366*7dd7cddfSDavid du Colombier 			res = vl->val.i << vr->val.i;
367*7dd7cddfSDavid du Colombier 			break;
368*7dd7cddfSDavid du Colombier 		case O_RSHIFT:
369*7dd7cddfSDavid du Colombier 		case O_RSHIFTASN:
370*7dd7cddfSDavid du Colombier 			res = vl->val.i >> vr->val.i;
371*7dd7cddfSDavid du Colombier 			break;
372*7dd7cddfSDavid du Colombier 		case O_LT:
373*7dd7cddfSDavid du Colombier 			res = vl->val.i < vr->val.i;
374*7dd7cddfSDavid du Colombier 			break;
375*7dd7cddfSDavid du Colombier 		case O_LE:
376*7dd7cddfSDavid du Colombier 			res = vl->val.i <= vr->val.i;
377*7dd7cddfSDavid du Colombier 			break;
378*7dd7cddfSDavid du Colombier 		case O_GT:
379*7dd7cddfSDavid du Colombier 			res = vl->val.i > vr->val.i;
380*7dd7cddfSDavid du Colombier 			break;
381*7dd7cddfSDavid du Colombier 		case O_GE:
382*7dd7cddfSDavid du Colombier 			res = vl->val.i >= vr->val.i;
383*7dd7cddfSDavid du Colombier 			break;
384*7dd7cddfSDavid du Colombier 		case O_EQ:
385*7dd7cddfSDavid du Colombier 			res = vl->val.i == vr->val.i;
386*7dd7cddfSDavid du Colombier 			break;
387*7dd7cddfSDavid du Colombier 		case O_NE:
388*7dd7cddfSDavid du Colombier 			res = vl->val.i != vr->val.i;
389*7dd7cddfSDavid du Colombier 			break;
390*7dd7cddfSDavid du Colombier 		case O_BAND:
391*7dd7cddfSDavid du Colombier 		case O_BANDASN:
392*7dd7cddfSDavid du Colombier 			res = vl->val.i & vr->val.i;
393*7dd7cddfSDavid du Colombier 			break;
394*7dd7cddfSDavid du Colombier 		case O_BXOR:
395*7dd7cddfSDavid du Colombier 		case O_BXORASN:
396*7dd7cddfSDavid du Colombier 			res = vl->val.i ^ vr->val.i;
397*7dd7cddfSDavid du Colombier 			break;
398*7dd7cddfSDavid du Colombier 		case O_BOR:
399*7dd7cddfSDavid du Colombier 		case O_BORASN:
400*7dd7cddfSDavid du Colombier 			res = vl->val.i | vr->val.i;
401*7dd7cddfSDavid du Colombier 			break;
402*7dd7cddfSDavid du Colombier 		case O_LAND:
403*7dd7cddfSDavid du Colombier 			if (!vl->val.i)
404*7dd7cddfSDavid du Colombier 				es->noassign++;
405*7dd7cddfSDavid du Colombier 			vr = intvar(es, evalexpr(es, ((int) prec) - 1));
406*7dd7cddfSDavid du Colombier 			res = vl->val.i && vr->val.i;
407*7dd7cddfSDavid du Colombier 			if (!vl->val.i)
408*7dd7cddfSDavid du Colombier 				es->noassign--;
409*7dd7cddfSDavid du Colombier 			break;
410*7dd7cddfSDavid du Colombier 		case O_LOR:
411*7dd7cddfSDavid du Colombier 			if (vl->val.i)
412*7dd7cddfSDavid du Colombier 				es->noassign++;
413*7dd7cddfSDavid du Colombier 			vr = intvar(es, evalexpr(es, ((int) prec) - 1));
414*7dd7cddfSDavid du Colombier 			res = vl->val.i || vr->val.i;
415*7dd7cddfSDavid du Colombier 			if (vl->val.i)
416*7dd7cddfSDavid du Colombier 				es->noassign--;
417*7dd7cddfSDavid du Colombier 			break;
418*7dd7cddfSDavid du Colombier 		case O_TERN:
419*7dd7cddfSDavid du Colombier 			{
420*7dd7cddfSDavid du Colombier 				int e = vl->val.i != 0;
421*7dd7cddfSDavid du Colombier 				if (!e)
422*7dd7cddfSDavid du Colombier 					es->noassign++;
423*7dd7cddfSDavid du Colombier 				vl = evalexpr(es, MAX_PREC);
424*7dd7cddfSDavid du Colombier 				if (!e)
425*7dd7cddfSDavid du Colombier 					es->noassign--;
426*7dd7cddfSDavid du Colombier 				if (es->tok != CTERN)
427*7dd7cddfSDavid du Colombier 					evalerr(es, ET_STR, "missing :");
428*7dd7cddfSDavid du Colombier 				token(es);
429*7dd7cddfSDavid du Colombier 				if (e)
430*7dd7cddfSDavid du Colombier 					es->noassign++;
431*7dd7cddfSDavid du Colombier 				vr = evalexpr(es, P_TERN);
432*7dd7cddfSDavid du Colombier 				if (e)
433*7dd7cddfSDavid du Colombier 					es->noassign--;
434*7dd7cddfSDavid du Colombier 				vl = e ? vl : vr;
435*7dd7cddfSDavid du Colombier 			}
436*7dd7cddfSDavid du Colombier 			break;
437*7dd7cddfSDavid du Colombier 		case O_ASN:
438*7dd7cddfSDavid du Colombier 			res = vr->val.i;
439*7dd7cddfSDavid du Colombier 			break;
440*7dd7cddfSDavid du Colombier 		case O_COMMA:
441*7dd7cddfSDavid du Colombier 			res = vr->val.i;
442*7dd7cddfSDavid du Colombier 			break;
443*7dd7cddfSDavid du Colombier 		}
444*7dd7cddfSDavid du Colombier 		if (IS_ASSIGNOP(op)) {
445*7dd7cddfSDavid du Colombier 			vr->val.i = res;
446*7dd7cddfSDavid du Colombier 			if (vasn->flag & INTEGER)
447*7dd7cddfSDavid du Colombier 				setint_v(vasn, vr);
448*7dd7cddfSDavid du Colombier 			else
449*7dd7cddfSDavid du Colombier 				setint(vasn, res);
450*7dd7cddfSDavid du Colombier 			vl = vr;
451*7dd7cddfSDavid du Colombier 		} else if (op != O_TERN)
452*7dd7cddfSDavid du Colombier 			vl->val.i = res;
453*7dd7cddfSDavid du Colombier 	}
454*7dd7cddfSDavid du Colombier 	return vl;
455*7dd7cddfSDavid du Colombier }
456*7dd7cddfSDavid du Colombier 
457*7dd7cddfSDavid du Colombier static void
token(es)458*7dd7cddfSDavid du Colombier token(es)
459*7dd7cddfSDavid du Colombier 	Expr_state *es;
460*7dd7cddfSDavid du Colombier {
461*7dd7cddfSDavid du Colombier 	const char *cp;
462*7dd7cddfSDavid du Colombier 	int c;
463*7dd7cddfSDavid du Colombier 	char *tvar;
464*7dd7cddfSDavid du Colombier 
465*7dd7cddfSDavid du Colombier 	/* skip white space */
466*7dd7cddfSDavid du Colombier 	for (cp = es->tokp; (c = *cp), isspace(c); cp++)
467*7dd7cddfSDavid du Colombier 		;
468*7dd7cddfSDavid du Colombier 	es->tokp = cp;
469*7dd7cddfSDavid du Colombier 
470*7dd7cddfSDavid du Colombier 	if (c == '\0')
471*7dd7cddfSDavid du Colombier 		es->tok = END;
472*7dd7cddfSDavid du Colombier 	else if (letter(c)) {
473*7dd7cddfSDavid du Colombier 		for (; letnum(c); c = *cp)
474*7dd7cddfSDavid du Colombier 			cp++;
475*7dd7cddfSDavid du Colombier 		if (c == '[') {
476*7dd7cddfSDavid du Colombier 			int len;
477*7dd7cddfSDavid du Colombier 
478*7dd7cddfSDavid du Colombier 			len = array_ref_len(cp);
479*7dd7cddfSDavid du Colombier 			if (len == 0)
480*7dd7cddfSDavid du Colombier 				evalerr(es, ET_STR, "missing ]");
481*7dd7cddfSDavid du Colombier 			cp += len;
482*7dd7cddfSDavid du Colombier 		}
483*7dd7cddfSDavid du Colombier #ifdef KSH
484*7dd7cddfSDavid du Colombier 		else if (c == '(' /*)*/ ) {
485*7dd7cddfSDavid du Colombier 		    /* todo: add math functions (all take single argument):
486*7dd7cddfSDavid du Colombier 		     * abs acos asin atan cos cosh exp int log sin sinh sqrt
487*7dd7cddfSDavid du Colombier 		     * tan tanh
488*7dd7cddfSDavid du Colombier 		     */
489*7dd7cddfSDavid du Colombier 		    ;
490*7dd7cddfSDavid du Colombier 		}
491*7dd7cddfSDavid du Colombier #endif /* KSH */
492*7dd7cddfSDavid du Colombier 		if (es->noassign) {
493*7dd7cddfSDavid du Colombier 			es->val = tempvar();
494*7dd7cddfSDavid du Colombier 			es->val->flag |= EXPRLVALUE;
495*7dd7cddfSDavid du Colombier 		} else {
496*7dd7cddfSDavid du Colombier 			tvar = str_nsave(es->tokp, cp - es->tokp, ATEMP);
497*7dd7cddfSDavid du Colombier 			es->val = global(tvar);
498*7dd7cddfSDavid du Colombier 			afree(tvar, ATEMP);
499*7dd7cddfSDavid du Colombier 		}
500*7dd7cddfSDavid du Colombier 		es->tok = VAR;
501*7dd7cddfSDavid du Colombier 	} else if (digit(c)) {
502*7dd7cddfSDavid du Colombier 		for (; c != '_' && (letnum(c) || c == '#'); c = *cp++)
503*7dd7cddfSDavid du Colombier 			;
504*7dd7cddfSDavid du Colombier 		tvar = str_nsave(es->tokp, --cp - es->tokp, ATEMP);
505*7dd7cddfSDavid du Colombier 		es->val = tempvar();
506*7dd7cddfSDavid du Colombier 		es->val->flag &= ~INTEGER;
507*7dd7cddfSDavid du Colombier 		es->val->type = 0;
508*7dd7cddfSDavid du Colombier 		es->val->val.s = tvar;
509*7dd7cddfSDavid du Colombier 		if (setint_v(es->val, es->val) == NULL)
510*7dd7cddfSDavid du Colombier 			evalerr(es, ET_BADLIT, tvar);
511*7dd7cddfSDavid du Colombier 		afree(tvar, ATEMP);
512*7dd7cddfSDavid du Colombier 		es->tok = LIT;
513*7dd7cddfSDavid du Colombier 	} else {
514*7dd7cddfSDavid du Colombier 		int i, n0;
515*7dd7cddfSDavid du Colombier 
516*7dd7cddfSDavid du Colombier 		for (i = 0; (n0 = opinfo[i].name[0]); i++)
517*7dd7cddfSDavid du Colombier 			if (c == n0
518*7dd7cddfSDavid du Colombier 			    && strncmp(cp, opinfo[i].name, opinfo[i].len) == 0)
519*7dd7cddfSDavid du Colombier 			{
520*7dd7cddfSDavid du Colombier 				es->tok = (enum token) i;
521*7dd7cddfSDavid du Colombier 				cp += opinfo[i].len;
522*7dd7cddfSDavid du Colombier 				break;
523*7dd7cddfSDavid du Colombier 			}
524*7dd7cddfSDavid du Colombier 		if (!n0)
525*7dd7cddfSDavid du Colombier 			es->tok = BAD;
526*7dd7cddfSDavid du Colombier 	}
527*7dd7cddfSDavid du Colombier 	es->tokp = cp;
528*7dd7cddfSDavid du Colombier }
529*7dd7cddfSDavid du Colombier 
530*7dd7cddfSDavid du Colombier /* Do a ++ or -- operation */
531*7dd7cddfSDavid du Colombier static struct tbl *
do_ppmm(es,op,vasn,is_prefix)532*7dd7cddfSDavid du Colombier do_ppmm(es, op, vasn, is_prefix)
533*7dd7cddfSDavid du Colombier 	Expr_state *es;
534*7dd7cddfSDavid du Colombier 	enum token op;
535*7dd7cddfSDavid du Colombier 	struct tbl *vasn;
536*7dd7cddfSDavid du Colombier 	bool_t is_prefix;
537*7dd7cddfSDavid du Colombier {
538*7dd7cddfSDavid du Colombier 	struct tbl *vl;
539*7dd7cddfSDavid du Colombier 	int oval;
540*7dd7cddfSDavid du Colombier 
541*7dd7cddfSDavid du Colombier 	assign_check(es, op, vasn);
542*7dd7cddfSDavid du Colombier 
543*7dd7cddfSDavid du Colombier 	vl = intvar(es, vasn);
544*7dd7cddfSDavid du Colombier 	oval = op == O_PLUSPLUS ? vl->val.i++ : vl->val.i--;
545*7dd7cddfSDavid du Colombier 	if (vasn->flag & INTEGER)
546*7dd7cddfSDavid du Colombier 		setint_v(vasn, vl);
547*7dd7cddfSDavid du Colombier 	else
548*7dd7cddfSDavid du Colombier 		setint(vasn, vl->val.i);
549*7dd7cddfSDavid du Colombier 	if (!is_prefix)		/* undo the inc/dec */
550*7dd7cddfSDavid du Colombier 		vl->val.i = oval;
551*7dd7cddfSDavid du Colombier 
552*7dd7cddfSDavid du Colombier 	return vl;
553*7dd7cddfSDavid du Colombier }
554*7dd7cddfSDavid du Colombier 
555*7dd7cddfSDavid du Colombier static void
assign_check(es,op,vasn)556*7dd7cddfSDavid du Colombier assign_check(es, op, vasn)
557*7dd7cddfSDavid du Colombier 	Expr_state *es;
558*7dd7cddfSDavid du Colombier 	enum token op;
559*7dd7cddfSDavid du Colombier 	struct tbl *vasn;
560*7dd7cddfSDavid du Colombier {
561*7dd7cddfSDavid du Colombier 	if (vasn->name[0] == '\0' && !(vasn->flag & EXPRLVALUE))
562*7dd7cddfSDavid du Colombier 		evalerr(es, ET_LVALUE, opinfo[(int) op].name);
563*7dd7cddfSDavid du Colombier 	else if (vasn->flag & RDONLY)
564*7dd7cddfSDavid du Colombier 		evalerr(es, ET_RDONLY, opinfo[(int) op].name);
565*7dd7cddfSDavid du Colombier }
566*7dd7cddfSDavid du Colombier 
567*7dd7cddfSDavid du Colombier static struct tbl *
tempvar()568*7dd7cddfSDavid du Colombier tempvar()
569*7dd7cddfSDavid du Colombier {
570*7dd7cddfSDavid du Colombier 	register struct tbl *vp;
571*7dd7cddfSDavid du Colombier 
572*7dd7cddfSDavid du Colombier 	vp = (struct tbl*) alloc(sizeof(struct tbl), ATEMP);
573*7dd7cddfSDavid du Colombier 	vp->flag = ISSET|INTEGER;
574*7dd7cddfSDavid du Colombier 	vp->type = 0;
575*7dd7cddfSDavid du Colombier 	vp->areap = ATEMP;
576*7dd7cddfSDavid du Colombier 	vp->val.i = 0;
577*7dd7cddfSDavid du Colombier 	vp->name[0] = '\0';
578*7dd7cddfSDavid du Colombier 	return vp;
579*7dd7cddfSDavid du Colombier }
580*7dd7cddfSDavid du Colombier 
581*7dd7cddfSDavid du Colombier /* cast (string) variable to temporary integer variable */
582*7dd7cddfSDavid du Colombier static struct tbl *
intvar(es,vp)583*7dd7cddfSDavid du Colombier intvar(es, vp)
584*7dd7cddfSDavid du Colombier 	Expr_state *es;
585*7dd7cddfSDavid du Colombier 	struct tbl *vp;
586*7dd7cddfSDavid du Colombier {
587*7dd7cddfSDavid du Colombier 	struct tbl *vq;
588*7dd7cddfSDavid du Colombier 
589*7dd7cddfSDavid du Colombier 	/* try to avoid replacing a temp var with another temp var */
590*7dd7cddfSDavid du Colombier 	if (vp->name[0] == '\0'
591*7dd7cddfSDavid du Colombier 	    && (vp->flag & (ISSET|INTEGER|EXPRLVALUE)) == (ISSET|INTEGER))
592*7dd7cddfSDavid du Colombier 		return vp;
593*7dd7cddfSDavid du Colombier 
594*7dd7cddfSDavid du Colombier 	vq = tempvar();
595*7dd7cddfSDavid du Colombier 	if (setint_v(vq, vp) == NULL) {
596*7dd7cddfSDavid du Colombier 		if (vp->flag & EXPRINEVAL)
597*7dd7cddfSDavid du Colombier 			evalerr(es, ET_RECURSIVE, vp->name);
598*7dd7cddfSDavid du Colombier 		es->evaling = vp;
599*7dd7cddfSDavid du Colombier 		vp->flag |= EXPRINEVAL;
600*7dd7cddfSDavid du Colombier 		v_evaluate(vq, str_val(vp), KSH_UNWIND_ERROR);
601*7dd7cddfSDavid du Colombier 		vp->flag &= ~EXPRINEVAL;
602*7dd7cddfSDavid du Colombier 		es->evaling = (struct tbl *) 0;
603*7dd7cddfSDavid du Colombier 	}
604*7dd7cddfSDavid du Colombier 	return vq;
605*7dd7cddfSDavid du Colombier }
606