[mirror_ubuntu-bionic-kernel.git] / net / netfilter / ipset / ip_set_hash_ipport.c

/* Copyright (C) 2003-2011 Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

/* Kernel module implementing an IP set type: the hash:ip,port 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 <net/tcp.h>

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

#define REVISION_MIN	0
#define REVISION_MAX	1 /* SCTP and UDPLITE support added */

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>");
IP_SET_MODULE_DESC("hash:ip,port", REVISION_MIN, REVISION_MAX);
MODULE_ALIAS("ip_set_hash:ip,port");

/* Type specific function prefix */
#define TYPE		hash_ipport

static bool
hash_ipport_same_set(const struct ip_set *a, const struct ip_set *b);

#define hash_ipport4_same_set	hash_ipport_same_set
#define hash_ipport6_same_set	hash_ipport_same_set

/* The type variant functions: IPv4 */

/* Member elements without timeout */
struct hash_ipport4_elem {
	__be32 ip;
	__be16 port;
	u8 proto;
	u8 padding;
};

/* Member elements with timeout support */
struct hash_ipport4_telem {
	__be32 ip;
	__be16 port;
	u8 proto;
	u8 padding;
	unsigned long timeout;
};

static inline bool
hash_ipport4_data_equal(const struct hash_ipport4_elem *ip1,
			const struct hash_ipport4_elem *ip2,
			u32 *multi)
{
	return ip1->ip == ip2->ip &&
	       ip1->port == ip2->port &&
	       ip1->proto == ip2->proto;
}

static inline bool
hash_ipport4_data_isnull(const struct hash_ipport4_elem *elem)
{
	return elem->proto == 0;
}

static inline void
hash_ipport4_data_copy(struct hash_ipport4_elem *dst,
		       const struct hash_ipport4_elem *src)
{
	dst->ip = src->ip;
	dst->port = src->port;
	dst->proto = src->proto;
}

static inline void
hash_ipport4_data_zero_out(struct hash_ipport4_elem *elem)
{
	elem->proto = 0;
}

static bool
hash_ipport4_data_list(struct sk_buff *skb,
		       const struct hash_ipport4_elem *data)
{
	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip) ||
	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto))
		goto nla_put_failure;
	return 0;

nla_put_failure:
	return 1;
}

static bool
hash_ipport4_data_tlist(struct sk_buff *skb,
			const struct hash_ipport4_elem *data)
{
	const struct hash_ipport4_telem *tdata =
		(const struct hash_ipport4_telem *)data;

	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, tdata->ip) ||
	    nla_put_net16(skb, IPSET_ATTR_PORT, tdata->port) ||
	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
	    nla_put_net32(skb, IPSET_ATTR_TIMEOUT,
			  htonl(ip_set_timeout_get(tdata->timeout))))
		goto nla_put_failure;
	return 0;

nla_put_failure:
	return 1;
}

#define PF		4
#define HOST_MASK	32
#include <linux/netfilter/ipset/ip_set_ahash.h>

static inline void
hash_ipport4_data_next(struct ip_set_hash *h,
		       const struct hash_ipport4_elem *d)
{
	h->next.ip = d->ip;
	h->next.port = d->port;
}

static int
hash_ipport4_kadt(struct ip_set *set, const struct sk_buff *skb,
		  const struct xt_action_param *par,
		  enum ipset_adt adt, const struct ip_set_adt_opt *opt)
{
	const struct ip_set_hash *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_ipport4_elem data = { };

	if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
				 &data.port, &data.proto))
		return -EINVAL;

	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &data.ip);

	return adtfn(set, &data, opt_timeout(opt, h), opt->cmdflags);
}

