diff mbox

[1/2] netfilter: ipset: Add hash:net,port,net module to kernel.

Message ID 1380392401-53264-1-git-send-email-oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa
State Superseded
Headers show

Commit Message

Oliver Smith Sept. 28, 2013, 6:20 p.m. UTC
From: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>

This adds a new set that provides similar functionality to ip,port,net
but permits arbitrary size subnets for both the first and last
parameter.

Signed-off-by: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
---
 kernel/net/netfilter/ipset/Kbuild                  |   2 +-
 .../net/netfilter/ipset/ip_set_hash_netportnet.c   | 592 +++++++++++++++++++++
 2 files changed, 593 insertions(+), 1 deletion(-)
 create mode 100644 kernel/net/netfilter/ipset/ip_set_hash_netportnet.c

Comments

Oliver Smith Sept. 28, 2013, 6:29 p.m. UTC | #1
Blah, ignore this patch, I forgot to add it to the Kconfig, will send a reroll 
in a minute.

On Saturday 28 September 2013 20:20:00 Oliver wrote:
> From: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
> 
> This adds a new set that provides similar functionality to ip,port,net
> but permits arbitrary size subnets for both the first and last
> parameter.
> 
> Signed-off-by: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
> ---
>  kernel/net/netfilter/ipset/Kbuild                  |   2 +-
>  .../net/netfilter/ipset/ip_set_hash_netportnet.c   | 592
> +++++++++++++++++++++ 2 files changed, 593 insertions(+), 1 deletion(-)
>  create mode 100644 kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> 
> diff --git a/kernel/net/netfilter/ipset/Kbuild
> b/kernel/net/netfilter/ipset/Kbuild index 6174c86..dc5cc05 100644
> --- a/kernel/net/netfilter/ipset/Kbuild
> +++ b/kernel/net/netfilter/ipset/Kbuild
> @@ -7,7 +7,7 @@ obj-m += ip_set_bitmap_ip.o ip_set_bitmap_ipmac.o
> ip_set_bitmap_port.o obj-m += ip_set_hash_ip.o ip_set_hash_ipport.o
> ip_set_hash_ipportip.o obj-m += ip_set_hash_ipportnet.o
>  obj-m += ip_set_hash_net.o ip_set_hash_netport.o ip_set_hash_netiface.o
> -obj-m += ip_set_hash_netnet.o
> +obj-m += ip_set_hash_netnet.o ip_set_hash_netportnet.o
>  obj-m += ip_set_list_set.o
> 
>  # It's for me...
> diff --git a/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c new file mode 100644
> index 0000000..cfb2997
> --- /dev/null
> +++ b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> @@ -0,0 +1,592 @@
> +/* Copyright (C) 2003-2013 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,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 <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_getport.h>
> +#include <linux/netfilter/ipset/ip_set_hash.h>
> +
> +#define IPSET_TYPE_REV_MIN	0
> +#define IPSET_TYPE_REV_MAX	0 /* Comments support added */
> +
> +MODULE_LICENSE("GPL");
> +MODULE_AUTHOR("Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>");
> +IP_SET_MODULE_DESC("hash:net,port,net", IPSET_TYPE_REV_MIN,
> IPSET_TYPE_REV_MAX); +MODULE_ALIAS("ip_set_hash:net,port,net");
> +
> +/* Type specific function prefix */
> +#define HTYPE		hash_netportnet
> +
> +/* We squeeze the "nomatch" flag into cidr: we don't support cidr == 0
> + * However this way we have to store internally cidr - 1,
> + * dancing back and forth.
> + */
> +#define IP_SET_HASH_WITH_PROTO
> +#define IP_SET_HASH_WITH_NETS
> +#define IPSET_NET_COUNT 2
> +
> +/* IPv4 variant */
> +
> +/* Member elements */
> +struct hash_netportnet4_elem {
> +	union {
> +		__be32 ip[2];
> +		__be64 ipcmp;
> +	};
> +	__be16 port;
> +	union {
> +		u8 cidr[2];
> +		u16 ccmp;
> +	};
> +	u8 nomatch:1;
> +	u8 proto;
> +};
> +
> +/* Common functions */
> +
> +static inline bool
> +hash_netportnet4_data_equal(const struct hash_netportnet4_elem *ip1,
> +			   const struct hash_netportnet4_elem *ip2,
> +			   u32 *multi)
> +{
> +	return ip1->ipcmp == ip2->ipcmp &&
> +	       ip1->ccmp == ip2->ccmp &&
> +	       ip1->port == ip2->port &&
> +	       ip1->proto == ip2->proto;
> +}
> +
> +static inline int
> +hash_netportnet4_do_data_match(const struct hash_netportnet4_elem *elem)
> +{
> +	return elem->nomatch ? -ENOTEMPTY : 1;
> +}
> +
> +static inline void
> +hash_netportnet4_data_set_flags(struct hash_netportnet4_elem *elem, u32
> flags) +{
> +	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
> +}
> +
> +static inline void
> +hash_netportnet4_data_reset_flags(struct hash_netportnet4_elem *elem, u8
> *flags) +{
> +	swap(*flags, elem->nomatch);
> +}
> +
> +static inline void
> +hash_netportnet4_data_reset_elem(struct hash_netportnet4_elem *elem,
> +				struct hash_netportnet4_elem *orig)
> +{
> +	elem->ip[1] = orig->ip[1];
> +}
> +
> +static inline void
> +hash_netportnet4_data_netmask(struct hash_netportnet4_elem *elem, u8 cidr,
> bool inner) +{
> +	if (inner) {
> +		elem->ip[1] &= ip_set_netmask(cidr);
> +		elem->cidr[1] = cidr;
> +	} else {
> +		elem->ip[0] &= ip_set_netmask(cidr);
> +		elem->cidr[0] = cidr;
> +	}
> +}
> +
> +static bool
> +hash_netportnet4_data_list(struct sk_buff *skb,
> +			  const struct hash_netportnet4_elem *data)
> +{
> +	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
> +
> +	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip[0]) ||
> +	    nla_put_ipaddr4(skb, IPSET_ATTR_IP2, data->ip[1]) ||
> +	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
> +	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
> +	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
> +	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
> +	    (flags &&
> +	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
> +		goto nla_put_failure;
> +	return 0;
> +
> +nla_put_failure:
> +	return 1;
> +}
> +
> +static inline void
> +hash_netportnet4_data_next(struct hash_netportnet4_elem *next,
> +			  const struct hash_netportnet4_elem *d)
> +{
> +	next->ipcmp = d->ipcmp;
> +	next->port = d->port;
> +}
> +
> +#define MTYPE		hash_netportnet4
> +#define PF		4
> +#define HOST_MASK	32
> +#include "ip_set_hash_gen.h"
> +
> +static int
> +hash_netportnet4_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_netportnet *h = set->data;
> +	ipset_adtfn adtfn = set->variant->adt[adt];
> +	struct hash_netportnet4_elem e = {
> +		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
> +		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
> +	};
> +	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
> +
> +	if (adt == IPSET_TEST)
> +		e.ccmp = (HOST_MASK << (sizeof(e.cidr[0]) * 8)) | HOST_MASK;
> +
> +	if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
> +				 &e.port, &e.proto))
> +		return -EINVAL;
> +
> +	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0]);
> +	ip4addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1]);
> +	e.ip[0] &= ip_set_netmask(e.cidr[0]);
> +	e.ip[1] &= ip_set_netmask(e.cidr[1]);
> +
> +	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
> +}
> +
> +static int
> +hash_netportnet4_uadt(struct ip_set *set, struct nlattr *tb[],
> +		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
> +{
> +	const struct hash_netportnet *h = set->data;
> +	ipset_adtfn adtfn = set->variant->adt[adt];
> +	struct hash_netportnet4_elem e = { .cidr[0] = HOST_MASK,
> +					   .cidr[1] = HOST_MASK };
> +	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
> +	u32 ip = 0, ip_to = 0, ip_last, p = 0, port, port_to;
> +	u32 ip2_from = 0, ip2_to = 0, ip2_last, ip2;
> +	bool with_ports = false;
> +	u8 cidr, cidr2;
> +	int ret;
> +
> +	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
> +		     !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) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES)))
> +		return -IPSET_ERR_PROTOCOL;
> +
> +	if (tb[IPSET_ATTR_LINENO])
> +		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
> +
> +	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP], &ip) ||
> +	      ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2], &ip2_from) ||
> +	      ip_set_get_extensions(set, tb, &ext);
> +	if (ret)
> +		return ret;
> +
> +	if (tb[IPSET_ATTR_CIDR]) {
> +		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
> +		if (!cidr || cidr > HOST_MASK)
> +			return -IPSET_ERR_INVALID_CIDR;
> +		e.cidr[0] = cidr;
> +	}
> +
> +	if (tb[IPSET_ATTR_CIDR2]) {
> +		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
> +		if (!cidr || cidr > HOST_MASK)
> +			return -IPSET_ERR_INVALID_CIDR;
> +		e.cidr[1] = cidr;
> +	}
> +
> +	if (tb[IPSET_ATTR_PORT])
> +		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
> +	else
> +		return -IPSET_ERR_PROTOCOL;
> +
> +	if (tb[IPSET_ATTR_PROTO]) {
> +		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
> +		with_ports = ip_set_proto_with_ports(e.proto);
> +
> +		if (e.proto == 0)
> +			return -IPSET_ERR_INVALID_PROTO;
> +	} else
> +		return -IPSET_ERR_MISSING_PROTO;
> +
> +	if (!(with_ports || e.proto == IPPROTO_ICMP))
> +		e.port = 0;
> +
> +	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);
> +	}
> +
> +	with_ports = with_ports && tb[IPSET_ATTR_PORT_TO];
> +	if (adt == IPSET_TEST ||
> +	    !(tb[IPSET_ATTR_IP_TO] || with_ports || tb[IPSET_ATTR_IP2_TO])) {
> +		e.ip[0] = htonl(ip & ip_set_hostmask(e.cidr[0]));
> +		e.ip[1] = htonl(ip2_from & ip_set_hostmask(e.cidr[1]));
> +		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 > ip_to)
> +			swap(ip, ip_to);
> +		if (unlikely(ip + UINT_MAX == ip_to))
> +			return -IPSET_ERR_HASH_RANGE;
> +	}
> +
> +	port_to = port = ntohs(e.port);
> +	if (tb[IPSET_ATTR_PORT_TO]) {
> +		port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
> +		if (port > port_to)
> +			swap(port, port_to);
> +	}
> +
> +	ip2_to = ip2_from;
> +	if (tb[IPSET_ATTR_IP2_TO]) {
> +		ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2_TO], &ip2_to);
> +		if (ret)
> +			return ret;
> +		if (ip2_from > ip2_to)
> +			swap(ip2_from, ip2_to);
> +		if (unlikely(ip2_from + UINT_MAX == ip2_to))
> +			return -IPSET_ERR_HASH_RANGE;
> +	}
> +
> +	if (retried)
> +		ip = ntohl(h->next.ip[0]);
> +
> +	while (!after(ip, ip_to)) {
> +		e.ip[0] = htonl(ip);
> +		ip_last = ip_set_range_to_cidr(ip, ip_to, &cidr);
> +		e.cidr[0] = cidr;
> +		p = retried && ip == ntohl(h->next.ip[0]) ? ntohs(h->next.port)
> +							  : port;
> +		for (; p <= port_to; p++) {
> +			e.port = htons(p);
> +			ip2 = (retried && ip == ntohl(h->next.ip[0]) &&
> +			       p == ntohs(h->next.port)) ? ntohl(h->next.ip[1])
> +							 : ip2_from;
> +			while (!after(ip2, ip2_to)) {
> +				e.ip[1] = htonl(ip2);
> +				ip2_last = ip_set_range_to_cidr(ip2, ip2_to,
> +								&cidr2);
> +				e.cidr[1] = cidr2;
> +				ret = adtfn(set, &e, &ext, &ext, flags);
> +				if (ret && !ip_set_eexist(ret, flags))
> +					return ret;
> +				else
> +					ret = 0;
> +				ip2 = ip2_last + 1;
> +			}
> +		}
> +		ip = ip_last + 1;
> +	}
> +	return ret;
> +}
> +
> +/* IPv6 variant */
> +
> +struct hash_netportnet6_elem {
> +	union nf_inet_addr ip[2];
> +	__be16 port;
> +	union {
> +		u8 cidr[2];
> +		u16 ccmp;
> +	};
> +	u8 nomatch:1;
> +	u8 proto;
> +};
> +
> +/* Common functions */
> +
> +static inline bool
> +hash_netportnet6_data_equal(const struct hash_netportnet6_elem *ip1,
> +			   const struct hash_netportnet6_elem *ip2,
> +			   u32 *multi)
> +{
> +	return ipv6_addr_equal(&ip1->ip[0].in6, &ip2->ip[0].in6) &&
> +	       ipv6_addr_equal(&ip1->ip[1].in6, &ip2->ip[1].in6) &&
> +	       ip1->ccmp == ip2->ccmp &&
> +	       ip1->port == ip2->port &&
> +	       ip1->proto == ip2->proto;
> +}
> +
> +static inline int
> +hash_netportnet6_do_data_match(const struct hash_netportnet6_elem *elem)
> +{
> +	return elem->nomatch ? -ENOTEMPTY : 1;
> +}
> +
> +static inline void
> +hash_netportnet6_data_set_flags(struct hash_netportnet6_elem *elem, u32
> flags) +{
> +	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
> +}
> +
> +static inline void
> +hash_netportnet6_data_reset_flags(struct hash_netportnet6_elem *elem, u8
> *flags) +{
> +	swap(*flags, elem->nomatch);
> +}
> +
> +static inline void
> +hash_netportnet6_data_reset_elem(struct hash_netportnet6_elem *elem,
> +				struct hash_netportnet6_elem *orig)
> +{
> +	elem->ip[1] = orig->ip[1];
> +}
> +
> +static inline void
> +hash_netportnet6_data_netmask(struct hash_netportnet6_elem *elem, u8 cidr,
> bool inner) +{
> +	if (inner) {
> +		ip6_netmask(&elem->ip[1], cidr);
> +		elem->cidr[1] = cidr;
> +	} else {
> +		ip6_netmask(&elem->ip[0], cidr);
> +		elem->cidr[0] = cidr;
> +	}
> +}
> +
> +static bool
> +hash_netportnet6_data_list(struct sk_buff *skb,
> +			  const struct hash_netportnet6_elem *data)
> +{
> +	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
> +
> +	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip[0].in6) ||
> +	    nla_put_ipaddr6(skb, IPSET_ATTR_IP2, &data->ip[1].in6) ||
> +	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
> +	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
> +	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
> +	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
> +	    (flags &&
> +	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
> +		goto nla_put_failure;
> +	return 0;
> +
> +nla_put_failure:
> +	return 1;
> +}
> +
> +static inline void
> +hash_netportnet6_data_next(struct hash_netportnet4_elem *next,
> +			  const struct hash_netportnet6_elem *d)
> +{
> +	next->port = d->port;
> +}
> +
> +#undef MTYPE
> +#undef PF
> +#undef HOST_MASK
> +
> +#define MTYPE		hash_netportnet6
> +#define PF		6
> +#define HOST_MASK	128
> +#define IP_SET_EMIT_CREATE
> +#include "ip_set_hash_gen.h"
> +
> +static int
> +hash_netportnet6_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_netportnet *h = set->data;
> +	ipset_adtfn adtfn = set->variant->adt[adt];
> +	struct hash_netportnet6_elem e = {
> +		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
> +		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
> +	};
> +	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
> +
> +	if (adt == IPSET_TEST)
> +		e.ccmp = (HOST_MASK << (sizeof(u8) * 8)) | HOST_MASK;
> +
> +	if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
> +				 &e.port, &e.proto))
> +		return -EINVAL;
> +
> +	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0].in6);
> +	ip6addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1].in6);
> +	ip6_netmask(&e.ip[0], e.cidr[0]);
> +	ip6_netmask(&e.ip[1], e.cidr[1]);
> +
> +	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
> +}
> +
> +static int
> +hash_netportnet6_uadt(struct ip_set *set, struct nlattr *tb[],
> +		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
> +{
> +	const struct hash_netportnet *h = set->data;
> +	ipset_adtfn adtfn = set->variant->adt[adt];
> +	struct hash_netportnet6_elem e = { .cidr[0] = HOST_MASK,
> +					   .cidr[1] = HOST_MASK };
> +	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
> +	u32 port, port_to;
> +	bool with_ports = false;
> +	int ret;
> +
> +	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
> +		     !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) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
> +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES) ||
> +		     tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
> +		return -IPSET_ERR_PROTOCOL;
> +	if (unlikely(tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
> +		return -IPSET_ERR_HASH_RANGE_UNSUPPORTED;
> +
> +	if (tb[IPSET_ATTR_LINENO])
> +		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
> +
> +	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip[0]) ||
> +	      ip_set_get_ipaddr6(tb[IPSET_ATTR_IP2], &e.ip[1]) ||
> +	      ip_set_get_extensions(set, tb, &ext);
> +	if (ret)
> +		return ret;
> +
> +	if (tb[IPSET_ATTR_CIDR])
> +		e.cidr[0] = nla_get_u8(tb[IPSET_ATTR_CIDR]);
> +
> +	if (tb[IPSET_ATTR_CIDR2])
> +		e.cidr[1] = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
> +
> +	if (unlikely(!e.cidr[0] || e.cidr[0] > HOST_MASK || !e.cidr[1] ||
> +		     e.cidr[1] > HOST_MASK))
> +		return -IPSET_ERR_INVALID_CIDR;
> +
> +	ip6_netmask(&e.ip[0], e.cidr[0]);
> +	ip6_netmask(&e.ip[1], e.cidr[1]);
> +
> +	if (tb[IPSET_ATTR_PORT])
> +		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
> +	else
> +		return -IPSET_ERR_PROTOCOL;
> +
> +	if (tb[IPSET_ATTR_PROTO]) {
> +		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
> +		with_ports = ip_set_proto_with_ports(e.proto);
> +
> +		if (e.proto == 0)
> +			return -IPSET_ERR_INVALID_PROTO;
> +	} else
> +		return -IPSET_ERR_MISSING_PROTO;
> +
> +	if (!(with_ports || e.proto == IPPROTO_ICMPV6))
> +		e.port = 0;
> +
> +	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 || !with_ports || !tb[IPSET_ATTR_PORT_TO]) {
> +		ret = adtfn(set, &e, &ext, &ext, flags);
> +		return ip_set_enomatch(ret, flags, adt, set) ? -ret :
> +		       ip_set_eexist(ret, flags) ? 0 : ret;
> +	}
> +
> +	port = ntohs(e.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++) {
> +		e.port = htons(port);
> +		ret = adtfn(set, &e, &ext, &ext, flags);
> +
> +		if (ret && !ip_set_eexist(ret, flags))
> +			return ret;
> +		else
> +			ret = 0;
> +	}
> +	return ret;
> +}
> +
> +static struct ip_set_type hash_netportnet_type __read_mostly = {
> +	.name		= "hash:net,port,net",
> +	.protocol	= IPSET_PROTOCOL,
> +	.features	= IPSET_TYPE_IP | IPSET_TYPE_PORT | IPSET_TYPE_IP2 |
> +			  IPSET_TYPE_NOMATCH,
> +	.dimension	= IPSET_DIM_THREE,
> +	.family		= NFPROTO_UNSPEC,
> +	.revision_min	= IPSET_TYPE_REV_MIN,
> +	.revision_max	= IPSET_TYPE_REV_MAX,
> +	.create		= hash_netportnet_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_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_IP2]	= { .type = NLA_NESTED },
> +		[IPSET_ATTR_IP2_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_CIDR2]	= { .type = NLA_U8 },
> +		[IPSET_ATTR_PROTO]	= { .type = NLA_U8 },
> +		[IPSET_ATTR_CADT_FLAGS]	= { .type = NLA_U32 },
> +		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
> +		[IPSET_ATTR_LINENO]	= { .type = NLA_U32 },
> +		[IPSET_ATTR_BYTES]	= { .type = NLA_U64 },
> +		[IPSET_ATTR_PACKETS]	= { .type = NLA_U64 },
> +		[IPSET_ATTR_COMMENT]	= { .type = NLA_NUL_STRING },
> +	},
> +	.me		= THIS_MODULE,
> +};
> +
> +static int __init
> +hash_netportnet_init(void)
> +{
> +	return ip_set_type_register(&hash_netportnet_type);
> +}
> +
> +static void __exit
> +hash_netportnet_fini(void)
> +{
> +	ip_set_type_unregister(&hash_netportnet_type);
> +}
> +
> +module_init(hash_netportnet_init);
> +module_exit(hash_netportnet_fini);
--
To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
the body of a message to majordomo@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Jozsef Kadlecsik Sept. 30, 2013, 6:25 p.m. UTC | #2
On Sat, 28 Sep 2013, Oliver wrote:

