net/netlink/genetlink.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * NETLINK      Generic Netlink Family
   4  *
   5  *              Authors:        Jamal Hadi Salim
   6  *                              Thomas Graf <[email protected]>
   7  *                              Johannes Berg <[email protected]>
   8  */
   9
  10 #include <linux/module.h>
  11 #include <linux/kernel.h>
  12 #include <linux/slab.h>
  13 #include <linux/errno.h>
  14 #include <linux/types.h>
  15 #include <linux/socket.h>
  16 #include <linux/string_helpers.h>
  17 #include <linux/skbuff.h>
  18 #include <linux/mutex.h>
  19 #include <linux/bitmap.h>
  20 #include <linux/rwsem.h>
  21 #include <linux/idr.h>
  22 #include <net/sock.h>
  23 #include <net/genetlink.h>
  24
  25 static DEFINE_MUTEX(genl_mutex); /* serialization of message processing */
  26 static DECLARE_RWSEM(cb_lock);
  27
  28 atomic_t genl_sk_destructing_cnt = ATOMIC_INIT(0);
  29 DECLARE_WAIT_QUEUE_HEAD(genl_sk_destructing_waitq);
  30
  31 void genl_lock(void)
  32 {
  33         mutex_lock(&genl_mutex);
  34 }
  35 EXPORT_SYMBOL(genl_lock);
  36
  37 void genl_unlock(void)
  38 {
  39         mutex_unlock(&genl_mutex);
  40 }
  41 EXPORT_SYMBOL(genl_unlock);
  42
  43 static void genl_lock_all(void)
  44 {
  45         down_write(&cb_lock);
  46         genl_lock();
  47 }
  48
  49 static void genl_unlock_all(void)
  50 {
  51         genl_unlock();
  52         up_write(&cb_lock);
  53 }
  54
  55 static DEFINE_IDR(genl_fam_idr);
  56
  57 /*
  58  * Bitmap of multicast groups that are currently in use.
  59  *
  60  * To avoid an allocation at boot of just one unsigned long,
  61  * declare it global instead.
  62  * Bit 0 is marked as already used since group 0 is invalid.
  63  * Bit 1 is marked as already used since the drop-monitor code
  64  * abuses the API and thinks it can statically use group 1.
  65  * That group will typically conflict with other groups that
  66  * any proper users use.
  67  * Bit 16 is marked as used since it's used for generic netlink
  68  * and the code no longer marks pre-reserved IDs as used.
  69  * Bit 17 is marked as already used since the VFS quota code
  70  * also abused this API and relied on family == group ID, we
  71  * cater to that by giving it a static family and group ID.
  72  * Bit 18 is marked as already used since the PMCRAID driver
  73  * did the same thing as the VFS quota code (maybe copied?)
  74  */
  75 static unsigned long mc_group_start = 0x3 | BIT(GENL_ID_CTRL) |
  76                                       BIT(GENL_ID_VFS_DQUOT) |
  77                                       BIT(GENL_ID_PMCRAID);
  78 static unsigned long *mc_groups = &mc_group_start;
  79 static unsigned long mc_groups_longs = 1;
  80
  81 /* We need the last attribute with non-zero ID therefore a 2-entry array */
  82 static struct nla_policy genl_policy_reject_all[] = {
  83         { .type = NLA_REJECT },
  84         { .type = NLA_REJECT },
  85 };
  86
  87 static int genl_ctrl_event(int event, const struct genl_family *family,
  88                            const struct genl_multicast_group *grp,
  89                            int grp_id);
  90
  91 static void
  92 genl_op_fill_in_reject_policy(const struct genl_family *family,
  93                               struct genl_ops *op)
  94 {
  95         BUILD_BUG_ON(ARRAY_SIZE(genl_policy_reject_all) - 1 != 1);
  96
  97         if (op->policy || op->cmd < family->resv_start_op)
  98                 return;
  99
 100         op->policy = genl_policy_reject_all;
 101         op->maxattr = 1;
 102 }
 103
 104 static void
 105 genl_op_fill_in_reject_policy_split(const struct genl_family *family,
 106                                     struct genl_split_ops *op)
 107 {
 108         if (op->policy)
 109                 return;
 110
 111         op->policy = genl_policy_reject_all;
 112         op->maxattr = 1;
 113 }
 114
 115 static const struct genl_family *genl_family_find_byid(unsigned int id)
 116 {
 117         return idr_find(&genl_fam_idr, id);
 118 }
 119
 120 static const struct genl_family *genl_family_find_byname(char *name)
 121 {
 122         const struct genl_family *family;
 123         unsigned int id;
 124
 125         idr_for_each_entry(&genl_fam_idr, family, id)
 126                 if (strcmp(family->name, name) == 0)
 127                         return family;
 128
 129         return NULL;
 130 }
 131
 132 struct genl_op_iter {
 133         const struct genl_family *family;
 134         struct genl_split_ops doit;
 135         struct genl_split_ops dumpit;
 136         int cmd_idx;
 137         int entry_idx;
 138         u32 cmd;
 139         u8 flags;
 140 };
 141
 142 static void genl_op_from_full(const struct genl_family *family,
 143                               unsigned int i, struct genl_ops *op)
 144 {
 145         *op = family->ops[i];
 146
 147         if (!op->maxattr)
 148                 op->maxattr = family->maxattr;
 149         if (!op->policy)
 150                 op->policy = family->policy;
 151
 152         genl_op_fill_in_reject_policy(family, op);
 153 }
 154
 155 static int genl_get_cmd_full(u32 cmd, const struct genl_family *family,
 156                              struct genl_ops *op)
 157 {
 158         int i;
 159
 160         for (i = 0; i < family->n_ops; i++)
 161                 if (family->ops[i].cmd == cmd) {
 162                         genl_op_from_full(family, i, op);
 163                         return 0;
 164                 }
 165
 166         return -ENOENT;
 167 }
 168
 169 static void genl_op_from_small(const struct genl_family *family,
 170                                unsigned int i, struct genl_ops *op)
 171 {
 172         memset(op, 0, sizeof(*op));
 173         op->doit        = family->small_ops[i].doit;
 174         op->dumpit      = family->small_ops[i].dumpit;
 175         op->cmd         = family->small_ops[i].cmd;
 176         op->internal_flags = family->small_ops[i].internal_flags;
 177         op->flags       = family->small_ops[i].flags;
 178         op->validate    = family->small_ops[i].validate;
 179
 180         op->maxattr = family->maxattr;
 181         op->policy = family->policy;
 182
 183         genl_op_fill_in_reject_policy(family, op);
 184 }
 185
 186 static int genl_get_cmd_small(u32 cmd, const struct genl_family *family,
 187                               struct genl_ops *op)
 188 {
 189         int i;
 190
 191         for (i = 0; i < family->n_small_ops; i++)
 192                 if (family->small_ops[i].cmd == cmd) {
 193                         genl_op_from_small(family, i, op);
 194                         return 0;
 195                 }
 196
 197         return -ENOENT;
 198 }
 199
 200 static void genl_op_from_split(struct genl_op_iter *iter)
 201 {
 202         const struct genl_family *family = iter->family;
 203         int i, cnt = 0;
 204
 205         i = iter->entry_idx - family->n_ops - family->n_small_ops;
 206
 207         if (family->split_ops[i + cnt].flags & GENL_CMD_CAP_DO) {
 208                 iter->doit = family->split_ops[i + cnt];
 209                 genl_op_fill_in_reject_policy_split(family, &iter->doit);
 210                 cnt++;
 211         } else {
 212                 memset(&iter->doit, 0, sizeof(iter->doit));
 213         }
 214
 215         if (i + cnt < family->n_split_ops &&
 216             family->split_ops[i + cnt].flags & GENL_CMD_CAP_DUMP) {
 217                 iter->dumpit = family->split_ops[i + cnt];
 218                 genl_op_fill_in_reject_policy_split(family, &iter->dumpit);
 219                 cnt++;
 220         } else {
 221                 memset(&iter->dumpit, 0, sizeof(iter->dumpit));
 222         }
 223
 224         WARN_ON(!cnt);
 225         iter->entry_idx += cnt;
 226 }
 227
 228 static int
 229 genl_get_cmd_split(u32 cmd, u8 flag, const struct genl_family *family,
 230                    struct genl_split_ops *op)
 231 {
 232         int i;
 233
 234         for (i = 0; i < family->n_split_ops; i++)
 235                 if (family->split_ops[i].cmd == cmd &&
 236                     family->split_ops[i].flags & flag) {
 237                         *op = family->split_ops[i];
 238                         return 0;
 239                 }
 240
 241         return -ENOENT;
 242 }
 243
 244 static int
 245 genl_cmd_full_to_split(struct genl_split_ops *op,
 246                        const struct genl_family *family,
 247                        const struct genl_ops *full, u8 flags)
 248 {
 249         if ((flags & GENL_CMD_CAP_DO && !full->doit) ||
 250             (flags & GENL_CMD_CAP_DUMP && !full->dumpit)) {
 251                 memset(op, 0, sizeof(*op));
 252                 return -ENOENT;
 253         }
 254
 255         if (flags & GENL_CMD_CAP_DUMP) {
 256                 op->start       = full->start;
 257                 op->dumpit      = full->dumpit;
 258                 op->done        = full->done;
 259         } else {
 260                 op->pre_doit    = family->pre_doit;
 261                 op->doit        = full->doit;
 262                 op->post_doit   = family->post_doit;
 263         }
 264
 265         if (flags & GENL_CMD_CAP_DUMP &&
 266             full->validate & GENL_DONT_VALIDATE_DUMP) {
 267                 op->policy      = NULL;
 268                 op->maxattr     = 0;
 269         } else {
 270                 op->policy      = full->policy;
 271                 op->maxattr     = full->maxattr;
 272         }
 273
 274         op->cmd                 = full->cmd;
 275         op->internal_flags      = full->internal_flags;
 276         op->flags               = full->flags;
 277         op->validate            = full->validate;
 278
 279         /* Make sure flags include the GENL_CMD_CAP_DO / GENL_CMD_CAP_DUMP */
 280         op->flags               |= flags;
 281
 282         return 0;
 283 }
 284
 285 /* Must make sure that op is initialized to 0 on failure */
 286 static int
 287 genl_get_cmd(u32 cmd, u8 flags, const struct genl_family *family,
 288              struct genl_split_ops *op)
 289 {
 290         struct genl_ops full;
 291         int err;
 292
 293         err = genl_get_cmd_full(cmd, family, &full);
 294         if (err == -ENOENT)
 295                 err = genl_get_cmd_small(cmd, family, &full);
 296         /* Found one of legacy forms */
 297         if (err == 0)
 298                 return genl_cmd_full_to_split(op, family, &full, flags);
 299
 300         err = genl_get_cmd_split(cmd, flags, family, op);
 301         if (err)
 302                 memset(op, 0, sizeof(*op));
 303         return err;
 304 }
 305
 306 /* For policy dumping only, get ops of both do and dump.
 307  * Fail if both are missing, genl_get_cmd() will zero-init in case of failure.
 308  */
 309 static int
 310 genl_get_cmd_both(u32 cmd, const struct genl_family *family,
 311                   struct genl_split_ops *doit, struct genl_split_ops *dumpit)
 312 {
 313         int err1, err2;
 314
 315         err1 = genl_get_cmd(cmd, GENL_CMD_CAP_DO, family, doit);
 316         err2 = genl_get_cmd(cmd, GENL_CMD_CAP_DUMP, family, dumpit);
 317
 318         return err1 && err2 ? -ENOENT : 0;
 319 }
 320
 321 static bool
 322 genl_op_iter_init(const struct genl_family *family, struct genl_op_iter *iter)
 323 {
 324         iter->family = family;
 325         iter->cmd_idx = 0;
 326         iter->entry_idx = 0;
 327
 328         iter->flags = 0;
 329
 330         return iter->family->n_ops +
 331                 iter->family->n_small_ops +
 332                 iter->family->n_split_ops;
 333 }
 334
 335 static bool genl_op_iter_next(struct genl_op_iter *iter)
 336 {
 337         const struct genl_family *family = iter->family;
 338         bool legacy_op = true;
 339         struct genl_ops op;
 340
 341         if (iter->entry_idx < family->n_ops) {
 342                 genl_op_from_full(family, iter->entry_idx, &op);
 343         } else if (iter->entry_idx < family->n_ops + family->n_small_ops) {
 344                 genl_op_from_small(family, iter->entry_idx - family->n_ops,
 345                                    &op);
 346         } else if (iter->entry_idx <
 347                    family->n_ops + family->n_small_ops + family->n_split_ops) {
 348                 legacy_op = false;
 349                 /* updates entry_idx */
 350                 genl_op_from_split(iter);
 351         } else {
 352                 return false;
 353         }
 354
 355         iter->cmd_idx++;
 356
 357         if (legacy_op) {
 358                 iter->entry_idx++;
 359
 360                 genl_cmd_full_to_split(&iter->doit, family,
 361                                        &op, GENL_CMD_CAP_DO);
 362                 genl_cmd_full_to_split(&iter->dumpit, family,
 363                                        &op, GENL_CMD_CAP_DUMP);
 364         }
 365
 366         iter->cmd = iter->doit.cmd | iter->dumpit.cmd;
 367         iter->flags = iter->doit.flags | iter->dumpit.flags;
 368
 369         return true;
 370 }
 371
 372 static void
 373 genl_op_iter_copy(struct genl_op_iter *dst, struct genl_op_iter *src)
 374 {
 375         *dst = *src;
 376 }
 377
 378 static unsigned int genl_op_iter_idx(struct genl_op_iter *iter)
 379 {
 380         return iter->cmd_idx;
 381 }
 382
 383 static int genl_allocate_reserve_groups(int n_groups, int *first_id)
 384 {
 385         unsigned long *new_groups;
 386         int start = 0;
 387         int i;
 388         int id;
 389         bool fits;
 390
 391         do {
 392                 if (start == 0)
 393                         id = find_first_zero_bit(mc_groups,
 394                                                  mc_groups_longs *
 395                                                  BITS_PER_LONG);
 396                 else
 397                         id = find_next_zero_bit(mc_groups,
 398                                                 mc_groups_longs * BITS_PER_LONG,
 399                                                 start);
 400
 401                 fits = true;
 402                 for (i = id;
 403                      i < min_t(int, id + n_groups,
 404                                mc_groups_longs * BITS_PER_LONG);
 405                      i++) {
 406                         if (test_bit(i, mc_groups)) {
 407                                 start = i;
 408                                 fits = false;
 409                                 break;
 410                         }
 411                 }
 412
 413                 if (id + n_groups > mc_groups_longs * BITS_PER_LONG) {
 414                         unsigned long new_longs = mc_groups_longs +
 415                                                   BITS_TO_LONGS(n_groups);
 416                         size_t nlen = new_longs * sizeof(unsigned long);
 417
 418                         if (mc_groups == &mc_group_start) {
 419                                 new_groups = kzalloc(nlen, GFP_KERNEL);
 420                                 if (!new_groups)
 421                                         return -ENOMEM;
 422                                 mc_groups = new_groups;
 423                                 *mc_groups = mc_group_start;
 424                         } else {
 425                                 new_groups = krealloc(mc_groups, nlen,
 426                                                       GFP_KERNEL);
 427                                 if (!new_groups)
 428                                         return -ENOMEM;
 429                                 mc_groups = new_groups;
 430                                 for (i = 0; i < BITS_TO_LONGS(n_groups); i++)
 431                                         mc_groups[mc_groups_longs + i] = 0;
 432                         }
 433                         mc_groups_longs = new_longs;
 434                 }
 435         } while (!fits);
 436
 437         for (i = id; i < id + n_groups; i++)
 438                 set_bit(i, mc_groups);
 439         *first_id = id;
 440         return 0;
 441 }
 442
 443 static struct genl_family genl_ctrl;
 444
 445 static int genl_validate_assign_mc_groups(struct genl_family *family)
 446 {
 447         int first_id;
 448         int n_groups = family->n_mcgrps;
 449         int err = 0, i;
 450         bool groups_allocated = false;
 451
 452         if (!n_groups)
 453                 return 0;
 454
 455         for (i = 0; i < n_groups; i++) {
 456                 const struct genl_multicast_group *grp = &family->mcgrps[i];
 457
 458                 if (WARN_ON(grp->name[0] == '\0'))
 459                         return -EINVAL;
 460                 if (WARN_ON(!string_is_terminated(grp->name, GENL_NAMSIZ)))
 461                         return -EINVAL;
 462         }
 463
 464         /* special-case our own group and hacks */
 465         if (family == &genl_ctrl) {
 466                 first_id = GENL_ID_CTRL;
 467                 BUG_ON(n_groups != 1);
 468         } else if (strcmp(family->name, "NET_DM") == 0) {
 469                 first_id = 1;
 470                 BUG_ON(n_groups != 1);
 471         } else if (family->id == GENL_ID_VFS_DQUOT) {
 472                 first_id = GENL_ID_VFS_DQUOT;
 473                 BUG_ON(n_groups != 1);
 474         } else if (family->id == GENL_ID_PMCRAID) {
 475                 first_id = GENL_ID_PMCRAID;
 476                 BUG_ON(n_groups != 1);
 477         } else {
 478                 groups_allocated = true;
 479                 err = genl_allocate_reserve_groups(n_groups, &first_id);
 480                 if (err)
 481                         return err;
 482         }
 483
 484         family->mcgrp_offset = first_id;
 485
 486         /* if still initializing, can't and don't need to realloc bitmaps */
 487         if (!init_net.genl_sock)
 488                 return 0;
 489
 490         if (family->netnsok) {
 491                 struct net *net;
 492
 493                 netlink_table_grab();
 494                 rcu_read_lock();
 495                 for_each_net_rcu(net) {
 496                         err = __netlink_change_ngroups(net->genl_sock,
 497                                         mc_groups_longs * BITS_PER_LONG);
 498                         if (err) {
 499                                 /*
 500                                  * No need to roll back, can only fail if
 501                                  * memory allocation fails and then the
 502                                  * number of _possible_ groups has been
 503                                  * increased on some sockets which is ok.
 504                                  */
 505                                 break;
 506                         }
 507                 }
 508                 rcu_read_unlock();
 509                 netlink_table_ungrab();
 510         } else {
 511                 err = netlink_change_ngroups(init_net.genl_sock,
 512                                              mc_groups_longs * BITS_PER_LONG);
 513         }
 514
 515         if (groups_allocated && err) {
 516                 for (i = 0; i < family->n_mcgrps; i++)
 517                         clear_bit(family->mcgrp_offset + i, mc_groups);
 518         }
 519
 520         return err;
 521 }
 522
 523 static void genl_unregister_mc_groups(const struct genl_family *family)
 524 {
 525         struct net *net;
 526         int i;
 527
 528         netlink_table_grab();
 529         rcu_read_lock();
 530         for_each_net_rcu(net) {
 531                 for (i = 0; i < family->n_mcgrps; i++)
 532                         __netlink_clear_multicast_users(
 533                                 net->genl_sock, family->mcgrp_offset + i);
 534         }
 535         rcu_read_unlock();
 536         netlink_table_ungrab();
 537
 538         for (i = 0; i < family->n_mcgrps; i++) {
 539                 int grp_id = family->mcgrp_offset + i;
 540
 541                 if (grp_id != 1)
 542                         clear_bit(grp_id, mc_groups);
 543                 genl_ctrl_event(CTRL_CMD_DELMCAST_GRP, family,
 544                                 &family->mcgrps[i], grp_id);
 545         }
 546 }
 547
 548 static bool genl_split_op_check(const struct genl_split_ops *op)
 549 {
 550         if (WARN_ON(hweight8(op->flags & (GENL_CMD_CAP_DO |
 551                                           GENL_CMD_CAP_DUMP)) != 1))
 552                 return true;
 553         return false;
 554 }
 555
 556 static int genl_validate_ops(const struct genl_family *family)
 557 {
 558         struct genl_op_iter i, j;
 559         unsigned int s;
 560
 561         if (WARN_ON(family->n_ops && !family->ops) ||
 562             WARN_ON(family->n_small_ops && !family->small_ops) ||
 563             WARN_ON(family->n_split_ops && !family->split_ops))
 564                 return -EINVAL;
 565
 566         for (genl_op_iter_init(family, &i); genl_op_iter_next(&i); ) {
 567                 if (!(i.flags & (GENL_CMD_CAP_DO | GENL_CMD_CAP_DUMP)))
 568                         return -EINVAL;
 569
 570                 if (WARN_ON(i.cmd >= family->resv_start_op &&
 571                             (i.doit.validate || i.dumpit.validate)))
 572                         return -EINVAL;
 573
 574                 genl_op_iter_copy(&j, &i);
 575                 while (genl_op_iter_next(&j)) {
 576                         if (i.cmd == j.cmd)
 577                                 return -EINVAL;
 578                 }
 579         }
 580
 581         if (family->n_split_ops) {
 582                 if (genl_split_op_check(&family->split_ops[0]))
 583                         return -EINVAL;
 584         }
 585
 586         for (s = 1; s < family->n_split_ops; s++) {
 587                 const struct genl_split_ops *a, *b;
 588
 589                 a = &family->split_ops[s - 1];
 590                 b = &family->split_ops[s];
 591
 592                 if (genl_split_op_check(b))
 593                         return -EINVAL;
 594
 595                 /* Check sort order */
 596                 if (a->cmd < b->cmd)
 597                         continue;
 598
 599                 if (a->internal_flags != b->internal_flags ||
 600                     ((a->flags ^ b->flags) & ~(GENL_CMD_CAP_DO |
 601                                                GENL_CMD_CAP_DUMP))) {
 602                         WARN_ON(1);
 603                         return -EINVAL;
 604                 }
 605
 606                 if ((a->flags & GENL_CMD_CAP_DO) &&
 607                     (b->flags & GENL_CMD_CAP_DUMP))
 608                         continue;
 609
 610                 WARN_ON(1);
 611                 return -EINVAL;
 612         }
 613
 614         return 0;
 615 }
 616
 617 /**
 618  * genl_register_family - register a generic netlink family
 619  * @family: generic netlink family
 620  *
 621  * Registers the specified family after validating it first. Only one
 622  * family may be registered with the same family name or identifier.
 623  *
 624  * The family's ops, multicast groups and module pointer must already
 625  * be assigned.
 626  *
 627  * Return 0 on success or a negative error code.
 628  */
 629 int genl_register_family(struct genl_family *family)
 630 {
 631         int err, i;
 632         int start = GENL_START_ALLOC, end = GENL_MAX_ID;
 633
 634         err = genl_validate_ops(family);
 635         if (err)
 636                 return err;
 637
 638         genl_lock_all();
 639
 640         if (genl_family_find_byname(family->name)) {
 641                 err = -EEXIST;
 642                 goto errout_locked;
 643         }
 644
 645         /*
 646          * Sadly, a few cases need to be special-cased
 647          * due to them having previously abused the API
 648          * and having used their family ID also as their
 649          * multicast group ID, so we use reserved IDs
 650          * for both to be sure we can do that mapping.
 651          */
 652         if (family == &genl_ctrl) {
 653                 /* and this needs to be special for initial family lookups */
 654                 start = end = GENL_ID_CTRL;
 655         } else if (strcmp(family->name, "pmcraid") == 0) {
 656                 start = end = GENL_ID_PMCRAID;
 657         } else if (strcmp(family->name, "VFS_DQUOT") == 0) {
 658                 start = end = GENL_ID_VFS_DQUOT;
 659         }
 660
 661         family->id = idr_alloc_cyclic(&genl_fam_idr, family,
 662                                       start, end + 1, GFP_KERNEL);
 663         if (family->id < 0) {
 664                 err = family->id;
 665                 goto errout_locked;
 666         }
 667
 668         err = genl_validate_assign_mc_groups(family);
 669         if (err)
 670                 goto errout_remove;
 671
 672         genl_unlock_all();
 673
 674         /* send all events */
 675         genl_ctrl_event(CTRL_CMD_NEWFAMILY, family, NULL, 0);
 676         for (i = 0; i < family->n_mcgrps; i++)
 677                 genl_ctrl_event(CTRL_CMD_NEWMCAST_GRP, family,
 678                                 &family->mcgrps[i], family->mcgrp_offset + i);
 679
 680         return 0;
 681
 682 errout_remove:
 683         idr_remove(&genl_fam_idr, family->id);
 684 errout_locked:
 685         genl_unlock_all();
 686         return err;
 687 }
 688 EXPORT_SYMBOL(genl_register_family);
 689
 690 /**
 691  * genl_unregister_family - unregister generic netlink family
 692  * @family: generic netlink family
 693  *
 694  * Unregisters the specified family.
 695  *
 696  * Returns 0 on success or a negative error code.
 697  */
 698 int genl_unregister_family(const struct genl_family *family)
 699 {
 700         genl_lock_all();
 701
 702         if (!genl_family_find_byid(family->id)) {
 703                 genl_unlock_all();
 704                 return -ENOENT;
 705         }
 706
 707         genl_unregister_mc_groups(family);
 708
 709         idr_remove(&genl_fam_idr, family->id);
 710
 711         up_write(&cb_lock);
 712         wait_event(genl_sk_destructing_waitq,
 713                    atomic_read(&genl_sk_destructing_cnt) == 0);
 714         genl_unlock();
 715
 716         genl_ctrl_event(CTRL_CMD_DELFAMILY, family, NULL, 0);
 717
 718         return 0;
 719 }
 720 EXPORT_SYMBOL(genl_unregister_family);
 721
 722 /**
 723  * genlmsg_put - Add generic netlink header to netlink message
 724  * @skb: socket buffer holding the message
 725  * @portid: netlink portid the message is addressed to
 726  * @seq: sequence number (usually the one of the sender)
 727  * @family: generic netlink family
 728  * @flags: netlink message flags
 729  * @cmd: generic netlink command
 730  *
 731  * Returns pointer to user specific header
 732  */
 733 void *genlmsg_put(struct sk_buff *skb, u32 portid, u32 seq,
 734                   const struct genl_family *family, int flags, u8 cmd)
 735 {
 736         struct nlmsghdr *nlh;
 737         struct genlmsghdr *hdr;
 738
 739         nlh = nlmsg_put(skb, portid, seq, family->id, GENL_HDRLEN +
 740                         family->hdrsize, flags);
 741         if (nlh == NULL)
 742                 return NULL;
 743
 744         hdr = nlmsg_data(nlh);
 745         hdr->cmd = cmd;
 746         hdr->version = family->version;
 747         hdr->reserved = 0;
 748
 749         return (char *) hdr + GENL_HDRLEN;
 750 }
 751 EXPORT_SYMBOL(genlmsg_put);
 752
 753 static struct genl_dumpit_info *genl_dumpit_info_alloc(void)
 754 {
 755         return kmalloc(sizeof(struct genl_dumpit_info), GFP_KERNEL);
 756 }
 757
 758 static void genl_dumpit_info_free(const struct genl_dumpit_info *info)
 759 {
 760         kfree(info);
 761 }
 762
 763 static struct nlattr **
 764 genl_family_rcv_msg_attrs_parse(const struct genl_family *family,
 765                                 struct nlmsghdr *nlh,
 766                                 struct netlink_ext_ack *extack,
 767                                 const struct genl_split_ops *ops,
 768                                 int hdrlen,
 769                                 enum genl_validate_flags no_strict_flag)
 770 {
 771         enum netlink_validation validate = ops->validate & no_strict_flag ?
 772                                            NL_VALIDATE_LIBERAL :
 773                                            NL_VALIDATE_STRICT;
 774         struct nlattr **attrbuf;
 775         int err;
 776
 777         if (!ops->maxattr)
 778                 return NULL;
 779
 780         attrbuf = kmalloc_array(ops->maxattr + 1,
 781                                 sizeof(struct nlattr *), GFP_KERNEL);
 782         if (!attrbuf)
 783                 return ERR_PTR(-ENOMEM);
 784
 785         err = __nlmsg_parse(nlh, hdrlen, attrbuf, ops->maxattr, ops->policy,
 786                             validate, extack);
 787         if (err) {
 788                 kfree(attrbuf);
 789                 return ERR_PTR(err);
 790         }
 791         return attrbuf;
 792 }
 793
 794 static void genl_family_rcv_msg_attrs_free(struct nlattr **attrbuf)
 795 {
 796         kfree(attrbuf);
 797 }
 798
 799 struct genl_start_context {
 800         const struct genl_family *family;
 801         struct nlmsghdr *nlh;
 802         struct netlink_ext_ack *extack;
 803         const struct genl_split_ops *ops;
 804         int hdrlen;
 805 };
 806
 807 static int genl_start(struct netlink_callback *cb)
 808 {
 809         struct genl_start_context *ctx = cb->data;
 810         const struct genl_split_ops *ops;
 811         struct genl_dumpit_info *info;
 812         struct nlattr **attrs = NULL;
 813         int rc = 0;
 814
 815         ops = ctx->ops;
 816         if (!(ops->validate & GENL_DONT_VALIDATE_DUMP) &&
 817             ctx->nlh->nlmsg_len < nlmsg_msg_size(ctx->hdrlen))
 818                 return -EINVAL;
 819
 820         attrs = genl_family_rcv_msg_attrs_parse(ctx->family, ctx->nlh, ctx->extack,
 821                                                 ops, ctx->hdrlen,
 822                                                 GENL_DONT_VALIDATE_DUMP_STRICT);
 823         if (IS_ERR(attrs))
 824                 return PTR_ERR(attrs);
 825
 826         info = genl_dumpit_info_alloc();
 827         if (!info) {
 828                 genl_family_rcv_msg_attrs_free(attrs);
 829                 return -ENOMEM;
 830         }
 831         info->family = ctx->family;
 832         info->op = *ops;
 833         info->attrs = attrs;
 834
 835         cb->data = info;
 836         if (ops->start) {
 837                 if (!ctx->family->parallel_ops)
 838                         genl_lock();
 839                 rc = ops->start(cb);
 840                 if (!ctx->family->parallel_ops)
 841                         genl_unlock();
 842         }
 843
 844         if (rc) {
 845                 genl_family_rcv_msg_attrs_free(info->attrs);
 846                 genl_dumpit_info_free(info);
 847                 cb->data = NULL;
 848         }
 849         return rc;
 850 }
 851
 852 static int genl_lock_dumpit(struct sk_buff *skb, struct netlink_callback *cb)
 853 {
 854         const struct genl_split_ops *ops = &genl_dumpit_info(cb)->op;
 855         int rc;
 856
 857         genl_lock();
 858         rc = ops->dumpit(skb, cb);
 859         genl_unlock();
 860         return rc;
 861 }
 862
 863 static int genl_lock_done(struct netlink_callback *cb)
 864 {
 865         const struct genl_dumpit_info *info = genl_dumpit_info(cb);
 866         const struct genl_split_ops *ops = &info->op;
 867         int rc = 0;
 868
 869         if (ops->done) {
 870                 genl_lock();
 871                 rc = ops->done(cb);
 872                 genl_unlock();
 873         }
 874         genl_family_rcv_msg_attrs_free(info->attrs);
 875         genl_dumpit_info_free(info);
 876         return rc;
 877 }
 878
 879 static int genl_parallel_done(struct netlink_callback *cb)
 880 {
 881         const struct genl_dumpit_info *info = genl_dumpit_info(cb);
 882         const struct genl_split_ops *ops = &info->op;
 883         int rc = 0;
 884
 885         if (ops->done)
 886                 rc = ops->done(cb);
 887         genl_family_rcv_msg_attrs_free(info->attrs);
 888         genl_dumpit_info_free(info);
 889         return rc;
 890 }
 891
 892 static int genl_family_rcv_msg_dumpit(const struct genl_family *family,
 893                                       struct sk_buff *skb,
 894                                       struct nlmsghdr *nlh,
 895                                       struct netlink_ext_ack *extack,
 896                                       const struct genl_split_ops *ops,
 897                                       int hdrlen, struct net *net)
 898 {
 899         struct genl_start_context ctx;
 900         int err;
 901
 902         ctx.family = family;
 903         ctx.nlh = nlh;
 904         ctx.extack = extack;
 905         ctx.ops = ops;
 906         ctx.hdrlen = hdrlen;
 907
 908         if (!family->parallel_ops) {
 909                 struct netlink_dump_control c = {
 910                         .module = family->module,
 911                         .data = &ctx,
 912                         .start = genl_start,
 913                         .dump = genl_lock_dumpit,
 914                         .done = genl_lock_done,
 915                 };
 916
 917                 genl_unlock();
 918                 err = __netlink_dump_start(net->genl_sock, skb, nlh, &c);
 919                 genl_lock();
 920         } else {
 921                 struct netlink_dump_control c = {
 922                         .module = family->module,
 923                         .data = &ctx,
 924                         .start = genl_start,
 925                         .dump = ops->dumpit,
 926                         .done = genl_parallel_done,
 927                 };
 928
 929                 err = __netlink_dump_start(net->genl_sock, skb, nlh, &c);
 930         }
 931
 932         return err;
 933 }
 934
 935 static int genl_family_rcv_msg_doit(const struct genl_family *family,
 936                                     struct sk_buff *skb,
 937                                     struct nlmsghdr *nlh,
 938                                     struct netlink_ext_ack *extack,
 939                                     const struct genl_split_ops *ops,
 940                                     int hdrlen, struct net *net)
 941 {
 942         struct nlattr **attrbuf;
 943         struct genl_info info;
 944         int err;
 945
 946         attrbuf = genl_family_rcv_msg_attrs_parse(family, nlh, extack,
 947                                                   ops, hdrlen,
 948                                                   GENL_DONT_VALIDATE_STRICT);
 949         if (IS_ERR(attrbuf))
 950                 return PTR_ERR(attrbuf);
 951
 952         info.snd_seq = nlh->nlmsg_seq;
 953         info.snd_portid = NETLINK_CB(skb).portid;
 954         info.nlhdr = nlh;
 955         info.genlhdr = nlmsg_data(nlh);
 956         info.userhdr = nlmsg_data(nlh) + GENL_HDRLEN;
 957         info.attrs = attrbuf;
 958         info.extack = extack;
 959         genl_info_net_set(&info, net);
 960         memset(&info.user_ptr, 0, sizeof(info.user_ptr));
 961
 962         if (ops->pre_doit) {
 963                 err = ops->pre_doit(ops, skb, &info);
 964                 if (err)
 965                         goto out;
 966         }
 967
 968         err = ops->doit(skb, &info);
 969
 970         if (ops->post_doit)
 971                 ops->post_doit(ops, skb, &info);
 972
 973 out:
 974         genl_family_rcv_msg_attrs_free(attrbuf);
 975
 976         return err;
 977 }
 978
 979 static int genl_header_check(const struct genl_family *family,
 980                              struct nlmsghdr *nlh, struct genlmsghdr *hdr,
 981                              struct netlink_ext_ack *extack)
 982 {
 983         u16 flags;
 984
 985         /* Only for commands added after we started validating */
 986         if (hdr->cmd < family->resv_start_op)
 987                 return 0;
 988
 989         if (hdr->reserved) {
 990                 NL_SET_ERR_MSG(extack, "genlmsghdr.reserved field is not 0");
 991                 return -EINVAL;
 992         }
 993
 994         /* Old netlink flags have pretty loose semantics, allow only the flags
 995          * consumed by the core where we can enforce the meaning.
 996          */
 997         flags = nlh->nlmsg_flags;
 998         if ((flags & NLM_F_DUMP) == NLM_F_DUMP) /* DUMP is 2 bits */
 999                 flags &= ~NLM_F_DUMP;
1000         if (flags & ~(NLM_F_REQUEST | NLM_F_ACK | NLM_F_ECHO)) {
1001                 NL_SET_ERR_MSG(extack,
1002                                "ambiguous or reserved bits set in nlmsg_flags");
1003                 return -EINVAL;
1004         }
1005
1006         return 0;
1007 }
1008
1009 static int genl_family_rcv_msg(const struct genl_family *family,
1010                                struct sk_buff *skb,
1011                                struct nlmsghdr *nlh,
1012                                struct netlink_ext_ack *extack)
1013 {
1014         struct net *net = sock_net(skb->sk);
1015         struct genlmsghdr *hdr = nlmsg_data(nlh);
1016         struct genl_split_ops op;
1017         int hdrlen;
1018         u8 flags;
1019
1020         /* this family doesn't exist in this netns */
1021         if (!family->netnsok && !net_eq(net, &init_net))
1022                 return -ENOENT;
1023
1024         hdrlen = GENL_HDRLEN + family->hdrsize;
1025         if (nlh->nlmsg_len < nlmsg_msg_size(hdrlen))
1026                 return -EINVAL;
1027
1028         if (genl_header_check(family, nlh, hdr, extack))
1029                 return -EINVAL;
1030
1031         flags = (nlh->nlmsg_flags & NLM_F_DUMP) == NLM_F_DUMP ?
1032                 GENL_CMD_CAP_DUMP : GENL_CMD_CAP_DO;
1033         if (genl_get_cmd(hdr->cmd, flags, family, &op))
1034                 return -EOPNOTSUPP;
1035
1036         if ((op.flags & GENL_ADMIN_PERM) &&
1037             !netlink_capable(skb, CAP_NET_ADMIN))
1038                 return -EPERM;
1039
1040         if ((op.flags & GENL_UNS_ADMIN_PERM) &&
1041             !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
1042                 return -EPERM;
1043
1044         if (flags & GENL_CMD_CAP_DUMP)
1045                 return genl_family_rcv_msg_dumpit(family, skb, nlh, extack,
1046                                                   &op, hdrlen, net);
1047         else
1048                 return genl_family_rcv_msg_doit(family, skb, nlh, extack,
1049                                                 &op, hdrlen, net);
1050 }
1051
1052 static int genl_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh,
1053                         struct netlink_ext_ack *extack)
1054 {
1055         const struct genl_family *family;
1056         int err;
1057
1058         family = genl_family_find_byid(nlh->nlmsg_type);
1059         if (family == NULL)
1060                 return -ENOENT;
1061
1062         if (!family->parallel_ops)
1063                 genl_lock();
1064
1065         err = genl_family_rcv_msg(family, skb, nlh, extack);
1066
1067         if (!family->parallel_ops)
1068                 genl_unlock();
1069
1070         return err;
1071 }
1072
1073 static void genl_rcv(struct sk_buff *skb)
1074 {
1075         down_read(&cb_lock);
1076         netlink_rcv_skb(skb, &genl_rcv_msg);
1077         up_read(&cb_lock);
1078 }
1079
1080 /**************************************************************************
1081  * Controller
1082  **************************************************************************/
1083
1084 static struct genl_family genl_ctrl;
1085
1086 static int ctrl_fill_info(const struct genl_family *family, u32 portid, u32 seq,
1087                           u32 flags, struct sk_buff *skb, u8 cmd)
1088 {
1089         struct genl_op_iter i;
1090         void *hdr;
1091
1092         hdr = genlmsg_put(skb, portid, seq, &genl_ctrl, flags, cmd);
1093         if (hdr == NULL)
1094                 return -1;
1095
1096         if (nla_put_string(skb, CTRL_ATTR_FAMILY_NAME, family->name) ||
1097             nla_put_u16(skb, CTRL_ATTR_FAMILY_ID, family->id) ||
1098             nla_put_u32(skb, CTRL_ATTR_VERSION, family->version) ||
1099             nla_put_u32(skb, CTRL_ATTR_HDRSIZE, family->hdrsize) ||
1100             nla_put_u32(skb, CTRL_ATTR_MAXATTR, family->maxattr))
1101                 goto nla_put_failure;
1102
1103         if (genl_op_iter_init(family, &i)) {
1104                 struct nlattr *nla_ops;
1105
1106                 nla_ops = nla_nest_start_noflag(skb, CTRL_ATTR_OPS);
1107                 if (nla_ops == NULL)
1108                         goto nla_put_failure;
1109
1110                 while (genl_op_iter_next(&i)) {
1111                         struct nlattr *nest;
1112                         u32 op_flags;
1113
1114                         op_flags = i.flags;
1115                         if (i.doit.policy || i.dumpit.policy)
1116                                 op_flags |= GENL_CMD_CAP_HASPOL;
1117
1118                         nest = nla_nest_start_noflag(skb, genl_op_iter_idx(&i));
1119                         if (nest == NULL)
1120                                 goto nla_put_failure;
1121
1122                         if (nla_put_u32(skb, CTRL_ATTR_OP_ID, i.cmd) ||
1123                             nla_put_u32(skb, CTRL_ATTR_OP_FLAGS, op_flags))
1124                                 goto nla_put_failure;
1125
1126                         nla_nest_end(skb, nest);
1127                 }
1128
1129                 nla_nest_end(skb, nla_ops);
1130         }
1131
1132         if (family->n_mcgrps) {
1133                 struct nlattr *nla_grps;
1134                 int i;
1135
1136                 nla_grps = nla_nest_start_noflag(skb, CTRL_ATTR_MCAST_GROUPS);
1137                 if (nla_grps == NULL)
1138                         goto nla_put_failure;
1139
1140                 for (i = 0; i < family->n_mcgrps; i++) {
1141                         struct nlattr *nest;
1142                         const struct genl_multicast_group *grp;
1143
1144                         grp = &family->mcgrps[i];
1145
1146                         nest = nla_nest_start_noflag(skb, i + 1);
1147                         if (nest == NULL)
1148                                 goto nla_put_failure;
1149
1150                         if (nla_put_u32(skb, CTRL_ATTR_MCAST_GRP_ID,
1151                                         family->mcgrp_offset + i) ||
1152                             nla_put_string(skb, CTRL_ATTR_MCAST_GRP_NAME,
1153                                            grp->name))
1154                                 goto nla_put_failure;
1155
1156                         nla_nest_end(skb, nest);
1157                 }
1158                 nla_nest_end(skb, nla_grps);
1159         }
1160
1161         genlmsg_end(skb, hdr);
1162         return 0;
1163
1164 nla_put_failure:
1165         genlmsg_cancel(skb, hdr);
1166         return -EMSGSIZE;
1167 }
1168
1169 static int ctrl_fill_mcgrp_info(const struct genl_family *family,
1170                                 const struct genl_multicast_group *grp,
1171                                 int grp_id, u32 portid, u32 seq, u32 flags,
1172                                 struct sk_buff *skb, u8 cmd)
1173 {
1174         void *hdr;
1175         struct nlattr *nla_grps;
1176         struct nlattr *nest;
1177
1178         hdr = genlmsg_put(skb, portid, seq, &genl_ctrl, flags, cmd);
1179         if (hdr == NULL)
1180                 return -1;
1181
1182         if (nla_put_string(skb, CTRL_ATTR_FAMILY_NAME, family->name) ||
1183             nla_put_u16(skb, CTRL_ATTR_FAMILY_ID, family->id))
1184                 goto nla_put_failure;
1185
1186         nla_grps = nla_nest_start_noflag(skb, CTRL_ATTR_MCAST_GROUPS);
1187         if (nla_grps == NULL)
1188                 goto nla_put_failure;
1189
1190         nest = nla_nest_start_noflag(skb, 1);
1191         if (nest == NULL)
1192                 goto nla_put_failure;
1193
1194         if (nla_put_u32(skb, CTRL_ATTR_MCAST_GRP_ID, grp_id) ||
1195             nla_put_string(skb, CTRL_ATTR_MCAST_GRP_NAME,
1196                            grp->name))
1197                 goto nla_put_failure;
1198
1199         nla_nest_end(skb, nest);
1200         nla_nest_end(skb, nla_grps);
1201
1202         genlmsg_end(skb, hdr);
1203         return 0;
1204
1205 nla_put_failure:
1206         genlmsg_cancel(skb, hdr);
1207         return -EMSGSIZE;
1208 }
1209
1210 static int ctrl_dumpfamily(struct sk_buff *skb, struct netlink_callback *cb)
1211 {
1212         int n = 0;
1213         struct genl_family *rt;
1214         struct net *net = sock_net(skb->sk);
1215         int fams_to_skip = cb->args[0];
1216         unsigned int id;
1217
1218         idr_for_each_entry(&genl_fam_idr, rt, id) {
1219                 if (!rt->netnsok && !net_eq(net, &init_net))
1220                         continue;
1221
1222                 if (n++ < fams_to_skip)
1223                         continue;
1224
1225                 if (ctrl_fill_info(rt, NETLINK_CB(cb->skb).portid,
1226                                    cb->nlh->nlmsg_seq, NLM_F_MULTI,
1227                                    skb, CTRL_CMD_NEWFAMILY) < 0) {
1228                         n--;
1229                         break;
1230                 }
1231         }
1232
1233         cb->args[0] = n;
1234         return skb->len;
1235 }
1236
1237 static struct sk_buff *ctrl_build_family_msg(const struct genl_family *family,
1238                                              u32 portid, int seq, u8 cmd)
1239 {
1240         struct sk_buff *skb;
1241         int err;
1242
1243         skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1244         if (skb == NULL)
1245                 return ERR_PTR(-ENOBUFS);
1246
1247         err = ctrl_fill_info(family, portid, seq, 0, skb, cmd);
1248         if (err < 0) {
1249                 nlmsg_free(skb);
1250                 return ERR_PTR(err);
1251         }
1252
1253         return skb;
1254 }
1255
1256 static struct sk_buff *
1257 ctrl_build_mcgrp_msg(const struct genl_family *family,
1258                      const struct genl_multicast_group *grp,
1259                      int grp_id, u32 portid, int seq, u8 cmd)
1260 {
1261         struct sk_buff *skb;
1262         int err;
1263
1264         skb = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
1265         if (skb == NULL)
1266                 return ERR_PTR(-ENOBUFS);
1267
1268         err = ctrl_fill_mcgrp_info(family, grp, grp_id, portid,
1269                                    seq, 0, skb, cmd);
1270         if (err < 0) {
1271                 nlmsg_free(skb);
1272                 return ERR_PTR(err);
1273         }
1274
1275         return skb;
1276 }
1277
1278 static const struct nla_policy ctrl_policy_family[] = {
1279         [CTRL_ATTR_FAMILY_ID]   = { .type = NLA_U16 },
1280         [CTRL_ATTR_FAMILY_NAME] = { .type = NLA_NUL_STRING,
1281                                     .len = GENL_NAMSIZ - 1 },
1282 };
1283
1284 static int ctrl_getfamily(struct sk_buff *skb, struct genl_info *info)
1285 {
1286         struct sk_buff *msg;
1287         const struct genl_family *res = NULL;
1288         int err = -EINVAL;
1289
1290         if (info->attrs[CTRL_ATTR_FAMILY_ID]) {
1291                 u16 id = nla_get_u16(info->attrs[CTRL_ATTR_FAMILY_ID]);
1292                 res = genl_family_find_byid(id);
1293                 err = -ENOENT;
1294         }
1295
1296         if (info->attrs[CTRL_ATTR_FAMILY_NAME]) {
1297                 char *name;
1298
1299                 name = nla_data(info->attrs[CTRL_ATTR_FAMILY_NAME]);
1300                 res = genl_family_find_byname(name);
1301 #ifdef CONFIG_MODULES
1302                 if (res == NULL) {
1303                         genl_unlock();
1304                         up_read(&cb_lock);
1305                         request_module("net-pf-%d-proto-%d-family-%s",
1306                                        PF_NETLINK, NETLINK_GENERIC, name);
1307                         down_read(&cb_lock);
1308                         genl_lock();
1309                         res = genl_family_find_byname(name);
1310                 }
1311 #endif
1312                 err = -ENOENT;
1313         }
1314
1315         if (res == NULL)
1316                 return err;
1317
1318         if (!res->netnsok && !net_eq(genl_info_net(info), &init_net)) {
1319                 /* family doesn't exist here */
1320                 return -ENOENT;
1321         }
1322
1323         msg = ctrl_build_family_msg(res, info->snd_portid, info->snd_seq,
1324                                     CTRL_CMD_NEWFAMILY);
1325         if (IS_ERR(msg))
1326                 return PTR_ERR(msg);
1327
1328         return genlmsg_reply(msg, info);
1329 }
1330
1331 static int genl_ctrl_event(int event, const struct genl_family *family,
1332                            const struct genl_multicast_group *grp,
1333                            int grp_id)
1334 {
1335         struct sk_buff *msg;
1336
1337         /* genl is still initialising */
1338         if (!init_net.genl_sock)
1339                 return 0;
1340
1341         switch (event) {
1342         case CTRL_CMD_NEWFAMILY:
1343         case CTRL_CMD_DELFAMILY:
1344                 WARN_ON(grp);
1345                 msg = ctrl_build_family_msg(family, 0, 0, event);
1346                 break;
1347         case CTRL_CMD_NEWMCAST_GRP:
1348         case CTRL_CMD_DELMCAST_GRP:
1349                 BUG_ON(!grp);
1350                 msg = ctrl_build_mcgrp_msg(family, grp, grp_id, 0, 0, event);
1351                 break;
1352         default:
1353                 return -EINVAL;
1354         }
1355
1356         if (IS_ERR(msg))
1357                 return PTR_ERR(msg);
1358
1359         if (!family->netnsok) {
1360                 genlmsg_multicast_netns(&genl_ctrl, &init_net, msg, 0,
1361                                         0, GFP_KERNEL);
1362         } else {
1363                 rcu_read_lock();
1364                 genlmsg_multicast_allns(&genl_ctrl, msg, 0,
1365                                         0, GFP_ATOMIC);
1366                 rcu_read_unlock();
1367         }
1368
1369         return 0;
1370 }
1371
1372 struct ctrl_dump_policy_ctx {
1373         struct netlink_policy_dump_state *state;
1374         const struct genl_family *rt;
1375         struct genl_op_iter *op_iter;
1376         u32 op;
1377         u16 fam_id;
1378         u8 dump_map:1,
1379            single_op:1;
1380 };
1381
1382 static const struct nla_policy ctrl_policy_policy[] = {
1383         [CTRL_ATTR_FAMILY_ID]   = { .type = NLA_U16 },
1384         [CTRL_ATTR_FAMILY_NAME] = { .type = NLA_NUL_STRING,
1385                                     .len = GENL_NAMSIZ - 1 },
1386         [CTRL_ATTR_OP]          = { .type = NLA_U32 },
1387 };
1388
1389 static int ctrl_dumppolicy_start(struct netlink_callback *cb)
1390 {
1391         const struct genl_dumpit_info *info = genl_dumpit_info(cb);
1392         struct ctrl_dump_policy_ctx *ctx = (void *)cb->ctx;
1393         struct nlattr **tb = info->attrs;
1394         const struct genl_family *rt;
1395         struct genl_op_iter i;
1396         int err;
1397
1398         BUILD_BUG_ON(sizeof(*ctx) > sizeof(cb->ctx));
1399
1400         if (!tb[CTRL_ATTR_FAMILY_ID] && !tb[CTRL_ATTR_FAMILY_NAME])
1401                 return -EINVAL;
1402
1403         if (tb[CTRL_ATTR_FAMILY_ID]) {
1404                 ctx->fam_id = nla_get_u16(tb[CTRL_ATTR_FAMILY_ID]);
1405         } else {
1406                 rt = genl_family_find_byname(
1407                         nla_data(tb[CTRL_ATTR_FAMILY_NAME]));
1408                 if (!rt)
1409                         return -ENOENT;
1410                 ctx->fam_id = rt->id;
1411         }
1412
1413         rt = genl_family_find_byid(ctx->fam_id);
1414         if (!rt)
1415                 return -ENOENT;
1416
1417         ctx->rt = rt;
1418
1419         if (tb[CTRL_ATTR_OP]) {
1420                 struct genl_split_ops doit, dump;
1421
1422                 ctx->single_op = true;
1423                 ctx->op = nla_get_u32(tb[CTRL_ATTR_OP]);
1424
1425                 err = genl_get_cmd_both(ctx->op, rt, &doit, &dump);
1426                 if (err) {
1427                         NL_SET_BAD_ATTR(cb->extack, tb[CTRL_ATTR_OP]);
1428                         return err;
1429                 }
1430
1431                 if (doit.policy) {
1432                         err = netlink_policy_dump_add_policy(&ctx->state,
1433                                                              doit.policy,
1434                                                              doit.maxattr);
1435                         if (err)
1436                                 goto err_free_state;
1437                 }
1438                 if (dump.policy) {
1439                         err = netlink_policy_dump_add_policy(&ctx->state,
1440                                                              dump.policy,
1441                                                              dump.maxattr);
1442                         if (err)
1443                                 goto err_free_state;
1444                 }
1445
1446                 if (!ctx->state)
1447                         return -ENODATA;
1448
1449                 ctx->dump_map = 1;
1450                 return 0;
1451         }
1452
1453         ctx->op_iter = kmalloc(sizeof(*ctx->op_iter), GFP_KERNEL);
1454         if (!ctx->op_iter)
1455                 return -ENOMEM;
1456
1457         genl_op_iter_init(rt, ctx->op_iter);
1458         ctx->dump_map = genl_op_iter_next(ctx->op_iter);
1459
1460         for (genl_op_iter_init(rt, &i); genl_op_iter_next(&i); ) {
1461                 if (i.doit.policy) {
1462                         err = netlink_policy_dump_add_policy(&ctx->state,
1463                                                              i.doit.policy,
1464                                                              i.doit.maxattr);
1465                         if (err)
1466                                 goto err_free_state;
1467                 }
1468                 if (i.dumpit.policy) {
1469                         err = netlink_policy_dump_add_policy(&ctx->state,
1470                                                              i.dumpit.policy,
1471                                                              i.dumpit.maxattr);
1472                         if (err)
1473                                 goto err_free_state;
1474                 }
1475         }
1476
1477         if (!ctx->state) {
1478                 err = -ENODATA;
1479                 goto err_free_op_iter;
1480         }
1481         return 0;
1482
1483 err_free_state:
1484         netlink_policy_dump_free(ctx->state);
1485 err_free_op_iter:
1486         kfree(ctx->op_iter);
1487         return err;
1488 }
1489
1490 static void *ctrl_dumppolicy_prep(struct sk_buff *skb,
1491                                   struct netlink_callback *cb)
1492 {
1493         struct ctrl_dump_policy_ctx *ctx = (void *)cb->ctx;
1494         void *hdr;
1495
1496         hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid,
1497                           cb->nlh->nlmsg_seq, &genl_ctrl,
1498                           NLM_F_MULTI, CTRL_CMD_GETPOLICY);
1499         if (!hdr)
1500                 return NULL;
1501
1502         if (nla_put_u16(skb, CTRL_ATTR_FAMILY_ID, ctx->fam_id))
1503                 return NULL;
1504
1505         return hdr;
1506 }
1507
1508 static int ctrl_dumppolicy_put_op(struct sk_buff *skb,
1509                                   struct netlink_callback *cb,
1510                                   struct genl_split_ops *doit,
1511                                   struct genl_split_ops *dumpit)
1512 {
1513         struct ctrl_dump_policy_ctx *ctx = (void *)cb->ctx;
1514         struct nlattr *nest_pol, *nest_op;
1515         void *hdr;
1516         int idx;
1517
1518         /* skip if we have nothing to show */
1519         if (!doit->policy && !dumpit->policy)
1520                 return 0;
1521
1522         hdr = ctrl_dumppolicy_prep(skb, cb);
1523         if (!hdr)
1524                 return -ENOBUFS;
1525
1526         nest_pol = nla_nest_start(skb, CTRL_ATTR_OP_POLICY);
1527         if (!nest_pol)
1528                 goto err;
1529
1530         nest_op = nla_nest_start(skb, doit->cmd);
1531         if (!nest_op)
1532                 goto err;
1533
1534         if (doit->policy) {
1535                 idx = netlink_policy_dump_get_policy_idx(ctx->state,
1536                                                          doit->policy,
1537                                                          doit->maxattr);
1538
1539                 if (nla_put_u32(skb, CTRL_ATTR_POLICY_DO, idx))
1540                         goto err;
1541         }
1542         if (dumpit->policy) {
1543                 idx = netlink_policy_dump_get_policy_idx(ctx->state,
1544                                                          dumpit->policy,
1545                                                          dumpit->maxattr);
1546
1547                 if (nla_put_u32(skb, CTRL_ATTR_POLICY_DUMP, idx))
1548                         goto err;
1549         }
1550
1551         nla_nest_end(skb, nest_op);
1552         nla_nest_end(skb, nest_pol);
1553         genlmsg_end(skb, hdr);
1554
1555         return 0;
1556 err:
1557         genlmsg_cancel(skb, hdr);
1558         return -ENOBUFS;
1559 }
1560
1561 static int ctrl_dumppolicy(struct sk_buff *skb, struct netlink_callback *cb)
1562 {
1563         struct ctrl_dump_policy_ctx *ctx = (void *)cb->ctx;
1564         void *hdr;
1565
1566         if (ctx->dump_map) {
1567                 if (ctx->single_op) {
1568                         struct genl_split_ops doit, dumpit;
1569
1570                         if (WARN_ON(genl_get_cmd_both(ctx->op, ctx->rt,
1571                                                       &doit, &dumpit)))
1572                                 return -ENOENT;
1573
1574                         if (ctrl_dumppolicy_put_op(skb, cb, &doit, &dumpit))
1575                                 return skb->len;
1576
1577                         /* done with the per-op policy index list */
1578                         ctx->dump_map = 0;
1579                 }
1580
1581                 while (ctx->dump_map) {
1582                         if (ctrl_dumppolicy_put_op(skb, cb,
1583                                                    &ctx->op_iter->doit,
1584                                                    &ctx->op_iter->dumpit))
1585                                 return skb->len;
1586
1587                         ctx->dump_map = genl_op_iter_next(ctx->op_iter);
1588                 }
1589         }
1590
1591         while (netlink_policy_dump_loop(ctx->state)) {
1592                 struct nlattr *nest;
1593
1594                 hdr = ctrl_dumppolicy_prep(skb, cb);
1595                 if (!hdr)
1596                         goto nla_put_failure;
1597
1598                 nest = nla_nest_start(skb, CTRL_ATTR_POLICY);
1599                 if (!nest)
1600                         goto nla_put_failure;
1601
1602                 if (netlink_policy_dump_write(skb, ctx->state))
1603                         goto nla_put_failure;
1604
1605                 nla_nest_end(skb, nest);
1606
1607                 genlmsg_end(skb, hdr);
1608         }
1609
1610         return skb->len;
1611
1612 nla_put_failure:
1613         genlmsg_cancel(skb, hdr);
1614         return skb->len;
1615 }
1616
1617 static int ctrl_dumppolicy_done(struct netlink_callback *cb)
1618 {
1619         struct ctrl_dump_policy_ctx *ctx = (void *)cb->ctx;
1620
1621         kfree(ctx->op_iter);
1622         netlink_policy_dump_free(ctx->state);
1623         return 0;
1624 }
1625
1626 static const struct genl_split_ops genl_ctrl_ops[] = {
1627         {
1628                 .cmd            = CTRL_CMD_GETFAMILY,
1629                 .validate       = GENL_DONT_VALIDATE_STRICT,
1630                 .policy         = ctrl_policy_family,
1631                 .maxattr        = ARRAY_SIZE(ctrl_policy_family) - 1,
1632                 .doit           = ctrl_getfamily,
1633                 .flags          = GENL_CMD_CAP_DO,
1634         },
1635         {
1636                 .cmd            = CTRL_CMD_GETFAMILY,
1637                 .validate       = GENL_DONT_VALIDATE_DUMP,
1638                 .policy         = ctrl_policy_family,
1639                 .maxattr        = ARRAY_SIZE(ctrl_policy_family) - 1,
1640                 .dumpit         = ctrl_dumpfamily,
1641                 .flags          = GENL_CMD_CAP_DUMP,
1642         },
1643         {
1644                 .cmd            = CTRL_CMD_GETPOLICY,
1645                 .policy         = ctrl_policy_policy,
1646                 .maxattr        = ARRAY_SIZE(ctrl_policy_policy) - 1,
1647                 .start          = ctrl_dumppolicy_start,
1648                 .dumpit         = ctrl_dumppolicy,
1649                 .done           = ctrl_dumppolicy_done,
1650                 .flags          = GENL_CMD_CAP_DUMP,
1651         },
1652 };
1653
1654 static const struct genl_multicast_group genl_ctrl_groups[] = {
1655         { .name = "notify", },
1656 };
1657
1658 static struct genl_family genl_ctrl __ro_after_init = {
1659         .module = THIS_MODULE,
1660         .split_ops = genl_ctrl_ops,
1661         .n_split_ops = ARRAY_SIZE(genl_ctrl_ops),
1662         .resv_start_op = CTRL_CMD_GETPOLICY + 1,
1663         .mcgrps = genl_ctrl_groups,
1664         .n_mcgrps = ARRAY_SIZE(genl_ctrl_groups),
1665         .id = GENL_ID_CTRL,
1666         .name = "nlctrl",
1667         .version = 0x2,
1668         .netnsok = true,
1669 };
1670
1671 static int genl_bind(struct net *net, int group)
1672 {
1673         const struct genl_family *family;
1674         unsigned int id;
1675         int ret = 0;
1676
1677         down_read(&cb_lock);
1678
1679         idr_for_each_entry(&genl_fam_idr, family, id) {
1680                 const struct genl_multicast_group *grp;
1681                 int i;
1682
1683                 if (family->n_mcgrps == 0)
1684                         continue;
1685
1686                 i = group - family->mcgrp_offset;
1687                 if (i < 0 || i >= family->n_mcgrps)
1688                         continue;
1689
1690                 grp = &family->mcgrps[i];
1691                 if ((grp->flags & GENL_UNS_ADMIN_PERM) &&
1692                     !ns_capable(net->user_ns, CAP_NET_ADMIN))
1693                         ret = -EPERM;
1694
1695                 break;
1696         }
1697
1698         up_read(&cb_lock);
1699         return ret;
1700 }
1701
1702 static int __net_init genl_pernet_init(struct net *net)
1703 {
1704         struct netlink_kernel_cfg cfg = {
1705                 .input          = genl_rcv,
1706                 .flags          = NL_CFG_F_NONROOT_RECV,
1707                 .bind           = genl_bind,
1708         };
1709
1710         /* we'll bump the group number right afterwards */
1711         net->genl_sock = netlink_kernel_create(net, NETLINK_GENERIC, &cfg);
1712
1713         if (!net->genl_sock && net_eq(net, &init_net))
1714                 panic("GENL: Cannot initialize generic netlink\n");
1715
1716         if (!net->genl_sock)
1717                 return -ENOMEM;
1718
1719         return 0;
1720 }
1721
1722 static void __net_exit genl_pernet_exit(struct net *net)
1723 {
1724         netlink_kernel_release(net->genl_sock);
1725         net->genl_sock = NULL;
1726 }
1727
1728 static struct pernet_operations genl_pernet_ops = {
1729         .init = genl_pernet_init,
1730         .exit = genl_pernet_exit,
1731 };
1732
1733 static int __init genl_init(void)
1734 {
1735         int err;
1736
1737         err = genl_register_family(&genl_ctrl);
1738         if (err < 0)
1739                 goto problem;
1740
1741         err = register_pernet_subsys(&genl_pernet_ops);
1742         if (err)
1743                 goto problem;
1744
1745         return 0;
1746
1747 problem:
1748         panic("GENL: Cannot register controller: %d\n", err);
1749 }
1750
1751 core_initcall(genl_init);
1752
1753 static int genlmsg_mcast(struct sk_buff *skb, u32 portid, unsigned long group,
1754                          gfp_t flags)
1755 {
1756         struct sk_buff *tmp;
1757         struct net *net, *prev = NULL;
1758         bool delivered = false;
1759         int err;
1760
1761         for_each_net_rcu(net) {
1762                 if (prev) {
1763                         tmp = skb_clone(skb, flags);
1764                         if (!tmp) {
1765                                 err = -ENOMEM;
1766                                 goto error;
1767                         }
1768                         err = nlmsg_multicast(prev->genl_sock, tmp,
1769                                               portid, group, flags);
1770                         if (!err)
1771                                 delivered = true;
1772                         else if (err != -ESRCH)
1773                                 goto error;
1774                 }
1775
1776                 prev = net;
1777         }
1778
1779         err = nlmsg_multicast(prev->genl_sock, skb, portid, group, flags);
1780         if (!err)
1781                 delivered = true;
1782         else if (err != -ESRCH)
1783                 return err;
1784         return delivered ? 0 : -ESRCH;
1785  error:
1786         kfree_skb(skb);
1787         return err;
1788 }
1789
1790 int genlmsg_multicast_allns(const struct genl_family *family,
1791                             struct sk_buff *skb, u32 portid,
1792                             unsigned int group, gfp_t flags)
1793 {
1794         if (WARN_ON_ONCE(group >= family->n_mcgrps))
1795                 return -EINVAL;
1796
1797         group = family->mcgrp_offset + group;
1798         return genlmsg_mcast(skb, portid, group, flags);
1799 }
1800 EXPORT_SYMBOL(genlmsg_multicast_allns);
1801
1802 void genl_notify(const struct genl_family *family, struct sk_buff *skb,
1803                  struct genl_info *info, u32 group, gfp_t flags)
1804 {
1805         struct net *net = genl_info_net(info);
1806         struct sock *sk = net->genl_sock;
1807
1808         if (WARN_ON_ONCE(group >= family->n_mcgrps))
1809                 return;
1810
1811         group = family->mcgrp_offset + group;
1812         nlmsg_notify(sk, skb, info->snd_portid, group,
1813                      nlmsg_report(info->nlhdr), flags);
1814 }
1815 EXPORT_SYMBOL(genl_notify);