static int
hash_ipport4_uadt(struct ip_set *set, struct nlattr *tb[],
		  enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
{
	const struct ip_set_hash *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_ipport4_elem data = { };
	u32 ip, ip_to = 0, p = 0, port, port_to;
	u32 timeout = h->timeout;
	bool with_ports = false;
	int ret;

	if (unlikely(!tb[IPSET_ATTR_IP] ||
		     !ip_set_attr_netorder(tb, IPSET_ATTR_PORT) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT)))
		return -IPSET_ERR_PROTOCOL;

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

	ret = ip_set_get_ipaddr4(tb[IPSET_ATTR_IP], &data.ip);
	if (ret)
		return ret;

	if (tb[IPSET_ATTR_PORT])
		data.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
	else
		return -IPSET_ERR_PROTOCOL;

	if (tb[IPSET_ATTR_PROTO]) {
		data.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
		with_ports = ip_set_proto_with_ports(data.proto);

		if (data.proto == 0)
			return -IPSET_ERR_INVALID_PROTO;
	} else
		return -IPSET_ERR_MISSING_PROTO;

	if (!(with_ports || data.proto == IPPROTO_ICMP))
		data.port = 0;

	if (tb[IPSET_ATTR_TIMEOUT]) {
		if (!with_timeout(h->timeout))
			return -IPSET_ERR_TIMEOUT;
		timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);
	}

	if (adt == IPSET_TEST ||
	    !(tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_CIDR] ||
	      tb[IPSET_ATTR_PORT_TO])) {
		ret = adtfn(set, &data, timeout, flags);
		return ip_set_eexist(ret, flags) ? 0 : ret;
	}

	ip = ntohl(data.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 > ip_to)
			swap(ip, ip_to);
	} else if (tb[IPSET_ATTR_CIDR]) {
		u8 cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);

		if (!cidr || cidr > 32)
			return -IPSET_ERR_INVALID_CIDR;
		ip_set_mask_from_to(ip, ip_to, cidr);
	} else
		ip_to = ip;

	port_to = port = ntohs(data.port);
	if (with_ports && tb[IPSET_ATTR_PORT_TO]) {
		port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
		if (port > port_to)
			swap(port, port_to);
	}

	if (retried)
		ip = ntohl(h->next.ip);
	for (; !before(ip_to, ip); ip++) {
		p = retried && ip == ntohl(h->next.ip) ? ntohs(h->next.port)
						       : port;
		for (; p <= port_to; p++) {
			data.ip = htonl(ip);
			data.port = htons(p);
			ret = adtfn(set, &data, timeout, flags);

			if (ret && !ip_set_eexist(ret, flags))
				return ret;
			else
				ret = 0;
		}
	}
	return ret;
}

static bool
hash_ipport_same_set(const struct ip_set *a, const struct ip_set *b)
{
	const struct ip_set_hash *x = a->data;
	const struct ip_set_hash *y = b->data;

	/* Resizing changes htable_bits, so we ignore it */
	return x->maxelem == y->maxelem &&
	       x->timeout == y->timeout;
}

/* The type variant functions: IPv6 */

struct hash_ipport6_elem {
	union nf_inet_addr ip;
	__be16 port;
	u8 proto;
	u8 padding;
};

struct hash_ipport6_telem {
	union nf_inet_addr ip;
	__be16 port;
	u8 proto;
	u8 padding;
	unsigned long timeout;
};

static inline bool
hash_ipport6_data_equal(const struct hash_ipport6_elem *ip1,
			const struct hash_ipport6_elem *ip2,
			u32 *multi)
{
	return ipv6_addr_cmp(&ip1->ip.in6, &ip2->ip.in6) == 0 &&
	       ip1->port == ip2->port &&
	       ip1->proto == ip2->proto;
}

static inline bool
hash_ipport6_data_isnull(const struct hash_ipport6_elem *elem)
{
	return elem->proto == 0;
}

static inline void
hash_ipport6_data_copy(struct hash_ipport6_elem *dst,
		       const struct hash_ipport6_elem *src)
{
	memcpy(dst, src, sizeof(*dst));
}

static inline void
hash_ipport6_data_zero_out(struct hash_ipport6_elem *elem)
{
	elem->proto = 0;
}

static bool
hash_ipport6_data_list(struct sk_buff *skb,
		       const struct hash_ipport6_elem *data)
{
	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip.in6) ||
	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto))
		goto nla_put_failure;
	return 0;

nla_put_failure:
	return 1;
}

static bool
hash_ipport6_data_tlist(struct sk_buff *skb,
			const struct hash_ipport6_elem *data)
{
	const struct hash_ipport6_telem *e =
		(const struct hash_ipport6_telem *)data;

	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &e->ip.in6) ||
	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
	    nla_put_net32(skb, IPSET_ATTR_TIMEOUT,
			  htonl(ip_set_timeout_get(e->timeout))))
		goto nla_put_failure;
	return 0;

nla_put_failure:
	return 1;
}

#undef PF
#undef HOST_MASK

#define PF		6
#define HOST_MASK	128
#include <linux/netfilter/ipset/ip_set_ahash.h>

static inline void
hash_ipport6_data_next(struct ip_set_hash *h,
		       const struct hash_ipport6_elem *d)
{
	h->next.port = d->port;
}

static int
hash_ipport6_kadt(struct ip_set *set, const struct sk_buff *skb,
		  const struct xt_action_param *par,
		  enum ipset_adt adt, const struct ip_set_adt_opt *opt)
{
	const struct ip_set_hash *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_ipport6_elem data = { };

	if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
				 &data.port, &data.proto))
		return -EINVAL;

	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &data.ip.in6);

	return adtfn(set, &data, opt_timeout(opt, h), opt->cmdflags);
}

