em_meta.c 22.9 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11
/*
 * net/sched/em_meta.c	Metadata ematch
 *
 *		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:	Thomas Graf <tgraf@suug.ch>
 *
 * ==========================================================================
12
 *
L
Linus Torvalds 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
 * 	The metadata ematch compares two meta objects where each object
 * 	represents either a meta value stored in the kernel or a static
 * 	value provided by userspace. The objects are not provided by
 * 	userspace itself but rather a definition providing the information
 * 	to build them. Every object is of a certain type which must be
 * 	equal to the object it is being compared to.
 *
 * 	The definition of a objects conists of the type (meta type), a
 * 	identifier (meta id) and additional type specific information.
 * 	The meta id is either TCF_META_TYPE_VALUE for values provided by
 * 	userspace or a index to the meta operations table consisting of
 * 	function pointers to type specific meta data collectors returning
 * 	the value of the requested meta value.
 *
 * 	         lvalue                                   rvalue
 * 	      +-----------+                           +-----------+
 * 	      | type: INT |                           | type: INT |
30
 * 	 def  | id: DEV   |                           | id: VALUE |
L
Linus Torvalds 已提交
31 32 33
 * 	      | data:     |                           | data: 3   |
 * 	      +-----------+                           +-----------+
 * 	            |                                       |
34
 * 	            ---> meta_ops[INT][DEV](...)            |
35
 *	                      |                             |
L
Linus Torvalds 已提交
36 37 38 39
 * 	            -----------                             |
 * 	            V                                       V
 * 	      +-----------+                           +-----------+
 * 	      | type: INT |                           | type: INT |
40
 * 	 obj  | id: DEV |                             | id: VALUE |
L
Linus Torvalds 已提交
41 42 43 44 45 46 47 48 49
 * 	      | data: 2   |<--data got filled out     | data: 3   |
 * 	      +-----------+                           +-----------+
 * 	            |                                         |
 * 	            --------------> 2  equals 3 <--------------
 *
 * 	This is a simplified schema, the complexity varies depending
 * 	on the meta type. Obviously, the length of the data must also
 * 	be provided for non-numeric types.
 *
L
Lucas De Marchi 已提交
50
 * 	Additionally, type dependent modifiers such as shift operators
L
Linus Torvalds 已提交
51 52 53 54 55 56 57
 * 	or mask may be applied to extend the functionaliy. As of now,
 * 	the variable length type supports shifting the byte string to
 * 	the right, eating up any number of octets and thus supporting
 * 	wildcard interface name comparisons such as "ppp%" matching
 * 	ppp0..9.
 *
 * 	NOTE: Certain meta values depend on other subsystems and are
58
 * 	      only available if that subsystem is enabled in the kernel.
L
Linus Torvalds 已提交
59 60
 */

61
#include <linux/slab.h>
L
Linus Torvalds 已提交
62 63 64 65 66 67 68
#include <linux/module.h>
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/string.h>
#include <linux/skbuff.h>
#include <linux/random.h>
S
Stephen Hemminger 已提交
69
#include <linux/if_vlan.h>
L
Linus Torvalds 已提交
70 71 72 73
#include <linux/tc_ematch/tc_em_meta.h>
#include <net/dst.h>
#include <net/route.h>
#include <net/pkt_cls.h>
74
#include <net/sock.h>
L
Linus Torvalds 已提交
75

E
Eric Dumazet 已提交
76
struct meta_obj {
L
Linus Torvalds 已提交
77 78 79 80
	unsigned long		value;
	unsigned int		len;
};

E
Eric Dumazet 已提交
81
struct meta_value {
L
Linus Torvalds 已提交
82 83 84 85 86
	struct tcf_meta_val	hdr;
	unsigned long		val;
	unsigned int		len;
};

E
Eric Dumazet 已提交
87
struct meta_match {
L
Linus Torvalds 已提交
88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 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 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170
	struct meta_value	lvalue;
	struct meta_value	rvalue;
};

static inline int meta_id(struct meta_value *v)
{
	return TCF_META_ID(v->hdr.kind);
}

static inline int meta_type(struct meta_value *v)
{
	return TCF_META_TYPE(v->hdr.kind);
}

#define META_COLLECTOR(FUNC) static void meta_##FUNC(struct sk_buff *skb, \
	struct tcf_pkt_info *info, struct meta_value *v, \
	struct meta_obj *dst, int *err)

