0aa17062b4f43ecd1078751675545995a2d83ed6
2 * smatch/smatch_comparison.c
4 * Copyright (C) 2012 Oracle.
6 * Licensed under the Open Software License version 1.1
11 * The point here is to store the relationships between two variables.
13 * To do that we create a state with the two variables in alphabetical order:
14 * ->name = "x vs y" and the state would be "<". On the false path the state
17 * Part of the trick of it is that if x or y is modified then we need to reset
18 * the state. We need to keep a list of all the states which depend on x and
19 * all the states which depend on y. The link_id code handles this.
21 * Future work: If we know that x is greater than y and y is greater than z
22 * then we know that x is greater than z.
26 #include "smatch_extra.h"
27 #include "smatch_slist.h"
29 static int compare_id
;
39 ALLOCATOR(compare_data
, "compare data");
41 int var_sym_eq(const char *a
, struct symbol
*a_sym
, const char *b
, struct symbol
*b_sym
)
45 if (strcmp(a
, b
) == 0)
50 static struct smatch_state
*alloc_compare_state(
51 const char *var1
, struct symbol
*sym1
,
53 const char *var2
, struct symbol
*sym2
)
55 struct smatch_state
*state
;
56 struct compare_data
*data
;
58 state
= __alloc_smatch_state(0);
59 state
->name
= alloc_sname(show_special(comparison
));
60 data
= __alloc_compare_data(0);
61 data
->var1
= alloc_sname(var1
);
63 data
->comparison
= comparison
;
64 data
->var2
= alloc_sname(var2
);
70 static int state_to_comparison(struct smatch_state
*state
)
72 if (!state
|| !state
->data
)
74 return ((struct compare_data
*)state
->data
)->comparison
;
78 * flip_op() reverses the op left and right. So "x >= y" becomes "y <= x".
80 static int flip_op(int op
)
87 case SPECIAL_UNSIGNED_LT
:
88 return SPECIAL_UNSIGNED_GT
;
91 case SPECIAL_UNSIGNED_LTE
:
92 return SPECIAL_UNSIGNED_GTE
;
95 case SPECIAL_NOTEQUAL
:
96 return SPECIAL_NOTEQUAL
;
99 case SPECIAL_UNSIGNED_GTE
:
100 return SPECIAL_UNSIGNED_LTE
;
103 case SPECIAL_UNSIGNED_GT
:
104 return SPECIAL_UNSIGNED_LT
;
106 sm_msg("internal smatch bug. unhandled comparison %d", op
);
111 static int falsify_op(int op
)
118 case SPECIAL_UNSIGNED_LT
:
119 return SPECIAL_UNSIGNED_GTE
;
122 case SPECIAL_UNSIGNED_LTE
:
123 return SPECIAL_UNSIGNED_GT
;
125 return SPECIAL_NOTEQUAL
;
126 case SPECIAL_NOTEQUAL
:
127 return SPECIAL_EQUAL
;
130 case SPECIAL_UNSIGNED_GTE
:
131 return SPECIAL_UNSIGNED_LT
;
134 case SPECIAL_UNSIGNED_GT
:
135 return SPECIAL_UNSIGNED_LTE
;
137 sm_msg("internal smatch bug. unhandled comparison %d", op
);
142 static int rl_comparison(struct range_list
*left_rl
, struct range_list
*right_rl
)
144 sval_t left_min
, left_max
, right_min
, right_max
;
146 if (!left_rl
|| !right_rl
)
149 left_min
= rl_min(left_rl
);
150 left_max
= rl_max(left_rl
);
151 right_min
= rl_min(right_rl
);
152 right_max
= rl_max(right_rl
);
154 if (left_min
.value
== left_max
.value
&&
155 right_min
.value
== right_max
.value
&&
156 left_min
.value
== right_min
.value
)
157 return SPECIAL_EQUAL
;
159 if (sval_cmp(left_max
, right_min
) < 0)
161 if (sval_cmp(left_max
, right_min
) == 0)
163 if (sval_cmp(left_min
, right_max
) > 0)
165 if (sval_cmp(left_min
, right_max
) == 0)
171 static struct range_list
*get_orig_rl(struct symbol
*sym
)
173 struct smatch_state
*state
;
175 if (!sym
|| !sym
->ident
)
177 state
= get_orig_estate(sym
->ident
->name
, sym
);
178 return estate_rl(state
);
181 static struct smatch_state
*unmatched_comparison(struct sm_state
*sm
)
183 struct compare_data
*data
= sm
->state
->data
;
184 struct range_list
*left_rl
, *right_rl
;
190 if (strstr(data
->var1
, " orig"))
191 left_rl
= get_orig_rl(data
->sym1
);
192 else if (!get_implied_rl_var_sym(data
->var1
, data
->sym1
, &left_rl
))
194 if (strstr(data
->var2
, " orig"))
195 right_rl
= get_orig_rl(data
->sym2
);
196 else if (!get_implied_rl_var_sym(data
->var2
, data
->sym2
, &right_rl
))
199 op
= rl_comparison(left_rl
, right_rl
);
201 return alloc_compare_state(data
->var1
, data
->sym1
, op
, data
->var2
, data
->sym2
);
206 /* remove_unsigned_from_comparison() is obviously a hack. */
207 static int remove_unsigned_from_comparison(int op
)
210 case SPECIAL_UNSIGNED_LT
:
212 case SPECIAL_UNSIGNED_LTE
:
214 case SPECIAL_UNSIGNED_GTE
:
216 case SPECIAL_UNSIGNED_GT
:
224 * This is for when you merge states "a < b" and "a == b", the result is that
225 * we can say for sure, "a <= b" after the merge.
227 static int merge_comparisons(int one
, int two
)
231 one
= remove_unsigned_from_comparison(one
);
232 two
= remove_unsigned_from_comparison(two
);
279 return SPECIAL_NOTEQUAL
;
290 * This is for if you have "a < b" and "b <= c". Or in other words,
291 * "a < b <= c". You would call this like get_combined_comparison('<', '<=').
292 * The return comparison would be '<'.
294 * This function is different from merge_comparisons(), for example:
295 * merge_comparison('<', '==') returns '<='
296 * get_combined_comparison('<', '==') returns '<'
298 static int combine_comparisons(int left_compare
, int right_compare
)
302 left_compare
= remove_unsigned_from_comparison(left_compare
);
303 right_compare
= remove_unsigned_from_comparison(right_compare
);
307 switch (left_compare
) {
316 return right_compare
;
325 switch (right_compare
) {
357 static struct smatch_state
*merge_compare_states(struct smatch_state
*s1
, struct smatch_state
*s2
)
359 struct compare_data
*data
= s1
->data
;
362 op
= merge_comparisons(state_to_comparison(s1
), state_to_comparison(s2
));
364 return alloc_compare_state(data
->var1
, data
->sym1
, op
, data
->var2
, data
->sym2
);
368 struct smatch_state
*alloc_link_state(struct string_list
*links
)
370 struct smatch_state
*state
;
371 static char buf
[256];
375 state
= __alloc_smatch_state(0);
378 FOR_EACH_PTR(links
, tmp
) {
380 snprintf(buf
, sizeof(buf
), "%s", tmp
);
382 snprintf(buf
, sizeof(buf
), "%s, %s", buf
, tmp
);
383 } END_FOR_EACH_PTR(tmp
);
385 state
->name
= alloc_sname(buf
);
390 static void save_start_states(struct statement
*stmt
)
392 struct symbol
*param
;
394 char state_name
[128];
395 struct smatch_state
*state
;
396 struct string_list
*links
;
399 FOR_EACH_PTR(cur_func_sym
->ctype
.base_type
->arguments
, param
) {
402 snprintf(orig
, sizeof(orig
), "%s orig", param
->ident
->name
);
403 snprintf(state_name
, sizeof(state_name
), "%s vs %s", param
->ident
->name
, orig
);
404 state
= alloc_compare_state(param
->ident
->name
, param
, SPECIAL_EQUAL
, alloc_sname(orig
), param
);
405 set_state(compare_id
, state_name
, NULL
, state
);
407 link
= alloc_sname(state_name
);
409 insert_string(&links
, link
);
410 state
= alloc_link_state(links
);
411 set_state(link_id
, param
->ident
->name
, param
, state
);
412 } END_FOR_EACH_PTR(param
);
415 static struct smatch_state
*merge_links(struct smatch_state
*s1
, struct smatch_state
*s2
)
417 struct smatch_state
*ret
;
418 struct string_list
*links
;
420 links
= combine_string_lists(s1
->data
, s2
->data
);
421 ret
= alloc_link_state(links
);
425 static void save_link_var_sym(const char *var
, struct symbol
*sym
, const char *link
)
427 struct smatch_state
*old_state
, *new_state
;
428 struct string_list
*links
;
431 old_state
= get_state(link_id
, var
, sym
);
433 links
= clone_str_list(old_state
->data
);
437 new = alloc_sname(link
);
438 insert_string(&links
, new);
440 new_state
= alloc_link_state(links
);
441 set_state(link_id
, var
, sym
, new_state
);
444 static void match_inc(struct sm_state
*sm
)
446 struct string_list
*links
;
447 struct smatch_state
*state
;
450 links
= sm
->state
->data
;
452 FOR_EACH_PTR(links
, tmp
) {
453 state
= get_state(compare_id
, tmp
, NULL
);
455 switch (state_to_comparison(state
)) {
458 case SPECIAL_UNSIGNED_GTE
:
460 case SPECIAL_UNSIGNED_GT
: {
461 struct compare_data
*data
= state
->data
;
462 struct smatch_state
*new;
464 new = alloc_compare_state(data
->var1
, data
->sym1
, '>', data
->var2
, data
->sym2
);
465 set_state(compare_id
, tmp
, NULL
, new);
469 set_state(compare_id
, tmp
, NULL
, &undefined
);
471 } END_FOR_EACH_PTR(tmp
);
474 static void match_dec(struct sm_state
*sm
)
476 struct string_list
*links
;
477 struct smatch_state
*state
;
480 links
= sm
->state
->data
;
482 FOR_EACH_PTR(links
, tmp
) {
483 state
= get_state(compare_id
, tmp
, NULL
);
485 switch (state_to_comparison(state
)) {
488 case SPECIAL_UNSIGNED_LTE
:
490 case SPECIAL_UNSIGNED_LT
: {
491 struct compare_data
*data
= state
->data
;
492 struct smatch_state
*new;
494 new = alloc_compare_state(data
->var1
, data
->sym1
, '<', data
->var2
, data
->sym2
);
495 set_state(compare_id
, tmp
, NULL
, new);
499 set_state(compare_id
, tmp
, NULL
, &undefined
);
501 } END_FOR_EACH_PTR(tmp
);
504 static int match_inc_dec(struct sm_state
*sm
, struct expression
*mod_expr
)
508 if (mod_expr
->type
!= EXPR_PREOP
&& mod_expr
->type
!= EXPR_POSTOP
)
511 if (mod_expr
->op
== SPECIAL_INCREMENT
) {
515 if (mod_expr
->op
== SPECIAL_DECREMENT
) {
522 static void match_modify(struct sm_state
*sm
, struct expression
*mod_expr
)
524 struct string_list
*links
;
527 if (match_inc_dec(sm
, mod_expr
))
530 links
= sm
->state
->data
;
532 FOR_EACH_PTR(links
, tmp
) {
533 set_state(compare_id
, tmp
, NULL
, &undefined
);
534 } END_FOR_EACH_PTR(tmp
);
535 set_state(link_id
, sm
->name
, sm
->sym
, &undefined
);
538 static char *chunk_to_var_sym(struct expression
*expr
, struct symbol
**sym
)
540 char *name
, *left_name
, *right_name
;
544 expr
= strip_expr(expr
);
548 name
= expr_to_var_sym(expr
, &tmp
);
557 if (expr
->type
!= EXPR_BINOP
)
559 if (expr
->op
!= '-' && expr
->op
!= '+')
562 left_name
= expr_to_var(expr
->left
);
565 right_name
= expr_to_var(expr
->right
);
567 free_string(left_name
);
570 snprintf(buf
, sizeof(buf
), "%s %s %s", left_name
, show_special(expr
->op
), right_name
);
571 free_string(left_name
);
572 free_string(right_name
);
573 return alloc_string(buf
);
576 static char *chunk_to_var(struct expression
*expr
)
578 return chunk_to_var_sym(expr
, NULL
);
581 static void save_link(struct expression
*expr
, char *link
)
586 expr
= strip_expr(expr
);
587 if (expr
->type
== EXPR_BINOP
) {
590 chunk
= chunk_to_var(expr
);
594 save_link(expr
->left
, link
);
595 save_link(expr
->right
, link
);
596 save_link_var_sym(chunk
, NULL
, link
);
600 var
= expr_to_var_sym(expr
, &sym
);
604 save_link_var_sym(var
, sym
, link
);
611 static void update_tf_links(const char *left_var
, struct symbol
*left_sym
,
613 const char *mid_var
, struct symbol
*mid_sym
,
614 struct string_list
*links
)
616 struct smatch_state
*state
;
617 struct smatch_state
*true_state
, *false_state
;
618 struct compare_data
*data
;
619 const char *right_var
;
620 struct symbol
*right_sym
;
621 int right_comparison
;
623 int false_comparison
;
625 char state_name
[256];
627 FOR_EACH_PTR(links
, tmp
) {
628 state
= get_state(compare_id
, tmp
, NULL
);
629 if (!state
|| !state
->data
)
632 right_comparison
= data
->comparison
;
633 right_var
= data
->var2
;
634 right_sym
= data
->sym2
;
635 if (var_sym_eq(mid_var
, mid_sym
, right_var
, right_sym
)) {
636 right_var
= data
->var1
;
637 right_sym
= data
->sym1
;
638 right_comparison
= flip_op(right_comparison
);
640 true_comparison
= combine_comparisons(left_comparison
, right_comparison
);
641 false_comparison
= combine_comparisons(falsify_op(left_comparison
), right_comparison
);
643 if (strcmp(left_var
, right_var
) > 0) {
644 struct symbol
*tmp_sym
= left_sym
;
645 const char *tmp_var
= left_var
;
647 left_var
= right_var
;
648 left_sym
= right_sym
;
651 true_comparison
= flip_op(true_comparison
);
652 false_comparison
= flip_op(false_comparison
);
655 if (!true_comparison
&& !false_comparison
)
659 true_state
= alloc_compare_state(left_var
, left_sym
, true_comparison
, right_var
, right_sym
);
662 if (false_comparison
)
663 false_state
= alloc_compare_state(left_var
, left_sym
, false_comparison
, right_var
, right_sym
);
667 snprintf(state_name
, sizeof(state_name
), "%s vs %s", left_var
, right_var
);
668 set_true_false_states(compare_id
, state_name
, NULL
, true_state
, false_state
);
669 save_link_var_sym(left_var
, left_sym
, state_name
);
670 save_link_var_sym(right_var
, right_sym
, state_name
);
671 } END_FOR_EACH_PTR(tmp
);
674 static void update_tf_data(struct compare_data
*tdata
)
676 struct smatch_state
*state
;
678 state
= get_state(link_id
, tdata
->var2
, tdata
->sym2
);
680 update_tf_links(tdata
->var1
, tdata
->sym1
, tdata
->comparison
, tdata
->var2
, tdata
->sym2
, state
->data
);
682 state
= get_state(link_id
, tdata
->var1
, tdata
->sym1
);
684 update_tf_links(tdata
->var2
, tdata
->sym2
, flip_op(tdata
->comparison
), tdata
->var1
, tdata
->sym1
, state
->data
);
687 static void match_compare(struct expression
*expr
)
691 struct symbol
*left_sym
, *right_sym
;
693 struct smatch_state
*true_state
, *false_state
;
694 char state_name
[256];
696 if (expr
->type
!= EXPR_COMPARE
)
698 left
= chunk_to_var_sym(expr
->left
, &left_sym
);
701 right
= chunk_to_var_sym(expr
->right
, &right_sym
);
705 if (strcmp(left
, right
) > 0) {
706 struct symbol
*tmp_sym
= left_sym
;
707 char *tmp_name
= left
;
710 left_sym
= right_sym
;
713 op
= flip_op(expr
->op
);
717 false_op
= falsify_op(op
);
718 snprintf(state_name
, sizeof(state_name
), "%s vs %s", left
, right
);
719 true_state
= alloc_compare_state(left
, left_sym
, op
, right
, right_sym
);
720 false_state
= alloc_compare_state(left
, left_sym
, false_op
, right
, right_sym
);
722 update_tf_data(true_state
->data
);
724 set_true_false_states(compare_id
, state_name
, NULL
, true_state
, false_state
);
725 save_link(expr
->left
, state_name
);
726 save_link(expr
->right
, state_name
);
732 static void add_comparison_var_sym(const char *left_name
, struct symbol
*left_sym
, int comparison
, const char *right_name
, struct symbol
*right_sym
)
734 struct smatch_state
*state
;
735 char state_name
[256];
737 if (strcmp(left_name
, right_name
) > 0) {
738 struct symbol
*tmp_sym
= left_sym
;
739 const char *tmp_name
= left_name
;
741 left_name
= right_name
;
742 left_sym
= right_sym
;
743 right_name
= tmp_name
;
745 comparison
= flip_op(comparison
);
747 snprintf(state_name
, sizeof(state_name
), "%s vs %s", left_name
, right_name
);
748 state
= alloc_compare_state(left_name
, left_sym
, comparison
, right_name
, right_sym
);
750 set_state(compare_id
, state_name
, NULL
, state
);
751 save_link_var_sym(left_name
, left_sym
, state_name
);
752 save_link_var_sym(right_name
, right_sym
, state_name
);
755 static void add_comparison(struct expression
*left
, int comparison
, struct expression
*right
)
757 char *left_name
= NULL
;
758 char *right_name
= NULL
;
759 struct symbol
*left_sym
, *right_sym
;
761 left_name
= chunk_to_var_sym(left
, &left_sym
);
764 right_name
= chunk_to_var_sym(right
, &right_sym
);
768 add_comparison_var_sym(left_name
, left_sym
, comparison
, right_name
, right_sym
);
771 free_string(left_name
);
772 free_string(right_name
);
775 static void match_assign_add(struct expression
*expr
)
777 struct expression
*right
;
778 struct expression
*r_left
, *r_right
;
779 sval_t left_tmp
, right_tmp
;
781 right
= strip_expr(expr
->right
);
782 r_left
= strip_expr(right
->left
);
783 r_right
= strip_expr(right
->right
);
785 get_absolute_min(r_left
, &left_tmp
);
786 get_absolute_min(r_right
, &right_tmp
);
788 if (left_tmp
.value
> 0)
789 add_comparison(expr
->left
, '>', r_right
);
790 else if (left_tmp
.value
== 0)
791 add_comparison(expr
->left
, SPECIAL_GTE
, r_right
);
793 if (right_tmp
.value
> 0)
794 add_comparison(expr
->left
, '>', r_left
);
795 else if (right_tmp
.value
== 0)
796 add_comparison(expr
->left
, SPECIAL_GTE
, r_left
);
799 static void match_assign_sub(struct expression
*expr
)
801 struct expression
*right
;
802 struct expression
*r_left
, *r_right
;
806 right
= strip_expr(expr
->right
);
807 r_left
= strip_expr(right
->left
);
808 r_right
= strip_expr(right
->right
);
810 if (get_absolute_min(r_right
, &min
) && sval_is_negative(min
))
813 comparison
= get_comparison(r_left
, r_right
);
815 switch (comparison
) {
818 if (implied_not_equal(r_right
, 0))
819 add_comparison(expr
->left
, '>', r_left
);
821 add_comparison(expr
->left
, SPECIAL_GTE
, r_left
);
826 static void match_assign_divide(struct expression
*expr
)
828 struct expression
*right
;
829 struct expression
*r_left
, *r_right
;
832 right
= strip_expr(expr
->right
);
833 r_left
= strip_expr(right
->left
);
834 r_right
= strip_expr(right
->right
);
835 if (!get_implied_min(r_right
, &min
) || min
.value
<= 1)
838 add_comparison(expr
->left
, '<', r_left
);
841 static void match_binop_assign(struct expression
*expr
)
843 struct expression
*right
;
845 right
= strip_expr(expr
->right
);
846 if (right
->op
== '+')
847 match_assign_add(expr
);
848 if (right
->op
== '-')
849 match_assign_sub(expr
);
850 if (right
->op
== '/')
851 match_assign_divide(expr
);
854 static void copy_comparisons(struct expression
*left
, struct expression
*right
)
856 struct string_list
*links
;
857 struct smatch_state
*state
;
858 struct compare_data
*data
;
859 struct symbol
*left_sym
, *right_sym
;
860 char *left_var
= NULL
;
861 char *right_var
= NULL
;
867 left_var
= chunk_to_var_sym(left
, &left_sym
);
870 right_var
= chunk_to_var_sym(right
, &right_sym
);
874 state
= get_state(link_id
, right_var
, right_sym
);
879 FOR_EACH_PTR(links
, tmp
) {
880 state
= get_state(compare_id
, tmp
, NULL
);
881 if (!state
|| !state
->data
)
884 comparison
= data
->comparison
;
887 if (var_sym_eq(var
, sym
, right_var
, right_sym
)) {
890 comparison
= flip_op(comparison
);
892 add_comparison_var_sym(left_var
, left_sym
, comparison
, var
, sym
);
893 } END_FOR_EACH_PTR(tmp
);
896 free_string(right_var
);
899 static void match_assign(struct expression
*expr
)
901 struct expression
*right
;
906 copy_comparisons(expr
->left
, expr
->right
);
907 add_comparison(expr
->left
, SPECIAL_EQUAL
, expr
->right
);
909 right
= strip_expr(expr
->right
);
910 if (right
->type
== EXPR_BINOP
)
911 match_binop_assign(expr
);
914 static int get_comparison_strings(char *one
, char *two
)
917 struct smatch_state
*state
;
921 if (strcmp(one
, two
) > 0) {
929 snprintf(buf
, sizeof(buf
), "%s vs %s", one
, two
);
930 state
= get_state(compare_id
, buf
, NULL
);
932 ret
= state_to_comparison(state
);
940 int get_comparison(struct expression
*a
, struct expression
*b
)
946 one
= chunk_to_var(a
);
949 two
= chunk_to_var(b
);
953 ret
= get_comparison_strings(one
, two
);
960 void __add_comparison_info(struct expression
*expr
, struct expression
*call
, const char *range
)
962 struct expression
*arg
;
964 const char *c
= range
;
966 if (!str_to_comparison_arg(c
, call
, &comparison
, &arg
))
968 add_comparison(expr
, comparison
, arg
);
971 static char *range_comparison_to_param_helper(struct expression
*expr
, char starts_with
)
973 struct symbol
*param
;
976 char *ret_str
= NULL
;
980 var
= chunk_to_var(expr
);
985 FOR_EACH_PTR(cur_func_sym
->ctype
.base_type
->arguments
, param
) {
989 snprintf(buf
, sizeof(buf
), "%s orig", param
->ident
->name
);
990 compare
= get_comparison_strings(var
, buf
);
993 if (show_special(compare
)[0] != starts_with
)
995 snprintf(buf
, sizeof(buf
), "[%sp%d]", show_special(compare
), i
);
996 ret_str
= alloc_sname(buf
);
998 } END_FOR_EACH_PTR(param
);
1005 char *expr_equal_to_param(struct expression
*expr
)
1007 return range_comparison_to_param_helper(expr
, '=');
1010 char *expr_lte_to_param(struct expression
*expr
)
1012 return range_comparison_to_param_helper(expr
, '<');
1015 static void free_data(struct symbol
*sym
)
1019 clear_compare_data_alloc();
1022 void register_comparison(int id
)
1025 add_hook(&match_compare
, CONDITION_HOOK
);
1026 add_hook(&match_assign
, ASSIGNMENT_HOOK
);
1027 add_hook(&save_start_states
, AFTER_DEF_HOOK
);
1028 add_unmatched_state_hook(compare_id
, unmatched_comparison
);
1029 add_merge_hook(compare_id
, &merge_compare_states
);
1030 add_hook(&free_data
, AFTER_FUNC_HOOK
);
1033 void register_comparison_links(int id
)
1036 add_merge_hook(link_id
, &merge_links
);
1037 add_modification_hook(link_id
, &match_modify
);