cls_api.c 20.4 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
static int tfilter_notify(struct net *net, struct sk_buff *oskb,
			  struct nlmsghdr *n, struct tcf_proto *tp,
105
			  unsigned long fh, int event, bool unicast);
L
Linus Torvalds 已提交
106

107 108
static void tfilter_notify_chain(struct net *net, struct sk_buff *oskb,
				 struct nlmsghdr *n,
109
				 struct tcf_chain *chain, int event)
110 111 112
{
	struct tcf_proto *tp;

113 114
	for (tp = rtnl_dereference(chain->filter_chain);
	     tp; tp = rtnl_dereference(tp->next))
115
		tfilter_notify(net, oskb, n, tp, 0, event, false);
116
}
L
Linus Torvalds 已提交
117 118 119

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

120
static inline u32 tcf_auto_prio(struct tcf_proto *tp)
L
Linus Torvalds 已提交
121
{
122
	u32 first = TC_H_MAKE(0xC0000000U, 0U);
L
Linus Torvalds 已提交
123 124

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

127
	return TC_H_MAJ(first);
L
Linus Torvalds 已提交
128 129
}

130
static struct tcf_proto *tcf_proto_create(const char *kind, u32 protocol,
131 132
					  u32 prio, u32 parent, struct Qdisc *q,
					  struct tcf_block *block)
133 134 135 136 137 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
{
	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;
168
	tp->block = block;
169 170 171 172 173 174 175 176 177 178 179 180 181

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

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

182
static void tcf_proto_destroy(struct tcf_proto *tp)
183
{
184 185 186
	tp->ops->destroy(tp);
	module_put(tp->ops->owner);
	kfree_rcu(tp, rcu);
187 188
}

189 190 191 192 193 194
static struct tcf_chain *tcf_chain_create(void)
{
	return kzalloc(sizeof(struct tcf_chain), GFP_KERNEL);
}

static void tcf_chain_destroy(struct tcf_chain *chain)
195 196 197
{
	struct tcf_proto *tp;

198 199
	while ((tp = rtnl_dereference(chain->filter_chain)) != NULL) {
		RCU_INIT_POINTER(chain->filter_chain, tp->next);
200
		tcf_proto_destroy(tp);
201
	}
202 203 204 205 206 207 208 209
	kfree(chain);
}

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;
210
}
211 212 213 214 215

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);
216
	int err;
217 218 219

	if (!block)
		return -ENOMEM;
220 221 222 223 224 225
	block->chain = tcf_chain_create();
	if (!block->chain) {
		err = -ENOMEM;
		goto err_chain_create;
	}
	tcf_chain_filter_chain_ptr_set(block->chain, p_filter_chain);
226 227
	*p_block = block;
	return 0;
228 229 230 231

err_chain_create:
	kfree(block);
	return err;
232 233 234 235 236 237 238
}
EXPORT_SYMBOL(tcf_block_get);

void tcf_block_put(struct tcf_block *block)
{
	if (!block)
		return;
239
	tcf_chain_destroy(block->chain);
240 241 242
	kfree(block);
}
EXPORT_SYMBOL(tcf_block_put);
243

244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 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;
	const struct tcf_proto *old_tp = tp;
	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
		if (unlikely(err == TC_ACT_RECLASSIFY && !compat_mode))
			goto reset;
#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;
	}

	tp = old_tp;
	protocol = tc_skb_protocol(skb);
	goto reclassify;
#endif
}
EXPORT_SYMBOL(tcf_classify);

292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350
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)
		*chain->p_filter_chain = tp;
	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)
		*chain->p_filter_chain = next;
	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;
}

L
Linus Torvalds 已提交
351 352
/* Add/change/delete/get a filter node */

353 354
static int tc_ctl_tfilter(struct sk_buff *skb, struct nlmsghdr *n,
			  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
355
{
356
	struct net *net = sock_net(skb->sk);
357
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
358 359 360
	struct tcmsg *t;
	u32 protocol;
	u32 prio;
361
	bool prio_allocate;
L
Linus Torvalds 已提交
362 363 364
	u32 parent;
	struct net_device *dev;
	struct Qdisc  *q;
365 366
	struct tcf_chain_info chain_info;
	struct tcf_chain *chain;
367
	struct tcf_block *block;
L
Linus Torvalds 已提交
368
	struct tcf_proto *tp;
369
	const struct Qdisc_class_ops *cops;
L
Linus Torvalds 已提交
370 371 372
	unsigned long cl;
	unsigned long fh;
	int err;
373
	int tp_created;
L
Linus Torvalds 已提交
374

375
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
376
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
377
		return -EPERM;
378

L
Linus Torvalds 已提交
379
replay:
380 381
	tp_created = 0;

382
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
383 384 385
	if (err < 0)
		return err;

386
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
387 388
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
389
	prio_allocate = false;
L
Linus Torvalds 已提交
390 391 392 393
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
394 395
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
396
			if (protocol || t->tcm_handle || tca[TCA_KIND])
397 398 399 400 401 402 403 404
				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);
405
				prio_allocate = true;
406 407 408 409
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
410
			return -ENOENT;
411
		}
