[thirdparty/kernel/stable.git] / net / netfilter / ipset / ip_set_hash_net.c

// SPDX-License-Identifier: GPL-2.0-only
/* Copyright (C) 2003-2013 Jozsef Kadlecsik <kadlec@netfilter.org> */

/* Kernel module implementing an IP set type: the hash:net type */

#include <linux/jhash.h>
#include <linux/module.h>
#include <linux/ip.h>
#include <linux/skbuff.h>
#include <linux/errno.h>
#include <linux/random.h>
#include <net/ip.h>
#include <net/ipv6.h>
#include <net/netlink.h>

#include <linux/netfilter.h>
#include <linux/netfilter/ipset/pfxlen.h>
#include <linux/netfilter/ipset/ip_set.h>
#include <linux/netfilter/ipset/ip_set_hash.h>

#define IPSET_TYPE_REV_MIN	0
/*				1    Range as input support for IPv4 added */
/*				2    nomatch flag support added */
/*				3    Counters support added */
/*				4    Comments support added */
/*				5    Forceadd support added */
/*				6    skbinfo support added */
#define IPSET_TYPE_REV_MAX	7 /* bucketsize, initval support added */

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@netfilter.org>");
IP_SET_MODULE_DESC("hash:net", IPSET_TYPE_REV_MIN, IPSET_TYPE_REV_MAX);
MODULE_ALIAS("ip_set_hash:net");

/* Type specific function prefix */
#define HTYPE		hash_net
#define IP_SET_HASH_WITH_NETS

/* IPv4 variant */

/* Member elements  */
struct hash_net4_elem {
	__be32 ip;
	u16 padding0;
	u8 nomatch;
	u8 cidr;
};

/* Common functions */

static bool
hash_net4_data_equal(const struct hash_net4_elem *ip1,
		     const struct hash_net4_elem *ip2,
		     u32 *multi)
{
	return ip1->ip == ip2->ip &&
	       ip1->cidr == ip2->cidr;
}

static int
hash_net4_do_data_match(const struct hash_net4_elem *elem)
{
	return elem->nomatch ? -ENOTEMPTY : 1;
}

static void
hash_net4_data_set_flags(struct hash_net4_elem *elem, u32 flags)
{
	elem->nomatch = (flags >> 16) & IPSET_FLAG_NOMATCH;
}

static void
hash_net4_data_reset_flags(struct hash_net4_elem *elem, u8 *flags)
{
	swap(*flags, elem->nomatch);
}

static void
hash_net4_data_netmask(struct hash_net4_elem *elem, u8 cidr)
{
	elem->ip &= ip_set_netmask(cidr);
	elem->cidr = cidr;
}

static bool
hash_net4_data_list(struct sk_buff *skb, const struct hash_net4_elem *data)
{
	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;

	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip) ||
	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr) ||
	    (flags &&
	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
		goto nla_put_failure;
	return false;

nla_put_failure:
	return true;
}

static void
hash_net4_data_next(struct hash_net4_elem *next,
		    const struct hash_net4_elem *d)
{
	next->ip = d->ip;
}

#define MTYPE		hash_net4
#define HOST_MASK	32
#include "ip_set_hash_gen.h"

static int
hash_net4_kadt(struct ip_set *set, const struct sk_buff *skb,
	       const struct xt_action_param *par,
	       enum ipset_adt adt, struct ip_set_adt_opt *opt)
{
	const struct hash_net4 *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_net4_elem e = {
		.cidr = INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
	};
	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);

	if (e.cidr == 0)
		return -EINVAL;
	if (adt == IPSET_TEST)
		e.cidr = HOST_MASK;

	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip);
	e.ip &= ip_set_netmask(e.cidr);

	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
}

