act_api.c 36.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * net/sched/act_api.c	Packet action API.
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 *
 * Author:	Jamal Hadi Salim
 *
 *
 */

#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/string.h>
#include <linux/errno.h>
18
#include <linux/slab.h>
L
Linus Torvalds 已提交
19 20 21
#include <linux/skbuff.h>
#include <linux/init.h>
#include <linux/kmod.h>
22
#include <linux/err.h>
23
#include <linux/module.h>
24 25
#include <linux/rhashtable.h>
#include <linux/list.h>
26 27
#include <net/net_namespace.h>
#include <net/sock.h>
L
Linus Torvalds 已提交
28
#include <net/sch_generic.h>
29
#include <net/pkt_cls.h>
L
Linus Torvalds 已提交
30
#include <net/act_api.h>
31
#include <net/netlink.h>
L
Linus Torvalds 已提交
32

33 34 35 36 37 38
static int tcf_action_goto_chain_init(struct tc_action *a, struct tcf_proto *tp)
{
	u32 chain_index = a->tcfa_action & TC_ACT_EXT_VAL_MASK;

	if (!tp)
		return -EINVAL;
39
	a->goto_chain = tcf_chain_get(tp->chain->block, chain_index, true);
40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
	if (!a->goto_chain)
		return -ENOMEM;
	return 0;
}

static void tcf_action_goto_chain_fini(struct tc_action *a)
{
	tcf_chain_put(a->goto_chain);
}

static void tcf_action_goto_chain_exec(const struct tc_action *a,
				       struct tcf_result *res)
{
	const struct tcf_chain *chain = a->goto_chain;

	res->goto_tp = rcu_dereference_bh(chain->filter_chain);
}

58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
static void tcf_free_cookie_rcu(struct rcu_head *p)
{
	struct tc_cookie *cookie = container_of(p, struct tc_cookie, rcu);

	kfree(cookie->data);
	kfree(cookie);
}

static void tcf_set_action_cookie(struct tc_cookie __rcu **old_cookie,
				  struct tc_cookie *new_cookie)
{
	struct tc_cookie *old;

	old = xchg(old_cookie, new_cookie);
	if (old)
		call_rcu(&old->rcu, tcf_free_cookie_rcu);
}

C
Cong Wang 已提交
76 77 78 79 80 81
/* XXX: For standalone actions, we don't need a RCU grace period either, because
 * actions are always connected to filters and filters are already destroyed in
 * RCU callbacks, so after a RCU grace period actions are already disconnected
 * from filters. Readers later can not find us.
 */
static void free_tcf(struct tc_action *p)
82 83 84
{
	free_percpu(p->cpu_bstats);
	free_percpu(p->cpu_qstats);
85

86
	tcf_set_action_cookie(&p->act_cookie, NULL);
87 88
	if (p->goto_chain)
		tcf_action_goto_chain_fini(p);
89

90 91 92
	kfree(p);
}

93
static void tcf_idr_remove(struct tcf_idrinfo *idrinfo, struct tc_action *p)
94
{
95
	spin_lock(&idrinfo->lock);
96
	idr_remove(&idrinfo->action_idr, p->tcfa_index);
97
	spin_unlock(&idrinfo->lock);
98
	gen_kill_estimator(&p->tcfa_rate_est);
C
Cong Wang 已提交
99
	free_tcf(p);
100 101
}

102
int __tcf_idr_release(struct tc_action *p, bool bind, bool strict)
103 104 105
{
	int ret = 0;

106 107
	ASSERT_RTNL();

108 109 110 111 112 113 114 115 116 117 118 119
	/* Release with strict==1 and bind==0 is only called through act API
	 * interface (classifiers always bind). Only case when action with
	 * positive reference count and zero bind count can exist is when it was
	 * also created with act API (unbinding last classifier will destroy the
	 * action if it was created by classifier). So only case when bind count
	 * can be changed after initial check is when unbound action is
	 * destroyed by act API while classifier binds to action with same id
	 * concurrently. This result either creation of new action(same behavior
	 * as before), or reusing existing action if concurrent process
	 * increments reference count before action is deleted. Both scenarios
	 * are acceptable.
	 */
120 121
	if (p) {
		if (bind)
122 123
			atomic_dec(&p->tcfa_bindcnt);
		else if (strict && atomic_read(&p->tcfa_bindcnt) > 0)
124
			return -EPERM;
125

126 127
		if (atomic_read(&p->tcfa_bindcnt) <= 0 &&
		    refcount_dec_and_test(&p->tcfa_refcnt)) {
128
			if (p->ops->cleanup)
129
				p->ops->cleanup(p);
130
			tcf_idr_remove(p->idrinfo, p);
131
			ret = ACT_P_DELETED;
132 133
		}
	}
134

135 136
	return ret;
}
137
EXPORT_SYMBOL(__tcf_idr_release);
138

139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
static size_t tcf_action_shared_attrs_size(const struct tc_action *act)
{
	u32 cookie_len = 0;

	if (act->act_cookie)
		cookie_len = nla_total_size(act->act_cookie->len);

	return  nla_total_size(0) /* action number nested */
		+ nla_total_size(IFNAMSIZ) /* TCA_ACT_KIND */
		+ cookie_len /* TCA_ACT_COOKIE */
		+ nla_total_size(0) /* TCA_ACT_STATS nested */
		/* TCA_STATS_BASIC */
		+ nla_total_size_64bit(sizeof(struct gnet_stats_basic))
		/* TCA_STATS_QUEUE */
		+ nla_total_size_64bit(sizeof(struct gnet_stats_queue))
		+ nla_total_size(0) /* TCA_OPTIONS nested */
		+ nla_total_size(sizeof(struct tcf_t)); /* TCA_GACT_TM */
}

static size_t tcf_action_full_attrs_size(size_t sz)
{
	return NLMSG_HDRLEN                     /* struct nlmsghdr */
		+ sizeof(struct tcamsg)
		+ nla_total_size(0)             /* TCA_ACT_TAB nested */
		+ sz;
}

static size_t tcf_action_fill_size(const struct tc_action *act)
{
	size_t sz = tcf_action_shared_attrs_size(act);

	if (act->ops->get_fill_size)
		return act->ops->get_fill_size(act) + sz;
	return sz;
}

175
static int tcf_dump_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb,
176
			   struct netlink_callback *cb)
177
{
178
	int err = 0, index = -1, s_i = 0, n_i = 0;
179
	u32 act_flags = cb->args[2];
180
	unsigned long jiffy_since = cb->args[3];
181
	struct nlattr *nest;
182 183 184
	struct idr *idr = &idrinfo->action_idr;
	struct tc_action *p;
	unsigned long id = 1;
185

186
	spin_lock(&idrinfo->lock);
187 188 189

	s_i = cb->args[0];

190
	idr_for_each_entry_ul(idr, p, id) {
191 192 193 194 195 196 197 198 199 200
		index++;
		if (index < s_i)
			continue;

		if (jiffy_since &&
		    time_after(jiffy_since,
			       (unsigned long)p->tcfa_tm.lastuse))
			continue;

		nest = nla_nest_start(skb, n_i);
201 202
		if (!nest) {
			index--;
203
			goto nla_put_failure;
204
		}
205 206 207 208 209
		err = tcf_action_dump_1(skb, p, 0, 0);
		if (err < 0) {
			index--;
			nlmsg_trim(skb, nest);
			goto done;
210
		}
211 212 213 214 215
		nla_nest_end(skb, nest);
		n_i++;
		if (!(act_flags & TCA_FLAG_LARGE_DUMP_ON) &&
		    n_i >= TCA_ACT_MAX_PRIO)
			goto done;
216 217
	}
done:
218 219 220
	if (index >= 0)
		cb->args[0] = index + 1;

221
	spin_unlock(&idrinfo->lock);
222 223 224 225
	if (n_i) {
		if (act_flags & TCA_FLAG_LARGE_DUMP_ON)
			cb->args[1] = n_i;
	}
226 227
	return n_i;

228
nla_put_failure:
229
	nla_nest_cancel(skb, nest);
230 231 232
	goto done;
}

