cls_api.c 24.0 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/*
 * net/sched/cls_api.c	Packet classifier 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.
 *
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
 *
 * Changes:
 *
 * Eduardo J. Blanco <ejbs@netlabs.com.uy> :990222: kmod support
 *
 */

#include <linux/module.h>
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/string.h>
#include <linux/errno.h>
22
#include <linux/err.h>
L
Linus Torvalds 已提交
23 24 25
#include <linux/skbuff.h>
#include <linux/init.h>
#include <linux/kmod.h>
26
#include <linux/err.h>
27
#include <linux/slab.h>
28 29
#include <net/net_namespace.h>
#include <net/sock.h>
30
#include <net/netlink.h>
L
Linus Torvalds 已提交
31 32 33 34
#include <net/pkt_sched.h>
#include <net/pkt_cls.h>

/* The list of all installed classifier types */
35
static LIST_HEAD(tcf_proto_base);
L
Linus Torvalds 已提交
36 37 38 39 40 41

/* Protects list of registered TC modules. It is pure SMP lock. */
static DEFINE_RWLOCK(cls_mod_lock);

/* Find classifier type by string name */

42
static const struct tcf_proto_ops *tcf_proto_lookup_ops(const char *kind)
L
Linus Torvalds 已提交
43
{
44
	const struct tcf_proto_ops *t, *res = NULL;
L
Linus Torvalds 已提交
45 46 47

	if (kind) {
		read_lock(&cls_mod_lock);
48
		list_for_each_entry(t, &tcf_proto_base, head) {
49
			if (strcmp(kind, t->kind) == 0) {
50 51
				if (try_module_get(t->owner))
					res = t;
L
Linus Torvalds 已提交
52 53 54 55 56
				break;
			}
		}
		read_unlock(&cls_mod_lock);
	}
57
	return res;
L
Linus Torvalds 已提交
58 59 60 61 62 63
}

/* Register(unregister) new classifier type */

int register_tcf_proto_ops(struct tcf_proto_ops *ops)
{
64
	struct tcf_proto_ops *t;
L
Linus Torvalds 已提交
65 66 67
	int rc = -EEXIST;

	write_lock(&cls_mod_lock);
68
	list_for_each_entry(t, &tcf_proto_base, head)
L
Linus Torvalds 已提交
69 70 71
		if (!strcmp(ops->kind, t->kind))
			goto out;

72
	list_add_tail(&ops->head, &tcf_proto_base);
L
Linus Torvalds 已提交
73 74 75 76 77
	rc = 0;
out:
	write_unlock(&cls_mod_lock);
	return rc;
}
78
EXPORT_SYMBOL(register_tcf_proto_ops);
L
Linus Torvalds 已提交
79 80 81

int unregister_tcf_proto_ops(struct tcf_proto_ops *ops)
{
82
	struct tcf_proto_ops *t;
L
Linus Torvalds 已提交
83 84
	int rc = -ENOENT;

85 86 87 88 89
	/* Wait for outstanding call_rcu()s, if any, from a
	 * tcf_proto_ops's destroy() handler.
	 */
	rcu_barrier();

L
Linus Torvalds 已提交
90
	write_lock(&cls_mod_lock);
91 92 93 94
	list_for_each_entry(t, &tcf_proto_base, head) {
		if (t == ops) {
			list_del(&t->head);
			rc = 0;
L
Linus Torvalds 已提交
95
			break;
96 97
		}
	}
L
Linus Torvalds 已提交
98 99 100
	write_unlock(&cls_mod_lock);
	return rc;
}
101
EXPORT_SYMBOL(unregister_tcf_proto_ops);
L
Linus Torvalds 已提交
102 103 104

/* Select new prio value from the range, managed by kernel. */

105
static inline u32 tcf_auto_prio(struct tcf_proto *tp)
L
Linus Torvalds 已提交
106
{
107
	u32 first = TC_H_MAKE(0xC0000000U, 0U);
L
Linus Torvalds 已提交
108 109

	if (tp)
E
Eric Dumazet 已提交
110
		first = tp->prio - 1;
L
Linus Torvalds 已提交
111

112
	return TC_H_MAJ(first);
L
Linus Torvalds 已提交
113 114
}