static int
hash_net4_uadt(struct ip_set *set, struct nlattr *tb[],
	       enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
{
	struct hash_net4 *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_net4_elem e = { .cidr = HOST_MASK };
	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
	u32 ip = 0, ip_to = 0, i = 0;
	int ret;

	if (tb[IPSET_ATTR_LINENO])
		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);

	if (unlikely(!tb[IPSET_ATTR_IP] ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS)))
		return -IPSET_ERR_PROTOCOL;

	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP], &ip);
	if (ret)
		return ret;

	ret = ip_set_get_extensions(set, tb, &ext);
	if (ret)
		return ret;

	if (tb[IPSET_ATTR_CIDR]) {
		e.cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
		if (!e.cidr || e.cidr > HOST_MASK)
			return -IPSET_ERR_INVALID_CIDR;
	}

	if (tb[IPSET_ATTR_CADT_FLAGS]) {
		u32 cadt_flags = ip_set_get_h32(tb[IPSET_ATTR_CADT_FLAGS]);

		if (cadt_flags & IPSET_FLAG_NOMATCH)
			flags |= (IPSET_FLAG_NOMATCH << 16);
	}

	if (adt == IPSET_TEST || !tb[IPSET_ATTR_IP_TO]) {
		e.ip = htonl(ip & ip_set_hostmask(e.cidr));
		ret = adtfn(set, &e, &ext, &ext, flags);
		return ip_set_enomatch(ret, flags, adt, set) ? -ret :
		       ip_set_eexist(ret, flags) ? 0 : ret;
	}

	ip_to = ip;
	if (tb[IPSET_ATTR_IP_TO]) {
		ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP_TO], &ip_to);
		if (ret)
			return ret;
		if (ip_to < ip)
			swap(ip, ip_to);
		if (ip + UINT_MAX == ip_to)
			return -IPSET_ERR_HASH_RANGE;
	}

	if (retried)
		ip = ntohl(h->next.ip);
	do {
		i++;
		e.ip = htonl(ip);
		if (i > IPSET_MAX_RANGE) {
			hash_net4_data_next(&h->next, &e);
			return -ERANGE;
		}
		ip = ip_set_range_to_cidr(ip, ip_to, &e.cidr);
		ret = adtfn(set, &e, &ext, &ext, flags);
		if (ret && !ip_set_eexist(ret, flags))
			return ret;

		ret = 0;
	} while (ip++ < ip_to);
	return ret;
}

/* IPv6 variant */

struct hash_net6_elem {
	union nf_inet_addr ip;
	u16 padding0;
	u8 nomatch;
	u8 cidr;
};

/* Common functions */

static bool
hash_net6_data_equal(const struct hash_net6_elem *ip1,
		     const struct hash_net6_elem *ip2,
		     u32 *multi)
{
	return ipv6_addr_equal(&ip1->ip.in6, &ip2->ip.in6) &&
	       ip1->cidr == ip2->cidr;
}

static int
hash_net6_do_data_match(const struct hash_net6_elem *elem)
{
	return elem->nomatch ? -ENOTEMPTY : 1;
}

static void
hash_net6_data_set_flags(struct hash_net6_elem *elem, u32 flags)
{
	elem->nomatch = (flags >> 16) & IPSET_FLAG_NOMATCH;
}

static void
hash_net6_data_reset_flags(struct hash_net6_elem *elem, u8 *flags)
{
	swap(*flags, elem->nomatch);
}

static void
hash_net6_data_netmask(struct hash_net6_elem *elem, u8 cidr)
{
	ip6_netmask(&elem->ip, cidr);
	elem->cidr = cidr;
}

static bool
hash_net6_data_list(struct sk_buff *skb, const struct hash_net6_elem *data)
{
	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;

	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip.in6) ||
	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr) ||
	    (flags &&
	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
		goto nla_put_failure;
	return false;

nla_put_failure:
	return true;
}

static void
hash_net6_data_next(struct hash_net6_elem *next,
		    const struct hash_net6_elem *d)
{
}

#undef MTYPE
#undef HOST_MASK

#define MTYPE		hash_net6
#define HOST_MASK	128
#define IP_SET_EMIT_CREATE
#include "ip_set_hash_gen.h"

static int
hash_net6_kadt(struct ip_set *set, const struct sk_buff *skb,
	       const struct xt_action_param *par,
	       enum ipset_adt adt, struct ip_set_adt_opt *opt)
{
	const struct hash_net6 *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_net6_elem e = {
		.cidr = INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
	};
	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);

	if (e.cidr == 0)
		return -EINVAL;
	if (adt == IPSET_TEST)
		e.cidr = HOST_MASK;

	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip.in6);
	ip6_netmask(&e.ip, e.cidr);

	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
}