/**************************************************************************
 * System status & misc
 **************************************************************************/

META_COLLECTOR(int_random)
{
	get_random_bytes(&dst->value, sizeof(dst->value));
}

static inline unsigned long fixed_loadavg(int load)
{
	int rnd_load = load + (FIXED_1/200);
	int rnd_frac = ((rnd_load & (FIXED_1-1)) * 100) >> FSHIFT;

	return ((rnd_load >> FSHIFT) * 100) + rnd_frac;
}

META_COLLECTOR(int_loadavg_0)
{
	dst->value = fixed_loadavg(avenrun[0]);
}

META_COLLECTOR(int_loadavg_1)
{
	dst->value = fixed_loadavg(avenrun[1]);
}

META_COLLECTOR(int_loadavg_2)
{
	dst->value = fixed_loadavg(avenrun[2]);
}

/**************************************************************************
 * Device names & indices
 **************************************************************************/

static inline int int_dev(struct net_device *dev, struct meta_obj *dst)
{
	if (unlikely(dev == NULL))
		return -1;

	dst->value = dev->ifindex;
	return 0;
}

static inline int var_dev(struct net_device *dev, struct meta_obj *dst)
{
	if (unlikely(dev == NULL))
		return -1;

	dst->value = (unsigned long) dev->name;
	dst->len = strlen(dev->name);
	return 0;
}

META_COLLECTOR(int_dev)
{
	*err = int_dev(skb->dev, dst);
}

META_COLLECTOR(var_dev)
{
	*err = var_dev(skb->dev, dst);
}

S
Stephen Hemminger 已提交
171 172 173 174 175 176
/**************************************************************************
 * vlan tag
 **************************************************************************/

META_COLLECTOR(int_vlan_tag)
{
177 178
	unsigned short tag;

179
	tag = skb_vlan_tag_get(skb);
180
	if (!tag && __vlan_get_tag(skb, &tag))
S
Stephen Hemminger 已提交
181 182 183 184 185 186 187
		*err = -1;
	else
		dst->value = tag;
}



L
Linus Torvalds 已提交
188 189 190 191 192 193 194 195 196 197 198 199
/**************************************************************************
 * skb attributes
 **************************************************************************/

META_COLLECTOR(int_priority)
{
	dst->value = skb->priority;
}

META_COLLECTOR(int_protocol)
{
	/* Let userspace take care of the byte ordering */
200
	dst->value = tc_skb_protocol(skb);
L
Linus Torvalds 已提交
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222
}

META_COLLECTOR(int_pkttype)
{
	dst->value = skb->pkt_type;
}

META_COLLECTOR(int_pktlen)
{
	dst->value = skb->len;
}

META_COLLECTOR(int_datalen)
{
	dst->value = skb->data_len;
}

META_COLLECTOR(int_maclen)
{
	dst->value = skb->mac_len;
}

223 224
META_COLLECTOR(int_rxhash)
{
225
	dst->value = skb_get_hash(skb);
226 227
}

L
Linus Torvalds 已提交
228 229 230 231
/**************************************************************************
 * Netfilter
 **************************************************************************/

T
Thomas Graf 已提交
232
META_COLLECTOR(int_mark)
L
Linus Torvalds 已提交
233
{
T
Thomas Graf 已提交
234
	dst->value = skb->mark;
235
}
L
Linus Torvalds 已提交
236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251

/**************************************************************************
 * Traffic Control
 **************************************************************************/

META_COLLECTOR(int_tcindex)
{
	dst->value = skb->tc_index;
}

/**************************************************************************
 * Routing
 **************************************************************************/

META_COLLECTOR(int_rtclassid)
{
E
Eric Dumazet 已提交
252
	if (unlikely(skb_dst(skb) == NULL))
L
Linus Torvalds 已提交
253 254
		*err = -1;
	else
255
#ifdef CONFIG_IP_ROUTE_CLASSID
E
Eric Dumazet 已提交
256
		dst->value = skb_dst(skb)->tclassid;
257 258
#else
		dst->value = 0;
L
Linus Torvalds 已提交
259
#endif
260
}
L
Linus Torvalds 已提交
261 262 263

META_COLLECTOR(int_rtiif)
{
E
Eric Dumazet 已提交
264
	if (unlikely(skb_rtable(skb) == NULL))
L
Linus Torvalds 已提交
265 266
		*err = -1;
	else
267
		dst->value = inet_iif(skb);
L
Linus Torvalds 已提交
268 269
}

