2 * Copyright (C) 2011 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
19 * There are a couple checks that try to see if a variable
20 * comes from the user. It would be better to unify them
21 * into one place. Also it we should follow the data down
22 * the call paths. Hence this file.
26 #include "smatch_slist.h"
27 #include "smatch_extra.h"
30 static int my_call_id
;
34 static const char * kstr_funcs
[] = {
35 "kstrtoull", "kstrtoll", "kstrtoul", "kstrtol", "kstrtouint",
36 "kstrtoint", "kstrtou64", "kstrtos64", "kstrtou32", "kstrtos32",
37 "kstrtou16", "kstrtos16", "kstrtou8", "kstrtos8", "kstrtoull_from_user"
38 "kstrtoll_from_user", "kstrtoul_from_user", "kstrtol_from_user",
39 "kstrtouint_from_user", "kstrtoint_from_user", "kstrtou16_from_user",
40 "kstrtos16_from_user", "kstrtou8_from_user", "kstrtos8_from_user",
41 "kstrtou64_from_user", "kstrtos64_from_user", "kstrtou32_from_user",
42 "kstrtos32_from_user",
45 static struct stree
*start_states
;
46 static struct stree_stack
*saved_stack
;
47 static void save_start_states(struct statement
*stmt
)
49 start_states
= clone_stree(__get_cur_stree());
52 static void free_start_states(void)
54 free_stree(&start_states
);
57 static void match_save_states(struct expression
*expr
)
59 push_stree(&saved_stack
, start_states
);
63 static void match_restore_states(struct expression
*expr
)
65 free_stree(&start_states
);
66 start_states
= pop_stree(&saved_stack
);
69 static struct smatch_state
*empty_state(struct sm_state
*sm
)
71 return alloc_estate_empty();
74 static void pre_merge_hook(struct sm_state
*sm
)
76 struct smatch_state
*user
;
77 struct smatch_state
*extra
;
78 struct range_list
*rl
;
80 extra
= get_state(SMATCH_EXTRA
, sm
->name
, sm
->sym
);
81 if (!extra
|| !estate_rl(extra
))
83 user
= get_state(my_id
, sm
->name
, sm
->sym
);
84 if (!user
|| !estate_rl(user
))
86 rl
= rl_intersection(estate_rl(user
), estate_rl(extra
));
87 set_state(my_id
, sm
->name
, sm
->sym
, alloc_estate_rl(clone_rl(rl
)));
90 static void tag_inner_struct_members(struct expression
*expr
, struct symbol
*member
)
92 struct expression
*edge_member
;
93 struct symbol
*base
= get_real_base_type(member
);
97 expr
= member_expression(expr
, '.', member
->ident
);
99 FOR_EACH_PTR(base
->symbol_list
, tmp
) {
102 type
= get_real_base_type(tmp
);
106 if (type
->type
== SYM_UNION
|| type
->type
== SYM_STRUCT
) {
107 tag_inner_struct_members(expr
, tmp
);
114 edge_member
= member_expression(expr
, '.', tmp
->ident
);
115 set_state_expr(my_id
, edge_member
, alloc_estate_whole(type
));
116 } END_FOR_EACH_PTR(tmp
);
121 static void tag_struct_members(struct symbol
*type
, struct expression
*expr
)
124 struct expression
*member
;
127 if (expr
->type
== EXPR_PREOP
&& expr
->op
== '&') {
128 expr
= strip_expr(expr
->unop
);
132 FOR_EACH_PTR(type
->symbol_list
, tmp
) {
133 type
= get_real_base_type(tmp
);
137 if (type
->type
== SYM_UNION
|| type
->type
== SYM_STRUCT
) {
138 tag_inner_struct_members(expr
, tmp
);
145 member
= member_expression(expr
, op
, tmp
->ident
);
146 set_state_expr(my_id
, member
, alloc_estate_whole(get_type(member
)));
147 } END_FOR_EACH_PTR(tmp
);
150 static void tag_base_type(struct expression
*expr
)
152 if (expr
->type
== EXPR_PREOP
&& expr
->op
== '&')
153 expr
= strip_expr(expr
->unop
);
155 expr
= deref_expression(expr
);
156 set_state_expr(my_id
, expr
, alloc_estate_whole(get_type(expr
)));
159 static void tag_as_user_data(struct expression
*expr
)
163 expr
= strip_expr(expr
);
165 type
= get_type(expr
);
166 if (!type
|| type
->type
!= SYM_PTR
)
168 type
= get_real_base_type(type
);
171 if (type
== &void_ctype
) {
172 set_state_expr(my_id
, deref_expression(expr
), alloc_estate_whole(&ulong_ctype
));
175 if (type
->type
== SYM_BASETYPE
)
177 if (type
->type
== SYM_STRUCT
) {
178 if (expr
->type
!= EXPR_PREOP
|| expr
->op
!= '&')
179 expr
= deref_expression(expr
);
180 tag_struct_members(type
, expr
);
184 static void match_user_copy(const char *fn
, struct expression
*expr
, void *_param
)
186 int param
= PTR_INT(_param
);
187 struct expression
*dest
;
189 dest
= get_argument_from_call_expr(expr
->args
, param
);
190 dest
= strip_expr(dest
);
193 tag_as_user_data(dest
);
196 static void match_sscanf(const char *fn
, struct expression
*expr
, void *unused
)
198 struct expression
*arg
;
202 FOR_EACH_PTR(expr
->args
, arg
) {
206 tag_as_user_data(arg
);
207 } END_FOR_EACH_PTR(arg
);
210 static int points_to_user_data(struct expression
*expr
)
212 struct smatch_state
*state
;
218 expr
= strip_expr(expr
);
220 if (expr
->type
== EXPR_BINOP
&& expr
->op
== '+') {
221 if (points_to_user_data(expr
->left
))
223 if (points_to_user_data(expr
->right
))
228 name
= expr_to_var_sym(expr
, &sym
);
231 snprintf(buf
, sizeof(buf
), "*%s", name
);
232 state
= get_state(my_id
, buf
, sym
);
233 if (state
&& estate_rl(state
))
240 static void set_points_to_user_data(struct expression
*expr
)
246 name
= expr_to_var_sym(expr
, &sym
);
249 snprintf(buf
, sizeof(buf
), "*%s", name
);
250 set_state(my_id
, buf
, sym
, alloc_estate_whole(&llong_ctype
));
255 static int is_skb_data(struct expression
*expr
)
259 expr
= strip_expr(expr
);
260 if (!expr
|| expr
->type
!= EXPR_DEREF
)
265 if (strcmp(expr
->member
->name
, "data") != 0)
268 sym
= expr_to_sym(expr
->deref
);
271 sym
= get_real_base_type(sym
);
272 if (!sym
|| sym
->type
!= SYM_PTR
)
274 sym
= get_real_base_type(sym
);
275 if (!sym
|| sym
->type
!= SYM_STRUCT
|| !sym
->ident
)
277 if (strcmp(sym
->ident
->name
, "sk_buff") != 0)
283 static int comes_from_skb_data(struct expression
*expr
)
285 expr
= strip_expr(expr
);
289 switch (expr
->type
) {
291 if (comes_from_skb_data(expr
->left
))
293 if (comes_from_skb_data(expr
->right
))
297 return comes_from_skb_data(expr
->unop
);
299 if (is_skb_data(expr
))
301 return comes_from_skb_data(expr
->deref
);
308 static int handle_struct_assignment(struct expression
*expr
)
310 struct expression
*right
;
313 type
= get_type(expr
->left
);
314 if (!type
|| type
->type
!= SYM_PTR
)
316 type
= get_real_base_type(type
);
317 if (!type
|| type
->type
!= SYM_STRUCT
)
321 * Ignore struct to struct assignments because for those we look at the
322 * individual members.
324 right
= strip_expr(expr
->right
);
325 type
= get_type(right
);
326 if (!type
|| type
->type
!= SYM_PTR
)
329 /* structs are handled else where */
330 type
= get_real_base_type(type
);
331 if (type
&& type
->type
== SYM_STRUCT
)
334 if (!points_to_user_data(right
) && !is_skb_data(right
))
337 tag_as_user_data(expr
->left
);
341 static int handle_get_user(struct expression
*expr
)
346 name
= get_macro_name(expr
->pos
);
347 if (!name
|| strcmp(name
, "get_user") != 0)
350 name
= expr_to_var(expr
->right
);
351 if (!name
|| strcmp(name
, "__val_gu") != 0)
353 set_state_expr(my_id
, expr
->left
, alloc_estate_whole(get_type(expr
->left
)));
360 static void match_assign(struct expression
*expr
)
362 struct range_list
*rl
;
364 if (is_fake_call(expr
->right
))
366 if (handle_get_user(expr
))
368 if (points_to_user_data(expr
->right
))
369 set_points_to_user_data(expr
->left
);
370 if (handle_struct_assignment(expr
))
373 if (expr
->right
->type
== EXPR_CALL
||
374 !get_user_rl(expr
->right
, &rl
))
375 goto clear_old_state
;
377 rl
= cast_rl(get_type(expr
->left
), rl
);
378 set_state_expr(my_id
, expr
->left
, alloc_estate_rl(rl
));
383 if (get_state_expr(my_id
, expr
->left
))
384 set_state_expr(my_id
, expr
->left
, alloc_estate_empty());
387 static void match_user_assign_function(const char *fn
, struct expression
*expr
, void *unused
)
389 tag_as_user_data(expr
->left
);
390 set_points_to_user_data(expr
->left
);
393 static int get_user_macro_rl(struct expression
*expr
, struct range_list
**rl
)
399 macro
= get_macro_name(expr
->pos
);
404 if (strcmp(macro
, "ntohl") == 0) {
405 *rl
= alloc_whole_rl(&uint_ctype
);
408 if (strcmp(macro
, "ntohs") == 0) {
409 *rl
= alloc_whole_rl(&ushort_ctype
);
415 static int user_data_flag
;
416 static struct range_list
*var_user_rl(struct expression
*expr
)
418 struct smatch_state
*state
;
419 struct range_list
*rl
;
420 struct range_list
*absolute_rl
;
422 if (get_user_macro_rl(expr
, &rl
))
425 if (comes_from_skb_data(expr
)) {
426 rl
= alloc_whole_rl(get_type(expr
));
430 state
= get_state_expr(my_id
, expr
);
431 if (state
&& estate_rl(state
)) {
432 rl
= estate_rl(state
);
439 absolute_rl
= var_to_absolute_rl(expr
);
440 return clone_rl(rl_intersection(rl
, absolute_rl
));
443 int get_user_rl(struct expression
*expr
, struct range_list
**rl
)
447 custom_get_absolute_rl(expr
, &var_user_rl
, rl
);
448 if (!user_data_flag
) {
455 static void match_call_info(struct expression
*expr
)
457 struct range_list
*rl
;
458 struct expression
*arg
;
462 FOR_EACH_PTR(expr
->args
, arg
) {
465 if (!get_user_rl(arg
, &rl
))
468 sql_insert_caller_info(expr
, USER_DATA3
, i
, "$", show_rl(rl
));
469 } END_FOR_EACH_PTR(arg
);
472 static void struct_member_callback(struct expression
*call
, int param
, char *printed_name
, struct sm_state
*sm
)
474 struct smatch_state
*state
;
475 struct range_list
*rl
;
477 if (strcmp(sm
->state
->name
, "") == 0)
480 state
= get_state(SMATCH_EXTRA
, sm
->name
, sm
->sym
);
481 if (!state
|| !estate_rl(state
))
482 rl
= estate_rl(sm
->state
);
484 rl
= rl_intersection(estate_rl(sm
->state
), estate_rl(state
));
486 sql_insert_caller_info(call
, USER_DATA3
, param
, printed_name
, show_rl(rl
));
489 static void set_param_user_data(const char *name
, struct symbol
*sym
, char *key
, char *value
)
491 struct range_list
*rl
= NULL
;
492 struct smatch_state
*state
;
496 if (strcmp(key
, "*$") == 0)
497 snprintf(fullname
, sizeof(fullname
), "*%s", name
);
498 else if (strncmp(key
, "$", 1) == 0)
499 snprintf(fullname
, 256, "%s%s", name
, key
+ 1);
503 type
= get_member_type_from_key(symbol_expression(sym
), key
);
505 /* if the caller passes a void pointer with user data */
506 if (strcmp(key
, "*$") == 0 && type
&& type
!= &void_ctype
) {
507 struct expression
*expr
= symbol_expression(sym
);
509 tag_as_user_data(expr
);
510 set_points_to_user_data(expr
);
513 str_to_rl(type
, value
, &rl
);
514 state
= alloc_estate_rl(rl
);
515 set_state(my_id
, fullname
, sym
, state
);
518 static void set_called(const char *name
, struct symbol
*sym
, char *key
, char *value
)
520 set_state(my_call_id
, "this_function", NULL
, &called
);
523 static void match_syscall_definition(struct symbol
*sym
)
530 macro
= get_macro_name(sym
->pos
);
532 (strncmp("SYSCALL_DEFINE", macro
, strlen("SYSCALL_DEFINE")) == 0 ||
533 strncmp("COMPAT_SYSCALL_DEFINE", macro
, strlen("COMPAT_SYSCALL_DEFINE")) == 0))
536 name
= get_function();
537 if (!option_no_db
&& get_state(my_call_id
, "this_function", NULL
) != &called
) {
538 if (name
&& strncmp(name
, "sys_", 4) == 0)
542 if (name
&& strncmp(name
, "compat_sys_", 11) == 0)
548 FOR_EACH_PTR(sym
->ctype
.base_type
->arguments
, arg
) {
549 set_state(my_id
, arg
->ident
->name
, arg
, alloc_estate_whole(get_real_base_type(arg
)));
550 } END_FOR_EACH_PTR(arg
);
553 static void returns_param_user_data(struct expression
*expr
, int param
, char *key
, char *value
)
555 struct expression
*arg
;
559 struct range_list
*rl
= NULL
;
561 while (expr
->type
== EXPR_ASSIGNMENT
)
562 expr
= strip_expr(expr
->right
);
563 if (expr
->type
!= EXPR_CALL
)
566 arg
= get_argument_from_call_expr(expr
->args
, param
);
569 type
= get_member_type_from_key(arg
, key
);
570 name
= get_variable_from_key(arg
, key
, &sym
);
574 call_results_to_rl(expr
, type
, value
, &rl
);
576 set_state(my_id
, name
, sym
, alloc_estate_rl(rl
));
581 static int has_empty_state(struct sm_state
*sm
)
583 struct sm_state
*tmp
;
585 FOR_EACH_PTR(sm
->possible
, tmp
) {
586 if (!estate_rl(tmp
->state
))
588 } END_FOR_EACH_PTR(tmp
);
593 static void param_set_to_user_data(int return_id
, char *return_ranges
, struct expression
*expr
)
596 struct smatch_state
*start_state
;
598 const char *param_name
;
600 FOR_EACH_MY_SM(my_id
, __get_cur_stree(), sm
) {
601 if (has_empty_state(sm
))
605 param
= get_param_num_from_sym(sm
->sym
);
607 if (expr_to_sym(expr
) == sm
->sym
)
614 start_state
= get_state_stree(start_states
, my_id
, sm
->name
, sm
->sym
);
615 if (start_state
&& estates_equiv(sm
->state
, start_state
))
619 param_name
= get_param_name(sm
);
622 if (strcmp(param_name
, "$") == 0)
625 sql_insert_return_states(return_id
, return_ranges
, USER_DATA3
,
626 param
, param_name
, show_rl(estate_rl(sm
->state
)));
627 } END_FOR_EACH_SM(sm
);
630 void check_user_data2(int id
)
636 if (option_project
!= PROJ_KERNEL
)
639 add_hook(&save_start_states
, AFTER_DEF_HOOK
);
640 add_hook(&free_start_states
, END_FUNC_HOOK
);
641 add_hook(&match_save_states
, INLINE_FN_START
);
642 add_hook(&match_restore_states
, INLINE_FN_END
);
644 add_unmatched_state_hook(my_id
, &empty_state
);
645 add_pre_merge_hook(my_id
, &pre_merge_hook
);
646 add_merge_hook(my_id
, &merge_estates
);
648 add_function_hook("copy_from_user", &match_user_copy
, INT_PTR(0));
649 add_function_hook("__copy_from_user", &match_user_copy
, INT_PTR(0));
650 add_function_hook("memcpy_fromiovec", &match_user_copy
, INT_PTR(0));
651 for (i
= 0; i
< ARRAY_SIZE(kstr_funcs
); i
++)
652 add_function_hook(kstr_funcs
[i
], &match_user_copy
, INT_PTR(2));
654 add_function_hook("sscanf", &match_sscanf
, NULL
);
656 add_function_assign_hook("memdup_user", &match_user_assign_function
, NULL
);
657 add_function_assign_hook("kmap_atomic", &match_user_assign_function
, NULL
);
658 add_function_assign_hook("skb_network_header", &match_user_assign_function
, NULL
);
660 add_hook(&match_syscall_definition
, AFTER_DEF_HOOK
);
662 add_hook(&match_assign
, ASSIGNMENT_HOOK
);
664 add_hook(&match_call_info
, FUNCTION_CALL_HOOK
);
665 add_member_info_callback(my_id
, struct_member_callback
);
666 select_caller_info_hook(set_param_user_data
, USER_DATA3
);
667 select_return_states_hook(USER_DATA3
, &returns_param_user_data
);
668 add_split_return_callback(¶m_set_to_user_data
);
671 void check_user_data3(int id
)
675 if (option_project
!= PROJ_KERNEL
)
677 select_caller_info_hook(set_called
, INTERNAL
);