> Blah, ignore this patch, I forgot to add it to the Kconfig, will send a 
> reroll in a minute.
> 
> On Saturday 28 September 2013 20:20:00 Oliver wrote:
> > From: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
> > 
> > This adds a new set that provides similar functionality to ip,port,net
> > but permits arbitrary size subnets for both the first and last
> > parameter.

Patch is applied, but I had to fix the whitespace issues reported by 
checkpatch. There's two other modifications as well:

> > Signed-off-by: Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
> > ---
> >  kernel/net/netfilter/ipset/Kbuild                  |   2 +-
> >  .../net/netfilter/ipset/ip_set_hash_netportnet.c   | 592
> > +++++++++++++++++++++ 2 files changed, 593 insertions(+), 1 deletion(-)
> >  create mode 100644 kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> > 
> > diff --git a/kernel/net/netfilter/ipset/Kbuild
> > b/kernel/net/netfilter/ipset/Kbuild index 6174c86..dc5cc05 100644
> > --- a/kernel/net/netfilter/ipset/Kbuild
> > +++ b/kernel/net/netfilter/ipset/Kbuild
> > @@ -7,7 +7,7 @@ obj-m += ip_set_bitmap_ip.o ip_set_bitmap_ipmac.o
> > ip_set_bitmap_port.o obj-m += ip_set_hash_ip.o ip_set_hash_ipport.o
> > ip_set_hash_ipportip.o obj-m += ip_set_hash_ipportnet.o
> >  obj-m += ip_set_hash_net.o ip_set_hash_netport.o ip_set_hash_netiface.o
> > -obj-m += ip_set_hash_netnet.o
> > +obj-m += ip_set_hash_netnet.o ip_set_hash_netportnet.o
> >  obj-m += ip_set_list_set.o
> > 
> >  # It's for me...
> > diff --git a/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> > b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c new file mode 100644
> > index 0000000..cfb2997
> > --- /dev/null
> > +++ b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
> > @@ -0,0 +1,592 @@
> > +/* Copyright (C) 2003-2013 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,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 <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_getport.h>
> > +#include <linux/netfilter/ipset/ip_set_hash.h>
> > +
> > +#define IPSET_TYPE_REV_MIN	0
> > +#define IPSET_TYPE_REV_MAX	0 /* Comments support added */
> > +
> > +MODULE_LICENSE("GPL");
> > +MODULE_AUTHOR("Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>");
> > +IP_SET_MODULE_DESC("hash:net,port,net", IPSET_TYPE_REV_MIN,
> > IPSET_TYPE_REV_MAX); +MODULE_ALIAS("ip_set_hash:net,port,net");
> > +
> > +/* Type specific function prefix */
> > +#define HTYPE		hash_netportnet
> > +
> > +/* We squeeze the "nomatch" flag into cidr: we don't support cidr == 0
> > + * However this way we have to store internally cidr - 1,
> > + * dancing back and forth.
> > + */