L
Linus Torvalds 已提交
412 413 414 415 416
	}

	/* Find head of filter chain. */

	/* Find link */
417
	dev = __dev_get_by_index(net, t->tcm_ifindex);
418
	if (dev == NULL)
L
Linus Torvalds 已提交
419 420 421 422
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
423
		q = dev->qdisc;
L
Linus Torvalds 已提交
424
		parent = q->handle;
425 426 427 428 429
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
430 431

	/* Is it classful? */
E
Eric Dumazet 已提交
432 433
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
434 435
		return -EINVAL;

436
	if (!cops->tcf_block)
437 438
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
439 440 441 442 443 444 445 446
	/* 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 */
447 448
	block = cops->tcf_block(q, cl);
	if (!block) {
449
		err = -EINVAL;
L
Linus Torvalds 已提交
450
		goto errout;
451
	}
452
	chain = block->chain;
453

454 455
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
456
		tcf_chain_destroy(chain);
457 458 459
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
460

461 462 463 464 465
	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 已提交
466 467 468 469 470
	}

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

471 472
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
473
			goto errout;
474
		}
L
Linus Torvalds 已提交
475

E
Eric Dumazet 已提交
476
		if (n->nlmsg_type != RTM_NEWTFILTER ||
477 478
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
479
			goto errout;
480
		}
L
Linus Torvalds 已提交
481

482
		if (prio_allocate)
483
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
484

485
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
486
				      protocol, prio, parent, q, block);
487 488
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
489 490
			goto errout;
		}
491
		tp_created = 1;
492 493
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
494
		goto errout;
495
	}
L
Linus Torvalds 已提交
496 497 498 499 500

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

	if (fh == 0) {
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
501
			tcf_chain_tp_remove(chain, &chain_info, tp);
502 503
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
504
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
505 506 507 508
			err = 0;
			goto errout;
		}

509
		if (n->nlmsg_type != RTM_NEWTFILTER ||
510 511
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
512
			goto errout;
513
		}
L
Linus Torvalds 已提交
514
	} else {
515 516
		bool last;

L
Linus Torvalds 已提交
517
		switch (n->nlmsg_type) {
518
		case RTM_NEWTFILTER:
519 520
			if (n->nlmsg_flags & NLM_F_EXCL) {
				if (tp_created)
521
					tcf_proto_destroy(tp);
522
				err = -EEXIST;
L
Linus Torvalds 已提交
523
				goto errout;
524
			}
L
Linus Torvalds 已提交
525 526
			break;
		case RTM_DELTFILTER:
527
			err = tp->ops->delete(tp, fh, &last);
528 529 530 531
			if (err)
				goto errout;
			tfilter_notify(net, skb, n, tp, t->tcm_handle,
				       RTM_DELTFILTER, false);
532
			if (last) {
533
				tcf_chain_tp_remove(chain, &chain_info, tp);
534 535
				tcf_proto_destroy(tp);
			}
536
			goto errout;
L
Linus Torvalds 已提交
537
		case RTM_GETTFILTER:
J
Jamal Hadi Salim 已提交
538
			err = tfilter_notify(net, skb, n, tp, fh,
539
					     RTM_NEWTFILTER, true);
L
Linus Torvalds 已提交
540 541 542 543 544 545 546
			goto errout;
		default:
			err = -EINVAL;
			goto errout;
		}
	}

547 548
	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);
549
	if (err == 0) {
550 551
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
552
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
553 554
	} else {
		if (tp_created)
555
			tcf_proto_destroy(tp);
556
	}
L
Linus Torvalds 已提交
557 558 559 560 561 562 563 564 565 566

errout:
	if (cl)
		cops->put(q, cl);
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

567 568 569
static int tcf_fill_node(struct net *net, struct sk_buff *skb,
			 struct tcf_proto *tp, unsigned long fh, u32 portid,
			 u32 seq, u16 flags, int event)