115
static struct tcf_proto *tcf_proto_create(const char *kind, u32 protocol,
116
					  u32 prio, u32 parent, struct Qdisc *q,
117
					  struct tcf_chain *chain)
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
{
	struct tcf_proto *tp;
	int err;

	tp = kzalloc(sizeof(*tp), GFP_KERNEL);
	if (!tp)
		return ERR_PTR(-ENOBUFS);

	err = -ENOENT;
	tp->ops = tcf_proto_lookup_ops(kind);
	if (!tp->ops) {
#ifdef CONFIG_MODULES
		rtnl_unlock();
		request_module("cls_%s", kind);
		rtnl_lock();
		tp->ops = tcf_proto_lookup_ops(kind);
		/* We dropped the RTNL semaphore in order to perform
		 * the module load. So, even if we succeeded in loading
		 * the module we have to replay the request. We indicate
		 * this using -EAGAIN.
		 */
		if (tp->ops) {
			module_put(tp->ops->owner);
			err = -EAGAIN;
		} else {
			err = -ENOENT;
		}
		goto errout;
#endif
	}
	tp->classify = tp->ops->classify;
	tp->protocol = protocol;
	tp->prio = prio;
	tp->classid = parent;
	tp->q = q;
153
	tp->chain = chain;
154 155 156 157 158 159 160 161 162 163 164 165 166

	err = tp->ops->init(tp);
	if (err) {
		module_put(tp->ops->owner);
		goto errout;
	}
	return tp;

errout:
	kfree(tp);
	return ERR_PTR(err);
}

167
static void tcf_proto_destroy(struct tcf_proto *tp)
168
{
169 170 171
	tp->ops->destroy(tp);
	module_put(tp->ops->owner);
	kfree_rcu(tp, rcu);
172 173
}

174 175
static struct tcf_chain *tcf_chain_create(struct tcf_block *block,
					  u32 chain_index)
176
{
177 178 179 180 181 182 183 184
	struct tcf_chain *chain;

	chain = kzalloc(sizeof(*chain), GFP_KERNEL);
	if (!chain)
		return NULL;
	list_add_tail(&chain->list, &block->chain_list);
	chain->block = block;
	chain->index = chain_index;
185
	chain->refcnt = 1;
186
	return chain;
187 188
}

J
Jiri Pirko 已提交
189
static void tcf_chain_flush(struct tcf_chain *chain)
190 191 192
{
	struct tcf_proto *tp;

193
	if (chain->p_filter_chain)
J
Jiri Pirko 已提交
194
		RCU_INIT_POINTER(*chain->p_filter_chain, NULL);
195 196
	while ((tp = rtnl_dereference(chain->filter_chain)) != NULL) {
		RCU_INIT_POINTER(chain->filter_chain, tp->next);
197
		tcf_chain_put(chain);
198
		tcf_proto_destroy(tp);
199
	}
J
Jiri Pirko 已提交
200 201 202 203
}

static void tcf_chain_destroy(struct tcf_chain *chain)
{
204 205 206
	list_del(&chain->list);
	kfree(chain);
}
207

208 209 210
static void tcf_chain_hold(struct tcf_chain *chain)
{
	++chain->refcnt;
211 212
}

213 214
struct tcf_chain *tcf_chain_get(struct tcf_block *block, u32 chain_index,
				bool create)
215 216 217 218
{
	struct tcf_chain *chain;

	list_for_each_entry(chain, &block->chain_list, list) {
219 220 221 222
		if (chain->index == chain_index) {
			tcf_chain_hold(chain);
			return chain;
		}
223
	}
224

225
	return create ? tcf_chain_create(block, chain_index) : NULL;
226 227 228 229 230
}
EXPORT_SYMBOL(tcf_chain_get);

void tcf_chain_put(struct tcf_chain *chain)
{
231
	if (--chain->refcnt == 0)
232 233 234 235
		tcf_chain_destroy(chain);
}
EXPORT_SYMBOL(tcf_chain_put);

236 237 238 239 240
static void
tcf_chain_filter_chain_ptr_set(struct tcf_chain *chain,
			       struct tcf_proto __rcu **p_filter_chain)
{
	chain->p_filter_chain = p_filter_chain;
241
}
242 243

int tcf_block_get(struct tcf_block **p_block,
244
		  struct tcf_proto __rcu **p_filter_chain, struct Qdisc *q)
245 246
{
	struct tcf_block *block = kzalloc(sizeof(*block), GFP_KERNEL);
247
	struct tcf_chain *chain;
248
	int err;
249 250 251

	if (!block)
		return -ENOMEM;
252 253 254 255
	INIT_LIST_HEAD(&block->chain_list);
	/* Create chain 0 by default, it has to be always present. */
	chain = tcf_chain_create(block, 0);
	if (!chain) {
256 257 258
		err = -ENOMEM;
		goto err_chain_create;
	}
259
	tcf_chain_filter_chain_ptr_set(chain, p_filter_chain);
260
	block->net = qdisc_net(q);
261
	block->q = q;
262 263
	*p_block = block;
	return 0;
264 265 266 267

err_chain_create:
	kfree(block);
	return err;
268 269 270 271 272
}
EXPORT_SYMBOL(tcf_block_get);