static int
hash_net6_uadt(struct ip_set *set, struct nlattr *tb[],
	       enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
{
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_net6_elem e = { .cidr = HOST_MASK };
	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
	int ret;

	if (tb[IPSET_ATTR_LINENO])
		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);

	if (unlikely(!tb[IPSET_ATTR_IP] ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS)))
		return -IPSET_ERR_PROTOCOL;
	if (unlikely(tb[IPSET_ATTR_IP_TO]))
		return -IPSET_ERR_HASH_RANGE_UNSUPPORTED;

	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip);
	if (ret)
		return ret;

	ret = ip_set_get_extensions(set, tb, &ext);
	if (ret)
		return ret;

	if (tb[IPSET_ATTR_CIDR]) {
		e.cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
		if (!e.cidr || e.cidr > HOST_MASK)
			return -IPSET_ERR_INVALID_CIDR;
	}

	ip6_netmask(&e.ip, e.cidr);

	if (tb[IPSET_ATTR_CADT_FLAGS]) {
		u32 cadt_flags = ip_set_get_h32(tb[IPSET_ATTR_CADT_FLAGS]);

		if (cadt_flags & IPSET_FLAG_NOMATCH)
			flags |= (IPSET_FLAG_NOMATCH << 16);
	}

	ret = adtfn(set, &e, &ext, &ext, flags);

	return ip_set_enomatch(ret, flags, adt, set) ? -ret :
	       ip_set_eexist(ret, flags) ? 0 : ret;
}

static struct ip_set_type hash_net_type __read_mostly = {
	.name		= "hash:net",
	.protocol	= IPSET_PROTOCOL,
	.features	= IPSET_TYPE_IP | IPSET_TYPE_NOMATCH,
	.dimension	= IPSET_DIM_ONE,
	.family		= NFPROTO_UNSPEC,
	.revision_min	= IPSET_TYPE_REV_MIN,
	.revision_max	= IPSET_TYPE_REV_MAX,
	.create_flags[IPSET_TYPE_REV_MAX] = IPSET_CREATE_FLAG_BUCKETSIZE,
	.create		= hash_net_create,
	.create_policy	= {
		[IPSET_ATTR_HASHSIZE]	= { .type = NLA_U32 },
		[IPSET_ATTR_MAXELEM]	= { .type = NLA_U32 },
		[IPSET_ATTR_INITVAL]	= { .type = NLA_U32 },
		[IPSET_ATTR_BUCKETSIZE]	= { .type = NLA_U8 },
		[IPSET_ATTR_RESIZE]	= { .type = NLA_U8  },
		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
		[IPSET_ATTR_CADT_FLAGS]	= { .type = NLA_U32 },
	},
	.adt_policy	= {
		[IPSET_ATTR_IP]		= { .type = NLA_NESTED },
		[IPSET_ATTR_IP_TO]	= { .type = NLA_NESTED },
		[IPSET_ATTR_CIDR]	= { .type = NLA_U8 },
		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
		[IPSET_ATTR_LINENO]	= { .type = NLA_U32 },
		[IPSET_ATTR_CADT_FLAGS]	= { .type = NLA_U32 },
		[IPSET_ATTR_BYTES]	= { .type = NLA_U64 },
		[IPSET_ATTR_PACKETS]	= { .type = NLA_U64 },
		[IPSET_ATTR_COMMENT]	= { .type = NLA_NUL_STRING,
					    .len  = IPSET_MAX_COMMENT_SIZE },
		[IPSET_ATTR_SKBMARK]	= { .type = NLA_U64 },
		[IPSET_ATTR_SKBPRIO]	= { .type = NLA_U32 },
		[IPSET_ATTR_SKBQUEUE]	= { .type = NLA_U16 },
	},
	.me		= THIS_MODULE,
};

static int __init
hash_net_init(void)
{
	return ip_set_type_register(&hash_net_type);
}

