smatch_hooks.c

   1 /*
   2  * sparse/smatch_hooks.c
   3  *
   4  * Copyright (C) 2006 Dan Carpenter.
   5  *
   6  *  Licensed under the Open Software License version 1.1
   7  *
   8  */
   9
  10 #include "smatch.h"
  11
  12 struct hook_container {
  13         int hook_type;
  14         int data_type;
  15         void * fn;
  16 };
  17 ALLOCATOR(hook_container, "hook functions");
  18 DECLARE_PTR_LIST(hook_func_list, struct hook_container);
  19 static struct hook_func_list *hook_funcs;
  20 static struct hook_func_list *merge_funcs;
  21
  22 void add_hook(void *func, enum hook_type type)
  23 {
  24         struct hook_container *container = __alloc_hook_container(0);
  25         container->hook_type = type;
  26         container->fn = func;
  27         switch(type) {
  28         case EXPR_HOOK:
  29                 container->data_type = EXPR_HOOK;
  30                 break;
  31         case STMT_HOOK:
  32                 container->data_type = STMT_HOOK;
  33                 break;
  34         case SYM_HOOK:
  35                 container->data_type = SYM_HOOK;
  36                 break;
  37         case DECLARATION_HOOK:
  38                 container->data_type = SYM_HOOK;
  39                 break;
  40         case ASSIGNMENT_HOOK:
  41                 container->data_type = EXPR_HOOK;
  42                 break;
  43         case OP_HOOK:
  44                 container->data_type = EXPR_HOOK;
  45                 break;
  46         case ASSIGNMENT_AFTER_HOOK:
  47                 container->data_type = EXPR_HOOK;
  48                 break;
  49         case CONDITION_HOOK:
  50                 container->data_type = EXPR_HOOK;
  51                 break;
  52         case WHOLE_CONDITION_HOOK:
  53                 container->data_type = EXPR_HOOK;
  54                 break;
  55         case FUNCTION_CALL_HOOK:
  56                 container->data_type = EXPR_HOOK;
  57                 break;
  58         case FUNCTION_CALL_AFTER_HOOK:
  59                 container->data_type = EXPR_HOOK;
  60                 break;
  61         case DEREF_HOOK:
  62                 container->data_type = EXPR_HOOK;
  63                 break;
  64         case BASE_HOOK:
  65                 container->data_type = SYM_HOOK;
  66                 break;
  67         case FUNC_DEF_HOOK:
  68                 container->data_type = SYM_HOOK;
  69                 break;
  70         case END_FUNC_HOOK:
  71                 container->data_type = SYM_HOOK;
  72                 break;
  73         case RETURN_HOOK:
  74                 container->data_type = STMT_HOOK;
  75                 break;
  76         }
  77         add_ptr_list(&hook_funcs, container);
  78 }
  79
  80 void add_merge_hook(int client_id, merge_func_t *func)
  81 {
  82         struct hook_container *container = __alloc_hook_container(0);
  83         container->data_type = client_id;
  84         container->fn = func;
  85         add_ptr_list(&merge_funcs, container);
  86 }
  87
  88
  89 static void pass_expr_to_client(void * fn, void * data)
  90 {
  91         typedef void (expr_func)(struct expression *expr);
  92         ((expr_func *) fn)((struct expression *) data);
  93 }
  94
  95 static void pass_stmt_to_client(void * fn, void * data)
  96 {
  97         typedef void (stmt_func)(struct statement *stmt);
  98         ((stmt_func *) fn)((struct statement *) data);
  99 }
 100
 101 static void pass_sym_to_client(void * fn, void * data)
 102 {
 103         typedef void (sym_func)(struct symbol *sym);
 104         ((sym_func *) fn)((struct symbol *) data);
 105 }
 106
 107 void __pass_to_client(void *data, enum hook_type type)
 108 {
 109         struct hook_container *container;
 110
 111         if (!data)
 112                 return;
 113
 114         FOR_EACH_PTR(hook_funcs, container) {
 115                 if (container->hook_type == type) {
 116                         switch(container->data_type) {
 117                         case EXPR_HOOK:
 118                                 pass_expr_to_client(container->fn, data);
 119                                 break;
 120                         case STMT_HOOK:
 121                                 pass_stmt_to_client(container->fn, data);
 122                                 break;
 123                         case SYM_HOOK:
 124                                 pass_sym_to_client(container->fn, data);
 125                                 break;
 126                         }
 127                 }
 128         } END_FOR_EACH_PTR(container);
 129 }
 130
 131 void __pass_declarations_to_client(struct symbol_list *sym_list)
 132 {
 133         struct symbol *sym;
 134         FOR_EACH_PTR(sym_list, sym) {
 135                 __pass_to_client(sym, DECLARATION_HOOK);
 136         } END_FOR_EACH_PTR(sym);
 137 }
 138
 139 int __has_merge_function(int client_id)
 140 {
 141         struct hook_container *tmp;
 142
 143         FOR_EACH_PTR(merge_funcs, tmp) {
 144                 if (tmp->data_type == client_id)
 145                         return 1;
 146         } END_FOR_EACH_PTR(tmp);
 147         return 0;
 148 }
 149
 150 int __client_merge_function(int owner, const char *name, struct symbol *sym,
 151                             int s1, int s2)
 152 {
 153         struct hook_container *tmp;
 154
 155         FOR_EACH_PTR(merge_funcs, tmp) {
 156                 if (tmp->data_type == owner)
 157                         return ((merge_func_t *) tmp->fn)(name, sym, s1, s2);
 158         } END_FOR_EACH_PTR(tmp);
 159         return UNDEFINED;
 160 }