1*1f5207b7SJohn Levon /*
2*1f5207b7SJohn Levon  * Copyright (C) 2014 Oracle.
3*1f5207b7SJohn Levon  *
4*1f5207b7SJohn Levon  * This program is free software; you can redistribute it and/or
5*1f5207b7SJohn Levon  * modify it under the terms of the GNU General Public License
6*1f5207b7SJohn Levon  * as published by the Free Software Foundation; either version 2
7*1f5207b7SJohn Levon  * of the License, or (at your option) any later version.
8*1f5207b7SJohn Levon  *
9*1f5207b7SJohn Levon  * This program is distributed in the hope that it will be useful,
10*1f5207b7SJohn Levon  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11*1f5207b7SJohn Levon  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12*1f5207b7SJohn Levon  * GNU General Public License for more details.
13*1f5207b7SJohn Levon  *
14*1f5207b7SJohn Levon  * You should have received a copy of the GNU General Public License
15*1f5207b7SJohn Levon  * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
16*1f5207b7SJohn Levon  */
17*1f5207b7SJohn Levon 
18*1f5207b7SJohn Levon #include "smatch.h"
19*1f5207b7SJohn Levon #include "smatch_slist.h"
20*1f5207b7SJohn Levon 
21*1f5207b7SJohn Levon STATE(capable);
22*1f5207b7SJohn Levon 
23*1f5207b7SJohn Levon static int capable_id;
24*1f5207b7SJohn Levon static int ns_capable_id;
25*1f5207b7SJohn Levon 
match_capable(const char * fn,struct expression * expr,void * _param)26*1f5207b7SJohn Levon static void match_capable(const char *fn, struct expression *expr, void *_param)
27*1f5207b7SJohn Levon {
28*1f5207b7SJohn Levon 	struct expression *arg;
29*1f5207b7SJohn Levon 	sval_t sval;
30*1f5207b7SJohn Levon 	char buf[32];
31*1f5207b7SJohn Levon 
32*1f5207b7SJohn Levon 	arg = get_argument_from_call_expr(expr->args, 0);
33*1f5207b7SJohn Levon 	if (!get_implied_value(arg, &sval))
34*1f5207b7SJohn Levon 		return;
35*1f5207b7SJohn Levon 	snprintf(buf, sizeof(buf), "%s", sval_to_str(sval));
36*1f5207b7SJohn Levon 	set_state(capable_id, buf, NULL, &capable);
37*1f5207b7SJohn Levon }
38*1f5207b7SJohn Levon 
match_ns_capable(const char * fn,struct expression * expr,void * _param)39*1f5207b7SJohn Levon static void match_ns_capable(const char *fn, struct expression *expr, void *_param)
40*1f5207b7SJohn Levon {
41*1f5207b7SJohn Levon 	struct expression *arg;
42*1f5207b7SJohn Levon 	sval_t sval;
43*1f5207b7SJohn Levon 	char buf[32];
44*1f5207b7SJohn Levon 
45*1f5207b7SJohn Levon 	if (get_function() && strcmp(get_function(), "capable") == 0)
46*1f5207b7SJohn Levon 		return;
47*1f5207b7SJohn Levon 
48*1f5207b7SJohn Levon 	arg = get_argument_from_call_expr(expr->args, 1);
49*1f5207b7SJohn Levon 	if (!get_implied_value(arg, &sval))
50*1f5207b7SJohn Levon 		return;
51*1f5207b7SJohn Levon 	snprintf(buf, sizeof(buf), "%s", sval_to_str(sval));
52*1f5207b7SJohn Levon 	set_state(ns_capable_id, buf, NULL, &capable);
53*1f5207b7SJohn Levon }
54*1f5207b7SJohn Levon 
save_call_info(struct expression * call)55*1f5207b7SJohn Levon static void save_call_info(struct expression *call)
56*1f5207b7SJohn Levon {
57*1f5207b7SJohn Levon 	struct sm_state *sm;
58*1f5207b7SJohn Levon 
59*1f5207b7SJohn Levon 	FOR_EACH_MY_SM(capable_id, __get_cur_stree(), sm) {
60*1f5207b7SJohn Levon 		if (sm->state == &capable)
61*1f5207b7SJohn Levon 			sql_insert_caller_info(call, CAPABLE, 0, sm->name, "");
62*1f5207b7SJohn Levon 	} END_FOR_EACH_SM(sm);
63*1f5207b7SJohn Levon 
64*1f5207b7SJohn Levon 	FOR_EACH_MY_SM(ns_capable_id, __get_cur_stree(), sm) {
65*1f5207b7SJohn Levon 		if (sm->state == &capable)
66*1f5207b7SJohn Levon 			sql_insert_caller_info(call, NS_CAPABLE, 0, sm->name, "");
67*1f5207b7SJohn Levon 	} END_FOR_EACH_SM(sm);
68*1f5207b7SJohn Levon }
69*1f5207b7SJohn Levon 
save_return_info(int return_id,char * return_ranges,struct expression * expr)70*1f5207b7SJohn Levon static void save_return_info(int return_id, char *return_ranges, struct expression *expr)
71*1f5207b7SJohn Levon {
72*1f5207b7SJohn Levon 	struct sm_state *sm;
73*1f5207b7SJohn Levon 
74*1f5207b7SJohn Levon 	FOR_EACH_MY_SM(capable_id, __get_cur_stree(), sm) {
75*1f5207b7SJohn Levon 		if (sm->state == &capable)
76*1f5207b7SJohn Levon 			sql_insert_return_states(return_id, return_ranges,
77*1f5207b7SJohn Levon 						 CAPABLE, 0, sm->name, "");
78*1f5207b7SJohn Levon 	} END_FOR_EACH_SM(sm);
79*1f5207b7SJohn Levon 
80*1f5207b7SJohn Levon 	FOR_EACH_MY_SM(ns_capable_id, __get_cur_stree(), sm) {
81*1f5207b7SJohn Levon 		if (sm->state == &capable)
82*1f5207b7SJohn Levon 			sql_insert_return_states(return_id, return_ranges,
83*1f5207b7SJohn Levon 						 CAPABLE, 0, sm->name, "");
84*1f5207b7SJohn Levon 	} END_FOR_EACH_SM(sm);
85*1f5207b7SJohn Levon }
86*1f5207b7SJohn Levon 
set_db_capable(const char * name,struct symbol * sym,char * key,char * value)87*1f5207b7SJohn Levon static void set_db_capable(const char *name, struct symbol *sym, char *key, char *value)
88*1f5207b7SJohn Levon {
89*1f5207b7SJohn Levon 	char buf[32];
90*1f5207b7SJohn Levon 
91*1f5207b7SJohn Levon 	snprintf(buf, sizeof(buf), "%s", key);
92*1f5207b7SJohn Levon 	set_state(capable_id, buf, NULL, &capable);
93*1f5207b7SJohn Levon }
94*1f5207b7SJohn Levon 
set_db_ns_capable(const char * name,struct symbol * sym,char * key,char * value)95*1f5207b7SJohn Levon static void set_db_ns_capable(const char *name, struct symbol *sym, char *key, char *value)
96*1f5207b7SJohn Levon {
97*1f5207b7SJohn Levon 	char buf[32];
98*1f5207b7SJohn Levon 
99*1f5207b7SJohn Levon 	snprintf(buf, sizeof(buf), "%s", key);
100*1f5207b7SJohn Levon 	set_state(ns_capable_id, buf, NULL, &capable);
101*1f5207b7SJohn Levon }
102*1f5207b7SJohn Levon 
check_capable(int id)103*1f5207b7SJohn Levon void check_capable(int id)
104*1f5207b7SJohn Levon {
105*1f5207b7SJohn Levon 	if (option_project != PROJ_KERNEL)
106*1f5207b7SJohn Levon 		return;
107*1f5207b7SJohn Levon 
108*1f5207b7SJohn Levon 	capable_id = id;
109*1f5207b7SJohn Levon 	add_function_hook("capable", &match_capable, INT_PTR(0));
110*1f5207b7SJohn Levon 
111*1f5207b7SJohn Levon 	add_hook(&save_call_info, FUNCTION_CALL_HOOK);
112*1f5207b7SJohn Levon 	add_split_return_callback(save_return_info);
113*1f5207b7SJohn Levon 	select_caller_info_hook(set_db_capable, CAPABLE);
114*1f5207b7SJohn Levon }
115*1f5207b7SJohn Levon 
check_ns_capable(int id)116*1f5207b7SJohn Levon void check_ns_capable(int id)
117*1f5207b7SJohn Levon {
118*1f5207b7SJohn Levon 	if (option_project != PROJ_KERNEL)
119*1f5207b7SJohn Levon 		return;
120*1f5207b7SJohn Levon 
121*1f5207b7SJohn Levon 	ns_capable_id = id;
122*1f5207b7SJohn Levon 	add_function_hook("ns_capable", &match_ns_capable, INT_PTR(0));
123*1f5207b7SJohn Levon 	select_caller_info_hook(set_db_ns_capable, NS_CAPABLE);
124*1f5207b7SJohn Levon }
125