1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _NET_FLOW_DISSECTOR_H
3 #define _NET_FLOW_DISSECTOR_H
4
5 #include <linux/types.h>
6 #include <linux/in6.h>
7 #include <linux/siphash.h>
8 #include <uapi/linux/if_ether.h>
9
10 /**
11 * struct flow_dissector_key_control:
12 * @thoff: Transport header offset
13 */
14 struct flow_dissector_key_control {
15 u16 thoff;
16 u16 addr_type;
17 u32 flags;
18 };
19
20 #define FLOW_DIS_IS_FRAGMENT BIT(0)
21 #define FLOW_DIS_FIRST_FRAG BIT(1)
22 #define FLOW_DIS_ENCAPSULATION BIT(2)
23
24 enum flow_dissect_ret {
25 FLOW_DISSECT_RET_OUT_GOOD,
26 FLOW_DISSECT_RET_OUT_BAD,
27 FLOW_DISSECT_RET_PROTO_AGAIN,
28 FLOW_DISSECT_RET_IPPROTO_AGAIN,
29 FLOW_DISSECT_RET_CONTINUE,
30 };
31
32 /**
33 * struct flow_dissector_key_basic:
34 * @thoff: Transport header offset
35 * @n_proto: Network header protocol (eg. IPv4/IPv6)
36 * @ip_proto: Transport header protocol (eg. TCP/UDP)
37 */
38 struct flow_dissector_key_basic {
39 __be16 n_proto;
40 u8 ip_proto;
41 u8 padding;
42 };
43
44 struct flow_dissector_key_tags {
45 u32 flow_label;
46 };
47
48 struct flow_dissector_key_vlan {
49 u16 vlan_id:12,
50 vlan_dei:1,
51 vlan_priority:3;
52 __be16 vlan_tpid;
53 };
54
55 struct flow_dissector_key_mpls {
56 u32 mpls_ttl:8,
57 mpls_bos:1,
58 mpls_tc:3,
59 mpls_label:20;
60 };
61
62 #define FLOW_DIS_TUN_OPTS_MAX 255
63 /**
64 * struct flow_dissector_key_enc_opts:
65 * @data: tunnel option data
66 * @len: length of tunnel option data
67 * @dst_opt_type: tunnel option type
68 */
69 struct flow_dissector_key_enc_opts {
70 u8 data[FLOW_DIS_TUN_OPTS_MAX]; /* Using IP_TUNNEL_OPTS_MAX is desired
71 * here but seems difficult to #include
72 */
73 u8 len;
74 __be16 dst_opt_type;
75 };
76
77 struct flow_dissector_key_keyid {
78 __be32 keyid;
79 };
80
81 /**
82 * struct flow_dissector_key_ipv4_addrs:
83 * @src: source ip address
84 * @dst: destination ip address
85 */
86 struct flow_dissector_key_ipv4_addrs {
87 /* (src,dst) must be grouped, in the same way than in IP header */
88 __be32 src;
89 __be32 dst;
90 };
91
92 /**
93 * struct flow_dissector_key_ipv6_addrs:
94 * @src: source ip address
95 * @dst: destination ip address
96 */
97 struct flow_dissector_key_ipv6_addrs {
98 /* (src,dst) must be grouped, in the same way than in IP header */
99 struct in6_addr src;
100 struct in6_addr dst;
101 };
102
103 /**
104 * struct flow_dissector_key_tipc:
105 * @key: source node address combined with selector
106 */
107 struct flow_dissector_key_tipc {
108 __be32 key;
109 };
110
111 /**
112 * struct flow_dissector_key_addrs:
113 * @v4addrs: IPv4 addresses
114 * @v6addrs: IPv6 addresses
115 */
116 struct flow_dissector_key_addrs {
117 union {
118 struct flow_dissector_key_ipv4_addrs v4addrs;
119 struct flow_dissector_key_ipv6_addrs v6addrs;
120 struct flow_dissector_key_tipc tipckey;
121 };
122 };
123
124 /**
125 * flow_dissector_key_arp:
126 * @ports: Operation, source and target addresses for an ARP header
127 * for Ethernet hardware addresses and IPv4 protocol addresses
128 * sip: Sender IP address
129 * tip: Target IP address
130 * op: Operation
131 * sha: Sender hardware address
132 * tpa: Target hardware address
133 */
134 struct flow_dissector_key_arp {
135 __u32 sip;
136 __u32 tip;
137 __u8 op;
138 unsigned char sha[ETH_ALEN];
139 unsigned char tha[ETH_ALEN];
140 };
141
142 /**
143 * flow_dissector_key_tp_ports:
144 * @ports: port numbers of Transport header
145 * src: source port number
146 * dst: destination port number
147 */
148 struct flow_dissector_key_ports {
149 union {
150 __be32 ports;
151 struct {
152 __be16 src;
153 __be16 dst;
154 };
155 };
156 };
157
158 /**
159 * flow_dissector_key_icmp:
160 * @ports: type and code of ICMP header
161 * icmp: ICMP type (high) and code (low)
162 * type: ICMP type
163 * code: ICMP code
164 */
165 struct flow_dissector_key_icmp {
166 union {
167 __be16 icmp;
168 struct {
169 u8 type;
170 u8 code;
171 };
172 };
173 };
174
175 /**
176 * struct flow_dissector_key_eth_addrs:
177 * @src: source Ethernet address
178 * @dst: destination Ethernet address
179 */
180 struct flow_dissector_key_eth_addrs {
181 /* (dst,src) must be grouped, in the same way than in ETH header */
182 unsigned char dst[ETH_ALEN];
183 unsigned char src[ETH_ALEN];
184 };
185
186 /**
187 * struct flow_dissector_key_tcp:
188 * @flags: flags
189 */
190 struct flow_dissector_key_tcp {
191 __be16 flags;
192 };
193
194 /**
195 * struct flow_dissector_key_ip:
196 * @tos: tos
197 * @ttl: ttl
198 */
199 struct flow_dissector_key_ip {
200 __u8 tos;
201 __u8 ttl;
202 };
203
204 /**
205 * struct flow_dissector_key_meta:
206 * @ingress_ifindex: ingress ifindex
207 */
208 struct flow_dissector_key_meta {
209 int ingress_ifindex;
210 };
211
212 /**
213 * struct flow_dissector_key_ct:
214 * @ct_state: conntrack state after converting with map
215 * @ct_mark: conttrack mark
216 * @ct_zone: conntrack zone
217 * @ct_labels: conntrack labels
218 */
219 struct flow_dissector_key_ct {
220 u16 ct_state;
221 u16 ct_zone;
222 u32 ct_mark;
223 u32 ct_labels[4];
224 };
225
226 enum flow_dissector_key_id {
227 FLOW_DISSECTOR_KEY_CONTROL, /* struct flow_dissector_key_control */
228 FLOW_DISSECTOR_KEY_BASIC, /* struct flow_dissector_key_basic */
229 FLOW_DISSECTOR_KEY_IPV4_ADDRS, /* struct flow_dissector_key_ipv4_addrs */
230 FLOW_DISSECTOR_KEY_IPV6_ADDRS, /* struct flow_dissector_key_ipv6_addrs */
231 FLOW_DISSECTOR_KEY_PORTS, /* struct flow_dissector_key_ports */
232 FLOW_DISSECTOR_KEY_ICMP, /* struct flow_dissector_key_icmp */
233 FLOW_DISSECTOR_KEY_ETH_ADDRS, /* struct flow_dissector_key_eth_addrs */
234 FLOW_DISSECTOR_KEY_TIPC, /* struct flow_dissector_key_tipc */
235 FLOW_DISSECTOR_KEY_ARP, /* struct flow_dissector_key_arp */
236 FLOW_DISSECTOR_KEY_VLAN, /* struct flow_dissector_key_vlan */
237 FLOW_DISSECTOR_KEY_FLOW_LABEL, /* struct flow_dissector_key_tags */
238 FLOW_DISSECTOR_KEY_GRE_KEYID, /* struct flow_dissector_key_keyid */
239 FLOW_DISSECTOR_KEY_MPLS_ENTROPY, /* struct flow_dissector_key_keyid */
240 FLOW_DISSECTOR_KEY_ENC_KEYID, /* struct flow_dissector_key_keyid */
241 FLOW_DISSECTOR_KEY_ENC_IPV4_ADDRS, /* struct flow_dissector_key_ipv4_addrs */
242 FLOW_DISSECTOR_KEY_ENC_IPV6_ADDRS, /* struct flow_dissector_key_ipv6_addrs */
243 FLOW_DISSECTOR_KEY_ENC_CONTROL, /* struct flow_dissector_key_control */
244 FLOW_DISSECTOR_KEY_ENC_PORTS, /* struct flow_dissector_key_ports */
245 FLOW_DISSECTOR_KEY_MPLS, /* struct flow_dissector_key_mpls */
246 FLOW_DISSECTOR_KEY_TCP, /* struct flow_dissector_key_tcp */
247 FLOW_DISSECTOR_KEY_IP, /* struct flow_dissector_key_ip */
248 FLOW_DISSECTOR_KEY_CVLAN, /* struct flow_dissector_key_vlan */
249 FLOW_DISSECTOR_KEY_ENC_IP, /* struct flow_dissector_key_ip */
250 FLOW_DISSECTOR_KEY_ENC_OPTS, /* struct flow_dissector_key_enc_opts */
251 FLOW_DISSECTOR_KEY_META, /* struct flow_dissector_key_meta */
252 FLOW_DISSECTOR_KEY_CT, /* struct flow_dissector_key_ct */
253
254 FLOW_DISSECTOR_KEY_MAX,
255 };
256
257 #define FLOW_DISSECTOR_F_PARSE_1ST_FRAG BIT(0)
258 #define FLOW_DISSECTOR_F_STOP_AT_FLOW_LABEL BIT(1)
259 #define FLOW_DISSECTOR_F_STOP_AT_ENCAP BIT(2)
260
261 struct flow_dissector_key {
262 enum flow_dissector_key_id key_id;
263 size_t offset; /* offset of struct flow_dissector_key_*
264 in target the struct */
265 };
266
267 struct flow_dissector {
268 unsigned int used_keys; /* each bit repesents presence of one key id */
269 unsigned short int offset[FLOW_DISSECTOR_KEY_MAX];
270 };
271
272 struct flow_keys_basic {
273 struct flow_dissector_key_control control;
274 struct flow_dissector_key_basic basic;
275 };
276
277 struct flow_keys {
278 struct flow_dissector_key_control control;
279 #define FLOW_KEYS_HASH_START_FIELD basic
280 struct flow_dissector_key_basic basic __aligned(SIPHASH_ALIGNMENT);
281 struct flow_dissector_key_tags tags;
282 struct flow_dissector_key_vlan vlan;
283 struct flow_dissector_key_vlan cvlan;
284 struct flow_dissector_key_keyid keyid;
285 struct flow_dissector_key_ports ports;
286 struct flow_dissector_key_addrs addrs;
287 };
288
289 #define FLOW_KEYS_HASH_OFFSET \
290 offsetof(struct flow_keys, FLOW_KEYS_HASH_START_FIELD)
291
292 __be32 flow_get_u32_src(const struct flow_keys *flow);
293 __be32 flow_get_u32_dst(const struct flow_keys *flow);
294
295 extern struct flow_dissector flow_keys_dissector;
296 extern struct flow_dissector flow_keys_basic_dissector;
297
298 /* struct flow_keys_digest:
299 *
300 * This structure is used to hold a digest of the full flow keys. This is a
301 * larger "hash" of a flow to allow definitively matching specific flows where
302 * the 32 bit skb->hash is not large enough. The size is limited to 16 bytes so
303 * that it can be used in CB of skb (see sch_choke for an example).
304 */
305 #define FLOW_KEYS_DIGEST_LEN 16
306 struct flow_keys_digest {
307 u8 data[FLOW_KEYS_DIGEST_LEN];
308 };
309
310 void make_flow_keys_digest(struct flow_keys_digest *digest,
311 const struct flow_keys *flow);
312
flow_keys_have_l4(const struct flow_keys * keys)313 static inline bool flow_keys_have_l4(const struct flow_keys *keys)
314 {
315 return (keys->ports.ports || keys->tags.flow_label);
316 }
317
318 u32 flow_hash_from_keys(struct flow_keys *keys);
319
dissector_uses_key(const struct flow_dissector * flow_dissector,enum flow_dissector_key_id key_id)320 static inline bool dissector_uses_key(const struct flow_dissector *flow_dissector,
321 enum flow_dissector_key_id key_id)
322 {
323 return flow_dissector->used_keys & (1 << key_id);
324 }
325
skb_flow_dissector_target(struct flow_dissector * flow_dissector,enum flow_dissector_key_id key_id,void * target_container)326 static inline void *skb_flow_dissector_target(struct flow_dissector *flow_dissector,
327 enum flow_dissector_key_id key_id,
328 void *target_container)
329 {
330 return ((char *)target_container) + flow_dissector->offset[key_id];
331 }
332
333 struct bpf_flow_dissector {
334 struct bpf_flow_keys *flow_keys;
335 const struct sk_buff *skb;
336 void *data;
337 void *data_end;
338 };
339
340 #endif
341