The copy&pasted comment above is not valid for this type, therefore I 
deleted it.

> > +#define IP_SET_HASH_WITH_PROTO
> > +#define IP_SET_HASH_WITH_NETS
> > +#define IPSET_NET_COUNT 2
> > +
> > +/* IPv4 variant */
> > +
> > +/* Member elements */
> > +struct hash_netportnet4_elem {
> > +	union {
> > +		__be32 ip[2];
> > +		__be64 ipcmp;
> > +	};
> > +	__be16 port;
> > +	union {
> > +		u8 cidr[2];
> > +		u16 ccmp;
> > +	};
> > +	u8 nomatch:1;
> > +	u8 proto;
> > +};
> > +
> > +/* Common functions */
> > +
> > +static inline bool
> > +hash_netportnet4_data_equal(const struct hash_netportnet4_elem *ip1,
> > +			   const struct hash_netportnet4_elem *ip2,
> > +			   u32 *multi)
> > +{
> > +	return ip1->ipcmp == ip2->ipcmp &&
> > +	       ip1->ccmp == ip2->ccmp &&
> > +	       ip1->port == ip2->port &&
> > +	       ip1->proto == ip2->proto;
> > +}
> > +
> > +static inline int
> > +hash_netportnet4_do_data_match(const struct hash_netportnet4_elem *elem)
> > +{
> > +	return elem->nomatch ? -ENOTEMPTY : 1;
> > +}
> > +
> > +static inline void
> > +hash_netportnet4_data_set_flags(struct hash_netportnet4_elem *elem, u32
> > flags) +{
> > +	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
> > +}
> > +
> > +static inline void
> > +hash_netportnet4_data_reset_flags(struct hash_netportnet4_elem *elem, u8
> > *flags) +{
> > +	swap(*flags, elem->nomatch);
> > +}
> > +
> > +static inline void
> > +hash_netportnet4_data_reset_elem(struct hash_netportnet4_elem *elem,
> > +				struct hash_netportnet4_elem *orig)
> > +{
> > +	elem->ip[1] = orig->ip[1];
> > +}
> > +
> > +static inline void
> > +hash_netportnet4_data_netmask(struct hash_netportnet4_elem *elem, u8 cidr,
> > bool inner) +{
> > +	if (inner) {
> > +		elem->ip[1] &= ip_set_netmask(cidr);
> > +		elem->cidr[1] = cidr;
> > +	} else {
> > +		elem->ip[0] &= ip_set_netmask(cidr);
> > +		elem->cidr[0] = cidr;
> > +	}
> > +}
> > +
> > +static bool
> > +hash_netportnet4_data_list(struct sk_buff *skb,
> > +			  const struct hash_netportnet4_elem *data)
> > +{
> > +	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
> > +
> > +	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip[0]) ||
> > +	    nla_put_ipaddr4(skb, IPSET_ATTR_IP2, data->ip[1]) ||
> > +	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
> > +	    (flags &&
> > +	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
> > +		goto nla_put_failure;
> > +	return 0;
> > +
> > +nla_put_failure:
> > +	return 1;
> > +}
> > +
> > +static inline void
> > +hash_netportnet4_data_next(struct hash_netportnet4_elem *next,
> > +			  const struct hash_netportnet4_elem *d)
> > +{
> > +	next->ipcmp = d->ipcmp;
> > +	next->port = d->port;
> > +}
> > +
> > +#define MTYPE		hash_netportnet4
> > +#define PF		4
> > +#define HOST_MASK	32
> > +#include "ip_set_hash_gen.h"
> > +
> > +static int
> > +hash_netportnet4_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_netportnet *h = set->data;
> > +	ipset_adtfn adtfn = set->variant->adt[adt];
> > +	struct hash_netportnet4_elem e = {
> > +		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
> > +		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
> > +	};
> > +	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
> > +
> > +	if (adt == IPSET_TEST)
> > +		e.ccmp = (HOST_MASK << (sizeof(e.cidr[0]) * 8)) | HOST_MASK;
> > +
> > +	if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
> > +				 &e.port, &e.proto))
> > +		return -EINVAL;
> > +
> > +	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0]);
> > +	ip4addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1]);
> > +	e.ip[0] &= ip_set_netmask(e.cidr[0]);
> > +	e.ip[1] &= ip_set_netmask(e.cidr[1]);
> > +
> > +	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
> > +}
> > +
> > +static int
> > +hash_netportnet4_uadt(struct ip_set *set, struct nlattr *tb[],
> > +		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
> > +{
> > +	const struct hash_netportnet *h = set->data;
> > +	ipset_adtfn adtfn = set->variant->adt[adt];
> > +	struct hash_netportnet4_elem e = { .cidr[0] = HOST_MASK,
> > +					   .cidr[1] = HOST_MASK };
> > +	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
> > +	u32 ip = 0, ip_to = 0, ip_last, p = 0, port, port_to;
> > +	u32 ip2_from = 0, ip2_to = 0, ip2_last, ip2;
> > +	bool with_ports = false;
> > +	u8 cidr, cidr2;
> > +	int ret;
> > +
> > +	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
> > +		     !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) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES)))
> > +		return -IPSET_ERR_PROTOCOL;
> > +
> > +	if (tb[IPSET_ATTR_LINENO])
> > +		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
> > +
> > +	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP], &ip) ||
> > +	      ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2], &ip2_from) ||
> > +	      ip_set_get_extensions(set, tb, &ext);
> > +	if (ret)
> > +		return ret;
> > +
> > +	if (tb[IPSET_ATTR_CIDR]) {
> > +		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
> > +		if (!cidr || cidr > HOST_MASK)
> > +			return -IPSET_ERR_INVALID_CIDR;
> > +		e.cidr[0] = cidr;
> > +	}
> > +
> > +	if (tb[IPSET_ATTR_CIDR2]) {
> > +		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
> > +		if (!cidr || cidr > HOST_MASK)
> > +			return -IPSET_ERR_INVALID_CIDR;
> > +		e.cidr[1] = cidr;
> > +	}
> > +
> > +	if (tb[IPSET_ATTR_PORT])
> > +		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
> > +	else
> > +		return -IPSET_ERR_PROTOCOL;
> > +
> > +	if (tb[IPSET_ATTR_PROTO]) {
> > +		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
> > +		with_ports = ip_set_proto_with_ports(e.proto);
> > +
> > +		if (e.proto == 0)
> > +			return -IPSET_ERR_INVALID_PROTO;
> > +	} else
> > +		return -IPSET_ERR_MISSING_PROTO;
> > +
> > +	if (!(with_ports || e.proto == IPPROTO_ICMP))
> > +		e.port = 0;
> > +
> > +	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);
> > +	}
> > +
> > +	with_ports = with_ports && tb[IPSET_ATTR_PORT_TO];
> > +	if (adt == IPSET_TEST ||
> > +	    !(tb[IPSET_ATTR_IP_TO] || with_ports || tb[IPSET_ATTR_IP2_TO])) {
> > +		e.ip[0] = htonl(ip & ip_set_hostmask(e.cidr[0]));
> > +		e.ip[1] = htonl(ip2_from & ip_set_hostmask(e.cidr[1]));
> > +		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 > ip_to)
> > +			swap(ip, ip_to);
> > +		if (unlikely(ip + UINT_MAX == ip_to))
> > +			return -IPSET_ERR_HASH_RANGE;
> > +	}
> > +
> > +	port_to = port = ntohs(e.port);
> > +	if (tb[IPSET_ATTR_PORT_TO]) {
> > +		port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
> > +		if (port > port_to)
> > +			swap(port, port_to);
> > +	}
> > +
> > +	ip2_to = ip2_from;
> > +	if (tb[IPSET_ATTR_IP2_TO]) {
> > +		ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2_TO], &ip2_to);
> > +		if (ret)
> > +			return ret;
> > +		if (ip2_from > ip2_to)
> > +			swap(ip2_from, ip2_to);
> > +		if (unlikely(ip2_from + UINT_MAX == ip2_to))
> > +			return -IPSET_ERR_HASH_RANGE;
> > +	}
> > +
> > +	if (retried)
> > +		ip = ntohl(h->next.ip[0]);
> > +
> > +	while (!after(ip, ip_to)) {
> > +		e.ip[0] = htonl(ip);
> > +		ip_last = ip_set_range_to_cidr(ip, ip_to, &cidr);
> > +		e.cidr[0] = cidr;
> > +		p = retried && ip == ntohl(h->next.ip[0]) ? ntohs(h->next.port)
> > +							  : port;
> > +		for (; p <= port_to; p++) {
> > +			e.port = htons(p);
> > +			ip2 = (retried && ip == ntohl(h->next.ip[0]) &&
> > +			       p == ntohs(h->next.port)) ? ntohl(h->next.ip[1])
> > +							 : ip2_from;
> > +			while (!after(ip2, ip2_to)) {
> > +				e.ip[1] = htonl(ip2);
> > +				ip2_last = ip_set_range_to_cidr(ip2, ip2_to,
> > +								&cidr2);
> > +				e.cidr[1] = cidr2;
> > +				ret = adtfn(set, &e, &ext, &ext, flags);
> > +				if (ret && !ip_set_eexist(ret, flags))
> > +					return ret;
> > +				else
> > +					ret = 0;
> > +				ip2 = ip2_last + 1;
> > +			}
> > +		}
> > +		ip = ip_last + 1;
> > +	}
> > +	return ret;
> > +}
> > +
> > +/* IPv6 variant */
> > +
> > +struct hash_netportnet6_elem {
> > +	union nf_inet_addr ip[2];
> > +	__be16 port;
> > +	union {
> > +		u8 cidr[2];
> > +		u16 ccmp;
> > +	};
> > +	u8 nomatch:1;
> > +	u8 proto;
> > +};
> > +
> > +/* Common functions */
> > +
> > +static inline bool
> > +hash_netportnet6_data_equal(const struct hash_netportnet6_elem *ip1,
> > +			   const struct hash_netportnet6_elem *ip2,
> > +			   u32 *multi)
> > +{
> > +	return ipv6_addr_equal(&ip1->ip[0].in6, &ip2->ip[0].in6) &&
> > +	       ipv6_addr_equal(&ip1->ip[1].in6, &ip2->ip[1].in6) &&
> > +	       ip1->ccmp == ip2->ccmp &&
> > +	       ip1->port == ip2->port &&
> > +	       ip1->proto == ip2->proto;
> > +}
> > +
> > +static inline int
> > +hash_netportnet6_do_data_match(const struct hash_netportnet6_elem *elem)
> > +{
> > +	return elem->nomatch ? -ENOTEMPTY : 1;
> > +}
> > +
> > +static inline void
> > +hash_netportnet6_data_set_flags(struct hash_netportnet6_elem *elem, u32
> > flags) +{
> > +	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
> > +}
> > +
> > +static inline void
> > +hash_netportnet6_data_reset_flags(struct hash_netportnet6_elem *elem, u8
> > *flags) +{
> > +	swap(*flags, elem->nomatch);
> > +}
> > +
> > +static inline void
> > +hash_netportnet6_data_reset_elem(struct hash_netportnet6_elem *elem,
> > +				struct hash_netportnet6_elem *orig)
> > +{
> > +	elem->ip[1] = orig->ip[1];
> > +}
> > +
> > +static inline void
> > +hash_netportnet6_data_netmask(struct hash_netportnet6_elem *elem, u8 cidr,
> > bool inner) +{
> > +	if (inner) {
> > +		ip6_netmask(&elem->ip[1], cidr);
> > +		elem->cidr[1] = cidr;
> > +	} else {
> > +		ip6_netmask(&elem->ip[0], cidr);
> > +		elem->cidr[0] = cidr;
> > +	}
> > +}
> > +
> > +static bool
> > +hash_netportnet6_data_list(struct sk_buff *skb,
> > +			  const struct hash_netportnet6_elem *data)
> > +{
> > +	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
> > +
> > +	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip[0].in6) ||
> > +	    nla_put_ipaddr6(skb, IPSET_ATTR_IP2, &data->ip[1].in6) ||
> > +	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
> > +	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
> > +	    (flags &&
> > +	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
> > +		goto nla_put_failure;
> > +	return 0;
> > +
> > +nla_put_failure:
> > +	return 1;
> > +}
> > +
> > +static inline void
> > +hash_netportnet6_data_next(struct hash_netportnet4_elem *next,
> > +			  const struct hash_netportnet6_elem *d)
> > +{
> > +	next->port = d->port;
> > +}
> > +
> > +#undef MTYPE
> > +#undef PF
> > +#undef HOST_MASK
> > +
> > +#define MTYPE		hash_netportnet6
> > +#define PF		6
> > +#define HOST_MASK	128
> > +#define IP_SET_EMIT_CREATE
> > +#include "ip_set_hash_gen.h"
> > +
> > +static int
> > +hash_netportnet6_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_netportnet *h = set->data;
> > +	ipset_adtfn adtfn = set->variant->adt[adt];
> > +	struct hash_netportnet6_elem e = {
> > +		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
> > +		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
> > +	};
> > +	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
> > +
> > +	if (adt == IPSET_TEST)
> > +		e.ccmp = (HOST_MASK << (sizeof(u8) * 8)) | HOST_MASK;
> > +
> > +	if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
> > +				 &e.port, &e.proto))
> > +		return -EINVAL;
> > +
> > +	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0].in6);
> > +	ip6addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1].in6);
> > +	ip6_netmask(&e.ip[0], e.cidr[0]);
> > +	ip6_netmask(&e.ip[1], e.cidr[1]);
> > +
> > +	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
> > +}
> > +
> > +static int
> > +hash_netportnet6_uadt(struct ip_set *set, struct nlattr *tb[],
> > +		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
> > +{
> > +	const struct hash_netportnet *h = set->data;
> > +	ipset_adtfn adtfn = set->variant->adt[adt];
> > +	struct hash_netportnet6_elem e = { .cidr[0] = HOST_MASK,
> > +					   .cidr[1] = HOST_MASK };
> > +	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
> > +	u32 port, port_to;
> > +	bool with_ports = false;
> > +	int ret;
> > +
> > +	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
> > +		     !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) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
> > +		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES) ||
> > +		     tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
> > +		return -IPSET_ERR_PROTOCOL;