static int
hash_ipport6_uadt(struct ip_set *set, struct nlattr *tb[],
		  enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
{
	const struct ip_set_hash *h = set->data;
	ipset_adtfn adtfn = set->variant->adt[adt];
	struct hash_ipport6_elem data = { };
	u32 port, port_to;
	u32 timeout = h->timeout;
	bool with_ports = false;
	int ret;

	if (unlikely(!tb[IPSET_ATTR_IP] ||
		     !ip_set_attr_netorder(tb, IPSET_ATTR_PORT) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT) ||
		     tb[IPSET_ATTR_IP_TO] ||
		     tb[IPSET_ATTR_CIDR]))
		return -IPSET_ERR_PROTOCOL;

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

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

	if (tb[IPSET_ATTR_PORT])
		data.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
	else
		return -IPSET_ERR_PROTOCOL;

	if (tb[IPSET_ATTR_PROTO]) {
		data.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
		with_ports = ip_set_proto_with_ports(data.proto);

		if (data.proto == 0)
			return -IPSET_ERR_INVALID_PROTO;
	} else
		return -IPSET_ERR_MISSING_PROTO;

	if (!(with_ports || data.proto == IPPROTO_ICMPV6))
		data.port = 0;

	if (tb[IPSET_ATTR_TIMEOUT]) {
		if (!with_timeout(h->timeout))
			return -IPSET_ERR_TIMEOUT;
		timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);
	}

	if (adt == IPSET_TEST || !with_ports || !tb[IPSET_ATTR_PORT_TO]) {
		ret = adtfn(set, &data, timeout, flags);
		return ip_set_eexist(ret, flags) ? 0 : ret;
	}

	port = ntohs(data.port);
	port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
	if (port > port_to)
		swap(port, port_to);

	if (retried)
		port = ntohs(h->next.port);
	for (; port <= port_to; port++) {
		data.port = htons(port);
		ret = adtfn(set, &data, timeout, flags);

		if (ret && !ip_set_eexist(ret, flags))
			return ret;
		else
			ret = 0;
	}
	return ret;
}

/* Create hash:ip type of sets */

static int
hash_ipport_create(struct ip_set *set, struct nlattr *tb[], u32 flags)
{
	struct ip_set_hash *h;
	u32 hashsize = IPSET_DEFAULT_HASHSIZE, maxelem = IPSET_DEFAULT_MAXELEM;
	u8 hbits;
	size_t hsize;

	if (!(set->family == NFPROTO_IPV4 || set->family == NFPROTO_IPV6))
		return -IPSET_ERR_INVALID_FAMILY;

	if (unlikely(!ip_set_optattr_netorder(tb, IPSET_ATTR_HASHSIZE) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_MAXELEM) ||
		     !ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT)))
		return -IPSET_ERR_PROTOCOL;

	if (tb[IPSET_ATTR_HASHSIZE]) {
		hashsize = ip_set_get_h32(tb[IPSET_ATTR_HASHSIZE]);
		if (hashsize < IPSET_MIMINAL_HASHSIZE)
			hashsize = IPSET_MIMINAL_HASHSIZE;
	}

	if (tb[IPSET_ATTR_MAXELEM])
		maxelem = ip_set_get_h32(tb[IPSET_ATTR_MAXELEM]);

	h = kzalloc(sizeof(*h), GFP_KERNEL);
	if (!h)
		return -ENOMEM;

	h->maxelem = maxelem;
	get_random_bytes(&h->initval, sizeof(h->initval));
	h->timeout = IPSET_NO_TIMEOUT;

	hbits = htable_bits(hashsize);
	hsize = htable_size(hbits);
	if (hsize == 0) {
		kfree(h);
		return -ENOMEM;
	}
	h->table = ip_set_alloc(hsize);
	if (!h->table) {
		kfree(h);
		return -ENOMEM;
	}
	h->table->htable_bits = hbits;

	set->data = h;

	if (tb[IPSET_ATTR_TIMEOUT]) {
		h->timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);

		set->variant = set->family == NFPROTO_IPV4
			? &hash_ipport4_tvariant : &hash_ipport6_tvariant;

		if (set->family == NFPROTO_IPV4)
			hash_ipport4_gc_init(set);
		else
			hash_ipport6_gc_init(set);
	} else {
		set->variant = set->family == NFPROTO_IPV4
			? &hash_ipport4_variant : &hash_ipport6_variant;
	}

	pr_debug("create %s hashsize %u (%u) maxelem %u: %p(%p)\n",
		 set->name, jhash_size(h->table->htable_bits),
		 h->table->htable_bits, h->maxelem, set->data, h->table);

	return 0;
}

static struct ip_set_type hash_ipport_type __read_mostly = {
	.name		= "hash:ip,port",
	.protocol	= IPSET_PROTOCOL,
	.features	= IPSET_TYPE_IP | IPSET_TYPE_PORT,
	.dimension	= IPSET_DIM_TWO,
	.family		= NFPROTO_UNSPEC,
	.revision_min	= REVISION_MIN,
	.revision_max	= REVISION_MAX,
	.create		= hash_ipport_create,
	.create_policy	= {
		[IPSET_ATTR_HASHSIZE]	= { .type = NLA_U32 },
		[IPSET_ATTR_MAXELEM]	= { .type = NLA_U32 },
		[IPSET_ATTR_PROBES]	= { .type = NLA_U8 },
		[IPSET_ATTR_RESIZE]	= { .type = NLA_U8  },
		[IPSET_ATTR_PROTO]	= { .type = NLA_U8 },
		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
	},
	.adt_policy	= {
		[IPSET_ATTR_IP]		= { .type = NLA_NESTED },
		[IPSET_ATTR_IP_TO]	= { .type = NLA_NESTED },
		[IPSET_ATTR_PORT]	= { .type = NLA_U16 },
		[IPSET_ATTR_PORT_TO]	= { .type = NLA_U16 },
		[IPSET_ATTR_CIDR]	= { .type = NLA_U8 },
		[IPSET_ATTR_PROTO]	= { .type = NLA_U8 },
		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
		[IPSET_ATTR_LINENO]	= { .type = NLA_U32 },
	},
	.me		= THIS_MODULE,
};