233
static int tcf_del_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb,
234
			  const struct tc_action_ops *ops)
235
{
236
	struct nlattr *nest;
237
	int n_i = 0;
238
	int ret = -EINVAL;
239 240 241
	struct idr *idr = &idrinfo->action_idr;
	struct tc_action *p;
	unsigned long id = 1;
242

243
	nest = nla_nest_start(skb, 0);
244 245
	if (nest == NULL)
		goto nla_put_failure;
246
	if (nla_put_string(skb, TCA_KIND, ops->kind))
247
		goto nla_put_failure;
248

249
	idr_for_each_entry_ul(idr, p, id) {
250 251
		ret = __tcf_idr_release(p, false, true);
		if (ret == ACT_P_DELETED) {
252
			module_put(ops->owner);
253 254 255
			n_i++;
		} else if (ret < 0) {
			goto nla_put_failure;
256 257
		}
	}
258 259
	if (nla_put_u32(skb, TCA_FCNT, n_i))
		goto nla_put_failure;
260
	nla_nest_end(skb, nest);
261 262

	return n_i;
263
nla_put_failure:
264
	nla_nest_cancel(skb, nest);
265
	return ret;
266 267
}

268 269
int tcf_generic_walker(struct tc_action_net *tn, struct sk_buff *skb,
		       struct netlink_callback *cb, int type,
270 271
		       const struct tc_action_ops *ops,
		       struct netlink_ext_ack *extack)
272
{
273
	struct tcf_idrinfo *idrinfo = tn->idrinfo;
274

275
	if (type == RTM_DELACTION) {
276
		return tcf_del_walker(idrinfo, skb, ops);
277
	} else if (type == RTM_GETACTION) {
278
		return tcf_dump_walker(idrinfo, skb, cb);
279
	} else {
280 281
		WARN(1, "tcf_generic_walker: unknown command %d\n", type);
		NL_SET_ERR_MSG(extack, "tcf_generic_walker: unknown command");
282 283 284
		return -EINVAL;
	}
}
285
EXPORT_SYMBOL(tcf_generic_walker);
286

287 288
static bool __tcf_idr_check(struct tc_action_net *tn, u32 index,
			    struct tc_action **a, int bind)
289
{
290 291
	struct tcf_idrinfo *idrinfo = tn->idrinfo;
	struct tc_action *p;
292

293
	spin_lock(&idrinfo->lock);
294
	p = idr_find(&idrinfo->action_idr, index);
295 296 297 298 299
	if (p) {
		refcount_inc(&p->tcfa_refcnt);
		if (bind)
			atomic_inc(&p->tcfa_bindcnt);
	}
300
	spin_unlock(&idrinfo->lock);
301

302 303 304 305 306
	if (p) {
		*a = p;
		return true;
	}
	return false;
307 308
}

309
int tcf_idr_search(struct tc_action_net *tn, struct tc_action **a, u32 index)
310
{
311
	return __tcf_idr_check(tn, index, a, 0);
312
}
313
EXPORT_SYMBOL(tcf_idr_search);
314

315 316
bool tcf_idr_check(struct tc_action_net *tn, u32 index, struct tc_action **a,
		   int bind)
317
{
318
	return __tcf_idr_check(tn, index, a, bind);
319
}
320
EXPORT_SYMBOL(tcf_idr_check);
321

322 323 324
int 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)
325
{
326
	struct tc_action *p = kzalloc(ops->size, GFP_KERNEL);
327 328
	struct tcf_idrinfo *idrinfo = tn->idrinfo;
	struct idr *idr = &idrinfo->action_idr;
329
	int err = -ENOMEM;
330 331

	if (unlikely(!p))
332
		return -ENOMEM;
333
	refcount_set(&p->tcfa_refcnt, 1);
334
	if (bind)
335
		atomic_set(&p->tcfa_bindcnt, 1);
336

337 338
	if (cpustats) {
		p->cpu_bstats = netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu);
339
		if (!p->cpu_bstats)
340
			goto err1;
341 342 343
		p->cpu_qstats = alloc_percpu(struct gnet_stats_queue);
		if (!p->cpu_qstats)
			goto err2;
344
	}
345
	spin_lock_init(&p->tcfa_lock);
346
	idr_preload(GFP_KERNEL);
347
	spin_lock(&idrinfo->lock);
348 349
	/* user doesn't specify an index */
	if (!index) {
350 351
		index = 1;
		err = idr_alloc_u32(idr, NULL, &index, UINT_MAX, GFP_ATOMIC);
352
	} else {
353
		err = idr_alloc_u32(idr, NULL, &index, index, GFP_ATOMIC);
354
	}
355
	spin_unlock(&idrinfo->lock);
356 357 358
	idr_preload_end();
	if (err)
		goto err3;
359

360
	p->tcfa_index = index;
361 362 363
	p->tcfa_tm.install = jiffies;
	p->tcfa_tm.lastuse = jiffies;
	p->tcfa_tm.firstuse = 0;
364
	if (est) {
365 366 367
		err = gen_new_estimator(&p->tcfa_bstats, p->cpu_bstats,
					&p->tcfa_rate_est,
					&p->tcfa_lock, NULL, est);
368 369
		if (err)
			goto err4;
370 371
	}

372
	p->idrinfo = idrinfo;
373 374 375
	p->ops = ops;
	INIT_LIST_HEAD(&p->list);
	*a = p;
376
	return 0;
377 378 379 380 381 382 383 384 385
err4:
	idr_remove(idr, index);
err3:
	free_percpu(p->cpu_qstats);
err2:
	free_percpu(p->cpu_bstats);
err1:
	kfree(p);
	return err;
386
}
387
EXPORT_SYMBOL(tcf_idr_create);
388

389
void tcf_idr_insert(struct tc_action_net *tn, struct tc_action *a)
390
{
391
	struct tcf_idrinfo *idrinfo = tn->idrinfo;
392

393
	spin_lock(&idrinfo->lock);
394
	idr_replace(&idrinfo->action_idr, a, a->tcfa_index);
395
	spin_unlock(&idrinfo->lock);
396
}
397
EXPORT_SYMBOL(tcf_idr_insert);
L
Linus Torvalds 已提交
398

399 400
void tcf_idrinfo_destroy(const struct tc_action_ops *ops,
			 struct tcf_idrinfo *idrinfo)
401
{
402 403 404 405
	struct idr *idr = &idrinfo->action_idr;
	struct tc_action *p;
	int ret;
	unsigned long id = 1;
406

407
	idr_for_each_entry_ul(idr, p, id) {
408 409 410 411 412
		ret = __tcf_idr_release(p, false, true);
		if (ret == ACT_P_DELETED)
			module_put(ops->owner);
		else if (ret < 0)
			return;
413
	}
414
	idr_destroy(&idrinfo->action_idr);
415
}
416
EXPORT_SYMBOL(tcf_idrinfo_destroy);
417

418
static LIST_HEAD(act_base);
L
Linus Torvalds 已提交
419 420
static DEFINE_RWLOCK(act_mod_lock);

421 422
int tcf_register_action(struct tc_action_ops *act,
			struct pernet_operations *ops)
