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 |
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
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 --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);