270 271 272 273
/**************************************************************************
 * Socket Attributes
 **************************************************************************/

274 275
#define skip_nonlocal(skb) \
	(unlikely(skb->sk == NULL))
276 277 278

META_COLLECTOR(int_sk_family)
{
279 280 281 282
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
283 284 285 286 287
	dst->value = skb->sk->sk_family;
}

META_COLLECTOR(int_sk_state)
{
288 289 290 291
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
292 293 294 295 296
	dst->value = skb->sk->sk_state;
}

META_COLLECTOR(int_sk_reuse)
{
297 298 299 300
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
301 302 303 304 305
	dst->value = skb->sk->sk_reuse;
}

META_COLLECTOR(int_sk_bound_if)
{
306 307 308 309
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
310 311 312 313 314 315
	/* No error if bound_dev_if is 0, legal userspace check */
	dst->value = skb->sk->sk_bound_dev_if;
}

META_COLLECTOR(var_sk_bound_if)
{
316 317 318 319
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
320

E
Eric Dumazet 已提交
321
	if (skb->sk->sk_bound_dev_if == 0) {
322 323
		dst->value = (unsigned long) "any";
		dst->len = 3;
E
Eric Dumazet 已提交
324
	} else {
325
		struct net_device *dev;
326

E
Eric Dumazet 已提交
327
		rcu_read_lock();
328 329
		dev = dev_get_by_index_rcu(sock_net(skb->sk),
					   skb->sk->sk_bound_dev_if);
330
		*err = var_dev(dev, dst);
E
Eric Dumazet 已提交
331 332
		rcu_read_unlock();
	}
333 334 335 336
}

META_COLLECTOR(int_sk_refcnt)
{
337 338 339 340
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
341 342 343 344 345
	dst->value = atomic_read(&skb->sk->sk_refcnt);
}

META_COLLECTOR(int_sk_rcvbuf)
{
346 347 348
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
349 350 351
		*err = -1;
		return;
	}
352
	dst->value = sk->sk_rcvbuf;
353 354 355 356
}

META_COLLECTOR(int_sk_shutdown)
{
357 358 359
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
360 361 362
		*err = -1;
		return;
	}
363
	dst->value = sk->sk_shutdown;
364 365 366 367
}

META_COLLECTOR(int_sk_proto)
{
368 369 370
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
371 372 373
		*err = -1;
		return;
	}
374
	dst->value = sk->sk_protocol;
375 376 377 378
}

META_COLLECTOR(int_sk_type)
{
379 380 381
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
382 383 384
		*err = -1;
		return;
	}
385
	dst->value = sk->sk_type;
386 387 388 389
}

META_COLLECTOR(int_sk_rmem_alloc)
{
390 391 392
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
393 394 395
		*err = -1;
		return;
	}
396
	dst->value = sk_rmem_alloc_get(sk);
397 398 399 400
}

META_COLLECTOR(int_sk_wmem_alloc)
{
401 402 403
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
404 405 406
		*err = -1;
		return;
	}
407
	dst->value = sk_wmem_alloc_get(sk);
408 409 410 411
}

META_COLLECTOR(int_sk_omem_alloc)
{
412 413 414
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
415 416 417
		*err = -1;
		return;
	}
418
	dst->value = atomic_read(&sk->sk_omem_alloc);
419 420 421 422
}

META_COLLECTOR(int_sk_rcv_qlen)
{
423 424 425
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
426 427 428
		*err = -1;
		return;
	}
429
	dst->value = sk->sk_receive_queue.qlen;
430 431 432 433
}

META_COLLECTOR(int_sk_snd_qlen)
{
434 435 436
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
437 438 439
		*err = -1;
		return;
	}
440
	dst->value = sk->sk_write_queue.qlen;
441 442 443 444
}

META_COLLECTOR(int_sk_wmem_queued)
{
445 446 447
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
448 449 450
		*err = -1;
		return;
	}
451
	dst->value = sk->sk_wmem_queued;
452 453 454 455
}

META_COLLECTOR(int_sk_fwd_alloc)
{
456 457 458
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
459 460 461
		*err = -1;
		return;
	}
462
	dst->value = sk->sk_forward_alloc;
463 464 465 466
}

META_COLLECTOR(int_sk_sndbuf)
{
467 468 469
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
470 471 472
		*err = -1;
		return;
	}
