helper: add get_last_statement_from_expression_stmt() [build fix]
[smatch.git] / smatch_math.c
blob6aec010043306014aed270c2e61375151c19216a
1 /*
2 * Copyright (C) 2010 Dan Carpenter.
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
18 #include "symbol.h"
19 #include "smatch.h"
20 #include "smatch_slist.h"
21 #include "smatch_extra.h"
23 static struct range_list *_get_rl(struct expression *expr, int implied, int *recurse_cnt);
24 static struct range_list *handle_variable(struct expression *expr, int implied, int *recurse_cnt);
25 static struct range_list *(*custom_handle_variable)(struct expression *expr);
27 static int get_implied_value_internal(struct expression *expr, sval_t *sval, int *recurse_cnt);
28 static int get_absolute_rl_internal(struct expression *expr, struct range_list **rl, int *recurse_cnt);
30 static sval_t zero = {.type = &int_ctype, {.value = 0} };
31 static sval_t one = {.type = &int_ctype, {.value = 1} };
33 struct range_list *rl_zero(void)
35 static struct range_list *zero_perm;
37 if (!zero_perm)
38 zero_perm = clone_rl_permanent(alloc_rl(zero, zero));
39 return zero_perm;
42 struct range_list *rl_one(void)
44 static struct range_list *one_perm;
46 if (!one_perm)
47 one_perm = clone_rl_permanent(alloc_rl(one, one));
49 return one_perm;
52 enum {
53 RL_EXACT,
54 RL_HARD,
55 RL_FUZZY,
56 RL_IMPLIED,
57 RL_ABSOLUTE,
58 RL_REAL_ABSOLUTE,
61 static struct range_list *last_stmt_rl(struct statement *stmt, int implied, int *recurse_cnt)
63 struct expression *expr;
65 if (!stmt)
66 return NULL;
68 stmt = last_ptr_list((struct ptr_list *)stmt->stmts);
69 if (stmt->type == STMT_LABEL) {
70 if (stmt->label_statement &&
71 stmt->label_statement->type == STMT_EXPRESSION)
72 expr = stmt->label_statement->expression;
73 else
74 return NULL;
75 } else if (stmt->type == STMT_EXPRESSION) {
76 expr = stmt->expression;
77 } else {
78 return NULL;
80 return _get_rl(expr, implied, recurse_cnt);
83 static struct range_list *handle_expression_statement_rl(struct expression *expr, int implied, int *recurse_cnt)
85 return last_stmt_rl(get_expression_statement(expr), implied, recurse_cnt);
88 static struct range_list *handle_ampersand_rl(struct expression *expr, int implied, int *recurse_cnt)
90 struct range_list *rl;
92 if (implied == RL_EXACT || implied == RL_HARD)
93 return NULL;
94 if (get_address_rl(expr, &rl))
95 return rl;
96 return alloc_rl(valid_ptr_min_sval, valid_ptr_max_sval);
99 static struct range_list *handle_negate_rl(struct expression *expr, int implied, int *recurse_cnt)
101 if (known_condition_true(expr->unop))
102 return rl_zero();
103 if (known_condition_false(expr->unop))
104 return rl_one();
106 if (implied == RL_EXACT)
107 return NULL;
109 if (implied_condition_true(expr->unop))
110 return rl_zero();
111 if (implied_condition_false(expr->unop))
112 return rl_one();
113 return alloc_rl(zero, one);
116 static struct range_list *handle_bitwise_negate(struct expression *expr, int implied, int *recurse_cnt)
118 struct range_list *rl;
119 sval_t sval;
121 rl = _get_rl(expr->unop, implied, recurse_cnt);
122 if (!rl_to_sval(rl, &sval))
123 return NULL;
124 sval = sval_preop(sval, '~');
125 sval_cast(get_type(expr->unop), sval);
126 return alloc_rl(sval, sval);
129 static struct range_list *handle_minus_preop(struct expression *expr, int implied, int *recurse_cnt)
131 struct range_list *rl;
132 sval_t min, max;
134 rl = _get_rl(expr->unop, implied, recurse_cnt);
135 min = sval_preop(rl_max(rl), '-');
136 max = sval_preop(rl_min(rl), '-');
137 return alloc_rl(min, max);
140 static struct range_list *handle_preop_rl(struct expression *expr, int implied, int *recurse_cnt)
142 switch (expr->op) {
143 case '&':
144 return handle_ampersand_rl(expr, implied, recurse_cnt);
145 case '!':
146 return handle_negate_rl(expr, implied, recurse_cnt);
147 case '~':
148 return handle_bitwise_negate(expr, implied, recurse_cnt);
149 case '-':
150 return handle_minus_preop(expr, implied, recurse_cnt);
151 case '*':
152 return handle_variable(expr, implied, recurse_cnt);
153 case '(':
154 return handle_expression_statement_rl(expr, implied, recurse_cnt);
155 default:
156 return NULL;
160 static struct range_list *handle_divide_rl(struct expression *expr, int implied, int *recurse_cnt)
162 struct range_list *left_rl, *right_rl;
163 struct symbol *type;
165 type = get_type(expr);
167 left_rl = _get_rl(expr->left, implied, recurse_cnt);
168 left_rl = cast_rl(type, left_rl);
169 right_rl = _get_rl(expr->right, implied, recurse_cnt);
170 right_rl = cast_rl(type, right_rl);
172 if (!left_rl || !right_rl)
173 return NULL;
175 if (implied != RL_REAL_ABSOLUTE) {
176 if (is_whole_rl(left_rl) || is_whole_rl(right_rl))
177 return NULL;
180 return rl_binop(left_rl, '/', right_rl);
183 static int handle_offset_subtraction(struct expression *expr)
185 struct expression *left, *right;
186 struct symbol *left_sym, *right_sym;
187 struct symbol *type;
188 int left_offset, right_offset;
190 type = get_type(expr);
191 if (!type || type->type != SYM_PTR)
192 return -1;
193 type = get_real_base_type(type);
194 if (!type || (type_bits(type) != 8 && (type != &void_ctype)))
195 return -1;
197 left = strip_expr(expr->left);
198 right = strip_expr(expr->right);
200 if (left->type != EXPR_PREOP || left->op != '&')
201 return -1;
202 left = strip_expr(left->unop);
204 left_sym = expr_to_sym(left);
205 right_sym = expr_to_sym(right);
206 if (!left_sym || left_sym != right_sym)
207 return -1;
209 left_offset = get_member_offset_from_deref(left);
210 if (right->type == EXPR_SYMBOL)
211 right_offset = 0;
212 else {
213 if (right->type != EXPR_PREOP || right->op != '&')
214 return -1;
215 right = strip_expr(right->unop);
216 right_offset = get_member_offset_from_deref(right);
218 if (left_offset < 0 || right_offset < 0)
219 return -1;
221 return left_offset - right_offset;
224 static struct range_list *handle_subtract_rl(struct expression *expr, int implied, int *recurse_cnt)
226 struct symbol *type;
227 struct range_list *left_orig, *right_orig;
228 struct range_list *left_rl, *right_rl;
229 sval_t max, min, tmp;
230 int comparison;
231 int offset;
233 type = get_type(expr);
235 offset = handle_offset_subtraction(expr);
236 if (offset >= 0) {
237 tmp.type = type;
238 tmp.value = offset;
240 return alloc_rl(tmp, tmp);
243 comparison = get_comparison(expr->left, expr->right);
245 left_orig = _get_rl(expr->left, implied, recurse_cnt);
246 left_rl = cast_rl(type, left_orig);
247 right_orig = _get_rl(expr->right, implied, recurse_cnt);
248 right_rl = cast_rl(type, right_orig);
250 if ((!left_rl || !right_rl) &&
251 (implied == RL_EXACT || implied == RL_HARD || implied == RL_FUZZY))
252 return NULL;
254 if (!left_rl)
255 left_rl = alloc_whole_rl(type);
256 if (!right_rl)
257 right_rl = alloc_whole_rl(type);
259 /* negative values complicate everything fix this later */
260 if (sval_is_negative(rl_min(right_rl)))
261 return NULL;
262 max = rl_max(left_rl);
263 min = sval_type_min(type);
265 switch (comparison) {
266 case '>':
267 case SPECIAL_UNSIGNED_GT:
268 min = sval_type_val(type, 1);
269 max = rl_max(left_rl);
270 break;
271 case SPECIAL_GTE:
272 case SPECIAL_UNSIGNED_GTE:
273 min = sval_type_val(type, 0);
274 max = rl_max(left_rl);
275 break;
276 case SPECIAL_EQUAL:
277 min = sval_type_val(type, 0);
278 max = sval_type_val(type, 0);
279 break;
280 case '<':
281 case SPECIAL_UNSIGNED_LT:
282 max = sval_type_val(type, -1);
283 break;
284 case SPECIAL_LTE:
285 case SPECIAL_UNSIGNED_LTE:
286 max = sval_type_val(type, 0);
287 break;
288 default:
289 if (!left_orig || !right_orig)
290 return NULL;
291 return rl_binop(left_rl, '-', right_rl);
294 if (!sval_binop_overflows(rl_min(left_rl), '-', rl_max(right_rl))) {
295 tmp = sval_binop(rl_min(left_rl), '-', rl_max(right_rl));
296 if (sval_cmp(tmp, min) > 0)
297 min = tmp;
300 if (!sval_is_max(rl_max(left_rl))) {
301 tmp = sval_binop(rl_max(left_rl), '-', rl_min(right_rl));
302 if (sval_cmp(tmp, max) < 0)
303 max = tmp;
306 if (sval_is_min(min) && sval_is_max(max))
307 return NULL;
309 return cast_rl(type, alloc_rl(min, max));
312 static struct range_list *handle_mod_rl(struct expression *expr, int implied, int *recurse_cnt)
314 struct range_list *rl;
315 sval_t left, right, sval;
317 if (implied == RL_EXACT) {
318 if (!get_implied_value(expr->right, &right))
319 return NULL;
320 if (!get_implied_value(expr->left, &left))
321 return NULL;
322 sval = sval_binop(left, '%', right);
323 return alloc_rl(sval, sval);
325 /* if we can't figure out the right side it's probably hopeless */
326 if (!get_implied_value_internal(expr->right, &right, recurse_cnt))
327 return NULL;
329 right = sval_cast(get_type(expr), right);
330 right.value--;
332 rl = _get_rl(expr->left, implied, recurse_cnt);
333 if (rl && rl_max(rl).uvalue < right.uvalue)
334 right.uvalue = rl_max(rl).uvalue;
336 return alloc_rl(sval_cast(right.type, zero), right);
339 static sval_t sval_lowest_set_bit(sval_t sval)
341 int i;
342 int found = 0;
344 for (i = 0; i < 64; i++) {
345 if (sval.uvalue & 1ULL << i) {
346 if (!found++)
347 continue;
348 sval.uvalue &= ~(1ULL << i);
351 return sval;
354 static struct range_list *handle_bitwise_AND(struct expression *expr, int implied, int *recurse_cnt)
356 struct symbol *type;
357 struct range_list *left_rl, *right_rl;
358 sval_t known;
359 int new_recurse;
361 if (implied != RL_IMPLIED && implied != RL_ABSOLUTE && implied != RL_REAL_ABSOLUTE)
362 return NULL;
364 type = get_type(expr);
366 if (get_implied_value_internal(expr->left, &known, recurse_cnt)) {
367 sval_t min;
369 min = sval_lowest_set_bit(known);
370 left_rl = alloc_rl(min, known);
371 left_rl = cast_rl(type, left_rl);
372 add_range(&left_rl, sval_type_val(type, 0), sval_type_val(type, 0));
373 } else {
374 left_rl = _get_rl(expr->left, implied, recurse_cnt);
375 if (left_rl) {
376 left_rl = cast_rl(type, left_rl);
377 left_rl = alloc_rl(sval_type_val(type, 0), rl_max(left_rl));
378 } else {
379 if (implied == RL_HARD)
380 return NULL;
381 left_rl = alloc_whole_rl(type);
385 new_recurse = *recurse_cnt;
386 if (*recurse_cnt >= 200)
387 new_recurse = 100; /* Let's try super hard to get the mask */
388 if (get_implied_value_internal(expr->right, &known, &new_recurse)) {
389 sval_t min, left_max, mod;
391 *recurse_cnt = new_recurse;
393 min = sval_lowest_set_bit(known);
394 right_rl = alloc_rl(min, known);
395 right_rl = cast_rl(type, right_rl);
396 add_range(&right_rl, sval_type_val(type, 0), sval_type_val(type, 0));
398 if (min.value != 0) {
399 left_max = rl_max(left_rl);
400 mod = sval_binop(left_max, '%', min);
401 if (mod.value) {
402 left_max = sval_binop(left_max, '-', mod);
403 left_max.value++;
404 if (left_max.value > 0 && sval_cmp(left_max, rl_max(left_rl)) < 0)
405 left_rl = remove_range(left_rl, left_max, rl_max(left_rl));
408 } else {
409 right_rl = _get_rl(expr->right, implied, recurse_cnt);
410 if (right_rl) {
411 right_rl = cast_rl(type, right_rl);
412 right_rl = alloc_rl(sval_type_val(type, 0), rl_max(right_rl));
413 } else {
414 if (implied == RL_HARD)
415 return NULL;
416 right_rl = alloc_whole_rl(type);
420 return rl_intersection(left_rl, right_rl);
423 static struct range_list *use_rl_binop(struct expression *expr, int implied, int *recurse_cnt)
425 struct symbol *type;
426 struct range_list *left_rl, *right_rl;
428 if (implied != RL_IMPLIED && implied != RL_ABSOLUTE && implied != RL_REAL_ABSOLUTE)
429 return NULL;
431 type = get_type(expr);
433 get_absolute_rl_internal(expr->left, &left_rl, recurse_cnt);
434 get_absolute_rl_internal(expr->right, &right_rl, recurse_cnt);
435 left_rl = cast_rl(type, left_rl);
436 right_rl = cast_rl(type, right_rl);
437 if (!left_rl || !right_rl)
438 return NULL;
440 return rl_binop(left_rl, expr->op, right_rl);
443 static struct range_list *handle_right_shift(struct expression *expr, int implied, int *recurse_cnt)
445 struct range_list *left_rl;
446 sval_t right;
447 sval_t min, max;
449 if (implied == RL_EXACT || implied == RL_HARD)
450 return NULL;
452 left_rl = _get_rl(expr->left, implied, recurse_cnt);
453 if (left_rl) {
454 max = rl_max(left_rl);
455 min = rl_min(left_rl);
456 } else {
457 if (implied == RL_FUZZY)
458 return NULL;
459 max = sval_type_max(get_type(expr->left));
460 min = sval_type_val(get_type(expr->left), 0);
463 if (get_implied_value_internal(expr->right, &right, recurse_cnt)) {
464 min = sval_binop(min, SPECIAL_RIGHTSHIFT, right);
465 max = sval_binop(max, SPECIAL_RIGHTSHIFT, right);
466 } else if (!sval_is_negative(min)) {
467 min.value = 0;
468 max = sval_type_max(max.type);
469 } else {
470 return NULL;
473 return alloc_rl(min, max);
476 static struct range_list *handle_left_shift(struct expression *expr, int implied, int *recurse_cnt)
478 struct range_list *left_rl, *res;
479 sval_t right;
480 sval_t min, max;
481 int add_zero = 0;
483 if (implied == RL_EXACT || implied == RL_HARD)
484 return NULL;
485 /* this is hopeless without the right side */
486 if (!get_implied_value_internal(expr->right, &right, recurse_cnt))
487 return NULL;
488 left_rl = _get_rl(expr->left, implied, recurse_cnt);
489 if (left_rl) {
490 max = rl_max(left_rl);
491 min = rl_min(left_rl);
492 if (min.value == 0) {
493 min.value = 1;
494 add_zero = 1;
496 } else {
497 if (implied == RL_FUZZY)
498 return NULL;
499 max = sval_type_max(get_type(expr->left));
500 min = sval_type_val(get_type(expr->left), 1);
501 add_zero = 1;
504 max = sval_binop(max, SPECIAL_LEFTSHIFT, right);
505 min = sval_binop(min, SPECIAL_LEFTSHIFT, right);
506 res = alloc_rl(min, max);
507 if (add_zero)
508 res = rl_union(res, rl_zero());
509 return res;
512 static struct range_list *handle_known_binop(struct expression *expr)
514 sval_t left, right;
516 if (!get_value(expr->left, &left))
517 return NULL;
518 if (!get_value(expr->right, &right))
519 return NULL;
520 left = sval_binop(left, expr->op, right);
521 return alloc_rl(left, left);
524 static int has_actual_ranges(struct range_list *rl)
526 struct data_range *tmp;
528 FOR_EACH_PTR(rl, tmp) {
529 if (sval_cmp(tmp->min, tmp->max) != 0)
530 return 1;
531 } END_FOR_EACH_PTR(tmp);
532 return 0;
535 static struct range_list *handle_implied_binop(struct range_list *left_rl, int op, struct range_list *right_rl)
537 struct range_list *res_rl;
538 struct data_range *left_drange, *right_drange;
539 sval_t res;
541 if (!left_rl || !right_rl)
542 return NULL;
543 if (has_actual_ranges(left_rl))
544 return NULL;
545 if (has_actual_ranges(right_rl))
546 return NULL;
548 if (ptr_list_size((struct ptr_list *)left_rl) * ptr_list_size((struct ptr_list *)right_rl) > 20)
549 return NULL;
551 res_rl = NULL;
553 FOR_EACH_PTR(left_rl, left_drange) {
554 FOR_EACH_PTR(right_rl, right_drange) {
555 if ((op == '%' || op == '/') &&
556 right_drange->min.value == 0)
557 return NULL;
558 res = sval_binop(left_drange->min, op, right_drange->min);
559 add_range(&res_rl, res, res);
560 } END_FOR_EACH_PTR(right_drange);
561 } END_FOR_EACH_PTR(left_drange);
563 return res_rl;
566 static struct range_list *handle_binop_rl(struct expression *expr, int implied, int *recurse_cnt)
568 struct smatch_state *state;
569 struct symbol *type;
570 struct range_list *left_rl, *right_rl, *rl;
571 sval_t min, max;
573 rl = handle_known_binop(expr);
574 if (rl)
575 return rl;
576 if (implied == RL_EXACT)
577 return NULL;
579 if (custom_handle_variable) {
580 rl = custom_handle_variable(expr);
581 if (rl)
582 return rl;
585 state = get_extra_state(expr);
586 if (state && !is_whole_rl(estate_rl(state))) {
587 if (implied != RL_HARD || estate_has_hard_max(state))
588 return clone_rl(estate_rl(state));
591 type = get_type(expr);
592 left_rl = _get_rl(expr->left, implied, recurse_cnt);
593 left_rl = cast_rl(type, left_rl);
594 right_rl = _get_rl(expr->right, implied, recurse_cnt);
595 right_rl = cast_rl(type, right_rl);
597 if (!left_rl && !right_rl)
598 return NULL;
600 rl = handle_implied_binop(left_rl, expr->op, right_rl);
601 if (rl)
602 return rl;
604 switch (expr->op) {
605 case '%':
606 return handle_mod_rl(expr, implied, recurse_cnt);
607 case '&':
608 return handle_bitwise_AND(expr, implied, recurse_cnt);
609 case '|':
610 case '^':
611 return use_rl_binop(expr, implied, recurse_cnt);
612 case SPECIAL_RIGHTSHIFT:
613 return handle_right_shift(expr, implied, recurse_cnt);
614 case SPECIAL_LEFTSHIFT:
615 return handle_left_shift(expr, implied, recurse_cnt);
616 case '-':
617 return handle_subtract_rl(expr, implied, recurse_cnt);
618 case '/':
619 return handle_divide_rl(expr, implied, recurse_cnt);
622 if (!left_rl || !right_rl)
623 return NULL;
625 if (sval_binop_overflows(rl_min(left_rl), expr->op, rl_min(right_rl)))
626 return NULL;
627 if (sval_binop_overflows(rl_max(left_rl), expr->op, rl_max(right_rl)))
628 return NULL;
630 min = sval_binop(rl_min(left_rl), expr->op, rl_min(right_rl));
631 max = sval_binop(rl_max(left_rl), expr->op, rl_max(right_rl));
633 return alloc_rl(min, max);
636 static int do_comparison(struct expression *expr)
638 struct range_list *left_ranges = NULL;
639 struct range_list *right_ranges = NULL;
640 int poss_true, poss_false;
641 struct symbol *type;
643 type = get_type(expr);
644 get_absolute_rl(expr->left, &left_ranges);
645 get_absolute_rl(expr->right, &right_ranges);
647 left_ranges = cast_rl(type, left_ranges);
648 right_ranges = cast_rl(type, right_ranges);
650 poss_true = possibly_true_rl(left_ranges, expr->op, right_ranges);
651 poss_false = possibly_false_rl(left_ranges, expr->op, right_ranges);
653 if (!poss_true && !poss_false)
654 return 0x0;
655 if (poss_true && !poss_false)
656 return 0x1;
657 if (!poss_true && poss_false)
658 return 0x2;
659 return 0x3;
662 static struct range_list *handle_comparison_rl(struct expression *expr, int implied, int *recurse_cnt)
664 sval_t left, right;
665 int res;
667 if (expr->op == SPECIAL_EQUAL && expr->left->type == EXPR_TYPE) {
668 struct symbol *left, *right;
670 left = get_real_base_type(expr->left->symbol);
671 right = get_real_base_type(expr->left->symbol);
672 if (left == right)
673 return rl_one();
674 return rl_zero();
677 if (get_value(expr->left, &left) && get_value(expr->right, &right)) {
678 struct data_range tmp_left, tmp_right;
680 tmp_left.min = left;
681 tmp_left.max = left;
682 tmp_right.min = right;
683 tmp_right.max = right;
684 if (true_comparison_range(&tmp_left, expr->op, &tmp_right))
685 return rl_one();
686 return rl_zero();
689 if (implied == RL_EXACT)
690 return NULL;
692 res = do_comparison(expr);
693 if (res == 1)
694 return rl_one();
695 if (res == 2)
696 return rl_zero();
698 return alloc_rl(zero, one);
701 static struct range_list *handle_logical_rl(struct expression *expr, int implied, int *recurse_cnt)
703 sval_t left, right;
704 int left_known = 0;
705 int right_known = 0;
707 if (implied == RL_EXACT) {
708 if (get_value(expr->left, &left))
709 left_known = 1;
710 if (get_value(expr->right, &right))
711 right_known = 1;
712 } else {
713 if (get_implied_value_internal(expr->left, &left, recurse_cnt))
714 left_known = 1;
715 if (get_implied_value_internal(expr->right, &right, recurse_cnt))
716 right_known = 1;
719 switch (expr->op) {
720 case SPECIAL_LOGICAL_OR:
721 if (left_known && left.value)
722 return rl_one();
723 if (right_known && right.value)
724 return rl_one();
725 if (left_known && right_known)
726 return rl_zero();
727 break;
728 case SPECIAL_LOGICAL_AND:
729 if (left_known && right_known) {
730 if (left.value && right.value)
731 return rl_one();
732 return rl_zero();
734 break;
735 default:
736 return NULL;
739 if (implied == RL_EXACT)
740 return NULL;
742 return alloc_rl(zero, one);
745 static struct range_list *handle_conditional_rl(struct expression *expr, int implied, int *recurse_cnt)
747 struct range_list *true_rl, *false_rl;
748 struct symbol *type;
749 int final_pass_orig = final_pass;
751 if (known_condition_true(expr->conditional))
752 return _get_rl(expr->cond_true, implied, recurse_cnt);
753 if (known_condition_false(expr->conditional))
754 return _get_rl(expr->cond_false, implied, recurse_cnt);
756 if (implied == RL_EXACT)
757 return NULL;
759 if (implied_condition_true(expr->conditional))
760 return _get_rl(expr->cond_true, implied, recurse_cnt);
761 if (implied_condition_false(expr->conditional))
762 return _get_rl(expr->cond_false, implied, recurse_cnt);
765 /* this becomes a problem with deeply nested conditional statements */
766 if (low_on_memory())
767 return NULL;
769 type = get_type(expr);
771 __push_fake_cur_stree();
772 final_pass = 0;
773 __split_whole_condition(expr->conditional);
774 true_rl = _get_rl(expr->cond_true, implied, recurse_cnt);
775 __push_true_states();
776 __use_false_states();
777 false_rl = _get_rl(expr->cond_false, implied, recurse_cnt);
778 __merge_true_states();
779 __free_fake_cur_stree();
780 final_pass = final_pass_orig;
782 if (!true_rl || !false_rl)
783 return NULL;
784 true_rl = cast_rl(type, true_rl);
785 false_rl = cast_rl(type, false_rl);
787 return rl_union(true_rl, false_rl);
790 static int get_fuzzy_max_helper(struct expression *expr, sval_t *max)
792 struct smatch_state *state;
793 sval_t sval;
795 if (get_hard_max(expr, &sval)) {
796 *max = sval;
797 return 1;
800 state = get_extra_state(expr);
801 if (!state || !estate_has_fuzzy_max(state))
802 return 0;
803 *max = sval_cast(get_type(expr), estate_get_fuzzy_max(state));
804 return 1;
807 static int get_fuzzy_min_helper(struct expression *expr, sval_t *min)
809 struct smatch_state *state;
810 sval_t sval;
812 state = get_extra_state(expr);
813 if (!state || !estate_rl(state))
814 return 0;
816 sval = estate_min(state);
817 if (sval_is_negative(sval) && sval_is_min(sval))
818 return 0;
820 if (sval_is_max(sval))
821 return 0;
823 *min = sval_cast(get_type(expr), sval);
824 return 1;
827 int get_const_value(struct expression *expr, sval_t *sval)
829 struct symbol *sym;
830 sval_t right;
832 if (expr->type != EXPR_SYMBOL || !expr->symbol)
833 return 0;
834 sym = expr->symbol;
835 if (!(sym->ctype.modifiers & MOD_CONST))
836 return 0;
837 if (get_value(sym->initializer, &right)) {
838 *sval = sval_cast(get_type(expr), right);
839 return 1;
841 return 0;
844 struct range_list *var_to_absolute_rl(struct expression *expr)
846 struct smatch_state *state;
847 struct range_list *rl;
849 state = get_extra_state(expr);
850 if (!state || is_whole_rl(estate_rl(state))) {
851 state = get_real_absolute_state(expr);
852 if (state && state->data && !estate_is_whole(state))
853 return clone_rl(estate_rl(state));
854 if (get_local_rl(expr, &rl) && !is_whole_rl(rl))
855 return rl;
856 if (get_db_type_rl(expr, &rl) && !is_whole_rl(rl))
857 return rl;
858 return alloc_whole_rl(get_type(expr));
860 /* err on the side of saying things are possible */
861 if (!estate_rl(state))
862 return alloc_whole_rl(get_type(expr));
863 return clone_rl(estate_rl(state));
866 static struct range_list *handle_variable(struct expression *expr, int implied, int *recurse_cnt)
868 struct smatch_state *state;
869 struct range_list *rl;
870 sval_t sval, min, max;
871 struct symbol *type;
873 if (get_const_value(expr, &sval))
874 return alloc_rl(sval, sval);
876 if (custom_handle_variable) {
877 rl = custom_handle_variable(expr);
878 if (!rl)
879 return var_to_absolute_rl(expr);
880 return rl;
883 if (implied == RL_EXACT)
884 return NULL;
886 type = get_type(expr);
887 if (type && type->type == SYM_FN)
888 return alloc_rl(fn_ptr_min, fn_ptr_max);
890 switch (implied) {
891 case RL_HARD:
892 case RL_IMPLIED:
893 case RL_ABSOLUTE:
894 state = get_extra_state(expr);
895 if (!state || !state->data) {
896 if (implied == RL_HARD)
897 return NULL;
898 if (get_local_rl(expr, &rl))
899 return rl;
900 if (get_db_type_rl(expr, &rl))
901 return rl;
902 return NULL;
904 if (implied == RL_HARD && !estate_has_hard_max(state))
905 return NULL;
906 return clone_rl(estate_rl(state));
907 case RL_REAL_ABSOLUTE: {
908 struct smatch_state *abs_state;
910 state = get_extra_state(expr);
911 abs_state = get_real_absolute_state(expr);
913 if (estate_rl(state) && estate_rl(abs_state)) {
914 return clone_rl(rl_intersection(estate_rl(state),
915 estate_rl(abs_state)));
916 } else if (estate_rl(state)) {
917 return clone_rl(estate_rl(state));
918 } else if (estate_is_empty(state)) {
920 * FIXME: we don't handle empty extra states correctly.
922 * The real abs rl is supposed to be filtered by the
923 * extra state if there is one. We don't bother keeping
924 * the abs state in sync all the time because we know it
925 * will be filtered later.
927 * It's not totally obvious to me how they should be
928 * handled. Perhaps we should take the whole rl and
929 * filter by the imaginary states. Perhaps we should
930 * just go with the empty state.
932 * Anyway what we currently do is return NULL here and
933 * that gets translated into the whole range in
934 * get_real_absolute_rl().
937 return NULL;
938 } else if (estate_rl(abs_state)) {
939 return clone_rl(estate_rl(abs_state));
942 if (get_local_rl(expr, &rl))
943 return rl;
944 if (get_db_type_rl(expr, &rl))
945 return rl;
946 return NULL;
948 case RL_FUZZY:
949 if (!get_fuzzy_min_helper(expr, &min))
950 min = sval_type_min(get_type(expr));
951 if (!get_fuzzy_max_helper(expr, &max))
952 return NULL;
953 /* fuzzy ranges are often inverted */
954 if (sval_cmp(min, max) > 0) {
955 sval = min;
956 min = max;
957 max = sval;
959 return alloc_rl(min, max);
961 return NULL;
964 static sval_t handle_sizeof(struct expression *expr)
966 struct symbol *sym;
967 sval_t ret;
969 ret = sval_blank(expr);
970 sym = expr->cast_type;
971 if (!sym) {
972 sym = evaluate_expression(expr->cast_expression);
973 if (!sym) {
974 __silence_warnings_for_stmt = true;
975 sym = &int_ctype;
977 #if 0
979 * Expressions of restricted types will possibly get
980 * promoted - check that here. I'm not sure how this works,
981 * the problem is that sizeof(le16) shouldn't be promoted and
982 * the original code did that... Let's if zero this out and
983 * see what breaks.
986 if (is_restricted_type(sym)) {
987 if (type_bits(sym) < bits_in_int)
988 sym = &int_ctype;
990 #endif
991 if (is_fouled_type(sym))
992 sym = &int_ctype;
994 examine_symbol_type(sym);
996 ret.type = size_t_ctype;
997 if (type_bits(sym) <= 0) /* sizeof(void) */ {
998 if (get_real_base_type(sym) == &void_ctype)
999 ret.value = 1;
1000 else
1001 ret.value = 0;
1002 } else
1003 ret.value = type_bytes(sym);
1005 return ret;
1008 static struct range_list *handle_strlen(struct expression *expr, int implied, int *recurse_cnt)
1010 struct range_list *rl;
1011 struct expression *arg;
1012 sval_t sval = { .type = &ulong_ctype };
1014 if (implied == RL_EXACT)
1015 return NULL;
1017 arg = get_argument_from_call_expr(expr->args, 0);
1018 if (!arg)
1019 return NULL;
1020 if (arg->type == EXPR_STRING) {
1021 sval.value = arg->string->length - 1;
1022 return alloc_rl(sval, sval);
1025 if (implied == RL_HARD || implied == RL_FUZZY)
1026 return NULL;
1028 if (get_implied_return(expr, &rl))
1029 return rl;
1031 return NULL;
1034 static struct range_list *handle_builtin_constant_p(struct expression *expr, int implied, int *recurse_cnt)
1036 struct expression *arg;
1037 struct range_list *rl;
1038 sval_t sval;
1040 arg = get_argument_from_call_expr(expr->args, 0);
1041 rl = _get_rl(arg, RL_EXACT, recurse_cnt);
1042 if (rl_to_sval(rl, &sval))
1043 return rl_one();
1044 return rl_zero();
1047 static struct range_list *handle_call_rl(struct expression *expr, int implied, int *recurse_cnt)
1049 struct range_list *rl;
1051 if (sym_name_is("__builtin_constant_p", expr->fn))
1052 return handle_builtin_constant_p(expr, implied, recurse_cnt);
1054 if (sym_name_is("__builtin_expect", expr->fn) ||
1055 sym_name_is("__builtin_bswap16", expr->fn) ||
1056 sym_name_is("__builtin_bswap32", expr->fn) ||
1057 sym_name_is("__builtin_bswap64", expr->fn)) {
1058 struct expression *arg;
1060 arg = get_argument_from_call_expr(expr->args, 0);
1061 return _get_rl(arg, implied, recurse_cnt);
1064 if (sym_name_is("strlen", expr->fn))
1065 return handle_strlen(expr, implied, recurse_cnt);
1067 if (implied == RL_EXACT || implied == RL_HARD || implied == RL_FUZZY)
1068 return NULL;
1070 if (custom_handle_variable) {
1071 rl = custom_handle_variable(expr);
1072 if (rl)
1073 return rl;
1076 if (get_implied_return(expr, &rl))
1077 return rl;
1078 return db_return_vals(expr);
1081 static struct range_list *handle_cast(struct expression *expr, int implied, int *recurse_cnt)
1083 struct range_list *rl;
1084 struct symbol *type;
1086 type = get_type(expr);
1087 rl = _get_rl(expr->cast_expression, implied, recurse_cnt);
1088 if (rl)
1089 return cast_rl(type, rl);
1090 if (implied == RL_ABSOLUTE || implied == RL_REAL_ABSOLUTE)
1091 return alloc_whole_rl(type);
1092 if (implied == RL_IMPLIED && type &&
1093 type_bits(type) > 0 && type_bits(type) < 32)
1094 return alloc_whole_rl(type);
1095 return NULL;
1098 static struct range_list *_get_rl(struct expression *expr, int implied, int *recurse_cnt)
1100 struct range_list *rl;
1101 struct symbol *type;
1102 sval_t sval;
1104 type = get_type(expr);
1105 expr = strip_parens(expr);
1106 if (!expr)
1107 return NULL;
1109 if (++(*recurse_cnt) >= 200)
1110 return NULL;
1112 switch(expr->type) {
1113 case EXPR_CAST:
1114 case EXPR_FORCE_CAST:
1115 case EXPR_IMPLIED_CAST:
1116 rl = handle_cast(expr, implied, recurse_cnt);
1117 goto out_cast;
1120 expr = strip_expr(expr);
1121 if (!expr)
1122 return NULL;
1124 switch (expr->type) {
1125 case EXPR_VALUE:
1126 sval = sval_from_val(expr, expr->value);
1127 rl = alloc_rl(sval, sval);
1128 break;
1129 case EXPR_PREOP:
1130 rl = handle_preop_rl(expr, implied, recurse_cnt);
1131 break;
1132 case EXPR_POSTOP:
1133 rl = _get_rl(expr->unop, implied, recurse_cnt);
1134 break;
1135 case EXPR_BINOP:
1136 rl = handle_binop_rl(expr, implied, recurse_cnt);
1137 break;
1138 case EXPR_COMPARE:
1139 rl = handle_comparison_rl(expr, implied, recurse_cnt);
1140 break;
1141 case EXPR_LOGICAL:
1142 rl = handle_logical_rl(expr, implied, recurse_cnt);
1143 break;
1144 case EXPR_PTRSIZEOF:
1145 case EXPR_SIZEOF:
1146 sval = handle_sizeof(expr);
1147 rl = alloc_rl(sval, sval);
1148 break;
1149 case EXPR_SELECT:
1150 case EXPR_CONDITIONAL:
1151 rl = handle_conditional_rl(expr, implied, recurse_cnt);
1152 break;
1153 case EXPR_CALL:
1154 rl = handle_call_rl(expr, implied, recurse_cnt);
1155 break;
1156 default:
1157 rl = handle_variable(expr, implied, recurse_cnt);
1160 out_cast:
1161 if (rl)
1162 return rl;
1163 if (type && (implied == RL_ABSOLUTE || implied == RL_REAL_ABSOLUTE))
1164 return alloc_whole_rl(type);
1165 return NULL;
1168 struct {
1169 struct expression *expr;
1170 struct range_list *rl;
1171 } cached_results[24];
1172 static int cache_idx;
1174 void clear_math_cache(void)
1176 memset(cached_results, 0, sizeof(cached_results));
1179 /* returns 1 if it can get a value literal or else returns 0 */
1180 int get_value(struct expression *expr, sval_t *sval)
1182 struct range_list *rl;
1183 int recurse_cnt = 0;
1184 sval_t tmp;
1185 int i;
1188 * This only handles RL_EXACT because other expr statements can be
1189 * different at different points. Like the list iterator, for example.
1191 for (i = 0; i < ARRAY_SIZE(cached_results); i++) {
1192 if (expr == cached_results[i].expr)
1193 return rl_to_sval(cached_results[i].rl, sval);
1196 rl = _get_rl(expr, RL_EXACT, &recurse_cnt);
1197 if (!rl_to_sval(rl, &tmp))
1198 rl = NULL;
1200 cached_results[cache_idx].expr = expr;
1201 cached_results[cache_idx].rl = rl;
1202 cache_idx = (cache_idx + 1) % ARRAY_SIZE(cached_results);
1204 if (!rl)
1205 return 0;
1207 *sval = tmp;
1208 return 1;
1211 static int get_implied_value_internal(struct expression *expr, sval_t *sval, int *recurse_cnt)
1213 struct range_list *rl;
1215 rl = _get_rl(expr, RL_IMPLIED, recurse_cnt);
1216 if (!rl_to_sval(rl, sval))
1217 return 0;
1218 return 1;
1221 int get_implied_value(struct expression *expr, sval_t *sval)
1223 struct range_list *rl;
1224 int recurse_cnt = 0;
1226 rl = _get_rl(expr, RL_IMPLIED, &recurse_cnt);
1227 if (!rl_to_sval(rl, sval))
1228 return 0;
1229 return 1;
1232 int get_implied_min(struct expression *expr, sval_t *sval)
1234 struct range_list *rl;
1235 int recurse_cnt = 0;
1237 rl = _get_rl(expr, RL_IMPLIED, &recurse_cnt);
1238 if (!rl)
1239 return 0;
1240 *sval = rl_min(rl);
1241 return 1;
1244 int get_implied_max(struct expression *expr, sval_t *sval)
1246 struct range_list *rl;
1247 int recurse_cnt = 0;
1249 rl = _get_rl(expr, RL_IMPLIED, &recurse_cnt);
1250 if (!rl)
1251 return 0;
1252 *sval = rl_max(rl);
1253 return 1;
1256 int get_implied_rl(struct expression *expr, struct range_list **rl)
1258 int recurse_cnt = 0;
1260 *rl = _get_rl(expr, RL_IMPLIED, &recurse_cnt);
1261 if (*rl)
1262 return 1;
1263 return 0;
1266 static int get_absolute_rl_internal(struct expression *expr, struct range_list **rl, int *recurse_cnt)
1268 *rl = _get_rl(expr, RL_ABSOLUTE, recurse_cnt);
1269 if (!*rl)
1270 *rl = alloc_whole_rl(get_type(expr));
1271 return 1;
1274 int get_absolute_rl(struct expression *expr, struct range_list **rl)
1276 int recurse_cnt = 0;
1278 *rl = _get_rl(expr, RL_ABSOLUTE, &recurse_cnt);
1279 if (!*rl)
1280 *rl = alloc_whole_rl(get_type(expr));
1281 return 1;
1284 int get_real_absolute_rl(struct expression *expr, struct range_list **rl)
1286 int recurse_cnt = 0;
1288 *rl = _get_rl(expr, RL_REAL_ABSOLUTE, &recurse_cnt);
1289 if (!*rl)
1290 *rl = alloc_whole_rl(get_type(expr));
1291 return 1;
1294 int custom_get_absolute_rl(struct expression *expr,
1295 struct range_list *(*fn)(struct expression *expr),
1296 struct range_list **rl)
1298 int recurse_cnt = 0;
1300 *rl = NULL;
1301 custom_handle_variable = fn;
1302 *rl = _get_rl(expr, RL_REAL_ABSOLUTE, &recurse_cnt);
1303 custom_handle_variable = NULL;
1304 return 1;
1307 int get_implied_rl_var_sym(const char *var, struct symbol *sym, struct range_list **rl)
1309 struct smatch_state *state;
1311 state = get_state(SMATCH_EXTRA, var, sym);
1312 *rl = estate_rl(state);
1313 if (*rl)
1314 return 1;
1315 return 0;
1318 int get_hard_max(struct expression *expr, sval_t *sval)
1320 struct range_list *rl;
1321 int recurse_cnt = 0;
1323 rl = _get_rl(expr, RL_HARD, &recurse_cnt);
1324 if (!rl)
1325 return 0;
1326 *sval = rl_max(rl);
1327 return 1;
1330 int get_fuzzy_min(struct expression *expr, sval_t *sval)
1332 struct range_list *rl;
1333 sval_t tmp;
1334 int recurse_cnt = 0;
1336 rl = _get_rl(expr, RL_FUZZY, &recurse_cnt);
1337 if (!rl)
1338 return 0;
1339 tmp = rl_min(rl);
1340 if (sval_is_negative(tmp) && sval_is_min(tmp))
1341 return 0;
1342 *sval = tmp;
1343 return 1;
1346 int get_fuzzy_max(struct expression *expr, sval_t *sval)
1348 struct range_list *rl;
1349 sval_t max;
1350 int recurse_cnt = 0;
1352 rl = _get_rl(expr, RL_FUZZY, &recurse_cnt);
1353 if (!rl)
1354 return 0;
1355 max = rl_max(rl);
1356 if (max.uvalue > INT_MAX - 10000)
1357 return 0;
1358 *sval = max;
1359 return 1;
1362 int get_absolute_min(struct expression *expr, sval_t *sval)
1364 struct range_list *rl;
1365 struct symbol *type;
1366 int recurse_cnt = 0;
1368 type = get_type(expr);
1369 if (!type)
1370 type = &llong_ctype; // FIXME: this is wrong but places assume get type can't fail.
1371 rl = _get_rl(expr, RL_REAL_ABSOLUTE, &recurse_cnt);
1372 if (rl)
1373 *sval = rl_min(rl);
1374 else
1375 *sval = sval_type_min(type);
1377 if (sval_cmp(*sval, sval_type_min(type)) < 0)
1378 *sval = sval_type_min(type);
1379 return 1;
1382 int get_absolute_max(struct expression *expr, sval_t *sval)
1384 struct range_list *rl;
1385 struct symbol *type;
1386 int recurse_cnt = 0;
1388 type = get_type(expr);
1389 if (!type)
1390 type = &llong_ctype;
1391 rl = _get_rl(expr, RL_REAL_ABSOLUTE, &recurse_cnt);
1392 if (rl)
1393 *sval = rl_max(rl);
1394 else
1395 *sval = sval_type_max(type);
1397 if (sval_cmp(sval_type_max(type), *sval) < 0)
1398 *sval = sval_type_max(type);
1399 return 1;
1402 int known_condition_true(struct expression *expr)
1404 sval_t tmp;
1406 if (!expr)
1407 return 0;
1409 if (get_value(expr, &tmp) && tmp.value)
1410 return 1;
1412 return 0;
1415 int known_condition_false(struct expression *expr)
1417 if (!expr)
1418 return 0;
1420 if (is_zero(expr))
1421 return 1;
1423 return 0;
1426 int implied_condition_true(struct expression *expr)
1428 sval_t tmp;
1430 if (!expr)
1431 return 0;
1433 if (known_condition_true(expr))
1434 return 1;
1435 if (get_implied_value(expr, &tmp) && tmp.value)
1436 return 1;
1438 if (expr->type == EXPR_POSTOP)
1439 return implied_condition_true(expr->unop);
1441 if (expr->type == EXPR_PREOP && expr->op == SPECIAL_DECREMENT)
1442 return implied_not_equal(expr->unop, 1);
1443 if (expr->type == EXPR_PREOP && expr->op == SPECIAL_INCREMENT)
1444 return implied_not_equal(expr->unop, -1);
1446 expr = strip_expr(expr);
1447 switch (expr->type) {
1448 case EXPR_COMPARE:
1449 if (do_comparison(expr) == 1)
1450 return 1;
1451 break;
1452 case EXPR_PREOP:
1453 if (expr->op == '!') {
1454 if (implied_condition_false(expr->unop))
1455 return 1;
1456 break;
1458 break;
1459 default:
1460 if (implied_not_equal(expr, 0) == 1)
1461 return 1;
1462 break;
1464 return 0;
1467 int implied_condition_false(struct expression *expr)
1469 struct expression *tmp;
1470 sval_t sval;
1472 if (!expr)
1473 return 0;
1475 if (known_condition_false(expr))
1476 return 1;
1478 switch (expr->type) {
1479 case EXPR_COMPARE:
1480 if (do_comparison(expr) == 2)
1481 return 1;
1482 case EXPR_PREOP:
1483 if (expr->op == '!') {
1484 if (implied_condition_true(expr->unop))
1485 return 1;
1486 break;
1488 tmp = strip_expr(expr);
1489 if (tmp != expr)
1490 return implied_condition_false(tmp);
1491 break;
1492 default:
1493 if (get_implied_value(expr, &sval) && sval.value == 0)
1494 return 1;
1495 break;
1497 return 0;
1500 int can_integer_overflow(struct symbol *type, struct expression *expr)
1502 int op;
1503 sval_t lmax, rmax, res;
1505 if (!type)
1506 type = &int_ctype;
1508 expr = strip_expr(expr);
1510 if (expr->type == EXPR_ASSIGNMENT) {
1511 switch(expr->op) {
1512 case SPECIAL_MUL_ASSIGN:
1513 op = '*';
1514 break;
1515 case SPECIAL_ADD_ASSIGN:
1516 op = '+';
1517 break;
1518 case SPECIAL_SHL_ASSIGN:
1519 op = SPECIAL_LEFTSHIFT;
1520 break;
1521 default:
1522 return 0;
1524 } else if (expr->type == EXPR_BINOP) {
1525 if (expr->op != '*' && expr->op != '+' && expr->op != SPECIAL_LEFTSHIFT)
1526 return 0;
1527 op = expr->op;
1528 } else {
1529 return 0;
1532 get_absolute_max(expr->left, &lmax);
1533 get_absolute_max(expr->right, &rmax);
1535 if (sval_binop_overflows(lmax, op, rmax))
1536 return 1;
1538 res = sval_binop(lmax, op, rmax);
1539 if (sval_cmp(res, sval_type_max(type)) > 0)
1540 return 1;
1541 return 0;