1
2
3
4
5
6
7
8
9
10
11#include <linux/module.h>
12#include <linux/init.h>
13#include <linux/list.h>
14#include <linux/skbuff.h>
15#include <linux/netlink.h>
16#include <linux/netfilter.h>
17#include <linux/netfilter/nfnetlink.h>
18#include <linux/netfilter/nf_tables.h>
19#include <net/netfilter/nf_tables_core.h>
20#include <net/netfilter/nf_tables.h>
21#include <net/net_namespace.h>
22#include <net/sock.h>
23
24static LIST_HEAD(nf_tables_expressions);
25
26
27
28
29
30
31
32
33
34int nft_register_afinfo(struct net *net, struct nft_af_info *afi)
35{
36 INIT_LIST_HEAD(&afi->tables);
37 nfnl_lock(NFNL_SUBSYS_NFTABLES);
38 list_add_tail_rcu(&afi->list, &net->nft.af_info);
39 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
40 return 0;
41}
42EXPORT_SYMBOL_GPL(nft_register_afinfo);
43
44
45
46
47
48
49
50
51void nft_unregister_afinfo(struct nft_af_info *afi)
52{
53 nfnl_lock(NFNL_SUBSYS_NFTABLES);
54 list_del_rcu(&afi->list);
55 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
56}
57EXPORT_SYMBOL_GPL(nft_unregister_afinfo);
58
59static struct nft_af_info *nft_afinfo_lookup(struct net *net, int family)
60{
61 struct nft_af_info *afi;
62
63 list_for_each_entry(afi, &net->nft.af_info, list) {
64 if (afi->family == family)
65 return afi;
66 }
67 return NULL;
68}
69
70static struct nft_af_info *
71nf_tables_afinfo_lookup(struct net *net, int family, bool autoload)
72{
73 struct nft_af_info *afi;
74
75 afi = nft_afinfo_lookup(net, family);
76 if (afi != NULL)
77 return afi;
78#ifdef CONFIG_MODULES
79 if (autoload) {
80 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
81 request_module("nft-afinfo-%u", family);
82 nfnl_lock(NFNL_SUBSYS_NFTABLES);
83 afi = nft_afinfo_lookup(net, family);
84 if (afi != NULL)
85 return ERR_PTR(-EAGAIN);
86 }
87#endif
88 return ERR_PTR(-EAFNOSUPPORT);
89}
90
91static void nft_ctx_init(struct nft_ctx *ctx,
92 const struct sk_buff *skb,
93 const struct nlmsghdr *nlh,
94 struct nft_af_info *afi,
95 struct nft_table *table,
96 struct nft_chain *chain,
97 const struct nlattr * const *nla)
98{
99 ctx->net = sock_net(skb->sk);
100 ctx->afi = afi;
101 ctx->table = table;
102 ctx->chain = chain;
103 ctx->nla = nla;
104 ctx->portid = NETLINK_CB(skb).portid;
105 ctx->report = nlmsg_report(nlh);
106 ctx->seq = nlh->nlmsg_seq;
107}
108
109static struct nft_trans *nft_trans_alloc(struct nft_ctx *ctx, int msg_type,
110 u32 size)
111{
112 struct nft_trans *trans;
113
114 trans = kzalloc(sizeof(struct nft_trans) + size, GFP_KERNEL);
115 if (trans == NULL)
116 return NULL;
117
118 trans->msg_type = msg_type;
119 trans->ctx = *ctx;
120
121 return trans;
122}
123
124static void nft_trans_destroy(struct nft_trans *trans)
125{
126 list_del(&trans->list);
127 kfree(trans);
128}
129
130int nft_register_basechain(struct nft_base_chain *basechain,
131 unsigned int hook_nops)
132{
133 struct net *net = read_pnet(&basechain->pnet);
134
135 if (basechain->flags & NFT_BASECHAIN_DISABLED)
136 return 0;
137
138 return nf_register_net_hooks(net, basechain->ops, hook_nops);
139}
140EXPORT_SYMBOL_GPL(nft_register_basechain);
141
142void nft_unregister_basechain(struct nft_base_chain *basechain,
143 unsigned int hook_nops)
144{
145 struct net *net = read_pnet(&basechain->pnet);
146
147 if (basechain->flags & NFT_BASECHAIN_DISABLED)
148 return;
149
150 nf_unregister_net_hooks(net, basechain->ops, hook_nops);
151}
152EXPORT_SYMBOL_GPL(nft_unregister_basechain);
153
154static int nf_tables_register_hooks(const struct nft_table *table,
155 struct nft_chain *chain,
156 unsigned int hook_nops)
157{
158 if (table->flags & NFT_TABLE_F_DORMANT ||
159 !(chain->flags & NFT_BASE_CHAIN))
160 return 0;
161
162 return nft_register_basechain(nft_base_chain(chain), hook_nops);
163}
164
165static void nf_tables_unregister_hooks(const struct nft_table *table,
166 struct nft_chain *chain,
167 unsigned int hook_nops)
168{
169 if (table->flags & NFT_TABLE_F_DORMANT ||
170 !(chain->flags & NFT_BASE_CHAIN))
171 return;
172
173 nft_unregister_basechain(nft_base_chain(chain), hook_nops);
174}
175
176
177#define NFT_TABLE_INACTIVE (1 << 15)
178
179static int nft_trans_table_add(struct nft_ctx *ctx, int msg_type)
180{
181 struct nft_trans *trans;
182
183 trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_table));
184 if (trans == NULL)
185 return -ENOMEM;
186
187 if (msg_type == NFT_MSG_NEWTABLE)
188 ctx->table->flags |= NFT_TABLE_INACTIVE;
189
190 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
191 return 0;
192}
193
194static int nft_deltable(struct nft_ctx *ctx)
195{
196 int err;
197
198 err = nft_trans_table_add(ctx, NFT_MSG_DELTABLE);
199 if (err < 0)
200 return err;
201
202 list_del_rcu(&ctx->table->list);
203 return err;
204}
205
206static int nft_trans_chain_add(struct nft_ctx *ctx, int msg_type)
207{
208 struct nft_trans *trans;
209
210 trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_chain));
211 if (trans == NULL)
212 return -ENOMEM;
213
214 if (msg_type == NFT_MSG_NEWCHAIN)
215 ctx->chain->flags |= NFT_CHAIN_INACTIVE;
216
217 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
218 return 0;
219}
220
221static int nft_delchain(struct nft_ctx *ctx)
222{
223 int err;
224
225 err = nft_trans_chain_add(ctx, NFT_MSG_DELCHAIN);
226 if (err < 0)
227 return err;
228
229 ctx->table->use--;
230 list_del_rcu(&ctx->chain->list);
231
232 return err;
233}
234
235static inline bool
236nft_rule_is_active(struct net *net, const struct nft_rule *rule)
237{
238 return (rule->genmask & nft_genmask_cur(net)) == 0;
239}
240
241static inline int
242nft_rule_is_active_next(struct net *net, const struct nft_rule *rule)
243{
244 return (rule->genmask & nft_genmask_next(net)) == 0;
245}
246
247static inline void
248nft_rule_activate_next(struct net *net, struct nft_rule *rule)
249{
250
251 rule->genmask = nft_genmask_cur(net);
252}
253
254static inline void
255nft_rule_deactivate_next(struct net *net, struct nft_rule *rule)
256{
257 rule->genmask = nft_genmask_next(net);
258}
259
260static inline void nft_rule_clear(struct net *net, struct nft_rule *rule)
261{
262 rule->genmask &= ~nft_genmask_next(net);
263}
264
265static int
266nf_tables_delrule_deactivate(struct nft_ctx *ctx, struct nft_rule *rule)
267{
268
269 if (nft_rule_is_active_next(ctx->net, rule)) {
270 nft_rule_deactivate_next(ctx->net, rule);
271 ctx->chain->use--;
272 return 0;
273 }
274 return -ENOENT;
275}
276
277static struct nft_trans *nft_trans_rule_add(struct nft_ctx *ctx, int msg_type,
278 struct nft_rule *rule)
279{
280 struct nft_trans *trans;
281
282 trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_rule));
283 if (trans == NULL)
284 return NULL;
285
286 nft_trans_rule(trans) = rule;
287 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
288
289 return trans;
290}
291
292static int nft_delrule(struct nft_ctx *ctx, struct nft_rule *rule)
293{
294 struct nft_trans *trans;
295 int err;
296
297 trans = nft_trans_rule_add(ctx, NFT_MSG_DELRULE, rule);
298 if (trans == NULL)
299 return -ENOMEM;
300
301 err = nf_tables_delrule_deactivate(ctx, rule);
302 if (err < 0) {
303 nft_trans_destroy(trans);
304 return err;
305 }
306
307 return 0;
308}
309
310static int nft_delrule_by_chain(struct nft_ctx *ctx)
311{
312 struct nft_rule *rule;
313 int err;
314
315 list_for_each_entry(rule, &ctx->chain->rules, list) {
316 err = nft_delrule(ctx, rule);
317 if (err < 0)
318 return err;
319 }
320 return 0;
321}
322
323
324#define NFT_SET_INACTIVE (1 << 15)
325
326static int nft_trans_set_add(struct nft_ctx *ctx, int msg_type,
327 struct nft_set *set)
328{
329 struct nft_trans *trans;
330
331 trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_set));
332 if (trans == NULL)
333 return -ENOMEM;
334
335 if (msg_type == NFT_MSG_NEWSET && ctx->nla[NFTA_SET_ID] != NULL) {
336 nft_trans_set_id(trans) =
337 ntohl(nla_get_be32(ctx->nla[NFTA_SET_ID]));
338 set->flags |= NFT_SET_INACTIVE;
339 }
340 nft_trans_set(trans) = set;
341 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
342
343 return 0;
344}
345
346static int nft_delset(struct nft_ctx *ctx, struct nft_set *set)
347{
348 int err;
349
350 err = nft_trans_set_add(ctx, NFT_MSG_DELSET, set);
351 if (err < 0)
352 return err;
353
354 list_del_rcu(&set->list);
355 ctx->table->use--;
356
357 return err;
358}
359
360
361
362
363
364static struct nft_table *nft_table_lookup(const struct nft_af_info *afi,
365 const struct nlattr *nla)
366{
367 struct nft_table *table;
368
369 list_for_each_entry(table, &afi->tables, list) {
370 if (!nla_strcmp(nla, table->name))
371 return table;
372 }
373 return NULL;
374}
375
376static struct nft_table *nf_tables_table_lookup(const struct nft_af_info *afi,
377 const struct nlattr *nla)
378{
379 struct nft_table *table;
380
381 if (nla == NULL)
382 return ERR_PTR(-EINVAL);
383
384 table = nft_table_lookup(afi, nla);
385 if (table != NULL)
386 return table;
387
388 return ERR_PTR(-ENOENT);
389}
390
391static inline u64 nf_tables_alloc_handle(struct nft_table *table)
392{
393 return ++table->hgenerator;
394}
395
396static const struct nf_chain_type *chain_type[AF_MAX][NFT_CHAIN_T_MAX];
397
398static const struct nf_chain_type *
399__nf_tables_chain_type_lookup(int family, const struct nlattr *nla)
400{
401 int i;
402
403 for (i = 0; i < NFT_CHAIN_T_MAX; i++) {
404 if (chain_type[family][i] != NULL &&
405 !nla_strcmp(nla, chain_type[family][i]->name))
406 return chain_type[family][i];
407 }
408 return NULL;
409}
410
411static const struct nf_chain_type *
412nf_tables_chain_type_lookup(const struct nft_af_info *afi,
413 const struct nlattr *nla,
414 bool autoload)
415{
416 const struct nf_chain_type *type;
417
418 type = __nf_tables_chain_type_lookup(afi->family, nla);
419 if (type != NULL)
420 return type;
421#ifdef CONFIG_MODULES
422 if (autoload) {
423 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
424 request_module("nft-chain-%u-%.*s", afi->family,
425 nla_len(nla), (const char *)nla_data(nla));
426 nfnl_lock(NFNL_SUBSYS_NFTABLES);
427 type = __nf_tables_chain_type_lookup(afi->family, nla);
428 if (type != NULL)
429 return ERR_PTR(-EAGAIN);
430 }
431#endif
432 return ERR_PTR(-ENOENT);
433}
434
435static const struct nla_policy nft_table_policy[NFTA_TABLE_MAX + 1] = {
436 [NFTA_TABLE_NAME] = { .type = NLA_STRING,
437 .len = NFT_TABLE_MAXNAMELEN - 1 },
438 [NFTA_TABLE_FLAGS] = { .type = NLA_U32 },
439};
440
441static int nf_tables_fill_table_info(struct sk_buff *skb, struct net *net,
442 u32 portid, u32 seq, int event, u32 flags,
443 int family, const struct nft_table *table)
444{
445 struct nlmsghdr *nlh;
446 struct nfgenmsg *nfmsg;
447
448 event |= NFNL_SUBSYS_NFTABLES << 8;
449 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg), flags);
450 if (nlh == NULL)
451 goto nla_put_failure;
452
453 nfmsg = nlmsg_data(nlh);
454 nfmsg->nfgen_family = family;
455 nfmsg->version = NFNETLINK_V0;
456 nfmsg->res_id = htons(net->nft.base_seq & 0xffff);
457
458 if (nla_put_string(skb, NFTA_TABLE_NAME, table->name) ||
459 nla_put_be32(skb, NFTA_TABLE_FLAGS, htonl(table->flags)) ||
460 nla_put_be32(skb, NFTA_TABLE_USE, htonl(table->use)))
461 goto nla_put_failure;
462
463 nlmsg_end(skb, nlh);
464 return 0;
465
466nla_put_failure:
467 nlmsg_trim(skb, nlh);
468 return -1;
469}
470
471static int nf_tables_table_notify(const struct nft_ctx *ctx, int event)
472{
473 struct sk_buff *skb;
474 int err;
475
476 if (!ctx->report &&
477 !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
478 return 0;
479
480 err = -ENOBUFS;
481 skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
482 if (skb == NULL)
483 goto err;
484
485 err = nf_tables_fill_table_info(skb, ctx->net, ctx->portid, ctx->seq,
486 event, 0, ctx->afi->family, ctx->table);
487 if (err < 0) {
488 kfree_skb(skb);
489 goto err;
490 }
491
492 err = nfnetlink_send(skb, ctx->net, ctx->portid, NFNLGRP_NFTABLES,
493 ctx->report, GFP_KERNEL);
494err:
495 if (err < 0) {
496 nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES,
497 err);
498 }
499 return err;
500}
501
502static int nf_tables_dump_tables(struct sk_buff *skb,
503 struct netlink_callback *cb)
504{
505 const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
506 const struct nft_af_info *afi;
507 const struct nft_table *table;
508 unsigned int idx = 0, s_idx = cb->args[0];
509 struct net *net = sock_net(skb->sk);
510 int family = nfmsg->nfgen_family;
511
512 rcu_read_lock();
513 cb->seq = net->nft.base_seq;
514
515 list_for_each_entry_rcu(afi, &net->nft.af_info, list) {
516 if (family != NFPROTO_UNSPEC && family != afi->family)
517 continue;
518
519 list_for_each_entry_rcu(table, &afi->tables, list) {
520 if (idx < s_idx)
521 goto cont;
522 if (idx > s_idx)
523 memset(&cb->args[1], 0,
524 sizeof(cb->args) - sizeof(cb->args[0]));
525 if (nf_tables_fill_table_info(skb, net,
526 NETLINK_CB(cb->skb).portid,
527 cb->nlh->nlmsg_seq,
528 NFT_MSG_NEWTABLE,
529 NLM_F_MULTI,
530 afi->family, table) < 0)
531 goto done;
532
533 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
534cont:
535 idx++;
536 }
537 }
538done:
539 rcu_read_unlock();
540 cb->args[0] = idx;
541 return skb->len;
542}
543
544static int nf_tables_gettable(struct sock *nlsk, struct sk_buff *skb,
545 const struct nlmsghdr *nlh,
546 const struct nlattr * const nla[])
547{
548 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
549 const struct nft_af_info *afi;
550 const struct nft_table *table;
551 struct sk_buff *skb2;
552 struct net *net = sock_net(skb->sk);
553 int family = nfmsg->nfgen_family;
554 int err;
555
556 if (nlh->nlmsg_flags & NLM_F_DUMP) {
557 struct netlink_dump_control c = {
558 .dump = nf_tables_dump_tables,
559 };
560 return netlink_dump_start(nlsk, skb, nlh, &c);
561 }
562
563 afi = nf_tables_afinfo_lookup(net, family, false);
564 if (IS_ERR(afi))
565 return PTR_ERR(afi);
566
567 table = nf_tables_table_lookup(afi, nla[NFTA_TABLE_NAME]);
568 if (IS_ERR(table))
569 return PTR_ERR(table);
570 if (table->flags & NFT_TABLE_INACTIVE)
571 return -ENOENT;
572
573 skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
574 if (!skb2)
575 return -ENOMEM;
576
577 err = nf_tables_fill_table_info(skb2, net, NETLINK_CB(skb).portid,
578 nlh->nlmsg_seq, NFT_MSG_NEWTABLE, 0,
579 family, table);
580 if (err < 0)
581 goto err;
582
583 return nlmsg_unicast(nlsk, skb2, NETLINK_CB(skb).portid);
584
585err:
586 kfree_skb(skb2);
587 return err;
588}
589
590static int nf_tables_table_enable(const struct nft_af_info *afi,
591 struct nft_table *table)
592{
593 struct nft_chain *chain;
594 int err, i = 0;
595
596 list_for_each_entry(chain, &table->chains, list) {
597 if (!(chain->flags & NFT_BASE_CHAIN))
598 continue;
599
600 err = nft_register_basechain(nft_base_chain(chain), afi->nops);
601 if (err < 0)
602 goto err;
603
604 i++;
605 }
606 return 0;
607err:
608 list_for_each_entry(chain, &table->chains, list) {
609 if (!(chain->flags & NFT_BASE_CHAIN))
610 continue;
611
612 if (i-- <= 0)
613 break;
614
615 nft_unregister_basechain(nft_base_chain(chain), afi->nops);
616 }
617 return err;
618}
619
620static void nf_tables_table_disable(const struct nft_af_info *afi,
621 struct nft_table *table)
622{
623 struct nft_chain *chain;
624
625 list_for_each_entry(chain, &table->chains, list) {
626 if (chain->flags & NFT_BASE_CHAIN)
627 nft_unregister_basechain(nft_base_chain(chain),
628 afi->nops);
629 }
630}
631
632static int nf_tables_updtable(struct nft_ctx *ctx)
633{
634 struct nft_trans *trans;
635 u32 flags;
636 int ret = 0;
637
638 if (!ctx->nla[NFTA_TABLE_FLAGS])
639 return 0;
640
641 flags = ntohl(nla_get_be32(ctx->nla[NFTA_TABLE_FLAGS]));
642 if (flags & ~NFT_TABLE_F_DORMANT)
643 return -EINVAL;
644
645 if (flags == ctx->table->flags)
646 return 0;
647
648 trans = nft_trans_alloc(ctx, NFT_MSG_NEWTABLE,
649 sizeof(struct nft_trans_table));
650 if (trans == NULL)
651 return -ENOMEM;
652
653 if ((flags & NFT_TABLE_F_DORMANT) &&
654 !(ctx->table->flags & NFT_TABLE_F_DORMANT)) {
655 nft_trans_table_enable(trans) = false;
656 } else if (!(flags & NFT_TABLE_F_DORMANT) &&
657 ctx->table->flags & NFT_TABLE_F_DORMANT) {
658 ret = nf_tables_table_enable(ctx->afi, ctx->table);
659 if (ret >= 0) {
660 ctx->table->flags &= ~NFT_TABLE_F_DORMANT;
661 nft_trans_table_enable(trans) = true;
662 }
663 }
664 if (ret < 0)
665 goto err;
666
667 nft_trans_table_update(trans) = true;
668 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
669 return 0;
670err:
671 nft_trans_destroy(trans);
672 return ret;
673}
674
675static int nf_tables_newtable(struct sock *nlsk, struct sk_buff *skb,
676 const struct nlmsghdr *nlh,
677 const struct nlattr * const nla[])
678{
679 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
680 const struct nlattr *name;
681 struct nft_af_info *afi;
682 struct nft_table *table;
683 struct net *net = sock_net(skb->sk);
684 int family = nfmsg->nfgen_family;
685 u32 flags = 0;
686 struct nft_ctx ctx;
687 int err;
688
689 afi = nf_tables_afinfo_lookup(net, family, true);
690 if (IS_ERR(afi))
691 return PTR_ERR(afi);
692
693 name = nla[NFTA_TABLE_NAME];
694 table = nf_tables_table_lookup(afi, name);
695 if (IS_ERR(table)) {
696 if (PTR_ERR(table) != -ENOENT)
697 return PTR_ERR(table);
698 table = NULL;
699 }
700
701 if (table != NULL) {
702 if (table->flags & NFT_TABLE_INACTIVE)
703 return -ENOENT;
704 if (nlh->nlmsg_flags & NLM_F_EXCL)
705 return -EEXIST;
706 if (nlh->nlmsg_flags & NLM_F_REPLACE)
707 return -EOPNOTSUPP;
708
709 nft_ctx_init(&ctx, skb, nlh, afi, table, NULL, nla);
710 return nf_tables_updtable(&ctx);
711 }
712
713 if (nla[NFTA_TABLE_FLAGS]) {
714 flags = ntohl(nla_get_be32(nla[NFTA_TABLE_FLAGS]));
715 if (flags & ~NFT_TABLE_F_DORMANT)
716 return -EINVAL;
717 }
718
719 err = -EAFNOSUPPORT;
720 if (!try_module_get(afi->owner))
721 goto err1;
722
723 err = -ENOMEM;
724 table = kzalloc(sizeof(*table), GFP_KERNEL);
725 if (table == NULL)
726 goto err2;
727
728 nla_strlcpy(table->name, name, NFT_TABLE_MAXNAMELEN);
729 INIT_LIST_HEAD(&table->chains);
730 INIT_LIST_HEAD(&table->sets);
731 table->flags = flags;
732
733 nft_ctx_init(&ctx, skb, nlh, afi, table, NULL, nla);
734 err = nft_trans_table_add(&ctx, NFT_MSG_NEWTABLE);
735 if (err < 0)
736 goto err3;
737
738 list_add_tail_rcu(&table->list, &afi->tables);
739 return 0;
740err3:
741 kfree(table);
742err2:
743 module_put(afi->owner);
744err1:
745 return err;
746}
747
748static int nft_flush_table(struct nft_ctx *ctx)
749{
750 int err;
751 struct nft_chain *chain, *nc;
752 struct nft_set *set, *ns;
753
754 list_for_each_entry(chain, &ctx->table->chains, list) {
755 ctx->chain = chain;
756
757 err = nft_delrule_by_chain(ctx);
758 if (err < 0)
759 goto out;
760 }
761
762 list_for_each_entry_safe(set, ns, &ctx->table->sets, list) {
763 if (set->flags & NFT_SET_ANONYMOUS &&
764 !list_empty(&set->bindings))
765 continue;
766
767 err = nft_delset(ctx, set);
768 if (err < 0)
769 goto out;
770 }
771
772 list_for_each_entry_safe(chain, nc, &ctx->table->chains, list) {
773 ctx->chain = chain;
774
775 err = nft_delchain(ctx);
776 if (err < 0)
777 goto out;
778 }
779
780 err = nft_deltable(ctx);
781out:
782 return err;
783}
784
785static int nft_flush(struct nft_ctx *ctx, int family)
786{
787 struct nft_af_info *afi;
788 struct nft_table *table, *nt;
789 const struct nlattr * const *nla = ctx->nla;
790 int err = 0;
791
792 list_for_each_entry(afi, &ctx->net->nft.af_info, list) {
793 if (family != AF_UNSPEC && afi->family != family)
794 continue;
795
796 ctx->afi = afi;
797 list_for_each_entry_safe(table, nt, &afi->tables, list) {
798 if (nla[NFTA_TABLE_NAME] &&
799 nla_strcmp(nla[NFTA_TABLE_NAME], table->name) != 0)
800 continue;
801
802 ctx->table = table;
803
804 err = nft_flush_table(ctx);
805 if (err < 0)
806 goto out;
807 }
808 }
809out:
810 return err;
811}
812
813static int nf_tables_deltable(struct sock *nlsk, struct sk_buff *skb,
814 const struct nlmsghdr *nlh,
815 const struct nlattr * const nla[])
816{
817 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
818 struct nft_af_info *afi;
819 struct nft_table *table;
820 struct net *net = sock_net(skb->sk);
821 int family = nfmsg->nfgen_family;
822 struct nft_ctx ctx;
823
824 nft_ctx_init(&ctx, skb, nlh, NULL, NULL, NULL, nla);
825 if (family == AF_UNSPEC || nla[NFTA_TABLE_NAME] == NULL)
826 return nft_flush(&ctx, family);
827
828 afi = nf_tables_afinfo_lookup(net, family, false);
829 if (IS_ERR(afi))
830 return PTR_ERR(afi);
831
832 table = nf_tables_table_lookup(afi, nla[NFTA_TABLE_NAME]);
833 if (IS_ERR(table))
834 return PTR_ERR(table);
835 if (table->flags & NFT_TABLE_INACTIVE)
836 return -ENOENT;
837
838 ctx.afi = afi;
839 ctx.table = table;
840
841 return nft_flush_table(&ctx);
842}
843
844static void nf_tables_table_destroy(struct nft_ctx *ctx)
845{
846 BUG_ON(ctx->table->use > 0);
847
848 kfree(ctx->table);
849 module_put(ctx->afi->owner);
850}
851
852int nft_register_chain_type(const struct nf_chain_type *ctype)
853{
854 int err = 0;
855
856 nfnl_lock(NFNL_SUBSYS_NFTABLES);
857 if (chain_type[ctype->family][ctype->type] != NULL) {
858 err = -EBUSY;
859 goto out;
860 }
861 chain_type[ctype->family][ctype->type] = ctype;
862out:
863 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
864 return err;
865}
866EXPORT_SYMBOL_GPL(nft_register_chain_type);
867
868void nft_unregister_chain_type(const struct nf_chain_type *ctype)
869{
870 nfnl_lock(NFNL_SUBSYS_NFTABLES);
871 chain_type[ctype->family][ctype->type] = NULL;
872 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
873}
874EXPORT_SYMBOL_GPL(nft_unregister_chain_type);
875
876
877
878
879
880static struct nft_chain *
881nf_tables_chain_lookup_byhandle(const struct nft_table *table, u64 handle)
882{
883 struct nft_chain *chain;
884
885 list_for_each_entry(chain, &table->chains, list) {
886 if (chain->handle == handle)
887 return chain;
888 }
889
890 return ERR_PTR(-ENOENT);
891}
892
893static struct nft_chain *nf_tables_chain_lookup(const struct nft_table *table,
894 const struct nlattr *nla)
895{
896 struct nft_chain *chain;
897
898 if (nla == NULL)
899 return ERR_PTR(-EINVAL);
900
901 list_for_each_entry(chain, &table->chains, list) {
902 if (!nla_strcmp(nla, chain->name))
903 return chain;
904 }
905
906 return ERR_PTR(-ENOENT);
907}
908
909static const struct nla_policy nft_chain_policy[NFTA_CHAIN_MAX + 1] = {
910 [NFTA_CHAIN_TABLE] = { .type = NLA_STRING },
911 [NFTA_CHAIN_HANDLE] = { .type = NLA_U64 },
912 [NFTA_CHAIN_NAME] = { .type = NLA_STRING,
913 .len = NFT_CHAIN_MAXNAMELEN - 1 },
914 [NFTA_CHAIN_HOOK] = { .type = NLA_NESTED },
915 [NFTA_CHAIN_POLICY] = { .type = NLA_U32 },
916 [NFTA_CHAIN_TYPE] = { .type = NLA_STRING },
917 [NFTA_CHAIN_COUNTERS] = { .type = NLA_NESTED },
918};
919
920static const struct nla_policy nft_hook_policy[NFTA_HOOK_MAX + 1] = {
921 [NFTA_HOOK_HOOKNUM] = { .type = NLA_U32 },
922 [NFTA_HOOK_PRIORITY] = { .type = NLA_U32 },
923 [NFTA_HOOK_DEV] = { .type = NLA_STRING,
924 .len = IFNAMSIZ - 1 },
925};
926
927static int nft_dump_stats(struct sk_buff *skb, struct nft_stats __percpu *stats)
928{
929 struct nft_stats *cpu_stats, total;
930 struct nlattr *nest;
931 unsigned int seq;
932 u64 pkts, bytes;
933 int cpu;
934
935 memset(&total, 0, sizeof(total));
936 for_each_possible_cpu(cpu) {
937 cpu_stats = per_cpu_ptr(stats, cpu);
938 do {
939 seq = u64_stats_fetch_begin_irq(&cpu_stats->syncp);
940 pkts = cpu_stats->pkts;
941 bytes = cpu_stats->bytes;
942 } while (u64_stats_fetch_retry_irq(&cpu_stats->syncp, seq));
943 total.pkts += pkts;
944 total.bytes += bytes;
945 }
946 nest = nla_nest_start(skb, NFTA_CHAIN_COUNTERS);
947 if (nest == NULL)
948 goto nla_put_failure;
949
950 if (nla_put_be64(skb, NFTA_COUNTER_PACKETS, cpu_to_be64(total.pkts)) ||
951 nla_put_be64(skb, NFTA_COUNTER_BYTES, cpu_to_be64(total.bytes)))
952 goto nla_put_failure;
953
954 nla_nest_end(skb, nest);
955 return 0;
956
957nla_put_failure:
958 return -ENOSPC;
959}
960
961static int nf_tables_fill_chain_info(struct sk_buff *skb, struct net *net,
962 u32 portid, u32 seq, int event, u32 flags,
963 int family, const struct nft_table *table,
964 const struct nft_chain *chain)
965{
966 struct nlmsghdr *nlh;
967 struct nfgenmsg *nfmsg;
968
969 event |= NFNL_SUBSYS_NFTABLES << 8;
970 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg), flags);
971 if (nlh == NULL)
972 goto nla_put_failure;
973
974 nfmsg = nlmsg_data(nlh);
975 nfmsg->nfgen_family = family;
976 nfmsg->version = NFNETLINK_V0;
977 nfmsg->res_id = htons(net->nft.base_seq & 0xffff);
978
979 if (nla_put_string(skb, NFTA_CHAIN_TABLE, table->name))
980 goto nla_put_failure;
981 if (nla_put_be64(skb, NFTA_CHAIN_HANDLE, cpu_to_be64(chain->handle)))
982 goto nla_put_failure;
983 if (nla_put_string(skb, NFTA_CHAIN_NAME, chain->name))
984 goto nla_put_failure;
985
986 if (chain->flags & NFT_BASE_CHAIN) {
987 const struct nft_base_chain *basechain = nft_base_chain(chain);
988 const struct nf_hook_ops *ops = &basechain->ops[0];
989 struct nlattr *nest;
990
991 nest = nla_nest_start(skb, NFTA_CHAIN_HOOK);
992 if (nest == NULL)
993 goto nla_put_failure;
994 if (nla_put_be32(skb, NFTA_HOOK_HOOKNUM, htonl(ops->hooknum)))
995 goto nla_put_failure;
996 if (nla_put_be32(skb, NFTA_HOOK_PRIORITY, htonl(ops->priority)))
997 goto nla_put_failure;
998 if (basechain->dev_name[0] &&
999 nla_put_string(skb, NFTA_HOOK_DEV, basechain->dev_name))
1000 goto nla_put_failure;
1001 nla_nest_end(skb, nest);
1002
1003 if (nla_put_be32(skb, NFTA_CHAIN_POLICY,
1004 htonl(basechain->policy)))
1005 goto nla_put_failure;
1006
1007 if (nla_put_string(skb, NFTA_CHAIN_TYPE, basechain->type->name))
1008 goto nla_put_failure;
1009
1010 if (nft_dump_stats(skb, nft_base_chain(chain)->stats))
1011 goto nla_put_failure;
1012 }
1013
1014 if (nla_put_be32(skb, NFTA_CHAIN_USE, htonl(chain->use)))
1015 goto nla_put_failure;
1016
1017 nlmsg_end(skb, nlh);
1018 return 0;
1019
1020nla_put_failure:
1021 nlmsg_trim(skb, nlh);
1022 return -1;
1023}
1024
1025static int nf_tables_chain_notify(const struct nft_ctx *ctx, int event)
1026{
1027 struct sk_buff *skb;
1028 int err;
1029
1030 if (!ctx->report &&
1031 !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
1032 return 0;
1033
1034 err = -ENOBUFS;
1035 skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
1036 if (skb == NULL)
1037 goto err;
1038
1039 err = nf_tables_fill_chain_info(skb, ctx->net, ctx->portid, ctx->seq,
1040 event, 0, ctx->afi->family, ctx->table,
1041 ctx->chain);
1042 if (err < 0) {
1043 kfree_skb(skb);
1044 goto err;
1045 }
1046
1047 err = nfnetlink_send(skb, ctx->net, ctx->portid, NFNLGRP_NFTABLES,
1048 ctx->report, GFP_KERNEL);
1049err:
1050 if (err < 0) {
1051 nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES,
1052 err);
1053 }
1054 return err;
1055}
1056
1057static int nf_tables_dump_chains(struct sk_buff *skb,
1058 struct netlink_callback *cb)
1059{
1060 const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
1061 const struct nft_af_info *afi;
1062 const struct nft_table *table;
1063 const struct nft_chain *chain;
1064 unsigned int idx = 0, s_idx = cb->args[0];
1065 struct net *net = sock_net(skb->sk);
1066 int family = nfmsg->nfgen_family;
1067
1068 rcu_read_lock();
1069 cb->seq = net->nft.base_seq;
1070
1071 list_for_each_entry_rcu(afi, &net->nft.af_info, list) {
1072 if (family != NFPROTO_UNSPEC && family != afi->family)
1073 continue;
1074
1075 list_for_each_entry_rcu(table, &afi->tables, list) {
1076 list_for_each_entry_rcu(chain, &table->chains, list) {
1077 if (idx < s_idx)
1078 goto cont;
1079 if (idx > s_idx)
1080 memset(&cb->args[1], 0,
1081 sizeof(cb->args) - sizeof(cb->args[0]));
1082 if (nf_tables_fill_chain_info(skb, net,
1083 NETLINK_CB(cb->skb).portid,
1084 cb->nlh->nlmsg_seq,
1085 NFT_MSG_NEWCHAIN,
1086 NLM_F_MULTI,
1087 afi->family, table, chain) < 0)
1088 goto done;
1089
1090 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
1091cont:
1092 idx++;
1093 }
1094 }
1095 }
1096done:
1097 rcu_read_unlock();
1098 cb->args[0] = idx;
1099 return skb->len;
1100}
1101
1102static int nf_tables_getchain(struct sock *nlsk, struct sk_buff *skb,
1103 const struct nlmsghdr *nlh,
1104 const struct nlattr * const nla[])
1105{
1106 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1107 const struct nft_af_info *afi;
1108 const struct nft_table *table;
1109 const struct nft_chain *chain;
1110 struct sk_buff *skb2;
1111 struct net *net = sock_net(skb->sk);
1112 int family = nfmsg->nfgen_family;
1113 int err;
1114
1115 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1116 struct netlink_dump_control c = {
1117 .dump = nf_tables_dump_chains,
1118 };
1119 return netlink_dump_start(nlsk, skb, nlh, &c);
1120 }
1121
1122 afi = nf_tables_afinfo_lookup(net, family, false);
1123 if (IS_ERR(afi))
1124 return PTR_ERR(afi);
1125
1126 table = nf_tables_table_lookup(afi, nla[NFTA_CHAIN_TABLE]);
1127 if (IS_ERR(table))
1128 return PTR_ERR(table);
1129 if (table->flags & NFT_TABLE_INACTIVE)
1130 return -ENOENT;
1131
1132 chain = nf_tables_chain_lookup(table, nla[NFTA_CHAIN_NAME]);
1133 if (IS_ERR(chain))
1134 return PTR_ERR(chain);
1135 if (chain->flags & NFT_CHAIN_INACTIVE)
1136 return -ENOENT;
1137
1138 skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1139 if (!skb2)
1140 return -ENOMEM;
1141
1142 err = nf_tables_fill_chain_info(skb2, net, NETLINK_CB(skb).portid,
1143 nlh->nlmsg_seq, NFT_MSG_NEWCHAIN, 0,
1144 family, table, chain);
1145 if (err < 0)
1146 goto err;
1147
1148 return nlmsg_unicast(nlsk, skb2, NETLINK_CB(skb).portid);
1149
1150err:
1151 kfree_skb(skb2);
1152 return err;
1153}
1154
1155static const struct nla_policy nft_counter_policy[NFTA_COUNTER_MAX + 1] = {
1156 [NFTA_COUNTER_PACKETS] = { .type = NLA_U64 },
1157 [NFTA_COUNTER_BYTES] = { .type = NLA_U64 },
1158};
1159
1160static struct nft_stats __percpu *nft_stats_alloc(const struct nlattr *attr)
1161{
1162 struct nlattr *tb[NFTA_COUNTER_MAX+1];
1163 struct nft_stats __percpu *newstats;
1164 struct nft_stats *stats;
1165 int err;
1166
1167 err = nla_parse_nested(tb, NFTA_COUNTER_MAX, attr, nft_counter_policy);
1168 if (err < 0)
1169 return ERR_PTR(err);
1170
1171 if (!tb[NFTA_COUNTER_BYTES] || !tb[NFTA_COUNTER_PACKETS])
1172 return ERR_PTR(-EINVAL);
1173
1174 newstats = netdev_alloc_pcpu_stats(struct nft_stats);
1175 if (newstats == NULL)
1176 return ERR_PTR(-ENOMEM);
1177
1178
1179
1180
1181 preempt_disable();
1182 stats = this_cpu_ptr(newstats);
1183 stats->bytes = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_BYTES]));
1184 stats->pkts = be64_to_cpu(nla_get_be64(tb[NFTA_COUNTER_PACKETS]));
1185 preempt_enable();
1186
1187 return newstats;
1188}
1189
1190static void nft_chain_stats_replace(struct nft_base_chain *chain,
1191 struct nft_stats __percpu *newstats)
1192{
1193 if (newstats == NULL)
1194 return;
1195
1196 if (chain->stats) {
1197 struct nft_stats __percpu *oldstats =
1198 nft_dereference(chain->stats);
1199
1200 rcu_assign_pointer(chain->stats, newstats);
1201 synchronize_rcu();
1202 free_percpu(oldstats);
1203 } else
1204 rcu_assign_pointer(chain->stats, newstats);
1205}
1206
1207static void nf_tables_chain_destroy(struct nft_chain *chain)
1208{
1209 BUG_ON(chain->use > 0);
1210
1211 if (chain->flags & NFT_BASE_CHAIN) {
1212 struct nft_base_chain *basechain = nft_base_chain(chain);
1213
1214 module_put(basechain->type->owner);
1215 free_percpu(basechain->stats);
1216 if (basechain->ops[0].dev != NULL)
1217 dev_put(basechain->ops[0].dev);
1218 kfree(basechain);
1219 } else {
1220 kfree(chain);
1221 }
1222}
1223
1224static int nf_tables_newchain(struct sock *nlsk, struct sk_buff *skb,
1225 const struct nlmsghdr *nlh,
1226 const struct nlattr * const nla[])
1227{
1228 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1229 const struct nlattr * uninitialized_var(name);
1230 struct nft_af_info *afi;
1231 struct nft_table *table;
1232 struct nft_chain *chain;
1233 struct nft_base_chain *basechain = NULL;
1234 struct nlattr *ha[NFTA_HOOK_MAX + 1];
1235 struct net *net = sock_net(skb->sk);
1236 int family = nfmsg->nfgen_family;
1237 struct net_device *dev = NULL;
1238 u8 policy = NF_ACCEPT;
1239 u64 handle = 0;
1240 unsigned int i;
1241 struct nft_stats __percpu *stats;
1242 int err;
1243 bool create;
1244 struct nft_ctx ctx;
1245
1246 create = nlh->nlmsg_flags & NLM_F_CREATE ? true : false;
1247
1248 afi = nf_tables_afinfo_lookup(net, family, true);
1249 if (IS_ERR(afi))
1250 return PTR_ERR(afi);
1251
1252 table = nf_tables_table_lookup(afi, nla[NFTA_CHAIN_TABLE]);
1253 if (IS_ERR(table))
1254 return PTR_ERR(table);
1255
1256 chain = NULL;
1257 name = nla[NFTA_CHAIN_NAME];
1258
1259 if (nla[NFTA_CHAIN_HANDLE]) {
1260 handle = be64_to_cpu(nla_get_be64(nla[NFTA_CHAIN_HANDLE]));
1261 chain = nf_tables_chain_lookup_byhandle(table, handle);
1262 if (IS_ERR(chain))
1263 return PTR_ERR(chain);
1264 } else {
1265 chain = nf_tables_chain_lookup(table, name);
1266 if (IS_ERR(chain)) {
1267 if (PTR_ERR(chain) != -ENOENT)
1268 return PTR_ERR(chain);
1269 chain = NULL;
1270 }
1271 }
1272
1273 if (nla[NFTA_CHAIN_POLICY]) {
1274 if ((chain != NULL &&
1275 !(chain->flags & NFT_BASE_CHAIN)))
1276 return -EOPNOTSUPP;
1277
1278 if (chain == NULL &&
1279 nla[NFTA_CHAIN_HOOK] == NULL)
1280 return -EOPNOTSUPP;
1281
1282 policy = ntohl(nla_get_be32(nla[NFTA_CHAIN_POLICY]));
1283 switch (policy) {
1284 case NF_DROP:
1285 case NF_ACCEPT:
1286 break;
1287 default:
1288 return -EINVAL;
1289 }
1290 }
1291
1292 if (chain != NULL) {
1293 struct nft_stats *stats = NULL;
1294 struct nft_trans *trans;
1295
1296 if (chain->flags & NFT_CHAIN_INACTIVE)
1297 return -ENOENT;
1298 if (nlh->nlmsg_flags & NLM_F_EXCL)
1299 return -EEXIST;
1300 if (nlh->nlmsg_flags & NLM_F_REPLACE)
1301 return -EOPNOTSUPP;
1302
1303 if (nla[NFTA_CHAIN_HANDLE] && name &&
1304 !IS_ERR(nf_tables_chain_lookup(table, nla[NFTA_CHAIN_NAME])))
1305 return -EEXIST;
1306
1307 if (nla[NFTA_CHAIN_COUNTERS]) {
1308 if (!(chain->flags & NFT_BASE_CHAIN))
1309 return -EOPNOTSUPP;
1310
1311 stats = nft_stats_alloc(nla[NFTA_CHAIN_COUNTERS]);
1312 if (IS_ERR(stats))
1313 return PTR_ERR(stats);
1314 }
1315
1316 nft_ctx_init(&ctx, skb, nlh, afi, table, chain, nla);
1317 trans = nft_trans_alloc(&ctx, NFT_MSG_NEWCHAIN,
1318 sizeof(struct nft_trans_chain));
1319 if (trans == NULL) {
1320 free_percpu(stats);
1321 return -ENOMEM;
1322 }
1323
1324 nft_trans_chain_stats(trans) = stats;
1325 nft_trans_chain_update(trans) = true;
1326
1327 if (nla[NFTA_CHAIN_POLICY])
1328 nft_trans_chain_policy(trans) = policy;
1329 else
1330 nft_trans_chain_policy(trans) = -1;
1331
1332 if (nla[NFTA_CHAIN_HANDLE] && name) {
1333 nla_strlcpy(nft_trans_chain_name(trans), name,
1334 NFT_CHAIN_MAXNAMELEN);
1335 }
1336 list_add_tail(&trans->list, &net->nft.commit_list);
1337 return 0;
1338 }
1339
1340 if (table->use == UINT_MAX)
1341 return -EOVERFLOW;
1342
1343 if (nla[NFTA_CHAIN_HOOK]) {
1344 const struct nf_chain_type *type;
1345 struct nf_hook_ops *ops;
1346 nf_hookfn *hookfn;
1347 u32 hooknum, priority;
1348
1349 type = chain_type[family][NFT_CHAIN_T_DEFAULT];
1350 if (nla[NFTA_CHAIN_TYPE]) {
1351 type = nf_tables_chain_type_lookup(afi,
1352 nla[NFTA_CHAIN_TYPE],
1353 create);
1354 if (IS_ERR(type))
1355 return PTR_ERR(type);
1356 }
1357
1358 err = nla_parse_nested(ha, NFTA_HOOK_MAX, nla[NFTA_CHAIN_HOOK],
1359 nft_hook_policy);
1360 if (err < 0)
1361 return err;
1362 if (ha[NFTA_HOOK_HOOKNUM] == NULL ||
1363 ha[NFTA_HOOK_PRIORITY] == NULL)
1364 return -EINVAL;
1365
1366 hooknum = ntohl(nla_get_be32(ha[NFTA_HOOK_HOOKNUM]));
1367 if (hooknum >= afi->nhooks)
1368 return -EINVAL;
1369 priority = ntohl(nla_get_be32(ha[NFTA_HOOK_PRIORITY]));
1370
1371 if (!(type->hook_mask & (1 << hooknum)))
1372 return -EOPNOTSUPP;
1373 if (!try_module_get(type->owner))
1374 return -ENOENT;
1375 hookfn = type->hooks[hooknum];
1376
1377 if (afi->flags & NFT_AF_NEEDS_DEV) {
1378 char ifname[IFNAMSIZ];
1379
1380 if (!ha[NFTA_HOOK_DEV]) {
1381 module_put(type->owner);
1382 return -EOPNOTSUPP;
1383 }
1384
1385 nla_strlcpy(ifname, ha[NFTA_HOOK_DEV], IFNAMSIZ);
1386 dev = dev_get_by_name(net, ifname);
1387 if (!dev) {
1388 module_put(type->owner);
1389 return -ENOENT;
1390 }
1391 } else if (ha[NFTA_HOOK_DEV]) {
1392 module_put(type->owner);
1393 return -EOPNOTSUPP;
1394 }
1395
1396 basechain = kzalloc(sizeof(*basechain), GFP_KERNEL);
1397 if (basechain == NULL) {
1398 module_put(type->owner);
1399 if (dev != NULL)
1400 dev_put(dev);
1401 return -ENOMEM;
1402 }
1403
1404 if (dev != NULL)
1405 strncpy(basechain->dev_name, dev->name, IFNAMSIZ);
1406
1407 if (nla[NFTA_CHAIN_COUNTERS]) {
1408 stats = nft_stats_alloc(nla[NFTA_CHAIN_COUNTERS]);
1409 if (IS_ERR(stats)) {
1410 module_put(type->owner);
1411 kfree(basechain);
1412 if (dev != NULL)
1413 dev_put(dev);
1414 return PTR_ERR(stats);
1415 }
1416 basechain->stats = stats;
1417 } else {
1418 stats = netdev_alloc_pcpu_stats(struct nft_stats);
1419 if (stats == NULL) {
1420 module_put(type->owner);
1421 kfree(basechain);
1422 if (dev != NULL)
1423 dev_put(dev);
1424 return -ENOMEM;
1425 }
1426 rcu_assign_pointer(basechain->stats, stats);
1427 }
1428
1429 write_pnet(&basechain->pnet, net);
1430 basechain->type = type;
1431 chain = &basechain->chain;
1432
1433 for (i = 0; i < afi->nops; i++) {
1434 ops = &basechain->ops[i];
1435 ops->pf = family;
1436 ops->owner = afi->owner;
1437 ops->hooknum = hooknum;
1438 ops->priority = priority;
1439 ops->priv = chain;
1440 ops->hook = afi->hooks[ops->hooknum];
1441 ops->dev = dev;
1442 if (hookfn)
1443 ops->hook = hookfn;
1444 if (afi->hook_ops_init)
1445 afi->hook_ops_init(ops, i);
1446 }
1447
1448 chain->flags |= NFT_BASE_CHAIN;
1449 basechain->policy = policy;
1450 } else {
1451 chain = kzalloc(sizeof(*chain), GFP_KERNEL);
1452 if (chain == NULL)
1453 return -ENOMEM;
1454 }
1455
1456 INIT_LIST_HEAD(&chain->rules);
1457 chain->handle = nf_tables_alloc_handle(table);
1458 chain->table = table;
1459 nla_strlcpy(chain->name, name, NFT_CHAIN_MAXNAMELEN);
1460
1461 err = nf_tables_register_hooks(table, chain, afi->nops);
1462 if (err < 0)
1463 goto err1;
1464
1465 nft_ctx_init(&ctx, skb, nlh, afi, table, chain, nla);
1466 err = nft_trans_chain_add(&ctx, NFT_MSG_NEWCHAIN);
1467 if (err < 0)
1468 goto err2;
1469
1470 table->use++;
1471 list_add_tail_rcu(&chain->list, &table->chains);
1472 return 0;
1473err2:
1474 nf_tables_unregister_hooks(table, chain, afi->nops);
1475err1:
1476 nf_tables_chain_destroy(chain);
1477 return err;
1478}
1479
1480static int nf_tables_delchain(struct sock *nlsk, struct sk_buff *skb,
1481 const struct nlmsghdr *nlh,
1482 const struct nlattr * const nla[])
1483{
1484 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1485 struct nft_af_info *afi;
1486 struct nft_table *table;
1487 struct nft_chain *chain;
1488 struct net *net = sock_net(skb->sk);
1489 int family = nfmsg->nfgen_family;
1490 struct nft_ctx ctx;
1491
1492 afi = nf_tables_afinfo_lookup(net, family, false);
1493 if (IS_ERR(afi))
1494 return PTR_ERR(afi);
1495
1496 table = nf_tables_table_lookup(afi, nla[NFTA_CHAIN_TABLE]);
1497 if (IS_ERR(table))
1498 return PTR_ERR(table);
1499 if (table->flags & NFT_TABLE_INACTIVE)
1500 return -ENOENT;
1501
1502 chain = nf_tables_chain_lookup(table, nla[NFTA_CHAIN_NAME]);
1503 if (IS_ERR(chain))
1504 return PTR_ERR(chain);
1505 if (chain->flags & NFT_CHAIN_INACTIVE)
1506 return -ENOENT;
1507 if (chain->use > 0)
1508 return -EBUSY;
1509
1510 nft_ctx_init(&ctx, skb, nlh, afi, table, chain, nla);
1511
1512 return nft_delchain(&ctx);
1513}
1514
1515
1516
1517
1518
1519
1520
1521
1522
1523
1524
1525
1526int nft_register_expr(struct nft_expr_type *type)
1527{
1528 nfnl_lock(NFNL_SUBSYS_NFTABLES);
1529 if (type->family == NFPROTO_UNSPEC)
1530 list_add_tail_rcu(&type->list, &nf_tables_expressions);
1531 else
1532 list_add_rcu(&type->list, &nf_tables_expressions);
1533 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1534 return 0;
1535}
1536EXPORT_SYMBOL_GPL(nft_register_expr);
1537
1538
1539
1540
1541
1542
1543
1544void nft_unregister_expr(struct nft_expr_type *type)
1545{
1546 nfnl_lock(NFNL_SUBSYS_NFTABLES);
1547 list_del_rcu(&type->list);
1548 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1549}
1550EXPORT_SYMBOL_GPL(nft_unregister_expr);
1551
1552static const struct nft_expr_type *__nft_expr_type_get(u8 family,
1553 struct nlattr *nla)
1554{
1555 const struct nft_expr_type *type;
1556
1557 list_for_each_entry(type, &nf_tables_expressions, list) {
1558 if (!nla_strcmp(nla, type->name) &&
1559 (!type->family || type->family == family))
1560 return type;
1561 }
1562 return NULL;
1563}
1564
1565static const struct nft_expr_type *nft_expr_type_get(u8 family,
1566 struct nlattr *nla)
1567{
1568 const struct nft_expr_type *type;
1569
1570 if (nla == NULL)
1571 return ERR_PTR(-EINVAL);
1572
1573 type = __nft_expr_type_get(family, nla);
1574 if (type != NULL && try_module_get(type->owner))
1575 return type;
1576
1577#ifdef CONFIG_MODULES
1578 if (type == NULL) {
1579 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1580 request_module("nft-expr-%u-%.*s", family,
1581 nla_len(nla), (char *)nla_data(nla));
1582 nfnl_lock(NFNL_SUBSYS_NFTABLES);
1583 if (__nft_expr_type_get(family, nla))
1584 return ERR_PTR(-EAGAIN);
1585
1586 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
1587 request_module("nft-expr-%.*s",
1588 nla_len(nla), (char *)nla_data(nla));
1589 nfnl_lock(NFNL_SUBSYS_NFTABLES);
1590 if (__nft_expr_type_get(family, nla))
1591 return ERR_PTR(-EAGAIN);
1592 }
1593#endif
1594 return ERR_PTR(-ENOENT);
1595}
1596
1597static const struct nla_policy nft_expr_policy[NFTA_EXPR_MAX + 1] = {
1598 [NFTA_EXPR_NAME] = { .type = NLA_STRING },
1599 [NFTA_EXPR_DATA] = { .type = NLA_NESTED },
1600};
1601
1602static int nf_tables_fill_expr_info(struct sk_buff *skb,
1603 const struct nft_expr *expr)
1604{
1605 if (nla_put_string(skb, NFTA_EXPR_NAME, expr->ops->type->name))
1606 goto nla_put_failure;
1607
1608 if (expr->ops->dump) {
1609 struct nlattr *data = nla_nest_start(skb, NFTA_EXPR_DATA);
1610 if (data == NULL)
1611 goto nla_put_failure;
1612 if (expr->ops->dump(skb, expr) < 0)
1613 goto nla_put_failure;
1614 nla_nest_end(skb, data);
1615 }
1616
1617 return skb->len;
1618
1619nla_put_failure:
1620 return -1;
1621};
1622
1623int nft_expr_dump(struct sk_buff *skb, unsigned int attr,
1624 const struct nft_expr *expr)
1625{
1626 struct nlattr *nest;
1627
1628 nest = nla_nest_start(skb, attr);
1629 if (!nest)
1630 goto nla_put_failure;
1631 if (nf_tables_fill_expr_info(skb, expr) < 0)
1632 goto nla_put_failure;
1633 nla_nest_end(skb, nest);
1634 return 0;
1635
1636nla_put_failure:
1637 return -1;
1638}
1639
1640struct nft_expr_info {
1641 const struct nft_expr_ops *ops;
1642 struct nlattr *tb[NFT_EXPR_MAXATTR + 1];
1643};
1644
1645static int nf_tables_expr_parse(const struct nft_ctx *ctx,
1646 const struct nlattr *nla,
1647 struct nft_expr_info *info)
1648{
1649 const struct nft_expr_type *type;
1650 const struct nft_expr_ops *ops;
1651 struct nlattr *tb[NFTA_EXPR_MAX + 1];
1652 int err;
1653
1654 err = nla_parse_nested(tb, NFTA_EXPR_MAX, nla, nft_expr_policy);
1655 if (err < 0)
1656 return err;
1657
1658 type = nft_expr_type_get(ctx->afi->family, tb[NFTA_EXPR_NAME]);
1659 if (IS_ERR(type))
1660 return PTR_ERR(type);
1661
1662 if (tb[NFTA_EXPR_DATA]) {
1663 err = nla_parse_nested(info->tb, type->maxattr,
1664 tb[NFTA_EXPR_DATA], type->policy);
1665 if (err < 0)
1666 goto err1;
1667 } else
1668 memset(info->tb, 0, sizeof(info->tb[0]) * (type->maxattr + 1));
1669
1670 if (type->select_ops != NULL) {
1671 ops = type->select_ops(ctx,
1672 (const struct nlattr * const *)info->tb);
1673 if (IS_ERR(ops)) {
1674 err = PTR_ERR(ops);
1675 goto err1;
1676 }
1677 } else
1678 ops = type->ops;
1679
1680 info->ops = ops;
1681 return 0;
1682
1683err1:
1684 module_put(type->owner);
1685 return err;
1686}
1687
1688static int nf_tables_newexpr(const struct nft_ctx *ctx,
1689 const struct nft_expr_info *info,
1690 struct nft_expr *expr)
1691{
1692 const struct nft_expr_ops *ops = info->ops;
1693 int err;
1694
1695 expr->ops = ops;
1696 if (ops->init) {
1697 err = ops->init(ctx, expr, (const struct nlattr **)info->tb);
1698 if (err < 0)
1699 goto err1;
1700 }
1701
1702 return 0;
1703
1704err1:
1705 expr->ops = NULL;
1706 return err;
1707}
1708
1709static void nf_tables_expr_destroy(const struct nft_ctx *ctx,
1710 struct nft_expr *expr)
1711{
1712 if (expr->ops->destroy)
1713 expr->ops->destroy(ctx, expr);
1714 module_put(expr->ops->type->owner);
1715}
1716
1717struct nft_expr *nft_expr_init(const struct nft_ctx *ctx,
1718 const struct nlattr *nla)
1719{
1720 struct nft_expr_info info;
1721 struct nft_expr *expr;
1722 int err;
1723
1724 err = nf_tables_expr_parse(ctx, nla, &info);
1725 if (err < 0)
1726 goto err1;
1727
1728 err = -ENOMEM;
1729 expr = kzalloc(info.ops->size, GFP_KERNEL);
1730 if (expr == NULL)
1731 goto err2;
1732
1733 err = nf_tables_newexpr(ctx, &info, expr);
1734 if (err < 0)
1735 goto err2;
1736
1737 return expr;
1738err2:
1739 module_put(info.ops->type->owner);
1740err1:
1741 return ERR_PTR(err);
1742}
1743
1744void nft_expr_destroy(const struct nft_ctx *ctx, struct nft_expr *expr)
1745{
1746 nf_tables_expr_destroy(ctx, expr);
1747 kfree(expr);
1748}
1749
1750
1751
1752
1753
1754static struct nft_rule *__nf_tables_rule_lookup(const struct nft_chain *chain,
1755 u64 handle)
1756{
1757 struct nft_rule *rule;
1758
1759
1760 list_for_each_entry(rule, &chain->rules, list) {
1761 if (handle == rule->handle)
1762 return rule;
1763 }
1764
1765 return ERR_PTR(-ENOENT);
1766}
1767
1768static struct nft_rule *nf_tables_rule_lookup(const struct nft_chain *chain,
1769 const struct nlattr *nla)
1770{
1771 if (nla == NULL)
1772 return ERR_PTR(-EINVAL);
1773
1774 return __nf_tables_rule_lookup(chain, be64_to_cpu(nla_get_be64(nla)));
1775}
1776
1777static const struct nla_policy nft_rule_policy[NFTA_RULE_MAX + 1] = {
1778 [NFTA_RULE_TABLE] = { .type = NLA_STRING },
1779 [NFTA_RULE_CHAIN] = { .type = NLA_STRING,
1780 .len = NFT_CHAIN_MAXNAMELEN - 1 },
1781 [NFTA_RULE_HANDLE] = { .type = NLA_U64 },
1782 [NFTA_RULE_EXPRESSIONS] = { .type = NLA_NESTED },
1783 [NFTA_RULE_COMPAT] = { .type = NLA_NESTED },
1784 [NFTA_RULE_POSITION] = { .type = NLA_U64 },
1785 [NFTA_RULE_USERDATA] = { .type = NLA_BINARY,
1786 .len = NFT_USERDATA_MAXLEN },
1787};
1788
1789static int nf_tables_fill_rule_info(struct sk_buff *skb, struct net *net,
1790 u32 portid, u32 seq, int event,
1791 u32 flags, int family,
1792 const struct nft_table *table,
1793 const struct nft_chain *chain,
1794 const struct nft_rule *rule)
1795{
1796 struct nlmsghdr *nlh;
1797 struct nfgenmsg *nfmsg;
1798 const struct nft_expr *expr, *next;
1799 struct nlattr *list;
1800 const struct nft_rule *prule;
1801 int type = event | NFNL_SUBSYS_NFTABLES << 8;
1802
1803 nlh = nlmsg_put(skb, portid, seq, type, sizeof(struct nfgenmsg),
1804 flags);
1805 if (nlh == NULL)
1806 goto nla_put_failure;
1807
1808 nfmsg = nlmsg_data(nlh);
1809 nfmsg->nfgen_family = family;
1810 nfmsg->version = NFNETLINK_V0;
1811 nfmsg->res_id = htons(net->nft.base_seq & 0xffff);
1812
1813 if (nla_put_string(skb, NFTA_RULE_TABLE, table->name))
1814 goto nla_put_failure;
1815 if (nla_put_string(skb, NFTA_RULE_CHAIN, chain->name))
1816 goto nla_put_failure;
1817 if (nla_put_be64(skb, NFTA_RULE_HANDLE, cpu_to_be64(rule->handle)))
1818 goto nla_put_failure;
1819
1820 if ((event != NFT_MSG_DELRULE) && (rule->list.prev != &chain->rules)) {
1821 prule = list_entry(rule->list.prev, struct nft_rule, list);
1822 if (nla_put_be64(skb, NFTA_RULE_POSITION,
1823 cpu_to_be64(prule->handle)))
1824 goto nla_put_failure;
1825 }
1826
1827 list = nla_nest_start(skb, NFTA_RULE_EXPRESSIONS);
1828 if (list == NULL)
1829 goto nla_put_failure;
1830 nft_rule_for_each_expr(expr, next, rule) {
1831 if (nft_expr_dump(skb, NFTA_LIST_ELEM, expr) < 0)
1832 goto nla_put_failure;
1833 }
1834 nla_nest_end(skb, list);
1835
1836 if (rule->udata) {
1837 struct nft_userdata *udata = nft_userdata(rule);
1838 if (nla_put(skb, NFTA_RULE_USERDATA, udata->len + 1,
1839 udata->data) < 0)
1840 goto nla_put_failure;
1841 }
1842
1843 nlmsg_end(skb, nlh);
1844 return 0;
1845
1846nla_put_failure:
1847 nlmsg_trim(skb, nlh);
1848 return -1;
1849}
1850
1851static int nf_tables_rule_notify(const struct nft_ctx *ctx,
1852 const struct nft_rule *rule,
1853 int event)
1854{
1855 struct sk_buff *skb;
1856 int err;
1857
1858 if (!ctx->report &&
1859 !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
1860 return 0;
1861
1862 err = -ENOBUFS;
1863 skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
1864 if (skb == NULL)
1865 goto err;
1866
1867 err = nf_tables_fill_rule_info(skb, ctx->net, ctx->portid, ctx->seq,
1868 event, 0, ctx->afi->family, ctx->table,
1869 ctx->chain, rule);
1870 if (err < 0) {
1871 kfree_skb(skb);
1872 goto err;
1873 }
1874
1875 err = nfnetlink_send(skb, ctx->net, ctx->portid, NFNLGRP_NFTABLES,
1876 ctx->report, GFP_KERNEL);
1877err:
1878 if (err < 0) {
1879 nfnetlink_set_err(ctx->net, ctx->portid, NFNLGRP_NFTABLES,
1880 err);
1881 }
1882 return err;
1883}
1884
1885static int nf_tables_dump_rules(struct sk_buff *skb,
1886 struct netlink_callback *cb)
1887{
1888 const struct nfgenmsg *nfmsg = nlmsg_data(cb->nlh);
1889 const struct nft_af_info *afi;
1890 const struct nft_table *table;
1891 const struct nft_chain *chain;
1892 const struct nft_rule *rule;
1893 unsigned int idx = 0, s_idx = cb->args[0];
1894 struct net *net = sock_net(skb->sk);
1895 int family = nfmsg->nfgen_family;
1896
1897 rcu_read_lock();
1898 cb->seq = net->nft.base_seq;
1899
1900 list_for_each_entry_rcu(afi, &net->nft.af_info, list) {
1901 if (family != NFPROTO_UNSPEC && family != afi->family)
1902 continue;
1903
1904 list_for_each_entry_rcu(table, &afi->tables, list) {
1905 list_for_each_entry_rcu(chain, &table->chains, list) {
1906 list_for_each_entry_rcu(rule, &chain->rules, list) {
1907 if (!nft_rule_is_active(net, rule))
1908 goto cont;
1909 if (idx < s_idx)
1910 goto cont;
1911 if (idx > s_idx)
1912 memset(&cb->args[1], 0,
1913 sizeof(cb->args) - sizeof(cb->args[0]));
1914 if (nf_tables_fill_rule_info(skb, net, NETLINK_CB(cb->skb).portid,
1915 cb->nlh->nlmsg_seq,
1916 NFT_MSG_NEWRULE,
1917 NLM_F_MULTI | NLM_F_APPEND,
1918 afi->family, table, chain, rule) < 0)
1919 goto done;
1920
1921 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
1922cont:
1923 idx++;
1924 }
1925 }
1926 }
1927 }
1928done:
1929 rcu_read_unlock();
1930
1931 cb->args[0] = idx;
1932 return skb->len;
1933}
1934
1935static int nf_tables_getrule(struct sock *nlsk, struct sk_buff *skb,
1936 const struct nlmsghdr *nlh,
1937 const struct nlattr * const nla[])
1938{
1939 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
1940 const struct nft_af_info *afi;
1941 const struct nft_table *table;
1942 const struct nft_chain *chain;
1943 const struct nft_rule *rule;
1944 struct sk_buff *skb2;
1945 struct net *net = sock_net(skb->sk);
1946 int family = nfmsg->nfgen_family;
1947 int err;
1948
1949 if (nlh->nlmsg_flags & NLM_F_DUMP) {
1950 struct netlink_dump_control c = {
1951 .dump = nf_tables_dump_rules,
1952 };
1953 return netlink_dump_start(nlsk, skb, nlh, &c);
1954 }
1955
1956 afi = nf_tables_afinfo_lookup(net, family, false);
1957 if (IS_ERR(afi))
1958 return PTR_ERR(afi);
1959
1960 table = nf_tables_table_lookup(afi, nla[NFTA_RULE_TABLE]);
1961 if (IS_ERR(table))
1962 return PTR_ERR(table);
1963 if (table->flags & NFT_TABLE_INACTIVE)
1964 return -ENOENT;
1965
1966 chain = nf_tables_chain_lookup(table, nla[NFTA_RULE_CHAIN]);
1967 if (IS_ERR(chain))
1968 return PTR_ERR(chain);
1969 if (chain->flags & NFT_CHAIN_INACTIVE)
1970 return -ENOENT;
1971
1972 rule = nf_tables_rule_lookup(chain, nla[NFTA_RULE_HANDLE]);
1973 if (IS_ERR(rule))
1974 return PTR_ERR(rule);
1975
1976 skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1977 if (!skb2)
1978 return -ENOMEM;
1979
1980 err = nf_tables_fill_rule_info(skb2, net, NETLINK_CB(skb).portid,
1981 nlh->nlmsg_seq, NFT_MSG_NEWRULE, 0,
1982 family, table, chain, rule);
1983 if (err < 0)
1984 goto err;
1985
1986 return nlmsg_unicast(nlsk, skb2, NETLINK_CB(skb).portid);
1987
1988err:
1989 kfree_skb(skb2);
1990 return err;
1991}
1992
1993static void nf_tables_rule_destroy(const struct nft_ctx *ctx,
1994 struct nft_rule *rule)
1995{
1996 struct nft_expr *expr;
1997
1998
1999
2000
2001
2002 expr = nft_expr_first(rule);
2003 while (expr->ops && expr != nft_expr_last(rule)) {
2004 nf_tables_expr_destroy(ctx, expr);
2005 expr = nft_expr_next(expr);
2006 }
2007 kfree(rule);
2008}
2009
2010#define NFT_RULE_MAXEXPRS 128
2011
2012static struct nft_expr_info *info;
2013
2014static int nf_tables_newrule(struct sock *nlsk, struct sk_buff *skb,
2015 const struct nlmsghdr *nlh,
2016 const struct nlattr * const nla[])
2017{
2018 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2019 struct nft_af_info *afi;
2020 struct net *net = sock_net(skb->sk);
2021 struct nft_table *table;
2022 struct nft_chain *chain;
2023 struct nft_rule *rule, *old_rule = NULL;
2024 struct nft_userdata *udata;
2025 struct nft_trans *trans = NULL;
2026 struct nft_expr *expr;
2027 struct nft_ctx ctx;
2028 struct nlattr *tmp;
2029 unsigned int size, i, n, ulen = 0, usize = 0;
2030 int err, rem;
2031 bool create;
2032 u64 handle, pos_handle;
2033
2034 create = nlh->nlmsg_flags & NLM_F_CREATE ? true : false;
2035
2036 afi = nf_tables_afinfo_lookup(net, nfmsg->nfgen_family, create);
2037 if (IS_ERR(afi))
2038 return PTR_ERR(afi);
2039
2040 table = nf_tables_table_lookup(afi, nla[NFTA_RULE_TABLE]);
2041 if (IS_ERR(table))
2042 return PTR_ERR(table);
2043
2044 chain = nf_tables_chain_lookup(table, nla[NFTA_RULE_CHAIN]);
2045 if (IS_ERR(chain))
2046 return PTR_ERR(chain);
2047
2048 if (nla[NFTA_RULE_HANDLE]) {
2049 handle = be64_to_cpu(nla_get_be64(nla[NFTA_RULE_HANDLE]));
2050 rule = __nf_tables_rule_lookup(chain, handle);
2051 if (IS_ERR(rule))
2052 return PTR_ERR(rule);
2053
2054 if (nlh->nlmsg_flags & NLM_F_EXCL)
2055 return -EEXIST;
2056 if (nlh->nlmsg_flags & NLM_F_REPLACE)
2057 old_rule = rule;
2058 else
2059 return -EOPNOTSUPP;
2060 } else {
2061 if (!create || nlh->nlmsg_flags & NLM_F_REPLACE)
2062 return -EINVAL;
2063 handle = nf_tables_alloc_handle(table);
2064
2065 if (chain->use == UINT_MAX)
2066 return -EOVERFLOW;
2067 }
2068
2069 if (nla[NFTA_RULE_POSITION]) {
2070 if (!(nlh->nlmsg_flags & NLM_F_CREATE))
2071 return -EOPNOTSUPP;
2072
2073 pos_handle = be64_to_cpu(nla_get_be64(nla[NFTA_RULE_POSITION]));
2074 old_rule = __nf_tables_rule_lookup(chain, pos_handle);
2075 if (IS_ERR(old_rule))
2076 return PTR_ERR(old_rule);
2077 }
2078
2079 nft_ctx_init(&ctx, skb, nlh, afi, table, chain, nla);
2080
2081 n = 0;
2082 size = 0;
2083 if (nla[NFTA_RULE_EXPRESSIONS]) {
2084 nla_for_each_nested(tmp, nla[NFTA_RULE_EXPRESSIONS], rem) {
2085 err = -EINVAL;
2086 if (nla_type(tmp) != NFTA_LIST_ELEM)
2087 goto err1;
2088 if (n == NFT_RULE_MAXEXPRS)
2089 goto err1;
2090 err = nf_tables_expr_parse(&ctx, tmp, &info[n]);
2091 if (err < 0)
2092 goto err1;
2093 size += info[n].ops->size;
2094 n++;
2095 }
2096 }
2097
2098 err = -EFBIG;
2099 if (size >= 1 << 12)
2100 goto err1;
2101
2102 if (nla[NFTA_RULE_USERDATA]) {
2103 ulen = nla_len(nla[NFTA_RULE_USERDATA]);
2104 if (ulen > 0)
2105 usize = sizeof(struct nft_userdata) + ulen;
2106 }
2107
2108 err = -ENOMEM;
2109 rule = kzalloc(sizeof(*rule) + size + usize, GFP_KERNEL);
2110 if (rule == NULL)
2111 goto err1;
2112
2113 nft_rule_activate_next(net, rule);
2114
2115 rule->handle = handle;
2116 rule->dlen = size;
2117 rule->udata = ulen ? 1 : 0;
2118
2119 if (ulen) {
2120 udata = nft_userdata(rule);
2121 udata->len = ulen - 1;
2122 nla_memcpy(udata->data, nla[NFTA_RULE_USERDATA], ulen);
2123 }
2124
2125 expr = nft_expr_first(rule);
2126 for (i = 0; i < n; i++) {
2127 err = nf_tables_newexpr(&ctx, &info[i], expr);
2128 if (err < 0)
2129 goto err2;
2130 info[i].ops = NULL;
2131 expr = nft_expr_next(expr);
2132 }
2133
2134 if (nlh->nlmsg_flags & NLM_F_REPLACE) {
2135 if (nft_rule_is_active_next(net, old_rule)) {
2136 trans = nft_trans_rule_add(&ctx, NFT_MSG_DELRULE,
2137 old_rule);
2138 if (trans == NULL) {
2139 err = -ENOMEM;
2140 goto err2;
2141 }
2142 nft_rule_deactivate_next(net, old_rule);
2143 chain->use--;
2144 list_add_tail_rcu(&rule->list, &old_rule->list);
2145 } else {
2146 err = -ENOENT;
2147 goto err2;
2148 }
2149 } else if (nlh->nlmsg_flags & NLM_F_APPEND)
2150 if (old_rule)
2151 list_add_rcu(&rule->list, &old_rule->list);
2152 else
2153 list_add_tail_rcu(&rule->list, &chain->rules);
2154 else {
2155 if (old_rule)
2156 list_add_tail_rcu(&rule->list, &old_rule->list);
2157 else
2158 list_add_rcu(&rule->list, &chain->rules);
2159 }
2160
2161 if (nft_trans_rule_add(&ctx, NFT_MSG_NEWRULE, rule) == NULL) {
2162 err = -ENOMEM;
2163 goto err3;
2164 }
2165 chain->use++;
2166 return 0;
2167
2168err3:
2169 list_del_rcu(&rule->list);
2170err2:
2171 nf_tables_rule_destroy(&ctx, rule);
2172err1:
2173 for (i = 0; i < n; i++) {
2174 if (info[i].ops != NULL)
2175 module_put(info[i].ops->type->owner);
2176 }
2177 return err;
2178}
2179
2180static int nf_tables_delrule(struct sock *nlsk, struct sk_buff *skb,
2181 const struct nlmsghdr *nlh,
2182 const struct nlattr * const nla[])
2183{
2184 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2185 struct nft_af_info *afi;
2186 struct net *net = sock_net(skb->sk);
2187 struct nft_table *table;
2188 struct nft_chain *chain = NULL;
2189 struct nft_rule *rule;
2190 int family = nfmsg->nfgen_family, err = 0;
2191 struct nft_ctx ctx;
2192
2193 afi = nf_tables_afinfo_lookup(net, family, false);
2194 if (IS_ERR(afi))
2195 return PTR_ERR(afi);
2196
2197 table = nf_tables_table_lookup(afi, nla[NFTA_RULE_TABLE]);
2198 if (IS_ERR(table))
2199 return PTR_ERR(table);
2200 if (table->flags & NFT_TABLE_INACTIVE)
2201 return -ENOENT;
2202
2203 if (nla[NFTA_RULE_CHAIN]) {
2204 chain = nf_tables_chain_lookup(table, nla[NFTA_RULE_CHAIN]);
2205 if (IS_ERR(chain))
2206 return PTR_ERR(chain);
2207 }
2208
2209 nft_ctx_init(&ctx, skb, nlh, afi, table, chain, nla);
2210
2211 if (chain) {
2212 if (nla[NFTA_RULE_HANDLE]) {
2213 rule = nf_tables_rule_lookup(chain,
2214 nla[NFTA_RULE_HANDLE]);
2215 if (IS_ERR(rule))
2216 return PTR_ERR(rule);
2217
2218 err = nft_delrule(&ctx, rule);
2219 } else {
2220 err = nft_delrule_by_chain(&ctx);
2221 }
2222 } else {
2223 list_for_each_entry(chain, &table->chains, list) {
2224 ctx.chain = chain;
2225 err = nft_delrule_by_chain(&ctx);
2226 if (err < 0)
2227 break;
2228 }
2229 }
2230
2231 return err;
2232}
2233
2234
2235
2236
2237
2238static LIST_HEAD(nf_tables_set_ops);
2239
2240int nft_register_set(struct nft_set_ops *ops)
2241{
2242 nfnl_lock(NFNL_SUBSYS_NFTABLES);
2243 list_add_tail_rcu(&ops->list, &nf_tables_set_ops);
2244 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
2245 return 0;
2246}
2247EXPORT_SYMBOL_GPL(nft_register_set);
2248
2249void nft_unregister_set(struct nft_set_ops *ops)
2250{
2251 nfnl_lock(NFNL_SUBSYS_NFTABLES);
2252 list_del_rcu(&ops->list);
2253 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
2254}
2255EXPORT_SYMBOL_GPL(nft_unregister_set);
2256
2257
2258
2259
2260
2261
2262static const struct nft_set_ops *
2263nft_select_set_ops(const struct nlattr * const nla[],
2264 const struct nft_set_desc *desc,
2265 enum nft_set_policies policy)
2266{
2267 const struct nft_set_ops *ops, *bops;
2268 struct nft_set_estimate est, best;
2269 u32 features;
2270
2271#ifdef CONFIG_MODULES
2272 if (list_empty(&nf_tables_set_ops)) {
2273 nfnl_unlock(NFNL_SUBSYS_NFTABLES);
2274 request_module("nft-set");
2275 nfnl_lock(NFNL_SUBSYS_NFTABLES);
2276 if (!list_empty(&nf_tables_set_ops))
2277 return ERR_PTR(-EAGAIN);
2278 }
2279#endif
2280 features = 0;
2281 if (nla[NFTA_SET_FLAGS] != NULL) {
2282 features = ntohl(nla_get_be32(nla[NFTA_SET_FLAGS]));
2283 features &= NFT_SET_INTERVAL | NFT_SET_MAP | NFT_SET_TIMEOUT;
2284 }
2285
2286 bops = NULL;
2287 best.size = ~0;
2288 best.class = ~0;
2289
2290 list_for_each_entry(ops, &nf_tables_set_ops, list) {
2291 if ((ops->features & features) != features)
2292 continue;
2293 if (!ops->estimate(desc, features, &est))
2294 continue;
2295
2296 switch (policy) {
2297 case NFT_SET_POL_PERFORMANCE:
2298 if (est.class < best.class)
2299 break;
2300 if (est.class == best.class && est.size < best.size)
2301 break;
2302 continue;
2303 case NFT_SET_POL_MEMORY:
2304 if (est.size < best.size)
2305 break;
2306 if (est.size == best.size && est.class < best.class)
2307 break;
2308 continue;
2309 default:
2310 break;
2311 }
2312
2313 if (!try_module_get(ops->owner))
2314 continue;
2315 if (bops != NULL)
2316 module_put(bops->owner);
2317
2318 bops = ops;
2319 best = est;
2320 }
2321
2322 if (bops != NULL)
2323 return bops;
2324
2325 return ERR_PTR(-EOPNOTSUPP);
2326}
2327
2328static const struct nla_policy nft_set_policy[NFTA_SET_MAX + 1] = {
2329 [NFTA_SET_TABLE] = { .type = NLA_STRING },
2330 [NFTA_SET_NAME] = { .type = NLA_STRING,
2331 .len = IFNAMSIZ - 1 },
2332 [NFTA_SET_FLAGS] = { .type = NLA_U32 },
2333 [NFTA_SET_KEY_TYPE] = { .type = NLA_U32 },
2334 [NFTA_SET_KEY_LEN] = { .type = NLA_U32 },
2335 [NFTA_SET_DATA_TYPE] = { .type = NLA_U32 },
2336 [NFTA_SET_DATA_LEN] = { .type = NLA_U32 },
2337 [NFTA_SET_POLICY] = { .type = NLA_U32 },
2338 [NFTA_SET_DESC] = { .type = NLA_NESTED },
2339 [NFTA_SET_ID] = { .type = NLA_U32 },
2340 [NFTA_SET_TIMEOUT] = { .type = NLA_U64 },
2341 [NFTA_SET_GC_INTERVAL] = { .type = NLA_U32 },
2342};
2343
2344static const struct nla_policy nft_set_desc_policy[NFTA_SET_DESC_MAX + 1] = {
2345 [NFTA_SET_DESC_SIZE] = { .type = NLA_U32 },
2346};
2347
2348static int nft_ctx_init_from_setattr(struct nft_ctx *ctx,
2349 const struct sk_buff *skb,
2350 const struct nlmsghdr *nlh,
2351 const struct nlattr * const nla[])
2352{
2353 struct net *net = sock_net(skb->sk);
2354 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2355 struct nft_af_info *afi = NULL;
2356 struct nft_table *table = NULL;
2357
2358 if (nfmsg->nfgen_family != NFPROTO_UNSPEC) {
2359 afi = nf_tables_afinfo_lookup(net, nfmsg->nfgen_family, false);
2360 if (IS_ERR(afi))
2361 return PTR_ERR(afi);
2362 }
2363
2364 if (nla[NFTA_SET_TABLE] != NULL) {
2365 if (afi == NULL)
2366 return -EAFNOSUPPORT;
2367
2368 table = nf_tables_table_lookup(afi, nla[NFTA_SET_TABLE]);
2369 if (IS_ERR(table))
2370 return PTR_ERR(table);
2371 if (table->flags & NFT_TABLE_INACTIVE)
2372 return -ENOENT;
2373 }
2374
2375 nft_ctx_init(ctx, skb, nlh, afi, table, NULL, nla);
2376 return 0;
2377}
2378
2379struct nft_set *nf_tables_set_lookup(const struct nft_table *table,
2380 const struct nlattr *nla)
2381{
2382 struct nft_set *set;
2383
2384 if (nla == NULL)
2385 return ERR_PTR(-EINVAL);
2386
2387 list_for_each_entry(set, &table->sets, list) {
2388 if (!nla_strcmp(nla, set->name))
2389 return set;
2390 }
2391 return ERR_PTR(-ENOENT);
2392}
2393
2394struct nft_set *nf_tables_set_lookup_byid(const struct net *net,
2395 const struct nlattr *nla)
2396{
2397 struct nft_trans *trans;
2398 u32 id = ntohl(nla_get_be32(nla));
2399
2400 list_for_each_entry(trans, &net->nft.commit_list, list) {
2401 if (trans->msg_type == NFT_MSG_NEWSET &&
2402 id == nft_trans_set_id(trans))
2403 return nft_trans_set(trans);
2404 }
2405 return ERR_PTR(-ENOENT);
2406}
2407
2408static int nf_tables_set_alloc_name(struct nft_ctx *ctx, struct nft_set *set,
2409 const char *name)
2410{
2411 const struct nft_set *i;
2412 const char *p;
2413 unsigned long *inuse;
2414 unsigned int n = 0, min = 0;
2415
2416 p = strnchr(name, IFNAMSIZ, '%');
2417 if (p != NULL) {
2418 if (p[1] != 'd' || strchr(p + 2, '%'))
2419 return -EINVAL;
2420
2421 inuse = (unsigned long *)get_zeroed_page(GFP_KERNEL);
2422 if (inuse == NULL)
2423 return -ENOMEM;
2424cont:
2425 list_for_each_entry(i, &ctx->table->sets, list) {
2426 int tmp;
2427
2428 if (!sscanf(i->name, name, &tmp))
2429 continue;
2430 if (tmp < min || tmp >= min + BITS_PER_BYTE * PAGE_SIZE)
2431 continue;
2432
2433 set_bit(tmp - min, inuse);
2434 }
2435
2436 n = find_first_zero_bit(inuse, BITS_PER_BYTE * PAGE_SIZE);
2437 if (n >= BITS_PER_BYTE * PAGE_SIZE) {
2438 min += BITS_PER_BYTE * PAGE_SIZE;
2439 memset(inuse, 0, PAGE_SIZE);
2440 goto cont;
2441 }
2442 free_page((unsigned long)inuse);
2443 }
2444
2445 snprintf(set->name, sizeof(set->name), name, min + n);
2446 list_for_each_entry(i, &ctx->table->sets, list) {
2447 if (!strcmp(set->name, i->name))
2448 return -ENFILE;
2449 }
2450 return 0;
2451}
2452
2453static int nf_tables_fill_set(struct sk_buff *skb, const struct nft_ctx *ctx,
2454 const struct nft_set *set, u16 event, u16 flags)
2455{
2456 struct nfgenmsg *nfmsg;
2457 struct nlmsghdr *nlh;
2458 struct nlattr *desc;
2459 u32 portid = ctx->portid;
2460 u32 seq = ctx->seq;
2461
2462 event |= NFNL_SUBSYS_NFTABLES << 8;
2463 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg),
2464 flags);
2465 if (nlh == NULL)
2466 goto nla_put_failure;
2467
2468 nfmsg = nlmsg_data(nlh);
2469 nfmsg->nfgen_family = ctx->afi->family;
2470 nfmsg->version = NFNETLINK_V0;
2471 nfmsg->res_id = htons(ctx->net->nft.base_seq & 0xffff);
2472
2473 if (nla_put_string(skb, NFTA_SET_TABLE, ctx->table->name))
2474 goto nla_put_failure;
2475 if (nla_put_string(skb, NFTA_SET_NAME, set->name))
2476 goto nla_put_failure;
2477 if (set->flags != 0)
2478 if (nla_put_be32(skb, NFTA_SET_FLAGS, htonl(set->flags)))
2479 goto nla_put_failure;
2480
2481 if (nla_put_be32(skb, NFTA_SET_KEY_TYPE, htonl(set->ktype)))
2482 goto nla_put_failure;
2483 if (nla_put_be32(skb, NFTA_SET_KEY_LEN, htonl(set->klen)))
2484 goto nla_put_failure;
2485 if (set->flags & NFT_SET_MAP) {
2486 if (nla_put_be32(skb, NFTA_SET_DATA_TYPE, htonl(set->dtype)))
2487 goto nla_put_failure;
2488 if (nla_put_be32(skb, NFTA_SET_DATA_LEN, htonl(set->dlen)))
2489 goto nla_put_failure;
2490 }
2491
2492 if (set->timeout &&
2493 nla_put_be64(skb, NFTA_SET_TIMEOUT, cpu_to_be64(set->timeout)))
2494 goto nla_put_failure;
2495 if (set->gc_int &&
2496 nla_put_be32(skb, NFTA_SET_GC_INTERVAL, htonl(set->gc_int)))
2497 goto nla_put_failure;
2498
2499 if (set->policy != NFT_SET_POL_PERFORMANCE) {
2500 if (nla_put_be32(skb, NFTA_SET_POLICY, htonl(set->policy)))
2501 goto nla_put_failure;
2502 }
2503
2504 desc = nla_nest_start(skb, NFTA_SET_DESC);
2505 if (desc == NULL)
2506 goto nla_put_failure;
2507 if (set->size &&
2508 nla_put_be32(skb, NFTA_SET_DESC_SIZE, htonl(set->size)))
2509 goto nla_put_failure;
2510 nla_nest_end(skb, desc);
2511
2512 nlmsg_end(skb, nlh);
2513 return 0;
2514
2515nla_put_failure:
2516 nlmsg_trim(skb, nlh);
2517 return -1;
2518}
2519
2520static int nf_tables_set_notify(const struct nft_ctx *ctx,
2521 const struct nft_set *set,
2522 int event, gfp_t gfp_flags)
2523{
2524 struct sk_buff *skb;
2525 u32 portid = ctx->portid;
2526 int err;
2527
2528 if (!ctx->report &&
2529 !nfnetlink_has_listeners(ctx->net, NFNLGRP_NFTABLES))
2530 return 0;
2531
2532 err = -ENOBUFS;
2533 skb = nlmsg_new(NLMSG_GOODSIZE, gfp_flags);
2534 if (skb == NULL)
2535 goto err;
2536
2537 err = nf_tables_fill_set(skb, ctx, set, event, 0);
2538 if (err < 0) {
2539 kfree_skb(skb);
2540 goto err;
2541 }
2542
2543 err = nfnetlink_send(skb, ctx->net, portid, NFNLGRP_NFTABLES,
2544 ctx->report, gfp_flags);
2545err:
2546 if (err < 0)
2547 nfnetlink_set_err(ctx->net, portid, NFNLGRP_NFTABLES, err);
2548 return err;
2549}
2550
2551static int nf_tables_dump_sets(struct sk_buff *skb, struct netlink_callback *cb)
2552{
2553 const struct nft_set *set;
2554 unsigned int idx, s_idx = cb->args[0];
2555 struct nft_af_info *afi;
2556 struct nft_table *table, *cur_table = (struct nft_table *)cb->args[2];
2557 struct net *net = sock_net(skb->sk);
2558 int cur_family = cb->args[3];
2559 struct nft_ctx *ctx = cb->data, ctx_set;
2560
2561 if (cb->args[1])
2562 return skb->len;
2563
2564 rcu_read_lock();
2565 cb->seq = net->nft.base_seq;
2566
2567 list_for_each_entry_rcu(afi, &net->nft.af_info, list) {
2568 if (ctx->afi && ctx->afi != afi)
2569 continue;
2570
2571 if (cur_family) {
2572 if (afi->family != cur_family)
2573 continue;
2574
2575 cur_family = 0;
2576 }
2577 list_for_each_entry_rcu(table, &afi->tables, list) {
2578 if (ctx->table && ctx->table != table)
2579 continue;
2580
2581 if (cur_table) {
2582 if (cur_table != table)
2583 continue;
2584
2585 cur_table = NULL;
2586 }
2587 idx = 0;
2588 list_for_each_entry_rcu(set, &table->sets, list) {
2589 if (idx < s_idx)
2590 goto cont;
2591
2592 ctx_set = *ctx;
2593 ctx_set.table = table;
2594 ctx_set.afi = afi;
2595 if (nf_tables_fill_set(skb, &ctx_set, set,
2596 NFT_MSG_NEWSET,
2597 NLM_F_MULTI) < 0) {
2598 cb->args[0] = idx;
2599 cb->args[2] = (unsigned long) table;
2600 cb->args[3] = afi->family;
2601 goto done;
2602 }
2603 nl_dump_check_consistent(cb, nlmsg_hdr(skb));
2604cont:
2605 idx++;
2606 }
2607 if (s_idx)
2608 s_idx = 0;
2609 }
2610 }
2611 cb->args[1] = 1;
2612done:
2613 rcu_read_unlock();
2614 return skb->len;
2615}
2616
2617static int nf_tables_dump_sets_done(struct netlink_callback *cb)
2618{
2619 kfree(cb->data);
2620 return 0;
2621}
2622
2623static int nf_tables_getset(struct sock *nlsk, struct sk_buff *skb,
2624 const struct nlmsghdr *nlh,
2625 const struct nlattr * const nla[])
2626{
2627 const struct nft_set *set;
2628 struct nft_ctx ctx;
2629 struct sk_buff *skb2;
2630 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2631 int err;
2632
2633
2634 err = nft_ctx_init_from_setattr(&ctx, skb, nlh, nla);
2635 if (err < 0)
2636 return err;
2637
2638 if (nlh->nlmsg_flags & NLM_F_DUMP) {
2639 struct netlink_dump_control c = {
2640 .dump = nf_tables_dump_sets,
2641 .done = nf_tables_dump_sets_done,
2642 };
2643 struct nft_ctx *ctx_dump;
2644
2645 ctx_dump = kmalloc(sizeof(*ctx_dump), GFP_KERNEL);
2646 if (ctx_dump == NULL)
2647 return -ENOMEM;
2648
2649 *ctx_dump = ctx;
2650 c.data = ctx_dump;
2651
2652 return netlink_dump_start(nlsk, skb, nlh, &c);
2653 }
2654
2655
2656 if (nfmsg->nfgen_family == NFPROTO_UNSPEC)
2657 return -EAFNOSUPPORT;
2658
2659 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_NAME]);
2660 if (IS_ERR(set))
2661 return PTR_ERR(set);
2662 if (set->flags & NFT_SET_INACTIVE)
2663 return -ENOENT;
2664
2665 skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
2666 if (skb2 == NULL)
2667 return -ENOMEM;
2668
2669 err = nf_tables_fill_set(skb2, &ctx, set, NFT_MSG_NEWSET, 0);
2670 if (err < 0)
2671 goto err;
2672
2673 return nlmsg_unicast(nlsk, skb2, NETLINK_CB(skb).portid);
2674
2675err:
2676 kfree_skb(skb2);
2677 return err;
2678}
2679
2680static int nf_tables_set_desc_parse(const struct nft_ctx *ctx,
2681 struct nft_set_desc *desc,
2682 const struct nlattr *nla)
2683{
2684 struct nlattr *da[NFTA_SET_DESC_MAX + 1];
2685 int err;
2686
2687 err = nla_parse_nested(da, NFTA_SET_DESC_MAX, nla, nft_set_desc_policy);
2688 if (err < 0)
2689 return err;
2690
2691 if (da[NFTA_SET_DESC_SIZE] != NULL)
2692 desc->size = ntohl(nla_get_be32(da[NFTA_SET_DESC_SIZE]));
2693
2694 return 0;
2695}
2696
2697static int nf_tables_newset(struct sock *nlsk, struct sk_buff *skb,
2698 const struct nlmsghdr *nlh,
2699 const struct nlattr * const nla[])
2700{
2701 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2702 const struct nft_set_ops *ops;
2703 struct nft_af_info *afi;
2704 struct net *net = sock_net(skb->sk);
2705 struct nft_table *table;
2706 struct nft_set *set;
2707 struct nft_ctx ctx;
2708 char name[IFNAMSIZ];
2709 unsigned int size;
2710 bool create;
2711 u64 timeout;
2712 u32 ktype, dtype, flags, policy, gc_int;
2713 struct nft_set_desc desc;
2714 int err;
2715
2716 if (nla[NFTA_SET_TABLE] == NULL ||
2717 nla[NFTA_SET_NAME] == NULL ||
2718 nla[NFTA_SET_KEY_LEN] == NULL ||
2719 nla[NFTA_SET_ID] == NULL)
2720 return -EINVAL;
2721
2722 memset(&desc, 0, sizeof(desc));
2723
2724 ktype = NFT_DATA_VALUE;
2725 if (nla[NFTA_SET_KEY_TYPE] != NULL) {
2726 ktype = ntohl(nla_get_be32(nla[NFTA_SET_KEY_TYPE]));
2727 if ((ktype & NFT_DATA_RESERVED_MASK) == NFT_DATA_RESERVED_MASK)
2728 return -EINVAL;
2729 }
2730
2731 desc.klen = ntohl(nla_get_be32(nla[NFTA_SET_KEY_LEN]));
2732 if (desc.klen == 0 || desc.klen > NFT_DATA_VALUE_MAXLEN)
2733 return -EINVAL;
2734
2735 flags = 0;
2736 if (nla[NFTA_SET_FLAGS] != NULL) {
2737 flags = ntohl(nla_get_be32(nla[NFTA_SET_FLAGS]));
2738 if (flags & ~(NFT_SET_ANONYMOUS | NFT_SET_CONSTANT |
2739 NFT_SET_INTERVAL | NFT_SET_TIMEOUT |
2740 NFT_SET_MAP | NFT_SET_EVAL))
2741 return -EINVAL;
2742
2743 if ((flags & (NFT_SET_MAP | NFT_SET_EVAL)) ==
2744 (NFT_SET_MAP | NFT_SET_EVAL))
2745 return -EOPNOTSUPP;
2746 }
2747
2748 dtype = 0;
2749 if (nla[NFTA_SET_DATA_TYPE] != NULL) {
2750 if (!(flags & NFT_SET_MAP))
2751 return -EINVAL;
2752
2753 dtype = ntohl(nla_get_be32(nla[NFTA_SET_DATA_TYPE]));
2754 if ((dtype & NFT_DATA_RESERVED_MASK) == NFT_DATA_RESERVED_MASK &&
2755 dtype != NFT_DATA_VERDICT)
2756 return -EINVAL;
2757
2758 if (dtype != NFT_DATA_VERDICT) {
2759 if (nla[NFTA_SET_DATA_LEN] == NULL)
2760 return -EINVAL;
2761 desc.dlen = ntohl(nla_get_be32(nla[NFTA_SET_DATA_LEN]));
2762 if (desc.dlen == 0 || desc.dlen > NFT_DATA_VALUE_MAXLEN)
2763 return -EINVAL;
2764 } else
2765 desc.dlen = sizeof(struct nft_verdict);
2766 } else if (flags & NFT_SET_MAP)
2767 return -EINVAL;
2768
2769 timeout = 0;
2770 if (nla[NFTA_SET_TIMEOUT] != NULL) {
2771 if (!(flags & NFT_SET_TIMEOUT))
2772 return -EINVAL;
2773 timeout = be64_to_cpu(nla_get_be64(nla[NFTA_SET_TIMEOUT]));
2774 }
2775 gc_int = 0;
2776 if (nla[NFTA_SET_GC_INTERVAL] != NULL) {
2777 if (!(flags & NFT_SET_TIMEOUT))
2778 return -EINVAL;
2779 gc_int = ntohl(nla_get_be32(nla[NFTA_SET_GC_INTERVAL]));
2780 }
2781
2782 policy = NFT_SET_POL_PERFORMANCE;
2783 if (nla[NFTA_SET_POLICY] != NULL)
2784 policy = ntohl(nla_get_be32(nla[NFTA_SET_POLICY]));
2785
2786 if (nla[NFTA_SET_DESC] != NULL) {
2787 err = nf_tables_set_desc_parse(&ctx, &desc, nla[NFTA_SET_DESC]);
2788 if (err < 0)
2789 return err;
2790 }
2791
2792 create = nlh->nlmsg_flags & NLM_F_CREATE ? true : false;
2793
2794 afi = nf_tables_afinfo_lookup(net, nfmsg->nfgen_family, create);
2795 if (IS_ERR(afi))
2796 return PTR_ERR(afi);
2797
2798 table = nf_tables_table_lookup(afi, nla[NFTA_SET_TABLE]);
2799 if (IS_ERR(table))
2800 return PTR_ERR(table);
2801
2802 nft_ctx_init(&ctx, skb, nlh, afi, table, NULL, nla);
2803
2804 set = nf_tables_set_lookup(table, nla[NFTA_SET_NAME]);
2805 if (IS_ERR(set)) {
2806 if (PTR_ERR(set) != -ENOENT)
2807 return PTR_ERR(set);
2808 set = NULL;
2809 }
2810
2811 if (set != NULL) {
2812 if (nlh->nlmsg_flags & NLM_F_EXCL)
2813 return -EEXIST;
2814 if (nlh->nlmsg_flags & NLM_F_REPLACE)
2815 return -EOPNOTSUPP;
2816 return 0;
2817 }
2818
2819 if (!(nlh->nlmsg_flags & NLM_F_CREATE))
2820 return -ENOENT;
2821
2822 ops = nft_select_set_ops(nla, &desc, policy);
2823 if (IS_ERR(ops))
2824 return PTR_ERR(ops);
2825
2826 size = 0;
2827 if (ops->privsize != NULL)
2828 size = ops->privsize(nla);
2829
2830 err = -ENOMEM;
2831 set = kzalloc(sizeof(*set) + size, GFP_KERNEL);
2832 if (set == NULL)
2833 goto err1;
2834
2835 nla_strlcpy(name, nla[NFTA_SET_NAME], sizeof(set->name));
2836 err = nf_tables_set_alloc_name(&ctx, set, name);
2837 if (err < 0)
2838 goto err2;
2839
2840 INIT_LIST_HEAD(&set->bindings);
2841 write_pnet(&set->pnet, net);
2842 set->ops = ops;
2843 set->ktype = ktype;
2844 set->klen = desc.klen;
2845 set->dtype = dtype;
2846 set->dlen = desc.dlen;
2847 set->flags = flags;
2848 set->size = desc.size;
2849 set->policy = policy;
2850 set->timeout = timeout;
2851 set->gc_int = gc_int;
2852
2853 err = ops->init(set, &desc, nla);
2854 if (err < 0)
2855 goto err2;
2856
2857 err = nft_trans_set_add(&ctx, NFT_MSG_NEWSET, set);
2858 if (err < 0)
2859 goto err2;
2860
2861 list_add_tail_rcu(&set->list, &table->sets);
2862 table->use++;
2863 return 0;
2864
2865err2:
2866 kfree(set);
2867err1:
2868 module_put(ops->owner);
2869 return err;
2870}
2871
2872static void nft_set_destroy(struct nft_set *set)
2873{
2874 set->ops->destroy(set);
2875 module_put(set->ops->owner);
2876 kfree(set);
2877}
2878
2879static void nf_tables_set_destroy(const struct nft_ctx *ctx, struct nft_set *set)
2880{
2881 list_del_rcu(&set->list);
2882 nf_tables_set_notify(ctx, set, NFT_MSG_DELSET, GFP_ATOMIC);
2883 nft_set_destroy(set);
2884}
2885
2886static int nf_tables_delset(struct sock *nlsk, struct sk_buff *skb,
2887 const struct nlmsghdr *nlh,
2888 const struct nlattr * const nla[])
2889{
2890 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
2891 struct nft_set *set;
2892 struct nft_ctx ctx;
2893 int err;
2894
2895 if (nfmsg->nfgen_family == NFPROTO_UNSPEC)
2896 return -EAFNOSUPPORT;
2897 if (nla[NFTA_SET_TABLE] == NULL)
2898 return -EINVAL;
2899
2900 err = nft_ctx_init_from_setattr(&ctx, skb, nlh, nla);
2901 if (err < 0)
2902 return err;
2903
2904 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_NAME]);
2905 if (IS_ERR(set))
2906 return PTR_ERR(set);
2907 if (set->flags & NFT_SET_INACTIVE)
2908 return -ENOENT;
2909 if (!list_empty(&set->bindings))
2910 return -EBUSY;
2911
2912 return nft_delset(&ctx, set);
2913}
2914
2915static int nf_tables_bind_check_setelem(const struct nft_ctx *ctx,
2916 const struct nft_set *set,
2917 const struct nft_set_iter *iter,
2918 const struct nft_set_elem *elem)
2919{
2920 const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
2921 enum nft_registers dreg;
2922
2923 dreg = nft_type_to_reg(set->dtype);
2924 return nft_validate_register_store(ctx, dreg, nft_set_ext_data(ext),
2925 set->dtype == NFT_DATA_VERDICT ?
2926 NFT_DATA_VERDICT : NFT_DATA_VALUE,
2927 set->dlen);
2928}
2929
2930int nf_tables_bind_set(const struct nft_ctx *ctx, struct nft_set *set,
2931 struct nft_set_binding *binding)
2932{
2933 struct nft_set_binding *i;
2934 struct nft_set_iter iter;
2935
2936 if (!list_empty(&set->bindings) && set->flags & NFT_SET_ANONYMOUS)
2937 return -EBUSY;
2938
2939 if (binding->flags & NFT_SET_MAP) {
2940
2941
2942
2943 list_for_each_entry(i, &set->bindings, list) {
2944 if (binding->flags & NFT_SET_MAP &&
2945 i->chain == binding->chain)
2946 goto bind;
2947 }
2948
2949 iter.skip = 0;
2950 iter.count = 0;
2951 iter.err = 0;
2952 iter.fn = nf_tables_bind_check_setelem;
2953
2954 set->ops->walk(ctx, set, &iter);
2955 if (iter.err < 0) {
2956
2957 if (set->flags & NFT_SET_ANONYMOUS)
2958 nf_tables_set_destroy(ctx, set);
2959
2960 return iter.err;
2961 }
2962 }
2963bind:
2964 binding->chain = ctx->chain;
2965 list_add_tail_rcu(&binding->list, &set->bindings);
2966 return 0;
2967}
2968
2969void nf_tables_unbind_set(const struct nft_ctx *ctx, struct nft_set *set,
2970 struct nft_set_binding *binding)
2971{
2972 list_del_rcu(&binding->list);
2973
2974 if (list_empty(&set->bindings) && set->flags & NFT_SET_ANONYMOUS &&
2975 !(set->flags & NFT_SET_INACTIVE))
2976 nf_tables_set_destroy(ctx, set);
2977}
2978
2979const struct nft_set_ext_type nft_set_ext_types[] = {
2980 [NFT_SET_EXT_KEY] = {
2981 .align = __alignof__(u32),
2982 },
2983 [NFT_SET_EXT_DATA] = {
2984 .align = __alignof__(u32),
2985 },
2986 [NFT_SET_EXT_EXPR] = {
2987 .align = __alignof__(struct nft_expr),
2988 },
2989 [NFT_SET_EXT_FLAGS] = {
2990 .len = sizeof(u8),
2991 .align = __alignof__(u8),
2992 },
2993 [NFT_SET_EXT_TIMEOUT] = {
2994 .len = sizeof(u64),
2995 .align = __alignof__(u64),
2996 },
2997 [NFT_SET_EXT_EXPIRATION] = {
2998 .len = sizeof(unsigned long),
2999 .align = __alignof__(unsigned long),
3000 },
3001 [NFT_SET_EXT_USERDATA] = {
3002 .len = sizeof(struct nft_userdata),
3003 .align = __alignof__(struct nft_userdata),
3004 },
3005};
3006EXPORT_SYMBOL_GPL(nft_set_ext_types);
3007
3008
3009
3010
3011
3012static const struct nla_policy nft_set_elem_policy[NFTA_SET_ELEM_MAX + 1] = {
3013 [NFTA_SET_ELEM_KEY] = { .type = NLA_NESTED },
3014 [NFTA_SET_ELEM_DATA] = { .type = NLA_NESTED },
3015 [NFTA_SET_ELEM_FLAGS] = { .type = NLA_U32 },
3016 [NFTA_SET_ELEM_TIMEOUT] = { .type = NLA_U64 },
3017 [NFTA_SET_ELEM_USERDATA] = { .type = NLA_BINARY,
3018 .len = NFT_USERDATA_MAXLEN },
3019};
3020
3021static const struct nla_policy nft_set_elem_list_policy[NFTA_SET_ELEM_LIST_MAX + 1] = {
3022 [NFTA_SET_ELEM_LIST_TABLE] = { .type = NLA_STRING },
3023 [NFTA_SET_ELEM_LIST_SET] = { .type = NLA_STRING },
3024 [NFTA_SET_ELEM_LIST_ELEMENTS] = { .type = NLA_NESTED },
3025 [NFTA_SET_ELEM_LIST_SET_ID] = { .type = NLA_U32 },
3026};
3027
3028static int nft_ctx_init_from_elemattr(struct nft_ctx *ctx,
3029 const struct sk_buff *skb,
3030 const struct nlmsghdr *nlh,
3031 const struct nlattr * const nla[],
3032 bool trans)
3033{
3034 const struct nfgenmsg *nfmsg = nlmsg_data(nlh);
3035 struct nft_af_info *afi;
3036 struct nft_table *table;
3037 struct net *net = sock_net(skb->sk);
3038
3039 afi = nf_tables_afinfo_lookup(net, nfmsg->nfgen_family, false);
3040 if (IS_ERR(afi))
3041 return PTR_ERR(afi);
3042
3043 table = nf_tables_table_lookup(afi, nla[NFTA_SET_ELEM_LIST_TABLE]);
3044 if (IS_ERR(table))
3045 return PTR_ERR(table);
3046 if (!trans && (table->flags & NFT_TABLE_INACTIVE))
3047 return -ENOENT;
3048
3049 nft_ctx_init(ctx, skb, nlh, afi, table, NULL, nla);
3050 return 0;
3051}
3052
3053static int nf_tables_fill_setelem(struct sk_buff *skb,
3054 const struct nft_set *set,
3055 const struct nft_set_elem *elem)
3056{
3057 const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
3058 unsigned char *b = skb_tail_pointer(skb);
3059 struct nlattr *nest;
3060
3061 nest = nla_nest_start(skb, NFTA_LIST_ELEM);
3062 if (nest == NULL)
3063 goto nla_put_failure;
3064
3065 if (nft_data_dump(skb, NFTA_SET_ELEM_KEY, nft_set_ext_key(ext),
3066 NFT_DATA_VALUE, set->klen) < 0)
3067 goto nla_put_failure;
3068
3069 if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA) &&
3070 nft_data_dump(skb, NFTA_SET_ELEM_DATA, nft_set_ext_data(ext),
3071 set->dtype == NFT_DATA_VERDICT ? NFT_DATA_VERDICT : NFT_DATA_VALUE,
3072 set->dlen) < 0)
3073 goto nla_put_failure;
3074
3075 if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPR) &&
3076 nft_expr_dump(skb, NFTA_SET_ELEM_EXPR, nft_set_ext_expr(ext)) < 0)
3077 goto nla_put_failure;
3078
3079 if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
3080 nla_put_be32(skb, NFTA_SET_ELEM_FLAGS,
3081 htonl(*nft_set_ext_flags(ext))))
3082 goto nla_put_failure;
3083
3084 if (nft_set_ext_exists(ext, NFT_SET_EXT_TIMEOUT) &&
3085 nla_put_be64(skb, NFTA_SET_ELEM_TIMEOUT,
3086 cpu_to_be64(*nft_set_ext_timeout(ext))))
3087 goto nla_put_failure;
3088
3089 if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPIRATION)) {
3090 unsigned long expires, now = jiffies;
3091
3092 expires = *nft_set_ext_expiration(ext);
3093 if (time_before(now, expires))
3094 expires -= now;
3095 else
3096 expires = 0;
3097
3098 if (nla_put_be64(skb, NFTA_SET_ELEM_EXPIRATION,
3099 cpu_to_be64(jiffies_to_msecs(expires))))
3100 goto nla_put_failure;
3101 }
3102
3103 if (nft_set_ext_exists(ext, NFT_SET_EXT_USERDATA)) {
3104 struct nft_userdata *udata;
3105
3106 udata = nft_set_ext_userdata(ext);
3107 if (nla_put(skb, NFTA_SET_ELEM_USERDATA,
3108 udata->len + 1, udata->data))
3109 goto nla_put_failure;
3110 }
3111
3112 nla_nest_end(skb, nest);
3113 return 0;
3114
3115nla_put_failure:
3116 nlmsg_trim(skb, b);
3117 return -EMSGSIZE;
3118}
3119
3120struct nft_set_dump_args {
3121 const struct netlink_callback *cb;
3122 struct nft_set_iter iter;
3123 struct sk_buff *skb;
3124};
3125
3126static int nf_tables_dump_setelem(const struct nft_ctx *ctx,
3127 const struct nft_set *set,
3128 const struct nft_set_iter *iter,
3129 const struct nft_set_elem *elem)
3130{
3131 struct nft_set_dump_args *args;
3132
3133 args = container_of(iter, struct nft_set_dump_args, iter);
3134 return nf_tables_fill_setelem(args->skb, set, elem);
3135}
3136
3137static int nf_tables_dump_set(struct sk_buff *skb, struct netlink_callback *cb)
3138{
3139 const struct nft_set *set;
3140 struct nft_set_dump_args args;
3141 struct nft_ctx ctx;
3142 struct nlattr *nla[NFTA_SET_ELEM_LIST_MAX + 1];
3143 struct nfgenmsg *nfmsg;
3144 struct nlmsghdr *nlh;
3145 struct nlattr *nest;
3146 u32 portid, seq;
3147 int event, err;
3148
3149 err = nlmsg_parse(cb->nlh, sizeof(struct nfgenmsg), nla,
3150 NFTA_SET_ELEM_LIST_MAX, nft_set_elem_list_policy);
3151 if (err < 0)
3152 return err;
3153
3154 err = nft_ctx_init_from_elemattr(&ctx, cb->skb, cb->nlh, (void *)nla,
3155 false);
3156 if (err < 0)
3157 return err;
3158
3159 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_ELEM_LIST_SET]);
3160 if (IS_ERR(set))
3161 return PTR_ERR(set);
3162 if (set->flags & NFT_SET_INACTIVE)
3163 return -ENOENT;
3164
3165 event = NFT_MSG_NEWSETELEM;
3166 event |= NFNL_SUBSYS_NFTABLES << 8;
3167 portid = NETLINK_CB(cb->skb).portid;
3168 seq = cb->nlh->nlmsg_seq;
3169
3170 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg),
3171 NLM_F_MULTI);
3172 if (nlh == NULL)
3173 goto nla_put_failure;
3174
3175 nfmsg = nlmsg_data(nlh);
3176 nfmsg->nfgen_family = ctx.afi->family;
3177 nfmsg->version = NFNETLINK_V0;
3178 nfmsg->res_id = htons(ctx.net->nft.base_seq & 0xffff);
3179
3180 if (nla_put_string(skb, NFTA_SET_ELEM_LIST_TABLE, ctx.table->name))
3181 goto nla_put_failure;
3182 if (nla_put_string(skb, NFTA_SET_ELEM_LIST_SET, set->name))
3183 goto nla_put_failure;
3184
3185 nest = nla_nest_start(skb, NFTA_SET_ELEM_LIST_ELEMENTS);
3186 if (nest == NULL)
3187 goto nla_put_failure;
3188
3189 args.cb = cb;
3190 args.skb = skb;
3191 args.iter.skip = cb->args[0];
3192 args.iter.count = 0;
3193 args.iter.err = 0;
3194 args.iter.fn = nf_tables_dump_setelem;
3195 set->ops->walk(&ctx, set, &args.iter);
3196
3197 nla_nest_end(skb, nest);
3198 nlmsg_end(skb, nlh);
3199
3200 if (args.iter.err && args.iter.err != -EMSGSIZE)
3201 return args.iter.err;
3202 if (args.iter.count == cb->args[0])
3203 return 0;
3204
3205 cb->args[0] = args.iter.count;
3206 return skb->len;
3207
3208nla_put_failure:
3209 return -ENOSPC;
3210}
3211
3212static int nf_tables_getsetelem(struct sock *nlsk, struct sk_buff *skb,
3213 const struct nlmsghdr *nlh,
3214 const struct nlattr * const nla[])
3215{
3216 const struct nft_set *set;
3217 struct nft_ctx ctx;
3218 int err;
3219
3220 err = nft_ctx_init_from_elemattr(&ctx, skb, nlh, nla, false);
3221 if (err < 0)
3222 return err;
3223
3224 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_ELEM_LIST_SET]);
3225 if (IS_ERR(set))
3226 return PTR_ERR(set);
3227 if (set->flags & NFT_SET_INACTIVE)
3228 return -ENOENT;
3229
3230 if (nlh->nlmsg_flags & NLM_F_DUMP) {
3231 struct netlink_dump_control c = {
3232 .dump = nf_tables_dump_set,
3233 };
3234 return netlink_dump_start(nlsk, skb, nlh, &c);
3235 }
3236 return -EOPNOTSUPP;
3237}
3238
3239static int nf_tables_fill_setelem_info(struct sk_buff *skb,
3240 const struct nft_ctx *ctx, u32 seq,
3241 u32 portid, int event, u16 flags,
3242 const struct nft_set *set,
3243 const struct nft_set_elem *elem)
3244{
3245 struct nfgenmsg *nfmsg;
3246 struct nlmsghdr *nlh;
3247 struct nlattr *nest;
3248 int err;
3249
3250 event |= NFNL_SUBSYS_NFTABLES << 8;
3251 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg),
3252 flags);
3253 if (nlh == NULL)
3254 goto nla_put_failure;
3255
3256 nfmsg = nlmsg_data(nlh);
3257 nfmsg->nfgen_family = ctx->afi->family;
3258 nfmsg->version = NFNETLINK_V0;
3259 nfmsg->res_id = htons(ctx->net->nft.base_seq & 0xffff);
3260
3261 if (nla_put_string(skb, NFTA_SET_TABLE, ctx->table->name))
3262 goto nla_put_failure;
3263 if (nla_put_string(skb, NFTA_SET_NAME, set->name))
3264 goto nla_put_failure;
3265
3266 nest = nla_nest_start(skb, NFTA_SET_ELEM_LIST_ELEMENTS);
3267 if (nest == NULL)
3268 goto nla_put_failure;
3269
3270 err = nf_tables_fill_setelem(skb, set, elem);
3271 if (err < 0)
3272 goto nla_put_failure;
3273
3274 nla_nest_end(skb, nest);
3275
3276 nlmsg_end(skb, nlh);
3277 return 0;
3278
3279nla_put_failure:
3280 nlmsg_trim(skb, nlh);
3281 return -1;
3282}
3283
3284static int nf_tables_setelem_notify(const struct nft_ctx *ctx,
3285 const struct nft_set *set,
3286 const struct nft_set_elem *elem,
3287 int event, u16 flags)
3288{
3289 struct net *net = ctx->net;
3290 u32 portid = ctx->portid;
3291 struct sk_buff *skb;
3292 int err;
3293
3294 if (!ctx->report && !nfnetlink_has_listeners(net, NFNLGRP_NFTABLES))
3295 return 0;
3296
3297 err = -ENOBUFS;
3298 skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
3299 if (skb == NULL)
3300 goto err;
3301
3302 err = nf_tables_fill_setelem_info(skb, ctx, 0, portid, event, flags,
3303 set, elem);
3304 if (err < 0) {
3305 kfree_skb(skb);
3306 goto err;
3307 }
3308
3309 err = nfnetlink_send(skb, net, portid, NFNLGRP_NFTABLES, ctx->report,
3310 GFP_KERNEL);
3311err:
3312 if (err < 0)
3313 nfnetlink_set_err(net, portid, NFNLGRP_NFTABLES, err);
3314 return err;
3315}
3316
3317static struct nft_trans *nft_trans_elem_alloc(struct nft_ctx *ctx,
3318 int msg_type,
3319 struct nft_set *set)
3320{
3321 struct nft_trans *trans;
3322
3323 trans = nft_trans_alloc(ctx, msg_type, sizeof(struct nft_trans_elem));
3324 if (trans == NULL)
3325 return NULL;
3326
3327 nft_trans_elem_set(trans) = set;
3328 return trans;
3329}
3330
3331void *nft_set_elem_init(const struct nft_set *set,
3332 const struct nft_set_ext_tmpl *tmpl,
3333 const u32 *key, const u32 *data,
3334 u64 timeout, gfp_t gfp)
3335{
3336 struct nft_set_ext *ext;
3337 void *elem;
3338
3339 elem = kzalloc(set->ops->elemsize + tmpl->len, gfp);
3340 if (elem == NULL)
3341 return NULL;
3342
3343 ext = nft_set_elem_ext(set, elem);
3344 nft_set_ext_init(ext, tmpl);
3345
3346 memcpy(nft_set_ext_key(ext), key, set->klen);
3347 if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA))
3348 memcpy(nft_set_ext_data(ext), data, set->dlen);
3349 if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPIRATION))
3350 *nft_set_ext_expiration(ext) =
3351 jiffies + msecs_to_jiffies(timeout);
3352 if (nft_set_ext_exists(ext, NFT_SET_EXT_TIMEOUT))
3353 *nft_set_ext_timeout(ext) = timeout;
3354
3355 return elem;
3356}
3357
3358void nft_set_elem_destroy(const struct nft_set *set, void *elem)
3359{
3360 struct nft_set_ext *ext = nft_set_elem_ext(set, elem);
3361
3362 nft_data_uninit(nft_set_ext_key(ext), NFT_DATA_VALUE);
3363 if (nft_set_ext_exists(ext, NFT_SET_EXT_DATA))
3364 nft_data_uninit(nft_set_ext_data(ext), set->dtype);
3365 if (nft_set_ext_exists(ext, NFT_SET_EXT_EXPR))
3366 nf_tables_expr_destroy(NULL, nft_set_ext_expr(ext));
3367
3368 kfree(elem);
3369}
3370EXPORT_SYMBOL_GPL(nft_set_elem_destroy);
3371
3372static int nft_add_set_elem(struct nft_ctx *ctx, struct nft_set *set,
3373 const struct nlattr *attr)
3374{
3375 struct nlattr *nla[NFTA_SET_ELEM_MAX + 1];
3376 struct nft_data_desc d1, d2;
3377 struct nft_set_ext_tmpl tmpl;
3378 struct nft_set_ext *ext;
3379 struct nft_set_elem elem;
3380 struct nft_set_binding *binding;
3381 struct nft_userdata *udata;
3382 struct nft_data data;
3383 enum nft_registers dreg;
3384 struct nft_trans *trans;
3385 u64 timeout;
3386 u32 flags;
3387 u8 ulen;
3388 int err;
3389
3390 err = nla_parse_nested(nla, NFTA_SET_ELEM_MAX, attr,
3391 nft_set_elem_policy);
3392 if (err < 0)
3393 return err;
3394
3395 if (nla[NFTA_SET_ELEM_KEY] == NULL)
3396 return -EINVAL;
3397
3398 nft_set_ext_prepare(&tmpl);
3399
3400 flags = 0;
3401 if (nla[NFTA_SET_ELEM_FLAGS] != NULL) {
3402 flags = ntohl(nla_get_be32(nla[NFTA_SET_ELEM_FLAGS]));
3403 if (flags & ~NFT_SET_ELEM_INTERVAL_END)
3404 return -EINVAL;
3405 if (!(set->flags & NFT_SET_INTERVAL) &&
3406 flags & NFT_SET_ELEM_INTERVAL_END)
3407 return -EINVAL;
3408 if (flags != 0)
3409 nft_set_ext_add(&tmpl, NFT_SET_EXT_FLAGS);
3410 }
3411
3412 if (set->flags & NFT_SET_MAP) {
3413 if (nla[NFTA_SET_ELEM_DATA] == NULL &&
3414 !(flags & NFT_SET_ELEM_INTERVAL_END))
3415 return -EINVAL;
3416 if (nla[NFTA_SET_ELEM_DATA] != NULL &&
3417 flags & NFT_SET_ELEM_INTERVAL_END)
3418 return -EINVAL;
3419 } else {
3420 if (nla[NFTA_SET_ELEM_DATA] != NULL)
3421 return -EINVAL;
3422 }
3423
3424 timeout = 0;
3425 if (nla[NFTA_SET_ELEM_TIMEOUT] != NULL) {
3426 if (!(set->flags & NFT_SET_TIMEOUT))
3427 return -EINVAL;
3428 timeout = be64_to_cpu(nla_get_be64(nla[NFTA_SET_ELEM_TIMEOUT]));
3429 } else if (set->flags & NFT_SET_TIMEOUT) {
3430 timeout = set->timeout;
3431 }
3432
3433 err = nft_data_init(ctx, &elem.key.val, sizeof(elem.key), &d1,
3434 nla[NFTA_SET_ELEM_KEY]);
3435 if (err < 0)
3436 goto err1;
3437 err = -EINVAL;
3438 if (d1.type != NFT_DATA_VALUE || d1.len != set->klen)
3439 goto err2;
3440
3441 nft_set_ext_add_length(&tmpl, NFT_SET_EXT_KEY, d1.len);
3442 if (timeout > 0) {
3443 nft_set_ext_add(&tmpl, NFT_SET_EXT_EXPIRATION);
3444 if (timeout != set->timeout)
3445 nft_set_ext_add(&tmpl, NFT_SET_EXT_TIMEOUT);
3446 }
3447
3448 if (nla[NFTA_SET_ELEM_DATA] != NULL) {
3449 err = nft_data_init(ctx, &data, sizeof(data), &d2,
3450 nla[NFTA_SET_ELEM_DATA]);
3451 if (err < 0)
3452 goto err2;
3453
3454 err = -EINVAL;
3455 if (set->dtype != NFT_DATA_VERDICT && d2.len != set->dlen)
3456 goto err3;
3457
3458 dreg = nft_type_to_reg(set->dtype);
3459 list_for_each_entry(binding, &set->bindings, list) {
3460 struct nft_ctx bind_ctx = {
3461 .afi = ctx->afi,
3462 .table = ctx->table,
3463 .chain = (struct nft_chain *)binding->chain,
3464 };
3465
3466 if (!(binding->flags & NFT_SET_MAP))
3467 continue;
3468
3469 err = nft_validate_register_store(&bind_ctx, dreg,
3470 &data,
3471 d2.type, d2.len);
3472 if (err < 0)
3473 goto err3;
3474 }
3475
3476 nft_set_ext_add_length(&tmpl, NFT_SET_EXT_DATA, d2.len);
3477 }
3478
3479
3480
3481
3482
3483 ulen = 0;
3484 if (nla[NFTA_SET_ELEM_USERDATA] != NULL) {
3485 ulen = nla_len(nla[NFTA_SET_ELEM_USERDATA]);
3486 if (ulen > 0)
3487 nft_set_ext_add_length(&tmpl, NFT_SET_EXT_USERDATA,
3488 ulen);
3489 }
3490
3491 err = -ENOMEM;
3492 elem.priv = nft_set_elem_init(set, &tmpl, elem.key.val.data, data.data,
3493 timeout, GFP_KERNEL);
3494 if (elem.priv == NULL)
3495 goto err3;
3496
3497 ext = nft_set_elem_ext(set, elem.priv);
3498 if (flags)
3499 *nft_set_ext_flags(ext) = flags;
3500 if (ulen > 0) {
3501 udata = nft_set_ext_userdata(ext);
3502 udata->len = ulen - 1;
3503 nla_memcpy(&udata->data, nla[NFTA_SET_ELEM_USERDATA], ulen);
3504 }
3505
3506 trans = nft_trans_elem_alloc(ctx, NFT_MSG_NEWSETELEM, set);
3507 if (trans == NULL)
3508 goto err4;
3509
3510 ext->genmask = nft_genmask_cur(ctx->net) | NFT_SET_ELEM_BUSY_MASK;
3511 err = set->ops->insert(set, &elem);
3512 if (err < 0)
3513 goto err5;
3514
3515 nft_trans_elem(trans) = elem;
3516 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
3517 return 0;
3518
3519err5:
3520 kfree(trans);
3521err4:
3522 kfree(elem.priv);
3523err3:
3524 if (nla[NFTA_SET_ELEM_DATA] != NULL)
3525 nft_data_uninit(&data, d2.type);
3526err2:
3527 nft_data_uninit(&elem.key.val, d1.type);
3528err1:
3529 return err;
3530}
3531
3532static int nf_tables_newsetelem(struct sock *nlsk, struct sk_buff *skb,
3533 const struct nlmsghdr *nlh,
3534 const struct nlattr * const nla[])
3535{
3536 struct net *net = sock_net(skb->sk);
3537 const struct nlattr *attr;
3538 struct nft_set *set;
3539 struct nft_ctx ctx;
3540 int rem, err = 0;
3541
3542 if (nla[NFTA_SET_ELEM_LIST_ELEMENTS] == NULL)
3543 return -EINVAL;
3544
3545 err = nft_ctx_init_from_elemattr(&ctx, skb, nlh, nla, true);
3546 if (err < 0)
3547 return err;
3548
3549 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_ELEM_LIST_SET]);
3550 if (IS_ERR(set)) {
3551 if (nla[NFTA_SET_ELEM_LIST_SET_ID]) {
3552 set = nf_tables_set_lookup_byid(net,
3553 nla[NFTA_SET_ELEM_LIST_SET_ID]);
3554 }
3555 if (IS_ERR(set))
3556 return PTR_ERR(set);
3557 }
3558
3559 if (!list_empty(&set->bindings) && set->flags & NFT_SET_CONSTANT)
3560 return -EBUSY;
3561
3562 nla_for_each_nested(attr, nla[NFTA_SET_ELEM_LIST_ELEMENTS], rem) {
3563 if (set->size &&
3564 !atomic_add_unless(&set->nelems, 1, set->size + set->ndeact))
3565 return -ENFILE;
3566
3567 err = nft_add_set_elem(&ctx, set, attr);
3568 if (err < 0) {
3569 atomic_dec(&set->nelems);
3570 break;
3571 }
3572 }
3573 return err;
3574}
3575
3576static int nft_del_setelem(struct nft_ctx *ctx, struct nft_set *set,
3577 const struct nlattr *attr)
3578{
3579 struct nlattr *nla[NFTA_SET_ELEM_MAX + 1];
3580 struct nft_data_desc desc;
3581 struct nft_set_elem elem;
3582 struct nft_trans *trans;
3583 int err;
3584
3585 err = nla_parse_nested(nla, NFTA_SET_ELEM_MAX, attr,
3586 nft_set_elem_policy);
3587 if (err < 0)
3588 goto err1;
3589
3590 err = -EINVAL;
3591 if (nla[NFTA_SET_ELEM_KEY] == NULL)
3592 goto err1;
3593
3594 err = nft_data_init(ctx, &elem.key.val, sizeof(elem.key), &desc,
3595 nla[NFTA_SET_ELEM_KEY]);
3596 if (err < 0)
3597 goto err1;
3598
3599 err = -EINVAL;
3600 if (desc.type != NFT_DATA_VALUE || desc.len != set->klen)
3601 goto err2;
3602
3603 trans = nft_trans_elem_alloc(ctx, NFT_MSG_DELSETELEM, set);
3604 if (trans == NULL) {
3605 err = -ENOMEM;
3606 goto err2;
3607 }
3608
3609 elem.priv = set->ops->deactivate(set, &elem);
3610 if (elem.priv == NULL) {
3611 err = -ENOENT;
3612 goto err3;
3613 }
3614
3615 nft_trans_elem(trans) = elem;
3616 list_add_tail(&trans->list, &ctx->net->nft.commit_list);
3617 return 0;
3618
3619err3:
3620 kfree(trans);
3621err2:
3622 nft_data_uninit(&elem.key.val, desc.type);
3623err1:
3624 return err;
3625}
3626
3627static int nf_tables_delsetelem(struct sock *nlsk, struct sk_buff *skb,
3628 const struct nlmsghdr *nlh,
3629 const struct nlattr * const nla[])
3630{
3631 const struct nlattr *attr;
3632 struct nft_set *set;
3633 struct nft_ctx ctx;
3634 int rem, err = 0;
3635
3636 if (nla[NFTA_SET_ELEM_LIST_ELEMENTS] == NULL)
3637 return -EINVAL;
3638
3639 err = nft_ctx_init_from_elemattr(&ctx, skb, nlh, nla, false);
3640 if (err < 0)
3641 return err;
3642
3643 set = nf_tables_set_lookup(ctx.table, nla[NFTA_SET_ELEM_LIST_SET]);
3644 if (IS_ERR(set))
3645 return PTR_ERR(set);
3646 if (!list_empty(&set->bindings) && set->flags & NFT_SET_CONSTANT)
3647 return -EBUSY;
3648
3649 nla_for_each_nested(attr, nla[NFTA_SET_ELEM_LIST_ELEMENTS], rem) {
3650 err = nft_del_setelem(&ctx, set, attr);
3651 if (err < 0)
3652 break;
3653
3654 set->ndeact++;
3655 }
3656 return err;
3657}
3658
3659void nft_set_gc_batch_release(struct rcu_head *rcu)
3660{
3661 struct nft_set_gc_batch *gcb;
3662 unsigned int i;
3663
3664 gcb = container_of(rcu, struct nft_set_gc_batch, head.rcu);
3665 for (i = 0; i < gcb->head.cnt; i++)
3666 nft_set_elem_destroy(gcb->head.set, gcb->elems[i]);
3667 kfree(gcb);
3668}
3669EXPORT_SYMBOL_GPL(nft_set_gc_batch_release);
3670
3671struct nft_set_gc_batch *nft_set_gc_batch_alloc(const struct nft_set *set,
3672 gfp_t gfp)
3673{
3674 struct nft_set_gc_batch *gcb;
3675
3676 gcb = kzalloc(sizeof(*gcb), gfp);
3677 if (gcb == NULL)
3678 return gcb;
3679 gcb->head.set = set;
3680 return gcb;
3681}
3682EXPORT_SYMBOL_GPL(nft_set_gc_batch_alloc);
3683
3684static int nf_tables_fill_gen_info(struct sk_buff *skb, struct net *net,
3685 u32 portid, u32 seq)
3686{
3687 struct nlmsghdr *nlh;
3688 struct nfgenmsg *nfmsg;
3689 int event = (NFNL_SUBSYS_NFTABLES << 8) | NFT_MSG_NEWGEN;
3690
3691 nlh = nlmsg_put(skb, portid, seq, event, sizeof(struct nfgenmsg), 0);
3692 if (nlh == NULL)
3693 goto nla_put_failure;
3694
3695 nfmsg = nlmsg_data(nlh);
3696 nfmsg->nfgen_family = AF_UNSPEC;
3697 nfmsg->version = NFNETLINK_V0;
3698 nfmsg->res_id = htons(net->nft.base_seq & 0xffff);
3699
3700 if (nla_put_be32(skb, NFTA_GEN_ID, htonl(net->nft.base_seq)))
3701 goto nla_put_failure;
3702
3703 nlmsg_end(skb, nlh);
3704 return 0;
3705
3706nla_put_failure:
3707 nlmsg_trim(skb, nlh);
3708 return -EMSGSIZE;
3709}
3710
3711static int nf_tables_gen_notify(struct net *net, struct sk_buff *skb, int event)
3712{
3713 struct nlmsghdr *nlh = nlmsg_hdr(skb);
3714 struct sk_buff *skb2;
3715 int err;
3716
3717 if (nlmsg_report(nlh) &&
3718 !nfnetlink_has_listeners(net, NFNLGRP_NFTABLES))
3719 return 0;
3720
3721 err = -ENOBUFS;
3722 skb2 = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
3723 if (skb2 == NULL)
3724 goto err;
3725
3726 err = nf_tables_fill_gen_info(skb2, net, NETLINK_CB(skb).portid,
3727 nlh->nlmsg_seq);
3728 if (err < 0) {
3729 kfree_skb(skb2);
3730 goto err;
3731 }
3732
3733 err = nfnetlink_send(skb2, net, NETLINK_CB(skb).portid,
3734 NFNLGRP_NFTABLES, nlmsg_report(nlh), GFP_KERNEL);
3735err:
3736 if (err < 0) {
3737 nfnetlink_set_err(net, NETLINK_CB(skb).portid, NFNLGRP_NFTABLES,
3738 err);
3739 }
3740 return err;
3741}
3742
3743static int nf_tables_getgen(struct sock *nlsk, struct sk_buff *skb,
3744 const struct nlmsghdr *nlh,
3745 const struct nlattr * const nla[])
3746{
3747 struct net *net = sock_net(skb->sk);
3748 struct sk_buff *skb2;
3749 int err;
3750
3751 skb2 = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
3752 if (skb2 == NULL)
3753 return -ENOMEM;
3754
3755 err = nf_tables_fill_gen_info(skb2, net, NETLINK_CB(skb).portid,
3756 nlh->nlmsg_seq);
3757 if (err < 0)
3758 goto err;
3759
3760 return nlmsg_unicast(nlsk, skb2, NETLINK_CB(skb).portid);
3761err:
3762 kfree_skb(skb2);
3763 return err;
3764}
3765
3766static const struct nfnl_callback nf_tables_cb[NFT_MSG_MAX] = {
3767 [NFT_MSG_NEWTABLE] = {
3768 .call_batch = nf_tables_newtable,
3769 .attr_count = NFTA_TABLE_MAX,
3770 .policy = nft_table_policy,
3771 },
3772 [NFT_MSG_GETTABLE] = {
3773 .call = nf_tables_gettable,
3774 .attr_count = NFTA_TABLE_MAX,
3775 .policy = nft_table_policy,
3776 },
3777 [NFT_MSG_DELTABLE] = {
3778 .call_batch = nf_tables_deltable,
3779 .attr_count = NFTA_TABLE_MAX,
3780 .policy = nft_table_policy,
3781 },
3782 [NFT_MSG_NEWCHAIN] = {
3783 .call_batch = nf_tables_newchain,
3784 .attr_count = NFTA_CHAIN_MAX,
3785 .policy = nft_chain_policy,
3786 },
3787 [NFT_MSG_GETCHAIN] = {
3788 .call = nf_tables_getchain,
3789 .attr_count = NFTA_CHAIN_MAX,
3790 .policy = nft_chain_policy,
3791 },
3792 [NFT_MSG_DELCHAIN] = {
3793 .call_batch = nf_tables_delchain,
3794 .attr_count = NFTA_CHAIN_MAX,
3795 .policy = nft_chain_policy,
3796 },
3797 [NFT_MSG_NEWRULE] = {
3798 .call_batch = nf_tables_newrule,
3799 .attr_count = NFTA_RULE_MAX,
3800 .policy = nft_rule_policy,
3801 },
3802 [NFT_MSG_GETRULE] = {
3803 .call = nf_tables_getrule,
3804 .attr_count = NFTA_RULE_MAX,
3805 .policy = nft_rule_policy,
3806 },
3807 [NFT_MSG_DELRULE] = {
3808 .call_batch = nf_tables_delrule,
3809 .attr_count = NFTA_RULE_MAX,
3810 .policy = nft_rule_policy,
3811 },
3812 [NFT_MSG_NEWSET] = {
3813 .call_batch = nf_tables_newset,
3814 .attr_count = NFTA_SET_MAX,
3815 .policy = nft_set_policy,
3816 },
3817 [NFT_MSG_GETSET] = {
3818 .call = nf_tables_getset,
3819 .attr_count = NFTA_SET_MAX,
3820 .policy = nft_set_policy,
3821 },
3822 [NFT_MSG_DELSET] = {
3823 .call_batch = nf_tables_delset,
3824 .attr_count = NFTA_SET_MAX,
3825 .policy = nft_set_policy,
3826 },
3827 [NFT_MSG_NEWSETELEM] = {
3828 .call_batch = nf_tables_newsetelem,
3829 .attr_count = NFTA_SET_ELEM_LIST_MAX,
3830 .policy = nft_set_elem_list_policy,
3831 },
3832 [NFT_MSG_GETSETELEM] = {
3833 .call = nf_tables_getsetelem,
3834 .attr_count = NFTA_SET_ELEM_LIST_MAX,
3835 .policy = nft_set_elem_list_policy,
3836 },
3837 [NFT_MSG_DELSETELEM] = {
3838 .call_batch = nf_tables_delsetelem,
3839 .attr_count = NFTA_SET_ELEM_LIST_MAX,
3840 .policy = nft_set_elem_list_policy,
3841 },
3842 [NFT_MSG_GETGEN] = {
3843 .call = nf_tables_getgen,
3844 },
3845};
3846
3847static void nft_chain_commit_update(struct nft_trans *trans)
3848{
3849 struct nft_base_chain *basechain;
3850
3851 if (nft_trans_chain_name(trans)[0])
3852 strcpy(trans->ctx.chain->name, nft_trans_chain_name(trans));
3853
3854 if (!(trans->ctx.chain->flags & NFT_BASE_CHAIN))
3855 return;
3856
3857 basechain = nft_base_chain(trans->ctx.chain);
3858 nft_chain_stats_replace(basechain, nft_trans_chain_stats(trans));
3859
3860 switch (nft_trans_chain_policy(trans)) {
3861 case NF_DROP:
3862 case NF_ACCEPT:
3863 basechain->policy = nft_trans_chain_policy(trans);
3864 break;
3865 }
3866}
3867
3868static void nf_tables_commit_release(struct nft_trans *trans)
3869{
3870 switch (trans->msg_type) {
3871 case NFT_MSG_DELTABLE:
3872 nf_tables_table_destroy(&trans->ctx);
3873 break;
3874 case NFT_MSG_DELCHAIN:
3875 nf_tables_chain_destroy(trans->ctx.chain);
3876 break;
3877 case NFT_MSG_DELRULE:
3878 nf_tables_rule_destroy(&trans->ctx, nft_trans_rule(trans));
3879 break;
3880 case NFT_MSG_DELSET:
3881 nft_set_destroy(nft_trans_set(trans));
3882 break;
3883 case NFT_MSG_DELSETELEM:
3884 nft_set_elem_destroy(nft_trans_elem_set(trans),
3885 nft_trans_elem(trans).priv);
3886 break;
3887 }
3888 kfree(trans);
3889}
3890
3891static int nf_tables_commit(struct sk_buff *skb)
3892{
3893 struct net *net = sock_net(skb->sk);
3894 struct nft_trans *trans, *next;
3895 struct nft_trans_elem *te;
3896
3897
3898 while (++net->nft.base_seq == 0);
3899
3900
3901 net->nft.gencursor = nft_gencursor_next(net);
3902
3903
3904
3905
3906 synchronize_rcu();
3907
3908 list_for_each_entry_safe(trans, next, &net->nft.commit_list, list) {
3909 switch (trans->msg_type) {
3910 case NFT_MSG_NEWTABLE:
3911 if (nft_trans_table_update(trans)) {
3912 if (!nft_trans_table_enable(trans)) {
3913 nf_tables_table_disable(trans->ctx.afi,
3914 trans->ctx.table);
3915 trans->ctx.table->flags |= NFT_TABLE_F_DORMANT;
3916 }
3917 } else {
3918 trans->ctx.table->flags &= ~NFT_TABLE_INACTIVE;
3919 }
3920 nf_tables_table_notify(&trans->ctx, NFT_MSG_NEWTABLE);
3921 nft_trans_destroy(trans);
3922 break;
3923 case NFT_MSG_DELTABLE:
3924 nf_tables_table_notify(&trans->ctx, NFT_MSG_DELTABLE);
3925 break;
3926 case NFT_MSG_NEWCHAIN:
3927 if (nft_trans_chain_update(trans))
3928 nft_chain_commit_update(trans);
3929 else
3930 trans->ctx.chain->flags &= ~NFT_CHAIN_INACTIVE;
3931
3932 nf_tables_chain_notify(&trans->ctx, NFT_MSG_NEWCHAIN);
3933 nft_trans_destroy(trans);
3934 break;
3935 case NFT_MSG_DELCHAIN:
3936 nf_tables_chain_notify(&trans->ctx, NFT_MSG_DELCHAIN);
3937 nf_tables_unregister_hooks(trans->ctx.table,
3938 trans->ctx.chain,
3939 trans->ctx.afi->nops);
3940 break;
3941 case NFT_MSG_NEWRULE:
3942 nft_rule_clear(trans->ctx.net, nft_trans_rule(trans));
3943 nf_tables_rule_notify(&trans->ctx,
3944 nft_trans_rule(trans),
3945 NFT_MSG_NEWRULE);
3946 nft_trans_destroy(trans);
3947 break;
3948 case NFT_MSG_DELRULE:
3949 list_del_rcu(&nft_trans_rule(trans)->list);
3950 nf_tables_rule_notify(&trans->ctx,
3951 nft_trans_rule(trans),
3952 NFT_MSG_DELRULE);
3953 break;
3954 case NFT_MSG_NEWSET:
3955 nft_trans_set(trans)->flags &= ~NFT_SET_INACTIVE;
3956
3957
3958
3959 if (nft_trans_set(trans)->flags & NFT_SET_ANONYMOUS &&
3960 !list_empty(&nft_trans_set(trans)->bindings))
3961 trans->ctx.table->use--;
3962
3963 nf_tables_set_notify(&trans->ctx, nft_trans_set(trans),
3964 NFT_MSG_NEWSET, GFP_KERNEL);
3965 nft_trans_destroy(trans);
3966 break;
3967 case NFT_MSG_DELSET:
3968 nf_tables_set_notify(&trans->ctx, nft_trans_set(trans),
3969 NFT_MSG_DELSET, GFP_KERNEL);
3970 break;
3971 case NFT_MSG_NEWSETELEM:
3972 te = (struct nft_trans_elem *)trans->data;
3973
3974 te->set->ops->activate(te->set, &te->elem);
3975 nf_tables_setelem_notify(&trans->ctx, te->set,
3976 &te->elem,
3977 NFT_MSG_NEWSETELEM, 0);
3978 nft_trans_destroy(trans);
3979 break;
3980 case NFT_MSG_DELSETELEM:
3981 te = (struct nft_trans_elem *)trans->data;
3982
3983 nf_tables_setelem_notify(&trans->ctx, te->set,
3984 &te->elem,
3985 NFT_MSG_DELSETELEM, 0);
3986 te->set->ops->remove(te->set, &te->elem);
3987 atomic_dec(&te->set->nelems);
3988 te->set->ndeact--;
3989 break;
3990 }
3991 }
3992
3993 synchronize_rcu();
3994
3995 list_for_each_entry_safe(trans, next, &net->nft.commit_list, list) {
3996 list_del(&trans->list);
3997 nf_tables_commit_release(trans);
3998 }
3999
4000 nf_tables_gen_notify(net, skb, NFT_MSG_NEWGEN);
4001
4002 return 0;
4003}
4004
4005static void nf_tables_abort_release(struct nft_trans *trans)
4006{
4007 switch (trans->msg_type) {
4008 case NFT_MSG_NEWTABLE:
4009 nf_tables_table_destroy(&trans->ctx);
4010 break;
4011 case NFT_MSG_NEWCHAIN:
4012 nf_tables_chain_destroy(trans->ctx.chain);
4013 break;
4014 case NFT_MSG_NEWRULE:
4015 nf_tables_rule_destroy(&trans->ctx, nft_trans_rule(trans));
4016 break;
4017 case NFT_MSG_NEWSET:
4018 nft_set_destroy(nft_trans_set(trans));
4019 break;
4020 case NFT_MSG_NEWSETELEM:
4021 nft_set_elem_destroy(nft_trans_elem_set(trans),
4022 nft_trans_elem(trans).priv);
4023 break;
4024 }
4025 kfree(trans);
4026}
4027
4028static int nf_tables_abort(struct sk_buff *skb)
4029{
4030 struct net *net = sock_net(skb->sk);
4031 struct nft_trans *trans, *next;
4032 struct nft_trans_elem *te;
4033
4034 list_for_each_entry_safe(trans, next, &net->nft.commit_list, list) {
4035 switch (trans->msg_type) {
4036 case NFT_MSG_NEWTABLE:
4037 if (nft_trans_table_update(trans)) {
4038 if (nft_trans_table_enable(trans)) {
4039 nf_tables_table_disable(trans->ctx.afi,
4040 trans->ctx.table);
4041 trans->ctx.table->flags |= NFT_TABLE_F_DORMANT;
4042 }
4043 nft_trans_destroy(trans);
4044 } else {
4045 list_del_rcu(&trans->ctx.table->list);
4046 }
4047 break;
4048 case NFT_MSG_DELTABLE:
4049 list_add_tail_rcu(&trans->ctx.table->list,
4050 &trans->ctx.afi->tables);
4051 nft_trans_destroy(trans);
4052 break;
4053 case NFT_MSG_NEWCHAIN:
4054 if (nft_trans_chain_update(trans)) {
4055 free_percpu(nft_trans_chain_stats(trans));
4056
4057 nft_trans_destroy(trans);
4058 } else {
4059 trans->ctx.table->use--;
4060 list_del_rcu(&trans->ctx.chain->list);
4061 nf_tables_unregister_hooks(trans->ctx.table,
4062 trans->ctx.chain,
4063 trans->ctx.afi->nops);
4064 }
4065 break;
4066 case NFT_MSG_DELCHAIN:
4067 trans->ctx.table->use++;
4068 list_add_tail_rcu(&trans->ctx.chain->list,
4069 &trans->ctx.table->chains);
4070 nft_trans_destroy(trans);
4071 break;
4072 case NFT_MSG_NEWRULE:
4073 trans->ctx.chain->use--;
4074 list_del_rcu(&nft_trans_rule(trans)->list);
4075 break;
4076 case NFT_MSG_DELRULE:
4077 trans->ctx.chain->use++;
4078 nft_rule_clear(trans->ctx.net, nft_trans_rule(trans));
4079 nft_trans_destroy(trans);
4080 break;
4081 case NFT_MSG_NEWSET:
4082 trans->ctx.table->use--;
4083 list_del_rcu(&nft_trans_set(trans)->list);
4084 break;
4085 case NFT_MSG_DELSET:
4086 trans->ctx.table->use++;
4087 list_add_tail_rcu(&nft_trans_set(trans)->list,
4088 &trans->ctx.table->sets);
4089 nft_trans_destroy(trans);
4090 break;
4091 case NFT_MSG_NEWSETELEM:
4092 te = (struct nft_trans_elem *)trans->data;
4093
4094 te->set->ops->remove(te->set, &te->elem);
4095 atomic_dec(&te->set->nelems);
4096 break;
4097 case NFT_MSG_DELSETELEM:
4098 te = (struct nft_trans_elem *)trans->data;
4099
4100 te->set->ops->activate(te->set, &te->elem);
4101 te->set->ndeact--;
4102
4103 nft_trans_destroy(trans);
4104 break;
4105 }
4106 }
4107
4108 synchronize_rcu();
4109
4110 list_for_each_entry_safe_reverse(trans, next,
4111 &net->nft.commit_list, list) {
4112 list_del(&trans->list);
4113 nf_tables_abort_release(trans);
4114 }
4115
4116 return 0;
4117}
4118
4119static const struct nfnetlink_subsystem nf_tables_subsys = {
4120 .name = "nf_tables",
4121 .subsys_id = NFNL_SUBSYS_NFTABLES,
4122 .cb_count = NFT_MSG_MAX,
4123 .cb = nf_tables_cb,
4124 .commit = nf_tables_commit,
4125 .abort = nf_tables_abort,
4126};
4127
4128int nft_chain_validate_dependency(const struct nft_chain *chain,
4129 enum nft_chain_type type)
4130{
4131 const struct nft_base_chain *basechain;
4132
4133 if (chain->flags & NFT_BASE_CHAIN) {
4134 basechain = nft_base_chain(chain);
4135 if (basechain->type->type != type)
4136 return -EOPNOTSUPP;
4137 }
4138 return 0;
4139}
4140EXPORT_SYMBOL_GPL(nft_chain_validate_dependency);
4141
4142int nft_chain_validate_hooks(const struct nft_chain *chain,
4143 unsigned int hook_flags)
4144{
4145 struct nft_base_chain *basechain;
4146
4147 if (chain->flags & NFT_BASE_CHAIN) {
4148 basechain = nft_base_chain(chain);
4149
4150 if ((1 << basechain->ops[0].hooknum) & hook_flags)
4151 return 0;
4152
4153 return -EOPNOTSUPP;
4154 }
4155
4156 return 0;
4157}
4158EXPORT_SYMBOL_GPL(nft_chain_validate_hooks);
4159
4160
4161
4162
4163
4164
4165
4166
4167
4168
4169static int nf_tables_check_loops(const struct nft_ctx *ctx,
4170 const struct nft_chain *chain);
4171
4172static int nf_tables_loop_check_setelem(const struct nft_ctx *ctx,
4173 const struct nft_set *set,
4174 const struct nft_set_iter *iter,
4175 const struct nft_set_elem *elem)
4176{
4177 const struct nft_set_ext *ext = nft_set_elem_ext(set, elem->priv);
4178 const struct nft_data *data;
4179
4180 if (nft_set_ext_exists(ext, NFT_SET_EXT_FLAGS) &&
4181 *nft_set_ext_flags(ext) & NFT_SET_ELEM_INTERVAL_END)
4182 return 0;
4183
4184 data = nft_set_ext_data(ext);
4185 switch (data->verdict.code) {
4186 case NFT_JUMP:
4187 case NFT_GOTO:
4188 return nf_tables_check_loops(ctx, data->verdict.chain);
4189 default:
4190 return 0;
4191 }
4192}
4193
4194static int nf_tables_check_loops(const struct nft_ctx *ctx,
4195 const struct nft_chain *chain)
4196{
4197 const struct nft_rule *rule;
4198 const struct nft_expr *expr, *last;
4199 const struct nft_set *set;
4200 struct nft_set_binding *binding;
4201 struct nft_set_iter iter;
4202
4203 if (ctx->chain == chain)
4204 return -ELOOP;
4205
4206 list_for_each_entry(rule, &chain->rules, list) {
4207 nft_rule_for_each_expr(expr, last, rule) {
4208 const struct nft_data *data = NULL;
4209 int err;
4210
4211 if (!expr->ops->validate)
4212 continue;
4213
4214 err = expr->ops->validate(ctx, expr, &data);
4215 if (err < 0)
4216 return err;
4217
4218 if (data == NULL)
4219 continue;
4220
4221 switch (data->verdict.code) {
4222 case NFT_JUMP:
4223 case NFT_GOTO:
4224 err = nf_tables_check_loops(ctx,
4225 data->verdict.chain);
4226 if (err < 0)
4227 return err;
4228 default:
4229 break;
4230 }
4231 }
4232 }
4233
4234 list_for_each_entry(set, &ctx->table->sets, list) {
4235 if (!(set->flags & NFT_SET_MAP) ||
4236 set->dtype != NFT_DATA_VERDICT)
4237 continue;
4238
4239 list_for_each_entry(binding, &set->bindings, list) {
4240 if (!(binding->flags & NFT_SET_MAP) ||
4241 binding->chain != chain)
4242 continue;
4243
4244 iter.skip = 0;
4245 iter.count = 0;
4246 iter.err = 0;
4247 iter.fn = nf_tables_loop_check_setelem;
4248
4249 set->ops->walk(ctx, set, &iter);
4250 if (iter.err < 0)
4251 return iter.err;
4252 }
4253 }
4254
4255 return 0;
4256}
4257
4258
4259
4260
4261
4262
4263
4264
4265
4266
4267unsigned int nft_parse_register(const struct nlattr *attr)
4268{
4269 unsigned int reg;
4270
4271 reg = ntohl(nla_get_be32(attr));
4272 switch (reg) {
4273 case NFT_REG_VERDICT...NFT_REG_4:
4274 return reg * NFT_REG_SIZE / NFT_REG32_SIZE;
4275 default:
4276 return reg + NFT_REG_SIZE / NFT_REG32_SIZE - NFT_REG32_00;
4277 }
4278}
4279EXPORT_SYMBOL_GPL(nft_parse_register);
4280
4281
4282
4283
4284
4285
4286
4287
4288
4289
4290
4291
4292int nft_dump_register(struct sk_buff *skb, unsigned int attr, unsigned int reg)
4293{
4294 if (reg % (NFT_REG_SIZE / NFT_REG32_SIZE) == 0)
4295 reg = reg / (NFT_REG_SIZE / NFT_REG32_SIZE);
4296 else
4297 reg = reg - NFT_REG_SIZE / NFT_REG32_SIZE + NFT_REG32_00;
4298
4299 return nla_put_be32(skb, attr, htonl(reg));
4300}
4301EXPORT_SYMBOL_GPL(nft_dump_register);
4302
4303
4304
4305
4306
4307
4308
4309
4310
4311
4312int nft_validate_register_load(enum nft_registers reg, unsigned int len)
4313{
4314 if (reg < NFT_REG_1 * NFT_REG_SIZE / NFT_REG32_SIZE)
4315 return -EINVAL;
4316 if (len == 0)
4317 return -EINVAL;
4318 if (reg * NFT_REG32_SIZE + len > FIELD_SIZEOF(struct nft_regs, data))
4319 return -ERANGE;
4320
4321 return 0;
4322}
4323EXPORT_SYMBOL_GPL(nft_validate_register_load);
4324
4325
4326
4327
4328
4329
4330
4331
4332
4333
4334
4335
4336
4337
4338
4339int nft_validate_register_store(const struct nft_ctx *ctx,
4340 enum nft_registers reg,
4341 const struct nft_data *data,
4342 enum nft_data_types type, unsigned int len)
4343{
4344 int err;
4345
4346 switch (reg) {
4347 case NFT_REG_VERDICT:
4348 if (type != NFT_DATA_VERDICT)
4349 return -EINVAL;
4350
4351 if (data != NULL &&
4352 (data->verdict.code == NFT_GOTO ||
4353 data->verdict.code == NFT_JUMP)) {
4354 err = nf_tables_check_loops(ctx, data->verdict.chain);
4355 if (err < 0)
4356 return err;
4357
4358 if (ctx->chain->level + 1 >
4359 data->verdict.chain->level) {
4360 if (ctx->chain->level + 1 == NFT_JUMP_STACK_SIZE)
4361 return -EMLINK;
4362 data->verdict.chain->level = ctx->chain->level + 1;
4363 }
4364 }
4365
4366 return 0;
4367 default:
4368 if (reg < NFT_REG_1 * NFT_REG_SIZE / NFT_REG32_SIZE)
4369 return -EINVAL;
4370 if (len == 0)
4371 return -EINVAL;
4372 if (reg * NFT_REG32_SIZE + len >
4373 FIELD_SIZEOF(struct nft_regs, data))
4374 return -ERANGE;
4375
4376 if (data != NULL && type != NFT_DATA_VALUE)
4377 return -EINVAL;
4378 return 0;
4379 }
4380}
4381EXPORT_SYMBOL_GPL(nft_validate_register_store);
4382
4383static const struct nla_policy nft_verdict_policy[NFTA_VERDICT_MAX + 1] = {
4384 [NFTA_VERDICT_CODE] = { .type = NLA_U32 },
4385 [NFTA_VERDICT_CHAIN] = { .type = NLA_STRING,
4386 .len = NFT_CHAIN_MAXNAMELEN - 1 },
4387};
4388
4389static int nft_verdict_init(const struct nft_ctx *ctx, struct nft_data *data,
4390 struct nft_data_desc *desc, const struct nlattr *nla)
4391{
4392 struct nlattr *tb[NFTA_VERDICT_MAX + 1];
4393 struct nft_chain *chain;
4394 int err;
4395
4396 err = nla_parse_nested(tb, NFTA_VERDICT_MAX, nla, nft_verdict_policy);
4397 if (err < 0)
4398 return err;
4399
4400 if (!tb[NFTA_VERDICT_CODE])
4401 return -EINVAL;
4402 data->verdict.code = ntohl(nla_get_be32(tb[NFTA_VERDICT_CODE]));
4403
4404 switch (data->verdict.code) {
4405 default:
4406 switch (data->verdict.code & NF_VERDICT_MASK) {
4407 case NF_ACCEPT:
4408 case NF_DROP:
4409 case NF_QUEUE:
4410 break;
4411 default:
4412 return -EINVAL;
4413 }
4414
4415 case NFT_CONTINUE:
4416 case NFT_BREAK:
4417 case NFT_RETURN:
4418 break;
4419 case NFT_JUMP:
4420 case NFT_GOTO:
4421 if (!tb[NFTA_VERDICT_CHAIN])
4422 return -EINVAL;
4423 chain = nf_tables_chain_lookup(ctx->table,
4424 tb[NFTA_VERDICT_CHAIN]);
4425 if (IS_ERR(chain))
4426 return PTR_ERR(chain);
4427 if (chain->flags & NFT_BASE_CHAIN)
4428 return -EOPNOTSUPP;
4429
4430 chain->use++;
4431 data->verdict.chain = chain;
4432 break;
4433 }
4434
4435 desc->len = sizeof(data->verdict);
4436 desc->type = NFT_DATA_VERDICT;
4437 return 0;
4438}
4439
4440static void nft_verdict_uninit(const struct nft_data *data)
4441{
4442 switch (data->verdict.code) {
4443 case NFT_JUMP:
4444 case NFT_GOTO:
4445 data->verdict.chain->use--;
4446 break;
4447 }
4448}
4449
4450static int nft_verdict_dump(struct sk_buff *skb, const struct nft_data *data)
4451{
4452 struct nlattr *nest;
4453
4454 nest = nla_nest_start(skb, NFTA_DATA_VERDICT);
4455 if (!nest)
4456 goto nla_put_failure;
4457
4458 if (nla_put_be32(skb, NFTA_VERDICT_CODE, htonl(data->verdict.code)))
4459 goto nla_put_failure;
4460
4461 switch (data->verdict.code) {
4462 case NFT_JUMP:
4463 case NFT_GOTO:
4464 if (nla_put_string(skb, NFTA_VERDICT_CHAIN,
4465 data->verdict.chain->name))
4466 goto nla_put_failure;
4467 }
4468 nla_nest_end(skb, nest);
4469 return 0;
4470
4471nla_put_failure:
4472 return -1;
4473}
4474
4475static int nft_value_init(const struct nft_ctx *ctx,
4476 struct nft_data *data, unsigned int size,
4477 struct nft_data_desc *desc, const struct nlattr *nla)
4478{
4479 unsigned int len;
4480
4481 len = nla_len(nla);
4482 if (len == 0)
4483 return -EINVAL;
4484 if (len > size)
4485 return -EOVERFLOW;
4486
4487 nla_memcpy(data->data, nla, len);
4488 desc->type = NFT_DATA_VALUE;
4489 desc->len = len;
4490 return 0;
4491}
4492
4493static int nft_value_dump(struct sk_buff *skb, const struct nft_data *data,
4494 unsigned int len)
4495{
4496 return nla_put(skb, NFTA_DATA_VALUE, len, data->data);
4497}
4498
4499static const struct nla_policy nft_data_policy[NFTA_DATA_MAX + 1] = {
4500 [NFTA_DATA_VALUE] = { .type = NLA_BINARY },
4501 [NFTA_DATA_VERDICT] = { .type = NLA_NESTED },
4502};
4503
4504
4505
4506
4507
4508
4509
4510
4511
4512
4513
4514
4515
4516
4517
4518
4519int nft_data_init(const struct nft_ctx *ctx,
4520 struct nft_data *data, unsigned int size,
4521 struct nft_data_desc *desc, const struct nlattr *nla)
4522{
4523 struct nlattr *tb[NFTA_DATA_MAX + 1];
4524 int err;
4525
4526 err = nla_parse_nested(tb, NFTA_DATA_MAX, nla, nft_data_policy);
4527 if (err < 0)
4528 return err;
4529
4530 if (tb[NFTA_DATA_VALUE])
4531 return nft_value_init(ctx, data, size, desc,
4532 tb[NFTA_DATA_VALUE]);
4533 if (tb[NFTA_DATA_VERDICT] && ctx != NULL)
4534 return nft_verdict_init(ctx, data, desc, tb[NFTA_DATA_VERDICT]);
4535 return -EINVAL;
4536}
4537EXPORT_SYMBOL_GPL(nft_data_init);
4538
4539
4540
4541
4542
4543
4544
4545
4546
4547
4548void nft_data_uninit(const struct nft_data *data, enum nft_data_types type)
4549{
4550 if (type < NFT_DATA_VERDICT)
4551 return;
4552 switch (type) {
4553 case NFT_DATA_VERDICT:
4554 return nft_verdict_uninit(data);
4555 default:
4556 WARN_ON(1);
4557 }
4558}
4559EXPORT_SYMBOL_GPL(nft_data_uninit);
4560
4561int nft_data_dump(struct sk_buff *skb, int attr, const struct nft_data *data,
4562 enum nft_data_types type, unsigned int len)
4563{
4564 struct nlattr *nest;
4565 int err;
4566
4567 nest = nla_nest_start(skb, attr);
4568 if (nest == NULL)
4569 return -1;
4570
4571 switch (type) {
4572 case NFT_DATA_VALUE:
4573 err = nft_value_dump(skb, data, len);
4574 break;
4575 case NFT_DATA_VERDICT:
4576 err = nft_verdict_dump(skb, data);
4577 break;
4578 default:
4579 err = -EINVAL;
4580 WARN_ON(1);
4581 }
4582
4583 nla_nest_end(skb, nest);
4584 return err;
4585}
4586EXPORT_SYMBOL_GPL(nft_data_dump);
4587
4588static int nf_tables_init_net(struct net *net)
4589{
4590 INIT_LIST_HEAD(&net->nft.af_info);
4591 INIT_LIST_HEAD(&net->nft.commit_list);
4592 net->nft.base_seq = 1;
4593 return 0;
4594}
4595
4596static struct pernet_operations nf_tables_net_ops = {
4597 .init = nf_tables_init_net,
4598};
4599
4600static int __init nf_tables_module_init(void)
4601{
4602 int err;
4603
4604 info = kmalloc(sizeof(struct nft_expr_info) * NFT_RULE_MAXEXPRS,
4605 GFP_KERNEL);
4606 if (info == NULL) {
4607 err = -ENOMEM;
4608 goto err1;
4609 }
4610
4611 err = nf_tables_core_module_init();
4612 if (err < 0)
4613 goto err2;
4614
4615 err = nfnetlink_subsys_register(&nf_tables_subsys);
4616 if (err < 0)
4617 goto err3;
4618
4619 pr_info("nf_tables: (c) 2007-2009 Patrick McHardy <kaber@trash.net>\n");
4620 return register_pernet_subsys(&nf_tables_net_ops);
4621err3:
4622 nf_tables_core_module_exit();
4623err2:
4624 kfree(info);
4625err1:
4626 return err;
4627}
4628
4629static void __exit nf_tables_module_exit(void)
4630{
4631 unregister_pernet_subsys(&nf_tables_net_ops);
4632 nfnetlink_subsys_unregister(&nf_tables_subsys);
4633 rcu_barrier();
4634 nf_tables_core_module_exit();
4635 kfree(info);
4636}
4637
4638module_init(nf_tables_module_init);
4639module_exit(nf_tables_module_exit);
4640
4641MODULE_LICENSE("GPL");
4642MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
4643MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_NFTABLES);
4644