473
	dst->value = sk->sk_sndbuf;
474 475 476 477
}

META_COLLECTOR(int_sk_alloc)
{
478 479 480
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
481 482 483
		*err = -1;
		return;
	}
484
	dst->value = (__force int) sk->sk_allocation;
485 486
}

487
META_COLLECTOR(int_sk_hash)
488
{
489 490 491 492
	if (skip_nonlocal(skb)) {
		*err = -1;
		return;
	}
493
	dst->value = skb->sk->sk_hash;
494 495 496 497
}

META_COLLECTOR(int_sk_lingertime)
{
498 499 500
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
501 502 503
		*err = -1;
		return;
	}
504
	dst->value = sk->sk_lingertime / HZ;
505 506 507 508
}

META_COLLECTOR(int_sk_err_qlen)
{
509 510 511
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
512 513 514
		*err = -1;
		return;
	}
515
	dst->value = sk->sk_error_queue.qlen;
516 517 518 519
}

META_COLLECTOR(int_sk_ack_bl)
{
520 521 522
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
523 524 525
		*err = -1;
		return;
	}
526
	dst->value = sk->sk_ack_backlog;
527 528 529 530
}

META_COLLECTOR(int_sk_max_ack_bl)
{
531 532 533
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
534 535 536
		*err = -1;
		return;
	}
537
	dst->value = sk->sk_max_ack_backlog;
538 539 540 541
}

META_COLLECTOR(int_sk_prio)
{
542 543 544
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
545 546 547
		*err = -1;
		return;
	}
548
	dst->value = sk->sk_priority;
549 550 551 552
}

META_COLLECTOR(int_sk_rcvlowat)
{
553 554 555
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
556 557 558
		*err = -1;
		return;
	}
559
	dst->value = sk->sk_rcvlowat;
560 561 562 563
}

META_COLLECTOR(int_sk_rcvtimeo)
{
564 565 566
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
567 568 569
		*err = -1;
		return;
	}
570
	dst->value = sk->sk_rcvtimeo / HZ;
571 572 573 574
}

META_COLLECTOR(int_sk_sndtimeo)
{
575 576 577
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
578 579 580
		*err = -1;
		return;
	}
581
	dst->value = sk->sk_sndtimeo / HZ;
582 583 584 585
}

META_COLLECTOR(int_sk_sendmsg_off)
{
586 587 588
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
589 590 591
		*err = -1;
		return;
	}
592
	dst->value = sk->sk_frag.offset;
593 594 595 596
}

META_COLLECTOR(int_sk_write_pend)
{
597 598 599
	const struct sock *sk = skb_to_full_sk(skb);

	if (!sk) {
600 601 602
		*err = -1;
		return;
	}
603
	dst->value = sk->sk_write_pending;
604 605
}

L
Linus Torvalds 已提交
606 607 608 609
/**************************************************************************
 * Meta value collectors assignment table
 **************************************************************************/

E
Eric Dumazet 已提交
610
struct meta_ops {
L
Linus Torvalds 已提交
611 612 613 614
	void		(*get)(struct sk_buff *, struct tcf_pkt_info *,
			       struct meta_value *, struct meta_obj *, int *);
};

615 616 617
#define META_ID(name) TCF_META_ID_##name
#define META_FUNC(name) { .get = meta_##name }

L
Linus Torvalds 已提交
618 619
/* Meta value operations table listing all meta value collectors and
 * assigns them to a type and meta id. */
