|
@@ -13,6 +13,7 @@
|
|
|
#include <linux/icmpv6.h>
|
|
|
#include <linux/netfilter_ipv6/ip6_tables.h>
|
|
|
#include <net/ip.h>
|
|
|
+#include <net/ipv6.h>
|
|
|
|
|
|
#include <linux/netfilter/ipset/ip_set_getport.h>
|
|
|
|
|
@@ -93,21 +94,23 @@ ip_set_get_ip4_port(const struct sk_buff *skb, bool src,
|
|
|
}
|
|
|
EXPORT_SYMBOL_GPL(ip_set_get_ip4_port);
|
|
|
|
|
|
+#if defined(CONFIG_IP6_NF_IPTABLES) || defined(CONFIG_IP6_NF_IPTABLES_MODULE)
|
|
|
bool
|
|
|
ip_set_get_ip6_port(const struct sk_buff *skb, bool src,
|
|
|
__be16 *port, u8 *proto)
|
|
|
{
|
|
|
- unsigned int protooff = 0;
|
|
|
- int protocol;
|
|
|
- unsigned short fragoff;
|
|
|
+ int protoff;
|
|
|
+ u8 nexthdr;
|
|
|
|
|
|
- protocol = ipv6_find_hdr(skb, &protooff, -1, &fragoff);
|
|
|
- if (protocol <= 0 || fragoff)
|
|
|
+ nexthdr = ipv6_hdr(skb)->nexthdr;
|
|
|
+ protoff = ipv6_skip_exthdr(skb, sizeof(struct ipv6hdr), &nexthdr);
|
|
|
+ if (protoff < 0)
|
|
|
return false;
|
|
|
|
|
|
- return get_port(skb, protocol, protooff, src, port, proto);
|
|
|
+ return get_port(skb, nexthdr, protoff, src, port, proto);
|
|
|
}
|
|
|
EXPORT_SYMBOL_GPL(ip_set_get_ip6_port);
|
|
|
+#endif
|
|
|
|
|
|
bool
|
|
|
ip_set_get_ip_port(const struct sk_buff *skb, u8 pf, bool src, __be16 *port)
|