1/* SPDX-License-Identifier: GPL-2.0 */ 2#ifndef _NF_TABLES_IPV4_H_ 3#define _NF_TABLES_IPV4_H_ 4 5#include <net/netfilter/nf_tables.h> 6#include <net/ip.h> 7 8static inline void nft_set_pktinfo_ipv4(struct nft_pktinfo *pkt) 9{ 10 struct iphdr *ip; 11 12 ip = ip_hdr(pkt->skb); 13 pkt->flags = NFT_PKTINFO_L4PROTO; 14 pkt->tprot = ip->protocol; 15 pkt->thoff = ip_hdrlen(pkt->skb); 16 pkt->fragoff = ntohs(ip->frag_off) & IP_OFFSET; 17} 18 19static inline int __nft_set_pktinfo_ipv4_validate(struct nft_pktinfo *pkt) 20{ 21 struct iphdr *iph, _iph; 22 u32 len, thoff; 23 24 iph = skb_header_pointer(pkt->skb, skb_network_offset(pkt->skb), 25 sizeof(*iph), &_iph); 26 if (!iph) 27 return -1; 28 29 if (iph->ihl < 5 || iph->version != 4) 30 return -1; 31 32 len = iph_totlen(pkt->skb, iph); 33 thoff = skb_network_offset(pkt->skb) + (iph->ihl * 4); 34 if (pkt->skb->len < len) 35 return -1; 36 else if (len < thoff) 37 return -1; 38 else if (thoff < sizeof(*iph)) 39 return -1; 40 41 pkt->flags = NFT_PKTINFO_L4PROTO; 42 pkt->tprot = iph->protocol; 43 pkt->thoff = thoff; 44 pkt->fragoff = ntohs(iph->frag_off) & IP_OFFSET; 45 46 return 0; 47} 48 49static inline void nft_set_pktinfo_ipv4_validate(struct nft_pktinfo *pkt) 50{ 51 if (__nft_set_pktinfo_ipv4_validate(pkt) < 0) 52 nft_set_pktinfo_unspec(pkt); 53} 54 55static inline int nft_set_pktinfo_ipv4_ingress(struct nft_pktinfo *pkt) 56{ 57 struct iphdr *iph; 58 u32 len, thoff; 59 60 if (!pskb_may_pull(pkt->skb, sizeof(*iph))) 61 return -1; 62 63 iph = ip_hdr(pkt->skb); 64 if (iph->ihl < 5 || iph->version != 4) 65 goto inhdr_error; 66 67 len = iph_totlen(pkt->skb, iph); 68 thoff = iph->ihl * 4; 69 if (pkt->skb->len < len) { 70 __IP_INC_STATS(nft_net(pkt), IPSTATS_MIB_INTRUNCATEDPKTS); 71 return -1; 72 } else if (len < thoff) { 73 goto inhdr_error; 74 } else if (thoff < sizeof(*iph)) { 75 return -1; 76 } 77 78 pkt->flags = NFT_PKTINFO_L4PROTO; 79 pkt->tprot = iph->protocol; 80 pkt->thoff = thoff; 81 pkt->fragoff = ntohs(iph->frag_off) & IP_OFFSET; 82 83 return 0; 84 85inhdr_error: 86 __IP_INC_STATS(nft_net(pkt), IPSTATS_MIB_INHDRERRORS); 87 return -1; 88} 89 90#endif 91