E
Eric Dumazet 已提交
620
static struct meta_ops __meta_ops[TCF_META_TYPE_MAX + 1][TCF_META_ID_MAX + 1] = {
L
Linus Torvalds 已提交
621
	[TCF_META_TYPE_VAR] = {
622 623
		[META_ID(DEV)]			= META_FUNC(var_dev),
		[META_ID(SK_BOUND_IF)] 		= META_FUNC(var_sk_bound_if),
L
Linus Torvalds 已提交
624 625
	},
	[TCF_META_TYPE_INT] = {
626 627 628 629 630 631 632 633 634 635 636
		[META_ID(RANDOM)]		= META_FUNC(int_random),
		[META_ID(LOADAVG_0)]		= META_FUNC(int_loadavg_0),
		[META_ID(LOADAVG_1)]		= META_FUNC(int_loadavg_1),
		[META_ID(LOADAVG_2)]		= META_FUNC(int_loadavg_2),
		[META_ID(DEV)]			= META_FUNC(int_dev),
		[META_ID(PRIORITY)]		= META_FUNC(int_priority),
		[META_ID(PROTOCOL)]		= META_FUNC(int_protocol),
		[META_ID(PKTTYPE)]		= META_FUNC(int_pkttype),
		[META_ID(PKTLEN)]		= META_FUNC(int_pktlen),
		[META_ID(DATALEN)]		= META_FUNC(int_datalen),
		[META_ID(MACLEN)]		= META_FUNC(int_maclen),
T
Thomas Graf 已提交
637
		[META_ID(NFMARK)]		= META_FUNC(int_mark),
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659
		[META_ID(TCINDEX)]		= META_FUNC(int_tcindex),
		[META_ID(RTCLASSID)]		= META_FUNC(int_rtclassid),
		[META_ID(RTIIF)]		= META_FUNC(int_rtiif),
		[META_ID(SK_FAMILY)]		= META_FUNC(int_sk_family),
		[META_ID(SK_STATE)]		= META_FUNC(int_sk_state),
		[META_ID(SK_REUSE)]		= META_FUNC(int_sk_reuse),
		[META_ID(SK_BOUND_IF)]		= META_FUNC(int_sk_bound_if),
		[META_ID(SK_REFCNT)]		= META_FUNC(int_sk_refcnt),
		[META_ID(SK_RCVBUF)]		= META_FUNC(int_sk_rcvbuf),
		[META_ID(SK_SNDBUF)]		= META_FUNC(int_sk_sndbuf),
		[META_ID(SK_SHUTDOWN)]		= META_FUNC(int_sk_shutdown),
		[META_ID(SK_PROTO)]		= META_FUNC(int_sk_proto),
		[META_ID(SK_TYPE)]		= META_FUNC(int_sk_type),
		[META_ID(SK_RMEM_ALLOC)]	= META_FUNC(int_sk_rmem_alloc),
		[META_ID(SK_WMEM_ALLOC)]	= META_FUNC(int_sk_wmem_alloc),
		[META_ID(SK_OMEM_ALLOC)]	= META_FUNC(int_sk_omem_alloc),
		[META_ID(SK_WMEM_QUEUED)]	= META_FUNC(int_sk_wmem_queued),
		[META_ID(SK_RCV_QLEN)]		= META_FUNC(int_sk_rcv_qlen),
		[META_ID(SK_SND_QLEN)]		= META_FUNC(int_sk_snd_qlen),
		[META_ID(SK_ERR_QLEN)]		= META_FUNC(int_sk_err_qlen),
		[META_ID(SK_FORWARD_ALLOCS)]	= META_FUNC(int_sk_fwd_alloc),
		[META_ID(SK_ALLOCS)]		= META_FUNC(int_sk_alloc),
660
		[META_ID(SK_HASH)]		= META_FUNC(int_sk_hash),
661 662 663 664 665 666 667 668 669
		[META_ID(SK_LINGERTIME)]	= META_FUNC(int_sk_lingertime),
		[META_ID(SK_ACK_BACKLOG)]	= META_FUNC(int_sk_ack_bl),
		[META_ID(SK_MAX_ACK_BACKLOG)]	= META_FUNC(int_sk_max_ack_bl),
		[META_ID(SK_PRIO)]		= META_FUNC(int_sk_prio),
		[META_ID(SK_RCVLOWAT)]		= META_FUNC(int_sk_rcvlowat),
		[META_ID(SK_RCVTIMEO)]		= META_FUNC(int_sk_rcvtimeo),
		[META_ID(SK_SNDTIMEO)]		= META_FUNC(int_sk_sndtimeo),
		[META_ID(SK_SENDMSG_OFF)]	= META_FUNC(int_sk_sendmsg_off),
		[META_ID(SK_WRITE_PENDING)]	= META_FUNC(int_sk_write_pend),
S
Stephen Hemminger 已提交
670
		[META_ID(VLAN_TAG)]		= META_FUNC(int_vlan_tag),
671
		[META_ID(RXHASH)]		= META_FUNC(int_rxhash),
L
Linus Torvalds 已提交
672 673 674
	}
};

E
Eric Dumazet 已提交
675
static inline struct meta_ops *meta_ops(struct meta_value *val)
L
Linus Torvalds 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693
{
	return &__meta_ops[meta_type(val)][meta_id(val)];
}