L
Linus Torvalds 已提交
423
{
424
	struct tc_action_ops *a;
425
	int ret;
L
Linus Torvalds 已提交
426

427
	if (!act->act || !act->dump || !act->init || !act->walk || !act->lookup)
428 429
		return -EINVAL;

430 431 432 433 434 435 436 437
	/* We have to register pernet ops before making the action ops visible,
	 * otherwise tcf_action_init_1() could get a partially initialized
	 * netns.
	 */
	ret = register_pernet_subsys(ops);
	if (ret)
		return ret;

L
Linus Torvalds 已提交
438
	write_lock(&act_mod_lock);
439
	list_for_each_entry(a, &act_base, head) {
L
Linus Torvalds 已提交
440 441
		if (act->type == a->type || (strcmp(act->kind, a->kind) == 0)) {
			write_unlock(&act_mod_lock);
442
			unregister_pernet_subsys(ops);
L
Linus Torvalds 已提交
443 444 445
			return -EEXIST;
		}
	}
446
	list_add_tail(&act->head, &act_base);
L
Linus Torvalds 已提交
447
	write_unlock(&act_mod_lock);
448

L
Linus Torvalds 已提交
449 450
	return 0;
}
451
EXPORT_SYMBOL(tcf_register_action);
L
Linus Torvalds 已提交
452

453 454
int tcf_unregister_action(struct tc_action_ops *act,
			  struct pernet_operations *ops)
L
Linus Torvalds 已提交
455
{
456
	struct tc_action_ops *a;
L
Linus Torvalds 已提交
457 458 459
	int err = -ENOENT;

	write_lock(&act_mod_lock);
460 461 462 463
	list_for_each_entry(a, &act_base, head) {
		if (a == act) {
			list_del(&act->head);
			err = 0;
L
Linus Torvalds 已提交
464
			break;
465
		}
L
Linus Torvalds 已提交
466 467
	}
	write_unlock(&act_mod_lock);
468 469
	if (!err)
		unregister_pernet_subsys(ops);
L
Linus Torvalds 已提交
470 471
	return err;
}
472
EXPORT_SYMBOL(tcf_unregister_action);
L
Linus Torvalds 已提交
473 474 475 476

/* lookup by name */
static struct tc_action_ops *tc_lookup_action_n(char *kind)
{
477
	struct tc_action_ops *a, *res = NULL;
L
Linus Torvalds 已提交
478 479 480

	if (kind) {
		read_lock(&act_mod_lock);
481
		list_for_each_entry(a, &act_base, head) {
L
Linus Torvalds 已提交
482
			if (strcmp(kind, a->kind) == 0) {
483 484
				if (try_module_get(a->owner))
					res = a;
L
Linus Torvalds 已提交
485 486 487 488 489
				break;
			}
		}
		read_unlock(&act_mod_lock);
	}
490
	return res;
L
Linus Torvalds 已提交
491 492
}

493 494
/* lookup by nlattr */
static struct tc_action_ops *tc_lookup_action(struct nlattr *kind)
L
Linus Torvalds 已提交
495
{
496
	struct tc_action_ops *a, *res = NULL;
L
Linus Torvalds 已提交
497 498 499

	if (kind) {
		read_lock(&act_mod_lock);
500
		list_for_each_entry(a, &act_base, head) {
501
			if (nla_strcmp(kind, a->kind) == 0) {
502 503
				if (try_module_get(a->owner))
					res = a;
L
Linus Torvalds 已提交
504 505 506 507 508
				break;
			}
		}
		read_unlock(&act_mod_lock);
	}
509
	return res;
L
Linus Torvalds 已提交
510 511
}

512 513
/*TCA_ACT_MAX_PRIO is 32, there count upto 32 */
#define TCA_ACT_MAX_PRIO_MASK 0x1FF
514 515
int tcf_action_exec(struct sk_buff *skb, struct tc_action **actions,
		    int nr_actions, struct tcf_result *res)
L
Linus Torvalds 已提交
516
{
517 518
	u32 jmp_prgcnt = 0;
	u32 jmp_ttl = TCA_ACT_MAX_PRIO; /*matches actions per filter */
519 520
	int i;
	int ret = TC_ACT_OK;
L
Linus Torvalds 已提交
521

522 523 524
	if (skb_skip_tc_classify(skb))
		return TC_ACT_OK;

525
restart_act_graph:
526 527 528
	for (i = 0; i < nr_actions; i++) {
		const struct tc_action *a = actions[i];

529 530 531 532
		if (jmp_prgcnt > 0) {
			jmp_prgcnt -= 1;
			continue;
		}
L
Linus Torvalds 已提交
533
repeat:
534 535 536
		ret = a->ops->act(skb, a, res);
		if (ret == TC_ACT_REPEAT)
			goto repeat;	/* we need a ttl - JHS */
537

538
		if (TC_ACT_EXT_CMP(ret, TC_ACT_JUMP)) {
539 540 541 542 543 544 545 546 547 548 549
			jmp_prgcnt = ret & TCA_ACT_MAX_PRIO_MASK;
			if (!jmp_prgcnt || (jmp_prgcnt > nr_actions)) {
				/* faulty opcode, stop pipeline */
				return TC_ACT_OK;
			} else {
				jmp_ttl -= 1;
				if (jmp_ttl > 0)
					goto restart_act_graph;
				else /* faulty graph, stop pipeline */
					return TC_ACT_OK;
			}
550 551
		} else if (TC_ACT_EXT_CMP(ret, TC_ACT_GOTO_CHAIN)) {
			tcf_action_goto_chain_exec(a, res);
552 553
		}

554
		if (ret != TC_ACT_PIPE)
555
			break;
L
Linus Torvalds 已提交
556
	}
557

L
Linus Torvalds 已提交
558 559
	return ret;
}
560
EXPORT_SYMBOL(tcf_action_exec);
L
Linus Torvalds 已提交
561

562
int tcf_action_destroy(struct list_head *actions, int bind)
L
Linus Torvalds 已提交
563
{
564
	const struct tc_action_ops *ops;
565
	struct tc_action *a, *tmp;
566
	int ret = 0;
L
Linus Torvalds 已提交
567

568
	list_for_each_entry_safe(a, tmp, actions, list) {
569
		ops = a->ops;
570
		ret = __tcf_idr_release(a, bind, true);
571
		if (ret == ACT_P_DELETED)
572
			module_put(ops->owner);
573 574
		else if (ret < 0)
			return ret;
L
Linus Torvalds 已提交
575
	}
576
	return ret;
L
Linus Torvalds 已提交
577 578 579 580 581 582 583 584 585 586 587 588
}

int
tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
{
	return a->ops->dump(skb, a, bind, ref);
}

int
tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
{
	int err = -EINVAL;
589
	unsigned char *b = skb_tail_pointer(skb);
590
	struct nlattr *nest;
591
	struct tc_cookie *cookie;
L
Linus Torvalds 已提交
592

593 594
	if (nla_put_string(skb, TCA_KIND, a->ops->kind))
		goto nla_put_failure;
L
Linus Torvalds 已提交
595
	if (tcf_action_copy_stats(skb, a, 0))
596
		goto nla_put_failure;
597 598 599 600 601 602

	rcu_read_lock();
	cookie = rcu_dereference(a->act_cookie);
	if (cookie) {
		if (nla_put(skb, TCA_ACT_COOKIE, cookie->len, cookie->data)) {
			rcu_read_unlock();
603
			goto nla_put_failure;
604
		}
605
	}
606
	rcu_read_unlock();
607

608 609 610
	nest = nla_nest_start(skb, TCA_OPTIONS);
	if (nest == NULL)
		goto nla_put_failure;
E
Eric Dumazet 已提交
611 612
	err = tcf_action_dump_old(skb, a, bind, ref);
	if (err > 0) {
613
		nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
614 615 616
		return err;
	}

617
nla_put_failure:
618
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
619 620
	return -1;
}
621
EXPORT_SYMBOL(tcf_action_dump_1);
L
Linus Torvalds 已提交
622