The testing of "tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]" defeats the 
intentional error reporting below, so I deleted it above.

Best regards,
Jozsef

> > +	if (unlikely(tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
> > +		return -IPSET_ERR_HASH_RANGE_UNSUPPORTED;
> > +
> > +	if (tb[IPSET_ATTR_LINENO])
> > +		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
> > +
> > +	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip[0]) ||
> > +	      ip_set_get_ipaddr6(tb[IPSET_ATTR_IP2], &e.ip[1]) ||
> > +	      ip_set_get_extensions(set, tb, &ext);
> > +	if (ret)
> > +		return ret;
> > +
> > +	if (tb[IPSET_ATTR_CIDR])
> > +		e.cidr[0] = nla_get_u8(tb[IPSET_ATTR_CIDR]);
> > +
> > +	if (tb[IPSET_ATTR_CIDR2])
> > +		e.cidr[1] = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
> > +
> > +	if (unlikely(!e.cidr[0] || e.cidr[0] > HOST_MASK || !e.cidr[1] ||
> > +		     e.cidr[1] > HOST_MASK))
> > +		return -IPSET_ERR_INVALID_CIDR;
> > +
> > +	ip6_netmask(&e.ip[0], e.cidr[0]);
> > +	ip6_netmask(&e.ip[1], e.cidr[1]);
> > +
> > +	if (tb[IPSET_ATTR_PORT])
> > +		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
> > +	else
> > +		return -IPSET_ERR_PROTOCOL;
> > +
> > +	if (tb[IPSET_ATTR_PROTO]) {
> > +		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
> > +		with_ports = ip_set_proto_with_ports(e.proto);
> > +
> > +		if (e.proto == 0)
> > +			return -IPSET_ERR_INVALID_PROTO;
> > +	} else
> > +		return -IPSET_ERR_MISSING_PROTO;
> > +
> > +	if (!(with_ports || e.proto == IPPROTO_ICMPV6))
> > +		e.port = 0;
> > +
> > +	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 || !with_ports || !tb[IPSET_ATTR_PORT_TO]) {
> > +		ret = adtfn(set, &e, &ext, &ext, flags);
> > +		return ip_set_enomatch(ret, flags, adt, set) ? -ret :
> > +		       ip_set_eexist(ret, flags) ? 0 : ret;
> > +	}
> > +
> > +	port = ntohs(e.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++) {
> > +		e.port = htons(port);
> > +		ret = adtfn(set, &e, &ext, &ext, flags);
> > +
> > +		if (ret && !ip_set_eexist(ret, flags))
> > +			return ret;
> > +		else
> > +			ret = 0;
> > +	}
> > +	return ret;
> > +}
> > +
> > +static struct ip_set_type hash_netportnet_type __read_mostly = {
> > +	.name		= "hash:net,port,net",
> > +	.protocol	= IPSET_PROTOCOL,
> > +	.features	= IPSET_TYPE_IP | IPSET_TYPE_PORT | IPSET_TYPE_IP2 |
> > +			  IPSET_TYPE_NOMATCH,
> > +	.dimension	= IPSET_DIM_THREE,
> > +	.family		= NFPROTO_UNSPEC,
> > +	.revision_min	= IPSET_TYPE_REV_MIN,
> > +	.revision_max	= IPSET_TYPE_REV_MAX,
> > +	.create		= hash_netportnet_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_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_IP2]	= { .type = NLA_NESTED },
> > +		[IPSET_ATTR_IP2_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_CIDR2]	= { .type = NLA_U8 },
> > +		[IPSET_ATTR_PROTO]	= { .type = NLA_U8 },
> > +		[IPSET_ATTR_CADT_FLAGS]	= { .type = NLA_U32 },
> > +		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
> > +		[IPSET_ATTR_LINENO]	= { .type = NLA_U32 },
> > +		[IPSET_ATTR_BYTES]	= { .type = NLA_U64 },
> > +		[IPSET_ATTR_PACKETS]	= { .type = NLA_U64 },
> > +		[IPSET_ATTR_COMMENT]	= { .type = NLA_NUL_STRING },
> > +	},
> > +	.me		= THIS_MODULE,
> > +};
> > +
> > +static int __init
> > +hash_netportnet_init(void)
> > +{
> > +	return ip_set_type_register(&hash_netportnet_type);
> > +}
> > +
> > +static void __exit
> > +hash_netportnet_fini(void)
> > +{
> > +	ip_set_type_unregister(&hash_netportnet_type);
> > +}
> > +
> > +module_init(hash_netportnet_init);
> > +module_exit(hash_netportnet_fini);
> --
> To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
> the body of a message to majordomo@vger.kernel.org
> More majordomo info at  http://vger.kernel.org/majordomo-info.html
> 

