xref: /openbmc/linux/net/sched/em_meta.c (revision cfe2f14c)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * net/sched/em_meta.c	Metadata ematch
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  *		This program is free software; you can redistribute it and/or
51da177e4SLinus Torvalds  *		modify it under the terms of the GNU General Public License
61da177e4SLinus Torvalds  *		as published by the Free Software Foundation; either version
71da177e4SLinus Torvalds  *		2 of the License, or (at your option) any later version.
81da177e4SLinus Torvalds  *
91da177e4SLinus Torvalds  * Authors:	Thomas Graf <tgraf@suug.ch>
101da177e4SLinus Torvalds  *
111da177e4SLinus Torvalds  * ==========================================================================
121da177e4SLinus Torvalds  *
131da177e4SLinus Torvalds  * 	The metadata ematch compares two meta objects where each object
141da177e4SLinus Torvalds  * 	represents either a meta value stored in the kernel or a static
151da177e4SLinus Torvalds  * 	value provided by userspace. The objects are not provided by
161da177e4SLinus Torvalds  * 	userspace itself but rather a definition providing the information
171da177e4SLinus Torvalds  * 	to build them. Every object is of a certain type which must be
181da177e4SLinus Torvalds  * 	equal to the object it is being compared to.
191da177e4SLinus Torvalds  *
201da177e4SLinus Torvalds  * 	The definition of a objects conists of the type (meta type), a
211da177e4SLinus Torvalds  * 	identifier (meta id) and additional type specific information.
221da177e4SLinus Torvalds  * 	The meta id is either TCF_META_TYPE_VALUE for values provided by
231da177e4SLinus Torvalds  * 	userspace or a index to the meta operations table consisting of
241da177e4SLinus Torvalds  * 	function pointers to type specific meta data collectors returning
251da177e4SLinus Torvalds  * 	the value of the requested meta value.
261da177e4SLinus Torvalds  *
271da177e4SLinus Torvalds  * 	         lvalue                                   rvalue
281da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
291da177e4SLinus Torvalds  * 	      | type: INT |                           | type: INT |
30261688d0SDavid S. Miller  * 	 def  | id: DEV   |                           | id: VALUE |
311da177e4SLinus Torvalds  * 	      | data:     |                           | data: 3   |
321da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
331da177e4SLinus Torvalds  * 	            |                                       |
34261688d0SDavid S. Miller  * 	            ---> meta_ops[INT][DEV](...)            |
351da177e4SLinus Torvalds  *	                      |                             |
361da177e4SLinus Torvalds  * 	            -----------                             |
371da177e4SLinus Torvalds  * 	            V                                       V
381da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
391da177e4SLinus Torvalds  * 	      | type: INT |                           | type: INT |
40261688d0SDavid S. Miller  * 	 obj  | id: DEV |                             | id: VALUE |
411da177e4SLinus Torvalds  * 	      | data: 2   |<--data got filled out     | data: 3   |
421da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
431da177e4SLinus Torvalds  * 	            |                                         |
441da177e4SLinus Torvalds  * 	            --------------> 2  equals 3 <--------------
451da177e4SLinus Torvalds  *
461da177e4SLinus Torvalds  * 	This is a simplified schema, the complexity varies depending
471da177e4SLinus Torvalds  * 	on the meta type. Obviously, the length of the data must also
481da177e4SLinus Torvalds  * 	be provided for non-numeric types.
491da177e4SLinus Torvalds  *
5025985edcSLucas De Marchi  * 	Additionally, type dependent modifiers such as shift operators
511da177e4SLinus Torvalds  * 	or mask may be applied to extend the functionaliy. As of now,
521da177e4SLinus Torvalds  * 	the variable length type supports shifting the byte string to
531da177e4SLinus Torvalds  * 	the right, eating up any number of octets and thus supporting
541da177e4SLinus Torvalds  * 	wildcard interface name comparisons such as "ppp%" matching
551da177e4SLinus Torvalds  * 	ppp0..9.
561da177e4SLinus Torvalds  *
571da177e4SLinus Torvalds  * 	NOTE: Certain meta values depend on other subsystems and are
583a4fa0a2SRobert P. J. Day  * 	      only available if that subsystem is enabled in the kernel.
591da177e4SLinus Torvalds  */
601da177e4SLinus Torvalds 
615a0e3ad6STejun Heo #include <linux/slab.h>
621da177e4SLinus Torvalds #include <linux/module.h>
631da177e4SLinus Torvalds #include <linux/types.h>
641da177e4SLinus Torvalds #include <linux/kernel.h>
651da177e4SLinus Torvalds #include <linux/sched.h>
661da177e4SLinus Torvalds #include <linux/string.h>
671da177e4SLinus Torvalds #include <linux/skbuff.h>
681da177e4SLinus Torvalds #include <linux/random.h>
693113e88cSStephen Hemminger #include <linux/if_vlan.h>
701da177e4SLinus Torvalds #include <linux/tc_ematch/tc_em_meta.h>
711da177e4SLinus Torvalds #include <net/dst.h>
721da177e4SLinus Torvalds #include <net/route.h>
731da177e4SLinus Torvalds #include <net/pkt_cls.h>
7448900629SThomas Graf #include <net/sock.h>
751da177e4SLinus Torvalds 
76cc7ec456SEric Dumazet struct meta_obj {
771da177e4SLinus Torvalds 	unsigned long		value;
781da177e4SLinus Torvalds 	unsigned int		len;
791da177e4SLinus Torvalds };
801da177e4SLinus Torvalds 
81cc7ec456SEric Dumazet struct meta_value {
821da177e4SLinus Torvalds 	struct tcf_meta_val	hdr;
831da177e4SLinus Torvalds 	unsigned long		val;
841da177e4SLinus Torvalds 	unsigned int		len;
851da177e4SLinus Torvalds };
861da177e4SLinus Torvalds 
87cc7ec456SEric Dumazet struct meta_match {
881da177e4SLinus Torvalds 	struct meta_value	lvalue;
891da177e4SLinus Torvalds 	struct meta_value	rvalue;
901da177e4SLinus Torvalds };
911da177e4SLinus Torvalds 
921da177e4SLinus Torvalds static inline int meta_id(struct meta_value *v)
931da177e4SLinus Torvalds {
941da177e4SLinus Torvalds 	return TCF_META_ID(v->hdr.kind);
951da177e4SLinus Torvalds }
961da177e4SLinus Torvalds 
971da177e4SLinus Torvalds static inline int meta_type(struct meta_value *v)
981da177e4SLinus Torvalds {
991da177e4SLinus Torvalds 	return TCF_META_TYPE(v->hdr.kind);
1001da177e4SLinus Torvalds }
1011da177e4SLinus Torvalds 
1021da177e4SLinus Torvalds #define META_COLLECTOR(FUNC) static void meta_##FUNC(struct sk_buff *skb, \
1031da177e4SLinus Torvalds 	struct tcf_pkt_info *info, struct meta_value *v, \
1041da177e4SLinus Torvalds 	struct meta_obj *dst, int *err)
1051da177e4SLinus Torvalds 
1061da177e4SLinus Torvalds /**************************************************************************
1071da177e4SLinus Torvalds  * System status & misc
1081da177e4SLinus Torvalds  **************************************************************************/
1091da177e4SLinus Torvalds 
1101da177e4SLinus Torvalds META_COLLECTOR(int_random)
1111da177e4SLinus Torvalds {
1121da177e4SLinus Torvalds 	get_random_bytes(&dst->value, sizeof(dst->value));
1131da177e4SLinus Torvalds }
1141da177e4SLinus Torvalds 
1151da177e4SLinus Torvalds static inline unsigned long fixed_loadavg(int load)
1161da177e4SLinus Torvalds {
1171da177e4SLinus Torvalds 	int rnd_load = load + (FIXED_1/200);
1181da177e4SLinus Torvalds 	int rnd_frac = ((rnd_load & (FIXED_1-1)) * 100) >> FSHIFT;
1191da177e4SLinus Torvalds 
1201da177e4SLinus Torvalds 	return ((rnd_load >> FSHIFT) * 100) + rnd_frac;
1211da177e4SLinus Torvalds }
1221da177e4SLinus Torvalds 
1231da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_0)
1241da177e4SLinus Torvalds {
1251da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[0]);
1261da177e4SLinus Torvalds }
1271da177e4SLinus Torvalds 
1281da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_1)
1291da177e4SLinus Torvalds {
1301da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[1]);
1311da177e4SLinus Torvalds }
1321da177e4SLinus Torvalds 
1331da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_2)
1341da177e4SLinus Torvalds {
1351da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[2]);
1361da177e4SLinus Torvalds }
1371da177e4SLinus Torvalds 
1381da177e4SLinus Torvalds /**************************************************************************
1391da177e4SLinus Torvalds  * Device names & indices
1401da177e4SLinus Torvalds  **************************************************************************/
1411da177e4SLinus Torvalds 
1421da177e4SLinus Torvalds static inline int int_dev(struct net_device *dev, struct meta_obj *dst)
1431da177e4SLinus Torvalds {
1441da177e4SLinus Torvalds 	if (unlikely(dev == NULL))
1451da177e4SLinus Torvalds 		return -1;
1461da177e4SLinus Torvalds 
1471da177e4SLinus Torvalds 	dst->value = dev->ifindex;
1481da177e4SLinus Torvalds 	return 0;
1491da177e4SLinus Torvalds }
1501da177e4SLinus Torvalds 
1511da177e4SLinus Torvalds static inline int var_dev(struct net_device *dev, struct meta_obj *dst)
1521da177e4SLinus Torvalds {
1531da177e4SLinus Torvalds 	if (unlikely(dev == NULL))
1541da177e4SLinus Torvalds 		return -1;
1551da177e4SLinus Torvalds 
1561da177e4SLinus Torvalds 	dst->value = (unsigned long) dev->name;
1571da177e4SLinus Torvalds 	dst->len = strlen(dev->name);
1581da177e4SLinus Torvalds 	return 0;
1591da177e4SLinus Torvalds }
1601da177e4SLinus Torvalds 
1611da177e4SLinus Torvalds META_COLLECTOR(int_dev)
1621da177e4SLinus Torvalds {
1631da177e4SLinus Torvalds 	*err = int_dev(skb->dev, dst);
1641da177e4SLinus Torvalds }
1651da177e4SLinus Torvalds 
1661da177e4SLinus Torvalds META_COLLECTOR(var_dev)
1671da177e4SLinus Torvalds {
1681da177e4SLinus Torvalds 	*err = var_dev(skb->dev, dst);
1691da177e4SLinus Torvalds }
1701da177e4SLinus Torvalds 
1711da177e4SLinus Torvalds /**************************************************************************
1723113e88cSStephen Hemminger  * vlan tag
1733113e88cSStephen Hemminger  **************************************************************************/
1743113e88cSStephen Hemminger 
1753113e88cSStephen Hemminger META_COLLECTOR(int_vlan_tag)
1763113e88cSStephen Hemminger {
1771a31f204SStephen Hemminger 	unsigned short tag;
1781a31f204SStephen Hemminger 
179df8a39deSJiri Pirko 	tag = skb_vlan_tag_get(skb);
1801a31f204SStephen Hemminger 	if (!tag && __vlan_get_tag(skb, &tag))
1813113e88cSStephen Hemminger 		*err = -1;
1823113e88cSStephen Hemminger 	else
1833113e88cSStephen Hemminger 		dst->value = tag;
1843113e88cSStephen Hemminger }
1853113e88cSStephen Hemminger 
1863113e88cSStephen Hemminger 
1873113e88cSStephen Hemminger 
1883113e88cSStephen Hemminger /**************************************************************************
1891da177e4SLinus Torvalds  * skb attributes
1901da177e4SLinus Torvalds  **************************************************************************/
1911da177e4SLinus Torvalds 
1921da177e4SLinus Torvalds META_COLLECTOR(int_priority)
1931da177e4SLinus Torvalds {
1941da177e4SLinus Torvalds 	dst->value = skb->priority;
1951da177e4SLinus Torvalds }
1961da177e4SLinus Torvalds 
1971da177e4SLinus Torvalds META_COLLECTOR(int_protocol)
1981da177e4SLinus Torvalds {
1991da177e4SLinus Torvalds 	/* Let userspace take care of the byte ordering */
200d8b9605dSJiri Pirko 	dst->value = tc_skb_protocol(skb);
2011da177e4SLinus Torvalds }
2021da177e4SLinus Torvalds 
2031da177e4SLinus Torvalds META_COLLECTOR(int_pkttype)
2041da177e4SLinus Torvalds {
2051da177e4SLinus Torvalds 	dst->value = skb->pkt_type;
2061da177e4SLinus Torvalds }
2071da177e4SLinus Torvalds 
2081da177e4SLinus Torvalds META_COLLECTOR(int_pktlen)
2091da177e4SLinus Torvalds {
2101da177e4SLinus Torvalds 	dst->value = skb->len;
2111da177e4SLinus Torvalds }
2121da177e4SLinus Torvalds 
2131da177e4SLinus Torvalds META_COLLECTOR(int_datalen)
2141da177e4SLinus Torvalds {
2151da177e4SLinus Torvalds 	dst->value = skb->data_len;
2161da177e4SLinus Torvalds }
2171da177e4SLinus Torvalds 
2181da177e4SLinus Torvalds META_COLLECTOR(int_maclen)
2191da177e4SLinus Torvalds {
2201da177e4SLinus Torvalds 	dst->value = skb->mac_len;
2211da177e4SLinus Torvalds }
2221da177e4SLinus Torvalds 
223c2e3143eSStephen Hemminger META_COLLECTOR(int_rxhash)
224c2e3143eSStephen Hemminger {
2253958afa1STom Herbert 	dst->value = skb_get_hash(skb);
226c2e3143eSStephen Hemminger }
227c2e3143eSStephen Hemminger 
2281da177e4SLinus Torvalds /**************************************************************************
2291da177e4SLinus Torvalds  * Netfilter
2301da177e4SLinus Torvalds  **************************************************************************/
2311da177e4SLinus Torvalds 
23282e91ffeSThomas Graf META_COLLECTOR(int_mark)
2331da177e4SLinus Torvalds {
23482e91ffeSThomas Graf 	dst->value = skb->mark;
2357686ee1aSPatrick McHardy }
2361da177e4SLinus Torvalds 
2371da177e4SLinus Torvalds /**************************************************************************
2381da177e4SLinus Torvalds  * Traffic Control
2391da177e4SLinus Torvalds  **************************************************************************/
2401da177e4SLinus Torvalds 
2411da177e4SLinus Torvalds META_COLLECTOR(int_tcindex)
2421da177e4SLinus Torvalds {
2431da177e4SLinus Torvalds 	dst->value = skb->tc_index;
2441da177e4SLinus Torvalds }
2451da177e4SLinus Torvalds 
2461da177e4SLinus Torvalds /**************************************************************************
2471da177e4SLinus Torvalds  * Routing
2481da177e4SLinus Torvalds  **************************************************************************/
2491da177e4SLinus Torvalds 
2501da177e4SLinus Torvalds META_COLLECTOR(int_rtclassid)
2511da177e4SLinus Torvalds {
252adf30907SEric Dumazet 	if (unlikely(skb_dst(skb) == NULL))
2531da177e4SLinus Torvalds 		*err = -1;
2541da177e4SLinus Torvalds 	else
255c7066f70SPatrick McHardy #ifdef CONFIG_IP_ROUTE_CLASSID
256adf30907SEric Dumazet 		dst->value = skb_dst(skb)->tclassid;
2577686ee1aSPatrick McHardy #else
2587686ee1aSPatrick McHardy 		dst->value = 0;
2591da177e4SLinus Torvalds #endif
2607686ee1aSPatrick McHardy }
2611da177e4SLinus Torvalds 
2621da177e4SLinus Torvalds META_COLLECTOR(int_rtiif)
2631da177e4SLinus Torvalds {
264511c3f92SEric Dumazet 	if (unlikely(skb_rtable(skb) == NULL))
2651da177e4SLinus Torvalds 		*err = -1;
2661da177e4SLinus Torvalds 	else
26792101b3bSDavid S. Miller 		dst->value = inet_iif(skb);
2681da177e4SLinus Torvalds }
2691da177e4SLinus Torvalds 
2701da177e4SLinus Torvalds /**************************************************************************
27148900629SThomas Graf  * Socket Attributes
27248900629SThomas Graf  **************************************************************************/
27348900629SThomas Graf 
2744f8f61ebSYang Yingliang #define skip_nonlocal(skb) \
2754f8f61ebSYang Yingliang 	(unlikely(skb->sk == NULL))
27648900629SThomas Graf 
27748900629SThomas Graf META_COLLECTOR(int_sk_family)
27848900629SThomas Graf {
2794f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2804f8f61ebSYang Yingliang 		*err = -1;
2814f8f61ebSYang Yingliang 		return;
2824f8f61ebSYang Yingliang 	}
28348900629SThomas Graf 	dst->value = skb->sk->sk_family;
28448900629SThomas Graf }
28548900629SThomas Graf 
28648900629SThomas Graf META_COLLECTOR(int_sk_state)
28748900629SThomas Graf {
2884f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2894f8f61ebSYang Yingliang 		*err = -1;
2904f8f61ebSYang Yingliang 		return;
2914f8f61ebSYang Yingliang 	}
29248900629SThomas Graf 	dst->value = skb->sk->sk_state;
29348900629SThomas Graf }
29448900629SThomas Graf 
29548900629SThomas Graf META_COLLECTOR(int_sk_reuse)
29648900629SThomas Graf {
2974f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2984f8f61ebSYang Yingliang 		*err = -1;
2994f8f61ebSYang Yingliang 		return;
3004f8f61ebSYang Yingliang 	}
30148900629SThomas Graf 	dst->value = skb->sk->sk_reuse;
30248900629SThomas Graf }
30348900629SThomas Graf 
30448900629SThomas Graf META_COLLECTOR(int_sk_bound_if)
30548900629SThomas Graf {
3064f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3074f8f61ebSYang Yingliang 		*err = -1;
3084f8f61ebSYang Yingliang 		return;
3094f8f61ebSYang Yingliang 	}
31048900629SThomas Graf 	/* No error if bound_dev_if is 0, legal userspace check */
31148900629SThomas Graf 	dst->value = skb->sk->sk_bound_dev_if;
31248900629SThomas Graf }
31348900629SThomas Graf 
31448900629SThomas Graf META_COLLECTOR(var_sk_bound_if)
31548900629SThomas Graf {
3164f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3174f8f61ebSYang Yingliang 		*err = -1;
3184f8f61ebSYang Yingliang 		return;
3194f8f61ebSYang Yingliang 	}
32048900629SThomas Graf 
32148900629SThomas Graf 	if (skb->sk->sk_bound_dev_if == 0) {
32248900629SThomas Graf 		dst->value = (unsigned long) "any";
32348900629SThomas Graf 		dst->len = 3;
32448900629SThomas Graf 	} else {
32548900629SThomas Graf 		struct net_device *dev;
32648900629SThomas Graf 
327d0075634SEric Dumazet 		rcu_read_lock();
3282939e275SEric Dumazet 		dev = dev_get_by_index_rcu(sock_net(skb->sk),
3292939e275SEric Dumazet 					   skb->sk->sk_bound_dev_if);
33048900629SThomas Graf 		*err = var_dev(dev, dst);
331d0075634SEric Dumazet 		rcu_read_unlock();
33248900629SThomas Graf 	}
33348900629SThomas Graf }
33448900629SThomas Graf 
33548900629SThomas Graf META_COLLECTOR(int_sk_refcnt)
33648900629SThomas Graf {
3374f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3384f8f61ebSYang Yingliang 		*err = -1;
3394f8f61ebSYang Yingliang 		return;
3404f8f61ebSYang Yingliang 	}
34148900629SThomas Graf 	dst->value = atomic_read(&skb->sk->sk_refcnt);
34248900629SThomas Graf }
34348900629SThomas Graf 
34448900629SThomas Graf META_COLLECTOR(int_sk_rcvbuf)
34548900629SThomas Graf {
34602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
34702a56c81SEric Dumazet 
34802a56c81SEric Dumazet 	if (!sk) {
3494f8f61ebSYang Yingliang 		*err = -1;
3504f8f61ebSYang Yingliang 		return;
3514f8f61ebSYang Yingliang 	}
35202a56c81SEric Dumazet 	dst->value = sk->sk_rcvbuf;
35348900629SThomas Graf }
35448900629SThomas Graf 
35548900629SThomas Graf META_COLLECTOR(int_sk_shutdown)
35648900629SThomas Graf {
35702a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
35802a56c81SEric Dumazet 
35902a56c81SEric Dumazet 	if (!sk) {
3604f8f61ebSYang Yingliang 		*err = -1;
3614f8f61ebSYang Yingliang 		return;
3624f8f61ebSYang Yingliang 	}
36302a56c81SEric Dumazet 	dst->value = sk->sk_shutdown;
36448900629SThomas Graf }
36548900629SThomas Graf 
36648900629SThomas Graf META_COLLECTOR(int_sk_proto)
36748900629SThomas Graf {
36802a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
36902a56c81SEric Dumazet 
37002a56c81SEric Dumazet 	if (!sk) {
3714f8f61ebSYang Yingliang 		*err = -1;
3724f8f61ebSYang Yingliang 		return;
3734f8f61ebSYang Yingliang 	}
37402a56c81SEric Dumazet 	dst->value = sk->sk_protocol;
37548900629SThomas Graf }
37648900629SThomas Graf 
37748900629SThomas Graf META_COLLECTOR(int_sk_type)
37848900629SThomas Graf {
37902a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
38002a56c81SEric Dumazet 
38102a56c81SEric Dumazet 	if (!sk) {
3824f8f61ebSYang Yingliang 		*err = -1;
3834f8f61ebSYang Yingliang 		return;
3844f8f61ebSYang Yingliang 	}
38502a56c81SEric Dumazet 	dst->value = sk->sk_type;
38648900629SThomas Graf }
38748900629SThomas Graf 
38848900629SThomas Graf META_COLLECTOR(int_sk_rmem_alloc)
38948900629SThomas Graf {
39002a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
39102a56c81SEric Dumazet 
39202a56c81SEric Dumazet 	if (!sk) {
3934f8f61ebSYang Yingliang 		*err = -1;
3944f8f61ebSYang Yingliang 		return;
3954f8f61ebSYang Yingliang 	}
39602a56c81SEric Dumazet 	dst->value = sk_rmem_alloc_get(sk);
39748900629SThomas Graf }
39848900629SThomas Graf 
39948900629SThomas Graf META_COLLECTOR(int_sk_wmem_alloc)
40048900629SThomas Graf {
40102a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
40202a56c81SEric Dumazet 
40302a56c81SEric Dumazet 	if (!sk) {
4044f8f61ebSYang Yingliang 		*err = -1;
4054f8f61ebSYang Yingliang 		return;
4064f8f61ebSYang Yingliang 	}
40702a56c81SEric Dumazet 	dst->value = sk_wmem_alloc_get(sk);
40848900629SThomas Graf }
40948900629SThomas Graf 
41048900629SThomas Graf META_COLLECTOR(int_sk_omem_alloc)
41148900629SThomas Graf {
41202a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
41302a56c81SEric Dumazet 
41402a56c81SEric Dumazet 	if (!sk) {
4154f8f61ebSYang Yingliang 		*err = -1;
4164f8f61ebSYang Yingliang 		return;
4174f8f61ebSYang Yingliang 	}
41802a56c81SEric Dumazet 	dst->value = atomic_read(&sk->sk_omem_alloc);
41948900629SThomas Graf }
42048900629SThomas Graf 
42148900629SThomas Graf META_COLLECTOR(int_sk_rcv_qlen)
42248900629SThomas Graf {
42302a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
42402a56c81SEric Dumazet 
42502a56c81SEric Dumazet 	if (!sk) {
4264f8f61ebSYang Yingliang 		*err = -1;
4274f8f61ebSYang Yingliang 		return;
4284f8f61ebSYang Yingliang 	}
42902a56c81SEric Dumazet 	dst->value = sk->sk_receive_queue.qlen;
43048900629SThomas Graf }
43148900629SThomas Graf 
43248900629SThomas Graf META_COLLECTOR(int_sk_snd_qlen)
43348900629SThomas Graf {
43402a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
43502a56c81SEric Dumazet 
43602a56c81SEric Dumazet 	if (!sk) {
4374f8f61ebSYang Yingliang 		*err = -1;
4384f8f61ebSYang Yingliang 		return;
4394f8f61ebSYang Yingliang 	}
44002a56c81SEric Dumazet 	dst->value = sk->sk_write_queue.qlen;
44148900629SThomas Graf }
44248900629SThomas Graf 
44348900629SThomas Graf META_COLLECTOR(int_sk_wmem_queued)
44448900629SThomas Graf {
44502a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
44602a56c81SEric Dumazet 
44702a56c81SEric Dumazet 	if (!sk) {
4484f8f61ebSYang Yingliang 		*err = -1;
4494f8f61ebSYang Yingliang 		return;
4504f8f61ebSYang Yingliang 	}
45102a56c81SEric Dumazet 	dst->value = sk->sk_wmem_queued;
45248900629SThomas Graf }
45348900629SThomas Graf 
45448900629SThomas Graf META_COLLECTOR(int_sk_fwd_alloc)
45548900629SThomas Graf {
45602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
45702a56c81SEric Dumazet 
45802a56c81SEric Dumazet 	if (!sk) {
4594f8f61ebSYang Yingliang 		*err = -1;
4604f8f61ebSYang Yingliang 		return;
4614f8f61ebSYang Yingliang 	}
46202a56c81SEric Dumazet 	dst->value = sk->sk_forward_alloc;
46348900629SThomas Graf }
46448900629SThomas Graf 
46548900629SThomas Graf META_COLLECTOR(int_sk_sndbuf)
46648900629SThomas Graf {
46702a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
46802a56c81SEric Dumazet 
46902a56c81SEric Dumazet 	if (!sk) {
4704f8f61ebSYang Yingliang 		*err = -1;
4714f8f61ebSYang Yingliang 		return;
4724f8f61ebSYang Yingliang 	}
47302a56c81SEric Dumazet 	dst->value = sk->sk_sndbuf;
47448900629SThomas Graf }
47548900629SThomas Graf 
47648900629SThomas Graf META_COLLECTOR(int_sk_alloc)
47748900629SThomas Graf {
47802a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
47902a56c81SEric Dumazet 
48002a56c81SEric Dumazet 	if (!sk) {
4814f8f61ebSYang Yingliang 		*err = -1;
4824f8f61ebSYang Yingliang 		return;
4834f8f61ebSYang Yingliang 	}
48402a56c81SEric Dumazet 	dst->value = (__force int) sk->sk_allocation;
48548900629SThomas Graf }
48648900629SThomas Graf 
48781c3d547SEric Dumazet META_COLLECTOR(int_sk_hash)
48848900629SThomas Graf {
4894f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
4904f8f61ebSYang Yingliang 		*err = -1;
4914f8f61ebSYang Yingliang 		return;
4924f8f61ebSYang Yingliang 	}
49381c3d547SEric Dumazet 	dst->value = skb->sk->sk_hash;
49448900629SThomas Graf }
49548900629SThomas Graf 
49648900629SThomas Graf META_COLLECTOR(int_sk_lingertime)
49748900629SThomas Graf {
49802a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
49902a56c81SEric Dumazet 
50002a56c81SEric Dumazet 	if (!sk) {
5014f8f61ebSYang Yingliang 		*err = -1;
5024f8f61ebSYang Yingliang 		return;
5034f8f61ebSYang Yingliang 	}
50402a56c81SEric Dumazet 	dst->value = sk->sk_lingertime / HZ;
50548900629SThomas Graf }
50648900629SThomas Graf 
50748900629SThomas Graf META_COLLECTOR(int_sk_err_qlen)
50848900629SThomas Graf {
50902a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
51002a56c81SEric Dumazet 
51102a56c81SEric Dumazet 	if (!sk) {
5124f8f61ebSYang Yingliang 		*err = -1;
5134f8f61ebSYang Yingliang 		return;
5144f8f61ebSYang Yingliang 	}
51502a56c81SEric Dumazet 	dst->value = sk->sk_error_queue.qlen;
51648900629SThomas Graf }
51748900629SThomas Graf 
51848900629SThomas Graf META_COLLECTOR(int_sk_ack_bl)
51948900629SThomas Graf {
52002a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
52102a56c81SEric Dumazet 
52202a56c81SEric Dumazet 	if (!sk) {
5234f8f61ebSYang Yingliang 		*err = -1;
5244f8f61ebSYang Yingliang 		return;
5254f8f61ebSYang Yingliang 	}
52602a56c81SEric Dumazet 	dst->value = sk->sk_ack_backlog;
52748900629SThomas Graf }
52848900629SThomas Graf 
52948900629SThomas Graf META_COLLECTOR(int_sk_max_ack_bl)
53048900629SThomas Graf {
53102a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
53202a56c81SEric Dumazet 
53302a56c81SEric Dumazet 	if (!sk) {
5344f8f61ebSYang Yingliang 		*err = -1;
5354f8f61ebSYang Yingliang 		return;
5364f8f61ebSYang Yingliang 	}
53702a56c81SEric Dumazet 	dst->value = sk->sk_max_ack_backlog;
53848900629SThomas Graf }
53948900629SThomas Graf 
54048900629SThomas Graf META_COLLECTOR(int_sk_prio)
54148900629SThomas Graf {
54202a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
54302a56c81SEric Dumazet 
54402a56c81SEric Dumazet 	if (!sk) {
5454f8f61ebSYang Yingliang 		*err = -1;
5464f8f61ebSYang Yingliang 		return;
5474f8f61ebSYang Yingliang 	}
54802a56c81SEric Dumazet 	dst->value = sk->sk_priority;
54948900629SThomas Graf }
55048900629SThomas Graf 
55148900629SThomas Graf META_COLLECTOR(int_sk_rcvlowat)
55248900629SThomas Graf {
55302a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
55402a56c81SEric Dumazet 
55502a56c81SEric Dumazet 	if (!sk) {
5564f8f61ebSYang Yingliang 		*err = -1;
5574f8f61ebSYang Yingliang 		return;
5584f8f61ebSYang Yingliang 	}
55902a56c81SEric Dumazet 	dst->value = sk->sk_rcvlowat;
56048900629SThomas Graf }
56148900629SThomas Graf 
56248900629SThomas Graf META_COLLECTOR(int_sk_rcvtimeo)
56348900629SThomas Graf {
56402a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
56502a56c81SEric Dumazet 
56602a56c81SEric Dumazet 	if (!sk) {
5674f8f61ebSYang Yingliang 		*err = -1;
5684f8f61ebSYang Yingliang 		return;
5694f8f61ebSYang Yingliang 	}
57002a56c81SEric Dumazet 	dst->value = sk->sk_rcvtimeo / HZ;
57148900629SThomas Graf }
57248900629SThomas Graf 
57348900629SThomas Graf META_COLLECTOR(int_sk_sndtimeo)
57448900629SThomas Graf {
57502a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
57602a56c81SEric Dumazet 
57702a56c81SEric Dumazet 	if (!sk) {
5784f8f61ebSYang Yingliang 		*err = -1;
5794f8f61ebSYang Yingliang 		return;
5804f8f61ebSYang Yingliang 	}
58102a56c81SEric Dumazet 	dst->value = sk->sk_sndtimeo / HZ;
58248900629SThomas Graf }
58348900629SThomas Graf 
58448900629SThomas Graf META_COLLECTOR(int_sk_sendmsg_off)
58548900629SThomas Graf {
58602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
58702a56c81SEric Dumazet 
58802a56c81SEric Dumazet 	if (!sk) {
5894f8f61ebSYang Yingliang 		*err = -1;
5904f8f61ebSYang Yingliang 		return;
5914f8f61ebSYang Yingliang 	}
59202a56c81SEric Dumazet 	dst->value = sk->sk_frag.offset;
59348900629SThomas Graf }
59448900629SThomas Graf 
59548900629SThomas Graf META_COLLECTOR(int_sk_write_pend)
59648900629SThomas Graf {
59702a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
59802a56c81SEric Dumazet 
59902a56c81SEric Dumazet 	if (!sk) {
6004f8f61ebSYang Yingliang 		*err = -1;
6014f8f61ebSYang Yingliang 		return;
6024f8f61ebSYang Yingliang 	}
60302a56c81SEric Dumazet 	dst->value = sk->sk_write_pending;
60448900629SThomas Graf }
60548900629SThomas Graf 
60648900629SThomas Graf /**************************************************************************
6071da177e4SLinus Torvalds  * Meta value collectors assignment table
6081da177e4SLinus Torvalds  **************************************************************************/
6091da177e4SLinus Torvalds 
610cc7ec456SEric Dumazet struct meta_ops {
6111da177e4SLinus Torvalds 	void		(*get)(struct sk_buff *, struct tcf_pkt_info *,
6121da177e4SLinus Torvalds 			       struct meta_value *, struct meta_obj *, int *);
6131da177e4SLinus Torvalds };
6141da177e4SLinus Torvalds 
61548900629SThomas Graf #define META_ID(name) TCF_META_ID_##name
61648900629SThomas Graf #define META_FUNC(name) { .get = meta_##name }
61748900629SThomas Graf 
6181da177e4SLinus Torvalds /* Meta value operations table listing all meta value collectors and
6191da177e4SLinus Torvalds  * assigns them to a type and meta id. */
6201da177e4SLinus Torvalds static struct meta_ops __meta_ops[TCF_META_TYPE_MAX + 1][TCF_META_ID_MAX + 1] = {
6211da177e4SLinus Torvalds 	[TCF_META_TYPE_VAR] = {
62248900629SThomas Graf 		[META_ID(DEV)]			= META_FUNC(var_dev),
62348900629SThomas Graf 		[META_ID(SK_BOUND_IF)] 		= META_FUNC(var_sk_bound_if),
6241da177e4SLinus Torvalds 	},
6251da177e4SLinus Torvalds 	[TCF_META_TYPE_INT] = {
62648900629SThomas Graf 		[META_ID(RANDOM)]		= META_FUNC(int_random),
62748900629SThomas Graf 		[META_ID(LOADAVG_0)]		= META_FUNC(int_loadavg_0),
62848900629SThomas Graf 		[META_ID(LOADAVG_1)]		= META_FUNC(int_loadavg_1),
62948900629SThomas Graf 		[META_ID(LOADAVG_2)]		= META_FUNC(int_loadavg_2),
63048900629SThomas Graf 		[META_ID(DEV)]			= META_FUNC(int_dev),
63148900629SThomas Graf 		[META_ID(PRIORITY)]		= META_FUNC(int_priority),
63248900629SThomas Graf 		[META_ID(PROTOCOL)]		= META_FUNC(int_protocol),
63348900629SThomas Graf 		[META_ID(PKTTYPE)]		= META_FUNC(int_pkttype),
63448900629SThomas Graf 		[META_ID(PKTLEN)]		= META_FUNC(int_pktlen),
63548900629SThomas Graf 		[META_ID(DATALEN)]		= META_FUNC(int_datalen),
63648900629SThomas Graf 		[META_ID(MACLEN)]		= META_FUNC(int_maclen),
63782e91ffeSThomas Graf 		[META_ID(NFMARK)]		= META_FUNC(int_mark),
63848900629SThomas Graf 		[META_ID(TCINDEX)]		= META_FUNC(int_tcindex),
63948900629SThomas Graf 		[META_ID(RTCLASSID)]		= META_FUNC(int_rtclassid),
64048900629SThomas Graf 		[META_ID(RTIIF)]		= META_FUNC(int_rtiif),
64148900629SThomas Graf 		[META_ID(SK_FAMILY)]		= META_FUNC(int_sk_family),
64248900629SThomas Graf 		[META_ID(SK_STATE)]		= META_FUNC(int_sk_state),
64348900629SThomas Graf 		[META_ID(SK_REUSE)]		= META_FUNC(int_sk_reuse),
64448900629SThomas Graf 		[META_ID(SK_BOUND_IF)]		= META_FUNC(int_sk_bound_if),
64548900629SThomas Graf 		[META_ID(SK_REFCNT)]		= META_FUNC(int_sk_refcnt),
64648900629SThomas Graf 		[META_ID(SK_RCVBUF)]		= META_FUNC(int_sk_rcvbuf),
64748900629SThomas Graf 		[META_ID(SK_SNDBUF)]		= META_FUNC(int_sk_sndbuf),
64848900629SThomas Graf 		[META_ID(SK_SHUTDOWN)]		= META_FUNC(int_sk_shutdown),
64948900629SThomas Graf 		[META_ID(SK_PROTO)]		= META_FUNC(int_sk_proto),
65048900629SThomas Graf 		[META_ID(SK_TYPE)]		= META_FUNC(int_sk_type),
65148900629SThomas Graf 		[META_ID(SK_RMEM_ALLOC)]	= META_FUNC(int_sk_rmem_alloc),
65248900629SThomas Graf 		[META_ID(SK_WMEM_ALLOC)]	= META_FUNC(int_sk_wmem_alloc),
65348900629SThomas Graf 		[META_ID(SK_OMEM_ALLOC)]	= META_FUNC(int_sk_omem_alloc),
65448900629SThomas Graf 		[META_ID(SK_WMEM_QUEUED)]	= META_FUNC(int_sk_wmem_queued),
65548900629SThomas Graf 		[META_ID(SK_RCV_QLEN)]		= META_FUNC(int_sk_rcv_qlen),
65648900629SThomas Graf 		[META_ID(SK_SND_QLEN)]		= META_FUNC(int_sk_snd_qlen),
65748900629SThomas Graf 		[META_ID(SK_ERR_QLEN)]		= META_FUNC(int_sk_err_qlen),
65848900629SThomas Graf 		[META_ID(SK_FORWARD_ALLOCS)]	= META_FUNC(int_sk_fwd_alloc),
65948900629SThomas Graf 		[META_ID(SK_ALLOCS)]		= META_FUNC(int_sk_alloc),
66081c3d547SEric Dumazet 		[META_ID(SK_HASH)]		= META_FUNC(int_sk_hash),
66148900629SThomas Graf 		[META_ID(SK_LINGERTIME)]	= META_FUNC(int_sk_lingertime),
66248900629SThomas Graf 		[META_ID(SK_ACK_BACKLOG)]	= META_FUNC(int_sk_ack_bl),
66348900629SThomas Graf 		[META_ID(SK_MAX_ACK_BACKLOG)]	= META_FUNC(int_sk_max_ack_bl),
66448900629SThomas Graf 		[META_ID(SK_PRIO)]		= META_FUNC(int_sk_prio),
66548900629SThomas Graf 		[META_ID(SK_RCVLOWAT)]		= META_FUNC(int_sk_rcvlowat),
66648900629SThomas Graf 		[META_ID(SK_RCVTIMEO)]		= META_FUNC(int_sk_rcvtimeo),
66748900629SThomas Graf 		[META_ID(SK_SNDTIMEO)]		= META_FUNC(int_sk_sndtimeo),
66848900629SThomas Graf 		[META_ID(SK_SENDMSG_OFF)]	= META_FUNC(int_sk_sendmsg_off),
66948900629SThomas Graf 		[META_ID(SK_WRITE_PENDING)]	= META_FUNC(int_sk_write_pend),
6703113e88cSStephen Hemminger 		[META_ID(VLAN_TAG)]		= META_FUNC(int_vlan_tag),
671c2e3143eSStephen Hemminger 		[META_ID(RXHASH)]		= META_FUNC(int_rxhash),
6721da177e4SLinus Torvalds 	}
6731da177e4SLinus Torvalds };
6741da177e4SLinus Torvalds 
6751da177e4SLinus Torvalds static inline struct meta_ops *meta_ops(struct meta_value *val)
6761da177e4SLinus Torvalds {
6771da177e4SLinus Torvalds 	return &__meta_ops[meta_type(val)][meta_id(val)];
6781da177e4SLinus Torvalds }
6791da177e4SLinus Torvalds 
6801da177e4SLinus Torvalds /**************************************************************************
6811da177e4SLinus Torvalds  * Type specific operations for TCF_META_TYPE_VAR
6821da177e4SLinus Torvalds  **************************************************************************/
6831da177e4SLinus Torvalds 
6841da177e4SLinus Torvalds static int meta_var_compare(struct meta_obj *a, struct meta_obj *b)
6851da177e4SLinus Torvalds {
6861da177e4SLinus Torvalds 	int r = a->len - b->len;
6871da177e4SLinus Torvalds 
6881da177e4SLinus Torvalds 	if (r == 0)
6891da177e4SLinus Torvalds 		r = memcmp((void *) a->value, (void *) b->value, a->len);
6901da177e4SLinus Torvalds 
6911da177e4SLinus Torvalds 	return r;
6921da177e4SLinus Torvalds }
6931da177e4SLinus Torvalds 
694add93b61SPatrick McHardy static int meta_var_change(struct meta_value *dst, struct nlattr *nla)
6951da177e4SLinus Torvalds {
696add93b61SPatrick McHardy 	int len = nla_len(nla);
6971da177e4SLinus Torvalds 
698add93b61SPatrick McHardy 	dst->val = (unsigned long)kmemdup(nla_data(nla), len, GFP_KERNEL);
6991da177e4SLinus Torvalds 	if (dst->val == 0UL)
7001da177e4SLinus Torvalds 		return -ENOMEM;
7011da177e4SLinus Torvalds 	dst->len = len;
7021da177e4SLinus Torvalds 	return 0;
7031da177e4SLinus Torvalds }
7041da177e4SLinus Torvalds 
7051da177e4SLinus Torvalds static void meta_var_destroy(struct meta_value *v)
7061da177e4SLinus Torvalds {
7071da177e4SLinus Torvalds 	kfree((void *) v->val);
7081da177e4SLinus Torvalds }
7091da177e4SLinus Torvalds 
7101da177e4SLinus Torvalds static void meta_var_apply_extras(struct meta_value *v,
7111da177e4SLinus Torvalds 				  struct meta_obj *dst)
7121da177e4SLinus Torvalds {
7131da177e4SLinus Torvalds 	int shift = v->hdr.shift;
7141da177e4SLinus Torvalds 
7151da177e4SLinus Torvalds 	if (shift && shift < dst->len)
7161da177e4SLinus Torvalds 		dst->len -= shift;
7171da177e4SLinus Torvalds }
7181da177e4SLinus Torvalds 
7191da177e4SLinus Torvalds static int meta_var_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
7201da177e4SLinus Torvalds {
7211b34ec43SDavid S. Miller 	if (v->val && v->len &&
7221b34ec43SDavid S. Miller 	    nla_put(skb, tlv, v->len, (void *) v->val))
7231b34ec43SDavid S. Miller 		goto nla_put_failure;
7241da177e4SLinus Torvalds 	return 0;
7251da177e4SLinus Torvalds 
726add93b61SPatrick McHardy nla_put_failure:
7271da177e4SLinus Torvalds 	return -1;
7281da177e4SLinus Torvalds }
7291da177e4SLinus Torvalds 
7301da177e4SLinus Torvalds /**************************************************************************
7311da177e4SLinus Torvalds  * Type specific operations for TCF_META_TYPE_INT
7321da177e4SLinus Torvalds  **************************************************************************/
7331da177e4SLinus Torvalds 
7341da177e4SLinus Torvalds static int meta_int_compare(struct meta_obj *a, struct meta_obj *b)
7351da177e4SLinus Torvalds {
7361da177e4SLinus Torvalds 	/* Let gcc optimize it, the unlikely is not really based on
7371da177e4SLinus Torvalds 	 * some numbers but jump free code for mismatches seems
7381da177e4SLinus Torvalds 	 * more logical. */
73998e56405SThomas Graf 	if (unlikely(a->value == b->value))
7401da177e4SLinus Torvalds 		return 0;
74198e56405SThomas Graf 	else if (a->value < b->value)
7421da177e4SLinus Torvalds 		return -1;
7431da177e4SLinus Torvalds 	else
7441da177e4SLinus Torvalds 		return 1;
7451da177e4SLinus Torvalds }
7461da177e4SLinus Torvalds 
747add93b61SPatrick McHardy static int meta_int_change(struct meta_value *dst, struct nlattr *nla)
7481da177e4SLinus Torvalds {
749add93b61SPatrick McHardy 	if (nla_len(nla) >= sizeof(unsigned long)) {
750add93b61SPatrick McHardy 		dst->val = *(unsigned long *) nla_data(nla);
7511da177e4SLinus Torvalds 		dst->len = sizeof(unsigned long);
752add93b61SPatrick McHardy 	} else if (nla_len(nla) == sizeof(u32)) {
7531587bac4SPatrick McHardy 		dst->val = nla_get_u32(nla);
7541da177e4SLinus Torvalds 		dst->len = sizeof(u32);
7551da177e4SLinus Torvalds 	} else
7561da177e4SLinus Torvalds 		return -EINVAL;
7571da177e4SLinus Torvalds 
7581da177e4SLinus Torvalds 	return 0;
7591da177e4SLinus Torvalds }
7601da177e4SLinus Torvalds 
7611da177e4SLinus Torvalds static void meta_int_apply_extras(struct meta_value *v,
7621da177e4SLinus Torvalds 				  struct meta_obj *dst)
7631da177e4SLinus Torvalds {
7641da177e4SLinus Torvalds 	if (v->hdr.shift)
7651da177e4SLinus Torvalds 		dst->value >>= v->hdr.shift;
7661da177e4SLinus Torvalds 
7671da177e4SLinus Torvalds 	if (v->val)
7681da177e4SLinus Torvalds 		dst->value &= v->val;
7691da177e4SLinus Torvalds }
7701da177e4SLinus Torvalds 
7711da177e4SLinus Torvalds static int meta_int_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
7721da177e4SLinus Torvalds {
7731b34ec43SDavid S. Miller 	if (v->len == sizeof(unsigned long)) {
7741b34ec43SDavid S. Miller 		if (nla_put(skb, tlv, sizeof(unsigned long), &v->val))
7751b34ec43SDavid S. Miller 			goto nla_put_failure;
7761b34ec43SDavid S. Miller 	} else if (v->len == sizeof(u32)) {
7771b34ec43SDavid S. Miller 		if (nla_put_u32(skb, tlv, v->val))
7781b34ec43SDavid S. Miller 			goto nla_put_failure;
7791b34ec43SDavid S. Miller 	}
7801da177e4SLinus Torvalds 
7811da177e4SLinus Torvalds 	return 0;
7821da177e4SLinus Torvalds 
783add93b61SPatrick McHardy nla_put_failure:
7841da177e4SLinus Torvalds 	return -1;
7851da177e4SLinus Torvalds }
7861da177e4SLinus Torvalds 
7871da177e4SLinus Torvalds /**************************************************************************
7881da177e4SLinus Torvalds  * Type specific operations table
7891da177e4SLinus Torvalds  **************************************************************************/
7901da177e4SLinus Torvalds 
791cc7ec456SEric Dumazet struct meta_type_ops {
7921da177e4SLinus Torvalds 	void	(*destroy)(struct meta_value *);
7931da177e4SLinus Torvalds 	int	(*compare)(struct meta_obj *, struct meta_obj *);
794add93b61SPatrick McHardy 	int	(*change)(struct meta_value *, struct nlattr *);
7951da177e4SLinus Torvalds 	void	(*apply_extras)(struct meta_value *, struct meta_obj *);
7961da177e4SLinus Torvalds 	int	(*dump)(struct sk_buff *, struct meta_value *, int);
7971da177e4SLinus Torvalds };
7981da177e4SLinus Torvalds 
799cfe2f14cSJulia Lawall static const struct meta_type_ops __meta_type_ops[TCF_META_TYPE_MAX + 1] = {
8001da177e4SLinus Torvalds 	[TCF_META_TYPE_VAR] = {
8011da177e4SLinus Torvalds 		.destroy = meta_var_destroy,
8021da177e4SLinus Torvalds 		.compare = meta_var_compare,
8031da177e4SLinus Torvalds 		.change = meta_var_change,
8041da177e4SLinus Torvalds 		.apply_extras = meta_var_apply_extras,
8051da177e4SLinus Torvalds 		.dump = meta_var_dump
8061da177e4SLinus Torvalds 	},
8071da177e4SLinus Torvalds 	[TCF_META_TYPE_INT] = {
8081da177e4SLinus Torvalds 		.compare = meta_int_compare,
8091da177e4SLinus Torvalds 		.change = meta_int_change,
8101da177e4SLinus Torvalds 		.apply_extras = meta_int_apply_extras,
8111da177e4SLinus Torvalds 		.dump = meta_int_dump
8121da177e4SLinus Torvalds 	}
8131da177e4SLinus Torvalds };
8141da177e4SLinus Torvalds 
815cfe2f14cSJulia Lawall static inline const struct meta_type_ops *meta_type_ops(struct meta_value *v)
8161da177e4SLinus Torvalds {
8171da177e4SLinus Torvalds 	return &__meta_type_ops[meta_type(v)];
8181da177e4SLinus Torvalds }
8191da177e4SLinus Torvalds 
8201da177e4SLinus Torvalds /**************************************************************************
8211da177e4SLinus Torvalds  * Core
8221da177e4SLinus Torvalds  **************************************************************************/
8231da177e4SLinus Torvalds 
824ed7af3b3SStephen Hemminger static int meta_get(struct sk_buff *skb, struct tcf_pkt_info *info,
8251da177e4SLinus Torvalds 		    struct meta_value *v, struct meta_obj *dst)
8261da177e4SLinus Torvalds {
8271da177e4SLinus Torvalds 	int err = 0;
8281da177e4SLinus Torvalds 
8291da177e4SLinus Torvalds 	if (meta_id(v) == TCF_META_ID_VALUE) {
8301da177e4SLinus Torvalds 		dst->value = v->val;
8311da177e4SLinus Torvalds 		dst->len = v->len;
8321da177e4SLinus Torvalds 		return 0;
8331da177e4SLinus Torvalds 	}
8341da177e4SLinus Torvalds 
8351da177e4SLinus Torvalds 	meta_ops(v)->get(skb, info, v, dst, &err);
8361da177e4SLinus Torvalds 	if (err < 0)
8371da177e4SLinus Torvalds 		return err;
8381da177e4SLinus Torvalds 
8391da177e4SLinus Torvalds 	if (meta_type_ops(v)->apply_extras)
8401da177e4SLinus Torvalds 		meta_type_ops(v)->apply_extras(v, dst);
8411da177e4SLinus Torvalds 
8421da177e4SLinus Torvalds 	return 0;
8431da177e4SLinus Torvalds }
8441da177e4SLinus Torvalds 
8451da177e4SLinus Torvalds static int em_meta_match(struct sk_buff *skb, struct tcf_ematch *m,
8461da177e4SLinus Torvalds 			 struct tcf_pkt_info *info)
8471da177e4SLinus Torvalds {
8481da177e4SLinus Torvalds 	int r;
8491da177e4SLinus Torvalds 	struct meta_match *meta = (struct meta_match *) m->data;
8501da177e4SLinus Torvalds 	struct meta_obj l_value, r_value;
8511da177e4SLinus Torvalds 
8521da177e4SLinus Torvalds 	if (meta_get(skb, info, &meta->lvalue, &l_value) < 0 ||
8531da177e4SLinus Torvalds 	    meta_get(skb, info, &meta->rvalue, &r_value) < 0)
8541da177e4SLinus Torvalds 		return 0;
8551da177e4SLinus Torvalds 
8561da177e4SLinus Torvalds 	r = meta_type_ops(&meta->lvalue)->compare(&l_value, &r_value);
8571da177e4SLinus Torvalds 
8581da177e4SLinus Torvalds 	switch (meta->lvalue.hdr.op) {
8591da177e4SLinus Torvalds 	case TCF_EM_OPND_EQ:
8601da177e4SLinus Torvalds 		return !r;
8611da177e4SLinus Torvalds 	case TCF_EM_OPND_LT:
8621da177e4SLinus Torvalds 		return r < 0;
8631da177e4SLinus Torvalds 	case TCF_EM_OPND_GT:
8641da177e4SLinus Torvalds 		return r > 0;
8651da177e4SLinus Torvalds 	}
8661da177e4SLinus Torvalds 
8671da177e4SLinus Torvalds 	return 0;
8681da177e4SLinus Torvalds }
8691da177e4SLinus Torvalds 
870ed7af3b3SStephen Hemminger static void meta_delete(struct meta_match *meta)
8711da177e4SLinus Torvalds {
87204f217acSStephen Hemminger 	if (meta) {
873cfe2f14cSJulia Lawall 		const struct meta_type_ops *ops = meta_type_ops(&meta->lvalue);
8741da177e4SLinus Torvalds 
8751da177e4SLinus Torvalds 		if (ops && ops->destroy) {
8761da177e4SLinus Torvalds 			ops->destroy(&meta->lvalue);
8771da177e4SLinus Torvalds 			ops->destroy(&meta->rvalue);
8781da177e4SLinus Torvalds 		}
87904f217acSStephen Hemminger 	}
8801da177e4SLinus Torvalds 
8811da177e4SLinus Torvalds 	kfree(meta);
8821da177e4SLinus Torvalds }
8831da177e4SLinus Torvalds 
884add93b61SPatrick McHardy static inline int meta_change_data(struct meta_value *dst, struct nlattr *nla)
8851da177e4SLinus Torvalds {
886add93b61SPatrick McHardy 	if (nla) {
887add93b61SPatrick McHardy 		if (nla_len(nla) == 0)
8881da177e4SLinus Torvalds 			return -EINVAL;
8891da177e4SLinus Torvalds 
890add93b61SPatrick McHardy 		return meta_type_ops(dst)->change(dst, nla);
8911da177e4SLinus Torvalds 	}
8921da177e4SLinus Torvalds 
8931da177e4SLinus Torvalds 	return 0;
8941da177e4SLinus Torvalds }
8951da177e4SLinus Torvalds 
8961da177e4SLinus Torvalds static inline int meta_is_supported(struct meta_value *val)
8971da177e4SLinus Torvalds {
898cc7ec456SEric Dumazet 	return !meta_id(val) || meta_ops(val)->get;
8991da177e4SLinus Torvalds }
9001da177e4SLinus Torvalds 
9017a9c1bd4SPatrick McHardy static const struct nla_policy meta_policy[TCA_EM_META_MAX + 1] = {
9027a9c1bd4SPatrick McHardy 	[TCA_EM_META_HDR]	= { .len = sizeof(struct tcf_meta_hdr) },
9037a9c1bd4SPatrick McHardy };
9047a9c1bd4SPatrick McHardy 
90582a470f1SJohn Fastabend static int em_meta_change(struct net *net, void *data, int len,
9061da177e4SLinus Torvalds 			  struct tcf_ematch *m)
9071da177e4SLinus Torvalds {
908cee63723SPatrick McHardy 	int err;
909add93b61SPatrick McHardy 	struct nlattr *tb[TCA_EM_META_MAX + 1];
9101da177e4SLinus Torvalds 	struct tcf_meta_hdr *hdr;
9111da177e4SLinus Torvalds 	struct meta_match *meta = NULL;
9121da177e4SLinus Torvalds 
9137a9c1bd4SPatrick McHardy 	err = nla_parse(tb, TCA_EM_META_MAX, data, len, meta_policy);
914cee63723SPatrick McHardy 	if (err < 0)
9151da177e4SLinus Torvalds 		goto errout;
9161da177e4SLinus Torvalds 
917cee63723SPatrick McHardy 	err = -EINVAL;
9187a9c1bd4SPatrick McHardy 	if (tb[TCA_EM_META_HDR] == NULL)
9191da177e4SLinus Torvalds 		goto errout;
920add93b61SPatrick McHardy 	hdr = nla_data(tb[TCA_EM_META_HDR]);
9211da177e4SLinus Torvalds 
9221da177e4SLinus Torvalds 	if (TCF_META_TYPE(hdr->left.kind) != TCF_META_TYPE(hdr->right.kind) ||
9231da177e4SLinus Torvalds 	    TCF_META_TYPE(hdr->left.kind) > TCF_META_TYPE_MAX ||
9241da177e4SLinus Torvalds 	    TCF_META_ID(hdr->left.kind) > TCF_META_ID_MAX ||
9251da177e4SLinus Torvalds 	    TCF_META_ID(hdr->right.kind) > TCF_META_ID_MAX)
9261da177e4SLinus Torvalds 		goto errout;
9271da177e4SLinus Torvalds 
9280da974f4SPanagiotis Issaris 	meta = kzalloc(sizeof(*meta), GFP_KERNEL);
9290c4e4020Sstephen hemminger 	if (meta == NULL) {
9300c4e4020Sstephen hemminger 		err = -ENOMEM;
9311da177e4SLinus Torvalds 		goto errout;
9320c4e4020Sstephen hemminger 	}
9331da177e4SLinus Torvalds 
9341da177e4SLinus Torvalds 	memcpy(&meta->lvalue.hdr, &hdr->left, sizeof(hdr->left));
9351da177e4SLinus Torvalds 	memcpy(&meta->rvalue.hdr, &hdr->right, sizeof(hdr->right));
9361da177e4SLinus Torvalds 
9371da177e4SLinus Torvalds 	if (!meta_is_supported(&meta->lvalue) ||
9381da177e4SLinus Torvalds 	    !meta_is_supported(&meta->rvalue)) {
9391da177e4SLinus Torvalds 		err = -EOPNOTSUPP;
9401da177e4SLinus Torvalds 		goto errout;
9411da177e4SLinus Torvalds 	}
9421da177e4SLinus Torvalds 
943add93b61SPatrick McHardy 	if (meta_change_data(&meta->lvalue, tb[TCA_EM_META_LVALUE]) < 0 ||
944add93b61SPatrick McHardy 	    meta_change_data(&meta->rvalue, tb[TCA_EM_META_RVALUE]) < 0)
9451da177e4SLinus Torvalds 		goto errout;
9461da177e4SLinus Torvalds 
9471da177e4SLinus Torvalds 	m->datalen = sizeof(*meta);
9481da177e4SLinus Torvalds 	m->data = (unsigned long) meta;
9491da177e4SLinus Torvalds 
9501da177e4SLinus Torvalds 	err = 0;
9511da177e4SLinus Torvalds errout:
9521da177e4SLinus Torvalds 	if (err && meta)
9531da177e4SLinus Torvalds 		meta_delete(meta);
9541da177e4SLinus Torvalds 	return err;
9551da177e4SLinus Torvalds }
9561da177e4SLinus Torvalds 
95782a470f1SJohn Fastabend static void em_meta_destroy(struct tcf_ematch *m)
9581da177e4SLinus Torvalds {
9591da177e4SLinus Torvalds 	if (m)
9601da177e4SLinus Torvalds 		meta_delete((struct meta_match *) m->data);
9611da177e4SLinus Torvalds }
9621da177e4SLinus Torvalds 
9631da177e4SLinus Torvalds static int em_meta_dump(struct sk_buff *skb, struct tcf_ematch *em)
9641da177e4SLinus Torvalds {
9651da177e4SLinus Torvalds 	struct meta_match *meta = (struct meta_match *) em->data;
9661da177e4SLinus Torvalds 	struct tcf_meta_hdr hdr;
967cfe2f14cSJulia Lawall 	const struct meta_type_ops *ops;
9681da177e4SLinus Torvalds 
9691da177e4SLinus Torvalds 	memset(&hdr, 0, sizeof(hdr));
9701da177e4SLinus Torvalds 	memcpy(&hdr.left, &meta->lvalue.hdr, sizeof(hdr.left));
9711da177e4SLinus Torvalds 	memcpy(&hdr.right, &meta->rvalue.hdr, sizeof(hdr.right));
9721da177e4SLinus Torvalds 
9731b34ec43SDavid S. Miller 	if (nla_put(skb, TCA_EM_META_HDR, sizeof(hdr), &hdr))
9741b34ec43SDavid S. Miller 		goto nla_put_failure;
9751da177e4SLinus Torvalds 
9761da177e4SLinus Torvalds 	ops = meta_type_ops(&meta->lvalue);
9771da177e4SLinus Torvalds 	if (ops->dump(skb, &meta->lvalue, TCA_EM_META_LVALUE) < 0 ||
9781da177e4SLinus Torvalds 	    ops->dump(skb, &meta->rvalue, TCA_EM_META_RVALUE) < 0)
979add93b61SPatrick McHardy 		goto nla_put_failure;
9801da177e4SLinus Torvalds 
9811da177e4SLinus Torvalds 	return 0;
9821da177e4SLinus Torvalds 
983add93b61SPatrick McHardy nla_put_failure:
9841da177e4SLinus Torvalds 	return -1;
9851da177e4SLinus Torvalds }
9861da177e4SLinus Torvalds 
9871da177e4SLinus Torvalds static struct tcf_ematch_ops em_meta_ops = {
9881da177e4SLinus Torvalds 	.kind	  = TCF_EM_META,
9891da177e4SLinus Torvalds 	.change	  = em_meta_change,
9901da177e4SLinus Torvalds 	.match	  = em_meta_match,
9911da177e4SLinus Torvalds 	.destroy  = em_meta_destroy,
9921da177e4SLinus Torvalds 	.dump	  = em_meta_dump,
9931da177e4SLinus Torvalds 	.owner	  = THIS_MODULE,
9941da177e4SLinus Torvalds 	.link	  = LIST_HEAD_INIT(em_meta_ops.link)
9951da177e4SLinus Torvalds };
9961da177e4SLinus Torvalds 
9971da177e4SLinus Torvalds static int __init init_em_meta(void)
9981da177e4SLinus Torvalds {
9991da177e4SLinus Torvalds 	return tcf_em_register(&em_meta_ops);
10001da177e4SLinus Torvalds }
10011da177e4SLinus Torvalds 
10021da177e4SLinus Torvalds static void __exit exit_em_meta(void)
10031da177e4SLinus Torvalds {
10041da177e4SLinus Torvalds 	tcf_em_unregister(&em_meta_ops);
10051da177e4SLinus Torvalds }
10061da177e4SLinus Torvalds 
10071da177e4SLinus Torvalds MODULE_LICENSE("GPL");
10081da177e4SLinus Torvalds 
10091da177e4SLinus Torvalds module_init(init_em_meta);
10101da177e4SLinus Torvalds module_exit(exit_em_meta);
1011db3d99c0SPatrick McHardy 
1012db3d99c0SPatrick McHardy MODULE_ALIAS_TCF_EMATCH(TCF_EM_META);
1013