smatch_returns.c

   1 /*
   2  * smatch/smatch_returns.c
   3  *
   4  * Copyright (C) 2011 Oracle.
   5  *
   6  * Licensed under the Open Software License version 1.1
   7  *
   8  */
   9
  10 #include "smatch.h"
  11 #include "smatch_slist.h"
  12
  13 static int my_id;
  14
  15 struct return_states_callback {
  16         void (*callback)(struct state_list *slist);
  17 };
  18 ALLOCATOR(return_states_callback, "return states callbacks");
  19 DECLARE_PTR_LIST(callback_list, struct return_states_callback);
  20 static struct callback_list *callback_list;
  21
  22 static struct state_list *all_return_states;
  23
  24 void all_return_states_hook(void (*callback)(struct state_list *slist))
  25 {
  26         struct return_states_callback *rs_cb = __alloc_return_states_callback(0);
  27
  28         rs_cb->callback = callback;
  29         add_ptr_list(&callback_list, rs_cb);
  30 }
  31
  32 static void call_hooks()
  33 {
  34         struct return_states_callback *rs_cb;
  35
  36         FOR_EACH_PTR(callback_list, rs_cb) {
  37                 rs_cb->callback(all_return_states);
  38         } END_FOR_EACH_PTR(rs_cb);
  39 }
  40
  41 static void match_return(struct expression *ret_value)
  42 {
  43         if (__inline_fn)
  44                 return;
  45         merge_slist(&all_return_states, __get_cur_slist());
  46 }
  47
  48 static void match_end_func(struct symbol *sym)
  49 {
  50         if (__inline_fn)
  51                 return;
  52         merge_slist(&all_return_states, __get_cur_slist());
  53         call_hooks();
  54         free_slist(&all_return_states);
  55 }
  56
  57 void register_returns(int id)
  58 {
  59         my_id = id;
  60
  61         add_hook(&match_return, RETURN_HOOK);
  62         add_hook(&match_end_func, END_FUNC_HOOK);
  63 }