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