void tcf_block_put(struct tcf_block *block)
{
273 274
	struct tcf_chain *chain, *tmp;

275 276
	if (!block)
		return;
277

278 279
	/* XXX: Standalone actions are not allowed to jump to any chain, and
	 * bound actions should be all removed after flushing. However,
280 281 282
	 * filters are destroyed in RCU callbacks, we have to hold the chains
	 * first, otherwise we would always race with RCU callbacks on this list
	 * without proper locking.
283
	 */
284 285 286 287 288 289 290 291 292 293 294

	/* Wait for existing RCU callbacks to cool down. */
	rcu_barrier();

	/* Hold a refcnt for all chains, except 0, in case they are gone. */
	list_for_each_entry(chain, &block->chain_list, list)
		if (chain->index)
			tcf_chain_hold(chain);

	/* No race on the list, because no chain could be destroyed. */
	list_for_each_entry(chain, &block->chain_list, list)
295
		tcf_chain_flush(chain);
296

297 298 299 300
	/* Wait for RCU callbacks to release the reference count. */
	rcu_barrier();

	/* At this point, all the chains should have refcnt == 1. */
301 302
	list_for_each_entry_safe(chain, tmp, &block->chain_list, list)
		tcf_chain_put(chain);
303 304 305
	kfree(block);
}
EXPORT_SYMBOL(tcf_block_put);
306

307 308 309 310 311 312 313 314 315 316
/* Main classifier routine: scans classifier chain attached
 * to this qdisc, (optionally) tests for protocol and asks
 * specific classifiers.
 */
int tcf_classify(struct sk_buff *skb, const struct tcf_proto *tp,
		 struct tcf_result *res, bool compat_mode)
{
	__be16 protocol = tc_skb_protocol(skb);
#ifdef CONFIG_NET_CLS_ACT
	const int max_reclassify_loop = 4;
317 318
	const struct tcf_proto *orig_tp = tp;
	const struct tcf_proto *first_tp;
319 320 321 322 323 324 325 326 327 328 329 330 331
	int limit = 0;

reclassify:
#endif
	for (; tp; tp = rcu_dereference_bh(tp->next)) {
		int err;

		if (tp->protocol != protocol &&
		    tp->protocol != htons(ETH_P_ALL))
			continue;

		err = tp->classify(skb, tp, res);
#ifdef CONFIG_NET_CLS_ACT
332
		if (unlikely(err == TC_ACT_RECLASSIFY && !compat_mode)) {
333
			first_tp = orig_tp;
334
			goto reset;
335
		} else if (unlikely(TC_ACT_EXT_CMP(err, TC_ACT_GOTO_CHAIN))) {
336
			first_tp = res->goto_tp;
337 338
			goto reset;
		}
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
#endif
		if (err >= 0)
			return err;
	}

	return TC_ACT_UNSPEC; /* signal: continue lookup */
#ifdef CONFIG_NET_CLS_ACT
reset:
	if (unlikely(limit++ >= max_reclassify_loop)) {
		net_notice_ratelimited("%s: reclassify loop, rule prio %u, protocol %02x\n",
				       tp->q->ops->id, tp->prio & 0xffff,
				       ntohs(tp->protocol));
		return TC_ACT_SHOT;
	}

354
	tp = first_tp;
355 356 357 358 359 360
	protocol = tc_skb_protocol(skb);
	goto reclassify;
#endif
}
EXPORT_SYMBOL(tcf_classify);

361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
struct tcf_chain_info {
	struct tcf_proto __rcu **pprev;
	struct tcf_proto __rcu *next;
};

static struct tcf_proto *tcf_chain_tp_prev(struct tcf_chain_info *chain_info)
{
	return rtnl_dereference(*chain_info->pprev);
}

static void tcf_chain_tp_insert(struct tcf_chain *chain,
				struct tcf_chain_info *chain_info,
				struct tcf_proto *tp)
{
	if (chain->p_filter_chain &&
	    *chain_info->pprev == chain->filter_chain)
377
		rcu_assign_pointer(*chain->p_filter_chain, tp);
378 379
	RCU_INIT_POINTER(tp->next, tcf_chain_tp_prev(chain_info));
	rcu_assign_pointer(*chain_info->pprev, tp);
380
	tcf_chain_hold(chain);
381 382 383 384 385 386 387 388 389
}

