2 /* Written by Pace Willisson (pace@blitz.com)
3 * and placed in the public domain.
5 * Largely rewritten by J.T. Conklin (jtc@wimsey.com)
7 * $FreeBSD: src/bin/expr/expr.y,v 1.14.2.3 2001/08/01 02:37:46 obrien Exp $
8 * $DragonFly: src/bin/expr/expr.y,v 1.3 2003/09/21 04:21:56 drhodus Exp $
11 #include <sys/types.h>
23 integer, numeric_string, string
36 int chk_div (quad_t, quad_t);
37 int chk_minus (quad_t, quad_t, quad_t);
38 int chk_plus (quad_t, quad_t, quad_t);
39 int chk_times (quad_t, quad_t, quad_t);
40 void free_value (struct val *);
41 int is_zero_or_null (struct val *);
42 int isstring (struct val *);
43 int main (int, char **);
44 struct val *make_integer (quad_t);
45 struct val *make_str (const char *);
46 struct val *op_and (struct val *, struct val *);
47 struct val *op_colon (struct val *, struct val *);
48 struct val *op_div (struct val *, struct val *);
49 struct val *op_eq (struct val *, struct val *);
50 struct val *op_ge (struct val *, struct val *);
51 struct val *op_gt (struct val *, struct val *);
52 struct val *op_le (struct val *, struct val *);
53 struct val *op_lt (struct val *, struct val *);
54 struct val *op_minus (struct val *, struct val *);
55 struct val *op_ne (struct val *, struct val *);
56 struct val *op_or (struct val *, struct val *);
57 struct val *op_plus (struct val *, struct val *);
58 struct val *op_rem (struct val *, struct val *);
59 struct val *op_times (struct val *, struct val *);
60 quad_t to_integer (struct val *);
61 void to_string (struct val *);
62 int yyerror (const char *);
76 %left <val> '=' '>' '<' GE LE NE
78 %left <val> '*' '/' '%'
82 %type <val> start expr
86 start: expr { result = $$; }
89 | '(' expr ')' { $$ = $2; }
90 | expr '|' expr { $$ = op_or ($1, $3); }
91 | expr '&' expr { $$ = op_and ($1, $3); }
92 | expr '=' expr { $$ = op_eq ($1, $3); }
93 | expr '>' expr { $$ = op_gt ($1, $3); }
94 | expr '<' expr { $$ = op_lt ($1, $3); }
95 | expr GE expr { $$ = op_ge ($1, $3); }
96 | expr LE expr { $$ = op_le ($1, $3); }
97 | expr NE expr { $$ = op_ne ($1, $3); }
98 | expr '+' expr { $$ = op_plus ($1, $3); }
99 | expr '-' expr { $$ = op_minus ($1, $3); }
100 | expr '*' expr { $$ = op_times ($1, $3); }
101 | expr '/' expr { $$ = op_div ($1, $3); }
102 | expr '%' expr { $$ = op_rem ($1, $3); }
103 | expr ':' expr { $$ = op_colon ($1, $3); }
115 vp = (struct val *) malloc (sizeof (*vp));
117 errx (2, "malloc() failed");
133 vp = (struct val *) malloc (sizeof (*vp));
134 if (vp == NULL || ((vp->u.s = strdup (s)) == NULL)) {
135 errx (2, "malloc() failed");
138 for(i = 1, isint = isdigit(s[0]) || s[0] == '-';
139 isint && i < strlen(s);
147 vp->type = numeric_string;
159 if (vp->type == string || vp->type == numeric_string)
170 if (vp->type == integer)
173 if (vp->type == string)
176 /* vp->type == numeric_string, make it numeric */
178 i = strtoq(vp->u.s, (char**)NULL, 10);
180 errx (2, "overflow");
194 if (vp->type == string || vp->type == numeric_string)
197 tmp = malloc ((size_t)25);
199 errx (2, "malloc() failed");
202 sprintf (tmp, "%lld", (long long)vp->u.i);
212 /* only TRUE if this string is not a valid integer */
213 return (vp->type == string);
227 if (strlen (p) == 1) {
228 if (strchr ("|&=<>+-*/%:()", *p))
230 } else if (strlen (p) == 2 && p[1] == '=') {
232 case '>': return (GE);
233 case '<': return (LE);
234 case '!': return (NE);
238 yylval.val = make_str (p);
246 if (vp->type == integer) {
247 return (vp->u.i == 0);
249 return (*vp->u.s == 0 || (to_integer (vp) && vp->u.i == 0));
259 setlocale (LC_ALL, "");
265 if (result->type == integer)
266 printf ("%lld\n", (long long)result->u.i);
268 printf ("%s\n", result->u.s);
270 return (is_zero_or_null (result));
275 const char *s __unused;
277 errx (2, "syntax error");
285 if (is_zero_or_null (a)) {
298 if (is_zero_or_null (a) || is_zero_or_null (b)) {
301 return (make_integer ((quad_t)0));
314 if (isstring (a) || isstring (b)) {
317 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) == 0));
321 r = make_integer ((quad_t)(a->u.i == b->u.i));
335 if (isstring (a) || isstring (b)) {
338 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) > 0));
342 r = make_integer ((quad_t)(a->u.i > b->u.i));
356 if (isstring (a) || isstring (b)) {
359 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) < 0));
363 r = make_integer ((quad_t)(a->u.i < b->u.i));
377 if (isstring (a) || isstring (b)) {
380 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) >= 0));
384 r = make_integer ((quad_t)(a->u.i >= b->u.i));
398 if (isstring (a) || isstring (b)) {
401 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) <= 0));
405 r = make_integer ((quad_t)(a->u.i <= b->u.i));
419 if (isstring (a) || isstring (b)) {
422 r = make_integer ((quad_t)(strcoll (a->u.s, b->u.s) != 0));
426 r = make_integer ((quad_t)(a->u.i != b->u.i));
438 /* sum of two positive numbers must be positive */
439 if (a > 0 && b > 0 && r <= 0)
441 /* sum of two negative numbers must be negative */
442 if (a < 0 && b < 0 && r >= 0)
444 /* all other cases are OK */
454 if (!to_integer (a) || !to_integer (b)) {
455 errx (2, "non-numeric argument");
458 r = make_integer (/*(quad_t)*/(a->u.i + b->u.i));
459 if (chk_plus (a->u.i, b->u.i, r->u.i)) {
460 errx (2, "overflow");
471 /* special case subtraction of QUAD_MIN */
478 /* this is allowed for b != QUAD_MIN */
479 return chk_plus (a, -b, r);
488 if (!to_integer (a) || !to_integer (b)) {
489 errx (2, "non-numeric argument");
492 r = make_integer (/*(quad_t)*/(a->u.i - b->u.i));
493 if (chk_minus (a->u.i, b->u.i, r->u.i)) {
494 errx (2, "overflow");
505 /* special case: first operand is 0, no overflow possible */
508 /* cerify that result of division matches second operand */
520 if (!to_integer (a) || !to_integer (b)) {
521 errx (2, "non-numeric argument");
524 r = make_integer (/*(quad_t)*/(a->u.i * b->u.i));
525 if (chk_times (a->u.i, b->u.i, r->u.i)) {
526 errx (2, "overflow");
537 /* div by zero has been taken care of before */
538 /* only QUAD_MIN / -1 causes overflow */
539 if (a == QUAD_MIN && b == -1)
541 /* everything else is OK */
551 if (!to_integer (a) || !to_integer (b)) {
552 errx (2, "non-numeric argument");
556 errx (2, "division by zero");
559 r = make_integer (/*(quad_t)*/(a->u.i / b->u.i));
560 if (chk_div (a->u.i, b->u.i)) {
561 errx (2, "overflow");
574 if (!to_integer (a) || !to_integer (b)) {
575 errx (2, "non-numeric argument");
579 errx (2, "division by zero");
582 r = make_integer (/*(quad_t)*/(a->u.i % b->u.i));
583 /* chk_rem necessary ??? */
599 /* coerce to both arguments to strings */
603 /* compile regular expression */
604 if ((eval = regcomp (&rp, b->u.s, 0)) != 0) {
605 regerror (eval, &rp, errbuf, sizeof(errbuf));
606 errx (2, "%s", errbuf);
609 /* compare string against pattern */
610 /* remember that patterns are anchored to the beginning of the line */
611 if (regexec(&rp, a->u.s, (size_t)2, rm, 0) == 0 && rm[0].rm_so == 0) {
612 if (rm[1].rm_so >= 0) {
613 *(a->u.s + rm[1].rm_eo) = '\0';
614 v = make_str (a->u.s + rm[1].rm_so);
617 v = make_integer ((quad_t)(rm[0].rm_eo - rm[0].rm_so));
620 if (rp.re_nsub == 0) {
621 v = make_integer ((quad_t)0);
627 /* free arguments and pattern buffer */