[linux-2.6-block.git] / net / xfrm / xfrm_hash.h

/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _XFRM_HASH_H
#define _XFRM_HASH_H

#include <linux/xfrm.h>
#include <linux/socket.h>
#include <linux/jhash.h>

static inline unsigned int __xfrm4_addr_hash(const xfrm_address_t *addr)
{
	return ntohl(addr->a4);
}

static inline unsigned int __xfrm6_addr_hash(const xfrm_address_t *addr)
{
	return jhash2((__force u32 *)addr->a6, 4, 0);
}

static inline unsigned int __xfrm4_daddr_saddr_hash(const xfrm_address_t *daddr,
						    const xfrm_address_t *saddr)
{
	u32 sum = (__force u32)daddr->a4 + (__force u32)saddr->a4;
	return ntohl((__force __be32)sum);
}

static inline unsigned int __xfrm6_daddr_saddr_hash(const xfrm_address_t *daddr,
						    const xfrm_address_t *saddr)
{
	return __xfrm6_addr_hash(daddr) ^ __xfrm6_addr_hash(saddr);
}

static inline u32 __bits2mask32(__u8 bits)
{
	u32 mask32 = 0xffffffff;

	if (bits == 0)
		mask32 = 0;
	else if (bits < 32)
		mask32 <<= (32 - bits);

	return mask32;
}

static inline unsigned int __xfrm4_dpref_spref_hash(const xfrm_address_t *daddr,
						    const xfrm_address_t *saddr,
						    __u8 dbits,
						    __u8 sbits)
{
	return jhash_2words(ntohl(daddr->a4) & __bits2mask32(dbits),
			    ntohl(saddr->a4) & __bits2mask32(sbits),
			    0);
}

static inline unsigned int __xfrm6_pref_hash(const xfrm_address_t *addr,
					     __u8 prefixlen)
{
	unsigned int pdw;
	unsigned int pbi;
	u32 initval = 0;

	pdw = prefixlen >> 5;     /* num of whole u32 in prefix */
	pbi = prefixlen &  0x1f;  /* num of bits in incomplete u32 in prefix */

	if (pbi) {
		__be32 mask;

		mask = htonl((0xffffffff) << (32 - pbi));

		initval = (__force u32)(addr->a6[pdw] & mask);
	}

	return jhash2((__force u32 *)addr->a6, pdw, initval);
}

static inline unsigned int __xfrm6_dpref_spref_hash(const xfrm_address_t *daddr,
						    const xfrm_address_t *saddr,
						    __u8 dbits,
						    __u8 sbits)
{
	return __xfrm6_pref_hash(daddr, dbits) ^
	       __xfrm6_pref_hash(saddr, sbits);
}

static inline unsigned int __xfrm_dst_hash(const xfrm_address_t *daddr,
					   const xfrm_address_t *saddr,
					   u32 reqid, unsigned short family,
					   unsigned int hmask)
{
	unsigned int h = family ^ reqid;
	switch (family) {
	case AF_INET:
		h ^= __xfrm4_daddr_saddr_hash(daddr, saddr);
		break;
	case AF_INET6:
		h ^= __xfrm6_daddr_saddr_hash(daddr, saddr);
		break;
	}
	return (h ^ (h >> 16)) & hmask;
}

static inline unsigned int __xfrm_src_hash(const xfrm_address_t *daddr,
					   const xfrm_address_t *saddr,
					   unsigned short family,
					   unsigned int hmask)
{
	unsigned int h = family;
	switch (family) {
	case AF_INET:
		h ^= __xfrm4_daddr_saddr_hash(daddr, saddr);
		break;
	case AF_INET6:
		h ^= __xfrm6_daddr_saddr_hash(daddr, saddr);
		break;
	}
	return (h ^ (h >> 16)) & hmask;
}

static inline unsigned int
__xfrm_spi_hash(const xfrm_address_t *daddr, __be32 spi, u8 proto,
		unsigned short family, unsigned int hmask)
{
	unsigned int h = (__force u32)spi ^ proto;
	switch (family) {
	case AF_INET:
		h ^= __xfrm4_addr_hash(daddr);
		break;
	case AF_INET6:
		h ^= __xfrm6_addr_hash(daddr);
		break;
	}
	return (h ^ (h >> 10) ^ (h >> 20)) & hmask;
}

static inline unsigned int
__xfrm_seq_hash(u32 seq, unsigned int hmask)
{
	unsigned int h = seq;
	return (h ^ (h >> 10) ^ (h >> 20)) & hmask;
}

static inline unsigned int __idx_hash(u32 index, unsigned int hmask)
{
	return (index ^ (index >> 8)) & hmask;
}

static inline unsigned int __sel_hash(const struct xfrm_selector *sel,
				      unsigned short family, unsigned int hmask,
				      u8 dbits, u8 sbits)
{
	const xfrm_address_t *daddr = &sel->daddr;
	const xfrm_address_t *saddr = &sel->saddr;
	unsigned int h = 0;

	switch (family) {
	case AF_INET:
		if (sel->prefixlen_d < dbits ||
		    sel->prefixlen_s < sbits)
			return hmask + 1;

		h = __xfrm4_dpref_spref_hash(daddr, saddr, dbits, sbits);
		break;

	case AF_INET6:
		if (sel->prefixlen_d < dbits ||
		    sel->prefixlen_s < sbits)
			return hmask + 1;

		h = __xfrm6_dpref_spref_hash(daddr, saddr, dbits, sbits);
		break;
	}
	h ^= (h >> 16);
	return h & hmask;
}