static void tcf_chain_tp_remove(struct tcf_chain *chain,
				struct tcf_chain_info *chain_info,
				struct tcf_proto *tp)
{
	struct tcf_proto *next = rtnl_dereference(chain_info->next);

	if (chain->p_filter_chain && tp == chain->filter_chain)
390
		RCU_INIT_POINTER(*chain->p_filter_chain, next);
391
	RCU_INIT_POINTER(*chain_info->pprev, next);
392
	tcf_chain_put(chain);
393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421
}

static struct tcf_proto *tcf_chain_tp_find(struct tcf_chain *chain,
					   struct tcf_chain_info *chain_info,
					   u32 protocol, u32 prio,
					   bool prio_allocate)
{
	struct tcf_proto **pprev;
	struct tcf_proto *tp;

	/* Check the chain for existence of proto-tcf with this priority */
	for (pprev = &chain->filter_chain;
	     (tp = rtnl_dereference(*pprev)); pprev = &tp->next) {
		if (tp->prio >= prio) {
			if (tp->prio == prio) {
				if (prio_allocate ||
				    (tp->protocol != protocol && protocol))
					return ERR_PTR(-EINVAL);
			} else {
				tp = NULL;
			}
			break;
		}
	}
	chain_info->pprev = pprev;
	chain_info->next = tp ? tp->next : NULL;
	return tp;
}

422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
static int tcf_fill_node(struct net *net, struct sk_buff *skb,
			 struct tcf_proto *tp, void *fh, u32 portid,
			 u32 seq, u16 flags, int event)
{
	struct tcmsg *tcm;
	struct nlmsghdr  *nlh;
	unsigned char *b = skb_tail_pointer(skb);

	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
	if (!nlh)
		goto out_nlmsg_trim;
	tcm = nlmsg_data(nlh);
	tcm->tcm_family = AF_UNSPEC;
	tcm->tcm__pad1 = 0;
	tcm->tcm__pad2 = 0;
	tcm->tcm_ifindex = qdisc_dev(tp->q)->ifindex;
	tcm->tcm_parent = tp->classid;
	tcm->tcm_info = TC_H_MAKE(tp->prio, tp->protocol);
	if (nla_put_string(skb, TCA_KIND, tp->ops->kind))
		goto nla_put_failure;
	if (nla_put_u32(skb, TCA_CHAIN, tp->chain->index))
		goto nla_put_failure;
	if (!fh) {
		tcm->tcm_handle = 0;
	} else {
		if (tp->ops->dump && tp->ops->dump(net, tp, fh, skb, tcm) < 0)
			goto nla_put_failure;
	}
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
	return skb->len;

out_nlmsg_trim:
nla_put_failure:
	nlmsg_trim(skb, b);
	return -1;
}