static void __exit
hash_net_fini(void)
{
	rcu_barrier();
	ip_set_type_unregister(&hash_net_type);
}

module_init(hash_net_init);
module_exit(hash_net_fini);
Commit	Line	Data
d2912cb1	1	// SPDX-License-Identifier: GPL-2.0-only
1c5ba67d	2	/* Copyright (C) 2003-2013 Jozsef Kadlecsik <kadlec@netfilter.org> */
b3837029 JK	3
	4	/* Kernel module implementing an IP set type: the hash:net type */
	5
	6	#include <linux/jhash.h>
	7	#include <linux/module.h>
	8	#include <linux/ip.h>
	9	#include <linux/skbuff.h>
	10	#include <linux/errno.h>
b3837029 JK	11	#include <linux/random.h>
	12	#include <net/ip.h>
	13	#include <net/ipv6.h>
	14	#include <net/netlink.h>
	15
	16	#include <linux/netfilter.h>
	17	#include <linux/netfilter/ipset/pfxlen.h>
	18	#include <linux/netfilter/ipset/ip_set.h>
b3837029 JK	19	#include <linux/netfilter/ipset/ip_set_hash.h>
b3837029 JK	20
35b8dcf8 JK	21	#define IPSET_TYPE_REV_MIN 0
	22	/* 1 Range as input support for IPv4 added */
	23	/* 2 nomatch flag support added */
fda75c6d	24	/* 3 Counters support added */
07cf8f5a	25	/* 4 Comments support added */
af331419	26	/* 5 Forceadd support added */
ccf0a4b7	27	/* 6 skbinfo support added */
3976ca10	28	#define IPSET_TYPE_REV_MAX 7 /* bucketsize, initval support added */
10111a6e	29
b3837029	30	MODULE_LICENSE("GPL");
fe03d474	31	MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@netfilter.org>");
35b8dcf8	32	IP_SET_MODULE_DESC("hash:net", IPSET_TYPE_REV_MIN, IPSET_TYPE_REV_MAX);
b3837029 JK	33	MODULE_ALIAS("ip_set_hash:net");
	34
	35	/* Type specific function prefix */
5d50e1d8 JK	36	#define HTYPE hash_net
5d50e1d8 JK	37	#define IP_SET_HASH_WITH_NETS
b3837029	38
03c8b234	39	/* IPv4 variant */
b3837029	40
5d50e1d8	41	/* Member elements */
b3837029 JK	42	struct hash_net4_elem {
	43	__be32 ip;
	44	u16 padding0;
2a7cef2a	45	u8 nomatch;
b3837029 JK	46	u8 cidr;
	47	};
	48
5d50e1d8 JK	49	/* Common functions */
5d50e1d8 JK	50
8dea982a	51	static bool
b3837029	52	hash_net4_data_equal(const struct hash_net4_elem *ip1,
89dc79b7 JK	53	const struct hash_net4_elem *ip2,
89dc79b7 JK	54	u32 *multi)
b3837029	55	{
2a7cef2a JK	56	return ip1->ip == ip2->ip &&
2a7cef2a JK	57	ip1->cidr == ip2->cidr;
b3837029 JK	58	}
b3837029 JK	59
8dea982a	60	static int
5d50e1d8	61	hash_net4_do_data_match(const struct hash_net4_elem *elem)
b3837029	62	{
5d50e1d8	63	return elem->nomatch ? -ENOTEMPTY : 1;
2a7cef2a JK	64	}
2a7cef2a JK	65
8dea982a	66	static void
5d50e1d8	67	hash_net4_data_set_flags(struct hash_net4_elem *elem, u32 flags)
2a7cef2a	68	{
5d50e1d8	69	elem->nomatch = (flags >> 16) & IPSET_FLAG_NOMATCH;
6eb4c7e9 JK	70	}
6eb4c7e9 JK	71
8dea982a	72	static void
5d50e1d8	73	hash_net4_data_reset_flags(struct hash_net4_elem elem, u8 flags)
2a7cef2a	74	{
5d50e1d8	75	swap(*flags, elem->nomatch);
b3837029 JK	76	}
b3837029 JK	77
8dea982a	78	static void
b3837029 JK	79	hash_net4_data_netmask(struct hash_net4_elem *elem, u8 cidr)
	80	{
	81	elem->ip &= ip_set_netmask(cidr);
	82	elem->cidr = cidr;
	83	}
	84
