cls_api.c 22.8 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
					  u32 prio, u32 parent, struct Qdisc *q,
132
					  struct tcf_chain *chain)
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->chain = chain;
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
static struct tcf_chain *tcf_chain_create(struct tcf_block *block,
					  u32 chain_index)
191
{
192 193 194 195 196 197 198 199 200 201
	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;
202 203
}

J
Jiri Pirko 已提交
204
static void tcf_chain_flush(struct tcf_chain *chain)
205 206 207
{
	struct tcf_proto *tp;

208
	if (chain->p_filter_chain)
J
Jiri Pirko 已提交
209
		RCU_INIT_POINTER(*chain->p_filter_chain, NULL);
210 211
	while ((tp = rtnl_dereference(chain->filter_chain)) != NULL) {
		RCU_INIT_POINTER(chain->filter_chain, tp->next);
212
		tcf_proto_destroy(tp);
213
	}
J
Jiri Pirko 已提交
214 215 216 217
}

static void tcf_chain_destroy(struct tcf_chain *chain)
{
218 219 220 221 222 223 224 225 226
	/* 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);
227 228
}

229 230
struct tcf_chain *tcf_chain_get(struct tcf_block *block, u32 chain_index,
				bool create)
231 232 233 234 235 236 237 238 239
{
	struct tcf_chain *chain;

	list_for_each_entry(chain, &block->chain_list, list) {
		if (chain->index == chain_index) {
			chain->refcnt++;
			return chain;
		}
	}
240 241 242 243
	if (create)
		return tcf_chain_create(block, chain_index);
	else
		return NULL;
244 245 246 247 248 249 250 251 252 253 254 255 256
}
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);

257 258 259 260 261
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;
262
}
263 264 265 266 267

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);
268
	struct tcf_chain *chain;
269
	int err;
270 271 272

	if (!block)
		return -ENOMEM;
273 274 275 276
	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) {
277 278 279
		err = -ENOMEM;
		goto err_chain_create;
	}
280
	tcf_chain_filter_chain_ptr_set(chain, p_filter_chain);
281 282
	*p_block = block;
	return 0;
283 284 285 286

err_chain_create:
	kfree(block);
	return err;
287 288 289 290 291
}
EXPORT_SYMBOL(tcf_block_get);

void tcf_block_put(struct tcf_block *block)
{
292 293
	struct tcf_chain *chain, *tmp;

294 295
	if (!block)
		return;
296

297 298
	list_for_each_entry_safe(chain, tmp, &block->chain_list, list) {
		tcf_chain_flush(chain);
299
		tcf_chain_destroy(chain);
300
	}
301 302 303
	kfree(block);
}
EXPORT_SYMBOL(tcf_block_put);
304

305 306 307 308 309 310 311 312 313 314
/* 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;
315 316
	const struct tcf_proto *orig_tp = tp;
	const struct tcf_proto *first_tp;
317 318 319 320 321 322 323 324 325 326 327 328 329
	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
330
		if (unlikely(err == TC_ACT_RECLASSIFY && !compat_mode)) {
331
			first_tp = orig_tp;
332
			goto reset;
333
		} else if (unlikely(TC_ACT_EXT_CMP(err, TC_ACT_GOTO_CHAIN))) {
334
			first_tp = res->goto_tp;
335 336
			goto reset;
		}
337 338 339 340 341 342 343 344 345 346 347 348 349 350 351
#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;
	}

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

359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
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)
375
		rcu_assign_pointer(*chain->p_filter_chain, tp);
376 377 378 379 380 381 382 383 384 385 386
	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)
387
		RCU_INIT_POINTER(*chain->p_filter_chain, next);
388 389 390 391 392 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
	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 已提交
418 419
/* Add/change/delete/get a filter node */

420 421
static int tc_ctl_tfilter(struct sk_buff *skb, struct nlmsghdr *n,
			  struct netlink_ext_ack *extack)
L
Linus Torvalds 已提交
422
{
423
	struct net *net = sock_net(skb->sk);
424
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
425 426 427
	struct tcmsg *t;
	u32 protocol;
	u32 prio;
428
	bool prio_allocate;
L
Linus Torvalds 已提交
429
	u32 parent;
430
	u32 chain_index;
L
Linus Torvalds 已提交
431 432
	struct net_device *dev;
	struct Qdisc  *q;
433
	struct tcf_chain_info chain_info;
434
	struct tcf_chain *chain = NULL;
435
	struct tcf_block *block;
L
Linus Torvalds 已提交
436
	struct tcf_proto *tp;
437
	const struct Qdisc_class_ops *cops;
L
Linus Torvalds 已提交
438 439 440
	unsigned long cl;
	unsigned long fh;
	int err;
441
	int tp_created;
L
Linus Torvalds 已提交
442

443
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
444
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
445
		return -EPERM;
446

L
Linus Torvalds 已提交
447
replay:
448 449
	tp_created = 0;

450
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
451 452 453
	if (err < 0)
		return err;

454
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
455 456
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
457
	prio_allocate = false;
L
Linus Torvalds 已提交
458 459 460 461
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
462 463
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
464
			if (protocol || t->tcm_handle || tca[TCA_KIND])
465 466 467 468 469 470 471 472
				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);
473
				prio_allocate = true;
474 475 476 477
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
478
			return -ENOENT;
479
		}
