net/netlabel/netlabel_mgmt.c

   1 /*
   2  * NetLabel Management Support
   3  *
   4  * This file defines the management functions for the NetLabel system.  The
   5  * NetLabel system manages static and dynamic label mappings for network
   6  * protocols such 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, write to the Free Software
  27  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  28  *
  29  */
  30
  31 #include <linux/types.h>
  32 #include <linux/socket.h>
  33 #include <linux/string.h>
  34 #include <linux/skbuff.h>
  35 #include <linux/in.h>
  36 #include <linux/in6.h>
  37 #include <linux/slab.h>
  38 #include <net/sock.h>
  39 #include <net/netlink.h>
  40 #include <net/genetlink.h>
  41 #include <net/ip.h>
  42 #include <net/ipv6.h>
  43 #include <net/netlabel.h>
  44 #include <net/cipso_ipv4.h>
  45 #include <linux/atomic.h>
  46
  47 #include "netlabel_domainhash.h"
  48 #include "netlabel_user.h"
  49 #include "netlabel_mgmt.h"
  50
  51 /* NetLabel configured protocol counter */
  52 atomic_t netlabel_mgmt_protocount = ATOMIC_INIT(0);
  53
  54 /* Argument struct for netlbl_domhsh_walk() */
  55 struct netlbl_domhsh_walk_arg {
  56         struct netlink_callback *nl_cb;
  57         struct sk_buff *skb;
  58         u32 seq;
  59 };
  60
  61 /* NetLabel Generic NETLINK CIPSOv4 family */
  62 static struct genl_family netlbl_mgmt_gnl_family = {
  63         .id = GENL_ID_GENERATE,
  64         .hdrsize = 0,
  65         .name = NETLBL_NLTYPE_MGMT_NAME,
  66         .version = NETLBL_PROTO_VERSION,
  67         .maxattr = NLBL_MGMT_A_MAX,
  68 };
  69
  70 /* NetLabel Netlink attribute policy */
  71 static const struct nla_policy netlbl_mgmt_genl_policy[NLBL_MGMT_A_MAX + 1] = {
  72         [NLBL_MGMT_A_DOMAIN] = { .type = NLA_NUL_STRING },
  73         [NLBL_MGMT_A_PROTOCOL] = { .type = NLA_U32 },
  74         [NLBL_MGMT_A_VERSION] = { .type = NLA_U32 },
  75         [NLBL_MGMT_A_CV4DOI] = { .type = NLA_U32 },
  76 };
  77
  78 /*
  79  * Helper Functions
  80  */
  81
  82 /**
  83  * netlbl_mgmt_add - Handle an ADD message
  84  * @info: the Generic NETLINK info block
  85  * @audit_info: NetLabel audit information
  86  *
  87  * Description:
  88  * Helper function for the ADD and ADDDEF messages to add the domain mappings
  89  * from the message to the hash table.  See netlabel.h for a description of the
  90  * message format.  Returns zero on success, negative values on failure.
  91  *
  92  */
  93 static int netlbl_mgmt_add_common(struct genl_info *info,
  94                                   struct netlbl_audit *audit_info)
  95 {
  96         int ret_val = -EINVAL;
  97         struct netlbl_dom_map *entry = NULL;
  98         struct netlbl_domaddr_map *addrmap = NULL;
  99         struct cipso_v4_doi *cipsov4 = NULL;
 100         u32 tmp_val;
 101
 102         entry = kzalloc(sizeof(*entry), GFP_KERNEL);
 103         if (entry == NULL) {
 104                 ret_val = -ENOMEM;
 105                 goto add_failure;
 106         }
 107         entry->def.type = nla_get_u32(info->attrs[NLBL_MGMT_A_PROTOCOL]);
 108         if (info->attrs[NLBL_MGMT_A_DOMAIN]) {
 109                 size_t tmp_size = nla_len(info->attrs[NLBL_MGMT_A_DOMAIN]);
 110                 entry->domain = kmalloc(tmp_size, GFP_KERNEL);
 111                 if (entry->domain == NULL) {
 112                         ret_val = -ENOMEM;
 113                         goto add_failure;
 114                 }
 115                 nla_strlcpy(entry->domain,
 116                             info->attrs[NLBL_MGMT_A_DOMAIN], tmp_size);
 117         }
 118
 119         /* NOTE: internally we allow/use a entry->def.type value of
 120          *       NETLBL_NLTYPE_ADDRSELECT but we don't currently allow users
 121          *       to pass that as a protocol value because we need to know the
 122          *       "real" protocol */
 123
 124         switch (entry->def.type) {
 125         case NETLBL_NLTYPE_UNLABELED:
 126                 break;
 127         case NETLBL_NLTYPE_CIPSOV4:
 128                 if (!info->attrs[NLBL_MGMT_A_CV4DOI])
 129                         goto add_failure;
 130
 131                 tmp_val = nla_get_u32(info->attrs[NLBL_MGMT_A_CV4DOI]);
 132                 cipsov4 = cipso_v4_doi_getdef(tmp_val);
 133                 if (cipsov4 == NULL)
 134                         goto add_failure;
 135                 entry->def.cipso = cipsov4;
 136                 break;
 137         default:
 138                 goto add_failure;
 139         }
 140
 141         if (info->attrs[NLBL_MGMT_A_IPV4ADDR]) {
 142                 struct in_addr *addr;
 143                 struct in_addr *mask;
 144                 struct netlbl_domaddr4_map *map;
 145
 146                 addrmap = kzalloc(sizeof(*addrmap), GFP_KERNEL);
 147                 if (addrmap == NULL) {
 148                         ret_val = -ENOMEM;
 149                         goto add_failure;
 150                 }
 151                 INIT_LIST_HEAD(&addrmap->list4);
 152                 INIT_LIST_HEAD(&addrmap->list6);
 153
 154                 if (nla_len(info->attrs[NLBL_MGMT_A_IPV4ADDR]) !=
 155                     sizeof(struct in_addr)) {
 156                         ret_val = -EINVAL;
 157                         goto add_failure;
 158                 }
 159                 if (nla_len(info->attrs[NLBL_MGMT_A_IPV4MASK]) !=
 160                     sizeof(struct in_addr)) {
 161                         ret_val = -EINVAL;
 162                         goto add_failure;
 163                 }
 164                 addr = nla_data(info->attrs[NLBL_MGMT_A_IPV4ADDR]);
 165                 mask = nla_data(info->attrs[NLBL_MGMT_A_IPV4MASK]);
 166
 167                 map = kzalloc(sizeof(*map), GFP_KERNEL);
 168                 if (map == NULL) {
 169                         ret_val = -ENOMEM;
 170                         goto add_failure;
 171                 }
 172                 map->list.addr = addr->s_addr & mask->s_addr;
 173                 map->list.mask = mask->s_addr;
 174                 map->list.valid = 1;
 175                 map->def.type = entry->def.type;
 176                 if (cipsov4)
 177                         map->def.cipso = cipsov4;
 178
 179                 ret_val = netlbl_af4list_add(&map->list, &addrmap->list4);
 180                 if (ret_val != 0) {
 181                         kfree(map);
 182                         goto add_failure;
 183                 }
 184
 185                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
 186                 entry->def.addrsel = addrmap;
 187 #if IS_ENABLED(CONFIG_IPV6)
 188         } else if (info->attrs[NLBL_MGMT_A_IPV6ADDR]) {
 189                 struct in6_addr *addr;
 190                 struct in6_addr *mask;
 191                 struct netlbl_domaddr6_map *map;
 192
 193                 addrmap = kzalloc(sizeof(*addrmap), GFP_KERNEL);
 194                 if (addrmap == NULL) {
 195                         ret_val = -ENOMEM;
 196                         goto add_failure;
 197                 }
 198                 INIT_LIST_HEAD(&addrmap->list4);
 199                 INIT_LIST_HEAD(&addrmap->list6);
 200
 201                 if (nla_len(info->attrs[NLBL_MGMT_A_IPV6ADDR]) !=
 202                     sizeof(struct in6_addr)) {
 203                         ret_val = -EINVAL;
 204                         goto add_failure;
 205                 }
 206                 if (nla_len(info->attrs[NLBL_MGMT_A_IPV6MASK]) !=
 207                     sizeof(struct in6_addr)) {
 208                         ret_val = -EINVAL;
 209                         goto add_failure;
 210                 }
 211                 addr = nla_data(info->attrs[NLBL_MGMT_A_IPV6ADDR]);
 212                 mask = nla_data(info->attrs[NLBL_MGMT_A_IPV6MASK]);
 213
 214                 map = kzalloc(sizeof(*map), GFP_KERNEL);
 215                 if (map == NULL) {
 216                         ret_val = -ENOMEM;
 217                         goto add_failure;
 218                 }
 219                 map->list.addr = *addr;
 220                 map->list.addr.s6_addr32[0] &= mask->s6_addr32[0];
 221                 map->list.addr.s6_addr32[1] &= mask->s6_addr32[1];
 222                 map->list.addr.s6_addr32[2] &= mask->s6_addr32[2];
 223                 map->list.addr.s6_addr32[3] &= mask->s6_addr32[3];
 224                 map->list.mask = *mask;
 225                 map->list.valid = 1;
 226                 map->def.type = entry->def.type;
 227
 228                 ret_val = netlbl_af6list_add(&map->list, &addrmap->list6);
 229                 if (ret_val != 0) {
 230                         kfree(map);
 231                         goto add_failure;
 232                 }
 233
 234                 entry->def.type = NETLBL_NLTYPE_ADDRSELECT;
 235                 entry->def.addrsel = addrmap;
 236 #endif /* IPv6 */
 237         }
 238
 239         ret_val = netlbl_domhsh_add(entry, audit_info);
 240         if (ret_val != 0)
 241                 goto add_failure;
 242
 243         return 0;
 244
 245 add_failure:
 246         if (cipsov4)
 247                 cipso_v4_doi_putdef(cipsov4);
 248         if (entry)
 249                 kfree(entry->domain);
 250         kfree(addrmap);
 251         kfree(entry);
 252         return ret_val;
 253 }
 254
 255 /**
 256  * netlbl_mgmt_listentry - List a NetLabel/LSM domain map entry
 257  * @skb: the NETLINK buffer
 258  * @entry: the map entry
 259  *
 260  * Description:
 261  * This function is a helper function used by the LISTALL and LISTDEF command
 262  * handlers.  The caller is responsible for ensuring that the RCU read lock
 263  * is held.  Returns zero on success, negative values on failure.
 264  *
 265  */
 266 static int netlbl_mgmt_listentry(struct sk_buff *skb,
 267                                  struct netlbl_dom_map *entry)
 268 {
 269         int ret_val = 0;
 270         struct nlattr *nla_a;
 271         struct nlattr *nla_b;
 272         struct netlbl_af4list *iter4;
 273 #if IS_ENABLED(CONFIG_IPV6)
 274         struct netlbl_af6list *iter6;
 275 #endif
 276
 277         if (entry->domain != NULL) {
 278                 ret_val = nla_put_string(skb,
 279                                          NLBL_MGMT_A_DOMAIN, entry->domain);
 280                 if (ret_val != 0)
 281                         return ret_val;
 282         }
 283
 284         switch (entry->def.type) {
 285         case NETLBL_NLTYPE_ADDRSELECT:
 286                 nla_a = nla_nest_start(skb, NLBL_MGMT_A_SELECTORLIST);
 287                 if (nla_a == NULL)
 288                         return -ENOMEM;
 289
 290                 netlbl_af4list_foreach_rcu(iter4, &entry->def.addrsel->list4) {
 291                         struct netlbl_domaddr4_map *map4;
 292                         struct in_addr addr_struct;
 293
 294                         nla_b = nla_nest_start(skb, NLBL_MGMT_A_ADDRSELECTOR);
 295                         if (nla_b == NULL)
 296                                 return -ENOMEM;
 297
 298                         addr_struct.s_addr = iter4->addr;
 299                         ret_val = nla_put(skb, NLBL_MGMT_A_IPV4ADDR,
 300                                           sizeof(struct in_addr),
 301                                           &addr_struct);
 302                         if (ret_val != 0)
 303                                 return ret_val;
 304                         addr_struct.s_addr = iter4->mask;
 305                         ret_val = nla_put(skb, NLBL_MGMT_A_IPV4MASK,
 306                                           sizeof(struct in_addr),
 307                                           &addr_struct);
 308                         if (ret_val != 0)
 309                                 return ret_val;
 310                         map4 = netlbl_domhsh_addr4_entry(iter4);
 311                         ret_val = nla_put_u32(skb, NLBL_MGMT_A_PROTOCOL,
 312                                               map4->def.type);
 313                         if (ret_val != 0)
 314                                 return ret_val;
 315                         switch (map4->def.type) {
 316                         case NETLBL_NLTYPE_CIPSOV4:
 317                                 ret_val = nla_put_u32(skb, NLBL_MGMT_A_CV4DOI,
 318                                                       map4->def.cipso->doi);
 319                                 if (ret_val != 0)
 320                                         return ret_val;
 321                                 break;
 322                         }
 323
 324                         nla_nest_end(skb, nla_b);
 325                 }
 326 #if IS_ENABLED(CONFIG_IPV6)
 327                 netlbl_af6list_foreach_rcu(iter6, &entry->def.addrsel->list6) {
 328                         struct netlbl_domaddr6_map *map6;
 329
 330                         nla_b = nla_nest_start(skb, NLBL_MGMT_A_ADDRSELECTOR);
 331                         if (nla_b == NULL)
 332                                 return -ENOMEM;
 333
 334                         ret_val = nla_put(skb, NLBL_MGMT_A_IPV6ADDR,
 335                                           sizeof(struct in6_addr),
 336                                           &iter6->addr);
 337                         if (ret_val != 0)
 338                                 return ret_val;
 339                         ret_val = nla_put(skb, NLBL_MGMT_A_IPV6MASK,
 340                                           sizeof(struct in6_addr),
 341                                           &iter6->mask);
 342                         if (ret_val != 0)
 343                                 return ret_val;
 344                         map6 = netlbl_domhsh_addr6_entry(iter6);
 345                         ret_val = nla_put_u32(skb, NLBL_MGMT_A_PROTOCOL,
 346                                               map6->def.type);
 347                         if (ret_val != 0)
 348                                 return ret_val;
 349
 350                         nla_nest_end(skb, nla_b);
 351                 }
 352 #endif /* IPv6 */
 353
 354                 nla_nest_end(skb, nla_a);
 355                 break;
 356         case NETLBL_NLTYPE_UNLABELED:
 357                 ret_val = nla_put_u32(skb,NLBL_MGMT_A_PROTOCOL,entry->def.type);
 358                 break;
 359         case NETLBL_NLTYPE_CIPSOV4:
 360                 ret_val = nla_put_u32(skb,NLBL_MGMT_A_PROTOCOL,entry->def.type);
 361                 if (ret_val != 0)
 362                         return ret_val;
 363                 ret_val = nla_put_u32(skb, NLBL_MGMT_A_CV4DOI,
 364                                       entry->def.cipso->doi);
 365                 break;
 366         }
 367
 368         return ret_val;
 369 }
 370
 371 /*
 372  * NetLabel Command Handlers
 373  */
 374
 375 /**
 376  * netlbl_mgmt_add - Handle an ADD message
 377  * @skb: the NETLINK buffer
 378  * @info: the Generic NETLINK info block
 379  *
 380  * Description:
 381  * Process a user generated ADD message and add the domains from the message
 382  * to the hash table.  See netlabel.h for a description of the message format.
 383  * Returns zero on success, negative values on failure.
 384  *
 385  */
 386 static int netlbl_mgmt_add(struct sk_buff *skb, struct genl_info *info)
 387 {
 388         struct netlbl_audit audit_info;
 389
 390         if ((!info->attrs[NLBL_MGMT_A_DOMAIN]) ||
 391             (!info->attrs[NLBL_MGMT_A_PROTOCOL]) ||
 392             (info->attrs[NLBL_MGMT_A_IPV4ADDR] &&
 393              info->attrs[NLBL_MGMT_A_IPV6ADDR]) ||
 394             (info->attrs[NLBL_MGMT_A_IPV4MASK] &&
 395              info->attrs[NLBL_MGMT_A_IPV6MASK]) ||
 396             ((info->attrs[NLBL_MGMT_A_IPV4ADDR] != NULL) ^
 397              (info->attrs[NLBL_MGMT_A_IPV4MASK] != NULL)) ||
 398             ((info->attrs[NLBL_MGMT_A_IPV6ADDR] != NULL) ^
 399              (info->attrs[NLBL_MGMT_A_IPV6MASK] != NULL)))
 400                 return -EINVAL;
 401
 402         netlbl_netlink_auditinfo(skb, &audit_info);
 403
 404         return netlbl_mgmt_add_common(info, &audit_info);
 405 }
 406
 407 /**
 408  * netlbl_mgmt_remove - Handle a REMOVE message
 409  * @skb: the NETLINK buffer
 410  * @info: the Generic NETLINK info block
 411  *
 412  * Description:
 413  * Process a user generated REMOVE message and remove the specified domain
 414  * mappings.  Returns zero on success, negative values on failure.
 415  *
 416  */
 417 static int netlbl_mgmt_remove(struct sk_buff *skb, struct genl_info *info)
 418 {
 419         char *domain;
 420         struct netlbl_audit audit_info;
 421
 422         if (!info->attrs[NLBL_MGMT_A_DOMAIN])
 423                 return -EINVAL;
 424
 425         netlbl_netlink_auditinfo(skb, &audit_info);
 426
 427         domain = nla_data(info->attrs[NLBL_MGMT_A_DOMAIN]);
 428         return netlbl_domhsh_remove(domain, &audit_info);
 429 }
 430
 431 /**
 432  * netlbl_mgmt_listall_cb - netlbl_domhsh_walk() callback for LISTALL
 433  * @entry: the domain mapping hash table entry
 434  * @arg: the netlbl_domhsh_walk_arg structure
 435  *
 436  * Description:
 437  * This function is designed to be used as a callback to the
 438  * netlbl_domhsh_walk() function for use in generating a response for a LISTALL
 439  * message.  Returns the size of the message on success, negative values on
 440  * failure.
 441  *
 442  */
 443 static int netlbl_mgmt_listall_cb(struct netlbl_dom_map *entry, void *arg)
 444 {
 445         int ret_val = -ENOMEM;
 446         struct netlbl_domhsh_walk_arg *cb_arg = arg;
 447         void *data;
 448
 449         data = genlmsg_put(cb_arg->skb, NETLINK_CB(cb_arg->nl_cb->skb).portid,
 450                            cb_arg->seq, &netlbl_mgmt_gnl_family,
 451                            NLM_F_MULTI, NLBL_MGMT_C_LISTALL);
 452         if (data == NULL)
 453                 goto listall_cb_failure;
 454
 455         ret_val = netlbl_mgmt_listentry(cb_arg->skb, entry);
 456         if (ret_val != 0)
 457                 goto listall_cb_failure;
 458
 459         cb_arg->seq++;
 460         return genlmsg_end(cb_arg->skb, data);
 461
 462 listall_cb_failure:
 463         genlmsg_cancel(cb_arg->skb, data);
 464         return ret_val;
 465 }
 466
 467 /**
 468  * netlbl_mgmt_listall - Handle a LISTALL message
 469  * @skb: the NETLINK buffer
 470  * @cb: the NETLINK callback
 471  *
 472  * Description:
 473  * Process a user generated LISTALL message and dumps the domain hash table in
 474  * a form suitable for use in a kernel generated LISTALL message.  Returns zero
 475  * on success, negative values on failure.
 476  *
 477  */
 478 static int netlbl_mgmt_listall(struct sk_buff *skb,
 479                                struct netlink_callback *cb)
 480 {
 481         struct netlbl_domhsh_walk_arg cb_arg;
 482         u32 skip_bkt = cb->args[0];
 483         u32 skip_chain = cb->args[1];
 484
 485         cb_arg.nl_cb = cb;
 486         cb_arg.skb = skb;
 487         cb_arg.seq = cb->nlh->nlmsg_seq;
 488
 489         netlbl_domhsh_walk(&skip_bkt,
 490                            &skip_chain,
 491                            netlbl_mgmt_listall_cb,
 492                            &cb_arg);
 493
 494         cb->args[0] = skip_bkt;
 495         cb->args[1] = skip_chain;
 496         return skb->len;
 497 }
 498
 499 /**
 500  * netlbl_mgmt_adddef - Handle an ADDDEF message
 501  * @skb: the NETLINK buffer
 502  * @info: the Generic NETLINK info block
 503  *
 504  * Description:
 505  * Process a user generated ADDDEF message and respond accordingly.  Returns
 506  * zero on success, negative values on failure.
 507  *
 508  */
 509 static int netlbl_mgmt_adddef(struct sk_buff *skb, struct genl_info *info)
 510 {
 511         struct netlbl_audit audit_info;
 512
 513         if ((!info->attrs[NLBL_MGMT_A_PROTOCOL]) ||
 514             (info->attrs[NLBL_MGMT_A_IPV4ADDR] &&
 515              info->attrs[NLBL_MGMT_A_IPV6ADDR]) ||
 516             (info->attrs[NLBL_MGMT_A_IPV4MASK] &&
 517              info->attrs[NLBL_MGMT_A_IPV6MASK]) ||
 518             ((info->attrs[NLBL_MGMT_A_IPV4ADDR] != NULL) ^
 519              (info->attrs[NLBL_MGMT_A_IPV4MASK] != NULL)) ||
 520             ((info->attrs[NLBL_MGMT_A_IPV6ADDR] != NULL) ^
 521              (info->attrs[NLBL_MGMT_A_IPV6MASK] != NULL)))
 522                 return -EINVAL;
 523
 524         netlbl_netlink_auditinfo(skb, &audit_info);
 525
 526         return netlbl_mgmt_add_common(info, &audit_info);
 527 }
 528
 529 /**
 530  * netlbl_mgmt_removedef - Handle a REMOVEDEF message
 531  * @skb: the NETLINK buffer
 532  * @info: the Generic NETLINK info block
 533  *
 534  * Description:
 535  * Process a user generated REMOVEDEF message and remove the default domain
 536  * mapping.  Returns zero on success, negative values on failure.
 537  *
 538  */
 539 static int netlbl_mgmt_removedef(struct sk_buff *skb, struct genl_info *info)
 540 {
 541         struct netlbl_audit audit_info;
 542
 543         netlbl_netlink_auditinfo(skb, &audit_info);
 544
 545         return netlbl_domhsh_remove_default(&audit_info);
 546 }
 547
 548 /**
 549  * netlbl_mgmt_listdef - Handle a LISTDEF message
 550  * @skb: the NETLINK buffer
 551  * @info: the Generic NETLINK info block
 552  *
 553  * Description:
 554  * Process a user generated LISTDEF message and dumps the default domain
 555  * mapping in a form suitable for use in a kernel generated LISTDEF message.
 556  * Returns zero on success, negative values on failure.
 557  *
 558  */
 559 static int netlbl_mgmt_listdef(struct sk_buff *skb, struct genl_info *info)
 560 {
 561         int ret_val = -ENOMEM;
 562         struct sk_buff *ans_skb = NULL;
 563         void *data;
 564         struct netlbl_dom_map *entry;
 565
 566         ans_skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 567         if (ans_skb == NULL)
 568                 return -ENOMEM;
 569         data = genlmsg_put_reply(ans_skb, info, &netlbl_mgmt_gnl_family,
 570                                  0, NLBL_MGMT_C_LISTDEF);
 571         if (data == NULL)
 572                 goto listdef_failure;
 573
 574         rcu_read_lock();
 575         entry = netlbl_domhsh_getentry(NULL);
 576         if (entry == NULL) {
 577                 ret_val = -ENOENT;
 578                 goto listdef_failure_lock;
 579         }
 580         ret_val = netlbl_mgmt_listentry(ans_skb, entry);
 581         rcu_read_unlock();
 582         if (ret_val != 0)
 583                 goto listdef_failure;
 584
 585         genlmsg_end(ans_skb, data);
 586         return genlmsg_reply(ans_skb, info);
 587
 588 listdef_failure_lock:
 589         rcu_read_unlock();
 590 listdef_failure:
 591         kfree_skb(ans_skb);
 592         return ret_val;
 593 }
 594
 595 /**
 596  * netlbl_mgmt_protocols_cb - Write an individual PROTOCOL message response
 597  * @skb: the skb to write to
 598  * @cb: the NETLINK callback
 599  * @protocol: the NetLabel protocol to use in the message
 600  *
 601  * Description:
 602  * This function is to be used in conjunction with netlbl_mgmt_protocols() to
 603  * answer a application's PROTOCOLS message.  Returns the size of the message
 604  * on success, negative values on failure.
 605  *
 606  */
 607 static int netlbl_mgmt_protocols_cb(struct sk_buff *skb,
 608                                     struct netlink_callback *cb,
 609                                     u32 protocol)
 610 {
 611         int ret_val = -ENOMEM;
 612         void *data;
 613
 614         data = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
 615                            &netlbl_mgmt_gnl_family, NLM_F_MULTI,
 616                            NLBL_MGMT_C_PROTOCOLS);
 617         if (data == NULL)
 618                 goto protocols_cb_failure;
 619
 620         ret_val = nla_put_u32(skb, NLBL_MGMT_A_PROTOCOL, protocol);
 621         if (ret_val != 0)
 622                 goto protocols_cb_failure;
 623
 624         return genlmsg_end(skb, data);
 625
 626 protocols_cb_failure:
 627         genlmsg_cancel(skb, data);
 628         return ret_val;
 629 }
 630
 631 /**
 632  * netlbl_mgmt_protocols - Handle a PROTOCOLS message
 633  * @skb: the NETLINK buffer
 634  * @cb: the NETLINK callback
 635  *
 636  * Description:
 637  * Process a user generated PROTOCOLS message and respond accordingly.
 638  *
 639  */
 640 static int netlbl_mgmt_protocols(struct sk_buff *skb,
 641                                  struct netlink_callback *cb)
 642 {
 643         u32 protos_sent = cb->args[0];
 644
 645         if (protos_sent == 0) {
 646                 if (netlbl_mgmt_protocols_cb(skb,
 647                                              cb,
 648                                              NETLBL_NLTYPE_UNLABELED) < 0)
 649                         goto protocols_return;
 650                 protos_sent++;
 651         }
 652         if (protos_sent == 1) {
 653                 if (netlbl_mgmt_protocols_cb(skb,
 654                                              cb,
 655                                              NETLBL_NLTYPE_CIPSOV4) < 0)
 656                         goto protocols_return;
 657                 protos_sent++;
 658         }
 659
 660 protocols_return:
 661         cb->args[0] = protos_sent;
 662         return skb->len;
 663 }
 664
 665 /**
 666  * netlbl_mgmt_version - Handle a VERSION message
 667  * @skb: the NETLINK buffer
 668  * @info: the Generic NETLINK info block
 669  *
 670  * Description:
 671  * Process a user generated VERSION message and respond accordingly.  Returns
 672  * zero on success, negative values on failure.
 673  *
 674  */
 675 static int netlbl_mgmt_version(struct sk_buff *skb, struct genl_info *info)
 676 {
 677         int ret_val = -ENOMEM;
 678         struct sk_buff *ans_skb = NULL;
 679         void *data;
 680
 681         ans_skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 682         if (ans_skb == NULL)
 683                 return -ENOMEM;
 684         data = genlmsg_put_reply(ans_skb, info, &netlbl_mgmt_gnl_family,
 685                                  0, NLBL_MGMT_C_VERSION);
 686         if (data == NULL)
 687                 goto version_failure;
 688
 689         ret_val = nla_put_u32(ans_skb,
 690                               NLBL_MGMT_A_VERSION,
 691                               NETLBL_PROTO_VERSION);
 692         if (ret_val != 0)
 693                 goto version_failure;
 694
 695         genlmsg_end(ans_skb, data);
 696         return genlmsg_reply(ans_skb, info);
 697
 698 version_failure:
 699         kfree_skb(ans_skb);
 700         return ret_val;
 701 }
 702
 703
 704 /*
 705  * NetLabel Generic NETLINK Command Definitions
 706  */
 707
 708 static struct genl_ops netlbl_mgmt_genl_ops[] = {
 709         {
 710         .cmd = NLBL_MGMT_C_ADD,
 711         .flags = GENL_ADMIN_PERM,
 712         .policy = netlbl_mgmt_genl_policy,
 713         .doit = netlbl_mgmt_add,
 714         .dumpit = NULL,
 715         },
 716         {
 717         .cmd = NLBL_MGMT_C_REMOVE,
 718         .flags = GENL_ADMIN_PERM,
 719         .policy = netlbl_mgmt_genl_policy,
 720         .doit = netlbl_mgmt_remove,
 721         .dumpit = NULL,
 722         },
 723         {
 724         .cmd = NLBL_MGMT_C_LISTALL,
 725         .flags = 0,
 726         .policy = netlbl_mgmt_genl_policy,
 727         .doit = NULL,
 728         .dumpit = netlbl_mgmt_listall,
 729         },
 730         {
 731         .cmd = NLBL_MGMT_C_ADDDEF,
 732         .flags = GENL_ADMIN_PERM,
 733         .policy = netlbl_mgmt_genl_policy,
 734         .doit = netlbl_mgmt_adddef,
 735         .dumpit = NULL,
 736         },
 737         {
 738         .cmd = NLBL_MGMT_C_REMOVEDEF,
 739         .flags = GENL_ADMIN_PERM,
 740         .policy = netlbl_mgmt_genl_policy,
 741         .doit = netlbl_mgmt_removedef,
 742         .dumpit = NULL,
 743         },
 744         {
 745         .cmd = NLBL_MGMT_C_LISTDEF,
 746         .flags = 0,
 747         .policy = netlbl_mgmt_genl_policy,
 748         .doit = netlbl_mgmt_listdef,
 749         .dumpit = NULL,
 750         },
 751         {
 752         .cmd = NLBL_MGMT_C_PROTOCOLS,
 753         .flags = 0,
 754         .policy = netlbl_mgmt_genl_policy,
 755         .doit = NULL,
 756         .dumpit = netlbl_mgmt_protocols,
 757         },
 758         {
 759         .cmd = NLBL_MGMT_C_VERSION,
 760         .flags = 0,
 761         .policy = netlbl_mgmt_genl_policy,
 762         .doit = netlbl_mgmt_version,
 763         .dumpit = NULL,
 764         },
 765 };
 766
 767 /*
 768  * NetLabel Generic NETLINK Protocol Functions
 769  */
 770
 771 /**
 772  * netlbl_mgmt_genl_init - Register the NetLabel management component
 773  *
 774  * Description:
 775  * Register the NetLabel management component with the Generic NETLINK
 776  * mechanism.  Returns zero on success, negative values on failure.
 777  *
 778  */
 779 int __init netlbl_mgmt_genl_init(void)
 780 {
 781         return genl_register_family_with_ops(&netlbl_mgmt_gnl_family,
 782                 netlbl_mgmt_genl_ops, ARRAY_SIZE(netlbl_mgmt_genl_ops));
 783 }