fill_db_untrusted: follow untrusted data down the call tree
[smatch.git] / smatch_hooks.c
blob218b9b5e623f86ce1858a3fe5aa088773e7cffb9
1 /*
2 * sparse/smatch_hooks.c
4 * Copyright (C) 2006 Dan Carpenter.
6 * Licensed under the Open Software License version 1.1
8 */
10 #include "smatch.h"
12 enum data_type {
13 EXPR_PTR,
14 STMT_PTR,
15 SYMBOL_PTR,
18 struct hook_container {
19 int hook_type;
20 enum data_type data_type;
21 void *fn;
23 ALLOCATOR(hook_container, "hook functions");
24 DECLARE_PTR_LIST(hook_func_list, struct hook_container);
25 static struct hook_func_list *hook_funcs;
26 static struct hook_func_list *merge_funcs;
27 static struct hook_func_list *unmatched_state_funcs;
29 struct scope_container {
30 void *fn;
31 void *data;
33 ALLOCATOR(scope_container, "scope hook functions");
34 DECLARE_PTR_LIST(scope_hook_list, struct scope_container);
35 DECLARE_PTR_LIST(scope_hook_stack, struct scope_hook_list);
36 static struct scope_hook_stack *scope_hooks;
38 void add_hook(void *func, enum hook_type type)
40 struct hook_container *container = __alloc_hook_container(0);
41 container->hook_type = type;
42 container->fn = func;
43 switch (type) {
44 case EXPR_HOOK:
45 container->data_type = EXPR_PTR;
46 break;
47 case STMT_HOOK:
48 container->data_type = STMT_PTR;
49 break;
50 case SYM_HOOK:
51 container->data_type = EXPR_PTR;
52 break;
53 case STRING_HOOK:
54 container->data_type = EXPR_PTR;
55 break;
56 case DECLARATION_HOOK:
57 container->data_type = SYMBOL_PTR;
58 break;
59 case ASSIGNMENT_HOOK:
60 container->data_type = EXPR_PTR;
61 break;
62 case RAW_ASSIGNMENT_HOOK:
63 container->data_type = EXPR_PTR;
64 break;
65 case CALL_ASSIGNMENT_HOOK:
66 container->data_type = EXPR_PTR;
67 break;
68 case MACRO_ASSIGNMENT_HOOK:
69 container->data_type = EXPR_PTR;
70 break;
71 case BINOP_HOOK:
72 container->data_type = EXPR_PTR;
73 break;
74 case OP_HOOK:
75 container->data_type = EXPR_PTR;
76 break;
77 case LOGIC_HOOK:
78 container->data_type = EXPR_PTR;
79 break;
80 case PRELOOP_HOOK:
81 container->data_type = STMT_PTR;
82 break;
83 case CONDITION_HOOK:
84 container->data_type = EXPR_PTR;
85 break;
86 case SELECT_HOOK:
87 container->data_type = EXPR_PTR;
88 break;
89 case WHOLE_CONDITION_HOOK:
90 container->data_type = EXPR_PTR;
91 break;
92 case FUNCTION_CALL_HOOK:
93 container->data_type = EXPR_PTR;
94 break;
95 case DEREF_HOOK:
96 container->data_type = EXPR_PTR;
97 break;
98 case CASE_HOOK:
99 /* nothing needed */
100 break;
101 case ASM_HOOK:
102 container->data_type = STMT_PTR;
103 break;
104 case BASE_HOOK:
105 container->data_type = SYMBOL_PTR;
106 break;
107 case FUNC_DEF_HOOK:
108 container->data_type = SYMBOL_PTR;
109 break;
110 case END_FUNC_HOOK:
111 container->data_type = SYMBOL_PTR;
112 break;
113 case RETURN_HOOK:
114 container->data_type = EXPR_PTR;
115 break;
116 case END_FILE_HOOK:
117 /* nothing needed... */
118 break;
120 add_ptr_list(&hook_funcs, container);
123 void add_merge_hook(int client_id, merge_func_t *func)
125 struct hook_container *container = __alloc_hook_container(0);
126 container->data_type = client_id;
127 container->fn = func;
128 add_ptr_list(&merge_funcs, container);
131 void add_unmatched_state_hook(int client_id, unmatched_func_t *func)
133 struct hook_container *container = __alloc_hook_container(0);
134 container->data_type = client_id;
135 container->fn = func;
136 add_ptr_list(&unmatched_state_funcs, container);
139 static void pass_to_client(void * fn)
141 typedef void (expr_func)();
142 ((expr_func *) fn)();
145 static void pass_expr_to_client(void * fn, void * data)
147 typedef void (expr_func)(struct expression *expr);
148 ((expr_func *) fn)((struct expression *) data);
151 static void pass_stmt_to_client(void * fn, void * data)
153 typedef void (stmt_func)(struct statement *stmt);
154 ((stmt_func *) fn)((struct statement *) data);
157 static void pass_sym_to_client(void * fn, void * data)
159 typedef void (sym_func)(struct symbol *sym);
160 ((sym_func *) fn)((struct symbol *) data);
163 void __pass_to_client(void *data, enum hook_type type)
165 struct hook_container *container;
167 FOR_EACH_PTR(hook_funcs, container) {
168 if (container->hook_type == type) {
169 switch (container->data_type) {
170 case EXPR_PTR:
171 pass_expr_to_client(container->fn, data);
172 break;
173 case STMT_PTR:
174 pass_stmt_to_client(container->fn, data);
175 break;
176 case SYMBOL_PTR:
177 pass_sym_to_client(container->fn, data);
178 break;
181 } END_FOR_EACH_PTR(container);
184 void __pass_to_client_no_data(enum hook_type type)
186 struct hook_container *container;
188 FOR_EACH_PTR(hook_funcs, container) {
189 if (container->hook_type == type)
190 pass_to_client(container->fn);
191 } END_FOR_EACH_PTR(container);
194 void __pass_case_to_client(struct expression *switch_expr,
195 struct expression *case_expr)
197 typedef void (case_func)(struct expression *switch_expr,
198 struct expression *case_expr);
199 struct hook_container *container;
201 FOR_EACH_PTR(hook_funcs, container) {
202 if (container->hook_type == CASE_HOOK) {
203 ((case_func *) container->fn)(switch_expr, case_expr);
205 } END_FOR_EACH_PTR(container);
208 int __has_merge_function(int client_id)
210 struct hook_container *tmp;
212 FOR_EACH_PTR(merge_funcs, tmp) {
213 if (tmp->data_type == client_id)
214 return 1;
215 } END_FOR_EACH_PTR(tmp);
216 return 0;
219 struct smatch_state *__client_merge_function(int owner, const char *name,
220 struct symbol *sym,
221 struct smatch_state *s1,
222 struct smatch_state *s2)
224 struct smatch_state *tmp_state;
225 struct hook_container *tmp;
227 /* Pass NULL states first and the rest alphabetically by name */
228 if (!s2 || (s1 && strcmp(s2->name, s1->name) < 0)) {
229 tmp_state = s1;
230 s1 = s2;
231 s2 = tmp_state;
234 FOR_EACH_PTR(merge_funcs, tmp) {
235 if (tmp->data_type == owner)
236 return ((merge_func_t *) tmp->fn)(name, sym, s1, s2);
237 } END_FOR_EACH_PTR(tmp);
238 return &undefined;
241 struct smatch_state *__client_unmatched_state_function(struct sm_state *sm)
243 struct hook_container *tmp;
245 FOR_EACH_PTR(unmatched_state_funcs, tmp) {
246 if (tmp->data_type == sm->owner)
247 return ((unmatched_func_t *) tmp->fn)(sm);
248 } END_FOR_EACH_PTR(tmp);
249 return &undefined;
252 static struct scope_hook_list *pop_scope_hook_list(struct scope_hook_stack **stack)
254 struct scope_hook_list *hook_list;
256 hook_list = last_ptr_list((struct ptr_list *)*stack);
257 delete_ptr_list_last((struct ptr_list **)stack);
258 return hook_list;
261 static void push_scope_hook_list(struct scope_hook_stack **stack, struct scope_hook_list *l)
263 add_ptr_list(stack, l);
266 void add_scope_hook(scope_hook *fn, void *data)
268 struct scope_hook_list *hook_list;
269 struct scope_container *new;
271 if (!scope_hooks)
272 return;
273 hook_list = pop_scope_hook_list(&scope_hooks);
274 new = __alloc_scope_container(0);
275 new->fn = fn;
276 new->data = data;
277 add_ptr_list(&hook_list, new);
278 push_scope_hook_list(&scope_hooks, hook_list);
281 void __push_scope_hooks(void)
283 push_scope_hook_list(&scope_hooks, NULL);
286 void __call_scope_hooks(void)
288 struct scope_hook_list *hook_list;
289 struct scope_container *tmp;
291 if (!scope_hooks)
292 return;
294 hook_list = pop_scope_hook_list(&scope_hooks);
295 FOR_EACH_PTR(hook_list, tmp) {
296 ((scope_hook *) tmp->fn)(tmp->data);
297 __free_scope_container(tmp);
298 } END_FOR_EACH_PTR(tmp);