• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * net/sched/act_api.c	Packet action API.
4  *
5  * Author:	Jamal Hadi Salim
6  */
7 
8 #include <linux/types.h>
9 #include <linux/kernel.h>
10 #include <linux/string.h>
11 #include <linux/errno.h>
12 #include <linux/slab.h>
13 #include <linux/skbuff.h>
14 #include <linux/init.h>
15 #include <linux/kmod.h>
16 #include <linux/err.h>
17 #include <linux/module.h>
18 #include <net/net_namespace.h>
19 #include <net/sock.h>
20 #include <net/sch_generic.h>
21 #include <net/pkt_cls.h>
22 #include <net/act_api.h>
23 #include <net/netlink.h>
24 
25 #ifdef CONFIG_INET
26 DEFINE_STATIC_KEY_FALSE(tcf_frag_xmit_count);
27 EXPORT_SYMBOL_GPL(tcf_frag_xmit_count);
28 #endif
29 
tcf_dev_queue_xmit(struct sk_buff * skb,int (* xmit)(struct sk_buff * skb))30 int tcf_dev_queue_xmit(struct sk_buff *skb, int (*xmit)(struct sk_buff *skb))
31 {
32 #ifdef CONFIG_INET
33 	if (static_branch_unlikely(&tcf_frag_xmit_count))
34 		return sch_frag_xmit_hook(skb, xmit);
35 #endif
36 
37 	return xmit(skb);
38 }
39 EXPORT_SYMBOL_GPL(tcf_dev_queue_xmit);
40 
tcf_action_goto_chain_exec(const struct tc_action * a,struct tcf_result * res)41 static void tcf_action_goto_chain_exec(const struct tc_action *a,
42 				       struct tcf_result *res)
43 {
44 	const struct tcf_chain *chain = rcu_dereference_bh(a->goto_chain);
45 
46 	res->goto_tp = rcu_dereference_bh(chain->filter_chain);
47 }
48 
tcf_free_cookie_rcu(struct rcu_head * p)49 static void tcf_free_cookie_rcu(struct rcu_head *p)
50 {
51 	struct tc_cookie *cookie = container_of(p, struct tc_cookie, rcu);
52 
53 	kfree(cookie->data);
54 	kfree(cookie);
55 }
56 
tcf_set_action_cookie(struct tc_cookie __rcu ** old_cookie,struct tc_cookie * new_cookie)57 static void tcf_set_action_cookie(struct tc_cookie __rcu **old_cookie,
58 				  struct tc_cookie *new_cookie)
59 {
60 	struct tc_cookie *old;
61 
62 	old = xchg((__force struct tc_cookie **)old_cookie, new_cookie);
63 	if (old)
64 		call_rcu(&old->rcu, tcf_free_cookie_rcu);
65 }
66 
tcf_action_check_ctrlact(int action,struct tcf_proto * tp,struct tcf_chain ** newchain,struct netlink_ext_ack * extack)67 int tcf_action_check_ctrlact(int action, struct tcf_proto *tp,
68 			     struct tcf_chain **newchain,
69 			     struct netlink_ext_ack *extack)
70 {
71 	int opcode = TC_ACT_EXT_OPCODE(action), ret = -EINVAL;
72 	u32 chain_index;
73 
74 	if (!opcode)
75 		ret = action > TC_ACT_VALUE_MAX ? -EINVAL : 0;
76 	else if (opcode <= TC_ACT_EXT_OPCODE_MAX || action == TC_ACT_UNSPEC)
77 		ret = 0;
78 	if (ret) {
79 		NL_SET_ERR_MSG(extack, "invalid control action");
80 		goto end;
81 	}
82 
83 	if (TC_ACT_EXT_CMP(action, TC_ACT_GOTO_CHAIN)) {
84 		chain_index = action & TC_ACT_EXT_VAL_MASK;
85 		if (!tp || !newchain) {
86 			ret = -EINVAL;
87 			NL_SET_ERR_MSG(extack,
88 				       "can't goto NULL proto/chain");
89 			goto end;
90 		}
91 		*newchain = tcf_chain_get_by_act(tp->chain->block, chain_index);
92 		if (!*newchain) {
93 			ret = -ENOMEM;
94 			NL_SET_ERR_MSG(extack,
95 				       "can't allocate goto_chain");
96 		}
97 	}
98 end:
99 	return ret;
100 }
101 EXPORT_SYMBOL(tcf_action_check_ctrlact);
102 
tcf_action_set_ctrlact(struct tc_action * a,int action,struct tcf_chain * goto_chain)103 struct tcf_chain *tcf_action_set_ctrlact(struct tc_action *a, int action,
104 					 struct tcf_chain *goto_chain)
105 {
106 	a->tcfa_action = action;
107 	goto_chain = rcu_replace_pointer(a->goto_chain, goto_chain, 1);
108 	return goto_chain;
109 }
110 EXPORT_SYMBOL(tcf_action_set_ctrlact);
111 
112 /* XXX: For standalone actions, we don't need a RCU grace period either, because
113  * actions are always connected to filters and filters are already destroyed in
114  * RCU callbacks, so after a RCU grace period actions are already disconnected
115  * from filters. Readers later can not find us.
116  */
free_tcf(struct tc_action * p)117 static void free_tcf(struct tc_action *p)
118 {
119 	struct tcf_chain *chain = rcu_dereference_protected(p->goto_chain, 1);
120 
121 	free_percpu(p->cpu_bstats);
122 	free_percpu(p->cpu_bstats_hw);
123 	free_percpu(p->cpu_qstats);
124 
125 	tcf_set_action_cookie(&p->act_cookie, NULL);
126 	if (chain)
127 		tcf_chain_put_by_act(chain);
128 
129 	kfree(p);
130 }
131 
tcf_action_cleanup(struct tc_action * p)132 static void tcf_action_cleanup(struct tc_action *p)
133 {
134 	if (p->ops->cleanup)
135 		p->ops->cleanup(p);
136 
137 	gen_kill_estimator(&p->tcfa_rate_est);
138 	free_tcf(p);
139 }
140 
__tcf_action_put(struct tc_action * p,bool bind)141 static int __tcf_action_put(struct tc_action *p, bool bind)
142 {
143 	struct tcf_idrinfo *idrinfo = p->idrinfo;
144 
145 	if (refcount_dec_and_mutex_lock(&p->tcfa_refcnt, &idrinfo->lock)) {
146 		if (bind)
147 			atomic_dec(&p->tcfa_bindcnt);
148 		idr_remove(&idrinfo->action_idr, p->tcfa_index);
149 		mutex_unlock(&idrinfo->lock);
150 
151 		tcf_action_cleanup(p);
152 		return 1;
153 	}
154 
155 	if (bind)
156 		atomic_dec(&p->tcfa_bindcnt);
157 
158 	return 0;
159 }
160 
__tcf_idr_release(struct tc_action * p,bool bind,bool strict)161 static int __tcf_idr_release(struct tc_action *p, bool bind, bool strict)
162 {
163 	int ret = 0;
164 
165 	/* Release with strict==1 and bind==0 is only called through act API
166 	 * interface (classifiers always bind). Only case when action with
167 	 * positive reference count and zero bind count can exist is when it was
168 	 * also created with act API (unbinding last classifier will destroy the
169 	 * action if it was created by classifier). So only case when bind count
170 	 * can be changed after initial check is when unbound action is
171 	 * destroyed by act API while classifier binds to action with same id
172 	 * concurrently. This result either creation of new action(same behavior
173 	 * as before), or reusing existing action if concurrent process
174 	 * increments reference count before action is deleted. Both scenarios
175 	 * are acceptable.
176 	 */
177 	if (p) {
178 		if (!bind && strict && atomic_read(&p->tcfa_bindcnt) > 0)
179 			return -EPERM;
180 
181 		if (__tcf_action_put(p, bind))
182 			ret = ACT_P_DELETED;
183 	}
184 
185 	return ret;
186 }
187 
tcf_idr_release(struct tc_action * a,bool bind)188 int tcf_idr_release(struct tc_action *a, bool bind)
189 {
190 	const struct tc_action_ops *ops = a->ops;
191 	int ret;
192 
193 	ret = __tcf_idr_release(a, bind, false);
194 	if (ret == ACT_P_DELETED)
195 		module_put(ops->owner);
196 	return ret;
197 }
198 EXPORT_SYMBOL(tcf_idr_release);
199 
tcf_action_shared_attrs_size(const struct tc_action * act)200 static size_t tcf_action_shared_attrs_size(const struct tc_action *act)
201 {
202 	struct tc_cookie *act_cookie;
203 	u32 cookie_len = 0;
204 
205 	rcu_read_lock();
206 	act_cookie = rcu_dereference(act->act_cookie);
207 
208 	if (act_cookie)
209 		cookie_len = nla_total_size(act_cookie->len);
210 	rcu_read_unlock();
211 
212 	return  nla_total_size(0) /* action number nested */
213 		+ nla_total_size(IFNAMSIZ) /* TCA_ACT_KIND */
214 		+ cookie_len /* TCA_ACT_COOKIE */
215 		+ nla_total_size(sizeof(struct nla_bitfield32)) /* TCA_ACT_HW_STATS */
216 		+ nla_total_size(0) /* TCA_ACT_STATS nested */
217 		+ nla_total_size(sizeof(struct nla_bitfield32)) /* TCA_ACT_FLAGS */
218 		/* TCA_STATS_BASIC */
219 		+ nla_total_size_64bit(sizeof(struct gnet_stats_basic))
220 		/* TCA_STATS_PKT64 */
221 		+ nla_total_size_64bit(sizeof(u64))
222 		/* TCA_STATS_QUEUE */
223 		+ nla_total_size_64bit(sizeof(struct gnet_stats_queue))
224 		+ nla_total_size(0) /* TCA_OPTIONS nested */
225 		+ nla_total_size(sizeof(struct tcf_t)); /* TCA_GACT_TM */
226 }
227 
tcf_action_full_attrs_size(size_t sz)228 static size_t tcf_action_full_attrs_size(size_t sz)
229 {
230 	return NLMSG_HDRLEN                     /* struct nlmsghdr */
231 		+ sizeof(struct tcamsg)
232 		+ nla_total_size(0)             /* TCA_ACT_TAB nested */
233 		+ sz;
234 }
235 
tcf_action_fill_size(const struct tc_action * act)236 static size_t tcf_action_fill_size(const struct tc_action *act)
237 {
238 	size_t sz = tcf_action_shared_attrs_size(act);
239 
240 	if (act->ops->get_fill_size)
241 		return act->ops->get_fill_size(act) + sz;
242 	return sz;
243 }
244 
245 static int
tcf_action_dump_terse(struct sk_buff * skb,struct tc_action * a,bool from_act)246 tcf_action_dump_terse(struct sk_buff *skb, struct tc_action *a, bool from_act)
247 {
248 	unsigned char *b = skb_tail_pointer(skb);
249 	struct tc_cookie *cookie;
250 
251 	if (nla_put_string(skb, TCA_KIND, a->ops->kind))
252 		goto nla_put_failure;
253 	if (tcf_action_copy_stats(skb, a, 0))
254 		goto nla_put_failure;
255 	if (from_act && nla_put_u32(skb, TCA_ACT_INDEX, a->tcfa_index))
256 		goto nla_put_failure;
257 
258 	rcu_read_lock();
259 	cookie = rcu_dereference(a->act_cookie);
260 	if (cookie) {
261 		if (nla_put(skb, TCA_ACT_COOKIE, cookie->len, cookie->data)) {
262 			rcu_read_unlock();
263 			goto nla_put_failure;
264 		}
265 	}
266 	rcu_read_unlock();
267 
268 	return 0;
269 
270 nla_put_failure:
271 	nlmsg_trim(skb, b);
272 	return -1;
273 }
274 
tcf_dump_walker(struct tcf_idrinfo * idrinfo,struct sk_buff * skb,struct netlink_callback * cb)275 static int tcf_dump_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb,
276 			   struct netlink_callback *cb)
277 {
278 	int err = 0, index = -1, s_i = 0, n_i = 0;
279 	u32 act_flags = cb->args[2];
280 	unsigned long jiffy_since = cb->args[3];
281 	struct nlattr *nest;
282 	struct idr *idr = &idrinfo->action_idr;
283 	struct tc_action *p;
284 	unsigned long id = 1;
285 	unsigned long tmp;
286 
287 	mutex_lock(&idrinfo->lock);
288 
289 	s_i = cb->args[0];
290 
291 	idr_for_each_entry_ul(idr, p, tmp, id) {
292 		index++;
293 		if (index < s_i)
294 			continue;
295 		if (IS_ERR(p))
296 			continue;
297 
298 		if (jiffy_since &&
299 		    time_after(jiffy_since,
300 			       (unsigned long)p->tcfa_tm.lastuse))
301 			continue;
302 
303 		nest = nla_nest_start_noflag(skb, n_i);
304 		if (!nest) {
305 			index--;
306 			goto nla_put_failure;
307 		}
308 		err = (act_flags & TCA_ACT_FLAG_TERSE_DUMP) ?
309 			tcf_action_dump_terse(skb, p, true) :
310 			tcf_action_dump_1(skb, p, 0, 0);
311 		if (err < 0) {
312 			index--;
313 			nlmsg_trim(skb, nest);
314 			goto done;
315 		}
316 		nla_nest_end(skb, nest);
317 		n_i++;
318 		if (!(act_flags & TCA_ACT_FLAG_LARGE_DUMP_ON) &&
319 		    n_i >= TCA_ACT_MAX_PRIO)
320 			goto done;
321 	}
322 done:
323 	if (index >= 0)
324 		cb->args[0] = index + 1;
325 
326 	mutex_unlock(&idrinfo->lock);
327 	if (n_i) {
328 		if (act_flags & TCA_ACT_FLAG_LARGE_DUMP_ON)
329 			cb->args[1] = n_i;
330 	}
331 	return n_i;
332 
333 nla_put_failure:
334 	nla_nest_cancel(skb, nest);
335 	goto done;
336 }
337 
tcf_idr_release_unsafe(struct tc_action * p)338 static int tcf_idr_release_unsafe(struct tc_action *p)
339 {
340 	if (atomic_read(&p->tcfa_bindcnt) > 0)
341 		return -EPERM;
342 
343 	if (refcount_dec_and_test(&p->tcfa_refcnt)) {
344 		idr_remove(&p->idrinfo->action_idr, p->tcfa_index);
345 		tcf_action_cleanup(p);
346 		return ACT_P_DELETED;
347 	}
348 
349 	return 0;
350 }
351 
tcf_del_walker(struct tcf_idrinfo * idrinfo,struct sk_buff * skb,const struct tc_action_ops * ops,struct netlink_ext_ack * extack)352 static int tcf_del_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb,
353 			  const struct tc_action_ops *ops,
354 			  struct netlink_ext_ack *extack)
355 {
356 	struct nlattr *nest;
357 	int n_i = 0;
358 	int ret = -EINVAL;
359 	struct idr *idr = &idrinfo->action_idr;
360 	struct tc_action *p;
361 	unsigned long id = 1;
362 	unsigned long tmp;
363 
364 	nest = nla_nest_start_noflag(skb, 0);
365 	if (nest == NULL)
366 		goto nla_put_failure;
367 	if (nla_put_string(skb, TCA_KIND, ops->kind))
368 		goto nla_put_failure;
369 
370 	ret = 0;
371 	mutex_lock(&idrinfo->lock);
372 	idr_for_each_entry_ul(idr, p, tmp, id) {
373 		if (IS_ERR(p))
374 			continue;
375 		ret = tcf_idr_release_unsafe(p);
376 		if (ret == ACT_P_DELETED)
377 			module_put(ops->owner);
378 		else if (ret < 0)
379 			break;
380 		n_i++;
381 	}
382 	mutex_unlock(&idrinfo->lock);
383 	if (ret < 0) {
384 		if (n_i)
385 			NL_SET_ERR_MSG(extack, "Unable to flush all TC actions");
386 		else
387 			goto nla_put_failure;
388 	}
389 
390 	ret = nla_put_u32(skb, TCA_FCNT, n_i);
391 	if (ret)
392 		goto nla_put_failure;
393 	nla_nest_end(skb, nest);
394 
395 	return n_i;
396 nla_put_failure:
397 	nla_nest_cancel(skb, nest);
398 	return ret;
399 }
400 
tcf_generic_walker(struct tc_action_net * tn,struct sk_buff * skb,struct netlink_callback * cb,int type,const struct tc_action_ops * ops,struct netlink_ext_ack * extack)401 int tcf_generic_walker(struct tc_action_net *tn, struct sk_buff *skb,
402 		       struct netlink_callback *cb, int type,
403 		       const struct tc_action_ops *ops,
404 		       struct netlink_ext_ack *extack)
405 {
406 	struct tcf_idrinfo *idrinfo = tn->idrinfo;
407 
408 	if (type == RTM_DELACTION) {
409 		return tcf_del_walker(idrinfo, skb, ops, extack);
410 	} else if (type == RTM_GETACTION) {
411 		return tcf_dump_walker(idrinfo, skb, cb);
412 	} else {
413 		WARN(1, "tcf_generic_walker: unknown command %d\n", type);
414 		NL_SET_ERR_MSG(extack, "tcf_generic_walker: unknown command");
415 		return -EINVAL;
416 	}
417 }
418 EXPORT_SYMBOL(tcf_generic_walker);
419 
tcf_idr_search(struct tc_action_net * tn,struct tc_action ** a,u32 index)420 int tcf_idr_search(struct tc_action_net *tn, struct tc_action **a, u32 index)
421 {
422 	struct tcf_idrinfo *idrinfo = tn->idrinfo;
423 	struct tc_action *p;
424 
425 	mutex_lock(&idrinfo->lock);
426 	p = idr_find(&idrinfo->action_idr, index);
427 	if (IS_ERR(p))
428 		p = NULL;
429 	else if (p)
430 		refcount_inc(&p->tcfa_refcnt);
431 	mutex_unlock(&idrinfo->lock);
432 
433 	if (p) {
434 		*a = p;
435 		return true;
436 	}
437 	return false;
438 }
439 EXPORT_SYMBOL(tcf_idr_search);
440 
tcf_idr_delete_index(struct tcf_idrinfo * idrinfo,u32 index)441 static int tcf_idr_delete_index(struct tcf_idrinfo *idrinfo, u32 index)
442 {
443 	struct tc_action *p;
444 	int ret = 0;
445 
446 	mutex_lock(&idrinfo->lock);
447 	p = idr_find(&idrinfo->action_idr, index);
448 	if (!p) {
449 		mutex_unlock(&idrinfo->lock);
450 		return -ENOENT;
451 	}
452 
453 	if (!atomic_read(&p->tcfa_bindcnt)) {
454 		if (refcount_dec_and_test(&p->tcfa_refcnt)) {
455 			struct module *owner = p->ops->owner;
456 
457 			WARN_ON(p != idr_remove(&idrinfo->action_idr,
458 						p->tcfa_index));
459 			mutex_unlock(&idrinfo->lock);
460 
461 			tcf_action_cleanup(p);
462 			module_put(owner);
463 			return 0;
464 		}
465 		ret = 0;
466 	} else {
467 		ret = -EPERM;
468 	}
469 
470 	mutex_unlock(&idrinfo->lock);
471 	return ret;
472 }
473 
tcf_idr_create(struct tc_action_net * tn,u32 index,struct nlattr * est,struct tc_action ** a,const struct tc_action_ops * ops,int bind,bool cpustats,u32 flags)474 int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est,
475 		   struct tc_action **a, const struct tc_action_ops *ops,
476 		   int bind, bool cpustats, u32 flags)
477 {
478 	struct tc_action *p = kzalloc(ops->size, GFP_KERNEL);
479 	struct tcf_idrinfo *idrinfo = tn->idrinfo;
480 	int err = -ENOMEM;
481 
482 	if (unlikely(!p))
483 		return -ENOMEM;
484 	refcount_set(&p->tcfa_refcnt, 1);
485 	if (bind)
486 		atomic_set(&p->tcfa_bindcnt, 1);
487 
488 	if (cpustats) {
489 		p->cpu_bstats = netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu);
490 		if (!p->cpu_bstats)
491 			goto err1;
492 		p->cpu_bstats_hw = netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu);
493 		if (!p->cpu_bstats_hw)
494 			goto err2;
495 		p->cpu_qstats = alloc_percpu(struct gnet_stats_queue);
496 		if (!p->cpu_qstats)
497 			goto err3;
498 	}
499 	spin_lock_init(&p->tcfa_lock);
500 	p->tcfa_index = index;
501 	p->tcfa_tm.install = jiffies;
502 	p->tcfa_tm.lastuse = jiffies;
503 	p->tcfa_tm.firstuse = 0;
504 	p->tcfa_flags = flags & TCA_ACT_FLAGS_USER_MASK;
505 	if (est) {
506 		err = gen_new_estimator(&p->tcfa_bstats, p->cpu_bstats,
507 					&p->tcfa_rate_est,
508 					&p->tcfa_lock, NULL, est);
509 		if (err)
510 			goto err4;
511 	}
512 
513 	p->idrinfo = idrinfo;
514 	__module_get(ops->owner);
515 	p->ops = ops;
516 	*a = p;
517 	return 0;
518 err4:
519 	free_percpu(p->cpu_qstats);
520 err3:
521 	free_percpu(p->cpu_bstats_hw);
522 err2:
523 	free_percpu(p->cpu_bstats);
524 err1:
525 	kfree(p);
526 	return err;
527 }
528 EXPORT_SYMBOL(tcf_idr_create);
529 
tcf_idr_create_from_flags(struct tc_action_net * tn,u32 index,struct nlattr * est,struct tc_action ** a,const struct tc_action_ops * ops,int bind,u32 flags)530 int tcf_idr_create_from_flags(struct tc_action_net *tn, u32 index,
531 			      struct nlattr *est, struct tc_action **a,
532 			      const struct tc_action_ops *ops, int bind,
533 			      u32 flags)
534 {
535 	/* Set cpustats according to actions flags. */
536 	return tcf_idr_create(tn, index, est, a, ops, bind,
537 			      !(flags & TCA_ACT_FLAGS_NO_PERCPU_STATS), flags);
538 }
539 EXPORT_SYMBOL(tcf_idr_create_from_flags);
540 
541 /* Cleanup idr index that was allocated but not initialized. */
542 
tcf_idr_cleanup(struct tc_action_net * tn,u32 index)543 void tcf_idr_cleanup(struct tc_action_net *tn, u32 index)
544 {
545 	struct tcf_idrinfo *idrinfo = tn->idrinfo;
546 
547 	mutex_lock(&idrinfo->lock);
548 	/* Remove ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc */
549 	WARN_ON(!IS_ERR(idr_remove(&idrinfo->action_idr, index)));
550 	mutex_unlock(&idrinfo->lock);
551 }
552 EXPORT_SYMBOL(tcf_idr_cleanup);
553 
554 /* Check if action with specified index exists. If actions is found, increments
555  * its reference and bind counters, and return 1. Otherwise insert temporary
556  * error pointer (to prevent concurrent users from inserting actions with same
557  * index) and return 0.
558  */
559 
tcf_idr_check_alloc(struct tc_action_net * tn,u32 * index,struct tc_action ** a,int bind)560 int tcf_idr_check_alloc(struct tc_action_net *tn, u32 *index,
561 			struct tc_action **a, int bind)
562 {
563 	struct tcf_idrinfo *idrinfo = tn->idrinfo;
564 	struct tc_action *p;
565 	int ret;
566 
567 again:
568 	mutex_lock(&idrinfo->lock);
569 	if (*index) {
570 		p = idr_find(&idrinfo->action_idr, *index);
571 		if (IS_ERR(p)) {
572 			/* This means that another process allocated
573 			 * index but did not assign the pointer yet.
574 			 */
575 			mutex_unlock(&idrinfo->lock);
576 			goto again;
577 		}
578 
579 		if (p) {
580 			refcount_inc(&p->tcfa_refcnt);
581 			if (bind)
582 				atomic_inc(&p->tcfa_bindcnt);
583 			*a = p;
584 			ret = 1;
585 		} else {
586 			*a = NULL;
587 			ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index,
588 					    *index, GFP_KERNEL);
589 			if (!ret)
590 				idr_replace(&idrinfo->action_idr,
591 					    ERR_PTR(-EBUSY), *index);
592 		}
593 	} else {
594 		*index = 1;
595 		*a = NULL;
596 		ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index,
597 				    UINT_MAX, GFP_KERNEL);
598 		if (!ret)
599 			idr_replace(&idrinfo->action_idr, ERR_PTR(-EBUSY),
600 				    *index);
601 	}
602 	mutex_unlock(&idrinfo->lock);
603 	return ret;
604 }
605 EXPORT_SYMBOL(tcf_idr_check_alloc);
606 
tcf_idrinfo_destroy(const struct tc_action_ops * ops,struct tcf_idrinfo * idrinfo)607 void tcf_idrinfo_destroy(const struct tc_action_ops *ops,
608 			 struct tcf_idrinfo *idrinfo)
609 {
610 	struct idr *idr = &idrinfo->action_idr;
611 	struct tc_action *p;
612 	int ret;
613 	unsigned long id = 1;
614 	unsigned long tmp;
615 
616 	idr_for_each_entry_ul(idr, p, tmp, id) {
617 		ret = __tcf_idr_release(p, false, true);
618 		if (ret == ACT_P_DELETED)
619 			module_put(ops->owner);
620 		else if (ret < 0)
621 			return;
622 	}
623 	idr_destroy(&idrinfo->action_idr);
624 }
625 EXPORT_SYMBOL(tcf_idrinfo_destroy);
626 
627 static LIST_HEAD(act_base);
628 static DEFINE_RWLOCK(act_mod_lock);
629 
tcf_register_action(struct tc_action_ops * act,struct pernet_operations * ops)630 int tcf_register_action(struct tc_action_ops *act,
631 			struct pernet_operations *ops)
632 {
633 	struct tc_action_ops *a;
634 	int ret;
635 
636 	if (!act->act || !act->dump || !act->init || !act->walk || !act->lookup)
637 		return -EINVAL;
638 
639 	/* We have to register pernet ops before making the action ops visible,
640 	 * otherwise tcf_action_init_1() could get a partially initialized
641 	 * netns.
642 	 */
643 	ret = register_pernet_subsys(ops);
644 	if (ret)
645 		return ret;
646 
647 	write_lock(&act_mod_lock);
648 	list_for_each_entry(a, &act_base, head) {
649 		if (act->id == a->id || (strcmp(act->kind, a->kind) == 0)) {
650 			write_unlock(&act_mod_lock);
651 			unregister_pernet_subsys(ops);
652 			return -EEXIST;
653 		}
654 	}
655 	list_add_tail(&act->head, &act_base);
656 	write_unlock(&act_mod_lock);
657 
658 	return 0;
659 }
660 EXPORT_SYMBOL(tcf_register_action);
661 
tcf_unregister_action(struct tc_action_ops * act,struct pernet_operations * ops)662 int tcf_unregister_action(struct tc_action_ops *act,
663 			  struct pernet_operations *ops)
664 {
665 	struct tc_action_ops *a;
666 	int err = -ENOENT;
667 
668 	write_lock(&act_mod_lock);
669 	list_for_each_entry(a, &act_base, head) {
670 		if (a == act) {
671 			list_del(&act->head);
672 			err = 0;
673 			break;
674 		}
675 	}
676 	write_unlock(&act_mod_lock);
677 	if (!err)
678 		unregister_pernet_subsys(ops);
679 	return err;
680 }
681 EXPORT_SYMBOL(tcf_unregister_action);
682 
683 /* lookup by name */
tc_lookup_action_n(char * kind)684 static struct tc_action_ops *tc_lookup_action_n(char *kind)
685 {
686 	struct tc_action_ops *a, *res = NULL;
687 
688 	if (kind) {
689 		read_lock(&act_mod_lock);
690 		list_for_each_entry(a, &act_base, head) {
691 			if (strcmp(kind, a->kind) == 0) {
692 				if (try_module_get(a->owner))
693 					res = a;
694 				break;
695 			}
696 		}
697 		read_unlock(&act_mod_lock);
698 	}
699 	return res;
700 }
701 
702 /* lookup by nlattr */
tc_lookup_action(struct nlattr * kind)703 static struct tc_action_ops *tc_lookup_action(struct nlattr *kind)
704 {
705 	struct tc_action_ops *a, *res = NULL;
706 
707 	if (kind) {
708 		read_lock(&act_mod_lock);
709 		list_for_each_entry(a, &act_base, head) {
710 			if (nla_strcmp(kind, a->kind) == 0) {
711 				if (try_module_get(a->owner))
712 					res = a;
713 				break;
714 			}
715 		}
716 		read_unlock(&act_mod_lock);
717 	}
718 	return res;
719 }
720 
721 /*TCA_ACT_MAX_PRIO is 32, there count up to 32 */
722 #define TCA_ACT_MAX_PRIO_MASK 0x1FF
tcf_action_exec(struct sk_buff * skb,struct tc_action ** actions,int nr_actions,struct tcf_result * res)723 int tcf_action_exec(struct sk_buff *skb, struct tc_action **actions,
724 		    int nr_actions, struct tcf_result *res)
725 {
726 	u32 jmp_prgcnt = 0;
727 	u32 jmp_ttl = TCA_ACT_MAX_PRIO; /*matches actions per filter */
728 	int i;
729 	int ret = TC_ACT_OK;
730 
731 	if (skb_skip_tc_classify(skb))
732 		return TC_ACT_OK;
733 
734 restart_act_graph:
735 	for (i = 0; i < nr_actions; i++) {
736 		const struct tc_action *a = actions[i];
737 		int repeat_ttl;
738 
739 		if (jmp_prgcnt > 0) {
740 			jmp_prgcnt -= 1;
741 			continue;
742 		}
743 
744 		repeat_ttl = 32;
745 repeat:
746 		ret = a->ops->act(skb, a, res);
747 
748 		if (unlikely(ret == TC_ACT_REPEAT)) {
749 			if (--repeat_ttl != 0)
750 				goto repeat;
751 			/* suspicious opcode, stop pipeline */
752 			net_warn_ratelimited("TC_ACT_REPEAT abuse ?\n");
753 			return TC_ACT_OK;
754 		}
755 
756 		if (TC_ACT_EXT_CMP(ret, TC_ACT_JUMP)) {
757 			jmp_prgcnt = ret & TCA_ACT_MAX_PRIO_MASK;
758 			if (!jmp_prgcnt || (jmp_prgcnt > nr_actions)) {
759 				/* faulty opcode, stop pipeline */
760 				return TC_ACT_OK;
761 			} else {
762 				jmp_ttl -= 1;
763 				if (jmp_ttl > 0)
764 					goto restart_act_graph;
765 				else /* faulty graph, stop pipeline */
766 					return TC_ACT_OK;
767 			}
768 		} else if (TC_ACT_EXT_CMP(ret, TC_ACT_GOTO_CHAIN)) {
769 			if (unlikely(!rcu_access_pointer(a->goto_chain))) {
770 				net_warn_ratelimited("can't go to NULL chain!\n");
771 				return TC_ACT_SHOT;
772 			}
773 			tcf_action_goto_chain_exec(a, res);
774 		}
775 
776 		if (ret != TC_ACT_PIPE)
777 			break;
778 	}
779 
780 	return ret;
781 }
782 EXPORT_SYMBOL(tcf_action_exec);
783 
tcf_action_destroy(struct tc_action * actions[],int bind)784 int tcf_action_destroy(struct tc_action *actions[], int bind)
785 {
786 	const struct tc_action_ops *ops;
787 	struct tc_action *a;
788 	int ret = 0, i;
789 
790 	for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) {
791 		a = actions[i];
792 		actions[i] = NULL;
793 		ops = a->ops;
794 		ret = __tcf_idr_release(a, bind, true);
795 		if (ret == ACT_P_DELETED)
796 			module_put(ops->owner);
797 		else if (ret < 0)
798 			return ret;
799 	}
800 	return ret;
801 }
802 
tcf_action_put(struct tc_action * p)803 static int tcf_action_put(struct tc_action *p)
804 {
805 	return __tcf_action_put(p, false);
806 }
807 
808 /* Put all actions in this array, skip those NULL's. */
tcf_action_put_many(struct tc_action * actions[])809 static void tcf_action_put_many(struct tc_action *actions[])
810 {
811 	int i;
812 
813 	for (i = 0; i < TCA_ACT_MAX_PRIO; i++) {
814 		struct tc_action *a = actions[i];
815 		const struct tc_action_ops *ops;
816 
817 		if (!a)
818 			continue;
819 		ops = a->ops;
820 		if (tcf_action_put(a))
821 			module_put(ops->owner);
822 	}
823 }
824 
825 int
tcf_action_dump_old(struct sk_buff * skb,struct tc_action * a,int bind,int ref)826 tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
827 {
828 	return a->ops->dump(skb, a, bind, ref);
829 }
830 
831 int
tcf_action_dump_1(struct sk_buff * skb,struct tc_action * a,int bind,int ref)832 tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
833 {
834 	int err = -EINVAL;
835 	unsigned char *b = skb_tail_pointer(skb);
836 	struct nlattr *nest;
837 
838 	if (tcf_action_dump_terse(skb, a, false))
839 		goto nla_put_failure;
840 
841 	if (a->hw_stats != TCA_ACT_HW_STATS_ANY &&
842 	    nla_put_bitfield32(skb, TCA_ACT_HW_STATS,
843 			       a->hw_stats, TCA_ACT_HW_STATS_ANY))
844 		goto nla_put_failure;
845 
846 	if (a->used_hw_stats_valid &&
847 	    nla_put_bitfield32(skb, TCA_ACT_USED_HW_STATS,
848 			       a->used_hw_stats, TCA_ACT_HW_STATS_ANY))
849 		goto nla_put_failure;
850 
851 	if (a->tcfa_flags &&
852 	    nla_put_bitfield32(skb, TCA_ACT_FLAGS,
853 			       a->tcfa_flags, a->tcfa_flags))
854 		goto nla_put_failure;
855 
856 	nest = nla_nest_start_noflag(skb, TCA_OPTIONS);
857 	if (nest == NULL)
858 		goto nla_put_failure;
859 	err = tcf_action_dump_old(skb, a, bind, ref);
860 	if (err > 0) {
861 		nla_nest_end(skb, nest);
862 		return err;
863 	}
864 
865 nla_put_failure:
866 	nlmsg_trim(skb, b);
867 	return -1;
868 }
869 EXPORT_SYMBOL(tcf_action_dump_1);
870 
tcf_action_dump(struct sk_buff * skb,struct tc_action * actions[],int bind,int ref,bool terse)871 int tcf_action_dump(struct sk_buff *skb, struct tc_action *actions[],
872 		    int bind, int ref, bool terse)
873 {
874 	struct tc_action *a;
875 	int err = -EINVAL, i;
876 	struct nlattr *nest;
877 
878 	for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) {
879 		a = actions[i];
880 		nest = nla_nest_start_noflag(skb, i + 1);
881 		if (nest == NULL)
882 			goto nla_put_failure;
883 		err = terse ? tcf_action_dump_terse(skb, a, false) :
884 			tcf_action_dump_1(skb, a, bind, ref);
885 		if (err < 0)
886 			goto errout;
887 		nla_nest_end(skb, nest);
888 	}
889 
890 	return 0;
891 
892 nla_put_failure:
893 	err = -EINVAL;
894 errout:
895 	nla_nest_cancel(skb, nest);
896 	return err;
897 }
898 
nla_memdup_cookie(struct nlattr ** tb)899 static struct tc_cookie *nla_memdup_cookie(struct nlattr **tb)
900 {
901 	struct tc_cookie *c = kzalloc(sizeof(*c), GFP_KERNEL);
902 	if (!c)
903 		return NULL;
904 
905 	c->data = nla_memdup(tb[TCA_ACT_COOKIE], GFP_KERNEL);
906 	if (!c->data) {
907 		kfree(c);
908 		return NULL;
909 	}
910 	c->len = nla_len(tb[TCA_ACT_COOKIE]);
911 
912 	return c;
913 }
914 
tcf_action_hw_stats_get(struct nlattr * hw_stats_attr)915 static u8 tcf_action_hw_stats_get(struct nlattr *hw_stats_attr)
916 {
917 	struct nla_bitfield32 hw_stats_bf;
918 
919 	/* If the user did not pass the attr, that means he does
920 	 * not care about the type. Return "any" in that case
921 	 * which is setting on all supported types.
922 	 */
923 	if (!hw_stats_attr)
924 		return TCA_ACT_HW_STATS_ANY;
925 	hw_stats_bf = nla_get_bitfield32(hw_stats_attr);
926 	return hw_stats_bf.value;
927 }
928 
929 static const struct nla_policy tcf_action_policy[TCA_ACT_MAX + 1] = {
930 	[TCA_ACT_KIND]		= { .type = NLA_STRING },
931 	[TCA_ACT_INDEX]		= { .type = NLA_U32 },
932 	[TCA_ACT_COOKIE]	= { .type = NLA_BINARY,
933 				    .len = TC_COOKIE_MAX_SIZE },
934 	[TCA_ACT_OPTIONS]	= { .type = NLA_NESTED },
935 	[TCA_ACT_FLAGS]		= NLA_POLICY_BITFIELD32(TCA_ACT_FLAGS_NO_PERCPU_STATS),
936 	[TCA_ACT_HW_STATS]	= NLA_POLICY_BITFIELD32(TCA_ACT_HW_STATS_ANY),
937 };
938 
tcf_idr_insert_many(struct tc_action * actions[])939 void tcf_idr_insert_many(struct tc_action *actions[])
940 {
941 	int i;
942 
943 	for (i = 0; i < TCA_ACT_MAX_PRIO; i++) {
944 		struct tc_action *a = actions[i];
945 		struct tcf_idrinfo *idrinfo;
946 
947 		if (!a)
948 			continue;
949 		idrinfo = a->idrinfo;
950 		mutex_lock(&idrinfo->lock);
951 		/* Replace ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc if
952 		 * it is just created, otherwise this is just a nop.
953 		 */
954 		idr_replace(&idrinfo->action_idr, a, a->tcfa_index);
955 		mutex_unlock(&idrinfo->lock);
956 	}
957 }
958 
tc_action_load_ops(struct nlattr * nla,bool police,bool rtnl_held,struct netlink_ext_ack * extack)959 struct tc_action_ops *tc_action_load_ops(struct nlattr *nla, bool police,
960 					 bool rtnl_held,
961 					 struct netlink_ext_ack *extack)
962 {
963 	struct nlattr *tb[TCA_ACT_MAX + 1];
964 	struct tc_action_ops *a_o;
965 	char act_name[IFNAMSIZ];
966 	struct nlattr *kind;
967 	int err;
968 
969 	if (!police) {
970 		err = nla_parse_nested_deprecated(tb, TCA_ACT_MAX, nla,
971 						  tcf_action_policy, extack);
972 		if (err < 0)
973 			return ERR_PTR(err);
974 		err = -EINVAL;
975 		kind = tb[TCA_ACT_KIND];
976 		if (!kind) {
977 			NL_SET_ERR_MSG(extack, "TC action kind must be specified");
978 			return ERR_PTR(err);
979 		}
980 		if (nla_strscpy(act_name, kind, IFNAMSIZ) < 0) {
981 			NL_SET_ERR_MSG(extack, "TC action name too long");
982 			return ERR_PTR(err);
983 		}
984 	} else {
985 		if (strlcpy(act_name, "police", IFNAMSIZ) >= IFNAMSIZ) {
986 			NL_SET_ERR_MSG(extack, "TC action name too long");
987 			return ERR_PTR(-EINVAL);
988 		}
989 	}
990 
991 	a_o = tc_lookup_action_n(act_name);
992 	if (a_o == NULL) {
993 #ifdef CONFIG_MODULES
994 		if (rtnl_held)
995 			rtnl_unlock();
996 		request_module("act_%s", act_name);
997 		if (rtnl_held)
998 			rtnl_lock();
999 
1000 		a_o = tc_lookup_action_n(act_name);
1001 
1002 		/* We dropped the RTNL semaphore in order to
1003 		 * perform the module load.  So, even if we
1004 		 * succeeded in loading the module we have to
1005 		 * tell the caller to replay the request.  We
1006 		 * indicate this using -EAGAIN.
1007 		 */
1008 		if (a_o != NULL) {
1009 			module_put(a_o->owner);
1010 			return ERR_PTR(-EAGAIN);
1011 		}
1012 #endif
1013 		NL_SET_ERR_MSG(extack, "Failed to load TC action module");
1014 		return ERR_PTR(-ENOENT);
1015 	}
1016 
1017 	return a_o;
1018 }
1019 
tcf_action_init_1(struct net * net,struct tcf_proto * tp,struct nlattr * nla,struct nlattr * est,struct tc_action_ops * a_o,int * init_res,u32 flags,struct netlink_ext_ack * extack)1020 struct tc_action *tcf_action_init_1(struct net *net, struct tcf_proto *tp,
1021 				    struct nlattr *nla, struct nlattr *est,
1022 				    struct tc_action_ops *a_o, int *init_res,
1023 				    u32 flags, struct netlink_ext_ack *extack)
1024 {
1025 	bool police = flags & TCA_ACT_FLAGS_POLICE;
1026 	struct nla_bitfield32 userflags = { 0, 0 };
1027 	u8 hw_stats = TCA_ACT_HW_STATS_ANY;
1028 	struct nlattr *tb[TCA_ACT_MAX + 1];
1029 	struct tc_cookie *cookie = NULL;
1030 	struct tc_action *a;
1031 	int err;
1032 
1033 	/* backward compatibility for policer */
1034 	if (!police) {
1035 		err = nla_parse_nested_deprecated(tb, TCA_ACT_MAX, nla,
1036 						  tcf_action_policy, extack);
1037 		if (err < 0)
1038 			return ERR_PTR(err);
1039 		if (tb[TCA_ACT_COOKIE]) {
1040 			cookie = nla_memdup_cookie(tb);
1041 			if (!cookie) {
1042 				NL_SET_ERR_MSG(extack, "No memory to generate TC cookie");
1043 				err = -ENOMEM;
1044 				goto err_out;
1045 			}
1046 		}
1047 		hw_stats = tcf_action_hw_stats_get(tb[TCA_ACT_HW_STATS]);
1048 		if (tb[TCA_ACT_FLAGS])
1049 			userflags = nla_get_bitfield32(tb[TCA_ACT_FLAGS]);
1050 
1051 		err = a_o->init(net, tb[TCA_ACT_OPTIONS], est, &a, tp,
1052 				userflags.value | flags, extack);
1053 	} else {
1054 		err = a_o->init(net, nla, est, &a, tp, userflags.value | flags,
1055 				extack);
1056 	}
1057 	if (err < 0)
1058 		goto err_out;
1059 	*init_res = err;
1060 
1061 	if (!police && tb[TCA_ACT_COOKIE])
1062 		tcf_set_action_cookie(&a->act_cookie, cookie);
1063 
1064 	if (!police)
1065 		a->hw_stats = hw_stats;
1066 
1067 	return a;
1068 
1069 err_out:
1070 	if (cookie) {
1071 		kfree(cookie->data);
1072 		kfree(cookie);
1073 	}
1074 	return ERR_PTR(err);
1075 }
1076 
1077 /* Returns numbers of initialized actions or negative error. */
1078 
tcf_action_init(struct net * net,struct tcf_proto * tp,struct nlattr * nla,struct nlattr * est,struct tc_action * actions[],int init_res[],size_t * attr_size,u32 flags,struct netlink_ext_ack * extack)1079 int tcf_action_init(struct net *net, struct tcf_proto *tp, struct nlattr *nla,
1080 		    struct nlattr *est, struct tc_action *actions[],
1081 		    int init_res[], size_t *attr_size, u32 flags,
1082 		    struct netlink_ext_ack *extack)
1083 {
1084 	struct tc_action_ops *ops[TCA_ACT_MAX_PRIO] = {};
1085 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1086 	struct tc_action *act;
1087 	size_t sz = 0;
1088 	int err;
1089 	int i;
1090 
1091 	err = nla_parse_nested_deprecated(tb, TCA_ACT_MAX_PRIO, nla, NULL,
1092 					  extack);
1093 	if (err < 0)
1094 		return err;
1095 
1096 	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
1097 		struct tc_action_ops *a_o;
1098 
1099 		a_o = tc_action_load_ops(tb[i], flags & TCA_ACT_FLAGS_POLICE,
1100 					 !(flags & TCA_ACT_FLAGS_NO_RTNL),
1101 					 extack);
1102 		if (IS_ERR(a_o)) {
1103 			err = PTR_ERR(a_o);
1104 			goto err_mod;
1105 		}
1106 		ops[i - 1] = a_o;
1107 	}
1108 
1109 	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
1110 		act = tcf_action_init_1(net, tp, tb[i], est, ops[i - 1],
1111 					&init_res[i - 1], flags, extack);
1112 		if (IS_ERR(act)) {
1113 			err = PTR_ERR(act);
1114 			goto err;
1115 		}
1116 		sz += tcf_action_fill_size(act);
1117 		/* Start from index 0 */
1118 		actions[i - 1] = act;
1119 	}
1120 
1121 	/* We have to commit them all together, because if any error happened in
1122 	 * between, we could not handle the failure gracefully.
1123 	 */
1124 	tcf_idr_insert_many(actions);
1125 
1126 	*attr_size = tcf_action_full_attrs_size(sz);
1127 	err = i - 1;
1128 	goto err_mod;
1129 
1130 err:
1131 	tcf_action_destroy(actions, flags & TCA_ACT_FLAGS_BIND);
1132 err_mod:
1133 	for (i = 0; i < TCA_ACT_MAX_PRIO; i++) {
1134 		if (ops[i])
1135 			module_put(ops[i]->owner);
1136 	}
1137 	return err;
1138 }
1139 
tcf_action_update_stats(struct tc_action * a,u64 bytes,u64 packets,u64 drops,bool hw)1140 void tcf_action_update_stats(struct tc_action *a, u64 bytes, u64 packets,
1141 			     u64 drops, bool hw)
1142 {
1143 	if (a->cpu_bstats) {
1144 		_bstats_cpu_update(this_cpu_ptr(a->cpu_bstats), bytes, packets);
1145 
1146 		this_cpu_ptr(a->cpu_qstats)->drops += drops;
1147 
1148 		if (hw)
1149 			_bstats_cpu_update(this_cpu_ptr(a->cpu_bstats_hw),
1150 					   bytes, packets);
1151 		return;
1152 	}
1153 
1154 	_bstats_update(&a->tcfa_bstats, bytes, packets);
1155 	a->tcfa_qstats.drops += drops;
1156 	if (hw)
1157 		_bstats_update(&a->tcfa_bstats_hw, bytes, packets);
1158 }
1159 EXPORT_SYMBOL(tcf_action_update_stats);
1160 
tcf_action_copy_stats(struct sk_buff * skb,struct tc_action * p,int compat_mode)1161 int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *p,
1162 			  int compat_mode)
1163 {
1164 	int err = 0;
1165 	struct gnet_dump d;
1166 
1167 	if (p == NULL)
1168 		goto errout;
1169 
1170 	/* compat_mode being true specifies a call that is supposed
1171 	 * to add additional backward compatibility statistic TLVs.
1172 	 */
1173 	if (compat_mode) {
1174 		if (p->type == TCA_OLD_COMPAT)
1175 			err = gnet_stats_start_copy_compat(skb, 0,
1176 							   TCA_STATS,
1177 							   TCA_XSTATS,
1178 							   &p->tcfa_lock, &d,
1179 							   TCA_PAD);
1180 		else
1181 			return 0;
1182 	} else
1183 		err = gnet_stats_start_copy(skb, TCA_ACT_STATS,
1184 					    &p->tcfa_lock, &d, TCA_ACT_PAD);
1185 
1186 	if (err < 0)
1187 		goto errout;
1188 
1189 	if (gnet_stats_copy_basic(NULL, &d, p->cpu_bstats, &p->tcfa_bstats) < 0 ||
1190 	    gnet_stats_copy_basic_hw(NULL, &d, p->cpu_bstats_hw,
1191 				     &p->tcfa_bstats_hw) < 0 ||
1192 	    gnet_stats_copy_rate_est(&d, &p->tcfa_rate_est) < 0 ||
1193 	    gnet_stats_copy_queue(&d, p->cpu_qstats,
1194 				  &p->tcfa_qstats,
1195 				  p->tcfa_qstats.qlen) < 0)
1196 		goto errout;
1197 
1198 	if (gnet_stats_finish_copy(&d) < 0)
1199 		goto errout;
1200 
1201 	return 0;
1202 
1203 errout:
1204 	return -1;
1205 }
1206 
tca_get_fill(struct sk_buff * skb,struct tc_action * actions[],u32 portid,u32 seq,u16 flags,int event,int bind,int ref)1207 static int tca_get_fill(struct sk_buff *skb, struct tc_action *actions[],
1208 			u32 portid, u32 seq, u16 flags, int event, int bind,
1209 			int ref)
1210 {
1211 	struct tcamsg *t;
1212 	struct nlmsghdr *nlh;
1213 	unsigned char *b = skb_tail_pointer(skb);
1214 	struct nlattr *nest;
1215 
1216 	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags);
1217 	if (!nlh)
1218 		goto out_nlmsg_trim;
1219 	t = nlmsg_data(nlh);
1220 	t->tca_family = AF_UNSPEC;
1221 	t->tca__pad1 = 0;
1222 	t->tca__pad2 = 0;
1223 
1224 	nest = nla_nest_start_noflag(skb, TCA_ACT_TAB);
1225 	if (!nest)
1226 		goto out_nlmsg_trim;
1227 
1228 	if (tcf_action_dump(skb, actions, bind, ref, false) < 0)
1229 		goto out_nlmsg_trim;
1230 
1231 	nla_nest_end(skb, nest);
1232 
1233 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1234 	return skb->len;
1235 
1236 out_nlmsg_trim:
1237 	nlmsg_trim(skb, b);
1238 	return -1;
1239 }
1240 
1241 static int
tcf_get_notify(struct net * net,u32 portid,struct nlmsghdr * n,struct tc_action * actions[],int event,struct netlink_ext_ack * extack)1242 tcf_get_notify(struct net *net, u32 portid, struct nlmsghdr *n,
1243 	       struct tc_action *actions[], int event,
1244 	       struct netlink_ext_ack *extack)
1245 {
1246 	struct sk_buff *skb;
1247 
1248 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1249 	if (!skb)
1250 		return -ENOBUFS;
1251 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, event,
1252 			 0, 1) <= 0) {
1253 		NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action");
1254 		kfree_skb(skb);
1255 		return -EINVAL;
1256 	}
1257 
1258 	return rtnl_unicast(skb, net, portid);
1259 }
1260 
tcf_action_get_1(struct net * net,struct nlattr * nla,struct nlmsghdr * n,u32 portid,struct netlink_ext_ack * extack)1261 static struct tc_action *tcf_action_get_1(struct net *net, struct nlattr *nla,
1262 					  struct nlmsghdr *n, u32 portid,
1263 					  struct netlink_ext_ack *extack)
1264 {
1265 	struct nlattr *tb[TCA_ACT_MAX + 1];
1266 	const struct tc_action_ops *ops;
1267 	struct tc_action *a;
1268 	int index;
1269 	int err;
1270 
1271 	err = nla_parse_nested_deprecated(tb, TCA_ACT_MAX, nla,
1272 					  tcf_action_policy, extack);
1273 	if (err < 0)
1274 		goto err_out;
1275 
1276 	err = -EINVAL;
1277 	if (tb[TCA_ACT_INDEX] == NULL ||
1278 	    nla_len(tb[TCA_ACT_INDEX]) < sizeof(index)) {
1279 		NL_SET_ERR_MSG(extack, "Invalid TC action index value");
1280 		goto err_out;
1281 	}
1282 	index = nla_get_u32(tb[TCA_ACT_INDEX]);
1283 
1284 	err = -EINVAL;
1285 	ops = tc_lookup_action(tb[TCA_ACT_KIND]);
1286 	if (!ops) { /* could happen in batch of actions */
1287 		NL_SET_ERR_MSG(extack, "Specified TC action kind not found");
1288 		goto err_out;
1289 	}
1290 	err = -ENOENT;
1291 	if (ops->lookup(net, &a, index) == 0) {
1292 		NL_SET_ERR_MSG(extack, "TC action with specified index not found");
1293 		goto err_mod;
1294 	}
1295 
1296 	module_put(ops->owner);
1297 	return a;
1298 
1299 err_mod:
1300 	module_put(ops->owner);
1301 err_out:
1302 	return ERR_PTR(err);
1303 }
1304 
tca_action_flush(struct net * net,struct nlattr * nla,struct nlmsghdr * n,u32 portid,struct netlink_ext_ack * extack)1305 static int tca_action_flush(struct net *net, struct nlattr *nla,
1306 			    struct nlmsghdr *n, u32 portid,
1307 			    struct netlink_ext_ack *extack)
1308 {
1309 	struct sk_buff *skb;
1310 	unsigned char *b;
1311 	struct nlmsghdr *nlh;
1312 	struct tcamsg *t;
1313 	struct netlink_callback dcb;
1314 	struct nlattr *nest;
1315 	struct nlattr *tb[TCA_ACT_MAX + 1];
1316 	const struct tc_action_ops *ops;
1317 	struct nlattr *kind;
1318 	int err = -ENOMEM;
1319 
1320 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
1321 	if (!skb)
1322 		return err;
1323 
1324 	b = skb_tail_pointer(skb);
1325 
1326 	err = nla_parse_nested_deprecated(tb, TCA_ACT_MAX, nla,
1327 					  tcf_action_policy, extack);
1328 	if (err < 0)
1329 		goto err_out;
1330 
1331 	err = -EINVAL;
1332 	kind = tb[TCA_ACT_KIND];
1333 	ops = tc_lookup_action(kind);
1334 	if (!ops) { /*some idjot trying to flush unknown action */
1335 		NL_SET_ERR_MSG(extack, "Cannot flush unknown TC action");
1336 		goto err_out;
1337 	}
1338 
1339 	nlh = nlmsg_put(skb, portid, n->nlmsg_seq, RTM_DELACTION,
1340 			sizeof(*t), 0);
1341 	if (!nlh) {
1342 		NL_SET_ERR_MSG(extack, "Failed to create TC action flush notification");
1343 		goto out_module_put;
1344 	}
1345 	t = nlmsg_data(nlh);
1346 	t->tca_family = AF_UNSPEC;
1347 	t->tca__pad1 = 0;
1348 	t->tca__pad2 = 0;
1349 
1350 	nest = nla_nest_start_noflag(skb, TCA_ACT_TAB);
1351 	if (!nest) {
1352 		NL_SET_ERR_MSG(extack, "Failed to add new netlink message");
1353 		goto out_module_put;
1354 	}
1355 
1356 	err = ops->walk(net, skb, &dcb, RTM_DELACTION, ops, extack);
1357 	if (err <= 0) {
1358 		nla_nest_cancel(skb, nest);
1359 		goto out_module_put;
1360 	}
1361 
1362 	nla_nest_end(skb, nest);
1363 
1364 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1365 	nlh->nlmsg_flags |= NLM_F_ROOT;
1366 	module_put(ops->owner);
1367 	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1368 			     n->nlmsg_flags & NLM_F_ECHO);
1369 	if (err < 0)
1370 		NL_SET_ERR_MSG(extack, "Failed to send TC action flush notification");
1371 
1372 	return err;
1373 
1374 out_module_put:
1375 	module_put(ops->owner);
1376 err_out:
1377 	kfree_skb(skb);
1378 	return err;
1379 }
1380 
tcf_action_delete(struct net * net,struct tc_action * actions[])1381 static int tcf_action_delete(struct net *net, struct tc_action *actions[])
1382 {
1383 	int i;
1384 
1385 	for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) {
1386 		struct tc_action *a = actions[i];
1387 		const struct tc_action_ops *ops = a->ops;
1388 		/* Actions can be deleted concurrently so we must save their
1389 		 * type and id to search again after reference is released.
1390 		 */
1391 		struct tcf_idrinfo *idrinfo = a->idrinfo;
1392 		u32 act_index = a->tcfa_index;
1393 
1394 		actions[i] = NULL;
1395 		if (tcf_action_put(a)) {
1396 			/* last reference, action was deleted concurrently */
1397 			module_put(ops->owner);
1398 		} else  {
1399 			int ret;
1400 
1401 			/* now do the delete */
1402 			ret = tcf_idr_delete_index(idrinfo, act_index);
1403 			if (ret < 0)
1404 				return ret;
1405 		}
1406 	}
1407 	return 0;
1408 }
1409 
1410 static int
tcf_del_notify(struct net * net,struct nlmsghdr * n,struct tc_action * actions[],u32 portid,size_t attr_size,struct netlink_ext_ack * extack)1411 tcf_del_notify(struct net *net, struct nlmsghdr *n, struct tc_action *actions[],
1412 	       u32 portid, size_t attr_size, struct netlink_ext_ack *extack)
1413 {
1414 	int ret;
1415 	struct sk_buff *skb;
1416 
1417 	skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size,
1418 			GFP_KERNEL);
1419 	if (!skb)
1420 		return -ENOBUFS;
1421 
1422 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, RTM_DELACTION,
1423 			 0, 2) <= 0) {
1424 		NL_SET_ERR_MSG(extack, "Failed to fill netlink TC action attributes");
1425 		kfree_skb(skb);
1426 		return -EINVAL;
1427 	}
1428 
1429 	/* now do the delete */
1430 	ret = tcf_action_delete(net, actions);
1431 	if (ret < 0) {
1432 		NL_SET_ERR_MSG(extack, "Failed to delete TC action");
1433 		kfree_skb(skb);
1434 		return ret;
1435 	}
1436 
1437 	ret = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1438 			     n->nlmsg_flags & NLM_F_ECHO);
1439 	return ret;
1440 }
1441 
1442 static int
tca_action_gd(struct net * net,struct nlattr * nla,struct nlmsghdr * n,u32 portid,int event,struct netlink_ext_ack * extack)1443 tca_action_gd(struct net *net, struct nlattr *nla, struct nlmsghdr *n,
1444 	      u32 portid, int event, struct netlink_ext_ack *extack)
1445 {
1446 	int i, ret;
1447 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1448 	struct tc_action *act;
1449 	size_t attr_size = 0;
1450 	struct tc_action *actions[TCA_ACT_MAX_PRIO] = {};
1451 
1452 	ret = nla_parse_nested_deprecated(tb, TCA_ACT_MAX_PRIO, nla, NULL,
1453 					  extack);
1454 	if (ret < 0)
1455 		return ret;
1456 
1457 	if (event == RTM_DELACTION && n->nlmsg_flags & NLM_F_ROOT) {
1458 		if (tb[1])
1459 			return tca_action_flush(net, tb[1], n, portid, extack);
1460 
1461 		NL_SET_ERR_MSG(extack, "Invalid netlink attributes while flushing TC action");
1462 		return -EINVAL;
1463 	}
1464 
1465 	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
1466 		act = tcf_action_get_1(net, tb[i], n, portid, extack);
1467 		if (IS_ERR(act)) {
1468 			ret = PTR_ERR(act);
1469 			goto err;
1470 		}
1471 		attr_size += tcf_action_fill_size(act);
1472 		actions[i - 1] = act;
1473 	}
1474 
1475 	attr_size = tcf_action_full_attrs_size(attr_size);
1476 
1477 	if (event == RTM_GETACTION)
1478 		ret = tcf_get_notify(net, portid, n, actions, event, extack);
1479 	else { /* delete */
1480 		ret = tcf_del_notify(net, n, actions, portid, attr_size, extack);
1481 		if (ret)
1482 			goto err;
1483 		return 0;
1484 	}
1485 err:
1486 	tcf_action_put_many(actions);
1487 	return ret;
1488 }
1489 
1490 static int
tcf_add_notify(struct net * net,struct nlmsghdr * n,struct tc_action * actions[],u32 portid,size_t attr_size,struct netlink_ext_ack * extack)1491 tcf_add_notify(struct net *net, struct nlmsghdr *n, struct tc_action *actions[],
1492 	       u32 portid, size_t attr_size, struct netlink_ext_ack *extack)
1493 {
1494 	struct sk_buff *skb;
1495 
1496 	skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size,
1497 			GFP_KERNEL);
1498 	if (!skb)
1499 		return -ENOBUFS;
1500 
1501 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, n->nlmsg_flags,
1502 			 RTM_NEWACTION, 0, 0) <= 0) {
1503 		NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action");
1504 		kfree_skb(skb);
1505 		return -EINVAL;
1506 	}
1507 
1508 	return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
1509 			      n->nlmsg_flags & NLM_F_ECHO);
1510 }
1511 
tcf_action_add(struct net * net,struct nlattr * nla,struct nlmsghdr * n,u32 portid,u32 flags,struct netlink_ext_ack * extack)1512 static int tcf_action_add(struct net *net, struct nlattr *nla,
1513 			  struct nlmsghdr *n, u32 portid, u32 flags,
1514 			  struct netlink_ext_ack *extack)
1515 {
1516 	size_t attr_size = 0;
1517 	int loop, ret, i;
1518 	struct tc_action *actions[TCA_ACT_MAX_PRIO] = {};
1519 	int init_res[TCA_ACT_MAX_PRIO] = {};
1520 
1521 	for (loop = 0; loop < 10; loop++) {
1522 		ret = tcf_action_init(net, NULL, nla, NULL, actions, init_res,
1523 				      &attr_size, flags, extack);
1524 		if (ret != -EAGAIN)
1525 			break;
1526 	}
1527 
1528 	if (ret < 0)
1529 		return ret;
1530 	ret = tcf_add_notify(net, n, actions, portid, attr_size, extack);
1531 
1532 	/* only put existing actions */
1533 	for (i = 0; i < TCA_ACT_MAX_PRIO; i++)
1534 		if (init_res[i] == ACT_P_CREATED)
1535 			actions[i] = NULL;
1536 	tcf_action_put_many(actions);
1537 
1538 	return ret;
1539 }
1540 
1541 static const struct nla_policy tcaa_policy[TCA_ROOT_MAX + 1] = {
1542 	[TCA_ROOT_FLAGS] = NLA_POLICY_BITFIELD32(TCA_ACT_FLAG_LARGE_DUMP_ON |
1543 						 TCA_ACT_FLAG_TERSE_DUMP),
1544 	[TCA_ROOT_TIME_DELTA]      = { .type = NLA_U32 },
1545 };
1546 
tc_ctl_action(struct sk_buff * skb,struct nlmsghdr * n,struct netlink_ext_ack * extack)1547 static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n,
1548 			 struct netlink_ext_ack *extack)
1549 {
1550 	struct net *net = sock_net(skb->sk);
1551 	struct nlattr *tca[TCA_ROOT_MAX + 1];
1552 	u32 portid = NETLINK_CB(skb).portid;
1553 	u32 flags = 0;
1554 	int ret = 0;
1555 
1556 	if ((n->nlmsg_type != RTM_GETACTION) &&
1557 	    !netlink_capable(skb, CAP_NET_ADMIN))
1558 		return -EPERM;
1559 
1560 	ret = nlmsg_parse_deprecated(n, sizeof(struct tcamsg), tca,
1561 				     TCA_ROOT_MAX, NULL, extack);
1562 	if (ret < 0)
1563 		return ret;
1564 
1565 	if (tca[TCA_ACT_TAB] == NULL) {
1566 		NL_SET_ERR_MSG(extack, "Netlink action attributes missing");
1567 		return -EINVAL;
1568 	}
1569 
1570 	/* n->nlmsg_flags & NLM_F_CREATE */
1571 	switch (n->nlmsg_type) {
1572 	case RTM_NEWACTION:
1573 		/* we are going to assume all other flags
1574 		 * imply create only if it doesn't exist
1575 		 * Note that CREATE | EXCL implies that
1576 		 * but since we want avoid ambiguity (eg when flags
1577 		 * is zero) then just set this
1578 		 */
1579 		if (n->nlmsg_flags & NLM_F_REPLACE)
1580 			flags = TCA_ACT_FLAGS_REPLACE;
1581 		ret = tcf_action_add(net, tca[TCA_ACT_TAB], n, portid, flags,
1582 				     extack);
1583 		break;
1584 	case RTM_DELACTION:
1585 		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1586 				    portid, RTM_DELACTION, extack);
1587 		break;
1588 	case RTM_GETACTION:
1589 		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1590 				    portid, RTM_GETACTION, extack);
1591 		break;
1592 	default:
1593 		BUG();
1594 	}
1595 
1596 	return ret;
1597 }
1598 
find_dump_kind(struct nlattr ** nla)1599 static struct nlattr *find_dump_kind(struct nlattr **nla)
1600 {
1601 	struct nlattr *tb1, *tb2[TCA_ACT_MAX + 1];
1602 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1603 	struct nlattr *kind;
1604 
1605 	tb1 = nla[TCA_ACT_TAB];
1606 	if (tb1 == NULL)
1607 		return NULL;
1608 
1609 	if (nla_parse_deprecated(tb, TCA_ACT_MAX_PRIO, nla_data(tb1), NLMSG_ALIGN(nla_len(tb1)), NULL, NULL) < 0)
1610 		return NULL;
1611 
1612 	if (tb[1] == NULL)
1613 		return NULL;
1614 	if (nla_parse_nested_deprecated(tb2, TCA_ACT_MAX, tb[1], tcf_action_policy, NULL) < 0)
1615 		return NULL;
1616 	kind = tb2[TCA_ACT_KIND];
1617 
1618 	return kind;
1619 }
1620 
tc_dump_action(struct sk_buff * skb,struct netlink_callback * cb)1621 static int tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb)
1622 {
1623 	struct net *net = sock_net(skb->sk);
1624 	struct nlmsghdr *nlh;
1625 	unsigned char *b = skb_tail_pointer(skb);
1626 	struct nlattr *nest;
1627 	struct tc_action_ops *a_o;
1628 	int ret = 0;
1629 	struct tcamsg *t = (struct tcamsg *) nlmsg_data(cb->nlh);
1630 	struct nlattr *tb[TCA_ROOT_MAX + 1];
1631 	struct nlattr *count_attr = NULL;
1632 	unsigned long jiffy_since = 0;
1633 	struct nlattr *kind = NULL;
1634 	struct nla_bitfield32 bf;
1635 	u32 msecs_since = 0;
1636 	u32 act_count = 0;
1637 
1638 	ret = nlmsg_parse_deprecated(cb->nlh, sizeof(struct tcamsg), tb,
1639 				     TCA_ROOT_MAX, tcaa_policy, cb->extack);
1640 	if (ret < 0)
1641 		return ret;
1642 
1643 	kind = find_dump_kind(tb);
1644 	if (kind == NULL) {
1645 		pr_info("tc_dump_action: action bad kind\n");
1646 		return 0;
1647 	}
1648 
1649 	a_o = tc_lookup_action(kind);
1650 	if (a_o == NULL)
1651 		return 0;
1652 
1653 	cb->args[2] = 0;
1654 	if (tb[TCA_ROOT_FLAGS]) {
1655 		bf = nla_get_bitfield32(tb[TCA_ROOT_FLAGS]);
1656 		cb->args[2] = bf.value;
1657 	}
1658 
1659 	if (tb[TCA_ROOT_TIME_DELTA]) {
1660 		msecs_since = nla_get_u32(tb[TCA_ROOT_TIME_DELTA]);
1661 	}
1662 
1663 	nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
1664 			cb->nlh->nlmsg_type, sizeof(*t), 0);
1665 	if (!nlh)
1666 		goto out_module_put;
1667 
1668 	if (msecs_since)
1669 		jiffy_since = jiffies - msecs_to_jiffies(msecs_since);
1670 
1671 	t = nlmsg_data(nlh);
1672 	t->tca_family = AF_UNSPEC;
1673 	t->tca__pad1 = 0;
1674 	t->tca__pad2 = 0;
1675 	cb->args[3] = jiffy_since;
1676 	count_attr = nla_reserve(skb, TCA_ROOT_COUNT, sizeof(u32));
1677 	if (!count_attr)
1678 		goto out_module_put;
1679 
1680 	nest = nla_nest_start_noflag(skb, TCA_ACT_TAB);
1681 	if (nest == NULL)
1682 		goto out_module_put;
1683 
1684 	ret = a_o->walk(net, skb, cb, RTM_GETACTION, a_o, NULL);
1685 	if (ret < 0)
1686 		goto out_module_put;
1687 
1688 	if (ret > 0) {
1689 		nla_nest_end(skb, nest);
1690 		ret = skb->len;
1691 		act_count = cb->args[1];
1692 		memcpy(nla_data(count_attr), &act_count, sizeof(u32));
1693 		cb->args[1] = 0;
1694 	} else
1695 		nlmsg_trim(skb, b);
1696 
1697 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1698 	if (NETLINK_CB(cb->skb).portid && ret)
1699 		nlh->nlmsg_flags |= NLM_F_MULTI;
1700 	module_put(a_o->owner);
1701 	return skb->len;
1702 
1703 out_module_put:
1704 	module_put(a_o->owner);
1705 	nlmsg_trim(skb, b);
1706 	return skb->len;
1707 }
1708 
tc_action_init(void)1709 static int __init tc_action_init(void)
1710 {
1711 	rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL, 0);
1712 	rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL, 0);
1713 	rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action,
1714 		      0);
1715 
1716 	return 0;
1717 }
1718 
1719 subsys_initcall(tc_action_init);
1720