vim_db: handle older versions of vim
[smatch.git] / smatch_type_val.c
blob51ef641f5a46add7693ef5013b838b3eda14ddfc
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;
68 struct symbol *type;
70 member = get_member_name(expr);
71 if (!member)
72 return 0;
74 db_vals = NULL;
75 run_sql(get_vals, NULL,
76 "select value from type_value where type = '%s';", member);
77 free_string(member);
78 if (!db_vals)
79 return 0;
80 type = get_type(expr);
81 str_to_rl(type, db_vals, &tmp);
82 free_string(db_vals);
83 if (is_whole_rl(tmp))
84 return 0;
85 *rl = tmp;
87 return 1;
90 static void add_type_val(char *member, struct range_list *rl)
92 struct smatch_state *old, *add, *new;
94 member = alloc_string(member);
95 old = get_state_stree(fn_type_val, my_id, member, NULL);
96 add = alloc_estate_rl(rl);
97 if (old)
98 new = merge_estates(old, add);
99 else
100 new = add;
101 set_state_stree(&fn_type_val, my_id, member, NULL, new);
104 static void add_fake_type_val(char *member, struct range_list *rl, int ignore)
106 struct smatch_state *old, *add, *new;
108 member = alloc_string(member);
109 old = get_state_stree(fn_type_val, my_id, member, NULL);
110 if (old && strcmp(old->name, "min-max") == 0)
111 return;
112 if (ignore && old && strcmp(old->name, "ignore") == 0)
113 return;
114 add = alloc_estate_rl(rl);
115 if (old) {
116 new = merge_estates(old, add);
117 } else {
118 new = add;
119 if (ignore)
120 new->name = alloc_string("ignore");
121 else
122 new->name = alloc_string("min-max");
124 set_state_stree(&fn_type_val, my_id, member, NULL, new);
127 static void add_global_type_val(char *member, struct range_list *rl)
129 struct smatch_state *old, *add, *new;
131 member = alloc_string(member);
132 old = get_state_stree(global_type_val, my_id, member, NULL);
133 add = alloc_estate_rl(rl);
134 if (old)
135 new = merge_estates(old, add);
136 else
137 new = add;
138 new = clone_estate_perm(new);
139 set_state_stree_perm(&global_type_val, my_id, member, NULL, new);
142 static int has_link_cb(void *has_link, int argc, char **argv, char **azColName)
144 *(int *)has_link = 1;
145 return 0;
148 static int is_ignored_fake_assignment(void)
150 struct expression *expr;
151 struct symbol *type;
152 char *member_name;
153 int has_link = 0;
155 expr = get_faked_expression();
156 if (!expr || expr->type != EXPR_ASSIGNMENT)
157 return 0;
158 if (!is_void_pointer(expr->right))
159 return 0;
160 member_name = get_member_name(expr->right);
161 if (!member_name)
162 return 0;
164 type = get_type(expr->left);
165 if (!type || type->type != SYM_PTR)
166 return 0;
167 type = get_real_base_type(type);
168 if (!type || type->type != SYM_STRUCT)
169 return 0;
171 run_sql(has_link_cb, &has_link,
172 "select * from data_info where type = %d and data = '%s' and value = '%s';",
173 TYPE_LINK, member_name, type_to_str(type));
174 return has_link;
177 static int is_ignored_macro(void)
179 struct expression *expr;
180 char *name;
182 expr = get_faked_expression();
183 if (!expr || expr->type != EXPR_ASSIGNMENT)
184 return 0;
185 name = get_macro_name(expr->right->pos);
186 if (!name)
187 return 0;
188 if (strcmp(name, "container_of") == 0)
189 return 1;
190 if (strcmp(name, "rb_entry") == 0)
191 return 1;
192 if (strcmp(name, "list_entry") == 0)
193 return 1;
194 if (strcmp(name, "list_first_entry") == 0)
195 return 1;
196 if (strcmp(name, "hlist_entry") == 0)
197 return 1;
198 if (strstr(name, "for_each"))
199 return 1;
200 return 0;
203 static int is_ignored_function(void)
205 struct expression *expr;
207 expr = get_faked_expression();
208 if (!expr || expr->type != EXPR_ASSIGNMENT)
209 return 0;
210 expr = strip_expr(expr->right);
211 if (!expr || expr->type != EXPR_CALL || expr->fn->type != EXPR_SYMBOL)
212 return 0;
214 if (sym_name_is("kmalloc", expr->fn))
215 return 1;
216 if (sym_name_is("netdev_priv", expr->fn))
217 return 1;
219 return 0;
222 static int is_uncasted_function(void)
224 struct expression *expr;
226 expr = get_faked_expression();
227 if (!expr || expr->type != EXPR_ASSIGNMENT)
228 return 0;
229 if (expr->right->type == EXPR_CALL)
230 return 1;
231 return 0;
234 static void match_assign_value(struct expression *expr)
236 char *member, *right_member;
237 struct range_list *rl;
238 struct symbol *type;
240 type = get_type(expr->left);
241 if (type && type->type == SYM_STRUCT)
242 return;
244 member = get_member_name(expr->left);
245 if (!member)
246 return;
248 /* if we're saying foo->mtu = bar->mtu then that doesn't add information */
249 right_member = get_member_name(expr->right);
250 if (right_member && strcmp(right_member, member) == 0)
251 goto free;
253 if (is_fake_call(expr->right)) {
254 if (is_ignored_macro())
255 goto free;
256 if (is_ignored_function())
257 goto free;
258 if (is_uncasted_function())
259 goto free;
260 add_fake_type_val(member, alloc_whole_rl(get_type(expr->left)), is_ignored_fake_assignment());
261 goto free;
264 if (expr->op != '=') {
265 add_type_val(member, alloc_whole_rl(get_type(expr->left)));
266 goto free;
268 get_absolute_rl(expr->right, &rl);
269 rl = cast_rl(type, rl);
270 add_type_val(member, rl);
271 free:
272 free_string(right_member);
273 free_string(member);
277 * If we too: int *p = &my_struct->member then abandon all hope of tracking
278 * my_struct->member.
280 static void match_assign_pointer(struct expression *expr)
282 struct expression *right;
283 char *member;
284 struct range_list *rl;
285 struct symbol *type;
287 right = strip_expr(expr->right);
288 if (right->type != EXPR_PREOP || right->op != '&')
289 return;
290 right = strip_expr(right->unop);
292 member = get_member_name(right);
293 if (!member)
294 return;
295 type = get_type(right);
296 rl = alloc_whole_rl(type);
297 add_type_val(member, rl);
298 free_string(member);
301 static void match_global_assign(struct expression *expr)
303 char *member;
304 struct range_list *rl;
305 struct symbol *type;
307 type = get_type(expr->left);
308 if (type && (type->type == SYM_ARRAY || type->type == SYM_STRUCT))
309 return;
310 member = get_member_name(expr->left);
311 if (!member)
312 return;
313 get_absolute_rl(expr->right, &rl);
314 add_global_type_val(member, rl);
315 free_string(member);
318 static void unop_expr(struct expression *expr)
320 struct range_list *rl;
321 char *member;
323 if (expr->op != SPECIAL_DECREMENT && expr->op != SPECIAL_INCREMENT)
324 return;
326 expr = strip_expr(expr->unop);
327 member = get_member_name(expr);
328 if (!member)
329 return;
330 rl = alloc_whole_rl(get_type(expr));
331 add_type_val(member, rl);
332 free_string(member);
335 static void asm_expr(struct statement *stmt)
337 struct expression *expr;
338 struct range_list *rl;
339 char *member;
340 int state = 0;
342 FOR_EACH_PTR(stmt->asm_outputs, expr) {
343 switch (state) {
344 case 0: /* identifier */
345 case 1: /* constraint */
346 state++;
347 continue;
348 case 2: /* expression */
349 state = 0;
350 member = get_member_name(expr);
351 if (!member)
352 continue;
353 rl = alloc_whole_rl(get_type(expr));
354 add_type_val(member, rl);
355 free_string(member);
356 continue;
358 } END_FOR_EACH_PTR(expr);
361 static void db_param_add(struct expression *expr, int param, char *key, char *value)
363 struct expression *arg;
364 struct symbol *type;
365 struct range_list *rl;
366 char *member;
368 if (strcmp(key, "*$") != 0)
369 return;
371 while (expr->type == EXPR_ASSIGNMENT)
372 expr = strip_expr(expr->right);
373 if (expr->type != EXPR_CALL)
374 return;
376 arg = get_argument_from_call_expr(expr->args, param);
377 arg = strip_expr(arg);
378 if (!arg)
379 return;
380 type = get_member_type_from_key(arg, key);
381 if (arg->type != EXPR_PREOP || arg->op != '&')
382 return;
383 arg = strip_expr(arg->unop);
385 member = get_member_name(arg);
386 if (!member)
387 return;
388 call_results_to_rl(expr, type, value, &rl);
389 add_type_val(member, rl);
390 free_string(member);
393 static void match_end_func_info(struct symbol *sym)
395 struct sm_state *sm;
397 FOR_EACH_SM(fn_type_val, sm) {
398 sql_insert_function_type_value(sm->name, sm->state->name);
399 } END_FOR_EACH_SM(sm);
402 static void match_after_func(struct symbol *sym)
404 free_stree(&fn_type_val);
407 static void match_end_file(struct symbol_list *sym_list)
409 struct sm_state *sm;
411 FOR_EACH_SM(global_type_val, sm) {
412 sql_insert_function_type_value(sm->name, sm->state->name);
413 } END_FOR_EACH_SM(sm);
416 void register_type_val(int id)
418 if (!option_info)
419 return;
421 my_id = id;
423 add_hook(&match_assign_value, ASSIGNMENT_HOOK);
424 add_hook(&match_assign_pointer, ASSIGNMENT_HOOK);
425 add_hook(&unop_expr, OP_HOOK);
426 add_hook(&asm_expr, ASM_HOOK);
427 select_return_states_hook(PARAM_ADD, &db_param_add);
428 select_return_states_hook(PARAM_SET, &db_param_add);
431 add_hook(&match_inline_start, INLINE_FN_START);
432 add_hook(&match_inline_end, INLINE_FN_END);
434 add_hook(&match_end_func_info, END_FUNC_HOOK);
435 add_hook(&match_after_func, AFTER_FUNC_HOOK);
437 add_hook(&match_global_assign, GLOBAL_ASSIGNMENT_HOOK);
438 add_hook(&match_end_file, END_FILE_HOOK);