cls_api.c 23.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
static struct workqueue_struct *tc_filter_wq;

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

87 88 89 90
	/* Wait for outstanding call_rcu()s, if any, from a
	 * tcf_proto_ops's destroy() handler.
	 */
	rcu_barrier();
91
	flush_workqueue(tc_filter_wq);
92

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

106 107 108 109 110 111
bool tcf_queue_work(struct work_struct *work)
{
	return queue_work(tc_filter_wq, work);
}
EXPORT_SYMBOL(tcf_queue_work);

L
Linus Torvalds 已提交
112 113
/* Select new prio value from the range, managed by kernel. */

114
static inline u32 tcf_auto_prio(struct tcf_proto *tp)
L
Linus Torvalds 已提交
115
{
116
	u32 first = TC_H_MAKE(0xC0000000U, 0U);
L
Linus Torvalds 已提交
117 118

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

121
	return TC_H_MAJ(first);
L
Linus Torvalds 已提交
122 123
}

124
static struct tcf_proto *tcf_proto_create(const char *kind, u32 protocol,
125
					  u32 prio, u32 parent, struct Qdisc *q,
126
					  struct tcf_chain *chain)
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 153 154 155 156 157 158 159 160 161
{
	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;
162
	tp->chain = chain;
163 164 165 166 167 168 169 170 171 172 173 174 175

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

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

176
static void tcf_proto_destroy(struct tcf_proto *tp)
177
{
178 179 180
	tp->ops->destroy(tp);
	module_put(tp->ops->owner);
	kfree_rcu(tp, rcu);
181 182
}

183 184
static struct tcf_chain *tcf_chain_create(struct tcf_block *block,
					  u32 chain_index)
185
{
186 187 188 189 190 191 192 193
	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;
194
	chain->refcnt = 1;
195
	return chain;
196 197
}

J
Jiri Pirko 已提交
198
static void tcf_chain_flush(struct tcf_chain *chain)
199 200 201
{
	struct tcf_proto *tp;

202
	if (chain->p_filter_chain)
J
Jiri Pirko 已提交
203
		RCU_INIT_POINTER(*chain->p_filter_chain, NULL);
204 205
	while ((tp = rtnl_dereference(chain->filter_chain)) != NULL) {
		RCU_INIT_POINTER(chain->filter_chain, tp->next);
206
		tcf_chain_put(chain);
207
		tcf_proto_destroy(tp);
208
	}
J
Jiri Pirko 已提交
209 210 211 212
}

static void tcf_chain_destroy(struct tcf_chain *chain)
{
213 214 215
	list_del(&chain->list);
	kfree(chain);
}
216

217 218 219
static void tcf_chain_hold(struct tcf_chain *chain)
{
	++chain->refcnt;
220 221
}

222 223
struct tcf_chain *tcf_chain_get(struct tcf_block *block, u32 chain_index,
				bool create)
224 225 226 227
{
	struct tcf_chain *chain;

	list_for_each_entry(chain, &block->chain_list, list) {
228 229 230 231
		if (chain->index == chain_index) {
			tcf_chain_hold(chain);
			return chain;
		}
232
	}
233

234
	return create ? tcf_chain_create(block, chain_index) : NULL;
235 236 237 238 239
}
EXPORT_SYMBOL(tcf_chain_get);

void tcf_chain_put(struct tcf_chain *chain)
{
240
	if (--chain->refcnt == 0)
241 242 243 244
		tcf_chain_destroy(chain);
}
EXPORT_SYMBOL(tcf_chain_put);

245 246 247 248 249
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;
250
}
251 252 253 254 255

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);
256
	struct tcf_chain *chain;
257
	int err;
258 259 260

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

err_chain_create:
	kfree(block);
	return err;
275 276 277
}
EXPORT_SYMBOL(tcf_block_get);

278
static void tcf_block_put_final(struct work_struct *work)
279
{
280
	struct tcf_block *block = container_of(work, struct tcf_block, work);
281 282
	struct tcf_chain *chain, *tmp;

283
	rtnl_lock();
284
	/* Only chain 0 should be still here. */
285 286 287 288 289
	list_for_each_entry_safe(chain, tmp, &block->chain_list, list)
		tcf_chain_put(chain);
	rtnl_unlock();
	kfree(block);
}
290