L
Linus Torvalds 已提交
480 481 482 483 484
	}

	/* Find head of filter chain. */

	/* Find link */
485
	dev = __dev_get_by_index(net, t->tcm_ifindex);
486
	if (dev == NULL)
L
Linus Torvalds 已提交
487 488 489 490
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
491
		q = dev->qdisc;
L
Linus Torvalds 已提交
492
		parent = q->handle;
493 494 495 496 497
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
498 499

	/* Is it classful? */
E
Eric Dumazet 已提交
500 501
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
502 503
		return -EINVAL;

504
	if (!cops->tcf_block)
505 506
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
507 508 509 510 511 512 513 514
	/* 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 */
515 516
	block = cops->tcf_block(q, cl);
	if (!block) {
517
		err = -EINVAL;
L
Linus Torvalds 已提交
518
		goto errout;
519
	}
520 521 522 523 524 525

	chain_index = tca[TCA_CHAIN] ? nla_get_u32(tca[TCA_CHAIN]) : 0;
	if (chain_index > TC_ACT_EXT_VAL_MASK) {
		err = -EINVAL;
		goto errout;
	}
526 527
	chain = tcf_chain_get(block, chain_index,
			      n->nlmsg_type == RTM_NEWTFILTER);
528
	if (!chain) {
529
		err = n->nlmsg_type == RTM_NEWTFILTER ? -ENOMEM : -EINVAL;
530 531
		goto errout;
	}
532

533 534
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
J
Jiri Pirko 已提交
535
		tcf_chain_flush(chain);
536 537 538
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
539

540 541 542 543 544
	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 已提交
545 546 547 548 549
	}

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

550 551
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
552
			goto errout;
553
		}
L
Linus Torvalds 已提交
554

E
Eric Dumazet 已提交
555
		if (n->nlmsg_type != RTM_NEWTFILTER ||
556 557
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
558
			goto errout;
559
		}
L
Linus Torvalds 已提交
560

561
		if (prio_allocate)
562
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
563

564
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
565
				      protocol, prio, parent, q, chain);
566 567
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
568 569
			goto errout;
		}
570
		tp_created = 1;
571 572
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
573
		goto errout;
574
	}
L
Linus Torvalds 已提交
575 576 577 578 579

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

	if (fh == 0) {
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
580
			tcf_chain_tp_remove(chain, &chain_info, tp);
581 582
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
583
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
584 585 586 587
			err = 0;
			goto errout;
		}

588
		if (n->nlmsg_type != RTM_NEWTFILTER ||
589 590
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
591
			goto errout;
592
		}