623 624
int tcf_action_dump(struct sk_buff *skb, struct list_head *actions,
		    int bind, int ref)
L
Linus Torvalds 已提交
625 626 627
{
	struct tc_action *a;
	int err = -EINVAL;
628
	struct nlattr *nest;
L
Linus Torvalds 已提交
629

630
	list_for_each_entry(a, actions, list) {
631 632 633
		nest = nla_nest_start(skb, a->order);
		if (nest == NULL)
			goto nla_put_failure;
L
Linus Torvalds 已提交
634 635
		err = tcf_action_dump_1(skb, a, bind, ref);
		if (err < 0)
636
			goto errout;
637
		nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
638 639 640 641
	}

	return 0;

642
nla_put_failure:
643 644
	err = -EINVAL;
errout:
645
	nla_nest_cancel(skb, nest);
646
	return err;
L
Linus Torvalds 已提交
647 648
}

649
static struct tc_cookie *nla_memdup_cookie(struct nlattr **tb)
650
{
651 652 653 654 655 656 657 658
	struct tc_cookie *c = kzalloc(sizeof(*c), GFP_KERNEL);
	if (!c)
		return NULL;

	c->data = nla_memdup(tb[TCA_ACT_COOKIE], GFP_KERNEL);
	if (!c->data) {
		kfree(c);
		return NULL;
659
	}
660
	c->len = nla_len(tb[TCA_ACT_COOKIE]);
661

662
	return c;
663 664
}

665 666
struct tc_action *tcf_action_init_1(struct net *net, struct tcf_proto *tp,
				    struct nlattr *nla, struct nlattr *est,
667
				    char *name, int ovr, int bind,
668
				    bool rtnl_held,
669
				    struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
670 671 672
{
	struct tc_action *a;
	struct tc_action_ops *a_o;
673
	struct tc_cookie *cookie = NULL;
L
Linus Torvalds 已提交
674
	char act_name[IFNAMSIZ];
E
Eric Dumazet 已提交
675
	struct nlattr *tb[TCA_ACT_MAX + 1];
676
	struct nlattr *kind;
677
	int err;
L
Linus Torvalds 已提交
678 679

	if (name == NULL) {
680
		err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack);
681
		if (err < 0)
L
Linus Torvalds 已提交
682
			goto err_out;
683
		err = -EINVAL;
684
		kind = tb[TCA_ACT_KIND];
685 686
		if (!kind) {
			NL_SET_ERR_MSG(extack, "TC action kind must be specified");
L
Linus Torvalds 已提交
687
			goto err_out;
688 689 690
		}
		if (nla_strlcpy(act_name, kind, IFNAMSIZ) >= IFNAMSIZ) {
			NL_SET_ERR_MSG(extack, "TC action name too long");
L
Linus Torvalds 已提交
691
			goto err_out;
692
		}
693 694 695
		if (tb[TCA_ACT_COOKIE]) {
			int cklen = nla_len(tb[TCA_ACT_COOKIE]);

696 697
			if (cklen > TC_COOKIE_MAX_SIZE) {
				NL_SET_ERR_MSG(extack, "TC cookie size above the maximum");
698
				goto err_out;
699
			}
700 701 702

			cookie = nla_memdup_cookie(tb);
			if (!cookie) {
703
				NL_SET_ERR_MSG(extack, "No memory to generate TC cookie");
704 705 706 707
				err = -ENOMEM;
				goto err_out;
			}
		}
L
Linus Torvalds 已提交
708
	} else {
709 710 711
		if (strlcpy(act_name, name, IFNAMSIZ) >= IFNAMSIZ) {
			NL_SET_ERR_MSG(extack, "TC action name too long");
			err = -EINVAL;
L
Linus Torvalds 已提交
712
			goto err_out;
713
		}
L
Linus Torvalds 已提交
714 715 716 717
	}

	a_o = tc_lookup_action_n(act_name);
	if (a_o == NULL) {
718
#ifdef CONFIG_MODULES
719 720
		if (rtnl_held)
			rtnl_unlock();
721
		request_module("act_%s", act_name);
722 723
		if (rtnl_held)
			rtnl_lock();
L
Linus Torvalds 已提交
724 725 726 727 728 729 730 731 732 733

		a_o = tc_lookup_action_n(act_name);

		/* We dropped the RTNL semaphore in order to
		 * perform the module load.  So, even if we
		 * succeeded in loading the module we have to
		 * tell the caller to replay the request.  We
		 * indicate this using -EAGAIN.
		 */
		if (a_o != NULL) {
734
			err = -EAGAIN;
L
Linus Torvalds 已提交
735 736 737
			goto err_mod;
		}
#endif
738
		NL_SET_ERR_MSG(extack, "Failed to load TC action module");
739
		err = -ENOENT;
L
Linus Torvalds 已提交
740 741 742 743 744
		goto err_out;
	}

	/* backward compatibility for policer */
	if (name == NULL)
745
		err = a_o->init(net, tb[TCA_ACT_OPTIONS], est, &a, ovr, bind,
746
				rtnl_held, extack);
L
Linus Torvalds 已提交
747
	else
748 749
		err = a_o->init(net, nla, est, &a, ovr, bind, rtnl_held,
				extack);
750
	if (err < 0)
751
		goto err_mod;
L
Linus Torvalds 已提交
752

753 754
	if (!name && tb[TCA_ACT_COOKIE])
		tcf_set_action_cookie(&a->act_cookie, cookie);
755

L
Linus Torvalds 已提交
756
	/* module count goes up only when brand new policy is created
E
Eric Dumazet 已提交
757 758 759
	 * if it exists and is only bound to in a_o->init() then
	 * ACT_P_CREATED is not returned (a zero is).
	 */
760
	if (err != ACT_P_CREATED)
L
Linus Torvalds 已提交
761 762
		module_put(a_o->owner);

763 764 765 766 767 768 769
	if (TC_ACT_EXT_CMP(a->tcfa_action, TC_ACT_GOTO_CHAIN)) {
		err = tcf_action_goto_chain_init(a, tp);
		if (err) {
			LIST_HEAD(actions);

			list_add_tail(&a->list, &actions);
			tcf_action_destroy(&actions, bind);
770
			NL_SET_ERR_MSG(extack, "Failed to init TC action chain");
771 772 773 774
			return ERR_PTR(err);
		}
	}

L
Linus Torvalds 已提交
775 776 777 778 779
	return a;

err_mod:
	module_put(a_o->owner);
err_out:
780 781 782 783
	if (cookie) {
		kfree(cookie->data);
		kfree(cookie);
	}
784
	return ERR_PTR(err);
L
Linus Torvalds 已提交
785 786
}

787 788 789 790 791 792 793 794
static void cleanup_a(struct list_head *actions, int ovr)
{
	struct tc_action *a;

	if (!ovr)
		return;

	list_for_each_entry(a, actions, list)
795
		refcount_dec(&a->tcfa_refcnt);
796 797
}