291
/* XXX: Standalone actions are not allowed to jump to any chain, and bound
292 293
 * actions should be all removed after flushing. However, filters are now
 * destroyed in tc filter workqueue with RTNL lock, they can not race here.
294
 */
295
void tcf_block_put(struct tcf_block *block)
296
{
297
	struct tcf_chain *chain, *tmp;
298

299 300
	if (!block)
		return;
301

302
	list_for_each_entry_safe(chain, tmp, &block->chain_list, list)
303
		tcf_chain_flush(chain);
304

305 306 307 308
	INIT_WORK(&block->work, tcf_block_put_final);
	/* Wait for RCU callbacks to release the reference count and make
	 * sure their works have been queued before this.
	 */
309
	rcu_barrier();
310
	tcf_queue_work(&block->work);
311 312
}
EXPORT_SYMBOL(tcf_block_put);
313

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

361
	tp = first_tp;
362 363 364 365 366 367
	protocol = tc_skb_protocol(skb);
	goto reclassify;
#endif
}
EXPORT_SYMBOL(tcf_classify);

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

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)
397
		RCU_INIT_POINTER(*chain->p_filter_chain, next);
398
	RCU_INIT_POINTER(*chain_info->pprev, next);
399
	tcf_chain_put(chain);
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
}

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

429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531
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 已提交
532 533
/* Add/change/delete/get a filter node */

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

557
	if ((n->nlmsg_type != RTM_GETTFILTER) &&
558
	    !netlink_ns_capable(skb, net->user_ns, CAP_NET_ADMIN))
559
		return -EPERM;
560

L
Linus Torvalds 已提交
561
replay:
562 563
	tp_created = 0;

564
	err = nlmsg_parse(n, sizeof(*t), tca, TCA_MAX, NULL, extack);
565 566 567
	if (err < 0)
		return err;

568
	t = nlmsg_data(n);
L
Linus Torvalds 已提交
569 570
	protocol = TC_H_MIN(t->tcm_info);
	prio = TC_H_MAJ(t->tcm_info);
571
	prio_allocate = false;
L
Linus Torvalds 已提交
572 573 574 575
	parent = t->tcm_parent;
	cl = 0;

	if (prio == 0) {
576 577
		switch (n->nlmsg_type) {
		case RTM_DELTFILTER:
578
			if (protocol || t->tcm_handle || tca[TCA_KIND])
579 580 581 582 583 584 585 586
				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);
587
				prio_allocate = true;
588 589 590 591
				break;
			}
			/* fall-through */
		default:
L
Linus Torvalds 已提交
592
			return -ENOENT;
593
		}
L
Linus Torvalds 已提交
594 595 596 597 598
	}

	/* Find head of filter chain. */

	/* Find link */
599
	dev = __dev_get_by_index(net, t->tcm_ifindex);
600
	if (dev == NULL)
L
Linus Torvalds 已提交
601 602 603 604
		return -ENODEV;

	/* Find qdisc */
	if (!parent) {
605
		q = dev->qdisc;
L
Linus Torvalds 已提交
606
		parent = q->handle;
607 608 609 610 611
	} else {
		q = qdisc_lookup(dev, TC_H_MAJ(t->tcm_parent));
		if (q == NULL)
			return -EINVAL;
	}
L
Linus Torvalds 已提交
612 613

	/* Is it classful? */
E
Eric Dumazet 已提交
614 615
	cops = q->ops->cl_ops;
	if (!cops)
L
Linus Torvalds 已提交
616 617
		return -EINVAL;

618
	if (!cops->tcf_block)
619 620
		return -EOPNOTSUPP;

L
Linus Torvalds 已提交
621 622
	/* Do we search for filter, attached to class? */
	if (TC_H_MIN(parent)) {
623
		cl = cops->find(q, parent);
L
Linus Torvalds 已提交
624 625 626 627 628
		if (cl == 0)
			return -ENOENT;
	}

	/* And the last stroke */
629 630
	block = cops->tcf_block(q, cl);
	if (!block) {
631
		err = -EINVAL;
L
Linus Torvalds 已提交
632
		goto errout;
633
	}