L
Linus Torvalds 已提交
570 571 572
{
	struct tcmsg *tcm;
	struct nlmsghdr  *nlh;
573
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
574

575
	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
576 577 578
	if (!nlh)
		goto out_nlmsg_trim;
	tcm = nlmsg_data(nlh);
L
Linus Torvalds 已提交
579
	tcm->tcm_family = AF_UNSPEC;
580
	tcm->tcm__pad1 = 0;
J
Jiri Pirko 已提交
581
	tcm->tcm__pad2 = 0;
582
	tcm->tcm_ifindex = qdisc_dev(tp->q)->ifindex;
L
Linus Torvalds 已提交
583 584
	tcm->tcm_parent = tp->classid;
	tcm->tcm_info = TC_H_MAKE(tp->prio, tp->protocol);
585 586
	if (nla_put_string(skb, TCA_KIND, tp->ops->kind))
		goto nla_put_failure;
L
Linus Torvalds 已提交
587 588 589
	tcm->tcm_handle = fh;
	if (RTM_DELTFILTER != event) {
		tcm->tcm_handle = 0;
590
		if (tp->ops->dump && tp->ops->dump(net, tp, fh, skb, tcm) < 0)
591
			goto nla_put_failure;
L
Linus Torvalds 已提交
592
	}
593
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
L
Linus Torvalds 已提交
594 595
	return skb->len;

596
out_nlmsg_trim:
597
nla_put_failure:
598
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
599 600 601
	return -1;
}

602 603
static int tfilter_notify(struct net *net, struct sk_buff *oskb,
			  struct nlmsghdr *n, struct tcf_proto *tp,
604
			  unsigned long fh, int event, bool unicast)
L
Linus Torvalds 已提交
605 606
{
	struct sk_buff *skb;
607
	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
L
Linus Torvalds 已提交
608 609 610 611 612

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

613 614
	if (tcf_fill_node(net, skb, tp, fh, portid, n->nlmsg_seq,
			  n->nlmsg_flags, event) <= 0) {
L
Linus Torvalds 已提交
615 616 617 618
		kfree_skb(skb);
		return -EINVAL;
	}

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

622
	return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
623
			      n->nlmsg_flags & NLM_F_ECHO);
L
Linus Torvalds 已提交
624 625
}

