cls_api.c 23.1 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
}

static void tcf_chain_destroy(struct tcf_chain *chain)
{
203 204 205 206 207 208 209 210 211
	/* May be already removed from the list by the previous call. */
	if (!list_empty(&chain->list))
		list_del_init(&chain->list);

	/* There might still be a reference held when we got here from
	 * tcf_block_put. Wait for the user to drop reference before free.
	 */
	if (!chain->refcnt)
		kfree(chain);
212 213
}

214 215
struct tcf_chain *tcf_chain_get(struct tcf_block *block, u32 chain_index,
				bool create)
216 217 218 219 220 221 222 223 224
{
	struct tcf_chain *chain;

	list_for_each_entry(chain, &block->chain_list, list) {
		if (chain->index == chain_index) {
			chain->refcnt++;
			return chain;
		}
	}
225 226 227 228
	if (create)
		return tcf_chain_create(block, chain_index);
	else
		return NULL;
229 230 231 232 233 234 235 236 237 238 239 240 241
}
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);

242 243 244 245 246
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;
247
}
248 249 250 251 252

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);
253
	struct tcf_chain *chain;
254
	int err;
255 256 257

	if (!block)
		return -ENOMEM;
258 259 260 261
	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) {
262 263 264
		err = -ENOMEM;
		goto err_chain_create;
	}
265
	tcf_chain_filter_chain_ptr_set(chain, p_filter_chain);
266 267
	*p_block = block;
	return 0;
268 269 270 271

err_chain_create:
	kfree(block);
	return err;
272 273 274 275 276
}
EXPORT_SYMBOL(tcf_block_get);

void tcf_block_put(struct tcf_block *block)
{
277 278
	struct tcf_chain *chain, *tmp;

279 280
	if (!block)
		return;
281

282 283
	list_for_each_entry_safe(chain, tmp, &block->chain_list, list) {
		tcf_chain_flush(chain);
284
		tcf_chain_destroy(chain);
285
	}
286 287 288
	kfree(block);
}
EXPORT_SYMBOL(tcf_block_put);
289

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

337
	tp = first_tp;
338 339 340 341 342 343
	protocol = tc_skb_protocol(skb);
	goto reclassify;
#endif
}
EXPORT_SYMBOL(tcf_classify);

344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359
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)
360
		rcu_assign_pointer(*chain->p_filter_chain, tp);
361 362 363 364 365 366 367 368 369 370 371
	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)
372
		RCU_INIT_POINTER(*chain->p_filter_chain, next);
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402
	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;
}

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 498 499 500 501 502 503 504 505
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 已提交
506 507
/* Add/change/delete/get a filter node */

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

531
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
532
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
533
		return -EPERM;
534

L
Linus Torvalds 已提交
535
replay:
536 537
	tp_created = 0;

538
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
539 540 541
	if (err < 0)
		return err;

542
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
543 544
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
545
	prio_allocate = false;
L
Linus Torvalds 已提交
546 547 548 549
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
550 551
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
552
			if (protocol || t->tcm_handle || tca[TCA_KIND])
553 554 555 556 557 558 559 560
				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);
561
				prio_allocate = true;
562 563 564 565
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
566
			return -ENOENT;
567
		}
L
Linus Torvalds 已提交
568 569 570 571 572
	}

	/* Find head of filter chain. */

	/* Find link */
573
	dev = __dev_get_by_index(net, t->tcm_ifindex);
574
	if (dev == NULL)
L
Linus Torvalds 已提交
575 576 577 578
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
579
		q = dev->qdisc;
L
Linus Torvalds 已提交
580
		parent = q->handle;
581 582 583 584 585
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
586 587

	/* Is it classful? */
E
Eric Dumazet 已提交
588 589
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
590 591
		return -EINVAL;

592
	if (!cops->tcf_block)
593 594
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
595 596
	/* Do we search for filter, attached to class? */
	if (TC_H_MIN(parent)) {
597
		cl = cops->find(q, parent);
L
Linus Torvalds 已提交
598 599 600 601 602
		if (cl == 0)
			return -ENOENT;
	}

	/* And the last stroke */
603 604
	block = cops->tcf_block(q, cl);
	if (!block) {
605
		err = -EINVAL;
L
Linus Torvalds 已提交
606
		goto errout;
607
	}