634 635 636 637 638 639

	chain_index = tca[TCA_CHAIN] ? nla_get_u32(tca[TCA_CHAIN]) : 0;
	if (chain_index > TC_ACT_EXT_VAL_MASK) {
		err = -EINVAL;
		goto errout;
	}
640 641
	chain = tcf_chain_get(block, chain_index,
			      n->nlmsg_type == RTM_NEWTFILTER);
642
	if (!chain) {
643
		err = n->nlmsg_type == RTM_NEWTFILTER ? -ENOMEM : -EINVAL;
644 645
		goto errout;
	}
646

647 648
	if (n->nlmsg_type == RTM_DELTFILTER && prio == 0) {
		tfilter_notify_chain(net, skb, n, chain, RTM_DELTFILTER);
J
Jiri Pirko 已提交
649
		tcf_chain_flush(chain);
650 651 652
		err = 0;
		goto errout;
	}
L
Linus Torvalds 已提交
653

654 655 656 657 658
	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 已提交
659 660 661 662 663
	}

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

664 665
		if (tca[TCA_KIND] == NULL || !protocol) {
			err = -EINVAL;
L
Linus Torvalds 已提交
666
			goto errout;
667
		}
L
Linus Torvalds 已提交
668

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

675
		if (prio_allocate)
676
			prio = tcf_auto_prio(tcf_chain_tp_prev(&chain_info));
L
Linus Torvalds 已提交
677

678
		tp = tcf_proto_create(nla_data(tca[TCA_KIND]),
679
				      protocol, prio, parent, q, chain);
680 681
		if (IS_ERR(tp)) {
			err = PTR_ERR(tp);
L
Linus Torvalds 已提交
682 683
			goto errout;
		}
684
		tp_created = 1;
685 686
	} else if (tca[TCA_KIND] && nla_strcmp(tca[TCA_KIND], tp->ops->kind)) {
		err = -EINVAL;
L
Linus Torvalds 已提交
687
		goto errout;
688
	}
L
Linus Torvalds 已提交
689 690 691

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

692
	if (!fh) {
L
Linus Torvalds 已提交
693
		if (n->nlmsg_type == RTM_DELTFILTER && t->tcm_handle == 0) {
694
			tcf_chain_tp_remove(chain, &chain_info, tp);
695 696
			tfilter_notify(net, skb, n, tp, fh,
				       RTM_DELTFILTER, false);
697
			tcf_proto_destroy(tp);
L
Linus Torvalds 已提交
698 699 700 701
			err = 0;
			goto errout;
		}

702
		if (n->nlmsg_type != RTM_NEWTFILTER ||
703 704
		    !(n->nlmsg_flags & NLM_F_CREATE)) {
			err = -ENOENT;
L
Linus Torvalds 已提交
705
			goto errout;
706
		}
L
Linus Torvalds 已提交
707
	} else {
708 709
		bool last;

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

739 740
	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);
741
	if (err == 0) {
742 743
		if (tp_created)
			tcf_chain_tp_insert(chain, &chain_info, tp);
744
		tfilter_notify(net, skb, n, tp, fh, RTM_NEWTFILTER, false);
745 746
	} else {
		if (tp_created)
747
			tcf_proto_destroy(tp);
748
	}
L
Linus Torvalds 已提交
749 750

errout:
751 752
	if (chain)
		tcf_chain_put(chain);
L
Linus Torvalds 已提交
753 754 755 756 757 758
	if (err == -EAGAIN)
		/* Replay the request. */
		goto replay;
	return err;
}

759
struct tcf_dump_args {
L
Linus Torvalds 已提交
760 761 762 763 764
	struct tcf_walker w;
	struct sk_buff *skb;
	struct netlink_callback *cb;
};

765
static int tcf_node_dump(struct tcf_proto *tp, void *n, struct tcf_walker *arg)
L
Linus Torvalds 已提交
766
{
767
	struct tcf_dump_args *a = (void *)arg;
768
	struct net *net = sock_net(a->skb->sk);
L
Linus Torvalds 已提交
769

770
	return tcf_fill_node(net, a->skb, tp, n, NETLINK_CB(a->cb->skb).portid,
J
Jamal Hadi Salim 已提交
771 772
			     a->cb->nlh->nlmsg_seq, NLM_F_MULTI,
			     RTM_NEWTFILTER);
L
Linus Torvalds 已提交
773 774
}