L
Linus Torvalds 已提交
593
	} else {
594 595
		bool last;

L
Linus Torvalds 已提交
596
		switch (n->nlmsg_type) {
597
		case RTM_NEWTFILTER:
598 599
			if (n->nlmsg_flags & NLM_F_EXCL) {
				if (tp_created)
600
					tcf_proto_destroy(tp);
601
				err = -EEXIST;
L
Linus Torvalds 已提交
602
				goto errout;
603
			}
L
Linus Torvalds 已提交
604 605
			break;
		case RTM_DELTFILTER:
606
			err = tp->ops->delete(tp, fh, &last);
607 608 609 610
			if (err)
				goto errout;
			tfilter_notify(net, skb, n, tp, t->tcm_handle,
				       RTM_DELTFILTER, false);
611
			if (last) {
612
				tcf_chain_tp_remove(chain, &chain_info, tp);
613 614
				tcf_proto_destroy(tp);
			}
615
			goto errout;
L
Linus Torvalds 已提交
616
		case RTM_GETTFILTER:
J
Jamal Hadi Salim 已提交
617
			err = tfilter_notify(net, skb, n, tp, fh,
618
					     RTM_NEWTFILTER, true);
L
Linus Torvalds 已提交
619 620 621 622 623 624 625
			goto errout;
		default:
			err = -EINVAL;
			goto errout;
		}
	}

626 627
	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);
628
	if (err == 0) {
629 630
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
631
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
632 633
	} else {
		if (tp_created)
634
			tcf_proto_destroy(tp);
635
	}
L
Linus Torvalds 已提交
636 637

errout:
638 639
	if (chain)
		tcf_chain_put(chain);
L
Linus Torvalds 已提交
640 641 642 643 644 645 646 647
	if (cl)
		cops->put(q, cl);
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

648 649 650
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 已提交
651 652 653
{
	struct tcmsg *tcm;
	struct nlmsghdr  *nlh;
654
	unsigned char *b = skb_tail_pointer(skb);
L
Linus Torvalds 已提交
655

656
	nlh = nlmsg_put(skb, portid, seq, event, sizeof(*tcm), flags);
657 658 659
	if (!nlh)
		goto out_nlmsg_trim;
	tcm = nlmsg_data(nlh);
L
Linus Torvalds 已提交
660
	tcm->tcm_family = AF_UNSPEC;
661
	tcm->tcm__pad1 = 0;
J
Jiri Pirko 已提交
662
	tcm->tcm__pad2 = 0;
663
	tcm->tcm_ifindex = qdisc_dev(tp->q)->ifindex;
L
Linus Torvalds 已提交
664 665
	tcm->tcm_parent = tp->classid;
	tcm->tcm_info = TC_H_MAKE(tp->prio, tp->protocol);
666 667
	if (nla_put_string(skb, TCA_KIND, tp->ops->kind))
		goto nla_put_failure;
668 669
	if (nla_put_u32(skb, TCA_CHAIN, tp->chain->index))
		goto nla_put_failure;
L
Linus Torvalds 已提交
670 671 672
	tcm->tcm_handle = fh;
	if (RTM_DELTFILTER != event) {
		tcm->tcm_handle = 0;
673
		if (tp->ops->dump && tp->ops->dump(net, tp, fh, skb, tcm) < 0)
674
			goto nla_put_failure;
L
Linus Torvalds 已提交
675
	}
676
	nlh->nlmsg_len = skb_tail_pointer(skb) - b;
L
Linus Torvalds 已提交
677 678
	return skb->len;

679
out_nlmsg_trim:
680
nla_put_failure:
681
	nlmsg_trim(skb, b);
L
Linus Torvalds 已提交
682 683 684
	return -1;
}

685 686
static int tfilter_notify(struct net *net, struct sk_buff *oskb,
			  struct nlmsghdr *n, struct tcf_proto *tp,
687
			  unsigned long fh, int event, bool unicast)
