1*1f5207b7SJohn Levon /* 2*1f5207b7SJohn Levon * Copyright (C) 2018 Oracle. All rights reserved. 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_extra.h" 20*1f5207b7SJohn Levon #include "smatch_slist.h" 21*1f5207b7SJohn Levon 22*1f5207b7SJohn Levon static int my_id; 23*1f5207b7SJohn Levon 24*1f5207b7SJohn Levon struct db_info { 25*1f5207b7SJohn Levon int count; 26*1f5207b7SJohn Levon struct symbol *type; 27*1f5207b7SJohn Levon struct range_list *rl; 28*1f5207b7SJohn Levon }; 29*1f5207b7SJohn Levon 30*1f5207b7SJohn Levon static int get_vals(void *_db_info, int argc, char **argv, char **azColName) 31*1f5207b7SJohn Levon { 32*1f5207b7SJohn Levon struct db_info *db_info = _db_info; 33*1f5207b7SJohn Levon struct range_list *rl; 34*1f5207b7SJohn Levon 35*1f5207b7SJohn Levon str_to_rl(db_info->type, argv[0], &rl); 36*1f5207b7SJohn Levon db_info->rl = rl_union(db_info->rl, rl); 37*1f5207b7SJohn Levon 38*1f5207b7SJohn Levon return 0; 39*1f5207b7SJohn Levon } 40*1f5207b7SJohn Levon 41*1f5207b7SJohn Levon static int is_file_local(struct expression *array) 42*1f5207b7SJohn Levon { 43*1f5207b7SJohn Levon struct symbol *sym = NULL; 44*1f5207b7SJohn Levon char *name; 45*1f5207b7SJohn Levon 46*1f5207b7SJohn Levon name = expr_to_str_sym(array, &sym); 47*1f5207b7SJohn Levon free_string(name); 48*1f5207b7SJohn Levon if (!sym) 49*1f5207b7SJohn Levon return 0; 50*1f5207b7SJohn Levon 51*1f5207b7SJohn Levon if ((sym->ctype.modifiers & MOD_TOPLEVEL) && 52*1f5207b7SJohn Levon (sym->ctype.modifiers & MOD_STATIC)) 53*1f5207b7SJohn Levon return 1; 54*1f5207b7SJohn Levon return 0; 55*1f5207b7SJohn Levon } 56*1f5207b7SJohn Levon 57*1f5207b7SJohn Levon static char *get_toplevel_name(struct expression *array) 58*1f5207b7SJohn Levon { 59*1f5207b7SJohn Levon char *name; 60*1f5207b7SJohn Levon char buf[128]; 61*1f5207b7SJohn Levon 62*1f5207b7SJohn Levon if (is_array(array)) 63*1f5207b7SJohn Levon array = get_array_base(array); 64*1f5207b7SJohn Levon 65*1f5207b7SJohn Levon if (!array || array->type != EXPR_SYMBOL) 66*1f5207b7SJohn Levon return NULL; 67*1f5207b7SJohn Levon if (!is_file_local(array)) 68*1f5207b7SJohn Levon return NULL; 69*1f5207b7SJohn Levon 70*1f5207b7SJohn Levon name = expr_to_str(array); 71*1f5207b7SJohn Levon snprintf(buf, sizeof(buf), "%s[]", name); 72*1f5207b7SJohn Levon free_string(name); 73*1f5207b7SJohn Levon 74*1f5207b7SJohn Levon return alloc_sname(buf); 75*1f5207b7SJohn Levon } 76*1f5207b7SJohn Levon 77*1f5207b7SJohn Levon static char *get_member_array(struct expression *array) 78*1f5207b7SJohn Levon { 79*1f5207b7SJohn Levon char *name; 80*1f5207b7SJohn Levon char buf[128]; 81*1f5207b7SJohn Levon 82*1f5207b7SJohn Levon name = get_member_name(array); 83*1f5207b7SJohn Levon if (!name) 84*1f5207b7SJohn Levon return NULL; 85*1f5207b7SJohn Levon snprintf(buf, sizeof(buf), "%s[]", name); 86*1f5207b7SJohn Levon free_string(name); 87*1f5207b7SJohn Levon return alloc_sname(buf); 88*1f5207b7SJohn Levon } 89*1f5207b7SJohn Levon 90*1f5207b7SJohn Levon static char *get_array_name(struct expression *array) 91*1f5207b7SJohn Levon { 92*1f5207b7SJohn Levon struct symbol *type; 93*1f5207b7SJohn Levon char *name; 94*1f5207b7SJohn Levon 95*1f5207b7SJohn Levon type = get_type(array); 96*1f5207b7SJohn Levon if (!type || type->type != SYM_ARRAY) 97*1f5207b7SJohn Levon return NULL; 98*1f5207b7SJohn Levon 99*1f5207b7SJohn Levon name = get_toplevel_name(array); 100*1f5207b7SJohn Levon if (name) 101*1f5207b7SJohn Levon return name; 102*1f5207b7SJohn Levon name = get_member_array(array); 103*1f5207b7SJohn Levon if (name) 104*1f5207b7SJohn Levon return name; 105*1f5207b7SJohn Levon 106*1f5207b7SJohn Levon return NULL; 107*1f5207b7SJohn Levon } 108*1f5207b7SJohn Levon 109*1f5207b7SJohn Levon int get_array_rl(struct expression *expr, struct range_list **rl) 110*1f5207b7SJohn Levon { 111*1f5207b7SJohn Levon struct expression *array; 112*1f5207b7SJohn Levon struct symbol *type; 113*1f5207b7SJohn Levon struct db_info db_info = {}; 114*1f5207b7SJohn Levon char *name; 115*1f5207b7SJohn Levon 116*1f5207b7SJohn Levon type = get_type(expr); 117*1f5207b7SJohn Levon if (!type || type->type != SYM_BASETYPE) 118*1f5207b7SJohn Levon return 0; 119*1f5207b7SJohn Levon db_info.type = type; 120*1f5207b7SJohn Levon 121*1f5207b7SJohn Levon array = get_array_base(expr); 122*1f5207b7SJohn Levon name = get_array_name(array); 123*1f5207b7SJohn Levon if (!name) 124*1f5207b7SJohn Levon return 0; 125*1f5207b7SJohn Levon 126*1f5207b7SJohn Levon if (is_file_local(array)) { 127*1f5207b7SJohn Levon run_sql(&get_vals, &db_info, 128*1f5207b7SJohn Levon "select value from sink_info where file = '%s' and static = 1 and sink_name = '%s' and type = %d;", 129*1f5207b7SJohn Levon get_filename(), name, DATA_VALUE); 130*1f5207b7SJohn Levon } else { 131*1f5207b7SJohn Levon run_sql(&get_vals, &db_info, 132*1f5207b7SJohn Levon "select value from sink_info where sink_name = '%s' and type = %d limit 10;", 133*1f5207b7SJohn Levon name, DATA_VALUE); 134*1f5207b7SJohn Levon } 135*1f5207b7SJohn Levon if (!db_info.rl || db_info.count >= 10) 136*1f5207b7SJohn Levon return 0; 137*1f5207b7SJohn Levon 138*1f5207b7SJohn Levon *rl = db_info.rl; 139*1f5207b7SJohn Levon return 1; 140*1f5207b7SJohn Levon } 141*1f5207b7SJohn Levon 142*1f5207b7SJohn Levon static struct range_list *get_saved_rl(struct symbol *type, char *name) 143*1f5207b7SJohn Levon { 144*1f5207b7SJohn Levon struct db_info db_info = {.type = type}; 145*1f5207b7SJohn Levon 146*1f5207b7SJohn Levon cache_sql(&get_vals, &db_info, "select value from sink_info where sink_name = '%s' and type = %d;", 147*1f5207b7SJohn Levon name, DATA_VALUE); 148*1f5207b7SJohn Levon return db_info.rl; 149*1f5207b7SJohn Levon } 150*1f5207b7SJohn Levon 151*1f5207b7SJohn Levon static void update_cache(char *name, int is_static, struct range_list *rl) 152*1f5207b7SJohn Levon { 153*1f5207b7SJohn Levon cache_sql(NULL, NULL, "delete from sink_info where sink_name = '%s' and type = %d;", 154*1f5207b7SJohn Levon name, DATA_VALUE); 155*1f5207b7SJohn Levon cache_sql(NULL, NULL, "insert into sink_info values ('%s', %d, '%s', %d, '', '%s');", 156*1f5207b7SJohn Levon get_filename(), is_static, name, DATA_VALUE, show_rl(rl)); 157*1f5207b7SJohn Levon } 158*1f5207b7SJohn Levon 159*1f5207b7SJohn Levon static void match_assign(struct expression *expr) 160*1f5207b7SJohn Levon { 161*1f5207b7SJohn Levon struct expression *left, *array; 162*1f5207b7SJohn Levon struct range_list *orig_rl, *rl; 163*1f5207b7SJohn Levon struct symbol *type; 164*1f5207b7SJohn Levon char *name; 165*1f5207b7SJohn Levon 166*1f5207b7SJohn Levon type = get_type(expr->right); 167*1f5207b7SJohn Levon if (!type || type->type != SYM_BASETYPE) 168*1f5207b7SJohn Levon return; 169*1f5207b7SJohn Levon 170*1f5207b7SJohn Levon left = strip_expr(expr->left); 171*1f5207b7SJohn Levon if (!is_array(left)) 172*1f5207b7SJohn Levon return; 173*1f5207b7SJohn Levon array = get_array_base(left); 174*1f5207b7SJohn Levon name = get_array_name(array); 175*1f5207b7SJohn Levon if (!name) 176*1f5207b7SJohn Levon return; 177*1f5207b7SJohn Levon 178*1f5207b7SJohn Levon if (expr->op != '=') { 179*1f5207b7SJohn Levon rl = alloc_whole_rl(type); 180*1f5207b7SJohn Levon } else { 181*1f5207b7SJohn Levon get_absolute_rl(expr->right, &rl); 182*1f5207b7SJohn Levon rl = cast_rl(type, rl); 183*1f5207b7SJohn Levon orig_rl = get_saved_rl(type, name); 184*1f5207b7SJohn Levon rl = rl_union(orig_rl, rl); 185*1f5207b7SJohn Levon } 186*1f5207b7SJohn Levon 187*1f5207b7SJohn Levon update_cache(name, is_file_local(array), rl); 188*1f5207b7SJohn Levon } 189*1f5207b7SJohn Levon 190*1f5207b7SJohn Levon void register_array_values(int id) 191*1f5207b7SJohn Levon { 192*1f5207b7SJohn Levon my_id = id; 193*1f5207b7SJohn Levon 194*1f5207b7SJohn Levon add_hook(&match_assign, ASSIGNMENT_HOOK); 195*1f5207b7SJohn Levon add_hook(&match_assign, GLOBAL_ASSIGNMENT_HOOK); 196*1f5207b7SJohn Levon } 197