check_info_leak.c

   1 /*
   2  * smatch/check_info_leak.c
   3  *
   4  * Copyright (C) 2010 Dan Carpenter.
   5  *
   6  * Licensed under the Open Software License version 1.1
   7  *
   8  */
   9
  10 #include "smatch.h"
  11 #include "smatch_slist.h"
  12
  13 static int my_id;
  14
  15 STATE(alloced);
  16 STATE(string);
  17
  18 static char *my_get_variable(struct expression *expr, struct symbol **sym)
  19 {
  20         char *name;
  21
  22         name = get_variable_from_expr(expr, sym);
  23         free_string(name);
  24         if (!name || !*sym)
  25                 return NULL;
  26
  27         return (*sym)->ident->name;
  28 }
  29
  30 static void match_kmalloc(const char *fn, struct expression *expr, void *unused)
  31 {
  32         char *name;
  33         struct symbol *sym;
  34
  35         name = my_get_variable(expr->left, &sym);
  36         if (!name)
  37                 return;
  38         set_state(my_id, name, sym, &alloced);
  39 }
  40
  41 static void match_strcpy(const char *fn, struct expression *expr, void *unused)
  42 {
  43         struct expression *dest;
  44         char *name;
  45         struct symbol *sym;
  46
  47         dest = get_argument_from_call_expr(expr->args, 0);
  48         name = my_get_variable(dest, &sym);
  49         if (!name || !sym)
  50                 return;
  51         if (!get_state(my_id, name, sym))
  52                 return;
  53         set_state(my_id, name, sym, &string);
  54 }
  55
  56 static void match_copy_to_user(const char *fn, struct expression *expr, void *unused)
  57 {
  58         struct expression *src;
  59         char *name;
  60         struct symbol *sym;
  61         struct sm_state *sm;
  62
  63         src = get_argument_from_call_expr(expr->args, 1);
  64         name = my_get_variable(src, &sym);
  65         if (!name || !sym)
  66                 return;
  67         sm = get_sm_state(my_id, name, sym);
  68         if (!sm || !slist_has_state(sm->possible, &string))
  69                 return;
  70         name = get_variable_from_expr(src, NULL);
  71         sm_msg("warn: possible info leak '%s'", name);
  72         free_string(name);
  73 }
  74
  75 void check_info_leak(int id)
  76 {
  77         if (option_project != PROJ_KERNEL)
  78                 return;
  79         my_id = id;
  80         add_function_assign_hook("kmalloc", &match_kmalloc, NULL);
  81         add_function_hook("strcpy", &match_strcpy, NULL);
  82         add_function_hook("strlcpy", &match_strcpy, NULL);
  83         add_function_hook("strlcat", &match_strcpy, NULL);
  84         add_function_hook("strncpy", &match_strcpy, NULL);
  85         add_function_hook("copy_to_user", &match_copy_to_user, NULL);
  86 }