L
Linus Torvalds 已提交
688 689
{
	struct sk_buff *skb;
690
	u32 portid = oskb ? NETLINK_CB(oskb).portid : 0;
L
Linus Torvalds 已提交
691 692 693 694 695

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

696 697
	if (tcf_fill_node(net, skb, tp, fh, portid, n->nlmsg_seq,
			  n->nlmsg_flags, event) <= 0) {
L
Linus Torvalds 已提交
698 699 700 701
		kfree_skb(skb);
		return -EINVAL;
	}

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

705
	return rtnetlink_send(skb, net, portid, RTNLGRP_TC,
706
			      n->nlmsg_flags & NLM_F_ECHO);
L
Linus Torvalds 已提交
707 708
}

709
struct tcf_dump_args {
L
Linus Torvalds 已提交
710 711 712 713 714
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

715 716
static int tcf_node_dump(struct tcf_proto *tp, unsigned long n,
			 struct tcf_walker *arg)
L
Linus Torvalds 已提交
717
{
718
	struct tcf_dump_args *a = (void *)arg;
719
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
720

721
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
722 723
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
724 725
}

726
static bool tcf_chain_dump(struct tcf_chain *chain, struct sk_buff *skb,
727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752
			   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)
753
				return false;
754 755 756 757 758 759 760 761 762 763 764 765 766 767

			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)
768
			return false;
769
	}
770
	return true;
771 772
}