775
static bool tcf_chain_dump(struct tcf_chain *chain, struct sk_buff *skb,
776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801
			   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)
802
				return false;
803 804 805 806 807 808 809 810 811 812 813 814 815 816

			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)
817
			return false;
818
	}
819
	return true;
820 821
}

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

838
	if (nlmsg_len(cb->nlh) < sizeof(*tcm))
L
Linus Torvalds 已提交
839
		return skb->len;
840 841 842 843 844

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

E
Eric Dumazet 已提交
845 846
	dev = __dev_get_by_index(net, tcm->tcm_ifindex);
	if (!dev)
L
Linus Torvalds 已提交
847 848 849
		return skb->len;

	if (!tcm->tcm_parent)
850
		q = dev->qdisc;
L
Linus Torvalds 已提交
851 852 853 854
	else
		q = qdisc_lookup(dev, TC_H_MAJ(tcm->tcm_parent));
	if (!q)
		goto out;
E
Eric Dumazet 已提交
855 856
	cops = q->ops->cl_ops;
	if (!cops)
857
		goto out;
858
	if (!cops->tcf_block)
859
		goto out;
L
Linus Torvalds 已提交
860
	if (TC_H_MIN(tcm->tcm_parent)) {
861
		cl = cops->find(q, tcm->tcm_parent);
L
Linus Torvalds 已提交
862
		if (cl == 0)
863
			goto out;
L
Linus Torvalds 已提交
864
	}
865 866
	block = cops->tcf_block(q, cl);
	if (!block)
867
		goto out;
L
Linus Torvalds 已提交
868

869 870
	index_start = cb->args[0];
	index = 0;
871 872 873 874 875 876 877 878 879

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

880
	cb->args[0] = index;
L
Linus Torvalds 已提交
881 882 883 884 885

out:
	return skb->len;
}

886
void tcf_exts_destroy(struct tcf_exts *exts)
L
Linus Torvalds 已提交
887 888
{
#ifdef CONFIG_NET_CLS_ACT
889 890
	LIST_HEAD(actions);

891
	ASSERT_RTNL();
892 893 894 895
	tcf_exts_to_list(exts, &actions);
	tcf_action_destroy(&actions, TCA_ACT_UNBIND);
	kfree(exts->actions);
	exts->nr_actions = 0;
L
Linus Torvalds 已提交
896 897
#endif
}
898
EXPORT_SYMBOL(tcf_exts_destroy);
L
Linus Torvalds 已提交
899

900
int tcf_exts_validate(struct net *net, struct tcf_proto *tp, struct nlattr **tb,
J
Jamal Hadi Salim 已提交
901
		      struct nlattr *rate_tlv, struct tcf_exts *exts, bool ovr)
L
Linus Torvalds 已提交
902 903 904 905 906
{
#ifdef CONFIG_NET_CLS_ACT
	{
		struct tc_action *act;

907
		if (exts->police && tb[exts->police]) {
908 909 910
			act = tcf_action_init_1(net, tp, tb[exts->police],
						rate_tlv, "police", ovr,
						TCA_ACT_BIND);
911 912
			if (IS_ERR(act))
				return PTR_ERR(act);
L
Linus Torvalds 已提交
913

914
			act->type = exts->type = TCA_OLD_COMPAT;
915 916
			exts->actions[0] = act;
			exts->nr_actions = 1;
917
		} else if (exts->action && tb[exts->action]) {
918 919 920
			LIST_HEAD(actions);
			int err, i = 0;

921 922
			err = tcf_action_init(net, tp, tb[exts->action],
					      rate_tlv, NULL, ovr, TCA_ACT_BIND,
J
Jamal Hadi Salim 已提交
923
					      &actions);
924 925
			if (err)
				return err;
926 927 928
			list_for_each_entry(act, &actions, list)
				exts->actions[i++] = act;
			exts->nr_actions = i;
L
Linus Torvalds 已提交
929 930 931
		}
	}
#else
932 933
	if ((exts->action && tb[exts->action]) ||
	    (exts->police && tb[exts->police]))
L
Linus Torvalds 已提交
934 935 936 937 938
		return -EOPNOTSUPP;
#endif

	return 0;
}
939
EXPORT_SYMBOL(tcf_exts_validate);
L
Linus Torvalds 已提交
940