608 609 610 611 612 613

	chain_index = tca[TCA_CHAIN] ? nla_get_u32(tca[TCA_CHAIN]) : 0;
	if (chain_index > TC_ACT_EXT_VAL_MASK) {
		err = -EINVAL;
		goto errout;
	}
614 615
	chain = tcf_chain_get(block, chain_index,
			      n->nlmsg_type == RTM_NEWTFILTER);
616
	if (!chain) {
617
		err = n->nlmsg_type == RTM_NEWTFILTER ? -ENOMEM : -EINVAL;
618 619
		goto errout;
	}
620

621 622
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
J
Jiri Pirko 已提交
623
		tcf_chain_flush(chain);
624 625 626
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
627

628 629 630 631 632
	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 已提交
633 634 635 636 637
	}

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

638 639
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
640
			goto errout;
641
		}
L
Linus Torvalds 已提交
642

E
Eric Dumazet 已提交
643
		if (n->nlmsg_type != RTM_NEWTFILTER ||
644 645
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
646
			goto errout;
647
		}
L
Linus Torvalds 已提交
648

649
		if (prio_allocate)
650
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
651

652
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
653
				      protocol, prio, parent, q, chain);
654 655
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
656 657
			goto errout;
		}
658
		tp_created = 1;
659 660
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
661
		goto errout;
662
	}
L
Linus Torvalds 已提交
663 664 665

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

666
	if (!fh) {
L
Linus Torvalds 已提交
667
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
668
			tcf_chain_tp_remove(chain, &chain_info, tp);
669 670
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
671
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
672 673 674 675
			err = 0;
			goto errout;
		}

676
		if (n->nlmsg_type != RTM_NEWTFILTER ||
677 678
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
679
			goto errout;
680
		}
L
Linus Torvalds 已提交
681
	} else {
682 683
		bool last;

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

713 714
	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);
715
	if (err == 0) {
716 717
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
718
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
719 720
	} else {
		if (tp_created)
721
			tcf_proto_destroy(tp);
722
	}
L
Linus Torvalds 已提交
723 724

errout:
725 726
	if (chain)
		tcf_chain_put(chain);
L
Linus Torvalds 已提交
727 728 729 730 731 732
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

733
struct tcf_dump_args {
L
Linus Torvalds 已提交
734 735 736 737 738
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

739
static int tcf_node_dump(struct tcf_proto *tp, void *n, struct tcf_walker *arg)
L
Linus Torvalds 已提交
740
{
741
	struct tcf_dump_args *a = (void *)arg;
742
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
743

744
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
745 746
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
747 748
}

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

			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)
791
			return false;
792
	}
793
	return true;
794 795
}

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

812
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
813
		return skb->len;
814 815 816 817 818

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

E
Eric Dumazet 已提交
819 820
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
821 822 823
		return skb->len;

	if (!tcm->tcm_parent)
824
		q = dev->qdisc;
L
Linus Torvalds 已提交
825 826 827 828
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
829 830
	cops = q->ops->cl_ops;
	if (!cops)
831
		goto out;
832
	if (!cops->tcf_block)
833
		goto out;
L
Linus Torvalds 已提交
834
	if (TC_H_MIN(tcm->tcm_parent)) {
835
		cl = cops->find(q, tcm->tcm_parent);
L
Linus Torvalds 已提交
836
		if (cl == 0)
837
			goto out;
L
Linus Torvalds 已提交
838
	}
839 840
	block = cops->tcf_block(q, cl);
	if (!block)
841
		goto out;
L
Linus Torvalds 已提交
842

843 844
	index_start = cb->args[0];
	index = 0;
845 846 847 848 849 850 851 852 853

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

854
	cb->args[0] = index;
L
Linus Torvalds 已提交
855 856 857 858 859

out:
	return skb->len;
}

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

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

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

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

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

	return 0;
}
912
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
913

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

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

925 926 927 928 929 930 931 932 933
#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
934

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

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

949
			nest = nla_nest_start(skb, exts->action);
950 951
			if (nest == NULL)
				goto nla_put_failure;
952 953 954

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

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

978

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

990 991 992 993 994 995 996
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);

997
	if (!tcf_exts_has_actions(exts))
998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
		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 已提交
1014 1015
static int __init tc_filter_init(void)
{
1016 1017
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, 0);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, 0);
1018
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
1019
		      tc_dump_tfilter, 0);
L
Linus Torvalds 已提交
1020 1021 1022 1023 1024

	return 0;
}

subsys_initcall(tc_filter_init);