798 799
int tcf_action_init(struct net *net, struct tcf_proto *tp, struct nlattr *nla,
		    struct nlattr *est, char *name, int ovr, int bind,
800
		    struct list_head *actions, size_t *attr_size,
801
		    bool rtnl_held, struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
802
{
E
Eric Dumazet 已提交
803
	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
804
	struct tc_action *act;
805
	size_t sz = 0;
806
	int err;
L
Linus Torvalds 已提交
807 808
	int i;

809
	err = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL, extack);
810
	if (err < 0)
811
		return err;
L
Linus Torvalds 已提交
812

813
	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
814
		act = tcf_action_init_1(net, tp, tb[i], est, name, ovr, bind,
815
					rtnl_held, extack);
816 817
		if (IS_ERR(act)) {
			err = PTR_ERR(act);
L
Linus Torvalds 已提交
818
			goto err;
819
		}
820
		act->order = i;
821
		sz += tcf_action_fill_size(act);
822
		if (ovr)
823
			refcount_inc(&act->tcfa_refcnt);
824
		list_add_tail(&act->list, actions);
L
Linus Torvalds 已提交
825
	}
826

827 828
	*attr_size = tcf_action_full_attrs_size(sz);

829 830 831 832
	/* Remove the temp refcnt which was necessary to protect against
	 * destroying an existing action which was being replaced
	 */
	cleanup_a(actions, ovr);
833
	return 0;
L
Linus Torvalds 已提交
834 835

err:
836 837
	tcf_action_destroy(actions, bind);
	return err;
L
Linus Torvalds 已提交
838 839
}

840
int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *p,
L
Linus Torvalds 已提交
841 842 843 844
			  int compat_mode)
{
	int err = 0;
	struct gnet_dump d;
845

846
	if (p == NULL)
L
Linus Torvalds 已提交
847 848 849
		goto errout;

	/* compat_mode being true specifies a call that is supposed
850
	 * to add additional backward compatibility statistic TLVs.
L
Linus Torvalds 已提交
851 852
	 */
	if (compat_mode) {
853
		if (p->type == TCA_OLD_COMPAT)
L
Linus Torvalds 已提交
854
			err = gnet_stats_start_copy_compat(skb, 0,
855 856
							   TCA_STATS,
							   TCA_XSTATS,
857
							   &p->tcfa_lock, &d,
858
							   TCA_PAD);
L
Linus Torvalds 已提交
859 860 861 862
		else
			return 0;
	} else
		err = gnet_stats_start_copy(skb, TCA_ACT_STATS,
863
					    &p->tcfa_lock, &d, TCA_ACT_PAD);
L
Linus Torvalds 已提交
864 865 866 867

	if (err < 0)
		goto errout;

868
	if (gnet_stats_copy_basic(NULL, &d, p->cpu_bstats, &p->tcfa_bstats) < 0 ||
869
	    gnet_stats_copy_rate_est(&d, &p->tcfa_rate_est) < 0 ||
870
	    gnet_stats_copy_queue(&d, p->cpu_qstats,
871 872
				  &p->tcfa_qstats,
				  p->tcfa_qstats.qlen) < 0)
L
Linus Torvalds 已提交
873 874 875 876 877 878 879 880 881 882 883
		goto errout;

	if (gnet_stats_finish_copy(&d) < 0)
		goto errout;

	return 0;

errout:
	return -1;
}

884 885 886
static int tca_get_fill(struct sk_buff *skb, struct list_head *actions,
			u32 portid, u32 seq, u16 flags, int event, int bind,
			int ref)
L
Linus Torvalds 已提交
887 888 889
{
	struct tcamsg *t;
	struct nlmsghdr *nlh;
890
	unsigned char *b = skb_tail_pointer(skb);
891
	struct nlattr *nest;
L
Linus Torvalds 已提交
892

893
	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags);
894 895 896
	if (!nlh)
		goto out_nlmsg_trim;
	t = nlmsg_data(nlh);
L
Linus Torvalds 已提交
897
	t->tca_family = AF_UNSPEC;
898 899
	t->tca__pad1 = 0;
	t->tca__pad2 = 0;
900

901
	nest = nla_nest_start(skb, TCA_ACT_TAB);
902
	if (!nest)
903
		goto out_nlmsg_trim;
L
Linus Torvalds 已提交
904

905
	if (tcf_action_dump(skb, actions, bind, ref) < 0)
906
		goto out_nlmsg_trim;
L
Linus Torvalds 已提交
907

908
	nla_nest_end(skb, nest);
909

910
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
L
Linus Torvalds 已提交
911 912
	return skb->len;

913
out_nlmsg_trim:
914
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
915 916 917 918
	return -1;
}

static int
919
tcf_get_notify(struct net *net, u32 portid, struct nlmsghdr *n,
920 921
	       struct list_head *actions, int event,
	       struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
922 923 924 925 926 927
{
	struct sk_buff *skb;

	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
	if (!skb)
		return -ENOBUFS;
928
	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, event,
929
			 0, 1) <= 0) {
930
		NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action");
L
Linus Torvalds 已提交
931 932 933
		kfree_skb(skb);
		return -EINVAL;
	}
934

935
	return rtnl_unicast(skb, net, portid);
L
Linus Torvalds 已提交
936 937
}

938
static struct tc_action *tcf_action_get_1(struct net *net, struct nlattr *nla,
939 940
					  struct nlmsghdr *n, u32 portid,
					  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
941
{
E
Eric Dumazet 已提交
942
	struct nlattr *tb[TCA_ACT_MAX + 1];
943
	const struct tc_action_ops *ops;
L
Linus Torvalds 已提交
944 945
	struct tc_action *a;
	int index;
946
	int err;
L
Linus Torvalds 已提交
947

948
	err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack);
949
	if (err < 0)
950
		goto err_out;
L
Linus Torvalds 已提交
951

952
	err = -EINVAL;
953
	if (tb[TCA_ACT_INDEX] == NULL ||
954 955
	    nla_len(tb[TCA_ACT_INDEX]) < sizeof(index)) {
		NL_SET_ERR_MSG(extack, "Invalid TC action index value");
956
		goto err_out;
957
	}
958
	index = nla_get_u32(tb[TCA_ACT_INDEX]);
L
Linus Torvalds 已提交
959

960
	err = -EINVAL;
961
	ops = tc_lookup_action(tb[TCA_ACT_KIND]);
962 963
	if (!ops) { /* could happen in batch of actions */
		NL_SET_ERR_MSG(extack, "Specified TC action not found");
964
		goto err_out;
965
	}
966
	err = -ENOENT;
967
	if (ops->lookup(net, &a, index, extack) == 0)
L
Linus Torvalds 已提交
968 969
		goto err_mod;

970
	module_put(ops->owner);
L
Linus Torvalds 已提交
971
	return a;
972

L
Linus Torvalds 已提交
973
err_mod:
974
	module_put(ops->owner);
975 976
err_out:
	return ERR_PTR(err);
L
Linus Torvalds 已提交
977 978
}