static int tfilter_notify(struct net *net, struct sk_buff *oskb,
			  struct nlmsghdr *n, struct tcf_proto *tp,
			  void *fh, int event, bool unicast)
{
	struct sk_buff *skb;
	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;

	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
	if (!skb)
		return -ENOBUFS;

	if (tcf_fill_node(net, skb, tp, fh, portid, n->nlmsg_seq,
			  n->nlmsg_flags, event) <= 0) {
		kfree_skb(skb);
		return -EINVAL;
	}

	if (unicast)
		return netlink_unicast(net->rtnl, skb, portid, MSG_DONTWAIT);

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

static int tfilter_del_notify(struct net *net, struct sk_buff *oskb,
			      struct nlmsghdr *n, struct tcf_proto *tp,
			      void *fh, bool unicast, bool *last)
{
	struct sk_buff *skb;
	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
	int err;

	skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
	if (!skb)
		return -ENOBUFS;

	if (tcf_fill_node(net, skb, tp, fh, portid, n->nlmsg_seq,
			  n->nlmsg_flags, RTM_DELTFILTER) <= 0) {
		kfree_skb(skb);
		return -EINVAL;
	}

	err = tp->ops->delete(tp, fh, last);
	if (err) {
		kfree_skb(skb);
		return err;
	}

	if (unicast)
		return netlink_unicast(net->rtnl, skb, portid, MSG_DONTWAIT);

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

static void tfilter_notify_chain(struct net *net, struct sk_buff *oskb,
				 struct nlmsghdr *n,
				 struct tcf_chain *chain, int event)
{
	struct tcf_proto *tp;

	for (tp = rtnl_dereference(chain->filter_chain);
	     tp; tp = rtnl_dereference(tp->next))
		tfilter_notify(net, oskb, n, tp, 0, event, false);
}

L
Linus Torvalds 已提交
525 526
/* Add/change/delete/get a filter node */

527 528
static int tc_ctl_tfilter(struct sk_buff *skb, struct nlmsghdr *n,
			  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
529
{
530
	struct net *net = sock_net(skb->sk);
531
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
532 533 534
	struct tcmsg *t;
	u32 protocol;
	u32 prio;
535
	bool prio_allocate;
L
Linus Torvalds 已提交
536
	u32 parent;
537
	u32 chain_index;
L
Linus Torvalds 已提交
538 539
	struct net_device *dev;
	struct Qdisc  *q;
540
	struct tcf_chain_info chain_info;
541
	struct tcf_chain *chain = NULL;
542
	struct tcf_block *block;
L
Linus Torvalds 已提交
543
	struct tcf_proto *tp;
544
	const struct Qdisc_class_ops *cops;
L
Linus Torvalds 已提交
545
	unsigned long cl;
546
	void *fh;
L
Linus Torvalds 已提交
547
	int err;
548
	int tp_created;
L
Linus Torvalds 已提交
549

550
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
551
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
552
		return -EPERM;
553

L
Linus Torvalds 已提交
554
replay:
555 556
	tp_created = 0;

557
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
558 559 560
	if (err < 0)
		return err;

561
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
562 563
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
564
	prio_allocate = false;
L
Linus Torvalds 已提交
565 566 567 568
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
569 570
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
571
			if (protocol || t->tcm_handle || tca[TCA_KIND])
572 573 574 575 576 577 578 579
				return -ENOENT;
			break;
		case RTM_NEWTFILTER:
			/* If no priority is provided by the user,
			 * we allocate one.
			 */
			if (n->nlmsg_flags & NLM_F_CREATE) {
				prio = TC_H_MAKE(0x80000000U, 0U);
580
				prio_allocate = true;
581 582 583 584
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
585
			return -ENOENT;
586
		}
L
Linus Torvalds 已提交
587 588 589 590 591
	}

	/* Find head of filter chain. */

	/* Find link */
592
	dev = __dev_get_by_index(net, t->tcm_ifindex);
593
	if (dev == NULL)
L
Linus Torvalds 已提交
594 595 596 597
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
598
		q = dev->qdisc;
L
Linus Torvalds 已提交
599
		parent = q->handle;
600 601 602 603 604
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
605 606

	/* Is it classful? */
E
Eric Dumazet 已提交
607 608
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
609 610
		return -EINVAL;

611
	if (!cops->tcf_block)
612 613
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
614 615
	/* Do we search for filter, attached to class? */
	if (TC_H_MIN(parent)) {
616
		cl = cops->find(q, parent);
L
Linus Torvalds 已提交
617 618 619 620 621
		if (cl == 0)
			return -ENOENT;
	}

	/* And the last stroke */
622 623
	block = cops->tcf_block(q, cl);
	if (!block) {
624
		err = -EINVAL;
L
Linus Torvalds 已提交
625
		goto errout;
626
	}
627 628 629 630 631 632

	chain_index = tca[TCA_CHAIN] ? nla_get_u32(tca[TCA_CHAIN]) : 0;
	if (chain_index > TC_ACT_EXT_VAL_MASK) {
		err = -EINVAL;
		goto errout;
	}
633 634
	chain = tcf_chain_get(block, chain_index,
			      n->nlmsg_type == RTM_NEWTFILTER);
635
	if (!chain) {
636
		err = n->nlmsg_type == RTM_NEWTFILTER ? -ENOMEM : -EINVAL;
637 638
		goto errout;
	}
639

640 641
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
J
Jiri Pirko 已提交
642
		tcf_chain_flush(chain);
643 644 645
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
646

647 648 649 650 651
	tp = tcf_chain_tp_find(chain, &chain_info, protocol,
			       prio, prio_allocate);
	if (IS_ERR(tp)) {
		err = PTR_ERR(tp);
		goto errout;
L
Linus Torvalds 已提交
652 653 654 655 656
	}

	if (tp == NULL) {
		/* Proto-tcf does not exist, create new one */

657 658
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
659
			goto errout;
660
		}
L
Linus Torvalds 已提交
661

E
Eric Dumazet 已提交
662
		if (n->nlmsg_type != RTM_NEWTFILTER ||
663 664
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
665
			goto errout;
666
		}
L
Linus Torvalds 已提交
667

668
		if (prio_allocate)
669
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
670

671
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
672
				      protocol, prio, parent, q, chain);
673 674
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
675 676
			goto errout;
		}
677
		tp_created = 1;
678 679
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
680
		goto errout;
681
	}
