xref: /linux/net/core/gso.c (revision 006c8b834aed8b02bd8fb9fb05e9e7ead37d1763)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 #include <linux/skbuff.h>
3 #include <linux/sctp.h>
4 #include <net/gso.h>
5 #include <net/gro.h>
6 
7 /**
8  *	skb_eth_gso_segment - segmentation handler for ethernet protocols.
9  *	@skb: buffer to segment
10  *	@features: features for the output path (see dev->features)
11  *	@type: Ethernet Protocol ID
12  */
13 struct sk_buff *skb_eth_gso_segment(struct sk_buff *skb,
14 				    netdev_features_t features, __be16 type)
15 {
16 	struct sk_buff *segs = ERR_PTR(-EPROTONOSUPPORT);
17 	struct packet_offload *ptype;
18 
19 	rcu_read_lock();
20 	list_for_each_entry_rcu(ptype, &net_hotdata.offload_base, list) {
21 		if (ptype->type == type && ptype->callbacks.gso_segment) {
22 			segs = ptype->callbacks.gso_segment(skb, features);
23 			break;
24 		}
25 	}
26 	rcu_read_unlock();
27 
28 	return segs;
29 }
30 EXPORT_SYMBOL(skb_eth_gso_segment);
31 
32 /**
33  *	skb_mac_gso_segment - mac layer segmentation handler.
34  *	@skb: buffer to segment
35  *	@features: features for the output path (see dev->features)
36  */
37 struct sk_buff *skb_mac_gso_segment(struct sk_buff *skb,
38 				    netdev_features_t features)
39 {
40 	struct sk_buff *segs = ERR_PTR(-EPROTONOSUPPORT);
41 	struct packet_offload *ptype;
42 	int vlan_depth = skb->mac_len;
43 	__be16 type = skb_network_protocol(skb, &vlan_depth);
44 
45 	if (unlikely(!type))
46 		return ERR_PTR(-EINVAL);
47 
48 	__skb_pull(skb, vlan_depth);
49 
50 	rcu_read_lock();
51 	list_for_each_entry_rcu(ptype, &net_hotdata.offload_base, list) {
52 		if (ptype->type == type && ptype->callbacks.gso_segment) {
53 			segs = ptype->callbacks.gso_segment(skb, features);
54 			break;
55 		}
56 	}
57 	rcu_read_unlock();
58 
59 	__skb_push(skb, skb->data - skb_mac_header(skb));
60 
61 	return segs;
62 }
63 EXPORT_SYMBOL(skb_mac_gso_segment);
64 /* openvswitch calls this on rx path, so we need a different check.
65  */
66 static bool skb_needs_check(const struct sk_buff *skb, bool tx_path)
67 {
68 	if (tx_path)
69 		return skb->ip_summed != CHECKSUM_PARTIAL &&
70 		       skb->ip_summed != CHECKSUM_UNNECESSARY;
71 
72 	return skb->ip_summed == CHECKSUM_NONE;
73 }
74 
75 /**
76  *	__skb_gso_segment - Perform segmentation on skb.
77  *	@skb: buffer to segment
78  *	@features: features for the output path (see dev->features)
79  *	@tx_path: whether it is called in TX path
80  *
81  *	This function segments the given skb and returns a list of segments.
82  *
83  *	It may return NULL if the skb requires no segmentation.  This is
84  *	only possible when GSO is used for verifying header integrity.
85  *
86  *	Segmentation preserves SKB_GSO_CB_OFFSET bytes of previous skb cb.
87  */
88 struct sk_buff *__skb_gso_segment(struct sk_buff *skb,
89 				  netdev_features_t features, bool tx_path)
90 {
91 	struct sk_buff *segs;
92 
93 	if (unlikely(skb_needs_check(skb, tx_path))) {
94 		int err;
95 
96 		/* We're going to init ->check field in TCP or UDP header */
97 		err = skb_cow_head(skb, 0);
98 		if (err < 0)
99 			return ERR_PTR(err);
100 	}
101 
102 	/* Only report GSO partial support if it will enable us to
103 	 * support segmentation on this frame without needing additional
104 	 * work.
105 	 */
106 	if (features & NETIF_F_GSO_PARTIAL) {
107 		netdev_features_t partial_features = NETIF_F_GSO_ROBUST;
108 		struct net_device *dev = skb->dev;
109 
110 		partial_features |= dev->features & dev->gso_partial_features;
111 		if (!skb_gso_ok(skb, features | partial_features))
112 			features &= ~NETIF_F_GSO_PARTIAL;
113 	}
114 
115 	BUILD_BUG_ON(SKB_GSO_CB_OFFSET +
116 		     sizeof(*SKB_GSO_CB(skb)) > sizeof(skb->cb));
117 
118 	SKB_GSO_CB(skb)->mac_offset = skb_headroom(skb);
119 	SKB_GSO_CB(skb)->encap_level = 0;
120 	SKB_GSO_CB(skb)->recursion_counter = 0;
121 
122 	skb_reset_mac_header(skb);
123 	skb_reset_mac_len(skb);
124 
125 	segs = skb_mac_gso_segment(skb, features);
126 
127 	if (segs != skb && unlikely(skb_needs_check(skb, tx_path) && !IS_ERR(segs)))
128 		skb_warn_bad_offload(skb);
129 
130 	return segs;
131 }
132 EXPORT_SYMBOL(__skb_gso_segment);
133 
134 /**
135  * skb_gso_transport_seglen - Return length of individual segments of a gso packet
136  *
137  * @skb: GSO skb
138  *
139  * skb_gso_transport_seglen is used to determine the real size of the
140  * individual segments, including Layer4 headers (TCP/UDP).
141  *
142  * The MAC/L2 or network (IP, IPv6) headers are not accounted for.
143  */
144 static unsigned int skb_gso_transport_seglen(const struct sk_buff *skb)
145 {
146 	const struct skb_shared_info *shinfo = skb_shinfo(skb);
147 	unsigned int thlen = 0;
148 
149 	if (skb->encapsulation) {
150 		thlen = skb_inner_transport_header(skb) -
151 			skb_transport_header(skb);
152 
153 		if (likely(shinfo->gso_type & (SKB_GSO_TCPV4 | SKB_GSO_TCPV6)))
154 			thlen += inner_tcp_hdrlen(skb);
155 	} else if (likely(shinfo->gso_type & (SKB_GSO_TCPV4 | SKB_GSO_TCPV6))) {
156 		thlen = tcp_hdrlen(skb);
157 	} else if (unlikely(skb_is_gso_sctp(skb))) {
158 		thlen = sizeof(struct sctphdr);
159 	} else if (shinfo->gso_type & SKB_GSO_UDP_L4) {
160 		thlen = sizeof(struct udphdr);
161 	}
162 	/* UFO sets gso_size to the size of the fragmentation
163 	 * payload, i.e. the size of the L4 (UDP) header is already
164 	 * accounted for.
165 	 */
166 	return thlen + shinfo->gso_size;
167 }
168 
169 /**
170  * skb_gso_network_seglen - Return length of individual segments of a gso packet
171  *
172  * @skb: GSO skb
173  *
174  * skb_gso_network_seglen is used to determine the real size of the
175  * individual segments, including Layer3 (IP, IPv6) and L4 headers (TCP/UDP).
176  *
177  * The MAC/L2 header is not accounted for.
178  */
179 static unsigned int skb_gso_network_seglen(const struct sk_buff *skb)
180 {
181 	unsigned int hdr_len = skb_transport_header(skb) -
182 			       skb_network_header(skb);
183 
184 	return hdr_len + skb_gso_transport_seglen(skb);
185 }
186 
187 /**
188  * skb_gso_mac_seglen - Return length of individual segments of a gso packet
189  *
190  * @skb: GSO skb
191  *
192  * skb_gso_mac_seglen is used to determine the real size of the
193  * individual segments, including MAC/L2, Layer3 (IP, IPv6) and L4
194  * headers (TCP/UDP).
195  */
196 static unsigned int skb_gso_mac_seglen(const struct sk_buff *skb)
197 {
198 	unsigned int hdr_len = skb_transport_header(skb) - skb_mac_header(skb);
199 
200 	return hdr_len + skb_gso_transport_seglen(skb);
201 }
202 
203 /**
204  * skb_gso_size_check - check the skb size, considering GSO_BY_FRAGS
205  *
206  * There are a couple of instances where we have a GSO skb, and we
207  * want to determine what size it would be after it is segmented.
208  *
209  * We might want to check:
210  * -    L3+L4+payload size (e.g. IP forwarding)
211  * - L2+L3+L4+payload size (e.g. sanity check before passing to driver)
212  *
213  * This is a helper to do that correctly considering GSO_BY_FRAGS.
214  *
215  * @skb: GSO skb
216  *
217  * @seg_len: The segmented length (from skb_gso_*_seglen). In the
218  *           GSO_BY_FRAGS case this will be [header sizes + GSO_BY_FRAGS].
219  *
220  * @max_len: The maximum permissible length.
221  *
222  * Returns true if the segmented length <= max length.
223  */
224 static inline bool skb_gso_size_check(const struct sk_buff *skb,
225 				      unsigned int seg_len,
226 				      unsigned int max_len) {
227 	const struct skb_shared_info *shinfo = skb_shinfo(skb);
228 	const struct sk_buff *iter;
229 
230 	if (shinfo->gso_size != GSO_BY_FRAGS)
231 		return seg_len <= max_len;
232 
233 	/* Undo this so we can re-use header sizes */
234 	seg_len -= GSO_BY_FRAGS;
235 
236 	skb_walk_frags(skb, iter) {
237 		if (seg_len + skb_headlen(iter) > max_len)
238 			return false;
239 	}
240 
241 	return true;
242 }
243 
244 /**
245  * skb_gso_validate_network_len - Will a split GSO skb fit into a given MTU?
246  *
247  * @skb: GSO skb
248  * @mtu: MTU to validate against
249  *
250  * skb_gso_validate_network_len validates if a given skb will fit a
251  * wanted MTU once split. It considers L3 headers, L4 headers, and the
252  * payload.
253  */
254 bool skb_gso_validate_network_len(const struct sk_buff *skb, unsigned int mtu)
255 {
256 	return skb_gso_size_check(skb, skb_gso_network_seglen(skb), mtu);
257 }
258 EXPORT_SYMBOL_GPL(skb_gso_validate_network_len);
259 
260 /**
261  * skb_gso_validate_mac_len - Will a split GSO skb fit in a given length?
262  *
263  * @skb: GSO skb
264  * @len: length to validate against
265  *
266  * skb_gso_validate_mac_len validates if a given skb will fit a wanted
267  * length once split, including L2, L3 and L4 headers and the payload.
268  */
269 bool skb_gso_validate_mac_len(const struct sk_buff *skb, unsigned int len)
270 {
271 	return skb_gso_size_check(skb, skb_gso_mac_seglen(skb), len);
272 }
273 EXPORT_SYMBOL_GPL(skb_gso_validate_mac_len);
274 
275