979
static int tca_action_flush(struct net *net, struct nlattr *nla,
980 981
			    struct nlmsghdr *n, u32 portid,
			    struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
982 983 984 985 986 987
{
	struct sk_buff *skb;
	unsigned char *b;
	struct nlmsghdr *nlh;
	struct tcamsg *t;
	struct netlink_callback dcb;
988
	struct nlattr *nest;
E
Eric Dumazet 已提交
989
	struct nlattr *tb[TCA_ACT_MAX + 1];
990
	const struct tc_action_ops *ops;
991
	struct nlattr *kind;
992
	int err = -ENOMEM;
L
Linus Torvalds 已提交
993 994

	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
995
	if (!skb)
996
		return err;
L
Linus Torvalds 已提交
997

998
	b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
999

1000
	err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack);
1001
	if (err < 0)
L
Linus Torvalds 已提交
1002 1003
		goto err_out;

1004
	err = -EINVAL;
1005
	kind = tb[TCA_ACT_KIND];
1006
	ops = tc_lookup_action(kind);
1007 1008
	if (!ops) { /*some idjot trying to flush unknown action */
		NL_SET_ERR_MSG(extack, "Cannot flush unknown TC action");
L
Linus Torvalds 已提交
1009
		goto err_out;
1010
	}
L
Linus Torvalds 已提交
1011

1012 1013
	nlh = nlmsg_put(skb, portid, n->nlmsg_seq, RTM_DELACTION,
			sizeof(*t), 0);
1014 1015
	if (!nlh) {
		NL_SET_ERR_MSG(extack, "Failed to create TC action flush notification");
1016
		goto out_module_put;
1017
	}
1018
	t = nlmsg_data(nlh);
L
Linus Torvalds 已提交
1019
	t->tca_family = AF_UNSPEC;
1020 1021
	t->tca__pad1 = 0;
	t->tca__pad2 = 0;
L
Linus Torvalds 已提交
1022

1023
	nest = nla_nest_start(skb, TCA_ACT_TAB);
1024 1025
	if (!nest) {
		NL_SET_ERR_MSG(extack, "Failed to add new netlink message");
1026
		goto out_module_put;
1027
	}
L
Linus Torvalds 已提交
1028

1029
	err = ops->walk(net, skb, &dcb, RTM_DELACTION, ops, extack);
1030 1031
	if (err <= 0) {
		nla_nest_cancel(skb, nest);
1032
		goto out_module_put;
1033
	}
L
Linus Torvalds 已提交
1034

1035
	nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
1036

1037
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
L
Linus Torvalds 已提交
1038
	nlh->nlmsg_flags |= NLM_F_ROOT;
1039
	module_put(ops->owner);
1040
	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
E
Eric Dumazet 已提交
1041
			     n->nlmsg_flags & NLM_F_ECHO);
L
Linus Torvalds 已提交
1042 1043
	if (err > 0)
		return 0;
1044 1045
	if (err < 0)
		NL_SET_ERR_MSG(extack, "Failed to send TC action flush notification");
L
Linus Torvalds 已提交
1046 1047 1048

	return err;

1049
out_module_put:
1050
	module_put(ops->owner);
L
Linus Torvalds 已提交
1051 1052 1053 1054 1055
err_out:
	kfree_skb(skb);
	return err;
}

1056 1057
static int
tcf_del_notify(struct net *net, struct nlmsghdr *n, struct list_head *actions,
1058
	       u32 portid, size_t attr_size, struct netlink_ext_ack *extack)
1059 1060 1061 1062
{
	int ret;
	struct sk_buff *skb;

1063 1064
	skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size,
			GFP_KERNEL);
1065 1066 1067 1068
	if (!skb)
		return -ENOBUFS;

	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, RTM_DELACTION,
1069
			 0, 2) <= 0) {
1070
		NL_SET_ERR_MSG(extack, "Failed to fill netlink TC action attributes");
1071 1072 1073 1074 1075
		kfree_skb(skb);
		return -EINVAL;
	}

	/* now do the delete */
1076 1077
	ret = tcf_action_destroy(actions, 0);
	if (ret < 0) {
1078
		NL_SET_ERR_MSG(extack, "Failed to delete TC action");
1079 1080 1081
		kfree_skb(skb);
		return ret;
	}
1082 1083 1084 1085 1086 1087 1088 1089

	ret = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
			     n->nlmsg_flags & NLM_F_ECHO);
	if (ret > 0)
		return 0;
	return ret;
}

L
Linus Torvalds 已提交
1090
static int
1091
tca_action_gd(struct net *net, struct nlattr *nla, struct nlmsghdr *n,
1092
	      u32 portid, int event, struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1093
{
1094
	int i, ret;
E
Eric Dumazet 已提交
1095
	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1096
	struct tc_action *act;
1097
	size_t attr_size = 0;
1098
	LIST_HEAD(actions);
L
Linus Torvalds 已提交
1099

1100
	ret = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL, extack);
1101 1102
	if (ret < 0)
		return ret;
L
Linus Torvalds 已提交
1103

E
Eric Dumazet 已提交
1104
	if (event == RTM_DELACTION && n->nlmsg_flags & NLM_F_ROOT) {
1105
		if (tb[1])
1106
			return tca_action_flush(net, tb[1], n, portid, extack);
1107

1108
		NL_SET_ERR_MSG(extack, "Invalid netlink attributes while flushing TC action");
1109
		return -EINVAL;
L
Linus Torvalds 已提交
1110 1111
	}

1112
	for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
1113
		act = tcf_action_get_1(net, tb[i], n, portid, extack);
1114 1115
		if (IS_ERR(act)) {
			ret = PTR_ERR(act);
L
Linus Torvalds 已提交
1116
			goto err;
1117
		}
1118
		act->order = i;
1119
		attr_size += tcf_action_fill_size(act);
1120
		list_add_tail(&act->list, &actions);
L
Linus Torvalds 已提交
1121
	}
1122 1123

	attr_size = tcf_action_full_attrs_size(attr_size);
L
Linus Torvalds 已提交
1124 1125

	if (event == RTM_GETACTION)
1126
		ret = tcf_get_notify(net, portid, n, &actions, event, extack);
L
Linus Torvalds 已提交
1127
	else { /* delete */
1128
		cleanup_a(&actions, 1); /* lookup took reference */
1129
		ret = tcf_del_notify(net, n, &actions, portid, attr_size, extack);
1130
		if (ret)
L
Linus Torvalds 已提交
1131 1132 1133 1134
			goto err;
		return ret;
	}
err:
1135
	tcf_action_destroy(&actions, 0);
L
Linus Torvalds 已提交
1136 1137 1138
	return ret;
}

1139 1140
static int
tcf_add_notify(struct net *net, struct nlmsghdr *n, struct list_head *actions,
1141
	       u32 portid, size_t attr_size, struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1142 1143 1144 1145
{
	struct sk_buff *skb;
	int err = 0;

1146 1147
	skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size,
			GFP_KERNEL);
L
Linus Torvalds 已提交
1148 1149 1150
	if (!skb)
		return -ENOBUFS;

1151 1152
	if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, n->nlmsg_flags,
			 RTM_NEWACTION, 0, 0) <= 0) {
1153
		NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action");
1154 1155 1156
		kfree_skb(skb);
		return -EINVAL;
	}
1157

1158 1159
	err = rtnetlink_send(skb, net, portid, RTNLGRP_TC,
			     n->nlmsg_flags & NLM_F_ECHO);
L
Linus Torvalds 已提交
1160 1161 1162 1163 1164
	if (err > 0)
		err = 0;
	return err;
}

