cls_api.c 22.9 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 185 186
	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;
	chain->refcnt = 1;
	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_proto_destroy(tp);
198
	}
J
Jiri Pirko 已提交
199 200 201 202 203 204
}

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

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

	list_for_each_entry(chain, &block->chain_list, list) {
		if (chain->index == chain_index) {
			chain->refcnt++;
			return chain;
		}
	}
219 220 221 222
	if (create)
		return tcf_chain_create(block, chain_index);
	else
		return NULL;
223 224 225 226 227 228 229 230 231 232 233 234 235
}
EXPORT_SYMBOL(tcf_chain_get);

void tcf_chain_put(struct tcf_chain *chain)
{
	/* Destroy unused chain, with exception of chain 0, which is the
	 * default one and has to be always present.
	 */
	if (--chain->refcnt == 0 && !chain->filter_chain && chain->index != 0)
		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 244 245 246

int tcf_block_get(struct tcf_block **p_block,
		  struct tcf_proto __rcu **p_filter_chain)
{
	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 261
	*p_block = block;
	return 0;
262 263 264 265

err_chain_create:
	kfree(block);
	return err;
266 267 268 269 270
}
EXPORT_SYMBOL(tcf_block_get);

void tcf_block_put(struct tcf_block *block)
{
271 272
	struct tcf_chain *chain, *tmp;

273 274
	if (!block)
		return;
275 276 277

	list_for_each_entry_safe(chain, tmp, &block->chain_list, list)
		tcf_chain_destroy(chain);
278 279 280
	kfree(block);
}
EXPORT_SYMBOL(tcf_block_put);
281

282 283 284 285 286 287 288 289 290 291
/* 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;
292 293
	const struct tcf_proto *orig_tp = tp;
	const struct tcf_proto *first_tp;
294 295 296 297 298 299 300 301 302 303 304 305 306
	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
307
		if (unlikely(err == TC_ACT_RECLASSIFY && !compat_mode)) {
308
			first_tp = orig_tp;
309
			goto reset;
310
		} else if (unlikely(TC_ACT_EXT_CMP(err, TC_ACT_GOTO_CHAIN))) {
311
			first_tp = res->goto_tp;
312 313
			goto reset;
		}
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
#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;
	}

329
	tp = first_tp;
330 331 332 333 334 335
	protocol = tc_skb_protocol(skb);
	goto reclassify;
#endif
}
EXPORT_SYMBOL(tcf_classify);

336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351
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)
352
		rcu_assign_pointer(*chain->p_filter_chain, tp);
353 354 355 356 357 358 359 360 361 362 363
	RCU_INIT_POINTER(tp->next, tcf_chain_tp_prev(chain_info));
	rcu_assign_pointer(*chain_info->pprev, tp);
}

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)
364
		RCU_INIT_POINTER(*chain->p_filter_chain, next);
365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394
	RCU_INIT_POINTER(*chain_info->pprev, next);
}

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;
}

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 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
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 已提交
498 499
/* Add/change/delete/get a filter node */

500 501
static int tc_ctl_tfilter(struct sk_buff *skb, struct nlmsghdr *n,
			  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
502
{
503
	struct net *net = sock_net(skb->sk);
504
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
505 506 507
	struct tcmsg *t;
	u32 protocol;
	u32 prio;
508
	bool prio_allocate;
L
Linus Torvalds 已提交
509
	u32 parent;
510
	u32 chain_index;
L
Linus Torvalds 已提交
511 512
	struct net_device *dev;
	struct Qdisc  *q;
513
	struct tcf_chain_info chain_info;
514
	struct tcf_chain *chain = NULL;
515
	struct tcf_block *block;
L
Linus Torvalds 已提交
516
	struct tcf_proto *tp;
517
	const struct Qdisc_class_ops *cops;
L
Linus Torvalds 已提交
518
	unsigned long cl;
519
	void *fh;
L
Linus Torvalds 已提交
520
	int err;
521
	int tp_created;
L
Linus Torvalds 已提交
522

523
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
524
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
525
		return -EPERM;
526

L
Linus Torvalds 已提交
527
replay:
528 529
	tp_created = 0;

530
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
531 532 533
	if (err < 0)
		return err;

534
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
535 536
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
537
	prio_allocate = false;
L
Linus Torvalds 已提交
538 539 540 541
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
542 543
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
544
			if (protocol || t->tcm_handle || tca[TCA_KIND])
545 546 547 548 549 550 551 552
				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);
553
				prio_allocate = true;
554 555 556 557
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
558
			return -ENOENT;
559
		}
L
Linus Torvalds 已提交
560 561 562 563 564
	}

	/* Find head of filter chain. */

	/* Find link */
