xref: /openbmc/linux/net/sched/em_meta.c (revision 2874c5fd)
12874c5fdSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
21da177e4SLinus Torvalds /*
31da177e4SLinus Torvalds  * net/sched/em_meta.c	Metadata ematch
41da177e4SLinus Torvalds  *
51da177e4SLinus Torvalds  * Authors:	Thomas Graf <tgraf@suug.ch>
61da177e4SLinus Torvalds  *
71da177e4SLinus Torvalds  * ==========================================================================
81da177e4SLinus Torvalds  *
91da177e4SLinus Torvalds  * 	The metadata ematch compares two meta objects where each object
101da177e4SLinus Torvalds  * 	represents either a meta value stored in the kernel or a static
111da177e4SLinus Torvalds  * 	value provided by userspace. The objects are not provided by
121da177e4SLinus Torvalds  * 	userspace itself but rather a definition providing the information
131da177e4SLinus Torvalds  * 	to build them. Every object is of a certain type which must be
141da177e4SLinus Torvalds  * 	equal to the object it is being compared to.
151da177e4SLinus Torvalds  *
161da177e4SLinus Torvalds  * 	The definition of a objects conists of the type (meta type), a
171da177e4SLinus Torvalds  * 	identifier (meta id) and additional type specific information.
181da177e4SLinus Torvalds  * 	The meta id is either TCF_META_TYPE_VALUE for values provided by
191da177e4SLinus Torvalds  * 	userspace or a index to the meta operations table consisting of
201da177e4SLinus Torvalds  * 	function pointers to type specific meta data collectors returning
211da177e4SLinus Torvalds  * 	the value of the requested meta value.
221da177e4SLinus Torvalds  *
231da177e4SLinus Torvalds  * 	         lvalue                                   rvalue
241da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
251da177e4SLinus Torvalds  * 	      | type: INT |                           | type: INT |
26261688d0SDavid S. Miller  * 	 def  | id: DEV   |                           | id: VALUE |
271da177e4SLinus Torvalds  * 	      | data:     |                           | data: 3   |
281da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
291da177e4SLinus Torvalds  * 	            |                                       |
30261688d0SDavid S. Miller  * 	            ---> meta_ops[INT][DEV](...)            |
311da177e4SLinus Torvalds  *	                      |                             |
321da177e4SLinus Torvalds  * 	            -----------                             |
331da177e4SLinus Torvalds  * 	            V                                       V
341da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
351da177e4SLinus Torvalds  * 	      | type: INT |                           | type: INT |
36261688d0SDavid S. Miller  * 	 obj  | id: DEV |                             | id: VALUE |
371da177e4SLinus Torvalds  * 	      | data: 2   |<--data got filled out     | data: 3   |
381da177e4SLinus Torvalds  * 	      +-----------+                           +-----------+
391da177e4SLinus Torvalds  * 	            |                                         |
401da177e4SLinus Torvalds  * 	            --------------> 2  equals 3 <--------------
411da177e4SLinus Torvalds  *
421da177e4SLinus Torvalds  * 	This is a simplified schema, the complexity varies depending
431da177e4SLinus Torvalds  * 	on the meta type. Obviously, the length of the data must also
441da177e4SLinus Torvalds  * 	be provided for non-numeric types.
451da177e4SLinus Torvalds  *
4625985edcSLucas De Marchi  * 	Additionally, type dependent modifiers such as shift operators
471da177e4SLinus Torvalds  * 	or mask may be applied to extend the functionaliy. As of now,
481da177e4SLinus Torvalds  * 	the variable length type supports shifting the byte string to
491da177e4SLinus Torvalds  * 	the right, eating up any number of octets and thus supporting
501da177e4SLinus Torvalds  * 	wildcard interface name comparisons such as "ppp%" matching
511da177e4SLinus Torvalds  * 	ppp0..9.
521da177e4SLinus Torvalds  *
531da177e4SLinus Torvalds  * 	NOTE: Certain meta values depend on other subsystems and are
543a4fa0a2SRobert P. J. Day  * 	      only available if that subsystem is enabled in the kernel.
551da177e4SLinus Torvalds  */
561da177e4SLinus Torvalds 
575a0e3ad6STejun Heo #include <linux/slab.h>
581da177e4SLinus Torvalds #include <linux/module.h>
591da177e4SLinus Torvalds #include <linux/types.h>
601da177e4SLinus Torvalds #include <linux/kernel.h>
611da177e4SLinus Torvalds #include <linux/sched.h>
624f17722cSIngo Molnar #include <linux/sched/loadavg.h>
631da177e4SLinus Torvalds #include <linux/string.h>
641da177e4SLinus Torvalds #include <linux/skbuff.h>
651da177e4SLinus Torvalds #include <linux/random.h>
663113e88cSStephen Hemminger #include <linux/if_vlan.h>
671da177e4SLinus Torvalds #include <linux/tc_ematch/tc_em_meta.h>
681da177e4SLinus Torvalds #include <net/dst.h>
691da177e4SLinus Torvalds #include <net/route.h>
701da177e4SLinus Torvalds #include <net/pkt_cls.h>
7148900629SThomas Graf #include <net/sock.h>
721da177e4SLinus Torvalds 
73cc7ec456SEric Dumazet struct meta_obj {
741da177e4SLinus Torvalds 	unsigned long		value;
751da177e4SLinus Torvalds 	unsigned int		len;
761da177e4SLinus Torvalds };
771da177e4SLinus Torvalds 
78cc7ec456SEric Dumazet struct meta_value {
791da177e4SLinus Torvalds 	struct tcf_meta_val	hdr;
801da177e4SLinus Torvalds 	unsigned long		val;
811da177e4SLinus Torvalds 	unsigned int		len;
821da177e4SLinus Torvalds };
831da177e4SLinus Torvalds 
84cc7ec456SEric Dumazet struct meta_match {
851da177e4SLinus Torvalds 	struct meta_value	lvalue;
861da177e4SLinus Torvalds 	struct meta_value	rvalue;
871da177e4SLinus Torvalds };
881da177e4SLinus Torvalds 
891da177e4SLinus Torvalds static inline int meta_id(struct meta_value *v)
901da177e4SLinus Torvalds {
911da177e4SLinus Torvalds 	return TCF_META_ID(v->hdr.kind);
921da177e4SLinus Torvalds }
931da177e4SLinus Torvalds 
941da177e4SLinus Torvalds static inline int meta_type(struct meta_value *v)
951da177e4SLinus Torvalds {
961da177e4SLinus Torvalds 	return TCF_META_TYPE(v->hdr.kind);
971da177e4SLinus Torvalds }
981da177e4SLinus Torvalds 
991da177e4SLinus Torvalds #define META_COLLECTOR(FUNC) static void meta_##FUNC(struct sk_buff *skb, \
1001da177e4SLinus Torvalds 	struct tcf_pkt_info *info, struct meta_value *v, \
1011da177e4SLinus Torvalds 	struct meta_obj *dst, int *err)
1021da177e4SLinus Torvalds 
1031da177e4SLinus Torvalds /**************************************************************************
1041da177e4SLinus Torvalds  * System status & misc
1051da177e4SLinus Torvalds  **************************************************************************/
1061da177e4SLinus Torvalds 
1071da177e4SLinus Torvalds META_COLLECTOR(int_random)
1081da177e4SLinus Torvalds {
1091da177e4SLinus Torvalds 	get_random_bytes(&dst->value, sizeof(dst->value));
1101da177e4SLinus Torvalds }
1111da177e4SLinus Torvalds 
1121da177e4SLinus Torvalds static inline unsigned long fixed_loadavg(int load)
1131da177e4SLinus Torvalds {
1141da177e4SLinus Torvalds 	int rnd_load = load + (FIXED_1/200);
1151da177e4SLinus Torvalds 	int rnd_frac = ((rnd_load & (FIXED_1-1)) * 100) >> FSHIFT;
1161da177e4SLinus Torvalds 
1171da177e4SLinus Torvalds 	return ((rnd_load >> FSHIFT) * 100) + rnd_frac;
1181da177e4SLinus Torvalds }
1191da177e4SLinus Torvalds 
1201da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_0)
1211da177e4SLinus Torvalds {
1221da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[0]);
1231da177e4SLinus Torvalds }
1241da177e4SLinus Torvalds 
1251da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_1)
1261da177e4SLinus Torvalds {
1271da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[1]);
1281da177e4SLinus Torvalds }
1291da177e4SLinus Torvalds 
1301da177e4SLinus Torvalds META_COLLECTOR(int_loadavg_2)
1311da177e4SLinus Torvalds {
1321da177e4SLinus Torvalds 	dst->value = fixed_loadavg(avenrun[2]);
1331da177e4SLinus Torvalds }
1341da177e4SLinus Torvalds 
1351da177e4SLinus Torvalds /**************************************************************************
1361da177e4SLinus Torvalds  * Device names & indices
1371da177e4SLinus Torvalds  **************************************************************************/
1381da177e4SLinus Torvalds 
1391da177e4SLinus Torvalds static inline int int_dev(struct net_device *dev, struct meta_obj *dst)
1401da177e4SLinus Torvalds {
1411da177e4SLinus Torvalds 	if (unlikely(dev == NULL))
1421da177e4SLinus Torvalds 		return -1;
1431da177e4SLinus Torvalds 
1441da177e4SLinus Torvalds 	dst->value = dev->ifindex;
1451da177e4SLinus Torvalds 	return 0;
1461da177e4SLinus Torvalds }
1471da177e4SLinus Torvalds 
1481da177e4SLinus Torvalds static inline int var_dev(struct net_device *dev, struct meta_obj *dst)
1491da177e4SLinus Torvalds {
1501da177e4SLinus Torvalds 	if (unlikely(dev == NULL))
1511da177e4SLinus Torvalds 		return -1;
1521da177e4SLinus Torvalds 
1531da177e4SLinus Torvalds 	dst->value = (unsigned long) dev->name;
1541da177e4SLinus Torvalds 	dst->len = strlen(dev->name);
1551da177e4SLinus Torvalds 	return 0;
1561da177e4SLinus Torvalds }
1571da177e4SLinus Torvalds 
1581da177e4SLinus Torvalds META_COLLECTOR(int_dev)
1591da177e4SLinus Torvalds {
1601da177e4SLinus Torvalds 	*err = int_dev(skb->dev, dst);
1611da177e4SLinus Torvalds }
1621da177e4SLinus Torvalds 
1631da177e4SLinus Torvalds META_COLLECTOR(var_dev)
1641da177e4SLinus Torvalds {
1651da177e4SLinus Torvalds 	*err = var_dev(skb->dev, dst);
1661da177e4SLinus Torvalds }
1671da177e4SLinus Torvalds 
1681da177e4SLinus Torvalds /**************************************************************************
1693113e88cSStephen Hemminger  * vlan tag
1703113e88cSStephen Hemminger  **************************************************************************/
1713113e88cSStephen Hemminger 
1723113e88cSStephen Hemminger META_COLLECTOR(int_vlan_tag)
1733113e88cSStephen Hemminger {
1741a31f204SStephen Hemminger 	unsigned short tag;
1751a31f204SStephen Hemminger 
176d65f2fa6SShmulik Ladkani 	if (skb_vlan_tag_present(skb))
177d65f2fa6SShmulik Ladkani 		dst->value = skb_vlan_tag_get(skb);
178d65f2fa6SShmulik Ladkani 	else if (!__vlan_get_tag(skb, &tag))
1793113e88cSStephen Hemminger 		dst->value = tag;
180d65f2fa6SShmulik Ladkani 	else
181d65f2fa6SShmulik Ladkani 		*err = -1;
1823113e88cSStephen Hemminger }
1833113e88cSStephen Hemminger 
1843113e88cSStephen Hemminger 
1853113e88cSStephen Hemminger 
1863113e88cSStephen Hemminger /**************************************************************************
1871da177e4SLinus Torvalds  * skb attributes
1881da177e4SLinus Torvalds  **************************************************************************/
1891da177e4SLinus Torvalds 
1901da177e4SLinus Torvalds META_COLLECTOR(int_priority)
1911da177e4SLinus Torvalds {
1921da177e4SLinus Torvalds 	dst->value = skb->priority;
1931da177e4SLinus Torvalds }
1941da177e4SLinus Torvalds 
1951da177e4SLinus Torvalds META_COLLECTOR(int_protocol)
1961da177e4SLinus Torvalds {
1971da177e4SLinus Torvalds 	/* Let userspace take care of the byte ordering */
198d8b9605dSJiri Pirko 	dst->value = tc_skb_protocol(skb);
1991da177e4SLinus Torvalds }
2001da177e4SLinus Torvalds 
2011da177e4SLinus Torvalds META_COLLECTOR(int_pkttype)
2021da177e4SLinus Torvalds {
2031da177e4SLinus Torvalds 	dst->value = skb->pkt_type;
2041da177e4SLinus Torvalds }
2051da177e4SLinus Torvalds 
2061da177e4SLinus Torvalds META_COLLECTOR(int_pktlen)
2071da177e4SLinus Torvalds {
2081da177e4SLinus Torvalds 	dst->value = skb->len;
2091da177e4SLinus Torvalds }
2101da177e4SLinus Torvalds 
2111da177e4SLinus Torvalds META_COLLECTOR(int_datalen)
2121da177e4SLinus Torvalds {
2131da177e4SLinus Torvalds 	dst->value = skb->data_len;
2141da177e4SLinus Torvalds }
2151da177e4SLinus Torvalds 
2161da177e4SLinus Torvalds META_COLLECTOR(int_maclen)
2171da177e4SLinus Torvalds {
2181da177e4SLinus Torvalds 	dst->value = skb->mac_len;
2191da177e4SLinus Torvalds }
2201da177e4SLinus Torvalds 
221c2e3143eSStephen Hemminger META_COLLECTOR(int_rxhash)
222c2e3143eSStephen Hemminger {
2233958afa1STom Herbert 	dst->value = skb_get_hash(skb);
224c2e3143eSStephen Hemminger }
225c2e3143eSStephen Hemminger 
2261da177e4SLinus Torvalds /**************************************************************************
2271da177e4SLinus Torvalds  * Netfilter
2281da177e4SLinus Torvalds  **************************************************************************/
2291da177e4SLinus Torvalds 
23082e91ffeSThomas Graf META_COLLECTOR(int_mark)
2311da177e4SLinus Torvalds {
23282e91ffeSThomas Graf 	dst->value = skb->mark;
2337686ee1aSPatrick McHardy }
2341da177e4SLinus Torvalds 
2351da177e4SLinus Torvalds /**************************************************************************
2361da177e4SLinus Torvalds  * Traffic Control
2371da177e4SLinus Torvalds  **************************************************************************/
2381da177e4SLinus Torvalds 
2391da177e4SLinus Torvalds META_COLLECTOR(int_tcindex)
2401da177e4SLinus Torvalds {
2411da177e4SLinus Torvalds 	dst->value = skb->tc_index;
2421da177e4SLinus Torvalds }
2431da177e4SLinus Torvalds 
2441da177e4SLinus Torvalds /**************************************************************************
2451da177e4SLinus Torvalds  * Routing
2461da177e4SLinus Torvalds  **************************************************************************/
2471da177e4SLinus Torvalds 
2481da177e4SLinus Torvalds META_COLLECTOR(int_rtclassid)
2491da177e4SLinus Torvalds {
250adf30907SEric Dumazet 	if (unlikely(skb_dst(skb) == NULL))
2511da177e4SLinus Torvalds 		*err = -1;
2521da177e4SLinus Torvalds 	else
253c7066f70SPatrick McHardy #ifdef CONFIG_IP_ROUTE_CLASSID
254adf30907SEric Dumazet 		dst->value = skb_dst(skb)->tclassid;
2557686ee1aSPatrick McHardy #else
2567686ee1aSPatrick McHardy 		dst->value = 0;
2571da177e4SLinus Torvalds #endif
2587686ee1aSPatrick McHardy }
2591da177e4SLinus Torvalds 
2601da177e4SLinus Torvalds META_COLLECTOR(int_rtiif)
2611da177e4SLinus Torvalds {
262511c3f92SEric Dumazet 	if (unlikely(skb_rtable(skb) == NULL))
2631da177e4SLinus Torvalds 		*err = -1;
2641da177e4SLinus Torvalds 	else
26592101b3bSDavid S. Miller 		dst->value = inet_iif(skb);
2661da177e4SLinus Torvalds }
2671da177e4SLinus Torvalds 
2681da177e4SLinus Torvalds /**************************************************************************
26948900629SThomas Graf  * Socket Attributes
27048900629SThomas Graf  **************************************************************************/
27148900629SThomas Graf 
2724f8f61ebSYang Yingliang #define skip_nonlocal(skb) \
2734f8f61ebSYang Yingliang 	(unlikely(skb->sk == NULL))
27448900629SThomas Graf 
27548900629SThomas Graf META_COLLECTOR(int_sk_family)
27648900629SThomas Graf {
2774f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2784f8f61ebSYang Yingliang 		*err = -1;
2794f8f61ebSYang Yingliang 		return;
2804f8f61ebSYang Yingliang 	}
28148900629SThomas Graf 	dst->value = skb->sk->sk_family;
28248900629SThomas Graf }
28348900629SThomas Graf 
28448900629SThomas Graf META_COLLECTOR(int_sk_state)
28548900629SThomas Graf {
2864f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2874f8f61ebSYang Yingliang 		*err = -1;
2884f8f61ebSYang Yingliang 		return;
2894f8f61ebSYang Yingliang 	}
29048900629SThomas Graf 	dst->value = skb->sk->sk_state;
29148900629SThomas Graf }
29248900629SThomas Graf 
29348900629SThomas Graf META_COLLECTOR(int_sk_reuse)
29448900629SThomas Graf {
2954f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
2964f8f61ebSYang Yingliang 		*err = -1;
2974f8f61ebSYang Yingliang 		return;
2984f8f61ebSYang Yingliang 	}
29948900629SThomas Graf 	dst->value = skb->sk->sk_reuse;
30048900629SThomas Graf }
30148900629SThomas Graf 
30248900629SThomas Graf META_COLLECTOR(int_sk_bound_if)
30348900629SThomas Graf {
3044f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3054f8f61ebSYang Yingliang 		*err = -1;
3064f8f61ebSYang Yingliang 		return;
3074f8f61ebSYang Yingliang 	}
30848900629SThomas Graf 	/* No error if bound_dev_if is 0, legal userspace check */
30948900629SThomas Graf 	dst->value = skb->sk->sk_bound_dev_if;
31048900629SThomas Graf }
31148900629SThomas Graf 
31248900629SThomas Graf META_COLLECTOR(var_sk_bound_if)
31348900629SThomas Graf {
3144f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3154f8f61ebSYang Yingliang 		*err = -1;
3164f8f61ebSYang Yingliang 		return;
3174f8f61ebSYang Yingliang 	}
31848900629SThomas Graf 
31948900629SThomas Graf 	if (skb->sk->sk_bound_dev_if == 0) {
32048900629SThomas Graf 		dst->value = (unsigned long) "any";
32148900629SThomas Graf 		dst->len = 3;
32248900629SThomas Graf 	} else {
32348900629SThomas Graf 		struct net_device *dev;
32448900629SThomas Graf 
325d0075634SEric Dumazet 		rcu_read_lock();
3262939e275SEric Dumazet 		dev = dev_get_by_index_rcu(sock_net(skb->sk),
3272939e275SEric Dumazet 					   skb->sk->sk_bound_dev_if);
32848900629SThomas Graf 		*err = var_dev(dev, dst);
329d0075634SEric Dumazet 		rcu_read_unlock();
33048900629SThomas Graf 	}
33148900629SThomas Graf }
33248900629SThomas Graf 
33348900629SThomas Graf META_COLLECTOR(int_sk_refcnt)
33448900629SThomas Graf {
3354f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
3364f8f61ebSYang Yingliang 		*err = -1;
3374f8f61ebSYang Yingliang 		return;
3384f8f61ebSYang Yingliang 	}
33941c6d650SReshetova, Elena 	dst->value = refcount_read(&skb->sk->sk_refcnt);
34048900629SThomas Graf }
34148900629SThomas Graf 
34248900629SThomas Graf META_COLLECTOR(int_sk_rcvbuf)
34348900629SThomas Graf {
34402a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
34502a56c81SEric Dumazet 
34602a56c81SEric Dumazet 	if (!sk) {
3474f8f61ebSYang Yingliang 		*err = -1;
3484f8f61ebSYang Yingliang 		return;
3494f8f61ebSYang Yingliang 	}
35002a56c81SEric Dumazet 	dst->value = sk->sk_rcvbuf;
35148900629SThomas Graf }
35248900629SThomas Graf 
35348900629SThomas Graf META_COLLECTOR(int_sk_shutdown)
35448900629SThomas Graf {
35502a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
35602a56c81SEric Dumazet 
35702a56c81SEric Dumazet 	if (!sk) {
3584f8f61ebSYang Yingliang 		*err = -1;
3594f8f61ebSYang Yingliang 		return;
3604f8f61ebSYang Yingliang 	}
36102a56c81SEric Dumazet 	dst->value = sk->sk_shutdown;
36248900629SThomas Graf }
36348900629SThomas Graf 
36448900629SThomas Graf META_COLLECTOR(int_sk_proto)
36548900629SThomas Graf {
36602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
36702a56c81SEric Dumazet 
36802a56c81SEric Dumazet 	if (!sk) {
3694f8f61ebSYang Yingliang 		*err = -1;
3704f8f61ebSYang Yingliang 		return;
3714f8f61ebSYang Yingliang 	}
37202a56c81SEric Dumazet 	dst->value = sk->sk_protocol;
37348900629SThomas Graf }
37448900629SThomas Graf 
37548900629SThomas Graf META_COLLECTOR(int_sk_type)
37648900629SThomas Graf {
37702a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
37802a56c81SEric Dumazet 
37902a56c81SEric Dumazet 	if (!sk) {
3804f8f61ebSYang Yingliang 		*err = -1;
3814f8f61ebSYang Yingliang 		return;
3824f8f61ebSYang Yingliang 	}
38302a56c81SEric Dumazet 	dst->value = sk->sk_type;
38448900629SThomas Graf }
38548900629SThomas Graf 
38648900629SThomas Graf META_COLLECTOR(int_sk_rmem_alloc)
38748900629SThomas Graf {
38802a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
38902a56c81SEric Dumazet 
39002a56c81SEric Dumazet 	if (!sk) {
3914f8f61ebSYang Yingliang 		*err = -1;
3924f8f61ebSYang Yingliang 		return;
3934f8f61ebSYang Yingliang 	}
39402a56c81SEric Dumazet 	dst->value = sk_rmem_alloc_get(sk);
39548900629SThomas Graf }
39648900629SThomas Graf 
39748900629SThomas Graf META_COLLECTOR(int_sk_wmem_alloc)
39848900629SThomas Graf {
39902a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
40002a56c81SEric Dumazet 
40102a56c81SEric Dumazet 	if (!sk) {
4024f8f61ebSYang Yingliang 		*err = -1;
4034f8f61ebSYang Yingliang 		return;
4044f8f61ebSYang Yingliang 	}
40502a56c81SEric Dumazet 	dst->value = sk_wmem_alloc_get(sk);
40648900629SThomas Graf }
40748900629SThomas Graf 
40848900629SThomas Graf META_COLLECTOR(int_sk_omem_alloc)
40948900629SThomas Graf {
41002a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
41102a56c81SEric Dumazet 
41202a56c81SEric Dumazet 	if (!sk) {
4134f8f61ebSYang Yingliang 		*err = -1;
4144f8f61ebSYang Yingliang 		return;
4154f8f61ebSYang Yingliang 	}
41602a56c81SEric Dumazet 	dst->value = atomic_read(&sk->sk_omem_alloc);
41748900629SThomas Graf }
41848900629SThomas Graf 
41948900629SThomas Graf META_COLLECTOR(int_sk_rcv_qlen)
42048900629SThomas Graf {
42102a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
42202a56c81SEric Dumazet 
42302a56c81SEric Dumazet 	if (!sk) {
4244f8f61ebSYang Yingliang 		*err = -1;
4254f8f61ebSYang Yingliang 		return;
4264f8f61ebSYang Yingliang 	}
42702a56c81SEric Dumazet 	dst->value = sk->sk_receive_queue.qlen;
42848900629SThomas Graf }
42948900629SThomas Graf 
43048900629SThomas Graf META_COLLECTOR(int_sk_snd_qlen)
43148900629SThomas Graf {
43202a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
43302a56c81SEric Dumazet 
43402a56c81SEric Dumazet 	if (!sk) {
4354f8f61ebSYang Yingliang 		*err = -1;
4364f8f61ebSYang Yingliang 		return;
4374f8f61ebSYang Yingliang 	}
43802a56c81SEric Dumazet 	dst->value = sk->sk_write_queue.qlen;
43948900629SThomas Graf }
44048900629SThomas Graf 
44148900629SThomas Graf META_COLLECTOR(int_sk_wmem_queued)
44248900629SThomas Graf {
44302a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
44402a56c81SEric Dumazet 
44502a56c81SEric Dumazet 	if (!sk) {
4464f8f61ebSYang Yingliang 		*err = -1;
4474f8f61ebSYang Yingliang 		return;
4484f8f61ebSYang Yingliang 	}
44902a56c81SEric Dumazet 	dst->value = sk->sk_wmem_queued;
45048900629SThomas Graf }
45148900629SThomas Graf 
45248900629SThomas Graf META_COLLECTOR(int_sk_fwd_alloc)
45348900629SThomas Graf {
45402a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
45502a56c81SEric Dumazet 
45602a56c81SEric Dumazet 	if (!sk) {
4574f8f61ebSYang Yingliang 		*err = -1;
4584f8f61ebSYang Yingliang 		return;
4594f8f61ebSYang Yingliang 	}
46002a56c81SEric Dumazet 	dst->value = sk->sk_forward_alloc;
46148900629SThomas Graf }
46248900629SThomas Graf 
46348900629SThomas Graf META_COLLECTOR(int_sk_sndbuf)
46448900629SThomas Graf {
46502a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
46602a56c81SEric Dumazet 
46702a56c81SEric Dumazet 	if (!sk) {
4684f8f61ebSYang Yingliang 		*err = -1;
4694f8f61ebSYang Yingliang 		return;
4704f8f61ebSYang Yingliang 	}
47102a56c81SEric Dumazet 	dst->value = sk->sk_sndbuf;
47248900629SThomas Graf }
47348900629SThomas Graf 
47448900629SThomas Graf META_COLLECTOR(int_sk_alloc)
47548900629SThomas Graf {
47602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
47702a56c81SEric Dumazet 
47802a56c81SEric Dumazet 	if (!sk) {
4794f8f61ebSYang Yingliang 		*err = -1;
4804f8f61ebSYang Yingliang 		return;
4814f8f61ebSYang Yingliang 	}
48202a56c81SEric Dumazet 	dst->value = (__force int) sk->sk_allocation;
48348900629SThomas Graf }
48448900629SThomas Graf 
48581c3d547SEric Dumazet META_COLLECTOR(int_sk_hash)
48648900629SThomas Graf {
4874f8f61ebSYang Yingliang 	if (skip_nonlocal(skb)) {
4884f8f61ebSYang Yingliang 		*err = -1;
4894f8f61ebSYang Yingliang 		return;
4904f8f61ebSYang Yingliang 	}
49181c3d547SEric Dumazet 	dst->value = skb->sk->sk_hash;
49248900629SThomas Graf }
49348900629SThomas Graf 
49448900629SThomas Graf META_COLLECTOR(int_sk_lingertime)
49548900629SThomas Graf {
49602a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
49702a56c81SEric Dumazet 
49802a56c81SEric Dumazet 	if (!sk) {
4994f8f61ebSYang Yingliang 		*err = -1;
5004f8f61ebSYang Yingliang 		return;
5014f8f61ebSYang Yingliang 	}
50202a56c81SEric Dumazet 	dst->value = sk->sk_lingertime / HZ;
50348900629SThomas Graf }
50448900629SThomas Graf 
50548900629SThomas Graf META_COLLECTOR(int_sk_err_qlen)
50648900629SThomas Graf {
50702a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
50802a56c81SEric Dumazet 
50902a56c81SEric Dumazet 	if (!sk) {
5104f8f61ebSYang Yingliang 		*err = -1;
5114f8f61ebSYang Yingliang 		return;
5124f8f61ebSYang Yingliang 	}
51302a56c81SEric Dumazet 	dst->value = sk->sk_error_queue.qlen;
51448900629SThomas Graf }
51548900629SThomas Graf 
51648900629SThomas Graf META_COLLECTOR(int_sk_ack_bl)
51748900629SThomas Graf {
51802a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
51902a56c81SEric Dumazet 
52002a56c81SEric Dumazet 	if (!sk) {
5214f8f61ebSYang Yingliang 		*err = -1;
5224f8f61ebSYang Yingliang 		return;
5234f8f61ebSYang Yingliang 	}
52402a56c81SEric Dumazet 	dst->value = sk->sk_ack_backlog;
52548900629SThomas Graf }
52648900629SThomas Graf 
52748900629SThomas Graf META_COLLECTOR(int_sk_max_ack_bl)
52848900629SThomas Graf {
52902a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
53002a56c81SEric Dumazet 
53102a56c81SEric Dumazet 	if (!sk) {
5324f8f61ebSYang Yingliang 		*err = -1;
5334f8f61ebSYang Yingliang 		return;
5344f8f61ebSYang Yingliang 	}
53502a56c81SEric Dumazet 	dst->value = sk->sk_max_ack_backlog;
53648900629SThomas Graf }
53748900629SThomas Graf 
53848900629SThomas Graf META_COLLECTOR(int_sk_prio)
53948900629SThomas Graf {
54002a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
54102a56c81SEric Dumazet 
54202a56c81SEric Dumazet 	if (!sk) {
5434f8f61ebSYang Yingliang 		*err = -1;
5444f8f61ebSYang Yingliang 		return;
5454f8f61ebSYang Yingliang 	}
54602a56c81SEric Dumazet 	dst->value = sk->sk_priority;
54748900629SThomas Graf }
54848900629SThomas Graf 
54948900629SThomas Graf META_COLLECTOR(int_sk_rcvlowat)
55048900629SThomas Graf {
55102a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
55202a56c81SEric Dumazet 
55302a56c81SEric Dumazet 	if (!sk) {
5544f8f61ebSYang Yingliang 		*err = -1;
5554f8f61ebSYang Yingliang 		return;
5564f8f61ebSYang Yingliang 	}
55702a56c81SEric Dumazet 	dst->value = sk->sk_rcvlowat;
55848900629SThomas Graf }
55948900629SThomas Graf 
56048900629SThomas Graf META_COLLECTOR(int_sk_rcvtimeo)
56148900629SThomas Graf {
56202a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
56302a56c81SEric Dumazet 
56402a56c81SEric Dumazet 	if (!sk) {
5654f8f61ebSYang Yingliang 		*err = -1;
5664f8f61ebSYang Yingliang 		return;
5674f8f61ebSYang Yingliang 	}
56802a56c81SEric Dumazet 	dst->value = sk->sk_rcvtimeo / HZ;
56948900629SThomas Graf }
57048900629SThomas Graf 
57148900629SThomas Graf META_COLLECTOR(int_sk_sndtimeo)
57248900629SThomas Graf {
57302a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
57402a56c81SEric Dumazet 
57502a56c81SEric Dumazet 	if (!sk) {
5764f8f61ebSYang Yingliang 		*err = -1;
5774f8f61ebSYang Yingliang 		return;
5784f8f61ebSYang Yingliang 	}
57902a56c81SEric Dumazet 	dst->value = sk->sk_sndtimeo / HZ;
58048900629SThomas Graf }
58148900629SThomas Graf 
58248900629SThomas Graf META_COLLECTOR(int_sk_sendmsg_off)
58348900629SThomas Graf {
58402a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
58502a56c81SEric Dumazet 
58602a56c81SEric Dumazet 	if (!sk) {
5874f8f61ebSYang Yingliang 		*err = -1;
5884f8f61ebSYang Yingliang 		return;
5894f8f61ebSYang Yingliang 	}
59002a56c81SEric Dumazet 	dst->value = sk->sk_frag.offset;
59148900629SThomas Graf }
59248900629SThomas Graf 
59348900629SThomas Graf META_COLLECTOR(int_sk_write_pend)
59448900629SThomas Graf {
59502a56c81SEric Dumazet 	const struct sock *sk = skb_to_full_sk(skb);
59602a56c81SEric Dumazet 
59702a56c81SEric Dumazet 	if (!sk) {
5984f8f61ebSYang Yingliang 		*err = -1;
5994f8f61ebSYang Yingliang 		return;
6004f8f61ebSYang Yingliang 	}
60102a56c81SEric Dumazet 	dst->value = sk->sk_write_pending;
60248900629SThomas Graf }
60348900629SThomas Graf 
60448900629SThomas Graf /**************************************************************************
6051da177e4SLinus Torvalds  * Meta value collectors assignment table
6061da177e4SLinus Torvalds  **************************************************************************/
6071da177e4SLinus Torvalds 
608cc7ec456SEric Dumazet struct meta_ops {
6091da177e4SLinus Torvalds 	void		(*get)(struct sk_buff *, struct tcf_pkt_info *,
6101da177e4SLinus Torvalds 			       struct meta_value *, struct meta_obj *, int *);
6111da177e4SLinus Torvalds };
6121da177e4SLinus Torvalds 
61348900629SThomas Graf #define META_ID(name) TCF_META_ID_##name
61448900629SThomas Graf #define META_FUNC(name) { .get = meta_##name }
61548900629SThomas Graf 
6161da177e4SLinus Torvalds /* Meta value operations table listing all meta value collectors and
6171da177e4SLinus Torvalds  * assigns them to a type and meta id. */
6181da177e4SLinus Torvalds static struct meta_ops __meta_ops[TCF_META_TYPE_MAX + 1][TCF_META_ID_MAX + 1] = {
6191da177e4SLinus Torvalds 	[TCF_META_TYPE_VAR] = {
62048900629SThomas Graf 		[META_ID(DEV)]			= META_FUNC(var_dev),
62148900629SThomas Graf 		[META_ID(SK_BOUND_IF)] 		= META_FUNC(var_sk_bound_if),
6221da177e4SLinus Torvalds 	},
6231da177e4SLinus Torvalds 	[TCF_META_TYPE_INT] = {
62448900629SThomas Graf 		[META_ID(RANDOM)]		= META_FUNC(int_random),
62548900629SThomas Graf 		[META_ID(LOADAVG_0)]		= META_FUNC(int_loadavg_0),
62648900629SThomas Graf 		[META_ID(LOADAVG_1)]		= META_FUNC(int_loadavg_1),
62748900629SThomas Graf 		[META_ID(LOADAVG_2)]		= META_FUNC(int_loadavg_2),
62848900629SThomas Graf 		[META_ID(DEV)]			= META_FUNC(int_dev),
62948900629SThomas Graf 		[META_ID(PRIORITY)]		= META_FUNC(int_priority),
63048900629SThomas Graf 		[META_ID(PROTOCOL)]		= META_FUNC(int_protocol),
63148900629SThomas Graf 		[META_ID(PKTTYPE)]		= META_FUNC(int_pkttype),
63248900629SThomas Graf 		[META_ID(PKTLEN)]		= META_FUNC(int_pktlen),
63348900629SThomas Graf 		[META_ID(DATALEN)]		= META_FUNC(int_datalen),
63448900629SThomas Graf 		[META_ID(MACLEN)]		= META_FUNC(int_maclen),
63582e91ffeSThomas Graf 		[META_ID(NFMARK)]		= META_FUNC(int_mark),
63648900629SThomas Graf 		[META_ID(TCINDEX)]		= META_FUNC(int_tcindex),
63748900629SThomas Graf 		[META_ID(RTCLASSID)]		= META_FUNC(int_rtclassid),
63848900629SThomas Graf 		[META_ID(RTIIF)]		= META_FUNC(int_rtiif),
63948900629SThomas Graf 		[META_ID(SK_FAMILY)]		= META_FUNC(int_sk_family),
64048900629SThomas Graf 		[META_ID(SK_STATE)]		= META_FUNC(int_sk_state),
64148900629SThomas Graf 		[META_ID(SK_REUSE)]		= META_FUNC(int_sk_reuse),
64248900629SThomas Graf 		[META_ID(SK_BOUND_IF)]		= META_FUNC(int_sk_bound_if),
64348900629SThomas Graf 		[META_ID(SK_REFCNT)]		= META_FUNC(int_sk_refcnt),
64448900629SThomas Graf 		[META_ID(SK_RCVBUF)]		= META_FUNC(int_sk_rcvbuf),
64548900629SThomas Graf 		[META_ID(SK_SNDBUF)]		= META_FUNC(int_sk_sndbuf),
64648900629SThomas Graf 		[META_ID(SK_SHUTDOWN)]		= META_FUNC(int_sk_shutdown),
64748900629SThomas Graf 		[META_ID(SK_PROTO)]		= META_FUNC(int_sk_proto),
64848900629SThomas Graf 		[META_ID(SK_TYPE)]		= META_FUNC(int_sk_type),
64948900629SThomas Graf 		[META_ID(SK_RMEM_ALLOC)]	= META_FUNC(int_sk_rmem_alloc),
65048900629SThomas Graf 		[META_ID(SK_WMEM_ALLOC)]	= META_FUNC(int_sk_wmem_alloc),
65148900629SThomas Graf 		[META_ID(SK_OMEM_ALLOC)]	= META_FUNC(int_sk_omem_alloc),
65248900629SThomas Graf 		[META_ID(SK_WMEM_QUEUED)]	= META_FUNC(int_sk_wmem_queued),
65348900629SThomas Graf 		[META_ID(SK_RCV_QLEN)]		= META_FUNC(int_sk_rcv_qlen),
65448900629SThomas Graf 		[META_ID(SK_SND_QLEN)]		= META_FUNC(int_sk_snd_qlen),
65548900629SThomas Graf 		[META_ID(SK_ERR_QLEN)]		= META_FUNC(int_sk_err_qlen),
65648900629SThomas Graf 		[META_ID(SK_FORWARD_ALLOCS)]	= META_FUNC(int_sk_fwd_alloc),
65748900629SThomas Graf 		[META_ID(SK_ALLOCS)]		= META_FUNC(int_sk_alloc),
65881c3d547SEric Dumazet 		[META_ID(SK_HASH)]		= META_FUNC(int_sk_hash),
65948900629SThomas Graf 		[META_ID(SK_LINGERTIME)]	= META_FUNC(int_sk_lingertime),
66048900629SThomas Graf 		[META_ID(SK_ACK_BACKLOG)]	= META_FUNC(int_sk_ack_bl),
66148900629SThomas Graf 		[META_ID(SK_MAX_ACK_BACKLOG)]	= META_FUNC(int_sk_max_ack_bl),
66248900629SThomas Graf 		[META_ID(SK_PRIO)]		= META_FUNC(int_sk_prio),
66348900629SThomas Graf 		[META_ID(SK_RCVLOWAT)]		= META_FUNC(int_sk_rcvlowat),
66448900629SThomas Graf 		[META_ID(SK_RCVTIMEO)]		= META_FUNC(int_sk_rcvtimeo),
66548900629SThomas Graf 		[META_ID(SK_SNDTIMEO)]		= META_FUNC(int_sk_sndtimeo),
66648900629SThomas Graf 		[META_ID(SK_SENDMSG_OFF)]	= META_FUNC(int_sk_sendmsg_off),
66748900629SThomas Graf 		[META_ID(SK_WRITE_PENDING)]	= META_FUNC(int_sk_write_pend),
6683113e88cSStephen Hemminger 		[META_ID(VLAN_TAG)]		= META_FUNC(int_vlan_tag),
669c2e3143eSStephen Hemminger 		[META_ID(RXHASH)]		= META_FUNC(int_rxhash),
6701da177e4SLinus Torvalds 	}
6711da177e4SLinus Torvalds };
6721da177e4SLinus Torvalds 
6731da177e4SLinus Torvalds static inline struct meta_ops *meta_ops(struct meta_value *val)
6741da177e4SLinus Torvalds {
6751da177e4SLinus Torvalds 	return &__meta_ops[meta_type(val)][meta_id(val)];
6761da177e4SLinus Torvalds }
6771da177e4SLinus Torvalds 
6781da177e4SLinus Torvalds /**************************************************************************
6791da177e4SLinus Torvalds  * Type specific operations for TCF_META_TYPE_VAR
6801da177e4SLinus Torvalds  **************************************************************************/
6811da177e4SLinus Torvalds 
6821da177e4SLinus Torvalds static int meta_var_compare(struct meta_obj *a, struct meta_obj *b)
6831da177e4SLinus Torvalds {
6841da177e4SLinus Torvalds 	int r = a->len - b->len;
6851da177e4SLinus Torvalds 
6861da177e4SLinus Torvalds 	if (r == 0)
6871da177e4SLinus Torvalds 		r = memcmp((void *) a->value, (void *) b->value, a->len);
6881da177e4SLinus Torvalds 
6891da177e4SLinus Torvalds 	return r;
6901da177e4SLinus Torvalds }
6911da177e4SLinus Torvalds 
692add93b61SPatrick McHardy static int meta_var_change(struct meta_value *dst, struct nlattr *nla)
6931da177e4SLinus Torvalds {
694add93b61SPatrick McHardy 	int len = nla_len(nla);
6951da177e4SLinus Torvalds 
696add93b61SPatrick McHardy 	dst->val = (unsigned long)kmemdup(nla_data(nla), len, GFP_KERNEL);
6971da177e4SLinus Torvalds 	if (dst->val == 0UL)
6981da177e4SLinus Torvalds 		return -ENOMEM;
6991da177e4SLinus Torvalds 	dst->len = len;
7001da177e4SLinus Torvalds 	return 0;
7011da177e4SLinus Torvalds }
7021da177e4SLinus Torvalds 
7031da177e4SLinus Torvalds static void meta_var_destroy(struct meta_value *v)
7041da177e4SLinus Torvalds {
7051da177e4SLinus Torvalds 	kfree((void *) v->val);
7061da177e4SLinus Torvalds }
7071da177e4SLinus Torvalds 
7081da177e4SLinus Torvalds static void meta_var_apply_extras(struct meta_value *v,
7091da177e4SLinus Torvalds 				  struct meta_obj *dst)
7101da177e4SLinus Torvalds {
7111da177e4SLinus Torvalds 	int shift = v->hdr.shift;
7121da177e4SLinus Torvalds 
7131da177e4SLinus Torvalds 	if (shift && shift < dst->len)
7141da177e4SLinus Torvalds 		dst->len -= shift;
7151da177e4SLinus Torvalds }
7161da177e4SLinus Torvalds 
7171da177e4SLinus Torvalds static int meta_var_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
7181da177e4SLinus Torvalds {
7191b34ec43SDavid S. Miller 	if (v->val && v->len &&
7201b34ec43SDavid S. Miller 	    nla_put(skb, tlv, v->len, (void *) v->val))
7211b34ec43SDavid S. Miller 		goto nla_put_failure;
7221da177e4SLinus Torvalds 	return 0;
7231da177e4SLinus Torvalds 
724add93b61SPatrick McHardy nla_put_failure:
7251da177e4SLinus Torvalds 	return -1;
7261da177e4SLinus Torvalds }
7271da177e4SLinus Torvalds 
7281da177e4SLinus Torvalds /**************************************************************************
7291da177e4SLinus Torvalds  * Type specific operations for TCF_META_TYPE_INT
7301da177e4SLinus Torvalds  **************************************************************************/
7311da177e4SLinus Torvalds 
7321da177e4SLinus Torvalds static int meta_int_compare(struct meta_obj *a, struct meta_obj *b)
7331da177e4SLinus Torvalds {
7341da177e4SLinus Torvalds 	/* Let gcc optimize it, the unlikely is not really based on
7351da177e4SLinus Torvalds 	 * some numbers but jump free code for mismatches seems
7361da177e4SLinus Torvalds 	 * more logical. */
73798e56405SThomas Graf 	if (unlikely(a->value == b->value))
7381da177e4SLinus Torvalds 		return 0;
73998e56405SThomas Graf 	else if (a->value < b->value)
7401da177e4SLinus Torvalds 		return -1;
7411da177e4SLinus Torvalds 	else
7421da177e4SLinus Torvalds 		return 1;
7431da177e4SLinus Torvalds }
7441da177e4SLinus Torvalds 
745add93b61SPatrick McHardy static int meta_int_change(struct meta_value *dst, struct nlattr *nla)
7461da177e4SLinus Torvalds {
747add93b61SPatrick McHardy 	if (nla_len(nla) >= sizeof(unsigned long)) {
748add93b61SPatrick McHardy 		dst->val = *(unsigned long *) nla_data(nla);
7491da177e4SLinus Torvalds 		dst->len = sizeof(unsigned long);
750add93b61SPatrick McHardy 	} else if (nla_len(nla) == sizeof(u32)) {
7511587bac4SPatrick McHardy 		dst->val = nla_get_u32(nla);
7521da177e4SLinus Torvalds 		dst->len = sizeof(u32);
7531da177e4SLinus Torvalds 	} else
7541da177e4SLinus Torvalds 		return -EINVAL;
7551da177e4SLinus Torvalds 
7561da177e4SLinus Torvalds 	return 0;
7571da177e4SLinus Torvalds }
7581da177e4SLinus Torvalds 
7591da177e4SLinus Torvalds static void meta_int_apply_extras(struct meta_value *v,
7601da177e4SLinus Torvalds 				  struct meta_obj *dst)
7611da177e4SLinus Torvalds {
7621da177e4SLinus Torvalds 	if (v->hdr.shift)
7631da177e4SLinus Torvalds 		dst->value >>= v->hdr.shift;
7641da177e4SLinus Torvalds 
7651da177e4SLinus Torvalds 	if (v->val)
7661da177e4SLinus Torvalds 		dst->value &= v->val;
7671da177e4SLinus Torvalds }
7681da177e4SLinus Torvalds 
7691da177e4SLinus Torvalds static int meta_int_dump(struct sk_buff *skb, struct meta_value *v, int tlv)
7701da177e4SLinus Torvalds {
7711b34ec43SDavid S. Miller 	if (v->len == sizeof(unsigned long)) {
7721b34ec43SDavid S. Miller 		if (nla_put(skb, tlv, sizeof(unsigned long), &v->val))
7731b34ec43SDavid S. Miller 			goto nla_put_failure;
7741b34ec43SDavid S. Miller 	} else if (v->len == sizeof(u32)) {
7751b34ec43SDavid S. Miller 		if (nla_put_u32(skb, tlv, v->val))
7761b34ec43SDavid S. Miller 			goto nla_put_failure;
7771b34ec43SDavid S. Miller 	}
7781da177e4SLinus Torvalds 
7791da177e4SLinus Torvalds 	return 0;
7801da177e4SLinus Torvalds 
781add93b61SPatrick McHardy nla_put_failure:
7821da177e4SLinus Torvalds 	return -1;
7831da177e4SLinus Torvalds }
7841da177e4SLinus Torvalds 
7851da177e4SLinus Torvalds /**************************************************************************
7861da177e4SLinus Torvalds  * Type specific operations table
7871da177e4SLinus Torvalds  **************************************************************************/
7881da177e4SLinus Torvalds 
789cc7ec456SEric Dumazet struct meta_type_ops {
7901da177e4SLinus Torvalds 	void	(*destroy)(struct meta_value *);
7911da177e4SLinus Torvalds 	int	(*compare)(struct meta_obj *, struct meta_obj *);
792add93b61SPatrick McHardy 	int	(*change)(struct meta_value *, struct nlattr *);
7931da177e4SLinus Torvalds 	void	(*apply_extras)(struct meta_value *, struct meta_obj *);
7941da177e4SLinus Torvalds 	int	(*dump)(struct sk_buff *, struct meta_value *, int);
7951da177e4SLinus Torvalds };
7961da177e4SLinus Torvalds 
797cfe2f14cSJulia Lawall static const struct meta_type_ops __meta_type_ops[TCF_META_TYPE_MAX + 1] = {
7981da177e4SLinus Torvalds 	[TCF_META_TYPE_VAR] = {
7991da177e4SLinus Torvalds 		.destroy = meta_var_destroy,
8001da177e4SLinus Torvalds 		.compare = meta_var_compare,
8011da177e4SLinus Torvalds 		.change = meta_var_change,
8021da177e4SLinus Torvalds 		.apply_extras = meta_var_apply_extras,
8031da177e4SLinus Torvalds 		.dump = meta_var_dump
8041da177e4SLinus Torvalds 	},
8051da177e4SLinus Torvalds 	[TCF_META_TYPE_INT] = {
8061da177e4SLinus Torvalds 		.compare = meta_int_compare,
8071da177e4SLinus Torvalds 		.change = meta_int_change,
8081da177e4SLinus Torvalds 		.apply_extras = meta_int_apply_extras,
8091da177e4SLinus Torvalds 		.dump = meta_int_dump
8101da177e4SLinus Torvalds 	}
8111da177e4SLinus Torvalds };
8121da177e4SLinus Torvalds 
813cfe2f14cSJulia Lawall static inline const struct meta_type_ops *meta_type_ops(struct meta_value *v)
8141da177e4SLinus Torvalds {
8151da177e4SLinus Torvalds 	return &__meta_type_ops[meta_type(v)];
8161da177e4SLinus Torvalds }
8171da177e4SLinus Torvalds 
8181da177e4SLinus Torvalds /**************************************************************************
8191da177e4SLinus Torvalds  * Core
8201da177e4SLinus Torvalds  **************************************************************************/
8211da177e4SLinus Torvalds 
822ed7af3b3SStephen Hemminger static int meta_get(struct sk_buff *skb, struct tcf_pkt_info *info,
8231da177e4SLinus Torvalds 		    struct meta_value *v, struct meta_obj *dst)
8241da177e4SLinus Torvalds {
8251da177e4SLinus Torvalds 	int err = 0;
8261da177e4SLinus Torvalds 
8271da177e4SLinus Torvalds 	if (meta_id(v) == TCF_META_ID_VALUE) {
8281da177e4SLinus Torvalds 		dst->value = v->val;
8291da177e4SLinus Torvalds 		dst->len = v->len;
8301da177e4SLinus Torvalds 		return 0;
8311da177e4SLinus Torvalds 	}
8321da177e4SLinus Torvalds 
8331da177e4SLinus Torvalds 	meta_ops(v)->get(skb, info, v, dst, &err);
8341da177e4SLinus Torvalds 	if (err < 0)
8351da177e4SLinus Torvalds 		return err;
8361da177e4SLinus Torvalds 
8371da177e4SLinus Torvalds 	if (meta_type_ops(v)->apply_extras)
8381da177e4SLinus Torvalds 		meta_type_ops(v)->apply_extras(v, dst);
8391da177e4SLinus Torvalds 
8401da177e4SLinus Torvalds 	return 0;
8411da177e4SLinus Torvalds }
8421da177e4SLinus Torvalds 
8431da177e4SLinus Torvalds static int em_meta_match(struct sk_buff *skb, struct tcf_ematch *m,
8441da177e4SLinus Torvalds 			 struct tcf_pkt_info *info)
8451da177e4SLinus Torvalds {
8461da177e4SLinus Torvalds 	int r;
8471da177e4SLinus Torvalds 	struct meta_match *meta = (struct meta_match *) m->data;
8481da177e4SLinus Torvalds 	struct meta_obj l_value, r_value;
8491da177e4SLinus Torvalds 
8501da177e4SLinus Torvalds 	if (meta_get(skb, info, &meta->lvalue, &l_value) < 0 ||
8511da177e4SLinus Torvalds 	    meta_get(skb, info, &meta->rvalue, &r_value) < 0)
8521da177e4SLinus Torvalds 		return 0;
8531da177e4SLinus Torvalds 
8541da177e4SLinus Torvalds 	r = meta_type_ops(&meta->lvalue)->compare(&l_value, &r_value);
8551da177e4SLinus Torvalds 
8561da177e4SLinus Torvalds 	switch (meta->lvalue.hdr.op) {
8571da177e4SLinus Torvalds 	case TCF_EM_OPND_EQ:
8581da177e4SLinus Torvalds 		return !r;
8591da177e4SLinus Torvalds 	case TCF_EM_OPND_LT:
8601da177e4SLinus Torvalds 		return r < 0;
8611da177e4SLinus Torvalds 	case TCF_EM_OPND_GT:
8621da177e4SLinus Torvalds 		return r > 0;
8631da177e4SLinus Torvalds 	}
8641da177e4SLinus Torvalds 
8651da177e4SLinus Torvalds 	return 0;
8661da177e4SLinus Torvalds }
8671da177e4SLinus Torvalds 
868ed7af3b3SStephen Hemminger static void meta_delete(struct meta_match *meta)
8691da177e4SLinus Torvalds {
87004f217acSStephen Hemminger 	if (meta) {
871cfe2f14cSJulia Lawall 		const struct meta_type_ops *ops = meta_type_ops(&meta->lvalue);
8721da177e4SLinus Torvalds 
8731da177e4SLinus Torvalds 		if (ops && ops->destroy) {
8741da177e4SLinus Torvalds 			ops->destroy(&meta->lvalue);
8751da177e4SLinus Torvalds 			ops->destroy(&meta->rvalue);
8761da177e4SLinus Torvalds 		}
87704f217acSStephen Hemminger 	}
8781da177e4SLinus Torvalds 
8791da177e4SLinus Torvalds 	kfree(meta);
8801da177e4SLinus Torvalds }
8811da177e4SLinus Torvalds 
882add93b61SPatrick McHardy static inline int meta_change_data(struct meta_value *dst, struct nlattr *nla)
8831da177e4SLinus Torvalds {
884add93b61SPatrick McHardy 	if (nla) {
885add93b61SPatrick McHardy 		if (nla_len(nla) == 0)
8861da177e4SLinus Torvalds 			return -EINVAL;
8871da177e4SLinus Torvalds 
888add93b61SPatrick McHardy 		return meta_type_ops(dst)->change(dst, nla);
8891da177e4SLinus Torvalds 	}
8901da177e4SLinus Torvalds 
8911da177e4SLinus Torvalds 	return 0;
8921da177e4SLinus Torvalds }
8931da177e4SLinus Torvalds 
8941da177e4SLinus Torvalds static inline int meta_is_supported(struct meta_value *val)
8951da177e4SLinus Torvalds {
896cc7ec456SEric Dumazet 	return !meta_id(val) || meta_ops(val)->get;
8971da177e4SLinus Torvalds }
8981da177e4SLinus Torvalds 
8997a9c1bd4SPatrick McHardy static const struct nla_policy meta_policy[TCA_EM_META_MAX + 1] = {
9007a9c1bd4SPatrick McHardy 	[TCA_EM_META_HDR]	= { .len = sizeof(struct tcf_meta_hdr) },
9017a9c1bd4SPatrick McHardy };
9027a9c1bd4SPatrick McHardy 
90382a470f1SJohn Fastabend static int em_meta_change(struct net *net, void *data, int len,
9041da177e4SLinus Torvalds 			  struct tcf_ematch *m)
9051da177e4SLinus Torvalds {
906cee63723SPatrick McHardy 	int err;
907add93b61SPatrick McHardy 	struct nlattr *tb[TCA_EM_META_MAX + 1];
9081da177e4SLinus Torvalds 	struct tcf_meta_hdr *hdr;
9091da177e4SLinus Torvalds 	struct meta_match *meta = NULL;
9101da177e4SLinus Torvalds 
9118cb08174SJohannes Berg 	err = nla_parse_deprecated(tb, TCA_EM_META_MAX, data, len,
9128cb08174SJohannes Berg 				   meta_policy, NULL);
913cee63723SPatrick McHardy 	if (err < 0)
9141da177e4SLinus Torvalds 		goto errout;
9151da177e4SLinus Torvalds 
916cee63723SPatrick McHardy 	err = -EINVAL;
9177a9c1bd4SPatrick McHardy 	if (tb[TCA_EM_META_HDR] == NULL)
9181da177e4SLinus Torvalds 		goto errout;
919add93b61SPatrick McHardy 	hdr = nla_data(tb[TCA_EM_META_HDR]);
9201da177e4SLinus Torvalds 
9211da177e4SLinus Torvalds 	if (TCF_META_TYPE(hdr->left.kind) != TCF_META_TYPE(hdr->right.kind) ||
9221da177e4SLinus Torvalds 	    TCF_META_TYPE(hdr->left.kind) > TCF_META_TYPE_MAX ||
9231da177e4SLinus Torvalds 	    TCF_META_ID(hdr->left.kind) > TCF_META_ID_MAX ||
9241da177e4SLinus Torvalds 	    TCF_META_ID(hdr->right.kind) > TCF_META_ID_MAX)
9251da177e4SLinus Torvalds 		goto errout;
9261da177e4SLinus Torvalds 
9270da974f4SPanagiotis Issaris 	meta = kzalloc(sizeof(*meta), GFP_KERNEL);
9280c4e4020Sstephen hemminger 	if (meta == NULL) {
9290c4e4020Sstephen hemminger 		err = -ENOMEM;
9301da177e4SLinus Torvalds 		goto errout;
9310c4e4020Sstephen hemminger 	}
9321da177e4SLinus Torvalds 
9331da177e4SLinus Torvalds 	memcpy(&meta->lvalue.hdr, &hdr->left, sizeof(hdr->left));
9341da177e4SLinus Torvalds 	memcpy(&meta->rvalue.hdr, &hdr->right, sizeof(hdr->right));
9351da177e4SLinus Torvalds 
9361da177e4SLinus Torvalds 	if (!meta_is_supported(&meta->lvalue) ||
9371da177e4SLinus Torvalds 	    !meta_is_supported(&meta->rvalue)) {
9381da177e4SLinus Torvalds 		err = -EOPNOTSUPP;
9391da177e4SLinus Torvalds 		goto errout;
9401da177e4SLinus Torvalds 	}
9411da177e4SLinus Torvalds 
942add93b61SPatrick McHardy 	if (meta_change_data(&meta->lvalue, tb[TCA_EM_META_LVALUE]) < 0 ||
943add93b61SPatrick McHardy 	    meta_change_data(&meta->rvalue, tb[TCA_EM_META_RVALUE]) < 0)
9441da177e4SLinus Torvalds 		goto errout;
9451da177e4SLinus Torvalds 
9461da177e4SLinus Torvalds 	m->datalen = sizeof(*meta);
9471da177e4SLinus Torvalds 	m->data = (unsigned long) meta;
9481da177e4SLinus Torvalds 
9491da177e4SLinus Torvalds 	err = 0;
9501da177e4SLinus Torvalds errout:
9511da177e4SLinus Torvalds 	if (err && meta)
9521da177e4SLinus Torvalds 		meta_delete(meta);
9531da177e4SLinus Torvalds 	return err;
9541da177e4SLinus Torvalds }
9551da177e4SLinus Torvalds 
95682a470f1SJohn Fastabend static void em_meta_destroy(struct tcf_ematch *m)
9571da177e4SLinus Torvalds {
9581da177e4SLinus Torvalds 	if (m)
9591da177e4SLinus Torvalds 		meta_delete((struct meta_match *) m->data);
9601da177e4SLinus Torvalds }
9611da177e4SLinus Torvalds 
9621da177e4SLinus Torvalds static int em_meta_dump(struct sk_buff *skb, struct tcf_ematch *em)
9631da177e4SLinus Torvalds {
9641da177e4SLinus Torvalds 	struct meta_match *meta = (struct meta_match *) em->data;
9651da177e4SLinus Torvalds 	struct tcf_meta_hdr hdr;
966cfe2f14cSJulia Lawall 	const struct meta_type_ops *ops;
9671da177e4SLinus Torvalds 
9681da177e4SLinus Torvalds 	memset(&hdr, 0, sizeof(hdr));
9691da177e4SLinus Torvalds 	memcpy(&hdr.left, &meta->lvalue.hdr, sizeof(hdr.left));
9701da177e4SLinus Torvalds 	memcpy(&hdr.right, &meta->rvalue.hdr, sizeof(hdr.right));
9711da177e4SLinus Torvalds 
9721b34ec43SDavid S. Miller 	if (nla_put(skb, TCA_EM_META_HDR, sizeof(hdr), &hdr))
9731b34ec43SDavid S. Miller 		goto nla_put_failure;
9741da177e4SLinus Torvalds 
9751da177e4SLinus Torvalds 	ops = meta_type_ops(&meta->lvalue);
9761da177e4SLinus Torvalds 	if (ops->dump(skb, &meta->lvalue, TCA_EM_META_LVALUE) < 0 ||
9771da177e4SLinus Torvalds 	    ops->dump(skb, &meta->rvalue, TCA_EM_META_RVALUE) < 0)
978add93b61SPatrick McHardy 		goto nla_put_failure;
9791da177e4SLinus Torvalds 
9801da177e4SLinus Torvalds 	return 0;
9811da177e4SLinus Torvalds 
982add93b61SPatrick McHardy nla_put_failure:
9831da177e4SLinus Torvalds 	return -1;
9841da177e4SLinus Torvalds }
9851da177e4SLinus Torvalds 
9861da177e4SLinus Torvalds static struct tcf_ematch_ops em_meta_ops = {
9871da177e4SLinus Torvalds 	.kind	  = TCF_EM_META,
9881da177e4SLinus Torvalds 	.change	  = em_meta_change,
9891da177e4SLinus Torvalds 	.match	  = em_meta_match,
9901da177e4SLinus Torvalds 	.destroy  = em_meta_destroy,
9911da177e4SLinus Torvalds 	.dump	  = em_meta_dump,
9921da177e4SLinus Torvalds 	.owner	  = THIS_MODULE,
9931da177e4SLinus Torvalds 	.link	  = LIST_HEAD_INIT(em_meta_ops.link)
9941da177e4SLinus Torvalds };
9951da177e4SLinus Torvalds 
9961da177e4SLinus Torvalds static int __init init_em_meta(void)
9971da177e4SLinus Torvalds {
9981da177e4SLinus Torvalds 	return tcf_em_register(&em_meta_ops);
9991da177e4SLinus Torvalds }
10001da177e4SLinus Torvalds 
10011da177e4SLinus Torvalds static void __exit exit_em_meta(void)
10021da177e4SLinus Torvalds {
10031da177e4SLinus Torvalds 	tcf_em_unregister(&em_meta_ops);
10041da177e4SLinus Torvalds }
10051da177e4SLinus Torvalds 
10061da177e4SLinus Torvalds MODULE_LICENSE("GPL");
10071da177e4SLinus Torvalds 
10081da177e4SLinus Torvalds module_init(init_em_meta);
10091da177e4SLinus Torvalds module_exit(exit_em_meta);
1010db3d99c0SPatrick McHardy 
1011db3d99c0SPatrick McHardy MODULE_ALIAS_TCF_EMATCH(TCF_EM_META);
1012