net/netfilter/nf_tables_api.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (c) 2007-2009 Patrick McHardy <kaber@trash.net>
   4  *
   5  * Development of this code funded by Astaro AG (http://www.astaro.com/)
   6  */
   7
   8 #include <linux/module.h>
   9 #include <linux/init.h>
  10 #include <linux/list.h>
  11 #include <linux/skbuff.h>
  12 #include <linux/netlink.h>
  13 #include <linux/vmalloc.h>
  14 #include <linux/rhashtable.h>
  15 #include <linux/audit.h>
  16 #include <linux/netfilter.h>
  17 #include <linux/netfilter/nfnetlink.h>
  18 #include <linux/netfilter/nf_tables.h>
  19 #include <net/netfilter/nf_flow_table.h>
  20 #include <net/netfilter/nf_tables_core.h>
  21 #include <net/netfilter/nf_tables.h>
  22 #include <net/netfilter/nf_tables_offload.h>
  23 #include <net/net_namespace.h>
  24 #include <net/sock.h>
  25
  26 #define NFT_MODULE_AUTOLOAD_LIMIT (MODULE_NAME_LEN - sizeof("nft-expr-255-"))
  27
  28 unsigned int nf_tables_net_id __read_mostly;
  29
  30 static LIST_HEAD(nf_tables_expressions);
  31 static LIST_HEAD(nf_tables_objects);
  32 static LIST_HEAD(nf_tables_flowtables);
  33 static LIST_HEAD(nf_tables_destroy_list);
  34 static DEFINE_SPINLOCK(nf_tables_destroy_list_lock);
  35
  36 enum {
  37         NFT_VALIDATE_SKIP       = 0,
  38         NFT_VALIDATE_NEED,
  39         NFT_VALIDATE_DO,
  40 };
  41
  42 static struct rhltable nft_objname_ht;
  43
  44 static u32 nft_chain_hash(const void *data, u32 len, u32 seed);
  45 static u32 nft_chain_hash_obj(const void *data, u32 len, u32 seed);
  46 static int nft_chain_hash_cmp(struct rhashtable_compare_arg *, const void *);
  47
  48 static u32 nft_objname_hash(const void *data, u32 len, u32 seed);
  49 static u32 nft_objname_hash_obj(const void *data, u32 len, u32 seed);
  50 static int nft_objname_hash_cmp(struct rhashtable_compare_arg *, const void *);
  51
  52 static const struct rhashtable_params nft_chain_ht_params = {
  53         .head_offset            = offsetof(struct nft_chain, rhlhead),
  54         .key_offset             = offsetof(struct nft_chain, name),
  55         .hashfn                 = nft_chain_hash,
  56         .obj_hashfn             = nft_chain_hash_obj,
  57         .obj_cmpfn              = nft_chain_hash_cmp,
  58         .automatic_shrinking    = true,
  59 };
  60
  61 static const struct rhashtable_params nft_objname_ht_params = {
  62         .head_offset            = offsetof(struct nft_object, rhlhead),
  63         .key_offset             = offsetof(struct nft_object, key),
  64         .hashfn                 = nft_objname_hash,
  65         .obj_hashfn             = nft_objname_hash_obj,
  66         .obj_cmpfn              = nft_objname_hash_cmp,
  67         .automatic_shrinking    = true,
  68 };
  69
  70 struct nft_audit_data {
  71         struct nft_table *table;
  72         int entries;
  73         int op;
  74         struct list_head list;
  75 };
  76
  77 static const u8 nft2audit_op[NFT_MSG_MAX] = { // enum nf_tables_msg_types
  78         [NFT_MSG_NEWTABLE]      = AUDIT_NFT_OP_TABLE_REGISTER,
  79         [NFT_MSG_GETTABLE]      = AUDIT_NFT_OP_INVALID,
  80         [NFT_MSG_DELTABLE]      = AUDIT_NFT_OP_TABLE_UNREGISTER,
  81         [NFT_MSG_NEWCHAIN]      = AUDIT_NFT_OP_CHAIN_REGISTER,
  82         [NFT_MSG_GETCHAIN]      = AUDIT_NFT_OP_INVALID,
  83         [NFT_MSG_DELCHAIN]      = AUDIT_NFT_OP_CHAIN_UNREGISTER,
  84         [NFT_MSG_NEWRULE]       = AUDIT_NFT_OP_RULE_REGISTER,
  85         [NFT_MSG_GETRULE]       = AUDIT_NFT_OP_INVALID,
  86         [NFT_MSG_DELRULE]       = AUDIT_NFT_OP_RULE_UNREGISTER,
  87         [NFT_MSG_NEWSET]        = AUDIT_NFT_OP_SET_REGISTER,
  88         [NFT_MSG_GETSET]        = AUDIT_NFT_OP_INVALID,
  89         [NFT_MSG_DELSET]        = AUDIT_NFT_OP_SET_UNREGISTER,
  90         [NFT_MSG_NEWSETELEM]    = AUDIT_NFT_OP_SETELEM_REGISTER,
  91         [NFT_MSG_GETSETELEM]    = AUDIT_NFT_OP_INVALID,
  92         [NFT_MSG_DELSETELEM]    = AUDIT_NFT_OP_SETELEM_UNREGISTER,
  93         [NFT_MSG_NEWGEN]        = AUDIT_NFT_OP_GEN_REGISTER,
  94         [NFT_MSG_GETGEN]        = AUDIT_NFT_OP_INVALID,
  95         [NFT_MSG_TRACE]         = AUDIT_NFT_OP_INVALID,
  96         [NFT_MSG_NEWOBJ]        = AUDIT_NFT_OP_OBJ_REGISTER,
  97         [NFT_MSG_GETOBJ]        = AUDIT_NFT_OP_INVALID,
  98         [NFT_MSG_DELOBJ]        = AUDIT_NFT_OP_OBJ_UNREGISTER,
  99         [NFT_MSG_GETOBJ_RESET]  = AUDIT_NFT_OP_OBJ_RESET,
 100         [NFT_MSG_NEWFLOWTABLE]  = AUDIT_NFT_OP_FLOWTABLE_REGISTER,
 101         [NFT_MSG_GETFLOWTABLE]  = AUDIT_NFT_OP_INVALID,
 102         [NFT_MSG_DELFLOWTABLE]  = AUDIT_NFT_OP_FLOWTABLE_UNREGISTER,
 103 };
 104
 105 static void nft_validate_state_update(struct nft_table *table, u8 new_validate_state)
 106 {
 107         switch (table->validate_state) {
 108         case NFT_VALIDATE_SKIP:
 109                 WARN_ON_ONCE(new_validate_state == NFT_VALIDATE_DO);
 110                 break;
 111         case NFT_VALIDATE_NEED:
 112                 break;
 113         case NFT_VALIDATE_DO:
 114                 if (new_validate_state == NFT_VALIDATE_NEED)
 115                         return;
 116         }
 117
 118         table->validate_state = new_validate_state;
 119 }
 120 static void nf_tables_trans_destroy_work(struct work_struct *w);
 121 static DECLARE_WORK(trans_destroy_work, nf_tables_trans_destroy_work);
 122
 123 static void nft_ctx_init(struct nft_ctx *ctx,
 124                          struct net *net,
 125                          const struct sk_buff *skb,
 126                          const struct nlmsghdr *nlh,
 127                          u8 family,
 128                          struct nft_table *table,
 129                          struct nft_chain *chain,
 130                          const struct nlattr * const *nla)
 131 {
 132         ctx->net        = net;
 133         ctx->family     = family;
 134         ctx->level      = 0;
 135         ctx->table      = table;
 136         ctx->chain      = chain;
 137         ctx->nla        = nla;
 138         ctx->portid     = NETLINK_CB(skb).portid;
 139         ctx->report     = nlmsg_report(nlh);
 140         ctx->flags      = nlh->nlmsg_flags;
 141         ctx->seq        = nlh->nlmsg_seq;
 142 }
 143
 144 static struct nft_trans *nft_trans_alloc_gfp(const struct nft_ctx *ctx,
 145                                              int msg_type, u32 size, gfp_t gfp)
 146 {
 147         struct nft_trans *trans;
 148
 149         trans = kzalloc(sizeof(struct nft_trans) + size, gfp);
 150         if (trans == NULL)
 151                 return NULL;
 152
 153         INIT_LIST_HEAD(&trans->list);
 154         INIT_LIST_HEAD(&trans->binding_list);
 155         trans->msg_type = msg_type;
 156         trans->ctx      = *ctx;
 157
 158         return trans;
 159 }
 160
 161 static struct nft_trans *nft_trans_alloc(const struct nft_ctx *ctx,
 162                                          int msg_type, u32 size)
 163 {
 164         return nft_trans_alloc_gfp(ctx, msg_type, size, GFP_KERNEL);
 165 }
 166
 167 static void nft_trans_list_del(struct nft_trans *trans)
 168 {
 169         list_del(&trans->list);
 170         list_del(&trans->binding_list);
 171 }
 172
 173 static void nft_trans_destroy(struct nft_trans *trans)
 174 {
 175         nft_trans_list_del(trans);
 176         kfree(trans);
 177 }
 178
 179 static void __nft_set_trans_bind(const struct nft_ctx *ctx, struct nft_set *set,
 180                                  bool bind)
 181 {
 182         struct nftables_pernet *nft_net;
 183         struct net *net = ctx->net;
 184         struct nft_trans *trans;
 185
 186         if (!nft_set_is_anonymous(set))
 187                 return;
 188
 189         nft_net = nft_pernet(net);
 190         list_for_each_entry_reverse(trans, &nft_net->commit_list, list) {
 191                 switch (trans->msg_type) {
 192                 case NFT_MSG_NEWSET:
 193                         if (nft_trans_set(trans) == set)
 194                                 nft_trans_set_bound(trans) = bind;
 195                         break;
 196                 case NFT_MSG_NEWSETELEM:
 197                         if (nft_trans_elem_set(trans) == set)
 198                                 nft_trans_elem_set_bound(trans) = bind;
 199                         break;
 200                 }
 201         }
 202 }
 203
 204 static void nft_set_trans_bind(const struct nft_ctx *ctx, struct nft_set *set)
 205 {
 206         return __nft_set_trans_bind(ctx, set, true);
 207 }
 208
 209 static void nft_set_trans_unbind(const struct nft_ctx *ctx, struct nft_set *set)
 210 {
 211         return __nft_set_trans_bind(ctx, set, false);
 212 }
 213
 214 static void __nft_chain_trans_bind(const struct nft_ctx *ctx,
 215                                    struct nft_chain *chain, bool bind)
 216 {
 217         struct nftables_pernet *nft_net;
 218         struct net *net = ctx->net;
 219         struct nft_trans *trans;
 220
 221         if (!nft_chain_binding(chain))
 222                 return;
 223
 224         nft_net = nft_pernet(net);
 225         list_for_each_entry_reverse(trans, &nft_net->commit_list, list) {
 226                 switch (trans->msg_type) {
 227                 case NFT_MSG_NEWCHAIN:
 228                         if (nft_trans_chain(trans) == chain)
 229                                 nft_trans_chain_bound(trans) = bind;
 230                         break;
 231                 case NFT_MSG_NEWRULE:
 232                         if (trans->ctx.chain == chain)
 233                                 nft_trans_rule_bound(trans) = bind;
 234                         break;
 235                 }
 236         }
 237 }
 238
 239 static void nft_chain_trans_bind(const struct nft_ctx *ctx,
 240                                  struct nft_chain *chain)
 241 {
 242         __nft_chain_trans_bind(ctx, chain, true);
 243 }
 244
 245 int nf_tables_bind_chain(const struct nft_ctx *ctx, struct nft_chain *chain)
 246 {
 247         if (!nft_chain_binding(chain))
 248                 return 0;
 249
 250         if (nft_chain_binding(ctx->chain))
 251                 return -EOPNOTSUPP;
 252
 253         if (chain->bound)
 254                 return -EBUSY;
 255
 256         chain->bound = true;
 257         chain->use++;
 258         nft_chain_trans_bind(ctx, chain);
 259
 260         return 0;
 261 }
 262
 263 void nf_tables_unbind_chain(const struct nft_ctx *ctx, struct nft_chain *chain)
 264 {
 265         __nft_chain_trans_bind(ctx, chain, false);
 266 }
 267
 268 static int nft_netdev_register_hooks(struct net *net,
 269                                      struct list_head *hook_list)
 270 {
 271         struct nft_hook *hook;
 272         int err, j;
 273
 274         j = 0;
 275         list_for_each_entry(hook, hook_list, list) {
 276                 err = nf_register_net_hook(net, &hook->ops);
 277                 if (err < 0)
 278                         goto err_register;
 279
 280                 j++;
 281         }
 282         return 0;
 283
 284 err_register:
 285         list_for_each_entry(hook, hook_list, list) {
 286                 if (j-- <= 0)
 287                         break;
 288
 289                 nf_unregister_net_hook(net, &hook->ops);
 290         }
 291         return err;
 292 }
 293
 294 static void nft_netdev_unregister_hooks(struct net *net,
 295                                         struct list_head *hook_list,
 296                                         bool release_netdev)
 297 {
 298         struct nft_hook *hook, *next;
 299
 300         list_for_each_entry_safe(hook, next, hook_list, list) {
 301                 nf_unregister_net_hook(net, &hook->ops);
 302                 if (release_netdev) {
 303                         list_del(&hook->list);
 304                         kfree_rcu(hook, rcu);
 305                 }
 306         }
 307 }
 308
 309 static int nf_tables_register_hook(struct net *net,
 310                                    const struct nft_table *table,
 311                                    struct nft_chain *chain)
 312 {
 313         struct nft_base_chain *basechain;
 314         const struct nf_hook_ops *ops;
 315
 316         if (table->flags & NFT_TABLE_F_DORMANT ||
 317             !nft_is_base_chain(chain))
 318                 return 0;
 319
 320         basechain = nft_base_chain(chain);
 321         ops = &basechain->ops;
 322
 323         if (basechain->type->ops_register)
 324                 return basechain->type->ops_register(net, ops);
 325
 326         if (nft_base_chain_netdev(table->family, basechain->ops.hooknum))
 327                 return nft_netdev_register_hooks(net, &basechain->hook_list);
 328
 329         return nf_register_net_hook(net, &basechain->ops);
 330 }
 331
 332 static void __nf_tables_unregister_hook(struct net *net,
 333                                         const struct nft_table *table,
 334                                         struct nft_chain *chain,
 335                                         bool release_netdev)
 336 {
 337         struct nft_base_chain *basechain;
 338         const struct nf_hook_ops *ops;
 339
 340         if (table->flags & NFT_TABLE_F_DORMANT ||
 341             !nft_is_base_chain(chain))
 342                 return;
 343         basechain = nft_base_chain(chain);
 344         ops = &basechain->ops;
 345
 346         if (basechain->type->ops_unregister)
 347                 return basechain->type->ops_unregister(net, ops);
 348
 349         if (nft_base_chain_netdev(table->family, basechain->ops.hooknum))
 350                 nft_netdev_unregister_hooks(net, &basechain->hook_list,
 351                                             release_netdev);
 352         else
 353                 nf_unregister_net_hook(net, &basechain->ops);
 354 }
 355
 356 static void nf_tables_unregister_hook(struct net *net,
 357                                       const struct nft_table *table,
 358                                       struct nft_chain *chain)
 359 {
 360         return __nf_tables_unregister_hook(net, table, chain, false);
 361 }
 362
 363 static void nft_trans_commit_list_add_tail(struct net *net, struct nft_trans *trans)
 364 {
 365         struct nftables_pernet *nft_net = nft_pernet(net);
 366
 367         switch (trans->msg_type) {
 368         case NFT_MSG_NEWSET:
 369                 if (!nft_trans_set_update(trans) &&
 370                     nft_set_is_anonymous(nft_trans_set(trans)))
 371                         list_add_tail(&trans->binding_list, &nft_net->binding_list);
 372                 break;
 373         case NFT_MSG_NEWCHAIN:
 374                 if (!nft_trans_chain_update(trans) &&
 375                     nft_chain_binding(nft_trans_chain(trans)))
 376                         list_add_tail(&trans->binding_list, &nft_net->binding_list);
 377                 break;
 378         }
 379
 380         list_add_tail(&trans->list, &nft_net->commit_list);
 381 }
 382
 383 static int nft_trans_table_add(struct nft_ctx *ctx, int msg_type)
 384 {
 385         struct nft_trans *trans;
 386
 387         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_table));
 388         if (trans == NULL)
 389                 return -ENOMEM;
 390
 391         if (msg_type == NFT_MSG_NEWTABLE)
 392                 nft_activate_next(ctx->net, ctx->table);
 393
 394         nft_trans_commit_list_add_tail(ctx->net, trans);
 395         return 0;
 396 }
 397
 398 static int nft_deltable(struct nft_ctx *ctx)
 399 {
 400         int err;
 401
 402         err = nft_trans_table_add(ctx, NFT_MSG_DELTABLE);
 403         if (err < 0)
 404                 return err;
 405
 406         nft_deactivate_next(ctx->net, ctx->table);
 407         return err;
 408 }
 409
 410 static struct nft_trans *nft_trans_chain_add(struct nft_ctx *ctx, int msg_type)
 411 {
 412         struct nft_trans *trans;
 413
 414         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_chain));
 415         if (trans == NULL)
 416                 return ERR_PTR(-ENOMEM);
 417
 418         if (msg_type == NFT_MSG_NEWCHAIN) {
 419                 nft_activate_next(ctx->net, ctx->chain);
 420
 421                 if (ctx->nla[NFTA_CHAIN_ID]) {
 422                         nft_trans_chain_id(trans) =
 423                                 ntohl(nla_get_be32(ctx->nla[NFTA_CHAIN_ID]));
 424                 }
 425         }
 426         nft_trans_chain(trans) = ctx->chain;
 427         nft_trans_commit_list_add_tail(ctx->net, trans);
 428
 429         return trans;
 430 }
 431
 432 static int nft_delchain(struct nft_ctx *ctx)
 433 {
 434         struct nft_trans *trans;
 435
 436         trans = nft_trans_chain_add(ctx, NFT_MSG_DELCHAIN);
 437         if (IS_ERR(trans))
 438                 return PTR_ERR(trans);
 439
 440         ctx->table->use--;
 441         nft_deactivate_next(ctx->net, ctx->chain);
 442
 443         return 0;
 444 }
 445
 446 void nft_rule_expr_activate(const struct nft_ctx *ctx, struct nft_rule *rule)
 447 {
 448         struct nft_expr *expr;
 449
 450         expr = nft_expr_first(rule);
 451         while (nft_expr_more(rule, expr)) {
 452                 if (expr->ops->activate)
 453                         expr->ops->activate(ctx, expr);
 454
 455                 expr = nft_expr_next(expr);
 456         }
 457 }
 458
 459 void nft_rule_expr_deactivate(const struct nft_ctx *ctx, struct nft_rule *rule,
 460                               enum nft_trans_phase phase)
 461 {
 462         struct nft_expr *expr;
 463
 464         expr = nft_expr_first(rule);
 465         while (nft_expr_more(rule, expr)) {
 466                 if (expr->ops->deactivate)
 467                         expr->ops->deactivate(ctx, expr, phase);
 468
 469                 expr = nft_expr_next(expr);
 470         }
 471 }
 472
 473 static int
 474 nf_tables_delrule_deactivate(struct nft_ctx *ctx, struct nft_rule *rule)
 475 {
 476         /* You cannot delete the same rule twice */
 477         if (nft_is_active_next(ctx->net, rule)) {
 478                 nft_deactivate_next(ctx->net, rule);
 479                 ctx->chain->use--;
 480                 return 0;
 481         }
 482         return -ENOENT;
 483 }
 484
 485 static struct nft_trans *nft_trans_rule_add(struct nft_ctx *ctx, int msg_type,
 486                                             struct nft_rule *rule)
 487 {
 488         struct nft_trans *trans;
 489
 490         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_rule));
 491         if (trans == NULL)
 492                 return NULL;
 493
 494         if (msg_type == NFT_MSG_NEWRULE && ctx->nla[NFTA_RULE_ID] != NULL) {
 495                 nft_trans_rule_id(trans) =
 496                         ntohl(nla_get_be32(ctx->nla[NFTA_RULE_ID]));
 497         }
 498         nft_trans_rule(trans) = rule;
 499         nft_trans_commit_list_add_tail(ctx->net, trans);
 500
 501         return trans;
 502 }
 503
 504 static int nft_delrule(struct nft_ctx *ctx, struct nft_rule *rule)
 505 {
 506         struct nft_flow_rule *flow;
 507         struct nft_trans *trans;
 508         int err;
 509
 510         trans = nft_trans_rule_add(ctx, NFT_MSG_DELRULE, rule);
 511         if (trans == NULL)
 512                 return -ENOMEM;
 513
 514         if (ctx->chain->flags & NFT_CHAIN_HW_OFFLOAD) {
 515                 flow = nft_flow_rule_create(ctx->net, rule);
 516                 if (IS_ERR(flow)) {
 517                         nft_trans_destroy(trans);
 518                         return PTR_ERR(flow);
 519                 }
 520
 521                 nft_trans_flow_rule(trans) = flow;
 522         }
 523
 524         err = nf_tables_delrule_deactivate(ctx, rule);
 525         if (err < 0) {
 526                 nft_trans_destroy(trans);
 527                 return err;
 528         }
 529         nft_rule_expr_deactivate(ctx, rule, NFT_TRANS_PREPARE);
 530
 531         return 0;
 532 }
 533
 534 static int nft_delrule_by_chain(struct nft_ctx *ctx)
 535 {
 536         struct nft_rule *rule;
 537         int err;
 538
 539         list_for_each_entry(rule, &ctx->chain->rules, list) {
 540                 if (!nft_is_active_next(ctx->net, rule))
 541                         continue;
 542
 543                 err = nft_delrule(ctx, rule);
 544                 if (err < 0)
 545                         return err;
 546         }
 547         return 0;
 548 }
 549
 550 static int __nft_trans_set_add(const struct nft_ctx *ctx, int msg_type,
 551                                struct nft_set *set,
 552                                const struct nft_set_desc *desc)
 553 {
 554         struct nft_trans *trans;
 555
 556         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_set));
 557         if (trans == NULL)
 558                 return -ENOMEM;
 559
 560         if (msg_type == NFT_MSG_NEWSET && ctx->nla[NFTA_SET_ID] && !desc) {
 561                 nft_trans_set_id(trans) =
 562                         ntohl(nla_get_be32(ctx->nla[NFTA_SET_ID]));
 563                 nft_activate_next(ctx->net, set);
 564         }
 565         nft_trans_set(trans) = set;
 566         if (desc) {
 567                 nft_trans_set_update(trans) = true;
 568                 nft_trans_set_gc_int(trans) = desc->gc_int;
 569                 nft_trans_set_timeout(trans) = desc->timeout;
 570         }
 571         nft_trans_commit_list_add_tail(ctx->net, trans);
 572
 573         return 0;
 574 }
 575
 576 static int nft_trans_set_add(const struct nft_ctx *ctx, int msg_type,
 577                              struct nft_set *set)
 578 {
 579         return __nft_trans_set_add(ctx, msg_type, set, NULL);
 580 }
 581
 582 static void nft_setelem_data_deactivate(const struct net *net,
 583                                         const struct nft_set *set,
 584                                         struct nft_set_elem *elem);
 585
 586 static int nft_mapelem_deactivate(const struct nft_ctx *ctx,
 587                                   struct nft_set *set,
 588                                   const struct nft_set_iter *iter,
 589                                   struct nft_set_elem *elem)
 590 {
 591         nft_setelem_data_deactivate(ctx->net, set, elem);
 592
 593         return 0;
 594 }
 595
 596 struct nft_set_elem_catchall {
 597         struct list_head        list;
 598         struct rcu_head         rcu;
 599         void                    *elem;
 600 };
 601
 602 static void nft_map_catchall_deactivate(const struct nft_ctx *ctx,
 603                                         struct nft_set *set)
 604 {
 605         u8 genmask = nft_genmask_next(ctx->net);
 606         struct nft_set_elem_catchall *catchall;
 607         struct nft_set_elem elem;
 608         struct nft_set_ext *ext;
 609
 610         list_for_each_entry(catchall, &set->catchall_list, list) {
 611                 ext = nft_set_elem_ext(set, catchall->elem);
 612                 if (!nft_set_elem_active(ext, genmask))
 613                         continue;
 614
 615                 elem.priv = catchall->elem;
 616                 nft_setelem_data_deactivate(ctx->net, set, &elem);
 617                 break;
 618         }
 619 }
 620
 621 static void nft_map_deactivate(const struct nft_ctx *ctx, struct nft_set *set)
 622 {
 623         struct nft_set_iter iter = {
 624                 .genmask        = nft_genmask_next(ctx->net),
 625                 .fn             = nft_mapelem_deactivate,
 626         };
 627
 628         set->ops->walk(ctx, set, &iter);
 629         WARN_ON_ONCE(iter.err);
 630
 631         nft_map_catchall_deactivate(ctx, set);
 632 }
 633
 634 static int nft_delset(const struct nft_ctx *ctx, struct nft_set *set)
 635 {
 636         int err;
 637
 638         err = nft_trans_set_add(ctx, NFT_MSG_DELSET, set);
 639         if (err < 0)
 640                 return err;
 641
 642         if (set->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
 643                 nft_map_deactivate(ctx, set);
 644
 645         nft_deactivate_next(ctx->net, set);
 646         ctx->table->use--;
 647
 648         return err;
 649 }
 650
 651 static int nft_trans_obj_add(struct nft_ctx *ctx, int msg_type,
 652                              struct nft_object *obj)
 653 {
 654         struct nft_trans *trans;
 655
 656         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_obj));
 657         if (trans == NULL)
 658                 return -ENOMEM;
 659
 660         if (msg_type == NFT_MSG_NEWOBJ)
 661                 nft_activate_next(ctx->net, obj);
 662
 663         nft_trans_obj(trans) = obj;
 664         nft_trans_commit_list_add_tail(ctx->net, trans);
 665
 666         return 0;
 667 }
 668
 669 static int nft_delobj(struct nft_ctx *ctx, struct nft_object *obj)
 670 {
 671         int err;
 672
 673         err = nft_trans_obj_add(ctx, NFT_MSG_DELOBJ, obj);
 674         if (err < 0)
 675                 return err;
 676
 677         nft_deactivate_next(ctx->net, obj);
 678         ctx->table->use--;
 679
 680         return err;
 681 }
 682
 683 static int nft_trans_flowtable_add(struct nft_ctx *ctx, int msg_type,
 684                                    struct nft_flowtable *flowtable)
 685 {
 686         struct nft_trans *trans;
 687
 688         trans = nft_trans_alloc(ctx, msg_type,
 689                                 sizeof(struct nft_trans_flowtable));
 690         if (trans == NULL)
 691                 return -ENOMEM;
 692
 693         if (msg_type == NFT_MSG_NEWFLOWTABLE)
 694                 nft_activate_next(ctx->net, flowtable);
 695
 696         INIT_LIST_HEAD(&nft_trans_flowtable_hooks(trans));
 697         nft_trans_flowtable(trans) = flowtable;
 698         nft_trans_commit_list_add_tail(ctx->net, trans);
 699
 700         return 0;
 701 }
 702
 703 static int nft_delflowtable(struct nft_ctx *ctx,
 704                             struct nft_flowtable *flowtable)
 705 {
 706         int err;
 707
 708         err = nft_trans_flowtable_add(ctx, NFT_MSG_DELFLOWTABLE, flowtable);
 709         if (err < 0)
 710                 return err;
 711
 712         nft_deactivate_next(ctx->net, flowtable);
 713         ctx->table->use--;
 714
 715         return err;
 716 }
 717
 718 static void __nft_reg_track_clobber(struct nft_regs_track *track, u8 dreg)
 719 {
 720         int i;
 721
 722         for (i = track->regs[dreg].num_reg; i > 0; i--)
 723                 __nft_reg_track_cancel(track, dreg - i);
 724 }
 725
 726 static void __nft_reg_track_update(struct nft_regs_track *track,
 727                                    const struct nft_expr *expr,
 728                                    u8 dreg, u8 num_reg)
 729 {
 730         track->regs[dreg].selector = expr;
 731         track->regs[dreg].bitwise = NULL;
 732         track->regs[dreg].num_reg = num_reg;
 733 }
 734
 735 void nft_reg_track_update(struct nft_regs_track *track,
 736                           const struct nft_expr *expr, u8 dreg, u8 len)
 737 {
 738         unsigned int regcount;
 739         int i;
 740
 741         __nft_reg_track_clobber(track, dreg);
 742
 743         regcount = DIV_ROUND_UP(len, NFT_REG32_SIZE);
 744         for (i = 0; i < regcount; i++, dreg++)
 745                 __nft_reg_track_update(track, expr, dreg, i);
 746 }
 747 EXPORT_SYMBOL_GPL(nft_reg_track_update);
 748
 749 void nft_reg_track_cancel(struct nft_regs_track *track, u8 dreg, u8 len)
 750 {
 751         unsigned int regcount;
 752         int i;
 753
 754         __nft_reg_track_clobber(track, dreg);
 755
 756         regcount = DIV_ROUND_UP(len, NFT_REG32_SIZE);
 757         for (i = 0; i < regcount; i++, dreg++)
 758                 __nft_reg_track_cancel(track, dreg);
 759 }
 760 EXPORT_SYMBOL_GPL(nft_reg_track_cancel);
 761
 762 void __nft_reg_track_cancel(struct nft_regs_track *track, u8 dreg)
 763 {
 764         track->regs[dreg].selector = NULL;
 765         track->regs[dreg].bitwise = NULL;
 766         track->regs[dreg].num_reg = 0;
 767 }
 768 EXPORT_SYMBOL_GPL(__nft_reg_track_cancel);
 769
 770 /*
 771  * Tables
 772  */
 773
 774 static struct nft_table *nft_table_lookup(const struct net *net,
 775                                           const struct nlattr *nla,
 776                                           u8 family, u8 genmask, u32 nlpid)
 777 {
 778         struct nftables_pernet *nft_net;
 779         struct nft_table *table;
 780
 781         if (nla == NULL)
 782                 return ERR_PTR(-EINVAL);
 783
 784         nft_net = nft_pernet(net);
 785         list_for_each_entry_rcu(table, &nft_net->tables, list,
 786                                 lockdep_is_held(&nft_net->commit_mutex)) {
 787                 if (!nla_strcmp(nla, table->name) &&
 788                     table->family == family &&
 789                     nft_active_genmask(table, genmask)) {
 790                         if (nft_table_has_owner(table) &&
 791                             nlpid && table->nlpid != nlpid)
 792                                 return ERR_PTR(-EPERM);
 793
 794                         return table;
 795                 }
 796         }
 797
 798         return ERR_PTR(-ENOENT);
 799 }
 800
 801 static struct nft_table *nft_table_lookup_byhandle(const struct net *net,
 802                                                    const struct nlattr *nla,
 803                                                    u8 genmask, u32 nlpid)
 804 {
 805         struct nftables_pernet *nft_net;
 806         struct nft_table *table;
 807
 808         nft_net = nft_pernet(net);
 809         list_for_each_entry(table, &nft_net->tables, list) {
 810                 if (be64_to_cpu(nla_get_be64(nla)) == table->handle &&
 811                     nft_active_genmask(table, genmask)) {
 812                         if (nft_table_has_owner(table) &&
 813                             nlpid && table->nlpid != nlpid)
 814                                 return ERR_PTR(-EPERM);
 815
 816                         return table;
 817                 }
 818         }
 819
 820         return ERR_PTR(-ENOENT);
 821 }
 822
 823 static inline u64 nf_tables_alloc_handle(struct nft_table *table)
 824 {
 825         return ++table->hgenerator;
 826 }
 827
 828 static const struct nft_chain_type *chain_type[NFPROTO_NUMPROTO][NFT_CHAIN_T_MAX];
 829
 830 static const struct nft_chain_type *
 831 __nft_chain_type_get(u8 family, enum nft_chain_types type)
 832 {
 833         if (family >= NFPROTO_NUMPROTO ||
 834             type >= NFT_CHAIN_T_MAX)
 835                 return NULL;
 836
 837         return chain_type[family][type];
 838 }
 839
 840 static const struct nft_chain_type *
 841 __nf_tables_chain_type_lookup(const struct nlattr *nla, u8 family)
 842 {
 843         const struct nft_chain_type *type;
 844         int i;
 845
 846         for (i = 0; i < NFT_CHAIN_T_MAX; i++) {
 847                 type = __nft_chain_type_get(family, i);
 848                 if (!type)
 849                         continue;
 850                 if (!nla_strcmp(nla, type->name))
 851                         return type;
 852         }
 853         return NULL;
 854 }
 855
 856 struct nft_module_request {
 857         struct list_head        list;
 858         char                    module[MODULE_NAME_LEN];
 859         bool                    done;
 860 };
 861
 862 #ifdef CONFIG_MODULES
 863 __printf(2, 3) int nft_request_module(struct net *net, const char *fmt,
 864                                       ...)
 865 {
 866         char module_name[MODULE_NAME_LEN];
 867         struct nftables_pernet *nft_net;
 868         struct nft_module_request *req;
 869         va_list args;
 870         int ret;
 871
 872         va_start(args, fmt);
 873         ret = vsnprintf(module_name, MODULE_NAME_LEN, fmt, args);
 874         va_end(args);
 875         if (ret >= MODULE_NAME_LEN)
 876                 return 0;
 877
 878         nft_net = nft_pernet(net);
 879         list_for_each_entry(req, &nft_net->module_list, list) {
 880                 if (!strcmp(req->module, module_name)) {
 881                         if (req->done)
 882                                 return 0;
 883
 884                         /* A request to load this module already exists. */
 885                         return -EAGAIN;
 886                 }
 887         }
 888
 889         req = kmalloc(sizeof(*req), GFP_KERNEL);
 890         if (!req)
 891                 return -ENOMEM;
 892
 893         req->done = false;
 894         strscpy(req->module, module_name, MODULE_NAME_LEN);
 895         list_add_tail(&req->list, &nft_net->module_list);
 896
 897         return -EAGAIN;
 898 }
 899 EXPORT_SYMBOL_GPL(nft_request_module);
 900 #endif
 901
 902 static void lockdep_nfnl_nft_mutex_not_held(void)
 903 {
 904 #ifdef CONFIG_PROVE_LOCKING
 905         if (debug_locks)
 906                 WARN_ON_ONCE(lockdep_nfnl_is_held(NFNL_SUBSYS_NFTABLES));
 907 #endif
 908 }
 909
 910 static const struct nft_chain_type *
 911 nf_tables_chain_type_lookup(struct net *net, const struct nlattr *nla,
 912                             u8 family, bool autoload)
 913 {
 914         const struct nft_chain_type *type;
 915
 916         type = __nf_tables_chain_type_lookup(nla, family);
 917         if (type != NULL)
 918                 return type;
 919
 920         lockdep_nfnl_nft_mutex_not_held();
 921 #ifdef CONFIG_MODULES
 922         if (autoload) {
 923                 if (nft_request_module(net, "nft-chain-%u-%.*s", family,
 924                                        nla_len(nla),
 925                                        (const char *)nla_data(nla)) == -EAGAIN)
 926                         return ERR_PTR(-EAGAIN);
 927         }
 928 #endif
 929         return ERR_PTR(-ENOENT);
 930 }
 931
 932 static __be16 nft_base_seq(const struct net *net)
 933 {
 934         struct nftables_pernet *nft_net = nft_pernet(net);
 935
 936         return htons(nft_net->base_seq & 0xffff);
 937 }
 938
 939 static const struct nla_policy nft_table_policy[NFTA_TABLE_MAX + 1] = {
 940         [NFTA_TABLE_NAME]       = { .type = NLA_STRING,
 941                                     .len = NFT_TABLE_MAXNAMELEN - 1 },
 942         [NFTA_TABLE_FLAGS]      = { .type = NLA_U32 },
 943         [NFTA_TABLE_HANDLE]     = { .type = NLA_U64 },
 944         [NFTA_TABLE_USERDATA]   = { .type = NLA_BINARY,
 945                                     .len = NFT_USERDATA_MAXLEN }
 946 };
 947
 948 static int nf_tables_fill_table_info(struct sk_buff *skb, struct net *net,
 949                                      u32 portid, u32 seq, int event, u32 flags,
 950                                      int family, const struct nft_table *table)
 951 {
 952         struct nlmsghdr *nlh;
 953
 954         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
 955         nlh = nfnl_msg_put(skb, portid, seq, event, flags, family,
 956                            NFNETLINK_V0, nft_base_seq(net));
 957         if (!nlh)
 958                 goto nla_put_failure;
 959
 960         if (nla_put_string(skb, NFTA_TABLE_NAME, table->name) ||
 961             nla_put_be32(skb, NFTA_TABLE_USE, htonl(table->use)) ||
 962             nla_put_be64(skb, NFTA_TABLE_HANDLE, cpu_to_be64(table->handle),
 963                          NFTA_TABLE_PAD))
 964                 goto nla_put_failure;
 965
 966         if (event == NFT_MSG_DELTABLE) {
 967                 nlmsg_end(skb, nlh);
 968                 return 0;
 969         }
 970
 971         if (nla_put_be32(skb, NFTA_TABLE_FLAGS,
 972                          htonl(table->flags & NFT_TABLE_F_MASK)))
 973                 goto nla_put_failure;
 974
 975         if (nft_table_has_owner(table) &&
 976             nla_put_be32(skb, NFTA_TABLE_OWNER, htonl(table->nlpid)))
 977                 goto nla_put_failure;
 978
 979         if (table->udata) {
 980                 if (nla_put(skb, NFTA_TABLE_USERDATA, table->udlen, table->udata))
 981                         goto nla_put_failure;
 982         }
 983
 984         nlmsg_end(skb, nlh);
 985         return 0;
 986
 987 nla_put_failure:
 988         nlmsg_trim(skb, nlh);
 989         return -1;
 990 }
 991
 992 struct nftnl_skb_parms {
 993         bool report;
 994 };
 995 #define NFT_CB(skb)     (*(struct nftnl_skb_parms*)&((skb)->cb))
 996
 997 static void nft_notify_enqueue(struct sk_buff *skb, bool report,
 998                                struct list_head *notify_list)
 999 {
1000         NFT_CB(skb).report = report;
1001         list_add_tail(&skb->list, notify_list);
1002 }
1003
1004 static void nf_tables_table_notify(const struct nft_ctx *ctx, int event)
1005 {
1006         struct nftables_pernet *nft_net;
1007         struct sk_buff *skb;
1008         u16 flags = 0;
1009         int err;
1010
1011         if (!ctx->report &&
1012             !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
1013                 return;
1014
1015         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
1016         if (skb == NULL)
1017                 goto err;
1018
1019         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
1020                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
1021
1022         err = nf_tables_fill_table_info(skb, ctx->net, ctx->portid, ctx->seq,
1023                                         event, flags, ctx->family, ctx->table);
1024         if (err < 0) {
1025                 kfree_skb(skb);
1026                 goto err;
1027         }
1028
1029         nft_net = nft_pernet(ctx->net);
1030         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
1031         return;
1032 err:
1033         nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES, -ENOBUFS);
1034 }
1035
1036 static int nf_tables_dump_tables(struct sk_buff *skb,
1037                                  struct netlink_callback *cb)
1038 {
1039         const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
1040         struct nftables_pernet *nft_net;
1041         const struct nft_table *table;
1042         unsigned int idx = 0, s_idx = cb->args[0];
1043         struct net *net = sock_net(skb->sk);
1044         int family = nfmsg->nfgen_family;
1045
1046         rcu_read_lock();
1047         nft_net = nft_pernet(net);
1048         cb->seq = READ_ONCE(nft_net->base_seq);
1049
1050         list_for_each_entry_rcu(table, &nft_net->tables, list) {
1051                 if (family != NFPROTO_UNSPEC && family != table->family)
1052                         continue;
1053
1054                 if (idx < s_idx)
1055                         goto cont;
1056                 if (idx > s_idx)
1057                         memset(&cb->args[1], 0,
1058                                sizeof(cb->args) - sizeof(cb->args[0]));
1059                 if (!nft_is_active(net, table))
1060                         continue;
1061                 if (nf_tables_fill_table_info(skb, net,
1062                                               NETLINK_CB(cb->skb).portid,
1063                                               cb->nlh->nlmsg_seq,
1064                                               NFT_MSG_NEWTABLE, NLM_F_MULTI,
1065                                               table->family, table) < 0)
1066                         goto done;
1067
1068                 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
1069 cont:
1070                 idx++;
1071         }
1072 done:
1073         rcu_read_unlock();
1074         cb->args[0] = idx;
1075         return skb->len;
1076 }
1077
1078 static int nft_netlink_dump_start_rcu(struct sock *nlsk, struct sk_buff *skb,
1079                                       const struct nlmsghdr *nlh,
1080                                       struct netlink_dump_control *c)
1081 {
1082         int err;
1083
1084         if (!try_module_get(THIS_MODULE))
1085                 return -EINVAL;
1086
1087         rcu_read_unlock();
1088         err = netlink_dump_start(nlsk, skb, nlh, c);
1089         rcu_read_lock();
1090         module_put(THIS_MODULE);
1091
1092         return err;
1093 }
1094
1095 /* called with rcu_read_lock held */
1096 static int nf_tables_gettable(struct sk_buff *skb, const struct nfnl_info *info,
1097                               const struct nlattr * const nla[])
1098 {
1099         struct netlink_ext_ack *extack = info->extack;
1100         u8 genmask = nft_genmask_cur(info->net);
1101         u8 family = info->nfmsg->nfgen_family;
1102         const struct nft_table *table;
1103         struct net *net = info->net;
1104         struct sk_buff *skb2;
1105         int err;
1106
1107         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
1108                 struct netlink_dump_control c = {
1109                         .dump = nf_tables_dump_tables,
1110                         .module = THIS_MODULE,
1111                 };
1112
1113                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
1114         }
1115
1116         table = nft_table_lookup(net, nla[NFTA_TABLE_NAME], family, genmask, 0);
1117         if (IS_ERR(table)) {
1118                 NL_SET_BAD_ATTR(extack, nla[NFTA_TABLE_NAME]);
1119                 return PTR_ERR(table);
1120         }
1121
1122         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
1123         if (!skb2)
1124                 return -ENOMEM;
1125
1126         err = nf_tables_fill_table_info(skb2, net, NETLINK_CB(skb).portid,
1127                                         info->nlh->nlmsg_seq, NFT_MSG_NEWTABLE,
1128                                         0, family, table);
1129         if (err < 0)
1130                 goto err_fill_table_info;
1131
1132         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
1133
1134 err_fill_table_info:
1135         kfree_skb(skb2);
1136         return err;
1137 }
1138
1139 static void nft_table_disable(struct net *net, struct nft_table *table, u32 cnt)
1140 {
1141         struct nft_chain *chain;
1142         u32 i = 0;
1143
1144         list_for_each_entry(chain, &table->chains, list) {
1145                 if (!nft_is_active_next(net, chain))
1146                         continue;
1147                 if (!nft_is_base_chain(chain))
1148                         continue;
1149
1150                 if (cnt && i++ == cnt)
1151                         break;
1152
1153                 nf_tables_unregister_hook(net, table, chain);
1154         }
1155 }
1156
1157 static int nf_tables_table_enable(struct net *net, struct nft_table *table)
1158 {
1159         struct nft_chain *chain;
1160         int err, i = 0;
1161
1162         list_for_each_entry(chain, &table->chains, list) {
1163                 if (!nft_is_active_next(net, chain))
1164                         continue;
1165                 if (!nft_is_base_chain(chain))
1166                         continue;
1167
1168                 err = nf_tables_register_hook(net, table, chain);
1169                 if (err < 0)
1170                         goto err_register_hooks;
1171
1172                 i++;
1173         }
1174         return 0;
1175
1176 err_register_hooks:
1177         if (i)
1178                 nft_table_disable(net, table, i);
1179         return err;
1180 }
1181
1182 static void nf_tables_table_disable(struct net *net, struct nft_table *table)
1183 {
1184         table->flags &= ~NFT_TABLE_F_DORMANT;
1185         nft_table_disable(net, table, 0);
1186         table->flags |= NFT_TABLE_F_DORMANT;
1187 }
1188
1189 #define __NFT_TABLE_F_INTERNAL          (NFT_TABLE_F_MASK + 1)
1190 #define __NFT_TABLE_F_WAS_DORMANT       (__NFT_TABLE_F_INTERNAL << 0)
1191 #define __NFT_TABLE_F_WAS_AWAKEN        (__NFT_TABLE_F_INTERNAL << 1)
1192 #define __NFT_TABLE_F_UPDATE            (__NFT_TABLE_F_WAS_DORMANT | \
1193                                          __NFT_TABLE_F_WAS_AWAKEN)
1194
1195 static int nf_tables_updtable(struct nft_ctx *ctx)
1196 {
1197         struct nft_trans *trans;
1198         u32 flags;
1199         int ret;
1200
1201         if (!ctx->nla[NFTA_TABLE_FLAGS])
1202                 return 0;
1203
1204         flags = ntohl(nla_get_be32(ctx->nla[NFTA_TABLE_FLAGS]));
1205         if (flags & ~NFT_TABLE_F_MASK)
1206                 return -EOPNOTSUPP;
1207
1208         if (flags == ctx->table->flags)
1209                 return 0;
1210
1211         if ((nft_table_has_owner(ctx->table) &&
1212              !(flags & NFT_TABLE_F_OWNER)) ||
1213             (!nft_table_has_owner(ctx->table) &&
1214              flags & NFT_TABLE_F_OWNER))
1215                 return -EOPNOTSUPP;
1216
1217         trans = nft_trans_alloc(ctx, NFT_MSG_NEWTABLE,
1218                                 sizeof(struct nft_trans_table));
1219         if (trans == NULL)
1220                 return -ENOMEM;
1221
1222         if ((flags & NFT_TABLE_F_DORMANT) &&
1223             !(ctx->table->flags & NFT_TABLE_F_DORMANT)) {
1224                 ctx->table->flags |= NFT_TABLE_F_DORMANT;
1225                 if (!(ctx->table->flags & __NFT_TABLE_F_UPDATE))
1226                         ctx->table->flags |= __NFT_TABLE_F_WAS_AWAKEN;
1227         } else if (!(flags & NFT_TABLE_F_DORMANT) &&
1228                    ctx->table->flags & NFT_TABLE_F_DORMANT) {
1229                 ctx->table->flags &= ~NFT_TABLE_F_DORMANT;
1230                 if (!(ctx->table->flags & __NFT_TABLE_F_UPDATE)) {
1231                         ret = nf_tables_table_enable(ctx->net, ctx->table);
1232                         if (ret < 0)
1233                                 goto err_register_hooks;
1234
1235                         ctx->table->flags |= __NFT_TABLE_F_WAS_DORMANT;
1236                 }
1237         }
1238
1239         nft_trans_table_update(trans) = true;
1240         nft_trans_commit_list_add_tail(ctx->net, trans);
1241
1242         return 0;
1243
1244 err_register_hooks:
1245         nft_trans_destroy(trans);
1246         return ret;
1247 }
1248
1249 static u32 nft_chain_hash(const void *data, u32 len, u32 seed)
1250 {
1251         const char *name = data;
1252
1253         return jhash(name, strlen(name), seed);
1254 }
1255
1256 static u32 nft_chain_hash_obj(const void *data, u32 len, u32 seed)
1257 {
1258         const struct nft_chain *chain = data;
1259
1260         return nft_chain_hash(chain->name, 0, seed);
1261 }
1262
1263 static int nft_chain_hash_cmp(struct rhashtable_compare_arg *arg,
1264                               const void *ptr)
1265 {
1266         const struct nft_chain *chain = ptr;
1267         const char *name = arg->key;
1268
1269         return strcmp(chain->name, name);
1270 }
1271
1272 static u32 nft_objname_hash(const void *data, u32 len, u32 seed)
1273 {
1274         const struct nft_object_hash_key *k = data;
1275
1276         seed ^= hash_ptr(k->table, 32);
1277
1278         return jhash(k->name, strlen(k->name), seed);
1279 }
1280
1281 static u32 nft_objname_hash_obj(const void *data, u32 len, u32 seed)
1282 {
1283         const struct nft_object *obj = data;
1284
1285         return nft_objname_hash(&obj->key, 0, seed);
1286 }
1287
1288 static int nft_objname_hash_cmp(struct rhashtable_compare_arg *arg,
1289                                 const void *ptr)
1290 {
1291         const struct nft_object_hash_key *k = arg->key;
1292         const struct nft_object *obj = ptr;
1293
1294         if (obj->key.table != k->table)
1295                 return -1;
1296
1297         return strcmp(obj->key.name, k->name);
1298 }
1299
1300 static bool nft_supported_family(u8 family)
1301 {
1302         return false
1303 #ifdef CONFIG_NF_TABLES_INET
1304                 || family == NFPROTO_INET
1305 #endif
1306 #ifdef CONFIG_NF_TABLES_IPV4
1307                 || family == NFPROTO_IPV4
1308 #endif
1309 #ifdef CONFIG_NF_TABLES_ARP
1310                 || family == NFPROTO_ARP
1311 #endif
1312 #ifdef CONFIG_NF_TABLES_NETDEV
1313                 || family == NFPROTO_NETDEV
1314 #endif
1315 #if IS_ENABLED(CONFIG_NF_TABLES_BRIDGE)
1316                 || family == NFPROTO_BRIDGE
1317 #endif
1318 #ifdef CONFIG_NF_TABLES_IPV6
1319                 || family == NFPROTO_IPV6
1320 #endif
1321                 ;
1322 }
1323
1324 static int nf_tables_newtable(struct sk_buff *skb, const struct nfnl_info *info,
1325                               const struct nlattr * const nla[])
1326 {
1327         struct nftables_pernet *nft_net = nft_pernet(info->net);
1328         struct netlink_ext_ack *extack = info->extack;
1329         u8 genmask = nft_genmask_next(info->net);
1330         u8 family = info->nfmsg->nfgen_family;
1331         struct net *net = info->net;
1332         const struct nlattr *attr;
1333         struct nft_table *table;
1334         struct nft_ctx ctx;
1335         u32 flags = 0;
1336         int err;
1337
1338         if (!nft_supported_family(family))
1339                 return -EOPNOTSUPP;
1340
1341         lockdep_assert_held(&nft_net->commit_mutex);
1342         attr = nla[NFTA_TABLE_NAME];
1343         table = nft_table_lookup(net, attr, family, genmask,
1344                                  NETLINK_CB(skb).portid);
1345         if (IS_ERR(table)) {
1346                 if (PTR_ERR(table) != -ENOENT)
1347                         return PTR_ERR(table);
1348         } else {
1349                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
1350                         NL_SET_BAD_ATTR(extack, attr);
1351                         return -EEXIST;
1352                 }
1353                 if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
1354                         return -EOPNOTSUPP;
1355
1356                 nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
1357
1358                 return nf_tables_updtable(&ctx);
1359         }
1360
1361         if (nla[NFTA_TABLE_FLAGS]) {
1362                 flags = ntohl(nla_get_be32(nla[NFTA_TABLE_FLAGS]));
1363                 if (flags & ~NFT_TABLE_F_MASK)
1364                         return -EOPNOTSUPP;
1365         }
1366
1367         err = -ENOMEM;
1368         table = kzalloc(sizeof(*table), GFP_KERNEL_ACCOUNT);
1369         if (table == NULL)
1370                 goto err_kzalloc;
1371
1372         table->validate_state = NFT_VALIDATE_SKIP;
1373         table->name = nla_strdup(attr, GFP_KERNEL_ACCOUNT);
1374         if (table->name == NULL)
1375                 goto err_strdup;
1376
1377         if (nla[NFTA_TABLE_USERDATA]) {
1378                 table->udata = nla_memdup(nla[NFTA_TABLE_USERDATA], GFP_KERNEL_ACCOUNT);
1379                 if (table->udata == NULL)
1380                         goto err_table_udata;
1381
1382                 table->udlen = nla_len(nla[NFTA_TABLE_USERDATA]);
1383         }
1384
1385         err = rhltable_init(&table->chains_ht, &nft_chain_ht_params);
1386         if (err)
1387                 goto err_chain_ht;
1388
1389         INIT_LIST_HEAD(&table->chains);
1390         INIT_LIST_HEAD(&table->sets);
1391         INIT_LIST_HEAD(&table->objects);
1392         INIT_LIST_HEAD(&table->flowtables);
1393         table->family = family;
1394         table->flags = flags;
1395         table->handle = ++nft_net->table_handle;
1396         if (table->flags & NFT_TABLE_F_OWNER)
1397                 table->nlpid = NETLINK_CB(skb).portid;
1398
1399         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
1400         err = nft_trans_table_add(&ctx, NFT_MSG_NEWTABLE);
1401         if (err < 0)
1402                 goto err_trans;
1403
1404         list_add_tail_rcu(&table->list, &nft_net->tables);
1405         return 0;
1406 err_trans:
1407         rhltable_destroy(&table->chains_ht);
1408 err_chain_ht:
1409         kfree(table->udata);
1410 err_table_udata:
1411         kfree(table->name);
1412 err_strdup:
1413         kfree(table);
1414 err_kzalloc:
1415         return err;
1416 }
1417
1418 static int nft_flush_table(struct nft_ctx *ctx)
1419 {
1420         struct nft_flowtable *flowtable, *nft;
1421         struct nft_chain *chain, *nc;
1422         struct nft_object *obj, *ne;
1423         struct nft_set *set, *ns;
1424         int err;
1425
1426         list_for_each_entry(chain, &ctx->table->chains, list) {
1427                 if (!nft_is_active_next(ctx->net, chain))
1428                         continue;
1429
1430                 if (nft_chain_is_bound(chain))
1431                         continue;
1432
1433                 ctx->chain = chain;
1434
1435                 err = nft_delrule_by_chain(ctx);
1436                 if (err < 0)
1437                         goto out;
1438         }
1439
1440         list_for_each_entry_safe(set, ns, &ctx->table->sets, list) {
1441                 if (!nft_is_active_next(ctx->net, set))
1442                         continue;
1443
1444                 if (nft_set_is_anonymous(set) &&
1445                     !list_empty(&set->bindings))
1446                         continue;
1447
1448                 err = nft_delset(ctx, set);
1449                 if (err < 0)
1450                         goto out;
1451         }
1452
1453         list_for_each_entry_safe(flowtable, nft, &ctx->table->flowtables, list) {
1454                 if (!nft_is_active_next(ctx->net, flowtable))
1455                         continue;
1456
1457                 err = nft_delflowtable(ctx, flowtable);
1458                 if (err < 0)
1459                         goto out;
1460         }
1461
1462         list_for_each_entry_safe(obj, ne, &ctx->table->objects, list) {
1463                 if (!nft_is_active_next(ctx->net, obj))
1464                         continue;
1465
1466                 err = nft_delobj(ctx, obj);
1467                 if (err < 0)
1468                         goto out;
1469         }
1470
1471         list_for_each_entry_safe(chain, nc, &ctx->table->chains, list) {
1472                 if (!nft_is_active_next(ctx->net, chain))
1473                         continue;
1474
1475                 if (nft_chain_is_bound(chain))
1476                         continue;
1477
1478                 ctx->chain = chain;
1479
1480                 err = nft_delchain(ctx);
1481                 if (err < 0)
1482                         goto out;
1483         }
1484
1485         err = nft_deltable(ctx);
1486 out:
1487         return err;
1488 }
1489
1490 static int nft_flush(struct nft_ctx *ctx, int family)
1491 {
1492         struct nftables_pernet *nft_net = nft_pernet(ctx->net);
1493         const struct nlattr * const *nla = ctx->nla;
1494         struct nft_table *table, *nt;
1495         int err = 0;
1496
1497         list_for_each_entry_safe(table, nt, &nft_net->tables, list) {
1498                 if (family != AF_UNSPEC && table->family != family)
1499                         continue;
1500
1501                 ctx->family = table->family;
1502
1503                 if (!nft_is_active_next(ctx->net, table))
1504                         continue;
1505
1506                 if (nft_table_has_owner(table) && table->nlpid != ctx->portid)
1507                         continue;
1508
1509                 if (nla[NFTA_TABLE_NAME] &&
1510                     nla_strcmp(nla[NFTA_TABLE_NAME], table->name) != 0)
1511                         continue;
1512
1513                 ctx->table = table;
1514
1515                 err = nft_flush_table(ctx);
1516                 if (err < 0)
1517                         goto out;
1518         }
1519 out:
1520         return err;
1521 }
1522
1523 static int nf_tables_deltable(struct sk_buff *skb, const struct nfnl_info *info,
1524                               const struct nlattr * const nla[])
1525 {
1526         struct netlink_ext_ack *extack = info->extack;
1527         u8 genmask = nft_genmask_next(info->net);
1528         u8 family = info->nfmsg->nfgen_family;
1529         struct net *net = info->net;
1530         const struct nlattr *attr;
1531         struct nft_table *table;
1532         struct nft_ctx ctx;
1533
1534         nft_ctx_init(&ctx, net, skb, info->nlh, 0, NULL, NULL, nla);
1535         if (family == AF_UNSPEC ||
1536             (!nla[NFTA_TABLE_NAME] && !nla[NFTA_TABLE_HANDLE]))
1537                 return nft_flush(&ctx, family);
1538
1539         if (nla[NFTA_TABLE_HANDLE]) {
1540                 attr = nla[NFTA_TABLE_HANDLE];
1541                 table = nft_table_lookup_byhandle(net, attr, genmask,
1542                                                   NETLINK_CB(skb).portid);
1543         } else {
1544                 attr = nla[NFTA_TABLE_NAME];
1545                 table = nft_table_lookup(net, attr, family, genmask,
1546                                          NETLINK_CB(skb).portid);
1547         }
1548
1549         if (IS_ERR(table)) {
1550                 if (PTR_ERR(table) == -ENOENT &&
1551                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYTABLE)
1552                         return 0;
1553
1554                 NL_SET_BAD_ATTR(extack, attr);
1555                 return PTR_ERR(table);
1556         }
1557
1558         if (info->nlh->nlmsg_flags & NLM_F_NONREC &&
1559             table->use > 0)
1560                 return -EBUSY;
1561
1562         ctx.family = family;
1563         ctx.table = table;
1564
1565         return nft_flush_table(&ctx);
1566 }
1567
1568 static void nf_tables_table_destroy(struct nft_ctx *ctx)
1569 {
1570         if (WARN_ON(ctx->table->use > 0))
1571                 return;
1572
1573         rhltable_destroy(&ctx->table->chains_ht);
1574         kfree(ctx->table->name);
1575         kfree(ctx->table->udata);
1576         kfree(ctx->table);
1577 }
1578
1579 void nft_register_chain_type(const struct nft_chain_type *ctype)
1580 {
1581         nfnl_lock(NFNL_SUBSYS_NFTABLES);
1582         if (WARN_ON(__nft_chain_type_get(ctype->family, ctype->type))) {
1583                 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1584                 return;
1585         }
1586         chain_type[ctype->family][ctype->type] = ctype;
1587         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1588 }
1589 EXPORT_SYMBOL_GPL(nft_register_chain_type);
1590
1591 void nft_unregister_chain_type(const struct nft_chain_type *ctype)
1592 {
1593         nfnl_lock(NFNL_SUBSYS_NFTABLES);
1594         chain_type[ctype->family][ctype->type] = NULL;
1595         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1596 }
1597 EXPORT_SYMBOL_GPL(nft_unregister_chain_type);
1598
1599 /*
1600  * Chains
1601  */
1602
1603 static struct nft_chain *
1604 nft_chain_lookup_byhandle(const struct nft_table *table, u64 handle, u8 genmask)
1605 {
1606         struct nft_chain *chain;
1607
1608         list_for_each_entry(chain, &table->chains, list) {
1609                 if (chain->handle == handle &&
1610                     nft_active_genmask(chain, genmask))
1611                         return chain;
1612         }
1613
1614         return ERR_PTR(-ENOENT);
1615 }
1616
1617 static bool lockdep_commit_lock_is_held(const struct net *net)
1618 {
1619 #ifdef CONFIG_PROVE_LOCKING
1620         struct nftables_pernet *nft_net = nft_pernet(net);
1621
1622         return lockdep_is_held(&nft_net->commit_mutex);
1623 #else
1624         return true;
1625 #endif
1626 }
1627
1628 static struct nft_chain *nft_chain_lookup(struct net *net,
1629                                           struct nft_table *table,
1630                                           const struct nlattr *nla, u8 genmask)
1631 {
1632         char search[NFT_CHAIN_MAXNAMELEN + 1];
1633         struct rhlist_head *tmp, *list;
1634         struct nft_chain *chain;
1635
1636         if (nla == NULL)
1637                 return ERR_PTR(-EINVAL);
1638
1639         nla_strscpy(search, nla, sizeof(search));
1640
1641         WARN_ON(!rcu_read_lock_held() &&
1642                 !lockdep_commit_lock_is_held(net));
1643
1644         chain = ERR_PTR(-ENOENT);
1645         rcu_read_lock();
1646         list = rhltable_lookup(&table->chains_ht, search, nft_chain_ht_params);
1647         if (!list)
1648                 goto out_unlock;
1649
1650         rhl_for_each_entry_rcu(chain, tmp, list, rhlhead) {
1651                 if (nft_active_genmask(chain, genmask))
1652                         goto out_unlock;
1653         }
1654         chain = ERR_PTR(-ENOENT);
1655 out_unlock:
1656         rcu_read_unlock();
1657         return chain;
1658 }
1659
1660 static const struct nla_policy nft_chain_policy[NFTA_CHAIN_MAX + 1] = {
1661         [NFTA_CHAIN_TABLE]      = { .type = NLA_STRING,
1662                                     .len = NFT_TABLE_MAXNAMELEN - 1 },
1663         [NFTA_CHAIN_HANDLE]     = { .type = NLA_U64 },
1664         [NFTA_CHAIN_NAME]       = { .type = NLA_STRING,
1665                                     .len = NFT_CHAIN_MAXNAMELEN - 1 },
1666         [NFTA_CHAIN_HOOK]       = { .type = NLA_NESTED },
1667         [NFTA_CHAIN_POLICY]     = { .type = NLA_U32 },
1668         [NFTA_CHAIN_TYPE]       = { .type = NLA_STRING,
1669                                     .len = NFT_MODULE_AUTOLOAD_LIMIT },
1670         [NFTA_CHAIN_COUNTERS]   = { .type = NLA_NESTED },
1671         [NFTA_CHAIN_FLAGS]      = { .type = NLA_U32 },
1672         [NFTA_CHAIN_ID]         = { .type = NLA_U32 },
1673         [NFTA_CHAIN_USERDATA]   = { .type = NLA_BINARY,
1674                                     .len = NFT_USERDATA_MAXLEN },
1675 };
1676
1677 static const struct nla_policy nft_hook_policy[NFTA_HOOK_MAX + 1] = {
1678         [NFTA_HOOK_HOOKNUM]     = { .type = NLA_U32 },
1679         [NFTA_HOOK_PRIORITY]    = { .type = NLA_U32 },
1680         [NFTA_HOOK_DEV]         = { .type = NLA_STRING,
1681                                     .len = IFNAMSIZ - 1 },
1682 };
1683
1684 static int nft_dump_stats(struct sk_buff *skb, struct nft_stats __percpu *stats)
1685 {
1686         struct nft_stats *cpu_stats, total;
1687         struct nlattr *nest;
1688         unsigned int seq;
1689         u64 pkts, bytes;
1690         int cpu;
1691
1692         if (!stats)
1693                 return 0;
1694
1695         memset(&total, 0, sizeof(total));
1696         for_each_possible_cpu(cpu) {
1697                 cpu_stats = per_cpu_ptr(stats, cpu);
1698                 do {
1699                         seq = u64_stats_fetch_begin(&cpu_stats->syncp);
1700                         pkts = cpu_stats->pkts;
1701                         bytes = cpu_stats->bytes;
1702                 } while (u64_stats_fetch_retry(&cpu_stats->syncp, seq));
1703                 total.pkts += pkts;
1704                 total.bytes += bytes;
1705         }
1706         nest = nla_nest_start_noflag(skb, NFTA_CHAIN_COUNTERS);
1707         if (nest == NULL)
1708                 goto nla_put_failure;
1709
1710         if (nla_put_be64(skb, NFTA_COUNTER_PACKETS, cpu_to_be64(total.pkts),
1711                          NFTA_COUNTER_PAD) ||
1712             nla_put_be64(skb, NFTA_COUNTER_BYTES, cpu_to_be64(total.bytes),
1713                          NFTA_COUNTER_PAD))
1714                 goto nla_put_failure;
1715
1716         nla_nest_end(skb, nest);
1717         return 0;
1718
1719 nla_put_failure:
1720         return -ENOSPC;
1721 }
1722
1723 static int nft_dump_basechain_hook(struct sk_buff *skb, int family,
1724                                    const struct nft_base_chain *basechain,
1725                                    const struct list_head *hook_list)
1726 {
1727         const struct nf_hook_ops *ops = &basechain->ops;
1728         struct nft_hook *hook, *first = NULL;
1729         struct nlattr *nest, *nest_devs;
1730         int n = 0;
1731
1732         nest = nla_nest_start_noflag(skb, NFTA_CHAIN_HOOK);
1733         if (nest == NULL)
1734                 goto nla_put_failure;
1735         if (nla_put_be32(skb, NFTA_HOOK_HOOKNUM, htonl(ops->hooknum)))
1736                 goto nla_put_failure;
1737         if (nla_put_be32(skb, NFTA_HOOK_PRIORITY, htonl(ops->priority)))
1738                 goto nla_put_failure;
1739
1740         if (nft_base_chain_netdev(family, ops->hooknum)) {
1741                 nest_devs = nla_nest_start_noflag(skb, NFTA_HOOK_DEVS);
1742                 if (!nest_devs)
1743                         goto nla_put_failure;
1744
1745                 if (!hook_list)
1746                         hook_list = &basechain->hook_list;
1747
1748                 list_for_each_entry(hook, hook_list, list) {
1749                         if (!first)
1750                                 first = hook;
1751
1752                         if (nla_put_string(skb, NFTA_DEVICE_NAME,
1753                                            hook->ops.dev->name))
1754                                 goto nla_put_failure;
1755                         n++;
1756                 }
1757                 nla_nest_end(skb, nest_devs);
1758
1759                 if (n == 1 &&
1760                     nla_put_string(skb, NFTA_HOOK_DEV, first->ops.dev->name))
1761                         goto nla_put_failure;
1762         }
1763         nla_nest_end(skb, nest);
1764
1765         return 0;
1766 nla_put_failure:
1767         return -1;
1768 }
1769
1770 static int nf_tables_fill_chain_info(struct sk_buff *skb, struct net *net,
1771                                      u32 portid, u32 seq, int event, u32 flags,
1772                                      int family, const struct nft_table *table,
1773                                      const struct nft_chain *chain,
1774                                      const struct list_head *hook_list)
1775 {
1776         struct nlmsghdr *nlh;
1777
1778         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
1779         nlh = nfnl_msg_put(skb, portid, seq, event, flags, family,
1780                            NFNETLINK_V0, nft_base_seq(net));
1781         if (!nlh)
1782                 goto nla_put_failure;
1783
1784         if (nla_put_string(skb, NFTA_CHAIN_TABLE, table->name) ||
1785             nla_put_string(skb, NFTA_CHAIN_NAME, chain->name) ||
1786             nla_put_be64(skb, NFTA_CHAIN_HANDLE, cpu_to_be64(chain->handle),
1787                          NFTA_CHAIN_PAD))
1788                 goto nla_put_failure;
1789
1790         if (event == NFT_MSG_DELCHAIN && !hook_list) {
1791                 nlmsg_end(skb, nlh);
1792                 return 0;
1793         }
1794
1795         if (nft_is_base_chain(chain)) {
1796                 const struct nft_base_chain *basechain = nft_base_chain(chain);
1797                 struct nft_stats __percpu *stats;
1798
1799                 if (nft_dump_basechain_hook(skb, family, basechain, hook_list))
1800                         goto nla_put_failure;
1801
1802                 if (nla_put_be32(skb, NFTA_CHAIN_POLICY,
1803                                  htonl(basechain->policy)))
1804                         goto nla_put_failure;
1805
1806                 if (nla_put_string(skb, NFTA_CHAIN_TYPE, basechain->type->name))
1807                         goto nla_put_failure;
1808
1809                 stats = rcu_dereference_check(basechain->stats,
1810                                               lockdep_commit_lock_is_held(net));
1811                 if (nft_dump_stats(skb, stats))
1812                         goto nla_put_failure;
1813         }
1814
1815         if (chain->flags &&
1816             nla_put_be32(skb, NFTA_CHAIN_FLAGS, htonl(chain->flags)))
1817                 goto nla_put_failure;
1818
1819         if (nla_put_be32(skb, NFTA_CHAIN_USE, htonl(chain->use)))
1820                 goto nla_put_failure;
1821
1822         if (chain->udata &&
1823             nla_put(skb, NFTA_CHAIN_USERDATA, chain->udlen, chain->udata))
1824                 goto nla_put_failure;
1825
1826         nlmsg_end(skb, nlh);
1827         return 0;
1828
1829 nla_put_failure:
1830         nlmsg_trim(skb, nlh);
1831         return -1;
1832 }
1833
1834 static void nf_tables_chain_notify(const struct nft_ctx *ctx, int event,
1835                                    const struct list_head *hook_list)
1836 {
1837         struct nftables_pernet *nft_net;
1838         struct sk_buff *skb;
1839         u16 flags = 0;
1840         int err;
1841
1842         if (!ctx->report &&
1843             !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
1844                 return;
1845
1846         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
1847         if (skb == NULL)
1848                 goto err;
1849
1850         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
1851                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
1852
1853         err = nf_tables_fill_chain_info(skb, ctx->net, ctx->portid, ctx->seq,
1854                                         event, flags, ctx->family, ctx->table,
1855                                         ctx->chain, hook_list);
1856         if (err < 0) {
1857                 kfree_skb(skb);
1858                 goto err;
1859         }
1860
1861         nft_net = nft_pernet(ctx->net);
1862         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
1863         return;
1864 err:
1865         nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES, -ENOBUFS);
1866 }
1867
1868 static int nf_tables_dump_chains(struct sk_buff *skb,
1869                                  struct netlink_callback *cb)
1870 {
1871         const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
1872         unsigned int idx = 0, s_idx = cb->args[0];
1873         struct net *net = sock_net(skb->sk);
1874         int family = nfmsg->nfgen_family;
1875         struct nftables_pernet *nft_net;
1876         const struct nft_table *table;
1877         const struct nft_chain *chain;
1878
1879         rcu_read_lock();
1880         nft_net = nft_pernet(net);
1881         cb->seq = READ_ONCE(nft_net->base_seq);
1882
1883         list_for_each_entry_rcu(table, &nft_net->tables, list) {
1884                 if (family != NFPROTO_UNSPEC && family != table->family)
1885                         continue;
1886
1887                 list_for_each_entry_rcu(chain, &table->chains, list) {
1888                         if (idx < s_idx)
1889                                 goto cont;
1890                         if (idx > s_idx)
1891                                 memset(&cb->args[1], 0,
1892                                        sizeof(cb->args) - sizeof(cb->args[0]));
1893                         if (!nft_is_active(net, chain))
1894                                 continue;
1895                         if (nf_tables_fill_chain_info(skb, net,
1896                                                       NETLINK_CB(cb->skb).portid,
1897                                                       cb->nlh->nlmsg_seq,
1898                                                       NFT_MSG_NEWCHAIN,
1899                                                       NLM_F_MULTI,
1900                                                       table->family, table,
1901                                                       chain, NULL) < 0)
1902                                 goto done;
1903
1904                         nl_dump_check_consistent(cb, nlmsg_hdr(skb));
1905 cont:
1906                         idx++;
1907                 }
1908         }
1909 done:
1910         rcu_read_unlock();
1911         cb->args[0] = idx;
1912         return skb->len;
1913 }
1914
1915 /* called with rcu_read_lock held */
1916 static int nf_tables_getchain(struct sk_buff *skb, const struct nfnl_info *info,
1917                               const struct nlattr * const nla[])
1918 {
1919         struct netlink_ext_ack *extack = info->extack;
1920         u8 genmask = nft_genmask_cur(info->net);
1921         u8 family = info->nfmsg->nfgen_family;
1922         const struct nft_chain *chain;
1923         struct net *net = info->net;
1924         struct nft_table *table;
1925         struct sk_buff *skb2;
1926         int err;
1927
1928         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
1929                 struct netlink_dump_control c = {
1930                         .dump = nf_tables_dump_chains,
1931                         .module = THIS_MODULE,
1932                 };
1933
1934                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
1935         }
1936
1937         table = nft_table_lookup(net, nla[NFTA_CHAIN_TABLE], family, genmask, 0);
1938         if (IS_ERR(table)) {
1939                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_TABLE]);
1940                 return PTR_ERR(table);
1941         }
1942
1943         chain = nft_chain_lookup(net, table, nla[NFTA_CHAIN_NAME], genmask);
1944         if (IS_ERR(chain)) {
1945                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_NAME]);
1946                 return PTR_ERR(chain);
1947         }
1948
1949         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
1950         if (!skb2)
1951                 return -ENOMEM;
1952
1953         err = nf_tables_fill_chain_info(skb2, net, NETLINK_CB(skb).portid,
1954                                         info->nlh->nlmsg_seq, NFT_MSG_NEWCHAIN,
1955                                         0, family, table, chain, NULL);
1956         if (err < 0)
1957                 goto err_fill_chain_info;
1958
1959         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
1960
1961 err_fill_chain_info:
1962         kfree_skb(skb2);
1963         return err;
1964 }
1965
1966 static const struct nla_policy nft_counter_policy[NFTA_COUNTER_MAX + 1] = {
1967         [NFTA_COUNTER_PACKETS]  = { .type = NLA_U64 },
1968         [NFTA_COUNTER_BYTES]    = { .type = NLA_U64 },
1969 };
1970
1971 static struct nft_stats __percpu *nft_stats_alloc(const struct nlattr *attr)
1972 {
1973         struct nlattr *tb[NFTA_COUNTER_MAX+1];
1974         struct nft_stats __percpu *newstats;
1975         struct nft_stats *stats;
1976         int err;
1977
1978         err = nla_parse_nested_deprecated(tb, NFTA_COUNTER_MAX, attr,
1979                                           nft_counter_policy, NULL);
1980         if (err < 0)
1981                 return ERR_PTR(err);
1982
1983         if (!tb[NFTA_COUNTER_BYTES] || !tb[NFTA_COUNTER_PACKETS])
1984                 return ERR_PTR(-EINVAL);
1985
1986         newstats = netdev_alloc_pcpu_stats(struct nft_stats);
1987         if (newstats == NULL)
1988                 return ERR_PTR(-ENOMEM);
1989
1990         /* Restore old counters on this cpu, no problem. Per-cpu statistics
1991          * are not exposed to userspace.
1992          */
1993         preempt_disable();
1994         stats = this_cpu_ptr(newstats);
1995         stats->bytes = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_BYTES]));
1996         stats->pkts = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_PACKETS]));
1997         preempt_enable();
1998
1999         return newstats;
2000 }
2001
2002 static void nft_chain_stats_replace(struct nft_trans *trans)
2003 {
2004         struct nft_base_chain *chain = nft_base_chain(trans->ctx.chain);
2005
2006         if (!nft_trans_chain_stats(trans))
2007                 return;
2008
2009         nft_trans_chain_stats(trans) =
2010                 rcu_replace_pointer(chain->stats, nft_trans_chain_stats(trans),
2011                                     lockdep_commit_lock_is_held(trans->ctx.net));
2012
2013         if (!nft_trans_chain_stats(trans))
2014                 static_branch_inc(&nft_counters_enabled);
2015 }
2016
2017 static void nf_tables_chain_free_chain_rules(struct nft_chain *chain)
2018 {
2019         struct nft_rule_blob *g0 = rcu_dereference_raw(chain->blob_gen_0);
2020         struct nft_rule_blob *g1 = rcu_dereference_raw(chain->blob_gen_1);
2021
2022         if (g0 != g1)
2023                 kvfree(g1);
2024         kvfree(g0);
2025
2026         /* should be NULL either via abort or via successful commit */
2027         WARN_ON_ONCE(chain->blob_next);
2028         kvfree(chain->blob_next);
2029 }
2030
2031 void nf_tables_chain_destroy(struct nft_ctx *ctx)
2032 {
2033         struct nft_chain *chain = ctx->chain;
2034         struct nft_hook *hook, *next;
2035
2036         if (WARN_ON(chain->use > 0))
2037                 return;
2038
2039         /* no concurrent access possible anymore */
2040         nf_tables_chain_free_chain_rules(chain);
2041
2042         if (nft_is_base_chain(chain)) {
2043                 struct nft_base_chain *basechain = nft_base_chain(chain);
2044
2045                 if (nft_base_chain_netdev(ctx->family, basechain->ops.hooknum)) {
2046                         list_for_each_entry_safe(hook, next,
2047                                                  &basechain->hook_list, list) {
2048                                 list_del_rcu(&hook->list);
2049                                 kfree_rcu(hook, rcu);
2050                         }
2051                 }
2052                 module_put(basechain->type->owner);
2053                 if (rcu_access_pointer(basechain->stats)) {
2054                         static_branch_dec(&nft_counters_enabled);
2055                         free_percpu(rcu_dereference_raw(basechain->stats));
2056                 }
2057                 kfree(chain->name);
2058                 kfree(chain->udata);
2059                 kfree(basechain);
2060         } else {
2061                 kfree(chain->name);
2062                 kfree(chain->udata);
2063                 kfree(chain);
2064         }
2065 }
2066
2067 static struct nft_hook *nft_netdev_hook_alloc(struct net *net,
2068                                               const struct nlattr *attr)
2069 {
2070         struct net_device *dev;
2071         char ifname[IFNAMSIZ];
2072         struct nft_hook *hook;
2073         int err;
2074
2075         hook = kmalloc(sizeof(struct nft_hook), GFP_KERNEL_ACCOUNT);
2076         if (!hook) {
2077                 err = -ENOMEM;
2078                 goto err_hook_alloc;
2079         }
2080
2081         nla_strscpy(ifname, attr, IFNAMSIZ);
2082         /* nf_tables_netdev_event() is called under rtnl_mutex, this is
2083          * indirectly serializing all the other holders of the commit_mutex with
2084          * the rtnl_mutex.
2085          */
2086         dev = __dev_get_by_name(net, ifname);
2087         if (!dev) {
2088                 err = -ENOENT;
2089                 goto err_hook_dev;
2090         }
2091         hook->ops.dev = dev;
2092
2093         return hook;
2094
2095 err_hook_dev:
2096         kfree(hook);
2097 err_hook_alloc:
2098         return ERR_PTR(err);
2099 }
2100
2101 static struct nft_hook *nft_hook_list_find(struct list_head *hook_list,
2102                                            const struct nft_hook *this)
2103 {
2104         struct nft_hook *hook;
2105
2106         list_for_each_entry(hook, hook_list, list) {
2107                 if (this->ops.dev == hook->ops.dev)
2108                         return hook;
2109         }
2110
2111         return NULL;
2112 }
2113
2114 static int nf_tables_parse_netdev_hooks(struct net *net,
2115                                         const struct nlattr *attr,
2116                                         struct list_head *hook_list,
2117                                         struct netlink_ext_ack *extack)
2118 {
2119         struct nft_hook *hook, *next;
2120         const struct nlattr *tmp;
2121         int rem, n = 0, err;
2122
2123         nla_for_each_nested(tmp, attr, rem) {
2124                 if (nla_type(tmp) != NFTA_DEVICE_NAME) {
2125                         err = -EINVAL;
2126                         goto err_hook;
2127                 }
2128
2129                 hook = nft_netdev_hook_alloc(net, tmp);
2130                 if (IS_ERR(hook)) {
2131                         NL_SET_BAD_ATTR(extack, tmp);
2132                         err = PTR_ERR(hook);
2133                         goto err_hook;
2134                 }
2135                 if (nft_hook_list_find(hook_list, hook)) {
2136                         NL_SET_BAD_ATTR(extack, tmp);
2137                         kfree(hook);
2138                         err = -EEXIST;
2139                         goto err_hook;
2140                 }
2141                 list_add_tail(&hook->list, hook_list);
2142                 n++;
2143
2144                 if (n == NFT_NETDEVICE_MAX) {
2145                         err = -EFBIG;
2146                         goto err_hook;
2147                 }
2148         }
2149
2150         return 0;
2151
2152 err_hook:
2153         list_for_each_entry_safe(hook, next, hook_list, list) {
2154                 list_del(&hook->list);
2155                 kfree(hook);
2156         }
2157         return err;
2158 }
2159
2160 struct nft_chain_hook {
2161         u32                             num;
2162         s32                             priority;
2163         const struct nft_chain_type     *type;
2164         struct list_head                list;
2165 };
2166
2167 static int nft_chain_parse_netdev(struct net *net, struct nlattr *tb[],
2168                                   struct list_head *hook_list,
2169                                   struct netlink_ext_ack *extack, u32 flags)
2170 {
2171         struct nft_hook *hook;
2172         int err;
2173
2174         if (tb[NFTA_HOOK_DEV]) {
2175                 hook = nft_netdev_hook_alloc(net, tb[NFTA_HOOK_DEV]);
2176                 if (IS_ERR(hook)) {
2177                         NL_SET_BAD_ATTR(extack, tb[NFTA_HOOK_DEV]);
2178                         return PTR_ERR(hook);
2179                 }
2180
2181                 list_add_tail(&hook->list, hook_list);
2182         } else if (tb[NFTA_HOOK_DEVS]) {
2183                 err = nf_tables_parse_netdev_hooks(net, tb[NFTA_HOOK_DEVS],
2184                                                    hook_list, extack);
2185                 if (err < 0)
2186                         return err;
2187
2188         }
2189
2190         if (flags & NFT_CHAIN_HW_OFFLOAD &&
2191             list_empty(hook_list))
2192                 return -EINVAL;
2193
2194         return 0;
2195 }
2196
2197 static int nft_chain_parse_hook(struct net *net,
2198                                 struct nft_base_chain *basechain,
2199                                 const struct nlattr * const nla[],
2200                                 struct nft_chain_hook *hook, u8 family,
2201                                 u32 flags, struct netlink_ext_ack *extack)
2202 {
2203         struct nftables_pernet *nft_net = nft_pernet(net);
2204         struct nlattr *ha[NFTA_HOOK_MAX + 1];
2205         const struct nft_chain_type *type;
2206         int err;
2207
2208         lockdep_assert_held(&nft_net->commit_mutex);
2209         lockdep_nfnl_nft_mutex_not_held();
2210
2211         err = nla_parse_nested_deprecated(ha, NFTA_HOOK_MAX,
2212                                           nla[NFTA_CHAIN_HOOK],
2213                                           nft_hook_policy, NULL);
2214         if (err < 0)
2215                 return err;
2216
2217         if (!basechain) {
2218                 if (!ha[NFTA_HOOK_HOOKNUM] ||
2219                     !ha[NFTA_HOOK_PRIORITY]) {
2220                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_NAME]);
2221                         return -ENOENT;
2222                 }
2223
2224                 hook->num = ntohl(nla_get_be32(ha[NFTA_HOOK_HOOKNUM]));
2225                 hook->priority = ntohl(nla_get_be32(ha[NFTA_HOOK_PRIORITY]));
2226
2227                 type = __nft_chain_type_get(family, NFT_CHAIN_T_DEFAULT);
2228                 if (!type)
2229                         return -EOPNOTSUPP;
2230
2231                 if (nla[NFTA_CHAIN_TYPE]) {
2232                         type = nf_tables_chain_type_lookup(net, nla[NFTA_CHAIN_TYPE],
2233                                                            family, true);
2234                         if (IS_ERR(type)) {
2235                                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_TYPE]);
2236                                 return PTR_ERR(type);
2237                         }
2238                 }
2239                 if (hook->num >= NFT_MAX_HOOKS || !(type->hook_mask & (1 << hook->num)))
2240                         return -EOPNOTSUPP;
2241
2242                 if (type->type == NFT_CHAIN_T_NAT &&
2243                     hook->priority <= NF_IP_PRI_CONNTRACK)
2244                         return -EOPNOTSUPP;
2245         } else {
2246                 if (ha[NFTA_HOOK_HOOKNUM]) {
2247                         hook->num = ntohl(nla_get_be32(ha[NFTA_HOOK_HOOKNUM]));
2248                         if (hook->num != basechain->ops.hooknum)
2249                                 return -EOPNOTSUPP;
2250                 }
2251                 if (ha[NFTA_HOOK_PRIORITY]) {
2252                         hook->priority = ntohl(nla_get_be32(ha[NFTA_HOOK_PRIORITY]));
2253                         if (hook->priority != basechain->ops.priority)
2254                                 return -EOPNOTSUPP;
2255                 }
2256
2257                 type = basechain->type;
2258         }
2259
2260         if (!try_module_get(type->owner)) {
2261                 if (nla[NFTA_CHAIN_TYPE])
2262                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_TYPE]);
2263                 return -ENOENT;
2264         }
2265
2266         hook->type = type;
2267
2268         INIT_LIST_HEAD(&hook->list);
2269         if (nft_base_chain_netdev(family, hook->num)) {
2270                 err = nft_chain_parse_netdev(net, ha, &hook->list, extack, flags);
2271                 if (err < 0) {
2272                         module_put(type->owner);
2273                         return err;
2274                 }
2275         } else if (ha[NFTA_HOOK_DEV] || ha[NFTA_HOOK_DEVS]) {
2276                 module_put(type->owner);
2277                 return -EOPNOTSUPP;
2278         }
2279
2280         return 0;
2281 }
2282
2283 static void nft_chain_release_hook(struct nft_chain_hook *hook)
2284 {
2285         struct nft_hook *h, *next;
2286
2287         list_for_each_entry_safe(h, next, &hook->list, list) {
2288                 list_del(&h->list);
2289                 kfree(h);
2290         }
2291         module_put(hook->type->owner);
2292 }
2293
2294 static void nft_last_rule(const struct nft_chain *chain, const void *ptr)
2295 {
2296         struct nft_rule_dp_last *lrule;
2297
2298         BUILD_BUG_ON(offsetof(struct nft_rule_dp_last, end) != 0);
2299
2300         lrule = (struct nft_rule_dp_last *)ptr;
2301         lrule->end.is_last = 1;
2302         lrule->chain = chain;
2303         /* blob size does not include the trailer rule */
2304 }
2305
2306 static struct nft_rule_blob *nf_tables_chain_alloc_rules(const struct nft_chain *chain,
2307                                                          unsigned int size)
2308 {
2309         struct nft_rule_blob *blob;
2310
2311         if (size > INT_MAX)
2312                 return NULL;
2313
2314         size += sizeof(struct nft_rule_blob) + sizeof(struct nft_rule_dp_last);
2315
2316         blob = kvmalloc(size, GFP_KERNEL_ACCOUNT);
2317         if (!blob)
2318                 return NULL;
2319
2320         blob->size = 0;
2321         nft_last_rule(chain, blob->data);
2322
2323         return blob;
2324 }
2325
2326 static void nft_basechain_hook_init(struct nf_hook_ops *ops, u8 family,
2327                                     const struct nft_chain_hook *hook,
2328                                     struct nft_chain *chain)
2329 {
2330         ops->pf                 = family;
2331         ops->hooknum            = hook->num;
2332         ops->priority           = hook->priority;
2333         ops->priv               = chain;
2334         ops->hook               = hook->type->hooks[ops->hooknum];
2335         ops->hook_ops_type      = NF_HOOK_OP_NF_TABLES;
2336 }
2337
2338 static int nft_basechain_init(struct nft_base_chain *basechain, u8 family,
2339                               struct nft_chain_hook *hook, u32 flags)
2340 {
2341         struct nft_chain *chain;
2342         struct nft_hook *h;
2343
2344         basechain->type = hook->type;
2345         INIT_LIST_HEAD(&basechain->hook_list);
2346         chain = &basechain->chain;
2347
2348         if (nft_base_chain_netdev(family, hook->num)) {
2349                 list_splice_init(&hook->list, &basechain->hook_list);
2350                 list_for_each_entry(h, &basechain->hook_list, list)
2351                         nft_basechain_hook_init(&h->ops, family, hook, chain);
2352         }
2353         nft_basechain_hook_init(&basechain->ops, family, hook, chain);
2354
2355         chain->flags |= NFT_CHAIN_BASE | flags;
2356         basechain->policy = NF_ACCEPT;
2357         if (chain->flags & NFT_CHAIN_HW_OFFLOAD &&
2358             !nft_chain_offload_support(basechain)) {
2359                 list_splice_init(&basechain->hook_list, &hook->list);
2360                 return -EOPNOTSUPP;
2361         }
2362
2363         flow_block_init(&basechain->flow_block);
2364
2365         return 0;
2366 }
2367
2368 int nft_chain_add(struct nft_table *table, struct nft_chain *chain)
2369 {
2370         int err;
2371
2372         err = rhltable_insert_key(&table->chains_ht, chain->name,
2373                                   &chain->rhlhead, nft_chain_ht_params);
2374         if (err)
2375                 return err;
2376
2377         list_add_tail_rcu(&chain->list, &table->chains);
2378
2379         return 0;
2380 }
2381
2382 static u64 chain_id;
2383
2384 static int nf_tables_addchain(struct nft_ctx *ctx, u8 family, u8 genmask,
2385                               u8 policy, u32 flags,
2386                               struct netlink_ext_ack *extack)
2387 {
2388         const struct nlattr * const *nla = ctx->nla;
2389         struct nft_table *table = ctx->table;
2390         struct nft_base_chain *basechain;
2391         struct net *net = ctx->net;
2392         char name[NFT_NAME_MAXLEN];
2393         struct nft_rule_blob *blob;
2394         struct nft_trans *trans;
2395         struct nft_chain *chain;
2396         int err;
2397
2398         if (table->use == UINT_MAX)
2399                 return -EOVERFLOW;
2400
2401         if (nla[NFTA_CHAIN_HOOK]) {
2402                 struct nft_stats __percpu *stats = NULL;
2403                 struct nft_chain_hook hook = {};
2404
2405                 if (flags & NFT_CHAIN_BINDING)
2406                         return -EOPNOTSUPP;
2407
2408                 err = nft_chain_parse_hook(net, NULL, nla, &hook, family, flags,
2409                                            extack);
2410                 if (err < 0)
2411                         return err;
2412
2413                 basechain = kzalloc(sizeof(*basechain), GFP_KERNEL_ACCOUNT);
2414                 if (basechain == NULL) {
2415                         nft_chain_release_hook(&hook);
2416                         return -ENOMEM;
2417                 }
2418                 chain = &basechain->chain;
2419
2420                 if (nla[NFTA_CHAIN_COUNTERS]) {
2421                         stats = nft_stats_alloc(nla[NFTA_CHAIN_COUNTERS]);
2422                         if (IS_ERR(stats)) {
2423                                 nft_chain_release_hook(&hook);
2424                                 kfree(basechain);
2425                                 return PTR_ERR(stats);
2426                         }
2427                         rcu_assign_pointer(basechain->stats, stats);
2428                 }
2429
2430                 err = nft_basechain_init(basechain, family, &hook, flags);
2431                 if (err < 0) {
2432                         nft_chain_release_hook(&hook);
2433                         kfree(basechain);
2434                         free_percpu(stats);
2435                         return err;
2436                 }
2437                 if (stats)
2438                         static_branch_inc(&nft_counters_enabled);
2439         } else {
2440                 if (flags & NFT_CHAIN_BASE)
2441                         return -EINVAL;
2442                 if (flags & NFT_CHAIN_HW_OFFLOAD)
2443                         return -EOPNOTSUPP;
2444
2445                 chain = kzalloc(sizeof(*chain), GFP_KERNEL_ACCOUNT);
2446                 if (chain == NULL)
2447                         return -ENOMEM;
2448
2449                 chain->flags = flags;
2450         }
2451         ctx->chain = chain;
2452
2453         INIT_LIST_HEAD(&chain->rules);
2454         chain->handle = nf_tables_alloc_handle(table);
2455         chain->table = table;
2456
2457         if (nla[NFTA_CHAIN_NAME]) {
2458                 chain->name = nla_strdup(nla[NFTA_CHAIN_NAME], GFP_KERNEL_ACCOUNT);
2459         } else {
2460                 if (!(flags & NFT_CHAIN_BINDING)) {
2461                         err = -EINVAL;
2462                         goto err_destroy_chain;
2463                 }
2464
2465                 snprintf(name, sizeof(name), "__chain%llu", ++chain_id);
2466                 chain->name = kstrdup(name, GFP_KERNEL_ACCOUNT);
2467         }
2468
2469         if (!chain->name) {
2470                 err = -ENOMEM;
2471                 goto err_destroy_chain;
2472         }
2473
2474         if (nla[NFTA_CHAIN_USERDATA]) {
2475                 chain->udata = nla_memdup(nla[NFTA_CHAIN_USERDATA], GFP_KERNEL_ACCOUNT);
2476                 if (chain->udata == NULL) {
2477                         err = -ENOMEM;
2478                         goto err_destroy_chain;
2479                 }
2480                 chain->udlen = nla_len(nla[NFTA_CHAIN_USERDATA]);
2481         }
2482
2483         blob = nf_tables_chain_alloc_rules(chain, 0);
2484         if (!blob) {
2485                 err = -ENOMEM;
2486                 goto err_destroy_chain;
2487         }
2488
2489         RCU_INIT_POINTER(chain->blob_gen_0, blob);
2490         RCU_INIT_POINTER(chain->blob_gen_1, blob);
2491
2492         err = nf_tables_register_hook(net, table, chain);
2493         if (err < 0)
2494                 goto err_destroy_chain;
2495
2496         trans = nft_trans_chain_add(ctx, NFT_MSG_NEWCHAIN);
2497         if (IS_ERR(trans)) {
2498                 err = PTR_ERR(trans);
2499                 goto err_unregister_hook;
2500         }
2501
2502         nft_trans_chain_policy(trans) = NFT_CHAIN_POLICY_UNSET;
2503         if (nft_is_base_chain(chain))
2504                 nft_trans_chain_policy(trans) = policy;
2505
2506         err = nft_chain_add(table, chain);
2507         if (err < 0) {
2508                 nft_trans_destroy(trans);
2509                 goto err_unregister_hook;
2510         }
2511
2512         table->use++;
2513
2514         return 0;
2515 err_unregister_hook:
2516         nf_tables_unregister_hook(net, table, chain);
2517 err_destroy_chain:
2518         nf_tables_chain_destroy(ctx);
2519
2520         return err;
2521 }
2522
2523 static int nf_tables_updchain(struct nft_ctx *ctx, u8 genmask, u8 policy,
2524                               u32 flags, const struct nlattr *attr,
2525                               struct netlink_ext_ack *extack)
2526 {
2527         const struct nlattr * const *nla = ctx->nla;
2528         struct nft_base_chain *basechain = NULL;
2529         struct nft_table *table = ctx->table;
2530         struct nft_chain *chain = ctx->chain;
2531         struct nft_chain_hook hook = {};
2532         struct nft_stats *stats = NULL;
2533         struct nft_hook *h, *next;
2534         struct nf_hook_ops *ops;
2535         struct nft_trans *trans;
2536         bool unregister = false;
2537         int err;
2538
2539         if (chain->flags ^ flags)
2540                 return -EOPNOTSUPP;
2541
2542         INIT_LIST_HEAD(&hook.list);
2543
2544         if (nla[NFTA_CHAIN_HOOK]) {
2545                 if (!nft_is_base_chain(chain)) {
2546                         NL_SET_BAD_ATTR(extack, attr);
2547                         return -EEXIST;
2548                 }
2549
2550                 basechain = nft_base_chain(chain);
2551                 err = nft_chain_parse_hook(ctx->net, basechain, nla, &hook,
2552                                            ctx->family, flags, extack);
2553                 if (err < 0)
2554                         return err;
2555
2556                 if (basechain->type != hook.type) {
2557                         nft_chain_release_hook(&hook);
2558                         NL_SET_BAD_ATTR(extack, attr);
2559                         return -EEXIST;
2560                 }
2561
2562                 if (nft_base_chain_netdev(ctx->family, basechain->ops.hooknum)) {
2563                         list_for_each_entry_safe(h, next, &hook.list, list) {
2564                                 h->ops.pf       = basechain->ops.pf;
2565                                 h->ops.hooknum  = basechain->ops.hooknum;
2566                                 h->ops.priority = basechain->ops.priority;
2567                                 h->ops.priv     = basechain->ops.priv;
2568                                 h->ops.hook     = basechain->ops.hook;
2569
2570                                 if (nft_hook_list_find(&basechain->hook_list, h)) {
2571                                         list_del(&h->list);
2572                                         kfree(h);
2573                                 }
2574                         }
2575                 } else {
2576                         ops = &basechain->ops;
2577                         if (ops->hooknum != hook.num ||
2578                             ops->priority != hook.priority) {
2579                                 nft_chain_release_hook(&hook);
2580                                 NL_SET_BAD_ATTR(extack, attr);
2581                                 return -EEXIST;
2582                         }
2583                 }
2584         }
2585
2586         if (nla[NFTA_CHAIN_HANDLE] &&
2587             nla[NFTA_CHAIN_NAME]) {
2588                 struct nft_chain *chain2;
2589
2590                 chain2 = nft_chain_lookup(ctx->net, table,
2591                                           nla[NFTA_CHAIN_NAME], genmask);
2592                 if (!IS_ERR(chain2)) {
2593                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_NAME]);
2594                         err = -EEXIST;
2595                         goto err_hooks;
2596                 }
2597         }
2598
2599         if (nla[NFTA_CHAIN_COUNTERS]) {
2600                 if (!nft_is_base_chain(chain)) {
2601                         err = -EOPNOTSUPP;
2602                         goto err_hooks;
2603                 }
2604
2605                 stats = nft_stats_alloc(nla[NFTA_CHAIN_COUNTERS]);
2606                 if (IS_ERR(stats)) {
2607                         err = PTR_ERR(stats);
2608                         goto err_hooks;
2609                 }
2610         }
2611
2612         if (!(table->flags & NFT_TABLE_F_DORMANT) &&
2613             nft_is_base_chain(chain) &&
2614             !list_empty(&hook.list)) {
2615                 basechain = nft_base_chain(chain);
2616                 ops = &basechain->ops;
2617
2618                 if (nft_base_chain_netdev(table->family, basechain->ops.hooknum)) {
2619                         err = nft_netdev_register_hooks(ctx->net, &hook.list);
2620                         if (err < 0)
2621                                 goto err_hooks;
2622                 }
2623         }
2624
2625         unregister = true;
2626         err = -ENOMEM;
2627         trans = nft_trans_alloc(ctx, NFT_MSG_NEWCHAIN,
2628                                 sizeof(struct nft_trans_chain));
2629         if (trans == NULL)
2630                 goto err_trans;
2631
2632         nft_trans_chain_stats(trans) = stats;
2633         nft_trans_chain_update(trans) = true;
2634
2635         if (nla[NFTA_CHAIN_POLICY])
2636                 nft_trans_chain_policy(trans) = policy;
2637         else
2638                 nft_trans_chain_policy(trans) = -1;
2639
2640         if (nla[NFTA_CHAIN_HANDLE] &&
2641             nla[NFTA_CHAIN_NAME]) {
2642                 struct nftables_pernet *nft_net = nft_pernet(ctx->net);
2643                 struct nft_trans *tmp;
2644                 char *name;
2645
2646                 err = -ENOMEM;
2647                 name = nla_strdup(nla[NFTA_CHAIN_NAME], GFP_KERNEL_ACCOUNT);
2648                 if (!name)
2649                         goto err_trans;
2650
2651                 err = -EEXIST;
2652                 list_for_each_entry(tmp, &nft_net->commit_list, list) {
2653                         if (tmp->msg_type == NFT_MSG_NEWCHAIN &&
2654                             tmp->ctx.table == table &&
2655                             nft_trans_chain_update(tmp) &&
2656                             nft_trans_chain_name(tmp) &&
2657                             strcmp(name, nft_trans_chain_name(tmp)) == 0) {
2658                                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_NAME]);
2659                                 kfree(name);
2660                                 goto err_trans;
2661                         }
2662                 }
2663
2664                 nft_trans_chain_name(trans) = name;
2665         }
2666
2667         nft_trans_basechain(trans) = basechain;
2668         INIT_LIST_HEAD(&nft_trans_chain_hooks(trans));
2669         list_splice(&hook.list, &nft_trans_chain_hooks(trans));
2670         if (nla[NFTA_CHAIN_HOOK])
2671                 module_put(hook.type->owner);
2672
2673         nft_trans_commit_list_add_tail(ctx->net, trans);
2674
2675         return 0;
2676
2677 err_trans:
2678         free_percpu(stats);
2679         kfree(trans);
2680 err_hooks:
2681         if (nla[NFTA_CHAIN_HOOK]) {
2682                 list_for_each_entry_safe(h, next, &hook.list, list) {
2683                         if (unregister)
2684                                 nf_unregister_net_hook(ctx->net, &h->ops);
2685                         list_del(&h->list);
2686                         kfree_rcu(h, rcu);
2687                 }
2688                 module_put(hook.type->owner);
2689         }
2690
2691         return err;
2692 }
2693
2694 static struct nft_chain *nft_chain_lookup_byid(const struct net *net,
2695                                                const struct nft_table *table,
2696                                                const struct nlattr *nla)
2697 {
2698         struct nftables_pernet *nft_net = nft_pernet(net);
2699         u32 id = ntohl(nla_get_be32(nla));
2700         struct nft_trans *trans;
2701
2702         list_for_each_entry(trans, &nft_net->commit_list, list) {
2703                 struct nft_chain *chain = trans->ctx.chain;
2704
2705                 if (trans->msg_type == NFT_MSG_NEWCHAIN &&
2706                     chain->table == table &&
2707                     id == nft_trans_chain_id(trans))
2708                         return chain;
2709         }
2710         return ERR_PTR(-ENOENT);
2711 }
2712
2713 static int nf_tables_newchain(struct sk_buff *skb, const struct nfnl_info *info,
2714                               const struct nlattr * const nla[])
2715 {
2716         struct nftables_pernet *nft_net = nft_pernet(info->net);
2717         struct netlink_ext_ack *extack = info->extack;
2718         u8 genmask = nft_genmask_next(info->net);
2719         u8 family = info->nfmsg->nfgen_family;
2720         struct nft_chain *chain = NULL;
2721         struct net *net = info->net;
2722         const struct nlattr *attr;
2723         struct nft_table *table;
2724         u8 policy = NF_ACCEPT;
2725         struct nft_ctx ctx;
2726         u64 handle = 0;
2727         u32 flags = 0;
2728
2729         lockdep_assert_held(&nft_net->commit_mutex);
2730
2731         table = nft_table_lookup(net, nla[NFTA_CHAIN_TABLE], family, genmask,
2732                                  NETLINK_CB(skb).portid);
2733         if (IS_ERR(table)) {
2734                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_TABLE]);
2735                 return PTR_ERR(table);
2736         }
2737
2738         chain = NULL;
2739         attr = nla[NFTA_CHAIN_NAME];
2740
2741         if (nla[NFTA_CHAIN_HANDLE]) {
2742                 handle = be64_to_cpu(nla_get_be64(nla[NFTA_CHAIN_HANDLE]));
2743                 chain = nft_chain_lookup_byhandle(table, handle, genmask);
2744                 if (IS_ERR(chain)) {
2745                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_HANDLE]);
2746                         return PTR_ERR(chain);
2747                 }
2748                 attr = nla[NFTA_CHAIN_HANDLE];
2749         } else if (nla[NFTA_CHAIN_NAME]) {
2750                 chain = nft_chain_lookup(net, table, attr, genmask);
2751                 if (IS_ERR(chain)) {
2752                         if (PTR_ERR(chain) != -ENOENT) {
2753                                 NL_SET_BAD_ATTR(extack, attr);
2754                                 return PTR_ERR(chain);
2755                         }
2756                         chain = NULL;
2757                 }
2758         } else if (!nla[NFTA_CHAIN_ID]) {
2759                 return -EINVAL;
2760         }
2761
2762         if (nla[NFTA_CHAIN_POLICY]) {
2763                 if (chain != NULL &&
2764                     !nft_is_base_chain(chain)) {
2765                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_POLICY]);
2766                         return -EOPNOTSUPP;
2767                 }
2768
2769                 if (chain == NULL &&
2770                     nla[NFTA_CHAIN_HOOK] == NULL) {
2771                         NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_POLICY]);
2772                         return -EOPNOTSUPP;
2773                 }
2774
2775                 policy = ntohl(nla_get_be32(nla[NFTA_CHAIN_POLICY]));
2776                 switch (policy) {
2777                 case NF_DROP:
2778                 case NF_ACCEPT:
2779                         break;
2780                 default:
2781                         return -EINVAL;
2782                 }
2783         }
2784
2785         if (nla[NFTA_CHAIN_FLAGS])
2786                 flags = ntohl(nla_get_be32(nla[NFTA_CHAIN_FLAGS]));
2787         else if (chain)
2788                 flags = chain->flags;
2789
2790         if (flags & ~NFT_CHAIN_FLAGS)
2791                 return -EOPNOTSUPP;
2792
2793         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, chain, nla);
2794
2795         if (chain != NULL) {
2796                 if (chain->flags & NFT_CHAIN_BINDING)
2797                         return -EINVAL;
2798
2799                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
2800                         NL_SET_BAD_ATTR(extack, attr);
2801                         return -EEXIST;
2802                 }
2803                 if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
2804                         return -EOPNOTSUPP;
2805
2806                 flags |= chain->flags & NFT_CHAIN_BASE;
2807                 return nf_tables_updchain(&ctx, genmask, policy, flags, attr,
2808                                           extack);
2809         }
2810
2811         return nf_tables_addchain(&ctx, family, genmask, policy, flags, extack);
2812 }
2813
2814 static int nft_delchain_hook(struct nft_ctx *ctx,
2815                              struct nft_base_chain *basechain,
2816                              struct netlink_ext_ack *extack)
2817 {
2818         const struct nft_chain *chain = &basechain->chain;
2819         const struct nlattr * const *nla = ctx->nla;
2820         struct nft_chain_hook chain_hook = {};
2821         struct nft_hook *this, *hook;
2822         LIST_HEAD(chain_del_list);
2823         struct nft_trans *trans;
2824         int err;
2825
2826         err = nft_chain_parse_hook(ctx->net, basechain, nla, &chain_hook,
2827                                    ctx->family, chain->flags, extack);
2828         if (err < 0)
2829                 return err;
2830
2831         list_for_each_entry(this, &chain_hook.list, list) {
2832                 hook = nft_hook_list_find(&basechain->hook_list, this);
2833                 if (!hook) {
2834                         err = -ENOENT;
2835                         goto err_chain_del_hook;
2836                 }
2837                 list_move(&hook->list, &chain_del_list);
2838         }
2839
2840         trans = nft_trans_alloc(ctx, NFT_MSG_DELCHAIN,
2841                                 sizeof(struct nft_trans_chain));
2842         if (!trans) {
2843                 err = -ENOMEM;
2844                 goto err_chain_del_hook;
2845         }
2846
2847         nft_trans_basechain(trans) = basechain;
2848         nft_trans_chain_update(trans) = true;
2849         INIT_LIST_HEAD(&nft_trans_chain_hooks(trans));
2850         list_splice(&chain_del_list, &nft_trans_chain_hooks(trans));
2851         nft_chain_release_hook(&chain_hook);
2852
2853         nft_trans_commit_list_add_tail(ctx->net, trans);
2854
2855         return 0;
2856
2857 err_chain_del_hook:
2858         list_splice(&chain_del_list, &basechain->hook_list);
2859         nft_chain_release_hook(&chain_hook);
2860
2861         return err;
2862 }
2863
2864 static int nf_tables_delchain(struct sk_buff *skb, const struct nfnl_info *info,
2865                               const struct nlattr * const nla[])
2866 {
2867         struct netlink_ext_ack *extack = info->extack;
2868         u8 genmask = nft_genmask_next(info->net);
2869         u8 family = info->nfmsg->nfgen_family;
2870         struct net *net = info->net;
2871         const struct nlattr *attr;
2872         struct nft_table *table;
2873         struct nft_chain *chain;
2874         struct nft_rule *rule;
2875         struct nft_ctx ctx;
2876         u64 handle;
2877         u32 use;
2878         int err;
2879
2880         table = nft_table_lookup(net, nla[NFTA_CHAIN_TABLE], family, genmask,
2881                                  NETLINK_CB(skb).portid);
2882         if (IS_ERR(table)) {
2883                 NL_SET_BAD_ATTR(extack, nla[NFTA_CHAIN_TABLE]);
2884                 return PTR_ERR(table);
2885         }
2886
2887         if (nla[NFTA_CHAIN_HANDLE]) {
2888                 attr = nla[NFTA_CHAIN_HANDLE];
2889                 handle = be64_to_cpu(nla_get_be64(attr));
2890                 chain = nft_chain_lookup_byhandle(table, handle, genmask);
2891         } else {
2892                 attr = nla[NFTA_CHAIN_NAME];
2893                 chain = nft_chain_lookup(net, table, attr, genmask);
2894         }
2895         if (IS_ERR(chain)) {
2896                 if (PTR_ERR(chain) == -ENOENT &&
2897                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYCHAIN)
2898                         return 0;
2899
2900                 NL_SET_BAD_ATTR(extack, attr);
2901                 return PTR_ERR(chain);
2902         }
2903
2904         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, chain, nla);
2905
2906         if (nla[NFTA_CHAIN_HOOK]) {
2907                 if (chain->flags & NFT_CHAIN_HW_OFFLOAD)
2908                         return -EOPNOTSUPP;
2909
2910                 if (nft_is_base_chain(chain)) {
2911                         struct nft_base_chain *basechain = nft_base_chain(chain);
2912
2913                         if (nft_base_chain_netdev(table->family, basechain->ops.hooknum))
2914                                 return nft_delchain_hook(&ctx, basechain, extack);
2915                 }
2916         }
2917
2918         if (info->nlh->nlmsg_flags & NLM_F_NONREC &&
2919             chain->use > 0)
2920                 return -EBUSY;
2921
2922         use = chain->use;
2923         list_for_each_entry(rule, &chain->rules, list) {
2924                 if (!nft_is_active_next(net, rule))
2925                         continue;
2926                 use--;
2927
2928                 err = nft_delrule(&ctx, rule);
2929                 if (err < 0)
2930                         return err;
2931         }
2932
2933         /* There are rules and elements that are still holding references to us,
2934          * we cannot do a recursive removal in this case.
2935          */
2936         if (use > 0) {
2937                 NL_SET_BAD_ATTR(extack, attr);
2938                 return -EBUSY;
2939         }
2940
2941         return nft_delchain(&ctx);
2942 }
2943
2944 /*
2945  * Expressions
2946  */
2947
2948 /**
2949  *      nft_register_expr - register nf_tables expr type
2950  *      @type: expr type
2951  *
2952  *      Registers the expr type for use with nf_tables. Returns zero on
2953  *      success or a negative errno code otherwise.
2954  */
2955 int nft_register_expr(struct nft_expr_type *type)
2956 {
2957         nfnl_lock(NFNL_SUBSYS_NFTABLES);
2958         if (type->family == NFPROTO_UNSPEC)
2959                 list_add_tail_rcu(&type->list, &nf_tables_expressions);
2960         else
2961                 list_add_rcu(&type->list, &nf_tables_expressions);
2962         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
2963         return 0;
2964 }
2965 EXPORT_SYMBOL_GPL(nft_register_expr);
2966
2967 /**
2968  *      nft_unregister_expr - unregister nf_tables expr type
2969  *      @type: expr type
2970  *
2971  *      Unregisters the expr typefor use with nf_tables.
2972  */
2973 void nft_unregister_expr(struct nft_expr_type *type)
2974 {
2975         nfnl_lock(NFNL_SUBSYS_NFTABLES);
2976         list_del_rcu(&type->list);
2977         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
2978 }
2979 EXPORT_SYMBOL_GPL(nft_unregister_expr);
2980
2981 static const struct nft_expr_type *__nft_expr_type_get(u8 family,
2982                                                        struct nlattr *nla)
2983 {
2984         const struct nft_expr_type *type, *candidate = NULL;
2985
2986         list_for_each_entry(type, &nf_tables_expressions, list) {
2987                 if (!nla_strcmp(nla, type->name)) {
2988                         if (!type->family && !candidate)
2989                                 candidate = type;
2990                         else if (type->family == family)
2991                                 candidate = type;
2992                 }
2993         }
2994         return candidate;
2995 }
2996
2997 #ifdef CONFIG_MODULES
2998 static int nft_expr_type_request_module(struct net *net, u8 family,
2999                                         struct nlattr *nla)
3000 {
3001         if (nft_request_module(net, "nft-expr-%u-%.*s", family,
3002                                nla_len(nla), (char *)nla_data(nla)) == -EAGAIN)
3003                 return -EAGAIN;
3004
3005         return 0;
3006 }
3007 #endif
3008
3009 static const struct nft_expr_type *nft_expr_type_get(struct net *net,
3010                                                      u8 family,
3011                                                      struct nlattr *nla)
3012 {
3013         const struct nft_expr_type *type;
3014
3015         if (nla == NULL)
3016                 return ERR_PTR(-EINVAL);
3017
3018         type = __nft_expr_type_get(family, nla);
3019         if (type != NULL && try_module_get(type->owner))
3020                 return type;
3021
3022         lockdep_nfnl_nft_mutex_not_held();
3023 #ifdef CONFIG_MODULES
3024         if (type == NULL) {
3025                 if (nft_expr_type_request_module(net, family, nla) == -EAGAIN)
3026                         return ERR_PTR(-EAGAIN);
3027
3028                 if (nft_request_module(net, "nft-expr-%.*s",
3029                                        nla_len(nla),
3030                                        (char *)nla_data(nla)) == -EAGAIN)
3031                         return ERR_PTR(-EAGAIN);
3032         }
3033 #endif
3034         return ERR_PTR(-ENOENT);
3035 }
3036
3037 static const struct nla_policy nft_expr_policy[NFTA_EXPR_MAX + 1] = {
3038         [NFTA_EXPR_NAME]        = { .type = NLA_STRING,
3039                                     .len = NFT_MODULE_AUTOLOAD_LIMIT },
3040         [NFTA_EXPR_DATA]        = { .type = NLA_NESTED },
3041 };
3042
3043 static int nf_tables_fill_expr_info(struct sk_buff *skb,
3044                                     const struct nft_expr *expr, bool reset)
3045 {
3046         if (nla_put_string(skb, NFTA_EXPR_NAME, expr->ops->type->name))
3047                 goto nla_put_failure;
3048
3049         if (expr->ops->dump) {
3050                 struct nlattr *data = nla_nest_start_noflag(skb,
3051                                                             NFTA_EXPR_DATA);
3052                 if (data == NULL)
3053                         goto nla_put_failure;
3054                 if (expr->ops->dump(skb, expr, reset) < 0)
3055                         goto nla_put_failure;
3056                 nla_nest_end(skb, data);
3057         }
3058
3059         return skb->len;
3060
3061 nla_put_failure:
3062         return -1;
3063 };
3064
3065 int nft_expr_dump(struct sk_buff *skb, unsigned int attr,
3066                   const struct nft_expr *expr, bool reset)
3067 {
3068         struct nlattr *nest;
3069
3070         nest = nla_nest_start_noflag(skb, attr);
3071         if (!nest)
3072                 goto nla_put_failure;
3073         if (nf_tables_fill_expr_info(skb, expr, reset) < 0)
3074                 goto nla_put_failure;
3075         nla_nest_end(skb, nest);
3076         return 0;
3077
3078 nla_put_failure:
3079         return -1;
3080 }
3081
3082 struct nft_expr_info {
3083         const struct nft_expr_ops       *ops;
3084         const struct nlattr             *attr;
3085         struct nlattr                   *tb[NFT_EXPR_MAXATTR + 1];
3086 };
3087
3088 static int nf_tables_expr_parse(const struct nft_ctx *ctx,
3089                                 const struct nlattr *nla,
3090                                 struct nft_expr_info *info)
3091 {
3092         const struct nft_expr_type *type;
3093         const struct nft_expr_ops *ops;
3094         struct nlattr *tb[NFTA_EXPR_MAX + 1];
3095         int err;
3096
3097         err = nla_parse_nested_deprecated(tb, NFTA_EXPR_MAX, nla,
3098                                           nft_expr_policy, NULL);
3099         if (err < 0)
3100                 return err;
3101
3102         type = nft_expr_type_get(ctx->net, ctx->family, tb[NFTA_EXPR_NAME]);
3103         if (IS_ERR(type))
3104                 return PTR_ERR(type);
3105
3106         if (tb[NFTA_EXPR_DATA]) {
3107                 err = nla_parse_nested_deprecated(info->tb, type->maxattr,
3108                                                   tb[NFTA_EXPR_DATA],
3109                                                   type->policy, NULL);
3110                 if (err < 0)
3111                         goto err1;
3112         } else
3113                 memset(info->tb, 0, sizeof(info->tb[0]) * (type->maxattr + 1));
3114
3115         if (type->select_ops != NULL) {
3116                 ops = type->select_ops(ctx,
3117                                        (const struct nlattr * const *)info->tb);
3118                 if (IS_ERR(ops)) {
3119                         err = PTR_ERR(ops);
3120 #ifdef CONFIG_MODULES
3121                         if (err == -EAGAIN)
3122                                 if (nft_expr_type_request_module(ctx->net,
3123                                                                  ctx->family,
3124                                                                  tb[NFTA_EXPR_NAME]) != -EAGAIN)
3125                                         err = -ENOENT;
3126 #endif
3127                         goto err1;
3128                 }
3129         } else
3130                 ops = type->ops;
3131
3132         info->attr = nla;
3133         info->ops = ops;
3134
3135         return 0;
3136
3137 err1:
3138         module_put(type->owner);
3139         return err;
3140 }
3141
3142 int nft_expr_inner_parse(const struct nft_ctx *ctx, const struct nlattr *nla,
3143                          struct nft_expr_info *info)
3144 {
3145         struct nlattr *tb[NFTA_EXPR_MAX + 1];
3146         const struct nft_expr_type *type;
3147         int err;
3148
3149         err = nla_parse_nested_deprecated(tb, NFTA_EXPR_MAX, nla,
3150                                           nft_expr_policy, NULL);
3151         if (err < 0)
3152                 return err;
3153
3154         if (!tb[NFTA_EXPR_DATA])
3155                 return -EINVAL;
3156
3157         type = __nft_expr_type_get(ctx->family, tb[NFTA_EXPR_NAME]);
3158         if (!type)
3159                 return -ENOENT;
3160
3161         if (!type->inner_ops)
3162                 return -EOPNOTSUPP;
3163
3164         err = nla_parse_nested_deprecated(info->tb, type->maxattr,
3165                                           tb[NFTA_EXPR_DATA],
3166                                           type->policy, NULL);
3167         if (err < 0)
3168                 goto err_nla_parse;
3169
3170         info->attr = nla;
3171         info->ops = type->inner_ops;
3172
3173         return 0;
3174
3175 err_nla_parse:
3176         return err;
3177 }
3178
3179 static int nf_tables_newexpr(const struct nft_ctx *ctx,
3180                              const struct nft_expr_info *expr_info,
3181                              struct nft_expr *expr)
3182 {
3183         const struct nft_expr_ops *ops = expr_info->ops;
3184         int err;
3185
3186         expr->ops = ops;
3187         if (ops->init) {
3188                 err = ops->init(ctx, expr, (const struct nlattr **)expr_info->tb);
3189                 if (err < 0)
3190                         goto err1;
3191         }
3192
3193         return 0;
3194 err1:
3195         expr->ops = NULL;
3196         return err;
3197 }
3198
3199 static void nf_tables_expr_destroy(const struct nft_ctx *ctx,
3200                                    struct nft_expr *expr)
3201 {
3202         const struct nft_expr_type *type = expr->ops->type;
3203
3204         if (expr->ops->destroy)
3205                 expr->ops->destroy(ctx, expr);
3206         module_put(type->owner);
3207 }
3208
3209 static struct nft_expr *nft_expr_init(const struct nft_ctx *ctx,
3210                                       const struct nlattr *nla)
3211 {
3212         struct nft_expr_info expr_info;
3213         struct nft_expr *expr;
3214         struct module *owner;
3215         int err;
3216
3217         err = nf_tables_expr_parse(ctx, nla, &expr_info);
3218         if (err < 0)
3219                 goto err_expr_parse;
3220
3221         err = -EOPNOTSUPP;
3222         if (!(expr_info.ops->type->flags & NFT_EXPR_STATEFUL))
3223                 goto err_expr_stateful;
3224
3225         err = -ENOMEM;
3226         expr = kzalloc(expr_info.ops->size, GFP_KERNEL_ACCOUNT);
3227         if (expr == NULL)
3228                 goto err_expr_stateful;
3229
3230         err = nf_tables_newexpr(ctx, &expr_info, expr);
3231         if (err < 0)
3232                 goto err_expr_new;
3233
3234         return expr;
3235 err_expr_new:
3236         kfree(expr);
3237 err_expr_stateful:
3238         owner = expr_info.ops->type->owner;
3239         if (expr_info.ops->type->release_ops)
3240                 expr_info.ops->type->release_ops(expr_info.ops);
3241
3242         module_put(owner);
3243 err_expr_parse:
3244         return ERR_PTR(err);
3245 }
3246
3247 int nft_expr_clone(struct nft_expr *dst, struct nft_expr *src)
3248 {
3249         int err;
3250
3251         if (src->ops->clone) {
3252                 dst->ops = src->ops;
3253                 err = src->ops->clone(dst, src);
3254                 if (err < 0)
3255                         return err;
3256         } else {
3257                 memcpy(dst, src, src->ops->size);
3258         }
3259
3260         __module_get(src->ops->type->owner);
3261
3262         return 0;
3263 }
3264
3265 void nft_expr_destroy(const struct nft_ctx *ctx, struct nft_expr *expr)
3266 {
3267         nf_tables_expr_destroy(ctx, expr);
3268         kfree(expr);
3269 }
3270
3271 /*
3272  * Rules
3273  */
3274
3275 static struct nft_rule *__nft_rule_lookup(const struct nft_chain *chain,
3276                                           u64 handle)
3277 {
3278         struct nft_rule *rule;
3279
3280         // FIXME: this sucks
3281         list_for_each_entry_rcu(rule, &chain->rules, list) {
3282                 if (handle == rule->handle)
3283                         return rule;
3284         }
3285
3286         return ERR_PTR(-ENOENT);
3287 }
3288
3289 static struct nft_rule *nft_rule_lookup(const struct nft_chain *chain,
3290                                         const struct nlattr *nla)
3291 {
3292         if (nla == NULL)
3293                 return ERR_PTR(-EINVAL);
3294
3295         return __nft_rule_lookup(chain, be64_to_cpu(nla_get_be64(nla)));
3296 }
3297
3298 static const struct nla_policy nft_rule_policy[NFTA_RULE_MAX + 1] = {
3299         [NFTA_RULE_TABLE]       = { .type = NLA_STRING,
3300                                     .len = NFT_TABLE_MAXNAMELEN - 1 },
3301         [NFTA_RULE_CHAIN]       = { .type = NLA_STRING,
3302                                     .len = NFT_CHAIN_MAXNAMELEN - 1 },
3303         [NFTA_RULE_HANDLE]      = { .type = NLA_U64 },
3304         [NFTA_RULE_EXPRESSIONS] = { .type = NLA_NESTED },
3305         [NFTA_RULE_COMPAT]      = { .type = NLA_NESTED },
3306         [NFTA_RULE_POSITION]    = { .type = NLA_U64 },
3307         [NFTA_RULE_USERDATA]    = { .type = NLA_BINARY,
3308                                     .len = NFT_USERDATA_MAXLEN },
3309         [NFTA_RULE_ID]          = { .type = NLA_U32 },
3310         [NFTA_RULE_POSITION_ID] = { .type = NLA_U32 },
3311         [NFTA_RULE_CHAIN_ID]    = { .type = NLA_U32 },
3312 };
3313
3314 static int nf_tables_fill_rule_info(struct sk_buff *skb, struct net *net,
3315                                     u32 portid, u32 seq, int event,
3316                                     u32 flags, int family,
3317                                     const struct nft_table *table,
3318                                     const struct nft_chain *chain,
3319                                     const struct nft_rule *rule, u64 handle,
3320                                     bool reset)
3321 {
3322         struct nlmsghdr *nlh;
3323         const struct nft_expr *expr, *next;
3324         struct nlattr *list;
3325         u16 type = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
3326
3327         nlh = nfnl_msg_put(skb, portid, seq, type, flags, family, NFNETLINK_V0,
3328                            nft_base_seq(net));
3329         if (!nlh)
3330                 goto nla_put_failure;
3331
3332         if (nla_put_string(skb, NFTA_RULE_TABLE, table->name))
3333                 goto nla_put_failure;
3334         if (nla_put_string(skb, NFTA_RULE_CHAIN, chain->name))
3335                 goto nla_put_failure;
3336         if (nla_put_be64(skb, NFTA_RULE_HANDLE, cpu_to_be64(rule->handle),
3337                          NFTA_RULE_PAD))
3338                 goto nla_put_failure;
3339
3340         if (event != NFT_MSG_DELRULE && handle) {
3341                 if (nla_put_be64(skb, NFTA_RULE_POSITION, cpu_to_be64(handle),
3342                                  NFTA_RULE_PAD))
3343                         goto nla_put_failure;
3344         }
3345
3346         if (chain->flags & NFT_CHAIN_HW_OFFLOAD)
3347                 nft_flow_rule_stats(chain, rule);
3348
3349         list = nla_nest_start_noflag(skb, NFTA_RULE_EXPRESSIONS);
3350         if (list == NULL)
3351                 goto nla_put_failure;
3352         nft_rule_for_each_expr(expr, next, rule) {
3353                 if (nft_expr_dump(skb, NFTA_LIST_ELEM, expr, reset) < 0)
3354                         goto nla_put_failure;
3355         }
3356         nla_nest_end(skb, list);
3357
3358         if (rule->udata) {
3359                 struct nft_userdata *udata = nft_userdata(rule);
3360                 if (nla_put(skb, NFTA_RULE_USERDATA, udata->len + 1,
3361                             udata->data) < 0)
3362                         goto nla_put_failure;
3363         }
3364
3365         nlmsg_end(skb, nlh);
3366         return 0;
3367
3368 nla_put_failure:
3369         nlmsg_trim(skb, nlh);
3370         return -1;
3371 }
3372
3373 static void nf_tables_rule_notify(const struct nft_ctx *ctx,
3374                                   const struct nft_rule *rule, int event)
3375 {
3376         struct nftables_pernet *nft_net = nft_pernet(ctx->net);
3377         const struct nft_rule *prule;
3378         struct sk_buff *skb;
3379         u64 handle = 0;
3380         u16 flags = 0;
3381         int err;
3382
3383         if (!ctx->report &&
3384             !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
3385                 return;
3386
3387         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
3388         if (skb == NULL)
3389                 goto err;
3390
3391         if (event == NFT_MSG_NEWRULE &&
3392             !list_is_first(&rule->list, &ctx->chain->rules) &&
3393             !list_is_last(&rule->list, &ctx->chain->rules)) {
3394                 prule = list_prev_entry(rule, list);
3395                 handle = prule->handle;
3396         }
3397         if (ctx->flags & (NLM_F_APPEND | NLM_F_REPLACE))
3398                 flags |= NLM_F_APPEND;
3399         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
3400                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
3401
3402         err = nf_tables_fill_rule_info(skb, ctx->net, ctx->portid, ctx->seq,
3403                                        event, flags, ctx->family, ctx->table,
3404                                        ctx->chain, rule, handle, false);
3405         if (err < 0) {
3406                 kfree_skb(skb);
3407                 goto err;
3408         }
3409
3410         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
3411         return;
3412 err:
3413         nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES, -ENOBUFS);
3414 }
3415
3416 struct nft_rule_dump_ctx {
3417         char *table;
3418         char *chain;
3419 };
3420
3421 static int __nf_tables_dump_rules(struct sk_buff *skb,
3422                                   unsigned int *idx,
3423                                   struct netlink_callback *cb,
3424                                   const struct nft_table *table,
3425                                   const struct nft_chain *chain,
3426                                   bool reset)
3427 {
3428         struct net *net = sock_net(skb->sk);
3429         const struct nft_rule *rule, *prule;
3430         unsigned int s_idx = cb->args[0];
3431         u64 handle;
3432
3433         prule = NULL;
3434         list_for_each_entry_rcu(rule, &chain->rules, list) {
3435                 if (!nft_is_active(net, rule))
3436                         goto cont_skip;
3437                 if (*idx < s_idx)
3438                         goto cont;
3439                 if (*idx > s_idx) {
3440                         memset(&cb->args[1], 0,
3441                                         sizeof(cb->args) - sizeof(cb->args[0]));
3442                 }
3443                 if (prule)
3444                         handle = prule->handle;
3445                 else
3446                         handle = 0;
3447
3448                 if (nf_tables_fill_rule_info(skb, net, NETLINK_CB(cb->skb).portid,
3449                                         cb->nlh->nlmsg_seq,
3450                                         NFT_MSG_NEWRULE,
3451                                         NLM_F_MULTI | NLM_F_APPEND,
3452                                         table->family,
3453                                         table, chain, rule, handle, reset) < 0)
3454                         return 1;
3455
3456                 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
3457 cont:
3458                 prule = rule;
3459 cont_skip:
3460                 (*idx)++;
3461         }
3462         return 0;
3463 }
3464
3465 static int nf_tables_dump_rules(struct sk_buff *skb,
3466                                 struct netlink_callback *cb)
3467 {
3468         const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
3469         const struct nft_rule_dump_ctx *ctx = cb->data;
3470         struct nft_table *table;
3471         const struct nft_chain *chain;
3472         unsigned int idx = 0;
3473         struct net *net = sock_net(skb->sk);
3474         int family = nfmsg->nfgen_family;
3475         struct nftables_pernet *nft_net;
3476         bool reset = false;
3477
3478         if (NFNL_MSG_TYPE(cb->nlh->nlmsg_type) == NFT_MSG_GETRULE_RESET)
3479                 reset = true;
3480
3481         rcu_read_lock();
3482         nft_net = nft_pernet(net);
3483         cb->seq = READ_ONCE(nft_net->base_seq);
3484
3485         list_for_each_entry_rcu(table, &nft_net->tables, list) {
3486                 if (family != NFPROTO_UNSPEC && family != table->family)
3487                         continue;
3488
3489                 if (ctx && ctx->table && strcmp(ctx->table, table->name) != 0)
3490                         continue;
3491
3492                 if (ctx && ctx->table && ctx->chain) {
3493                         struct rhlist_head *list, *tmp;
3494
3495                         list = rhltable_lookup(&table->chains_ht, ctx->chain,
3496                                                nft_chain_ht_params);
3497                         if (!list)
3498                                 goto done;
3499
3500                         rhl_for_each_entry_rcu(chain, tmp, list, rhlhead) {
3501                                 if (!nft_is_active(net, chain))
3502                                         continue;
3503                                 __nf_tables_dump_rules(skb, &idx,
3504                                                        cb, table, chain, reset);
3505                                 break;
3506                         }
3507                         goto done;
3508                 }
3509
3510                 list_for_each_entry_rcu(chain, &table->chains, list) {
3511                         if (__nf_tables_dump_rules(skb, &idx,
3512                                                    cb, table, chain, reset))
3513                                 goto done;
3514                 }
3515
3516                 if (ctx && ctx->table)
3517                         break;
3518         }
3519 done:
3520         rcu_read_unlock();
3521
3522         cb->args[0] = idx;
3523         return skb->len;
3524 }
3525
3526 static int nf_tables_dump_rules_start(struct netlink_callback *cb)
3527 {
3528         const struct nlattr * const *nla = cb->data;
3529         struct nft_rule_dump_ctx *ctx = NULL;
3530
3531         if (nla[NFTA_RULE_TABLE] || nla[NFTA_RULE_CHAIN]) {
3532                 ctx = kzalloc(sizeof(*ctx), GFP_ATOMIC);
3533                 if (!ctx)
3534                         return -ENOMEM;
3535
3536                 if (nla[NFTA_RULE_TABLE]) {
3537                         ctx->table = nla_strdup(nla[NFTA_RULE_TABLE],
3538                                                         GFP_ATOMIC);
3539                         if (!ctx->table) {
3540                                 kfree(ctx);
3541                                 return -ENOMEM;
3542                         }
3543                 }
3544                 if (nla[NFTA_RULE_CHAIN]) {
3545                         ctx->chain = nla_strdup(nla[NFTA_RULE_CHAIN],
3546                                                 GFP_ATOMIC);
3547                         if (!ctx->chain) {
3548                                 kfree(ctx->table);
3549                                 kfree(ctx);
3550                                 return -ENOMEM;
3551                         }
3552                 }
3553         }
3554
3555         cb->data = ctx;
3556         return 0;
3557 }
3558
3559 static int nf_tables_dump_rules_done(struct netlink_callback *cb)
3560 {
3561         struct nft_rule_dump_ctx *ctx = cb->data;
3562
3563         if (ctx) {
3564                 kfree(ctx->table);
3565                 kfree(ctx->chain);
3566                 kfree(ctx);
3567         }
3568         return 0;
3569 }
3570
3571 /* called with rcu_read_lock held */
3572 static int nf_tables_getrule(struct sk_buff *skb, const struct nfnl_info *info,
3573                              const struct nlattr * const nla[])
3574 {
3575         struct netlink_ext_ack *extack = info->extack;
3576         u8 genmask = nft_genmask_cur(info->net);
3577         u8 family = info->nfmsg->nfgen_family;
3578         const struct nft_chain *chain;
3579         const struct nft_rule *rule;
3580         struct net *net = info->net;
3581         struct nft_table *table;
3582         struct sk_buff *skb2;
3583         bool reset = false;
3584         int err;
3585
3586         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
3587                 struct netlink_dump_control c = {
3588                         .start= nf_tables_dump_rules_start,
3589                         .dump = nf_tables_dump_rules,
3590                         .done = nf_tables_dump_rules_done,
3591                         .module = THIS_MODULE,
3592                         .data = (void *)nla,
3593                 };
3594
3595                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
3596         }
3597
3598         table = nft_table_lookup(net, nla[NFTA_RULE_TABLE], family, genmask, 0);
3599         if (IS_ERR(table)) {
3600                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_TABLE]);
3601                 return PTR_ERR(table);
3602         }
3603
3604         chain = nft_chain_lookup(net, table, nla[NFTA_RULE_CHAIN], genmask);
3605         if (IS_ERR(chain)) {
3606                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_CHAIN]);
3607                 return PTR_ERR(chain);
3608         }
3609
3610         rule = nft_rule_lookup(chain, nla[NFTA_RULE_HANDLE]);
3611         if (IS_ERR(rule)) {
3612                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_HANDLE]);
3613                 return PTR_ERR(rule);
3614         }
3615
3616         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
3617         if (!skb2)
3618                 return -ENOMEM;
3619
3620         if (NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_GETRULE_RESET)
3621                 reset = true;
3622
3623         err = nf_tables_fill_rule_info(skb2, net, NETLINK_CB(skb).portid,
3624                                        info->nlh->nlmsg_seq, NFT_MSG_NEWRULE, 0,
3625                                        family, table, chain, rule, 0, reset);
3626         if (err < 0)
3627                 goto err_fill_rule_info;
3628
3629         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
3630
3631 err_fill_rule_info:
3632         kfree_skb(skb2);
3633         return err;
3634 }
3635
3636 void nf_tables_rule_destroy(const struct nft_ctx *ctx, struct nft_rule *rule)
3637 {
3638         struct nft_expr *expr, *next;
3639
3640         /*
3641          * Careful: some expressions might not be initialized in case this
3642          * is called on error from nf_tables_newrule().
3643          */
3644         expr = nft_expr_first(rule);
3645         while (nft_expr_more(rule, expr)) {
3646                 next = nft_expr_next(expr);
3647                 nf_tables_expr_destroy(ctx, expr);
3648                 expr = next;
3649         }
3650         kfree(rule);
3651 }
3652
3653 static void nf_tables_rule_release(const struct nft_ctx *ctx, struct nft_rule *rule)
3654 {
3655         nft_rule_expr_deactivate(ctx, rule, NFT_TRANS_RELEASE);
3656         nf_tables_rule_destroy(ctx, rule);
3657 }
3658
3659 int nft_chain_validate(const struct nft_ctx *ctx, const struct nft_chain *chain)
3660 {
3661         struct nft_expr *expr, *last;
3662         const struct nft_data *data;
3663         struct nft_rule *rule;
3664         int err;
3665
3666         if (ctx->level == NFT_JUMP_STACK_SIZE)
3667                 return -EMLINK;
3668
3669         list_for_each_entry(rule, &chain->rules, list) {
3670                 if (!nft_is_active_next(ctx->net, rule))
3671                         continue;
3672
3673                 nft_rule_for_each_expr(expr, last, rule) {
3674                         if (!expr->ops->validate)
3675                                 continue;
3676
3677                         err = expr->ops->validate(ctx, expr, &data);
3678                         if (err < 0)
3679                                 return err;
3680                 }
3681
3682                 cond_resched();
3683         }
3684
3685         return 0;
3686 }
3687 EXPORT_SYMBOL_GPL(nft_chain_validate);
3688
3689 static int nft_table_validate(struct net *net, const struct nft_table *table)
3690 {
3691         struct nft_chain *chain;
3692         struct nft_ctx ctx = {
3693                 .net    = net,
3694                 .family = table->family,
3695         };
3696         int err;
3697
3698         list_for_each_entry(chain, &table->chains, list) {
3699                 if (!nft_is_base_chain(chain))
3700                         continue;
3701
3702                 ctx.chain = chain;
3703                 err = nft_chain_validate(&ctx, chain);
3704                 if (err < 0)
3705                         return err;
3706         }
3707
3708         return 0;
3709 }
3710
3711 int nft_setelem_validate(const struct nft_ctx *ctx, struct nft_set *set,
3712                          const struct nft_set_iter *iter,
3713                          struct nft_set_elem *elem)
3714 {
3715         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
3716         struct nft_ctx *pctx = (struct nft_ctx *)ctx;
3717         const struct nft_data *data;
3718         int err;
3719
3720         if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
3721             *nft_set_ext_flags(ext) & NFT_SET_ELEM_INTERVAL_END)
3722                 return 0;
3723
3724         data = nft_set_ext_data(ext);
3725         switch (data->verdict.code) {
3726         case NFT_JUMP:
3727         case NFT_GOTO:
3728                 pctx->level++;
3729                 err = nft_chain_validate(ctx, data->verdict.chain);
3730                 if (err < 0)
3731                         return err;
3732                 pctx->level--;
3733                 break;
3734         default:
3735                 break;
3736         }
3737
3738         return 0;
3739 }
3740
3741 int nft_set_catchall_validate(const struct nft_ctx *ctx, struct nft_set *set)
3742 {
3743         u8 genmask = nft_genmask_next(ctx->net);
3744         struct nft_set_elem_catchall *catchall;
3745         struct nft_set_elem elem;
3746         struct nft_set_ext *ext;
3747         int ret = 0;
3748
3749         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
3750                 ext = nft_set_elem_ext(set, catchall->elem);
3751                 if (!nft_set_elem_active(ext, genmask))
3752                         continue;
3753
3754                 elem.priv = catchall->elem;
3755                 ret = nft_setelem_validate(ctx, set, NULL, &elem);
3756                 if (ret < 0)
3757                         return ret;
3758         }
3759
3760         return ret;
3761 }
3762
3763 static struct nft_rule *nft_rule_lookup_byid(const struct net *net,
3764                                              const struct nft_chain *chain,
3765                                              const struct nlattr *nla);
3766
3767 #define NFT_RULE_MAXEXPRS       128
3768
3769 static int nf_tables_newrule(struct sk_buff *skb, const struct nfnl_info *info,
3770                              const struct nlattr * const nla[])
3771 {
3772         struct nftables_pernet *nft_net = nft_pernet(info->net);
3773         struct netlink_ext_ack *extack = info->extack;
3774         unsigned int size, i, n, ulen = 0, usize = 0;
3775         u8 genmask = nft_genmask_next(info->net);
3776         struct nft_rule *rule, *old_rule = NULL;
3777         struct nft_expr_info *expr_info = NULL;
3778         u8 family = info->nfmsg->nfgen_family;
3779         struct nft_flow_rule *flow = NULL;
3780         struct net *net = info->net;
3781         struct nft_userdata *udata;
3782         struct nft_table *table;
3783         struct nft_chain *chain;
3784         struct nft_trans *trans;
3785         u64 handle, pos_handle;
3786         struct nft_expr *expr;
3787         struct nft_ctx ctx;
3788         struct nlattr *tmp;
3789         int err, rem;
3790
3791         lockdep_assert_held(&nft_net->commit_mutex);
3792
3793         table = nft_table_lookup(net, nla[NFTA_RULE_TABLE], family, genmask,
3794                                  NETLINK_CB(skb).portid);
3795         if (IS_ERR(table)) {
3796                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_TABLE]);
3797                 return PTR_ERR(table);
3798         }
3799
3800         if (nla[NFTA_RULE_CHAIN]) {
3801                 chain = nft_chain_lookup(net, table, nla[NFTA_RULE_CHAIN],
3802                                          genmask);
3803                 if (IS_ERR(chain)) {
3804                         NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_CHAIN]);
3805                         return PTR_ERR(chain);
3806                 }
3807                 if (nft_chain_is_bound(chain))
3808                         return -EOPNOTSUPP;
3809
3810         } else if (nla[NFTA_RULE_CHAIN_ID]) {
3811                 chain = nft_chain_lookup_byid(net, table, nla[NFTA_RULE_CHAIN_ID]);
3812                 if (IS_ERR(chain)) {
3813                         NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_CHAIN_ID]);
3814                         return PTR_ERR(chain);
3815                 }
3816         } else {
3817                 return -EINVAL;
3818         }
3819
3820         if (nla[NFTA_RULE_HANDLE]) {
3821                 handle = be64_to_cpu(nla_get_be64(nla[NFTA_RULE_HANDLE]));
3822                 rule = __nft_rule_lookup(chain, handle);
3823                 if (IS_ERR(rule)) {
3824                         NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_HANDLE]);
3825                         return PTR_ERR(rule);
3826                 }
3827
3828                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
3829                         NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_HANDLE]);
3830                         return -EEXIST;
3831                 }
3832                 if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
3833                         old_rule = rule;
3834                 else
3835                         return -EOPNOTSUPP;
3836         } else {
3837                 if (!(info->nlh->nlmsg_flags & NLM_F_CREATE) ||
3838                     info->nlh->nlmsg_flags & NLM_F_REPLACE)
3839                         return -EINVAL;
3840                 handle = nf_tables_alloc_handle(table);
3841
3842                 if (chain->use == UINT_MAX)
3843                         return -EOVERFLOW;
3844
3845                 if (nla[NFTA_RULE_POSITION]) {
3846                         pos_handle = be64_to_cpu(nla_get_be64(nla[NFTA_RULE_POSITION]));
3847                         old_rule = __nft_rule_lookup(chain, pos_handle);
3848                         if (IS_ERR(old_rule)) {
3849                                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_POSITION]);
3850                                 return PTR_ERR(old_rule);
3851                         }
3852                 } else if (nla[NFTA_RULE_POSITION_ID]) {
3853                         old_rule = nft_rule_lookup_byid(net, chain, nla[NFTA_RULE_POSITION_ID]);
3854                         if (IS_ERR(old_rule)) {
3855                                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_POSITION_ID]);
3856                                 return PTR_ERR(old_rule);
3857                         }
3858                 }
3859         }
3860
3861         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, chain, nla);
3862
3863         n = 0;
3864         size = 0;
3865         if (nla[NFTA_RULE_EXPRESSIONS]) {
3866                 expr_info = kvmalloc_array(NFT_RULE_MAXEXPRS,
3867                                            sizeof(struct nft_expr_info),
3868                                            GFP_KERNEL);
3869                 if (!expr_info)
3870                         return -ENOMEM;
3871
3872                 nla_for_each_nested(tmp, nla[NFTA_RULE_EXPRESSIONS], rem) {
3873                         err = -EINVAL;
3874                         if (nla_type(tmp) != NFTA_LIST_ELEM)
3875                                 goto err_release_expr;
3876                         if (n == NFT_RULE_MAXEXPRS)
3877                                 goto err_release_expr;
3878                         err = nf_tables_expr_parse(&ctx, tmp, &expr_info[n]);
3879                         if (err < 0) {
3880                                 NL_SET_BAD_ATTR(extack, tmp);
3881                                 goto err_release_expr;
3882                         }
3883                         size += expr_info[n].ops->size;
3884                         n++;
3885                 }
3886         }
3887         /* Check for overflow of dlen field */
3888         err = -EFBIG;
3889         if (size >= 1 << 12)
3890                 goto err_release_expr;
3891
3892         if (nla[NFTA_RULE_USERDATA]) {
3893                 ulen = nla_len(nla[NFTA_RULE_USERDATA]);
3894                 if (ulen > 0)
3895                         usize = sizeof(struct nft_userdata) + ulen;
3896         }
3897
3898         err = -ENOMEM;
3899         rule = kzalloc(sizeof(*rule) + size + usize, GFP_KERNEL_ACCOUNT);
3900         if (rule == NULL)
3901                 goto err_release_expr;
3902
3903         nft_activate_next(net, rule);
3904
3905         rule->handle = handle;
3906         rule->dlen   = size;
3907         rule->udata  = ulen ? 1 : 0;
3908
3909         if (ulen) {
3910                 udata = nft_userdata(rule);
3911                 udata->len = ulen - 1;
3912                 nla_memcpy(udata->data, nla[NFTA_RULE_USERDATA], ulen);
3913         }
3914
3915         expr = nft_expr_first(rule);
3916         for (i = 0; i < n; i++) {
3917                 err = nf_tables_newexpr(&ctx, &expr_info[i], expr);
3918                 if (err < 0) {
3919                         NL_SET_BAD_ATTR(extack, expr_info[i].attr);
3920                         goto err_release_rule;
3921                 }
3922
3923                 if (expr_info[i].ops->validate)
3924                         nft_validate_state_update(table, NFT_VALIDATE_NEED);
3925
3926                 expr_info[i].ops = NULL;
3927                 expr = nft_expr_next(expr);
3928         }
3929
3930         if (chain->flags & NFT_CHAIN_HW_OFFLOAD) {
3931                 flow = nft_flow_rule_create(net, rule);
3932                 if (IS_ERR(flow)) {
3933                         err = PTR_ERR(flow);
3934                         goto err_release_rule;
3935                 }
3936         }
3937
3938         if (info->nlh->nlmsg_flags & NLM_F_REPLACE) {
3939                 err = nft_delrule(&ctx, old_rule);
3940                 if (err < 0)
3941                         goto err_destroy_flow_rule;
3942
3943                 trans = nft_trans_rule_add(&ctx, NFT_MSG_NEWRULE, rule);
3944                 if (trans == NULL) {
3945                         err = -ENOMEM;
3946                         goto err_destroy_flow_rule;
3947                 }
3948                 list_add_tail_rcu(&rule->list, &old_rule->list);
3949         } else {
3950                 trans = nft_trans_rule_add(&ctx, NFT_MSG_NEWRULE, rule);
3951                 if (!trans) {
3952                         err = -ENOMEM;
3953                         goto err_destroy_flow_rule;
3954                 }
3955
3956                 if (info->nlh->nlmsg_flags & NLM_F_APPEND) {
3957                         if (old_rule)
3958                                 list_add_rcu(&rule->list, &old_rule->list);
3959                         else
3960                                 list_add_tail_rcu(&rule->list, &chain->rules);
3961                  } else {
3962                         if (old_rule)
3963                                 list_add_tail_rcu(&rule->list, &old_rule->list);
3964                         else
3965                                 list_add_rcu(&rule->list, &chain->rules);
3966                 }
3967         }
3968         kvfree(expr_info);
3969         chain->use++;
3970
3971         if (flow)
3972                 nft_trans_flow_rule(trans) = flow;
3973
3974         if (table->validate_state == NFT_VALIDATE_DO)
3975                 return nft_table_validate(net, table);
3976
3977         return 0;
3978
3979 err_destroy_flow_rule:
3980         if (flow)
3981                 nft_flow_rule_destroy(flow);
3982 err_release_rule:
3983         nft_rule_expr_deactivate(&ctx, rule, NFT_TRANS_PREPARE_ERROR);
3984         nf_tables_rule_destroy(&ctx, rule);
3985 err_release_expr:
3986         for (i = 0; i < n; i++) {
3987                 if (expr_info[i].ops) {
3988                         module_put(expr_info[i].ops->type->owner);
3989                         if (expr_info[i].ops->type->release_ops)
3990                                 expr_info[i].ops->type->release_ops(expr_info[i].ops);
3991                 }
3992         }
3993         kvfree(expr_info);
3994
3995         return err;
3996 }
3997
3998 static struct nft_rule *nft_rule_lookup_byid(const struct net *net,
3999                                              const struct nft_chain *chain,
4000                                              const struct nlattr *nla)
4001 {
4002         struct nftables_pernet *nft_net = nft_pernet(net);
4003         u32 id = ntohl(nla_get_be32(nla));
4004         struct nft_trans *trans;
4005
4006         list_for_each_entry(trans, &nft_net->commit_list, list) {
4007                 if (trans->msg_type == NFT_MSG_NEWRULE &&
4008                     trans->ctx.chain == chain &&
4009                     id == nft_trans_rule_id(trans))
4010                         return nft_trans_rule(trans);
4011         }
4012         return ERR_PTR(-ENOENT);
4013 }
4014
4015 static int nf_tables_delrule(struct sk_buff *skb, const struct nfnl_info *info,
4016                              const struct nlattr * const nla[])
4017 {
4018         struct netlink_ext_ack *extack = info->extack;
4019         u8 genmask = nft_genmask_next(info->net);
4020         u8 family = info->nfmsg->nfgen_family;
4021         struct nft_chain *chain = NULL;
4022         struct net *net = info->net;
4023         struct nft_table *table;
4024         struct nft_rule *rule;
4025         struct nft_ctx ctx;
4026         int err = 0;
4027
4028         table = nft_table_lookup(net, nla[NFTA_RULE_TABLE], family, genmask,
4029                                  NETLINK_CB(skb).portid);
4030         if (IS_ERR(table)) {
4031                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_TABLE]);
4032                 return PTR_ERR(table);
4033         }
4034
4035         if (nla[NFTA_RULE_CHAIN]) {
4036                 chain = nft_chain_lookup(net, table, nla[NFTA_RULE_CHAIN],
4037                                          genmask);
4038                 if (IS_ERR(chain)) {
4039                         if (PTR_ERR(chain) == -ENOENT &&
4040                             NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYRULE)
4041                                 return 0;
4042
4043                         NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_CHAIN]);
4044                         return PTR_ERR(chain);
4045                 }
4046                 if (nft_chain_is_bound(chain))
4047                         return -EOPNOTSUPP;
4048         }
4049
4050         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, chain, nla);
4051
4052         if (chain) {
4053                 if (nla[NFTA_RULE_HANDLE]) {
4054                         rule = nft_rule_lookup(chain, nla[NFTA_RULE_HANDLE]);
4055                         if (IS_ERR(rule)) {
4056                                 if (PTR_ERR(rule) == -ENOENT &&
4057                                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYRULE)
4058                                         return 0;
4059
4060                                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_HANDLE]);
4061                                 return PTR_ERR(rule);
4062                         }
4063
4064                         err = nft_delrule(&ctx, rule);
4065                 } else if (nla[NFTA_RULE_ID]) {
4066                         rule = nft_rule_lookup_byid(net, chain, nla[NFTA_RULE_ID]);
4067                         if (IS_ERR(rule)) {
4068                                 NL_SET_BAD_ATTR(extack, nla[NFTA_RULE_ID]);
4069                                 return PTR_ERR(rule);
4070                         }
4071
4072                         err = nft_delrule(&ctx, rule);
4073                 } else {
4074                         err = nft_delrule_by_chain(&ctx);
4075                 }
4076         } else {
4077                 list_for_each_entry(chain, &table->chains, list) {
4078                         if (!nft_is_active_next(net, chain))
4079                                 continue;
4080
4081                         ctx.chain = chain;
4082                         err = nft_delrule_by_chain(&ctx);
4083                         if (err < 0)
4084                                 break;
4085                 }
4086         }
4087
4088         return err;
4089 }
4090
4091 /*
4092  * Sets
4093  */
4094 static const struct nft_set_type *nft_set_types[] = {
4095         &nft_set_hash_fast_type,
4096         &nft_set_hash_type,
4097         &nft_set_rhash_type,
4098         &nft_set_bitmap_type,
4099         &nft_set_rbtree_type,
4100 #if defined(CONFIG_X86_64) && !defined(CONFIG_UML)
4101         &nft_set_pipapo_avx2_type,
4102 #endif
4103         &nft_set_pipapo_type,
4104 };
4105
4106 #define NFT_SET_FEATURES        (NFT_SET_INTERVAL | NFT_SET_MAP | \
4107                                  NFT_SET_TIMEOUT | NFT_SET_OBJECT | \
4108                                  NFT_SET_EVAL)
4109
4110 static bool nft_set_ops_candidate(const struct nft_set_type *type, u32 flags)
4111 {
4112         return (flags & type->features) == (flags & NFT_SET_FEATURES);
4113 }
4114
4115 /*
4116  * Select a set implementation based on the data characteristics and the
4117  * given policy. The total memory use might not be known if no size is
4118  * given, in that case the amount of memory per element is used.
4119  */
4120 static const struct nft_set_ops *
4121 nft_select_set_ops(const struct nft_ctx *ctx,
4122                    const struct nlattr * const nla[],
4123                    const struct nft_set_desc *desc)
4124 {
4125         struct nftables_pernet *nft_net = nft_pernet(ctx->net);
4126         const struct nft_set_ops *ops, *bops;
4127         struct nft_set_estimate est, best;
4128         const struct nft_set_type *type;
4129         u32 flags = 0;
4130         int i;
4131
4132         lockdep_assert_held(&nft_net->commit_mutex);
4133         lockdep_nfnl_nft_mutex_not_held();
4134
4135         if (nla[NFTA_SET_FLAGS] != NULL)
4136                 flags = ntohl(nla_get_be32(nla[NFTA_SET_FLAGS]));
4137
4138         bops        = NULL;
4139         best.size   = ~0;
4140         best.lookup = ~0;
4141         best.space  = ~0;
4142
4143         for (i = 0; i < ARRAY_SIZE(nft_set_types); i++) {
4144                 type = nft_set_types[i];
4145                 ops = &type->ops;
4146
4147                 if (!nft_set_ops_candidate(type, flags))
4148                         continue;
4149                 if (!ops->estimate(desc, flags, &est))
4150                         continue;
4151
4152                 switch (desc->policy) {
4153                 case NFT_SET_POL_PERFORMANCE:
4154                         if (est.lookup < best.lookup)
4155                                 break;
4156                         if (est.lookup == best.lookup &&
4157                             est.space < best.space)
4158                                 break;
4159                         continue;
4160                 case NFT_SET_POL_MEMORY:
4161                         if (!desc->size) {
4162                                 if (est.space < best.space)
4163                                         break;
4164                                 if (est.space == best.space &&
4165                                     est.lookup < best.lookup)
4166                                         break;
4167                         } else if (est.size < best.size || !bops) {
4168                                 break;
4169                         }
4170                         continue;
4171                 default:
4172                         break;
4173                 }
4174
4175                 bops = ops;
4176                 best = est;
4177         }
4178
4179         if (bops != NULL)
4180                 return bops;
4181
4182         return ERR_PTR(-EOPNOTSUPP);
4183 }
4184
4185 static const struct nla_policy nft_set_policy[NFTA_SET_MAX + 1] = {
4186         [NFTA_SET_TABLE]                = { .type = NLA_STRING,
4187                                             .len = NFT_TABLE_MAXNAMELEN - 1 },
4188         [NFTA_SET_NAME]                 = { .type = NLA_STRING,
4189                                             .len = NFT_SET_MAXNAMELEN - 1 },
4190         [NFTA_SET_FLAGS]                = { .type = NLA_U32 },
4191         [NFTA_SET_KEY_TYPE]             = { .type = NLA_U32 },
4192         [NFTA_SET_KEY_LEN]              = { .type = NLA_U32 },
4193         [NFTA_SET_DATA_TYPE]            = { .type = NLA_U32 },
4194         [NFTA_SET_DATA_LEN]             = { .type = NLA_U32 },
4195         [NFTA_SET_POLICY]               = { .type = NLA_U32 },
4196         [NFTA_SET_DESC]                 = { .type = NLA_NESTED },
4197         [NFTA_SET_ID]                   = { .type = NLA_U32 },
4198         [NFTA_SET_TIMEOUT]              = { .type = NLA_U64 },
4199         [NFTA_SET_GC_INTERVAL]          = { .type = NLA_U32 },
4200         [NFTA_SET_USERDATA]             = { .type = NLA_BINARY,
4201                                             .len  = NFT_USERDATA_MAXLEN },
4202         [NFTA_SET_OBJ_TYPE]             = { .type = NLA_U32 },
4203         [NFTA_SET_HANDLE]               = { .type = NLA_U64 },
4204         [NFTA_SET_EXPR]                 = { .type = NLA_NESTED },
4205         [NFTA_SET_EXPRESSIONS]          = { .type = NLA_NESTED },
4206 };
4207
4208 static const struct nla_policy nft_set_desc_policy[NFTA_SET_DESC_MAX + 1] = {
4209         [NFTA_SET_DESC_SIZE]            = { .type = NLA_U32 },
4210         [NFTA_SET_DESC_CONCAT]          = { .type = NLA_NESTED },
4211 };
4212
4213 static struct nft_set *nft_set_lookup(const struct nft_table *table,
4214                                       const struct nlattr *nla, u8 genmask)
4215 {
4216         struct nft_set *set;
4217
4218         if (nla == NULL)
4219                 return ERR_PTR(-EINVAL);
4220
4221         list_for_each_entry_rcu(set, &table->sets, list) {
4222                 if (!nla_strcmp(nla, set->name) &&
4223                     nft_active_genmask(set, genmask))
4224                         return set;
4225         }
4226         return ERR_PTR(-ENOENT);
4227 }
4228
4229 static struct nft_set *nft_set_lookup_byhandle(const struct nft_table *table,
4230                                                const struct nlattr *nla,
4231                                                u8 genmask)
4232 {
4233         struct nft_set *set;
4234
4235         list_for_each_entry(set, &table->sets, list) {
4236                 if (be64_to_cpu(nla_get_be64(nla)) == set->handle &&
4237                     nft_active_genmask(set, genmask))
4238                         return set;
4239         }
4240         return ERR_PTR(-ENOENT);
4241 }
4242
4243 static struct nft_set *nft_set_lookup_byid(const struct net *net,
4244                                            const struct nft_table *table,
4245                                            const struct nlattr *nla, u8 genmask)
4246 {
4247         struct nftables_pernet *nft_net = nft_pernet(net);
4248         u32 id = ntohl(nla_get_be32(nla));
4249         struct nft_trans *trans;
4250
4251         list_for_each_entry(trans, &nft_net->commit_list, list) {
4252                 if (trans->msg_type == NFT_MSG_NEWSET) {
4253                         struct nft_set *set = nft_trans_set(trans);
4254
4255                         if (id == nft_trans_set_id(trans) &&
4256                             set->table == table &&
4257                             nft_active_genmask(set, genmask))
4258                                 return set;
4259                 }
4260         }
4261         return ERR_PTR(-ENOENT);
4262 }
4263
4264 struct nft_set *nft_set_lookup_global(const struct net *net,
4265                                       const struct nft_table *table,
4266                                       const struct nlattr *nla_set_name,
4267                                       const struct nlattr *nla_set_id,
4268                                       u8 genmask)
4269 {
4270         struct nft_set *set;
4271
4272         set = nft_set_lookup(table, nla_set_name, genmask);
4273         if (IS_ERR(set)) {
4274                 if (!nla_set_id)
4275                         return set;
4276
4277                 set = nft_set_lookup_byid(net, table, nla_set_id, genmask);
4278         }
4279         return set;
4280 }
4281 EXPORT_SYMBOL_GPL(nft_set_lookup_global);
4282
4283 static int nf_tables_set_alloc_name(struct nft_ctx *ctx, struct nft_set *set,
4284                                     const char *name)
4285 {
4286         const struct nft_set *i;
4287         const char *p;
4288         unsigned long *inuse;
4289         unsigned int n = 0, min = 0;
4290
4291         p = strchr(name, '%');
4292         if (p != NULL) {
4293                 if (p[1] != 'd' || strchr(p + 2, '%'))
4294                         return -EINVAL;
4295
4296                 inuse = (unsigned long *)get_zeroed_page(GFP_KERNEL);
4297                 if (inuse == NULL)
4298                         return -ENOMEM;
4299 cont:
4300                 list_for_each_entry(i, &ctx->table->sets, list) {
4301                         int tmp;
4302
4303                         if (!nft_is_active_next(ctx->net, i))
4304                                 continue;
4305                         if (!sscanf(i->name, name, &tmp))
4306                                 continue;
4307                         if (tmp < min || tmp >= min + BITS_PER_BYTE * PAGE_SIZE)
4308                                 continue;
4309
4310                         set_bit(tmp - min, inuse);
4311                 }
4312
4313                 n = find_first_zero_bit(inuse, BITS_PER_BYTE * PAGE_SIZE);
4314                 if (n >= BITS_PER_BYTE * PAGE_SIZE) {
4315                         min += BITS_PER_BYTE * PAGE_SIZE;
4316                         memset(inuse, 0, PAGE_SIZE);
4317                         goto cont;
4318                 }
4319                 free_page((unsigned long)inuse);
4320         }
4321
4322         set->name = kasprintf(GFP_KERNEL_ACCOUNT, name, min + n);
4323         if (!set->name)
4324                 return -ENOMEM;
4325
4326         list_for_each_entry(i, &ctx->table->sets, list) {
4327                 if (!nft_is_active_next(ctx->net, i))
4328                         continue;
4329                 if (!strcmp(set->name, i->name)) {
4330                         kfree(set->name);
4331                         set->name = NULL;
4332                         return -ENFILE;
4333                 }
4334         }
4335         return 0;
4336 }
4337
4338 int nf_msecs_to_jiffies64(const struct nlattr *nla, u64 *result)
4339 {
4340         u64 ms = be64_to_cpu(nla_get_be64(nla));
4341         u64 max = (u64)(~((u64)0));
4342
4343         max = div_u64(max, NSEC_PER_MSEC);
4344         if (ms >= max)
4345                 return -ERANGE;
4346
4347         ms *= NSEC_PER_MSEC;
4348         *result = nsecs_to_jiffies64(ms);
4349         return 0;
4350 }
4351
4352 __be64 nf_jiffies64_to_msecs(u64 input)
4353 {
4354         return cpu_to_be64(jiffies64_to_msecs(input));
4355 }
4356
4357 static int nf_tables_fill_set_concat(struct sk_buff *skb,
4358                                      const struct nft_set *set)
4359 {
4360         struct nlattr *concat, *field;
4361         int i;
4362
4363         concat = nla_nest_start_noflag(skb, NFTA_SET_DESC_CONCAT);
4364         if (!concat)
4365                 return -ENOMEM;
4366
4367         for (i = 0; i < set->field_count; i++) {
4368                 field = nla_nest_start_noflag(skb, NFTA_LIST_ELEM);
4369                 if (!field)
4370                         return -ENOMEM;
4371
4372                 if (nla_put_be32(skb, NFTA_SET_FIELD_LEN,
4373                                  htonl(set->field_len[i])))
4374                         return -ENOMEM;
4375
4376                 nla_nest_end(skb, field);
4377         }
4378
4379         nla_nest_end(skb, concat);
4380
4381         return 0;
4382 }
4383
4384 static int nf_tables_fill_set(struct sk_buff *skb, const struct nft_ctx *ctx,
4385                               const struct nft_set *set, u16 event, u16 flags)
4386 {
4387         u64 timeout = READ_ONCE(set->timeout);
4388         u32 gc_int = READ_ONCE(set->gc_int);
4389         u32 portid = ctx->portid;
4390         struct nlmsghdr *nlh;
4391         struct nlattr *nest;
4392         u32 seq = ctx->seq;
4393         int i;
4394
4395         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
4396         nlh = nfnl_msg_put(skb, portid, seq, event, flags, ctx->family,
4397                            NFNETLINK_V0, nft_base_seq(ctx->net));
4398         if (!nlh)
4399                 goto nla_put_failure;
4400
4401         if (nla_put_string(skb, NFTA_SET_TABLE, ctx->table->name))
4402                 goto nla_put_failure;
4403         if (nla_put_string(skb, NFTA_SET_NAME, set->name))
4404                 goto nla_put_failure;
4405         if (nla_put_be64(skb, NFTA_SET_HANDLE, cpu_to_be64(set->handle),
4406                          NFTA_SET_PAD))
4407                 goto nla_put_failure;
4408
4409         if (event == NFT_MSG_DELSET) {
4410                 nlmsg_end(skb, nlh);
4411                 return 0;
4412         }
4413
4414         if (set->flags != 0)
4415                 if (nla_put_be32(skb, NFTA_SET_FLAGS, htonl(set->flags)))
4416                         goto nla_put_failure;
4417
4418         if (nla_put_be32(skb, NFTA_SET_KEY_TYPE, htonl(set->ktype)))
4419                 goto nla_put_failure;
4420         if (nla_put_be32(skb, NFTA_SET_KEY_LEN, htonl(set->klen)))
4421                 goto nla_put_failure;
4422         if (set->flags & NFT_SET_MAP) {
4423                 if (nla_put_be32(skb, NFTA_SET_DATA_TYPE, htonl(set->dtype)))
4424                         goto nla_put_failure;
4425                 if (nla_put_be32(skb, NFTA_SET_DATA_LEN, htonl(set->dlen)))
4426                         goto nla_put_failure;
4427         }
4428         if (set->flags & NFT_SET_OBJECT &&
4429             nla_put_be32(skb, NFTA_SET_OBJ_TYPE, htonl(set->objtype)))
4430                 goto nla_put_failure;
4431
4432         if (timeout &&
4433             nla_put_be64(skb, NFTA_SET_TIMEOUT,
4434                          nf_jiffies64_to_msecs(timeout),
4435                          NFTA_SET_PAD))
4436                 goto nla_put_failure;
4437         if (gc_int &&
4438             nla_put_be32(skb, NFTA_SET_GC_INTERVAL, htonl(gc_int)))
4439                 goto nla_put_failure;
4440
4441         if (set->policy != NFT_SET_POL_PERFORMANCE) {
4442                 if (nla_put_be32(skb, NFTA_SET_POLICY, htonl(set->policy)))
4443                         goto nla_put_failure;
4444         }
4445
4446         if (set->udata &&
4447             nla_put(skb, NFTA_SET_USERDATA, set->udlen, set->udata))
4448                 goto nla_put_failure;
4449
4450         nest = nla_nest_start_noflag(skb, NFTA_SET_DESC);
4451         if (!nest)
4452                 goto nla_put_failure;
4453         if (set->size &&
4454             nla_put_be32(skb, NFTA_SET_DESC_SIZE, htonl(set->size)))
4455                 goto nla_put_failure;
4456
4457         if (set->field_count > 1 &&
4458             nf_tables_fill_set_concat(skb, set))
4459                 goto nla_put_failure;
4460
4461         nla_nest_end(skb, nest);
4462
4463         if (set->num_exprs == 1) {
4464                 nest = nla_nest_start_noflag(skb, NFTA_SET_EXPR);
4465                 if (nf_tables_fill_expr_info(skb, set->exprs[0], false) < 0)
4466                         goto nla_put_failure;
4467
4468                 nla_nest_end(skb, nest);
4469         } else if (set->num_exprs > 1) {
4470                 nest = nla_nest_start_noflag(skb, NFTA_SET_EXPRESSIONS);
4471                 if (nest == NULL)
4472                         goto nla_put_failure;
4473
4474                 for (i = 0; i < set->num_exprs; i++) {
4475                         if (nft_expr_dump(skb, NFTA_LIST_ELEM,
4476                                           set->exprs[i], false) < 0)
4477                                 goto nla_put_failure;
4478                 }
4479                 nla_nest_end(skb, nest);
4480         }
4481
4482         nlmsg_end(skb, nlh);
4483         return 0;
4484
4485 nla_put_failure:
4486         nlmsg_trim(skb, nlh);
4487         return -1;
4488 }
4489
4490 static void nf_tables_set_notify(const struct nft_ctx *ctx,
4491                                  const struct nft_set *set, int event,
4492                                  gfp_t gfp_flags)
4493 {
4494         struct nftables_pernet *nft_net = nft_pernet(ctx->net);
4495         u32 portid = ctx->portid;
4496         struct sk_buff *skb;
4497         u16 flags = 0;
4498         int err;
4499
4500         if (!ctx->report &&
4501             !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
4502                 return;
4503
4504         skb = nlmsg_new(NLMSG_GOODSIZE, gfp_flags);
4505         if (skb == NULL)
4506                 goto err;
4507
4508         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
4509                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
4510
4511         err = nf_tables_fill_set(skb, ctx, set, event, flags);
4512         if (err < 0) {
4513                 kfree_skb(skb);
4514                 goto err;
4515         }
4516
4517         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
4518         return;
4519 err:
4520         nfnetlink_set_err(ctx->net, portid, NFNLGRP_NFTABLES, -ENOBUFS);
4521 }
4522
4523 static int nf_tables_dump_sets(struct sk_buff *skb, struct netlink_callback *cb)
4524 {
4525         const struct nft_set *set;
4526         unsigned int idx, s_idx = cb->args[0];
4527         struct nft_table *table, *cur_table = (struct nft_table *)cb->args[2];
4528         struct net *net = sock_net(skb->sk);
4529         struct nft_ctx *ctx = cb->data, ctx_set;
4530         struct nftables_pernet *nft_net;
4531
4532         if (cb->args[1])
4533                 return skb->len;
4534
4535         rcu_read_lock();
4536         nft_net = nft_pernet(net);
4537         cb->seq = READ_ONCE(nft_net->base_seq);
4538
4539         list_for_each_entry_rcu(table, &nft_net->tables, list) {
4540                 if (ctx->family != NFPROTO_UNSPEC &&
4541                     ctx->family != table->family)
4542                         continue;
4543
4544                 if (ctx->table && ctx->table != table)
4545                         continue;
4546
4547                 if (cur_table) {
4548                         if (cur_table != table)
4549                                 continue;
4550
4551                         cur_table = NULL;
4552                 }
4553                 idx = 0;
4554                 list_for_each_entry_rcu(set, &table->sets, list) {
4555                         if (idx < s_idx)
4556                                 goto cont;
4557                         if (!nft_is_active(net, set))
4558                                 goto cont;
4559
4560                         ctx_set = *ctx;
4561                         ctx_set.table = table;
4562                         ctx_set.family = table->family;
4563
4564                         if (nf_tables_fill_set(skb, &ctx_set, set,
4565                                                NFT_MSG_NEWSET,
4566                                                NLM_F_MULTI) < 0) {
4567                                 cb->args[0] = idx;
4568                                 cb->args[2] = (unsigned long) table;
4569                                 goto done;
4570                         }
4571                         nl_dump_check_consistent(cb, nlmsg_hdr(skb));
4572 cont:
4573                         idx++;
4574                 }
4575                 if (s_idx)
4576                         s_idx = 0;
4577         }
4578         cb->args[1] = 1;
4579 done:
4580         rcu_read_unlock();
4581         return skb->len;
4582 }
4583
4584 static int nf_tables_dump_sets_start(struct netlink_callback *cb)
4585 {
4586         struct nft_ctx *ctx_dump = NULL;
4587
4588         ctx_dump = kmemdup(cb->data, sizeof(*ctx_dump), GFP_ATOMIC);
4589         if (ctx_dump == NULL)
4590                 return -ENOMEM;
4591
4592         cb->data = ctx_dump;
4593         return 0;
4594 }
4595
4596 static int nf_tables_dump_sets_done(struct netlink_callback *cb)
4597 {
4598         kfree(cb->data);
4599         return 0;
4600 }
4601
4602 /* called with rcu_read_lock held */
4603 static int nf_tables_getset(struct sk_buff *skb, const struct nfnl_info *info,
4604                             const struct nlattr * const nla[])
4605 {
4606         struct netlink_ext_ack *extack = info->extack;
4607         u8 genmask = nft_genmask_cur(info->net);
4608         u8 family = info->nfmsg->nfgen_family;
4609         struct nft_table *table = NULL;
4610         struct net *net = info->net;
4611         const struct nft_set *set;
4612         struct sk_buff *skb2;
4613         struct nft_ctx ctx;
4614         int err;
4615
4616         if (nla[NFTA_SET_TABLE]) {
4617                 table = nft_table_lookup(net, nla[NFTA_SET_TABLE], family,
4618                                          genmask, 0);
4619                 if (IS_ERR(table)) {
4620                         NL_SET_BAD_ATTR(extack, nla[NFTA_SET_TABLE]);
4621                         return PTR_ERR(table);
4622                 }
4623         }
4624
4625         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
4626
4627         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
4628                 struct netlink_dump_control c = {
4629                         .start = nf_tables_dump_sets_start,
4630                         .dump = nf_tables_dump_sets,
4631                         .done = nf_tables_dump_sets_done,
4632                         .data = &ctx,
4633                         .module = THIS_MODULE,
4634                 };
4635
4636                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
4637         }
4638
4639         /* Only accept unspec with dump */
4640         if (info->nfmsg->nfgen_family == NFPROTO_UNSPEC)
4641                 return -EAFNOSUPPORT;
4642         if (!nla[NFTA_SET_TABLE])
4643                 return -EINVAL;
4644
4645         set = nft_set_lookup(table, nla[NFTA_SET_NAME], genmask);
4646         if (IS_ERR(set))
4647                 return PTR_ERR(set);
4648
4649         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
4650         if (skb2 == NULL)
4651                 return -ENOMEM;
4652
4653         err = nf_tables_fill_set(skb2, &ctx, set, NFT_MSG_NEWSET, 0);
4654         if (err < 0)
4655                 goto err_fill_set_info;
4656
4657         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
4658
4659 err_fill_set_info:
4660         kfree_skb(skb2);
4661         return err;
4662 }
4663
4664 static const struct nla_policy nft_concat_policy[NFTA_SET_FIELD_MAX + 1] = {
4665         [NFTA_SET_FIELD_LEN]    = { .type = NLA_U32 },
4666 };
4667
4668 static int nft_set_desc_concat_parse(const struct nlattr *attr,
4669                                      struct nft_set_desc *desc)
4670 {
4671         struct nlattr *tb[NFTA_SET_FIELD_MAX + 1];
4672         u32 len;
4673         int err;
4674
4675         if (desc->field_count >= ARRAY_SIZE(desc->field_len))
4676                 return -E2BIG;
4677
4678         err = nla_parse_nested_deprecated(tb, NFTA_SET_FIELD_MAX, attr,
4679                                           nft_concat_policy, NULL);
4680         if (err < 0)
4681                 return err;
4682
4683         if (!tb[NFTA_SET_FIELD_LEN])
4684                 return -EINVAL;
4685
4686         len = ntohl(nla_get_be32(tb[NFTA_SET_FIELD_LEN]));
4687         if (!len || len > U8_MAX)
4688                 return -EINVAL;
4689
4690         desc->field_len[desc->field_count++] = len;
4691
4692         return 0;
4693 }
4694
4695 static int nft_set_desc_concat(struct nft_set_desc *desc,
4696                                const struct nlattr *nla)
4697 {
4698         struct nlattr *attr;
4699         u32 num_regs = 0;
4700         int rem, err, i;
4701
4702         nla_for_each_nested(attr, nla, rem) {
4703                 if (nla_type(attr) != NFTA_LIST_ELEM)
4704                         return -EINVAL;
4705
4706                 err = nft_set_desc_concat_parse(attr, desc);
4707                 if (err < 0)
4708                         return err;
4709         }
4710
4711         for (i = 0; i < desc->field_count; i++)
4712                 num_regs += DIV_ROUND_UP(desc->field_len[i], sizeof(u32));
4713
4714         if (num_regs > NFT_REG32_COUNT)
4715                 return -E2BIG;
4716
4717         return 0;
4718 }
4719
4720 static int nf_tables_set_desc_parse(struct nft_set_desc *desc,
4721                                     const struct nlattr *nla)
4722 {
4723         struct nlattr *da[NFTA_SET_DESC_MAX + 1];
4724         int err;
4725
4726         err = nla_parse_nested_deprecated(da, NFTA_SET_DESC_MAX, nla,
4727                                           nft_set_desc_policy, NULL);
4728         if (err < 0)
4729                 return err;
4730
4731         if (da[NFTA_SET_DESC_SIZE] != NULL)
4732                 desc->size = ntohl(nla_get_be32(da[NFTA_SET_DESC_SIZE]));
4733         if (da[NFTA_SET_DESC_CONCAT])
4734                 err = nft_set_desc_concat(desc, da[NFTA_SET_DESC_CONCAT]);
4735
4736         return err;
4737 }
4738
4739 static int nft_set_expr_alloc(struct nft_ctx *ctx, struct nft_set *set,
4740                               const struct nlattr * const *nla,
4741                               struct nft_expr **exprs, int *num_exprs,
4742                               u32 flags)
4743 {
4744         struct nft_expr *expr;
4745         int err, i;
4746
4747         if (nla[NFTA_SET_EXPR]) {
4748                 expr = nft_set_elem_expr_alloc(ctx, set, nla[NFTA_SET_EXPR]);
4749                 if (IS_ERR(expr)) {
4750                         err = PTR_ERR(expr);
4751                         goto err_set_expr_alloc;
4752                 }
4753                 exprs[0] = expr;
4754                 (*num_exprs)++;
4755         } else if (nla[NFTA_SET_EXPRESSIONS]) {
4756                 struct nlattr *tmp;
4757                 int left;
4758
4759                 if (!(flags & NFT_SET_EXPR)) {
4760                         err = -EINVAL;
4761                         goto err_set_expr_alloc;
4762                 }
4763                 i = 0;
4764                 nla_for_each_nested(tmp, nla[NFTA_SET_EXPRESSIONS], left) {
4765                         if (i == NFT_SET_EXPR_MAX) {
4766                                 err = -E2BIG;
4767                                 goto err_set_expr_alloc;
4768                         }
4769                         if (nla_type(tmp) != NFTA_LIST_ELEM) {
4770                                 err = -EINVAL;
4771                                 goto err_set_expr_alloc;
4772                         }
4773                         expr = nft_set_elem_expr_alloc(ctx, set, tmp);
4774                         if (IS_ERR(expr)) {
4775                                 err = PTR_ERR(expr);
4776                                 goto err_set_expr_alloc;
4777                         }
4778                         exprs[i++] = expr;
4779                         (*num_exprs)++;
4780                 }
4781         }
4782
4783         return 0;
4784
4785 err_set_expr_alloc:
4786         for (i = 0; i < *num_exprs; i++)
4787                 nft_expr_destroy(ctx, exprs[i]);
4788
4789         return err;
4790 }
4791
4792 static bool nft_set_is_same(const struct nft_set *set,
4793                             const struct nft_set_desc *desc,
4794                             struct nft_expr *exprs[], u32 num_exprs, u32 flags)
4795 {
4796         int i;
4797
4798         if (set->ktype != desc->ktype ||
4799             set->dtype != desc->dtype ||
4800             set->flags != flags ||
4801             set->klen != desc->klen ||
4802             set->dlen != desc->dlen ||
4803             set->field_count != desc->field_count ||
4804             set->num_exprs != num_exprs)
4805                 return false;
4806
4807         for (i = 0; i < desc->field_count; i++) {
4808                 if (set->field_len[i] != desc->field_len[i])
4809                         return false;
4810         }
4811
4812         for (i = 0; i < num_exprs; i++) {
4813                 if (set->exprs[i]->ops != exprs[i]->ops)
4814                         return false;
4815         }
4816
4817         return true;
4818 }
4819
4820 static int nf_tables_newset(struct sk_buff *skb, const struct nfnl_info *info,
4821                             const struct nlattr * const nla[])
4822 {
4823         struct netlink_ext_ack *extack = info->extack;
4824         u8 genmask = nft_genmask_next(info->net);
4825         u8 family = info->nfmsg->nfgen_family;
4826         const struct nft_set_ops *ops;
4827         struct net *net = info->net;
4828         struct nft_set_desc desc;
4829         struct nft_table *table;
4830         unsigned char *udata;
4831         struct nft_set *set;
4832         struct nft_ctx ctx;
4833         size_t alloc_size;
4834         int num_exprs = 0;
4835         char *name;
4836         int err, i;
4837         u16 udlen;
4838         u32 flags;
4839         u64 size;
4840
4841         if (nla[NFTA_SET_TABLE] == NULL ||
4842             nla[NFTA_SET_NAME] == NULL ||
4843             nla[NFTA_SET_KEY_LEN] == NULL ||
4844             nla[NFTA_SET_ID] == NULL)
4845                 return -EINVAL;
4846
4847         memset(&desc, 0, sizeof(desc));
4848
4849         desc.ktype = NFT_DATA_VALUE;
4850         if (nla[NFTA_SET_KEY_TYPE] != NULL) {
4851                 desc.ktype = ntohl(nla_get_be32(nla[NFTA_SET_KEY_TYPE]));
4852                 if ((desc.ktype & NFT_DATA_RESERVED_MASK) == NFT_DATA_RESERVED_MASK)
4853                         return -EINVAL;
4854         }
4855
4856         desc.klen = ntohl(nla_get_be32(nla[NFTA_SET_KEY_LEN]));
4857         if (desc.klen == 0 || desc.klen > NFT_DATA_VALUE_MAXLEN)
4858                 return -EINVAL;
4859
4860         flags = 0;
4861         if (nla[NFTA_SET_FLAGS] != NULL) {
4862                 flags = ntohl(nla_get_be32(nla[NFTA_SET_FLAGS]));
4863                 if (flags & ~(NFT_SET_ANONYMOUS | NFT_SET_CONSTANT |
4864                               NFT_SET_INTERVAL | NFT_SET_TIMEOUT |
4865                               NFT_SET_MAP | NFT_SET_EVAL |
4866                               NFT_SET_OBJECT | NFT_SET_CONCAT | NFT_SET_EXPR))
4867                         return -EOPNOTSUPP;
4868                 /* Only one of these operations is supported */
4869                 if ((flags & (NFT_SET_MAP | NFT_SET_OBJECT)) ==
4870                              (NFT_SET_MAP | NFT_SET_OBJECT))
4871                         return -EOPNOTSUPP;
4872                 if ((flags & (NFT_SET_EVAL | NFT_SET_OBJECT)) ==
4873                              (NFT_SET_EVAL | NFT_SET_OBJECT))
4874                         return -EOPNOTSUPP;
4875         }
4876
4877         desc.dtype = 0;
4878         if (nla[NFTA_SET_DATA_TYPE] != NULL) {
4879                 if (!(flags & NFT_SET_MAP))
4880                         return -EINVAL;
4881
4882                 desc.dtype = ntohl(nla_get_be32(nla[NFTA_SET_DATA_TYPE]));
4883                 if ((desc.dtype & NFT_DATA_RESERVED_MASK) == NFT_DATA_RESERVED_MASK &&
4884                     desc.dtype != NFT_DATA_VERDICT)
4885                         return -EINVAL;
4886
4887                 if (desc.dtype != NFT_DATA_VERDICT) {
4888                         if (nla[NFTA_SET_DATA_LEN] == NULL)
4889                                 return -EINVAL;
4890                         desc.dlen = ntohl(nla_get_be32(nla[NFTA_SET_DATA_LEN]));
4891                         if (desc.dlen == 0 || desc.dlen > NFT_DATA_VALUE_MAXLEN)
4892                                 return -EINVAL;
4893                 } else
4894                         desc.dlen = sizeof(struct nft_verdict);
4895         } else if (flags & NFT_SET_MAP)
4896                 return -EINVAL;
4897
4898         if (nla[NFTA_SET_OBJ_TYPE] != NULL) {
4899                 if (!(flags & NFT_SET_OBJECT))
4900                         return -EINVAL;
4901
4902                 desc.objtype = ntohl(nla_get_be32(nla[NFTA_SET_OBJ_TYPE]));
4903                 if (desc.objtype == NFT_OBJECT_UNSPEC ||
4904                     desc.objtype > NFT_OBJECT_MAX)
4905                         return -EOPNOTSUPP;
4906         } else if (flags & NFT_SET_OBJECT)
4907                 return -EINVAL;
4908         else
4909                 desc.objtype = NFT_OBJECT_UNSPEC;
4910
4911         desc.timeout = 0;
4912         if (nla[NFTA_SET_TIMEOUT] != NULL) {
4913                 if (!(flags & NFT_SET_TIMEOUT))
4914                         return -EINVAL;
4915
4916                 if (flags & NFT_SET_ANONYMOUS)
4917                         return -EOPNOTSUPP;
4918
4919                 err = nf_msecs_to_jiffies64(nla[NFTA_SET_TIMEOUT], &desc.timeout);
4920                 if (err)
4921                         return err;
4922         }
4923         desc.gc_int = 0;
4924         if (nla[NFTA_SET_GC_INTERVAL] != NULL) {
4925                 if (!(flags & NFT_SET_TIMEOUT))
4926                         return -EINVAL;
4927
4928                 if (flags & NFT_SET_ANONYMOUS)
4929                         return -EOPNOTSUPP;
4930
4931                 desc.gc_int = ntohl(nla_get_be32(nla[NFTA_SET_GC_INTERVAL]));
4932         }
4933
4934         desc.policy = NFT_SET_POL_PERFORMANCE;
4935         if (nla[NFTA_SET_POLICY] != NULL)
4936                 desc.policy = ntohl(nla_get_be32(nla[NFTA_SET_POLICY]));
4937
4938         if (nla[NFTA_SET_DESC] != NULL) {
4939                 err = nf_tables_set_desc_parse(&desc, nla[NFTA_SET_DESC]);
4940                 if (err < 0)
4941                         return err;
4942
4943                 if (desc.field_count > 1 && !(flags & NFT_SET_CONCAT))
4944                         return -EINVAL;
4945         } else if (flags & NFT_SET_CONCAT) {
4946                 return -EINVAL;
4947         }
4948
4949         if (nla[NFTA_SET_EXPR] || nla[NFTA_SET_EXPRESSIONS])
4950                 desc.expr = true;
4951
4952         table = nft_table_lookup(net, nla[NFTA_SET_TABLE], family, genmask,
4953                                  NETLINK_CB(skb).portid);
4954         if (IS_ERR(table)) {
4955                 NL_SET_BAD_ATTR(extack, nla[NFTA_SET_TABLE]);
4956                 return PTR_ERR(table);
4957         }
4958
4959         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
4960
4961         set = nft_set_lookup(table, nla[NFTA_SET_NAME], genmask);
4962         if (IS_ERR(set)) {
4963                 if (PTR_ERR(set) != -ENOENT) {
4964                         NL_SET_BAD_ATTR(extack, nla[NFTA_SET_NAME]);
4965                         return PTR_ERR(set);
4966                 }
4967         } else {
4968                 struct nft_expr *exprs[NFT_SET_EXPR_MAX] = {};
4969
4970                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
4971                         NL_SET_BAD_ATTR(extack, nla[NFTA_SET_NAME]);
4972                         return -EEXIST;
4973                 }
4974                 if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
4975                         return -EOPNOTSUPP;
4976
4977                 if (nft_set_is_anonymous(set))
4978                         return -EOPNOTSUPP;
4979
4980                 err = nft_set_expr_alloc(&ctx, set, nla, exprs, &num_exprs, flags);
4981                 if (err < 0)
4982                         return err;
4983
4984                 err = 0;
4985                 if (!nft_set_is_same(set, &desc, exprs, num_exprs, flags)) {
4986                         NL_SET_BAD_ATTR(extack, nla[NFTA_SET_NAME]);
4987                         err = -EEXIST;
4988                 }
4989
4990                 for (i = 0; i < num_exprs; i++)
4991                         nft_expr_destroy(&ctx, exprs[i]);
4992
4993                 if (err < 0)
4994                         return err;
4995
4996                 return __nft_trans_set_add(&ctx, NFT_MSG_NEWSET, set, &desc);
4997         }
4998
4999         if (!(info->nlh->nlmsg_flags & NLM_F_CREATE))
5000                 return -ENOENT;
5001
5002         ops = nft_select_set_ops(&ctx, nla, &desc);
5003         if (IS_ERR(ops))
5004                 return PTR_ERR(ops);
5005
5006         udlen = 0;
5007         if (nla[NFTA_SET_USERDATA])
5008                 udlen = nla_len(nla[NFTA_SET_USERDATA]);
5009
5010         size = 0;
5011         if (ops->privsize != NULL)
5012                 size = ops->privsize(nla, &desc);
5013         alloc_size = sizeof(*set) + size + udlen;
5014         if (alloc_size < size || alloc_size > INT_MAX)
5015                 return -ENOMEM;
5016         set = kvzalloc(alloc_size, GFP_KERNEL_ACCOUNT);
5017         if (!set)
5018                 return -ENOMEM;
5019
5020         name = nla_strdup(nla[NFTA_SET_NAME], GFP_KERNEL_ACCOUNT);
5021         if (!name) {
5022                 err = -ENOMEM;
5023                 goto err_set_name;
5024         }
5025
5026         err = nf_tables_set_alloc_name(&ctx, set, name);
5027         kfree(name);
5028         if (err < 0)
5029                 goto err_set_name;
5030
5031         udata = NULL;
5032         if (udlen) {
5033                 udata = set->data + size;
5034                 nla_memcpy(udata, nla[NFTA_SET_USERDATA], udlen);
5035         }
5036
5037         INIT_LIST_HEAD(&set->bindings);
5038         INIT_LIST_HEAD(&set->catchall_list);
5039         set->table = table;
5040         write_pnet(&set->net, net);
5041         set->ops = ops;
5042         set->ktype = desc.ktype;
5043         set->klen = desc.klen;
5044         set->dtype = desc.dtype;
5045         set->objtype = desc.objtype;
5046         set->dlen = desc.dlen;
5047         set->flags = flags;
5048         set->size = desc.size;
5049         set->policy = desc.policy;
5050         set->udlen = udlen;
5051         set->udata = udata;
5052         set->timeout = desc.timeout;
5053         set->gc_int = desc.gc_int;
5054
5055         set->field_count = desc.field_count;
5056         for (i = 0; i < desc.field_count; i++)
5057                 set->field_len[i] = desc.field_len[i];
5058
5059         err = ops->init(set, &desc, nla);
5060         if (err < 0)
5061                 goto err_set_init;
5062
5063         err = nft_set_expr_alloc(&ctx, set, nla, set->exprs, &num_exprs, flags);
5064         if (err < 0)
5065                 goto err_set_destroy;
5066
5067         set->num_exprs = num_exprs;
5068         set->handle = nf_tables_alloc_handle(table);
5069         INIT_LIST_HEAD(&set->pending_update);
5070
5071         err = nft_trans_set_add(&ctx, NFT_MSG_NEWSET, set);
5072         if (err < 0)
5073                 goto err_set_expr_alloc;
5074
5075         list_add_tail_rcu(&set->list, &table->sets);
5076         table->use++;
5077         return 0;
5078
5079 err_set_expr_alloc:
5080         for (i = 0; i < set->num_exprs; i++)
5081                 nft_expr_destroy(&ctx, set->exprs[i]);
5082 err_set_destroy:
5083         ops->destroy(&ctx, set);
5084 err_set_init:
5085         kfree(set->name);
5086 err_set_name:
5087         kvfree(set);
5088         return err;
5089 }
5090
5091 static void nft_set_catchall_destroy(const struct nft_ctx *ctx,
5092                                      struct nft_set *set)
5093 {
5094         struct nft_set_elem_catchall *next, *catchall;
5095
5096         list_for_each_entry_safe(catchall, next, &set->catchall_list, list) {
5097                 list_del_rcu(&catchall->list);
5098                 nf_tables_set_elem_destroy(ctx, set, catchall->elem);
5099                 kfree_rcu(catchall, rcu);
5100         }
5101 }
5102
5103 static void nft_set_destroy(const struct nft_ctx *ctx, struct nft_set *set)
5104 {
5105         int i;
5106
5107         if (WARN_ON(set->use > 0))
5108                 return;
5109
5110         for (i = 0; i < set->num_exprs; i++)
5111                 nft_expr_destroy(ctx, set->exprs[i]);
5112
5113         set->ops->destroy(ctx, set);
5114         nft_set_catchall_destroy(ctx, set);
5115         kfree(set->name);
5116         kvfree(set);
5117 }
5118
5119 static int nf_tables_delset(struct sk_buff *skb, const struct nfnl_info *info,
5120                             const struct nlattr * const nla[])
5121 {
5122         struct netlink_ext_ack *extack = info->extack;
5123         u8 genmask = nft_genmask_next(info->net);
5124         u8 family = info->nfmsg->nfgen_family;
5125         struct net *net = info->net;
5126         const struct nlattr *attr;
5127         struct nft_table *table;
5128         struct nft_set *set;
5129         struct nft_ctx ctx;
5130
5131         if (info->nfmsg->nfgen_family == NFPROTO_UNSPEC)
5132                 return -EAFNOSUPPORT;
5133
5134         table = nft_table_lookup(net, nla[NFTA_SET_TABLE], family,
5135                                  genmask, NETLINK_CB(skb).portid);
5136         if (IS_ERR(table)) {
5137                 NL_SET_BAD_ATTR(extack, nla[NFTA_SET_TABLE]);
5138                 return PTR_ERR(table);
5139         }
5140
5141         if (nla[NFTA_SET_HANDLE]) {
5142                 attr = nla[NFTA_SET_HANDLE];
5143                 set = nft_set_lookup_byhandle(table, attr, genmask);
5144         } else {
5145                 attr = nla[NFTA_SET_NAME];
5146                 set = nft_set_lookup(table, attr, genmask);
5147         }
5148
5149         if (IS_ERR(set)) {
5150                 if (PTR_ERR(set) == -ENOENT &&
5151                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYSET)
5152                         return 0;
5153
5154                 NL_SET_BAD_ATTR(extack, attr);
5155                 return PTR_ERR(set);
5156         }
5157         if (set->use ||
5158             (info->nlh->nlmsg_flags & NLM_F_NONREC &&
5159              atomic_read(&set->nelems) > 0)) {
5160                 NL_SET_BAD_ATTR(extack, attr);
5161                 return -EBUSY;
5162         }
5163
5164         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
5165
5166         return nft_delset(&ctx, set);
5167 }
5168
5169 static int nft_validate_register_store(const struct nft_ctx *ctx,
5170                                        enum nft_registers reg,
5171                                        const struct nft_data *data,
5172                                        enum nft_data_types type,
5173                                        unsigned int len);
5174
5175 static int nft_setelem_data_validate(const struct nft_ctx *ctx,
5176                                      struct nft_set *set,
5177                                      struct nft_set_elem *elem)
5178 {
5179         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
5180         enum nft_registers dreg;
5181
5182         dreg = nft_type_to_reg(set->dtype);
5183         return nft_validate_register_store(ctx, dreg, nft_set_ext_data(ext),
5184                                            set->dtype == NFT_DATA_VERDICT ?
5185                                            NFT_DATA_VERDICT : NFT_DATA_VALUE,
5186                                            set->dlen);
5187 }
5188
5189 static int nf_tables_bind_check_setelem(const struct nft_ctx *ctx,
5190                                         struct nft_set *set,
5191                                         const struct nft_set_iter *iter,
5192                                         struct nft_set_elem *elem)
5193 {
5194         return nft_setelem_data_validate(ctx, set, elem);
5195 }
5196
5197 static int nft_set_catchall_bind_check(const struct nft_ctx *ctx,
5198                                        struct nft_set *set)
5199 {
5200         u8 genmask = nft_genmask_next(ctx->net);
5201         struct nft_set_elem_catchall *catchall;
5202         struct nft_set_elem elem;
5203         struct nft_set_ext *ext;
5204         int ret = 0;
5205
5206         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
5207                 ext = nft_set_elem_ext(set, catchall->elem);
5208                 if (!nft_set_elem_active(ext, genmask))
5209                         continue;
5210
5211                 elem.priv = catchall->elem;
5212                 ret = nft_setelem_data_validate(ctx, set, &elem);
5213                 if (ret < 0)
5214                         break;
5215         }
5216
5217         return ret;
5218 }
5219
5220 int nf_tables_bind_set(const struct nft_ctx *ctx, struct nft_set *set,
5221                        struct nft_set_binding *binding)
5222 {
5223         struct nft_set_binding *i;
5224         struct nft_set_iter iter;
5225
5226         if (set->use == UINT_MAX)
5227                 return -EOVERFLOW;
5228
5229         if (!list_empty(&set->bindings) && nft_set_is_anonymous(set))
5230                 return -EBUSY;
5231
5232         if (binding->flags & NFT_SET_MAP) {
5233                 /* If the set is already bound to the same chain all
5234                  * jumps are already validated for that chain.
5235                  */
5236                 list_for_each_entry(i, &set->bindings, list) {
5237                         if (i->flags & NFT_SET_MAP &&
5238                             i->chain == binding->chain)
5239                                 goto bind;
5240                 }
5241
5242                 iter.genmask    = nft_genmask_next(ctx->net);
5243                 iter.skip       = 0;
5244                 iter.count      = 0;
5245                 iter.err        = 0;
5246                 iter.fn         = nf_tables_bind_check_setelem;
5247
5248                 set->ops->walk(ctx, set, &iter);
5249                 if (!iter.err)
5250                         iter.err = nft_set_catchall_bind_check(ctx, set);
5251
5252                 if (iter.err < 0)
5253                         return iter.err;
5254         }
5255 bind:
5256         binding->chain = ctx->chain;
5257         list_add_tail_rcu(&binding->list, &set->bindings);
5258         nft_set_trans_bind(ctx, set);
5259         set->use++;
5260
5261         return 0;
5262 }
5263 EXPORT_SYMBOL_GPL(nf_tables_bind_set);
5264
5265 static void nf_tables_unbind_set(const struct nft_ctx *ctx, struct nft_set *set,
5266                                  struct nft_set_binding *binding, bool event)
5267 {
5268         list_del_rcu(&binding->list);
5269
5270         if (list_empty(&set->bindings) && nft_set_is_anonymous(set)) {
5271                 list_del_rcu(&set->list);
5272                 if (event)
5273                         nf_tables_set_notify(ctx, set, NFT_MSG_DELSET,
5274                                              GFP_KERNEL);
5275         }
5276 }
5277
5278 static void nft_setelem_data_activate(const struct net *net,
5279                                       const struct nft_set *set,
5280                                       struct nft_set_elem *elem);
5281
5282 static int nft_mapelem_activate(const struct nft_ctx *ctx,
5283                                 struct nft_set *set,
5284                                 const struct nft_set_iter *iter,
5285                                 struct nft_set_elem *elem)
5286 {
5287         nft_setelem_data_activate(ctx->net, set, elem);
5288
5289         return 0;
5290 }
5291
5292 static void nft_map_catchall_activate(const struct nft_ctx *ctx,
5293                                       struct nft_set *set)
5294 {
5295         u8 genmask = nft_genmask_next(ctx->net);
5296         struct nft_set_elem_catchall *catchall;
5297         struct nft_set_elem elem;
5298         struct nft_set_ext *ext;
5299
5300         list_for_each_entry(catchall, &set->catchall_list, list) {
5301                 ext = nft_set_elem_ext(set, catchall->elem);
5302                 if (!nft_set_elem_active(ext, genmask))
5303                         continue;
5304
5305                 elem.priv = catchall->elem;
5306                 nft_setelem_data_activate(ctx->net, set, &elem);
5307                 break;
5308         }
5309 }
5310
5311 static void nft_map_activate(const struct nft_ctx *ctx, struct nft_set *set)
5312 {
5313         struct nft_set_iter iter = {
5314                 .genmask        = nft_genmask_next(ctx->net),
5315                 .fn             = nft_mapelem_activate,
5316         };
5317
5318         set->ops->walk(ctx, set, &iter);
5319         WARN_ON_ONCE(iter.err);
5320
5321         nft_map_catchall_activate(ctx, set);
5322 }
5323
5324 void nf_tables_activate_set(const struct nft_ctx *ctx, struct nft_set *set)
5325 {
5326         if (nft_set_is_anonymous(set)) {
5327                 if (set->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
5328                         nft_map_activate(ctx, set);
5329
5330                 nft_clear(ctx->net, set);
5331         }
5332
5333         set->use++;
5334 }
5335 EXPORT_SYMBOL_GPL(nf_tables_activate_set);
5336
5337 void nf_tables_deactivate_set(const struct nft_ctx *ctx, struct nft_set *set,
5338                               struct nft_set_binding *binding,
5339                               enum nft_trans_phase phase)
5340 {
5341         switch (phase) {
5342         case NFT_TRANS_PREPARE_ERROR:
5343                 nft_set_trans_unbind(ctx, set);
5344                 if (nft_set_is_anonymous(set))
5345                         nft_deactivate_next(ctx->net, set);
5346
5347                 set->use--;
5348                 break;
5349         case NFT_TRANS_PREPARE:
5350                 if (nft_set_is_anonymous(set)) {
5351                         if (set->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
5352                                 nft_map_deactivate(ctx, set);
5353
5354                         nft_deactivate_next(ctx->net, set);
5355                 }
5356                 set->use--;
5357                 return;
5358         case NFT_TRANS_ABORT:
5359         case NFT_TRANS_RELEASE:
5360                 if (nft_set_is_anonymous(set) &&
5361                     set->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
5362                         nft_map_deactivate(ctx, set);
5363
5364                 set->use--;
5365                 fallthrough;
5366         default:
5367                 nf_tables_unbind_set(ctx, set, binding,
5368                                      phase == NFT_TRANS_COMMIT);
5369         }
5370 }
5371 EXPORT_SYMBOL_GPL(nf_tables_deactivate_set);
5372
5373 void nf_tables_destroy_set(const struct nft_ctx *ctx, struct nft_set *set)
5374 {
5375         if (list_empty(&set->bindings) && nft_set_is_anonymous(set))
5376                 nft_set_destroy(ctx, set);
5377 }
5378 EXPORT_SYMBOL_GPL(nf_tables_destroy_set);
5379
5380 const struct nft_set_ext_type nft_set_ext_types[] = {
5381         [NFT_SET_EXT_KEY]               = {
5382                 .align  = __alignof__(u32),
5383         },
5384         [NFT_SET_EXT_DATA]              = {
5385                 .align  = __alignof__(u32),
5386         },
5387         [NFT_SET_EXT_EXPRESSIONS]       = {
5388                 .align  = __alignof__(struct nft_set_elem_expr),
5389         },
5390         [NFT_SET_EXT_OBJREF]            = {
5391                 .len    = sizeof(struct nft_object *),
5392                 .align  = __alignof__(struct nft_object *),
5393         },
5394         [NFT_SET_EXT_FLAGS]             = {
5395                 .len    = sizeof(u8),
5396                 .align  = __alignof__(u8),
5397         },
5398         [NFT_SET_EXT_TIMEOUT]           = {
5399                 .len    = sizeof(u64),
5400                 .align  = __alignof__(u64),
5401         },
5402         [NFT_SET_EXT_EXPIRATION]        = {
5403                 .len    = sizeof(u64),
5404                 .align  = __alignof__(u64),
5405         },
5406         [NFT_SET_EXT_USERDATA]          = {
5407                 .len    = sizeof(struct nft_userdata),
5408                 .align  = __alignof__(struct nft_userdata),
5409         },
5410         [NFT_SET_EXT_KEY_END]           = {
5411                 .align  = __alignof__(u32),
5412         },
5413 };
5414
5415 /*
5416  * Set elements
5417  */
5418
5419 static const struct nla_policy nft_set_elem_policy[NFTA_SET_ELEM_MAX + 1] = {
5420         [NFTA_SET_ELEM_KEY]             = { .type = NLA_NESTED },
5421         [NFTA_SET_ELEM_DATA]            = { .type = NLA_NESTED },
5422         [NFTA_SET_ELEM_FLAGS]           = { .type = NLA_U32 },
5423         [NFTA_SET_ELEM_TIMEOUT]         = { .type = NLA_U64 },
5424         [NFTA_SET_ELEM_EXPIRATION]      = { .type = NLA_U64 },
5425         [NFTA_SET_ELEM_USERDATA]        = { .type = NLA_BINARY,
5426                                             .len = NFT_USERDATA_MAXLEN },
5427         [NFTA_SET_ELEM_EXPR]            = { .type = NLA_NESTED },
5428         [NFTA_SET_ELEM_OBJREF]          = { .type = NLA_STRING,
5429                                             .len = NFT_OBJ_MAXNAMELEN - 1 },
5430         [NFTA_SET_ELEM_KEY_END]         = { .type = NLA_NESTED },
5431         [NFTA_SET_ELEM_EXPRESSIONS]     = { .type = NLA_NESTED },
5432 };
5433
5434 static const struct nla_policy nft_set_elem_list_policy[NFTA_SET_ELEM_LIST_MAX + 1] = {
5435         [NFTA_SET_ELEM_LIST_TABLE]      = { .type = NLA_STRING,
5436                                             .len = NFT_TABLE_MAXNAMELEN - 1 },
5437         [NFTA_SET_ELEM_LIST_SET]        = { .type = NLA_STRING,
5438                                             .len = NFT_SET_MAXNAMELEN - 1 },
5439         [NFTA_SET_ELEM_LIST_ELEMENTS]   = { .type = NLA_NESTED },
5440         [NFTA_SET_ELEM_LIST_SET_ID]     = { .type = NLA_U32 },
5441 };
5442
5443 static int nft_set_elem_expr_dump(struct sk_buff *skb,
5444                                   const struct nft_set *set,
5445                                   const struct nft_set_ext *ext)
5446 {
5447         struct nft_set_elem_expr *elem_expr;
5448         u32 size, num_exprs = 0;
5449         struct nft_expr *expr;
5450         struct nlattr *nest;
5451
5452         elem_expr = nft_set_ext_expr(ext);
5453         nft_setelem_expr_foreach(expr, elem_expr, size)
5454                 num_exprs++;
5455
5456         if (num_exprs == 1) {
5457                 expr = nft_setelem_expr_at(elem_expr, 0);
5458                 if (nft_expr_dump(skb, NFTA_SET_ELEM_EXPR, expr, false) < 0)
5459                         return -1;
5460
5461                 return 0;
5462         } else if (num_exprs > 1) {
5463                 nest = nla_nest_start_noflag(skb, NFTA_SET_ELEM_EXPRESSIONS);
5464                 if (nest == NULL)
5465                         goto nla_put_failure;
5466
5467                 nft_setelem_expr_foreach(expr, elem_expr, size) {
5468                         expr = nft_setelem_expr_at(elem_expr, size);
5469                         if (nft_expr_dump(skb, NFTA_LIST_ELEM, expr, false) < 0)
5470                                 goto nla_put_failure;
5471                 }
5472                 nla_nest_end(skb, nest);
5473         }
5474         return 0;
5475
5476 nla_put_failure:
5477         return -1;
5478 }
5479
5480 static int nf_tables_fill_setelem(struct sk_buff *skb,
5481                                   const struct nft_set *set,
5482                                   const struct nft_set_elem *elem)
5483 {
5484         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
5485         unsigned char *b = skb_tail_pointer(skb);
5486         struct nlattr *nest;
5487
5488         nest = nla_nest_start_noflag(skb, NFTA_LIST_ELEM);
5489         if (nest == NULL)
5490                 goto nla_put_failure;
5491
5492         if (nft_set_ext_exists(ext, NFT_SET_EXT_KEY) &&
5493             nft_data_dump(skb, NFTA_SET_ELEM_KEY, nft_set_ext_key(ext),
5494                           NFT_DATA_VALUE, set->klen) < 0)
5495                 goto nla_put_failure;
5496
5497         if (nft_set_ext_exists(ext, NFT_SET_EXT_KEY_END) &&
5498             nft_data_dump(skb, NFTA_SET_ELEM_KEY_END, nft_set_ext_key_end(ext),
5499                           NFT_DATA_VALUE, set->klen) < 0)
5500                 goto nla_put_failure;
5501
5502         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA) &&
5503             nft_data_dump(skb, NFTA_SET_ELEM_DATA, nft_set_ext_data(ext),
5504                           set->dtype == NFT_DATA_VERDICT ? NFT_DATA_VERDICT : NFT_DATA_VALUE,
5505                           set->dlen) < 0)
5506                 goto nla_put_failure;
5507
5508         if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPRESSIONS) &&
5509             nft_set_elem_expr_dump(skb, set, ext))
5510                 goto nla_put_failure;
5511
5512         if (nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF) &&
5513             nla_put_string(skb, NFTA_SET_ELEM_OBJREF,
5514                            (*nft_set_ext_obj(ext))->key.name) < 0)
5515                 goto nla_put_failure;
5516
5517         if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
5518             nla_put_be32(skb, NFTA_SET_ELEM_FLAGS,
5519                          htonl(*nft_set_ext_flags(ext))))
5520                 goto nla_put_failure;
5521
5522         if (nft_set_ext_exists(ext, NFT_SET_EXT_TIMEOUT) &&
5523             nla_put_be64(skb, NFTA_SET_ELEM_TIMEOUT,
5524                          nf_jiffies64_to_msecs(*nft_set_ext_timeout(ext)),
5525                          NFTA_SET_ELEM_PAD))
5526                 goto nla_put_failure;
5527
5528         if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPIRATION)) {
5529                 u64 expires, now = get_jiffies_64();
5530
5531                 expires = *nft_set_ext_expiration(ext);
5532                 if (time_before64(now, expires))
5533                         expires -= now;
5534                 else
5535                         expires = 0;
5536
5537                 if (nla_put_be64(skb, NFTA_SET_ELEM_EXPIRATION,
5538                                  nf_jiffies64_to_msecs(expires),
5539                                  NFTA_SET_ELEM_PAD))
5540                         goto nla_put_failure;
5541         }
5542
5543         if (nft_set_ext_exists(ext, NFT_SET_EXT_USERDATA)) {
5544                 struct nft_userdata *udata;
5545
5546                 udata = nft_set_ext_userdata(ext);
5547                 if (nla_put(skb, NFTA_SET_ELEM_USERDATA,
5548                             udata->len + 1, udata->data))
5549                         goto nla_put_failure;
5550         }
5551
5552         nla_nest_end(skb, nest);
5553         return 0;
5554
5555 nla_put_failure:
5556         nlmsg_trim(skb, b);
5557         return -EMSGSIZE;
5558 }
5559
5560 struct nft_set_dump_args {
5561         const struct netlink_callback   *cb;
5562         struct nft_set_iter             iter;
5563         struct sk_buff                  *skb;
5564 };
5565
5566 static int nf_tables_dump_setelem(const struct nft_ctx *ctx,
5567                                   struct nft_set *set,
5568                                   const struct nft_set_iter *iter,
5569                                   struct nft_set_elem *elem)
5570 {
5571         struct nft_set_dump_args *args;
5572
5573         args = container_of(iter, struct nft_set_dump_args, iter);
5574         return nf_tables_fill_setelem(args->skb, set, elem);
5575 }
5576
5577 struct nft_set_dump_ctx {
5578         const struct nft_set    *set;
5579         struct nft_ctx          ctx;
5580 };
5581
5582 static int nft_set_catchall_dump(struct net *net, struct sk_buff *skb,
5583                                  const struct nft_set *set)
5584 {
5585         struct nft_set_elem_catchall *catchall;
5586         u8 genmask = nft_genmask_cur(net);
5587         struct nft_set_elem elem;
5588         struct nft_set_ext *ext;
5589         int ret = 0;
5590
5591         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
5592                 ext = nft_set_elem_ext(set, catchall->elem);
5593                 if (!nft_set_elem_active(ext, genmask) ||
5594                     nft_set_elem_expired(ext))
5595                         continue;
5596
5597                 elem.priv = catchall->elem;
5598                 ret = nf_tables_fill_setelem(skb, set, &elem);
5599                 break;
5600         }
5601
5602         return ret;
5603 }
5604
5605 static int nf_tables_dump_set(struct sk_buff *skb, struct netlink_callback *cb)
5606 {
5607         struct nft_set_dump_ctx *dump_ctx = cb->data;
5608         struct net *net = sock_net(skb->sk);
5609         struct nftables_pernet *nft_net;
5610         struct nft_table *table;
5611         struct nft_set *set;
5612         struct nft_set_dump_args args;
5613         bool set_found = false;
5614         struct nlmsghdr *nlh;
5615         struct nlattr *nest;
5616         u32 portid, seq;
5617         int event;
5618
5619         rcu_read_lock();
5620         nft_net = nft_pernet(net);
5621         cb->seq = READ_ONCE(nft_net->base_seq);
5622
5623         list_for_each_entry_rcu(table, &nft_net->tables, list) {
5624                 if (dump_ctx->ctx.family != NFPROTO_UNSPEC &&
5625                     dump_ctx->ctx.family != table->family)
5626                         continue;
5627
5628                 if (table != dump_ctx->ctx.table)
5629                         continue;
5630
5631                 list_for_each_entry_rcu(set, &table->sets, list) {
5632                         if (set == dump_ctx->set) {
5633                                 set_found = true;
5634                                 break;
5635                         }
5636                 }
5637                 break;
5638         }
5639
5640         if (!set_found) {
5641                 rcu_read_unlock();
5642                 return -ENOENT;
5643         }
5644
5645         event  = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, NFT_MSG_NEWSETELEM);
5646         portid = NETLINK_CB(cb->skb).portid;
5647         seq    = cb->nlh->nlmsg_seq;
5648
5649         nlh = nfnl_msg_put(skb, portid, seq, event, NLM_F_MULTI,
5650                            table->family, NFNETLINK_V0, nft_base_seq(net));
5651         if (!nlh)
5652                 goto nla_put_failure;
5653
5654         if (nla_put_string(skb, NFTA_SET_ELEM_LIST_TABLE, table->name))
5655                 goto nla_put_failure;
5656         if (nla_put_string(skb, NFTA_SET_ELEM_LIST_SET, set->name))
5657                 goto nla_put_failure;
5658
5659         nest = nla_nest_start_noflag(skb, NFTA_SET_ELEM_LIST_ELEMENTS);
5660         if (nest == NULL)
5661                 goto nla_put_failure;
5662
5663         args.cb                 = cb;
5664         args.skb                = skb;
5665         args.iter.genmask       = nft_genmask_cur(net);
5666         args.iter.skip          = cb->args[0];
5667         args.iter.count         = 0;
5668         args.iter.err           = 0;
5669         args.iter.fn            = nf_tables_dump_setelem;
5670         set->ops->walk(&dump_ctx->ctx, set, &args.iter);
5671
5672         if (!args.iter.err && args.iter.count == cb->args[0])
5673                 args.iter.err = nft_set_catchall_dump(net, skb, set);
5674         rcu_read_unlock();
5675
5676         nla_nest_end(skb, nest);
5677         nlmsg_end(skb, nlh);
5678
5679         if (args.iter.err && args.iter.err != -EMSGSIZE)
5680                 return args.iter.err;
5681         if (args.iter.count == cb->args[0])
5682                 return 0;
5683
5684         cb->args[0] = args.iter.count;
5685         return skb->len;
5686
5687 nla_put_failure:
5688         rcu_read_unlock();
5689         return -ENOSPC;
5690 }
5691
5692 static int nf_tables_dump_set_start(struct netlink_callback *cb)
5693 {
5694         struct nft_set_dump_ctx *dump_ctx = cb->data;
5695
5696         cb->data = kmemdup(dump_ctx, sizeof(*dump_ctx), GFP_ATOMIC);
5697
5698         return cb->data ? 0 : -ENOMEM;
5699 }
5700
5701 static int nf_tables_dump_set_done(struct netlink_callback *cb)
5702 {
5703         kfree(cb->data);
5704         return 0;
5705 }
5706
5707 static int nf_tables_fill_setelem_info(struct sk_buff *skb,
5708                                        const struct nft_ctx *ctx, u32 seq,
5709                                        u32 portid, int event, u16 flags,
5710                                        const struct nft_set *set,
5711                                        const struct nft_set_elem *elem)
5712 {
5713         struct nlmsghdr *nlh;
5714         struct nlattr *nest;
5715         int err;
5716
5717         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
5718         nlh = nfnl_msg_put(skb, portid, seq, event, flags, ctx->family,
5719                            NFNETLINK_V0, nft_base_seq(ctx->net));
5720         if (!nlh)
5721                 goto nla_put_failure;
5722
5723         if (nla_put_string(skb, NFTA_SET_TABLE, ctx->table->name))
5724                 goto nla_put_failure;
5725         if (nla_put_string(skb, NFTA_SET_NAME, set->name))
5726                 goto nla_put_failure;
5727
5728         nest = nla_nest_start_noflag(skb, NFTA_SET_ELEM_LIST_ELEMENTS);
5729         if (nest == NULL)
5730                 goto nla_put_failure;
5731
5732         err = nf_tables_fill_setelem(skb, set, elem);
5733         if (err < 0)
5734                 goto nla_put_failure;
5735
5736         nla_nest_end(skb, nest);
5737
5738         nlmsg_end(skb, nlh);
5739         return 0;
5740
5741 nla_put_failure:
5742         nlmsg_trim(skb, nlh);
5743         return -1;
5744 }
5745
5746 static int nft_setelem_parse_flags(const struct nft_set *set,
5747                                    const struct nlattr *attr, u32 *flags)
5748 {
5749         if (attr == NULL)
5750                 return 0;
5751
5752         *flags = ntohl(nla_get_be32(attr));
5753         if (*flags & ~(NFT_SET_ELEM_INTERVAL_END | NFT_SET_ELEM_CATCHALL))
5754                 return -EOPNOTSUPP;
5755         if (!(set->flags & NFT_SET_INTERVAL) &&
5756             *flags & NFT_SET_ELEM_INTERVAL_END)
5757                 return -EINVAL;
5758         if ((*flags & (NFT_SET_ELEM_INTERVAL_END | NFT_SET_ELEM_CATCHALL)) ==
5759             (NFT_SET_ELEM_INTERVAL_END | NFT_SET_ELEM_CATCHALL))
5760                 return -EINVAL;
5761
5762         return 0;
5763 }
5764
5765 static int nft_setelem_parse_key(struct nft_ctx *ctx, struct nft_set *set,
5766                                  struct nft_data *key, struct nlattr *attr)
5767 {
5768         struct nft_data_desc desc = {
5769                 .type   = NFT_DATA_VALUE,
5770                 .size   = NFT_DATA_VALUE_MAXLEN,
5771                 .len    = set->klen,
5772         };
5773
5774         return nft_data_init(ctx, key, &desc, attr);
5775 }
5776
5777 static int nft_setelem_parse_data(struct nft_ctx *ctx, struct nft_set *set,
5778                                   struct nft_data_desc *desc,
5779                                   struct nft_data *data,
5780                                   struct nlattr *attr)
5781 {
5782         u32 dtype;
5783
5784         if (set->dtype == NFT_DATA_VERDICT)
5785                 dtype = NFT_DATA_VERDICT;
5786         else
5787                 dtype = NFT_DATA_VALUE;
5788
5789         desc->type = dtype;
5790         desc->size = NFT_DATA_VALUE_MAXLEN;
5791         desc->len = set->dlen;
5792         desc->flags = NFT_DATA_DESC_SETELEM;
5793
5794         return nft_data_init(ctx, data, desc, attr);
5795 }
5796
5797 static void *nft_setelem_catchall_get(const struct net *net,
5798                                       const struct nft_set *set)
5799 {
5800         struct nft_set_elem_catchall *catchall;
5801         u8 genmask = nft_genmask_cur(net);
5802         struct nft_set_ext *ext;
5803         void *priv = NULL;
5804
5805         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
5806                 ext = nft_set_elem_ext(set, catchall->elem);
5807                 if (!nft_set_elem_active(ext, genmask) ||
5808                     nft_set_elem_expired(ext))
5809                         continue;
5810
5811                 priv = catchall->elem;
5812                 break;
5813         }
5814
5815         return priv;
5816 }
5817
5818 static int nft_setelem_get(struct nft_ctx *ctx, struct nft_set *set,
5819                            struct nft_set_elem *elem, u32 flags)
5820 {
5821         void *priv;
5822
5823         if (!(flags & NFT_SET_ELEM_CATCHALL)) {
5824                 priv = set->ops->get(ctx->net, set, elem, flags);
5825                 if (IS_ERR(priv))
5826                         return PTR_ERR(priv);
5827         } else {
5828                 priv = nft_setelem_catchall_get(ctx->net, set);
5829                 if (!priv)
5830                         return -ENOENT;
5831         }
5832         elem->priv = priv;
5833
5834         return 0;
5835 }
5836
5837 static int nft_get_set_elem(struct nft_ctx *ctx, struct nft_set *set,
5838                             const struct nlattr *attr)
5839 {
5840         struct nlattr *nla[NFTA_SET_ELEM_MAX + 1];
5841         struct nft_set_elem elem;
5842         struct sk_buff *skb;
5843         uint32_t flags = 0;
5844         int err;
5845
5846         err = nla_parse_nested_deprecated(nla, NFTA_SET_ELEM_MAX, attr,
5847                                           nft_set_elem_policy, NULL);
5848         if (err < 0)
5849                 return err;
5850
5851         err = nft_setelem_parse_flags(set, nla[NFTA_SET_ELEM_FLAGS], &flags);
5852         if (err < 0)
5853                 return err;
5854
5855         if (!nla[NFTA_SET_ELEM_KEY] && !(flags & NFT_SET_ELEM_CATCHALL))
5856                 return -EINVAL;
5857
5858         if (nla[NFTA_SET_ELEM_KEY]) {
5859                 err = nft_setelem_parse_key(ctx, set, &elem.key.val,
5860                                             nla[NFTA_SET_ELEM_KEY]);
5861                 if (err < 0)
5862                         return err;
5863         }
5864
5865         if (nla[NFTA_SET_ELEM_KEY_END]) {
5866                 err = nft_setelem_parse_key(ctx, set, &elem.key_end.val,
5867                                             nla[NFTA_SET_ELEM_KEY_END]);
5868                 if (err < 0)
5869                         return err;
5870         }
5871
5872         err = nft_setelem_get(ctx, set, &elem, flags);
5873         if (err < 0)
5874                 return err;
5875
5876         err = -ENOMEM;
5877         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_ATOMIC);
5878         if (skb == NULL)
5879                 return err;
5880
5881         err = nf_tables_fill_setelem_info(skb, ctx, ctx->seq, ctx->portid,
5882                                           NFT_MSG_NEWSETELEM, 0, set, &elem);
5883         if (err < 0)
5884                 goto err_fill_setelem;
5885
5886         return nfnetlink_unicast(skb, ctx->net, ctx->portid);
5887
5888 err_fill_setelem:
5889         kfree_skb(skb);
5890         return err;
5891 }
5892
5893 /* called with rcu_read_lock held */
5894 static int nf_tables_getsetelem(struct sk_buff *skb,
5895                                 const struct nfnl_info *info,
5896                                 const struct nlattr * const nla[])
5897 {
5898         struct netlink_ext_ack *extack = info->extack;
5899         u8 genmask = nft_genmask_cur(info->net);
5900         u8 family = info->nfmsg->nfgen_family;
5901         struct net *net = info->net;
5902         struct nft_table *table;
5903         struct nft_set *set;
5904         struct nlattr *attr;
5905         struct nft_ctx ctx;
5906         int rem, err = 0;
5907
5908         table = nft_table_lookup(net, nla[NFTA_SET_ELEM_LIST_TABLE], family,
5909                                  genmask, 0);
5910         if (IS_ERR(table)) {
5911                 NL_SET_BAD_ATTR(extack, nla[NFTA_SET_ELEM_LIST_TABLE]);
5912                 return PTR_ERR(table);
5913         }
5914
5915         set = nft_set_lookup(table, nla[NFTA_SET_ELEM_LIST_SET], genmask);
5916         if (IS_ERR(set))
5917                 return PTR_ERR(set);
5918
5919         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
5920
5921         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
5922                 struct netlink_dump_control c = {
5923                         .start = nf_tables_dump_set_start,
5924                         .dump = nf_tables_dump_set,
5925                         .done = nf_tables_dump_set_done,
5926                         .module = THIS_MODULE,
5927                 };
5928                 struct nft_set_dump_ctx dump_ctx = {
5929                         .set = set,
5930                         .ctx = ctx,
5931                 };
5932
5933                 c.data = &dump_ctx;
5934                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
5935         }
5936
5937         if (!nla[NFTA_SET_ELEM_LIST_ELEMENTS])
5938                 return -EINVAL;
5939
5940         nla_for_each_nested(attr, nla[NFTA_SET_ELEM_LIST_ELEMENTS], rem) {
5941                 err = nft_get_set_elem(&ctx, set, attr);
5942                 if (err < 0) {
5943                         NL_SET_BAD_ATTR(extack, attr);
5944                         break;
5945                 }
5946         }
5947
5948         return err;
5949 }
5950
5951 static void nf_tables_setelem_notify(const struct nft_ctx *ctx,
5952                                      const struct nft_set *set,
5953                                      const struct nft_set_elem *elem,
5954                                      int event)
5955 {
5956         struct nftables_pernet *nft_net;
5957         struct net *net = ctx->net;
5958         u32 portid = ctx->portid;
5959         struct sk_buff *skb;
5960         u16 flags = 0;
5961         int err;
5962
5963         if (!ctx->report && !nfnetlink_has_listeners(net, NFNLGRP_NFTABLES))
5964                 return;
5965
5966         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
5967         if (skb == NULL)
5968                 goto err;
5969
5970         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
5971                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
5972
5973         err = nf_tables_fill_setelem_info(skb, ctx, 0, portid, event, flags,
5974                                           set, elem);
5975         if (err < 0) {
5976                 kfree_skb(skb);
5977                 goto err;
5978         }
5979
5980         nft_net = nft_pernet(net);
5981         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
5982         return;
5983 err:
5984         nfnetlink_set_err(net, portid, NFNLGRP_NFTABLES, -ENOBUFS);
5985 }
5986
5987 static struct nft_trans *nft_trans_elem_alloc(struct nft_ctx *ctx,
5988                                               int msg_type,
5989                                               struct nft_set *set)
5990 {
5991         struct nft_trans *trans;
5992
5993         trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_elem));
5994         if (trans == NULL)
5995                 return NULL;
5996
5997         nft_trans_elem_set(trans) = set;
5998         return trans;
5999 }
6000
6001 struct nft_expr *nft_set_elem_expr_alloc(const struct nft_ctx *ctx,
6002                                          const struct nft_set *set,
6003                                          const struct nlattr *attr)
6004 {
6005         struct nft_expr *expr;
6006         int err;
6007
6008         expr = nft_expr_init(ctx, attr);
6009         if (IS_ERR(expr))
6010                 return expr;
6011
6012         err = -EOPNOTSUPP;
6013         if (expr->ops->type->flags & NFT_EXPR_GC) {
6014                 if (set->flags & NFT_SET_TIMEOUT)
6015                         goto err_set_elem_expr;
6016                 if (!set->ops->gc_init)
6017                         goto err_set_elem_expr;
6018                 set->ops->gc_init(set);
6019         }
6020
6021         return expr;
6022
6023 err_set_elem_expr:
6024         nft_expr_destroy(ctx, expr);
6025         return ERR_PTR(err);
6026 }
6027
6028 static int nft_set_ext_check(const struct nft_set_ext_tmpl *tmpl, u8 id, u32 len)
6029 {
6030         len += nft_set_ext_types[id].len;
6031         if (len > tmpl->ext_len[id] ||
6032             len > U8_MAX)
6033                 return -1;
6034
6035         return 0;
6036 }
6037
6038 static int nft_set_ext_memcpy(const struct nft_set_ext_tmpl *tmpl, u8 id,
6039                               void *to, const void *from, u32 len)
6040 {
6041         if (nft_set_ext_check(tmpl, id, len) < 0)
6042                 return -1;
6043
6044         memcpy(to, from, len);
6045
6046         return 0;
6047 }
6048
6049 void *nft_set_elem_init(const struct nft_set *set,
6050                         const struct nft_set_ext_tmpl *tmpl,
6051                         const u32 *key, const u32 *key_end,
6052                         const u32 *data, u64 timeout, u64 expiration, gfp_t gfp)
6053 {
6054         struct nft_set_ext *ext;
6055         void *elem;
6056
6057         elem = kzalloc(set->ops->elemsize + tmpl->len, gfp);
6058         if (elem == NULL)
6059                 return ERR_PTR(-ENOMEM);
6060
6061         ext = nft_set_elem_ext(set, elem);
6062         nft_set_ext_init(ext, tmpl);
6063
6064         if (nft_set_ext_exists(ext, NFT_SET_EXT_KEY) &&
6065             nft_set_ext_memcpy(tmpl, NFT_SET_EXT_KEY,
6066                                nft_set_ext_key(ext), key, set->klen) < 0)
6067                 goto err_ext_check;
6068
6069         if (nft_set_ext_exists(ext, NFT_SET_EXT_KEY_END) &&
6070             nft_set_ext_memcpy(tmpl, NFT_SET_EXT_KEY_END,
6071                                nft_set_ext_key_end(ext), key_end, set->klen) < 0)
6072                 goto err_ext_check;
6073
6074         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA) &&
6075             nft_set_ext_memcpy(tmpl, NFT_SET_EXT_DATA,
6076                                nft_set_ext_data(ext), data, set->dlen) < 0)
6077                 goto err_ext_check;
6078
6079         if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPIRATION)) {
6080                 *nft_set_ext_expiration(ext) = get_jiffies_64() + expiration;
6081                 if (expiration == 0)
6082                         *nft_set_ext_expiration(ext) += timeout;
6083         }
6084         if (nft_set_ext_exists(ext, NFT_SET_EXT_TIMEOUT))
6085                 *nft_set_ext_timeout(ext) = timeout;
6086
6087         return elem;
6088
6089 err_ext_check:
6090         kfree(elem);
6091
6092         return ERR_PTR(-EINVAL);
6093 }
6094
6095 static void __nft_set_elem_expr_destroy(const struct nft_ctx *ctx,
6096                                         struct nft_expr *expr)
6097 {
6098         if (expr->ops->destroy_clone) {
6099                 expr->ops->destroy_clone(ctx, expr);
6100                 module_put(expr->ops->type->owner);
6101         } else {
6102                 nf_tables_expr_destroy(ctx, expr);
6103         }
6104 }
6105
6106 static void nft_set_elem_expr_destroy(const struct nft_ctx *ctx,
6107                                       struct nft_set_elem_expr *elem_expr)
6108 {
6109         struct nft_expr *expr;
6110         u32 size;
6111
6112         nft_setelem_expr_foreach(expr, elem_expr, size)
6113                 __nft_set_elem_expr_destroy(ctx, expr);
6114 }
6115
6116 /* Drop references and destroy. Called from gc, dynset and abort path. */
6117 void nft_set_elem_destroy(const struct nft_set *set, void *elem,
6118                           bool destroy_expr)
6119 {
6120         struct nft_set_ext *ext = nft_set_elem_ext(set, elem);
6121         struct nft_ctx ctx = {
6122                 .net    = read_pnet(&set->net),
6123                 .family = set->table->family,
6124         };
6125
6126         nft_data_release(nft_set_ext_key(ext), NFT_DATA_VALUE);
6127         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA))
6128                 nft_data_release(nft_set_ext_data(ext), set->dtype);
6129         if (destroy_expr && nft_set_ext_exists(ext, NFT_SET_EXT_EXPRESSIONS))
6130                 nft_set_elem_expr_destroy(&ctx, nft_set_ext_expr(ext));
6131
6132         if (nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF))
6133                 (*nft_set_ext_obj(ext))->use--;
6134         kfree(elem);
6135 }
6136 EXPORT_SYMBOL_GPL(nft_set_elem_destroy);
6137
6138 /* Destroy element. References have been already dropped in the preparation
6139  * path via nft_setelem_data_deactivate().
6140  */
6141 void nf_tables_set_elem_destroy(const struct nft_ctx *ctx,
6142                                 const struct nft_set *set, void *elem)
6143 {
6144         struct nft_set_ext *ext = nft_set_elem_ext(set, elem);
6145
6146         if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPRESSIONS))
6147                 nft_set_elem_expr_destroy(ctx, nft_set_ext_expr(ext));
6148
6149         kfree(elem);
6150 }
6151
6152 int nft_set_elem_expr_clone(const struct nft_ctx *ctx, struct nft_set *set,
6153                             struct nft_expr *expr_array[])
6154 {
6155         struct nft_expr *expr;
6156         int err, i, k;
6157
6158         for (i = 0; i < set->num_exprs; i++) {
6159                 expr = kzalloc(set->exprs[i]->ops->size, GFP_KERNEL_ACCOUNT);
6160                 if (!expr)
6161                         goto err_expr;
6162
6163                 err = nft_expr_clone(expr, set->exprs[i]);
6164                 if (err < 0) {
6165                         kfree(expr);
6166                         goto err_expr;
6167                 }
6168                 expr_array[i] = expr;
6169         }
6170
6171         return 0;
6172
6173 err_expr:
6174         for (k = i - 1; k >= 0; k--)
6175                 nft_expr_destroy(ctx, expr_array[k]);
6176
6177         return -ENOMEM;
6178 }
6179
6180 static int nft_set_elem_expr_setup(struct nft_ctx *ctx,
6181                                    const struct nft_set_ext_tmpl *tmpl,
6182                                    const struct nft_set_ext *ext,
6183                                    struct nft_expr *expr_array[],
6184                                    u32 num_exprs)
6185 {
6186         struct nft_set_elem_expr *elem_expr = nft_set_ext_expr(ext);
6187         u32 len = sizeof(struct nft_set_elem_expr);
6188         struct nft_expr *expr;
6189         int i, err;
6190
6191         if (num_exprs == 0)
6192                 return 0;
6193
6194         for (i = 0; i < num_exprs; i++)
6195                 len += expr_array[i]->ops->size;
6196
6197         if (nft_set_ext_check(tmpl, NFT_SET_EXT_EXPRESSIONS, len) < 0)
6198                 return -EINVAL;
6199
6200         for (i = 0; i < num_exprs; i++) {
6201                 expr = nft_setelem_expr_at(elem_expr, elem_expr->size);
6202                 err = nft_expr_clone(expr, expr_array[i]);
6203                 if (err < 0)
6204                         goto err_elem_expr_setup;
6205
6206                 elem_expr->size += expr_array[i]->ops->size;
6207                 nft_expr_destroy(ctx, expr_array[i]);
6208                 expr_array[i] = NULL;
6209         }
6210
6211         return 0;
6212
6213 err_elem_expr_setup:
6214         for (; i < num_exprs; i++) {
6215                 nft_expr_destroy(ctx, expr_array[i]);
6216                 expr_array[i] = NULL;
6217         }
6218
6219         return -ENOMEM;
6220 }
6221
6222 struct nft_set_ext *nft_set_catchall_lookup(const struct net *net,
6223                                             const struct nft_set *set)
6224 {
6225         struct nft_set_elem_catchall *catchall;
6226         u8 genmask = nft_genmask_cur(net);
6227         struct nft_set_ext *ext;
6228
6229         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
6230                 ext = nft_set_elem_ext(set, catchall->elem);
6231                 if (nft_set_elem_active(ext, genmask) &&
6232                     !nft_set_elem_expired(ext))
6233                         return ext;
6234         }
6235
6236         return NULL;
6237 }
6238 EXPORT_SYMBOL_GPL(nft_set_catchall_lookup);
6239
6240 void *nft_set_catchall_gc(const struct nft_set *set)
6241 {
6242         struct nft_set_elem_catchall *catchall, *next;
6243         struct nft_set_ext *ext;
6244         void *elem = NULL;
6245
6246         list_for_each_entry_safe(catchall, next, &set->catchall_list, list) {
6247                 ext = nft_set_elem_ext(set, catchall->elem);
6248
6249                 if (!nft_set_elem_expired(ext) ||
6250                     nft_set_elem_mark_busy(ext))
6251                         continue;
6252
6253                 elem = catchall->elem;
6254                 list_del_rcu(&catchall->list);
6255                 kfree_rcu(catchall, rcu);
6256                 break;
6257         }
6258
6259         return elem;
6260 }
6261 EXPORT_SYMBOL_GPL(nft_set_catchall_gc);
6262
6263 static int nft_setelem_catchall_insert(const struct net *net,
6264                                        struct nft_set *set,
6265                                        const struct nft_set_elem *elem,
6266                                        struct nft_set_ext **pext)
6267 {
6268         struct nft_set_elem_catchall *catchall;
6269         u8 genmask = nft_genmask_next(net);
6270         struct nft_set_ext *ext;
6271
6272         list_for_each_entry(catchall, &set->catchall_list, list) {
6273                 ext = nft_set_elem_ext(set, catchall->elem);
6274                 if (nft_set_elem_active(ext, genmask)) {
6275                         *pext = ext;
6276                         return -EEXIST;
6277                 }
6278         }
6279
6280         catchall = kmalloc(sizeof(*catchall), GFP_KERNEL);
6281         if (!catchall)
6282                 return -ENOMEM;
6283
6284         catchall->elem = elem->priv;
6285         list_add_tail_rcu(&catchall->list, &set->catchall_list);
6286
6287         return 0;
6288 }
6289
6290 static int nft_setelem_insert(const struct net *net,
6291                               struct nft_set *set,
6292                               const struct nft_set_elem *elem,
6293                               struct nft_set_ext **ext, unsigned int flags)
6294 {
6295         int ret;
6296
6297         if (flags & NFT_SET_ELEM_CATCHALL)
6298                 ret = nft_setelem_catchall_insert(net, set, elem, ext);
6299         else
6300                 ret = set->ops->insert(net, set, elem, ext);
6301
6302         return ret;
6303 }
6304
6305 static bool nft_setelem_is_catchall(const struct nft_set *set,
6306                                     const struct nft_set_elem *elem)
6307 {
6308         struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
6309
6310         if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
6311             *nft_set_ext_flags(ext) & NFT_SET_ELEM_CATCHALL)
6312                 return true;
6313
6314         return false;
6315 }
6316
6317 static void nft_setelem_activate(struct net *net, struct nft_set *set,
6318                                  struct nft_set_elem *elem)
6319 {
6320         struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
6321
6322         if (nft_setelem_is_catchall(set, elem)) {
6323                 nft_set_elem_change_active(net, set, ext);
6324                 nft_set_elem_clear_busy(ext);
6325         } else {
6326                 set->ops->activate(net, set, elem);
6327         }
6328 }
6329
6330 static int nft_setelem_catchall_deactivate(const struct net *net,
6331                                            struct nft_set *set,
6332                                            struct nft_set_elem *elem)
6333 {
6334         struct nft_set_elem_catchall *catchall;
6335         struct nft_set_ext *ext;
6336
6337         list_for_each_entry(catchall, &set->catchall_list, list) {
6338                 ext = nft_set_elem_ext(set, catchall->elem);
6339                 if (!nft_is_active(net, ext) ||
6340                     nft_set_elem_mark_busy(ext))
6341                         continue;
6342
6343                 kfree(elem->priv);
6344                 elem->priv = catchall->elem;
6345                 nft_set_elem_change_active(net, set, ext);
6346                 return 0;
6347         }
6348
6349         return -ENOENT;
6350 }
6351
6352 static int __nft_setelem_deactivate(const struct net *net,
6353                                     struct nft_set *set,
6354                                     struct nft_set_elem *elem)
6355 {
6356         void *priv;
6357
6358         priv = set->ops->deactivate(net, set, elem);
6359         if (!priv)
6360                 return -ENOENT;
6361
6362         kfree(elem->priv);
6363         elem->priv = priv;
6364         set->ndeact++;
6365
6366         return 0;
6367 }
6368
6369 static int nft_setelem_deactivate(const struct net *net,
6370                                   struct nft_set *set,
6371                                   struct nft_set_elem *elem, u32 flags)
6372 {
6373         int ret;
6374
6375         if (flags & NFT_SET_ELEM_CATCHALL)
6376                 ret = nft_setelem_catchall_deactivate(net, set, elem);
6377         else
6378                 ret = __nft_setelem_deactivate(net, set, elem);
6379
6380         return ret;
6381 }
6382
6383 static void nft_setelem_catchall_remove(const struct net *net,
6384                                         const struct nft_set *set,
6385                                         const struct nft_set_elem *elem)
6386 {
6387         struct nft_set_elem_catchall *catchall, *next;
6388
6389         list_for_each_entry_safe(catchall, next, &set->catchall_list, list) {
6390                 if (catchall->elem == elem->priv) {
6391                         list_del_rcu(&catchall->list);
6392                         kfree_rcu(catchall, rcu);
6393                         break;
6394                 }
6395         }
6396 }
6397
6398 static void nft_setelem_remove(const struct net *net,
6399                                const struct nft_set *set,
6400                                const struct nft_set_elem *elem)
6401 {
6402         if (nft_setelem_is_catchall(set, elem))
6403                 nft_setelem_catchall_remove(net, set, elem);
6404         else
6405                 set->ops->remove(net, set, elem);
6406 }
6407
6408 static bool nft_setelem_valid_key_end(const struct nft_set *set,
6409                                       struct nlattr **nla, u32 flags)
6410 {
6411         if ((set->flags & (NFT_SET_CONCAT | NFT_SET_INTERVAL)) ==
6412                           (NFT_SET_CONCAT | NFT_SET_INTERVAL)) {
6413                 if (flags & NFT_SET_ELEM_INTERVAL_END)
6414                         return false;
6415
6416                 if (nla[NFTA_SET_ELEM_KEY_END] &&
6417                     flags & NFT_SET_ELEM_CATCHALL)
6418                         return false;
6419         } else {
6420                 if (nla[NFTA_SET_ELEM_KEY_END])
6421                         return false;
6422         }
6423
6424         return true;
6425 }
6426
6427 static int nft_add_set_elem(struct nft_ctx *ctx, struct nft_set *set,
6428                             const struct nlattr *attr, u32 nlmsg_flags)
6429 {
6430         struct nft_expr *expr_array[NFT_SET_EXPR_MAX] = {};
6431         struct nlattr *nla[NFTA_SET_ELEM_MAX + 1];
6432         u8 genmask = nft_genmask_next(ctx->net);
6433         u32 flags = 0, size = 0, num_exprs = 0;
6434         struct nft_set_ext_tmpl tmpl;
6435         struct nft_set_ext *ext, *ext2;
6436         struct nft_set_elem elem;
6437         struct nft_set_binding *binding;
6438         struct nft_object *obj = NULL;
6439         struct nft_userdata *udata;
6440         struct nft_data_desc desc;
6441         enum nft_registers dreg;
6442         struct nft_trans *trans;
6443         u64 timeout;
6444         u64 expiration;
6445         int err, i;
6446         u8 ulen;
6447
6448         err = nla_parse_nested_deprecated(nla, NFTA_SET_ELEM_MAX, attr,
6449                                           nft_set_elem_policy, NULL);
6450         if (err < 0)
6451                 return err;
6452
6453         nft_set_ext_prepare(&tmpl);
6454
6455         err = nft_setelem_parse_flags(set, nla[NFTA_SET_ELEM_FLAGS], &flags);
6456         if (err < 0)
6457                 return err;
6458
6459         if (((flags & NFT_SET_ELEM_CATCHALL) && nla[NFTA_SET_ELEM_KEY]) ||
6460             (!(flags & NFT_SET_ELEM_CATCHALL) && !nla[NFTA_SET_ELEM_KEY]))
6461                 return -EINVAL;
6462
6463         if (flags != 0) {
6464                 err = nft_set_ext_add(&tmpl, NFT_SET_EXT_FLAGS);
6465                 if (err < 0)
6466                         return err;
6467         }
6468
6469         if (set->flags & NFT_SET_MAP) {
6470                 if (nla[NFTA_SET_ELEM_DATA] == NULL &&
6471                     !(flags & NFT_SET_ELEM_INTERVAL_END))
6472                         return -EINVAL;
6473         } else {
6474                 if (nla[NFTA_SET_ELEM_DATA] != NULL)
6475                         return -EINVAL;
6476         }
6477
6478         if (set->flags & NFT_SET_OBJECT) {
6479                 if (!nla[NFTA_SET_ELEM_OBJREF] &&
6480                     !(flags & NFT_SET_ELEM_INTERVAL_END))
6481                         return -EINVAL;
6482         } else {
6483                 if (nla[NFTA_SET_ELEM_OBJREF])
6484                         return -EINVAL;
6485         }
6486
6487         if (!nft_setelem_valid_key_end(set, nla, flags))
6488                 return -EINVAL;
6489
6490         if ((flags & NFT_SET_ELEM_INTERVAL_END) &&
6491              (nla[NFTA_SET_ELEM_DATA] ||
6492               nla[NFTA_SET_ELEM_OBJREF] ||
6493               nla[NFTA_SET_ELEM_TIMEOUT] ||
6494               nla[NFTA_SET_ELEM_EXPIRATION] ||
6495               nla[NFTA_SET_ELEM_USERDATA] ||
6496               nla[NFTA_SET_ELEM_EXPR] ||
6497               nla[NFTA_SET_ELEM_KEY_END] ||
6498               nla[NFTA_SET_ELEM_EXPRESSIONS]))
6499                 return -EINVAL;
6500
6501         timeout = 0;
6502         if (nla[NFTA_SET_ELEM_TIMEOUT] != NULL) {
6503                 if (!(set->flags & NFT_SET_TIMEOUT))
6504                         return -EINVAL;
6505                 err = nf_msecs_to_jiffies64(nla[NFTA_SET_ELEM_TIMEOUT],
6506                                             &timeout);
6507                 if (err)
6508                         return err;
6509         } else if (set->flags & NFT_SET_TIMEOUT &&
6510                    !(flags & NFT_SET_ELEM_INTERVAL_END)) {
6511                 timeout = READ_ONCE(set->timeout);
6512         }
6513
6514         expiration = 0;
6515         if (nla[NFTA_SET_ELEM_EXPIRATION] != NULL) {
6516                 if (!(set->flags & NFT_SET_TIMEOUT))
6517                         return -EINVAL;
6518                 err = nf_msecs_to_jiffies64(nla[NFTA_SET_ELEM_EXPIRATION],
6519                                             &expiration);
6520                 if (err)
6521                         return err;
6522         }
6523
6524         if (nla[NFTA_SET_ELEM_EXPR]) {
6525                 struct nft_expr *expr;
6526
6527                 if (set->num_exprs && set->num_exprs != 1)
6528                         return -EOPNOTSUPP;
6529
6530                 expr = nft_set_elem_expr_alloc(ctx, set,
6531                                                nla[NFTA_SET_ELEM_EXPR]);
6532                 if (IS_ERR(expr))
6533                         return PTR_ERR(expr);
6534
6535                 expr_array[0] = expr;
6536                 num_exprs = 1;
6537
6538                 if (set->num_exprs && set->exprs[0]->ops != expr->ops) {
6539                         err = -EOPNOTSUPP;
6540                         goto err_set_elem_expr;
6541                 }
6542         } else if (nla[NFTA_SET_ELEM_EXPRESSIONS]) {
6543                 struct nft_expr *expr;
6544                 struct nlattr *tmp;
6545                 int left;
6546
6547                 i = 0;
6548                 nla_for_each_nested(tmp, nla[NFTA_SET_ELEM_EXPRESSIONS], left) {
6549                         if (i == NFT_SET_EXPR_MAX ||
6550                             (set->num_exprs && set->num_exprs == i)) {
6551                                 err = -E2BIG;
6552                                 goto err_set_elem_expr;
6553                         }
6554                         if (nla_type(tmp) != NFTA_LIST_ELEM) {
6555                                 err = -EINVAL;
6556                                 goto err_set_elem_expr;
6557                         }
6558                         expr = nft_set_elem_expr_alloc(ctx, set, tmp);
6559                         if (IS_ERR(expr)) {
6560                                 err = PTR_ERR(expr);
6561                                 goto err_set_elem_expr;
6562                         }
6563                         expr_array[i] = expr;
6564                         num_exprs++;
6565
6566                         if (set->num_exprs && expr->ops != set->exprs[i]->ops) {
6567                                 err = -EOPNOTSUPP;
6568                                 goto err_set_elem_expr;
6569                         }
6570                         i++;
6571                 }
6572                 if (set->num_exprs && set->num_exprs != i) {
6573                         err = -EOPNOTSUPP;
6574                         goto err_set_elem_expr;
6575                 }
6576         } else if (set->num_exprs > 0 &&
6577                    !(flags & NFT_SET_ELEM_INTERVAL_END)) {
6578                 err = nft_set_elem_expr_clone(ctx, set, expr_array);
6579                 if (err < 0)
6580                         goto err_set_elem_expr_clone;
6581
6582                 num_exprs = set->num_exprs;
6583         }
6584
6585         if (nla[NFTA_SET_ELEM_KEY]) {
6586                 err = nft_setelem_parse_key(ctx, set, &elem.key.val,
6587                                             nla[NFTA_SET_ELEM_KEY]);
6588                 if (err < 0)
6589                         goto err_set_elem_expr;
6590
6591                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_KEY, set->klen);
6592                 if (err < 0)
6593                         goto err_parse_key;
6594         }
6595
6596         if (nla[NFTA_SET_ELEM_KEY_END]) {
6597                 err = nft_setelem_parse_key(ctx, set, &elem.key_end.val,
6598                                             nla[NFTA_SET_ELEM_KEY_END]);
6599                 if (err < 0)
6600                         goto err_parse_key;
6601
6602                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_KEY_END, set->klen);
6603                 if (err < 0)
6604                         goto err_parse_key_end;
6605         }
6606
6607         if (timeout > 0) {
6608                 err = nft_set_ext_add(&tmpl, NFT_SET_EXT_EXPIRATION);
6609                 if (err < 0)
6610                         goto err_parse_key_end;
6611
6612                 if (timeout != READ_ONCE(set->timeout)) {
6613                         err = nft_set_ext_add(&tmpl, NFT_SET_EXT_TIMEOUT);
6614                         if (err < 0)
6615                                 goto err_parse_key_end;
6616                 }
6617         }
6618
6619         if (num_exprs) {
6620                 for (i = 0; i < num_exprs; i++)
6621                         size += expr_array[i]->ops->size;
6622
6623                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_EXPRESSIONS,
6624                                              sizeof(struct nft_set_elem_expr) + size);
6625                 if (err < 0)
6626                         goto err_parse_key_end;
6627         }
6628
6629         if (nla[NFTA_SET_ELEM_OBJREF] != NULL) {
6630                 obj = nft_obj_lookup(ctx->net, ctx->table,
6631                                      nla[NFTA_SET_ELEM_OBJREF],
6632                                      set->objtype, genmask);
6633                 if (IS_ERR(obj)) {
6634                         err = PTR_ERR(obj);
6635                         goto err_parse_key_end;
6636                 }
6637                 err = nft_set_ext_add(&tmpl, NFT_SET_EXT_OBJREF);
6638                 if (err < 0)
6639                         goto err_parse_key_end;
6640         }
6641
6642         if (nla[NFTA_SET_ELEM_DATA] != NULL) {
6643                 err = nft_setelem_parse_data(ctx, set, &desc, &elem.data.val,
6644                                              nla[NFTA_SET_ELEM_DATA]);
6645                 if (err < 0)
6646                         goto err_parse_key_end;
6647
6648                 dreg = nft_type_to_reg(set->dtype);
6649                 list_for_each_entry(binding, &set->bindings, list) {
6650                         struct nft_ctx bind_ctx = {
6651                                 .net    = ctx->net,
6652                                 .family = ctx->family,
6653                                 .table  = ctx->table,
6654                                 .chain  = (struct nft_chain *)binding->chain,
6655                         };
6656
6657                         if (!(binding->flags & NFT_SET_MAP))
6658                                 continue;
6659
6660                         err = nft_validate_register_store(&bind_ctx, dreg,
6661                                                           &elem.data.val,
6662                                                           desc.type, desc.len);
6663                         if (err < 0)
6664                                 goto err_parse_data;
6665
6666                         if (desc.type == NFT_DATA_VERDICT &&
6667                             (elem.data.val.verdict.code == NFT_GOTO ||
6668                              elem.data.val.verdict.code == NFT_JUMP))
6669                                 nft_validate_state_update(ctx->table,
6670                                                           NFT_VALIDATE_NEED);
6671                 }
6672
6673                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_DATA, desc.len);
6674                 if (err < 0)
6675                         goto err_parse_data;
6676         }
6677
6678         /* The full maximum length of userdata can exceed the maximum
6679          * offset value (U8_MAX) for following extensions, therefor it
6680          * must be the last extension added.
6681          */
6682         ulen = 0;
6683         if (nla[NFTA_SET_ELEM_USERDATA] != NULL) {
6684                 ulen = nla_len(nla[NFTA_SET_ELEM_USERDATA]);
6685                 if (ulen > 0) {
6686                         err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_USERDATA,
6687                                                      ulen);
6688                         if (err < 0)
6689                                 goto err_parse_data;
6690                 }
6691         }
6692
6693         elem.priv = nft_set_elem_init(set, &tmpl, elem.key.val.data,
6694                                       elem.key_end.val.data, elem.data.val.data,
6695                                       timeout, expiration, GFP_KERNEL_ACCOUNT);
6696         if (IS_ERR(elem.priv)) {
6697                 err = PTR_ERR(elem.priv);
6698                 goto err_parse_data;
6699         }
6700
6701         ext = nft_set_elem_ext(set, elem.priv);
6702         if (flags)
6703                 *nft_set_ext_flags(ext) = flags;
6704
6705         if (obj) {
6706                 *nft_set_ext_obj(ext) = obj;
6707                 obj->use++;
6708         }
6709         if (ulen > 0) {
6710                 if (nft_set_ext_check(&tmpl, NFT_SET_EXT_USERDATA, ulen) < 0) {
6711                         err = -EINVAL;
6712                         goto err_elem_free;
6713                 }
6714                 udata = nft_set_ext_userdata(ext);
6715                 udata->len = ulen - 1;
6716                 nla_memcpy(&udata->data, nla[NFTA_SET_ELEM_USERDATA], ulen);
6717         }
6718         err = nft_set_elem_expr_setup(ctx, &tmpl, ext, expr_array, num_exprs);
6719         if (err < 0)
6720                 goto err_elem_free;
6721
6722         trans = nft_trans_elem_alloc(ctx, NFT_MSG_NEWSETELEM, set);
6723         if (trans == NULL) {
6724                 err = -ENOMEM;
6725                 goto err_elem_free;
6726         }
6727
6728         ext->genmask = nft_genmask_cur(ctx->net) | NFT_SET_ELEM_BUSY_MASK;
6729
6730         err = nft_setelem_insert(ctx->net, set, &elem, &ext2, flags);
6731         if (err) {
6732                 if (err == -EEXIST) {
6733                         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA) ^
6734                             nft_set_ext_exists(ext2, NFT_SET_EXT_DATA) ||
6735                             nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF) ^
6736                             nft_set_ext_exists(ext2, NFT_SET_EXT_OBJREF))
6737                                 goto err_element_clash;
6738                         if ((nft_set_ext_exists(ext, NFT_SET_EXT_DATA) &&
6739                              nft_set_ext_exists(ext2, NFT_SET_EXT_DATA) &&
6740                              memcmp(nft_set_ext_data(ext),
6741                                     nft_set_ext_data(ext2), set->dlen) != 0) ||
6742                             (nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF) &&
6743                              nft_set_ext_exists(ext2, NFT_SET_EXT_OBJREF) &&
6744                              *nft_set_ext_obj(ext) != *nft_set_ext_obj(ext2)))
6745                                 goto err_element_clash;
6746                         else if (!(nlmsg_flags & NLM_F_EXCL))
6747                                 err = 0;
6748                 } else if (err == -ENOTEMPTY) {
6749                         /* ENOTEMPTY reports overlapping between this element
6750                          * and an existing one.
6751                          */
6752                         err = -EEXIST;
6753                 }
6754                 goto err_element_clash;
6755         }
6756
6757         if (!(flags & NFT_SET_ELEM_CATCHALL) && set->size &&
6758             !atomic_add_unless(&set->nelems, 1, set->size + set->ndeact)) {
6759                 err = -ENFILE;
6760                 goto err_set_full;
6761         }
6762
6763         nft_trans_elem(trans) = elem;
6764         nft_trans_commit_list_add_tail(ctx->net, trans);
6765         return 0;
6766
6767 err_set_full:
6768         nft_setelem_remove(ctx->net, set, &elem);
6769 err_element_clash:
6770         kfree(trans);
6771 err_elem_free:
6772         nft_set_elem_destroy(set, elem.priv, true);
6773 err_parse_data:
6774         if (nla[NFTA_SET_ELEM_DATA] != NULL)
6775                 nft_data_release(&elem.data.val, desc.type);
6776 err_parse_key_end:
6777         nft_data_release(&elem.key_end.val, NFT_DATA_VALUE);
6778 err_parse_key:
6779         nft_data_release(&elem.key.val, NFT_DATA_VALUE);
6780 err_set_elem_expr:
6781         for (i = 0; i < num_exprs && expr_array[i]; i++)
6782                 nft_expr_destroy(ctx, expr_array[i]);
6783 err_set_elem_expr_clone:
6784         return err;
6785 }
6786
6787 static int nf_tables_newsetelem(struct sk_buff *skb,
6788                                 const struct nfnl_info *info,
6789                                 const struct nlattr * const nla[])
6790 {
6791         struct netlink_ext_ack *extack = info->extack;
6792         u8 genmask = nft_genmask_next(info->net);
6793         u8 family = info->nfmsg->nfgen_family;
6794         struct net *net = info->net;
6795         const struct nlattr *attr;
6796         struct nft_table *table;
6797         struct nft_set *set;
6798         struct nft_ctx ctx;
6799         int rem, err;
6800
6801         if (nla[NFTA_SET_ELEM_LIST_ELEMENTS] == NULL)
6802                 return -EINVAL;
6803
6804         table = nft_table_lookup(net, nla[NFTA_SET_ELEM_LIST_TABLE], family,
6805                                  genmask, NETLINK_CB(skb).portid);
6806         if (IS_ERR(table)) {
6807                 NL_SET_BAD_ATTR(extack, nla[NFTA_SET_ELEM_LIST_TABLE]);
6808                 return PTR_ERR(table);
6809         }
6810
6811         set = nft_set_lookup_global(net, table, nla[NFTA_SET_ELEM_LIST_SET],
6812                                     nla[NFTA_SET_ELEM_LIST_SET_ID], genmask);
6813         if (IS_ERR(set))
6814                 return PTR_ERR(set);
6815
6816         if (!list_empty(&set->bindings) &&
6817             (set->flags & (NFT_SET_CONSTANT | NFT_SET_ANONYMOUS)))
6818                 return -EBUSY;
6819
6820         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
6821
6822         nla_for_each_nested(attr, nla[NFTA_SET_ELEM_LIST_ELEMENTS], rem) {
6823                 err = nft_add_set_elem(&ctx, set, attr, info->nlh->nlmsg_flags);
6824                 if (err < 0) {
6825                         NL_SET_BAD_ATTR(extack, attr);
6826                         return err;
6827                 }
6828         }
6829
6830         if (table->validate_state == NFT_VALIDATE_DO)
6831                 return nft_table_validate(net, table);
6832
6833         return 0;
6834 }
6835
6836 /**
6837  *      nft_data_hold - hold a nft_data item
6838  *
6839  *      @data: struct nft_data to release
6840  *      @type: type of data
6841  *
6842  *      Hold a nft_data item. NFT_DATA_VALUE types can be silently discarded,
6843  *      NFT_DATA_VERDICT bumps the reference to chains in case of NFT_JUMP and
6844  *      NFT_GOTO verdicts. This function must be called on active data objects
6845  *      from the second phase of the commit protocol.
6846  */
6847 void nft_data_hold(const struct nft_data *data, enum nft_data_types type)
6848 {
6849         struct nft_chain *chain;
6850
6851         if (type == NFT_DATA_VERDICT) {
6852                 switch (data->verdict.code) {
6853                 case NFT_JUMP:
6854                 case NFT_GOTO:
6855                         chain = data->verdict.chain;
6856                         chain->use++;
6857                         break;
6858                 }
6859         }
6860 }
6861
6862 static void nft_setelem_data_activate(const struct net *net,
6863                                       const struct nft_set *set,
6864                                       struct nft_set_elem *elem)
6865 {
6866         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
6867
6868         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA))
6869                 nft_data_hold(nft_set_ext_data(ext), set->dtype);
6870         if (nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF))
6871                 (*nft_set_ext_obj(ext))->use++;
6872 }
6873
6874 static void nft_setelem_data_deactivate(const struct net *net,
6875                                         const struct nft_set *set,
6876                                         struct nft_set_elem *elem)
6877 {
6878         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
6879
6880         if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA))
6881                 nft_data_release(nft_set_ext_data(ext), set->dtype);
6882         if (nft_set_ext_exists(ext, NFT_SET_EXT_OBJREF))
6883                 (*nft_set_ext_obj(ext))->use--;
6884 }
6885
6886 static int nft_del_setelem(struct nft_ctx *ctx, struct nft_set *set,
6887                            const struct nlattr *attr)
6888 {
6889         struct nlattr *nla[NFTA_SET_ELEM_MAX + 1];
6890         struct nft_set_ext_tmpl tmpl;
6891         struct nft_set_elem elem;
6892         struct nft_set_ext *ext;
6893         struct nft_trans *trans;
6894         u32 flags = 0;
6895         int err;
6896
6897         err = nla_parse_nested_deprecated(nla, NFTA_SET_ELEM_MAX, attr,
6898                                           nft_set_elem_policy, NULL);
6899         if (err < 0)
6900                 return err;
6901
6902         err = nft_setelem_parse_flags(set, nla[NFTA_SET_ELEM_FLAGS], &flags);
6903         if (err < 0)
6904                 return err;
6905
6906         if (!nla[NFTA_SET_ELEM_KEY] && !(flags & NFT_SET_ELEM_CATCHALL))
6907                 return -EINVAL;
6908
6909         if (!nft_setelem_valid_key_end(set, nla, flags))
6910                 return -EINVAL;
6911
6912         nft_set_ext_prepare(&tmpl);
6913
6914         if (flags != 0) {
6915                 err = nft_set_ext_add(&tmpl, NFT_SET_EXT_FLAGS);
6916                 if (err < 0)
6917                         return err;
6918         }
6919
6920         if (nla[NFTA_SET_ELEM_KEY]) {
6921                 err = nft_setelem_parse_key(ctx, set, &elem.key.val,
6922                                             nla[NFTA_SET_ELEM_KEY]);
6923                 if (err < 0)
6924                         return err;
6925
6926                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_KEY, set->klen);
6927                 if (err < 0)
6928                         goto fail_elem;
6929         }
6930
6931         if (nla[NFTA_SET_ELEM_KEY_END]) {
6932                 err = nft_setelem_parse_key(ctx, set, &elem.key_end.val,
6933                                             nla[NFTA_SET_ELEM_KEY_END]);
6934                 if (err < 0)
6935                         goto fail_elem;
6936
6937                 err = nft_set_ext_add_length(&tmpl, NFT_SET_EXT_KEY_END, set->klen);
6938                 if (err < 0)
6939                         goto fail_elem_key_end;
6940         }
6941
6942         err = -ENOMEM;
6943         elem.priv = nft_set_elem_init(set, &tmpl, elem.key.val.data,
6944                                       elem.key_end.val.data, NULL, 0, 0,
6945                                       GFP_KERNEL_ACCOUNT);
6946         if (IS_ERR(elem.priv)) {
6947                 err = PTR_ERR(elem.priv);
6948                 goto fail_elem_key_end;
6949         }
6950
6951         ext = nft_set_elem_ext(set, elem.priv);
6952         if (flags)
6953                 *nft_set_ext_flags(ext) = flags;
6954
6955         trans = nft_trans_elem_alloc(ctx, NFT_MSG_DELSETELEM, set);
6956         if (trans == NULL)
6957                 goto fail_trans;
6958
6959         err = nft_setelem_deactivate(ctx->net, set, &elem, flags);
6960         if (err < 0)
6961                 goto fail_ops;
6962
6963         nft_setelem_data_deactivate(ctx->net, set, &elem);
6964
6965         nft_trans_elem(trans) = elem;
6966         nft_trans_commit_list_add_tail(ctx->net, trans);
6967         return 0;
6968
6969 fail_ops:
6970         kfree(trans);
6971 fail_trans:
6972         kfree(elem.priv);
6973 fail_elem_key_end:
6974         nft_data_release(&elem.key_end.val, NFT_DATA_VALUE);
6975 fail_elem:
6976         nft_data_release(&elem.key.val, NFT_DATA_VALUE);
6977         return err;
6978 }
6979
6980 static int nft_setelem_flush(const struct nft_ctx *ctx,
6981                              struct nft_set *set,
6982                              const struct nft_set_iter *iter,
6983                              struct nft_set_elem *elem)
6984 {
6985         struct nft_trans *trans;
6986         int err;
6987
6988         trans = nft_trans_alloc_gfp(ctx, NFT_MSG_DELSETELEM,
6989                                     sizeof(struct nft_trans_elem), GFP_ATOMIC);
6990         if (!trans)
6991                 return -ENOMEM;
6992
6993         if (!set->ops->flush(ctx->net, set, elem->priv)) {
6994                 err = -ENOENT;
6995                 goto err1;
6996         }
6997         set->ndeact++;
6998
6999         nft_setelem_data_deactivate(ctx->net, set, elem);
7000         nft_trans_elem_set(trans) = set;
7001         nft_trans_elem(trans) = *elem;
7002         nft_trans_commit_list_add_tail(ctx->net, trans);
7003
7004         return 0;
7005 err1:
7006         kfree(trans);
7007         return err;
7008 }
7009
7010 static int __nft_set_catchall_flush(const struct nft_ctx *ctx,
7011                                     struct nft_set *set,
7012                                     struct nft_set_elem *elem)
7013 {
7014         struct nft_trans *trans;
7015
7016         trans = nft_trans_alloc_gfp(ctx, NFT_MSG_DELSETELEM,
7017                                     sizeof(struct nft_trans_elem), GFP_KERNEL);
7018         if (!trans)
7019                 return -ENOMEM;
7020
7021         nft_setelem_data_deactivate(ctx->net, set, elem);
7022         nft_trans_elem_set(trans) = set;
7023         nft_trans_elem(trans) = *elem;
7024         nft_trans_commit_list_add_tail(ctx->net, trans);
7025
7026         return 0;
7027 }
7028
7029 static int nft_set_catchall_flush(const struct nft_ctx *ctx,
7030                                   struct nft_set *set)
7031 {
7032         u8 genmask = nft_genmask_next(ctx->net);
7033         struct nft_set_elem_catchall *catchall;
7034         struct nft_set_elem elem;
7035         struct nft_set_ext *ext;
7036         int ret = 0;
7037
7038         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
7039                 ext = nft_set_elem_ext(set, catchall->elem);
7040                 if (!nft_set_elem_active(ext, genmask) ||
7041                     nft_set_elem_mark_busy(ext))
7042                         continue;
7043
7044                 elem.priv = catchall->elem;
7045                 ret = __nft_set_catchall_flush(ctx, set, &elem);
7046                 if (ret < 0)
7047                         break;
7048         }
7049
7050         return ret;
7051 }
7052
7053 static int nft_set_flush(struct nft_ctx *ctx, struct nft_set *set, u8 genmask)
7054 {
7055         struct nft_set_iter iter = {
7056                 .genmask        = genmask,
7057                 .fn             = nft_setelem_flush,
7058         };
7059
7060         set->ops->walk(ctx, set, &iter);
7061         if (!iter.err)
7062                 iter.err = nft_set_catchall_flush(ctx, set);
7063
7064         return iter.err;
7065 }
7066
7067 static int nf_tables_delsetelem(struct sk_buff *skb,
7068                                 const struct nfnl_info *info,
7069                                 const struct nlattr * const nla[])
7070 {
7071         struct netlink_ext_ack *extack = info->extack;
7072         u8 genmask = nft_genmask_next(info->net);
7073         u8 family = info->nfmsg->nfgen_family;
7074         struct net *net = info->net;
7075         const struct nlattr *attr;
7076         struct nft_table *table;
7077         struct nft_set *set;
7078         struct nft_ctx ctx;
7079         int rem, err = 0;
7080
7081         table = nft_table_lookup(net, nla[NFTA_SET_ELEM_LIST_TABLE], family,
7082                                  genmask, NETLINK_CB(skb).portid);
7083         if (IS_ERR(table)) {
7084                 NL_SET_BAD_ATTR(extack, nla[NFTA_SET_ELEM_LIST_TABLE]);
7085                 return PTR_ERR(table);
7086         }
7087
7088         set = nft_set_lookup(table, nla[NFTA_SET_ELEM_LIST_SET], genmask);
7089         if (IS_ERR(set))
7090                 return PTR_ERR(set);
7091
7092         if (!list_empty(&set->bindings) &&
7093             (set->flags & (NFT_SET_CONSTANT | NFT_SET_ANONYMOUS)))
7094                 return -EBUSY;
7095
7096         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
7097
7098         if (!nla[NFTA_SET_ELEM_LIST_ELEMENTS])
7099                 return nft_set_flush(&ctx, set, genmask);
7100
7101         nla_for_each_nested(attr, nla[NFTA_SET_ELEM_LIST_ELEMENTS], rem) {
7102                 err = nft_del_setelem(&ctx, set, attr);
7103                 if (err == -ENOENT &&
7104                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYSETELEM)
7105                         continue;
7106
7107                 if (err < 0) {
7108                         NL_SET_BAD_ATTR(extack, attr);
7109                         break;
7110                 }
7111         }
7112         return err;
7113 }
7114
7115 void nft_set_gc_batch_release(struct rcu_head *rcu)
7116 {
7117         struct nft_set_gc_batch *gcb;
7118         unsigned int i;
7119
7120         gcb = container_of(rcu, struct nft_set_gc_batch, head.rcu);
7121         for (i = 0; i < gcb->head.cnt; i++)
7122                 nft_set_elem_destroy(gcb->head.set, gcb->elems[i], true);
7123         kfree(gcb);
7124 }
7125
7126 struct nft_set_gc_batch *nft_set_gc_batch_alloc(const struct nft_set *set,
7127                                                 gfp_t gfp)
7128 {
7129         struct nft_set_gc_batch *gcb;
7130
7131         gcb = kzalloc(sizeof(*gcb), gfp);
7132         if (gcb == NULL)
7133                 return gcb;
7134         gcb->head.set = set;
7135         return gcb;
7136 }
7137
7138 /*
7139  * Stateful objects
7140  */
7141
7142 /**
7143  *      nft_register_obj- register nf_tables stateful object type
7144  *      @obj_type: object type
7145  *
7146  *      Registers the object type for use with nf_tables. Returns zero on
7147  *      success or a negative errno code otherwise.
7148  */
7149 int nft_register_obj(struct nft_object_type *obj_type)
7150 {
7151         if (obj_type->type == NFT_OBJECT_UNSPEC)
7152                 return -EINVAL;
7153
7154         nfnl_lock(NFNL_SUBSYS_NFTABLES);
7155         list_add_rcu(&obj_type->list, &nf_tables_objects);
7156         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
7157         return 0;
7158 }
7159 EXPORT_SYMBOL_GPL(nft_register_obj);
7160
7161 /**
7162  *      nft_unregister_obj - unregister nf_tables object type
7163  *      @obj_type: object type
7164  *
7165  *      Unregisters the object type for use with nf_tables.
7166  */
7167 void nft_unregister_obj(struct nft_object_type *obj_type)
7168 {
7169         nfnl_lock(NFNL_SUBSYS_NFTABLES);
7170         list_del_rcu(&obj_type->list);
7171         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
7172 }
7173 EXPORT_SYMBOL_GPL(nft_unregister_obj);
7174
7175 struct nft_object *nft_obj_lookup(const struct net *net,
7176                                   const struct nft_table *table,
7177                                   const struct nlattr *nla, u32 objtype,
7178                                   u8 genmask)
7179 {
7180         struct nft_object_hash_key k = { .table = table };
7181         char search[NFT_OBJ_MAXNAMELEN];
7182         struct rhlist_head *tmp, *list;
7183         struct nft_object *obj;
7184
7185         nla_strscpy(search, nla, sizeof(search));
7186         k.name = search;
7187
7188         WARN_ON_ONCE(!rcu_read_lock_held() &&
7189                      !lockdep_commit_lock_is_held(net));
7190
7191         rcu_read_lock();
7192         list = rhltable_lookup(&nft_objname_ht, &k, nft_objname_ht_params);
7193         if (!list)
7194                 goto out;
7195
7196         rhl_for_each_entry_rcu(obj, tmp, list, rhlhead) {
7197                 if (objtype == obj->ops->type->type &&
7198                     nft_active_genmask(obj, genmask)) {
7199                         rcu_read_unlock();
7200                         return obj;
7201                 }
7202         }
7203 out:
7204         rcu_read_unlock();
7205         return ERR_PTR(-ENOENT);
7206 }
7207 EXPORT_SYMBOL_GPL(nft_obj_lookup);
7208
7209 static struct nft_object *nft_obj_lookup_byhandle(const struct nft_table *table,
7210                                                   const struct nlattr *nla,
7211                                                   u32 objtype, u8 genmask)
7212 {
7213         struct nft_object *obj;
7214
7215         list_for_each_entry(obj, &table->objects, list) {
7216                 if (be64_to_cpu(nla_get_be64(nla)) == obj->handle &&
7217                     objtype == obj->ops->type->type &&
7218                     nft_active_genmask(obj, genmask))
7219                         return obj;
7220         }
7221         return ERR_PTR(-ENOENT);
7222 }
7223
7224 static const struct nla_policy nft_obj_policy[NFTA_OBJ_MAX + 1] = {
7225         [NFTA_OBJ_TABLE]        = { .type = NLA_STRING,
7226                                     .len = NFT_TABLE_MAXNAMELEN - 1 },
7227         [NFTA_OBJ_NAME]         = { .type = NLA_STRING,
7228                                     .len = NFT_OBJ_MAXNAMELEN - 1 },
7229         [NFTA_OBJ_TYPE]         = { .type = NLA_U32 },
7230         [NFTA_OBJ_DATA]         = { .type = NLA_NESTED },
7231         [NFTA_OBJ_HANDLE]       = { .type = NLA_U64},
7232         [NFTA_OBJ_USERDATA]     = { .type = NLA_BINARY,
7233                                     .len = NFT_USERDATA_MAXLEN },
7234 };
7235
7236 static struct nft_object *nft_obj_init(const struct nft_ctx *ctx,
7237                                        const struct nft_object_type *type,
7238                                        const struct nlattr *attr)
7239 {
7240         struct nlattr **tb;
7241         const struct nft_object_ops *ops;
7242         struct nft_object *obj;
7243         int err = -ENOMEM;
7244
7245         tb = kmalloc_array(type->maxattr + 1, sizeof(*tb), GFP_KERNEL);
7246         if (!tb)
7247                 goto err1;
7248
7249         if (attr) {
7250                 err = nla_parse_nested_deprecated(tb, type->maxattr, attr,
7251                                                   type->policy, NULL);
7252                 if (err < 0)
7253                         goto err2;
7254         } else {
7255                 memset(tb, 0, sizeof(tb[0]) * (type->maxattr + 1));
7256         }
7257
7258         if (type->select_ops) {
7259                 ops = type->select_ops(ctx, (const struct nlattr * const *)tb);
7260                 if (IS_ERR(ops)) {
7261                         err = PTR_ERR(ops);
7262                         goto err2;
7263                 }
7264         } else {
7265                 ops = type->ops;
7266         }
7267
7268         err = -ENOMEM;
7269         obj = kzalloc(sizeof(*obj) + ops->size, GFP_KERNEL_ACCOUNT);
7270         if (!obj)
7271                 goto err2;
7272
7273         err = ops->init(ctx, (const struct nlattr * const *)tb, obj);
7274         if (err < 0)
7275                 goto err3;
7276
7277         obj->ops = ops;
7278
7279         kfree(tb);
7280         return obj;
7281 err3:
7282         kfree(obj);
7283 err2:
7284         kfree(tb);
7285 err1:
7286         return ERR_PTR(err);
7287 }
7288
7289 static int nft_object_dump(struct sk_buff *skb, unsigned int attr,
7290                            struct nft_object *obj, bool reset)
7291 {
7292         struct nlattr *nest;
7293
7294         nest = nla_nest_start_noflag(skb, attr);
7295         if (!nest)
7296                 goto nla_put_failure;
7297         if (obj->ops->dump(skb, obj, reset) < 0)
7298                 goto nla_put_failure;
7299         nla_nest_end(skb, nest);
7300         return 0;
7301
7302 nla_put_failure:
7303         return -1;
7304 }
7305
7306 static const struct nft_object_type *__nft_obj_type_get(u32 objtype)
7307 {
7308         const struct nft_object_type *type;
7309
7310         list_for_each_entry(type, &nf_tables_objects, list) {
7311                 if (objtype == type->type)
7312                         return type;
7313         }
7314         return NULL;
7315 }
7316
7317 static const struct nft_object_type *
7318 nft_obj_type_get(struct net *net, u32 objtype)
7319 {
7320         const struct nft_object_type *type;
7321
7322         type = __nft_obj_type_get(objtype);
7323         if (type != NULL && try_module_get(type->owner))
7324                 return type;
7325
7326         lockdep_nfnl_nft_mutex_not_held();
7327 #ifdef CONFIG_MODULES
7328         if (type == NULL) {
7329                 if (nft_request_module(net, "nft-obj-%u", objtype) == -EAGAIN)
7330                         return ERR_PTR(-EAGAIN);
7331         }
7332 #endif
7333         return ERR_PTR(-ENOENT);
7334 }
7335
7336 static int nf_tables_updobj(const struct nft_ctx *ctx,
7337                             const struct nft_object_type *type,
7338                             const struct nlattr *attr,
7339                             struct nft_object *obj)
7340 {
7341         struct nft_object *newobj;
7342         struct nft_trans *trans;
7343         int err = -ENOMEM;
7344
7345         if (!try_module_get(type->owner))
7346                 return -ENOENT;
7347
7348         trans = nft_trans_alloc(ctx, NFT_MSG_NEWOBJ,
7349                                 sizeof(struct nft_trans_obj));
7350         if (!trans)
7351                 goto err_trans;
7352
7353         newobj = nft_obj_init(ctx, type, attr);
7354         if (IS_ERR(newobj)) {
7355                 err = PTR_ERR(newobj);
7356                 goto err_free_trans;
7357         }
7358
7359         nft_trans_obj(trans) = obj;
7360         nft_trans_obj_update(trans) = true;
7361         nft_trans_obj_newobj(trans) = newobj;
7362         nft_trans_commit_list_add_tail(ctx->net, trans);
7363
7364         return 0;
7365
7366 err_free_trans:
7367         kfree(trans);
7368 err_trans:
7369         module_put(type->owner);
7370         return err;
7371 }
7372
7373 static int nf_tables_newobj(struct sk_buff *skb, const struct nfnl_info *info,
7374                             const struct nlattr * const nla[])
7375 {
7376         struct netlink_ext_ack *extack = info->extack;
7377         u8 genmask = nft_genmask_next(info->net);
7378         u8 family = info->nfmsg->nfgen_family;
7379         const struct nft_object_type *type;
7380         struct net *net = info->net;
7381         struct nft_table *table;
7382         struct nft_object *obj;
7383         struct nft_ctx ctx;
7384         u32 objtype;
7385         int err;
7386
7387         if (!nla[NFTA_OBJ_TYPE] ||
7388             !nla[NFTA_OBJ_NAME] ||
7389             !nla[NFTA_OBJ_DATA])
7390                 return -EINVAL;
7391
7392         table = nft_table_lookup(net, nla[NFTA_OBJ_TABLE], family, genmask,
7393                                  NETLINK_CB(skb).portid);
7394         if (IS_ERR(table)) {
7395                 NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_TABLE]);
7396                 return PTR_ERR(table);
7397         }
7398
7399         objtype = ntohl(nla_get_be32(nla[NFTA_OBJ_TYPE]));
7400         obj = nft_obj_lookup(net, table, nla[NFTA_OBJ_NAME], objtype, genmask);
7401         if (IS_ERR(obj)) {
7402                 err = PTR_ERR(obj);
7403                 if (err != -ENOENT) {
7404                         NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_NAME]);
7405                         return err;
7406                 }
7407         } else {
7408                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
7409                         NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_NAME]);
7410                         return -EEXIST;
7411                 }
7412                 if (info->nlh->nlmsg_flags & NLM_F_REPLACE)
7413                         return -EOPNOTSUPP;
7414
7415                 type = __nft_obj_type_get(objtype);
7416                 if (WARN_ON_ONCE(!type))
7417                         return -ENOENT;
7418
7419                 nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
7420
7421                 return nf_tables_updobj(&ctx, type, nla[NFTA_OBJ_DATA], obj);
7422         }
7423
7424         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
7425
7426         type = nft_obj_type_get(net, objtype);
7427         if (IS_ERR(type))
7428                 return PTR_ERR(type);
7429
7430         obj = nft_obj_init(&ctx, type, nla[NFTA_OBJ_DATA]);
7431         if (IS_ERR(obj)) {
7432                 err = PTR_ERR(obj);
7433                 goto err_init;
7434         }
7435         obj->key.table = table;
7436         obj->handle = nf_tables_alloc_handle(table);
7437
7438         obj->key.name = nla_strdup(nla[NFTA_OBJ_NAME], GFP_KERNEL_ACCOUNT);
7439         if (!obj->key.name) {
7440                 err = -ENOMEM;
7441                 goto err_strdup;
7442         }
7443
7444         if (nla[NFTA_OBJ_USERDATA]) {
7445                 obj->udata = nla_memdup(nla[NFTA_OBJ_USERDATA], GFP_KERNEL_ACCOUNT);
7446                 if (obj->udata == NULL)
7447                         goto err_userdata;
7448
7449                 obj->udlen = nla_len(nla[NFTA_OBJ_USERDATA]);
7450         }
7451
7452         err = nft_trans_obj_add(&ctx, NFT_MSG_NEWOBJ, obj);
7453         if (err < 0)
7454                 goto err_trans;
7455
7456         err = rhltable_insert(&nft_objname_ht, &obj->rhlhead,
7457                               nft_objname_ht_params);
7458         if (err < 0)
7459                 goto err_obj_ht;
7460
7461         list_add_tail_rcu(&obj->list, &table->objects);
7462         table->use++;
7463         return 0;
7464 err_obj_ht:
7465         /* queued in transaction log */
7466         INIT_LIST_HEAD(&obj->list);
7467         return err;
7468 err_trans:
7469         kfree(obj->udata);
7470 err_userdata:
7471         kfree(obj->key.name);
7472 err_strdup:
7473         if (obj->ops->destroy)
7474                 obj->ops->destroy(&ctx, obj);
7475         kfree(obj);
7476 err_init:
7477         module_put(type->owner);
7478         return err;
7479 }
7480
7481 static int nf_tables_fill_obj_info(struct sk_buff *skb, struct net *net,
7482                                    u32 portid, u32 seq, int event, u32 flags,
7483                                    int family, const struct nft_table *table,
7484                                    struct nft_object *obj, bool reset)
7485 {
7486         struct nlmsghdr *nlh;
7487
7488         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
7489         nlh = nfnl_msg_put(skb, portid, seq, event, flags, family,
7490                            NFNETLINK_V0, nft_base_seq(net));
7491         if (!nlh)
7492                 goto nla_put_failure;
7493
7494         if (nla_put_string(skb, NFTA_OBJ_TABLE, table->name) ||
7495             nla_put_string(skb, NFTA_OBJ_NAME, obj->key.name) ||
7496             nla_put_be64(skb, NFTA_OBJ_HANDLE, cpu_to_be64(obj->handle),
7497                          NFTA_OBJ_PAD))
7498                 goto nla_put_failure;
7499
7500         if (event == NFT_MSG_DELOBJ) {
7501                 nlmsg_end(skb, nlh);
7502                 return 0;
7503         }
7504
7505         if (nla_put_be32(skb, NFTA_OBJ_TYPE, htonl(obj->ops->type->type)) ||
7506             nla_put_be32(skb, NFTA_OBJ_USE, htonl(obj->use)) ||
7507             nft_object_dump(skb, NFTA_OBJ_DATA, obj, reset))
7508                 goto nla_put_failure;
7509
7510         if (obj->udata &&
7511             nla_put(skb, NFTA_OBJ_USERDATA, obj->udlen, obj->udata))
7512                 goto nla_put_failure;
7513
7514         nlmsg_end(skb, nlh);
7515         return 0;
7516
7517 nla_put_failure:
7518         nlmsg_trim(skb, nlh);
7519         return -1;
7520 }
7521
7522 struct nft_obj_filter {
7523         char            *table;
7524         u32             type;
7525 };
7526
7527 static int nf_tables_dump_obj(struct sk_buff *skb, struct netlink_callback *cb)
7528 {
7529         const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
7530         const struct nft_table *table;
7531         unsigned int idx = 0, s_idx = cb->args[0];
7532         struct nft_obj_filter *filter = cb->data;
7533         struct net *net = sock_net(skb->sk);
7534         int family = nfmsg->nfgen_family;
7535         struct nftables_pernet *nft_net;
7536         struct nft_object *obj;
7537         bool reset = false;
7538
7539         if (NFNL_MSG_TYPE(cb->nlh->nlmsg_type) == NFT_MSG_GETOBJ_RESET)
7540                 reset = true;
7541
7542         rcu_read_lock();
7543         nft_net = nft_pernet(net);
7544         cb->seq = READ_ONCE(nft_net->base_seq);
7545
7546         list_for_each_entry_rcu(table, &nft_net->tables, list) {
7547                 if (family != NFPROTO_UNSPEC && family != table->family)
7548                         continue;
7549
7550                 list_for_each_entry_rcu(obj, &table->objects, list) {
7551                         if (!nft_is_active(net, obj))
7552                                 goto cont;
7553                         if (idx < s_idx)
7554                                 goto cont;
7555                         if (idx > s_idx)
7556                                 memset(&cb->args[1], 0,
7557                                        sizeof(cb->args) - sizeof(cb->args[0]));
7558                         if (filter && filter->table &&
7559                             strcmp(filter->table, table->name))
7560                                 goto cont;
7561                         if (filter &&
7562                             filter->type != NFT_OBJECT_UNSPEC &&
7563                             obj->ops->type->type != filter->type)
7564                                 goto cont;
7565                         if (reset) {
7566                                 char *buf = kasprintf(GFP_ATOMIC,
7567                                                       "%s:%u",
7568                                                       table->name,
7569                                                       nft_net->base_seq);
7570
7571                                 audit_log_nfcfg(buf,
7572                                                 family,
7573                                                 obj->handle,
7574                                                 AUDIT_NFT_OP_OBJ_RESET,
7575                                                 GFP_ATOMIC);
7576                                 kfree(buf);
7577                         }
7578
7579                         if (nf_tables_fill_obj_info(skb, net, NETLINK_CB(cb->skb).portid,
7580                                                     cb->nlh->nlmsg_seq,
7581                                                     NFT_MSG_NEWOBJ,
7582                                                     NLM_F_MULTI | NLM_F_APPEND,
7583                                                     table->family, table,
7584                                                     obj, reset) < 0)
7585                                 goto done;
7586
7587                         nl_dump_check_consistent(cb, nlmsg_hdr(skb));
7588 cont:
7589                         idx++;
7590                 }
7591         }
7592 done:
7593         rcu_read_unlock();
7594
7595         cb->args[0] = idx;
7596         return skb->len;
7597 }
7598
7599 static int nf_tables_dump_obj_start(struct netlink_callback *cb)
7600 {
7601         const struct nlattr * const *nla = cb->data;
7602         struct nft_obj_filter *filter = NULL;
7603
7604         if (nla[NFTA_OBJ_TABLE] || nla[NFTA_OBJ_TYPE]) {
7605                 filter = kzalloc(sizeof(*filter), GFP_ATOMIC);
7606                 if (!filter)
7607                         return -ENOMEM;
7608
7609                 if (nla[NFTA_OBJ_TABLE]) {
7610                         filter->table = nla_strdup(nla[NFTA_OBJ_TABLE], GFP_ATOMIC);
7611                         if (!filter->table) {
7612                                 kfree(filter);
7613                                 return -ENOMEM;
7614                         }
7615                 }
7616
7617                 if (nla[NFTA_OBJ_TYPE])
7618                         filter->type = ntohl(nla_get_be32(nla[NFTA_OBJ_TYPE]));
7619         }
7620
7621         cb->data = filter;
7622         return 0;
7623 }
7624
7625 static int nf_tables_dump_obj_done(struct netlink_callback *cb)
7626 {
7627         struct nft_obj_filter *filter = cb->data;
7628
7629         if (filter) {
7630                 kfree(filter->table);
7631                 kfree(filter);
7632         }
7633
7634         return 0;
7635 }
7636
7637 /* called with rcu_read_lock held */
7638 static int nf_tables_getobj(struct sk_buff *skb, const struct nfnl_info *info,
7639                             const struct nlattr * const nla[])
7640 {
7641         struct netlink_ext_ack *extack = info->extack;
7642         u8 genmask = nft_genmask_cur(info->net);
7643         u8 family = info->nfmsg->nfgen_family;
7644         const struct nft_table *table;
7645         struct net *net = info->net;
7646         struct nft_object *obj;
7647         struct sk_buff *skb2;
7648         bool reset = false;
7649         u32 objtype;
7650         int err;
7651
7652         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
7653                 struct netlink_dump_control c = {
7654                         .start = nf_tables_dump_obj_start,
7655                         .dump = nf_tables_dump_obj,
7656                         .done = nf_tables_dump_obj_done,
7657                         .module = THIS_MODULE,
7658                         .data = (void *)nla,
7659                 };
7660
7661                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
7662         }
7663
7664         if (!nla[NFTA_OBJ_NAME] ||
7665             !nla[NFTA_OBJ_TYPE])
7666                 return -EINVAL;
7667
7668         table = nft_table_lookup(net, nla[NFTA_OBJ_TABLE], family, genmask, 0);
7669         if (IS_ERR(table)) {
7670                 NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_TABLE]);
7671                 return PTR_ERR(table);
7672         }
7673
7674         objtype = ntohl(nla_get_be32(nla[NFTA_OBJ_TYPE]));
7675         obj = nft_obj_lookup(net, table, nla[NFTA_OBJ_NAME], objtype, genmask);
7676         if (IS_ERR(obj)) {
7677                 NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_NAME]);
7678                 return PTR_ERR(obj);
7679         }
7680
7681         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
7682         if (!skb2)
7683                 return -ENOMEM;
7684
7685         if (NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_GETOBJ_RESET)
7686                 reset = true;
7687
7688         if (reset) {
7689                 const struct nftables_pernet *nft_net;
7690                 char *buf;
7691
7692                 nft_net = nft_pernet(net);
7693                 buf = kasprintf(GFP_ATOMIC, "%s:%u", table->name, nft_net->base_seq);
7694
7695                 audit_log_nfcfg(buf,
7696                                 family,
7697                                 obj->handle,
7698                                 AUDIT_NFT_OP_OBJ_RESET,
7699                                 GFP_ATOMIC);
7700                 kfree(buf);
7701         }
7702
7703         err = nf_tables_fill_obj_info(skb2, net, NETLINK_CB(skb).portid,
7704                                       info->nlh->nlmsg_seq, NFT_MSG_NEWOBJ, 0,
7705                                       family, table, obj, reset);
7706         if (err < 0)
7707                 goto err_fill_obj_info;
7708
7709         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
7710
7711 err_fill_obj_info:
7712         kfree_skb(skb2);
7713         return err;
7714 }
7715
7716 static void nft_obj_destroy(const struct nft_ctx *ctx, struct nft_object *obj)
7717 {
7718         if (obj->ops->destroy)
7719                 obj->ops->destroy(ctx, obj);
7720
7721         module_put(obj->ops->type->owner);
7722         kfree(obj->key.name);
7723         kfree(obj->udata);
7724         kfree(obj);
7725 }
7726
7727 static int nf_tables_delobj(struct sk_buff *skb, const struct nfnl_info *info,
7728                             const struct nlattr * const nla[])
7729 {
7730         struct netlink_ext_ack *extack = info->extack;
7731         u8 genmask = nft_genmask_next(info->net);
7732         u8 family = info->nfmsg->nfgen_family;
7733         struct net *net = info->net;
7734         const struct nlattr *attr;
7735         struct nft_table *table;
7736         struct nft_object *obj;
7737         struct nft_ctx ctx;
7738         u32 objtype;
7739
7740         if (!nla[NFTA_OBJ_TYPE] ||
7741             (!nla[NFTA_OBJ_NAME] && !nla[NFTA_OBJ_HANDLE]))
7742                 return -EINVAL;
7743
7744         table = nft_table_lookup(net, nla[NFTA_OBJ_TABLE], family, genmask,
7745                                  NETLINK_CB(skb).portid);
7746         if (IS_ERR(table)) {
7747                 NL_SET_BAD_ATTR(extack, nla[NFTA_OBJ_TABLE]);
7748                 return PTR_ERR(table);
7749         }
7750
7751         objtype = ntohl(nla_get_be32(nla[NFTA_OBJ_TYPE]));
7752         if (nla[NFTA_OBJ_HANDLE]) {
7753                 attr = nla[NFTA_OBJ_HANDLE];
7754                 obj = nft_obj_lookup_byhandle(table, attr, objtype, genmask);
7755         } else {
7756                 attr = nla[NFTA_OBJ_NAME];
7757                 obj = nft_obj_lookup(net, table, attr, objtype, genmask);
7758         }
7759
7760         if (IS_ERR(obj)) {
7761                 if (PTR_ERR(obj) == -ENOENT &&
7762                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYOBJ)
7763                         return 0;
7764
7765                 NL_SET_BAD_ATTR(extack, attr);
7766                 return PTR_ERR(obj);
7767         }
7768         if (obj->use > 0) {
7769                 NL_SET_BAD_ATTR(extack, attr);
7770                 return -EBUSY;
7771         }
7772
7773         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
7774
7775         return nft_delobj(&ctx, obj);
7776 }
7777
7778 void nft_obj_notify(struct net *net, const struct nft_table *table,
7779                     struct nft_object *obj, u32 portid, u32 seq, int event,
7780                     u16 flags, int family, int report, gfp_t gfp)
7781 {
7782         struct nftables_pernet *nft_net = nft_pernet(net);
7783         struct sk_buff *skb;
7784         int err;
7785         char *buf = kasprintf(gfp, "%s:%u",
7786                               table->name, nft_net->base_seq);
7787
7788         audit_log_nfcfg(buf,
7789                         family,
7790                         obj->handle,
7791                         event == NFT_MSG_NEWOBJ ?
7792                                  AUDIT_NFT_OP_OBJ_REGISTER :
7793                                  AUDIT_NFT_OP_OBJ_UNREGISTER,
7794                         gfp);
7795         kfree(buf);
7796
7797         if (!report &&
7798             !nfnetlink_has_listeners(net, NFNLGRP_NFTABLES))
7799                 return;
7800
7801         skb = nlmsg_new(NLMSG_GOODSIZE, gfp);
7802         if (skb == NULL)
7803                 goto err;
7804
7805         err = nf_tables_fill_obj_info(skb, net, portid, seq, event,
7806                                       flags & (NLM_F_CREATE | NLM_F_EXCL),
7807                                       family, table, obj, false);
7808         if (err < 0) {
7809                 kfree_skb(skb);
7810                 goto err;
7811         }
7812
7813         nft_notify_enqueue(skb, report, &nft_net->notify_list);
7814         return;
7815 err:
7816         nfnetlink_set_err(net, portid, NFNLGRP_NFTABLES, -ENOBUFS);
7817 }
7818 EXPORT_SYMBOL_GPL(nft_obj_notify);
7819
7820 static void nf_tables_obj_notify(const struct nft_ctx *ctx,
7821                                  struct nft_object *obj, int event)
7822 {
7823         nft_obj_notify(ctx->net, ctx->table, obj, ctx->portid, ctx->seq, event,
7824                        ctx->flags, ctx->family, ctx->report, GFP_KERNEL);
7825 }
7826
7827 /*
7828  * Flow tables
7829  */
7830 void nft_register_flowtable_type(struct nf_flowtable_type *type)
7831 {
7832         nfnl_lock(NFNL_SUBSYS_NFTABLES);
7833         list_add_tail_rcu(&type->list, &nf_tables_flowtables);
7834         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
7835 }
7836 EXPORT_SYMBOL_GPL(nft_register_flowtable_type);
7837
7838 void nft_unregister_flowtable_type(struct nf_flowtable_type *type)
7839 {
7840         nfnl_lock(NFNL_SUBSYS_NFTABLES);
7841         list_del_rcu(&type->list);
7842         nfnl_unlock(NFNL_SUBSYS_NFTABLES);
7843 }
7844 EXPORT_SYMBOL_GPL(nft_unregister_flowtable_type);
7845
7846 static const struct nla_policy nft_flowtable_policy[NFTA_FLOWTABLE_MAX + 1] = {
7847         [NFTA_FLOWTABLE_TABLE]          = { .type = NLA_STRING,
7848                                             .len = NFT_NAME_MAXLEN - 1 },
7849         [NFTA_FLOWTABLE_NAME]           = { .type = NLA_STRING,
7850                                             .len = NFT_NAME_MAXLEN - 1 },
7851         [NFTA_FLOWTABLE_HOOK]           = { .type = NLA_NESTED },
7852         [NFTA_FLOWTABLE_HANDLE]         = { .type = NLA_U64 },
7853         [NFTA_FLOWTABLE_FLAGS]          = { .type = NLA_U32 },
7854 };
7855
7856 struct nft_flowtable *nft_flowtable_lookup(const struct nft_table *table,
7857                                            const struct nlattr *nla, u8 genmask)
7858 {
7859         struct nft_flowtable *flowtable;
7860
7861         list_for_each_entry_rcu(flowtable, &table->flowtables, list) {
7862                 if (!nla_strcmp(nla, flowtable->name) &&
7863                     nft_active_genmask(flowtable, genmask))
7864                         return flowtable;
7865         }
7866         return ERR_PTR(-ENOENT);
7867 }
7868 EXPORT_SYMBOL_GPL(nft_flowtable_lookup);
7869
7870 void nf_tables_deactivate_flowtable(const struct nft_ctx *ctx,
7871                                     struct nft_flowtable *flowtable,
7872                                     enum nft_trans_phase phase)
7873 {
7874         switch (phase) {
7875         case NFT_TRANS_PREPARE_ERROR:
7876         case NFT_TRANS_PREPARE:
7877         case NFT_TRANS_ABORT:
7878         case NFT_TRANS_RELEASE:
7879                 flowtable->use--;
7880                 fallthrough;
7881         default:
7882                 return;
7883         }
7884 }
7885 EXPORT_SYMBOL_GPL(nf_tables_deactivate_flowtable);
7886
7887 static struct nft_flowtable *
7888 nft_flowtable_lookup_byhandle(const struct nft_table *table,
7889                               const struct nlattr *nla, u8 genmask)
7890 {
7891        struct nft_flowtable *flowtable;
7892
7893        list_for_each_entry(flowtable, &table->flowtables, list) {
7894                if (be64_to_cpu(nla_get_be64(nla)) == flowtable->handle &&
7895                    nft_active_genmask(flowtable, genmask))
7896                        return flowtable;
7897        }
7898        return ERR_PTR(-ENOENT);
7899 }
7900
7901 struct nft_flowtable_hook {
7902         u32                     num;
7903         int                     priority;
7904         struct list_head        list;
7905 };
7906
7907 static const struct nla_policy nft_flowtable_hook_policy[NFTA_FLOWTABLE_HOOK_MAX + 1] = {
7908         [NFTA_FLOWTABLE_HOOK_NUM]       = { .type = NLA_U32 },
7909         [NFTA_FLOWTABLE_HOOK_PRIORITY]  = { .type = NLA_U32 },
7910         [NFTA_FLOWTABLE_HOOK_DEVS]      = { .type = NLA_NESTED },
7911 };
7912
7913 static int nft_flowtable_parse_hook(const struct nft_ctx *ctx,
7914                                     const struct nlattr * const nla[],
7915                                     struct nft_flowtable_hook *flowtable_hook,
7916                                     struct nft_flowtable *flowtable,
7917                                     struct netlink_ext_ack *extack, bool add)
7918 {
7919         struct nlattr *tb[NFTA_FLOWTABLE_HOOK_MAX + 1];
7920         struct nft_hook *hook;
7921         int hooknum, priority;
7922         int err;
7923
7924         INIT_LIST_HEAD(&flowtable_hook->list);
7925
7926         err = nla_parse_nested_deprecated(tb, NFTA_FLOWTABLE_HOOK_MAX,
7927                                           nla[NFTA_FLOWTABLE_HOOK],
7928                                           nft_flowtable_hook_policy, NULL);
7929         if (err < 0)
7930                 return err;
7931
7932         if (add) {
7933                 if (!tb[NFTA_FLOWTABLE_HOOK_NUM] ||
7934                     !tb[NFTA_FLOWTABLE_HOOK_PRIORITY]) {
7935                         NL_SET_BAD_ATTR(extack, nla[NFTA_FLOWTABLE_NAME]);
7936                         return -ENOENT;
7937                 }
7938
7939                 hooknum = ntohl(nla_get_be32(tb[NFTA_FLOWTABLE_HOOK_NUM]));
7940                 if (hooknum != NF_NETDEV_INGRESS)
7941                         return -EOPNOTSUPP;
7942
7943                 priority = ntohl(nla_get_be32(tb[NFTA_FLOWTABLE_HOOK_PRIORITY]));
7944
7945                 flowtable_hook->priority        = priority;
7946                 flowtable_hook->num             = hooknum;
7947         } else {
7948                 if (tb[NFTA_FLOWTABLE_HOOK_NUM]) {
7949                         hooknum = ntohl(nla_get_be32(tb[NFTA_FLOWTABLE_HOOK_NUM]));
7950                         if (hooknum != flowtable->hooknum)
7951                                 return -EOPNOTSUPP;
7952                 }
7953
7954                 if (tb[NFTA_FLOWTABLE_HOOK_PRIORITY]) {
7955                         priority = ntohl(nla_get_be32(tb[NFTA_FLOWTABLE_HOOK_PRIORITY]));
7956                         if (priority != flowtable->data.priority)
7957                                 return -EOPNOTSUPP;
7958                 }
7959
7960                 flowtable_hook->priority        = flowtable->data.priority;
7961                 flowtable_hook->num             = flowtable->hooknum;
7962         }
7963
7964         if (tb[NFTA_FLOWTABLE_HOOK_DEVS]) {
7965                 err = nf_tables_parse_netdev_hooks(ctx->net,
7966                                                    tb[NFTA_FLOWTABLE_HOOK_DEVS],
7967                                                    &flowtable_hook->list,
7968                                                    extack);
7969                 if (err < 0)
7970                         return err;
7971         }
7972
7973         list_for_each_entry(hook, &flowtable_hook->list, list) {
7974                 hook->ops.pf            = NFPROTO_NETDEV;
7975                 hook->ops.hooknum       = flowtable_hook->num;
7976                 hook->ops.priority      = flowtable_hook->priority;
7977                 hook->ops.priv          = &flowtable->data;
7978                 hook->ops.hook          = flowtable->data.type->hook;
7979         }
7980
7981         return err;
7982 }
7983
7984 static const struct nf_flowtable_type *__nft_flowtable_type_get(u8 family)
7985 {
7986         const struct nf_flowtable_type *type;
7987
7988         list_for_each_entry(type, &nf_tables_flowtables, list) {
7989                 if (family == type->family)
7990                         return type;
7991         }
7992         return NULL;
7993 }
7994
7995 static const struct nf_flowtable_type *
7996 nft_flowtable_type_get(struct net *net, u8 family)
7997 {
7998         const struct nf_flowtable_type *type;
7999
8000         type = __nft_flowtable_type_get(family);
8001         if (type != NULL && try_module_get(type->owner))
8002                 return type;
8003
8004         lockdep_nfnl_nft_mutex_not_held();
8005 #ifdef CONFIG_MODULES
8006         if (type == NULL) {
8007                 if (nft_request_module(net, "nf-flowtable-%u", family) == -EAGAIN)
8008                         return ERR_PTR(-EAGAIN);
8009         }
8010 #endif
8011         return ERR_PTR(-ENOENT);
8012 }
8013
8014 /* Only called from error and netdev event paths. */
8015 static void nft_unregister_flowtable_hook(struct net *net,
8016                                           struct nft_flowtable *flowtable,
8017                                           struct nft_hook *hook)
8018 {
8019         nf_unregister_net_hook(net, &hook->ops);
8020         flowtable->data.type->setup(&flowtable->data, hook->ops.dev,
8021                                     FLOW_BLOCK_UNBIND);
8022 }
8023
8024 static void __nft_unregister_flowtable_net_hooks(struct net *net,
8025                                                  struct list_head *hook_list,
8026                                                  bool release_netdev)
8027 {
8028         struct nft_hook *hook, *next;
8029
8030         list_for_each_entry_safe(hook, next, hook_list, list) {
8031                 nf_unregister_net_hook(net, &hook->ops);
8032                 if (release_netdev) {
8033                         list_del(&hook->list);
8034                         kfree_rcu(hook, rcu);
8035                 }
8036         }
8037 }
8038
8039 static void nft_unregister_flowtable_net_hooks(struct net *net,
8040                                                struct list_head *hook_list)
8041 {
8042         __nft_unregister_flowtable_net_hooks(net, hook_list, false);
8043 }
8044
8045 static int nft_register_flowtable_net_hooks(struct net *net,
8046                                             struct nft_table *table,
8047                                             struct list_head *hook_list,
8048                                             struct nft_flowtable *flowtable)
8049 {
8050         struct nft_hook *hook, *hook2, *next;
8051         struct nft_flowtable *ft;
8052         int err, i = 0;
8053
8054         list_for_each_entry(hook, hook_list, list) {
8055                 list_for_each_entry(ft, &table->flowtables, list) {
8056                         if (!nft_is_active_next(net, ft))
8057                                 continue;
8058
8059                         list_for_each_entry(hook2, &ft->hook_list, list) {
8060                                 if (hook->ops.dev == hook2->ops.dev &&
8061                                     hook->ops.pf == hook2->ops.pf) {
8062                                         err = -EEXIST;
8063                                         goto err_unregister_net_hooks;
8064                                 }
8065                         }
8066                 }
8067
8068                 err = flowtable->data.type->setup(&flowtable->data,
8069                                                   hook->ops.dev,
8070                                                   FLOW_BLOCK_BIND);
8071                 if (err < 0)
8072                         goto err_unregister_net_hooks;
8073
8074                 err = nf_register_net_hook(net, &hook->ops);
8075                 if (err < 0) {
8076                         flowtable->data.type->setup(&flowtable->data,
8077                                                     hook->ops.dev,
8078                                                     FLOW_BLOCK_UNBIND);
8079                         goto err_unregister_net_hooks;
8080                 }
8081
8082                 i++;
8083         }
8084
8085         return 0;
8086
8087 err_unregister_net_hooks:
8088         list_for_each_entry_safe(hook, next, hook_list, list) {
8089                 if (i-- <= 0)
8090                         break;
8091
8092                 nft_unregister_flowtable_hook(net, flowtable, hook);
8093                 list_del_rcu(&hook->list);
8094                 kfree_rcu(hook, rcu);
8095         }
8096
8097         return err;
8098 }
8099
8100 static void nft_hooks_destroy(struct list_head *hook_list)
8101 {
8102         struct nft_hook *hook, *next;
8103
8104         list_for_each_entry_safe(hook, next, hook_list, list) {
8105                 list_del_rcu(&hook->list);
8106                 kfree_rcu(hook, rcu);
8107         }
8108 }
8109
8110 static int nft_flowtable_update(struct nft_ctx *ctx, const struct nlmsghdr *nlh,
8111                                 struct nft_flowtable *flowtable,
8112                                 struct netlink_ext_ack *extack)
8113 {
8114         const struct nlattr * const *nla = ctx->nla;
8115         struct nft_flowtable_hook flowtable_hook;
8116         struct nft_hook *hook, *next;
8117         struct nft_trans *trans;
8118         bool unregister = false;
8119         u32 flags;
8120         int err;
8121
8122         err = nft_flowtable_parse_hook(ctx, nla, &flowtable_hook, flowtable,
8123                                        extack, false);
8124         if (err < 0)
8125                 return err;
8126
8127         list_for_each_entry_safe(hook, next, &flowtable_hook.list, list) {
8128                 if (nft_hook_list_find(&flowtable->hook_list, hook)) {
8129                         list_del(&hook->list);
8130                         kfree(hook);
8131                 }
8132         }
8133
8134         if (nla[NFTA_FLOWTABLE_FLAGS]) {
8135                 flags = ntohl(nla_get_be32(nla[NFTA_FLOWTABLE_FLAGS]));
8136                 if (flags & ~NFT_FLOWTABLE_MASK) {
8137                         err = -EOPNOTSUPP;
8138                         goto err_flowtable_update_hook;
8139                 }
8140                 if ((flowtable->data.flags & NFT_FLOWTABLE_HW_OFFLOAD) ^
8141                     (flags & NFT_FLOWTABLE_HW_OFFLOAD)) {
8142                         err = -EOPNOTSUPP;
8143                         goto err_flowtable_update_hook;
8144                 }
8145         } else {
8146                 flags = flowtable->data.flags;
8147         }
8148
8149         err = nft_register_flowtable_net_hooks(ctx->net, ctx->table,
8150                                                &flowtable_hook.list, flowtable);
8151         if (err < 0)
8152                 goto err_flowtable_update_hook;
8153
8154         trans = nft_trans_alloc(ctx, NFT_MSG_NEWFLOWTABLE,
8155                                 sizeof(struct nft_trans_flowtable));
8156         if (!trans) {
8157                 unregister = true;
8158                 err = -ENOMEM;
8159                 goto err_flowtable_update_hook;
8160         }
8161
8162         nft_trans_flowtable_flags(trans) = flags;
8163         nft_trans_flowtable(trans) = flowtable;
8164         nft_trans_flowtable_update(trans) = true;
8165         INIT_LIST_HEAD(&nft_trans_flowtable_hooks(trans));
8166         list_splice(&flowtable_hook.list, &nft_trans_flowtable_hooks(trans));
8167
8168         nft_trans_commit_list_add_tail(ctx->net, trans);
8169
8170         return 0;
8171
8172 err_flowtable_update_hook:
8173         list_for_each_entry_safe(hook, next, &flowtable_hook.list, list) {
8174                 if (unregister)
8175                         nft_unregister_flowtable_hook(ctx->net, flowtable, hook);
8176                 list_del_rcu(&hook->list);
8177                 kfree_rcu(hook, rcu);
8178         }
8179
8180         return err;
8181
8182 }
8183
8184 static int nf_tables_newflowtable(struct sk_buff *skb,
8185                                   const struct nfnl_info *info,
8186                                   const struct nlattr * const nla[])
8187 {
8188         struct netlink_ext_ack *extack = info->extack;
8189         struct nft_flowtable_hook flowtable_hook;
8190         u8 genmask = nft_genmask_next(info->net);
8191         u8 family = info->nfmsg->nfgen_family;
8192         const struct nf_flowtable_type *type;
8193         struct nft_flowtable *flowtable;
8194         struct nft_hook *hook, *next;
8195         struct net *net = info->net;
8196         struct nft_table *table;
8197         struct nft_ctx ctx;
8198         int err;
8199
8200         if (!nla[NFTA_FLOWTABLE_TABLE] ||
8201             !nla[NFTA_FLOWTABLE_NAME] ||
8202             !nla[NFTA_FLOWTABLE_HOOK])
8203                 return -EINVAL;
8204
8205         table = nft_table_lookup(net, nla[NFTA_FLOWTABLE_TABLE], family,
8206                                  genmask, NETLINK_CB(skb).portid);
8207         if (IS_ERR(table)) {
8208                 NL_SET_BAD_ATTR(extack, nla[NFTA_FLOWTABLE_TABLE]);
8209                 return PTR_ERR(table);
8210         }
8211
8212         flowtable = nft_flowtable_lookup(table, nla[NFTA_FLOWTABLE_NAME],
8213                                          genmask);
8214         if (IS_ERR(flowtable)) {
8215                 err = PTR_ERR(flowtable);
8216                 if (err != -ENOENT) {
8217                         NL_SET_BAD_ATTR(extack, nla[NFTA_FLOWTABLE_NAME]);
8218                         return err;
8219                 }
8220         } else {
8221                 if (info->nlh->nlmsg_flags & NLM_F_EXCL) {
8222                         NL_SET_BAD_ATTR(extack, nla[NFTA_FLOWTABLE_NAME]);
8223                         return -EEXIST;
8224                 }
8225
8226                 nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
8227
8228                 return nft_flowtable_update(&ctx, info->nlh, flowtable, extack);
8229         }
8230
8231         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
8232
8233         flowtable = kzalloc(sizeof(*flowtable), GFP_KERNEL_ACCOUNT);
8234         if (!flowtable)
8235                 return -ENOMEM;
8236
8237         flowtable->table = table;
8238         flowtable->handle = nf_tables_alloc_handle(table);
8239         INIT_LIST_HEAD(&flowtable->hook_list);
8240
8241         flowtable->name = nla_strdup(nla[NFTA_FLOWTABLE_NAME], GFP_KERNEL_ACCOUNT);
8242         if (!flowtable->name) {
8243                 err = -ENOMEM;
8244                 goto err1;
8245         }
8246
8247         type = nft_flowtable_type_get(net, family);
8248         if (IS_ERR(type)) {
8249                 err = PTR_ERR(type);
8250                 goto err2;
8251         }
8252
8253         if (nla[NFTA_FLOWTABLE_FLAGS]) {
8254                 flowtable->data.flags =
8255                         ntohl(nla_get_be32(nla[NFTA_FLOWTABLE_FLAGS]));
8256                 if (flowtable->data.flags & ~NFT_FLOWTABLE_MASK) {
8257                         err = -EOPNOTSUPP;
8258                         goto err3;
8259                 }
8260         }
8261
8262         write_pnet(&flowtable->data.net, net);
8263         flowtable->data.type = type;
8264         err = type->init(&flowtable->data);
8265         if (err < 0)
8266                 goto err3;
8267
8268         err = nft_flowtable_parse_hook(&ctx, nla, &flowtable_hook, flowtable,
8269                                        extack, true);
8270         if (err < 0)
8271                 goto err4;
8272
8273         list_splice(&flowtable_hook.list, &flowtable->hook_list);
8274         flowtable->data.priority = flowtable_hook.priority;
8275         flowtable->hooknum = flowtable_hook.num;
8276
8277         err = nft_register_flowtable_net_hooks(ctx.net, table,
8278                                                &flowtable->hook_list,
8279                                                flowtable);
8280         if (err < 0) {
8281                 nft_hooks_destroy(&flowtable->hook_list);
8282                 goto err4;
8283         }
8284
8285         err = nft_trans_flowtable_add(&ctx, NFT_MSG_NEWFLOWTABLE, flowtable);
8286         if (err < 0)
8287                 goto err5;
8288
8289         list_add_tail_rcu(&flowtable->list, &table->flowtables);
8290         table->use++;
8291
8292         return 0;
8293 err5:
8294         list_for_each_entry_safe(hook, next, &flowtable->hook_list, list) {
8295                 nft_unregister_flowtable_hook(net, flowtable, hook);
8296                 list_del_rcu(&hook->list);
8297                 kfree_rcu(hook, rcu);
8298         }
8299 err4:
8300         flowtable->data.type->free(&flowtable->data);
8301 err3:
8302         module_put(type->owner);
8303 err2:
8304         kfree(flowtable->name);
8305 err1:
8306         kfree(flowtable);
8307         return err;
8308 }
8309
8310 static void nft_flowtable_hook_release(struct nft_flowtable_hook *flowtable_hook)
8311 {
8312         struct nft_hook *this, *next;
8313
8314         list_for_each_entry_safe(this, next, &flowtable_hook->list, list) {
8315                 list_del(&this->list);
8316                 kfree(this);
8317         }
8318 }
8319
8320 static int nft_delflowtable_hook(struct nft_ctx *ctx,
8321                                  struct nft_flowtable *flowtable,
8322                                  struct netlink_ext_ack *extack)
8323 {
8324         const struct nlattr * const *nla = ctx->nla;
8325         struct nft_flowtable_hook flowtable_hook;
8326         LIST_HEAD(flowtable_del_list);
8327         struct nft_hook *this, *hook;
8328         struct nft_trans *trans;
8329         int err;
8330
8331         err = nft_flowtable_parse_hook(ctx, nla, &flowtable_hook, flowtable,
8332                                        extack, false);
8333         if (err < 0)
8334                 return err;
8335
8336         list_for_each_entry(this, &flowtable_hook.list, list) {
8337                 hook = nft_hook_list_find(&flowtable->hook_list, this);
8338                 if (!hook) {
8339                         err = -ENOENT;
8340                         goto err_flowtable_del_hook;
8341                 }
8342                 list_move(&hook->list, &flowtable_del_list);
8343         }
8344
8345         trans = nft_trans_alloc(ctx, NFT_MSG_DELFLOWTABLE,
8346                                 sizeof(struct nft_trans_flowtable));
8347         if (!trans) {
8348                 err = -ENOMEM;
8349                 goto err_flowtable_del_hook;
8350         }
8351
8352         nft_trans_flowtable(trans) = flowtable;
8353         nft_trans_flowtable_update(trans) = true;
8354         INIT_LIST_HEAD(&nft_trans_flowtable_hooks(trans));
8355         list_splice(&flowtable_del_list, &nft_trans_flowtable_hooks(trans));
8356         nft_flowtable_hook_release(&flowtable_hook);
8357
8358         nft_trans_commit_list_add_tail(ctx->net, trans);
8359
8360         return 0;
8361
8362 err_flowtable_del_hook:
8363         list_splice(&flowtable_del_list, &flowtable->hook_list);
8364         nft_flowtable_hook_release(&flowtable_hook);
8365
8366         return err;
8367 }
8368
8369 static int nf_tables_delflowtable(struct sk_buff *skb,
8370                                   const struct nfnl_info *info,
8371                                   const struct nlattr * const nla[])
8372 {
8373         struct netlink_ext_ack *extack = info->extack;
8374         u8 genmask = nft_genmask_next(info->net);
8375         u8 family = info->nfmsg->nfgen_family;
8376         struct nft_flowtable *flowtable;
8377         struct net *net = info->net;
8378         const struct nlattr *attr;
8379         struct nft_table *table;
8380         struct nft_ctx ctx;
8381
8382         if (!nla[NFTA_FLOWTABLE_TABLE] ||
8383             (!nla[NFTA_FLOWTABLE_NAME] &&
8384              !nla[NFTA_FLOWTABLE_HANDLE]))
8385                 return -EINVAL;
8386
8387         table = nft_table_lookup(net, nla[NFTA_FLOWTABLE_TABLE], family,
8388                                  genmask, NETLINK_CB(skb).portid);
8389         if (IS_ERR(table)) {
8390                 NL_SET_BAD_ATTR(extack, nla[NFTA_FLOWTABLE_TABLE]);
8391                 return PTR_ERR(table);
8392         }
8393
8394         if (nla[NFTA_FLOWTABLE_HANDLE]) {
8395                 attr = nla[NFTA_FLOWTABLE_HANDLE];
8396                 flowtable = nft_flowtable_lookup_byhandle(table, attr, genmask);
8397         } else {
8398                 attr = nla[NFTA_FLOWTABLE_NAME];
8399                 flowtable = nft_flowtable_lookup(table, attr, genmask);
8400         }
8401
8402         if (IS_ERR(flowtable)) {
8403                 if (PTR_ERR(flowtable) == -ENOENT &&
8404                     NFNL_MSG_TYPE(info->nlh->nlmsg_type) == NFT_MSG_DESTROYFLOWTABLE)
8405                         return 0;
8406
8407                 NL_SET_BAD_ATTR(extack, attr);
8408                 return PTR_ERR(flowtable);
8409         }
8410
8411         nft_ctx_init(&ctx, net, skb, info->nlh, family, table, NULL, nla);
8412
8413         if (nla[NFTA_FLOWTABLE_HOOK])
8414                 return nft_delflowtable_hook(&ctx, flowtable, extack);
8415
8416         if (flowtable->use > 0) {
8417                 NL_SET_BAD_ATTR(extack, attr);
8418                 return -EBUSY;
8419         }
8420
8421         return nft_delflowtable(&ctx, flowtable);
8422 }
8423
8424 static int nf_tables_fill_flowtable_info(struct sk_buff *skb, struct net *net,
8425                                          u32 portid, u32 seq, int event,
8426                                          u32 flags, int family,
8427                                          struct nft_flowtable *flowtable,
8428                                          struct list_head *hook_list)
8429 {
8430         struct nlattr *nest, *nest_devs;
8431         struct nft_hook *hook;
8432         struct nlmsghdr *nlh;
8433
8434         event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, event);
8435         nlh = nfnl_msg_put(skb, portid, seq, event, flags, family,
8436                            NFNETLINK_V0, nft_base_seq(net));
8437         if (!nlh)
8438                 goto nla_put_failure;
8439
8440         if (nla_put_string(skb, NFTA_FLOWTABLE_TABLE, flowtable->table->name) ||
8441             nla_put_string(skb, NFTA_FLOWTABLE_NAME, flowtable->name) ||
8442             nla_put_be64(skb, NFTA_FLOWTABLE_HANDLE, cpu_to_be64(flowtable->handle),
8443                          NFTA_FLOWTABLE_PAD))
8444                 goto nla_put_failure;
8445
8446         if (event == NFT_MSG_DELFLOWTABLE && !hook_list) {
8447                 nlmsg_end(skb, nlh);
8448                 return 0;
8449         }
8450
8451         if (nla_put_be32(skb, NFTA_FLOWTABLE_USE, htonl(flowtable->use)) ||
8452             nla_put_be32(skb, NFTA_FLOWTABLE_FLAGS, htonl(flowtable->data.flags)))
8453                 goto nla_put_failure;
8454
8455         nest = nla_nest_start_noflag(skb, NFTA_FLOWTABLE_HOOK);
8456         if (!nest)
8457                 goto nla_put_failure;
8458         if (nla_put_be32(skb, NFTA_FLOWTABLE_HOOK_NUM, htonl(flowtable->hooknum)) ||
8459             nla_put_be32(skb, NFTA_FLOWTABLE_HOOK_PRIORITY, htonl(flowtable->data.priority)))
8460                 goto nla_put_failure;
8461
8462         nest_devs = nla_nest_start_noflag(skb, NFTA_FLOWTABLE_HOOK_DEVS);
8463         if (!nest_devs)
8464                 goto nla_put_failure;
8465
8466         if (!hook_list)
8467                 hook_list = &flowtable->hook_list;
8468
8469         list_for_each_entry_rcu(hook, hook_list, list) {
8470                 if (nla_put_string(skb, NFTA_DEVICE_NAME, hook->ops.dev->name))
8471                         goto nla_put_failure;
8472         }
8473         nla_nest_end(skb, nest_devs);
8474         nla_nest_end(skb, nest);
8475
8476         nlmsg_end(skb, nlh);
8477         return 0;
8478
8479 nla_put_failure:
8480         nlmsg_trim(skb, nlh);
8481         return -1;
8482 }
8483
8484 struct nft_flowtable_filter {
8485         char            *table;
8486 };
8487
8488 static int nf_tables_dump_flowtable(struct sk_buff *skb,
8489                                     struct netlink_callback *cb)
8490 {
8491         const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
8492         struct nft_flowtable_filter *filter = cb->data;
8493         unsigned int idx = 0, s_idx = cb->args[0];
8494         struct net *net = sock_net(skb->sk);
8495         int family = nfmsg->nfgen_family;
8496         struct nft_flowtable *flowtable;
8497         struct nftables_pernet *nft_net;
8498         const struct nft_table *table;
8499
8500         rcu_read_lock();
8501         nft_net = nft_pernet(net);
8502         cb->seq = READ_ONCE(nft_net->base_seq);
8503
8504         list_for_each_entry_rcu(table, &nft_net->tables, list) {
8505                 if (family != NFPROTO_UNSPEC && family != table->family)
8506                         continue;
8507
8508                 list_for_each_entry_rcu(flowtable, &table->flowtables, list) {
8509                         if (!nft_is_active(net, flowtable))
8510                                 goto cont;
8511                         if (idx < s_idx)
8512                                 goto cont;
8513                         if (idx > s_idx)
8514                                 memset(&cb->args[1], 0,
8515                                        sizeof(cb->args) - sizeof(cb->args[0]));
8516                         if (filter && filter->table &&
8517                             strcmp(filter->table, table->name))
8518                                 goto cont;
8519
8520                         if (nf_tables_fill_flowtable_info(skb, net, NETLINK_CB(cb->skb).portid,
8521                                                           cb->nlh->nlmsg_seq,
8522                                                           NFT_MSG_NEWFLOWTABLE,
8523                                                           NLM_F_MULTI | NLM_F_APPEND,
8524                                                           table->family,
8525                                                           flowtable, NULL) < 0)
8526                                 goto done;
8527
8528                         nl_dump_check_consistent(cb, nlmsg_hdr(skb));
8529 cont:
8530                         idx++;
8531                 }
8532         }
8533 done:
8534         rcu_read_unlock();
8535
8536         cb->args[0] = idx;
8537         return skb->len;
8538 }
8539
8540 static int nf_tables_dump_flowtable_start(struct netlink_callback *cb)
8541 {
8542         const struct nlattr * const *nla = cb->data;
8543         struct nft_flowtable_filter *filter = NULL;
8544
8545         if (nla[NFTA_FLOWTABLE_TABLE]) {
8546                 filter = kzalloc(sizeof(*filter), GFP_ATOMIC);
8547                 if (!filter)
8548                         return -ENOMEM;
8549
8550                 filter->table = nla_strdup(nla[NFTA_FLOWTABLE_TABLE],
8551                                            GFP_ATOMIC);
8552                 if (!filter->table) {
8553                         kfree(filter);
8554                         return -ENOMEM;
8555                 }
8556         }
8557
8558         cb->data = filter;
8559         return 0;
8560 }
8561
8562 static int nf_tables_dump_flowtable_done(struct netlink_callback *cb)
8563 {
8564         struct nft_flowtable_filter *filter = cb->data;
8565
8566         if (!filter)
8567                 return 0;
8568
8569         kfree(filter->table);
8570         kfree(filter);
8571
8572         return 0;
8573 }
8574
8575 /* called with rcu_read_lock held */
8576 static int nf_tables_getflowtable(struct sk_buff *skb,
8577                                   const struct nfnl_info *info,
8578                                   const struct nlattr * const nla[])
8579 {
8580         u8 genmask = nft_genmask_cur(info->net);
8581         u8 family = info->nfmsg->nfgen_family;
8582         struct nft_flowtable *flowtable;
8583         const struct nft_table *table;
8584         struct net *net = info->net;
8585         struct sk_buff *skb2;
8586         int err;
8587
8588         if (info->nlh->nlmsg_flags & NLM_F_DUMP) {
8589                 struct netlink_dump_control c = {
8590                         .start = nf_tables_dump_flowtable_start,
8591                         .dump = nf_tables_dump_flowtable,
8592                         .done = nf_tables_dump_flowtable_done,
8593                         .module = THIS_MODULE,
8594                         .data = (void *)nla,
8595                 };
8596
8597                 return nft_netlink_dump_start_rcu(info->sk, skb, info->nlh, &c);
8598         }
8599
8600         if (!nla[NFTA_FLOWTABLE_NAME])
8601                 return -EINVAL;
8602
8603         table = nft_table_lookup(net, nla[NFTA_FLOWTABLE_TABLE], family,
8604                                  genmask, 0);
8605         if (IS_ERR(table))
8606                 return PTR_ERR(table);
8607
8608         flowtable = nft_flowtable_lookup(table, nla[NFTA_FLOWTABLE_NAME],
8609                                          genmask);
8610         if (IS_ERR(flowtable))
8611                 return PTR_ERR(flowtable);
8612
8613         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
8614         if (!skb2)
8615                 return -ENOMEM;
8616
8617         err = nf_tables_fill_flowtable_info(skb2, net, NETLINK_CB(skb).portid,
8618                                             info->nlh->nlmsg_seq,
8619                                             NFT_MSG_NEWFLOWTABLE, 0, family,
8620                                             flowtable, NULL);
8621         if (err < 0)
8622                 goto err_fill_flowtable_info;
8623
8624         return nfnetlink_unicast(skb2, net, NETLINK_CB(skb).portid);
8625
8626 err_fill_flowtable_info:
8627         kfree_skb(skb2);
8628         return err;
8629 }
8630
8631 static void nf_tables_flowtable_notify(struct nft_ctx *ctx,
8632                                        struct nft_flowtable *flowtable,
8633                                        struct list_head *hook_list, int event)
8634 {
8635         struct nftables_pernet *nft_net = nft_pernet(ctx->net);
8636         struct sk_buff *skb;
8637         u16 flags = 0;
8638         int err;
8639
8640         if (!ctx->report &&
8641             !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
8642                 return;
8643
8644         skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
8645         if (skb == NULL)
8646                 goto err;
8647
8648         if (ctx->flags & (NLM_F_CREATE | NLM_F_EXCL))
8649                 flags |= ctx->flags & (NLM_F_CREATE | NLM_F_EXCL);
8650
8651         err = nf_tables_fill_flowtable_info(skb, ctx->net, ctx->portid,
8652                                             ctx->seq, event, flags,
8653                                             ctx->family, flowtable, hook_list);
8654         if (err < 0) {
8655                 kfree_skb(skb);
8656                 goto err;
8657         }
8658
8659         nft_notify_enqueue(skb, ctx->report, &nft_net->notify_list);
8660         return;
8661 err:
8662         nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES, -ENOBUFS);
8663 }
8664
8665 static void nf_tables_flowtable_destroy(struct nft_flowtable *flowtable)
8666 {
8667         struct nft_hook *hook, *next;
8668
8669         flowtable->data.type->free(&flowtable->data);
8670         list_for_each_entry_safe(hook, next, &flowtable->hook_list, list) {
8671                 flowtable->data.type->setup(&flowtable->data, hook->ops.dev,
8672                                             FLOW_BLOCK_UNBIND);
8673                 list_del_rcu(&hook->list);
8674                 kfree(hook);
8675         }
8676         kfree(flowtable->name);
8677         module_put(flowtable->data.type->owner);
8678         kfree(flowtable);
8679 }
8680
8681 static int nf_tables_fill_gen_info(struct sk_buff *skb, struct net *net,
8682                                    u32 portid, u32 seq)
8683 {
8684         struct nftables_pernet *nft_net = nft_pernet(net);
8685         struct nlmsghdr *nlh;
8686         char buf[TASK_COMM_LEN];
8687         int event = nfnl_msg_type(NFNL_SUBSYS_NFTABLES, NFT_MSG_NEWGEN);
8688
8689         nlh = nfnl_msg_put(skb, portid, seq, event, 0, AF_UNSPEC,
8690                            NFNETLINK_V0, nft_base_seq(net));
8691         if (!nlh)
8692                 goto nla_put_failure;
8693
8694         if (nla_put_be32(skb, NFTA_GEN_ID, htonl(nft_net->base_seq)) ||
8695             nla_put_be32(skb, NFTA_GEN_PROC_PID, htonl(task_pid_nr(current))) ||
8696             nla_put_string(skb, NFTA_GEN_PROC_NAME, get_task_comm(buf, current)))
8697                 goto nla_put_failure;
8698
8699         nlmsg_end(skb, nlh);
8700         return 0;
8701
8702 nla_put_failure:
8703         nlmsg_trim(skb, nlh);
8704         return -EMSGSIZE;
8705 }
8706
8707 static void nft_flowtable_event(unsigned long event, struct net_device *dev,
8708                                 struct nft_flowtable *flowtable)
8709 {
8710         struct nft_hook *hook;
8711
8712         list_for_each_entry(hook, &flowtable->hook_list, list) {
8713                 if (hook->ops.dev != dev)
8714                         continue;
8715
8716                 /* flow_offload_netdev_event() cleans up entries for us. */
8717                 nft_unregister_flowtable_hook(dev_net(dev), flowtable, hook);
8718                 list_del_rcu(&hook->list);
8719                 kfree_rcu(hook, rcu);
8720                 break;
8721         }
8722 }
8723
8724 static int nf_tables_flowtable_event(struct notifier_block *this,
8725                                      unsigned long event, void *ptr)
8726 {
8727         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
8728         struct nft_flowtable *flowtable;
8729         struct nftables_pernet *nft_net;
8730         struct nft_table *table;
8731         struct net *net;
8732
8733         if (event != NETDEV_UNREGISTER)
8734                 return 0;
8735
8736         net = dev_net(dev);
8737         nft_net = nft_pernet(net);
8738         mutex_lock(&nft_net->commit_mutex);
8739         list_for_each_entry(table, &nft_net->tables, list) {
8740                 list_for_each_entry(flowtable, &table->flowtables, list) {
8741                         nft_flowtable_event(event, dev, flowtable);
8742                 }
8743         }
8744         mutex_unlock(&nft_net->commit_mutex);
8745
8746         return NOTIFY_DONE;
8747 }
8748
8749 static struct notifier_block nf_tables_flowtable_notifier = {
8750         .notifier_call  = nf_tables_flowtable_event,
8751 };
8752
8753 static void nf_tables_gen_notify(struct net *net, struct sk_buff *skb,
8754                                  int event)
8755 {
8756         struct nlmsghdr *nlh = nlmsg_hdr(skb);
8757         struct sk_buff *skb2;
8758         int err;
8759
8760         if (!nlmsg_report(nlh) &&
8761             !nfnetlink_has_listeners(net, NFNLGRP_NFTABLES))
8762                 return;
8763
8764         skb2 = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
8765         if (skb2 == NULL)
8766                 goto err;
8767
8768         err = nf_tables_fill_gen_info(skb2, net, NETLINK_CB(skb).portid,
8769                                       nlh->nlmsg_seq);
8770         if (err < 0) {
8771                 kfree_skb(skb2);
8772                 goto err;
8773         }
8774
8775         nfnetlink_send(skb2, net, NETLINK_CB(skb).portid, NFNLGRP_NFTABLES,
8776                        nlmsg_report(nlh), GFP_KERNEL);
8777         return;
8778 err:
8779         nfnetlink_set_err(net, NETLINK_CB(skb).portid, NFNLGRP_NFTABLES,
8780                           -ENOBUFS);
8781 }
8782
8783 static int nf_tables_getgen(struct sk_buff *skb, const struct nfnl_info *info,
8784                             const struct nlattr * const nla[])
8785 {
8786         struct sk_buff *skb2;
8787         int err;
8788
8789         skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_ATOMIC);
8790         if (skb2 == NULL)
8791                 return -ENOMEM;
8792
8793         err = nf_tables_fill_gen_info(skb2, info->net, NETLINK_CB(skb).portid,
8794                                       info->nlh->nlmsg_seq);
8795         if (err < 0)
8796                 goto err_fill_gen_info;
8797
8798         return nfnetlink_unicast(skb2, info->net, NETLINK_CB(skb).portid);
8799
8800 err_fill_gen_info:
8801         kfree_skb(skb2);
8802         return err;
8803 }
8804
8805 static const struct nfnl_callback nf_tables_cb[NFT_MSG_MAX] = {
8806         [NFT_MSG_NEWTABLE] = {
8807                 .call           = nf_tables_newtable,
8808                 .type           = NFNL_CB_BATCH,
8809                 .attr_count     = NFTA_TABLE_MAX,
8810                 .policy         = nft_table_policy,
8811         },
8812         [NFT_MSG_GETTABLE] = {
8813                 .call           = nf_tables_gettable,
8814                 .type           = NFNL_CB_RCU,
8815                 .attr_count     = NFTA_TABLE_MAX,
8816                 .policy         = nft_table_policy,
8817         },
8818         [NFT_MSG_DELTABLE] = {
8819                 .call           = nf_tables_deltable,
8820                 .type           = NFNL_CB_BATCH,
8821                 .attr_count     = NFTA_TABLE_MAX,
8822                 .policy         = nft_table_policy,
8823         },
8824         [NFT_MSG_DESTROYTABLE] = {
8825                 .call           = nf_tables_deltable,
8826                 .type           = NFNL_CB_BATCH,
8827                 .attr_count     = NFTA_TABLE_MAX,
8828                 .policy         = nft_table_policy,
8829         },
8830         [NFT_MSG_NEWCHAIN] = {
8831                 .call           = nf_tables_newchain,
8832                 .type           = NFNL_CB_BATCH,
8833                 .attr_count     = NFTA_CHAIN_MAX,
8834                 .policy         = nft_chain_policy,
8835         },
8836         [NFT_MSG_GETCHAIN] = {
8837                 .call           = nf_tables_getchain,
8838                 .type           = NFNL_CB_RCU,
8839                 .attr_count     = NFTA_CHAIN_MAX,
8840                 .policy         = nft_chain_policy,
8841         },
8842         [NFT_MSG_DELCHAIN] = {
8843                 .call           = nf_tables_delchain,
8844                 .type           = NFNL_CB_BATCH,
8845                 .attr_count     = NFTA_CHAIN_MAX,
8846                 .policy         = nft_chain_policy,
8847         },
8848         [NFT_MSG_DESTROYCHAIN] = {
8849                 .call           = nf_tables_delchain,
8850                 .type           = NFNL_CB_BATCH,
8851                 .attr_count     = NFTA_CHAIN_MAX,
8852                 .policy         = nft_chain_policy,
8853         },
8854         [NFT_MSG_NEWRULE] = {
8855                 .call           = nf_tables_newrule,
8856                 .type           = NFNL_CB_BATCH,
8857                 .attr_count     = NFTA_RULE_MAX,
8858                 .policy         = nft_rule_policy,
8859         },
8860         [NFT_MSG_GETRULE] = {
8861                 .call           = nf_tables_getrule,
8862                 .type           = NFNL_CB_RCU,
8863                 .attr_count     = NFTA_RULE_MAX,
8864                 .policy         = nft_rule_policy,
8865         },
8866         [NFT_MSG_GETRULE_RESET] = {
8867                 .call           = nf_tables_getrule,
8868                 .type           = NFNL_CB_RCU,
8869                 .attr_count     = NFTA_RULE_MAX,
8870                 .policy         = nft_rule_policy,
8871         },
8872         [NFT_MSG_DELRULE] = {
8873                 .call           = nf_tables_delrule,
8874                 .type           = NFNL_CB_BATCH,
8875                 .attr_count     = NFTA_RULE_MAX,
8876                 .policy         = nft_rule_policy,
8877         },
8878         [NFT_MSG_DESTROYRULE] = {
8879                 .call           = nf_tables_delrule,
8880                 .type           = NFNL_CB_BATCH,
8881                 .attr_count     = NFTA_RULE_MAX,
8882                 .policy         = nft_rule_policy,
8883         },
8884         [NFT_MSG_NEWSET] = {
8885                 .call           = nf_tables_newset,
8886                 .type           = NFNL_CB_BATCH,
8887                 .attr_count     = NFTA_SET_MAX,
8888                 .policy         = nft_set_policy,
8889         },
8890         [NFT_MSG_GETSET] = {
8891                 .call           = nf_tables_getset,
8892                 .type           = NFNL_CB_RCU,
8893                 .attr_count     = NFTA_SET_MAX,
8894                 .policy         = nft_set_policy,
8895         },
8896         [NFT_MSG_DELSET] = {
8897                 .call           = nf_tables_delset,
8898                 .type           = NFNL_CB_BATCH,
8899                 .attr_count     = NFTA_SET_MAX,
8900                 .policy         = nft_set_policy,
8901         },
8902         [NFT_MSG_DESTROYSET] = {
8903                 .call           = nf_tables_delset,
8904                 .type           = NFNL_CB_BATCH,
8905                 .attr_count     = NFTA_SET_MAX,
8906                 .policy         = nft_set_policy,
8907         },
8908         [NFT_MSG_NEWSETELEM] = {
8909                 .call           = nf_tables_newsetelem,
8910                 .type           = NFNL_CB_BATCH,
8911                 .attr_count     = NFTA_SET_ELEM_LIST_MAX,
8912                 .policy         = nft_set_elem_list_policy,
8913         },
8914         [NFT_MSG_GETSETELEM] = {
8915                 .call           = nf_tables_getsetelem,
8916                 .type           = NFNL_CB_RCU,
8917                 .attr_count     = NFTA_SET_ELEM_LIST_MAX,
8918                 .policy         = nft_set_elem_list_policy,
8919         },
8920         [NFT_MSG_DELSETELEM] = {
8921                 .call           = nf_tables_delsetelem,
8922                 .type           = NFNL_CB_BATCH,
8923                 .attr_count     = NFTA_SET_ELEM_LIST_MAX,
8924                 .policy         = nft_set_elem_list_policy,
8925         },
8926         [NFT_MSG_DESTROYSETELEM] = {
8927                 .call           = nf_tables_delsetelem,
8928                 .type           = NFNL_CB_BATCH,
8929                 .attr_count     = NFTA_SET_ELEM_LIST_MAX,
8930                 .policy         = nft_set_elem_list_policy,
8931         },
8932         [NFT_MSG_GETGEN] = {
8933                 .call           = nf_tables_getgen,
8934                 .type           = NFNL_CB_RCU,
8935         },
8936         [NFT_MSG_NEWOBJ] = {
8937                 .call           = nf_tables_newobj,
8938                 .type           = NFNL_CB_BATCH,
8939                 .attr_count     = NFTA_OBJ_MAX,
8940                 .policy         = nft_obj_policy,
8941         },
8942         [NFT_MSG_GETOBJ] = {
8943                 .call           = nf_tables_getobj,
8944                 .type           = NFNL_CB_RCU,
8945                 .attr_count     = NFTA_OBJ_MAX,
8946                 .policy         = nft_obj_policy,
8947         },
8948         [NFT_MSG_DELOBJ] = {
8949                 .call           = nf_tables_delobj,
8950                 .type           = NFNL_CB_BATCH,
8951                 .attr_count     = NFTA_OBJ_MAX,
8952                 .policy         = nft_obj_policy,
8953         },
8954         [NFT_MSG_DESTROYOBJ] = {
8955                 .call           = nf_tables_delobj,
8956                 .type           = NFNL_CB_BATCH,
8957                 .attr_count     = NFTA_OBJ_MAX,
8958                 .policy         = nft_obj_policy,
8959         },
8960         [NFT_MSG_GETOBJ_RESET] = {
8961                 .call           = nf_tables_getobj,
8962                 .type           = NFNL_CB_RCU,
8963                 .attr_count     = NFTA_OBJ_MAX,
8964                 .policy         = nft_obj_policy,
8965         },
8966         [NFT_MSG_NEWFLOWTABLE] = {
8967                 .call           = nf_tables_newflowtable,
8968                 .type           = NFNL_CB_BATCH,
8969                 .attr_count     = NFTA_FLOWTABLE_MAX,
8970                 .policy         = nft_flowtable_policy,
8971         },
8972         [NFT_MSG_GETFLOWTABLE] = {
8973                 .call           = nf_tables_getflowtable,
8974                 .type           = NFNL_CB_RCU,
8975                 .attr_count     = NFTA_FLOWTABLE_MAX,
8976                 .policy         = nft_flowtable_policy,
8977         },
8978         [NFT_MSG_DELFLOWTABLE] = {
8979                 .call           = nf_tables_delflowtable,
8980                 .type           = NFNL_CB_BATCH,
8981                 .attr_count     = NFTA_FLOWTABLE_MAX,
8982                 .policy         = nft_flowtable_policy,
8983         },
8984         [NFT_MSG_DESTROYFLOWTABLE] = {
8985                 .call           = nf_tables_delflowtable,
8986                 .type           = NFNL_CB_BATCH,
8987                 .attr_count     = NFTA_FLOWTABLE_MAX,
8988                 .policy         = nft_flowtable_policy,
8989         },
8990 };
8991
8992 static int nf_tables_validate(struct net *net)
8993 {
8994         struct nftables_pernet *nft_net = nft_pernet(net);
8995         struct nft_table *table;
8996
8997         list_for_each_entry(table, &nft_net->tables, list) {
8998                 switch (table->validate_state) {
8999                 case NFT_VALIDATE_SKIP:
9000                         continue;
9001                 case NFT_VALIDATE_NEED:
9002                         nft_validate_state_update(table, NFT_VALIDATE_DO);
9003                         fallthrough;
9004                 case NFT_VALIDATE_DO:
9005                         if (nft_table_validate(net, table) < 0)
9006                                 return -EAGAIN;
9007
9008                         nft_validate_state_update(table, NFT_VALIDATE_SKIP);
9009                 }
9010
9011                 break;
9012         }
9013
9014         return 0;
9015 }
9016
9017 /* a drop policy has to be deferred until all rules have been activated,
9018  * otherwise a large ruleset that contains a drop-policy base chain will
9019  * cause all packets to get dropped until the full transaction has been
9020  * processed.
9021  *
9022  * We defer the drop policy until the transaction has been finalized.
9023  */
9024 static void nft_chain_commit_drop_policy(struct nft_trans *trans)
9025 {
9026         struct nft_base_chain *basechain;
9027
9028         if (nft_trans_chain_policy(trans) != NF_DROP)
9029                 return;
9030
9031         if (!nft_is_base_chain(trans->ctx.chain))
9032                 return;
9033
9034         basechain = nft_base_chain(trans->ctx.chain);
9035         basechain->policy = NF_DROP;
9036 }
9037
9038 static void nft_chain_commit_update(struct nft_trans *trans)
9039 {
9040         struct nft_base_chain *basechain;
9041
9042         if (nft_trans_chain_name(trans)) {
9043                 rhltable_remove(&trans->ctx.table->chains_ht,
9044                                 &trans->ctx.chain->rhlhead,
9045                                 nft_chain_ht_params);
9046                 swap(trans->ctx.chain->name, nft_trans_chain_name(trans));
9047                 rhltable_insert_key(&trans->ctx.table->chains_ht,
9048                                     trans->ctx.chain->name,
9049                                     &trans->ctx.chain->rhlhead,
9050                                     nft_chain_ht_params);
9051         }
9052
9053         if (!nft_is_base_chain(trans->ctx.chain))
9054                 return;
9055
9056         nft_chain_stats_replace(trans);
9057
9058         basechain = nft_base_chain(trans->ctx.chain);
9059
9060         switch (nft_trans_chain_policy(trans)) {
9061         case NF_DROP:
9062         case NF_ACCEPT:
9063                 basechain->policy = nft_trans_chain_policy(trans);
9064                 break;
9065         }
9066 }
9067
9068 static void nft_obj_commit_update(struct nft_trans *trans)
9069 {
9070         struct nft_object *newobj;
9071         struct nft_object *obj;
9072
9073         obj = nft_trans_obj(trans);
9074         newobj = nft_trans_obj_newobj(trans);
9075
9076         if (obj->ops->update)
9077                 obj->ops->update(obj, newobj);
9078
9079         nft_obj_destroy(&trans->ctx, newobj);
9080 }
9081
9082 static void nft_commit_release(struct nft_trans *trans)
9083 {
9084         switch (trans->msg_type) {
9085         case NFT_MSG_DELTABLE:
9086         case NFT_MSG_DESTROYTABLE:
9087                 nf_tables_table_destroy(&trans->ctx);
9088                 break;
9089         case NFT_MSG_NEWCHAIN:
9090                 free_percpu(nft_trans_chain_stats(trans));
9091                 kfree(nft_trans_chain_name(trans));
9092                 break;
9093         case NFT_MSG_DELCHAIN:
9094         case NFT_MSG_DESTROYCHAIN:
9095                 if (nft_trans_chain_update(trans))
9096                         nft_hooks_destroy(&nft_trans_chain_hooks(trans));
9097                 else
9098                         nf_tables_chain_destroy(&trans->ctx);
9099                 break;
9100         case NFT_MSG_DELRULE:
9101         case NFT_MSG_DESTROYRULE:
9102                 nf_tables_rule_destroy(&trans->ctx, nft_trans_rule(trans));
9103                 break;
9104         case NFT_MSG_DELSET:
9105         case NFT_MSG_DESTROYSET:
9106                 nft_set_destroy(&trans->ctx, nft_trans_set(trans));
9107                 break;
9108         case NFT_MSG_DELSETELEM:
9109         case NFT_MSG_DESTROYSETELEM:
9110                 nf_tables_set_elem_destroy(&trans->ctx,
9111                                            nft_trans_elem_set(trans),
9112                                            nft_trans_elem(trans).priv);
9113                 break;
9114         case NFT_MSG_DELOBJ:
9115         case NFT_MSG_DESTROYOBJ:
9116                 nft_obj_destroy(&trans->ctx, nft_trans_obj(trans));
9117                 break;
9118         case NFT_MSG_DELFLOWTABLE:
9119         case NFT_MSG_DESTROYFLOWTABLE:
9120                 if (nft_trans_flowtable_update(trans))
9121                         nft_hooks_destroy(&nft_trans_flowtable_hooks(trans));
9122                 else
9123                         nf_tables_flowtable_destroy(nft_trans_flowtable(trans));
9124                 break;
9125         }
9126
9127         if (trans->put_net)
9128                 put_net(trans->ctx.net);
9129
9130         kfree(trans);
9131 }
9132
9133 static void nf_tables_trans_destroy_work(struct work_struct *w)
9134 {
9135         struct nft_trans *trans, *next;
9136         LIST_HEAD(head);
9137
9138         spin_lock(&nf_tables_destroy_list_lock);
9139         list_splice_init(&nf_tables_destroy_list, &head);
9140         spin_unlock(&nf_tables_destroy_list_lock);
9141
9142         if (list_empty(&head))
9143                 return;
9144
9145         synchronize_rcu();
9146
9147         list_for_each_entry_safe(trans, next, &head, list) {
9148                 nft_trans_list_del(trans);
9149                 nft_commit_release(trans);
9150         }
9151 }
9152
9153 void nf_tables_trans_destroy_flush_work(void)
9154 {
9155         flush_work(&trans_destroy_work);
9156 }
9157 EXPORT_SYMBOL_GPL(nf_tables_trans_destroy_flush_work);
9158
9159 static bool nft_expr_reduce(struct nft_regs_track *track,
9160                             const struct nft_expr *expr)
9161 {
9162         return false;
9163 }
9164
9165 static int nf_tables_commit_chain_prepare(struct net *net, struct nft_chain *chain)
9166 {
9167         const struct nft_expr *expr, *last;
9168         struct nft_regs_track track = {};
9169         unsigned int size, data_size;
9170         void *data, *data_boundary;
9171         struct nft_rule_dp *prule;
9172         struct nft_rule *rule;
9173
9174         /* already handled or inactive chain? */
9175         if (chain->blob_next || !nft_is_active_next(net, chain))
9176                 return 0;
9177
9178         data_size = 0;
9179         list_for_each_entry(rule, &chain->rules, list) {
9180                 if (nft_is_active_next(net, rule)) {
9181                         data_size += sizeof(*prule) + rule->dlen;
9182                         if (data_size > INT_MAX)
9183                                 return -ENOMEM;
9184                 }
9185         }
9186
9187         chain->blob_next = nf_tables_chain_alloc_rules(chain, data_size);
9188         if (!chain->blob_next)
9189                 return -ENOMEM;
9190
9191         data = (void *)chain->blob_next->data;
9192         data_boundary = data + data_size;
9193         size = 0;
9194
9195         list_for_each_entry(rule, &chain->rules, list) {
9196                 if (!nft_is_active_next(net, rule))
9197                         continue;
9198
9199                 prule = (struct nft_rule_dp *)data;
9200                 data += offsetof(struct nft_rule_dp, data);
9201                 if (WARN_ON_ONCE(data > data_boundary))
9202                         return -ENOMEM;
9203
9204                 size = 0;
9205                 track.last = nft_expr_last(rule);
9206                 nft_rule_for_each_expr(expr, last, rule) {
9207                         track.cur = expr;
9208
9209                         if (nft_expr_reduce(&track, expr)) {
9210                                 expr = track.cur;
9211                                 continue;
9212                         }
9213
9214                         if (WARN_ON_ONCE(data + size + expr->ops->size > data_boundary))
9215                                 return -ENOMEM;
9216
9217                         memcpy(data + size, expr, expr->ops->size);
9218                         size += expr->ops->size;
9219                 }
9220                 if (WARN_ON_ONCE(size >= 1 << 12))
9221                         return -ENOMEM;
9222
9223                 prule->handle = rule->handle;
9224                 prule->dlen = size;
9225                 prule->is_last = 0;
9226
9227                 data += size;
9228                 size = 0;
9229                 chain->blob_next->size += (unsigned long)(data - (void *)prule);
9230         }
9231
9232         if (WARN_ON_ONCE(data > data_boundary))
9233                 return -ENOMEM;
9234
9235         prule = (struct nft_rule_dp *)data;
9236         nft_last_rule(chain, prule);
9237
9238         return 0;
9239 }
9240
9241 static void nf_tables_commit_chain_prepare_cancel(struct net *net)
9242 {
9243         struct nftables_pernet *nft_net = nft_pernet(net);
9244         struct nft_trans *trans, *next;
9245
9246         list_for_each_entry_safe(trans, next, &nft_net->commit_list, list) {
9247                 struct nft_chain *chain = trans->ctx.chain;
9248
9249                 if (trans->msg_type == NFT_MSG_NEWRULE ||
9250                     trans->msg_type == NFT_MSG_DELRULE) {
9251                         kvfree(chain->blob_next);
9252                         chain->blob_next = NULL;
9253                 }
9254         }
9255 }
9256
9257 static void __nf_tables_commit_chain_free_rules(struct rcu_head *h)
9258 {
9259         struct nft_rule_dp_last *l = container_of(h, struct nft_rule_dp_last, h);
9260
9261         kvfree(l->blob);
9262 }
9263
9264 static void nf_tables_commit_chain_free_rules_old(struct nft_rule_blob *blob)
9265 {
9266         struct nft_rule_dp_last *last;
9267
9268         /* last rule trailer is after end marker */
9269         last = (void *)blob + sizeof(*blob) + blob->size;
9270         last->blob = blob;
9271
9272         call_rcu(&last->h, __nf_tables_commit_chain_free_rules);
9273 }
9274
9275 static void nf_tables_commit_chain(struct net *net, struct nft_chain *chain)
9276 {
9277         struct nft_rule_blob *g0, *g1;
9278         bool next_genbit;
9279
9280         next_genbit = nft_gencursor_next(net);
9281
9282         g0 = rcu_dereference_protected(chain->blob_gen_0,
9283                                        lockdep_commit_lock_is_held(net));
9284         g1 = rcu_dereference_protected(chain->blob_gen_1,
9285                                        lockdep_commit_lock_is_held(net));
9286
9287         /* No changes to this chain? */
9288         if (chain->blob_next == NULL) {
9289                 /* chain had no change in last or next generation */
9290                 if (g0 == g1)
9291                         return;
9292                 /*
9293                  * chain had no change in this generation; make sure next
9294                  * one uses same rules as current generation.
9295                  */
9296                 if (next_genbit) {
9297                         rcu_assign_pointer(chain->blob_gen_1, g0);
9298                         nf_tables_commit_chain_free_rules_old(g1);
9299                 } else {
9300                         rcu_assign_pointer(chain->blob_gen_0, g1);
9301                         nf_tables_commit_chain_free_rules_old(g0);
9302                 }
9303
9304                 return;
9305         }
9306
9307         if (next_genbit)
9308                 rcu_assign_pointer(chain->blob_gen_1, chain->blob_next);
9309         else
9310                 rcu_assign_pointer(chain->blob_gen_0, chain->blob_next);
9311
9312         chain->blob_next = NULL;
9313
9314         if (g0 == g1)
9315                 return;
9316
9317         if (next_genbit)
9318                 nf_tables_commit_chain_free_rules_old(g1);
9319         else
9320                 nf_tables_commit_chain_free_rules_old(g0);
9321 }
9322
9323 static void nft_obj_del(struct nft_object *obj)
9324 {
9325         rhltable_remove(&nft_objname_ht, &obj->rhlhead, nft_objname_ht_params);
9326         list_del_rcu(&obj->list);
9327 }
9328
9329 void nft_chain_del(struct nft_chain *chain)
9330 {
9331         struct nft_table *table = chain->table;
9332
9333         WARN_ON_ONCE(rhltable_remove(&table->chains_ht, &chain->rhlhead,
9334                                      nft_chain_ht_params));
9335         list_del_rcu(&chain->list);
9336 }
9337
9338 static void nf_tables_module_autoload_cleanup(struct net *net)
9339 {
9340         struct nftables_pernet *nft_net = nft_pernet(net);
9341         struct nft_module_request *req, *next;
9342
9343         WARN_ON_ONCE(!list_empty(&nft_net->commit_list));
9344         list_for_each_entry_safe(req, next, &nft_net->module_list, list) {
9345                 WARN_ON_ONCE(!req->done);
9346                 list_del(&req->list);
9347                 kfree(req);
9348         }
9349 }
9350
9351 static void nf_tables_commit_release(struct net *net)
9352 {
9353         struct nftables_pernet *nft_net = nft_pernet(net);
9354         struct nft_trans *trans;
9355
9356         /* all side effects have to be made visible.
9357          * For example, if a chain named 'foo' has been deleted, a
9358          * new transaction must not find it anymore.
9359          *
9360          * Memory reclaim happens asynchronously from work queue
9361          * to prevent expensive synchronize_rcu() in commit phase.
9362          */
9363         if (list_empty(&nft_net->commit_list)) {
9364                 nf_tables_module_autoload_cleanup(net);
9365                 mutex_unlock(&nft_net->commit_mutex);
9366                 return;
9367         }
9368
9369         trans = list_last_entry(&nft_net->commit_list,
9370                                 struct nft_trans, list);
9371         get_net(trans->ctx.net);
9372         WARN_ON_ONCE(trans->put_net);
9373
9374         trans->put_net = true;
9375         spin_lock(&nf_tables_destroy_list_lock);
9376         list_splice_tail_init(&nft_net->commit_list, &nf_tables_destroy_list);
9377         spin_unlock(&nf_tables_destroy_list_lock);
9378
9379         nf_tables_module_autoload_cleanup(net);
9380         schedule_work(&trans_destroy_work);
9381
9382         mutex_unlock(&nft_net->commit_mutex);
9383 }
9384
9385 static void nft_commit_notify(struct net *net, u32 portid)
9386 {
9387         struct nftables_pernet *nft_net = nft_pernet(net);
9388         struct sk_buff *batch_skb = NULL, *nskb, *skb;
9389         unsigned char *data;
9390         int len;
9391
9392         list_for_each_entry_safe(skb, nskb, &nft_net->notify_list, list) {
9393                 if (!batch_skb) {
9394 new_batch:
9395                         batch_skb = skb;
9396                         len = NLMSG_GOODSIZE - skb->len;
9397                         list_del(&skb->list);
9398                         continue;
9399                 }
9400                 len -= skb->len;
9401                 if (len > 0 && NFT_CB(skb).report == NFT_CB(batch_skb).report) {
9402                         data = skb_put(batch_skb, skb->len);
9403                         memcpy(data, skb->data, skb->len);
9404                         list_del(&skb->list);
9405                         kfree_skb(skb);
9406                         continue;
9407                 }
9408                 nfnetlink_send(batch_skb, net, portid, NFNLGRP_NFTABLES,
9409                                NFT_CB(batch_skb).report, GFP_KERNEL);
9410                 goto new_batch;
9411         }
9412
9413         if (batch_skb) {
9414                 nfnetlink_send(batch_skb, net, portid, NFNLGRP_NFTABLES,
9415                                NFT_CB(batch_skb).report, GFP_KERNEL);
9416         }
9417
9418         WARN_ON_ONCE(!list_empty(&nft_net->notify_list));
9419 }
9420
9421 static int nf_tables_commit_audit_alloc(struct list_head *adl,
9422                                         struct nft_table *table)
9423 {
9424         struct nft_audit_data *adp;
9425
9426         list_for_each_entry(adp, adl, list) {
9427                 if (adp->table == table)
9428                         return 0;
9429         }
9430         adp = kzalloc(sizeof(*adp), GFP_KERNEL);
9431         if (!adp)
9432                 return -ENOMEM;
9433         adp->table = table;
9434         list_add(&adp->list, adl);
9435         return 0;
9436 }
9437
9438 static void nf_tables_commit_audit_free(struct list_head *adl)
9439 {
9440         struct nft_audit_data *adp, *adn;
9441
9442         list_for_each_entry_safe(adp, adn, adl, list) {
9443                 list_del(&adp->list);
9444                 kfree(adp);
9445         }
9446 }
9447
9448 static void nf_tables_commit_audit_collect(struct list_head *adl,
9449                                            struct nft_table *table, u32 op)
9450 {
9451         struct nft_audit_data *adp;
9452
9453         list_for_each_entry(adp, adl, list) {
9454                 if (adp->table == table)
9455                         goto found;
9456         }
9457         WARN_ONCE(1, "table=%s not expected in commit list", table->name);
9458         return;
9459 found:
9460         adp->entries++;
9461         if (!adp->op || adp->op > op)
9462                 adp->op = op;
9463 }
9464
9465 #define AUNFTABLENAMELEN (NFT_TABLE_MAXNAMELEN + 22)
9466
9467 static void nf_tables_commit_audit_log(struct list_head *adl, u32 generation)
9468 {
9469         struct nft_audit_data *adp, *adn;
9470         char aubuf[AUNFTABLENAMELEN];
9471
9472         list_for_each_entry_safe(adp, adn, adl, list) {
9473                 snprintf(aubuf, AUNFTABLENAMELEN, "%s:%u", adp->table->name,
9474                          generation);
9475                 audit_log_nfcfg(aubuf, adp->table->family, adp->entries,
9476                                 nft2audit_op[adp->op], GFP_KERNEL);
9477                 list_del(&adp->list);
9478                 kfree(adp);
9479         }
9480 }
9481
9482 static void nft_set_commit_update(struct list_head *set_update_list)
9483 {
9484         struct nft_set *set, *next;
9485
9486         list_for_each_entry_safe(set, next, set_update_list, pending_update) {
9487                 list_del_init(&set->pending_update);
9488
9489                 if (!set->ops->commit)
9490                         continue;
9491
9492                 set->ops->commit(set);
9493         }
9494 }
9495
9496 static int nf_tables_commit(struct net *net, struct sk_buff *skb)
9497 {
9498         struct nftables_pernet *nft_net = nft_pernet(net);
9499         struct nft_trans *trans, *next;
9500         LIST_HEAD(set_update_list);
9501         struct nft_trans_elem *te;
9502         struct nft_chain *chain;
9503         struct nft_table *table;
9504         unsigned int base_seq;
9505         LIST_HEAD(adl);
9506         int err;
9507
9508         if (list_empty(&nft_net->commit_list)) {
9509                 mutex_unlock(&nft_net->commit_mutex);
9510                 return 0;
9511         }
9512
9513         list_for_each_entry(trans, &nft_net->binding_list, binding_list) {
9514                 switch (trans->msg_type) {
9515                 case NFT_MSG_NEWSET:
9516                         if (!nft_trans_set_update(trans) &&
9517                             nft_set_is_anonymous(nft_trans_set(trans)) &&
9518                             !nft_trans_set_bound(trans)) {
9519                                 pr_warn_once("nftables ruleset with unbound set\n");
9520                                 return -EINVAL;
9521                         }
9522                         break;
9523                 case NFT_MSG_NEWCHAIN:
9524                         if (!nft_trans_chain_update(trans) &&
9525                             nft_chain_binding(nft_trans_chain(trans)) &&
9526                             !nft_trans_chain_bound(trans)) {
9527                                 pr_warn_once("nftables ruleset with unbound chain\n");
9528                                 return -EINVAL;
9529                         }
9530                         break;
9531                 }
9532         }
9533
9534         /* 0. Validate ruleset, otherwise roll back for error reporting. */
9535         if (nf_tables_validate(net) < 0)
9536                 return -EAGAIN;
9537
9538         err = nft_flow_rule_offload_commit(net);
9539         if (err < 0)
9540                 return err;
9541
9542         /* 1.  Allocate space for next generation rules_gen_X[] */
9543         list_for_each_entry_safe(trans, next, &nft_net->commit_list, list) {
9544                 int ret;
9545
9546                 ret = nf_tables_commit_audit_alloc(&adl, trans->ctx.table);
9547                 if (ret) {
9548                         nf_tables_commit_chain_prepare_cancel(net);
9549                         nf_tables_commit_audit_free(&adl);
9550                         return ret;
9551                 }
9552                 if (trans->msg_type == NFT_MSG_NEWRULE ||
9553                     trans->msg_type == NFT_MSG_DELRULE) {
9554                         chain = trans->ctx.chain;
9555
9556                         ret = nf_tables_commit_chain_prepare(net, chain);
9557                         if (ret < 0) {
9558                                 nf_tables_commit_chain_prepare_cancel(net);
9559                                 nf_tables_commit_audit_free(&adl);
9560                                 return ret;
9561                         }
9562                 }
9563         }
9564
9565         /* step 2.  Make rules_gen_X visible to packet path */
9566         list_for_each_entry(table, &nft_net->tables, list) {
9567                 list_for_each_entry(chain, &table->chains, list)
9568                         nf_tables_commit_chain(net, chain);
9569         }
9570
9571         /*
9572          * Bump generation counter, invalidate any dump in progress.
9573          * Cannot fail after this point.
9574          */
9575         base_seq = READ_ONCE(nft_net->base_seq);
9576         while (++base_seq == 0)
9577                 ;
9578
9579         WRITE_ONCE(nft_net->base_seq, base_seq);
9580
9581         /* step 3. Start new generation, rules_gen_X now in use. */
9582         net->nft.gencursor = nft_gencursor_next(net);
9583
9584         list_for_each_entry_safe(trans, next, &nft_net->commit_list, list) {
9585                 nf_tables_commit_audit_collect(&adl, trans->ctx.table,
9586                                                trans->msg_type);
9587                 switch (trans->msg_type) {
9588                 case NFT_MSG_NEWTABLE:
9589                         if (nft_trans_table_update(trans)) {
9590                                 if (!(trans->ctx.table->flags & __NFT_TABLE_F_UPDATE)) {
9591                                         nft_trans_destroy(trans);
9592                                         break;
9593                                 }
9594                                 if (trans->ctx.table->flags & NFT_TABLE_F_DORMANT)
9595                                         nf_tables_table_disable(net, trans->ctx.table);
9596
9597                                 trans->ctx.table->flags &= ~__NFT_TABLE_F_UPDATE;
9598                         } else {
9599                                 nft_clear(net, trans->ctx.table);
9600                         }
9601                         nf_tables_table_notify(&trans->ctx, NFT_MSG_NEWTABLE);
9602                         nft_trans_destroy(trans);
9603                         break;
9604                 case NFT_MSG_DELTABLE:
9605                 case NFT_MSG_DESTROYTABLE:
9606                         list_del_rcu(&trans->ctx.table->list);
9607                         nf_tables_table_notify(&trans->ctx, trans->msg_type);
9608                         break;
9609                 case NFT_MSG_NEWCHAIN:
9610                         if (nft_trans_chain_update(trans)) {
9611                                 nft_chain_commit_update(trans);
9612                                 nf_tables_chain_notify(&trans->ctx, NFT_MSG_NEWCHAIN,
9613                                                        &nft_trans_chain_hooks(trans));
9614                                 list_splice(&nft_trans_chain_hooks(trans),
9615                                             &nft_trans_basechain(trans)->hook_list);
9616                                 /* trans destroyed after rcu grace period */
9617                         } else {
9618                                 nft_chain_commit_drop_policy(trans);
9619                                 nft_clear(net, trans->ctx.chain);
9620                                 nf_tables_chain_notify(&trans->ctx, NFT_MSG_NEWCHAIN, NULL);
9621                                 nft_trans_destroy(trans);
9622                         }
9623                         break;
9624                 case NFT_MSG_DELCHAIN:
9625                 case NFT_MSG_DESTROYCHAIN:
9626                         if (nft_trans_chain_update(trans)) {
9627                                 nf_tables_chain_notify(&trans->ctx, NFT_MSG_DELCHAIN,
9628                                                        &nft_trans_chain_hooks(trans));
9629                                 nft_netdev_unregister_hooks(net,
9630                                                             &nft_trans_chain_hooks(trans),
9631                                                             true);
9632                         } else {
9633                                 nft_chain_del(trans->ctx.chain);
9634                                 nf_tables_chain_notify(&trans->ctx, NFT_MSG_DELCHAIN,
9635                                                        NULL);
9636                                 nf_tables_unregister_hook(trans->ctx.net,
9637                                                           trans->ctx.table,
9638                                                           trans->ctx.chain);
9639                         }
9640                         break;
9641                 case NFT_MSG_NEWRULE:
9642                         nft_clear(trans->ctx.net, nft_trans_rule(trans));
9643                         nf_tables_rule_notify(&trans->ctx,
9644                                               nft_trans_rule(trans),
9645                                               NFT_MSG_NEWRULE);
9646                         if (trans->ctx.chain->flags & NFT_CHAIN_HW_OFFLOAD)
9647                                 nft_flow_rule_destroy(nft_trans_flow_rule(trans));
9648
9649                         nft_trans_destroy(trans);
9650                         break;
9651                 case NFT_MSG_DELRULE:
9652                 case NFT_MSG_DESTROYRULE:
9653                         list_del_rcu(&nft_trans_rule(trans)->list);
9654                         nf_tables_rule_notify(&trans->ctx,
9655                                               nft_trans_rule(trans),
9656                                               trans->msg_type);
9657                         nft_rule_expr_deactivate(&trans->ctx,
9658                                                  nft_trans_rule(trans),
9659                                                  NFT_TRANS_COMMIT);
9660
9661                         if (trans->ctx.chain->flags & NFT_CHAIN_HW_OFFLOAD)
9662                                 nft_flow_rule_destroy(nft_trans_flow_rule(trans));
9663                         break;
9664                 case NFT_MSG_NEWSET:
9665                         if (nft_trans_set_update(trans)) {
9666                                 struct nft_set *set = nft_trans_set(trans);
9667
9668                                 WRITE_ONCE(set->timeout, nft_trans_set_timeout(trans));
9669                                 WRITE_ONCE(set->gc_int, nft_trans_set_gc_int(trans));
9670                         } else {
9671                                 nft_clear(net, nft_trans_set(trans));
9672                                 /* This avoids hitting -EBUSY when deleting the table
9673                                  * from the transaction.
9674                                  */
9675                                 if (nft_set_is_anonymous(nft_trans_set(trans)) &&
9676                                     !list_empty(&nft_trans_set(trans)->bindings))
9677                                         trans->ctx.table->use--;
9678                         }
9679                         nf_tables_set_notify(&trans->ctx, nft_trans_set(trans),
9680                                              NFT_MSG_NEWSET, GFP_KERNEL);
9681                         nft_trans_destroy(trans);
9682                         break;
9683                 case NFT_MSG_DELSET:
9684                 case NFT_MSG_DESTROYSET:
9685                         list_del_rcu(&nft_trans_set(trans)->list);
9686                         nf_tables_set_notify(&trans->ctx, nft_trans_set(trans),
9687                                              trans->msg_type, GFP_KERNEL);
9688                         break;
9689                 case NFT_MSG_NEWSETELEM:
9690                         te = (struct nft_trans_elem *)trans->data;
9691
9692                         nft_setelem_activate(net, te->set, &te->elem);
9693                         nf_tables_setelem_notify(&trans->ctx, te->set,
9694                                                  &te->elem,
9695                                                  NFT_MSG_NEWSETELEM);
9696                         if (te->set->ops->commit &&
9697                             list_empty(&te->set->pending_update)) {
9698                                 list_add_tail(&te->set->pending_update,
9699                                               &set_update_list);
9700                         }
9701                         nft_trans_destroy(trans);
9702                         break;
9703                 case NFT_MSG_DELSETELEM:
9704                 case NFT_MSG_DESTROYSETELEM:
9705                         te = (struct nft_trans_elem *)trans->data;
9706
9707                         nf_tables_setelem_notify(&trans->ctx, te->set,
9708                                                  &te->elem,
9709                                                  trans->msg_type);
9710                         nft_setelem_remove(net, te->set, &te->elem);
9711                         if (!nft_setelem_is_catchall(te->set, &te->elem)) {
9712                                 atomic_dec(&te->set->nelems);
9713                                 te->set->ndeact--;
9714                         }
9715                         if (te->set->ops->commit &&
9716                             list_empty(&te->set->pending_update)) {
9717                                 list_add_tail(&te->set->pending_update,
9718                                               &set_update_list);
9719                         }
9720                         break;
9721                 case NFT_MSG_NEWOBJ:
9722                         if (nft_trans_obj_update(trans)) {
9723                                 nft_obj_commit_update(trans);
9724                                 nf_tables_obj_notify(&trans->ctx,
9725                                                      nft_trans_obj(trans),
9726                                                      NFT_MSG_NEWOBJ);
9727                         } else {
9728                                 nft_clear(net, nft_trans_obj(trans));
9729                                 nf_tables_obj_notify(&trans->ctx,
9730                                                      nft_trans_obj(trans),
9731                                                      NFT_MSG_NEWOBJ);
9732                                 nft_trans_destroy(trans);
9733                         }
9734                         break;
9735                 case NFT_MSG_DELOBJ:
9736                 case NFT_MSG_DESTROYOBJ:
9737                         nft_obj_del(nft_trans_obj(trans));
9738                         nf_tables_obj_notify(&trans->ctx, nft_trans_obj(trans),
9739                                              trans->msg_type);
9740                         break;
9741                 case NFT_MSG_NEWFLOWTABLE:
9742                         if (nft_trans_flowtable_update(trans)) {
9743                                 nft_trans_flowtable(trans)->data.flags =
9744                                         nft_trans_flowtable_flags(trans);
9745                                 nf_tables_flowtable_notify(&trans->ctx,
9746                                                            nft_trans_flowtable(trans),
9747                                                            &nft_trans_flowtable_hooks(trans),
9748                                                            NFT_MSG_NEWFLOWTABLE);
9749                                 list_splice(&nft_trans_flowtable_hooks(trans),
9750                                             &nft_trans_flowtable(trans)->hook_list);
9751                         } else {
9752                                 nft_clear(net, nft_trans_flowtable(trans));
9753                                 nf_tables_flowtable_notify(&trans->ctx,
9754                                                            nft_trans_flowtable(trans),
9755                                                            NULL,
9756                                                            NFT_MSG_NEWFLOWTABLE);
9757                         }
9758                         nft_trans_destroy(trans);
9759                         break;
9760                 case NFT_MSG_DELFLOWTABLE:
9761                 case NFT_MSG_DESTROYFLOWTABLE:
9762                         if (nft_trans_flowtable_update(trans)) {
9763                                 nf_tables_flowtable_notify(&trans->ctx,
9764                                                            nft_trans_flowtable(trans),
9765                                                            &nft_trans_flowtable_hooks(trans),
9766                                                            trans->msg_type);
9767                                 nft_unregister_flowtable_net_hooks(net,
9768                                                                    &nft_trans_flowtable_hooks(trans));
9769                         } else {
9770                                 list_del_rcu(&nft_trans_flowtable(trans)->list);
9771                                 nf_tables_flowtable_notify(&trans->ctx,
9772                                                            nft_trans_flowtable(trans),
9773                                                            NULL,
9774                                                            trans->msg_type);
9775                                 nft_unregister_flowtable_net_hooks(net,
9776                                                 &nft_trans_flowtable(trans)->hook_list);
9777                         }
9778                         break;
9779                 }
9780         }
9781
9782         nft_set_commit_update(&set_update_list);
9783
9784         nft_commit_notify(net, NETLINK_CB(skb).portid);
9785         nf_tables_gen_notify(net, skb, NFT_MSG_NEWGEN);
9786         nf_tables_commit_audit_log(&adl, nft_net->base_seq);
9787         nf_tables_commit_release(net);
9788
9789         return 0;
9790 }
9791
9792 static void nf_tables_module_autoload(struct net *net)
9793 {
9794         struct nftables_pernet *nft_net = nft_pernet(net);
9795         struct nft_module_request *req, *next;
9796         LIST_HEAD(module_list);
9797
9798         list_splice_init(&nft_net->module_list, &module_list);
9799         mutex_unlock(&nft_net->commit_mutex);
9800         list_for_each_entry_safe(req, next, &module_list, list) {
9801                 request_module("%s", req->module);
9802                 req->done = true;
9803         }
9804         mutex_lock(&nft_net->commit_mutex);
9805         list_splice(&module_list, &nft_net->module_list);
9806 }
9807
9808 static void nf_tables_abort_release(struct nft_trans *trans)
9809 {
9810         switch (trans->msg_type) {
9811         case NFT_MSG_NEWTABLE:
9812                 nf_tables_table_destroy(&trans->ctx);
9813                 break;
9814         case NFT_MSG_NEWCHAIN:
9815                 if (nft_trans_chain_update(trans))
9816                         nft_hooks_destroy(&nft_trans_chain_hooks(trans));
9817                 else
9818                         nf_tables_chain_destroy(&trans->ctx);
9819                 break;
9820         case NFT_MSG_NEWRULE:
9821                 nf_tables_rule_destroy(&trans->ctx, nft_trans_rule(trans));
9822                 break;
9823         case NFT_MSG_NEWSET:
9824                 nft_set_destroy(&trans->ctx, nft_trans_set(trans));
9825                 break;
9826         case NFT_MSG_NEWSETELEM:
9827                 nft_set_elem_destroy(nft_trans_elem_set(trans),
9828                                      nft_trans_elem(trans).priv, true);
9829                 break;
9830         case NFT_MSG_NEWOBJ:
9831                 nft_obj_destroy(&trans->ctx, nft_trans_obj(trans));
9832                 break;
9833         case NFT_MSG_NEWFLOWTABLE:
9834                 if (nft_trans_flowtable_update(trans))
9835                         nft_hooks_destroy(&nft_trans_flowtable_hooks(trans));
9836                 else
9837                         nf_tables_flowtable_destroy(nft_trans_flowtable(trans));
9838                 break;
9839         }
9840         kfree(trans);
9841 }
9842
9843 static void nft_set_abort_update(struct list_head *set_update_list)
9844 {
9845         struct nft_set *set, *next;
9846
9847         list_for_each_entry_safe(set, next, set_update_list, pending_update) {
9848                 list_del_init(&set->pending_update);
9849
9850                 if (!set->ops->abort)
9851                         continue;
9852
9853                 set->ops->abort(set);
9854         }
9855 }
9856
9857 static int __nf_tables_abort(struct net *net, enum nfnl_abort_action action)
9858 {
9859         struct nftables_pernet *nft_net = nft_pernet(net);
9860         struct nft_trans *trans, *next;
9861         LIST_HEAD(set_update_list);
9862         struct nft_trans_elem *te;
9863
9864         if (action == NFNL_ABORT_VALIDATE &&
9865             nf_tables_validate(net) < 0)
9866                 return -EAGAIN;
9867
9868         list_for_each_entry_safe_reverse(trans, next, &nft_net->commit_list,
9869                                          list) {
9870                 switch (trans->msg_type) {
9871                 case NFT_MSG_NEWTABLE:
9872                         if (nft_trans_table_update(trans)) {
9873                                 if (!(trans->ctx.table->flags & __NFT_TABLE_F_UPDATE)) {
9874                                         nft_trans_destroy(trans);
9875                                         break;
9876                                 }
9877                                 if (trans->ctx.table->flags & __NFT_TABLE_F_WAS_DORMANT) {
9878                                         nf_tables_table_disable(net, trans->ctx.table);
9879                                         trans->ctx.table->flags |= NFT_TABLE_F_DORMANT;
9880                                 } else if (trans->ctx.table->flags & __NFT_TABLE_F_WAS_AWAKEN) {
9881                                         trans->ctx.table->flags &= ~NFT_TABLE_F_DORMANT;
9882                                 }
9883                                 trans->ctx.table->flags &= ~__NFT_TABLE_F_UPDATE;
9884                                 nft_trans_destroy(trans);
9885                         } else {
9886                                 list_del_rcu(&trans->ctx.table->list);
9887                         }
9888                         break;
9889                 case NFT_MSG_DELTABLE:
9890                 case NFT_MSG_DESTROYTABLE:
9891                         nft_clear(trans->ctx.net, trans->ctx.table);
9892                         nft_trans_destroy(trans);
9893                         break;
9894                 case NFT_MSG_NEWCHAIN:
9895                         if (nft_trans_chain_update(trans)) {
9896                                 nft_netdev_unregister_hooks(net,
9897                                                             &nft_trans_chain_hooks(trans),
9898                                                             true);
9899                                 free_percpu(nft_trans_chain_stats(trans));
9900                                 kfree(nft_trans_chain_name(trans));
9901                                 nft_trans_destroy(trans);
9902                         } else {
9903                                 if (nft_trans_chain_bound(trans)) {
9904                                         nft_trans_destroy(trans);
9905                                         break;
9906                                 }
9907                                 trans->ctx.table->use--;
9908                                 nft_chain_del(trans->ctx.chain);
9909                                 nf_tables_unregister_hook(trans->ctx.net,
9910                                                           trans->ctx.table,
9911                                                           trans->ctx.chain);
9912                         }
9913                         break;
9914                 case NFT_MSG_DELCHAIN:
9915                 case NFT_MSG_DESTROYCHAIN:
9916                         if (nft_trans_chain_update(trans)) {
9917                                 list_splice(&nft_trans_chain_hooks(trans),
9918                                             &nft_trans_basechain(trans)->hook_list);
9919                         } else {
9920                                 trans->ctx.table->use++;
9921                                 nft_clear(trans->ctx.net, trans->ctx.chain);
9922                         }
9923                         nft_trans_destroy(trans);
9924                         break;
9925                 case NFT_MSG_NEWRULE:
9926                         if (nft_trans_rule_bound(trans)) {
9927                                 nft_trans_destroy(trans);
9928                                 break;
9929                         }
9930                         trans->ctx.chain->use--;
9931                         list_del_rcu(&nft_trans_rule(trans)->list);
9932                         nft_rule_expr_deactivate(&trans->ctx,
9933                                                  nft_trans_rule(trans),
9934                                                  NFT_TRANS_ABORT);
9935                         if (trans->ctx.chain->flags & NFT_CHAIN_HW_OFFLOAD)
9936                                 nft_flow_rule_destroy(nft_trans_flow_rule(trans));
9937                         break;
9938                 case NFT_MSG_DELRULE:
9939                 case NFT_MSG_DESTROYRULE:
9940                         trans->ctx.chain->use++;
9941                         nft_clear(trans->ctx.net, nft_trans_rule(trans));
9942                         nft_rule_expr_activate(&trans->ctx, nft_trans_rule(trans));
9943                         if (trans->ctx.chain->flags & NFT_CHAIN_HW_OFFLOAD)
9944                                 nft_flow_rule_destroy(nft_trans_flow_rule(trans));
9945
9946                         nft_trans_destroy(trans);
9947                         break;
9948                 case NFT_MSG_NEWSET:
9949                         if (nft_trans_set_update(trans)) {
9950                                 nft_trans_destroy(trans);
9951                                 break;
9952                         }
9953                         trans->ctx.table->use--;
9954                         if (nft_trans_set_bound(trans)) {
9955                                 nft_trans_destroy(trans);
9956                                 break;
9957                         }
9958                         list_del_rcu(&nft_trans_set(trans)->list);
9959                         break;
9960                 case NFT_MSG_DELSET:
9961                 case NFT_MSG_DESTROYSET:
9962                         trans->ctx.table->use++;
9963                         nft_clear(trans->ctx.net, nft_trans_set(trans));
9964                         if (nft_trans_set(trans)->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
9965                                 nft_map_activate(&trans->ctx, nft_trans_set(trans));
9966
9967                         nft_trans_destroy(trans);
9968                         break;
9969                 case NFT_MSG_NEWSETELEM:
9970                         if (nft_trans_elem_set_bound(trans)) {
9971                                 nft_trans_destroy(trans);
9972                                 break;
9973                         }
9974                         te = (struct nft_trans_elem *)trans->data;
9975                         nft_setelem_remove(net, te->set, &te->elem);
9976                         if (!nft_setelem_is_catchall(te->set, &te->elem))
9977                                 atomic_dec(&te->set->nelems);
9978
9979                         if (te->set->ops->abort &&
9980                             list_empty(&te->set->pending_update)) {
9981                                 list_add_tail(&te->set->pending_update,
9982                                               &set_update_list);
9983                         }
9984                         break;
9985                 case NFT_MSG_DELSETELEM:
9986                 case NFT_MSG_DESTROYSETELEM:
9987                         te = (struct nft_trans_elem *)trans->data;
9988
9989                         nft_setelem_data_activate(net, te->set, &te->elem);
9990                         nft_setelem_activate(net, te->set, &te->elem);
9991                         if (!nft_setelem_is_catchall(te->set, &te->elem))
9992                                 te->set->ndeact--;
9993
9994                         if (te->set->ops->abort &&
9995                             list_empty(&te->set->pending_update)) {
9996                                 list_add_tail(&te->set->pending_update,
9997                                               &set_update_list);
9998                         }
9999                         nft_trans_destroy(trans);
10000                         break;
10001                 case NFT_MSG_NEWOBJ:
10002                         if (nft_trans_obj_update(trans)) {
10003                                 nft_obj_destroy(&trans->ctx, nft_trans_obj_newobj(trans));
10004                                 nft_trans_destroy(trans);
10005                         } else {
10006                                 trans->ctx.table->use--;
10007                                 nft_obj_del(nft_trans_obj(trans));
10008                         }
10009                         break;
10010                 case NFT_MSG_DELOBJ:
10011                 case NFT_MSG_DESTROYOBJ:
10012                         trans->ctx.table->use++;
10013                         nft_clear(trans->ctx.net, nft_trans_obj(trans));
10014                         nft_trans_destroy(trans);
10015                         break;
10016                 case NFT_MSG_NEWFLOWTABLE:
10017                         if (nft_trans_flowtable_update(trans)) {
10018                                 nft_unregister_flowtable_net_hooks(net,
10019                                                 &nft_trans_flowtable_hooks(trans));
10020                         } else {
10021                                 trans->ctx.table->use--;
10022                                 list_del_rcu(&nft_trans_flowtable(trans)->list);
10023                                 nft_unregister_flowtable_net_hooks(net,
10024                                                 &nft_trans_flowtable(trans)->hook_list);
10025                         }
10026                         break;
10027                 case NFT_MSG_DELFLOWTABLE:
10028                 case NFT_MSG_DESTROYFLOWTABLE:
10029                         if (nft_trans_flowtable_update(trans)) {
10030                                 list_splice(&nft_trans_flowtable_hooks(trans),
10031                                             &nft_trans_flowtable(trans)->hook_list);
10032                         } else {
10033                                 trans->ctx.table->use++;
10034                                 nft_clear(trans->ctx.net, nft_trans_flowtable(trans));
10035                         }
10036                         nft_trans_destroy(trans);
10037                         break;
10038                 }
10039         }
10040
10041         nft_set_abort_update(&set_update_list);
10042
10043         synchronize_rcu();
10044
10045         list_for_each_entry_safe_reverse(trans, next,
10046                                          &nft_net->commit_list, list) {
10047                 nft_trans_list_del(trans);
10048                 nf_tables_abort_release(trans);
10049         }
10050
10051         if (action == NFNL_ABORT_AUTOLOAD)
10052                 nf_tables_module_autoload(net);
10053         else
10054                 nf_tables_module_autoload_cleanup(net);
10055
10056         return 0;
10057 }
10058
10059 static int nf_tables_abort(struct net *net, struct sk_buff *skb,
10060                            enum nfnl_abort_action action)
10061 {
10062         struct nftables_pernet *nft_net = nft_pernet(net);
10063         int ret = __nf_tables_abort(net, action);
10064
10065         mutex_unlock(&nft_net->commit_mutex);
10066
10067         return ret;
10068 }
10069
10070 static bool nf_tables_valid_genid(struct net *net, u32 genid)
10071 {
10072         struct nftables_pernet *nft_net = nft_pernet(net);
10073         bool genid_ok;
10074
10075         mutex_lock(&nft_net->commit_mutex);
10076
10077         genid_ok = genid == 0 || nft_net->base_seq == genid;
10078         if (!genid_ok)
10079                 mutex_unlock(&nft_net->commit_mutex);
10080
10081         /* else, commit mutex has to be released by commit or abort function */
10082         return genid_ok;
10083 }
10084
10085 static const struct nfnetlink_subsystem nf_tables_subsys = {
10086         .name           = "nf_tables",
10087         .subsys_id      = NFNL_SUBSYS_NFTABLES,
10088         .cb_count       = NFT_MSG_MAX,
10089         .cb             = nf_tables_cb,
10090         .commit         = nf_tables_commit,
10091         .abort          = nf_tables_abort,
10092         .valid_genid    = nf_tables_valid_genid,
10093         .owner          = THIS_MODULE,
10094 };
10095
10096 int nft_chain_validate_dependency(const struct nft_chain *chain,
10097                                   enum nft_chain_types type)
10098 {
10099         const struct nft_base_chain *basechain;
10100
10101         if (nft_is_base_chain(chain)) {
10102                 basechain = nft_base_chain(chain);
10103                 if (basechain->type->type != type)
10104                         return -EOPNOTSUPP;
10105         }
10106         return 0;
10107 }
10108 EXPORT_SYMBOL_GPL(nft_chain_validate_dependency);
10109
10110 int nft_chain_validate_hooks(const struct nft_chain *chain,
10111                              unsigned int hook_flags)
10112 {
10113         struct nft_base_chain *basechain;
10114
10115         if (nft_is_base_chain(chain)) {
10116                 basechain = nft_base_chain(chain);
10117
10118                 if ((1 << basechain->ops.hooknum) & hook_flags)
10119                         return 0;
10120
10121                 return -EOPNOTSUPP;
10122         }
10123
10124         return 0;
10125 }
10126 EXPORT_SYMBOL_GPL(nft_chain_validate_hooks);
10127
10128 /*
10129  * Loop detection - walk through the ruleset beginning at the destination chain
10130  * of a new jump until either the source chain is reached (loop) or all
10131  * reachable chains have been traversed.
10132  *
10133  * The loop check is performed whenever a new jump verdict is added to an
10134  * expression or verdict map or a verdict map is bound to a new chain.
10135  */
10136
10137 static int nf_tables_check_loops(const struct nft_ctx *ctx,
10138                                  const struct nft_chain *chain);
10139
10140 static int nft_check_loops(const struct nft_ctx *ctx,
10141                            const struct nft_set_ext *ext)
10142 {
10143         const struct nft_data *data;
10144         int ret;
10145
10146         data = nft_set_ext_data(ext);
10147         switch (data->verdict.code) {
10148         case NFT_JUMP:
10149         case NFT_GOTO:
10150                 ret = nf_tables_check_loops(ctx, data->verdict.chain);
10151                 break;
10152         default:
10153                 ret = 0;
10154                 break;
10155         }
10156
10157         return ret;
10158 }
10159
10160 static int nf_tables_loop_check_setelem(const struct nft_ctx *ctx,
10161                                         struct nft_set *set,
10162                                         const struct nft_set_iter *iter,
10163                                         struct nft_set_elem *elem)
10164 {
10165         const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
10166
10167         if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
10168             *nft_set_ext_flags(ext) & NFT_SET_ELEM_INTERVAL_END)
10169                 return 0;
10170
10171         return nft_check_loops(ctx, ext);
10172 }
10173
10174 static int nft_set_catchall_loops(const struct nft_ctx *ctx,
10175                                   struct nft_set *set)
10176 {
10177         u8 genmask = nft_genmask_next(ctx->net);
10178         struct nft_set_elem_catchall *catchall;
10179         struct nft_set_ext *ext;
10180         int ret = 0;
10181
10182         list_for_each_entry_rcu(catchall, &set->catchall_list, list) {
10183                 ext = nft_set_elem_ext(set, catchall->elem);
10184                 if (!nft_set_elem_active(ext, genmask))
10185                         continue;
10186
10187                 ret = nft_check_loops(ctx, ext);
10188                 if (ret < 0)
10189                         return ret;
10190         }
10191
10192         return ret;
10193 }
10194
10195 static int nf_tables_check_loops(const struct nft_ctx *ctx,
10196                                  const struct nft_chain *chain)
10197 {
10198         const struct nft_rule *rule;
10199         const struct nft_expr *expr, *last;
10200         struct nft_set *set;
10201         struct nft_set_binding *binding;
10202         struct nft_set_iter iter;
10203
10204         if (ctx->chain == chain)
10205                 return -ELOOP;
10206
10207         list_for_each_entry(rule, &chain->rules, list) {
10208                 nft_rule_for_each_expr(expr, last, rule) {
10209                         struct nft_immediate_expr *priv;
10210                         const struct nft_data *data;
10211                         int err;
10212
10213                         if (strcmp(expr->ops->type->name, "immediate"))
10214                                 continue;
10215
10216                         priv = nft_expr_priv(expr);
10217                         if (priv->dreg != NFT_REG_VERDICT)
10218                                 continue;
10219
10220                         data = &priv->data;
10221                         switch (data->verdict.code) {
10222                         case NFT_JUMP:
10223                         case NFT_GOTO:
10224                                 err = nf_tables_check_loops(ctx,
10225                                                         data->verdict.chain);
10226                                 if (err < 0)
10227                                         return err;
10228                                 break;
10229                         default:
10230                                 break;
10231                         }
10232                 }
10233         }
10234
10235         list_for_each_entry(set, &ctx->table->sets, list) {
10236                 if (!nft_is_active_next(ctx->net, set))
10237                         continue;
10238                 if (!(set->flags & NFT_SET_MAP) ||
10239                     set->dtype != NFT_DATA_VERDICT)
10240                         continue;
10241
10242                 list_for_each_entry(binding, &set->bindings, list) {
10243                         if (!(binding->flags & NFT_SET_MAP) ||
10244                             binding->chain != chain)
10245                                 continue;
10246
10247                         iter.genmask    = nft_genmask_next(ctx->net);
10248                         iter.skip       = 0;
10249                         iter.count      = 0;
10250                         iter.err        = 0;
10251                         iter.fn         = nf_tables_loop_check_setelem;
10252
10253                         set->ops->walk(ctx, set, &iter);
10254                         if (!iter.err)
10255                                 iter.err = nft_set_catchall_loops(ctx, set);
10256
10257                         if (iter.err < 0)
10258                                 return iter.err;
10259                 }
10260         }
10261
10262         return 0;
10263 }
10264
10265 /**
10266  *      nft_parse_u32_check - fetch u32 attribute and check for maximum value
10267  *
10268  *      @attr: netlink attribute to fetch value from
10269  *      @max: maximum value to be stored in dest
10270  *      @dest: pointer to the variable
10271  *
10272  *      Parse, check and store a given u32 netlink attribute into variable.
10273  *      This function returns -ERANGE if the value goes over maximum value.
10274  *      Otherwise a 0 is returned and the attribute value is stored in the
10275  *      destination variable.
10276  */
10277 int nft_parse_u32_check(const struct nlattr *attr, int max, u32 *dest)
10278 {
10279         u32 val;
10280
10281         val = ntohl(nla_get_be32(attr));
10282         if (val > max)
10283                 return -ERANGE;
10284
10285         *dest = val;
10286         return 0;
10287 }
10288 EXPORT_SYMBOL_GPL(nft_parse_u32_check);
10289
10290 static int nft_parse_register(const struct nlattr *attr, u32 *preg)
10291 {
10292         unsigned int reg;
10293
10294         reg = ntohl(nla_get_be32(attr));
10295         switch (reg) {
10296         case NFT_REG_VERDICT...NFT_REG_4:
10297                 *preg = reg * NFT_REG_SIZE / NFT_REG32_SIZE;
10298                 break;
10299         case NFT_REG32_00...NFT_REG32_15:
10300                 *preg = reg + NFT_REG_SIZE / NFT_REG32_SIZE - NFT_REG32_00;
10301                 break;
10302         default:
10303                 return -ERANGE;
10304         }
10305
10306         return 0;
10307 }
10308
10309 /**
10310  *      nft_dump_register - dump a register value to a netlink attribute
10311  *
10312  *      @skb: socket buffer
10313  *      @attr: attribute number
10314  *      @reg: register number
10315  *
10316  *      Construct a netlink attribute containing the register number. For
10317  *      compatibility reasons, register numbers being a multiple of 4 are
10318  *      translated to the corresponding 128 bit register numbers.
10319  */
10320 int nft_dump_register(struct sk_buff *skb, unsigned int attr, unsigned int reg)
10321 {
10322         if (reg % (NFT_REG_SIZE / NFT_REG32_SIZE) == 0)
10323                 reg = reg / (NFT_REG_SIZE / NFT_REG32_SIZE);
10324         else
10325                 reg = reg - NFT_REG_SIZE / NFT_REG32_SIZE + NFT_REG32_00;
10326
10327         return nla_put_be32(skb, attr, htonl(reg));
10328 }
10329 EXPORT_SYMBOL_GPL(nft_dump_register);
10330
10331 static int nft_validate_register_load(enum nft_registers reg, unsigned int len)
10332 {
10333         if (reg < NFT_REG_1 * NFT_REG_SIZE / NFT_REG32_SIZE)
10334                 return -EINVAL;
10335         if (len == 0)
10336                 return -EINVAL;
10337         if (reg * NFT_REG32_SIZE + len > sizeof_field(struct nft_regs, data))
10338                 return -ERANGE;
10339
10340         return 0;
10341 }
10342
10343 int nft_parse_register_load(const struct nlattr *attr, u8 *sreg, u32 len)
10344 {
10345         u32 reg;
10346         int err;
10347
10348         err = nft_parse_register(attr, &reg);
10349         if (err < 0)
10350                 return err;
10351
10352         err = nft_validate_register_load(reg, len);
10353         if (err < 0)
10354                 return err;
10355
10356         *sreg = reg;
10357         return 0;
10358 }
10359 EXPORT_SYMBOL_GPL(nft_parse_register_load);
10360
10361 static int nft_validate_register_store(const struct nft_ctx *ctx,
10362                                        enum nft_registers reg,
10363                                        const struct nft_data *data,
10364                                        enum nft_data_types type,
10365                                        unsigned int len)
10366 {
10367         int err;
10368
10369         switch (reg) {
10370         case NFT_REG_VERDICT:
10371                 if (type != NFT_DATA_VERDICT)
10372                         return -EINVAL;
10373
10374                 if (data != NULL &&
10375                     (data->verdict.code == NFT_GOTO ||
10376                      data->verdict.code == NFT_JUMP)) {
10377                         err = nf_tables_check_loops(ctx, data->verdict.chain);
10378                         if (err < 0)
10379                                 return err;
10380                 }
10381
10382                 return 0;
10383         default:
10384                 if (reg < NFT_REG_1 * NFT_REG_SIZE / NFT_REG32_SIZE)
10385                         return -EINVAL;
10386                 if (len == 0)
10387                         return -EINVAL;
10388                 if (reg * NFT_REG32_SIZE + len >
10389                     sizeof_field(struct nft_regs, data))
10390                         return -ERANGE;
10391
10392                 if (data != NULL && type != NFT_DATA_VALUE)
10393                         return -EINVAL;
10394                 return 0;
10395         }
10396 }
10397
10398 int nft_parse_register_store(const struct nft_ctx *ctx,
10399                              const struct nlattr *attr, u8 *dreg,
10400                              const struct nft_data *data,
10401                              enum nft_data_types type, unsigned int len)
10402 {
10403         int err;
10404         u32 reg;
10405
10406         err = nft_parse_register(attr, &reg);
10407         if (err < 0)
10408                 return err;
10409
10410         err = nft_validate_register_store(ctx, reg, data, type, len);
10411         if (err < 0)
10412                 return err;
10413
10414         *dreg = reg;
10415         return 0;
10416 }
10417 EXPORT_SYMBOL_GPL(nft_parse_register_store);
10418
10419 static const struct nla_policy nft_verdict_policy[NFTA_VERDICT_MAX + 1] = {
10420         [NFTA_VERDICT_CODE]     = { .type = NLA_U32 },
10421         [NFTA_VERDICT_CHAIN]    = { .type = NLA_STRING,
10422                                     .len = NFT_CHAIN_MAXNAMELEN - 1 },
10423         [NFTA_VERDICT_CHAIN_ID] = { .type = NLA_U32 },
10424 };
10425
10426 static int nft_verdict_init(const struct nft_ctx *ctx, struct nft_data *data,
10427                             struct nft_data_desc *desc, const struct nlattr *nla)
10428 {
10429         u8 genmask = nft_genmask_next(ctx->net);
10430         struct nlattr *tb[NFTA_VERDICT_MAX + 1];
10431         struct nft_chain *chain;
10432         int err;
10433
10434         err = nla_parse_nested_deprecated(tb, NFTA_VERDICT_MAX, nla,
10435                                           nft_verdict_policy, NULL);
10436         if (err < 0)
10437                 return err;
10438
10439         if (!tb[NFTA_VERDICT_CODE])
10440                 return -EINVAL;
10441         data->verdict.code = ntohl(nla_get_be32(tb[NFTA_VERDICT_CODE]));
10442
10443         switch (data->verdict.code) {
10444         default:
10445                 switch (data->verdict.code & NF_VERDICT_MASK) {
10446                 case NF_ACCEPT:
10447                 case NF_DROP:
10448                 case NF_QUEUE:
10449                         break;
10450                 default:
10451                         return -EINVAL;
10452                 }
10453                 fallthrough;
10454         case NFT_CONTINUE:
10455         case NFT_BREAK:
10456         case NFT_RETURN:
10457                 break;
10458         case NFT_JUMP:
10459         case NFT_GOTO:
10460                 if (tb[NFTA_VERDICT_CHAIN]) {
10461                         chain = nft_chain_lookup(ctx->net, ctx->table,
10462                                                  tb[NFTA_VERDICT_CHAIN],
10463                                                  genmask);
10464                 } else if (tb[NFTA_VERDICT_CHAIN_ID]) {
10465                         chain = nft_chain_lookup_byid(ctx->net, ctx->table,
10466                                                       tb[NFTA_VERDICT_CHAIN_ID]);
10467                         if (IS_ERR(chain))
10468                                 return PTR_ERR(chain);
10469                 } else {
10470                         return -EINVAL;
10471                 }
10472
10473                 if (IS_ERR(chain))
10474                         return PTR_ERR(chain);
10475                 if (nft_is_base_chain(chain))
10476                         return -EOPNOTSUPP;
10477                 if (nft_chain_is_bound(chain))
10478                         return -EINVAL;
10479                 if (desc->flags & NFT_DATA_DESC_SETELEM &&
10480                     chain->flags & NFT_CHAIN_BINDING)
10481                         return -EINVAL;
10482
10483                 chain->use++;
10484                 data->verdict.chain = chain;
10485                 break;
10486         }
10487
10488         desc->len = sizeof(data->verdict);
10489
10490         return 0;
10491 }
10492
10493 static void nft_verdict_uninit(const struct nft_data *data)
10494 {
10495         struct nft_chain *chain;
10496
10497         switch (data->verdict.code) {
10498         case NFT_JUMP:
10499         case NFT_GOTO:
10500                 chain = data->verdict.chain;
10501                 chain->use--;
10502                 break;
10503         }
10504 }
10505
10506 int nft_verdict_dump(struct sk_buff *skb, int type, const struct nft_verdict *v)
10507 {
10508         struct nlattr *nest;
10509
10510         nest = nla_nest_start_noflag(skb, type);
10511         if (!nest)
10512                 goto nla_put_failure;
10513
10514         if (nla_put_be32(skb, NFTA_VERDICT_CODE, htonl(v->code)))
10515                 goto nla_put_failure;
10516
10517         switch (v->code) {
10518         case NFT_JUMP:
10519         case NFT_GOTO:
10520                 if (nla_put_string(skb, NFTA_VERDICT_CHAIN,
10521                                    v->chain->name))
10522                         goto nla_put_failure;
10523         }
10524         nla_nest_end(skb, nest);
10525         return 0;
10526
10527 nla_put_failure:
10528         return -1;
10529 }
10530
10531 static int nft_value_init(const struct nft_ctx *ctx,
10532                           struct nft_data *data, struct nft_data_desc *desc,
10533                           const struct nlattr *nla)
10534 {
10535         unsigned int len;
10536
10537         len = nla_len(nla);
10538         if (len == 0)
10539                 return -EINVAL;
10540         if (len > desc->size)
10541                 return -EOVERFLOW;
10542         if (desc->len) {
10543                 if (len != desc->len)
10544                         return -EINVAL;
10545         } else {
10546                 desc->len = len;
10547         }
10548
10549         nla_memcpy(data->data, nla, len);
10550
10551         return 0;
10552 }
10553
10554 static int nft_value_dump(struct sk_buff *skb, const struct nft_data *data,
10555                           unsigned int len)
10556 {
10557         return nla_put(skb, NFTA_DATA_VALUE, len, data->data);
10558 }
10559
10560 static const struct nla_policy nft_data_policy[NFTA_DATA_MAX + 1] = {
10561         [NFTA_DATA_VALUE]       = { .type = NLA_BINARY },
10562         [NFTA_DATA_VERDICT]     = { .type = NLA_NESTED },
10563 };
10564
10565 /**
10566  *      nft_data_init - parse nf_tables data netlink attributes
10567  *
10568  *      @ctx: context of the expression using the data
10569  *      @data: destination struct nft_data
10570  *      @desc: data description
10571  *      @nla: netlink attribute containing data
10572  *
10573  *      Parse the netlink data attributes and initialize a struct nft_data.
10574  *      The type and length of data are returned in the data description.
10575  *
10576  *      The caller can indicate that it only wants to accept data of type
10577  *      NFT_DATA_VALUE by passing NULL for the ctx argument.
10578  */
10579 int nft_data_init(const struct nft_ctx *ctx, struct nft_data *data,
10580                   struct nft_data_desc *desc, const struct nlattr *nla)
10581 {
10582         struct nlattr *tb[NFTA_DATA_MAX + 1];
10583         int err;
10584
10585         if (WARN_ON_ONCE(!desc->size))
10586                 return -EINVAL;
10587
10588         err = nla_parse_nested_deprecated(tb, NFTA_DATA_MAX, nla,
10589                                           nft_data_policy, NULL);
10590         if (err < 0)
10591                 return err;
10592
10593         if (tb[NFTA_DATA_VALUE]) {
10594                 if (desc->type != NFT_DATA_VALUE)
10595                         return -EINVAL;
10596
10597                 err = nft_value_init(ctx, data, desc, tb[NFTA_DATA_VALUE]);
10598         } else if (tb[NFTA_DATA_VERDICT] && ctx != NULL) {
10599                 if (desc->type != NFT_DATA_VERDICT)
10600                         return -EINVAL;
10601
10602                 err = nft_verdict_init(ctx, data, desc, tb[NFTA_DATA_VERDICT]);
10603         } else {
10604                 err = -EINVAL;
10605         }
10606
10607         return err;
10608 }
10609 EXPORT_SYMBOL_GPL(nft_data_init);
10610
10611 /**
10612  *      nft_data_release - release a nft_data item
10613  *
10614  *      @data: struct nft_data to release
10615  *      @type: type of data
10616  *
10617  *      Release a nft_data item. NFT_DATA_VALUE types can be silently discarded,
10618  *      all others need to be released by calling this function.
10619  */
10620 void nft_data_release(const struct nft_data *data, enum nft_data_types type)
10621 {
10622         if (type < NFT_DATA_VERDICT)
10623                 return;
10624         switch (type) {
10625         case NFT_DATA_VERDICT:
10626                 return nft_verdict_uninit(data);
10627         default:
10628                 WARN_ON(1);
10629         }
10630 }
10631 EXPORT_SYMBOL_GPL(nft_data_release);
10632
10633 int nft_data_dump(struct sk_buff *skb, int attr, const struct nft_data *data,
10634                   enum nft_data_types type, unsigned int len)
10635 {
10636         struct nlattr *nest;
10637         int err;
10638
10639         nest = nla_nest_start_noflag(skb, attr);
10640         if (nest == NULL)
10641                 return -1;
10642
10643         switch (type) {
10644         case NFT_DATA_VALUE:
10645                 err = nft_value_dump(skb, data, len);
10646                 break;
10647         case NFT_DATA_VERDICT:
10648                 err = nft_verdict_dump(skb, NFTA_DATA_VERDICT, &data->verdict);
10649                 break;
10650         default:
10651                 err = -EINVAL;
10652                 WARN_ON(1);
10653         }
10654
10655         nla_nest_end(skb, nest);
10656         return err;
10657 }
10658 EXPORT_SYMBOL_GPL(nft_data_dump);
10659
10660 int __nft_release_basechain(struct nft_ctx *ctx)
10661 {
10662         struct nft_rule *rule, *nr;
10663
10664         if (WARN_ON(!nft_is_base_chain(ctx->chain)))
10665                 return 0;
10666
10667         nf_tables_unregister_hook(ctx->net, ctx->chain->table, ctx->chain);
10668         list_for_each_entry_safe(rule, nr, &ctx->chain->rules, list) {
10669                 list_del(&rule->list);
10670                 ctx->chain->use--;
10671                 nf_tables_rule_release(ctx, rule);
10672         }
10673         nft_chain_del(ctx->chain);
10674         ctx->table->use--;
10675         nf_tables_chain_destroy(ctx);
10676
10677         return 0;
10678 }
10679 EXPORT_SYMBOL_GPL(__nft_release_basechain);
10680
10681 static void __nft_release_hook(struct net *net, struct nft_table *table)
10682 {
10683         struct nft_flowtable *flowtable;
10684         struct nft_chain *chain;
10685
10686         list_for_each_entry(chain, &table->chains, list)
10687                 __nf_tables_unregister_hook(net, table, chain, true);
10688         list_for_each_entry(flowtable, &table->flowtables, list)
10689                 __nft_unregister_flowtable_net_hooks(net, &flowtable->hook_list,
10690                                                      true);
10691 }
10692
10693 static void __nft_release_hooks(struct net *net)
10694 {
10695         struct nftables_pernet *nft_net = nft_pernet(net);
10696         struct nft_table *table;
10697
10698         list_for_each_entry(table, &nft_net->tables, list) {
10699                 if (nft_table_has_owner(table))
10700                         continue;
10701
10702                 __nft_release_hook(net, table);
10703         }
10704 }
10705
10706 static void __nft_release_table(struct net *net, struct nft_table *table)
10707 {
10708         struct nft_flowtable *flowtable, *nf;
10709         struct nft_chain *chain, *nc;
10710         struct nft_object *obj, *ne;
10711         struct nft_rule *rule, *nr;
10712         struct nft_set *set, *ns;
10713         struct nft_ctx ctx = {
10714                 .net    = net,
10715                 .family = NFPROTO_NETDEV,
10716         };
10717
10718         ctx.family = table->family;
10719         ctx.table = table;
10720         list_for_each_entry(chain, &table->chains, list) {
10721                 ctx.chain = chain;
10722                 list_for_each_entry_safe(rule, nr, &chain->rules, list) {
10723                         list_del(&rule->list);
10724                         chain->use--;
10725                         nf_tables_rule_release(&ctx, rule);
10726                 }
10727         }
10728         list_for_each_entry_safe(flowtable, nf, &table->flowtables, list) {
10729                 list_del(&flowtable->list);
10730                 table->use--;
10731                 nf_tables_flowtable_destroy(flowtable);
10732         }
10733         list_for_each_entry_safe(set, ns, &table->sets, list) {
10734                 list_del(&set->list);
10735                 table->use--;
10736                 if (set->flags & (NFT_SET_MAP | NFT_SET_OBJECT))
10737                         nft_map_deactivate(&ctx, set);
10738
10739                 nft_set_destroy(&ctx, set);
10740         }
10741         list_for_each_entry_safe(obj, ne, &table->objects, list) {
10742                 nft_obj_del(obj);
10743                 table->use--;
10744                 nft_obj_destroy(&ctx, obj);
10745         }
10746         list_for_each_entry_safe(chain, nc, &table->chains, list) {
10747                 ctx.chain = chain;
10748                 nft_chain_del(chain);
10749                 table->use--;
10750                 nf_tables_chain_destroy(&ctx);
10751         }
10752         nf_tables_table_destroy(&ctx);
10753 }
10754
10755 static void __nft_release_tables(struct net *net)
10756 {
10757         struct nftables_pernet *nft_net = nft_pernet(net);
10758         struct nft_table *table, *nt;
10759
10760         list_for_each_entry_safe(table, nt, &nft_net->tables, list) {
10761                 if (nft_table_has_owner(table))
10762                         continue;
10763
10764                 list_del(&table->list);
10765
10766                 __nft_release_table(net, table);
10767         }
10768 }
10769
10770 static int nft_rcv_nl_event(struct notifier_block *this, unsigned long event,
10771                             void *ptr)
10772 {
10773         struct nft_table *table, *to_delete[8];
10774         struct nftables_pernet *nft_net;
10775         struct netlink_notify *n = ptr;
10776         struct net *net = n->net;
10777         unsigned int deleted;
10778         bool restart = false;
10779
10780         if (event != NETLINK_URELEASE || n->protocol != NETLINK_NETFILTER)
10781                 return NOTIFY_DONE;
10782
10783         nft_net = nft_pernet(net);
10784         deleted = 0;
10785         mutex_lock(&nft_net->commit_mutex);
10786         if (!list_empty(&nf_tables_destroy_list))
10787                 rcu_barrier();
10788 again:
10789         list_for_each_entry(table, &nft_net->tables, list) {
10790                 if (nft_table_has_owner(table) &&
10791                     n->portid == table->nlpid) {
10792                         __nft_release_hook(net, table);
10793                         list_del_rcu(&table->list);
10794                         to_delete[deleted++] = table;
10795                         if (deleted >= ARRAY_SIZE(to_delete))
10796                                 break;
10797                 }
10798         }
10799         if (deleted) {
10800                 restart = deleted >= ARRAY_SIZE(to_delete);
10801                 synchronize_rcu();
10802                 while (deleted)
10803                         __nft_release_table(net, to_delete[--deleted]);
10804
10805                 if (restart)
10806                         goto again;
10807         }
10808         mutex_unlock(&nft_net->commit_mutex);
10809
10810         return NOTIFY_DONE;
10811 }
10812
10813 static struct notifier_block nft_nl_notifier = {
10814         .notifier_call  = nft_rcv_nl_event,
10815 };
10816
10817 static int __net_init nf_tables_init_net(struct net *net)
10818 {
10819         struct nftables_pernet *nft_net = nft_pernet(net);
10820
10821         INIT_LIST_HEAD(&nft_net->tables);
10822         INIT_LIST_HEAD(&nft_net->commit_list);
10823         INIT_LIST_HEAD(&nft_net->binding_list);
10824         INIT_LIST_HEAD(&nft_net->module_list);
10825         INIT_LIST_HEAD(&nft_net->notify_list);
10826         mutex_init(&nft_net->commit_mutex);
10827         nft_net->base_seq = 1;
10828
10829         return 0;
10830 }
10831
10832 static void __net_exit nf_tables_pre_exit_net(struct net *net)
10833 {
10834         struct nftables_pernet *nft_net = nft_pernet(net);
10835
10836         mutex_lock(&nft_net->commit_mutex);
10837         __nft_release_hooks(net);
10838         mutex_unlock(&nft_net->commit_mutex);
10839 }
10840
10841 static void __net_exit nf_tables_exit_net(struct net *net)
10842 {
10843         struct nftables_pernet *nft_net = nft_pernet(net);
10844
10845         mutex_lock(&nft_net->commit_mutex);
10846         if (!list_empty(&nft_net->commit_list) ||
10847             !list_empty(&nft_net->module_list))
10848                 __nf_tables_abort(net, NFNL_ABORT_NONE);
10849         __nft_release_tables(net);
10850         mutex_unlock(&nft_net->commit_mutex);
10851         WARN_ON_ONCE(!list_empty(&nft_net->tables));
10852         WARN_ON_ONCE(!list_empty(&nft_net->module_list));
10853         WARN_ON_ONCE(!list_empty(&nft_net->notify_list));
10854 }
10855
10856 static struct pernet_operations nf_tables_net_ops = {
10857         .init           = nf_tables_init_net,
10858         .pre_exit       = nf_tables_pre_exit_net,
10859         .exit           = nf_tables_exit_net,
10860         .id             = &nf_tables_net_id,
10861         .size           = sizeof(struct nftables_pernet),
10862 };
10863
10864 static int __init nf_tables_module_init(void)
10865 {
10866         int err;
10867
10868         err = register_pernet_subsys(&nf_tables_net_ops);
10869         if (err < 0)
10870                 return err;
10871
10872         err = nft_chain_filter_init();
10873         if (err < 0)
10874                 goto err_chain_filter;
10875
10876         err = nf_tables_core_module_init();
10877         if (err < 0)
10878                 goto err_core_module;
10879
10880         err = register_netdevice_notifier(&nf_tables_flowtable_notifier);
10881         if (err < 0)
10882                 goto err_netdev_notifier;
10883
10884         err = rhltable_init(&nft_objname_ht, &nft_objname_ht_params);
10885         if (err < 0)
10886                 goto err_rht_objname;
10887
10888         err = nft_offload_init();
10889         if (err < 0)
10890                 goto err_offload;
10891
10892         err = netlink_register_notifier(&nft_nl_notifier);
10893         if (err < 0)
10894                 goto err_netlink_notifier;
10895
10896         /* must be last */
10897         err = nfnetlink_subsys_register(&nf_tables_subsys);
10898         if (err < 0)
10899                 goto err_nfnl_subsys;
10900
10901         nft_chain_route_init();
10902
10903         return err;
10904
10905 err_nfnl_subsys:
10906         netlink_unregister_notifier(&nft_nl_notifier);
10907 err_netlink_notifier:
10908         nft_offload_exit();
10909 err_offload:
10910         rhltable_destroy(&nft_objname_ht);
10911 err_rht_objname:
10912         unregister_netdevice_notifier(&nf_tables_flowtable_notifier);
10913 err_netdev_notifier:
10914         nf_tables_core_module_exit();
10915 err_core_module:
10916         nft_chain_filter_fini();
10917 err_chain_filter:
10918         unregister_pernet_subsys(&nf_tables_net_ops);
10919         return err;
10920 }
10921
10922 static void __exit nf_tables_module_exit(void)
10923 {
10924         nfnetlink_subsys_unregister(&nf_tables_subsys);
10925         netlink_unregister_notifier(&nft_nl_notifier);
10926         nft_offload_exit();
10927         unregister_netdevice_notifier(&nf_tables_flowtable_notifier);
10928         nft_chain_filter_fini();
10929         nft_chain_route_fini();
10930         unregister_pernet_subsys(&nf_tables_net_ops);
10931         cancel_work_sync(&trans_destroy_work);
10932         rcu_barrier();
10933         rhltable_destroy(&nft_objname_ht);
10934         nf_tables_core_module_exit();
10935 }
10936
10937 module_init(nf_tables_module_init);
10938 module_exit(nf_tables_module_exit);
10939
10940 MODULE_LICENSE("GPL");
10941 MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
10942 MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_NFTABLES);