check_nospec.c

   1 /*
   2  * Copyright (C) 2018 Oracle.
   3  *
   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.
   8  *
   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.
  13  *
  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
  16  */
  17
  18 #include <stdlib.h>
  19 #include "parse.h"
  20 #include "smatch.h"
  21 #include "smatch_slist.h"
  22 #include "smatch_extra.h"
  23
  24 static int my_id;
  25
  26 STATE(nospec);
  27
  28 static struct smatch_state *unmatched_state(struct sm_state *sm)
  29 {
  30         struct range_list *rl;
  31
  32         if (__in_function_def && !get_user_rl_var_sym(sm->name, sm->sym, &rl))
  33                 return &nospec;
  34         return &undefined;
  35 }
  36
  37 bool is_nospec(struct expression *expr)
  38 {
  39         char *macro;
  40
  41         if (get_state_expr(my_id, expr) == &nospec)
  42                 return true;
  43         macro = get_macro_name(expr->pos);
  44         if (macro && strcmp(macro, "array_index_nospec") == 0)
  45                 return true;
  46         return false;
  47 }
  48
  49 static void nospec_assign(struct expression *expr)
  50 {
  51         if (is_nospec(expr->right))
  52                 set_state_expr(my_id, expr->left, &nospec);
  53 }
  54
  55 static void set_param_nospec(const char *name, struct symbol *sym, char *key, char *value)
  56 {
  57         char fullname[256];
  58
  59         if (strcmp(key, "*$") == 0)
  60                 snprintf(fullname, sizeof(fullname), "*%s", name);
  61         else if (strncmp(key, "$", 1) == 0)
  62                 snprintf(fullname, 256, "%s%s", name, key + 1);
  63         else
  64                 return;
  65
  66         set_state(my_id, fullname, sym, &nospec);
  67 }
  68
  69 static void match_call_info(struct expression *expr)
  70 {
  71         struct expression *arg;
  72         int i = 0;
  73
  74         FOR_EACH_PTR(expr->args, arg) {
  75                 if (get_state_expr(my_id, arg) == &nospec)
  76                         sql_insert_caller_info(expr, NOSPEC, i, "$", "");
  77                 i++;
  78         } END_FOR_EACH_PTR(arg);
  79 }
  80
  81 static void struct_member_callback(struct expression *call, int param, char *printed_name, struct sm_state *sm)
  82 {
  83         sql_insert_caller_info(call, NOSPEC, param, printed_name, "");
  84 }
  85
  86 static void returned_struct_members(int return_id, char *return_ranges, struct expression *expr)
  87 {
  88         struct symbol *returned_sym;
  89         struct sm_state *sm;
  90         const char *param_name;
  91         int param;
  92
  93         returned_sym = expr_to_sym(expr);
  94
  95         FOR_EACH_MY_SM(my_id, __get_cur_stree(), sm) {
  96                 param = get_param_num_from_sym(sm->sym);
  97                 if (param < 0) {
  98                         if (!returned_sym || returned_sym != sm->sym)
  99                                 continue;
 100                         param = -1;
 101                 }
 102
 103                 param_name = get_param_name(sm);
 104                 if (!param_name)
 105                         continue;
 106                 if (param != -1 && strcmp(param_name, "$") == 0)
 107                         continue;
 108
 109                 sql_insert_return_states(return_id, return_ranges, NOSPEC, param, param_name, "");
 110         } END_FOR_EACH_SM(sm);
 111 }
 112
 113 void check_nospec(int id)
 114 {
 115         my_id = id;
 116
 117         add_hook(&nospec_assign, ASSIGNMENT_HOOK);
 118
 119         select_caller_info_hook(set_param_nospec, NOSPEC);
 120         add_unmatched_state_hook(my_id, &unmatched_state);
 121
 122         add_hook(&match_call_info, FUNCTION_CALL_HOOK);
 123         add_member_info_callback(my_id, struct_member_callback);
 124         add_split_return_callback(&returned_struct_members);
 125 }