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