• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *	IPV4 GSO/GRO offload support
4  *	Linux INET implementation
5  *
6  *	UDPv4 GSO support
7  */
8 
9 #include <linux/skbuff.h>
10 #include <net/udp.h>
11 #include <net/protocol.h>
12 #include <net/inet_common.h>
13 #include <net/ip6_checksum.h>
14 
__skb_udp_tunnel_segment(struct sk_buff * skb,netdev_features_t features,struct sk_buff * (* gso_inner_segment)(struct sk_buff * skb,netdev_features_t features),__be16 new_protocol,bool is_ipv6)15 static struct sk_buff *__skb_udp_tunnel_segment(struct sk_buff *skb,
16 	netdev_features_t features,
17 	struct sk_buff *(*gso_inner_segment)(struct sk_buff *skb,
18 					     netdev_features_t features),
19 	__be16 new_protocol, bool is_ipv6)
20 {
21 	int tnl_hlen = skb_inner_mac_header(skb) - skb_transport_header(skb);
22 	bool remcsum, need_csum, offload_csum, gso_partial;
23 	struct sk_buff *segs = ERR_PTR(-EINVAL);
24 	struct udphdr *uh = udp_hdr(skb);
25 	u16 mac_offset = skb->mac_header;
26 	__be16 protocol = skb->protocol;
27 	u16 mac_len = skb->mac_len;
28 	int udp_offset, outer_hlen;
29 	__wsum partial;
30 	bool need_ipsec;
31 
32 	if (unlikely(!pskb_may_pull(skb, tnl_hlen)))
33 		goto out;
34 
35 	/* Adjust partial header checksum to negate old length.
36 	 * We cannot rely on the value contained in uh->len as it is
37 	 * possible that the actual value exceeds the boundaries of the
38 	 * 16 bit length field due to the header being added outside of an
39 	 * IP or IPv6 frame that was already limited to 64K - 1.
40 	 */
41 	if (skb_shinfo(skb)->gso_type & SKB_GSO_PARTIAL)
42 		partial = (__force __wsum)uh->len;
43 	else
44 		partial = (__force __wsum)htonl(skb->len);
45 	partial = csum_sub(csum_unfold(uh->check), partial);
46 
47 	/* setup inner skb. */
48 	skb->encapsulation = 0;
49 	SKB_GSO_CB(skb)->encap_level = 0;
50 	__skb_pull(skb, tnl_hlen);
51 	skb_reset_mac_header(skb);
52 	skb_set_network_header(skb, skb_inner_network_offset(skb));
53 	skb->mac_len = skb_inner_network_offset(skb);
54 	skb->protocol = new_protocol;
55 
56 	need_csum = !!(skb_shinfo(skb)->gso_type & SKB_GSO_UDP_TUNNEL_CSUM);
57 	skb->encap_hdr_csum = need_csum;
58 
59 	remcsum = !!(skb_shinfo(skb)->gso_type & SKB_GSO_TUNNEL_REMCSUM);
60 	skb->remcsum_offload = remcsum;
61 
62 	need_ipsec = skb_dst(skb) && dst_xfrm(skb_dst(skb));
63 	/* Try to offload checksum if possible */
64 	offload_csum = !!(need_csum &&
65 			  !need_ipsec &&
66 			  (skb->dev->features &
67 			   (is_ipv6 ? (NETIF_F_HW_CSUM | NETIF_F_IPV6_CSUM) :
68 				      (NETIF_F_HW_CSUM | NETIF_F_IP_CSUM))));
69 
70 	features &= skb->dev->hw_enc_features;
71 
72 	/* The only checksum offload we care about from here on out is the
73 	 * outer one so strip the existing checksum feature flags and
74 	 * instead set the flag based on our outer checksum offload value.
75 	 */
76 	if (remcsum) {
77 		features &= ~NETIF_F_CSUM_MASK;
78 		if (!need_csum || offload_csum)
79 			features |= NETIF_F_HW_CSUM;
80 	}
81 
82 	/* segment inner packet. */
83 	segs = gso_inner_segment(skb, features);
84 	if (IS_ERR_OR_NULL(segs)) {
85 		skb_gso_error_unwind(skb, protocol, tnl_hlen, mac_offset,
86 				     mac_len);
87 		goto out;
88 	}
89 
90 	gso_partial = !!(skb_shinfo(segs)->gso_type & SKB_GSO_PARTIAL);
91 
92 	outer_hlen = skb_tnl_header_len(skb);
93 	udp_offset = outer_hlen - tnl_hlen;
94 	skb = segs;
95 	do {
96 		unsigned int len;
97 
98 		if (remcsum)
99 			skb->ip_summed = CHECKSUM_NONE;
100 
101 		/* Set up inner headers if we are offloading inner checksum */
102 		if (skb->ip_summed == CHECKSUM_PARTIAL) {
103 			skb_reset_inner_headers(skb);
104 			skb->encapsulation = 1;
105 		}
106 
107 		skb->mac_len = mac_len;
108 		skb->protocol = protocol;
109 
110 		__skb_push(skb, outer_hlen);
111 		skb_reset_mac_header(skb);
112 		skb_set_network_header(skb, mac_len);
113 		skb_set_transport_header(skb, udp_offset);
114 		len = skb->len - udp_offset;
115 		uh = udp_hdr(skb);
116 
117 		/* If we are only performing partial GSO the inner header
118 		 * will be using a length value equal to only one MSS sized
119 		 * segment instead of the entire frame.
120 		 */
121 		if (gso_partial && skb_is_gso(skb)) {
122 			uh->len = htons(skb_shinfo(skb)->gso_size +
123 					SKB_GSO_CB(skb)->data_offset +
124 					skb->head - (unsigned char *)uh);
125 		} else {
126 			uh->len = htons(len);
127 		}
128 
129 		if (!need_csum)
130 			continue;
131 
132 		uh->check = ~csum_fold(csum_add(partial,
133 				       (__force __wsum)htonl(len)));
134 
135 		if (skb->encapsulation || !offload_csum) {
136 			uh->check = gso_make_checksum(skb, ~uh->check);
137 			if (uh->check == 0)
138 				uh->check = CSUM_MANGLED_0;
139 		} else {
140 			skb->ip_summed = CHECKSUM_PARTIAL;
141 			skb->csum_start = skb_transport_header(skb) - skb->head;
142 			skb->csum_offset = offsetof(struct udphdr, check);
143 		}
144 	} while ((skb = skb->next));
145 out:
146 	return segs;
147 }
148 
skb_udp_tunnel_segment(struct sk_buff * skb,netdev_features_t features,bool is_ipv6)149 struct sk_buff *skb_udp_tunnel_segment(struct sk_buff *skb,
150 				       netdev_features_t features,
151 				       bool is_ipv6)
152 {
153 	__be16 protocol = skb->protocol;
154 	const struct net_offload **offloads;
155 	const struct net_offload *ops;
156 	struct sk_buff *segs = ERR_PTR(-EINVAL);
157 	struct sk_buff *(*gso_inner_segment)(struct sk_buff *skb,
158 					     netdev_features_t features);
159 
160 	rcu_read_lock();
161 
162 	switch (skb->inner_protocol_type) {
163 	case ENCAP_TYPE_ETHER:
164 		protocol = skb->inner_protocol;
165 		gso_inner_segment = skb_mac_gso_segment;
166 		break;
167 	case ENCAP_TYPE_IPPROTO:
168 		offloads = is_ipv6 ? inet6_offloads : inet_offloads;
169 		ops = rcu_dereference(offloads[skb->inner_ipproto]);
170 		if (!ops || !ops->callbacks.gso_segment)
171 			goto out_unlock;
172 		gso_inner_segment = ops->callbacks.gso_segment;
173 		break;
174 	default:
175 		goto out_unlock;
176 	}
177 
178 	segs = __skb_udp_tunnel_segment(skb, features, gso_inner_segment,
179 					protocol, is_ipv6);
180 
181 out_unlock:
182 	rcu_read_unlock();
183 
184 	return segs;
185 }
186 EXPORT_SYMBOL(skb_udp_tunnel_segment);
187 
__udpv4_gso_segment_csum(struct sk_buff * seg,__be32 * oldip,__be32 * newip,__be16 * oldport,__be16 * newport)188 static void __udpv4_gso_segment_csum(struct sk_buff *seg,
189 				     __be32 *oldip, __be32 *newip,
190 				     __be16 *oldport, __be16 *newport)
191 {
192 	struct udphdr *uh;
193 	struct iphdr *iph;
194 
195 	if (*oldip == *newip && *oldport == *newport)
196 		return;
197 
198 	uh = udp_hdr(seg);
199 	iph = ip_hdr(seg);
200 
201 	if (uh->check) {
202 		inet_proto_csum_replace4(&uh->check, seg, *oldip, *newip,
203 					 true);
204 		inet_proto_csum_replace2(&uh->check, seg, *oldport, *newport,
205 					 false);
206 		if (!uh->check)
207 			uh->check = CSUM_MANGLED_0;
208 	}
209 	*oldport = *newport;
210 
211 	csum_replace4(&iph->check, *oldip, *newip);
212 	*oldip = *newip;
213 }
214 
__udpv4_gso_segment_list_csum(struct sk_buff * segs)215 static struct sk_buff *__udpv4_gso_segment_list_csum(struct sk_buff *segs)
216 {
217 	struct sk_buff *seg;
218 	struct udphdr *uh, *uh2;
219 	struct iphdr *iph, *iph2;
220 
221 	seg = segs;
222 	uh = udp_hdr(seg);
223 	iph = ip_hdr(seg);
224 
225 	if ((udp_hdr(seg)->dest == udp_hdr(seg->next)->dest) &&
226 	    (udp_hdr(seg)->source == udp_hdr(seg->next)->source) &&
227 	    (ip_hdr(seg)->daddr == ip_hdr(seg->next)->daddr) &&
228 	    (ip_hdr(seg)->saddr == ip_hdr(seg->next)->saddr))
229 		return segs;
230 
231 	while ((seg = seg->next)) {
232 		uh2 = udp_hdr(seg);
233 		iph2 = ip_hdr(seg);
234 
235 		__udpv4_gso_segment_csum(seg,
236 					 &iph2->saddr, &iph->saddr,
237 					 &uh2->source, &uh->source);
238 		__udpv4_gso_segment_csum(seg,
239 					 &iph2->daddr, &iph->daddr,
240 					 &uh2->dest, &uh->dest);
241 	}
242 
243 	return segs;
244 }
245 
__udp_gso_segment_list(struct sk_buff * skb,netdev_features_t features,bool is_ipv6)246 static struct sk_buff *__udp_gso_segment_list(struct sk_buff *skb,
247 					      netdev_features_t features,
248 					      bool is_ipv6)
249 {
250 	unsigned int mss = skb_shinfo(skb)->gso_size;
251 
252 	skb = skb_segment_list(skb, features, skb_mac_header_len(skb));
253 	if (IS_ERR(skb))
254 		return skb;
255 
256 	udp_hdr(skb)->len = htons(sizeof(struct udphdr) + mss);
257 
258 	return is_ipv6 ? skb : __udpv4_gso_segment_list_csum(skb);
259 }
260 
__udp_gso_segment(struct sk_buff * gso_skb,netdev_features_t features,bool is_ipv6)261 struct sk_buff *__udp_gso_segment(struct sk_buff *gso_skb,
262 				  netdev_features_t features, bool is_ipv6)
263 {
264 	struct sock *sk = gso_skb->sk;
265 	unsigned int sum_truesize = 0;
266 	struct sk_buff *segs, *seg;
267 	struct udphdr *uh;
268 	unsigned int mss;
269 	bool copy_dtor;
270 	__sum16 check;
271 	__be16 newlen;
272 
273 	if (skb_shinfo(gso_skb)->gso_type & SKB_GSO_FRAGLIST) {
274 		 /* Detect modified geometry and pass those to skb_segment. */
275 		if (skb_pagelen(gso_skb) - sizeof(*uh) == skb_shinfo(gso_skb)->gso_size)
276 			return __udp_gso_segment_list(gso_skb, features, is_ipv6);
277 
278 		 /* Setup csum, as fraglist skips this in udp4_gro_receive. */
279 		gso_skb->csum_start = skb_transport_header(gso_skb) - gso_skb->head;
280 		gso_skb->csum_offset = offsetof(struct udphdr, check);
281 		gso_skb->ip_summed = CHECKSUM_PARTIAL;
282 
283 		uh = udp_hdr(gso_skb);
284 		if (is_ipv6)
285 			uh->check = ~udp_v6_check(gso_skb->len,
286 						  &ipv6_hdr(gso_skb)->saddr,
287 						  &ipv6_hdr(gso_skb)->daddr, 0);
288 		else
289 			uh->check = ~udp_v4_check(gso_skb->len,
290 						  ip_hdr(gso_skb)->saddr,
291 						  ip_hdr(gso_skb)->daddr, 0);
292 	}
293 
294 	mss = skb_shinfo(gso_skb)->gso_size;
295 	if (gso_skb->len <= sizeof(*uh) + mss)
296 		return ERR_PTR(-EINVAL);
297 
298 	skb_pull(gso_skb, sizeof(*uh));
299 
300 	/* clear destructor to avoid skb_segment assigning it to tail */
301 	copy_dtor = gso_skb->destructor == sock_wfree;
302 	if (copy_dtor)
303 		gso_skb->destructor = NULL;
304 
305 	segs = skb_segment(gso_skb, features);
306 	if (IS_ERR_OR_NULL(segs)) {
307 		if (copy_dtor)
308 			gso_skb->destructor = sock_wfree;
309 		return segs;
310 	}
311 
312 	/* GSO partial and frag_list segmentation only requires splitting
313 	 * the frame into an MSS multiple and possibly a remainder, both
314 	 * cases return a GSO skb. So update the mss now.
315 	 */
316 	if (skb_is_gso(segs))
317 		mss *= skb_shinfo(segs)->gso_segs;
318 
319 	seg = segs;
320 	uh = udp_hdr(seg);
321 
322 	/* preserve TX timestamp flags and TS key for first segment */
323 	skb_shinfo(seg)->tskey = skb_shinfo(gso_skb)->tskey;
324 	skb_shinfo(seg)->tx_flags |=
325 			(skb_shinfo(gso_skb)->tx_flags & SKBTX_ANY_TSTAMP);
326 
327 	/* compute checksum adjustment based on old length versus new */
328 	newlen = htons(sizeof(*uh) + mss);
329 	check = csum16_add(csum16_sub(uh->check, uh->len), newlen);
330 
331 	for (;;) {
332 		if (copy_dtor) {
333 			seg->destructor = sock_wfree;
334 			seg->sk = sk;
335 			sum_truesize += seg->truesize;
336 		}
337 
338 		if (!seg->next)
339 			break;
340 
341 		uh->len = newlen;
342 		uh->check = check;
343 
344 		if (seg->ip_summed == CHECKSUM_PARTIAL)
345 			gso_reset_checksum(seg, ~check);
346 		else
347 			uh->check = gso_make_checksum(seg, ~check) ? :
348 				    CSUM_MANGLED_0;
349 
350 		seg = seg->next;
351 		uh = udp_hdr(seg);
352 	}
353 
354 	/* last packet can be partial gso_size, account for that in checksum */
355 	newlen = htons(skb_tail_pointer(seg) - skb_transport_header(seg) +
356 		       seg->data_len);
357 	check = csum16_add(csum16_sub(uh->check, uh->len), newlen);
358 
359 	uh->len = newlen;
360 	uh->check = check;
361 
362 	if (seg->ip_summed == CHECKSUM_PARTIAL)
363 		gso_reset_checksum(seg, ~check);
364 	else
365 		uh->check = gso_make_checksum(seg, ~check) ? : CSUM_MANGLED_0;
366 
367 	/* update refcount for the packet */
368 	if (copy_dtor) {
369 		int delta = sum_truesize - gso_skb->truesize;
370 
371 		/* In some pathological cases, delta can be negative.
372 		 * We need to either use refcount_add() or refcount_sub_and_test()
373 		 */
374 		if (likely(delta >= 0))
375 			refcount_add(delta, &sk->sk_wmem_alloc);
376 		else
377 			WARN_ON_ONCE(refcount_sub_and_test(-delta, &sk->sk_wmem_alloc));
378 	}
379 	return segs;
380 }
381 EXPORT_SYMBOL_GPL(__udp_gso_segment);
382 
udp4_ufo_fragment(struct sk_buff * skb,netdev_features_t features)383 static struct sk_buff *udp4_ufo_fragment(struct sk_buff *skb,
384 					 netdev_features_t features)
385 {
386 	struct sk_buff *segs = ERR_PTR(-EINVAL);
387 	unsigned int mss;
388 	__wsum csum;
389 	struct udphdr *uh;
390 	struct iphdr *iph;
391 
392 	if (skb->encapsulation &&
393 	    (skb_shinfo(skb)->gso_type &
394 	     (SKB_GSO_UDP_TUNNEL|SKB_GSO_UDP_TUNNEL_CSUM))) {
395 		segs = skb_udp_tunnel_segment(skb, features, false);
396 		goto out;
397 	}
398 
399 	if (!(skb_shinfo(skb)->gso_type & (SKB_GSO_UDP | SKB_GSO_UDP_L4)))
400 		goto out;
401 
402 	if (!pskb_may_pull(skb, sizeof(struct udphdr)))
403 		goto out;
404 
405 	if (skb_shinfo(skb)->gso_type & SKB_GSO_UDP_L4)
406 		return __udp_gso_segment(skb, features, false);
407 
408 	mss = skb_shinfo(skb)->gso_size;
409 	if (unlikely(skb->len <= mss))
410 		goto out;
411 
412 	/* Do software UFO. Complete and fill in the UDP checksum as
413 	 * HW cannot do checksum of UDP packets sent as multiple
414 	 * IP fragments.
415 	 */
416 
417 	uh = udp_hdr(skb);
418 	iph = ip_hdr(skb);
419 
420 	uh->check = 0;
421 	csum = skb_checksum(skb, 0, skb->len, 0);
422 	uh->check = udp_v4_check(skb->len, iph->saddr, iph->daddr, csum);
423 	if (uh->check == 0)
424 		uh->check = CSUM_MANGLED_0;
425 
426 	skb->ip_summed = CHECKSUM_UNNECESSARY;
427 
428 	/* If there is no outer header we can fake a checksum offload
429 	 * due to the fact that we have already done the checksum in
430 	 * software prior to segmenting the frame.
431 	 */
432 	if (!skb->encap_hdr_csum)
433 		features |= NETIF_F_HW_CSUM;
434 
435 	/* Fragment the skb. IP headers of the fragments are updated in
436 	 * inet_gso_segment()
437 	 */
438 	segs = skb_segment(skb, features);
439 out:
440 	return segs;
441 }
442 
443 #define UDP_GRO_CNT_MAX 64
udp_gro_receive_segment(struct list_head * head,struct sk_buff * skb)444 static struct sk_buff *udp_gro_receive_segment(struct list_head *head,
445 					       struct sk_buff *skb)
446 {
447 	struct udphdr *uh = udp_gro_udphdr(skb);
448 	struct sk_buff *pp = NULL;
449 	struct udphdr *uh2;
450 	struct sk_buff *p;
451 	unsigned int ulen;
452 	int ret = 0;
453 
454 	/* requires non zero csum, for symmetry with GSO */
455 	if (!uh->check) {
456 		NAPI_GRO_CB(skb)->flush = 1;
457 		return NULL;
458 	}
459 
460 	/* Do not deal with padded or malicious packets, sorry ! */
461 	ulen = ntohs(uh->len);
462 	if (ulen <= sizeof(*uh) || ulen != skb_gro_len(skb)) {
463 		NAPI_GRO_CB(skb)->flush = 1;
464 		return NULL;
465 	}
466 	/* pull encapsulating udp header */
467 	skb_gro_pull(skb, sizeof(struct udphdr));
468 
469 	list_for_each_entry(p, head, list) {
470 		if (!NAPI_GRO_CB(p)->same_flow)
471 			continue;
472 
473 		uh2 = udp_hdr(p);
474 
475 		/* Match ports only, as csum is always non zero */
476 		if ((*(u32 *)&uh->source != *(u32 *)&uh2->source)) {
477 			NAPI_GRO_CB(p)->same_flow = 0;
478 			continue;
479 		}
480 
481 		if (NAPI_GRO_CB(skb)->is_flist != NAPI_GRO_CB(p)->is_flist) {
482 			NAPI_GRO_CB(skb)->flush = 1;
483 			return p;
484 		}
485 
486 		/* Terminate the flow on len mismatch or if it grow "too much".
487 		 * Under small packet flood GRO count could elsewhere grow a lot
488 		 * leading to excessive truesize values.
489 		 * On len mismatch merge the first packet shorter than gso_size,
490 		 * otherwise complete the GRO packet.
491 		 */
492 		if (ulen > ntohs(uh2->len)) {
493 			pp = p;
494 		} else {
495 			if (NAPI_GRO_CB(skb)->is_flist) {
496 				if (!pskb_may_pull(skb, skb_gro_offset(skb))) {
497 					NAPI_GRO_CB(skb)->flush = 1;
498 					return NULL;
499 				}
500 				if ((skb->ip_summed != p->ip_summed) ||
501 				    (skb->csum_level != p->csum_level)) {
502 					NAPI_GRO_CB(skb)->flush = 1;
503 					return NULL;
504 				}
505 				ret = skb_gro_receive_list(p, skb);
506 			} else {
507 				skb_gro_postpull_rcsum(skb, uh,
508 						       sizeof(struct udphdr));
509 
510 				ret = skb_gro_receive(p, skb);
511 			}
512 		}
513 
514 		if (ret || ulen != ntohs(uh2->len) ||
515 		    NAPI_GRO_CB(p)->count >= UDP_GRO_CNT_MAX)
516 			pp = p;
517 
518 		return pp;
519 	}
520 
521 	/* mismatch, but we never need to flush */
522 	return NULL;
523 }
524 
udp_gro_receive(struct list_head * head,struct sk_buff * skb,struct udphdr * uh,struct sock * sk)525 struct sk_buff *udp_gro_receive(struct list_head *head, struct sk_buff *skb,
526 				struct udphdr *uh, struct sock *sk)
527 {
528 	struct sk_buff *pp = NULL;
529 	struct sk_buff *p;
530 	struct udphdr *uh2;
531 	unsigned int off = skb_gro_offset(skb);
532 	int flush = 1;
533 
534 	/* We can do L4 aggregation only if the packet can't land in a tunnel
535 	 * otherwise we could corrupt the inner stream. Detecting such packets
536 	 * cannot be foolproof and the aggregation might still happen in some
537 	 * cases. Such packets should be caught in udp_unexpected_gso later.
538 	 */
539 	NAPI_GRO_CB(skb)->is_flist = 0;
540 	if (skb->dev->features & NETIF_F_GRO_FRAGLIST)
541 		NAPI_GRO_CB(skb)->is_flist = sk ? !udp_sk(sk)->gro_enabled: 1;
542 
543 	if ((sk && udp_sk(sk)->gro_enabled) || NAPI_GRO_CB(skb)->is_flist) {
544 		pp = call_gro_receive(udp_gro_receive_segment, head, skb);
545 		return pp;
546 	}
547 
548 	if (!sk || NAPI_GRO_CB(skb)->encap_mark ||
549 	    (uh->check && skb->ip_summed != CHECKSUM_PARTIAL &&
550 	     NAPI_GRO_CB(skb)->csum_cnt == 0 &&
551 	     !NAPI_GRO_CB(skb)->csum_valid) ||
552 	    !udp_sk(sk)->gro_receive)
553 		goto out;
554 
555 	/* mark that this skb passed once through the tunnel gro layer */
556 	NAPI_GRO_CB(skb)->encap_mark = 1;
557 
558 	flush = 0;
559 
560 	list_for_each_entry(p, head, list) {
561 		if (!NAPI_GRO_CB(p)->same_flow)
562 			continue;
563 
564 		uh2 = (struct udphdr   *)(p->data + off);
565 
566 		/* Match ports and either checksums are either both zero
567 		 * or nonzero.
568 		 */
569 		if ((*(u32 *)&uh->source != *(u32 *)&uh2->source) ||
570 		    (!uh->check ^ !uh2->check)) {
571 			NAPI_GRO_CB(p)->same_flow = 0;
572 			continue;
573 		}
574 	}
575 
576 	skb_gro_pull(skb, sizeof(struct udphdr)); /* pull encapsulating udp header */
577 	skb_gro_postpull_rcsum(skb, uh, sizeof(struct udphdr));
578 	pp = call_gro_receive_sk(udp_sk(sk)->gro_receive, sk, head, skb);
579 
580 out:
581 	skb_gro_flush_final(skb, pp, flush);
582 	return pp;
583 }
584 EXPORT_SYMBOL(udp_gro_receive);
585 
udp4_gro_lookup_skb(struct sk_buff * skb,__be16 sport,__be16 dport)586 static struct sock *udp4_gro_lookup_skb(struct sk_buff *skb, __be16 sport,
587 					__be16 dport)
588 {
589 	const struct iphdr *iph = skb_gro_network_header(skb);
590 
591 	return __udp4_lib_lookup(dev_net(skb->dev), iph->saddr, sport,
592 				 iph->daddr, dport, inet_iif(skb),
593 				 inet_sdif(skb), &udp_table, NULL);
594 }
595 
596 INDIRECT_CALLABLE_SCOPE
udp4_gro_receive(struct list_head * head,struct sk_buff * skb)597 struct sk_buff *udp4_gro_receive(struct list_head *head, struct sk_buff *skb)
598 {
599 	struct udphdr *uh = udp_gro_udphdr(skb);
600 	struct sock *sk = NULL;
601 	struct sk_buff *pp;
602 
603 	if (unlikely(!uh))
604 		goto flush;
605 
606 	/* Don't bother verifying checksum if we're going to flush anyway. */
607 	if (NAPI_GRO_CB(skb)->flush)
608 		goto skip;
609 
610 	if (skb_gro_checksum_validate_zero_check(skb, IPPROTO_UDP, uh->check,
611 						 inet_gro_compute_pseudo))
612 		goto flush;
613 	else if (uh->check)
614 		skb_gro_checksum_try_convert(skb, IPPROTO_UDP,
615 					     inet_gro_compute_pseudo);
616 skip:
617 	NAPI_GRO_CB(skb)->is_ipv6 = 0;
618 	rcu_read_lock();
619 
620 	if (static_branch_unlikely(&udp_encap_needed_key))
621 		sk = udp4_gro_lookup_skb(skb, uh->source, uh->dest);
622 
623 	pp = udp_gro_receive(head, skb, uh, sk);
624 	rcu_read_unlock();
625 	return pp;
626 
627 flush:
628 	NAPI_GRO_CB(skb)->flush = 1;
629 	return NULL;
630 }
631 
udp_gro_complete_segment(struct sk_buff * skb)632 static int udp_gro_complete_segment(struct sk_buff *skb)
633 {
634 	struct udphdr *uh = udp_hdr(skb);
635 
636 	skb->csum_start = (unsigned char *)uh - skb->head;
637 	skb->csum_offset = offsetof(struct udphdr, check);
638 	skb->ip_summed = CHECKSUM_PARTIAL;
639 
640 	skb_shinfo(skb)->gso_segs = NAPI_GRO_CB(skb)->count;
641 	skb_shinfo(skb)->gso_type |= SKB_GSO_UDP_L4;
642 
643 	if (skb->encapsulation)
644 		skb->inner_transport_header = skb->transport_header;
645 
646 	return 0;
647 }
648 
udp_gro_complete(struct sk_buff * skb,int nhoff,udp_lookup_t lookup)649 int udp_gro_complete(struct sk_buff *skb, int nhoff,
650 		     udp_lookup_t lookup)
651 {
652 	__be16 newlen = htons(skb->len - nhoff);
653 	struct udphdr *uh = (struct udphdr *)(skb->data + nhoff);
654 	int err = -ENOSYS;
655 	struct sock *sk;
656 
657 	uh->len = newlen;
658 
659 	rcu_read_lock();
660 	sk = INDIRECT_CALL_INET(lookup, udp6_lib_lookup_skb,
661 				udp4_lib_lookup_skb, skb, uh->source, uh->dest);
662 	if (sk && udp_sk(sk)->gro_complete) {
663 		skb_shinfo(skb)->gso_type = uh->check ? SKB_GSO_UDP_TUNNEL_CSUM
664 					: SKB_GSO_UDP_TUNNEL;
665 
666 		/* Set encapsulation before calling into inner gro_complete()
667 		 * functions to make them set up the inner offsets.
668 		 */
669 		skb->encapsulation = 1;
670 		err = udp_sk(sk)->gro_complete(sk, skb,
671 				nhoff + sizeof(struct udphdr));
672 	} else {
673 		err = udp_gro_complete_segment(skb);
674 	}
675 	rcu_read_unlock();
676 
677 	if (skb->remcsum_offload)
678 		skb_shinfo(skb)->gso_type |= SKB_GSO_TUNNEL_REMCSUM;
679 
680 	return err;
681 }
682 EXPORT_SYMBOL(udp_gro_complete);
683 
udp4_gro_complete(struct sk_buff * skb,int nhoff)684 INDIRECT_CALLABLE_SCOPE int udp4_gro_complete(struct sk_buff *skb, int nhoff)
685 {
686 	const struct iphdr *iph = ip_hdr(skb);
687 	struct udphdr *uh = (struct udphdr *)(skb->data + nhoff);
688 
689 	if (NAPI_GRO_CB(skb)->is_flist) {
690 		uh->len = htons(skb->len - nhoff);
691 
692 		skb_shinfo(skb)->gso_type |= (SKB_GSO_FRAGLIST|SKB_GSO_UDP_L4);
693 		skb_shinfo(skb)->gso_segs = NAPI_GRO_CB(skb)->count;
694 
695 		if (skb->ip_summed == CHECKSUM_UNNECESSARY) {
696 			if (skb->csum_level < SKB_MAX_CSUM_LEVEL)
697 				skb->csum_level++;
698 		} else {
699 			skb->ip_summed = CHECKSUM_UNNECESSARY;
700 			skb->csum_level = 0;
701 		}
702 
703 		return 0;
704 	}
705 
706 	if (uh->check)
707 		uh->check = ~udp_v4_check(skb->len - nhoff, iph->saddr,
708 					  iph->daddr, 0);
709 
710 	return udp_gro_complete(skb, nhoff, udp4_lib_lookup_skb);
711 }
712 
713 static const struct net_offload udpv4_offload = {
714 	.callbacks = {
715 		.gso_segment = udp4_ufo_fragment,
716 		.gro_receive  =	udp4_gro_receive,
717 		.gro_complete =	udp4_gro_complete,
718 	},
719 };
720 
udpv4_offload_init(void)721 int __init udpv4_offload_init(void)
722 {
723 	return inet_add_offload(&udpv4_offload, IPPROTO_UDP);
724 }
725