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