/**************************************************************************
 * Type specific operations for TCF_META_TYPE_VAR
 **************************************************************************/

static int meta_var_compare(struct meta_obj *a, struct meta_obj *b)
{
	int r = a->len - b->len;

	if (r == 0)
		r = memcmp((void *) a->value, (void *) b->value, a->len);

	return r;
}

694
static int meta_var_change(struct meta_value *dst, struct nlattr *nla)
L
Linus Torvalds 已提交
695
{
696
	int len = nla_len(nla);
L
Linus Torvalds 已提交
697

698
	dst->val = (unsigned long)kmemdup(nla_data(nla), len, GFP_KERNEL);
L
Linus Torvalds 已提交
699 700 701 702 703 704 705 706
	if (dst->val == 0UL)
		return -ENOMEM;
	dst->len = len;
	return 0;
}

static void meta_var_destroy(struct meta_value *v)
{
J
Jesper Juhl 已提交
707
	kfree((void *) v->val);
L
Linus Torvalds 已提交
708 709 710 711 712 713 714 715 716 717 718 719 720
}

static void meta_var_apply_extras(struct meta_value *v,
				  struct meta_obj *dst)
{
	int shift = v->hdr.shift;

	if (shift && shift < dst->len)
		dst->len -= shift;
}

static int meta_var_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
{
721 722 723
	if (v->val && v->len &&
	    nla_put(skb, tlv, v->len, (void *) v->val))
		goto nla_put_failure;
L
Linus Torvalds 已提交
724 725
	return 0;

726
nla_put_failure:
L
Linus Torvalds 已提交
727 728 729 730 731 732 733 734 735 736 737 738
	return -1;
}

/**************************************************************************
 * Type specific operations for TCF_META_TYPE_INT
 **************************************************************************/

static int meta_int_compare(struct meta_obj *a, struct meta_obj *b)
{
	/* Let gcc optimize it, the unlikely is not really based on
	 * some numbers but jump free code for mismatches seems
	 * more logical. */
739
	if (unlikely(a->value == b->value))
L
Linus Torvalds 已提交
740
		return 0;
741
	else if (a->value < b->value)
L
Linus Torvalds 已提交
742 743 744 745 746
		return -1;
	else
		return 1;
}

747
static int meta_int_change(struct meta_value *dst, struct nlattr *nla)
L
Linus Torvalds 已提交
748
{
749 750
	if (nla_len(nla) >= sizeof(unsigned long)) {
		dst->val = *(unsigned long *) nla_data(nla);
L
Linus Torvalds 已提交
751
		dst->len = sizeof(unsigned long);
752
	} else if (nla_len(nla) == sizeof(u32)) {
753
		dst->val = nla_get_u32(nla);
L
Linus Torvalds 已提交
754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772
		dst->len = sizeof(u32);
	} else
		return -EINVAL;

	return 0;
}

static void meta_int_apply_extras(struct meta_value *v,
				  struct meta_obj *dst)
{
	if (v->hdr.shift)
		dst->value >>= v->hdr.shift;

	if (v->val)
		dst->value &= v->val;
}

static int meta_int_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
{
773 774 775 776 777 778 779
	if (v->len == sizeof(unsigned long)) {
		if (nla_put(skb, tlv, sizeof(unsigned long), &v->val))
			goto nla_put_failure;
	} else if (v->len == sizeof(u32)) {
		if (nla_put_u32(skb, tlv, v->val))
			goto nla_put_failure;
	}
L
Linus Torvalds 已提交
780 781 782

	return 0;

783
nla_put_failure:
L
Linus Torvalds 已提交
784 785 786 787 788 789 790
	return -1;
}

/**************************************************************************
 * Type specific operations table
 **************************************************************************/

E
Eric Dumazet 已提交
791
struct meta_type_ops {
L
Linus Torvalds 已提交
792 793
	void	(*destroy)(struct meta_value *);
	int	(*compare)(struct meta_obj *, struct meta_obj *);
794
	int	(*change)(struct meta_value *, struct nlattr *);
L
Linus Torvalds 已提交
795 796 797 798
	void	(*apply_extras)(struct meta_value *, struct meta_obj *);
	int	(*dump)(struct sk_buff *, struct meta_value *, int);
};