-
E-mail  : kadlec@blackhole.kfki.hu, kadlecsik.jozsef@wigner.mta.hu
PGP key : http://www.kfki.hu/~kadlec/pgp_public_key.txt
Address : Wigner Research Centre for Physics, Hungarian Academy of Sciences
          H-1525 Budapest 114, POB. 49, Hungary
--
To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in
the body of a message to majordomo@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
diff mbox

Patch

diff --git a/kernel/net/netfilter/ipset/Kbuild b/kernel/net/netfilter/ipset/Kbuild
index 6174c86..dc5cc05 100644
--- a/kernel/net/netfilter/ipset/Kbuild
+++ b/kernel/net/netfilter/ipset/Kbuild
@@ -7,7 +7,7 @@  obj-m += ip_set_bitmap_ip.o ip_set_bitmap_ipmac.o ip_set_bitmap_port.o
 obj-m += ip_set_hash_ip.o ip_set_hash_ipport.o ip_set_hash_ipportip.o
 obj-m += ip_set_hash_ipportnet.o
 obj-m += ip_set_hash_net.o ip_set_hash_netport.o ip_set_hash_netiface.o
-obj-m += ip_set_hash_netnet.o
+obj-m += ip_set_hash_netnet.o ip_set_hash_netportnet.o
 obj-m += ip_set_list_set.o
 
 # It's for me...