941
void tcf_exts_change(struct tcf_exts *dst, struct tcf_exts *src)
L
Linus Torvalds 已提交
942 943
{
#ifdef CONFIG_NET_CLS_ACT
944 945
	struct tcf_exts old = *dst;

946
	*dst = *src;
947
	tcf_exts_destroy(&old);
L
Linus Torvalds 已提交
948 949
#endif
}
950
EXPORT_SYMBOL(tcf_exts_change);
L
Linus Torvalds 已提交
951

952 953 954 955 956 957 958 959 960
#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
961

962
int tcf_exts_dump(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
963 964
{
#ifdef CONFIG_NET_CLS_ACT
965 966
	struct nlattr *nest;

967
	if (exts->action && tcf_exts_has_actions(exts)) {
L
Linus Torvalds 已提交
968 969 970 971 972
		/*
		 * 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
		 */
973
		if (exts->type != TCA_OLD_COMPAT) {
974 975
			LIST_HEAD(actions);

976
			nest = nla_nest_start(skb, exts->action);
977 978
			if (nest == NULL)
				goto nla_put_failure;
979 980 981

			tcf_exts_to_list(exts, &actions);
			if (tcf_action_dump(skb, &actions, 0, 0) < 0)
982
				goto nla_put_failure;
983
			nla_nest_end(skb, nest);
984
		} else if (exts->police) {
985
			struct tc_action *act = tcf_exts_first_act(exts);
986
			nest = nla_nest_start(skb, exts->police);
987
			if (nest == NULL || !act)
988
				goto nla_put_failure;
989
			if (tcf_action_dump_old(skb, act, 0, 0) < 0)
990
				goto nla_put_failure;
991
			nla_nest_end(skb, nest);
L
Linus Torvalds 已提交
992 993 994
		}
	}
	return 0;
995 996 997

nla_put_failure:
	nla_nest_cancel(skb, nest);
L
Linus Torvalds 已提交
998
	return -1;
999 1000 1001
#else
	return 0;
#endif
L
Linus Torvalds 已提交
1002
}
1003
EXPORT_SYMBOL(tcf_exts_dump);
L
Linus Torvalds 已提交
1004

1005

1006
int tcf_exts_dump_stats(struct sk_buff *skb, struct tcf_exts *exts)
L
Linus Torvalds 已提交
1007 1008
{
#ifdef CONFIG_NET_CLS_ACT
1009
	struct tc_action *a = tcf_exts_first_act(exts);
1010
	if (a != NULL && tcf_action_copy_stats(skb, a, 1) < 0)
1011
		return -1;
L
Linus Torvalds 已提交
1012 1013 1014
#endif
	return 0;
}
1015
EXPORT_SYMBOL(tcf_exts_dump_stats);
L
Linus Torvalds 已提交
1016

1017 1018 1019 1020 1021 1022 1023
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);

1024
	if (!tcf_exts_has_actions(exts))
1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040
		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 已提交
1041 1042
static int __init tc_filter_init(void)
{
1043 1044 1045 1046
	tc_filter_wq = alloc_ordered_workqueue("tc_filter_workqueue", 0);
	if (!tc_filter_wq)
		return -ENOMEM;

1047 1048
	rtnl_register(PF_UNSPEC, RTM_NEWTFILTER, tc_ctl_tfilter, NULL, 0);
	rtnl_register(PF_UNSPEC, RTM_DELTFILTER, tc_ctl_tfilter, NULL, 0);
1049
	rtnl_register(PF_UNSPEC, RTM_GETTFILTER, tc_ctl_tfilter,
1050
		      tc_dump_tfilter, 0);
L
Linus Torvalds 已提交
1051 1052 1053 1054 1055

	return 0;
}

subsys_initcall(tc_filter_init);