565
	dev = __dev_get_by_index(net, t->tcm_ifindex);
566
	if (dev == NULL)
L
Linus Torvalds 已提交
567 568 569 570
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
571
		q = dev->qdisc;
L
Linus Torvalds 已提交
572
		parent = q->handle;
573 574 575 576 577
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
578 579

	/* Is it classful? */
E
Eric Dumazet 已提交
580 581
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
582 583
		return -EINVAL;

584
	if (!cops->tcf_block)
585 586
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
587 588 589 590 591 592 593 594
	/* Do we search for filter, attached to class? */
	if (TC_H_MIN(parent)) {
		cl = cops->get(q, parent);
		if (cl == 0)
			return -ENOENT;
	}

	/* And the last stroke */
595 596
	block = cops->tcf_block(q, cl);
	if (!block) {
597
		err = -EINVAL;
L
Linus Torvalds 已提交
598
		goto errout;
599
	}
600 601 602 603 604 605

	chain_index = tca[TCA_CHAIN] ? nla_get_u32(tca[TCA_CHAIN]) : 0;
	if (chain_index > TC_ACT_EXT_VAL_MASK) {
		err = -EINVAL;
		goto errout;
	}
606 607
	chain = tcf_chain_get(block, chain_index,
			      n->nlmsg_type == RTM_NEWTFILTER);
608
	if (!chain) {
609
		err = n->nlmsg_type == RTM_NEWTFILTER ? -ENOMEM : -EINVAL;
610 611
		goto errout;
	}
612

613 614
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
J
Jiri Pirko 已提交
615
		tcf_chain_flush(chain);
616 617 618
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
619

620 621 622 623 624
	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 已提交
625 626 627 628 629
	}

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

630 631
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
632
			goto errout;
633
		}
L
Linus Torvalds 已提交
634

E
Eric Dumazet 已提交
635
		if (n->nlmsg_type != RTM_NEWTFILTER ||
636 637
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
638
			goto errout;
639
		}
L
Linus Torvalds 已提交
640

641
		if (prio_allocate)
642
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
643

644
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
645
				      protocol, prio, parent, q, chain);
646 647
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
648 649
			goto errout;
		}
650
		tp_created = 1;
651 652
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
653
		goto errout;
654
	}
L
Linus Torvalds 已提交
655 656 657

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

658
	if (!fh) {
L
Linus Torvalds 已提交
659
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
660
			tcf_chain_tp_remove(chain, &chain_info, tp);
661 662
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
663
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
664 665 666 667
			err = 0;
			goto errout;
		}

668
		if (n->nlmsg_type != RTM_NEWTFILTER ||
669 670
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
671
			goto errout;
672
		}
L
Linus Torvalds 已提交
673
	} else {
674 675
		bool last;

L
Linus Torvalds 已提交
676
		switch (n->nlmsg_type) {
677
		case RTM_NEWTFILTER:
678 679
			if (n->nlmsg_flags & NLM_F_EXCL) {
				if (tp_created)
680
					tcf_proto_destroy(tp);
681
				err = -EEXIST;
L
Linus Torvalds 已提交
682
				goto errout;
683
			}
L
Linus Torvalds 已提交
684 685
			break;
		case RTM_DELTFILTER:
686 687
			err = tfilter_del_notify(net, skb, n, tp, fh, false,
						 &last);
688 689
			if (err)
				goto errout;
690
			if (last) {
691
				tcf_chain_tp_remove(chain, &chain_info, tp);
692 693
				tcf_proto_destroy(tp);
			}
694
			goto errout;
L
Linus Torvalds 已提交
695
		case RTM_GETTFILTER:
J
Jamal Hadi Salim 已提交
696
			err = tfilter_notify(net, skb, n, tp, fh,
697
					     RTM_NEWTFILTER, true);
L
Linus Torvalds 已提交
698 699 700 701 702 703 704
			goto errout;
		default:
			err = -EINVAL;
			goto errout;
		}
	}

