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