static int __init
hash_ipport_init(void)
{
	return ip_set_type_register(&hash_ipport_type);
}

static void __exit
hash_ipport_fini(void)
{
	ip_set_type_unregister(&hash_ipport_type);
}

module_init(hash_ipport_init);
module_exit(hash_ipport_fini);
Commit	Line	Data
07896ed3 JK	1	/* Copyright (C) 2003-2011 Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>
	2	*
	3	* This program is free software; you can redistribute it and/or modify
	4	* it under the terms of the GNU General Public License version 2 as
	5	* published by the Free Software Foundation.
	6	*/
	7
	8	/* Kernel module implementing an IP set type: the hash:ip,port type */
	9
	10	#include <linux/jhash.h>
	11	#include <linux/module.h>
	12	#include <linux/ip.h>
	13	#include <linux/skbuff.h>
	14	#include <linux/errno.h>
07896ed3 JK	15	#include <linux/random.h>
	16	#include <net/ip.h>
	17	#include <net/ipv6.h>
	18	#include <net/netlink.h>
	19	#include <net/tcp.h>
	20
	21	#include <linux/netfilter.h>
	22	#include <linux/netfilter/ipset/pfxlen.h>
	23	#include <linux/netfilter/ipset/ip_set.h>
	24	#include <linux/netfilter/ipset/ip_set_timeout.h>
	25	#include <linux/netfilter/ipset/ip_set_getport.h>
	26	#include <linux/netfilter/ipset/ip_set_hash.h>
	27
10111a6e JK	28	#define REVISION_MIN 0
	29	#define REVISION_MAX 1 /* SCTP and UDPLITE support added */
	30
07896ed3 JK	31	MODULE_LICENSE("GPL");
07896ed3 JK	32	MODULE_AUTHOR("Jozsef Kadlecsik <kadlec@blackhole.kfki.hu>");
10111a6e	33	IP_SET_MODULE_DESC("hash:ip,port", REVISION_MIN, REVISION_MAX);
07896ed3 JK	34	MODULE_ALIAS("ip_set_hash:ip,port");
	35
	36	/* Type specific function prefix */
	37	#define TYPE hash_ipport
	38
	39	static bool
	40	hash_ipport_same_set(const struct ip_set a, const struct ip_set b);
	41
	42	#define hash_ipport4_same_set hash_ipport_same_set
	43	#define hash_ipport6_same_set hash_ipport_same_set
	44
	45	/* The type variant functions: IPv4 */
	46
	47	/* Member elements without timeout */
	48	struct hash_ipport4_elem {
	49	__be32 ip;
	50	__be16 port;
	51	u8 proto;
	52	u8 padding;
	53	};
	54
	55	/* Member elements with timeout support */
	56	struct hash_ipport4_telem {
	57	__be32 ip;
	58	__be16 port;
	59	u8 proto;
	60	u8 padding;
	61	unsigned long timeout;
	62	};
	63
	64	static inline bool
	65	hash_ipport4_data_equal(const struct hash_ipport4_elem *ip1,
89dc79b7 JK	66	const struct hash_ipport4_elem *ip2,
89dc79b7 JK	67	u32 *multi)
07896ed3 JK	68	{
	69	return ip1->ip == ip2->ip &&
	70	ip1->port == ip2->port &&
	71	ip1->proto == ip2->proto;
	72	}
	73
	74	static inline bool
	75	hash_ipport4_data_isnull(const struct hash_ipport4_elem *elem)
	76	{
	77	return elem->proto == 0;
	78	}
	79
	80	static inline void
	81	hash_ipport4_data_copy(struct hash_ipport4_elem *dst,
	82	const struct hash_ipport4_elem *src)
	83	{
	84	dst->ip = src->ip;
	85	dst->port = src->port;
	86	dst->proto = src->proto;
	87	}
	88
	89	static inline void
	90	hash_ipport4_data_zero_out(struct hash_ipport4_elem *elem)
	91	{
	92	elem->proto = 0;
	93	}
	94
	95	static bool
	96	hash_ipport4_data_list(struct sk_buff *skb,
	97	const struct hash_ipport4_elem *data)
	98	{
7cf7899d DM	99	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip) \|\|
	100	nla_put_net16(skb, IPSET_ATTR_PORT, data->port) \|\|
	101	nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto))
	102	goto nla_put_failure;