b3837029 JK	85	static bool
	86	hash_net4_data_list(struct sk_buff skb, const struct hash_net4_elem data)
	87	{
2a7cef2a JK	88	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
2a7cef2a JK	89
7cf7899d DM	90	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip) \|\|
	91	nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr) \|\|
	92	(flags &&
	93	nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
	94	goto nla_put_failure;
728a7e69	95	return false;
b3837029 JK	96
b3837029 JK	97	nla_put_failure:
728a7e69	98	return true;
b3837029 JK	99	}
b3837029 JK	100
8dea982a	101	static void
5d50e1d8 JK	102	hash_net4_data_next(struct hash_net4_elem *next,
5d50e1d8 JK	103	const struct hash_net4_elem *d)
b3837029	104	{
5d50e1d8	105	next->ip = d->ip;
b3837029 JK	106	}
b3837029 JK	107
5d50e1d8	108	#define MTYPE hash_net4
b3837029	109	#define HOST_MASK 32
5d50e1d8	110	#include "ip_set_hash_gen.h"
3d14b171	111
b3837029 JK	112	static int
b3837029 JK	113	hash_net4_kadt(struct ip_set set, const struct sk_buff skb,
b66554cf	114	const struct xt_action_param *par,
5d50e1d8	115	enum ipset_adt adt, struct ip_set_adt_opt *opt)
b3837029	116	{
21956ab2	117	const struct hash_net4 *h = set->data;
b3837029	118	ipset_adtfn adtfn = set->variant->adt[adt];
5d50e1d8	119	struct hash_net4_elem e = {
f690cbae	120	.cidr = INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
9b03a5ef	121	};
ca134ce8	122	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
b3837029	123
5d50e1d8	124	if (e.cidr == 0)
b3837029 JK	125	return -EINVAL;
b3837029 JK	126	if (adt == IPSET_TEST)
5d50e1d8	127	e.cidr = HOST_MASK;
b3837029	128
5d50e1d8 JK	129	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip);
5d50e1d8 JK	130	e.ip &= ip_set_netmask(e.cidr);
b3837029	131
5d50e1d8	132	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
b3837029 JK	133	}
	134
	135	static int
	136	hash_net4_uadt(struct ip_set set, struct nlattr tb[],
3d14b171	137	enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
b3837029	138	{
5e29dc36	139	struct hash_net4 *h = set->data;
b3837029	140	ipset_adtfn adtfn = set->variant->adt[adt];
5d50e1d8	141	struct hash_net4_elem e = { .cidr = HOST_MASK };
ca134ce8	142	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
5e29dc36	143	u32 ip = 0, ip_to = 0, i = 0;
b3837029 JK	144	int ret;
b3837029 JK	145
a212e08e SP	146	if (tb[IPSET_ATTR_LINENO])
	147	*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
	148
b3837029	149	if (unlikely(!tb[IPSET_ATTR_IP] \|\|
7dd37bc8	150	!ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS)))
b3837029 JK	151	return -IPSET_ERR_PROTOCOL;
b3837029 JK	152
8e55d2e5 SP	153	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP], &ip);
	154	if (ret)
	155	return ret;
	156
	157	ret = ip_set_get_extensions(set, tb, &ext);
b3837029 JK	158	if (ret)
	159	return ret;
	160
