include/net/netfilter/nf_tables_ipv4.h

   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
   8 static 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
  19 static 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
  49 static 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
  55 static 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
  85 inhdr_error:
  86         __IP_INC_STATS(nft_net(pkt), IPSTATS_MIB_INHDRERRORS);
  87         return -1;
  88 }
  89
  90 #endif