1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _NF_TABLES_IPV6_H_ 3 #define _NF_TABLES_IPV6_H_ 4 5 #include <linux/netfilter_ipv6/ip6_tables.h> 6 #include <net/ipv6.h> 7 #include <net/netfilter/nf_tables.h> 8 9 static inline void nft_set_pktinfo_ipv6(struct nft_pktinfo *pkt, 10 struct sk_buff *skb) 11 { 12 unsigned int flags = IP6_FH_F_AUTH; 13 int protohdr, thoff = 0; 14 unsigned short frag_off; 15 16 protohdr = ipv6_find_hdr(pkt->skb, &thoff, -1, &frag_off, &flags); 17 if (protohdr < 0) { 18 nft_set_pktinfo_unspec(pkt, skb); 19 return; 20 } 21 22 pkt->tprot_set = true; 23 pkt->tprot = protohdr; 24 pkt->xt.thoff = thoff; 25 pkt->xt.fragoff = frag_off; 26 } 27 28 static inline int __nft_set_pktinfo_ipv6_validate(struct nft_pktinfo *pkt, 29 struct sk_buff *skb) 30 { 31 #if IS_ENABLED(CONFIG_IPV6) 32 unsigned int flags = IP6_FH_F_AUTH; 33 struct ipv6hdr *ip6h, _ip6h; 34 unsigned int thoff = 0; 35 unsigned short frag_off; 36 int protohdr; 37 u32 pkt_len; 38 39 ip6h = skb_header_pointer(skb, skb_network_offset(skb), sizeof(*ip6h), 40 &_ip6h); 41 if (!ip6h) 42 return -1; 43 44 if (ip6h->version != 6) 45 return -1; 46 47 pkt_len = ntohs(ip6h->payload_len); 48 if (pkt_len + sizeof(*ip6h) > skb->len) 49 return -1; 50 51 protohdr = ipv6_find_hdr(pkt->skb, &thoff, -1, &frag_off, &flags); 52 if (protohdr < 0) 53 return -1; 54 55 pkt->tprot_set = true; 56 pkt->tprot = protohdr; 57 pkt->xt.thoff = thoff; 58 pkt->xt.fragoff = frag_off; 59 60 return 0; 61 #else 62 return -1; 63 #endif 64 } 65 66 static inline void nft_set_pktinfo_ipv6_validate(struct nft_pktinfo *pkt, 67 struct sk_buff *skb) 68 { 69 if (__nft_set_pktinfo_ipv6_validate(pkt, skb) < 0) 70 nft_set_pktinfo_unspec(pkt, skb); 71 } 72 73 #endif 74