J
Jamal Hadi Salim 已提交
1165
static int tcf_action_add(struct net *net, struct nlattr *nla,
1166 1167
			  struct nlmsghdr *n, u32 portid, int ovr,
			  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1168
{
1169
	size_t attr_size = 0;
L
Linus Torvalds 已提交
1170
	int ret = 0;
1171
	LIST_HEAD(actions);
L
Linus Torvalds 已提交
1172

1173
	ret = tcf_action_init(net, NULL, nla, NULL, NULL, ovr, 0, &actions,
1174
			      &attr_size, true, extack);
1175
	if (ret)
1176
		return ret;
L
Linus Torvalds 已提交
1177

1178
	return tcf_add_notify(net, n, &actions, portid, attr_size, extack);
L
Linus Torvalds 已提交
1179 1180
}

1181 1182 1183 1184
static u32 tcaa_root_flags_allowed = TCA_FLAG_LARGE_DUMP_ON;
static const struct nla_policy tcaa_policy[TCA_ROOT_MAX + 1] = {
	[TCA_ROOT_FLAGS] = { .type = NLA_BITFIELD32,
			     .validation_data = &tcaa_root_flags_allowed },
1185
	[TCA_ROOT_TIME_DELTA]      = { .type = NLA_U32 },
1186 1187
};

1188 1189
static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n,
			 struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
1190
{
1191
	struct net *net = sock_net(skb->sk);
1192
	struct nlattr *tca[TCA_ROOT_MAX + 1];
1193
	u32 portid = skb ? NETLINK_CB(skb).portid : 0;
L
Linus Torvalds 已提交
1194 1195
	int ret = 0, ovr = 0;

1196 1197
	if ((n->nlmsg_type != RTM_GETACTION) &&
	    !netlink_capable(skb, CAP_NET_ADMIN))
1198 1199
		return -EPERM;

1200
	ret = nlmsg_parse(n, sizeof(struct tcamsg), tca, TCA_ROOT_MAX, NULL,
1201
			  extack);
1202 1203 1204 1205
	if (ret < 0)
		return ret;

	if (tca[TCA_ACT_TAB] == NULL) {
1206
		NL_SET_ERR_MSG(extack, "Netlink action attributes missing");
L
Linus Torvalds 已提交
1207 1208 1209
		return -EINVAL;
	}

E
Eric Dumazet 已提交
1210
	/* n->nlmsg_flags & NLM_F_CREATE */
L
Linus Torvalds 已提交
1211 1212 1213
	switch (n->nlmsg_type) {
	case RTM_NEWACTION:
		/* we are going to assume all other flags
L
Lucas De Marchi 已提交
1214
		 * imply create only if it doesn't exist
L
Linus Torvalds 已提交
1215 1216 1217 1218
		 * Note that CREATE | EXCL implies that
		 * but since we want avoid ambiguity (eg when flags
		 * is zero) then just set this
		 */
E
Eric Dumazet 已提交
1219
		if (n->nlmsg_flags & NLM_F_REPLACE)
L
Linus Torvalds 已提交
1220 1221
			ovr = 1;
replay:
1222 1223
		ret = tcf_action_add(net, tca[TCA_ACT_TAB], n, portid, ovr,
				     extack);
L
Linus Torvalds 已提交
1224 1225 1226 1227
		if (ret == -EAGAIN)
			goto replay;
		break;
	case RTM_DELACTION:
1228
		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1229
				    portid, RTM_DELACTION, extack);
L
Linus Torvalds 已提交
1230 1231
		break;
	case RTM_GETACTION:
1232
		ret = tca_action_gd(net, tca[TCA_ACT_TAB], n,
1233
				    portid, RTM_GETACTION, extack);
L
Linus Torvalds 已提交
1234 1235 1236 1237 1238 1239 1240 1241
		break;
	default:
		BUG();
	}

	return ret;
}

1242
static struct nlattr *find_dump_kind(struct nlattr **nla)
L
Linus Torvalds 已提交
1243
{
E
Eric Dumazet 已提交
1244
	struct nlattr *tb1, *tb2[TCA_ACT_MAX + 1];
1245 1246
	struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
	struct nlattr *kind;
L
Linus Torvalds 已提交
1247

1248
	tb1 = nla[TCA_ACT_TAB];
L
Linus Torvalds 已提交
1249 1250 1251
	if (tb1 == NULL)
		return NULL;

1252
	if (nla_parse(tb, TCA_ACT_MAX_PRIO, nla_data(tb1),
1253
		      NLMSG_ALIGN(nla_len(tb1)), NULL, NULL) < 0)
L
Linus Torvalds 已提交
1254 1255
		return NULL;

1256 1257
	if (tb[1] == NULL)
		return NULL;
1258
	if (nla_parse_nested(tb2, TCA_ACT_MAX, tb[1], NULL, NULL) < 0)
L
Linus Torvalds 已提交
1259
		return NULL;
1260
	kind = tb2[TCA_ACT_KIND];
L
Linus Torvalds 已提交
1261

1262
	return kind;
L
Linus Torvalds 已提交
1263 1264
}

J
Jamal Hadi Salim 已提交
1265
static int tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb)
L
Linus Torvalds 已提交
1266
{
1267
	struct net *net = sock_net(skb->sk);
L
Linus Torvalds 已提交
1268
	struct nlmsghdr *nlh;
1269
	unsigned char *b = skb_tail_pointer(skb);
1270
	struct nlattr *nest;
L
Linus Torvalds 已提交
1271 1272
	struct tc_action_ops *a_o;
	int ret = 0;
1273
	struct tcamsg *t = (struct tcamsg *) nlmsg_data(cb->nlh);
1274 1275
	struct nlattr *tb[TCA_ROOT_MAX + 1];
	struct nlattr *count_attr = NULL;
1276
	unsigned long jiffy_since = 0;
1277 1278
	struct nlattr *kind = NULL;
	struct nla_bitfield32 bf;
1279
	u32 msecs_since = 0;
1280 1281 1282 1283 1284 1285
	u32 act_count = 0;

	ret = nlmsg_parse(cb->nlh, sizeof(struct tcamsg), tb, TCA_ROOT_MAX,
			  tcaa_policy, NULL);
	if (ret < 0)
		return ret;
L
Linus Torvalds 已提交
1286

1287
	kind = find_dump_kind(tb);
L
Linus Torvalds 已提交
1288
	if (kind == NULL) {
1289
		pr_info("tc_dump_action: action bad kind\n");
L
Linus Torvalds 已提交
1290 1291 1292
		return 0;
	}

1293
	a_o = tc_lookup_action(kind);
E
Eric Dumazet 已提交
1294
	if (a_o == NULL)
L
Linus Torvalds 已提交
1295 1296
		return 0;

1297 1298 1299 1300 1301 1302
	cb->args[2] = 0;
	if (tb[TCA_ROOT_FLAGS]) {
		bf = nla_get_bitfield32(tb[TCA_ROOT_FLAGS]);
		cb->args[2] = bf.value;
	}

1303 1304 1305 1306
	if (tb[TCA_ROOT_TIME_DELTA]) {
		msecs_since = nla_get_u32(tb[TCA_ROOT_TIME_DELTA]);
	}

1307
	nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
1308 1309 1310
			cb->nlh->nlmsg_type, sizeof(*t), 0);
	if (!nlh)
		goto out_module_put;
1311

1312 1313 1314
	if (msecs_since)
		jiffy_since = jiffies - msecs_to_jiffies(msecs_since);

1315
	t = nlmsg_data(nlh);
L
Linus Torvalds 已提交
1316
	t->tca_family = AF_UNSPEC;
1317 1318
	t->tca__pad1 = 0;
	t->tca__pad2 = 0;
1319
	cb->args[3] = jiffy_since;
1320 1321 1322
	count_attr = nla_reserve(skb, TCA_ROOT_COUNT, sizeof(u32));
	if (!count_attr)
		goto out_module_put;
L
Linus Torvalds 已提交
1323

1324 1325
	nest = nla_nest_start(skb, TCA_ACT_TAB);
	if (nest == NULL)
1326
		goto out_module_put;
L
Linus Torvalds 已提交
1327

1328
	ret = a_o->walk(net, skb, cb, RTM_GETACTION, a_o, NULL);
L
Linus Torvalds 已提交
1329
	if (ret < 0)
1330
		goto out_module_put;
L
Linus Torvalds 已提交
1331 1332

	if (ret > 0) {
1333
		nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
1334
		ret = skb->len;
1335 1336 1337
		act_count = cb->args[1];
		memcpy(nla_data(count_attr), &act_count, sizeof(u32));
		cb->args[1] = 0;
L
Linus Torvalds 已提交
1338
	} else
1339
		nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1340

1341
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
1342
	if (NETLINK_CB(cb->skb).portid && ret)
L
Linus Torvalds 已提交
1343 1344 1345 1346
		nlh->nlmsg_flags |= NLM_F_MULTI;
	module_put(a_o->owner);
	return skb->len;