E
Eric Dumazet 已提交
773
/* called with RTNL */
L
Linus Torvalds 已提交
774 775
static int tc_dump_tfilter(struct sk_buff *skb, struct netlink_callback *cb)
{
776
	struct net *net = sock_net(skb->sk);
777
	struct nlattr *tca[TCA_MAX + 1];
L
Linus Torvalds 已提交
778 779
	struct net_device *dev;
	struct Qdisc *q;
780
	struct tcf_block *block;
781
	struct tcf_chain *chain;
782
	struct tcmsg *tcm = nlmsg_data(cb->nlh);
L
Linus Torvalds 已提交
783
	unsigned long cl = 0;
784
	const struct Qdisc_class_ops *cops;
785 786
	long index_start;
	long index;
787
	int err;
L
Linus Torvalds 已提交
788

789
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
790
		return skb->len;
791 792 793 794 795

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

E
Eric Dumazet 已提交
796 797
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
798 799 800
		return skb->len;

	if (!tcm->tcm_parent)
801
		q = dev->qdisc;
L
Linus Torvalds 已提交
802 803 804 805
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
806 807
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
808
		goto errout;
809
	if (!cops->tcf_block)
810
		goto errout;
L
Linus Torvalds 已提交
811 812 813 814 815
	if (TC_H_MIN(tcm->tcm_parent)) {
		cl = cops->get(q, tcm->tcm_parent);
		if (cl == 0)
			goto errout;
	}
816 817
	block = cops->tcf_block(q, cl);
	if (!block)
L
Linus Torvalds 已提交
818 819
		goto errout;

820 821
	index_start = cb->args[0];
	index = 0;
822 823 824 825 826 827 828 829 830

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

831
	cb->args[0] = index;
L
Linus Torvalds 已提交
832 833 834 835 836 837 838 839

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

840
void tcf_exts_destroy(struct tcf_exts *exts)
L
Linus Torvalds 已提交
841 842
{
#ifdef CONFIG_NET_CLS_ACT
843 844 845 846 847 848
	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 已提交
849 850
#endif
}
851
EXPORT_SYMBOL(tcf_exts_destroy);
L
Linus Torvalds 已提交
852

853
int tcf_exts_validate(struct net *net, struct tcf_proto *tp, struct nlattr **tb,
J
Jamal Hadi Salim 已提交
854
		      struct nlattr *rate_tlv, struct tcf_exts *exts, bool ovr)
L
Linus Torvalds 已提交
855 856 857 858 859
{
#ifdef CONFIG_NET_CLS_ACT
	{
		struct tc_action *act;

860
		if (exts->police && tb[exts->police]) {
861 862 863
			act = tcf_action_init_1(net, tp, tb[exts->police],
						rate_tlv, "police", ovr,
						TCA_ACT_BIND);
864 865
			if (IS_ERR(act))
				return PTR_ERR(act);
L
Linus Torvalds 已提交
866

867
			act->type = exts->type = TCA_OLD_COMPAT;
868 869
			exts->actions[0] = act;
			exts->nr_actions = 1;
870
		} else if (exts->action && tb[exts->action]) {
871 872 873
			LIST_HEAD(actions);
			int err, i = 0;

874 875
			err = tcf_action_init(net, tp, tb[exts->action],
					      rate_tlv, NULL, ovr, TCA_ACT_BIND,
J
Jamal Hadi Salim 已提交
876
					      &actions);
877 878
			if (err)
				return err;
879 880 881
			list_for_each_entry(act, &actions, list)
				exts->actions[i++] = act;
			exts->nr_actions = i;
L
Linus Torvalds 已提交
882 883 884
		}
	}
#else
885 886
	if ((exts->action && tb[exts->action]) ||
	    (exts->police && tb[exts->police]))
L
Linus Torvalds 已提交
887 888 889 890 891
		return -EOPNOTSUPP;
#endif

	return 0;
}
892
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
893

894 895
void tcf_exts_change(struct tcf_proto *tp, struct tcf_exts *dst,
		     struct tcf_exts *src)
L
Linus Torvalds 已提交
896 897
{
#ifdef CONFIG_NET_CLS_ACT
898 899
	struct tcf_exts old = *dst;

900
	tcf_tree_lock(tp);
901 902
	dst->nr_actions = src->nr_actions;
	dst->actions = src->actions;
903
	dst->type = src->type;
904
	tcf_tree_unlock(tp);
905 906

	tcf_exts_destroy(&old);
L
Linus Torvalds 已提交
907 908
#endif
}
909
EXPORT_SYMBOL(tcf_exts_change);
L
Linus Torvalds 已提交
910

911 912 913 914 915 916 917 918 919
#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
920

921
int tcf_exts_dump(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
922 923
{
#ifdef CONFIG_NET_CLS_ACT
924 925
	struct nlattr *nest;

926
	if (exts->action && exts->nr_actions) {
L
Linus Torvalds 已提交
927 928 929 930 931
		/*
		 * 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
		 */
932
		if (exts->type != TCA_OLD_COMPAT) {
933 934
			LIST_HEAD(actions);

935
			nest = nla_nest_start(skb, exts->action);
936 937
			if (nest == NULL)
				goto nla_put_failure;
938 939 940

			tcf_exts_to_list(exts, &actions);
			if (tcf_action_dump(skb, &actions, 0, 0) < 0)
941
				goto nla_put_failure;
942
			nla_nest_end(skb, nest);
943
		} else if (exts->police) {
944
			struct tc_action *act = tcf_exts_first_act(exts);
945
			nest = nla_nest_start(skb, exts->police);
946
			if (nest == NULL || !act)
947
				goto nla_put_failure;
948
			if (tcf_action_dump_old(skb, act, 0, 0) < 0)
949
				goto nla_put_failure;
950
			nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
951 952 953
		}
	}
	return 0;
954 955 956

nla_put_failure:
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
957
	return -1;
958 959 960
#else
	return 0;
#endif
L
Linus Torvalds 已提交
961
}
962
EXPORT_SYMBOL(tcf_exts_dump);
L
Linus Torvalds 已提交
963

964

965
int tcf_exts_dump_stats(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
966 967
{
#ifdef CONFIG_NET_CLS_ACT
968
	struct tc_action *a = tcf_exts_first_act(exts);
969
	if (a != NULL && tcf_action_copy_stats(skb, a, 1) < 0)
970
		return -1;
L
Linus Torvalds 已提交
971 972 973
#endif
	return 0;
}
974
EXPORT_SYMBOL(tcf_exts_dump_stats);
L
Linus Torvalds 已提交
975

976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999
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 已提交
1000 1001
static int __init tc_filter_init(void)
{
1002 1003
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, NULL);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, NULL);
1004
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
1005
		      tc_dump_tfilter, NULL);
L
Linus Torvalds 已提交
1006 1007 1008 1009 1010

	return 0;
}

subsys_initcall(tc_filter_init);