comparison: handle preops like "if (++a == b)"
[smatch.git] / smatch_type_val.c
blobc3818b0a3d993a120082a997f37984c437eb9b7b
1 /*
2 * Copyright (C) 2013 Oracle.
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 * The plan here is to save all the possible values store to a given struct
20 * member.
22 * We will load all the values in to the function_type_val table first then
23 * run a script on that and load all the resulting values into the type_val
24 * table.
26 * So in this file we want to take the union of everything assigned to the
27 * struct member and insert it into the function_type_val at the end.
29 * You would think that we could use smatch_modification_hooks.c or
30 * extra_modification_hook() here to get the information here but in the end we
31 * need to code everything again a third time.
35 #include "smatch.h"
36 #include "smatch_slist.h"
37 #include "smatch_extra.h"
39 static int my_id;
41 struct stree_stack *fn_type_val_stack;
42 struct stree *fn_type_val;
43 struct stree *global_type_val;
45 static char *db_vals;
46 static int get_vals(void *unused, int argc, char **argv, char **azColName)
48 db_vals = alloc_string(argv[0]);
49 return 0;
52 static void match_inline_start(struct expression *expr)
54 push_stree(&fn_type_val_stack, fn_type_val);
55 fn_type_val = NULL;
58 static void match_inline_end(struct expression *expr)
60 free_stree(&fn_type_val);
61 fn_type_val = pop_stree(&fn_type_val_stack);
64 int get_db_type_rl(struct expression *expr, struct range_list **rl)
66 char *member;
67 struct range_list *tmp;
69 member = get_member_name(expr);
70 if (!member)
71 return 0;
73 db_vals = NULL;
74 run_sql(get_vals, NULL,
75 "select value from type_value where type = '%s';", member);
76 free_string(member);
77 if (!db_vals)
78 return 0;
79 str_to_rl(&llong_ctype, db_vals, &tmp);
80 tmp = cast_rl(get_type(expr), tmp);
81 if (is_whole_rl(tmp))
82 return 0;
83 *rl = tmp;
84 free_string(db_vals);
86 return 1;
89 static void add_type_val(char *member, struct range_list *rl)
91 struct smatch_state *old, *add, *new;
93 member = alloc_string(member);
94 old = get_state_stree(fn_type_val, my_id, member, NULL);
95 add = alloc_estate_rl(rl);
96 if (old)
97 new = merge_estates(old, add);
98 else
99 new = add;
100 set_state_stree(&fn_type_val, my_id, member, NULL, new);
103 static void add_fake_type_val(char *member, struct range_list *rl, int ignore)
105 struct smatch_state *old, *add, *new;
107 member = alloc_string(member);
108 old = get_state_stree(fn_type_val, my_id, member, NULL);
109 if (old && strcmp(old->name, "min-max") == 0)
110 return;
111 if (ignore && old && strcmp(old->name, "ignore") == 0)
112 return;
113 add = alloc_estate_rl(rl);
114 if (old) {
115 new = merge_estates(old, add);
116 } else {
117 new = add;
118 if (ignore)
119 new->name = alloc_string("ignore");
120 else
121 new->name = alloc_string("min-max");
123 set_state_stree(&fn_type_val, my_id, member, NULL, new);
126 static void add_global_type_val(char *member, struct range_list *rl)
128 struct smatch_state *old, *add, *new;
130 member = alloc_string(member);
131 old = get_state_stree(global_type_val, my_id, member, NULL);
132 add = alloc_estate_rl(rl);
133 if (old)
134 new = merge_estates(old, add);
135 else
136 new = add;
137 new = clone_estate_perm(new);
138 set_state_stree_perm(&global_type_val, my_id, member, NULL, new);
141 static int has_link_cb(void *has_link, int argc, char **argv, char **azColName)
143 *(int *)has_link = 1;
144 return 0;
147 static int is_ignored_fake_assignment(void)
149 struct expression *expr;
150 struct symbol *type;
151 char *member_name;
152 int has_link = 0;
154 expr = get_faked_expression();
155 if (!expr || expr->type != EXPR_ASSIGNMENT)
156 return 0;
157 if (!is_void_pointer(expr->right))
158 return 0;
159 member_name = get_member_name(expr->right);
160 if (!member_name)
161 return 0;
163 type = get_type(expr->left);
164 if (!type || type->type != SYM_PTR)
165 return 0;
166 type = get_real_base_type(type);
167 if (!type || type->type != SYM_STRUCT)
168 return 0;
170 run_sql(has_link_cb, &has_link,
171 "select * from data_info where type = %d and data = '%s' and value = '%s';",
172 TYPE_LINK, member_name, type_to_str(type));
173 return has_link;
176 static int is_ignored_container_of(void)
178 struct expression *expr;
179 char *name;
181 expr = get_faked_expression();
182 if (!expr || expr->type != EXPR_ASSIGNMENT)
183 return 0;
184 name = get_macro_name(expr->right->pos);
185 if (!name)
186 return 0;
187 if (strcmp(name, "container_of") == 0)
188 return 1;
189 return 0;
192 static void match_assign_value(struct expression *expr)
194 char *member, *right_member;
195 struct range_list *rl;
196 struct symbol *type;
198 type = get_type(expr->left);
199 if (type && type->type == SYM_STRUCT)
200 return;
202 member = get_member_name(expr->left);
203 if (!member)
204 return;
206 /* if we're saying foo->mtu = bar->mtu then that doesn't add information */
207 right_member = get_member_name(expr->right);
208 if (right_member && strcmp(right_member, member) == 0)
209 goto free;
211 if (is_fake_call(expr->right)) {
212 if (is_ignored_container_of())
213 goto free;
214 add_fake_type_val(member, alloc_whole_rl(get_type(expr->left)), is_ignored_fake_assignment());
215 goto free;
218 if (expr->op != '=') {
219 add_type_val(member, alloc_whole_rl(get_type(expr->left)));
220 goto free;
222 get_absolute_rl(expr->right, &rl);
223 rl = cast_rl(type, rl);
224 add_type_val(member, rl);
225 free:
226 free_string(right_member);
227 free_string(member);
231 * If we too: int *p = &my_struct->member then abandon all hope of tracking
232 * my_struct->member.
234 static void match_assign_pointer(struct expression *expr)
236 struct expression *right;
237 char *member;
238 struct range_list *rl;
239 struct symbol *type;
241 right = strip_expr(expr->right);
242 if (right->type != EXPR_PREOP || right->op != '&')
243 return;
244 right = strip_expr(right->unop);
246 member = get_member_name(right);
247 if (!member)
248 return;
249 type = get_type(right);
250 rl = alloc_whole_rl(type);
251 add_type_val(member, rl);
252 free_string(member);
255 static void match_global_assign(struct expression *expr)
257 char *member;
258 struct range_list *rl;
260 member = get_member_name(expr->left);
261 if (!member)
262 return;
263 get_absolute_rl(expr->right, &rl);
264 add_global_type_val(member, rl);
265 free_string(member);
268 static void unop_expr(struct expression *expr)
270 struct range_list *rl;
271 char *member;
273 if (expr->op != SPECIAL_DECREMENT && expr->op != SPECIAL_INCREMENT)
274 return;
276 expr = strip_expr(expr->unop);
277 member = get_member_name(expr);
278 if (!member)
279 return;
280 rl = alloc_whole_rl(get_type(expr));
281 add_type_val(member, rl);
282 free_string(member);
285 static void asm_expr(struct statement *stmt)
287 struct expression *expr;
288 struct range_list *rl;
289 char *member;
290 int state = 0;
292 FOR_EACH_PTR(stmt->asm_outputs, expr) {
293 switch (state) {
294 case 0: /* identifier */
295 case 1: /* constraint */
296 state++;
297 continue;
298 case 2: /* expression */
299 state = 0;
300 member = get_member_name(expr);
301 if (!member)
302 continue;
303 rl = alloc_whole_rl(get_type(expr));
304 add_type_val(member, rl);
305 free_string(member);
306 continue;
308 } END_FOR_EACH_PTR(expr);
311 static void db_param_add(struct expression *expr, int param, char *key, char *value)
313 struct expression *arg;
314 struct symbol *type;
315 struct range_list *rl;
316 char *member;
318 if (strcmp(key, "*$") != 0)
319 return;
321 while (expr->type == EXPR_ASSIGNMENT)
322 expr = strip_expr(expr->right);
323 if (expr->type != EXPR_CALL)
324 return;
326 arg = get_argument_from_call_expr(expr->args, param);
327 arg = strip_expr(arg);
328 if (!arg)
329 return;
330 type = get_member_type_from_key(arg, key);
331 if (arg->type != EXPR_PREOP || arg->op != '&')
332 return;
333 arg = strip_expr(arg->unop);
335 member = get_member_name(arg);
336 if (!member)
337 return;
338 call_results_to_rl(expr, type, value, &rl);
339 add_type_val(member, rl);
340 free_string(member);
343 static void match_end_func_info(struct symbol *sym)
345 struct sm_state *sm;
347 FOR_EACH_SM(fn_type_val, sm) {
348 sql_insert_function_type_value(sm->name, sm->state->name);
349 } END_FOR_EACH_SM(sm);
351 free_stree(&fn_type_val);
354 static void match_end_file(struct symbol_list *sym_list)
356 struct sm_state *sm;
358 FOR_EACH_SM(global_type_val, sm) {
359 sql_insert_function_type_value(sm->name, sm->state->name);
360 } END_FOR_EACH_SM(sm);
363 void register_type_val(int id)
365 if (!option_info)
366 return;
368 my_id = id;
370 add_hook(&match_assign_value, ASSIGNMENT_HOOK);
371 add_hook(&match_assign_pointer, ASSIGNMENT_HOOK);
372 add_hook(&unop_expr, OP_HOOK);
373 add_hook(&asm_expr, ASM_HOOK);
374 select_return_states_hook(ADDED_VALUE, &db_param_add);
376 add_hook(&match_inline_start, INLINE_FN_START);
377 add_hook(&match_inline_end, INLINE_FN_END);
379 add_hook(&match_end_func_info, END_FUNC_HOOK);
381 add_hook(&match_global_assign, GLOBAL_ASSIGNMENT_HOOK);
382 add_hook(&match_end_file, END_FILE_HOOK);