705 706
	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);
707
	if (err == 0) {
708 709
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
710
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
711 712
	} else {
		if (tp_created)
713
			tcf_proto_destroy(tp);
714
	}
L
Linus Torvalds 已提交
715 716

errout:
717 718
	if (chain)
		tcf_chain_put(chain);
L
Linus Torvalds 已提交
719 720 721 722 723 724 725 726
	if (cl)
		cops->put(q, cl);
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

727
struct tcf_dump_args {
L
Linus Torvalds 已提交
728 729 730 731 732
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

733
static int tcf_node_dump(struct tcf_proto *tp, void *n, struct tcf_walker *arg)
L
Linus Torvalds 已提交
734
{
735
	struct tcf_dump_args *a = (void *)arg;
736
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
737

738
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
739 740
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
741 742
}

743
static bool tcf_chain_dump(struct tcf_chain *chain, struct sk_buff *skb,
744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769
			   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)
770
				return false;
771 772 773 774 775 776 777 778 779 780 781 782 783 784

			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)
785
			return false;
786
	}
787
	return true;
788 789
}

E
Eric Dumazet 已提交
790
/* called with RTNL */
L
Linus Torvalds 已提交
791 792
static int tc_dump_tfilter(struct sk_buff *skb, struct netlink_callback *cb)
{
793
	struct net *net = sock_net(skb->sk);
794
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
795 796
	struct net_device *dev;
	struct Qdisc *q;
797
	struct tcf_block *block;
798
	struct tcf_chain *chain;
799
	struct tcmsg *tcm = nlmsg_data(cb->nlh);
L
Linus Torvalds 已提交
800
	unsigned long cl = 0;
801
	const struct Qdisc_class_ops *cops;
802 803
	long index_start;
	long index;
804
	int err;
L
Linus Torvalds 已提交
805

806
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
807
		return skb->len;
808 809 810 811 812

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

E
Eric Dumazet 已提交
813 814
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
815 816 817
		return skb->len;

	if (!tcm->tcm_parent)
818
		q = dev->qdisc;
L
Linus Torvalds 已提交
819 820 821 822
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
823 824
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
825
		goto errout;
826
	if (!cops->tcf_block)
827
		goto errout;
L
Linus Torvalds 已提交
828 829 830 831 832
	if (TC_H_MIN(tcm->tcm_parent)) {
		cl = cops->get(q, tcm->tcm_parent);
		if (cl == 0)
			goto errout;
	}
833 834
	block = cops->tcf_block(q, cl);
	if (!block)
L
Linus Torvalds 已提交
835 836
		goto errout;

837 838
	index_start = cb->args[0];
	index = 0;
839 840 841 842 843 844 845 846 847

	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;
	}

848
	cb->args[0] = index;
L
Linus Torvalds 已提交
849 850 851 852 853 854 855 856

errout:
	if (cl)
		cops->put(q, cl);
out:
	return skb->len;
}

857
void tcf_exts_destroy(struct tcf_exts *exts)
L
Linus Torvalds 已提交
858 859
{
#ifdef CONFIG_NET_CLS_ACT
860 861 862 863 864 865
	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 已提交
866 867
#endif
}
868
EXPORT_SYMBOL(tcf_exts_destroy);
L
Linus Torvalds 已提交
869

870
int tcf_exts_validate(struct net *net, struct tcf_proto *tp, struct nlattr **tb,
J
Jamal Hadi Salim 已提交
871
		      struct nlattr *rate_tlv, struct tcf_exts *exts, bool ovr)
L
Linus Torvalds 已提交
872 873 874 875 876
{
#ifdef CONFIG_NET_CLS_ACT
	{
		struct tc_action *act;

877
		if (exts->police && tb[exts->police]) {
878 879 880
			act = tcf_action_init_1(net, tp, tb[exts->police],
						rate_tlv, "police", ovr,
						TCA_ACT_BIND);
881 882
			if (IS_ERR(act))
				return PTR_ERR(act);
L
Linus Torvalds 已提交
883

884
			act->type = exts->type = TCA_OLD_COMPAT;
885 886
			exts->actions[0] = act;
			exts->nr_actions = 1;
887
		} else if (exts->action && tb[exts->action]) {
888 889 890
			LIST_HEAD(actions);
			int err, i = 0;

891 892
			err = tcf_action_init(net, tp, tb[exts->action],
					      rate_tlv, NULL, ovr, TCA_ACT_BIND,
J
Jamal Hadi Salim 已提交
893
					      &actions);
894 895
			if (err)
				return err;
896 897 898
			list_for_each_entry(act, &actions, list)
				exts->actions[i++] = act;
			exts->nr_actions = i;
L
Linus Torvalds 已提交
899 900 901
		}
	}
#else
902 903
	if ((exts->action && tb[exts->action]) ||
	    (exts->police && tb[exts->police]))
L
Linus Torvalds 已提交
904 905 906 907 908
		return -EOPNOTSUPP;
#endif

	return 0;
}
909
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
910

