math: store all constant EXPR_BINOP results
[smatch.git] / smatch_extra.c
blob1cdfef8164493e32b271ff4432350a8af3c871e3
1 /*
2 * Copyright (C) 2008 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
19 * smatch_extra.c is supposed to track the value of every variable.
23 #define _GNU_SOURCE
24 #include <string.h>
26 #include <stdlib.h>
27 #include <errno.h>
28 #ifndef __USE_ISOC99
29 #define __USE_ISOC99
30 #endif
31 #include <limits.h>
32 #include "parse.h"
33 #include "smatch.h"
34 #include "smatch_slist.h"
35 #include "smatch_extra.h"
37 static int my_id;
38 static int link_id;
40 static void match_link_modify(struct sm_state *sm, struct expression *mod_expr);
42 struct string_list *__ignored_macros = NULL;
43 int in_warn_on_macro(void)
45 struct statement *stmt;
46 char *tmp;
47 char *macro;
49 stmt = get_current_statement();
50 if (!stmt)
51 return 0;
52 macro = get_macro_name(stmt->pos);
53 if (!macro)
54 return 0;
56 FOR_EACH_PTR(__ignored_macros, tmp) {
57 if (!strcmp(tmp, macro))
58 return 1;
59 } END_FOR_EACH_PTR(tmp);
60 return 0;
63 typedef void (mod_hook)(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state);
64 DECLARE_PTR_LIST(void_fn_list, mod_hook *);
65 static struct void_fn_list *extra_mod_hooks;
66 static struct void_fn_list *extra_nomod_hooks;
68 void add_extra_mod_hook(mod_hook *fn)
70 mod_hook **p = malloc(sizeof(mod_hook *));
71 *p = fn;
72 add_ptr_list(&extra_mod_hooks, p);
75 void add_extra_nomod_hook(mod_hook *fn)
77 mod_hook **p = malloc(sizeof(mod_hook *));
78 *p = fn;
79 add_ptr_list(&extra_nomod_hooks, p);
82 void call_extra_hooks(struct void_fn_list *hooks, const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
84 mod_hook **fn;
86 FOR_EACH_PTR(hooks, fn) {
87 (*fn)(name, sym, expr, state);
88 } END_FOR_EACH_PTR(fn);
91 void call_extra_mod_hooks(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
93 call_extra_hooks(extra_mod_hooks, name, sym, expr, state);
96 void call_extra_nomod_hooks(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
98 call_extra_hooks(extra_nomod_hooks, name, sym, expr, state);
101 static void set_union_info(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
103 struct symbol *type, *tmp, *inner_type, *inner, *new_type;
104 struct expression *deref, *member_expr;
105 struct smatch_state *new;
106 int offset, inner_offset;
107 static bool in_recurse;
108 char *member_name;
110 if (__in_fake_assign)
111 return;
113 if (in_recurse)
114 return;
115 in_recurse = true;
117 if (!expr || expr->type != EXPR_DEREF || !expr->member)
118 goto done;
119 offset = get_member_offset_from_deref(expr);
120 if (offset < 0)
121 goto done;
123 deref = strip_expr(expr->deref);
124 type = get_type(deref);
125 if (type_is_ptr(type))
126 type = get_real_base_type(type);
127 if (!type || type->type != SYM_STRUCT)
128 goto done;
130 FOR_EACH_PTR(type->symbol_list, tmp) {
131 inner_type = get_real_base_type(tmp);
132 if (!inner_type || inner_type->type != SYM_UNION)
133 continue;
135 inner = first_ptr_list((struct ptr_list *)inner_type->symbol_list);
136 if (!inner || !inner->ident)
137 continue;
139 inner_offset = get_member_offset(type, inner->ident->name);
140 if (inner_offset < offset)
141 continue;
142 if (inner_offset > offset)
143 goto done;
145 FOR_EACH_PTR(inner_type->symbol_list, inner) {
146 struct symbol *tmp_type;
148 if (!inner->ident || inner->ident == expr->member)
149 continue;
150 tmp_type = get_real_base_type(inner);
151 if (tmp_type && tmp_type->type == SYM_STRUCT)
152 continue;
153 member_expr = deref;
154 if (tmp->ident)
155 member_expr = member_expression(member_expr, '.', tmp->ident);
156 member_expr = member_expression(member_expr, expr->op, inner->ident);
157 member_name = expr_to_var(member_expr);
158 if (!member_name)
159 continue;
160 new_type = get_real_base_type(inner);
161 new = alloc_estate_rl(cast_rl(new_type, estate_rl(state)));
162 set_extra_mod_helper(member_name, sym, member_expr, new);
163 free_string(member_name);
164 } END_FOR_EACH_PTR(inner);
165 } END_FOR_EACH_PTR(tmp);
167 done:
168 in_recurse = false;
171 static bool in_param_set;
172 void set_extra_mod_helper(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
174 struct expression *faked;
176 if (!expr)
177 expr = gen_expression_from_name_sym(name, sym);
178 remove_from_equiv(name, sym);
179 set_union_info(name, sym, expr, state);
180 call_extra_mod_hooks(name, sym, expr, state);
181 faked = get_faked_expression();
182 if (!faked ||
183 (faked->type == EXPR_ASSIGNMENT && is_fresh_alloc(faked->right)))
184 update_mtag_data(expr, state);
185 if ((__in_fake_assign || in_param_set) &&
186 estate_is_unknown(state) && !get_state(SMATCH_EXTRA, name, sym))
187 return;
188 set_state(SMATCH_EXTRA, name, sym, state);
191 static void set_extra_nomod_helper(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
193 call_extra_nomod_hooks(name, sym, expr, state);
194 set_state(SMATCH_EXTRA, name, sym, state);
197 static char *get_pointed_at(const char *name, struct symbol *sym, struct symbol **new_sym)
199 struct expression *assigned;
202 * Imagine we have an assignment: "foo = &addr;" then the other name
203 * of "*foo" is addr.
206 if (name[0] != '*')
207 return NULL;
208 if (strcmp(name + 1, sym->ident->name) != 0)
209 return NULL;
211 assigned = get_assigned_expr_name_sym(sym->ident->name, sym);
212 if (!assigned)
213 return NULL;
214 assigned = strip_parens(assigned);
215 if (assigned->type != EXPR_PREOP || assigned->op != '&')
216 return NULL;
218 return expr_to_var_sym(assigned->unop, new_sym);
221 char *get_other_name_sym_from_chunk(const char *name, const char *chunk, int len, struct symbol *sym, struct symbol **new_sym)
223 struct expression *assigned;
224 char *orig_name = NULL;
225 char buf[256];
226 char *ret;
228 assigned = get_assigned_expr_name_sym(chunk, sym);
229 if (!assigned)
230 return NULL;
231 if (assigned->type == EXPR_CALL)
232 return map_call_to_other_name_sym(name, sym, new_sym);
233 if (assigned->type == EXPR_PREOP && assigned->op == '&') {
235 orig_name = expr_to_var_sym(assigned, new_sym);
236 if (!orig_name || !*new_sym)
237 goto free;
239 snprintf(buf, sizeof(buf), "%s.%s", orig_name + 1, name + len);
240 ret = alloc_string(buf);
241 free_string(orig_name);
242 return ret;
245 orig_name = expr_to_var_sym(assigned, new_sym);
246 if (!orig_name || !*new_sym)
247 goto free;
249 snprintf(buf, sizeof(buf), "%s->%s", orig_name, name + len);
250 ret = alloc_string(buf);
251 free_string(orig_name);
252 return ret;
253 free:
254 free_string(orig_name);
255 return NULL;
258 static char *get_long_name_sym(const char *name, struct symbol *sym, struct symbol **new_sym, bool use_stack)
260 struct expression *tmp;
261 struct sm_state *sm;
262 char buf[256];
265 * Just prepend the name with a different name/sym and return that.
266 * For example, if we set "foo->bar = bar;" then the other name
267 * for "bar->baz" is "foo->bar->baz". Or if we have "foo = bar;" then
268 * the other name for "bar" is "foo". A third option is if we have
269 * "foo = bar;" then another name for "*bar" is "*foo".
272 FOR_EACH_MY_SM(check_assigned_expr_id, __get_cur_stree(), sm) {
273 tmp = sm->state->data;
274 if (!tmp || tmp->type != EXPR_SYMBOL)
275 continue;
276 if (tmp->symbol == sym)
277 goto found;
278 } END_FOR_EACH_SM(sm);
280 return NULL;
282 found:
283 if (!use_stack && name[tmp->symbol->ident->len] != '-')
284 return NULL;
286 if (name[0] == '*' && strcmp(name + 1, tmp->symbol_name->name) == 0)
287 snprintf(buf, sizeof(buf), "*%s", sm->name);
288 else if (name[tmp->symbol->ident->len] == '-' ||
289 name[tmp->symbol->ident->len] == '.')
290 snprintf(buf, sizeof(buf), "%s%s", sm->name, name + tmp->symbol->ident->len);
291 else if (strcmp(name, tmp->symbol_name->name) == 0)
292 snprintf(buf, sizeof(buf), "%s", sm->name);
293 else
294 return NULL;
296 *new_sym = sm->sym;
297 return alloc_string(buf);
300 char *get_other_name_sym_helper(const char *name, struct symbol *sym, struct symbol **new_sym, bool use_stack)
302 char buf[256];
303 char *ret;
304 int len;
306 *new_sym = NULL;
308 if (!sym || !sym->ident)
309 return NULL;
311 ret = get_pointed_at(name, sym, new_sym);
312 if (ret)
313 return ret;
315 ret = map_long_to_short_name_sym(name, sym, new_sym, use_stack);
316 if (ret)
317 return ret;
319 len = snprintf(buf, sizeof(buf), "%s", name);
320 if (len >= sizeof(buf) - 2)
321 return NULL;
323 while (use_stack && len >= 1) {
324 if (buf[len] == '>' && buf[len - 1] == '-') {
325 len--;
326 buf[len] = '\0';
327 ret = get_other_name_sym_from_chunk(name, buf, len + 2, sym, new_sym);
328 if (ret)
329 return ret;
331 len--;
334 ret = get_long_name_sym(name, sym, new_sym, use_stack);
335 if (ret)
336 return ret;
338 return NULL;
341 char *get_other_name_sym(const char *name, struct symbol *sym, struct symbol **new_sym)
343 return get_other_name_sym_helper(name, sym, new_sym, true);
346 char *get_other_name_sym_nostack(const char *name, struct symbol *sym, struct symbol **new_sym)
348 return get_other_name_sym_helper(name, sym, new_sym, false);
351 void set_extra_mod(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
353 char *new_name;
354 struct symbol *new_sym;
356 set_extra_mod_helper(name, sym, expr, state);
357 new_name = get_other_name_sym_nostack(name, sym, &new_sym);
358 if (new_name && new_sym)
359 set_extra_mod_helper(new_name, new_sym, NULL, state);
360 free_string(new_name);
363 static struct expression *chunk_get_array_base(struct expression *expr)
366 * The problem with is_array() is that it only returns true for things
367 * like foo[1] but not for foo[1].bar.
370 expr = strip_expr(expr);
371 while (expr && expr->type == EXPR_DEREF)
372 expr = strip_expr(expr->deref);
373 return get_array_base(expr);
376 static int chunk_has_array(struct expression *expr)
378 return !!chunk_get_array_base(expr);
381 static void clear_array_states(struct expression *array)
383 struct sm_state *sm;
385 sm = get_sm_state_expr(link_id, array);
386 if (sm)
387 match_link_modify(sm, NULL);
390 static void set_extra_array_mod(struct expression *expr, struct smatch_state *state)
392 struct expression *array;
393 struct var_sym_list *vsl;
394 struct var_sym *vs;
395 char *name;
396 struct symbol *sym;
398 array = chunk_get_array_base(expr);
400 name = expr_to_chunk_sym_vsl(expr, &sym, &vsl);
401 if (!name || !vsl) {
402 clear_array_states(array);
403 goto free;
406 FOR_EACH_PTR(vsl, vs) {
407 store_link(link_id, vs->var, vs->sym, name, sym);
408 } END_FOR_EACH_PTR(vs);
410 call_extra_mod_hooks(name, sym, expr, state);
411 set_state(SMATCH_EXTRA, name, sym, state);
412 free:
413 free_string(name);
416 void set_extra_expr_mod(struct expression *expr, struct smatch_state *state)
418 struct symbol *sym;
419 char *name;
421 if (chunk_has_array(expr)) {
422 set_extra_array_mod(expr, state);
423 return;
426 expr = strip_expr(expr);
427 name = expr_to_var_sym(expr, &sym);
428 if (!name || !sym)
429 goto free;
430 set_extra_mod(name, sym, expr, state);
431 free:
432 free_string(name);
435 void set_extra_nomod(const char *name, struct symbol *sym, struct expression *expr, struct smatch_state *state)
437 char *new_name;
438 struct symbol *new_sym;
439 struct relation *rel;
440 struct smatch_state *orig_state;
442 orig_state = get_state(SMATCH_EXTRA, name, sym);
444 /* don't save unknown states if leaving it blank is the same */
445 if (!orig_state && estate_is_unknown(state))
446 return;
448 new_name = get_other_name_sym(name, sym, &new_sym);
449 if (new_name && new_sym)
450 set_extra_nomod_helper(new_name, new_sym, expr, state);
451 free_string(new_name);
453 if (!estate_related(orig_state)) {
454 set_extra_nomod_helper(name, sym, expr, state);
455 return;
458 set_related(state, estate_related(orig_state));
459 FOR_EACH_PTR(estate_related(orig_state), rel) {
460 struct smatch_state *estate;
462 estate = get_state(SMATCH_EXTRA, rel->name, rel->sym);
463 if (!estate)
464 continue;
465 set_extra_nomod_helper(rel->name, rel->sym, expr, clone_estate_cast(estate_type(estate), state));
466 } END_FOR_EACH_PTR(rel);
469 void set_extra_nomod_vsl(const char *name, struct symbol *sym, struct var_sym_list *vsl, struct expression *expr, struct smatch_state *state)
471 struct var_sym *vs;
473 FOR_EACH_PTR(vsl, vs) {
474 store_link(link_id, vs->var, vs->sym, name, sym);
475 } END_FOR_EACH_PTR(vs);
477 set_extra_nomod(name, sym, expr, state);
481 * This is for return_implies_state() hooks which modify a SMATCH_EXTRA state
483 void set_extra_expr_nomod(struct expression *expr, struct smatch_state *state)
485 struct var_sym_list *vsl;
486 struct var_sym *vs;
487 char *name;
488 struct symbol *sym;
490 name = expr_to_chunk_sym_vsl(expr, &sym, &vsl);
491 if (!name || !vsl)
492 goto free;
493 FOR_EACH_PTR(vsl, vs) {
494 store_link(link_id, vs->var, vs->sym, name, sym);
495 } END_FOR_EACH_PTR(vs);
497 set_extra_nomod(name, sym, expr, state);
498 free:
499 free_string(name);
502 static void set_extra_true_false(const char *name, struct symbol *sym,
503 struct smatch_state *true_state,
504 struct smatch_state *false_state)
506 char *new_name;
507 struct symbol *new_sym;
508 struct relation *rel;
509 struct smatch_state *orig_state;
511 if (!true_state && !false_state)
512 return;
514 if (in_warn_on_macro())
515 return;
517 new_name = get_other_name_sym(name, sym, &new_sym);
518 if (new_name && new_sym)
519 set_true_false_states(SMATCH_EXTRA, new_name, new_sym, true_state, false_state);
520 free_string(new_name);
522 orig_state = get_state(SMATCH_EXTRA, name, sym);
524 if (!estate_related(orig_state)) {
525 set_true_false_states(SMATCH_EXTRA, name, sym, true_state, false_state);
526 return;
529 if (true_state)
530 set_related(true_state, estate_related(orig_state));
531 if (false_state)
532 set_related(false_state, estate_related(orig_state));
534 FOR_EACH_PTR(estate_related(orig_state), rel) {
535 set_true_false_states(SMATCH_EXTRA, rel->name, rel->sym,
536 true_state, false_state);
537 } END_FOR_EACH_PTR(rel);
540 static void set_extra_chunk_true_false(struct expression *expr,
541 struct smatch_state *true_state,
542 struct smatch_state *false_state)
544 struct var_sym_list *vsl;
545 struct var_sym *vs;
546 struct symbol *type;
547 char *name;
548 struct symbol *sym;
550 if (in_warn_on_macro())
551 return;
553 type = get_type(expr);
554 if (!type)
555 return;
557 name = expr_to_chunk_sym_vsl(expr, &sym, &vsl);
558 if (!name || !vsl)
559 goto free;
560 FOR_EACH_PTR(vsl, vs) {
561 store_link(link_id, vs->var, vs->sym, name, sym);
562 } END_FOR_EACH_PTR(vs);
564 set_true_false_states(SMATCH_EXTRA, name, sym,
565 clone_estate(true_state),
566 clone_estate(false_state));
567 free:
568 free_string(name);
571 static void set_extra_expr_true_false(struct expression *expr,
572 struct smatch_state *true_state,
573 struct smatch_state *false_state)
575 char *name;
576 struct symbol *sym;
577 sval_t sval;
579 if (!true_state && !false_state)
580 return;
582 if (get_value(expr, &sval))
583 return;
585 expr = strip_expr(expr);
586 name = expr_to_var_sym(expr, &sym);
587 if (!name || !sym) {
588 free_string(name);
589 set_extra_chunk_true_false(expr, true_state, false_state);
590 return;
592 set_extra_true_false(name, sym, true_state, false_state);
593 free_string(name);
596 static int get_countdown_info(struct expression *condition, struct expression **unop, int *op, sval_t *right)
598 struct expression *unop_expr;
599 int comparison;
600 sval_t limit;
602 right->type = &int_ctype;
603 right->value = 0;
605 condition = strip_expr(condition);
607 if (condition->type == EXPR_COMPARE) {
608 comparison = remove_unsigned_from_comparison(condition->op);
610 if (comparison != SPECIAL_GTE && comparison != '>')
611 return 0;
612 if (!get_value(condition->right, &limit))
613 return 0;
615 unop_expr = condition->left;
616 if (unop_expr->type != EXPR_PREOP && unop_expr->type != EXPR_POSTOP)
617 return 0;
618 if (unop_expr->op != SPECIAL_DECREMENT)
619 return 0;
621 *unop = unop_expr;
622 *op = comparison;
623 *right = limit;
625 return 1;
628 if (condition->type != EXPR_PREOP && condition->type != EXPR_POSTOP)
629 return 0;
630 if (condition->op != SPECIAL_DECREMENT)
631 return 0;
633 *unop = condition;
634 *op = '>';
636 return 1;
639 static struct sm_state *handle_canonical_while_count_down(struct statement *loop)
641 struct expression *iter_var;
642 struct expression *condition, *unop;
643 struct symbol *type;
644 struct sm_state *sm;
645 struct smatch_state *estate;
646 int op;
647 sval_t start, right;
649 right.type = &int_ctype;
650 right.value = 0;
652 condition = strip_expr(loop->iterator_pre_condition);
653 if (!condition)
654 return NULL;
656 if (!get_countdown_info(condition, &unop, &op, &right))
657 return NULL;
659 iter_var = unop->unop;
661 sm = get_sm_state_expr(SMATCH_EXTRA, iter_var);
662 if (!sm)
663 return NULL;
664 if (sval_cmp(estate_min(sm->state), right) < 0)
665 return NULL;
666 start = estate_max(sm->state);
668 type = get_type(iter_var);
669 right = sval_cast(type, right);
670 start = sval_cast(type, start);
672 if (sval_cmp(start, right) <= 0)
673 return NULL;
674 if (!sval_is_max(start))
675 start.value--;
677 if (op == SPECIAL_GTE)
678 right.value--;
680 if (unop->type == EXPR_PREOP) {
681 right.value++;
682 estate = alloc_estate_range(right, start);
683 if (estate_has_hard_max(sm->state))
684 estate_set_hard_max(estate);
685 estate_copy_fuzzy_max(estate, sm->state);
686 set_extra_expr_mod(iter_var, estate);
688 if (unop->type == EXPR_POSTOP) {
689 estate = alloc_estate_range(right, start);
690 if (estate_has_hard_max(sm->state))
691 estate_set_hard_max(estate);
692 estate_copy_fuzzy_max(estate, sm->state);
693 set_extra_expr_mod(iter_var, estate);
695 return get_sm_state_expr(SMATCH_EXTRA, iter_var);
698 static struct sm_state *handle_canonical_for_inc(struct expression *iter_expr,
699 struct expression *condition)
701 struct expression *iter_var;
702 struct sm_state *sm;
703 struct smatch_state *estate;
704 sval_t start, end, max;
705 struct symbol *type;
707 iter_var = iter_expr->unop;
708 sm = get_sm_state_expr(SMATCH_EXTRA, iter_var);
709 if (!sm)
710 return NULL;
711 if (!estate_get_single_value(sm->state, &start))
712 return NULL;
713 if (!get_implied_value(condition->right, &end))
714 return NULL;
716 if (get_sm_state_expr(SMATCH_EXTRA, condition->left) != sm)
717 return NULL;
719 switch (condition->op) {
720 case SPECIAL_UNSIGNED_LT:
721 case SPECIAL_NOTEQUAL:
722 case '<':
723 if (!sval_is_min(end))
724 end.value--;
725 break;
726 case SPECIAL_UNSIGNED_LTE:
727 case SPECIAL_LTE:
728 break;
729 default:
730 return NULL;
732 if (sval_cmp(end, start) < 0)
733 return NULL;
734 type = get_type(iter_var);
735 start = sval_cast(type, start);
736 end = sval_cast(type, end);
737 estate = alloc_estate_range(start, end);
738 if (get_hard_max(condition->right, &max)) {
739 if (!get_macro_name(condition->pos))
740 estate_set_hard_max(estate);
741 if (condition->op == '<' ||
742 condition->op == SPECIAL_UNSIGNED_LT ||
743 condition->op == SPECIAL_NOTEQUAL)
744 max.value--;
745 max = sval_cast(type, max);
746 estate_set_fuzzy_max(estate, max);
748 set_extra_expr_mod(iter_var, estate);
749 return get_sm_state_expr(SMATCH_EXTRA, iter_var);
752 static struct sm_state *handle_canonical_for_dec(struct expression *iter_expr,
753 struct expression *condition)
755 struct expression *iter_var;
756 struct sm_state *sm;
757 struct smatch_state *estate;
758 sval_t start, end;
760 iter_var = iter_expr->unop;
761 sm = get_sm_state_expr(SMATCH_EXTRA, iter_var);
762 if (!sm)
763 return NULL;
764 if (!estate_get_single_value(sm->state, &start))
765 return NULL;
766 if (!get_implied_min(condition->right, &end))
767 end = sval_type_min(get_type(iter_var));
768 end = sval_cast(estate_type(sm->state), end);
769 if (get_sm_state_expr(SMATCH_EXTRA, condition->left) != sm)
770 return NULL;
772 switch (condition->op) {
773 case SPECIAL_NOTEQUAL:
774 case '>':
775 if (!sval_is_max(end))
776 end.value++;
777 break;
778 case SPECIAL_GTE:
779 break;
780 default:
781 return NULL;
783 if (sval_cmp(end, start) > 0)
784 return NULL;
785 estate = alloc_estate_range(end, start);
786 estate_set_hard_max(estate);
787 estate_set_fuzzy_max(estate, estate_get_fuzzy_max(estate));
788 set_extra_expr_mod(iter_var, estate);
789 return get_sm_state_expr(SMATCH_EXTRA, iter_var);
792 static struct sm_state *handle_canonical_for_loops(struct statement *loop)
794 struct expression *iter_expr;
795 struct expression *condition;
797 if (!loop->iterator_post_statement)
798 return NULL;
799 if (loop->iterator_post_statement->type != STMT_EXPRESSION)
800 return NULL;
801 iter_expr = loop->iterator_post_statement->expression;
802 if (!loop->iterator_pre_condition)
803 return NULL;
804 if (loop->iterator_pre_condition->type != EXPR_COMPARE)
805 return NULL;
806 condition = loop->iterator_pre_condition;
808 if (iter_expr->op == SPECIAL_INCREMENT)
809 return handle_canonical_for_inc(iter_expr, condition);
810 if (iter_expr->op == SPECIAL_DECREMENT)
811 return handle_canonical_for_dec(iter_expr, condition);
812 return NULL;
815 struct sm_state *__extra_handle_canonical_loops(struct statement *loop, struct stree **stree)
817 struct sm_state *ret;
820 * Canonical loops are a hack. The proper way to handle this is to
821 * use two passes, but unfortunately, doing two passes makes parsing
822 * code twice as slow.
824 * What we do is we set the inside state here, which overwrites whatever
825 * __extra_match_condition() does. Then we set the outside state in
826 * __extra_pre_loop_hook_after().
829 __push_fake_cur_stree();
830 if (!loop->iterator_post_statement)
831 ret = handle_canonical_while_count_down(loop);
832 else
833 ret = handle_canonical_for_loops(loop);
834 *stree = __pop_fake_cur_stree();
835 return ret;
838 int __iterator_unchanged(struct sm_state *sm)
840 if (!sm)
841 return 0;
842 if (get_sm_state(my_id, sm->name, sm->sym) == sm)
843 return 1;
844 return 0;
847 static void while_count_down_after(struct sm_state *sm, struct expression *condition)
849 struct expression *unop;
850 int op;
851 sval_t limit, after_value;
853 if (!get_countdown_info(condition, &unop, &op, &limit))
854 return;
855 after_value = estate_min(sm->state);
856 after_value.value--;
857 set_extra_mod(sm->name, sm->sym, condition->unop, alloc_estate_sval(after_value));
860 void __extra_pre_loop_hook_after(struct sm_state *sm,
861 struct statement *iterator,
862 struct expression *condition)
864 struct expression *iter_expr;
865 sval_t limit;
866 struct smatch_state *state;
868 if (!iterator) {
869 while_count_down_after(sm, condition);
870 return;
873 iter_expr = iterator->expression;
875 if (condition->type != EXPR_COMPARE)
876 return;
877 if (iter_expr->op == SPECIAL_INCREMENT) {
878 limit = sval_binop(estate_max(sm->state), '+',
879 sval_type_val(estate_type(sm->state), 1));
880 } else {
881 limit = sval_binop(estate_min(sm->state), '-',
882 sval_type_val(estate_type(sm->state), 1));
884 limit = sval_cast(estate_type(sm->state), limit);
885 if (!estate_has_hard_max(sm->state) && !__has_breaks()) {
886 if (iter_expr->op == SPECIAL_INCREMENT)
887 state = alloc_estate_range(estate_min(sm->state), limit);
888 else
889 state = alloc_estate_range(limit, estate_max(sm->state));
890 } else {
891 state = alloc_estate_sval(limit);
893 if (!estate_has_hard_max(sm->state)) {
894 estate_clear_hard_max(state);
896 if (estate_has_fuzzy_max(sm->state)) {
897 sval_t hmax = estate_get_fuzzy_max(sm->state);
898 sval_t max = estate_max(sm->state);
900 if (sval_cmp(hmax, max) != 0)
901 estate_clear_fuzzy_max(state);
902 } else if (!estate_has_fuzzy_max(sm->state)) {
903 estate_clear_fuzzy_max(state);
906 set_extra_mod(sm->name, sm->sym, iter_expr, state);
909 static bool get_global_rl(const char *name, struct symbol *sym, struct range_list **rl)
911 struct expression *expr;
913 if (!sym || !(sym->ctype.modifiers & MOD_TOPLEVEL) || !sym->ident)
914 return false;
915 if (strcmp(sym->ident->name, name) != 0)
916 return false;
918 expr = symbol_expression(sym);
919 return get_implied_rl(expr, rl);
922 static struct stree *unmatched_stree;
923 static struct smatch_state *unmatched_state(struct sm_state *sm)
925 struct smatch_state *state;
926 struct range_list *rl;
928 if (unmatched_stree) {
929 state = get_state_stree(unmatched_stree, SMATCH_EXTRA, sm->name, sm->sym);
930 if (state)
931 return state;
933 if (parent_is_gone_var_sym(sm->name, sm->sym))
934 return alloc_estate_empty();
935 if (get_global_rl(sm->name, sm->sym, &rl))
936 return alloc_estate_rl(rl);
937 return alloc_estate_whole(estate_type(sm->state));
940 static void clear_the_pointed_at(struct expression *expr)
942 struct stree *stree;
943 char *name;
944 struct symbol *sym;
945 struct sm_state *tmp;
947 name = expr_to_var_sym(expr, &sym);
948 if (!name || !sym)
949 goto free;
951 stree = __get_cur_stree();
952 FOR_EACH_MY_SM(SMATCH_EXTRA, stree, tmp) {
953 if (tmp->name[0] != '*')
954 continue;
955 if (tmp->sym != sym)
956 continue;
957 if (strcmp(tmp->name + 1, name) != 0)
958 continue;
959 set_extra_mod(tmp->name, tmp->sym, expr, alloc_estate_whole(estate_type(tmp->state)));
960 } END_FOR_EACH_SM(tmp);
962 free:
963 free_string(name);
966 static int is_const_param(struct expression *expr, int param)
968 struct symbol *type;
970 type = get_arg_type(expr, param);
971 if (!type)
972 return 0;
973 if (type->ctype.modifiers & MOD_CONST)
974 return 1;
975 return 0;
978 static void match_function_call(struct expression *expr)
980 struct expression *arg;
981 struct expression *tmp;
982 int param = -1;
984 /* if we have the db this is handled in smatch_function_hooks.c */
985 if (!option_no_db)
986 return;
987 if (inlinable(expr->fn))
988 return;
990 FOR_EACH_PTR(expr->args, arg) {
991 param++;
992 if (is_const_param(expr->fn, param))
993 continue;
994 tmp = strip_expr(arg);
995 if (tmp->type == EXPR_PREOP && tmp->op == '&')
996 set_extra_expr_mod(tmp->unop, alloc_estate_whole(get_type(tmp->unop)));
997 else
998 clear_the_pointed_at(tmp);
999 } END_FOR_EACH_PTR(arg);
1002 int values_fit_type(struct expression *left, struct expression *right)
1004 struct range_list *rl;
1005 struct symbol *type;
1007 type = get_type(left);
1008 if (!type)
1009 return 0;
1010 get_absolute_rl(right, &rl);
1011 if (type == rl_type(rl))
1012 return 1;
1013 if (type_unsigned(type) && sval_is_negative(rl_min(rl)))
1014 return 0;
1015 if (sval_cmp(sval_type_min(type), rl_min(rl)) > 0)
1016 return 0;
1017 if (sval_cmp(sval_type_max(type), rl_max(rl)) < 0)
1018 return 0;
1019 return 1;
1022 static void save_chunk_info(struct expression *left, struct expression *right)
1024 struct var_sym_list *vsl;
1025 struct var_sym *vs;
1026 struct expression *add_expr;
1027 struct symbol *type;
1028 sval_t sval;
1029 char *name;
1030 struct symbol *sym;
1032 if (right->type != EXPR_BINOP || right->op != '-')
1033 return;
1034 if (!get_value(right->left, &sval))
1035 return;
1036 if (!expr_to_sym(right->right))
1037 return;
1039 add_expr = binop_expression(left, '+', right->right);
1040 type = get_type(add_expr);
1041 if (!type)
1042 return;
1043 name = expr_to_chunk_sym_vsl(add_expr, &sym, &vsl);
1044 if (!name || !vsl)
1045 goto free;
1046 FOR_EACH_PTR(vsl, vs) {
1047 store_link(link_id, vs->var, vs->sym, name, sym);
1048 } END_FOR_EACH_PTR(vs);
1050 set_state(SMATCH_EXTRA, name, sym, alloc_estate_sval(sval_cast(type, sval)));
1051 free:
1052 free_string(name);
1055 static void do_array_assign(struct expression *left, int op, struct expression *right)
1057 struct range_list *rl;
1059 if (op == '=') {
1060 get_absolute_rl(right, &rl);
1061 rl = cast_rl(get_type(left), rl);
1062 } else {
1063 rl = alloc_whole_rl(get_type(left));
1066 set_extra_array_mod(left, alloc_estate_rl(rl));
1069 static void match_vanilla_assign(struct expression *left, struct expression *right)
1071 struct range_list *orig_rl = NULL;
1072 struct range_list *rl = NULL;
1073 struct symbol *right_sym;
1074 struct symbol *left_type;
1075 struct symbol *right_type;
1076 char *right_name = NULL;
1077 struct symbol *sym;
1078 char *name;
1079 sval_t sval, max;
1080 struct smatch_state *state;
1081 int comparison;
1083 if (is_struct(left))
1084 return;
1086 save_chunk_info(left, right);
1088 name = expr_to_var_sym(left, &sym);
1089 if (!name) {
1090 if (chunk_has_array(left))
1091 do_array_assign(left, '=', right);
1092 return;
1095 left_type = get_type(left);
1096 right_type = get_type(right);
1098 right_name = expr_to_var_sym(right, &right_sym);
1100 if (!__in_fake_assign &&
1101 !(right->type == EXPR_PREOP && right->op == '&') &&
1102 right_name && right_sym &&
1103 values_fit_type(left, strip_expr(right)) &&
1104 !has_symbol(right, sym)) {
1105 set_equiv(left, right);
1106 goto free;
1109 if (get_implied_value(right, &sval)) {
1110 state = alloc_estate_sval(sval_cast(left_type, sval));
1111 goto done;
1114 if (__in_fake_assign) {
1115 struct smatch_state *right_state;
1116 sval_t sval;
1118 if (get_value(right, &sval)) {
1119 sval = sval_cast(left_type, sval);
1120 state = alloc_estate_sval(sval);
1121 goto done;
1124 right_state = get_state(SMATCH_EXTRA, right_name, right_sym);
1125 if (right_state) {
1126 /* simple assignment */
1127 state = clone_estate(right_state);
1128 goto done;
1131 state = alloc_estate_rl(alloc_whole_rl(left_type));
1132 goto done;
1135 comparison = get_comparison_no_extra(left, right);
1136 if (comparison) {
1137 comparison = flip_comparison(comparison);
1138 get_implied_rl(left, &orig_rl);
1141 if (get_implied_rl(right, &rl)) {
1142 rl = cast_rl(left_type, rl);
1143 if (orig_rl)
1144 filter_by_comparison(&rl, comparison, orig_rl);
1145 state = alloc_estate_rl(rl);
1146 if (get_hard_max(right, &max)) {
1147 estate_set_hard_max(state);
1148 estate_set_fuzzy_max(state, max);
1150 } else {
1151 rl = alloc_whole_rl(right_type);
1152 rl = cast_rl(left_type, rl);
1153 if (orig_rl)
1154 filter_by_comparison(&rl, comparison, orig_rl);
1155 state = alloc_estate_rl(rl);
1158 done:
1159 set_extra_mod(name, sym, left, state);
1160 free:
1161 free_string(right_name);
1164 static void match_assign(struct expression *expr)
1166 struct range_list *rl = NULL;
1167 struct expression *left;
1168 struct expression *right;
1169 struct expression *binop_expr;
1170 struct symbol *left_type;
1171 struct symbol *sym;
1172 char *name;
1174 left = strip_expr(expr->left);
1176 right = strip_parens(expr->right);
1177 if (right->type == EXPR_CALL && sym_name_is("__builtin_expect", right->fn))
1178 right = get_argument_from_call_expr(right->args, 0);
1179 while (right->type == EXPR_ASSIGNMENT && right->op == '=')
1180 right = strip_parens(right->left);
1182 if (expr->op == '=' && is_condition(expr->right))
1183 return; /* handled in smatch_condition.c */
1184 if (expr->op == '=' && right->type == EXPR_CALL)
1185 return; /* handled in smatch_function_hooks.c */
1186 if (expr->op == '=') {
1187 match_vanilla_assign(left, right);
1188 return;
1191 name = expr_to_var_sym(left, &sym);
1192 if (!name)
1193 return;
1195 left_type = get_type(left);
1197 switch (expr->op) {
1198 case SPECIAL_ADD_ASSIGN:
1199 case SPECIAL_SUB_ASSIGN:
1200 case SPECIAL_AND_ASSIGN:
1201 case SPECIAL_MOD_ASSIGN:
1202 case SPECIAL_SHL_ASSIGN:
1203 case SPECIAL_SHR_ASSIGN:
1204 case SPECIAL_OR_ASSIGN:
1205 case SPECIAL_XOR_ASSIGN:
1206 case SPECIAL_MUL_ASSIGN:
1207 case SPECIAL_DIV_ASSIGN:
1208 binop_expr = binop_expression(expr->left,
1209 op_remove_assign(expr->op),
1210 expr->right);
1211 get_absolute_rl(binop_expr, &rl);
1212 rl = cast_rl(left_type, rl);
1213 if (inside_loop()) {
1214 if (expr->op == SPECIAL_ADD_ASSIGN)
1215 add_range(&rl, rl_max(rl), sval_type_max(rl_type(rl)));
1217 if (expr->op == SPECIAL_SUB_ASSIGN &&
1218 !sval_is_negative(rl_min(rl))) {
1219 sval_t zero = { .type = rl_type(rl) };
1221 add_range(&rl, rl_min(rl), zero);
1224 set_extra_mod(name, sym, left, alloc_estate_rl(rl));
1225 goto free;
1227 set_extra_mod(name, sym, left, alloc_estate_whole(left_type));
1228 free:
1229 free_string(name);
1232 static struct smatch_state *increment_state(struct smatch_state *state)
1234 sval_t min = estate_min(state);
1235 sval_t max = estate_max(state);
1237 if (!estate_rl(state))
1238 return NULL;
1240 if (inside_loop())
1241 max = sval_type_max(max.type);
1243 if (!sval_is_min(min) && !sval_is_max(min))
1244 min.value++;
1245 if (!sval_is_min(max) && !sval_is_max(max))
1246 max.value++;
1247 return alloc_estate_range(min, max);
1250 static struct smatch_state *decrement_state(struct smatch_state *state)
1252 sval_t min = estate_min(state);
1253 sval_t max = estate_max(state);
1255 if (!estate_rl(state))
1256 return NULL;
1258 if (inside_loop())
1259 min = sval_type_min(min.type);
1261 if (!sval_is_min(min) && !sval_is_max(min))
1262 min.value--;
1263 if (!sval_is_min(max) && !sval_is_max(max))
1264 max.value--;
1265 return alloc_estate_range(min, max);
1268 static void clear_pointed_at_state(struct expression *expr)
1270 struct symbol *type;
1273 * ALERT: This is sort of a mess. If it's is a struct assigment like
1274 * "foo = bar;", then that's handled by smatch_struct_assignment.c.
1275 * the same thing for p++ where "p" is a struct. Most modifications
1276 * are handled by the assignment hook or the db. Smatch_extra.c doesn't
1277 * use smatch_modification.c because we have to get the ordering right
1278 * or something. So if you have p++ where p is a pointer to a standard
1279 * c type then we handle that here. What a mess.
1281 expr = strip_expr(expr);
1282 type = get_type(expr);
1283 if (!type || type->type != SYM_PTR)
1284 return;
1285 type = get_real_base_type(type);
1286 if (!type || type->type != SYM_BASETYPE)
1287 return;
1288 set_extra_expr_nomod(deref_expression(expr), alloc_estate_whole(type));
1291 static void unop_expr(struct expression *expr)
1293 struct smatch_state *state;
1295 if (expr->smatch_flags & Handled)
1296 return;
1298 switch (expr->op) {
1299 case SPECIAL_INCREMENT:
1300 state = get_state_expr(SMATCH_EXTRA, expr->unop);
1301 state = increment_state(state);
1302 if (!state)
1303 state = alloc_estate_whole(get_type(expr));
1304 set_extra_expr_mod(expr->unop, state);
1305 clear_pointed_at_state(expr->unop);
1306 break;
1307 case SPECIAL_DECREMENT:
1308 state = get_state_expr(SMATCH_EXTRA, expr->unop);
1309 state = decrement_state(state);
1310 if (!state)
1311 state = alloc_estate_whole(get_type(expr));
1312 set_extra_expr_mod(expr->unop, state);
1313 clear_pointed_at_state(expr->unop);
1314 break;
1315 default:
1316 return;
1320 static void asm_expr(struct statement *stmt)
1322 struct asm_operand *op;
1323 struct symbol *type;
1325 FOR_EACH_PTR(stmt->asm_outputs, op) {
1326 type = get_type(strip_expr(op->expr));
1327 set_extra_expr_mod(op->expr, alloc_estate_whole(type));
1328 } END_FOR_EACH_PTR(op);
1331 static void check_dereference(struct expression *expr)
1333 struct smatch_state *state;
1335 if (__in_fake_assign)
1336 return;
1337 if (outside_of_function())
1338 return;
1339 state = get_extra_state(expr);
1340 if (state) {
1341 struct range_list *rl;
1343 rl = rl_intersection(estate_rl(state), valid_ptr_rl);
1344 if (rl_equiv(rl, estate_rl(state)))
1345 return;
1346 set_extra_expr_nomod(expr, alloc_estate_rl(rl));
1347 } else {
1348 struct range_list *rl;
1350 if (get_mtag_rl(expr, &rl))
1351 rl = rl_intersection(rl, valid_ptr_rl);
1352 else
1353 rl = clone_rl(valid_ptr_rl);
1355 set_extra_expr_nomod(expr, alloc_estate_rl(rl));
1359 static void match_dereferences(struct expression *expr)
1361 if (expr->type != EXPR_PREOP)
1362 return;
1363 if (getting_address(expr))
1364 return;
1365 /* it's saying that foo[1] = bar dereferences foo[1] */
1366 if (is_array(expr))
1367 return;
1368 check_dereference(expr->unop);
1371 static void match_pointer_as_array(struct expression *expr)
1373 if (!is_array(expr))
1374 return;
1375 check_dereference(get_array_base(expr));
1378 static void find_dereferences(struct expression *expr)
1380 while (expr->type == EXPR_PREOP) {
1381 if (expr->op == '*')
1382 check_dereference(expr->unop);
1383 expr = strip_expr(expr->unop);
1387 static void set_param_dereferenced(struct expression *call, struct expression *arg, char *key, char *unused)
1389 struct symbol *sym;
1390 char *name;
1392 name = get_variable_from_key(arg, key, &sym);
1393 if (name && sym) {
1394 struct smatch_state *orig, *new;
1395 struct range_list *rl;
1397 orig = get_state(SMATCH_EXTRA, name, sym);
1398 if (orig) {
1399 rl = rl_intersection(estate_rl(orig),
1400 alloc_rl(valid_ptr_min_sval,
1401 valid_ptr_max_sval));
1402 new = alloc_estate_rl(rl);
1403 } else {
1404 new = alloc_estate_range(valid_ptr_min_sval, valid_ptr_max_sval);
1407 set_extra_nomod(name, sym, NULL, new);
1409 free_string(name);
1411 find_dereferences(arg);
1414 static sval_t add_one(sval_t sval)
1416 sval.value++;
1417 return sval;
1420 static int handle_postop_inc(struct expression *left, int op, struct expression *right)
1422 struct statement *stmt;
1423 struct expression *cond;
1424 struct smatch_state *true_state, *false_state;
1425 struct symbol *type;
1426 sval_t start;
1427 sval_t limit;
1430 * If we're decrementing here then that's a canonical while count down
1431 * so it's handled already. We're only handling loops like:
1432 * i = 0;
1433 * do { ... } while (i++ < 3);
1436 if (left->type != EXPR_POSTOP || left->op != SPECIAL_INCREMENT)
1437 return 0;
1439 stmt = __cur_stmt->parent;
1440 if (!stmt)
1441 return 0;
1442 if (stmt->type == STMT_COMPOUND)
1443 stmt = stmt->parent;
1444 if (!stmt || stmt->type != STMT_ITERATOR || !stmt->iterator_post_condition)
1445 return 0;
1447 cond = strip_expr(stmt->iterator_post_condition);
1448 if (cond->type != EXPR_COMPARE || cond->op != op)
1449 return 0;
1450 if (left != strip_expr(cond->left) || right != strip_expr(cond->right))
1451 return 0;
1453 if (!get_implied_value(left->unop, &start))
1454 return 0;
1455 if (!get_implied_value(right, &limit))
1456 return 0;
1457 type = get_type(left->unop);
1458 limit = sval_cast(type, limit);
1459 if (sval_cmp(start, limit) > 0)
1460 return 0;
1462 switch (op) {
1463 case '<':
1464 case SPECIAL_UNSIGNED_LT:
1465 break;
1466 case SPECIAL_LTE:
1467 case SPECIAL_UNSIGNED_LTE:
1468 limit = add_one(limit);
1469 default:
1470 return 0;
1474 true_state = alloc_estate_range(add_one(start), limit);
1475 false_state = alloc_estate_range(add_one(limit), add_one(limit));
1477 /* Currently we just discard the false state but when two passes is
1478 * implimented correctly then it will use it.
1481 set_extra_expr_true_false(left->unop, true_state, false_state);
1483 return 1;
1486 bool is_impossible_variable(struct expression *expr)
1488 struct smatch_state *state;
1490 state = get_extra_state(expr);
1491 if (state && !estate_rl(state))
1492 return true;
1493 return false;
1496 static bool in_macro(struct expression *left, struct expression *right)
1498 if (!left || !right)
1499 return 0;
1500 if (left->pos.line != right->pos.line || left->pos.pos != right->pos.pos)
1501 return 0;
1502 if (get_macro_name(left->pos))
1503 return 1;
1504 return 0;
1507 static void handle_comparison(struct symbol *type, struct expression *left, int op, struct expression *right)
1509 struct range_list *left_orig;
1510 struct range_list *left_true;
1511 struct range_list *left_false;
1512 struct range_list *right_orig;
1513 struct range_list *right_true;
1514 struct range_list *right_false;
1515 struct smatch_state *left_true_state;
1516 struct smatch_state *left_false_state;
1517 struct smatch_state *right_true_state;
1518 struct smatch_state *right_false_state;
1519 sval_t dummy, hard_max;
1520 int left_postop = 0;
1521 int right_postop = 0;
1523 if (left->op == SPECIAL_INCREMENT || left->op == SPECIAL_DECREMENT) {
1524 if (left->type == EXPR_POSTOP) {
1525 left->smatch_flags |= Handled;
1526 left_postop = left->op;
1527 if (handle_postop_inc(left, op, right))
1528 return;
1530 left = strip_parens(left->unop);
1532 while (left->type == EXPR_ASSIGNMENT)
1533 left = strip_parens(left->left);
1535 if (right->op == SPECIAL_INCREMENT || right->op == SPECIAL_DECREMENT) {
1536 if (right->type == EXPR_POSTOP) {
1537 right->smatch_flags |= Handled;
1538 right_postop = right->op;
1540 right = strip_parens(right->unop);
1543 if (is_impossible_variable(left) || is_impossible_variable(right))
1544 return;
1546 get_real_absolute_rl(left, &left_orig);
1547 left_orig = cast_rl(type, left_orig);
1549 get_real_absolute_rl(right, &right_orig);
1550 right_orig = cast_rl(type, right_orig);
1552 split_comparison_rl(left_orig, op, right_orig, &left_true, &left_false, &right_true, &right_false);
1554 left_true = rl_truncate_cast(get_type(strip_expr(left)), left_true);
1555 left_false = rl_truncate_cast(get_type(strip_expr(left)), left_false);
1556 right_true = rl_truncate_cast(get_type(strip_expr(right)), right_true);
1557 right_false = rl_truncate_cast(get_type(strip_expr(right)), right_false);
1559 if (!left_true || !left_false) {
1560 struct range_list *tmp_true, *tmp_false;
1562 split_comparison_rl(alloc_whole_rl(type), op, right_orig, &tmp_true, &tmp_false, NULL, NULL);
1563 tmp_true = rl_truncate_cast(get_type(strip_expr(left)), tmp_true);
1564 tmp_false = rl_truncate_cast(get_type(strip_expr(left)), tmp_false);
1565 if (tmp_true && tmp_false)
1566 __save_imaginary_state(left, tmp_true, tmp_false);
1569 if (!right_true || !right_false) {
1570 struct range_list *tmp_true, *tmp_false;
1572 split_comparison_rl(alloc_whole_rl(type), op, right_orig, NULL, NULL, &tmp_true, &tmp_false);
1573 tmp_true = rl_truncate_cast(get_type(strip_expr(right)), tmp_true);
1574 tmp_false = rl_truncate_cast(get_type(strip_expr(right)), tmp_false);
1575 if (tmp_true && tmp_false)
1576 __save_imaginary_state(right, tmp_true, tmp_false);
1579 left_true_state = alloc_estate_rl(left_true);
1580 left_false_state = alloc_estate_rl(left_false);
1581 right_true_state = alloc_estate_rl(right_true);
1582 right_false_state = alloc_estate_rl(right_false);
1584 switch (op) {
1585 case '<':
1586 case SPECIAL_UNSIGNED_LT:
1587 case SPECIAL_UNSIGNED_LTE:
1588 case SPECIAL_LTE:
1589 if (get_implied_value(right, &dummy) && !in_macro(left, right))
1590 estate_set_hard_max(left_true_state);
1591 if (get_implied_value(left, &dummy) && !in_macro(left, right))
1592 estate_set_hard_max(right_false_state);
1593 break;
1594 case '>':
1595 case SPECIAL_UNSIGNED_GT:
1596 case SPECIAL_UNSIGNED_GTE:
1597 case SPECIAL_GTE:
1598 if (get_implied_value(left, &dummy) && !in_macro(left, right))
1599 estate_set_hard_max(right_true_state);
1600 if (get_implied_value(right, &dummy) && !in_macro(left, right))
1601 estate_set_hard_max(left_false_state);
1602 break;
1605 switch (op) {
1606 case '<':
1607 case SPECIAL_UNSIGNED_LT:
1608 case SPECIAL_UNSIGNED_LTE:
1609 case SPECIAL_LTE:
1610 if (get_hard_max(right, &hard_max)) {
1611 if (op == '<' || op == SPECIAL_UNSIGNED_LT)
1612 hard_max.value--;
1613 estate_set_fuzzy_max(left_true_state, hard_max);
1615 if (get_implied_value(right, &hard_max)) {
1616 if (op == SPECIAL_UNSIGNED_LTE ||
1617 op == SPECIAL_LTE)
1618 hard_max.value++;
1619 estate_set_fuzzy_max(left_false_state, hard_max);
1621 if (get_hard_max(left, &hard_max)) {
1622 if (op == SPECIAL_UNSIGNED_LTE ||
1623 op == SPECIAL_LTE)
1624 hard_max.value--;
1625 estate_set_fuzzy_max(right_false_state, hard_max);
1627 if (get_implied_value(left, &hard_max)) {
1628 if (op == '<' || op == SPECIAL_UNSIGNED_LT)
1629 hard_max.value++;
1630 estate_set_fuzzy_max(right_true_state, hard_max);
1632 break;
1633 case '>':
1634 case SPECIAL_UNSIGNED_GT:
1635 case SPECIAL_UNSIGNED_GTE:
1636 case SPECIAL_GTE:
1637 if (get_hard_max(left, &hard_max)) {
1638 if (op == '>' || op == SPECIAL_UNSIGNED_GT)
1639 hard_max.value--;
1640 estate_set_fuzzy_max(right_true_state, hard_max);
1642 if (get_implied_value(left, &hard_max)) {
1643 if (op == SPECIAL_UNSIGNED_GTE ||
1644 op == SPECIAL_GTE)
1645 hard_max.value++;
1646 estate_set_fuzzy_max(right_false_state, hard_max);
1648 if (get_hard_max(right, &hard_max)) {
1649 if (op == SPECIAL_UNSIGNED_LTE ||
1650 op == SPECIAL_LTE)
1651 hard_max.value--;
1652 estate_set_fuzzy_max(left_false_state, hard_max);
1654 if (get_implied_value(right, &hard_max)) {
1655 if (op == '>' ||
1656 op == SPECIAL_UNSIGNED_GT)
1657 hard_max.value++;
1658 estate_set_fuzzy_max(left_true_state, hard_max);
1660 break;
1661 case SPECIAL_EQUAL:
1662 if (get_hard_max(left, &hard_max))
1663 estate_set_fuzzy_max(right_true_state, hard_max);
1664 if (get_hard_max(right, &hard_max))
1665 estate_set_fuzzy_max(left_true_state, hard_max);
1666 break;
1669 if (get_hard_max(left, &hard_max)) {
1670 estate_set_hard_max(left_true_state);
1671 estate_set_hard_max(left_false_state);
1673 if (get_hard_max(right, &hard_max)) {
1674 estate_set_hard_max(right_true_state);
1675 estate_set_hard_max(right_false_state);
1678 if (left_postop == SPECIAL_INCREMENT) {
1679 left_true_state = increment_state(left_true_state);
1680 left_false_state = increment_state(left_false_state);
1682 if (left_postop == SPECIAL_DECREMENT) {
1683 left_true_state = decrement_state(left_true_state);
1684 left_false_state = decrement_state(left_false_state);
1686 if (right_postop == SPECIAL_INCREMENT) {
1687 right_true_state = increment_state(right_true_state);
1688 right_false_state = increment_state(right_false_state);
1690 if (right_postop == SPECIAL_DECREMENT) {
1691 right_true_state = decrement_state(right_true_state);
1692 right_false_state = decrement_state(right_false_state);
1695 if (estate_rl(left_true_state) && estates_equiv(left_true_state, left_false_state)) {
1696 left_true_state = NULL;
1697 left_false_state = NULL;
1700 if (estate_rl(right_true_state) && estates_equiv(right_true_state, right_false_state)) {
1701 right_true_state = NULL;
1702 right_false_state = NULL;
1705 /* Don't introduce new states for known true/false conditions */
1706 if (rl_equiv(left_orig, estate_rl(left_true_state)))
1707 left_true_state = NULL;
1708 if (rl_equiv(left_orig, estate_rl(left_false_state)))
1709 left_false_state = NULL;
1710 if (rl_equiv(right_orig, estate_rl(right_true_state)))
1711 right_true_state = NULL;
1712 if (rl_equiv(right_orig, estate_rl(right_false_state)))
1713 right_false_state = NULL;
1715 set_extra_expr_true_false(left, left_true_state, left_false_state);
1716 set_extra_expr_true_false(right, right_true_state, right_false_state);
1719 static int is_simple_math(struct expression *expr)
1721 if (!expr)
1722 return 0;
1723 if (expr->type != EXPR_BINOP)
1724 return 0;
1725 switch (expr->op) {
1726 case '+':
1727 case '-':
1728 case '*':
1729 return 1;
1731 return 0;
1734 static int flip_op(int op)
1736 /* We only care about simple math */
1737 switch (op) {
1738 case '+':
1739 return '-';
1740 case '-':
1741 return '+';
1742 case '*':
1743 return '/';
1745 return 0;
1748 static void move_known_to_rl(struct expression **expr_p, struct range_list **rl_p)
1750 struct expression *expr = *expr_p;
1751 struct range_list *rl = *rl_p;
1752 sval_t sval;
1754 if (!is_simple_math(expr))
1755 return;
1757 if (get_implied_value(expr->right, &sval)) {
1758 *expr_p = expr->left;
1759 *rl_p = rl_binop(rl, flip_op(expr->op), alloc_rl(sval, sval));
1760 move_known_to_rl(expr_p, rl_p);
1761 return;
1763 if (expr->op == '-')
1764 return;
1765 if (get_implied_value(expr->left, &sval)) {
1766 *expr_p = expr->right;
1767 *rl_p = rl_binop(rl, flip_op(expr->op), alloc_rl(sval, sval));
1768 move_known_to_rl(expr_p, rl_p);
1769 return;
1773 static void move_known_values(struct expression **left_p, struct expression **right_p)
1775 struct expression *left = *left_p;
1776 struct expression *right = *right_p;
1777 sval_t sval, dummy;
1779 if (get_implied_value(left, &sval)) {
1780 if (!is_simple_math(right))
1781 return;
1782 if (get_implied_value(right, &dummy))
1783 return;
1784 if (right->op == '*') {
1785 sval_t divisor;
1787 if (!get_value(right->right, &divisor))
1788 return;
1789 if (divisor.value == 0)
1790 return;
1791 *left_p = binop_expression(left, invert_op(right->op), right->right);
1792 *right_p = right->left;
1793 return;
1795 if (right->op == '+' && get_value(right->left, &sval)) {
1796 *left_p = binop_expression(left, invert_op(right->op), right->left);
1797 *right_p = right->right;
1798 return;
1800 if (get_value(right->right, &sval)) {
1801 *left_p = binop_expression(left, invert_op(right->op), right->right);
1802 *right_p = right->left;
1803 return;
1805 return;
1807 if (get_implied_value(right, &sval)) {
1808 if (!is_simple_math(left))
1809 return;
1810 if (get_implied_value(left, &dummy))
1811 return;
1812 if (left->op == '*') {
1813 sval_t divisor;
1815 if (!get_value(left->right, &divisor))
1816 return;
1817 if (divisor.value == 0)
1818 return;
1819 *right_p = binop_expression(right, invert_op(left->op), left->right);
1820 *left_p = left->left;
1821 return;
1823 if (left->op == '+' && get_value(left->left, &sval)) {
1824 *right_p = binop_expression(right, invert_op(left->op), left->left);
1825 *left_p = left->right;
1826 return;
1829 if (get_value(left->right, &sval)) {
1830 *right_p = binop_expression(right, invert_op(left->op), left->right);
1831 *left_p = left->left;
1832 return;
1834 return;
1839 * The reason for do_simple_algebra() is to solve things like:
1840 * if (foo > 66 || foo + bar > 64) {
1841 * "foo" is not really a known variable so it won't be handled by
1842 * move_known_variables() but it's a super common idiom.
1845 static int do_simple_algebra(struct expression **left_p, struct expression **right_p)
1847 struct expression *left = *left_p;
1848 struct expression *right = *right_p;
1849 struct range_list *rl;
1850 sval_t tmp;
1852 if (left->type != EXPR_BINOP || left->op != '+')
1853 return 0;
1854 if (can_integer_overflow(get_type(left), left))
1855 return 0;
1856 if (!get_implied_value(right, &tmp))
1857 return 0;
1859 if (!get_implied_value(left->left, &tmp) &&
1860 get_implied_rl(left->left, &rl) &&
1861 !is_whole_rl(rl)) {
1862 *right_p = binop_expression(right, '-', left->left);
1863 *left_p = left->right;
1864 return 1;
1866 if (!get_implied_value(left->right, &tmp) &&
1867 get_implied_rl(left->right, &rl) &&
1868 !is_whole_rl(rl)) {
1869 *right_p = binop_expression(right, '-', left->right);
1870 *left_p = left->left;
1871 return 1;
1874 return 0;
1877 static int match_func_comparison(struct expression *expr)
1879 struct expression *left = strip_expr(expr->left);
1880 struct expression *right = strip_expr(expr->right);
1882 if (left->type == EXPR_CALL || right->type == EXPR_CALL) {
1883 function_comparison(left, expr->op, right);
1884 return 1;
1887 return 0;
1890 /* Handle conditions like "if (foo + bar < foo) {" */
1891 static int handle_integer_overflow_test(struct expression *expr)
1893 struct expression *left, *right;
1894 struct symbol *type;
1895 sval_t left_min, right_min, min, max;
1897 if (expr->op != '<' && expr->op != SPECIAL_UNSIGNED_LT)
1898 return 0;
1900 left = strip_parens(expr->left);
1901 right = strip_parens(expr->right);
1903 if (left->op != '+')
1904 return 0;
1906 type = get_type(expr);
1907 if (!type)
1908 return 0;
1909 if (type_positive_bits(type) == 32) {
1910 max.type = &uint_ctype;
1911 max.uvalue = (unsigned int)-1;
1912 } else if (type_positive_bits(type) == 64) {
1913 max.type = &ulong_ctype;
1914 max.value = (unsigned long long)-1;
1915 } else {
1916 return 0;
1919 if (!expr_equiv(left->left, right) && !expr_equiv(left->right, right))
1920 return 0;
1922 get_absolute_min(left->left, &left_min);
1923 get_absolute_min(left->right, &right_min);
1924 min = sval_binop(left_min, '+', right_min);
1926 type = get_type(left);
1927 min = sval_cast(type, min);
1928 max = sval_cast(type, max);
1930 set_extra_chunk_true_false(left, NULL, alloc_estate_range(min, max));
1931 return 1;
1934 static void match_comparison(struct expression *expr)
1936 struct expression *left_orig = strip_parens(expr->left);
1937 struct expression *right_orig = strip_parens(expr->right);
1938 struct expression *left, *right, *tmp;
1939 struct expression *prev;
1940 struct symbol *type;
1941 int redo, count;
1943 if (match_func_comparison(expr))
1944 return;
1946 type = get_type(expr);
1947 if (!type)
1948 type = &llong_ctype;
1950 if (handle_integer_overflow_test(expr))
1951 return;
1953 left = left_orig;
1954 right = right_orig;
1955 move_known_values(&left, &right);
1956 handle_comparison(type, left, expr->op, right);
1958 left = left_orig;
1959 right = right_orig;
1960 if (do_simple_algebra(&left, &right))
1961 handle_comparison(type, left, expr->op, right);
1963 prev = get_assigned_expr(left_orig);
1964 if (is_simple_math(prev) && has_variable(prev, left_orig) == 0) {
1965 left = prev;
1966 right = right_orig;
1967 move_known_values(&left, &right);
1968 handle_comparison(type, left, expr->op, right);
1971 prev = get_assigned_expr(right_orig);
1972 if (is_simple_math(prev) && has_variable(prev, right_orig) == 0) {
1973 left = left_orig;
1974 right = prev;
1975 move_known_values(&left, &right);
1976 handle_comparison(type, left, expr->op, right);
1979 redo = 0;
1980 left = left_orig;
1981 right = right_orig;
1982 if (get_last_expr_from_expression_stmt(left_orig)) {
1983 left = get_last_expr_from_expression_stmt(left_orig);
1984 redo = 1;
1986 if (get_last_expr_from_expression_stmt(right_orig)) {
1987 right = get_last_expr_from_expression_stmt(right_orig);
1988 redo = 1;
1991 if (!redo)
1992 return;
1994 count = 0;
1995 while ((tmp = get_assigned_expr(left))) {
1996 if (count++ > 3)
1997 break;
1998 left = strip_expr(tmp);
2000 count = 0;
2001 while ((tmp = get_assigned_expr(right))) {
2002 if (count++ > 3)
2003 break;
2004 right = strip_expr(tmp);
2007 handle_comparison(type, left, expr->op, right);
2010 static sval_t get_high_mask(sval_t known)
2012 sval_t ret;
2013 int i;
2015 ret = known;
2016 ret.value = 0;
2018 for (i = type_bits(known.type) - 1; i >= 0; i--) {
2019 if (known.uvalue & (1ULL << i))
2020 ret.uvalue |= (1ULL << i);
2021 else
2022 return ret;
2025 return ret;
2028 static bool handle_bit_test(struct expression *expr)
2030 struct range_list *orig_rl, *rl;
2031 struct expression *shift, *mask, *var;
2032 struct bit_info *bit_info;
2033 sval_t sval;
2034 sval_t high = { .type = &int_ctype };
2035 sval_t low = { .type = &int_ctype };
2037 shift = strip_expr(expr->right);
2038 mask = strip_expr(expr->left);
2039 if (shift->type != EXPR_BINOP || shift->op != SPECIAL_LEFTSHIFT) {
2040 shift = strip_expr(expr->left);
2041 mask = strip_expr(expr->right);
2042 if (shift->type != EXPR_BINOP || shift->op != SPECIAL_LEFTSHIFT)
2043 return false;
2045 if (!get_implied_value(shift->left, &sval) || sval.value != 1)
2046 return false;
2047 var = strip_expr(shift->right);
2049 bit_info = get_bit_info(mask);
2050 if (!bit_info)
2051 return false;
2052 if (!bit_info->possible)
2053 return false;
2055 get_absolute_rl(var, &orig_rl);
2056 if (sval_is_negative(rl_min(orig_rl)) ||
2057 rl_max(orig_rl).uvalue > type_bits(get_type(shift->left)))
2058 return false;
2060 low.value = ffsll(bit_info->possible);
2061 high.value = sm_fls64(bit_info->possible);
2062 rl = alloc_rl(low, high);
2063 rl = cast_rl(get_type(var), rl);
2064 rl = rl_intersection(orig_rl, rl);
2065 if (!rl)
2066 return false;
2068 set_extra_expr_true_false(shift->right, alloc_estate_rl(rl), NULL);
2070 return true;
2073 static void handle_AND_op(struct expression *var, sval_t known)
2075 struct range_list *orig_rl;
2076 struct range_list *true_rl = NULL;
2077 struct range_list *false_rl = NULL;
2078 int bit;
2079 sval_t low_mask = known;
2080 sval_t high_mask;
2081 sval_t max;
2083 get_absolute_rl(var, &orig_rl);
2085 if (known.value > 0) {
2086 bit = ffsll(known.value) - 1;
2087 low_mask.uvalue = (1ULL << bit) - 1;
2088 true_rl = remove_range(orig_rl, sval_type_val(known.type, 0), low_mask);
2090 high_mask = get_high_mask(known);
2091 if (high_mask.value) {
2092 bit = ffsll(high_mask.value) - 1;
2093 low_mask.uvalue = (1ULL << bit) - 1;
2095 false_rl = orig_rl;
2096 if (sval_is_negative(rl_min(orig_rl)))
2097 false_rl = remove_range(false_rl, sval_type_min(known.type), sval_type_val(known.type, -1));
2098 false_rl = remove_range(false_rl, low_mask, sval_type_max(known.type));
2099 if (type_signed(high_mask.type) && type_unsigned(rl_type(false_rl))) {
2100 false_rl = remove_range(false_rl,
2101 sval_type_val(rl_type(false_rl), sval_type_max(known.type).uvalue),
2102 sval_type_val(rl_type(false_rl), -1));
2104 } else if (known.value == 1 &&
2105 get_hard_max(var, &max) &&
2106 sval_cmp(max, rl_max(orig_rl)) == 0 &&
2107 max.value & 1) {
2108 false_rl = remove_range(orig_rl, max, max);
2110 set_extra_expr_true_false(var,
2111 true_rl ? alloc_estate_rl(true_rl) : NULL,
2112 false_rl ? alloc_estate_rl(false_rl) : NULL);
2115 static void handle_AND_condition(struct expression *expr)
2117 sval_t known;
2119 if (handle_bit_test(expr))
2120 return;
2122 if (get_implied_value(expr->left, &known))
2123 handle_AND_op(expr->right, known);
2124 else if (get_implied_value(expr->right, &known))
2125 handle_AND_op(expr->left, known);
2128 static void handle_MOD_condition(struct expression *expr)
2130 struct range_list *orig_rl;
2131 struct range_list *true_rl;
2132 struct range_list *false_rl = NULL;
2133 sval_t right;
2134 sval_t zero = {
2135 .value = 0,
2138 if (!get_implied_value(expr->right, &right) || right.value == 0)
2139 return;
2140 get_absolute_rl(expr->left, &orig_rl);
2142 zero.type = rl_type(orig_rl);
2144 /* We're basically dorking around the min and max here */
2145 true_rl = remove_range(orig_rl, zero, zero);
2146 if (!sval_is_max(rl_max(true_rl)) &&
2147 !(rl_max(true_rl).value % right.value))
2148 true_rl = remove_range(true_rl, rl_max(true_rl), rl_max(true_rl));
2150 if (rl_equiv(true_rl, orig_rl))
2151 true_rl = NULL;
2153 if (sval_is_positive(rl_min(orig_rl)) &&
2154 (rl_max(orig_rl).value - rl_min(orig_rl).value) / right.value < 5) {
2155 sval_t add;
2156 int i;
2158 add = rl_min(orig_rl);
2159 add.value += right.value - (add.value % right.value);
2160 add.value -= right.value;
2162 for (i = 0; i < 5; i++) {
2163 add.value += right.value;
2164 if (add.value > rl_max(orig_rl).value)
2165 break;
2166 add_range(&false_rl, add, add);
2168 } else {
2169 if (rl_min(orig_rl).uvalue != 0 &&
2170 rl_min(orig_rl).uvalue < right.uvalue) {
2171 sval_t chop = right;
2172 chop.value--;
2173 false_rl = remove_range(orig_rl, zero, chop);
2176 if (!sval_is_max(rl_max(orig_rl)) &&
2177 (rl_max(orig_rl).value % right.value)) {
2178 sval_t chop = rl_max(orig_rl);
2179 chop.value -= chop.value % right.value;
2180 chop.value++;
2181 if (!false_rl)
2182 false_rl = clone_rl(orig_rl);
2183 false_rl = remove_range(false_rl, chop, rl_max(orig_rl));
2187 set_extra_expr_true_false(expr->left,
2188 true_rl ? alloc_estate_rl(true_rl) : NULL,
2189 false_rl ? alloc_estate_rl(false_rl) : NULL);
2192 /* this is actually hooked from smatch_implied.c... it's hacky, yes */
2193 void __extra_match_condition(struct expression *expr)
2195 expr = strip_expr(expr);
2196 switch (expr->type) {
2197 case EXPR_CALL:
2198 function_comparison(expr, SPECIAL_NOTEQUAL, zero_expr());
2199 return;
2200 case EXPR_PREOP:
2201 case EXPR_SYMBOL:
2202 case EXPR_DEREF:
2203 handle_comparison(get_type(expr), expr, SPECIAL_NOTEQUAL, zero_expr());
2204 return;
2205 case EXPR_COMPARE:
2206 match_comparison(expr);
2207 return;
2208 case EXPR_ASSIGNMENT:
2209 __extra_match_condition(expr->left);
2210 return;
2211 case EXPR_BINOP:
2212 if (expr->op == '&')
2213 handle_AND_condition(expr);
2214 if (expr->op == '%')
2215 handle_MOD_condition(expr);
2216 return;
2220 static void assume_indexes_are_valid(struct expression *expr)
2222 struct expression *array_expr;
2223 int array_size;
2224 struct expression *offset;
2225 struct symbol *offset_type;
2226 struct range_list *rl_before;
2227 struct range_list *rl_after;
2228 struct range_list *filter = NULL;
2229 sval_t size;
2231 expr = strip_expr(expr);
2232 if (!is_array(expr))
2233 return;
2235 offset = get_array_offset(expr);
2236 offset_type = get_type(offset);
2237 if (offset_type && type_signed(offset_type)) {
2238 filter = alloc_rl(sval_type_min(offset_type),
2239 sval_type_val(offset_type, -1));
2242 array_expr = get_array_base(expr);
2243 array_size = get_real_array_size(array_expr);
2244 if (array_size > 1) {
2245 size = sval_type_val(offset_type, array_size);
2246 add_range(&filter, size, sval_type_max(offset_type));
2249 if (!filter)
2250 return;
2251 get_absolute_rl(offset, &rl_before);
2252 rl_after = rl_filter(rl_before, filter);
2253 if (rl_equiv(rl_before, rl_after))
2254 return;
2255 set_extra_expr_nomod(offset, alloc_estate_rl(rl_after));
2258 /* returns 1 if it is not possible for expr to be value, otherwise returns 0 */
2259 int implied_not_equal(struct expression *expr, long long val)
2261 return !possibly_false(expr, SPECIAL_NOTEQUAL, value_expr(val));
2264 int implied_not_equal_name_sym(char *name, struct symbol *sym, long long val)
2266 struct smatch_state *estate;
2268 estate = get_state(SMATCH_EXTRA, name, sym);
2269 if (!estate)
2270 return 0;
2271 if (!rl_has_sval(estate_rl(estate), sval_type_val(estate_type(estate), 0)))
2272 return 1;
2273 return 0;
2276 int parent_is_null_var_sym(const char *name, struct symbol *sym)
2278 char buf[256];
2279 char *start;
2280 char *end;
2281 struct smatch_state *state;
2283 strncpy(buf, name, sizeof(buf) - 1);
2284 buf[sizeof(buf) - 1] = '\0';
2286 start = &buf[0];
2287 while (*start == '*') {
2288 start++;
2289 state = __get_state(SMATCH_EXTRA, start, sym);
2290 if (!state)
2291 continue;
2292 if (!estate_rl(state))
2293 return 1;
2294 if (estate_min(state).value == 0 &&
2295 estate_max(state).value == 0)
2296 return 1;
2299 start = &buf[0];
2300 while (*start == '&')
2301 start++;
2303 while ((end = strrchr(start, '-'))) {
2304 *end = '\0';
2305 state = __get_state(SMATCH_EXTRA, start, sym);
2306 if (!state)
2307 continue;
2308 if (estate_min(state).value == 0 &&
2309 estate_max(state).value == 0)
2310 return 1;
2312 return 0;
2315 int parent_is_null(struct expression *expr)
2317 struct symbol *sym;
2318 char *var;
2319 int ret = 0;
2321 expr = strip_expr(expr);
2322 var = expr_to_var_sym(expr, &sym);
2323 if (!var || !sym)
2324 goto free;
2325 ret = parent_is_null_var_sym(var, sym);
2326 free:
2327 free_string(var);
2328 return ret;
2331 static int param_used_callback(void *found, int argc, char **argv, char **azColName)
2333 *(int *)found = 1;
2334 return 0;
2337 static int is_kzalloc_info(struct sm_state *sm)
2339 sval_t sval;
2342 * kzalloc() information is treated as special because so there is just
2343 * a lot of stuff initialized to zero and it makes building the database
2344 * take hours and hours.
2346 * In theory, we should just remove this line and not pass any unused
2347 * information, but I'm not sure enough that this code works so I want
2348 * to hold off on that for now.
2350 if (!estate_get_single_value(sm->state, &sval))
2351 return 0;
2352 if (sval.value != 0)
2353 return 0;
2354 return 1;
2357 static int is_really_long(struct sm_state *sm)
2359 const char *p;
2360 int cnt = 0;
2362 p = sm->name;
2363 while ((p = strstr(p, "->"))) {
2364 p += 2;
2365 cnt++;
2368 if (cnt < 3 ||
2369 strlen(sm->name) < 40)
2370 return 0;
2371 return 1;
2374 static int filter_unused_param_value_info(struct expression *call, int param, char *printed_name, struct sm_state *sm)
2376 int found = 0;
2378 /* for function pointers assume everything is used */
2379 if (call->fn->type != EXPR_SYMBOL)
2380 return 0;
2382 if (strcmp(printed_name, "$") == 0 ||
2383 strcmp(printed_name, "*$") == 0)
2384 return 0;
2387 * This is to handle __builtin_mul_overflow(). In an ideal world we
2388 * would only need this for invalid code.
2391 if (!call->fn->symbol)
2392 return 0;
2394 if (!is_kzalloc_info(sm) && !is_really_long(sm))
2395 return 0;
2397 run_sql(&param_used_callback, &found,
2398 "select * from return_implies where %s and type = %d and parameter = %d and key = '%s';",
2399 get_static_filter(call->fn->symbol), PARAM_USED, param, printed_name);
2400 if (found)
2401 return 0;
2403 /* If the database is not built yet, then assume everything is used */
2404 run_sql(&param_used_callback, &found,
2405 "select * from return_implies where %s and type = %d;",
2406 get_static_filter(call->fn->symbol), PARAM_USED);
2407 if (!found)
2408 return 0;
2410 return 1;
2413 struct range_list *intersect_with_real_abs_var_sym(const char *name, struct symbol *sym, struct range_list *start)
2415 struct smatch_state *state;
2418 * Here is the difference between implied value and real absolute, say
2419 * you have:
2421 * int a = (u8)x;
2423 * Then you know that a is 0-255. That's real absolute. But you don't
2424 * know for sure that it actually goes up to 255. So it's not implied.
2425 * Implied indicates a degree of certainty.
2427 * But then say you cap "a" at 8. That means you know it goes up to
2428 * 8. So now the implied value is s32min-8. But you can combine it
2429 * with the real absolute to say that actually it's 0-8.
2431 * We are combining it here. But now that I think about it, this is
2432 * probably not the ideal place to combine it because it should proably
2433 * be done earlier. Oh well, this is an improvement on what was there
2434 * before so I'm going to commit this code.
2438 state = get_real_absolute_state_var_sym(name, sym);
2439 if (!state || !estate_rl(state))
2440 return start;
2442 return rl_intersection(estate_rl(state), start);
2445 struct range_list *intersect_with_real_abs_expr(struct expression *expr, struct range_list *start)
2447 struct smatch_state *state;
2448 struct range_list *abs_rl;
2450 state = get_real_absolute_state(expr);
2451 if (!state || !estate_rl(state))
2452 return start;
2454 abs_rl = cast_rl(rl_type(start), estate_rl(state));
2455 return rl_intersection(abs_rl, start);
2458 static void struct_member_callback(struct expression *call, int param, char *printed_name, struct sm_state *sm)
2460 struct range_list *rl;
2461 sval_t dummy;
2463 if (estate_is_whole(sm->state) || !estate_rl(sm->state))
2464 return;
2465 if (filter_unused_param_value_info(call, param, printed_name, sm))
2466 return;
2467 rl = estate_rl(sm->state);
2468 rl = intersect_with_real_abs_var_sym(sm->name, sm->sym, rl);
2469 if (!rl)
2470 return;
2471 sql_insert_caller_info(call, PARAM_VALUE, param, printed_name, show_rl(rl));
2472 if (!estate_get_single_value(sm->state, &dummy)) {
2473 if (estate_has_hard_max(sm->state))
2474 sql_insert_caller_info(call, HARD_MAX, param, printed_name,
2475 sval_to_str(estate_max(sm->state)));
2476 if (estate_has_fuzzy_max(sm->state))
2477 sql_insert_caller_info(call, FUZZY_MAX, param, printed_name,
2478 sval_to_str(estate_get_fuzzy_max(sm->state)));
2482 static void returned_struct_members(int return_id, char *return_ranges, struct expression *expr)
2484 struct symbol *returned_sym;
2485 char *returned_name;
2486 struct sm_state *sm;
2487 char *compare_str;
2488 char name_buf[256];
2489 char val_buf[256];
2490 int len;
2492 // FIXME handle *$
2494 if (!is_pointer(expr))
2495 return;
2497 returned_name = expr_to_var_sym(expr, &returned_sym);
2498 if (!returned_name || !returned_sym)
2499 goto free;
2500 len = strlen(returned_name);
2502 FOR_EACH_MY_SM(my_id, __get_cur_stree(), sm) {
2503 if (!estate_rl(sm->state))
2504 continue;
2505 if (returned_sym != sm->sym)
2506 continue;
2507 if (strncmp(returned_name, sm->name, len) != 0)
2508 continue;
2509 if (sm->name[len] != '-')
2510 continue;
2512 snprintf(name_buf, sizeof(name_buf), "$%s", sm->name + len);
2514 compare_str = name_sym_to_param_comparison(sm->name, sm->sym);
2515 if (!compare_str && estate_is_whole(sm->state))
2516 continue;
2517 snprintf(val_buf, sizeof(val_buf), "%s%s", sm->state->name, compare_str ?: "");
2519 sql_insert_return_states(return_id, return_ranges, PARAM_VALUE,
2520 -1, name_buf, val_buf);
2521 } END_FOR_EACH_SM(sm);
2523 free:
2524 free_string(returned_name);
2527 static void db_limited_before(void)
2529 unmatched_stree = clone_stree(__get_cur_stree());
2532 static void db_limited_after(void)
2534 free_stree(&unmatched_stree);
2537 static int basically_the_same(struct range_list *orig, struct range_list *new)
2539 if (rl_equiv(orig, new))
2540 return 1;
2543 * The whole range is essentially the same as 0,4096-27777777777 so
2544 * don't overwrite the implications just to store that.
2547 if (rl_type(orig)->type == SYM_PTR &&
2548 is_whole_rl(orig) &&
2549 rl_min(new).value == 0 &&
2550 rl_max(new).value == valid_ptr_max)
2551 return 1;
2552 return 0;
2555 static void db_param_limit_binops(struct expression *arg, char *key, struct range_list *rl)
2557 struct range_list *left_rl;
2558 sval_t zero = { .type = rl_type(rl), };
2559 sval_t sval;
2561 if (arg->op != '*')
2562 return;
2563 if (!get_implied_value(arg->right, &sval))
2564 return;
2565 if (can_integer_overflow(get_type(arg), arg))
2566 return;
2568 left_rl = rl_binop(rl, '/', alloc_rl(sval, sval));
2569 if (!rl_has_sval(rl, zero))
2570 left_rl = remove_range(left_rl, zero, zero);
2572 set_extra_expr_nomod(arg->left, alloc_estate_rl(left_rl));
2575 static void db_param_limit_filter(struct expression *expr, int param, char *key, char *value, enum info_type op)
2577 struct expression *arg;
2578 char *name;
2579 struct symbol *sym;
2580 struct var_sym_list *vsl = NULL;
2581 struct sm_state *sm;
2582 struct symbol *compare_type, *var_type;
2583 struct range_list *rl;
2584 struct range_list *limit;
2585 struct range_list *new;
2586 char *other_name;
2587 struct symbol *other_sym;
2589 while (expr->type == EXPR_ASSIGNMENT)
2590 expr = strip_expr(expr->right);
2591 if (expr->type != EXPR_CALL)
2592 return;
2594 arg = get_argument_from_call_expr(expr->args, param);
2595 if (!arg)
2596 return;
2598 if (strcmp(key, "$") == 0)
2599 compare_type = get_arg_type(expr->fn, param);
2600 else
2601 compare_type = get_member_type_from_key(arg, key);
2603 call_results_to_rl(expr, compare_type, value, &limit);
2604 if (strcmp(key, "$") == 0)
2605 move_known_to_rl(&arg, &limit);
2606 name = get_chunk_from_key(arg, key, &sym, &vsl);
2607 if (!name)
2608 return;
2609 if (op != PARAM_LIMIT && !sym)
2610 goto free;
2612 sm = get_sm_state(SMATCH_EXTRA, name, sym);
2613 if (sm)
2614 rl = estate_rl(sm->state);
2615 else
2616 rl = alloc_whole_rl(compare_type);
2618 if (op == PARAM_LIMIT && !rl_fits_in_type(rl, compare_type))
2619 goto free;
2621 new = rl_intersection(rl, limit);
2623 var_type = get_member_type_from_key(arg, key);
2624 new = cast_rl(var_type, new);
2626 /* We want to preserve the implications here */
2627 if (sm && basically_the_same(rl, new))
2628 goto free;
2629 other_name = get_other_name_sym(name, sym, &other_sym);
2631 if (op == PARAM_LIMIT)
2632 set_extra_nomod_vsl(name, sym, vsl, NULL, alloc_estate_rl(new));
2633 else
2634 set_extra_mod(name, sym, NULL, alloc_estate_rl(new));
2636 if (other_name && other_sym) {
2637 if (op == PARAM_LIMIT)
2638 set_extra_nomod_vsl(other_name, other_sym, vsl, NULL, alloc_estate_rl(new));
2639 else
2640 set_extra_mod(other_name, other_sym, NULL, alloc_estate_rl(new));
2643 if (op == PARAM_LIMIT && arg->type == EXPR_BINOP)
2644 db_param_limit_binops(arg, key, new);
2645 free:
2646 free_string(name);
2649 static void db_param_limit(struct expression *expr, int param, char *key, char *value)
2651 db_param_limit_filter(expr, param, key, value, PARAM_LIMIT);
2654 static void db_param_filter(struct expression *expr, int param, char *key, char *value)
2656 db_param_limit_filter(expr, param, key, value, PARAM_FILTER);
2659 static void db_param_add_set(struct expression *expr, int param, char *key, char *value, enum info_type op)
2661 struct expression *arg, *gen_expr;
2662 char *name;
2663 char *other_name = NULL;
2664 struct symbol *sym, *other_sym;
2665 struct symbol *param_type, *arg_type;
2666 struct smatch_state *state;
2667 struct range_list *new = NULL;
2668 struct range_list *added = NULL;
2670 while (expr->type == EXPR_ASSIGNMENT)
2671 expr = strip_expr(expr->right);
2672 if (expr->type != EXPR_CALL)
2673 return;
2675 arg = get_argument_from_call_expr(expr->args, param);
2676 if (!arg)
2677 return;
2679 arg_type = get_arg_type_from_key(expr->fn, param, arg, key);
2680 param_type = get_member_type_from_key(arg, key);
2681 if (param_type && param_type->type == SYM_STRUCT)
2682 return;
2683 name = get_variable_from_key(arg, key, &sym);
2684 if (!name || !sym)
2685 goto free;
2686 gen_expr = gen_expression_from_key(arg, key);
2688 state = get_state(SMATCH_EXTRA, name, sym);
2689 if (state)
2690 new = estate_rl(state);
2692 call_results_to_rl(expr, arg_type, value, &added);
2693 added = cast_rl(param_type, added);
2694 if (op == PARAM_SET)
2695 new = added;
2696 else
2697 new = rl_union(new, added);
2699 other_name = get_other_name_sym_nostack(name, sym, &other_sym);
2700 set_extra_mod(name, sym, gen_expr, alloc_estate_rl(new));
2701 if (other_name && other_sym)
2702 set_extra_mod(other_name, other_sym, gen_expr, alloc_estate_rl(new));
2703 free:
2704 free_string(other_name);
2705 free_string(name);
2708 static void db_param_add(struct expression *expr, int param, char *key, char *value)
2710 in_param_set = true;
2711 db_param_add_set(expr, param, key, value, PARAM_ADD);
2712 in_param_set = false;
2715 static void db_param_set(struct expression *expr, int param, char *key, char *value)
2717 in_param_set = true;
2718 db_param_add_set(expr, param, key, value, PARAM_SET);
2719 in_param_set = false;
2722 static void match_lost_param(struct expression *call, int param)
2724 struct expression *arg;
2726 if (is_const_param(call->fn, param))
2727 return;
2729 arg = get_argument_from_call_expr(call->args, param);
2730 if (!arg)
2731 return;
2733 arg = strip_expr(arg);
2734 if (arg->type == EXPR_PREOP && arg->op == '&')
2735 set_extra_expr_mod(arg->unop, alloc_estate_whole(get_type(arg->unop)));
2736 else
2737 ; /* if pointer then set struct members, maybe?*/
2740 static void db_param_value(struct expression *expr, int param, char *key, char *value)
2742 struct expression *call;
2743 char *name;
2744 struct symbol *sym;
2745 struct symbol *type;
2746 struct range_list *rl = NULL;
2748 if (param != -1)
2749 return;
2751 call = expr;
2752 while (call->type == EXPR_ASSIGNMENT)
2753 call = strip_expr(call->right);
2754 if (call->type != EXPR_CALL)
2755 return;
2757 type = get_member_type_from_key(expr->left, key);
2758 name = get_variable_from_key(expr->left, key, &sym);
2759 if (!name || !sym)
2760 goto free;
2762 call_results_to_rl(call, type, value, &rl);
2764 set_extra_mod(name, sym, NULL, alloc_estate_rl(rl));
2765 free:
2766 free_string(name);
2769 static void match_call_info(struct expression *expr)
2771 struct smatch_state *state;
2772 struct range_list *rl = NULL;
2773 struct expression *arg;
2774 struct symbol *type;
2775 sval_t dummy;
2776 int i = 0;
2778 FOR_EACH_PTR(expr->args, arg) {
2779 type = get_arg_type(expr->fn, i);
2781 get_absolute_rl(arg, &rl);
2782 rl = cast_rl(type, rl);
2784 if (!is_whole_rl(rl)) {
2785 rl = intersect_with_real_abs_expr(arg, rl);
2786 sql_insert_caller_info(expr, PARAM_VALUE, i, "$", show_rl(rl));
2788 state = get_state_expr(SMATCH_EXTRA, arg);
2789 if (!estate_get_single_value(state, &dummy) && estate_has_hard_max(state)) {
2790 sql_insert_caller_info(expr, HARD_MAX, i, "$",
2791 sval_to_str(estate_max(state)));
2793 if (estate_has_fuzzy_max(state)) {
2794 sql_insert_caller_info(expr, FUZZY_MAX, i, "$",
2795 sval_to_str(estate_get_fuzzy_max(state)));
2797 i++;
2798 } END_FOR_EACH_PTR(arg);
2801 static void set_param_value(const char *name, struct symbol *sym, char *key, char *value)
2803 struct expression *expr;
2804 struct range_list *rl = NULL;
2805 struct smatch_state *state;
2806 struct symbol *type;
2807 char *key_orig = key;
2808 char *fullname;
2809 sval_t dummy;
2811 expr = symbol_expression(sym);
2812 fullname = get_variable_from_key(expr, key, NULL);
2813 if (!fullname)
2814 return;
2816 type = get_member_type_from_key(expr, key_orig);
2817 str_to_rl(type, value, &rl);
2818 state = alloc_estate_rl(rl);
2819 if (estate_get_single_value(state, &dummy))
2820 estate_set_hard_max(state);
2821 set_state(SMATCH_EXTRA, fullname, sym, state);
2824 static void set_param_fuzzy_max(const char *name, struct symbol *sym, char *key, char *value)
2826 struct expression *expr;
2827 struct range_list *rl = NULL;
2828 struct smatch_state *state;
2829 struct symbol *type;
2830 char *fullname;
2831 sval_t max;
2833 expr = symbol_expression(sym);
2834 fullname = get_variable_from_key(expr, key, NULL);
2835 if (!fullname)
2836 return;
2838 state = get_state(SMATCH_EXTRA, fullname, sym);
2839 if (!state)
2840 return;
2841 type = estate_type(state);
2842 str_to_rl(type, value, &rl);
2843 if (!rl_to_sval(rl, &max))
2844 return;
2845 estate_set_fuzzy_max(state, max);
2848 static void set_param_hard_max(const char *name, struct symbol *sym, char *key, char *value)
2850 struct smatch_state *state;
2851 struct expression *expr;
2852 char *fullname;
2854 expr = symbol_expression(sym);
2855 fullname = get_variable_from_key(expr, key, NULL);
2856 if (!fullname)
2857 return;
2859 state = get_state(SMATCH_EXTRA, fullname, sym);
2860 if (!state)
2861 return;
2862 estate_set_hard_max(state);
2865 static struct sm_state *get_sm_from_call(struct expression *expr)
2867 char buf[32];
2869 if (is_fake_call(expr))
2870 return NULL;
2872 snprintf(buf, sizeof(buf), "return %p", expr);
2873 return get_sm_state(SMATCH_EXTRA, buf, NULL);
2876 struct sm_state *get_extra_sm_state(struct expression *expr)
2878 char *name;
2879 struct symbol *sym;
2880 struct sm_state *ret = NULL;
2882 expr = strip_expr(expr);
2883 if (!expr)
2884 return NULL;
2886 if (expr->type == EXPR_CALL)
2887 return get_sm_from_call(expr);
2889 name = expr_to_known_chunk_sym(expr, &sym);
2890 if (!name)
2891 goto free;
2893 ret = get_sm_state(SMATCH_EXTRA, name, sym);
2894 free:
2895 free_string(name);
2896 return ret;
2899 struct smatch_state *get_extra_state(struct expression *expr)
2901 struct sm_state *sm;
2903 sm = get_extra_sm_state(expr);
2904 if (!sm)
2905 return NULL;
2906 return sm->state;
2909 void register_smatch_extra(int id)
2911 my_id = id;
2913 set_dynamic_states(my_id);
2914 add_merge_hook(my_id, &merge_estates);
2915 add_unmatched_state_hook(my_id, &unmatched_state);
2916 select_caller_info_hook(set_param_value, PARAM_VALUE);
2917 select_caller_info_hook(set_param_fuzzy_max, FUZZY_MAX);
2918 select_caller_info_hook(set_param_hard_max, HARD_MAX);
2919 select_return_states_before(&db_limited_before);
2920 select_return_states_hook(PARAM_LIMIT, &db_param_limit);
2921 select_return_states_hook(PARAM_FILTER, &db_param_filter);
2922 select_return_states_hook(PARAM_ADD, &db_param_add);
2923 select_return_states_hook(PARAM_SET, &db_param_set);
2924 add_lost_param_hook(&match_lost_param);
2925 select_return_states_hook(PARAM_VALUE, &db_param_value);
2926 select_return_states_after(&db_limited_after);
2929 static void match_link_modify(struct sm_state *sm, struct expression *mod_expr)
2931 struct var_sym_list *links;
2932 struct var_sym *tmp;
2933 struct smatch_state *state;
2935 links = sm->state->data;
2937 FOR_EACH_PTR(links, tmp) {
2938 if (sm->sym == tmp->sym &&
2939 strcmp(sm->name, tmp->var) == 0)
2940 continue;
2941 state = get_state(SMATCH_EXTRA, tmp->var, tmp->sym);
2942 if (!state)
2943 continue;
2944 set_state(SMATCH_EXTRA, tmp->var, tmp->sym, alloc_estate_whole(estate_type(state)));
2945 } END_FOR_EACH_PTR(tmp);
2946 set_state(link_id, sm->name, sm->sym, &undefined);
2949 void register_smatch_extra_links(int id)
2951 link_id = id;
2952 set_dynamic_states(link_id);
2955 void register_smatch_extra_late(int id)
2957 add_merge_hook(link_id, &merge_link_states);
2958 add_modification_hook(link_id, &match_link_modify);
2959 add_hook(&match_dereferences, DEREF_HOOK);
2960 add_hook(&match_pointer_as_array, OP_HOOK);
2961 select_return_implies_hook(DEREFERENCE, &set_param_dereferenced);
2962 add_hook(&match_function_call, FUNCTION_CALL_HOOK);
2963 add_hook(&match_assign, ASSIGNMENT_HOOK);
2964 add_hook(&match_assign, GLOBAL_ASSIGNMENT_HOOK);
2965 add_hook(&unop_expr, OP_HOOK);
2966 add_hook(&asm_expr, ASM_HOOK);
2968 add_hook(&match_call_info, FUNCTION_CALL_HOOK);
2969 add_member_info_callback(my_id, struct_member_callback);
2970 add_split_return_callback(&returned_struct_members);
2972 // add_hook(&assume_indexes_are_valid, OP_HOOK);