L
Linus Torvalds 已提交
682 683 684

	fh = tp->ops->get(tp, t->tcm_handle);

685
	if (!fh) {
L
Linus Torvalds 已提交
686
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
687
			tcf_chain_tp_remove(chain, &chain_info, tp);
688 689
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
690
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
691 692 693 694
			err = 0;
			goto errout;
		}

695
		if (n->nlmsg_type != RTM_NEWTFILTER ||
696 697
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
698
			goto errout;
699
		}
L
Linus Torvalds 已提交
700
	} else {
701 702
		bool last;

L
Linus Torvalds 已提交
703
		switch (n->nlmsg_type) {
704
		case RTM_NEWTFILTER:
705 706
			if (n->nlmsg_flags & NLM_F_EXCL) {
				if (tp_created)
707
					tcf_proto_destroy(tp);
708
				err = -EEXIST;
L
Linus Torvalds 已提交
709
				goto errout;
710
			}
L
Linus Torvalds 已提交
711 712
			break;
		case RTM_DELTFILTER:
713 714
			err = tfilter_del_notify(net, skb, n, tp, fh, false,
						 &last);
715 716
			if (err)
				goto errout;
717
			if (last) {
718
				tcf_chain_tp_remove(chain, &chain_info, tp);
719 720
				tcf_proto_destroy(tp);
			}
721
			goto errout;
L
Linus Torvalds 已提交
722
		case RTM_GETTFILTER:
J
Jamal Hadi Salim 已提交
723
			err = tfilter_notify(net, skb, n, tp, fh,
724
					     RTM_NEWTFILTER, true);
L
Linus Torvalds 已提交
725 726 727 728 729 730 731
			goto errout;
		default:
			err = -EINVAL;
			goto errout;
		}
	}

732 733
	err = tp->ops->change(net, skb, tp, cl, t->tcm_handle, tca, &fh,
			      n->nlmsg_flags & NLM_F_CREATE ? TCA_ACT_NOREPLACE : TCA_ACT_REPLACE);
734
	if (err == 0) {
735 736
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
737
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
738 739
	} else {
		if (tp_created)
740
			tcf_proto_destroy(tp);
741
	}
L
Linus Torvalds 已提交
742 743

errout:
744 745
	if (chain)
		tcf_chain_put(chain);
L
Linus Torvalds 已提交
746 747 748 749 750 751
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

