net/netlabel/netlabel_kapi.c

   1 /*
   2  * NetLabel Kernel API
   3  *
   4  * This file defines the kernel API for the NetLabel system.  The NetLabel
   5  * system manages static and dynamic label mappings for network protocols such
   6  * as CIPSO and RIPSO.
   7  *
   8  * Author: Paul Moore <paul@paul-moore.com>
   9  *
  10  */
  11
  12 /*
  13  * (c) Copyright Hewlett-Packard Development Company, L.P., 2006, 2008
  14  *
  15  * This program is free software;  you can redistribute it and/or modify
  16  * it under the terms of the GNU General Public License as published by
  17  * the Free Software Foundation; either version 2 of the License, or
  18  * (at your option) any later version.
  19  *
  20  * This program is distributed in the hope that it will be useful,
  21  * but WITHOUT ANY WARRANTY;  without even the implied warranty of
  22  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
  23  * the GNU General Public License for more details.
  24  *
  25  * You should have received a copy of the GNU General Public License
  26  * along with this program;  if not, see <http://www.gnu.org/licenses/>.
  27  *
  28  */
  29
  30 #include <linux/init.h>
  31 #include <linux/types.h>
  32 #include <linux/slab.h>
  33 #include <linux/audit.h>
  34 #include <linux/in.h>
  35 #include <linux/in6.h>
  36 #include <net/ip.h>
  37 #include <net/ipv6.h>
  38 #include <net/netlabel.h>
  39 #include <net/cipso_ipv4.h>
  40 #include <net/calipso.h>
  41 #include <asm/bug.h>
  42 #include <linux/atomic.h>
  43
  44 #include "netlabel_domainhash.h"
  45 #include "netlabel_unlabeled.h"
  46 #include "netlabel_cipso_v4.h"
  47 #include "netlabel_calipso.h"
  48 #include "netlabel_user.h"
  49 #include "netlabel_mgmt.h"
  50 #include "netlabel_addrlist.h"
  51
  52 /*
  53  * Configuration Functions
  54  */
  55
  56 /**
  57  * netlbl_cfg_map_del - Remove a NetLabel/LSM domain mapping
  58  * @domain: the domain mapping to remove
  59  * @family: address family
  60  * @addr: IP address
  61  * @mask: IP address mask
  62  * @audit_info: NetLabel audit information
  63  *
  64  * Description:
  65  * Removes a NetLabel/LSM domain mapping.  A @domain value of NULL causes the
  66  * default domain mapping to be removed.  Returns zero on success, negative
  67  * values on failure.
  68  *
  69  */
  70 int netlbl_cfg_map_del(const char *domain,
  71                        u16 family,
  72                        const void *addr,
  73                        const void *mask,
  74                        struct netlbl_audit *audit_info)
  75 {
  76         if (addr == NULL && mask == NULL) {
  77                 return netlbl_domhsh_remove(domain, family, audit_info);
  78         } else if (addr != NULL && mask != NULL) {
  79                 switch (family) {
  80                 case AF_INET:
  81                         return netlbl_domhsh_remove_af4(domain, addr, mask,
  82                                                         audit_info);
  83 #if IS_ENABLED(CONFIG_IPV6)
  84                 case AF_INET6:
  85                         return netlbl_domhsh_remove_af6(domain, addr, mask,
  86                                                         audit_info);
  87 #endif /* IPv6 */
  88                 default:
  89                         return -EPFNOSUPPORT;
  90                 }
  91         } else
  92                 return -EINVAL;
  93 }
  94
  95 /**
  96  * netlbl_cfg_unlbl_map_add - Add a new unlabeled mapping
  97  * @domain: the domain mapping to add
  98  * @family: address family
  99  * @addr: IP address
 100  * @mask: IP address mask
 101  * @audit_info: NetLabel audit information
 102  *
 103  * Description:
 104  * Adds a new unlabeled NetLabel/LSM domain mapping.  A @domain value of NULL
 105  * causes a new default domain mapping to be added.  Returns zero on success,
 106  * negative values on failure.
 107  *
 108  */
 109 int netlbl_cfg_unlbl_map_add(const char *domain,
 110                              u16 family,
 111                              const void *addr,
 112                              const void *mask,
 113                              struct netlbl_audit *audit_info)
 114 {
 115         int ret_val = -ENOMEM;
 116         struct netlbl_dom_map *entry;
 117         struct netlbl_domaddr_map *addrmap = NULL;
 118         struct netlbl_domaddr4_map *map4 = NULL;
 119         struct netlbl_domaddr6_map *map6 = NULL;
 120
 121         entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
 122         if (entry == NULL)
 123                 return -ENOMEM;
 124         if (domain != NULL) {
 125                 entry->domain = kstrdup(domain, GFP_ATOMIC);
 126                 if (entry->domain == NULL)
 127                         goto cfg_unlbl_map_add_failure;
 128         }
 129         entry->family = family;
 130
 131         if (addr == NULL && mask == NULL)
 132                 entry->def.type = NETLBL_NLTYPE_UNLABELED;
 133         else if (addr != NULL && mask != NULL) {
 134                 addrmap = kzalloc(sizeof(*addrmap), GFP_ATOMIC);
 135                 if (addrmap == NULL)
 136                         goto cfg_unlbl_map_add_failure;
 137                 INIT_LIST_HEAD(&addrmap->list4);
 138                 INIT_LIST_HEAD(&addrmap->list6);
 139
 140                 switch (family) {
 141                 case AF_INET: {
 142                         const struct in_addr *addr4 = addr;
 143                         const struct in_addr *mask4 = mask;
 144                         map4 = kzalloc(sizeof(*map4), GFP_ATOMIC);
 145                         if (map4 == NULL)
 146                                 goto cfg_unlbl_map_add_failure;
 147                         map4->def.type = NETLBL_NLTYPE_UNLABELED;
 148                         map4->list.addr = addr4->s_addr & mask4->s_addr;
 149                         map4->list.mask = mask4->s_addr;
 150                         map4->list.valid = 1;
 151                         ret_val = netlbl_af4list_add(&map4->list,
 152                                                      &addrmap->list4);
 153                         if (ret_val != 0)
 154                                 goto cfg_unlbl_map_add_failure;
 155                         break;
 156                         }
 157 #if IS_ENABLED(CONFIG_IPV6)
 158                 case AF_INET6: {
 159                         const struct in6_addr *addr6 = addr;
 160                         const struct in6_addr *mask6 = mask;
 161                         map6 = kzalloc(sizeof(*map6), GFP_ATOMIC);
 162                         if (map6 == NULL)
 163                                 goto cfg_unlbl_map_add_failure;
 164                         map6->def.type = NETLBL_NLTYPE_UNLABELED;
 165                         map6->list.addr = *addr6;
 166                         map6->list.addr.s6_addr32[0] &= mask6->s6_addr32[0];
 167                         map6->list.addr.s6_addr32[1] &= mask6->s6_addr32[1];
 168                         map6->list.addr.s6_addr32[2] &= mask6->s6_addr32[2];
 169                         map6->list.addr.s6_addr32[3] &= mask6->s6_addr32[3];
 170                         map6->list.mask = *mask6;
 171                         map6->list.valid = 1;
 172                         ret_val = netlbl_af6list_add(&map6->list,
 173                                                      &addrmap->list6);
 174                         if (ret_val != 0)
 175                                 goto cfg_unlbl_map_add_failure;
 176                         break;
 177                         }
 178 #endif /* IPv6 */
 179                 default:
 180                         goto cfg_unlbl_map_add_failure;
 181                 }
 182
 183                 entry->def.addrsel = addrmap;
 184                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
 185         } else {
 186                 ret_val = -EINVAL;
 187                 goto cfg_unlbl_map_add_failure;
 188         }
 189
 190         ret_val = netlbl_domhsh_add(entry, audit_info);
 191         if (ret_val != 0)
 192                 goto cfg_unlbl_map_add_failure;
 193
 194         return 0;
 195
 196 cfg_unlbl_map_add_failure:
 197         kfree(entry->domain);
 198         kfree(entry);
 199         kfree(addrmap);
 200         kfree(map4);
 201         kfree(map6);
 202         return ret_val;
 203 }
 204
 205
 206 /**
 207  * netlbl_cfg_unlbl_static_add - Adds a new static label
 208  * @net: network namespace
 209  * @dev_name: interface name
 210  * @addr: IP address in network byte order (struct in[6]_addr)
 211  * @mask: address mask in network byte order (struct in[6]_addr)
 212  * @family: address family
 213  * @secid: LSM secid value for the entry
 214  * @audit_info: NetLabel audit information
 215  *
 216  * Description:
 217  * Adds a new NetLabel static label to be used when protocol provided labels
 218  * are not present on incoming traffic.  If @dev_name is NULL then the default
 219  * interface will be used.  Returns zero on success, negative values on failure.
 220  *
 221  */
 222 int netlbl_cfg_unlbl_static_add(struct net *net,
 223                                 const char *dev_name,
 224                                 const void *addr,
 225                                 const void *mask,
 226                                 u16 family,
 227                                 u32 secid,
 228                                 struct netlbl_audit *audit_info)
 229 {
 230         u32 addr_len;
 231
 232         switch (family) {
 233         case AF_INET:
 234                 addr_len = sizeof(struct in_addr);
 235                 break;
 236 #if IS_ENABLED(CONFIG_IPV6)
 237         case AF_INET6:
 238                 addr_len = sizeof(struct in6_addr);
 239                 break;
 240 #endif /* IPv6 */
 241         default:
 242                 return -EPFNOSUPPORT;
 243         }
 244
 245         return netlbl_unlhsh_add(net,
 246                                  dev_name, addr, mask, addr_len,
 247                                  secid, audit_info);
 248 }
 249
 250 /**
 251  * netlbl_cfg_unlbl_static_del - Removes an existing static label
 252  * @net: network namespace
 253  * @dev_name: interface name
 254  * @addr: IP address in network byte order (struct in[6]_addr)
 255  * @mask: address mask in network byte order (struct in[6]_addr)
 256  * @family: address family
 257  * @audit_info: NetLabel audit information
 258  *
 259  * Description:
 260  * Removes an existing NetLabel static label used when protocol provided labels
 261  * are not present on incoming traffic.  If @dev_name is NULL then the default
 262  * interface will be used.  Returns zero on success, negative values on failure.
 263  *
 264  */
 265 int netlbl_cfg_unlbl_static_del(struct net *net,
 266                                 const char *dev_name,
 267                                 const void *addr,
 268                                 const void *mask,
 269                                 u16 family,
 270                                 struct netlbl_audit *audit_info)
 271 {
 272         u32 addr_len;
 273
 274         switch (family) {
 275         case AF_INET:
 276                 addr_len = sizeof(struct in_addr);
 277                 break;
 278 #if IS_ENABLED(CONFIG_IPV6)
 279         case AF_INET6:
 280                 addr_len = sizeof(struct in6_addr);
 281                 break;
 282 #endif /* IPv6 */
 283         default:
 284                 return -EPFNOSUPPORT;
 285         }
 286
 287         return netlbl_unlhsh_remove(net,
 288                                     dev_name, addr, mask, addr_len,
 289                                     audit_info);
 290 }
 291
 292 /**
 293  * netlbl_cfg_cipsov4_add - Add a new CIPSOv4 DOI definition
 294  * @doi_def: CIPSO DOI definition
 295  * @audit_info: NetLabel audit information
 296  *
 297  * Description:
 298  * Add a new CIPSO DOI definition as defined by @doi_def.  Returns zero on
 299  * success and negative values on failure.
 300  *
 301  */
 302 int netlbl_cfg_cipsov4_add(struct cipso_v4_doi *doi_def,
 303                            struct netlbl_audit *audit_info)
 304 {
 305         return cipso_v4_doi_add(doi_def, audit_info);
 306 }
 307
 308 /**
 309  * netlbl_cfg_cipsov4_del - Remove an existing CIPSOv4 DOI definition
 310  * @doi: CIPSO DOI
 311  * @audit_info: NetLabel audit information
 312  *
 313  * Description:
 314  * Remove an existing CIPSO DOI definition matching @doi.  Returns zero on
 315  * success and negative values on failure.
 316  *
 317  */
 318 void netlbl_cfg_cipsov4_del(u32 doi, struct netlbl_audit *audit_info)
 319 {
 320         cipso_v4_doi_remove(doi, audit_info);
 321 }
 322
 323 /**
 324  * netlbl_cfg_cipsov4_map_add - Add a new CIPSOv4 DOI mapping
 325  * @doi: the CIPSO DOI
 326  * @domain: the domain mapping to add
 327  * @addr: IP address
 328  * @mask: IP address mask
 329  * @audit_info: NetLabel audit information
 330  *
 331  * Description:
 332  * Add a new NetLabel/LSM domain mapping for the given CIPSO DOI to the NetLabel
 333  * subsystem.  A @domain value of NULL adds a new default domain mapping.
 334  * Returns zero on success, negative values on failure.
 335  *
 336  */
 337 int netlbl_cfg_cipsov4_map_add(u32 doi,
 338                                const char *domain,
 339                                const struct in_addr *addr,
 340                                const struct in_addr *mask,
 341                                struct netlbl_audit *audit_info)
 342 {
 343         int ret_val = -ENOMEM;
 344         struct cipso_v4_doi *doi_def;
 345         struct netlbl_dom_map *entry;
 346         struct netlbl_domaddr_map *addrmap = NULL;
 347         struct netlbl_domaddr4_map *addrinfo = NULL;
 348
 349         doi_def = cipso_v4_doi_getdef(doi);
 350         if (doi_def == NULL)
 351                 return -ENOENT;
 352
 353         entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
 354         if (entry == NULL)
 355                 goto out_entry;
 356         entry->family = AF_INET;
 357         if (domain != NULL) {
 358                 entry->domain = kstrdup(domain, GFP_ATOMIC);
 359                 if (entry->domain == NULL)
 360                         goto out_domain;
 361         }
 362
 363         if (addr == NULL && mask == NULL) {
 364                 entry->def.cipso = doi_def;
 365                 entry->def.type = NETLBL_NLTYPE_CIPSOV4;
 366         } else if (addr != NULL && mask != NULL) {
 367                 addrmap = kzalloc(sizeof(*addrmap), GFP_ATOMIC);
 368                 if (addrmap == NULL)
 369                         goto out_addrmap;
 370                 INIT_LIST_HEAD(&addrmap->list4);
 371                 INIT_LIST_HEAD(&addrmap->list6);
 372
 373                 addrinfo = kzalloc(sizeof(*addrinfo), GFP_ATOMIC);
 374                 if (addrinfo == NULL)
 375                         goto out_addrinfo;
 376                 addrinfo->def.cipso = doi_def;
 377                 addrinfo->def.type = NETLBL_NLTYPE_CIPSOV4;
 378                 addrinfo->list.addr = addr->s_addr & mask->s_addr;
 379                 addrinfo->list.mask = mask->s_addr;
 380                 addrinfo->list.valid = 1;
 381                 ret_val = netlbl_af4list_add(&addrinfo->list, &addrmap->list4);
 382                 if (ret_val != 0)
 383                         goto cfg_cipsov4_map_add_failure;
 384
 385                 entry->def.addrsel = addrmap;
 386                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
 387         } else {
 388                 ret_val = -EINVAL;
 389                 goto out_addrmap;
 390         }
 391
 392         ret_val = netlbl_domhsh_add(entry, audit_info);
 393         if (ret_val != 0)
 394                 goto cfg_cipsov4_map_add_failure;
 395
 396         return 0;
 397
 398 cfg_cipsov4_map_add_failure:
 399         kfree(addrinfo);
 400 out_addrinfo:
 401         kfree(addrmap);
 402 out_addrmap:
 403         kfree(entry->domain);
 404 out_domain:
 405         kfree(entry);
 406 out_entry:
 407         cipso_v4_doi_putdef(doi_def);
 408         return ret_val;
 409 }
 410
 411 /**
 412  * netlbl_cfg_calipso_add - Add a new CALIPSO DOI definition
 413  * @doi_def: CALIPSO DOI definition
 414  * @audit_info: NetLabel audit information
 415  *
 416  * Description:
 417  * Add a new CALIPSO DOI definition as defined by @doi_def.  Returns zero on
 418  * success and negative values on failure.
 419  *
 420  */
 421 int netlbl_cfg_calipso_add(struct calipso_doi *doi_def,
 422                            struct netlbl_audit *audit_info)
 423 {
 424 #if IS_ENABLED(CONFIG_IPV6)
 425         return calipso_doi_add(doi_def, audit_info);
 426 #else /* IPv6 */
 427         return -ENOSYS;
 428 #endif /* IPv6 */
 429 }
 430
 431 /**
 432  * netlbl_cfg_calipso_del - Remove an existing CALIPSO DOI definition
 433  * @doi: CALIPSO DOI
 434  * @audit_info: NetLabel audit information
 435  *
 436  * Description:
 437  * Remove an existing CALIPSO DOI definition matching @doi.  Returns zero on
 438  * success and negative values on failure.
 439  *
 440  */
 441 void netlbl_cfg_calipso_del(u32 doi, struct netlbl_audit *audit_info)
 442 {
 443 #if IS_ENABLED(CONFIG_IPV6)
 444         calipso_doi_remove(doi, audit_info);
 445 #endif /* IPv6 */
 446 }
 447
 448 /**
 449  * netlbl_cfg_calipso_map_add - Add a new CALIPSO DOI mapping
 450  * @doi: the CALIPSO DOI
 451  * @domain: the domain mapping to add
 452  * @addr: IP address
 453  * @mask: IP address mask
 454  * @audit_info: NetLabel audit information
 455  *
 456  * Description:
 457  * Add a new NetLabel/LSM domain mapping for the given CALIPSO DOI to the
 458  * NetLabel subsystem.  A @domain value of NULL adds a new default domain
 459  * mapping.  Returns zero on success, negative values on failure.
 460  *
 461  */
 462 int netlbl_cfg_calipso_map_add(u32 doi,
 463                                const char *domain,
 464                                const struct in6_addr *addr,
 465                                const struct in6_addr *mask,
 466                                struct netlbl_audit *audit_info)
 467 {
 468 #if IS_ENABLED(CONFIG_IPV6)
 469         int ret_val = -ENOMEM;
 470         struct calipso_doi *doi_def;
 471         struct netlbl_dom_map *entry;
 472         struct netlbl_domaddr_map *addrmap = NULL;
 473         struct netlbl_domaddr6_map *addrinfo = NULL;
 474
 475         doi_def = calipso_doi_getdef(doi);
 476         if (doi_def == NULL)
 477                 return -ENOENT;
 478
 479         entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
 480         if (entry == NULL)
 481                 goto out_entry;
 482         entry->family = AF_INET6;
 483         if (domain != NULL) {
 484                 entry->domain = kstrdup(domain, GFP_ATOMIC);
 485                 if (entry->domain == NULL)
 486                         goto out_domain;
 487         }
 488
 489         if (addr == NULL && mask == NULL) {
 490                 entry->def.calipso = doi_def;
 491                 entry->def.type = NETLBL_NLTYPE_CALIPSO;
 492         } else if (addr != NULL && mask != NULL) {
 493                 addrmap = kzalloc(sizeof(*addrmap), GFP_ATOMIC);
 494                 if (addrmap == NULL)
 495                         goto out_addrmap;
 496                 INIT_LIST_HEAD(&addrmap->list4);
 497                 INIT_LIST_HEAD(&addrmap->list6);
 498
 499                 addrinfo = kzalloc(sizeof(*addrinfo), GFP_ATOMIC);
 500                 if (addrinfo == NULL)
 501                         goto out_addrinfo;
 502                 addrinfo->def.calipso = doi_def;
 503                 addrinfo->def.type = NETLBL_NLTYPE_CALIPSO;
 504                 addrinfo->list.addr = *addr;
 505                 addrinfo->list.addr.s6_addr32[0] &= mask->s6_addr32[0];
 506                 addrinfo->list.addr.s6_addr32[1] &= mask->s6_addr32[1];
 507                 addrinfo->list.addr.s6_addr32[2] &= mask->s6_addr32[2];
 508                 addrinfo->list.addr.s6_addr32[3] &= mask->s6_addr32[3];
 509                 addrinfo->list.mask = *mask;
 510                 addrinfo->list.valid = 1;
 511                 ret_val = netlbl_af6list_add(&addrinfo->list, &addrmap->list6);
 512                 if (ret_val != 0)
 513                         goto cfg_calipso_map_add_failure;
 514
 515                 entry->def.addrsel = addrmap;
 516                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
 517         } else {
 518                 ret_val = -EINVAL;
 519                 goto out_addrmap;
 520         }
 521
 522         ret_val = netlbl_domhsh_add(entry, audit_info);
 523         if (ret_val != 0)
 524                 goto cfg_calipso_map_add_failure;
 525
 526         return 0;
 527
 528 cfg_calipso_map_add_failure:
 529         kfree(addrinfo);
 530 out_addrinfo:
 531         kfree(addrmap);
 532 out_addrmap:
 533         kfree(entry->domain);
 534 out_domain:
 535         kfree(entry);
 536 out_entry:
 537         calipso_doi_putdef(doi_def);
 538         return ret_val;
 539 #else /* IPv6 */
 540         return -ENOSYS;
 541 #endif /* IPv6 */
 542 }
 543
 544 /*
 545  * Security Attribute Functions
 546  */
 547
 548 #define _CM_F_NONE      0x00000000
 549 #define _CM_F_ALLOC     0x00000001
 550 #define _CM_F_WALK      0x00000002
 551
 552 /**
 553  * _netlbl_catmap_getnode - Get a individual node from a catmap
 554  * @catmap: pointer to the category bitmap
 555  * @offset: the requested offset
 556  * @cm_flags: catmap flags, see _CM_F_*
 557  * @gfp_flags: memory allocation flags
 558  *
 559  * Description:
 560  * Iterate through the catmap looking for the node associated with @offset.
 561  * If the _CM_F_ALLOC flag is set in @cm_flags and there is no associated node,
 562  * one will be created and inserted into the catmap.  If the _CM_F_WALK flag is
 563  * set in @cm_flags and there is no associated node, the next highest node will
 564  * be returned.  Returns a pointer to the node on success, NULL on failure.
 565  *
 566  */
 567 static struct netlbl_lsm_catmap *_netlbl_catmap_getnode(
 568                                              struct netlbl_lsm_catmap **catmap,
 569                                              u32 offset,
 570                                              unsigned int cm_flags,
 571                                              gfp_t gfp_flags)
 572 {
 573         struct netlbl_lsm_catmap *iter = *catmap;
 574         struct netlbl_lsm_catmap *prev = NULL;
 575
 576         if (iter == NULL)
 577                 goto catmap_getnode_alloc;
 578         if (offset < iter->startbit)
 579                 goto catmap_getnode_walk;
 580         while (iter && offset >= (iter->startbit + NETLBL_CATMAP_SIZE)) {
 581                 prev = iter;
 582                 iter = iter->next;
 583         }
 584         if (iter == NULL || offset < iter->startbit)
 585                 goto catmap_getnode_walk;
 586
 587         return iter;
 588
 589 catmap_getnode_walk:
 590         if (cm_flags & _CM_F_WALK)
 591                 return iter;
 592 catmap_getnode_alloc:
 593         if (!(cm_flags & _CM_F_ALLOC))
 594                 return NULL;
 595
 596         iter = netlbl_catmap_alloc(gfp_flags);
 597         if (iter == NULL)
 598                 return NULL;
 599         iter->startbit = offset & ~(NETLBL_CATMAP_SIZE - 1);
 600
 601         if (prev == NULL) {
 602                 iter->next = *catmap;
 603                 *catmap = iter;
 604         } else {
 605                 iter->next = prev->next;
 606                 prev->next = iter;
 607         }
 608
 609         return iter;
 610 }
 611
 612 /**
 613  * netlbl_catmap_walk - Walk a LSM secattr catmap looking for a bit
 614  * @catmap: the category bitmap
 615  * @offset: the offset to start searching at, in bits
 616  *
 617  * Description:
 618  * This function walks a LSM secattr category bitmap starting at @offset and
 619  * returns the spot of the first set bit or -ENOENT if no bits are set.
 620  *
 621  */
 622 int netlbl_catmap_walk(struct netlbl_lsm_catmap *catmap, u32 offset)
 623 {
 624         struct netlbl_lsm_catmap *iter = catmap;
 625         u32 idx;
 626         u32 bit;
 627         NETLBL_CATMAP_MAPTYPE bitmap;
 628
 629         iter = _netlbl_catmap_getnode(&catmap, offset, _CM_F_WALK, 0);
 630         if (iter == NULL)
 631                 return -ENOENT;
 632         if (offset > iter->startbit) {
 633                 offset -= iter->startbit;
 634                 idx = offset / NETLBL_CATMAP_MAPSIZE;
 635                 bit = offset % NETLBL_CATMAP_MAPSIZE;
 636         } else {
 637                 idx = 0;
 638                 bit = 0;
 639         }
 640         bitmap = iter->bitmap[idx] >> bit;
 641
 642         for (;;) {
 643                 if (bitmap != 0) {
 644                         while ((bitmap & NETLBL_CATMAP_BIT) == 0) {
 645                                 bitmap >>= 1;
 646                                 bit++;
 647                         }
 648                         return iter->startbit +
 649                                (NETLBL_CATMAP_MAPSIZE * idx) + bit;
 650                 }
 651                 if (++idx >= NETLBL_CATMAP_MAPCNT) {
 652                         if (iter->next != NULL) {
 653                                 iter = iter->next;
 654                                 idx = 0;
 655                         } else
 656                                 return -ENOENT;
 657                 }
 658                 bitmap = iter->bitmap[idx];
 659                 bit = 0;
 660         }
 661
 662         return -ENOENT;
 663 }
 664 EXPORT_SYMBOL(netlbl_catmap_walk);
 665
 666 /**
 667  * netlbl_catmap_walkrng - Find the end of a string of set bits
 668  * @catmap: the category bitmap
 669  * @offset: the offset to start searching at, in bits
 670  *
 671  * Description:
 672  * This function walks a LSM secattr category bitmap starting at @offset and
 673  * returns the spot of the first cleared bit or -ENOENT if the offset is past
 674  * the end of the bitmap.
 675  *
 676  */
 677 int netlbl_catmap_walkrng(struct netlbl_lsm_catmap *catmap, u32 offset)
 678 {
 679         struct netlbl_lsm_catmap *iter;
 680         struct netlbl_lsm_catmap *prev = NULL;
 681         u32 idx;
 682         u32 bit;
 683         NETLBL_CATMAP_MAPTYPE bitmask;
 684         NETLBL_CATMAP_MAPTYPE bitmap;
 685
 686         iter = _netlbl_catmap_getnode(&catmap, offset, _CM_F_WALK, 0);
 687         if (iter == NULL)
 688                 return -ENOENT;
 689         if (offset > iter->startbit) {
 690                 offset -= iter->startbit;
 691                 idx = offset / NETLBL_CATMAP_MAPSIZE;
 692                 bit = offset % NETLBL_CATMAP_MAPSIZE;
 693         } else {
 694                 idx = 0;
 695                 bit = 0;
 696         }
 697         bitmask = NETLBL_CATMAP_BIT << bit;
 698
 699         for (;;) {
 700                 bitmap = iter->bitmap[idx];
 701                 while (bitmask != 0 && (bitmap & bitmask) != 0) {
 702                         bitmask <<= 1;
 703                         bit++;
 704                 }
 705
 706                 if (prev && idx == 0 && bit == 0)
 707                         return prev->startbit + NETLBL_CATMAP_SIZE - 1;
 708                 else if (bitmask != 0)
 709                         return iter->startbit +
 710                                 (NETLBL_CATMAP_MAPSIZE * idx) + bit - 1;
 711                 else if (++idx >= NETLBL_CATMAP_MAPCNT) {
 712                         if (iter->next == NULL)
 713                                 return iter->startbit + NETLBL_CATMAP_SIZE - 1;
 714                         prev = iter;
 715                         iter = iter->next;
 716                         idx = 0;
 717                 }
 718                 bitmask = NETLBL_CATMAP_BIT;
 719                 bit = 0;
 720         }
 721
 722         return -ENOENT;
 723 }
 724
 725 /**
 726  * netlbl_catmap_getlong - Export an unsigned long bitmap
 727  * @catmap: pointer to the category bitmap
 728  * @offset: pointer to the requested offset
 729  * @bitmap: the exported bitmap
 730  *
 731  * Description:
 732  * Export a bitmap with an offset greater than or equal to @offset and return
 733  * it in @bitmap.  The @offset must be aligned to an unsigned long and will be
 734  * updated on return if different from what was requested; if the catmap is
 735  * empty at the requested offset and beyond, the @offset is set to (u32)-1.
 736  * Returns zero on sucess, negative values on failure.
 737  *
 738  */
 739 int netlbl_catmap_getlong(struct netlbl_lsm_catmap *catmap,
 740                           u32 *offset,
 741                           unsigned long *bitmap)
 742 {
 743         struct netlbl_lsm_catmap *iter;
 744         u32 off = *offset;
 745         u32 idx;
 746
 747         /* only allow aligned offsets */
 748         if ((off & (BITS_PER_LONG - 1)) != 0)
 749                 return -EINVAL;
 750
 751         /* a null catmap is equivalent to an empty one */
 752         if (!catmap) {
 753                 *offset = (u32)-1;
 754                 return 0;
 755         }
 756
 757         if (off < catmap->startbit) {
 758                 off = catmap->startbit;
 759                 *offset = off;
 760         }
 761         iter = _netlbl_catmap_getnode(&catmap, off, _CM_F_WALK, 0);
 762         if (iter == NULL) {
 763                 *offset = (u32)-1;
 764                 return 0;
 765         }
 766
 767         if (off < iter->startbit) {
 768                 *offset = iter->startbit;
 769                 off = 0;
 770         } else
 771                 off -= iter->startbit;
 772         idx = off / NETLBL_CATMAP_MAPSIZE;
 773         *bitmap = iter->bitmap[idx] >> (off % NETLBL_CATMAP_MAPSIZE);
 774
 775         return 0;
 776 }
 777
 778 /**
 779  * netlbl_catmap_setbit - Set a bit in a LSM secattr catmap
 780  * @catmap: pointer to the category bitmap
 781  * @bit: the bit to set
 782  * @flags: memory allocation flags
 783  *
 784  * Description:
 785  * Set the bit specified by @bit in @catmap.  Returns zero on success,
 786  * negative values on failure.
 787  *
 788  */
 789 int netlbl_catmap_setbit(struct netlbl_lsm_catmap **catmap,
 790                          u32 bit,
 791                          gfp_t flags)
 792 {
 793         struct netlbl_lsm_catmap *iter;
 794         u32 idx;
 795
 796         iter = _netlbl_catmap_getnode(catmap, bit, _CM_F_ALLOC, flags);
 797         if (iter == NULL)
 798                 return -ENOMEM;
 799
 800         bit -= iter->startbit;
 801         idx = bit / NETLBL_CATMAP_MAPSIZE;
 802         iter->bitmap[idx] |= NETLBL_CATMAP_BIT << (bit % NETLBL_CATMAP_MAPSIZE);
 803
 804         return 0;
 805 }
 806 EXPORT_SYMBOL(netlbl_catmap_setbit);
 807
 808 /**
 809  * netlbl_catmap_setrng - Set a range of bits in a LSM secattr catmap
 810  * @catmap: pointer to the category bitmap
 811  * @start: the starting bit
 812  * @end: the last bit in the string
 813  * @flags: memory allocation flags
 814  *
 815  * Description:
 816  * Set a range of bits, starting at @start and ending with @end.  Returns zero
 817  * on success, negative values on failure.
 818  *
 819  */
 820 int netlbl_catmap_setrng(struct netlbl_lsm_catmap **catmap,
 821                          u32 start,
 822                          u32 end,
 823                          gfp_t flags)
 824 {
 825         int rc = 0;
 826         u32 spot = start;
 827
 828         while (rc == 0 && spot <= end) {
 829                 if (((spot & (BITS_PER_LONG - 1)) == 0) &&
 830                     ((end - spot) > BITS_PER_LONG)) {
 831                         rc = netlbl_catmap_setlong(catmap,
 832                                                    spot,
 833                                                    (unsigned long)-1,
 834                                                    flags);
 835                         spot += BITS_PER_LONG;
 836                 } else
 837                         rc = netlbl_catmap_setbit(catmap, spot++, flags);
 838         }
 839
 840         return rc;
 841 }
 842
 843 /**
 844  * netlbl_catmap_setlong - Import an unsigned long bitmap
 845  * @catmap: pointer to the category bitmap
 846  * @offset: offset to the start of the imported bitmap
 847  * @bitmap: the bitmap to import
 848  * @flags: memory allocation flags
 849  *
 850  * Description:
 851  * Import the bitmap specified in @bitmap into @catmap, using the offset
 852  * in @offset.  The offset must be aligned to an unsigned long.  Returns zero
 853  * on success, negative values on failure.
 854  *
 855  */
 856 int netlbl_catmap_setlong(struct netlbl_lsm_catmap **catmap,
 857                           u32 offset,
 858                           unsigned long bitmap,
 859                           gfp_t flags)
 860 {
 861         struct netlbl_lsm_catmap *iter;
 862         u32 idx;
 863
 864         /* only allow aligned offsets */
 865         if ((offset & (BITS_PER_LONG - 1)) != 0)
 866                 return -EINVAL;
 867
 868         iter = _netlbl_catmap_getnode(catmap, offset, _CM_F_ALLOC, flags);
 869         if (iter == NULL)
 870                 return -ENOMEM;
 871
 872         offset -= iter->startbit;
 873         idx = offset / NETLBL_CATMAP_MAPSIZE;
 874         iter->bitmap[idx] |= (NETLBL_CATMAP_MAPTYPE)bitmap
 875                              << (offset % NETLBL_CATMAP_MAPSIZE);
 876
 877         return 0;
 878 }
 879
 880 /* Bitmap functions
 881  */
 882
 883 /**
 884  * netlbl_bitmap_walk - Walk a bitmap looking for a bit
 885  * @bitmap: the bitmap
 886  * @bitmap_len: length in bits
 887  * @offset: starting offset
 888  * @state: if non-zero, look for a set (1) bit else look for a cleared (0) bit
 889  *
 890  * Description:
 891  * Starting at @offset, walk the bitmap from left to right until either the
 892  * desired bit is found or we reach the end.  Return the bit offset, -1 if
 893  * not found, or -2 if error.
 894  */
 895 int netlbl_bitmap_walk(const unsigned char *bitmap, u32 bitmap_len,
 896                        u32 offset, u8 state)
 897 {
 898         u32 bit_spot;
 899         u32 byte_offset;
 900         unsigned char bitmask;
 901         unsigned char byte;
 902
 903         byte_offset = offset / 8;
 904         byte = bitmap[byte_offset];
 905         bit_spot = offset;
 906         bitmask = 0x80 >> (offset % 8);
 907
 908         while (bit_spot < bitmap_len) {
 909                 if ((state && (byte & bitmask) == bitmask) ||
 910                     (state == 0 && (byte & bitmask) == 0))
 911                         return bit_spot;
 912
 913                 if (++bit_spot >= bitmap_len)
 914                         return -1;
 915                 bitmask >>= 1;
 916                 if (bitmask == 0) {
 917                         byte = bitmap[++byte_offset];
 918                         bitmask = 0x80;
 919                 }
 920         }
 921
 922         return -1;
 923 }
 924 EXPORT_SYMBOL(netlbl_bitmap_walk);
 925
 926 /**
 927  * netlbl_bitmap_setbit - Sets a single bit in a bitmap
 928  * @bitmap: the bitmap
 929  * @bit: the bit
 930  * @state: if non-zero, set the bit (1) else clear the bit (0)
 931  *
 932  * Description:
 933  * Set a single bit in the bitmask.  Returns zero on success, negative values
 934  * on error.
 935  */
 936 void netlbl_bitmap_setbit(unsigned char *bitmap, u32 bit, u8 state)
 937 {
 938         u32 byte_spot;
 939         u8 bitmask;
 940
 941         /* gcc always rounds to zero when doing integer division */
 942         byte_spot = bit / 8;
 943         bitmask = 0x80 >> (bit % 8);
 944         if (state)
 945                 bitmap[byte_spot] |= bitmask;
 946         else
 947                 bitmap[byte_spot] &= ~bitmask;
 948 }
 949 EXPORT_SYMBOL(netlbl_bitmap_setbit);
 950
 951 /*
 952  * LSM Functions
 953  */
 954
 955 /**
 956  * netlbl_enabled - Determine if the NetLabel subsystem is enabled
 957  *
 958  * Description:
 959  * The LSM can use this function to determine if it should use NetLabel
 960  * security attributes in it's enforcement mechanism.  Currently, NetLabel is
 961  * considered to be enabled when it's configuration contains a valid setup for
 962  * at least one labeled protocol (i.e. NetLabel can understand incoming
 963  * labeled packets of at least one type); otherwise NetLabel is considered to
 964  * be disabled.
 965  *
 966  */
 967 int netlbl_enabled(void)
 968 {
 969         /* At some point we probably want to expose this mechanism to the user
 970          * as well so that admins can toggle NetLabel regardless of the
 971          * configuration */
 972         return (atomic_read(&netlabel_mgmt_protocount) > 0);
 973 }
 974
 975 /**
 976  * netlbl_sock_setattr - Label a socket using the correct protocol
 977  * @sk: the socket to label
 978  * @family: protocol family
 979  * @secattr: the security attributes
 980  *
 981  * Description:
 982  * Attach the correct label to the given socket using the security attributes
 983  * specified in @secattr.  This function requires exclusive access to @sk,
 984  * which means it either needs to be in the process of being created or locked.
 985  * Returns zero on success, -EDESTADDRREQ if the domain is configured to use
 986  * network address selectors (can't blindly label the socket), and negative
 987  * values on all other failures.
 988  *
 989  */
 990 int netlbl_sock_setattr(struct sock *sk,
 991                         u16 family,
 992                         const struct netlbl_lsm_secattr *secattr)
 993 {
 994         int ret_val;
 995         struct netlbl_dom_map *dom_entry;
 996
 997         rcu_read_lock();
 998         dom_entry = netlbl_domhsh_getentry(secattr->domain, family);
 999         if (dom_entry == NULL) {
1000                 ret_val = -ENOENT;
1001                 goto socket_setattr_return;
1002         }
1003         switch (family) {
1004         case AF_INET:
1005                 switch (dom_entry->def.type) {
1006                 case NETLBL_NLTYPE_ADDRSELECT:
1007                         ret_val = -EDESTADDRREQ;
1008                         break;
1009                 case NETLBL_NLTYPE_CIPSOV4:
1010                         ret_val = cipso_v4_sock_setattr(sk,
1011                                                         dom_entry->def.cipso,
1012                                                         secattr);
1013                         break;
1014                 case NETLBL_NLTYPE_UNLABELED:
1015                         ret_val = 0;
1016                         break;
1017                 default:
1018                         ret_val = -ENOENT;
1019                 }
1020                 break;
1021 #if IS_ENABLED(CONFIG_IPV6)
1022         case AF_INET6:
1023                 switch (dom_entry->def.type) {
1024                 case NETLBL_NLTYPE_ADDRSELECT:
1025                         ret_val = -EDESTADDRREQ;
1026                         break;
1027                 case NETLBL_NLTYPE_CALIPSO:
1028                         ret_val = calipso_sock_setattr(sk,
1029                                                        dom_entry->def.calipso,
1030                                                        secattr);
1031                         break;
1032                 case NETLBL_NLTYPE_UNLABELED:
1033                         ret_val = 0;
1034                         break;
1035                 default:
1036                         ret_val = -ENOENT;
1037                 }
1038                 break;
1039 #endif /* IPv6 */
1040         default:
1041                 ret_val = -EPROTONOSUPPORT;
1042         }
1043
1044 socket_setattr_return:
1045         rcu_read_unlock();
1046         return ret_val;
1047 }
1048
1049 /**
1050  * netlbl_sock_delattr - Delete all the NetLabel labels on a socket
1051  * @sk: the socket
1052  *
1053  * Description:
1054  * Remove all the NetLabel labeling from @sk.  The caller is responsible for
1055  * ensuring that @sk is locked.
1056  *
1057  */
1058 void netlbl_sock_delattr(struct sock *sk)
1059 {
1060         switch (sk->sk_family) {
1061         case AF_INET:
1062                 cipso_v4_sock_delattr(sk);
1063                 break;
1064 #if IS_ENABLED(CONFIG_IPV6)
1065         case AF_INET6:
1066                 calipso_sock_delattr(sk);
1067                 break;
1068 #endif /* IPv6 */
1069         }
1070 }
1071
1072 /**
1073  * netlbl_sock_getattr - Determine the security attributes of a sock
1074  * @sk: the sock
1075  * @secattr: the security attributes
1076  *
1077  * Description:
1078  * Examines the given sock to see if any NetLabel style labeling has been
1079  * applied to the sock, if so it parses the socket label and returns the
1080  * security attributes in @secattr.  Returns zero on success, negative values
1081  * on failure.
1082  *
1083  */
1084 int netlbl_sock_getattr(struct sock *sk,
1085                         struct netlbl_lsm_secattr *secattr)
1086 {
1087         int ret_val;
1088
1089         switch (sk->sk_family) {
1090         case AF_INET:
1091                 ret_val = cipso_v4_sock_getattr(sk, secattr);
1092                 break;
1093 #if IS_ENABLED(CONFIG_IPV6)
1094         case AF_INET6:
1095                 ret_val = calipso_sock_getattr(sk, secattr);
1096                 break;
1097 #endif /* IPv6 */
1098         default:
1099                 ret_val = -EPROTONOSUPPORT;
1100         }
1101
1102         return ret_val;
1103 }
1104
1105 /**
1106  * netlbl_conn_setattr - Label a connected socket using the correct protocol
1107  * @sk: the socket to label
1108  * @addr: the destination address
1109  * @secattr: the security attributes
1110  *
1111  * Description:
1112  * Attach the correct label to the given connected socket using the security
1113  * attributes specified in @secattr.  The caller is responsible for ensuring
1114  * that @sk is locked.  Returns zero on success, negative values on failure.
1115  *
1116  */
1117 int netlbl_conn_setattr(struct sock *sk,
1118                         struct sockaddr *addr,
1119                         const struct netlbl_lsm_secattr *secattr)
1120 {
1121         int ret_val;
1122         struct sockaddr_in *addr4;
1123 #if IS_ENABLED(CONFIG_IPV6)
1124         struct sockaddr_in6 *addr6;
1125 #endif
1126         struct netlbl_dommap_def *entry;
1127
1128         rcu_read_lock();
1129         switch (addr->sa_family) {
1130         case AF_INET:
1131                 addr4 = (struct sockaddr_in *)addr;
1132                 entry = netlbl_domhsh_getentry_af4(secattr->domain,
1133                                                    addr4->sin_addr.s_addr);
1134                 if (entry == NULL) {
1135                         ret_val = -ENOENT;
1136                         goto conn_setattr_return;
1137                 }
1138                 switch (entry->type) {
1139                 case NETLBL_NLTYPE_CIPSOV4:
1140                         ret_val = cipso_v4_sock_setattr(sk,
1141                                                         entry->cipso, secattr);
1142                         break;
1143                 case NETLBL_NLTYPE_UNLABELED:
1144                         /* just delete the protocols we support for right now
1145                          * but we could remove other protocols if needed */
1146                         netlbl_sock_delattr(sk);
1147                         ret_val = 0;
1148                         break;
1149                 default:
1150                         ret_val = -ENOENT;
1151                 }
1152                 break;
1153 #if IS_ENABLED(CONFIG_IPV6)
1154         case AF_INET6:
1155                 addr6 = (struct sockaddr_in6 *)addr;
1156                 entry = netlbl_domhsh_getentry_af6(secattr->domain,
1157                                                    &addr6->sin6_addr);
1158                 if (entry == NULL) {
1159                         ret_val = -ENOENT;
1160                         goto conn_setattr_return;
1161                 }
1162                 switch (entry->type) {
1163                 case NETLBL_NLTYPE_CALIPSO:
1164                         ret_val = calipso_sock_setattr(sk,
1165                                                        entry->calipso, secattr);
1166                         break;
1167                 case NETLBL_NLTYPE_UNLABELED:
1168                         /* just delete the protocols we support for right now
1169                          * but we could remove other protocols if needed */
1170                         netlbl_sock_delattr(sk);
1171                         ret_val = 0;
1172                         break;
1173                 default:
1174                         ret_val = -ENOENT;
1175                 }
1176                 break;
1177 #endif /* IPv6 */
1178         default:
1179                 ret_val = -EPROTONOSUPPORT;
1180         }
1181
1182 conn_setattr_return:
1183         rcu_read_unlock();
1184         return ret_val;
1185 }
1186
1187 /**
1188  * netlbl_req_setattr - Label a request socket using the correct protocol
1189  * @req: the request socket to label
1190  * @secattr: the security attributes
1191  *
1192  * Description:
1193  * Attach the correct label to the given socket using the security attributes
1194  * specified in @secattr.  Returns zero on success, negative values on failure.
1195  *
1196  */
1197 int netlbl_req_setattr(struct request_sock *req,
1198                        const struct netlbl_lsm_secattr *secattr)
1199 {
1200         int ret_val;
1201         struct netlbl_dommap_def *entry;
1202         struct inet_request_sock *ireq = inet_rsk(req);
1203
1204         rcu_read_lock();
1205         switch (req->rsk_ops->family) {
1206         case AF_INET:
1207                 entry = netlbl_domhsh_getentry_af4(secattr->domain,
1208                                                    ireq->ir_rmt_addr);
1209                 if (entry == NULL) {
1210                         ret_val = -ENOENT;
1211                         goto req_setattr_return;
1212                 }
1213                 switch (entry->type) {
1214                 case NETLBL_NLTYPE_CIPSOV4:
1215                         ret_val = cipso_v4_req_setattr(req,
1216                                                        entry->cipso, secattr);
1217                         break;
1218                 case NETLBL_NLTYPE_UNLABELED:
1219                         netlbl_req_delattr(req);
1220                         ret_val = 0;
1221                         break;
1222                 default:
1223                         ret_val = -ENOENT;
1224                 }
1225                 break;
1226 #if IS_ENABLED(CONFIG_IPV6)
1227         case AF_INET6:
1228                 entry = netlbl_domhsh_getentry_af6(secattr->domain,
1229                                                    &ireq->ir_v6_rmt_addr);
1230                 if (entry == NULL) {
1231                         ret_val = -ENOENT;
1232                         goto req_setattr_return;
1233                 }
1234                 switch (entry->type) {
1235                 case NETLBL_NLTYPE_CALIPSO:
1236                         ret_val = calipso_req_setattr(req,
1237                                                       entry->calipso, secattr);
1238                         break;
1239                 case NETLBL_NLTYPE_UNLABELED:
1240                         netlbl_req_delattr(req);
1241                         ret_val = 0;
1242                         break;
1243                 default:
1244                         ret_val = -ENOENT;
1245                 }
1246                 break;
1247 #endif /* IPv6 */
1248         default:
1249                 ret_val = -EPROTONOSUPPORT;
1250         }
1251
1252 req_setattr_return:
1253         rcu_read_unlock();
1254         return ret_val;
1255 }
1256
1257 /**
1258 * netlbl_req_delattr - Delete all the NetLabel labels on a socket
1259 * @req: the socket
1260 *
1261 * Description:
1262 * Remove all the NetLabel labeling from @req.
1263 *
1264 */
1265 void netlbl_req_delattr(struct request_sock *req)
1266 {
1267         switch (req->rsk_ops->family) {
1268         case AF_INET:
1269                 cipso_v4_req_delattr(req);
1270                 break;
1271 #if IS_ENABLED(CONFIG_IPV6)
1272         case AF_INET6:
1273                 calipso_req_delattr(req);
1274                 break;
1275 #endif /* IPv6 */
1276         }
1277 }
1278
1279 /**
1280  * netlbl_skbuff_setattr - Label a packet using the correct protocol
1281  * @skb: the packet
1282  * @family: protocol family
1283  * @secattr: the security attributes
1284  *
1285  * Description:
1286  * Attach the correct label to the given packet using the security attributes
1287  * specified in @secattr.  Returns zero on success, negative values on failure.
1288  *
1289  */
1290 int netlbl_skbuff_setattr(struct sk_buff *skb,
1291                           u16 family,
1292                           const struct netlbl_lsm_secattr *secattr)
1293 {
1294         int ret_val;
1295         struct iphdr *hdr4;
1296 #if IS_ENABLED(CONFIG_IPV6)
1297         struct ipv6hdr *hdr6;
1298 #endif
1299         struct netlbl_dommap_def *entry;
1300
1301         rcu_read_lock();
1302         switch (family) {
1303         case AF_INET:
1304                 hdr4 = ip_hdr(skb);
1305                 entry = netlbl_domhsh_getentry_af4(secattr->domain,
1306                                                    hdr4->daddr);
1307                 if (entry == NULL) {
1308                         ret_val = -ENOENT;
1309                         goto skbuff_setattr_return;
1310                 }
1311                 switch (entry->type) {
1312                 case NETLBL_NLTYPE_CIPSOV4:
1313                         ret_val = cipso_v4_skbuff_setattr(skb, entry->cipso,
1314                                                           secattr);
1315                         break;
1316                 case NETLBL_NLTYPE_UNLABELED:
1317                         /* just delete the protocols we support for right now
1318                          * but we could remove other protocols if needed */
1319                         ret_val = cipso_v4_skbuff_delattr(skb);
1320                         break;
1321                 default:
1322                         ret_val = -ENOENT;
1323                 }
1324                 break;
1325 #if IS_ENABLED(CONFIG_IPV6)
1326         case AF_INET6:
1327                 hdr6 = ipv6_hdr(skb);
1328                 entry = netlbl_domhsh_getentry_af6(secattr->domain,
1329                                                    &hdr6->daddr);
1330                 if (entry == NULL) {
1331                         ret_val = -ENOENT;
1332                         goto skbuff_setattr_return;
1333                 }
1334                 switch (entry->type) {
1335                 case NETLBL_NLTYPE_CALIPSO:
1336                         ret_val = calipso_skbuff_setattr(skb, entry->calipso,
1337                                                          secattr);
1338                         break;
1339                 case NETLBL_NLTYPE_UNLABELED:
1340                         /* just delete the protocols we support for right now
1341                          * but we could remove other protocols if needed */
1342                         ret_val = calipso_skbuff_delattr(skb);
1343                         break;
1344                 default:
1345                         ret_val = -ENOENT;
1346                 }
1347                 break;
1348 #endif /* IPv6 */
1349         default:
1350                 ret_val = -EPROTONOSUPPORT;
1351         }
1352
1353 skbuff_setattr_return:
1354         rcu_read_unlock();
1355         return ret_val;
1356 }
1357
1358 /**
1359  * netlbl_skbuff_getattr - Determine the security attributes of a packet
1360  * @skb: the packet
1361  * @family: protocol family
1362  * @secattr: the security attributes
1363  *
1364  * Description:
1365  * Examines the given packet to see if a recognized form of packet labeling
1366  * is present, if so it parses the packet label and returns the security
1367  * attributes in @secattr.  Returns zero on success, negative values on
1368  * failure.
1369  *
1370  */
1371 int netlbl_skbuff_getattr(const struct sk_buff *skb,
1372                           u16 family,
1373                           struct netlbl_lsm_secattr *secattr)
1374 {
1375         unsigned char *ptr;
1376
1377         switch (family) {
1378         case AF_INET:
1379                 ptr = cipso_v4_optptr(skb);
1380                 if (ptr && cipso_v4_getattr(ptr, secattr) == 0)
1381                         return 0;
1382                 break;
1383 #if IS_ENABLED(CONFIG_IPV6)
1384         case AF_INET6:
1385                 ptr = calipso_optptr(skb);
1386                 if (ptr && calipso_getattr(ptr, secattr) == 0)
1387                         return 0;
1388                 break;
1389 #endif /* IPv6 */
1390         }
1391
1392         return netlbl_unlabel_getattr(skb, family, secattr);
1393 }
1394
1395 /**
1396  * netlbl_skbuff_err - Handle a LSM error on a sk_buff
1397  * @skb: the packet
1398  * @family: the family
1399  * @error: the error code
1400  * @gateway: true if host is acting as a gateway, false otherwise
1401  *
1402  * Description:
1403  * Deal with a LSM problem when handling the packet in @skb, typically this is
1404  * a permission denied problem (-EACCES).  The correct action is determined
1405  * according to the packet's labeling protocol.
1406  *
1407  */
1408 void netlbl_skbuff_err(struct sk_buff *skb, u16 family, int error, int gateway)
1409 {
1410         switch (family) {
1411         case AF_INET:
1412                 if (cipso_v4_optptr(skb))
1413                         cipso_v4_error(skb, error, gateway);
1414                 break;
1415         }
1416 }
1417
1418 /**
1419  * netlbl_cache_invalidate - Invalidate all of the NetLabel protocol caches
1420  *
1421  * Description:
1422  * For all of the NetLabel protocols that support some form of label mapping
1423  * cache, invalidate the cache.  Returns zero on success, negative values on
1424  * error.
1425  *
1426  */
1427 void netlbl_cache_invalidate(void)
1428 {
1429         cipso_v4_cache_invalidate();
1430 #if IS_ENABLED(CONFIG_IPV6)
1431         calipso_cache_invalidate();
1432 #endif /* IPv6 */
1433 }
1434
1435 /**
1436  * netlbl_cache_add - Add an entry to a NetLabel protocol cache
1437  * @skb: the packet
1438  * @family: the family
1439  * @secattr: the packet's security attributes
1440  *
1441  * Description:
1442  * Add the LSM security attributes for the given packet to the underlying
1443  * NetLabel protocol's label mapping cache.  Returns zero on success, negative
1444  * values on error.
1445  *
1446  */
1447 int netlbl_cache_add(const struct sk_buff *skb, u16 family,
1448                      const struct netlbl_lsm_secattr *secattr)
1449 {
1450         unsigned char *ptr;
1451
1452         if ((secattr->flags & NETLBL_SECATTR_CACHE) == 0)
1453                 return -ENOMSG;
1454
1455         switch (family) {
1456         case AF_INET:
1457                 ptr = cipso_v4_optptr(skb);
1458                 if (ptr)
1459                         return cipso_v4_cache_add(ptr, secattr);
1460                 break;
1461 #if IS_ENABLED(CONFIG_IPV6)
1462         case AF_INET6:
1463                 ptr = calipso_optptr(skb);
1464                 if (ptr)
1465                         return calipso_cache_add(ptr, secattr);
1466                 break;
1467 #endif /* IPv6 */
1468         }
1469         return -ENOMSG;
1470 }
1471
1472 /*
1473  * Protocol Engine Functions
1474  */
1475
1476 /**
1477  * netlbl_audit_start - Start an audit message
1478  * @type: audit message type
1479  * @audit_info: NetLabel audit information
1480  *
1481  * Description:
1482  * Start an audit message using the type specified in @type and fill the audit
1483  * message with some fields common to all NetLabel audit messages.  This
1484  * function should only be used by protocol engines, not LSMs.  Returns a
1485  * pointer to the audit buffer on success, NULL on failure.
1486  *
1487  */
1488 struct audit_buffer *netlbl_audit_start(int type,
1489                                         struct netlbl_audit *audit_info)
1490 {
1491         return netlbl_audit_start_common(type, audit_info);
1492 }
1493 EXPORT_SYMBOL(netlbl_audit_start);
1494
1495 /*
1496  * Setup Functions
1497  */
1498
1499 /**
1500  * netlbl_init - Initialize NetLabel
1501  *
1502  * Description:
1503  * Perform the required NetLabel initialization before first use.
1504  *
1505  */
1506 static int __init netlbl_init(void)
1507 {
1508         int ret_val;
1509
1510         printk(KERN_INFO "NetLabel: Initializing\n");
1511         printk(KERN_INFO "NetLabel:  domain hash size = %u\n",
1512                (1 << NETLBL_DOMHSH_BITSIZE));
1513         printk(KERN_INFO "NetLabel:  protocols = UNLABELED CIPSOv4 CALIPSO\n");
1514
1515         ret_val = netlbl_domhsh_init(NETLBL_DOMHSH_BITSIZE);
1516         if (ret_val != 0)
1517                 goto init_failure;
1518
1519         ret_val = netlbl_unlabel_init(NETLBL_UNLHSH_BITSIZE);
1520         if (ret_val != 0)
1521                 goto init_failure;
1522
1523         ret_val = netlbl_netlink_init();
1524         if (ret_val != 0)
1525                 goto init_failure;
1526
1527         ret_val = netlbl_unlabel_defconf();
1528         if (ret_val != 0)
1529                 goto init_failure;
1530         printk(KERN_INFO "NetLabel:  unlabeled traffic allowed by default\n");
1531
1532         return 0;
1533
1534 init_failure:
1535         panic("NetLabel: failed to initialize properly (%d)\n", ret_val);
1536 }
1537
1538 subsys_initcall(netlbl_init);