07896ed3 JK	103	return 0;
	104
	105	nla_put_failure:
	106	return 1;
	107	}
	108
	109	static bool
	110	hash_ipport4_data_tlist(struct sk_buff *skb,
	111	const struct hash_ipport4_elem *data)
	112	{
	113	const struct hash_ipport4_telem *tdata =
	114	(const struct hash_ipport4_telem *)data;
	115
7cf7899d DM	116	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, tdata->ip) \|\|
	117	nla_put_net16(skb, IPSET_ATTR_PORT, tdata->port) \|\|
	118	nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) \|\|
	119	nla_put_net32(skb, IPSET_ATTR_TIMEOUT,
	120	htonl(ip_set_timeout_get(tdata->timeout))))
	121	goto nla_put_failure;
07896ed3 JK	122	return 0;
	123
	124	nla_put_failure:
	125	return 1;
	126	}
	127
	128	#define PF 4
	129	#define HOST_MASK 32
	130	#include <linux/netfilter/ipset/ip_set_ahash.h>
	131
3d14b171 JK	132	static inline void
	133	hash_ipport4_data_next(struct ip_set_hash *h,
	134	const struct hash_ipport4_elem *d)
	135	{
6e27c9b4 JK	136	h->next.ip = d->ip;
6e27c9b4 JK	137	h->next.port = d->port;
3d14b171 JK	138	}
3d14b171 JK	139
07896ed3 JK	140	static int
07896ed3 JK	141	hash_ipport4_kadt(struct ip_set set, const struct sk_buff skb,
b66554cf	142	const struct xt_action_param *par,
ac8cc925	143	enum ipset_adt adt, const struct ip_set_adt_opt *opt)
07896ed3 JK	144	{
	145	const struct ip_set_hash *h = set->data;
	146	ipset_adtfn adtfn = set->variant->adt[adt];
	147	struct hash_ipport4_elem data = { };
	148
ac8cc925	149	if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
07896ed3 JK	150	&data.port, &data.proto))
	151	return -EINVAL;
	152
ac8cc925	153	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &data.ip);
07896ed3	154
ac8cc925	155	return adtfn(set, &data, opt_timeout(opt, h), opt->cmdflags);
07896ed3 JK	156	}
	157
	158	static int
	159	hash_ipport4_uadt(struct ip_set set, struct nlattr tb[],
3d14b171	160	enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
07896ed3 JK	161	{
	162	const struct ip_set_hash *h = set->data;
	163	ipset_adtfn adtfn = set->variant->adt[adt];
	164	struct hash_ipport4_elem data = { };
648ae8e5	165	u32 ip, ip_to = 0, p = 0, port, port_to;
07896ed3	166	u32 timeout = h->timeout;
5e0c1eb7	167	bool with_ports = false;
07896ed3 JK	168	int ret;
	169
	170	if (unlikely(!tb[IPSET_ATTR_IP] \|\|
	171	!ip_set_attr_netorder(tb, IPSET_ATTR_PORT) \|\|
	172	!ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO) \|\|
	173	!ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT)))
	174	return -IPSET_ERR_PROTOCOL;
	175
	176	if (tb[IPSET_ATTR_LINENO])
	177	*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
	178
	179	ret = ip_set_get_ipaddr4(tb[IPSET_ATTR_IP], &data.ip);
	180	if (ret)
	181	return ret;
	182
	183	if (tb[IPSET_ATTR_PORT])
	184	data.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
	185	else
	186	return -IPSET_ERR_PROTOCOL;
	187
	188	if (tb[IPSET_ATTR_PROTO]) {
	189	data.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
5e0c1eb7	190	with_ports = ip_set_proto_with_ports(data.proto);
07896ed3 JK	191
	192	if (data.proto == 0)
	193	return -IPSET_ERR_INVALID_PROTO;
	194	} else
	195	return -IPSET_ERR_MISSING_PROTO;
	196
5e0c1eb7	197	if (!(with_ports \|\| data.proto == IPPROTO_ICMP))
07896ed3	198	data.port = 0;
07896ed3 JK	199
	200	if (tb[IPSET_ATTR_TIMEOUT]) {
	201	if (!with_timeout(h->timeout))
	202	return -IPSET_ERR_TIMEOUT;
	203	timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);
	204	}
	205
	206	if (adt == IPSET_TEST \|\|
07896ed3 JK	207	!(tb[IPSET_ATTR_IP_TO] \|\| tb[IPSET_ATTR_CIDR] \|\|
07896ed3 JK	208	tb[IPSET_ATTR_PORT_TO])) {
5416219e	209	ret = adtfn(set, &data, timeout, flags);
07896ed3 JK	210	return ip_set_eexist(ret, flags) ? 0 : ret;
	211	}
	212
	213	ip = ntohl(data.ip);
	214	if (tb[IPSET_ATTR_IP_TO]) {
	215	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP_TO], &ip_to);
	216	if (ret)
	217	return ret;
	218	if (ip > ip_to)
	219	swap(ip, ip_to);
	220	} else if (tb[IPSET_ATTR_CIDR]) {
	221	u8 cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
	222
b9fed748	223	if (!cidr \|\| cidr > 32)
07896ed3	224	return -IPSET_ERR_INVALID_CIDR;
e6146e86	225	ip_set_mask_from_to(ip, ip_to, cidr);
07896ed3 JK	226	} else
	227	ip_to = ip;
	228