d0d9e0a5	161	if (tb[IPSET_ATTR_CIDR]) {
5d50e1d8 JK	162	e.cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
5d50e1d8 JK	163	if (!e.cidr \|\| e.cidr > HOST_MASK)
d0d9e0a5 JK	164	return -IPSET_ERR_INVALID_CIDR;
d0d9e0a5 JK	165	}
b3837029	166
43c56e59	167	if (tb[IPSET_ATTR_CADT_FLAGS]) {
2a7cef2a	168	u32 cadt_flags = ip_set_get_h32(tb[IPSET_ATTR_CADT_FLAGS]);
ca0f6a5c	169
2a7cef2a	170	if (cadt_flags & IPSET_FLAG_NOMATCH)
43c56e59	171	flags \|= (IPSET_FLAG_NOMATCH << 16);
2a7cef2a JK	172	}
2a7cef2a JK	173
d0d9e0a5	174	if (adt == IPSET_TEST \|\| !tb[IPSET_ATTR_IP_TO]) {
5d50e1d8 JK	175	e.ip = htonl(ip & ip_set_hostmask(e.cidr));
5d50e1d8 JK	176	ret = adtfn(set, &e, &ext, &ext, flags);
ca0f6a5c	177	return ip_set_enomatch(ret, flags, adt, set) ? -ret :
43c56e59	178	ip_set_eexist(ret, flags) ? 0 : ret;
d0d9e0a5	179	}
b3837029	180
d0d9e0a5 JK	181	ip_to = ip;
	182	if (tb[IPSET_ATTR_IP_TO]) {
	183	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP_TO], &ip_to);
	184	if (ret)
	185	return ret;
	186	if (ip_to < ip)
	187	swap(ip, ip_to);
	188	if (ip + UINT_MAX == ip_to)
	189	return -IPSET_ERR_HASH_RANGE;
	190	}
5f7b51bf	191
d0d9e0a5	192	if (retried)
6e27c9b4	193	ip = ntohl(h->next.ip);
0b8d9073	194	do {
5e29dc36	195	i++;
5d50e1d8	196	e.ip = htonl(ip);
5e29dc36 JK	197	if (i > IPSET_MAX_RANGE) {
	198	hash_net4_data_next(&h->next, &e);
	199	return -ERANGE;
	200	}
0b8d9073	201	ip = ip_set_range_to_cidr(ip, ip_to, &e.cidr);
5d50e1d8	202	ret = adtfn(set, &e, &ext, &ext, flags);
d0d9e0a5 JK	203	if (ret && !ip_set_eexist(ret, flags))
d0d9e0a5 JK	204	return ret;
ca0f6a5c JK	205
ca0f6a5c JK	206	ret = 0;
0b8d9073	207	} while (ip++ < ip_to);
d0d9e0a5	208	return ret;
b3837029 JK	209	}
b3837029 JK	210
03c8b234	211	/* IPv6 variant */
b3837029 JK	212
	213	struct hash_net6_elem {
	214	union nf_inet_addr ip;
	215	u16 padding0;
2a7cef2a	216	u8 nomatch;
b3837029 JK	217	u8 cidr;
	218	};
	219
5d50e1d8 JK	220	/* Common functions */
5d50e1d8 JK	221
8dea982a	222	static bool
b3837029	223	hash_net6_data_equal(const struct hash_net6_elem *ip1,
89dc79b7 JK	224	const struct hash_net6_elem *ip2,
89dc79b7 JK	225	u32 *multi)
b3837029	226	{
29e3b160	227	return ipv6_addr_equal(&ip1->ip.in6, &ip2->ip.in6) &&
b3837029 JK	228	ip1->cidr == ip2->cidr;
	229	}
	230