799
static const struct meta_type_ops __meta_type_ops[TCF_META_TYPE_MAX + 1] = {
L
Linus Torvalds 已提交
800 801 802 803 804 805 806 807 808 809 810 811 812 813 814
	[TCF_META_TYPE_VAR] = {
		.destroy = meta_var_destroy,
		.compare = meta_var_compare,
		.change = meta_var_change,
		.apply_extras = meta_var_apply_extras,
		.dump = meta_var_dump
	},
	[TCF_META_TYPE_INT] = {
		.compare = meta_int_compare,
		.change = meta_int_change,
		.apply_extras = meta_int_apply_extras,
		.dump = meta_int_dump
	}
};

815
static inline const struct meta_type_ops *meta_type_ops(struct meta_value *v)
L
Linus Torvalds 已提交
816 817 818 819 820 821 822 823
{
	return &__meta_type_ops[meta_type(v)];
}

/**************************************************************************
 * Core
 **************************************************************************/

824 825
static int meta_get(struct sk_buff *skb, struct tcf_pkt_info *info,
		    struct meta_value *v, struct meta_obj *dst)
L
Linus Torvalds 已提交
826 827 828 829 830 831 832 833 834 835 836 837 838 839
{
	int err = 0;

	if (meta_id(v) == TCF_META_ID_VALUE) {
		dst->value = v->val;
		dst->len = v->len;
		return 0;
	}

	meta_ops(v)->get(skb, info, v, dst, &err);
	if (err < 0)
		return err;

	if (meta_type_ops(v)->apply_extras)
E
Eric Dumazet 已提交
840
		meta_type_ops(v)->apply_extras(v, dst);
L
Linus Torvalds 已提交
841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858

	return 0;
}

static int em_meta_match(struct sk_buff *skb, struct tcf_ematch *m,
			 struct tcf_pkt_info *info)
{
	int r;
	struct meta_match *meta = (struct meta_match *) m->data;
	struct meta_obj l_value, r_value;

	if (meta_get(skb, info, &meta->lvalue, &l_value) < 0 ||
	    meta_get(skb, info, &meta->rvalue, &r_value) < 0)
		return 0;

	r = meta_type_ops(&meta->lvalue)->compare(&l_value, &r_value);

	switch (meta->lvalue.hdr.op) {
E
Eric Dumazet 已提交
859 860 861 862 863 864
	case TCF_EM_OPND_EQ:
		return !r;
	case TCF_EM_OPND_LT:
		return r < 0;
	case TCF_EM_OPND_GT:
		return r > 0;
L
Linus Torvalds 已提交
865 866 867 868 869
	}

	return 0;
}

870
static void meta_delete(struct meta_match *meta)
L
Linus Torvalds 已提交
871
{
S
Stephen Hemminger 已提交
872
	if (meta) {
873
		const struct meta_type_ops *ops = meta_type_ops(&meta->lvalue);
L
Linus Torvalds 已提交
874

S
Stephen Hemminger 已提交
875 876 877 878
		if (ops && ops->destroy) {
			ops->destroy(&meta->lvalue);
			ops->destroy(&meta->rvalue);
		}
L
Linus Torvalds 已提交
879 880 881 882 883
	}

	kfree(meta);
}

884
static inline int meta_change_data(struct meta_value *dst, struct nlattr *nla)
L
Linus Torvalds 已提交
885
{
886 887
	if (nla) {
		if (nla_len(nla) == 0)
L
Linus Torvalds 已提交
888 889
			return -EINVAL;

890
		return meta_type_ops(dst)->change(dst, nla);
L
Linus Torvalds 已提交
891 892 893 894 895 896 897
	}

	return 0;
}

static inline int meta_is_supported(struct meta_value *val)
{
E
Eric Dumazet 已提交
898
	return !meta_id(val) || meta_ops(val)->get;
L
Linus Torvalds 已提交
899 900
}

901 902 903 904
static const struct nla_policy meta_policy[TCA_EM_META_MAX + 1] = {
	[TCA_EM_META_HDR]	= { .len = sizeof(struct tcf_meta_hdr) },
};

905
static int em_meta_change(struct net *net, void *data, int len,
L
Linus Torvalds 已提交
906 907
			  struct tcf_ematch *m)
{
908
	int err;
909
	struct nlattr *tb[TCA_EM_META_MAX + 1];
L
Linus Torvalds 已提交
910 911
	struct tcf_meta_hdr *hdr;
	struct meta_match *meta = NULL;
912

913
	err = nla_parse(tb, TCA_EM_META_MAX, data, len, meta_policy);
914
	if (err < 0)
L
Linus Torvalds 已提交
915 916
		goto errout;

917
	err = -EINVAL;
918
	if (tb[TCA_EM_META_HDR] == NULL)
L
Linus Torvalds 已提交
919
		goto errout;
920
	hdr = nla_data(tb[TCA_EM_META_HDR]);
L
Linus Torvalds 已提交
921 922 923 924 925 926 927

