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