752
struct tcf_dump_args {
L
Linus Torvalds 已提交
753 754 755 756 757
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

758
static int tcf_node_dump(struct tcf_proto *tp, void *n, struct tcf_walker *arg)
L
Linus Torvalds 已提交
759
{
760
	struct tcf_dump_args *a = (void *)arg;
761
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
762

763
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
764 765
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
766 767
}

768
static bool tcf_chain_dump(struct tcf_chain *chain, struct sk_buff *skb,
769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794
			   struct netlink_callback *cb,
			   long index_start, long *p_index)
{
	struct net *net = sock_net(skb->sk);
	struct tcmsg *tcm = nlmsg_data(cb->nlh);
	struct tcf_dump_args arg;
	struct tcf_proto *tp;

	for (tp = rtnl_dereference(chain->filter_chain);
	     tp; tp = rtnl_dereference(tp->next), (*p_index)++) {
		if (*p_index < index_start)
			continue;
		if (TC_H_MAJ(tcm->tcm_info) &&
		    TC_H_MAJ(tcm->tcm_info) != tp->prio)
			continue;
		if (TC_H_MIN(tcm->tcm_info) &&
		    TC_H_MIN(tcm->tcm_info) != tp->protocol)
			continue;
		if (*p_index > index_start)
			memset(&cb->args[1], 0,
			       sizeof(cb->args) - sizeof(cb->args[0]));
		if (cb->args[1] == 0) {
			if (tcf_fill_node(net, skb, tp, 0,
					  NETLINK_CB(cb->skb).portid,
					  cb->nlh->nlmsg_seq, NLM_F_MULTI,
					  RTM_NEWTFILTER) <= 0)
795
				return false;
796 797 798 799 800 801 802 803 804 805 806 807 808 809

			cb->args[1] = 1;
		}
		if (!tp->ops->walk)
			continue;
		arg.w.fn = tcf_node_dump;
		arg.skb = skb;
		arg.cb = cb;
		arg.w.stop = 0;
		arg.w.skip = cb->args[1] - 1;
		arg.w.count = 0;
		tp->ops->walk(tp, &arg.w);
		cb->args[1] = arg.w.count + 1;
		if (arg.w.stop)
810
			return false;
811
	}
812
	return true;
813 814
}

E
Eric Dumazet 已提交
815
/* called with RTNL */
L
Linus Torvalds 已提交
816 817
static int tc_dump_tfilter(struct sk_buff *skb, struct netlink_callback *cb)
{
818
	struct net *net = sock_net(skb->sk);
819
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
820 821
	struct net_device *dev;
	struct Qdisc *q;
822
	struct tcf_block *block;
823
	struct tcf_chain *chain;
824
	struct tcmsg *tcm = nlmsg_data(cb->nlh);
L
Linus Torvalds 已提交
825
	unsigned long cl = 0;
826
	const struct Qdisc_class_ops *cops;
827 828
	long index_start;
	long index;
829
	int err;
L
Linus Torvalds 已提交
830

831
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
832
		return skb->len;
833 834 835 836 837

	err = nlmsg_parse(cb->nlh, sizeof(*tcm), tca, TCA_MAX, NULL, NULL);
	if (err)
		return err;

E
Eric Dumazet 已提交
838 839
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
840 841 842
		return skb->len;

	if (!tcm->tcm_parent)
843
		q = dev->qdisc;
L
Linus Torvalds 已提交
844 845 846 847
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
848 849
	cops = q->ops->cl_ops;
	if (!cops)
850
		goto out;
851
	if (!cops->tcf_block)
852
		goto out;
L
Linus Torvalds 已提交
853
	if (TC_H_MIN(tcm->tcm_parent)) {
854
		cl = cops->find(q, tcm->tcm_parent);
L
Linus Torvalds 已提交
855
		if (cl == 0)
856
			goto out;
L
Linus Torvalds 已提交
857
	}
858 859
	block = cops->tcf_block(q, cl);
	if (!block)
860
		goto out;
L
Linus Torvalds 已提交
861

862 863
	index_start = cb->args[0];
	index = 0;
864 865 866 867 868 869 870 871 872

	list_for_each_entry(chain, &block->chain_list, list) {
		if (tca[TCA_CHAIN] &&
		    nla_get_u32(tca[TCA_CHAIN]) != chain->index)
			continue;
		if (!tcf_chain_dump(chain, skb, cb, index_start, &index))
			break;
	}

873
	cb->args[0] = index;
L
Linus Torvalds 已提交
874 875 876 877 878

out:
	return skb->len;
}

879
void tcf_exts_destroy(struct tcf_exts *exts)
L
Linus Torvalds 已提交
880 881
{
#ifdef CONFIG_NET_CLS_ACT
882 883 884 885 886 887
	LIST_HEAD(actions);

	tcf_exts_to_list(exts, &actions);
	tcf_action_destroy(&actions, TCA_ACT_UNBIND);
	kfree(exts->actions);
	exts->nr_actions = 0;
L
Linus Torvalds 已提交
888 889
#endif
}
890
EXPORT_SYMBOL(tcf_exts_destroy);
L
Linus Torvalds 已提交
891

892
int tcf_exts_validate(struct net *net, struct tcf_proto *tp, struct nlattr **tb,
J
Jamal Hadi Salim 已提交
893
		      struct nlattr *rate_tlv, struct tcf_exts *exts, bool ovr)
L
Linus Torvalds 已提交
894 895 896 897 898
{
#ifdef CONFIG_NET_CLS_ACT
	{
		struct tc_action *act;

899
		if (exts->police && tb[exts->police]) {
900 901 902
			act = tcf_action_init_1(net, tp, tb[exts->police],
						rate_tlv, "police", ovr,
						TCA_ACT_BIND);
903 904
			if (IS_ERR(act))
				return PTR_ERR(act);
L
Linus Torvalds 已提交
905

906
			act->type = exts->type = TCA_OLD_COMPAT;
907 908
			exts->actions[0] = act;
			exts->nr_actions = 1;
909
		} else if (exts->action && tb[exts->action]) {
910 911 912
			LIST_HEAD(actions);
			int err, i = 0;

913 914
			err = tcf_action_init(net, tp, tb[exts->action],
					      rate_tlv, NULL, ovr, TCA_ACT_BIND,
J
Jamal Hadi Salim 已提交
915
					      &actions);
916 917
			if (err)
				return err;
918 919 920
			list_for_each_entry(act, &actions, list)
				exts->actions[i++] = act;
			exts->nr_actions = i;
L
Linus Torvalds 已提交
921 922 923
		}
	}
#else
924 925
	if ((exts->action && tb[exts->action]) ||
	    (exts->police && tb[exts->police]))
L
Linus Torvalds 已提交
926 927 928 929 930
		return -EOPNOTSUPP;
#endif

	return 0;
}
931
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
932

933
void tcf_exts_change(struct tcf_exts *dst, struct tcf_exts *src)
L
Linus Torvalds 已提交
934 935
{
#ifdef CONFIG_NET_CLS_ACT
936 937
	struct tcf_exts old = *dst;

938
	*dst = *src;
939
	tcf_exts_destroy(&old);
L
Linus Torvalds 已提交
940 941
#endif
}
942
EXPORT_SYMBOL(tcf_exts_change);
L
Linus Torvalds 已提交
943

944 945 946 947 948 949 950 951 952
#ifdef CONFIG_NET_CLS_ACT
static struct tc_action *tcf_exts_first_act(struct tcf_exts *exts)
{
	if (exts->nr_actions == 0)
		return NULL;
	else
		return exts->actions[0];
}
#endif
953

954
int tcf_exts_dump(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
955 956
{
#ifdef CONFIG_NET_CLS_ACT
957 958
	struct nlattr *nest;

959
	if (exts->action && tcf_exts_has_actions(exts)) {
L
Linus Torvalds 已提交
960 961 962 963 964
		/*
		 * again for backward compatible mode - we want
		 * to work with both old and new modes of entering
		 * tc data even if iproute2  was newer - jhs
		 */
965
		if (exts->type != TCA_OLD_COMPAT) {
966 967
			LIST_HEAD(actions);

968
			nest = nla_nest_start(skb, exts->action);
969 970
			if (nest == NULL)
				goto nla_put_failure;
971 972 973

			tcf_exts_to_list(exts, &actions);
			if (tcf_action_dump(skb, &actions, 0, 0) < 0)
974
				goto nla_put_failure;
975
			nla_nest_end(skb, nest);
976
		} else if (exts->police) {
977
			struct tc_action *act = tcf_exts_first_act(exts);
978
			nest = nla_nest_start(skb, exts->police);
979
			if (nest == NULL || !act)
980
				goto nla_put_failure;
981
			if (tcf_action_dump_old(skb, act, 0, 0) < 0)
982
				goto nla_put_failure;
983
			nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
984 985 986
		}
	}
	return 0;
987 988 989

nla_put_failure:
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
990
	return -1;
991 992 993
#else
	return 0;
#endif
L
Linus Torvalds 已提交
994
}
995
EXPORT_SYMBOL(tcf_exts_dump);
L
Linus Torvalds 已提交
996

997

998
int tcf_exts_dump_stats(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
999 1000
{
#ifdef CONFIG_NET_CLS_ACT
1001
	struct tc_action *a = tcf_exts_first_act(exts);
1002
	if (a != NULL && tcf_action_copy_stats(skb, a, 1) < 0)
1003
		return -1;
L
Linus Torvalds 已提交
1004 1005 1006
#endif
	return 0;
}
1007
EXPORT_SYMBOL(tcf_exts_dump_stats);
L
Linus Torvalds 已提交
1008

1009 1010 1011
static int tc_exts_setup_cb_egdev_call(struct tcf_exts *exts,
				       enum tc_setup_type type,
				       void *type_data, bool err_stop)
1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
{
	int ok_count = 0;
#ifdef CONFIG_NET_CLS_ACT
	const struct tc_action *a;
	struct net_device *dev;
	LIST_HEAD(actions);
	int ret;

	if (!tcf_exts_has_actions(exts))
		return 0;

	tcf_exts_to_list(exts, &actions);
	list_for_each_entry(a, &actions, list) {
		if (!a->ops->get_dev)
			continue;
		dev = a->ops->get_dev(a);
		if (!dev || !tc_can_offload(dev))
			continue;
		ret = tc_setup_cb_egdev_call(dev, type, type_data, err_stop);
		if (ret < 0)
			return ret;
		ok_count += ret;
	}
#endif
	return ok_count;
}
1038 1039 1040 1041 1042 1043 1044

int tc_setup_cb_call(struct tcf_exts *exts, enum tc_setup_type type,
		     void *type_data, bool err_stop)
{
	return tc_exts_setup_cb_egdev_call(exts, type, type_data, err_stop);
}
EXPORT_SYMBOL(tc_setup_cb_call);
1045

L
Linus Torvalds 已提交
1046 1047
static int __init tc_filter_init(void)
{
1048 1049
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, 0);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, 0);
1050
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
1051
		      tc_dump_tfilter, 0);
L
Linus Torvalds 已提交
1052 1053 1054 1055 1056

	return 0;
}

subsys_initcall(tc_filter_init);