	if (TCF_META_TYPE(hdr->left.kind) != TCF_META_TYPE(hdr->right.kind) ||
	    TCF_META_TYPE(hdr->left.kind) > TCF_META_TYPE_MAX ||
	    TCF_META_ID(hdr->left.kind) > TCF_META_ID_MAX ||
	    TCF_META_ID(hdr->right.kind) > TCF_META_ID_MAX)
		goto errout;

928
	meta = kzalloc(sizeof(*meta), GFP_KERNEL);
929 930
	if (meta == NULL) {
		err = -ENOMEM;
L
Linus Torvalds 已提交
931
		goto errout;
932
	}
L
Linus Torvalds 已提交
933 934 935 936 937 938 939 940 941 942

	memcpy(&meta->lvalue.hdr, &hdr->left, sizeof(hdr->left));
	memcpy(&meta->rvalue.hdr, &hdr->right, sizeof(hdr->right));

	if (!meta_is_supported(&meta->lvalue) ||
	    !meta_is_supported(&meta->rvalue)) {
		err = -EOPNOTSUPP;
		goto errout;
	}

943 944
	if (meta_change_data(&meta->lvalue, tb[TCA_EM_META_LVALUE]) < 0 ||
	    meta_change_data(&meta->rvalue, tb[TCA_EM_META_RVALUE]) < 0)
L
Linus Torvalds 已提交
945 946 947 948 949 950 951 952 953 954 955 956
		goto errout;

	m->datalen = sizeof(*meta);
	m->data = (unsigned long) meta;

	err = 0;
errout:
	if (err && meta)
		meta_delete(meta);
	return err;
}

957
static void em_meta_destroy(struct tcf_ematch *m)
L
Linus Torvalds 已提交
958 959 960 961 962 963 964 965 966
{
	if (m)
		meta_delete((struct meta_match *) m->data);
}

static int em_meta_dump(struct sk_buff *skb, struct tcf_ematch *em)
{
	struct meta_match *meta = (struct meta_match *) em->data;
	struct tcf_meta_hdr hdr;
967
	const struct meta_type_ops *ops;
L
Linus Torvalds 已提交
968 969 970 971 972

	memset(&hdr, 0, sizeof(hdr));
	memcpy(&hdr.left, &meta->lvalue.hdr, sizeof(hdr.left));
	memcpy(&hdr.right, &meta->rvalue.hdr, sizeof(hdr.right));

973 974
	if (nla_put(skb, TCA_EM_META_HDR, sizeof(hdr), &hdr))
		goto nla_put_failure;
L
Linus Torvalds 已提交
975 976 977 978

	ops = meta_type_ops(&meta->lvalue);
	if (ops->dump(skb, &meta->lvalue, TCA_EM_META_LVALUE) < 0 ||
	    ops->dump(skb, &meta->rvalue, TCA_EM_META_RVALUE) < 0)
979
		goto nla_put_failure;
L
Linus Torvalds 已提交
980 981 982

	return 0;

983
nla_put_failure:
L
Linus Torvalds 已提交
984
	return -1;
985
}
L
Linus Torvalds 已提交
986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001

static struct tcf_ematch_ops em_meta_ops = {
	.kind	  = TCF_EM_META,
	.change	  = em_meta_change,
	.match	  = em_meta_match,
	.destroy  = em_meta_destroy,
	.dump	  = em_meta_dump,
	.owner	  = THIS_MODULE,
	.link	  = LIST_HEAD_INIT(em_meta_ops.link)
};

static int __init init_em_meta(void)
{
	return tcf_em_register(&em_meta_ops);
}

1002
static void __exit exit_em_meta(void)
L
Linus Torvalds 已提交
1003 1004 1005 1006 1007 1008 1009 1010
{
	tcf_em_unregister(&em_meta_ops);
}

MODULE_LICENSE("GPL");

module_init(init_em_meta);
module_exit(exit_em_meta);
1011 1012

MODULE_ALIAS_TCF_EMATCH(TCF_EM_META);