static inline unsigned int __addr_hash(const xfrm_address_t *daddr,
				       const xfrm_address_t *saddr,
				       unsigned short family,
				       unsigned int hmask,
				       u8 dbits, u8 sbits)
{
	unsigned int h = 0;

	switch (family) {
	case AF_INET:
		h = __xfrm4_dpref_spref_hash(daddr, saddr, dbits, sbits);
		break;

	case AF_INET6:
		h = __xfrm6_dpref_spref_hash(daddr, saddr, dbits, sbits);
		break;
	}
	h ^= (h >> 16);
	return h & hmask;
}

struct hlist_head *xfrm_hash_alloc(unsigned int sz);
void xfrm_hash_free(struct hlist_head *n, unsigned int sz);

#endif /* _XFRM_HASH_H */
Commit	Line	Data
b2441318	1	/* SPDX-License-Identifier: GPL-2.0 */
44e36b42 DM	2	#ifndef _XFRM_HASH_H
	3	#define _XFRM_HASH_H
	4
	5	#include <linux/xfrm.h>
	6	#include <linux/socket.h>
b58555f1	7	#include <linux/jhash.h>
44e36b42	8
5f803b58	9	static inline unsigned int __xfrm4_addr_hash(const xfrm_address_t *addr)
44e36b42 DM	10	{
	11	return ntohl(addr->a4);
	12	}
	13
5f803b58	14	static inline unsigned int __xfrm6_addr_hash(const xfrm_address_t *addr)
44e36b42	15	{
8d4b6bce	16	return jhash2((__force u32 *)addr->a6, 4, 0);
44e36b42 DM	17	}
44e36b42 DM	18
5f803b58 DM	19	static inline unsigned int __xfrm4_daddr_saddr_hash(const xfrm_address_t *daddr,
5f803b58 DM	20	const xfrm_address_t *saddr)
44e36b42	21	{
0eae88f3 ED	22	u32 sum = (__force u32)daddr->a4 + (__force u32)saddr->a4;
0eae88f3 ED	23	return ntohl((__force __be32)sum);
44e36b42 DM	24	}
44e36b42 DM	25
5f803b58 DM	26	static inline unsigned int __xfrm6_daddr_saddr_hash(const xfrm_address_t *daddr,
5f803b58 DM	27	const xfrm_address_t *saddr)
44e36b42	28	{
8d4b6bce	29	return __xfrm6_addr_hash(daddr) ^ __xfrm6_addr_hash(saddr);
44e36b42 DM	30	}
44e36b42 DM	31
b58555f1 CG	32	static inline u32 __bits2mask32(__u8 bits)
	33	{
	34	u32 mask32 = 0xffffffff;
	35
	36	if (bits == 0)
	37	mask32 = 0;
	38	else if (bits < 32)
	39	mask32 <<= (32 - bits);
	40
	41	return mask32;
	42	}
	43
	44	static inline unsigned int __xfrm4_dpref_spref_hash(const xfrm_address_t *daddr,
	45	const xfrm_address_t *saddr,
	46	__u8 dbits,
	47	__u8 sbits)
	48	{
	49	return jhash_2words(ntohl(daddr->a4) & __bits2mask32(dbits),
	50	ntohl(saddr->a4) & __bits2mask32(sbits),
	51	0);
	52	}
	53
	54	static inline unsigned int __xfrm6_pref_hash(const xfrm_address_t *addr,
	55	__u8 prefixlen)
	56	{
d7f69466 AD	57	unsigned int pdw;
d7f69466 AD	58	unsigned int pbi;
b58555f1 CG	59	u32 initval = 0;
	60
	61	pdw = prefixlen >> 5; /* num of whole u32 in prefix */
	62	pbi = prefixlen & 0x1f; /* num of bits in incomplete u32 in prefix */
	63
	64	if (pbi) {
	65	__be32 mask;
	66
	67	mask = htonl((0xffffffff) << (32 - pbi));
	68
	69	initval = (__force u32)(addr->a6[pdw] & mask);
	70	}
	71
	72	return jhash2((__force u32 *)addr->a6, pdw, initval);
	73	}
	74
	75	static inline unsigned int __xfrm6_dpref_spref_hash(const xfrm_address_t *daddr,
	76	const xfrm_address_t *saddr,
	77	__u8 dbits,
	78	__u8 sbits)
	79	{
	80	return __xfrm6_pref_hash(daddr, dbits) ^
	81	__xfrm6_pref_hash(saddr, sbits);
	82	}
	83