5e0c1eb7 JK	229	port_to = port = ntohs(data.port);
5e0c1eb7 JK	230	if (with_ports && tb[IPSET_ATTR_PORT_TO]) {
07896ed3 JK	231	port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
	232	if (port > port_to)
	233	swap(port, port_to);
5e0c1eb7	234	}
07896ed3	235
3d14b171	236	if (retried)
6e27c9b4	237	ip = ntohl(h->next.ip);
3d14b171	238	for (; !before(ip_to, ip); ip++) {
6e27c9b4 JK	239	p = retried && ip == ntohl(h->next.ip) ? ntohs(h->next.port)
6e27c9b4 JK	240	: port;
3d14b171	241	for (; p <= port_to; p++) {
07896ed3 JK	242	data.ip = htonl(ip);
07896ed3 JK	243	data.port = htons(p);
5416219e	244	ret = adtfn(set, &data, timeout, flags);
07896ed3 JK	245
	246	if (ret && !ip_set_eexist(ret, flags))
	247	return ret;
	248	else
	249	ret = 0;
	250	}
3d14b171	251	}
07896ed3 JK	252	return ret;
	253	}
	254
	255	static bool
	256	hash_ipport_same_set(const struct ip_set a, const struct ip_set b)
	257	{
	258	const struct ip_set_hash *x = a->data;
	259	const struct ip_set_hash *y = b->data;
	260
	261	/* Resizing changes htable_bits, so we ignore it */
	262	return x->maxelem == y->maxelem &&
	263	x->timeout == y->timeout;
	264	}
	265
	266	/* The type variant functions: IPv6 */
	267
	268	struct hash_ipport6_elem {
	269	union nf_inet_addr ip;
	270	__be16 port;
	271	u8 proto;
	272	u8 padding;
	273	};
	274
	275	struct hash_ipport6_telem {
	276	union nf_inet_addr ip;
	277	__be16 port;
	278	u8 proto;
	279	u8 padding;
	280	unsigned long timeout;
	281	};
	282
	283	static inline bool
	284	hash_ipport6_data_equal(const struct hash_ipport6_elem *ip1,
89dc79b7 JK	285	const struct hash_ipport6_elem *ip2,
89dc79b7 JK	286	u32 *multi)
07896ed3 JK	287	{
	288	return ipv6_addr_cmp(&ip1->ip.in6, &ip2->ip.in6) == 0 &&
	289	ip1->port == ip2->port &&
	290	ip1->proto == ip2->proto;
	291	}
	292
	293	static inline bool
	294	hash_ipport6_data_isnull(const struct hash_ipport6_elem *elem)
	295	{
	296	return elem->proto == 0;
	297	}
	298
	299	static inline void
	300	hash_ipport6_data_copy(struct hash_ipport6_elem *dst,
	301	const struct hash_ipport6_elem *src)
	302	{
	303	memcpy(dst, src, sizeof(*dst));
	304	}
	305
	306	static inline void
	307	hash_ipport6_data_zero_out(struct hash_ipport6_elem *elem)
	308	{
	309	elem->proto = 0;
	310	}
	311
	312	static bool
	313	hash_ipport6_data_list(struct sk_buff *skb,
	314	const struct hash_ipport6_elem *data)
	315	{
7cf7899d DM	316	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip.in6) \|\|
	317	nla_put_net16(skb, IPSET_ATTR_PORT, data->port) \|\|
	318	nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto))
	319	goto nla_put_failure;
07896ed3 JK	320	return 0;
	321
	322	nla_put_failure:
	323	return 1;
	324	}
	325
	326	static bool
	327	hash_ipport6_data_tlist(struct sk_buff *skb,
	328	const struct hash_ipport6_elem *data)
	329	{
	330	const struct hash_ipport6_telem *e =
	331	(const struct hash_ipport6_telem *)data;
	332
7cf7899d DM	333	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &e->ip.in6) \|\|
	334	nla_put_net16(skb, IPSET_ATTR_PORT, data->port) \|\|
	335	nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) \|\|
	336	nla_put_net32(skb, IPSET_ATTR_TIMEOUT,
	337	htonl(ip_set_timeout_get(e->timeout))))
	338	goto nla_put_failure;
07896ed3 JK	339	return 0;
	340
	341	nla_put_failure:
	342	return 1;
	343	}
	344
	345	#undef PF
	346	#undef HOST_MASK
	347
	348	#define PF 6
	349	#define HOST_MASK 128
	350	#include <linux/netfilter/ipset/ip_set_ahash.h>
	351
