Warning: This file is not a C or C++ file. It does not have highlighting.
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 |
91 |
Warning: This file is not a C or C++ file. It does not have highlighting.