8dea982a	231	static int
5d50e1d8	232	hash_net6_do_data_match(const struct hash_net6_elem *elem)
2a7cef2a	233	{
5d50e1d8	234	return elem->nomatch ? -ENOTEMPTY : 1;
6eb4c7e9 JK	235	}
6eb4c7e9 JK	236
8dea982a	237	static void
5d50e1d8	238	hash_net6_data_set_flags(struct hash_net6_elem *elem, u32 flags)
2a7cef2a	239	{
5d50e1d8	240	elem->nomatch = (flags >> 16) & IPSET_FLAG_NOMATCH;
b3837029 JK	241	}
b3837029 JK	242
8dea982a	243	static void
5d50e1d8	244	hash_net6_data_reset_flags(struct hash_net6_elem elem, u8 flags)
b3837029	245	{
5d50e1d8	246	swap(*flags, elem->nomatch);
b3837029 JK	247	}
b3837029 JK	248
8dea982a	249	static void
b3837029 JK	250	hash_net6_data_netmask(struct hash_net6_elem *elem, u8 cidr)
	251	{
	252	ip6_netmask(&elem->ip, cidr);
	253	elem->cidr = cidr;
	254	}
	255
	256	static bool
	257	hash_net6_data_list(struct sk_buff skb, const struct hash_net6_elem data)
	258	{
2a7cef2a JK	259	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
2a7cef2a JK	260
7cf7899d DM	261	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip.in6) \|\|
	262	nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr) \|\|
	263	(flags &&
	264	nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
	265	goto nla_put_failure;
728a7e69	266	return false;
b3837029 JK	267
b3837029 JK	268	nla_put_failure:
728a7e69	269	return true;
b3837029 JK	270	}
b3837029 JK	271
8dea982a	272	static void
21956ab2	273	hash_net6_data_next(struct hash_net6_elem *next,
5d50e1d8	274	const struct hash_net6_elem *d)
b3837029	275	{
b3837029 JK	276	}
b3837029 JK	277
5d50e1d8	278	#undef MTYPE
b3837029 JK	279	#undef HOST_MASK
b3837029 JK	280
5d50e1d8	281	#define MTYPE hash_net6
b3837029	282	#define HOST_MASK 128
5d50e1d8 JK	283	#define IP_SET_EMIT_CREATE
5d50e1d8 JK	284	#include "ip_set_hash_gen.h"
3d14b171	285
b3837029 JK	286	static int
b3837029 JK	287	hash_net6_kadt(struct ip_set set, const struct sk_buff skb,
b66554cf	288	const struct xt_action_param *par,
5d50e1d8	289	enum ipset_adt adt, struct ip_set_adt_opt *opt)
b3837029	290	{
21956ab2	291	const struct hash_net6 *h = set->data;
b3837029	292	ipset_adtfn adtfn = set->variant->adt[adt];
5d50e1d8	293	struct hash_net6_elem e = {
f690cbae	294	.cidr = INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
9b03a5ef	295	};
ca134ce8	296	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
b3837029	297
5d50e1d8	298	if (e.cidr == 0)
b3837029 JK	299	return -EINVAL;
b3837029 JK	300	if (adt == IPSET_TEST)
5d50e1d8	301	e.cidr = HOST_MASK;
b3837029	302
5d50e1d8 JK	303	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip.in6);
5d50e1d8 JK	304	ip6_netmask(&e.ip, e.cidr);
b3837029	305
5d50e1d8	306	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
b3837029 JK	307	}
	308
	309	static int
	310	hash_net6_uadt(struct ip_set set, struct nlattr tb[],
3d14b171	311	enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
b3837029	312	{
b3837029	313	ipset_adtfn adtfn = set->variant->adt[adt];
5d50e1d8	314	struct hash_net6_elem e = { .cidr = HOST_MASK };
ca134ce8	315	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
b3837029 JK	316	int ret;
b3837029 JK	317
a212e08e SP	318	if (tb[IPSET_ATTR_LINENO])
	319	*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
	320
b3837029	321	if (unlikely(!tb[IPSET_ATTR_IP] \|\|
7dd37bc8	322	!ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS)))
b3837029	323	return -IPSET_ERR_PROTOCOL;
d0d9e0a5 JK	324	if (unlikely(tb[IPSET_ATTR_IP_TO]))
d0d9e0a5 JK	325	return -IPSET_ERR_HASH_RANGE_UNSUPPORTED;
b3837029	326
8e55d2e5 SP	327	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip);
	328	if (ret)
	329	return ret;
	330
	331	ret = ip_set_get_extensions(set, tb, &ext);
b3837029 JK	332	if (ret)
	333	return ret;
	334
aff22758	335	if (tb[IPSET_ATTR_CIDR]) {
5d50e1d8	336	e.cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
aff22758 SP	337	if (!e.cidr \|\| e.cidr > HOST_MASK)
	338	return -IPSET_ERR_INVALID_CIDR;
	339	}