diff --git a/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
new file mode 100644
index 0000000..cfb2997
--- /dev/null
+++ b/kernel/net/netfilter/ipset/ip_set_hash_netportnet.c
@@ -0,0 +1,592 @@ 
+/* Copyright (C) 2003-2013 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,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 <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_getport.h>
+#include <linux/netfilter/ipset/ip_set_hash.h>
+
+#define IPSET_TYPE_REV_MIN	0
+#define IPSET_TYPE_REV_MAX	0 /* Comments support added */
+
+MODULE_LICENSE("GPL");
+MODULE_AUTHOR("Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>");
+IP_SET_MODULE_DESC("hash:net,port,net", IPSET_TYPE_REV_MIN, IPSET_TYPE_REV_MAX);
+MODULE_ALIAS("ip_set_hash:net,port,net");
+
+/* Type specific function prefix */
+#define HTYPE		hash_netportnet
+
+/* We squeeze the "nomatch" flag into cidr: we don't support cidr == 0
+ * However this way we have to store internally cidr - 1,
+ * dancing back and forth.
+ */
+#define IP_SET_HASH_WITH_PROTO
+#define IP_SET_HASH_WITH_NETS
+#define IPSET_NET_COUNT 2
+
+/* IPv4 variant */
+
+/* Member elements */
+struct hash_netportnet4_elem {
+	union {
+		__be32 ip[2];
+		__be64 ipcmp;
+	};
+	__be16 port;
+	union {
+		u8 cidr[2];
+		u16 ccmp;
+	};
+	u8 nomatch:1;
+	u8 proto;
+};
+
+/* Common functions */
+
+static inline bool
+hash_netportnet4_data_equal(const struct hash_netportnet4_elem *ip1,
+			   const struct hash_netportnet4_elem *ip2,
+			   u32 *multi)
+{
+	return ip1->ipcmp == ip2->ipcmp &&
+	       ip1->ccmp == ip2->ccmp &&
+	       ip1->port == ip2->port &&
+	       ip1->proto == ip2->proto;
+}
+
+static inline int
+hash_netportnet4_do_data_match(const struct hash_netportnet4_elem *elem)
+{
+	return elem->nomatch ? -ENOTEMPTY : 1;
+}
+
+static inline void
+hash_netportnet4_data_set_flags(struct hash_netportnet4_elem *elem, u32 flags)
+{
+	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
+}
+
+static inline void
+hash_netportnet4_data_reset_flags(struct hash_netportnet4_elem *elem, u8 *flags)
+{
+	swap(*flags, elem->nomatch);
+}
+
+static inline void
+hash_netportnet4_data_reset_elem(struct hash_netportnet4_elem *elem,
+				struct hash_netportnet4_elem *orig)
+{
+	elem->ip[1] = orig->ip[1];
+}
+
+static inline void
+hash_netportnet4_data_netmask(struct hash_netportnet4_elem *elem, u8 cidr, bool inner)
+{
+	if (inner) {
+		elem->ip[1] &= ip_set_netmask(cidr);
+		elem->cidr[1] = cidr;
+	} else {
+		elem->ip[0] &= ip_set_netmask(cidr);
+		elem->cidr[0] = cidr;
+	}
+}
+
+static bool
+hash_netportnet4_data_list(struct sk_buff *skb,
+			  const struct hash_netportnet4_elem *data)
+{
+	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
+
+	if (nla_put_ipaddr4(skb, IPSET_ATTR_IP, data->ip[0]) ||
+	    nla_put_ipaddr4(skb, IPSET_ATTR_IP2, data->ip[1]) ||
+	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
+	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
+	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
+	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
+	    (flags &&
+	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
+		goto nla_put_failure;
+	return 0;
+
+nla_put_failure:
+	return 1;
+}
+
+static inline void
+hash_netportnet4_data_next(struct hash_netportnet4_elem *next,
+			  const struct hash_netportnet4_elem *d)
+{
+	next->ipcmp = d->ipcmp;
+	next->port = d->port;
+}
+
+#define MTYPE		hash_netportnet4
+#define PF		4
+#define HOST_MASK	32
+#include "ip_set_hash_gen.h"
+
+static int
+hash_netportnet4_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_netportnet *h = set->data;
+	ipset_adtfn adtfn = set->variant->adt[adt];
+	struct hash_netportnet4_elem e = {
+		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
+		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
+	};
+	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
+
+	if (adt == IPSET_TEST)
+		e.ccmp = (HOST_MASK << (sizeof(e.cidr[0]) * 8)) | HOST_MASK;
+
+	if (!ip_set_get_ip4_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
+				 &e.port, &e.proto))
+		return -EINVAL;
+
+	ip4addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0]);
+	ip4addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1]);
+	e.ip[0] &= ip_set_netmask(e.cidr[0]);
+	e.ip[1] &= ip_set_netmask(e.cidr[1]);
+
+	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
+}
+
+static int
+hash_netportnet4_uadt(struct ip_set *set, struct nlattr *tb[],
+		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
+{
+	const struct hash_netportnet *h = set->data;
+	ipset_adtfn adtfn = set->variant->adt[adt];
+	struct hash_netportnet4_elem e = { .cidr[0] = HOST_MASK,
+					   .cidr[1] = HOST_MASK };
+	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
+	u32 ip = 0, ip_to = 0, ip_last, p = 0, port, port_to;
+	u32 ip2_from = 0, ip2_to = 0, ip2_last, ip2;
+	bool with_ports = false;
+	u8 cidr, cidr2;
+	int ret;
+
+	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
+		     !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) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES)))
+		return -IPSET_ERR_PROTOCOL;
+
+	if (tb[IPSET_ATTR_LINENO])
+		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
+
+	ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP], &ip) ||
+	      ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2], &ip2_from) ||
+	      ip_set_get_extensions(set, tb, &ext);
+	if (ret)
+		return ret;
+
+	if (tb[IPSET_ATTR_CIDR]) {
+		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR]);
+		if (!cidr || cidr > HOST_MASK)
+			return -IPSET_ERR_INVALID_CIDR;
+		e.cidr[0] = cidr;
+	}
+
+	if (tb[IPSET_ATTR_CIDR2]) {
+		cidr = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
+		if (!cidr || cidr > HOST_MASK)
+			return -IPSET_ERR_INVALID_CIDR;
+		e.cidr[1] = cidr;
+	}
+
+	if (tb[IPSET_ATTR_PORT])
+		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
+	else
+		return -IPSET_ERR_PROTOCOL;
+
+	if (tb[IPSET_ATTR_PROTO]) {
+		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
+		with_ports = ip_set_proto_with_ports(e.proto);
+
+		if (e.proto == 0)
+			return -IPSET_ERR_INVALID_PROTO;
+	} else
+		return -IPSET_ERR_MISSING_PROTO;
+
+	if (!(with_ports || e.proto == IPPROTO_ICMP))
+		e.port = 0;
+
+	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);
+	}
+
+	with_ports = with_ports && tb[IPSET_ATTR_PORT_TO];
+	if (adt == IPSET_TEST ||
+	    !(tb[IPSET_ATTR_IP_TO] || with_ports || tb[IPSET_ATTR_IP2_TO])) {
+		e.ip[0] = htonl(ip & ip_set_hostmask(e.cidr[0]));
+		e.ip[1] = htonl(ip2_from & ip_set_hostmask(e.cidr[1]));
+		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 > ip_to)
+			swap(ip, ip_to);
+		if (unlikely(ip + UINT_MAX == ip_to))
+			return -IPSET_ERR_HASH_RANGE;
+	}
+
+	port_to = port = ntohs(e.port);
+	if (tb[IPSET_ATTR_PORT_TO]) {
+		port_to = ip_set_get_h16(tb[IPSET_ATTR_PORT_TO]);
+		if (port > port_to)
+			swap(port, port_to);
+	}
+
+	ip2_to = ip2_from;
+	if (tb[IPSET_ATTR_IP2_TO]) {
+		ret = ip_set_get_hostipaddr4(tb[IPSET_ATTR_IP2_TO], &ip2_to);
+		if (ret)
+			return ret;
+		if (ip2_from > ip2_to)
+			swap(ip2_from, ip2_to);
+		if (unlikely(ip2_from + UINT_MAX == ip2_to))
+			return -IPSET_ERR_HASH_RANGE;
+	}
+
+	if (retried)
+		ip = ntohl(h->next.ip[0]);
+
+	while (!after(ip, ip_to)) {
+		e.ip[0] = htonl(ip);
+		ip_last = ip_set_range_to_cidr(ip, ip_to, &cidr);
+		e.cidr[0] = cidr;
+		p = retried && ip == ntohl(h->next.ip[0]) ? ntohs(h->next.port)
+							  : port;
+		for (; p <= port_to; p++) {
+			e.port = htons(p);
+			ip2 = (retried && ip == ntohl(h->next.ip[0]) &&
+			       p == ntohs(h->next.port)) ? ntohl(h->next.ip[1])
+							 : ip2_from;
+			while (!after(ip2, ip2_to)) {
+				e.ip[1] = htonl(ip2);
+				ip2_last = ip_set_range_to_cidr(ip2, ip2_to,
+								&cidr2);
+				e.cidr[1] = cidr2;
+				ret = adtfn(set, &e, &ext, &ext, flags);
+				if (ret && !ip_set_eexist(ret, flags))
+					return ret;
+				else
+					ret = 0;
+				ip2 = ip2_last + 1;
+			}
+		}
+		ip = ip_last + 1;
+	}
+	return ret;
+}
+
+/* IPv6 variant */
+
+struct hash_netportnet6_elem {
+	union nf_inet_addr ip[2];
+	__be16 port;
+	union {
+		u8 cidr[2];
+		u16 ccmp;
+	};
+	u8 nomatch:1;
+	u8 proto;
+};
+
+/* Common functions */
+
+static inline bool
+hash_netportnet6_data_equal(const struct hash_netportnet6_elem *ip1,
+			   const struct hash_netportnet6_elem *ip2,
+			   u32 *multi)
+{
+	return ipv6_addr_equal(&ip1->ip[0].in6, &ip2->ip[0].in6) &&
+	       ipv6_addr_equal(&ip1->ip[1].in6, &ip2->ip[1].in6) &&
+	       ip1->ccmp == ip2->ccmp &&
+	       ip1->port == ip2->port &&
+	       ip1->proto == ip2->proto;
+}
+
+static inline int
+hash_netportnet6_do_data_match(const struct hash_netportnet6_elem *elem)
+{
+	return elem->nomatch ? -ENOTEMPTY : 1;
+}
+
+static inline void
+hash_netportnet6_data_set_flags(struct hash_netportnet6_elem *elem, u32 flags)
+{
+	elem->nomatch = !!((flags >> 16) & IPSET_FLAG_NOMATCH);
+}
+
+static inline void
+hash_netportnet6_data_reset_flags(struct hash_netportnet6_elem *elem, u8 *flags)
+{
+	swap(*flags, elem->nomatch);
+}
+
+static inline void
+hash_netportnet6_data_reset_elem(struct hash_netportnet6_elem *elem,
+				struct hash_netportnet6_elem *orig)
+{
+	elem->ip[1] = orig->ip[1];
+}
+
+static inline void
+hash_netportnet6_data_netmask(struct hash_netportnet6_elem *elem, u8 cidr, bool inner)
+{
+	if (inner) {
+		ip6_netmask(&elem->ip[1], cidr);
+		elem->cidr[1] = cidr;
+	} else {
+		ip6_netmask(&elem->ip[0], cidr);
+		elem->cidr[0] = cidr;
+	}
+}
+
+static bool
+hash_netportnet6_data_list(struct sk_buff *skb,
+			  const struct hash_netportnet6_elem *data)
+{
+	u32 flags = data->nomatch ? IPSET_FLAG_NOMATCH : 0;
+
+	if (nla_put_ipaddr6(skb, IPSET_ATTR_IP, &data->ip[0].in6) ||
+	    nla_put_ipaddr6(skb, IPSET_ATTR_IP2, &data->ip[1].in6) ||
+	    nla_put_net16(skb, IPSET_ATTR_PORT, data->port) ||
+	    nla_put_u8(skb, IPSET_ATTR_CIDR, data->cidr[0]) ||
+	    nla_put_u8(skb, IPSET_ATTR_CIDR2, data->cidr[1]) ||
+	    nla_put_u8(skb, IPSET_ATTR_PROTO, data->proto) ||
+	    (flags &&
+	     nla_put_net32(skb, IPSET_ATTR_CADT_FLAGS, htonl(flags))))
+		goto nla_put_failure;
+	return 0;
+
+nla_put_failure:
+	return 1;
+}
+
+static inline void
+hash_netportnet6_data_next(struct hash_netportnet4_elem *next,
+			  const struct hash_netportnet6_elem *d)
+{
+	next->port = d->port;
+}
+
+#undef MTYPE
+#undef PF
+#undef HOST_MASK
+
+#define MTYPE		hash_netportnet6
+#define PF		6
+#define HOST_MASK	128
+#define IP_SET_EMIT_CREATE
+#include "ip_set_hash_gen.h"
+
+static int
+hash_netportnet6_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_netportnet *h = set->data;
+	ipset_adtfn adtfn = set->variant->adt[adt];
+	struct hash_netportnet6_elem e = {
+		.cidr[0] = IP_SET_INIT_CIDR(h->nets[0].cidr[0], HOST_MASK),
+		.cidr[1] = IP_SET_INIT_CIDR(h->nets[0].cidr[1], HOST_MASK),
+	};
+	struct ip_set_ext ext = IP_SET_INIT_KEXT(skb, opt, set);
+
+	if (adt == IPSET_TEST)
+		e.ccmp = (HOST_MASK << (sizeof(u8) * 8)) | HOST_MASK;
+
+	if (!ip_set_get_ip6_port(skb, opt->flags & IPSET_DIM_TWO_SRC,
+				 &e.port, &e.proto))
+		return -EINVAL;
+
+	ip6addrptr(skb, opt->flags & IPSET_DIM_ONE_SRC, &e.ip[0].in6);
+	ip6addrptr(skb, opt->flags & IPSET_DIM_THREE_SRC, &e.ip[1].in6);
+	ip6_netmask(&e.ip[0], e.cidr[0]);
+	ip6_netmask(&e.ip[1], e.cidr[1]);
+
+	return adtfn(set, &e, &ext, &opt->ext, opt->cmdflags);
+}
+
+static int
+hash_netportnet6_uadt(struct ip_set *set, struct nlattr *tb[],
+		     enum ipset_adt adt, u32 *lineno, u32 flags, bool retried)
+{
+	const struct hash_netportnet *h = set->data;
+	ipset_adtfn adtfn = set->variant->adt[adt];
+	struct hash_netportnet6_elem e = { .cidr[0] = HOST_MASK,
+					   .cidr[1] = HOST_MASK };
+	struct ip_set_ext ext = IP_SET_INIT_UEXT(set);
+	u32 port, port_to;
+	bool with_ports = false;
+	int ret;
+
+	if (unlikely(!tb[IPSET_ATTR_IP] || !tb[IPSET_ATTR_IP2] ||
+		     !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) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_CADT_FLAGS) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_PACKETS) ||
+		     !ip_set_optattr_netorder(tb, IPSET_ATTR_BYTES) ||
+		     tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
+		return -IPSET_ERR_PROTOCOL;
+	if (unlikely(tb[IPSET_ATTR_IP_TO] || tb[IPSET_ATTR_IP2_TO]))
+		return -IPSET_ERR_HASH_RANGE_UNSUPPORTED;
+
+	if (tb[IPSET_ATTR_LINENO])
+		*lineno = nla_get_u32(tb[IPSET_ATTR_LINENO]);
+
+	ret = ip_set_get_ipaddr6(tb[IPSET_ATTR_IP], &e.ip[0]) ||
+	      ip_set_get_ipaddr6(tb[IPSET_ATTR_IP2], &e.ip[1]) ||
+	      ip_set_get_extensions(set, tb, &ext);
+	if (ret)
+		return ret;
+
+	if (tb[IPSET_ATTR_CIDR])
+		e.cidr[0] = nla_get_u8(tb[IPSET_ATTR_CIDR]);
+	
+	if (tb[IPSET_ATTR_CIDR2])
+		e.cidr[1] = nla_get_u8(tb[IPSET_ATTR_CIDR2]);
+
+	if (unlikely(!e.cidr[0] || e.cidr[0] > HOST_MASK || !e.cidr[1] ||
+		     e.cidr[1] > HOST_MASK))
+		return -IPSET_ERR_INVALID_CIDR;
+
+	ip6_netmask(&e.ip[0], e.cidr[0]);
+	ip6_netmask(&e.ip[1], e.cidr[1]);
+
+	if (tb[IPSET_ATTR_PORT])
+		e.port = nla_get_be16(tb[IPSET_ATTR_PORT]);
+	else
+		return -IPSET_ERR_PROTOCOL;
+
+	if (tb[IPSET_ATTR_PROTO]) {
+		e.proto = nla_get_u8(tb[IPSET_ATTR_PROTO]);
+		with_ports = ip_set_proto_with_ports(e.proto);
+
+		if (e.proto == 0)
+			return -IPSET_ERR_INVALID_PROTO;
+	} else
+		return -IPSET_ERR_MISSING_PROTO;
+
+	if (!(with_ports || e.proto == IPPROTO_ICMPV6))
+		e.port = 0;
+
+	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 || !with_ports || !tb[IPSET_ATTR_PORT_TO]) {
+		ret = adtfn(set, &e, &ext, &ext, flags);
+		return ip_set_enomatch(ret, flags, adt, set) ? -ret :
+		       ip_set_eexist(ret, flags) ? 0 : ret;
+	}
+
+	port = ntohs(e.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++) {
+		e.port = htons(port);
+		ret = adtfn(set, &e, &ext, &ext, flags);
+
+		if (ret && !ip_set_eexist(ret, flags))
+			return ret;
+		else
+			ret = 0;
+	}
+	return ret;
+}
+
+static struct ip_set_type hash_netportnet_type __read_mostly = {
+	.name		= "hash:net,port,net",
+	.protocol	= IPSET_PROTOCOL,
+	.features	= IPSET_TYPE_IP | IPSET_TYPE_PORT | IPSET_TYPE_IP2 |
+			  IPSET_TYPE_NOMATCH,
+	.dimension	= IPSET_DIM_THREE,
+	.family		= NFPROTO_UNSPEC,
+	.revision_min	= IPSET_TYPE_REV_MIN,
+	.revision_max	= IPSET_TYPE_REV_MAX,
+	.create		= hash_netportnet_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_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_IP2]	= { .type = NLA_NESTED },
+		[IPSET_ATTR_IP2_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_CIDR2]	= { .type = NLA_U8 },
+		[IPSET_ATTR_PROTO]	= { .type = NLA_U8 },
+		[IPSET_ATTR_CADT_FLAGS]	= { .type = NLA_U32 },
+		[IPSET_ATTR_TIMEOUT]	= { .type = NLA_U32 },
+		[IPSET_ATTR_LINENO]	= { .type = NLA_U32 },
+		[IPSET_ATTR_BYTES]	= { .type = NLA_U64 },
+		[IPSET_ATTR_PACKETS]	= { .type = NLA_U64 },
+		[IPSET_ATTR_COMMENT]	= { .type = NLA_NUL_STRING },
+	},
+	.me		= THIS_MODULE,
+};
+
+static int __init
+hash_netportnet_init(void)
+{
+	return ip_set_type_register(&hash_netportnet_type);
+}
+
+static void __exit
+hash_netportnet_fini(void)
+{
+	ip_set_type_unregister(&hash_netportnet_type);
+}
+
+module_init(hash_netportnet_init);
+module_exit(hash_netportnet_fini);