626
struct tcf_dump_args {
L
Linus Torvalds 已提交
627 628 629 630 631
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

632 633
static int tcf_node_dump(struct tcf_proto *tp, unsigned long n,
			 struct tcf_walker *arg)
L
Linus Torvalds 已提交
634
{
635
	struct tcf_dump_args *a = (void *)arg;
636
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
637

638
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
639 640
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
641 642
}

643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688
static void tcf_chain_dump(struct tcf_chain *chain, struct sk_buff *skb,
			   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)
				break;

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

E
Eric Dumazet 已提交
689
/* called with RTNL */
L
Linus Torvalds 已提交
690 691
static int tc_dump_tfilter(struct sk_buff *skb, struct netlink_callback *cb)
{
692
	struct net *net = sock_net(skb->sk);
L
Linus Torvalds 已提交
693 694
	struct net_device *dev;
	struct Qdisc *q;
695
	struct tcf_block *block;
696
	struct tcf_chain *chain;
697
	struct tcmsg *tcm = nlmsg_data(cb->nlh);
L
Linus Torvalds 已提交
698
	unsigned long cl = 0;
699
	const struct Qdisc_class_ops *cops;
700 701
	long index_start;
	long index;
L
Linus Torvalds 已提交
702

703
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
704
		return skb->len;
E
Eric Dumazet 已提交
705 706
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
707 708 709
		return skb->len;

	if (!tcm->tcm_parent)
710
		q = dev->qdisc;
L
Linus Torvalds 已提交
711 712 713 714
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
715 716
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
717
		goto errout;
718
	if (!cops->tcf_block)
719
		goto errout;
L
Linus Torvalds 已提交
720 721 722 723 724
	if (TC_H_MIN(tcm->tcm_parent)) {
		cl = cops->get(q, tcm->tcm_parent);
		if (cl == 0)
			goto errout;
	}
725 726
	block = cops->tcf_block(q, cl);
	if (!block)
L
Linus Torvalds 已提交
727
		goto errout;
728
	chain = block->chain;
L
Linus Torvalds 已提交
729

730 731 732 733
	index_start = cb->args[0];
	index = 0;
	tcf_chain_dump(chain, skb, cb, index_start, &index);
	cb->args[0] = index;
L
Linus Torvalds 已提交
734 735 736 737 738 739 740 741

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

742
void tcf_exts_destroy(struct tcf_exts *exts)
L
Linus Torvalds 已提交
743 744
{
#ifdef CONFIG_NET_CLS_ACT
745 746 747 748 749 750
	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 已提交
751 752
#endif
}
753
EXPORT_SYMBOL(tcf_exts_destroy);
L
Linus Torvalds 已提交
754

755
int tcf_exts_validate(struct net *net, struct tcf_proto *tp, struct nlattr **tb,
J
Jamal Hadi Salim 已提交
756
		      struct nlattr *rate_tlv, struct tcf_exts *exts, bool ovr)
L
Linus Torvalds 已提交
757 758 759 760 761
{
#ifdef CONFIG_NET_CLS_ACT
	{
		struct tc_action *act;

762 763
		if (exts->police && tb[exts->police]) {
			act = tcf_action_init_1(net, tb[exts->police], rate_tlv,
J
Jamal Hadi Salim 已提交
764
						"police", ovr, TCA_ACT_BIND);
765 766
			if (IS_ERR(act))
				return PTR_ERR(act);
L
Linus Torvalds 已提交
767

768
			act->type = exts->type = TCA_OLD_COMPAT;
769 770
			exts->actions[0] = act;
			exts->nr_actions = 1;
771
		} else if (exts->action && tb[exts->action]) {
772 773 774
			LIST_HEAD(actions);
			int err, i = 0;

775
			err = tcf_action_init(net, tb[exts->action], rate_tlv,
J
Jamal Hadi Salim 已提交
776 777
					      NULL, ovr, TCA_ACT_BIND,
					      &actions);
778 779
			if (err)
				return err;
780 781 782
			list_for_each_entry(act, &actions, list)
				exts->actions[i++] = act;
			exts->nr_actions = i;
L
Linus Torvalds 已提交
783 784 785
		}
	}
#else
786 787
	if ((exts->action && tb[exts->action]) ||
	    (exts->police && tb[exts->police]))
L
Linus Torvalds 已提交
788 789 790 791 792
		return -EOPNOTSUPP;
#endif

	return 0;
}
793
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
794

795 796
void tcf_exts_change(struct tcf_proto *tp, struct tcf_exts *dst,
		     struct tcf_exts *src)
L
Linus Torvalds 已提交
797 798
{
#ifdef CONFIG_NET_CLS_ACT
799 800
	struct tcf_exts old = *dst;

801
	tcf_tree_lock(tp);
802 803
	dst->nr_actions = src->nr_actions;
	dst->actions = src->actions;
804
	dst->type = src->type;
805
	tcf_tree_unlock(tp);
806 807

	tcf_exts_destroy(&old);
L
Linus Torvalds 已提交
808 809
#endif
}
810
EXPORT_SYMBOL(tcf_exts_change);
L
Linus Torvalds 已提交
811

812 813 814 815 816 817 818 819 820
#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
821

822
int tcf_exts_dump(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
823 824
{
#ifdef CONFIG_NET_CLS_ACT
825 826
	struct nlattr *nest;

827
	if (exts->action && exts->nr_actions) {
L
Linus Torvalds 已提交
828 829 830 831 832
		/*
		 * 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
		 */
833
		if (exts->type != TCA_OLD_COMPAT) {
834 835
			LIST_HEAD(actions);

836
			nest = nla_nest_start(skb, exts->action);
837 838
			if (nest == NULL)
				goto nla_put_failure;
839 840 841

			tcf_exts_to_list(exts, &actions);
			if (tcf_action_dump(skb, &actions, 0, 0) < 0)
842
				goto nla_put_failure;
843
			nla_nest_end(skb, nest);
844
		} else if (exts->police) {
845
			struct tc_action *act = tcf_exts_first_act(exts);
846
			nest = nla_nest_start(skb, exts->police);
847
			if (nest == NULL || !act)
848
				goto nla_put_failure;
849
			if (tcf_action_dump_old(skb, act, 0, 0) < 0)
850
				goto nla_put_failure;
851
			nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
852 853 854
		}
	}
	return 0;
855 856 857

nla_put_failure:
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
858
	return -1;
859 860 861
#else
	return 0;
#endif
L
Linus Torvalds 已提交
862
}
863
EXPORT_SYMBOL(tcf_exts_dump);
L
Linus Torvalds 已提交
864

865

866
int tcf_exts_dump_stats(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
867 868
{
#ifdef CONFIG_NET_CLS_ACT
869
	struct tc_action *a = tcf_exts_first_act(exts);
870
	if (a != NULL && tcf_action_copy_stats(skb, a, 1) < 0)
871
		return -1;
L
Linus Torvalds 已提交
872 873 874
#endif
	return 0;
}
875
EXPORT_SYMBOL(tcf_exts_dump_stats);
L
Linus Torvalds 已提交
876

877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900
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);

	if (tc_no_actions(exts))
		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 已提交
901 902
static int __init tc_filter_init(void)
{
903 904
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, NULL);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, NULL);
905
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
906
		      tc_dump_tfilter, NULL);
L
Linus Torvalds 已提交
907 908 909 910 911

	return 0;
}

subsys_initcall(tc_filter_init);