security/integrity/ima/ima_iint.c

   1 /*
   2  * Copyright (C) 2008 IBM Corporation
   3  *
   4  * Authors:
   5  * Mimi Zohar <zohar@us.ibm.com>
   6  *
   7  * This program is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License as
   9  * published by the Free Software Foundation, version 2 of the
  10  * License.
  11  *
  12  * File: ima_iint.c
  13  *      - implements the IMA hooks: ima_inode_alloc, ima_inode_free
  14  *      - cache integrity information associated with an inode
  15  *        using a rbtree tree.
  16  */
  17 #include <linux/slab.h>
  18 #include <linux/module.h>
  19 #include <linux/spinlock.h>
  20 #include <linux/rbtree.h>
  21 #include "ima.h"
  22
  23 static struct rb_root ima_iint_tree = RB_ROOT;
  24 static DEFINE_SPINLOCK(ima_iint_lock);
  25 static struct kmem_cache *iint_cache __read_mostly;
  26
  27 int iint_initialized = 0;
  28
  29 /*
  30  * __ima_iint_find - return the iint associated with an inode
  31  */
  32 static struct ima_iint_cache *__ima_iint_find(struct inode *inode)
  33 {
  34         struct ima_iint_cache *iint;
  35         struct rb_node *n = ima_iint_tree.rb_node;
  36
  37         assert_spin_locked(&ima_iint_lock);
  38
  39         while (n) {
  40                 iint = rb_entry(n, struct ima_iint_cache, rb_node);
  41
  42                 if (inode < iint->inode)
  43                         n = n->rb_left;
  44                 else if (inode > iint->inode)
  45                         n = n->rb_right;
  46                 else
  47                         break;
  48         }
  49         if (!n)
  50                 return NULL;
  51
  52         return iint;
  53 }
  54
  55 /*
  56  * ima_iint_find_get - return the iint associated with an inode
  57  *
  58  * ima_iint_find_get gets a reference to the iint. Caller must
  59  * remember to put the iint reference.
  60  */
  61 struct ima_iint_cache *ima_iint_find_get(struct inode *inode)
  62 {
  63         struct ima_iint_cache *iint;
  64
  65         spin_lock(&ima_iint_lock);
  66         iint = __ima_iint_find(inode);
  67         if (iint)
  68                 kref_get(&iint->refcount);
  69         spin_unlock(&ima_iint_lock);
  70
  71         return iint;
  72 }
  73
  74 /**
  75  * ima_inode_alloc - allocate an iint associated with an inode
  76  * @inode: pointer to the inode
  77  */
  78 int ima_inode_alloc(struct inode *inode)
  79 {
  80         struct rb_node **p;
  81         struct rb_node *new_node, *parent = NULL;
  82         struct ima_iint_cache *new_iint, *test_iint;
  83         int rc;
  84
  85         new_iint = kmem_cache_alloc(iint_cache, GFP_NOFS);
  86         if (!new_iint)
  87                 return -ENOMEM;
  88
  89         new_iint->inode = inode;
  90         new_node = &new_iint->rb_node;
  91
  92         spin_lock(&ima_iint_lock);
  93
  94         p = &ima_iint_tree.rb_node;
  95         while (*p) {
  96                 parent = *p;
  97                 test_iint = rb_entry(parent, struct ima_iint_cache, rb_node);
  98
  99                 rc = -EEXIST;
 100                 if (inode < test_iint->inode)
 101                         p = &(*p)->rb_left;
 102                 else if (inode > test_iint->inode)
 103                         p = &(*p)->rb_right;
 104                 else
 105                         goto out_err;
 106         }
 107
 108         rb_link_node(new_node, parent, p);
 109         rb_insert_color(new_node, &ima_iint_tree);
 110
 111         spin_unlock(&ima_iint_lock);
 112
 113         return 0;
 114 out_err:
 115         spin_unlock(&ima_iint_lock);
 116         kref_put(&new_iint->refcount, iint_free);
 117         return rc;
 118 }
 119
 120 /* iint_free - called when the iint refcount goes to zero */
 121 void iint_free(struct kref *kref)
 122 {
 123         struct ima_iint_cache *iint = container_of(kref, struct ima_iint_cache,
 124                                                    refcount);
 125         iint->version = 0;
 126         iint->flags = 0UL;
 127         if (iint->readcount != 0) {
 128                 printk(KERN_INFO "%s: readcount: %ld\n", __func__,
 129                        iint->readcount);
 130                 iint->readcount = 0;
 131         }
 132         if (iint->writecount != 0) {
 133                 printk(KERN_INFO "%s: writecount: %ld\n", __func__,
 134                        iint->writecount);
 135                 iint->writecount = 0;
 136         }
 137         kref_init(&iint->refcount);
 138         kmem_cache_free(iint_cache, iint);
 139 }
 140
 141 /**
 142  * ima_inode_free - called on security_inode_free
 143  * @inode: pointer to the inode
 144  *
 145  * Free the integrity information(iint) associated with an inode.
 146  */
 147 void ima_inode_free(struct inode *inode)
 148 {
 149         struct ima_iint_cache *iint;
 150
 151         spin_lock(&ima_iint_lock);
 152         iint = __ima_iint_find(inode);
 153         if (iint)
 154                 rb_erase(&iint->rb_node, &ima_iint_tree);
 155         spin_unlock(&ima_iint_lock);
 156         if (iint)
 157                 kref_put(&iint->refcount, iint_free);
 158 }
 159
 160 static void init_once(void *foo)
 161 {
 162         struct ima_iint_cache *iint = foo;
 163
 164         memset(iint, 0, sizeof *iint);
 165         iint->version = 0;
 166         iint->flags = 0UL;
 167         mutex_init(&iint->mutex);
 168         iint->readcount = 0;
 169         iint->writecount = 0;
 170         kref_init(&iint->refcount);
 171 }
 172
 173 static int __init ima_iintcache_init(void)
 174 {
 175         iint_cache =
 176             kmem_cache_create("iint_cache", sizeof(struct ima_iint_cache), 0,
 177                               SLAB_PANIC, init_once);
 178         iint_initialized = 1;
 179         return 0;
 180 }
 181 security_initcall(ima_iintcache_init);