b3837029	340
5d50e1d8	341	ip6_netmask(&e.ip, e.cidr);
b3837029	342
43c56e59	343	if (tb[IPSET_ATTR_CADT_FLAGS]) {
2a7cef2a	344	u32 cadt_flags = ip_set_get_h32(tb[IPSET_ATTR_CADT_FLAGS]);
ca0f6a5c	345
2a7cef2a	346	if (cadt_flags & IPSET_FLAG_NOMATCH)
43c56e59	347	flags \|= (IPSET_FLAG_NOMATCH << 16);
2a7cef2a JK	348	}
2a7cef2a JK	349
5d50e1d8	350	ret = adtfn(set, &e, &ext, &ext, flags);
b3837029	351
0f1799ba	352	return ip_set_enomatch(ret, flags, adt, set) ? -ret :
43c56e59	353	ip_set_eexist(ret, flags) ? 0 : ret;
b3837029 JK	354	}
b3837029 JK	355
b3837029 JK	356	static struct ip_set_type hash_net_type __read_mostly = {
	357	.name = "hash:net",
	358	.protocol = IPSET_PROTOCOL,
3e0304a5	359	.features = IPSET_TYPE_IP \| IPSET_TYPE_NOMATCH,
b3837029	360	.dimension = IPSET_DIM_ONE,
c15f1c83	361	.family = NFPROTO_UNSPEC,
35b8dcf8 JK	362	.revision_min = IPSET_TYPE_REV_MIN,
35b8dcf8 JK	363	.revision_max = IPSET_TYPE_REV_MAX,
ccf0a4b7	364	.create_flags[IPSET_TYPE_REV_MAX] = IPSET_CREATE_FLAG_BUCKETSIZE,
b3837029 JK	365	.create = hash_net_create,
	366	.create_policy = {
	367	[IPSET_ATTR_HASHSIZE] = { .type = NLA_U32 },
	368	[IPSET_ATTR_MAXELEM] = { .type = NLA_U32 },
3976ca10	369	[IPSET_ATTR_INITVAL] = { .type = NLA_U32 },
ccf0a4b7	370	[IPSET_ATTR_BUCKETSIZE] = { .type = NLA_U8 },
b3837029 JK	371	[IPSET_ATTR_RESIZE] = { .type = NLA_U8 },
b3837029 JK	372	[IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 },
5d50e1d8	373	[IPSET_ATTR_CADT_FLAGS] = { .type = NLA_U32 },
b3837029 JK	374	},
	375	.adt_policy = {
	376	[IPSET_ATTR_IP] = { .type = NLA_NESTED },
d0d9e0a5	377	[IPSET_ATTR_IP_TO] = { .type = NLA_NESTED },
b3837029 JK	378	[IPSET_ATTR_CIDR] = { .type = NLA_U8 },
b3837029 JK	379	[IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 },
12899756	380	[IPSET_ATTR_LINENO] = { .type = NLA_U32 },
2a7cef2a	381	[IPSET_ATTR_CADT_FLAGS] = { .type = NLA_U32 },
00d71b27 JK	382	[IPSET_ATTR_BYTES] = { .type = NLA_U64 },
00d71b27 JK	383	[IPSET_ATTR_PACKETS] = { .type = NLA_U64 },
03726186 SP	384	[IPSET_ATTR_COMMENT] = { .type = NLA_NUL_STRING,
03726186 SP	385	.len = IPSET_MAX_COMMENT_SIZE },
af331419 AD	386	[IPSET_ATTR_SKBMARK] = { .type = NLA_U64 },
	387	[IPSET_ATTR_SKBPRIO] = { .type = NLA_U32 },
	388	[IPSET_ATTR_SKBQUEUE] = { .type = NLA_U16 },
b3837029 JK	389	},
	390	.me = THIS_MODULE,
	391	};
	392
	393	static int __init
	394	hash_net_init(void)
	395	{
	396	return ip_set_type_register(&hash_net_type);
	397	}
	398
	399	static void __exit
	400	hash_net_fini(void)
	401	{
18f84d41	402	rcu_barrier();
b3837029 JK	403	ip_set_type_unregister(&hash_net_type);
	404	}
	405
	406	module_init(hash_net_init);
	407	module_exit(hash_net_fini);