db/smdb.py: add a try catch block printing the return states
[smatch.git] / smatch_flow.c
blob4b0983cb124b4f365dfeed7eb76e640b25a98167
1 /*
2 * Copyright (C) 2006,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
18 #define _GNU_SOURCE 1
19 #include <unistd.h>
20 #include <stdio.h>
21 #include "token.h"
22 #include "scope.h"
23 #include "smatch.h"
24 #include "smatch_expression_stacks.h"
25 #include "smatch_extra.h"
26 #include "smatch_slist.h"
28 int __in_fake_assign;
29 int __in_fake_struct_assign;
30 int __in_fake_var_assign;
31 int __fake_state_cnt;
32 int in_fake_env;
33 int final_pass;
34 int __inline_call;
35 struct expression *__inline_fn;
37 static int __smatch_lineno = 0;
39 static char *base_file;
40 static const char *filename;
41 static char *pathname;
42 static char *full_filename;
43 static char *full_base_file;
44 static char *cur_func;
45 static unsigned int loop_count;
46 static int last_goto_statement_handled;
47 int __expr_stmt_count;
48 int __in_function_def;
49 int __in_unmatched_hook;
50 static struct expression_list *switch_expr_stack = NULL;
51 static struct expression_list *post_op_stack = NULL;
53 static struct ptr_list *fn_data_list;
54 static struct ptr_list *backup;
56 struct expression_list *big_expression_stack;
57 struct statement_list *big_statement_stack;
58 struct statement *__prev_stmt;
59 struct statement *__cur_stmt;
60 struct statement *__next_stmt;
61 int __in_pre_condition = 0;
62 int __bail_on_rest_of_function = 0;
63 static struct timeval fn_start_time;
64 static struct timeval outer_fn_start_time;
65 char *get_function(void) { return cur_func; }
66 int get_lineno(void) { return __smatch_lineno; }
67 int inside_loop(void) { return !!loop_count; }
68 int definitely_inside_loop(void) { return !!(loop_count & ~0x08000000); }
69 struct expression *get_switch_expr(void) { return top_expression(switch_expr_stack); }
70 int in_expression_statement(void) { return !!__expr_stmt_count; }
72 static void split_symlist(struct symbol_list *sym_list);
73 static void split_declaration(struct symbol_list *sym_list);
74 static void split_expr_list(struct expression_list *expr_list, struct expression *parent);
75 static void split_args(struct expression *expr);
76 static struct expression *fake_a_variable_assign(struct symbol *type, struct expression *call, struct expression *expr, int nr);
77 static void add_inline_function(struct symbol *sym);
78 static void parse_inline(struct expression *expr);
80 int option_assume_loops = 0;
81 int option_two_passes = 0;
82 struct symbol *cur_func_sym = NULL;
83 struct stree *global_states;
85 const unsigned long valid_ptr_min = 4096;
86 unsigned long valid_ptr_max = ULONG_MAX & ~(MTAG_OFFSET_MASK);
87 const sval_t valid_ptr_min_sval = {
88 .type = &ptr_ctype,
89 {.value = 4096},
91 sval_t valid_ptr_max_sval = {
92 .type = &ptr_ctype,
93 {.value = ULONG_MAX & ~(MTAG_OFFSET_MASK)},
95 struct range_list *valid_ptr_rl;
97 void alloc_valid_ptr_rl(void)
99 valid_ptr_max = sval_type_max(&ulong_ctype).value & ~(MTAG_OFFSET_MASK);
100 valid_ptr_max_sval.value = valid_ptr_max;
102 valid_ptr_rl = alloc_rl(valid_ptr_min_sval, valid_ptr_max_sval);
103 valid_ptr_rl = cast_rl(&ptr_ctype, valid_ptr_rl);
104 valid_ptr_rl = clone_rl_permanent(valid_ptr_rl);
107 int outside_of_function(void)
109 return cur_func_sym == NULL;
112 const char *get_filename(void)
114 if (option_info && option_full_path)
115 return full_base_file;
116 if (option_info)
117 return base_file;
118 if (option_full_path)
119 return full_filename;
120 return filename;
123 const char *get_base_file(void)
125 if (option_full_path)
126 return full_base_file;
127 return base_file;
130 static void set_position(struct position pos)
132 int len;
133 static int prev_stream = -1;
135 if (in_fake_env)
136 return;
138 if (pos.stream == 0 && pos.line == 0)
139 return;
141 __smatch_lineno = pos.line;
143 if (pos.stream == prev_stream)
144 return;
146 filename = stream_name(pos.stream);
148 free(full_filename);
149 pathname = getcwd(NULL, 0);
150 if (pathname) {
151 len = strlen(pathname) + 1 + strlen(filename) + 1;
152 full_filename = malloc(len);
153 snprintf(full_filename, len, "%s/%s", pathname, filename);
154 } else {
155 full_filename = alloc_string(filename);
157 free(pathname);
160 int is_assigned_call(struct expression *expr)
162 struct expression *parent = expr_get_parent_expr(expr);
164 if (parent &&
165 parent->type == EXPR_ASSIGNMENT &&
166 parent->op == '=' &&
167 strip_expr(parent->right) == expr)
168 return 1;
170 return 0;
173 int is_fake_assigned_call(struct expression *expr)
175 struct expression *parent = expr_get_fake_parent_expr(expr);
177 if (parent &&
178 parent->type == EXPR_ASSIGNMENT &&
179 parent->op == '=' &&
180 strip_expr(parent->right) == expr)
181 return 1;
183 return 0;
186 static bool is_inline_func(struct expression *expr)
188 if (expr->type != EXPR_SYMBOL || !expr->symbol)
189 return false;
190 if (!expr->symbol->definition)
191 return false;
192 if (expr->symbol->definition->ctype.modifiers & MOD_INLINE)
193 return true;
195 return 0;
198 static int is_noreturn_func(struct expression *expr)
200 if (expr->type != EXPR_SYMBOL || !expr->symbol)
201 return 0;
202 if (expr->symbol->ctype.modifiers & MOD_NORETURN)
203 return 1;
204 return 0;
207 static int save_func_time(void *_rl, int argc, char **argv, char **azColName)
209 unsigned long *rl = _rl;
211 *rl = strtoul(argv[0], NULL, 10);
212 return 0;
215 static int get_func_time(struct symbol *sym)
217 unsigned long time = 0;
219 run_sql(&save_func_time, &time,
220 "select key from return_implies where %s and type = %d;",
221 get_static_filter(sym), FUNC_TIME);
223 return time;
226 static int inline_budget = 20;
228 int inlinable(struct expression *expr)
230 struct symbol *sym;
231 struct statement *last_stmt = NULL;
233 if (__inline_fn) /* don't nest */
234 return 0;
236 if (expr->type != EXPR_SYMBOL || !expr->symbol)
237 return 0;
238 if (is_no_inline_function(expr->symbol->ident->name))
239 return 0;
240 sym = get_base_type(expr->symbol);
241 if (sym->stmt && sym->stmt->type == STMT_COMPOUND) {
242 if (ptr_list_size((struct ptr_list *)sym->stmt->stmts) > 10)
243 return 0;
244 if (sym->stmt->type != STMT_COMPOUND)
245 return 0;
246 last_stmt = last_ptr_list((struct ptr_list *)sym->stmt->stmts);
248 if (sym->inline_stmt && sym->inline_stmt->type == STMT_COMPOUND) {
249 if (ptr_list_size((struct ptr_list *)sym->inline_stmt->stmts) > 10)
250 return 0;
251 if (sym->inline_stmt->type != STMT_COMPOUND)
252 return 0;
253 last_stmt = last_ptr_list((struct ptr_list *)sym->inline_stmt->stmts);
256 if (!last_stmt)
257 return 0;
259 /* the magic numbers in this function are pulled out of my bum. */
260 if (last_stmt->pos.line > sym->pos.line + inline_budget)
261 return 0;
263 if (get_func_time(expr->symbol) >= 2)
264 return 0;
266 return 1;
269 void __process_post_op_stack(void)
271 struct expression *expr;
273 FOR_EACH_PTR(post_op_stack, expr) {
274 __pass_to_client(expr, OP_HOOK);
275 } END_FOR_EACH_PTR(expr);
277 __free_ptr_list((struct ptr_list **)&post_op_stack);
280 static int handle_comma_assigns(struct expression *expr)
282 struct expression *right;
283 struct expression *assign;
285 right = strip_expr(expr->right);
286 if (right->type != EXPR_COMMA)
287 return 0;
289 __split_expr(right->left);
290 __process_post_op_stack();
292 assign = assign_expression(expr->left, '=', right->right);
293 __split_expr(assign);
295 return 1;
298 /* This is to handle *p++ = foo; assignments */
299 static int handle_postop_assigns(struct expression *expr)
301 struct expression *left, *fake_left;
302 struct expression *assign;
304 left = strip_expr(expr->left);
305 if (left->type != EXPR_PREOP || left->op != '*')
306 return 0;
307 left = strip_expr(left->unop);
308 if (left->type != EXPR_POSTOP)
309 return 0;
311 fake_left = deref_expression(strip_expr(left->unop));
312 assign = assign_expression(fake_left, '=', expr->right);
314 __split_expr(assign);
315 __split_expr(expr->left);
317 return 1;
320 static int prev_expression_is_getting_address(struct expression *expr)
322 struct expression *parent;
324 do {
325 parent = expr_get_parent_expr(expr);
327 if (!parent)
328 return 0;
329 if (parent->type == EXPR_PREOP && parent->op == '&')
330 return 1;
331 if (parent->type == EXPR_PREOP && parent->op == '(')
332 goto next;
333 if (parent->type == EXPR_DEREF && parent->op == '.')
334 goto next;
336 return 0;
337 next:
338 expr = parent;
339 } while (1);
342 static void handle_builtin_overflow_func(struct expression *expr)
344 struct expression *a, *b, *res, *assign;
345 int op;
347 if (sym_name_is("__builtin_add_overflow", expr->fn))
348 op = '+';
349 else if (sym_name_is("__builtin_sub_overflow", expr->fn))
350 op = '-';
351 else if (sym_name_is("__builtin_mul_overflow", expr->fn))
352 op = '*';
353 else
354 return;
356 a = get_argument_from_call_expr(expr->args, 0);
357 b = get_argument_from_call_expr(expr->args, 1);
358 res = get_argument_from_call_expr(expr->args, 2);
360 assign = assign_expression(deref_expression(res), '=', binop_expression(a, op, b));
361 __split_expr(assign);
364 static int handle__builtin_choose_expr(struct expression *expr)
366 struct expression *const_expr, *expr1, *expr2;
367 sval_t sval;
369 if (!sym_name_is("__builtin_choose_expr", expr->fn))
370 return 0;
372 const_expr = get_argument_from_call_expr(expr->args, 0);
373 expr1 = get_argument_from_call_expr(expr->args, 1);
374 expr2 = get_argument_from_call_expr(expr->args, 2);
376 if (!get_value(const_expr, &sval) || !expr1 || !expr2)
377 return 0;
378 if (sval.value)
379 __split_expr(expr1);
380 else
381 __split_expr(expr2);
382 return 1;
385 static int handle__builtin_choose_expr_assigns(struct expression *expr)
387 struct expression *const_expr, *right, *expr1, *expr2, *fake;
388 sval_t sval;
390 right = strip_expr(expr->right);
391 if (right->type != EXPR_CALL)
392 return 0;
393 if (!sym_name_is("__builtin_choose_expr", right->fn))
394 return 0;
396 const_expr = get_argument_from_call_expr(right->args, 0);
397 expr1 = get_argument_from_call_expr(right->args, 1);
398 expr2 = get_argument_from_call_expr(right->args, 2);
400 if (!get_value(const_expr, &sval) || !expr1 || !expr2)
401 return 0;
403 fake = assign_expression(expr->left, '=', sval.value ? expr1 : expr2);
404 __split_expr(fake);
405 return 1;
408 void __split_expr(struct expression *expr)
410 if (!expr)
411 return;
413 // sm_msg(" Debug expr_type %d %s", expr->type, show_special(expr->op));
415 if (__in_fake_assign && expr->type != EXPR_ASSIGNMENT)
416 return;
417 if (__in_fake_assign >= 4) /* don't allow too much nesting */
418 return;
420 push_expression(&big_expression_stack, expr);
421 set_position(expr->pos);
422 __pass_to_client(expr, EXPR_HOOK);
424 switch (expr->type) {
425 case EXPR_PREOP:
426 expr_set_parent_expr(expr->unop, expr);
428 if (expr->op == '*' &&
429 !prev_expression_is_getting_address(expr))
430 __pass_to_client(expr, DEREF_HOOK);
431 __split_expr(expr->unop);
432 __pass_to_client(expr, OP_HOOK);
433 break;
434 case EXPR_POSTOP:
435 expr_set_parent_expr(expr->unop, expr);
437 __split_expr(expr->unop);
438 push_expression(&post_op_stack, expr);
439 break;
440 case EXPR_STATEMENT:
441 __expr_stmt_count++;
442 if (expr->statement && !expr->statement) {
443 stmt_set_parent_stmt(expr->statement,
444 last_ptr_list((struct ptr_list *)big_statement_stack));
446 __split_stmt(expr->statement);
447 __expr_stmt_count--;
448 break;
449 case EXPR_LOGICAL:
450 case EXPR_COMPARE:
451 expr_set_parent_expr(expr->left, expr);
452 expr_set_parent_expr(expr->right, expr);
454 __pass_to_client(expr, LOGIC_HOOK);
455 __handle_logic(expr);
456 break;
457 case EXPR_BINOP:
458 expr_set_parent_expr(expr->left, expr);
459 expr_set_parent_expr(expr->right, expr);
461 __pass_to_client(expr, BINOP_HOOK);
462 case EXPR_COMMA:
463 expr_set_parent_expr(expr->left, expr);
464 expr_set_parent_expr(expr->right, expr);
466 __split_expr(expr->left);
467 __process_post_op_stack();
468 __split_expr(expr->right);
469 break;
470 case EXPR_ASSIGNMENT: {
471 struct expression *right;
473 expr_set_parent_expr(expr->left, expr);
474 expr_set_parent_expr(expr->right, expr);
476 right = strip_expr(expr->right);
477 if (!right)
478 break;
480 __pass_to_client(expr, RAW_ASSIGNMENT_HOOK);
482 /* foo = !bar() */
483 if (__handle_condition_assigns(expr))
484 goto after_assign;
485 /* foo = (x < 5 ? foo : 5); */
486 if (__handle_select_assigns(expr))
487 goto after_assign;
488 /* foo = ({frob(); frob(); frob(); 1;}) */
489 if (__handle_expr_statement_assigns(expr))
490 break; // FIXME: got after
491 /* foo = (3, 4); */
492 if (handle_comma_assigns(expr))
493 goto after_assign;
494 if (handle__builtin_choose_expr_assigns(expr))
495 goto after_assign;
496 if (handle_postop_assigns(expr))
497 break; /* no need to goto after_assign */
499 __split_expr(expr->right);
500 if (outside_of_function())
501 __pass_to_client(expr, GLOBAL_ASSIGNMENT_HOOK);
502 else
503 __pass_to_client(expr, ASSIGNMENT_HOOK);
505 __fake_struct_member_assignments(expr);
507 /* Re-examine ->right for inlines. See the commit message */
508 right = strip_expr(expr->right);
509 if (expr->op == '=' && right->type == EXPR_CALL)
510 __pass_to_client(expr, CALL_ASSIGNMENT_HOOK);
512 after_assign:
513 if (get_macro_name(right->pos) &&
514 get_macro_name(expr->pos) != get_macro_name(right->pos))
515 __pass_to_client(expr, MACRO_ASSIGNMENT_HOOK);
517 __pass_to_client(expr, ASSIGNMENT_HOOK_AFTER);
518 __split_expr(expr->left);
519 break;
521 case EXPR_DEREF:
522 expr_set_parent_expr(expr->deref, expr);
524 __pass_to_client(expr, DEREF_HOOK);
525 __split_expr(expr->deref);
526 break;
527 case EXPR_SLICE:
528 expr_set_parent_expr(expr->base, expr);
530 __split_expr(expr->base);
531 break;
532 case EXPR_CAST:
533 case EXPR_FORCE_CAST:
534 expr_set_parent_expr(expr->cast_expression, expr);
536 __pass_to_client(expr, CAST_HOOK);
537 __split_expr(expr->cast_expression);
538 break;
539 case EXPR_SIZEOF:
540 if (expr->cast_expression)
541 __pass_to_client(strip_parens(expr->cast_expression),
542 SIZEOF_HOOK);
543 break;
544 case EXPR_OFFSETOF:
545 case EXPR_ALIGNOF:
546 break;
547 case EXPR_CONDITIONAL:
548 case EXPR_SELECT:
549 expr_set_parent_expr(expr->conditional, expr);
550 expr_set_parent_expr(expr->cond_true, expr);
551 expr_set_parent_expr(expr->cond_false, expr);
553 if (known_condition_true(expr->conditional)) {
554 __split_expr(expr->cond_true);
555 break;
557 if (known_condition_false(expr->conditional)) {
558 __split_expr(expr->cond_false);
559 break;
561 __pass_to_client(expr, SELECT_HOOK);
562 __split_whole_condition(expr->conditional);
563 __split_expr(expr->cond_true);
564 __push_true_states();
565 __use_false_states();
566 __split_expr(expr->cond_false);
567 __merge_true_states();
568 break;
569 case EXPR_CALL:
570 expr_set_parent_expr(expr->fn, expr);
572 if (sym_name_is("__builtin_constant_p", expr->fn))
573 break;
574 if (handle__builtin_choose_expr(expr))
575 break;
576 __split_expr(expr->fn);
577 split_args(expr);
578 if (is_inline_func(expr->fn))
579 add_inline_function(expr->fn->symbol->definition);
580 if (inlinable(expr->fn))
581 __inline_call = 1;
582 __process_post_op_stack();
583 __pass_to_client(expr, FUNCTION_CALL_HOOK_BEFORE);
584 __pass_to_client(expr, FUNCTION_CALL_HOOK);
585 __inline_call = 0;
586 if (inlinable(expr->fn))
587 parse_inline(expr);
588 __pass_to_client(expr, CALL_HOOK_AFTER_INLINE);
589 if (is_noreturn_func(expr->fn))
590 nullify_path();
591 if (!expr_get_parent_expr(expr))
592 __discard_fake_states(expr);
593 handle_builtin_overflow_func(expr);
594 break;
595 case EXPR_INITIALIZER:
596 split_expr_list(expr->expr_list, expr);
597 break;
598 case EXPR_IDENTIFIER:
599 expr_set_parent_expr(expr->ident_expression, expr);
600 __split_expr(expr->ident_expression);
601 break;
602 case EXPR_INDEX:
603 expr_set_parent_expr(expr->idx_expression, expr);
604 __split_expr(expr->idx_expression);
605 break;
606 case EXPR_POS:
607 expr_set_parent_expr(expr->init_expr, expr);
608 __split_expr(expr->init_expr);
609 break;
610 case EXPR_SYMBOL:
611 __pass_to_client(expr, SYM_HOOK);
612 break;
613 case EXPR_STRING:
614 __pass_to_client(expr, STRING_HOOK);
615 break;
616 default:
617 break;
619 __pass_to_client(expr, EXPR_HOOK_AFTER);
620 pop_expression(&big_expression_stack);
623 static int is_forever_loop(struct statement *stmt)
625 struct expression *expr;
626 sval_t sval;
628 expr = strip_expr(stmt->iterator_pre_condition);
629 if (!expr)
630 expr = stmt->iterator_post_condition;
631 if (!expr) {
632 /* this is a for(;;) loop... */
633 return 1;
636 if (get_value(expr, &sval) && sval.value != 0)
637 return 1;
639 return 0;
642 static int loop_num;
643 static char *get_loop_name(int num)
645 char buf[256];
647 snprintf(buf, 255, "-loop%d", num);
648 buf[255] = '\0';
649 return alloc_sname(buf);
653 * Pre Loops are while and for loops.
655 static void handle_pre_loop(struct statement *stmt)
657 int once_through; /* we go through the loop at least once */
658 struct sm_state *extra_sm = NULL;
659 int unchanged = 0;
660 char *loop_name;
661 struct stree *stree = NULL;
662 struct sm_state *sm = NULL;
664 loop_name = get_loop_name(loop_num);
665 loop_num++;
667 __split_stmt(stmt->iterator_pre_statement);
668 __prev_stmt = stmt->iterator_pre_statement;
670 once_through = implied_condition_true(stmt->iterator_pre_condition);
672 loop_count++;
673 __push_continues();
674 __push_breaks();
676 __merge_gotos(loop_name, NULL);
678 extra_sm = __extra_handle_canonical_loops(stmt, &stree);
679 __in_pre_condition++;
680 __pass_to_client(stmt, PRELOOP_HOOK);
681 __split_whole_condition(stmt->iterator_pre_condition);
682 __in_pre_condition--;
683 FOR_EACH_SM(stree, sm) {
684 set_state(sm->owner, sm->name, sm->sym, sm->state);
685 } END_FOR_EACH_SM(sm);
686 free_stree(&stree);
687 if (extra_sm)
688 extra_sm = get_sm_state(extra_sm->owner, extra_sm->name, extra_sm->sym);
690 if (option_assume_loops)
691 once_through = 1;
693 __split_stmt(stmt->iterator_statement);
694 if (is_forever_loop(stmt)) {
695 __merge_continues();
696 __save_gotos(loop_name, NULL);
698 __push_fake_cur_stree();
699 __split_stmt(stmt->iterator_post_statement);
700 stree = __pop_fake_cur_stree();
702 __discard_false_states();
703 __use_breaks();
705 if (!__path_is_null())
706 __merge_stree_into_cur(stree);
707 free_stree(&stree);
708 } else {
709 __merge_continues();
710 unchanged = __iterator_unchanged(extra_sm);
711 __split_stmt(stmt->iterator_post_statement);
712 __prev_stmt = stmt->iterator_post_statement;
713 __cur_stmt = stmt;
715 __save_gotos(loop_name, NULL);
716 __in_pre_condition++;
717 __split_whole_condition(stmt->iterator_pre_condition);
718 __in_pre_condition--;
719 nullify_path();
720 __merge_false_states();
721 if (once_through)
722 __discard_false_states();
723 else
724 __merge_false_states();
726 if (extra_sm && unchanged)
727 __extra_pre_loop_hook_after(extra_sm,
728 stmt->iterator_post_statement,
729 stmt->iterator_pre_condition);
730 __merge_breaks();
732 loop_count--;
736 * Post loops are do {} while();
738 static void handle_post_loop(struct statement *stmt)
740 char *loop_name;
742 loop_name = get_loop_name(loop_num);
743 loop_num++;
744 loop_count++;
746 __push_continues();
747 __push_breaks();
748 __merge_gotos(loop_name, NULL);
749 __split_stmt(stmt->iterator_statement);
750 __merge_continues();
751 if (!expr_is_zero(stmt->iterator_post_condition))
752 __save_gotos(loop_name, NULL);
754 if (is_forever_loop(stmt)) {
755 __use_breaks();
756 } else {
757 __split_whole_condition(stmt->iterator_post_condition);
758 __use_false_states();
759 __merge_breaks();
761 loop_count--;
764 static int empty_statement(struct statement *stmt)
766 if (!stmt)
767 return 0;
768 if (stmt->type == STMT_EXPRESSION && !stmt->expression)
769 return 1;
770 return 0;
773 static int last_stmt_on_same_line(void)
775 struct statement *stmt;
776 int i = 0;
778 FOR_EACH_PTR_REVERSE(big_statement_stack, stmt) {
779 if (!i++)
780 continue;
781 if (stmt->pos.line == get_lineno())
782 return 1;
783 return 0;
784 } END_FOR_EACH_PTR_REVERSE(stmt);
785 return 0;
788 static void split_asm_ops(struct asm_operand_list *ops)
790 struct asm_operand *op;
792 FOR_EACH_PTR(ops, op) {
793 __split_expr(op->expr);
794 } END_FOR_EACH_PTR(op);
797 static int is_case_val(struct statement *stmt, sval_t sval)
799 sval_t case_sval;
801 if (stmt->type != STMT_CASE)
802 return 0;
803 if (!stmt->case_expression) {
804 __set_default();
805 return 1;
807 if (!get_value(stmt->case_expression, &case_sval))
808 return 0;
809 if (case_sval.value == sval.value)
810 return 1;
811 return 0;
814 static struct range_list *get_case_rl(struct expression *switch_expr,
815 struct expression *case_expr,
816 struct expression *case_to)
818 sval_t start, end;
819 struct range_list *rl = NULL;
820 struct symbol *switch_type;
822 switch_type = get_type(switch_expr);
823 if (get_value(case_to, &end) && get_value(case_expr, &start)) {
824 start = sval_cast(switch_type, start);
825 end = sval_cast(switch_type, end);
826 add_range(&rl, start, end);
827 } else if (get_value(case_expr, &start)) {
828 start = sval_cast(switch_type, start);
829 add_range(&rl, start, start);
832 return rl;
835 static void split_known_switch(struct statement *stmt, sval_t sval)
837 struct statement *tmp;
838 struct range_list *rl;
840 __split_expr(stmt->switch_expression);
841 sval = sval_cast(get_type(stmt->switch_expression), sval);
843 push_expression(&switch_expr_stack, stmt->switch_expression);
844 __save_switch_states(top_expression(switch_expr_stack));
845 nullify_path();
846 __push_default();
847 __push_breaks();
849 stmt = stmt->switch_statement;
851 __push_scope_hooks();
852 FOR_EACH_PTR(stmt->stmts, tmp) {
853 __smatch_lineno = tmp->pos.line;
854 if (is_case_val(tmp, sval)) {
855 rl = alloc_rl(sval, sval);
856 __merge_switches(top_expression(switch_expr_stack), rl);
857 __pass_case_to_client(top_expression(switch_expr_stack), rl);
859 if (__path_is_null())
860 continue;
861 __split_stmt(tmp);
862 if (__path_is_null()) {
863 __set_default();
864 goto out;
866 } END_FOR_EACH_PTR(tmp);
867 out:
868 __call_scope_hooks();
869 if (!__pop_default())
870 __merge_switches(top_expression(switch_expr_stack), NULL);
871 __discard_switches();
872 __merge_breaks();
873 pop_expression(&switch_expr_stack);
876 static void split_case(struct statement *stmt)
878 struct range_list *rl = NULL;
880 expr_set_parent_stmt(stmt->case_expression, stmt);
881 expr_set_parent_stmt(stmt->case_to, stmt);
883 rl = get_case_rl(top_expression(switch_expr_stack),
884 stmt->case_expression, stmt->case_to);
885 while (stmt->case_statement->type == STMT_CASE) {
886 struct range_list *tmp;
888 tmp = get_case_rl(top_expression(switch_expr_stack),
889 stmt->case_statement->case_expression,
890 stmt->case_statement->case_to);
891 if (!tmp)
892 break;
893 rl = rl_union(rl, tmp);
894 if (!stmt->case_expression)
895 __set_default();
896 stmt = stmt->case_statement;
899 __merge_switches(top_expression(switch_expr_stack), rl);
901 if (!stmt->case_expression)
902 __set_default();
903 __split_stmt(stmt->case_statement);
906 int time_parsing_function(void)
908 return ms_since(&fn_start_time) / 1000;
911 bool taking_too_long(void)
913 if ((ms_since(&outer_fn_start_time) / 1000) > 60 * 5) /* five minutes */
914 return 1;
915 return 0;
918 int is_last_stmt(struct statement *cur_stmt)
920 struct symbol *fn;
921 struct statement *stmt;
923 if (!cur_func_sym)
924 return 0;
925 fn = get_base_type(cur_func_sym);
926 if (!fn)
927 return 0;
928 stmt = fn->stmt;
929 if (!stmt)
930 stmt = fn->inline_stmt;
931 if (!stmt || stmt->type != STMT_COMPOUND)
932 return 0;
933 stmt = last_ptr_list((struct ptr_list *)stmt->stmts);
934 if (stmt && stmt->type == STMT_LABEL)
935 stmt = stmt->label_statement;
936 if (stmt == cur_stmt)
937 return 1;
938 return 0;
941 static void handle_backward_goto(struct statement *goto_stmt)
943 const char *goto_name, *label_name;
944 struct statement *func_stmt;
945 struct symbol *base_type = get_base_type(cur_func_sym);
946 struct statement *tmp;
947 int found = 0;
949 if (!option_info)
950 return;
951 if (last_goto_statement_handled)
952 return;
953 last_goto_statement_handled = 1;
955 if (!goto_stmt->goto_label ||
956 goto_stmt->goto_label->type != SYM_LABEL ||
957 !goto_stmt->goto_label->ident)
958 return;
959 goto_name = goto_stmt->goto_label->ident->name;
961 func_stmt = base_type->stmt;
962 if (!func_stmt)
963 func_stmt = base_type->inline_stmt;
964 if (!func_stmt)
965 return;
966 if (func_stmt->type != STMT_COMPOUND)
967 return;
969 FOR_EACH_PTR(func_stmt->stmts, tmp) {
970 if (!found) {
971 if (tmp->type != STMT_LABEL)
972 continue;
973 if (!tmp->label_identifier ||
974 tmp->label_identifier->type != SYM_LABEL ||
975 !tmp->label_identifier->ident)
976 continue;
977 label_name = tmp->label_identifier->ident->name;
978 if (strcmp(goto_name, label_name) != 0)
979 continue;
980 found = 1;
982 __split_stmt(tmp);
983 } END_FOR_EACH_PTR(tmp);
986 static void fake_a_return(void)
988 struct expression *ret = NULL;
990 nullify_path();
991 __unnullify_path();
993 if (cur_func_return_type() != &void_ctype)
994 ret = unknown_value_expression(NULL);
996 __pass_to_client(ret, RETURN_HOOK);
997 nullify_path();
1000 static void split_ret_value(struct expression *expr)
1002 struct symbol *type;
1004 if (!expr)
1005 return;
1007 type = get_real_base_type(cur_func_sym);
1008 type = get_real_base_type(type);
1009 expr = fake_a_variable_assign(type, NULL, expr, -1);
1011 __in_fake_var_assign++;
1012 __split_expr(expr);
1013 __in_fake_var_assign--;
1016 static void fake_an_empty_default(struct position pos)
1018 static struct statement none = {};
1020 none.pos = pos;
1021 none.type = STMT_NONE;
1022 __merge_switches(top_expression(switch_expr_stack), NULL);
1023 __split_stmt(&none);
1026 static void split_compound(struct statement *stmt)
1028 struct statement *prev = NULL;
1029 struct statement *cur = NULL;
1030 struct statement *next;
1032 __push_scope_hooks();
1034 FOR_EACH_PTR(stmt->stmts, next) {
1035 /* just set them all ahead of time */
1036 stmt_set_parent_stmt(next, stmt);
1038 if (cur) {
1039 __prev_stmt = prev;
1040 __next_stmt = next;
1041 __cur_stmt = cur;
1042 __split_stmt(cur);
1044 prev = cur;
1045 cur = next;
1046 } END_FOR_EACH_PTR(next);
1047 if (cur) {
1048 __prev_stmt = prev;
1049 __cur_stmt = cur;
1050 __next_stmt = NULL;
1051 __split_stmt(cur);
1055 * For function scope, then delay calling the scope hooks until the
1056 * end of function hooks can run. I'm not positive this is the right
1057 * thing...
1059 if (!is_last_stmt(cur))
1060 __call_scope_hooks();
1064 * This is a hack, work around for detecting empty functions.
1066 static int need_delayed_scope_hooks(void)
1068 struct symbol *fn = get_base_type(cur_func_sym);
1069 struct statement *stmt;
1071 if (!fn)
1072 return 0;
1073 stmt = fn->stmt;
1074 if (!stmt)
1075 stmt = fn->inline_stmt;
1076 if (stmt && stmt->type == STMT_COMPOUND)
1077 return 1;
1078 return 0;
1081 void __split_label_stmt(struct statement *stmt)
1083 if (stmt->label_identifier &&
1084 stmt->label_identifier->type == SYM_LABEL &&
1085 stmt->label_identifier->ident) {
1086 loop_count |= 0x0800000;
1087 __merge_gotos(stmt->label_identifier->ident->name, stmt->label_identifier);
1091 static void find_asm_gotos(struct statement *stmt)
1093 struct symbol *sym;
1095 FOR_EACH_PTR(stmt->asm_labels, sym) {
1096 __save_gotos(sym->ident->name, sym);
1097 } END_FOR_EACH_PTR(sym);
1100 void __split_stmt(struct statement *stmt)
1102 static int indent_cnt;
1103 sval_t sval;
1104 struct timeval start, stop;
1106 gettimeofday(&start, NULL);
1108 if (!stmt)
1109 goto out;
1111 if (!__in_fake_assign)
1112 __silence_warnings_for_stmt = false;
1114 if (__bail_on_rest_of_function || is_skipped_function())
1115 return;
1117 if (out_of_memory() || taking_too_long()) {
1118 gettimeofday(&start, NULL);
1120 __bail_on_rest_of_function = 1;
1121 final_pass = 1;
1122 sm_perror("Function too hairy. Giving up. %lu seconds",
1123 start.tv_sec - fn_start_time.tv_sec);
1124 fake_a_return();
1125 final_pass = 0; /* turn off sm_msg() from here */
1126 return;
1129 indent_cnt++;
1131 add_ptr_list(&big_statement_stack, stmt);
1132 free_expression_stack(&big_expression_stack);
1133 set_position(stmt->pos);
1134 __pass_to_client(stmt, STMT_HOOK);
1136 switch (stmt->type) {
1137 case STMT_DECLARATION:
1138 split_declaration(stmt->declaration);
1139 break;
1140 case STMT_RETURN:
1141 expr_set_parent_stmt(stmt->ret_value, stmt);
1143 split_ret_value(stmt->ret_value);
1144 __pass_to_client(stmt->ret_value, RETURN_HOOK);
1145 __process_post_op_stack();
1146 nullify_path();
1147 break;
1148 case STMT_EXPRESSION:
1149 expr_set_parent_stmt(stmt->expression, stmt);
1150 expr_set_parent_stmt(stmt->context, stmt);
1152 __split_expr(stmt->expression);
1153 break;
1154 case STMT_COMPOUND:
1155 split_compound(stmt);
1156 break;
1157 case STMT_IF:
1158 stmt_set_parent_stmt(stmt->if_true, stmt);
1159 stmt_set_parent_stmt(stmt->if_false, stmt);
1160 expr_set_parent_stmt(stmt->if_conditional, stmt);
1162 if (known_condition_true(stmt->if_conditional)) {
1163 __split_stmt(stmt->if_true);
1164 break;
1166 if (known_condition_false(stmt->if_conditional)) {
1167 __split_stmt(stmt->if_false);
1168 break;
1170 __split_whole_condition(stmt->if_conditional);
1171 __split_stmt(stmt->if_true);
1172 if (empty_statement(stmt->if_true) &&
1173 last_stmt_on_same_line() &&
1174 !get_macro_name(stmt->if_true->pos))
1175 sm_warning("if();");
1176 __push_true_states();
1177 __use_false_states();
1178 __split_stmt(stmt->if_false);
1179 __merge_true_states();
1180 break;
1181 case STMT_ITERATOR:
1182 stmt_set_parent_stmt(stmt->iterator_pre_statement, stmt);
1183 stmt_set_parent_stmt(stmt->iterator_statement, stmt);
1184 stmt_set_parent_stmt(stmt->iterator_post_statement, stmt);
1185 expr_set_parent_stmt(stmt->iterator_pre_condition, stmt);
1186 expr_set_parent_stmt(stmt->iterator_post_condition, stmt);
1188 if (stmt->iterator_pre_condition)
1189 handle_pre_loop(stmt);
1190 else if (stmt->iterator_post_condition)
1191 handle_post_loop(stmt);
1192 else {
1193 // these are for(;;) type loops.
1194 handle_pre_loop(stmt);
1196 break;
1197 case STMT_SWITCH:
1198 stmt_set_parent_stmt(stmt->switch_statement, stmt);
1199 expr_set_parent_stmt(stmt->switch_expression, stmt);
1201 if (get_value(stmt->switch_expression, &sval)) {
1202 split_known_switch(stmt, sval);
1203 break;
1205 __split_expr(stmt->switch_expression);
1206 push_expression(&switch_expr_stack, stmt->switch_expression);
1207 __save_switch_states(top_expression(switch_expr_stack));
1208 nullify_path();
1209 __push_default();
1210 __push_breaks();
1211 __split_stmt(stmt->switch_statement);
1212 if (!__pop_default() && have_remaining_cases())
1213 fake_an_empty_default(stmt->pos);
1214 __discard_switches();
1215 __merge_breaks();
1216 pop_expression(&switch_expr_stack);
1217 break;
1218 case STMT_CASE:
1219 split_case(stmt);
1220 break;
1221 case STMT_LABEL:
1222 __split_label_stmt(stmt);
1223 __split_stmt(stmt->label_statement);
1224 break;
1225 case STMT_GOTO:
1226 expr_set_parent_stmt(stmt->goto_expression, stmt);
1228 __split_expr(stmt->goto_expression);
1229 if (stmt->goto_label && stmt->goto_label->type == SYM_NODE) {
1230 if (!strcmp(stmt->goto_label->ident->name, "break")) {
1231 __process_breaks();
1232 } else if (!strcmp(stmt->goto_label->ident->name,
1233 "continue")) {
1234 __process_continues();
1236 } else if (stmt->goto_label &&
1237 stmt->goto_label->type == SYM_LABEL &&
1238 stmt->goto_label->ident) {
1239 __save_gotos(stmt->goto_label->ident->name, stmt->goto_label);
1241 nullify_path();
1242 if (is_last_stmt(stmt))
1243 handle_backward_goto(stmt);
1244 break;
1245 case STMT_NONE:
1246 break;
1247 case STMT_ASM:
1248 expr_set_parent_stmt(stmt->asm_string, stmt);
1250 find_asm_gotos(stmt);
1251 __pass_to_client(stmt, ASM_HOOK);
1252 __split_expr(stmt->asm_string);
1253 split_asm_ops(stmt->asm_outputs);
1254 split_asm_ops(stmt->asm_inputs);
1255 split_expr_list(stmt->asm_clobbers, NULL);
1256 break;
1257 case STMT_CONTEXT:
1258 break;
1259 case STMT_RANGE:
1260 __split_expr(stmt->range_expression);
1261 __split_expr(stmt->range_low);
1262 __split_expr(stmt->range_high);
1263 break;
1265 __pass_to_client(stmt, STMT_HOOK_AFTER);
1266 if (--indent_cnt == 1)
1267 __discard_fake_states(NULL);
1269 out:
1270 __process_post_op_stack();
1272 gettimeofday(&stop, NULL);
1273 if (option_time_stmt && stmt)
1274 sm_msg("stmt_time%s: %ld",
1275 stmt->type == STMT_COMPOUND ? "_block" : "",
1276 stop.tv_sec - start.tv_sec);
1279 static void split_expr_list(struct expression_list *expr_list, struct expression *parent)
1281 struct expression *expr;
1283 FOR_EACH_PTR(expr_list, expr) {
1284 expr_set_parent_expr(expr, parent);
1285 __split_expr(expr);
1286 __process_post_op_stack();
1287 } END_FOR_EACH_PTR(expr);
1290 static bool cast_arg(struct symbol *type, struct expression *arg)
1292 struct symbol *orig;
1294 if (!type)
1295 return false;
1297 arg = strip_parens(arg);
1298 if (arg != strip_expr(arg))
1299 return true;
1301 orig = get_type(arg);
1302 if (!orig)
1303 return true;
1304 if (orig == type)
1305 return false;
1307 if (orig->type == SYM_ARRAY && type->type == SYM_PTR)
1308 return true;
1311 * I would have expected that we could just do use (orig == type) but I
1312 * guess for pointers we need to get the basetype to do that comparison.
1316 if (orig->type != SYM_PTR ||
1317 type->type != SYM_PTR) {
1318 if (type_fits(type, orig))
1319 return false;
1320 return true;
1322 orig = get_real_base_type(orig);
1323 type = get_real_base_type(type);
1324 if (orig == type)
1325 return false;
1327 return true;
1330 static struct expression *fake_a_variable_assign(struct symbol *type, struct expression *call, struct expression *expr, int nr)
1332 struct expression *var, *assign, *parent;
1333 char buf[64];
1334 bool cast;
1336 if (!expr || !cur_func_sym)
1337 return NULL;
1339 if (expr->type == EXPR_ASSIGNMENT)
1340 return expr;
1342 cast = cast_arg(type, expr);
1344 * Using expr_to_sym() here is a hack. We want to say that we don't
1345 * need to assign frob(foo) or frob(foo->bar) if the types are right.
1346 * It turns out faking these assignments is way more expensive than I
1347 * would have imagined. I'm not sure why exactly.
1350 if (!cast) {
1352 * if the code is "return *p;" where "p" is a user pointer then
1353 * we want to create a fake assignment so that it sets the state
1354 * in check_kernel_user_data.c.
1357 if (expr->type != EXPR_PREOP &&
1358 expr->op != '*' && expr->op != '&' &&
1359 expr_to_sym(expr))
1360 return expr;
1363 if (nr == -1)
1364 snprintf(buf, sizeof(buf), "__sm_fake_%p", expr);
1365 else
1366 snprintf(buf, sizeof(buf), "__fake_param_%p_%d", call, nr);
1367 var = fake_variable(type, buf);
1368 assign = assign_expression(var, '=', expr);
1369 assign->smatch_flags |= Fake;
1371 parent = expr_get_parent_expr(expr);
1372 expr_set_parent_expr(assign, parent);
1373 expr_set_parent_expr(expr, assign);
1375 __fake_state_cnt++;
1377 return assign;
1380 static void split_args(struct expression *expr)
1382 struct expression *arg, *tmp;
1383 struct symbol *type;
1384 int i;
1386 i = -1;
1387 FOR_EACH_PTR(expr->args, arg) {
1388 i++;
1389 expr_set_parent_expr(arg, expr);
1390 type = get_arg_type(expr->fn, i);
1391 tmp = fake_a_variable_assign(type, expr, arg, i);
1392 if (tmp != arg)
1393 __in_fake_var_assign++;
1394 __split_expr(tmp);
1395 if (tmp != arg)
1396 __in_fake_var_assign--;
1397 __process_post_op_stack();
1398 } END_FOR_EACH_PTR(arg);
1401 static void split_sym(struct symbol *sym)
1403 if (!sym)
1404 return;
1405 if (!(sym->namespace & NS_SYMBOL))
1406 return;
1408 __split_stmt(sym->stmt);
1409 __split_expr(sym->array_size);
1410 split_symlist(sym->arguments);
1411 split_symlist(sym->symbol_list);
1412 __split_stmt(sym->inline_stmt);
1413 split_symlist(sym->inline_symbol_list);
1416 static void split_symlist(struct symbol_list *sym_list)
1418 struct symbol *sym;
1420 FOR_EACH_PTR(sym_list, sym) {
1421 split_sym(sym);
1422 } END_FOR_EACH_PTR(sym);
1425 typedef void (fake_cb)(struct expression *expr);
1427 static int member_to_number(struct expression *expr, struct ident *member)
1429 struct symbol *type, *tmp;
1430 char *name;
1431 int i;
1433 if (!member)
1434 return -1;
1435 name = member->name;
1437 type = get_type(expr);
1438 if (!type || type->type != SYM_STRUCT)
1439 return -1;
1441 i = -1;
1442 FOR_EACH_PTR(type->symbol_list, tmp) {
1443 i++;
1444 if (!tmp->ident)
1445 continue;
1446 if (strcmp(name, tmp->ident->name) == 0)
1447 return i;
1448 } END_FOR_EACH_PTR(tmp);
1449 return -1;
1452 static struct ident *number_to_member(struct expression *expr, int num)
1454 struct symbol *type, *member;
1455 int i = 0;
1457 type = get_type(expr);
1458 if (!type || type->type != SYM_STRUCT)
1459 return NULL;
1461 FOR_EACH_PTR(type->symbol_list, member) {
1462 if (i == num)
1463 return member->ident;
1464 i++;
1465 } END_FOR_EACH_PTR(member);
1466 return NULL;
1469 static void fake_element_assigns_helper(struct expression *array, struct expression_list *expr_list, fake_cb *fake_cb);
1471 static void set_inner_struct_members(struct expression *expr, struct symbol *member)
1473 struct expression *edge_member, *assign;
1474 struct symbol *base = get_real_base_type(member);
1475 struct symbol *tmp;
1477 if (member->ident)
1478 expr = member_expression(expr, '.', member->ident);
1480 FOR_EACH_PTR(base->symbol_list, tmp) {
1481 struct symbol *type;
1483 type = get_real_base_type(tmp);
1484 if (!type)
1485 continue;
1487 edge_member = member_expression(expr, '.', tmp->ident);
1488 if (get_extra_state(edge_member))
1489 continue;
1491 if (type->type == SYM_UNION || type->type == SYM_STRUCT) {
1492 set_inner_struct_members(expr, tmp);
1493 continue;
1496 if (!tmp->ident)
1497 continue;
1499 assign = assign_expression(edge_member, '=', zero_expr());
1500 __split_expr(assign);
1501 } END_FOR_EACH_PTR(tmp);
1506 static void set_unset_to_zero(struct symbol *type, struct expression *expr)
1508 struct symbol *tmp;
1509 struct expression *member = NULL;
1510 struct expression *assign;
1511 int op = '*';
1513 if (expr->type == EXPR_PREOP && expr->op == '&') {
1514 expr = strip_expr(expr->unop);
1515 op = '.';
1518 FOR_EACH_PTR(type->symbol_list, tmp) {
1519 type = get_real_base_type(tmp);
1520 if (!type)
1521 continue;
1523 if (tmp->ident) {
1524 member = member_expression(expr, op, tmp->ident);
1525 if (get_extra_state(member))
1526 continue;
1529 if (type->type == SYM_UNION || type->type == SYM_STRUCT) {
1530 set_inner_struct_members(expr, tmp);
1531 continue;
1533 if (type->type == SYM_ARRAY)
1534 continue;
1535 if (!tmp->ident)
1536 continue;
1538 assign = assign_expression(member, '=', zero_expr());
1539 __split_expr(assign);
1540 } END_FOR_EACH_PTR(tmp);
1543 static void fake_member_assigns_helper(struct expression *symbol, struct expression_list *members, fake_cb *fake_cb)
1545 struct expression *deref, *assign, *tmp, *right;
1546 struct symbol *struct_type, *type;
1547 struct ident *member;
1548 int member_idx;
1550 struct_type = get_type(symbol);
1551 if (!struct_type ||
1552 (struct_type->type != SYM_STRUCT && struct_type->type != SYM_UNION))
1553 return;
1556 * We're parsing an initializer that could look something like this:
1557 * struct foo foo = {
1558 * 42,
1559 * .whatever.xxx = 11,
1560 * .zzz = 12,
1561 * };
1563 * So what we have here is a list with 42, .whatever, and .zzz. We need
1564 * to break it up into left and right sides of the assignments.
1567 member_idx = 0;
1568 FOR_EACH_PTR(members, tmp) {
1569 deref = NULL;
1570 if (tmp->type == EXPR_IDENTIFIER) {
1571 member_idx = member_to_number(symbol, tmp->expr_ident);
1572 while (tmp->type == EXPR_IDENTIFIER) {
1573 member = tmp->expr_ident;
1574 tmp = tmp->ident_expression;
1575 if (deref)
1576 deref = member_expression(deref, '.', member);
1577 else
1578 deref = member_expression(symbol, '.', member);
1580 } else {
1581 member = number_to_member(symbol, member_idx);
1582 deref = member_expression(symbol, '.', member);
1584 right = tmp;
1585 member_idx++;
1586 if (right->type == EXPR_INITIALIZER) {
1587 type = get_type(deref);
1588 if (type && type->type == SYM_ARRAY)
1589 fake_element_assigns_helper(deref, right->expr_list, fake_cb);
1590 else
1591 fake_member_assigns_helper(deref, right->expr_list, fake_cb);
1592 } else {
1593 assign = assign_expression(deref, '=', right);
1594 fake_cb(assign);
1596 } END_FOR_EACH_PTR(tmp);
1598 set_unset_to_zero(struct_type, symbol);
1601 static void fake_member_assigns(struct symbol *sym, fake_cb *fake_cb)
1603 fake_member_assigns_helper(symbol_expression(sym),
1604 sym->initializer->expr_list, fake_cb);
1607 static void fake_element_assigns_helper(struct expression *array, struct expression_list *expr_list, fake_cb *fake_cb)
1609 struct expression *offset, *binop, *assign, *tmp;
1610 struct symbol *type;
1611 int idx;
1613 if (ptr_list_size((struct ptr_list *)expr_list) > 1000)
1614 return;
1616 idx = 0;
1617 FOR_EACH_PTR(expr_list, tmp) {
1618 if (tmp->type == EXPR_INDEX) {
1619 if (tmp->idx_from != tmp->idx_to)
1620 return;
1621 idx = tmp->idx_from;
1622 if (!tmp->idx_expression)
1623 goto next;
1624 tmp = tmp->idx_expression;
1626 offset = value_expr(idx);
1627 binop = array_element_expression(array, offset);
1628 if (tmp->type == EXPR_INITIALIZER) {
1629 type = get_type(binop);
1630 if (type && type->type == SYM_ARRAY)
1631 fake_element_assigns_helper(binop, tmp->expr_list, fake_cb);
1632 else
1633 fake_member_assigns_helper(binop, tmp->expr_list, fake_cb);
1634 } else {
1635 assign = assign_expression(binop, '=', tmp);
1636 fake_cb(assign);
1638 next:
1639 idx++;
1640 } END_FOR_EACH_PTR(tmp);
1643 static void fake_element_assigns(struct symbol *sym, fake_cb *fake_cb)
1645 fake_element_assigns_helper(symbol_expression(sym), sym->initializer->expr_list, fake_cb);
1648 static void fake_assign_expr(struct symbol *sym)
1650 struct expression *assign, *symbol;
1652 symbol = symbol_expression(sym);
1653 assign = assign_expression(symbol, '=', sym->initializer);
1654 __split_expr(assign);
1657 static void do_initializer_stuff(struct symbol *sym)
1659 if (!sym->initializer)
1660 return;
1662 if (sym->initializer->type == EXPR_INITIALIZER) {
1663 if (get_real_base_type(sym)->type == SYM_ARRAY)
1664 fake_element_assigns(sym, __split_expr);
1665 else
1666 fake_member_assigns(sym, __split_expr);
1667 } else {
1668 fake_assign_expr(sym);
1672 static void split_declaration(struct symbol_list *sym_list)
1674 struct symbol *sym;
1676 FOR_EACH_PTR(sym_list, sym) {
1677 __pass_to_client(sym, DECLARATION_HOOK);
1678 do_initializer_stuff(sym);
1679 split_sym(sym);
1680 } END_FOR_EACH_PTR(sym);
1683 static void call_global_assign_hooks(struct expression *assign)
1685 __pass_to_client(assign, GLOBAL_ASSIGNMENT_HOOK);
1688 static void fake_global_assign(struct symbol *sym)
1690 struct expression *assign, *symbol;
1692 if (get_real_base_type(sym)->type == SYM_ARRAY) {
1693 if (sym->initializer && sym->initializer->type == EXPR_INITIALIZER) {
1694 fake_element_assigns(sym, call_global_assign_hooks);
1695 } else if (sym->initializer) {
1696 symbol = symbol_expression(sym);
1697 assign = assign_expression(symbol, '=', sym->initializer);
1698 __pass_to_client(assign, GLOBAL_ASSIGNMENT_HOOK);
1699 } else {
1700 fake_element_assigns_helper(symbol_expression(sym), NULL, call_global_assign_hooks);
1702 } else if (get_real_base_type(sym)->type == SYM_STRUCT) {
1703 if (sym->initializer && sym->initializer->type == EXPR_INITIALIZER) {
1704 fake_member_assigns(sym, call_global_assign_hooks);
1705 } else if (sym->initializer) {
1706 symbol = symbol_expression(sym);
1707 assign = assign_expression(symbol, '=', sym->initializer);
1708 __pass_to_client(assign, GLOBAL_ASSIGNMENT_HOOK);
1709 } else {
1710 fake_member_assigns_helper(symbol_expression(sym), NULL, call_global_assign_hooks);
1712 } else {
1713 symbol = symbol_expression(sym);
1714 if (sym->initializer) {
1715 assign = assign_expression(symbol, '=', sym->initializer);
1716 __split_expr(assign);
1717 } else {
1718 assign = assign_expression(symbol, '=', zero_expr());
1720 __pass_to_client(assign, GLOBAL_ASSIGNMENT_HOOK);
1724 static void start_function_definition(struct symbol *sym)
1726 __in_function_def = 1;
1727 __pass_to_client(sym, FUNC_DEF_HOOK);
1728 __in_function_def = 0;
1729 __pass_to_client(sym, AFTER_DEF_HOOK);
1733 void add_function_data(unsigned long *fn_data)
1735 __add_ptr_list(&fn_data_list, fn_data);
1738 static void clear_function_data(void)
1740 unsigned long *tmp;
1742 FOR_EACH_PTR(fn_data_list, tmp) {
1743 *tmp = 0;
1744 } END_FOR_EACH_PTR(tmp);
1747 static void record_func_time(void)
1749 struct timeval stop;
1750 int func_time;
1751 char buf[32];
1753 gettimeofday(&stop, NULL);
1754 func_time = stop.tv_sec - fn_start_time.tv_sec;
1755 snprintf(buf, sizeof(buf), "%d", func_time);
1756 sql_insert_return_implies(FUNC_TIME, 0, "", buf);
1757 if (option_time && func_time > 2) {
1758 final_pass++;
1759 sm_msg("func_time: %d", func_time);
1760 final_pass--;
1764 static void split_function(struct symbol *sym)
1766 struct symbol *base_type = get_base_type(sym);
1768 if (!base_type->stmt && !base_type->inline_stmt)
1769 return;
1771 gettimeofday(&outer_fn_start_time, NULL);
1772 gettimeofday(&fn_start_time, NULL);
1773 cur_func_sym = sym;
1774 if (sym->ident)
1775 cur_func = sym->ident->name;
1776 set_position(sym->pos);
1777 clear_function_data();
1778 loop_count = 0;
1779 last_goto_statement_handled = 0;
1780 sm_debug("new function: %s\n", cur_func);
1781 __stree_id = 0;
1782 if (option_two_passes) {
1783 __unnullify_path();
1784 loop_num = 0;
1785 final_pass = 0;
1786 start_function_definition(sym);
1787 __split_stmt(base_type->stmt);
1788 __split_stmt(base_type->inline_stmt);
1789 nullify_path();
1791 __unnullify_path();
1792 loop_num = 0;
1793 final_pass = 1;
1794 start_function_definition(sym);
1795 __split_stmt(base_type->stmt);
1796 __split_stmt(base_type->inline_stmt);
1797 __pass_to_client(sym, END_FUNC_HOOK);
1798 if (need_delayed_scope_hooks())
1799 __call_scope_hooks();
1800 __pass_to_client(sym, AFTER_FUNC_HOOK);
1801 sym->parsed = true;
1803 clear_all_states();
1805 record_func_time();
1807 cur_func_sym = NULL;
1808 cur_func = NULL;
1809 free_data_info_allocs();
1810 free_expression_stack(&switch_expr_stack);
1811 __free_ptr_list((struct ptr_list **)&big_statement_stack);
1812 __bail_on_rest_of_function = 0;
1815 static void save_flow_state(void)
1817 unsigned long *tmp;
1819 __add_ptr_list(&backup, INT_PTR(loop_num << 2));
1820 __add_ptr_list(&backup, INT_PTR(loop_count << 2));
1821 __add_ptr_list(&backup, INT_PTR(final_pass << 2));
1823 __add_ptr_list(&backup, big_statement_stack);
1824 __add_ptr_list(&backup, big_expression_stack);
1825 __add_ptr_list(&backup, big_condition_stack);
1826 __add_ptr_list(&backup, switch_expr_stack);
1828 __add_ptr_list(&backup, cur_func_sym);
1830 __add_ptr_list(&backup, __prev_stmt);
1831 __add_ptr_list(&backup, __cur_stmt);
1832 __add_ptr_list(&backup, __next_stmt);
1834 FOR_EACH_PTR(fn_data_list, tmp) {
1835 __add_ptr_list(&backup, (void *)*tmp);
1836 } END_FOR_EACH_PTR(tmp);
1839 static void *pop_backup(void)
1841 void *ret;
1843 ret = last_ptr_list(backup);
1844 delete_ptr_list_last(&backup);
1845 return ret;
1848 static void restore_flow_state(void)
1850 unsigned long *tmp;
1852 FOR_EACH_PTR_REVERSE(fn_data_list, tmp) {
1853 *tmp = (unsigned long)pop_backup();
1854 } END_FOR_EACH_PTR_REVERSE(tmp);
1856 __next_stmt = pop_backup();
1857 __cur_stmt = pop_backup();
1858 __prev_stmt = pop_backup();
1860 cur_func_sym = pop_backup();
1861 switch_expr_stack = pop_backup();
1862 big_condition_stack = pop_backup();
1863 big_expression_stack = pop_backup();
1864 big_statement_stack = pop_backup();
1865 final_pass = PTR_INT(pop_backup()) >> 2;
1866 loop_count = PTR_INT(pop_backup()) >> 2;
1867 loop_num = PTR_INT(pop_backup()) >> 2;
1870 static void parse_inline(struct expression *call)
1872 struct symbol *base_type;
1873 char *cur_func_bak = cur_func; /* not aligned correctly for backup */
1874 struct timeval time_backup = fn_start_time;
1875 struct expression *orig_inline = __inline_fn;
1876 int orig_budget;
1878 if (out_of_memory() || taking_too_long())
1879 return;
1881 save_flow_state();
1883 __pass_to_client(call, INLINE_FN_START);
1884 final_pass = 0; /* don't print anything */
1885 __inline_fn = call;
1886 orig_budget = inline_budget;
1887 inline_budget = inline_budget - 5;
1889 base_type = get_base_type(call->fn->symbol);
1890 cur_func_sym = call->fn->symbol;
1891 if (call->fn->symbol->ident)
1892 cur_func = call->fn->symbol->ident->name;
1893 else
1894 cur_func = NULL;
1895 set_position(call->fn->symbol->pos);
1897 save_all_states();
1898 big_statement_stack = NULL;
1899 big_expression_stack = NULL;
1900 big_condition_stack = NULL;
1901 switch_expr_stack = NULL;
1903 sm_debug("inline function: %s\n", cur_func);
1904 __unnullify_path();
1905 clear_function_data();
1906 loop_num = 0;
1907 loop_count = 0;
1908 start_function_definition(call->fn->symbol);
1909 __split_stmt(base_type->stmt);
1910 __split_stmt(base_type->inline_stmt);
1911 __pass_to_client(call->fn->symbol, END_FUNC_HOOK);
1912 __pass_to_client(call->fn->symbol, AFTER_FUNC_HOOK);
1913 call->fn->symbol->parsed = true;
1915 free_expression_stack(&switch_expr_stack);
1916 __free_ptr_list((struct ptr_list **)&big_statement_stack);
1917 nullify_path();
1918 free_goto_stack();
1920 restore_flow_state();
1921 fn_start_time = time_backup;
1922 cur_func = cur_func_bak;
1924 restore_all_states();
1925 set_position(call->pos);
1926 __inline_fn = orig_inline;
1927 inline_budget = orig_budget;
1928 __pass_to_client(call, INLINE_FN_END);
1931 static struct symbol_list *inlines_called;
1932 static void add_inline_function(struct symbol *sym)
1934 static struct symbol_list *already_added;
1935 struct symbol *tmp;
1937 FOR_EACH_PTR(already_added, tmp) {
1938 if (tmp == sym)
1939 return;
1940 } END_FOR_EACH_PTR(tmp);
1942 add_ptr_list(&already_added, sym);
1943 add_ptr_list(&inlines_called, sym);
1946 static void process_inlines(void)
1948 struct symbol *tmp;
1950 FOR_EACH_PTR(inlines_called, tmp) {
1951 split_function(tmp);
1952 } END_FOR_EACH_PTR(tmp);
1953 free_ptr_list(&inlines_called);
1956 static struct symbol *get_last_scoped_symbol(struct symbol_list *big_list, int use_static)
1958 struct symbol *sym;
1960 FOR_EACH_PTR_REVERSE(big_list, sym) {
1961 if (!sym->scope)
1962 continue;
1963 if (use_static && sym->ctype.modifiers & MOD_STATIC)
1964 return sym;
1965 if (!use_static && !(sym->ctype.modifiers & MOD_STATIC))
1966 return sym;
1967 } END_FOR_EACH_PTR_REVERSE(sym);
1969 return NULL;
1972 static bool interesting_function(struct symbol *sym)
1974 static int prev_stream = -1;
1975 static bool prev_answer;
1976 const char *filename;
1977 int len;
1979 if (!(sym->ctype.modifiers & MOD_INLINE))
1980 return true;
1982 if (sym->pos.stream == prev_stream)
1983 return prev_answer;
1985 prev_stream = sym->pos.stream;
1986 prev_answer = false;
1988 filename = stream_name(sym->pos.stream);
1989 len = strlen(filename);
1990 if (len > 0 && filename[len - 1] == 'c')
1991 prev_answer = true;
1992 return prev_answer;
1995 static void split_inlines_in_scope(struct symbol *sym)
1997 struct symbol *base;
1998 struct symbol_list *scope_list;
1999 int stream;
2001 scope_list = sym->scope->symbols;
2002 stream = sym->pos.stream;
2004 /* find the last static symbol in the file */
2005 FOR_EACH_PTR_REVERSE(scope_list, sym) {
2006 if (sym->pos.stream != stream)
2007 continue;
2008 if (sym->type != SYM_NODE)
2009 continue;
2010 base = get_base_type(sym);
2011 if (!base)
2012 continue;
2013 if (base->type != SYM_FN)
2014 continue;
2015 if (!base->inline_stmt)
2016 continue;
2017 if (!interesting_function(sym))
2018 continue;
2019 add_inline_function(sym);
2020 } END_FOR_EACH_PTR_REVERSE(sym);
2022 process_inlines();
2025 static void split_inlines(struct symbol_list *sym_list)
2027 struct symbol *sym;
2029 sym = get_last_scoped_symbol(sym_list, 0);
2030 if (sym)
2031 split_inlines_in_scope(sym);
2032 sym = get_last_scoped_symbol(sym_list, 1);
2033 if (sym)
2034 split_inlines_in_scope(sym);
2037 static struct stree *clone_estates_perm(struct stree *orig)
2039 struct stree *ret = NULL;
2040 struct sm_state *tmp;
2042 FOR_EACH_SM(orig, tmp) {
2043 set_state_stree_perm(&ret, tmp->owner, tmp->name, tmp->sym, clone_estate_perm(tmp->state));
2044 } END_FOR_EACH_SM(tmp);
2046 return ret;
2049 struct position last_pos;
2050 static void split_c_file_functions(struct symbol_list *sym_list)
2052 struct symbol *sym;
2054 __unnullify_path();
2055 FOR_EACH_PTR(sym_list, sym) {
2056 set_position(sym->pos);
2057 if (sym->type != SYM_NODE || get_base_type(sym)->type != SYM_FN) {
2058 __pass_to_client(sym, BASE_HOOK);
2059 fake_global_assign(sym);
2061 } END_FOR_EACH_PTR(sym);
2062 global_states = clone_estates_perm(get_all_states_stree(SMATCH_EXTRA));
2063 nullify_path();
2065 FOR_EACH_PTR(sym_list, sym) {
2066 set_position(sym->pos);
2067 last_pos = sym->pos;
2068 if (!interesting_function(sym))
2069 continue;
2070 if (sym->type == SYM_NODE && get_base_type(sym)->type == SYM_FN) {
2071 split_function(sym);
2072 process_inlines();
2074 last_pos = sym->pos;
2075 } END_FOR_EACH_PTR(sym);
2076 split_inlines(sym_list);
2077 __pass_to_client(sym_list, END_FILE_HOOK);
2080 static int final_before_fake;
2081 void init_fake_env(void)
2083 if (!in_fake_env)
2084 final_before_fake = final_pass;
2085 in_fake_env++;
2086 __push_fake_cur_stree();
2087 final_pass = 0;
2090 void end_fake_env(void)
2092 __pop_fake_cur_stree();
2093 in_fake_env--;
2094 if (!in_fake_env)
2095 final_pass = final_before_fake;
2098 static void open_output_files(char *base_file)
2100 char buf[256];
2102 snprintf(buf, sizeof(buf), "%s.smatch", base_file);
2103 sm_outfd = fopen(buf, "w");
2104 if (!sm_outfd)
2105 sm_fatal("Cannot open %s", buf);
2107 if (!option_info)
2108 return;
2110 snprintf(buf, sizeof(buf), "%s.smatch.sql", base_file);
2111 sql_outfd = fopen(buf, "w");
2112 if (!sql_outfd)
2113 sm_fatal("Error: Cannot open %s", buf);
2115 snprintf(buf, sizeof(buf), "%s.smatch.caller_info", base_file);
2116 caller_info_fd = fopen(buf, "w");
2117 if (!caller_info_fd)
2118 sm_fatal("Error: Cannot open %s", buf);
2121 void smatch(struct string_list *filelist)
2123 struct symbol_list *sym_list;
2124 struct timeval stop, start;
2125 char *path;
2126 int len;
2128 gettimeofday(&start, NULL);
2130 FOR_EACH_PTR_NOTAG(filelist, base_file) {
2131 path = getcwd(NULL, 0);
2132 free(full_base_file);
2133 if (path) {
2134 len = strlen(path) + 1 + strlen(base_file) + 1;
2135 full_base_file = malloc(len);
2136 snprintf(full_base_file, len, "%s/%s", path, base_file);
2137 } else {
2138 full_base_file = alloc_string(base_file);
2140 if (option_file_output)
2141 open_output_files(base_file);
2142 sym_list = sparse_keep_tokens(base_file);
2143 split_c_file_functions(sym_list);
2144 } END_FOR_EACH_PTR_NOTAG(base_file);
2146 gettimeofday(&stop, NULL);
2148 set_position(last_pos);
2149 final_pass = 1;
2150 if (option_time)
2151 sm_msg("time: %lu", stop.tv_sec - start.tv_sec);
2152 if (option_mem)
2153 sm_msg("mem: %luKb", get_max_memory());