3d14b171 JK	352	static inline void
	353	hash_ipport6_data_next(struct ip_set_hash *h,
	354	const struct hash_ipport6_elem *d)
	355	{
6e27c9b4	356	h->next.port = d->port;
3d14b171 JK	357	}
3d14b171 JK	358
07896ed3 JK	359	static int
07896ed3 JK	360	hash_ipport6_kadt(struct ip_set set, const struct sk_buff skb,
b66554cf	361	const struct xt_action_param *par,
ac8cc925	362	enum ipset_adt adt, const struct ip_set_adt_opt *opt)
07896ed3 JK	363	{
	364	const struct ip_set_hash *h = set->data;
	365	ipset_adtfn adtfn = set->variant->adt[adt];
	366	struct hash_ipport6_elem data = { };
	367
ac8cc925	368	if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
07896ed3 JK	369	&data.port, &data.proto))
	370	return -EINVAL;
	371
ac8cc925	372	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &data.ip.in6);
07896ed3	373
ac8cc925	374	return adtfn(set, &data, opt_timeout(opt, h), opt->cmdflags);
07896ed3 JK	375	}
	376
	377	static int
	378	hash_ipport6_uadt(struct ip_set set, struct nlattr tb[],
3d14b171	379	enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
07896ed3 JK	380	{
	381	const struct ip_set_hash *h = set->data;
	382	ipset_adtfn adtfn = set->variant->adt[adt];
	383	struct hash_ipport6_elem data = { };
	384	u32 port, port_to;
	385	u32 timeout = h->timeout;
5e0c1eb7	386	bool with_ports = false;
07896ed3 JK	387	int ret;
	388
	389	if (unlikely(!tb[IPSET_ATTR_IP] \|\|
	390	!ip_set_attr_netorder(tb, IPSET_ATTR_PORT) \|\|
	391	!ip_set_optattr_netorder(tb, IPSET_ATTR_PORT_TO) \|\|
	392	!ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT) \|\|
	393	tb[IPSET_ATTR_IP_TO] \|\|
	394	tb[IPSET_ATTR_CIDR]))
	395	return -IPSET_ERR_PROTOCOL;
	396
	397	if (tb[IPSET_ATTR_LINENO])
	398	*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
	399
	400	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &data.ip);
	401	if (ret)
	402	return ret;
	403
	404	if (tb[IPSET_ATTR_PORT])
	405	data.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
	406	else
	407	return -IPSET_ERR_PROTOCOL;
	408
	409	if (tb[IPSET_ATTR_PROTO]) {
	410	data.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
5e0c1eb7	411	with_ports = ip_set_proto_with_ports(data.proto);
07896ed3 JK	412
	413	if (data.proto == 0)
	414	return -IPSET_ERR_INVALID_PROTO;
	415	} else
	416	return -IPSET_ERR_MISSING_PROTO;
	417
5e0c1eb7	418	if (!(with_ports \|\| data.proto == IPPROTO_ICMPV6))
07896ed3	419	data.port = 0;
07896ed3 JK	420
	421	if (tb[IPSET_ATTR_TIMEOUT]) {
	422	if (!with_timeout(h->timeout))
	423	return -IPSET_ERR_TIMEOUT;
	424	timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);
	425	}
	426
5e0c1eb7	427	if (adt == IPSET_TEST \|\| !with_ports \|\| !tb[IPSET_ATTR_PORT_TO]) {
5416219e	428	ret = adtfn(set, &data, timeout, flags);
07896ed3 JK	429	return ip_set_eexist(ret, flags) ? 0 : ret;
	430	}
	431
	432	port = ntohs(data.port);
	433	port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
	434	if (port > port_to)
	435	swap(port, port_to);
	436