5f803b58 DM	84	static inline unsigned int __xfrm_dst_hash(const xfrm_address_t *daddr,
5f803b58 DM	85	const xfrm_address_t *saddr,
44e36b42 DM	86	u32 reqid, unsigned short family,
	87	unsigned int hmask)
	88	{
	89	unsigned int h = family ^ reqid;
	90	switch (family) {
	91	case AF_INET:
	92	h ^= __xfrm4_daddr_saddr_hash(daddr, saddr);
	93	break;
	94	case AF_INET6:
	95	h ^= __xfrm6_daddr_saddr_hash(daddr, saddr);
	96	break;
	97	}
	98	return (h ^ (h >> 16)) & hmask;
	99	}
	100
95c96174 ED	101	static inline unsigned int __xfrm_src_hash(const xfrm_address_t *daddr,
	102	const xfrm_address_t *saddr,
	103	unsigned short family,
	104	unsigned int hmask)
44e36b42 DM	105	{
	106	unsigned int h = family;
	107	switch (family) {
	108	case AF_INET:
667bbcb6	109	h ^= __xfrm4_daddr_saddr_hash(daddr, saddr);
44e36b42 DM	110	break;
44e36b42 DM	111	case AF_INET6:
667bbcb6	112	h ^= __xfrm6_daddr_saddr_hash(daddr, saddr);
44e36b42	113	break;
ccbd6a5a	114	}
44e36b42 DM	115	return (h ^ (h >> 16)) & hmask;
	116	}
	117
	118	static inline unsigned int
5f803b58 DM	119	__xfrm_spi_hash(const xfrm_address_t *daddr, __be32 spi, u8 proto,
5f803b58 DM	120	unsigned short family, unsigned int hmask)
44e36b42	121	{
8122adf0	122	unsigned int h = (__force u32)spi ^ proto;
44e36b42 DM	123	switch (family) {
	124	case AF_INET:
	125	h ^= __xfrm4_addr_hash(daddr);
	126	break;
	127	case AF_INET6:
	128	h ^= __xfrm6_addr_hash(daddr);
	129	break;
	130	}
	131	return (h ^ (h >> 10) ^ (h >> 20)) & hmask;
	132	}
	133
fe9f1d87 SD	134	static inline unsigned int
	135	__xfrm_seq_hash(u32 seq, unsigned int hmask)
	136	{
	137	unsigned int h = seq;
	138	return (h ^ (h >> 10) ^ (h >> 20)) & hmask;
	139	}
	140
44e36b42 DM	141	static inline unsigned int __idx_hash(u32 index, unsigned int hmask)
	142	{
	143	return (index ^ (index >> 8)) & hmask;
	144	}
	145
5f803b58	146	static inline unsigned int __sel_hash(const struct xfrm_selector *sel,
b58555f1 CG	147	unsigned short family, unsigned int hmask,
b58555f1 CG	148	u8 dbits, u8 sbits)
44e36b42	149	{
5f803b58 DM	150	const xfrm_address_t *daddr = &sel->daddr;
5f803b58 DM	151	const xfrm_address_t *saddr = &sel->saddr;
44e36b42 DM	152	unsigned int h = 0;
	153
	154	switch (family) {
	155	case AF_INET:
b58555f1 CG	156	if (sel->prefixlen_d < dbits \|\|
b58555f1 CG	157	sel->prefixlen_s < sbits)
44e36b42 DM	158	return hmask + 1;
44e36b42 DM	159
b58555f1	160	h = __xfrm4_dpref_spref_hash(daddr, saddr, dbits, sbits);
44e36b42 DM	161	break;
	162
	163	case AF_INET6:
b58555f1 CG	164	if (sel->prefixlen_d < dbits \|\|
b58555f1 CG	165	sel->prefixlen_s < sbits)
44e36b42 DM	166	return hmask + 1;
44e36b42 DM	167
b58555f1	168	h = __xfrm6_dpref_spref_hash(daddr, saddr, dbits, sbits);
44e36b42	169	break;
ccbd6a5a	170	}
44e36b42 DM	171	h ^= (h >> 16);
	172	return h & hmask;
	173	}
	174
5f803b58 DM	175	static inline unsigned int __addr_hash(const xfrm_address_t *daddr,
5f803b58 DM	176	const xfrm_address_t *saddr,
b58555f1 CG	177	unsigned short family,
	178	unsigned int hmask,
	179	u8 dbits, u8 sbits)
44e36b42 DM	180	{
	181	unsigned int h = 0;
	182
	183	switch (family) {
	184	case AF_INET:
b58555f1	185	h = __xfrm4_dpref_spref_hash(daddr, saddr, dbits, sbits);
44e36b42 DM	186	break;
	187
	188	case AF_INET6:
b58555f1	189	h = __xfrm6_dpref_spref_hash(daddr, saddr, dbits, sbits);
44e36b42	190	break;
ccbd6a5a	191	}
44e36b42 DM	192	h ^= (h >> 16);
	193	return h & hmask;
	194	}
	195
c1b1203d JP	196	struct hlist_head *xfrm_hash_alloc(unsigned int sz);
c1b1203d JP	197	void xfrm_hash_free(struct hlist_head *n, unsigned int sz);
44e36b42 DM	198
44e36b42 DM	199	#endif /* _XFRM_HASH_H */