xref: /openbmc/linux/net/sched/act_api.c (revision 174cd4b1)
1 /*
2  * net/sched/act_api.c	Packet action API.
3  *
4  *		This program is free software; you can redistribute it and/or
5  *		modify it under the terms of the GNU General Public License
6  *		as published by the Free Software Foundation; either version
7  *		2 of the License, or (at your option) any later version.
8  *
9  * Author:	Jamal Hadi Salim
10  *
11  *
12  */
13 
14 #include <linux/types.h>
15 #include <linux/kernel.h>
16 #include <linux/string.h>
17 #include <linux/errno.h>
18 #include <linux/slab.h>
19 #include <linux/skbuff.h>
20 #include <linux/init.h>
21 #include <linux/kmod.h>
22 #include <linux/err.h>
23 #include <linux/module.h>
24 #include <net/net_namespace.h>
25 #include <net/sock.h>
26 #include <net/sch_generic.h>
27 #include <net/pkt_cls.h>
28 #include <net/act_api.h>
29 #include <net/netlink.h>
30 
31 static void free_tcf(struct rcu_head *head)
32 {
33 	struct tc_action *p = container_of(head, struct tc_action, tcfa_rcu);
34 
35 	free_percpu(p->cpu_bstats);
36 	free_percpu(p->cpu_qstats);
37 
38 	if (p->act_cookie) {
39 		kfree(p->act_cookie->data);
40 		kfree(p->act_cookie);
41 	}
42 
43 	kfree(p);
44 }
45 
46 static void tcf_hash_destroy(struct tcf_hashinfo *hinfo, struct tc_action *p)
47 {
48 	spin_lock_bh(&hinfo->lock);
49 	hlist_del(&p->tcfa_head);
50 	spin_unlock_bh(&hinfo->lock);
51 	gen_kill_estimator(&p->tcfa_rate_est);
52 	/*
53 	 * gen_estimator est_timer() might access p->tcfa_lock
54 	 * or bstats, wait a RCU grace period before freeing p
55 	 */
56 	call_rcu(&p->tcfa_rcu, free_tcf);
57 }
58 
59 int __tcf_hash_release(struct tc_action *p, bool bind, bool strict)
60 {
61 	int ret = 0;
62 
63 	if (p) {
64 		if (bind)
65 			p->tcfa_bindcnt--;
66 		else if (strict && p->tcfa_bindcnt > 0)
67 			return -EPERM;
68 
69 		p->tcfa_refcnt--;
70 		if (p->tcfa_bindcnt <= 0 && p->tcfa_refcnt <= 0) {
71 			if (p->ops->cleanup)
72 				p->ops->cleanup(p, bind);
73 			tcf_hash_destroy(p->hinfo, p);
74 			ret = ACT_P_DELETED;
75 		}
76 	}
77 
78 	return ret;
79 }
80 EXPORT_SYMBOL(__tcf_hash_release);
81 
82 static int tcf_dump_walker(struct tcf_hashinfo *hinfo, struct sk_buff *skb,
83 			   struct netlink_callback *cb)
84 {
85 	int err = 0, index = -1, i = 0, s_i = 0, n_i = 0;
86 	struct nlattr *nest;
87 
88 	spin_lock_bh(&hinfo->lock);
89 
90 	s_i = cb->args[0];
91 
92 	for (i = 0; i < (hinfo->hmask + 1); i++) {
93 		struct hlist_head *head;
94 		struct tc_action *p;
95 
96 		head = &hinfo->htab[tcf_hash(i, hinfo->hmask)];
97 
98 		hlist_for_each_entry_rcu(p, head, tcfa_head) {
99 			index++;
100 			if (index < s_i)
101 				continue;
102 
103 			nest = nla_nest_start(skb, n_i);
104 			if (nest == NULL)
105 				goto nla_put_failure;
106 			err = tcf_action_dump_1(skb, p, 0, 0);
107 			if (err < 0) {
108 				index--;
109 				nlmsg_trim(skb, nest);
110 				goto done;
111 			}
112 			nla_nest_end(skb, nest);
113 			n_i++;
114 			if (n_i >= TCA_ACT_MAX_PRIO)
115 				goto done;
116 		}
117 	}
118 done:
119 	spin_unlock_bh(&hinfo->lock);
120 	if (n_i)
121 		cb->args[0] += n_i;
122 	return n_i;
123 
124 nla_put_failure:
125 	nla_nest_cancel(skb, nest);
126 	goto done;
127 }
128 
129 static int tcf_del_walker(struct tcf_hashinfo *hinfo, struct sk_buff *skb,
130 			  const struct tc_action_ops *ops)
131 {
132 	struct nlattr *nest;
133 	int i = 0, n_i = 0;
134 	int ret = -EINVAL;
135 
136 	nest = nla_nest_start(skb, 0);
137 	if (nest == NULL)
138 		goto nla_put_failure;
139 	if (nla_put_string(skb, TCA_KIND, ops->kind))
140 		goto nla_put_failure;
141 	for (i = 0; i < (hinfo->hmask + 1); i++) {
142 		struct hlist_head *head;
143 		struct hlist_node *n;
144 		struct tc_action *p;
145 
146 		head = &hinfo->htab[tcf_hash(i, hinfo->hmask)];
147 		hlist_for_each_entry_safe(p, n, head, tcfa_head) {
148 			ret = __tcf_hash_release(p, false, true);
149 			if (ret == ACT_P_DELETED) {
150 				module_put(p->ops->owner);
151 				n_i++;
152 			} else if (ret < 0)
153 				goto nla_put_failure;
154 		}
155 	}
156 	if (nla_put_u32(skb, TCA_FCNT, n_i))
157 		goto nla_put_failure;
158 	nla_nest_end(skb, nest);
159 
160 	return n_i;
161 nla_put_failure:
162 	nla_nest_cancel(skb, nest);
163 	return ret;
164 }
165 
166 int tcf_generic_walker(struct tc_action_net *tn, struct sk_buff *skb,
167 		       struct netlink_callback *cb, int type,
168 		       const struct tc_action_ops *ops)
169 {
170 	struct tcf_hashinfo *hinfo = tn->hinfo;
171 
172 	if (type == RTM_DELACTION) {
173 		return tcf_del_walker(hinfo, skb, ops);
174 	} else if (type == RTM_GETACTION) {
175 		return tcf_dump_walker(hinfo, skb, cb);
176 	} else {
177 		WARN(1, "tcf_generic_walker: unknown action %d\n", type);
178 		return -EINVAL;
179 	}
180 }
181 EXPORT_SYMBOL(tcf_generic_walker);
182 
183 static struct tc_action *tcf_hash_lookup(u32 index, struct tcf_hashinfo *hinfo)
184 {
185 	struct tc_action *p = NULL;
186 	struct hlist_head *head;
187 
188 	spin_lock_bh(&hinfo->lock);
189 	head = &hinfo->htab[tcf_hash(index, hinfo->hmask)];
190 	hlist_for_each_entry_rcu(p, head, tcfa_head)
191 		if (p->tcfa_index == index)
192 			break;
193 	spin_unlock_bh(&hinfo->lock);
194 
195 	return p;
196 }
197 
198 u32 tcf_hash_new_index(struct tc_action_net *tn)
199 {
200 	struct tcf_hashinfo *hinfo = tn->hinfo;
201 	u32 val = hinfo->index;
202 
203 	do {
204 		if (++val == 0)
205 			val = 1;
206 	} while (tcf_hash_lookup(val, hinfo));
207 
208 	hinfo->index = val;
209 	return val;
210 }
211 EXPORT_SYMBOL(tcf_hash_new_index);
212 
213 int tcf_hash_search(struct tc_action_net *tn, struct tc_action **a, u32 index)
214 {
215 	struct tcf_hashinfo *hinfo = tn->hinfo;
216 	struct tc_action *p = tcf_hash_lookup(index, hinfo);
217 
218 	if (p) {
219 		*a = p;
220 		return 1;
221 	}
222 	return 0;
223 }
224 EXPORT_SYMBOL(tcf_hash_search);
225 
226 bool tcf_hash_check(struct tc_action_net *tn, u32 index, struct tc_action **a,
227 		    int bind)
228 {
229 	struct tcf_hashinfo *hinfo = tn->hinfo;
230 	struct tc_action *p = NULL;
231 
232 	if (index && (p = tcf_hash_lookup(index, hinfo)) != NULL) {
233 		if (bind)
234 			p->tcfa_bindcnt++;
235 		p->tcfa_refcnt++;
236 		*a = p;
237 		return true;
238 	}
239 	return false;
240 }
241 EXPORT_SYMBOL(tcf_hash_check);
242 
243 void tcf_hash_cleanup(struct tc_action *a, struct nlattr *est)
244 {
245 	if (est)
246 		gen_kill_estimator(&a->tcfa_rate_est);
247 	call_rcu(&a->tcfa_rcu, free_tcf);
248 }
249 EXPORT_SYMBOL(tcf_hash_cleanup);
250 
251 int tcf_hash_create(struct tc_action_net *tn, u32 index, struct nlattr *est,
252 		    struct tc_action **a, const struct tc_action_ops *ops,
253 		    int bind, bool cpustats)
254 {
255 	struct tc_action *p = kzalloc(ops->size, GFP_KERNEL);
256 	struct tcf_hashinfo *hinfo = tn->hinfo;
257 	int err = -ENOMEM;
258 
259 	if (unlikely(!p))
260 		return -ENOMEM;
261 	p->tcfa_refcnt = 1;
262 	if (bind)
263 		p->tcfa_bindcnt = 1;
264 
265 	if (cpustats) {
266 		p->cpu_bstats = netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu);
267 		if (!p->cpu_bstats) {
268 err1:
269 			kfree(p);
270 			return err;
271 		}
272 		p->cpu_qstats = alloc_percpu(struct gnet_stats_queue);
273 		if (!p->cpu_qstats) {
274 err2:
275 			free_percpu(p->cpu_bstats);
276 			goto err1;
277 		}
278 	}
279 	spin_lock_init(&p->tcfa_lock);
280 	INIT_HLIST_NODE(&p->tcfa_head);
281 	p->tcfa_index = index ? index : tcf_hash_new_index(tn);
282 	p->tcfa_tm.install = jiffies;
283 	p->tcfa_tm.lastuse = jiffies;
284 	p->tcfa_tm.firstuse = 0;
285 	if (est) {
286 		err = gen_new_estimator(&p->tcfa_bstats, p->cpu_bstats,
287 					&p->tcfa_rate_est,
288 					&p->tcfa_lock, NULL, est);
289 		if (err) {
290 			free_percpu(p->cpu_qstats);
291 			goto err2;
292 		}
293 	}
294 
295 	p->hinfo = hinfo;
296 	p->ops = ops;
297 	INIT_LIST_HEAD(&p->list);
298 	*a = p;
299 	return 0;
300 }
301 EXPORT_SYMBOL(tcf_hash_create);
302 
303 void tcf_hash_insert(struct tc_action_net *tn, struct tc_action *a)
304 {
305 	struct tcf_hashinfo *hinfo = tn->hinfo;
306 	unsigned int h = tcf_hash(a->tcfa_index, hinfo->hmask);
307 
308 	spin_lock_bh(&hinfo->lock);
309 	hlist_add_head(&a->tcfa_head, &hinfo->htab[h]);
310 	spin_unlock_bh(&hinfo->lock);
311 }
312 EXPORT_SYMBOL(tcf_hash_insert);
313 
314 void tcf_hashinfo_destroy(const struct tc_action_ops *ops,
315 			  struct tcf_hashinfo *hinfo)
316 {
317 	int i;
318 
319 	for (i = 0; i < hinfo->hmask + 1; i++) {
320 		struct tc_action *p;
321 		struct hlist_node *n;
322 
323 		hlist_for_each_entry_safe(p, n, &hinfo->htab[i], tcfa_head) {
324 			int ret;
325 
326 			ret = __tcf_hash_release(p, false, true);
327 			if (ret == ACT_P_DELETED)
328 				module_put(ops->owner);
329 			else if (ret < 0)
330 				return;
331 		}
332 	}
333 	kfree(hinfo->htab);
334 }
335 EXPORT_SYMBOL(tcf_hashinfo_destroy);
336 
337 static LIST_HEAD(act_base);
338 static DEFINE_RWLOCK(act_mod_lock);
339 
340 int tcf_register_action(struct tc_action_ops *act,
341 			struct pernet_operations *ops)
342 {
343 	struct tc_action_ops *a;
344 	int ret;
345 
346 	if (!act->act || !act->dump || !act->init || !act->walk || !act->lookup)
347 		return -EINVAL;
348 
349 	/* We have to register pernet ops before making the action ops visible,
350 	 * otherwise tcf_action_init_1() could get a partially initialized
351 	 * netns.
352 	 */
353 	ret = register_pernet_subsys(ops);
354 	if (ret)
355 		return ret;
356 
357 	write_lock(&act_mod_lock);
358 	list_for_each_entry(a, &act_base, head) {
359 		if (act->type == a->type || (strcmp(act->kind, a->kind) == 0)) {
360 			write_unlock(&act_mod_lock);
361 			unregister_pernet_subsys(ops);
362 			return -EEXIST;
363 		}
364 	}
365 	list_add_tail(&act->head, &act_base);
366 	write_unlock(&act_mod_lock);
367 
368 	return 0;
369 }
370 EXPORT_SYMBOL(tcf_register_action);
371 
372 int tcf_unregister_action(struct tc_action_ops *act,
373 			  struct pernet_operations *ops)
374 {
375 	struct tc_action_ops *a;
376 	int err = -ENOENT;
377 
378 	write_lock(&act_mod_lock);
379 	list_for_each_entry(a, &act_base, head) {
380 		if (a == act) {
381 			list_del(&act->head);
382 			err = 0;
383 			break;
384 		}
385 	}
386 	write_unlock(&act_mod_lock);
387 	if (!err)
388 		unregister_pernet_subsys(ops);
389 	return err;
390 }
391 EXPORT_SYMBOL(tcf_unregister_action);
392 
393 /* lookup by name */
394 static struct tc_action_ops *tc_lookup_action_n(char *kind)
395 {
396 	struct tc_action_ops *a, *res = NULL;
397 
398 	if (kind) {
399 		read_lock(&act_mod_lock);
400 		list_for_each_entry(a, &act_base, head) {
401 			if (strcmp(kind, a->kind) == 0) {
402 				if (try_module_get(a->owner))
403 					res = a;
404 				break;
405 			}
406 		}
407 		read_unlock(&act_mod_lock);
408 	}
409 	return res;
410 }
411 
412 /* lookup by nlattr */
413 static struct tc_action_ops *tc_lookup_action(struct nlattr *kind)
414 {
415 	struct tc_action_ops *a, *res = NULL;
416 
417 	if (kind) {
418 		read_lock(&act_mod_lock);
419 		list_for_each_entry(a, &act_base, head) {
420 			if (nla_strcmp(kind, a->kind) == 0) {
421 				if (try_module_get(a->owner))
422 					res = a;
423 				break;
424 			}
425 		}
426 		read_unlock(&act_mod_lock);
427 	}
428 	return res;
429 }
430 
431 int tcf_action_exec(struct sk_buff *skb, struct tc_action **actions,
432 		    int nr_actions, struct tcf_result *res)
433 {
434 	int ret = -1, i;
435 
436 	if (skb_skip_tc_classify(skb))
437 		return TC_ACT_OK;
438 
439 	for (i = 0; i < nr_actions; i++) {
440 		const struct tc_action *a = actions[i];
441 
442 repeat:
443 		ret = a->ops->act(skb, a, res);
444 		if (ret == TC_ACT_REPEAT)
445 			goto repeat;	/* we need a ttl - JHS */
446 		if (ret != TC_ACT_PIPE)
447 			break;
448 	}
449 	return ret;
450 }
451 EXPORT_SYMBOL(tcf_action_exec);
452 
453 int tcf_action_destroy(struct list_head *actions, int bind)
454 {
455 	struct tc_action *a, *tmp;
456 	int ret = 0;
457 
458 	list_for_each_entry_safe(a, tmp, actions, list) {
459 		ret = __tcf_hash_release(a, bind, true);
460 		if (ret == ACT_P_DELETED)
461 			module_put(a->ops->owner);
462 		else if (ret < 0)
463 			return ret;
464 	}
465 	return ret;
466 }
467 
468 int
469 tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
470 {
471 	return a->ops->dump(skb, a, bind, ref);
472 }
473 
474 int
475 tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
476 {
477 	int err = -EINVAL;
478 	unsigned char *b = skb_tail_pointer(skb);
479 	struct nlattr *nest;
480 
481 	if (nla_put_string(skb, TCA_KIND, a->ops->kind))
482 		goto nla_put_failure;
483 	if (tcf_action_copy_stats(skb, a, 0))
484 		goto nla_put_failure;
485 	if (a->act_cookie) {
486 		if (nla_put(skb, TCA_ACT_COOKIE, a->act_cookie->len,
487 			    a->act_cookie->data))
488 			goto nla_put_failure;
489 	}
490 
491 	nest = nla_nest_start(skb, TCA_OPTIONS);
492 	if (nest == NULL)
493 		goto nla_put_failure;
494 	err = tcf_action_dump_old(skb, a, bind, ref);
495 	if (err > 0) {
496 		nla_nest_end(skb, nest);
497 		return err;
498 	}
499 
500 nla_put_failure:
501 	nlmsg_trim(skb, b);
502 	return -1;
503 }
504 EXPORT_SYMBOL(tcf_action_dump_1);
505 
506 int tcf_action_dump(struct sk_buff *skb, struct list_head *actions,
507 		    int bind, int ref)
508 {
509 	struct tc_action *a;
510 	int err = -EINVAL;
511 	struct nlattr *nest;
512 
513 	list_for_each_entry(a, actions, list) {
514 		nest = nla_nest_start(skb, a->order);
515 		if (nest == NULL)
516 			goto nla_put_failure;
517 		err = tcf_action_dump_1(skb, a, bind, ref);
518 		if (err < 0)
519 			goto errout;
520 		nla_nest_end(skb, nest);
521 	}
522 
523 	return 0;
524 
525 nla_put_failure:
526 	err = -EINVAL;
527 errout:
528 	nla_nest_cancel(skb, nest);
529 	return err;
530 }
531 
532 static int nla_memdup_cookie(struct tc_action *a, struct nlattr **tb)
533 {
534 	a->act_cookie = kzalloc(sizeof(*a->act_cookie), GFP_KERNEL);
535 	if (!a->act_cookie)
536 		return -ENOMEM;
537 
538 	a->act_cookie->data = nla_memdup(tb[TCA_ACT_COOKIE], GFP_KERNEL);
539 	if (!a->act_cookie->data) {
540 		kfree(a->act_cookie);
541 		return -ENOMEM;
542 	}
543 	a->act_cookie->len = nla_len(tb[TCA_ACT_COOKIE]);
544 
545 	return 0;
546 }
547 
548 struct tc_action *tcf_action_init_1(struct net *net, struct nlattr *nla,
549 				    struct nlattr *est, char *name, int ovr,
550 				    int bind)
551 {
552 	struct tc_action *a;
553 	struct tc_action_ops *a_o;
554 	char act_name[IFNAMSIZ];
555 	struct nlattr *tb[TCA_ACT_MAX + 1];
556 	struct nlattr *kind;
557 	int err;
558 
559 	if (name == NULL) {
560 		err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
561 		if (err < 0)
562 			goto err_out;
563 		err = -EINVAL;
564 		kind = tb[TCA_ACT_KIND];
565 		if (kind == NULL)
566 			goto err_out;
567 		if (nla_strlcpy(act_name, kind, IFNAMSIZ) >= IFNAMSIZ)
568 			goto err_out;
569 	} else {
570 		err = -EINVAL;
571 		if (strlcpy(act_name, name, IFNAMSIZ) >= IFNAMSIZ)
572 			goto err_out;
573 	}
574 
575 	a_o = tc_lookup_action_n(act_name);
576 	if (a_o == NULL) {
577 #ifdef CONFIG_MODULES
578 		rtnl_unlock();
579 		request_module("act_%s", act_name);
580 		rtnl_lock();
581 
582 		a_o = tc_lookup_action_n(act_name);
583 
584 		/* We dropped the RTNL semaphore in order to
585 		 * perform the module load.  So, even if we
586 		 * succeeded in loading the module we have to
587 		 * tell the caller to replay the request.  We
588 		 * indicate this using -EAGAIN.
589 		 */
590 		if (a_o != NULL) {
591 			err = -EAGAIN;
592 			goto err_mod;
593 		}
594 #endif
595 		err = -ENOENT;
596 		goto err_out;
597 	}
598 
599 	/* backward compatibility for policer */
600 	if (name == NULL)
601 		err = a_o->init(net, tb[TCA_ACT_OPTIONS], est, &a, ovr, bind);
602 	else
603 		err = a_o->init(net, nla, est, &a, ovr, bind);
604 	if (err < 0)
605 		goto err_mod;
606 
607 	if (tb[TCA_ACT_COOKIE]) {
608 		int cklen = nla_len(tb[TCA_ACT_COOKIE]);
609 
610 		if (cklen > TC_COOKIE_MAX_SIZE) {
611 			err = -EINVAL;
612 			tcf_hash_release(a, bind);
613 			goto err_mod;
614 		}
615 
616 		if (nla_memdup_cookie(a, tb) < 0) {
617 			err = -ENOMEM;
618 			tcf_hash_release(a, bind);
619 			goto err_mod;
620 		}
621 	}
622 
623 	/* module count goes up only when brand new policy is created
624 	 * if it exists and is only bound to in a_o->init() then
625 	 * ACT_P_CREATED is not returned (a zero is).
626 	 */
627 	if (err != ACT_P_CREATED)
628 		module_put(a_o->owner);
629 
630 	return a;
631 
632 err_mod:
633 	module_put(a_o->owner);
634 err_out:
635 	return ERR_PTR(err);
636 }
637 
638 static void cleanup_a(struct list_head *actions, int ovr)
639 {
640 	struct tc_action *a;
641 
642 	if (!ovr)
643 		return;
644 
645 	list_for_each_entry(a, actions, list)
646 		a->tcfa_refcnt--;
647 }
648 
649 int tcf_action_init(struct net *net, struct nlattr *nla, struct nlattr *est,
650 		    char *name, int ovr, int bind, struct list_head *actions)
651 {
652 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
653 	struct tc_action *act;
654 	int err;
655 	int i;
656 
657 	err = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL);
658 	if (err < 0)
659 		return err;
660 
661 	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
662 		act = tcf_action_init_1(net, tb[i], est, name, ovr, bind);
663 		if (IS_ERR(act)) {
664 			err = PTR_ERR(act);
665 			goto err;
666 		}
667 		act->order = i;
668 		if (ovr)
669 			act->tcfa_refcnt++;
670 		list_add_tail(&act->list, actions);
671 	}
672 
673 	/* Remove the temp refcnt which was necessary to protect against
674 	 * destroying an existing action which was being replaced
675 	 */
676 	cleanup_a(actions, ovr);
677 	return 0;
678 
679 err:
680 	tcf_action_destroy(actions, bind);
681 	return err;
682 }
683 
684 int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *p,
685 			  int compat_mode)
686 {
687 	int err = 0;
688 	struct gnet_dump d;
689 
690 	if (p == NULL)
691 		goto errout;
692 
693 	/* compat_mode being true specifies a call that is supposed
694 	 * to add additional backward compatibility statistic TLVs.
695 	 */
696 	if (compat_mode) {
697 		if (p->type == TCA_OLD_COMPAT)
698 			err = gnet_stats_start_copy_compat(skb, 0,
699 							   TCA_STATS,
700 							   TCA_XSTATS,
701 							   &p->tcfa_lock, &d,
702 							   TCA_PAD);
703 		else
704 			return 0;
705 	} else
706 		err = gnet_stats_start_copy(skb, TCA_ACT_STATS,
707 					    &p->tcfa_lock, &d, TCA_ACT_PAD);
708 
709 	if (err < 0)
710 		goto errout;
711 
712 	if (gnet_stats_copy_basic(NULL, &d, p->cpu_bstats, &p->tcfa_bstats) < 0 ||
713 	    gnet_stats_copy_rate_est(&d, &p->tcfa_rate_est) < 0 ||
714 	    gnet_stats_copy_queue(&d, p->cpu_qstats,
715 				  &p->tcfa_qstats,
716 				  p->tcfa_qstats.qlen) < 0)
717 		goto errout;
718 
719 	if (gnet_stats_finish_copy(&d) < 0)
720 		goto errout;
721 
722 	return 0;
723 
724 errout:
725 	return -1;
726 }
727 
728 static int tca_get_fill(struct sk_buff *skb, struct list_head *actions,
729 			u32 portid, u32 seq, u16 flags, int event, int bind,
730 			int ref)
731 {
732 	struct tcamsg *t;
733 	struct nlmsghdr *nlh;
734 	unsigned char *b = skb_tail_pointer(skb);
735 	struct nlattr *nest;
736 
737 	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags);
738 	if (!nlh)
739 		goto out_nlmsg_trim;
740 	t = nlmsg_data(nlh);
741 	t->tca_family = AF_UNSPEC;
742 	t->tca__pad1 = 0;
743 	t->tca__pad2 = 0;
744 
745 	nest = nla_nest_start(skb, TCA_ACT_TAB);
746 	if (nest == NULL)
747 		goto out_nlmsg_trim;
748 
749 	if (tcf_action_dump(skb, actions, bind, ref) < 0)
750 		goto out_nlmsg_trim;
751 
752 	nla_nest_end(skb, nest);
753 
754 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
755 	return skb->len;
756 
757 out_nlmsg_trim:
758 	nlmsg_trim(skb, b);
759 	return -1;
760 }
761 
762 static int
763 act_get_notify(struct net *net, u32 portid, struct nlmsghdr *n,
764 	       struct list_head *actions, int event)
765 {
766 	struct sk_buff *skb;
767 
768 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
769 	if (!skb)
770 		return -ENOBUFS;
771 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, event,
772 			 0, 0) <= 0) {
773 		kfree_skb(skb);
774 		return -EINVAL;
775 	}
776 
777 	return rtnl_unicast(skb, net, portid);
778 }
779 
780 static struct tc_action *tcf_action_get_1(struct net *net, struct nlattr *nla,
781 					  struct nlmsghdr *n, u32 portid)
782 {
783 	struct nlattr *tb[TCA_ACT_MAX + 1];
784 	const struct tc_action_ops *ops;
785 	struct tc_action *a;
786 	int index;
787 	int err;
788 
789 	err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
790 	if (err < 0)
791 		goto err_out;
792 
793 	err = -EINVAL;
794 	if (tb[TCA_ACT_INDEX] == NULL ||
795 	    nla_len(tb[TCA_ACT_INDEX]) < sizeof(index))
796 		goto err_out;
797 	index = nla_get_u32(tb[TCA_ACT_INDEX]);
798 
799 	err = -EINVAL;
800 	ops = tc_lookup_action(tb[TCA_ACT_KIND]);
801 	if (!ops) /* could happen in batch of actions */
802 		goto err_out;
803 	err = -ENOENT;
804 	if (ops->lookup(net, &a, index) == 0)
805 		goto err_mod;
806 
807 	module_put(ops->owner);
808 	return a;
809 
810 err_mod:
811 	module_put(ops->owner);
812 err_out:
813 	return ERR_PTR(err);
814 }
815 
816 static int tca_action_flush(struct net *net, struct nlattr *nla,
817 			    struct nlmsghdr *n, u32 portid)
818 {
819 	struct sk_buff *skb;
820 	unsigned char *b;
821 	struct nlmsghdr *nlh;
822 	struct tcamsg *t;
823 	struct netlink_callback dcb;
824 	struct nlattr *nest;
825 	struct nlattr *tb[TCA_ACT_MAX + 1];
826 	const struct tc_action_ops *ops;
827 	struct nlattr *kind;
828 	int err = -ENOMEM;
829 
830 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
831 	if (!skb) {
832 		pr_debug("tca_action_flush: failed skb alloc\n");
833 		return err;
834 	}
835 
836 	b = skb_tail_pointer(skb);
837 
838 	err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
839 	if (err < 0)
840 		goto err_out;
841 
842 	err = -EINVAL;
843 	kind = tb[TCA_ACT_KIND];
844 	ops = tc_lookup_action(kind);
845 	if (!ops) /*some idjot trying to flush unknown action */
846 		goto err_out;
847 
848 	nlh = nlmsg_put(skb, portid, n->nlmsg_seq, RTM_DELACTION,
849 			sizeof(*t), 0);
850 	if (!nlh)
851 		goto out_module_put;
852 	t = nlmsg_data(nlh);
853 	t->tca_family = AF_UNSPEC;
854 	t->tca__pad1 = 0;
855 	t->tca__pad2 = 0;
856 
857 	nest = nla_nest_start(skb, TCA_ACT_TAB);
858 	if (nest == NULL)
859 		goto out_module_put;
860 
861 	err = ops->walk(net, skb, &dcb, RTM_DELACTION, ops);
862 	if (err <= 0)
863 		goto out_module_put;
864 
865 	nla_nest_end(skb, nest);
866 
867 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
868 	nlh->nlmsg_flags |= NLM_F_ROOT;
869 	module_put(ops->owner);
870 	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
871 			     n->nlmsg_flags & NLM_F_ECHO);
872 	if (err > 0)
873 		return 0;
874 
875 	return err;
876 
877 out_module_put:
878 	module_put(ops->owner);
879 err_out:
880 	kfree_skb(skb);
881 	return err;
882 }
883 
884 static int
885 tcf_del_notify(struct net *net, struct nlmsghdr *n, struct list_head *actions,
886 	       u32 portid)
887 {
888 	int ret;
889 	struct sk_buff *skb;
890 
891 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
892 	if (!skb)
893 		return -ENOBUFS;
894 
895 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, RTM_DELACTION,
896 			 0, 1) <= 0) {
897 		kfree_skb(skb);
898 		return -EINVAL;
899 	}
900 
901 	/* now do the delete */
902 	ret = tcf_action_destroy(actions, 0);
903 	if (ret < 0) {
904 		kfree_skb(skb);
905 		return ret;
906 	}
907 
908 	ret = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
909 			     n->nlmsg_flags & NLM_F_ECHO);
910 	if (ret > 0)
911 		return 0;
912 	return ret;
913 }
914 
915 static int
916 tca_action_gd(struct net *net, struct nlattr *nla, struct nlmsghdr *n,
917 	      u32 portid, int event)
918 {
919 	int i, ret;
920 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
921 	struct tc_action *act;
922 	LIST_HEAD(actions);
923 
924 	ret = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL);
925 	if (ret < 0)
926 		return ret;
927 
928 	if (event == RTM_DELACTION && n->nlmsg_flags & NLM_F_ROOT) {
929 		if (tb[1] != NULL)
930 			return tca_action_flush(net, tb[1], n, portid);
931 		else
932 			return -EINVAL;
933 	}
934 
935 	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
936 		act = tcf_action_get_1(net, tb[i], n, portid);
937 		if (IS_ERR(act)) {
938 			ret = PTR_ERR(act);
939 			goto err;
940 		}
941 		act->order = i;
942 		list_add_tail(&act->list, &actions);
943 	}
944 
945 	if (event == RTM_GETACTION)
946 		ret = act_get_notify(net, portid, n, &actions, event);
947 	else { /* delete */
948 		ret = tcf_del_notify(net, n, &actions, portid);
949 		if (ret)
950 			goto err;
951 		return ret;
952 	}
953 err:
954 	if (event != RTM_GETACTION)
955 		tcf_action_destroy(&actions, 0);
956 	return ret;
957 }
958 
959 static int
960 tcf_add_notify(struct net *net, struct nlmsghdr *n, struct list_head *actions,
961 	       u32 portid)
962 {
963 	struct sk_buff *skb;
964 	int err = 0;
965 
966 	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
967 	if (!skb)
968 		return -ENOBUFS;
969 
970 	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, n->nlmsg_flags,
971 			 RTM_NEWACTION, 0, 0) <= 0) {
972 		kfree_skb(skb);
973 		return -EINVAL;
974 	}
975 
976 	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
977 			     n->nlmsg_flags & NLM_F_ECHO);
978 	if (err > 0)
979 		err = 0;
980 	return err;
981 }
982 
983 static int tcf_action_add(struct net *net, struct nlattr *nla,
984 			  struct nlmsghdr *n, u32 portid, int ovr)
985 {
986 	int ret = 0;
987 	LIST_HEAD(actions);
988 
989 	ret = tcf_action_init(net, nla, NULL, NULL, ovr, 0, &actions);
990 	if (ret)
991 		return ret;
992 
993 	return tcf_add_notify(net, n, &actions, portid);
994 }
995 
996 static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n)
997 {
998 	struct net *net = sock_net(skb->sk);
999 	struct nlattr *tca[TCA_ACT_MAX + 1];
1000 	u32 portid = skb ? NETLINK_CB(skb).portid : 0;
1001 	int ret = 0, ovr = 0;
1002 
1003 	if ((n->nlmsg_type != RTM_GETACTION) &&
1004 	    !netlink_capable(skb, CAP_NET_ADMIN))
1005 		return -EPERM;
1006 
1007 	ret = nlmsg_parse(n, sizeof(struct tcamsg), tca, TCA_ACT_MAX, NULL);
1008 	if (ret < 0)
1009 		return ret;
1010 
1011 	if (tca[TCA_ACT_TAB] == NULL) {
1012 		pr_notice("tc_ctl_action: received NO action attribs\n");
1013 		return -EINVAL;
1014 	}
1015 
1016 	/* n->nlmsg_flags & NLM_F_CREATE */
1017 	switch (n->nlmsg_type) {
1018 	case RTM_NEWACTION:
1019 		/* we are going to assume all other flags
1020 		 * imply create only if it doesn't exist
1021 		 * Note that CREATE | EXCL implies that
1022 		 * but since we want avoid ambiguity (eg when flags
1023 		 * is zero) then just set this
1024 		 */
1025 		if (n->nlmsg_flags & NLM_F_REPLACE)
1026 			ovr = 1;
1027 replay:
1028 		ret = tcf_action_add(net, tca[TCA_ACT_TAB], n, portid, ovr);
1029 		if (ret == -EAGAIN)
1030 			goto replay;
1031 		break;
1032 	case RTM_DELACTION:
1033 		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1034 				    portid, RTM_DELACTION);
1035 		break;
1036 	case RTM_GETACTION:
1037 		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1038 				    portid, RTM_GETACTION);
1039 		break;
1040 	default:
1041 		BUG();
1042 	}
1043 
1044 	return ret;
1045 }
1046 
1047 static struct nlattr *find_dump_kind(const struct nlmsghdr *n)
1048 {
1049 	struct nlattr *tb1, *tb2[TCA_ACT_MAX + 1];
1050 	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1051 	struct nlattr *nla[TCAA_MAX + 1];
1052 	struct nlattr *kind;
1053 
1054 	if (nlmsg_parse(n, sizeof(struct tcamsg), nla, TCAA_MAX, NULL) < 0)
1055 		return NULL;
1056 	tb1 = nla[TCA_ACT_TAB];
1057 	if (tb1 == NULL)
1058 		return NULL;
1059 
1060 	if (nla_parse(tb, TCA_ACT_MAX_PRIO, nla_data(tb1),
1061 		      NLMSG_ALIGN(nla_len(tb1)), NULL) < 0)
1062 		return NULL;
1063 
1064 	if (tb[1] == NULL)
1065 		return NULL;
1066 	if (nla_parse_nested(tb2, TCA_ACT_MAX, tb[1], NULL) < 0)
1067 		return NULL;
1068 	kind = tb2[TCA_ACT_KIND];
1069 
1070 	return kind;
1071 }
1072 
1073 static int tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb)
1074 {
1075 	struct net *net = sock_net(skb->sk);
1076 	struct nlmsghdr *nlh;
1077 	unsigned char *b = skb_tail_pointer(skb);
1078 	struct nlattr *nest;
1079 	struct tc_action_ops *a_o;
1080 	int ret = 0;
1081 	struct tcamsg *t = (struct tcamsg *) nlmsg_data(cb->nlh);
1082 	struct nlattr *kind = find_dump_kind(cb->nlh);
1083 
1084 	if (kind == NULL) {
1085 		pr_info("tc_dump_action: action bad kind\n");
1086 		return 0;
1087 	}
1088 
1089 	a_o = tc_lookup_action(kind);
1090 	if (a_o == NULL)
1091 		return 0;
1092 
1093 	nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
1094 			cb->nlh->nlmsg_type, sizeof(*t), 0);
1095 	if (!nlh)
1096 		goto out_module_put;
1097 	t = nlmsg_data(nlh);
1098 	t->tca_family = AF_UNSPEC;
1099 	t->tca__pad1 = 0;
1100 	t->tca__pad2 = 0;
1101 
1102 	nest = nla_nest_start(skb, TCA_ACT_TAB);
1103 	if (nest == NULL)
1104 		goto out_module_put;
1105 
1106 	ret = a_o->walk(net, skb, cb, RTM_GETACTION, a_o);
1107 	if (ret < 0)
1108 		goto out_module_put;
1109 
1110 	if (ret > 0) {
1111 		nla_nest_end(skb, nest);
1112 		ret = skb->len;
1113 	} else
1114 		nlmsg_trim(skb, b);
1115 
1116 	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1117 	if (NETLINK_CB(cb->skb).portid && ret)
1118 		nlh->nlmsg_flags |= NLM_F_MULTI;
1119 	module_put(a_o->owner);
1120 	return skb->len;
1121 
1122 out_module_put:
1123 	module_put(a_o->owner);
1124 	nlmsg_trim(skb, b);
1125 	return skb->len;
1126 }
1127 
1128 static int __init tc_action_init(void)
1129 {
1130 	rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL, NULL);
1131 	rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL, NULL);
1132 	rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action,
1133 		      NULL);
1134 
1135 	return 0;
1136 }
1137 
1138 subsys_initcall(tc_action_init);
1139