1347
out_module_put:
L
Linus Torvalds 已提交
1348
	module_put(a_o->owner);
1349
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
1350 1351 1352
	return skb->len;
}

1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401
struct tcf_action_net {
	struct rhashtable egdev_ht;
};

static unsigned int tcf_action_net_id;

struct tcf_action_egdev_cb {
	struct list_head list;
	tc_setup_cb_t *cb;
	void *cb_priv;
};

struct tcf_action_egdev {
	struct rhash_head ht_node;
	const struct net_device *dev;
	unsigned int refcnt;
	struct list_head cb_list;
};

static const struct rhashtable_params tcf_action_egdev_ht_params = {
	.key_offset = offsetof(struct tcf_action_egdev, dev),
	.head_offset = offsetof(struct tcf_action_egdev, ht_node),
	.key_len = sizeof(const struct net_device *),
};

static struct tcf_action_egdev *
tcf_action_egdev_lookup(const struct net_device *dev)
{
	struct net *net = dev_net(dev);
	struct tcf_action_net *tan = net_generic(net, tcf_action_net_id);

	return rhashtable_lookup_fast(&tan->egdev_ht, &dev,
				      tcf_action_egdev_ht_params);
}

static struct tcf_action_egdev *
tcf_action_egdev_get(const struct net_device *dev)
{
	struct tcf_action_egdev *egdev;
	struct tcf_action_net *tan;

	egdev = tcf_action_egdev_lookup(dev);
	if (egdev)
		goto inc_ref;

	egdev = kzalloc(sizeof(*egdev), GFP_KERNEL);
	if (!egdev)
		return NULL;
	INIT_LIST_HEAD(&egdev->cb_list);
1402
	egdev->dev = dev;
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
	tan = net_generic(dev_net(dev), tcf_action_net_id);
	rhashtable_insert_fast(&tan->egdev_ht, &egdev->ht_node,
			       tcf_action_egdev_ht_params);

inc_ref:
	egdev->refcnt++;
	return egdev;
}

static void tcf_action_egdev_put(struct tcf_action_egdev *egdev)
{
	struct tcf_action_net *tan;

	if (--egdev->refcnt)
		return;
	tan = net_generic(dev_net(egdev->dev), tcf_action_net_id);
	rhashtable_remove_fast(&tan->egdev_ht, &egdev->ht_node,
			       tcf_action_egdev_ht_params);
	kfree(egdev);
}

static struct tcf_action_egdev_cb *
tcf_action_egdev_cb_lookup(struct tcf_action_egdev *egdev,
			   tc_setup_cb_t *cb, void *cb_priv)
{
	struct tcf_action_egdev_cb *egdev_cb;

	list_for_each_entry(egdev_cb, &egdev->cb_list, list)
		if (egdev_cb->cb == cb && egdev_cb->cb_priv == cb_priv)
			return egdev_cb;
	return NULL;
}

static int tcf_action_egdev_cb_call(struct tcf_action_egdev *egdev,
				    enum tc_setup_type type,
				    void *type_data, bool err_stop)
{
	struct tcf_action_egdev_cb *egdev_cb;
	int ok_count = 0;
	int err;

	list_for_each_entry(egdev_cb, &egdev->cb_list, list) {
		err = egdev_cb->cb(type, type_data, egdev_cb->cb_priv);
		if (err) {
			if (err_stop)
				return err;
		} else {
			ok_count++;
		}
	}
	return ok_count;
}

static int tcf_action_egdev_cb_add(struct tcf_action_egdev *egdev,
				   tc_setup_cb_t *cb, void *cb_priv)
{
	struct tcf_action_egdev_cb *egdev_cb;

	egdev_cb = tcf_action_egdev_cb_lookup(egdev, cb, cb_priv);
	if (WARN_ON(egdev_cb))
		return -EEXIST;
	egdev_cb = kzalloc(sizeof(*egdev_cb), GFP_KERNEL);
	if (!egdev_cb)
		return -ENOMEM;
	egdev_cb->cb = cb;
	egdev_cb->cb_priv = cb_priv;
	list_add(&egdev_cb->list, &egdev->cb_list);
	return 0;
}

static void tcf_action_egdev_cb_del(struct tcf_action_egdev *egdev,
				    tc_setup_cb_t *cb, void *cb_priv)
{
	struct tcf_action_egdev_cb *egdev_cb;

	egdev_cb = tcf_action_egdev_cb_lookup(egdev, cb, cb_priv);
	if (WARN_ON(!egdev_cb))
		return;
	list_del(&egdev_cb->list);
	kfree(egdev_cb);
}

static int __tc_setup_cb_egdev_register(const struct net_device *dev,
					tc_setup_cb_t *cb, void *cb_priv)
{
	struct tcf_action_egdev *egdev = tcf_action_egdev_get(dev);
	int err;

	if (!egdev)
		return -ENOMEM;
	err = tcf_action_egdev_cb_add(egdev, cb, cb_priv);
	if (err)
		goto err_cb_add;
	return 0;

err_cb_add:
	tcf_action_egdev_put(egdev);
	return err;
}
int tc_setup_cb_egdev_register(const struct net_device *dev,
			       tc_setup_cb_t *cb, void *cb_priv)
{
	int err;

	rtnl_lock();
	err = __tc_setup_cb_egdev_register(dev, cb, cb_priv);
	rtnl_unlock();
	return err;
}
EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_register);

static void __tc_setup_cb_egdev_unregister(const struct net_device *dev,
					   tc_setup_cb_t *cb, void *cb_priv)
{
	struct tcf_action_egdev *egdev = tcf_action_egdev_lookup(dev);

	if (WARN_ON(!egdev))
		return;
	tcf_action_egdev_cb_del(egdev, cb, cb_priv);
	tcf_action_egdev_put(egdev);
}
void tc_setup_cb_egdev_unregister(const struct net_device *dev,
				  tc_setup_cb_t *cb, void *cb_priv)
{
	rtnl_lock();
	__tc_setup_cb_egdev_unregister(dev, cb, cb_priv);
	rtnl_unlock();
}
EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_unregister);

int tc_setup_cb_egdev_call(const struct net_device *dev,
			   enum tc_setup_type type, void *type_data,
			   bool err_stop)
{
	struct tcf_action_egdev *egdev = tcf_action_egdev_lookup(dev);

	if (!egdev)
		return 0;
	return tcf_action_egdev_cb_call(egdev, type, type_data, err_stop);
}
EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_call);

static __net_init int tcf_action_net_init(struct net *net)
{
	struct tcf_action_net *tan = net_generic(net, tcf_action_net_id);

	return rhashtable_init(&tan->egdev_ht, &tcf_action_egdev_ht_params);
}

static void __net_exit tcf_action_net_exit(struct net *net)
{
	struct tcf_action_net *tan = net_generic(net, tcf_action_net_id);

	rhashtable_destroy(&tan->egdev_ht);
}

static struct pernet_operations tcf_action_net_ops = {
	.init = tcf_action_net_init,
	.exit = tcf_action_net_exit,
	.id = &tcf_action_net_id,
	.size = sizeof(struct tcf_action_net),
};

L
Linus Torvalds 已提交
1566 1567
static int __init tc_action_init(void)
{
1568 1569 1570 1571 1572 1573
	int err;

	err = register_pernet_subsys(&tcf_action_net_ops);
	if (err)
		return err;

1574 1575
	rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL, 0);
	rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL, 0);
1576
	rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action,
1577
		      0);
L
Linus Torvalds 已提交
1578 1579 1580 1581 1582

	return 0;
}

subsys_initcall(tc_action_init);