net/netlabel/netlabel_domainhash.c

   1 /*
   2  * NetLabel Domain Hash Table
   3  *
   4  * This file manages the domain hash table that NetLabel uses to determine
   5  * which network labeling protocol to use for a given domain.  The NetLabel
   6  * system manages static and dynamic label mappings for network protocols such
   7  * as CIPSO and RIPSO.
   8  *
   9  * Author: Paul Moore <paul.moore@hp.com>
  10  *
  11  */
  12
  13 /*
  14  * (c) Copyright Hewlett-Packard Development Company, L.P., 2006
  15  *
  16  * This program is free software;  you can redistribute it and/or modify
  17  * it under the terms of the GNU General Public License as published by
  18  * the Free Software Foundation; either version 2 of the License, or
  19  * (at your option) any later version.
  20  *
  21  * This program is distributed in the hope that it will be useful,
  22  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
  23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
  24  * the GNU General Public License for more details.
  25  *
  26  * You should have received a copy of the GNU General Public License
  27  * along with this program;  if not, write to the Free Software
  28  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  29  *
  30  */
  31
  32 #include <linux/types.h>
  33 #include <linux/rculist.h>
  34 #include <linux/skbuff.h>
  35 #include <linux/spinlock.h>
  36 #include <linux/string.h>
  37 #include <linux/audit.h>
  38 #include <net/netlabel.h>
  39 #include <net/cipso_ipv4.h>
  40 #include <asm/bug.h>
  41
  42 #include "netlabel_mgmt.h"
  43 #include "netlabel_domainhash.h"
  44 #include "netlabel_user.h"
  45
  46 struct netlbl_domhsh_tbl {
  47         struct list_head *tbl;
  48         u32 size;
  49 };
  50
  51 /* Domain hash table */
  52 /* XXX - updates should be so rare that having one spinlock for the entire
  53  * hash table should be okay */
  54 static DEFINE_SPINLOCK(netlbl_domhsh_lock);
  55 static struct netlbl_domhsh_tbl *netlbl_domhsh = NULL;
  56 static struct netlbl_dom_map *netlbl_domhsh_def = NULL;
  57
  58 /*
  59  * Domain Hash Table Helper Functions
  60  */
  61
  62 /**
  63  * netlbl_domhsh_free_entry - Frees a domain hash table entry
  64  * @entry: the entry's RCU field
  65  *
  66  * Description:
  67  * This function is designed to be used as a callback to the call_rcu()
  68  * function so that the memory allocated to a hash table entry can be released
  69  * safely.
  70  *
  71  */
  72 static void netlbl_domhsh_free_entry(struct rcu_head *entry)
  73 {
  74         struct netlbl_dom_map *ptr;
  75
  76         ptr = container_of(entry, struct netlbl_dom_map, rcu);
  77         kfree(ptr->domain);
  78         kfree(ptr);
  79 }
  80
  81 /**
  82  * netlbl_domhsh_hash - Hashing function for the domain hash table
  83  * @domain: the domain name to hash
  84  *
  85  * Description:
  86  * This is the hashing function for the domain hash table, it returns the
  87  * correct bucket number for the domain.  The caller is responsibile for
  88  * calling the rcu_read_[un]lock() functions.
  89  *
  90  */
  91 static u32 netlbl_domhsh_hash(const char *key)
  92 {
  93         u32 iter;
  94         u32 val;
  95         u32 len;
  96
  97         /* This is taken (with slight modification) from
  98          * security/selinux/ss/symtab.c:symhash() */
  99
 100         for (iter = 0, val = 0, len = strlen(key); iter < len; iter++)
 101                 val = (val << 4 | (val >> (8 * sizeof(u32) - 4))) ^ key[iter];
 102         return val & (rcu_dereference(netlbl_domhsh)->size - 1);
 103 }
 104
 105 /**
 106  * netlbl_domhsh_search - Search for a domain entry
 107  * @domain: the domain
 108  *
 109  * Description:
 110  * Searches the domain hash table and returns a pointer to the hash table
 111  * entry if found, otherwise NULL is returned.  The caller is responsibile for
 112  * the rcu hash table locks (i.e. the caller much call rcu_read_[un]lock()).
 113  *
 114  */
 115 static struct netlbl_dom_map *netlbl_domhsh_search(const char *domain)
 116 {
 117         u32 bkt;
 118         struct netlbl_dom_map *iter;
 119
 120         if (domain != NULL) {
 121                 bkt = netlbl_domhsh_hash(domain);
 122                 list_for_each_entry_rcu(iter,
 123                                      &rcu_dereference(netlbl_domhsh)->tbl[bkt],
 124                                      list)
 125                         if (iter->valid && strcmp(iter->domain, domain) == 0)
 126                                 return iter;
 127         }
 128
 129         return NULL;
 130 }
 131
 132 /**
 133  * netlbl_domhsh_search_def - Search for a domain entry
 134  * @domain: the domain
 135  * @def: return default if no match is found
 136  *
 137  * Description:
 138  * Searches the domain hash table and returns a pointer to the hash table
 139  * entry if an exact match is found, if an exact match is not present in the
 140  * hash table then the default entry is returned if valid otherwise NULL is
 141  * returned.  The caller is responsibile for the rcu hash table locks
 142  * (i.e. the caller much call rcu_read_[un]lock()).
 143  *
 144  */
 145 static struct netlbl_dom_map *netlbl_domhsh_search_def(const char *domain)
 146 {
 147         struct netlbl_dom_map *entry;
 148
 149         entry = netlbl_domhsh_search(domain);
 150         if (entry == NULL) {
 151                 entry = rcu_dereference(netlbl_domhsh_def);
 152                 if (entry != NULL && !entry->valid)
 153                         entry = NULL;
 154         }
 155
 156         return entry;
 157 }
 158
 159 /*
 160  * Domain Hash Table Functions
 161  */
 162
 163 /**
 164  * netlbl_domhsh_init - Init for the domain hash
 165  * @size: the number of bits to use for the hash buckets
 166  *
 167  * Description:
 168  * Initializes the domain hash table, should be called only by
 169  * netlbl_user_init() during initialization.  Returns zero on success, non-zero
 170  * values on error.
 171  *
 172  */
 173 int __init netlbl_domhsh_init(u32 size)
 174 {
 175         u32 iter;
 176         struct netlbl_domhsh_tbl *hsh_tbl;
 177
 178         if (size == 0)
 179                 return -EINVAL;
 180
 181         hsh_tbl = kmalloc(sizeof(*hsh_tbl), GFP_KERNEL);
 182         if (hsh_tbl == NULL)
 183                 return -ENOMEM;
 184         hsh_tbl->size = 1 << size;
 185         hsh_tbl->tbl = kcalloc(hsh_tbl->size,
 186                                sizeof(struct list_head),
 187                                GFP_KERNEL);
 188         if (hsh_tbl->tbl == NULL) {
 189                 kfree(hsh_tbl);
 190                 return -ENOMEM;
 191         }
 192         for (iter = 0; iter < hsh_tbl->size; iter++)
 193                 INIT_LIST_HEAD(&hsh_tbl->tbl[iter]);
 194
 195         spin_lock(&netlbl_domhsh_lock);
 196         rcu_assign_pointer(netlbl_domhsh, hsh_tbl);
 197         spin_unlock(&netlbl_domhsh_lock);
 198
 199         return 0;
 200 }
 201
 202 /**
 203  * netlbl_domhsh_add - Adds a entry to the domain hash table
 204  * @entry: the entry to add
 205  * @audit_info: NetLabel audit information
 206  *
 207  * Description:
 208  * Adds a new entry to the domain hash table and handles any updates to the
 209  * lower level protocol handler (i.e. CIPSO).  Returns zero on success,
 210  * negative on failure.
 211  *
 212  */
 213 int netlbl_domhsh_add(struct netlbl_dom_map *entry,
 214                       struct netlbl_audit *audit_info)
 215 {
 216         int ret_val;
 217         u32 bkt;
 218         struct audit_buffer *audit_buf;
 219
 220         switch (entry->type) {
 221         case NETLBL_NLTYPE_UNLABELED:
 222                 ret_val = 0;
 223                 break;
 224         case NETLBL_NLTYPE_CIPSOV4:
 225                 ret_val = cipso_v4_doi_domhsh_add(entry->type_def.cipsov4,
 226                                                   entry->domain);
 227                 break;
 228         default:
 229                 return -EINVAL;
 230         }
 231         if (ret_val != 0)
 232                 return ret_val;
 233
 234         entry->valid = 1;
 235         INIT_RCU_HEAD(&entry->rcu);
 236
 237         rcu_read_lock();
 238         spin_lock(&netlbl_domhsh_lock);
 239         if (entry->domain != NULL) {
 240                 bkt = netlbl_domhsh_hash(entry->domain);
 241                 if (netlbl_domhsh_search(entry->domain) == NULL)
 242                         list_add_tail_rcu(&entry->list,
 243                                     &rcu_dereference(netlbl_domhsh)->tbl[bkt]);
 244                 else
 245                         ret_val = -EEXIST;
 246         } else {
 247                 INIT_LIST_HEAD(&entry->list);
 248                 if (rcu_dereference(netlbl_domhsh_def) == NULL)
 249                         rcu_assign_pointer(netlbl_domhsh_def, entry);
 250                 else
 251                         ret_val = -EEXIST;
 252         }
 253         spin_unlock(&netlbl_domhsh_lock);
 254         audit_buf = netlbl_audit_start_common(AUDIT_MAC_MAP_ADD, audit_info);
 255         if (audit_buf != NULL) {
 256                 audit_log_format(audit_buf,
 257                                  " nlbl_domain=%s",
 258                                  entry->domain ? entry->domain : "(default)");
 259                 switch (entry->type) {
 260                 case NETLBL_NLTYPE_UNLABELED:
 261                         audit_log_format(audit_buf, " nlbl_protocol=unlbl");
 262                         break;
 263                 case NETLBL_NLTYPE_CIPSOV4:
 264                         audit_log_format(audit_buf,
 265                                          " nlbl_protocol=cipsov4 cipso_doi=%u",
 266                                          entry->type_def.cipsov4->doi);
 267                         break;
 268                 }
 269                 audit_log_format(audit_buf, " res=%u", ret_val == 0 ? 1 : 0);
 270                 audit_log_end(audit_buf);
 271         }
 272         rcu_read_unlock();
 273
 274         if (ret_val != 0) {
 275                 switch (entry->type) {
 276                 case NETLBL_NLTYPE_CIPSOV4:
 277                         if (cipso_v4_doi_domhsh_remove(entry->type_def.cipsov4,
 278                                                        entry->domain) != 0)
 279                                 BUG();
 280                         break;
 281                 }
 282         }
 283
 284         return ret_val;
 285 }
 286
 287 /**
 288  * netlbl_domhsh_add_default - Adds the default entry to the domain hash table
 289  * @entry: the entry to add
 290  * @audit_info: NetLabel audit information
 291  *
 292  * Description:
 293  * Adds a new default entry to the domain hash table and handles any updates
 294  * to the lower level protocol handler (i.e. CIPSO).  Returns zero on success,
 295  * negative on failure.
 296  *
 297  */
 298 int netlbl_domhsh_add_default(struct netlbl_dom_map *entry,
 299                               struct netlbl_audit *audit_info)
 300 {
 301         return netlbl_domhsh_add(entry, audit_info);
 302 }
 303
 304 /**
 305  * netlbl_domhsh_remove - Removes an entry from the domain hash table
 306  * @domain: the domain to remove
 307  * @audit_info: NetLabel audit information
 308  *
 309  * Description:
 310  * Removes an entry from the domain hash table and handles any updates to the
 311  * lower level protocol handler (i.e. CIPSO).  Returns zero on success,
 312  * negative on failure.
 313  *
 314  */
 315 int netlbl_domhsh_remove(const char *domain, struct netlbl_audit *audit_info)
 316 {
 317         int ret_val = -ENOENT;
 318         struct netlbl_dom_map *entry;
 319         struct audit_buffer *audit_buf;
 320
 321         rcu_read_lock();
 322         if (domain)
 323                 entry = netlbl_domhsh_search(domain);
 324         else
 325                 entry = netlbl_domhsh_search_def(domain);
 326         if (entry == NULL)
 327                 goto remove_return;
 328         switch (entry->type) {
 329         case NETLBL_NLTYPE_CIPSOV4:
 330                 cipso_v4_doi_domhsh_remove(entry->type_def.cipsov4,
 331                                            entry->domain);
 332                 break;
 333         }
 334         spin_lock(&netlbl_domhsh_lock);
 335         if (entry->valid) {
 336                 entry->valid = 0;
 337                 if (entry != rcu_dereference(netlbl_domhsh_def))
 338                         list_del_rcu(&entry->list);
 339                 else
 340                         rcu_assign_pointer(netlbl_domhsh_def, NULL);
 341                 ret_val = 0;
 342         }
 343         spin_unlock(&netlbl_domhsh_lock);
 344
 345         audit_buf = netlbl_audit_start_common(AUDIT_MAC_MAP_DEL, audit_info);
 346         if (audit_buf != NULL) {
 347                 audit_log_format(audit_buf,
 348                                  " nlbl_domain=%s res=%u",
 349                                  entry->domain ? entry->domain : "(default)",
 350                                  ret_val == 0 ? 1 : 0);
 351                 audit_log_end(audit_buf);
 352         }
 353
 354 remove_return:
 355         rcu_read_unlock();
 356         if (ret_val == 0)
 357                 call_rcu(&entry->rcu, netlbl_domhsh_free_entry);
 358         return ret_val;
 359 }
 360
 361 /**
 362  * netlbl_domhsh_remove_default - Removes the default entry from the table
 363  * @audit_info: NetLabel audit information
 364  *
 365  * Description:
 366  * Removes/resets the default entry for the domain hash table and handles any
 367  * updates to the lower level protocol handler (i.e. CIPSO).  Returns zero on
 368  * success, non-zero on failure.
 369  *
 370  */
 371 int netlbl_domhsh_remove_default(struct netlbl_audit *audit_info)
 372 {
 373         return netlbl_domhsh_remove(NULL, audit_info);
 374 }
 375
 376 /**
 377  * netlbl_domhsh_getentry - Get an entry from the domain hash table
 378  * @domain: the domain name to search for
 379  *
 380  * Description:
 381  * Look through the domain hash table searching for an entry to match @domain,
 382  * return a pointer to a copy of the entry or NULL.  The caller is responsibile
 383  * for ensuring that rcu_read_[un]lock() is called.
 384  *
 385  */
 386 struct netlbl_dom_map *netlbl_domhsh_getentry(const char *domain)
 387 {
 388         return netlbl_domhsh_search_def(domain);
 389 }
 390
 391 /**
 392  * netlbl_domhsh_walk - Iterate through the domain mapping hash table
 393  * @skip_bkt: the number of buckets to skip at the start
 394  * @skip_chain: the number of entries to skip in the first iterated bucket
 395  * @callback: callback for each entry
 396  * @cb_arg: argument for the callback function
 397  *
 398  * Description:
 399  * Interate over the domain mapping hash table, skipping the first @skip_bkt
 400  * buckets and @skip_chain entries.  For each entry in the table call
 401  * @callback, if @callback returns a negative value stop 'walking' through the
 402  * table and return.  Updates the values in @skip_bkt and @skip_chain on
 403  * return.  Returns zero on succcess, negative values on failure.
 404  *
 405  */
 406 int netlbl_domhsh_walk(u32 *skip_bkt,
 407                      u32 *skip_chain,
 408                      int (*callback) (struct netlbl_dom_map *entry, void *arg),
 409                      void *cb_arg)
 410 {
 411         int ret_val = -ENOENT;
 412         u32 iter_bkt;
 413         struct netlbl_dom_map *iter_entry;
 414         u32 chain_cnt = 0;
 415
 416         rcu_read_lock();
 417         for (iter_bkt = *skip_bkt;
 418              iter_bkt < rcu_dereference(netlbl_domhsh)->size;
 419              iter_bkt++, chain_cnt = 0) {
 420                 list_for_each_entry_rcu(iter_entry,
 421                                 &rcu_dereference(netlbl_domhsh)->tbl[iter_bkt],
 422                                 list)
 423                         if (iter_entry->valid) {
 424                                 if (chain_cnt++ < *skip_chain)
 425                                         continue;
 426                                 ret_val = callback(iter_entry, cb_arg);
 427                                 if (ret_val < 0) {
 428                                         chain_cnt--;
 429                                         goto walk_return;
 430                                 }
 431                         }
 432         }
 433
 434 walk_return:
 435         rcu_read_unlock();
 436         *skip_bkt = iter_bkt;
 437         *skip_chain = chain_cnt;
 438         return ret_val;
 439 }