3d14b171	437	if (retried)
6e27c9b4	438	port = ntohs(h->next.port);
07896ed3 JK	439	for (; port <= port_to; port++) {
07896ed3 JK	440	data.port = htons(port);
5416219e	441	ret = adtfn(set, &data, timeout, flags);
07896ed3 JK	442
	443	if (ret && !ip_set_eexist(ret, flags))
	444	return ret;
	445	else
	446	ret = 0;
	447	}
	448	return ret;
	449	}
	450
	451	/* Create hash:ip type of sets */
	452
	453	static int
	454	hash_ipport_create(struct ip_set set, struct nlattr tb[], u32 flags)
	455	{
	456	struct ip_set_hash *h;
	457	u32 hashsize = IPSET_DEFAULT_HASHSIZE, maxelem = IPSET_DEFAULT_MAXELEM;
	458	u8 hbits;
26a5d3cc	459	size_t hsize;
07896ed3	460
c15f1c83	461	if (!(set->family == NFPROTO_IPV4 \|\| set->family == NFPROTO_IPV6))
07896ed3 JK	462	return -IPSET_ERR_INVALID_FAMILY;
	463
	464	if (unlikely(!ip_set_optattr_netorder(tb, IPSET_ATTR_HASHSIZE) \|\|
	465	!ip_set_optattr_netorder(tb, IPSET_ATTR_MAXELEM) \|\|
	466	!ip_set_optattr_netorder(tb, IPSET_ATTR_TIMEOUT)))
	467	return -IPSET_ERR_PROTOCOL;
	468
	469	if (tb[IPSET_ATTR_HASHSIZE]) {
	470	hashsize = ip_set_get_h32(tb[IPSET_ATTR_HASHSIZE]);
	471	if (hashsize < IPSET_MIMINAL_HASHSIZE)
	472	hashsize = IPSET_MIMINAL_HASHSIZE;
	473	}
	474
	475	if (tb[IPSET_ATTR_MAXELEM])
	476	maxelem = ip_set_get_h32(tb[IPSET_ATTR_MAXELEM]);
	477
	478	h = kzalloc(sizeof(*h), GFP_KERNEL);
	479	if (!h)
	480	return -ENOMEM;
	481
	482	h->maxelem = maxelem;
	483	get_random_bytes(&h->initval, sizeof(h->initval));
	484	h->timeout = IPSET_NO_TIMEOUT;
	485
	486	hbits = htable_bits(hashsize);
26a5d3cc JK	487	hsize = htable_size(hbits);
	488	if (hsize == 0) {
	489	kfree(h);
	490	return -ENOMEM;
	491	}
	492	h->table = ip_set_alloc(hsize);
07896ed3 JK	493	if (!h->table) {
	494	kfree(h);
	495	return -ENOMEM;
	496	}
	497	h->table->htable_bits = hbits;
	498
	499	set->data = h;
	500
	501	if (tb[IPSET_ATTR_TIMEOUT]) {
	502	h->timeout = ip_set_timeout_uget(tb[IPSET_ATTR_TIMEOUT]);
	503
c15f1c83	504	set->variant = set->family == NFPROTO_IPV4
07896ed3 JK	505	? &hash_ipport4_tvariant : &hash_ipport6_tvariant;
07896ed3 JK	506
c15f1c83	507	if (set->family == NFPROTO_IPV4)
07896ed3 JK	508	hash_ipport4_gc_init(set);
	509	else
	510	hash_ipport6_gc_init(set);
	511	} else {
c15f1c83	512	set->variant = set->family == NFPROTO_IPV4
07896ed3 JK	513	? &hash_ipport4_variant : &hash_ipport6_variant;
	514	}
	515
	516	pr_debug("create %s hashsize %u (%u) maxelem %u: %p(%p)\n",
	517	set->name, jhash_size(h->table->htable_bits),
	518	h->table->htable_bits, h->maxelem, set->data, h->table);
	519
	520	return 0;
	521	}
	522
	523	static struct ip_set_type hash_ipport_type __read_mostly = {
	524	.name = "hash:ip,port",
	525	.protocol = IPSET_PROTOCOL,
	526	.features = IPSET_TYPE_IP \| IPSET_TYPE_PORT,
	527	.dimension = IPSET_DIM_TWO,
c15f1c83	528	.family = NFPROTO_UNSPEC,
10111a6e JK	529	.revision_min = REVISION_MIN,
10111a6e JK	530	.revision_max = REVISION_MAX,
07896ed3 JK	531	.create = hash_ipport_create,
	532	.create_policy = {
	533	[IPSET_ATTR_HASHSIZE] = { .type = NLA_U32 },
	534	[IPSET_ATTR_MAXELEM] = { .type = NLA_U32 },
	535	[IPSET_ATTR_PROBES] = { .type = NLA_U8 },
	536	[IPSET_ATTR_RESIZE] = { .type = NLA_U8 },
	537	[IPSET_ATTR_PROTO] = { .type = NLA_U8 },
	538	[IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 },
	539	},
	540	.adt_policy = {
	541	[IPSET_ATTR_IP] = { .type = NLA_NESTED },
	542	[IPSET_ATTR_IP_TO] = { .type = NLA_NESTED },
	543	[IPSET_ATTR_PORT] = { .type = NLA_U16 },
	544	[IPSET_ATTR_PORT_TO] = { .type = NLA_U16 },
	545	[IPSET_ATTR_CIDR] = { .type = NLA_U8 },
	546	[IPSET_ATTR_PROTO] = { .type = NLA_U8 },
	547	[IPSET_ATTR_TIMEOUT] = { .type = NLA_U32 },
	548	[IPSET_ATTR_LINENO] = { .type = NLA_U32 },
	549	},
	550	.me = THIS_MODULE,
	551	};
	552
	553	static int __init
	554	hash_ipport_init(void)
	555	{
	556	return ip_set_type_register(&hash_ipport_type);
	557	}
	558
	559	static void __exit
	560	hash_ipport_fini(void)
	561	{
	562	ip_set_type_unregister(&hash_ipport_type);
	563	}
	564
	565	module_init(hash_ipport_init);
	566	module_exit(hash_ipport_fini);