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.6 2005/11/06 11:44:02 swildner 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); }
110 make_integer
(quad_t i
)
114 vp
= (struct val
*) malloc
(sizeof
(*vp
));
116 errx
(2, "malloc() failed");
125 make_str
(const char *s
)
131 vp
= (struct val
*) malloc
(sizeof
(*vp
));
132 if
(vp
== NULL ||
((vp
->u.s
= strdup
(s
)) == NULL
)) {
133 errx
(2, "malloc() failed");
136 for
(i
= 1, isint
= isdigit
(s
[0]) || s
[0] == '-';
137 isint
&& i
< strlen
(s
);
145 vp
->type
= numeric_string
;
154 free_value
(struct val
*vp
)
156 if
(vp
->type
== string || vp
->type
== numeric_string
)
162 to_integer
(struct val
*vp
)
166 if
(vp
->type
== integer
)
169 if
(vp
->type
== string)
172 /* vp->type == numeric_string, make it numeric */
174 i
= strtoll
(vp
->u.s
, NULL
, 10);
176 errx
(2, "overflow");
185 to_string
(struct val
*vp
)
189 if
(vp
->type
== string || vp
->type
== numeric_string
)
192 tmp
= malloc
((size_t)25);
194 errx
(2, "malloc() failed");
197 sprintf
(tmp
, "%lld", (long long)vp
->u.i
);
204 isstring
(struct val
*vp
)
206 /* only TRUE if this string is not a valid integer */
207 return
(vp
->type
== string);
221 if
(strlen
(p
) == 1) {
222 if
(strchr
("|&=<>+-*/%:()", *p
))
224 } else if
(strlen
(p
) == 2 && p
[1] == '=') {
226 case
'>': return
(GE
);
227 case
'<': return
(LE
);
228 case
'!': return
(NE
);
232 yylval.val
= make_str
(p
);
237 is_zero_or_null
(struct val
*vp
)
239 if
(vp
->type
== integer
) {
240 return
(vp
->u.i
== 0);
242 return
(*vp
->u.s
== 0 ||
(to_integer
(vp
) && vp
->u.i
== 0));
251 "usage: expr expression\n");
256 main
(int argc
, char **argv
)
258 setlocale
(LC_ALL
, "");
260 if
(argc
> 1 && strcmp
(argv
[1], "--"))
269 if
(result
->type
== integer
)
270 printf
("%lld\n", (long long)result
->u.i
);
272 printf
("%s\n", result
->u.s
);
274 return
(is_zero_or_null
(result
));
278 yyerror(const char *s __unused
)
280 errx
(2, "syntax error");
285 op_or
(struct val
*a
, struct val
*b
)
287 if
(is_zero_or_null
(a
)) {
297 op_and
(struct val
*a
, struct val
*b
)
299 if
(is_zero_or_null
(a
) || is_zero_or_null
(b
)) {
302 return
(make_integer
((quad_t
)0));
310 op_eq
(struct val
*a
, struct val
*b
)
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
));
330 op_gt
(struct val
*a
, struct val
*b
)
334 if
(isstring
(a
) || isstring
(b
)) {
337 r
= make_integer
((quad_t
)(strcoll
(a
->u.s
, b
->u.s
) > 0));
341 r
= make_integer
((quad_t
)(a
->u.i
> b
->u.i
));
350 op_lt
(struct val
*a
, struct val
*b
)
354 if
(isstring
(a
) || isstring
(b
)) {
357 r
= make_integer
((quad_t
)(strcoll
(a
->u.s
, b
->u.s
) < 0));
361 r
= make_integer
((quad_t
)(a
->u.i
< b
->u.i
));
370 op_ge
(struct val
*a
, struct val
*b
)
374 if
(isstring
(a
) || isstring
(b
)) {
377 r
= make_integer
((quad_t
)(strcoll
(a
->u.s
, b
->u.s
) >= 0));
381 r
= make_integer
((quad_t
)(a
->u.i
>= b
->u.i
));
390 op_le
(struct val
*a
, struct val
*b
)
394 if
(isstring
(a
) || isstring
(b
)) {
397 r
= make_integer
((quad_t
)(strcoll
(a
->u.s
, b
->u.s
) <= 0));
401 r
= make_integer
((quad_t
)(a
->u.i
<= b
->u.i
));
410 op_ne
(struct val
*a
, struct val
*b
)
414 if
(isstring
(a
) || isstring
(b
)) {
417 r
= make_integer
((quad_t
)(strcoll
(a
->u.s
, b
->u.s
) != 0));
421 r
= make_integer
((quad_t
)(a
->u.i
!= b
->u.i
));
430 chk_plus
(quad_t a
, quad_t b
, quad_t r
)
432 /* sum of two positive numbers must be positive */
433 if
(a
> 0 && b
> 0 && r
<= 0)
435 /* sum of two negative numbers must be negative */
436 if
(a
< 0 && b
< 0 && r
>= 0)
438 /* all other cases are OK */
443 op_plus
(struct val
*a
, struct val
*b
)
447 if
(!to_integer
(a
) ||
!to_integer
(b
)) {
448 errx
(2, "non-numeric argument");
451 r
= make_integer
(/*(quad_t)*/(a
->u.i
+ b
->u.i
));
452 if
(chk_plus
(a
->u.i
, b
->u.i
, r
->u.i
)) {
453 errx
(2, "overflow");
461 chk_minus
(quad_t a
, quad_t b
, quad_t r
)
463 /* special case subtraction of QUAD_MIN */
470 /* this is allowed for b != QUAD_MIN */
471 return chk_plus
(a
, -b
, r
);
475 op_minus
(struct val
*a
, struct val
*b
)
479 if
(!to_integer
(a
) ||
!to_integer
(b
)) {
480 errx
(2, "non-numeric argument");
483 r
= make_integer
(/*(quad_t)*/(a
->u.i
- b
->u.i
));
484 if
(chk_minus
(a
->u.i
, b
->u.i
, r
->u.i
)) {
485 errx
(2, "overflow");
493 chk_times
(quad_t a
, quad_t b
, quad_t r
)
495 /* special case: first operand is 0, no overflow possible */
498 /* cerify that result of division matches second operand */
505 op_times
(struct val
*a
, struct val
*b
)
509 if
(!to_integer
(a
) ||
!to_integer
(b
)) {
510 errx
(2, "non-numeric argument");
513 r
= make_integer
(/*(quad_t)*/(a
->u.i
* b
->u.i
));
514 if
(chk_times
(a
->u.i
, b
->u.i
, r
->u.i
)) {
515 errx
(2, "overflow");
523 chk_div
(quad_t a
, quad_t b
)
525 /* div by zero has been taken care of before */
526 /* only QUAD_MIN / -1 causes overflow */
527 if
(a
== QUAD_MIN
&& b
== -1)
529 /* everything else is OK */
534 op_div
(struct val
*a
, struct val
*b
)
538 if
(!to_integer
(a
) ||
!to_integer
(b
)) {
539 errx
(2, "non-numeric argument");
543 errx
(2, "division by zero");
546 r
= make_integer
(/*(quad_t)*/(a
->u.i
/ b
->u.i
));
547 if
(chk_div
(a
->u.i
, b
->u.i
)) {
548 errx
(2, "overflow");
556 op_rem
(struct val
*a
, struct val
*b
)
560 if
(!to_integer
(a
) ||
!to_integer
(b
)) {
561 errx
(2, "non-numeric argument");
565 errx
(2, "division by zero");
568 r
= make_integer
(/*(quad_t)*/(a
->u.i % b
->u.i
));
569 /* chk_rem necessary ??? */
576 op_colon
(struct val
*a
, struct val
*b
)
584 /* coerce to both arguments to strings */
588 /* compile regular expression */
589 if
((eval
= regcomp
(&rp
, b
->u.s
, 0)) != 0) {
590 regerror
(eval
, &rp
, errbuf
, sizeof
(errbuf
));
591 errx
(2, "%s", errbuf
);
594 /* compare string against pattern */
595 /* remember that patterns are anchored to the beginning of the line */
596 if
(regexec
(&rp
, a
->u.s
, (size_t)2, rm
, 0) == 0 && rm
[0].rm_so
== 0) {
597 if
(rm
[1].rm_so
>= 0) {
598 *(a
->u.s
+ rm
[1].rm_eo
) = '\0';
599 v
= make_str
(a
->u.s
+ rm
[1].rm_so
);
602 v
= make_integer
((quad_t
)(rm
[0].rm_eo
- rm
[0].rm_so
));
605 if
(rp.re_nsub
== 0) {
606 v
= make_integer
((quad_t
)0);
612 /* free arguments and pattern buffer */