911
void tcf_exts_change(struct tcf_exts *dst, struct tcf_exts *src)
L
Linus Torvalds 已提交
912 913
{
#ifdef CONFIG_NET_CLS_ACT
914 915
	struct tcf_exts old = *dst;

916
	*dst = *src;
917
	tcf_exts_destroy(&old);
L
Linus Torvalds 已提交
918 919
#endif
}
920
EXPORT_SYMBOL(tcf_exts_change);
L
Linus Torvalds 已提交
921

922 923 924 925 926 927 928 929 930
#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
931

932
int tcf_exts_dump(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
933 934
{
#ifdef CONFIG_NET_CLS_ACT
935 936
	struct nlattr *nest;

937
	if (exts->action && tcf_exts_has_actions(exts)) {
L
Linus Torvalds 已提交
938 939 940 941 942
		/*
		 * 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
		 */
943
		if (exts->type != TCA_OLD_COMPAT) {
944 945
			LIST_HEAD(actions);

946
			nest = nla_nest_start(skb, exts->action);
947 948
			if (nest == NULL)
				goto nla_put_failure;
949 950 951

			tcf_exts_to_list(exts, &actions);
			if (tcf_action_dump(skb, &actions, 0, 0) < 0)
952
				goto nla_put_failure;
953
			nla_nest_end(skb, nest);
954
		} else if (exts->police) {
955
			struct tc_action *act = tcf_exts_first_act(exts);
956
			nest = nla_nest_start(skb, exts->police);
957
			if (nest == NULL || !act)
958
				goto nla_put_failure;
959
			if (tcf_action_dump_old(skb, act, 0, 0) < 0)
960
				goto nla_put_failure;
961
			nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
962 963 964
		}
	}
	return 0;
965 966 967

nla_put_failure:
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
968
	return -1;
969 970 971
#else
	return 0;
#endif
L
Linus Torvalds 已提交
972
}
973
EXPORT_SYMBOL(tcf_exts_dump);
L
Linus Torvalds 已提交
974

975

976
int tcf_exts_dump_stats(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
977 978
{
#ifdef CONFIG_NET_CLS_ACT
979
	struct tc_action *a = tcf_exts_first_act(exts);
980
	if (a != NULL && tcf_action_copy_stats(skb, a, 1) < 0)
981
		return -1;
L
Linus Torvalds 已提交
982 983 984
#endif
	return 0;
}
985
EXPORT_SYMBOL(tcf_exts_dump_stats);
L
Linus Torvalds 已提交
986

987 988 989 990 991 992 993
int tcf_exts_get_dev(struct net_device *dev, struct tcf_exts *exts,
		     struct net_device **hw_dev)
{
#ifdef CONFIG_NET_CLS_ACT
	const struct tc_action *a;
	LIST_HEAD(actions);

994
	if (!tcf_exts_has_actions(exts))
995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
		return -EINVAL;

	tcf_exts_to_list(exts, &actions);
	list_for_each_entry(a, &actions, list) {
		if (a->ops->get_dev) {
			a->ops->get_dev(a, dev_net(dev), hw_dev);
			break;
		}
	}
	if (*hw_dev)
		return 0;
#endif
	return -EOPNOTSUPP;
}
EXPORT_SYMBOL(tcf_exts_get_dev);

L
Linus Torvalds 已提交
1011 1012
static int __init tc_filter_init(void)
{
1013 1014
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, 0);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, 0);
1015
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
1016
		      tc_dump_tfilter, 0);
L
Linus Torvalds 已提交
1017 1018 1019 1020 1021

	return 0;
}

subsys_initcall(tc_filter_init);