blob: 7164569c1ec8076a89c29e378ca769344d8122c4 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * Routines having to do with the 'struct sk_buff' memory handlers.
3 *
Alan Cox113aa832008-10-13 19:01:08 -07004 * Authors: Alan Cox <alan@lxorguk.ukuu.org.uk>
Linus Torvalds1da177e2005-04-16 15:20:36 -07005 * Florian La Roche <rzsfl@rz.uni-sb.de>
6 *
Linus Torvalds1da177e2005-04-16 15:20:36 -07007 * Fixes:
8 * Alan Cox : Fixed the worst of the load
9 * balancer bugs.
10 * Dave Platt : Interrupt stacking fix.
11 * Richard Kooijman : Timestamp fixes.
12 * Alan Cox : Changed buffer format.
13 * Alan Cox : destructor hook for AF_UNIX etc.
14 * Linus Torvalds : Better skb_clone.
15 * Alan Cox : Added skb_copy.
16 * Alan Cox : Added all the changed routines Linus
17 * only put in the headers
18 * Ray VanTassle : Fixed --skb->lock in free
19 * Alan Cox : skb_copy copy arp field
20 * Andi Kleen : slabified it.
21 * Robert Olsson : Removed skb_head_pool
22 *
23 * NOTE:
24 * The __skb_ routines should be called with interrupts
25 * disabled, or you better be *real* sure that the operation is atomic
26 * with respect to whatever list is being frobbed (e.g. via lock_sock()
27 * or via disabling bottom half handlers, etc).
28 *
29 * This program is free software; you can redistribute it and/or
30 * modify it under the terms of the GNU General Public License
31 * as published by the Free Software Foundation; either version
32 * 2 of the License, or (at your option) any later version.
33 */
34
35/*
36 * The functions in this file will not compile correctly with gcc 2.4.x
37 */
38
Joe Perchese005d192012-05-16 19:58:40 +000039#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
40
Linus Torvalds1da177e2005-04-16 15:20:36 -070041#include <linux/module.h>
42#include <linux/types.h>
43#include <linux/kernel.h>
Vegard Nossumfe55f6d2008-08-30 12:16:35 +020044#include <linux/kmemcheck.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070045#include <linux/mm.h>
46#include <linux/interrupt.h>
47#include <linux/in.h>
48#include <linux/inet.h>
49#include <linux/slab.h>
Florian Westphalde960aa2014-01-26 10:58:16 +010050#include <linux/tcp.h>
51#include <linux/udp.h>
Marcelo Ricardo Leitner90017ac2016-06-02 15:05:43 -030052#include <linux/sctp.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070053#include <linux/netdevice.h>
54#ifdef CONFIG_NET_CLS_ACT
55#include <net/pkt_sched.h>
56#endif
57#include <linux/string.h>
58#include <linux/skbuff.h>
Jens Axboe9c55e012007-11-06 23:30:13 -080059#include <linux/splice.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070060#include <linux/cache.h>
61#include <linux/rtnetlink.h>
62#include <linux/init.h>
David Howells716ea3a2007-04-02 20:19:53 -070063#include <linux/scatterlist.h>
Patrick Ohlyac45f602009-02-12 05:03:37 +000064#include <linux/errqueue.h>
Linus Torvalds268bb0c2011-05-20 12:50:29 -070065#include <linux/prefetch.h>
Vlad Yasevich0d5501c2014-08-08 14:42:13 -040066#include <linux/if_vlan.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070067
68#include <net/protocol.h>
69#include <net/dst.h>
70#include <net/sock.h>
71#include <net/checksum.h>
Paul Durranted1f50c2014-01-09 10:02:46 +000072#include <net/ip6_checksum.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070073#include <net/xfrm.h>
74
75#include <asm/uaccess.h>
Steven Rostedtad8d75f2009-04-14 19:39:12 -040076#include <trace/events/skb.h>
Eric Dumazet51c56b02012-04-05 11:35:15 +020077#include <linux/highmem.h>
Willem de Bruijnb245be12015-01-30 13:29:32 -050078#include <linux/capability.h>
79#include <linux/user_namespace.h>
Al Viroa1f8e7f72006-10-19 16:08:53 -040080
Eric Dumazetd7e88832012-04-30 08:10:34 +000081struct kmem_cache *skbuff_head_cache __read_mostly;
Christoph Lametere18b8902006-12-06 20:33:20 -080082static struct kmem_cache *skbuff_fclone_cache __read_mostly;
Hans Westgaard Ry5f74f822016-02-03 09:26:57 +010083int sysctl_max_skb_frags __read_mostly = MAX_SKB_FRAGS;
84EXPORT_SYMBOL(sysctl_max_skb_frags);
Linus Torvalds1da177e2005-04-16 15:20:36 -070085
Linus Torvalds1da177e2005-04-16 15:20:36 -070086/**
Jean Sacrenf05de732013-02-11 13:30:38 +000087 * skb_panic - private function for out-of-line support
88 * @skb: buffer
89 * @sz: size
90 * @addr: address
James Hogan99d58512013-02-13 11:20:27 +000091 * @msg: skb_over_panic or skb_under_panic
Linus Torvalds1da177e2005-04-16 15:20:36 -070092 *
Jean Sacrenf05de732013-02-11 13:30:38 +000093 * Out-of-line support for skb_put() and skb_push().
94 * Called via the wrapper skb_over_panic() or skb_under_panic().
95 * Keep out of line to prevent kernel bloat.
96 * __builtin_return_address is not used because it is not always reliable.
Linus Torvalds1da177e2005-04-16 15:20:36 -070097 */
Jean Sacrenf05de732013-02-11 13:30:38 +000098static void skb_panic(struct sk_buff *skb, unsigned int sz, void *addr,
James Hogan99d58512013-02-13 11:20:27 +000099 const char msg[])
Linus Torvalds1da177e2005-04-16 15:20:36 -0700100{
Joe Perchese005d192012-05-16 19:58:40 +0000101 pr_emerg("%s: text:%p len:%d put:%d head:%p data:%p tail:%#lx end:%#lx dev:%s\n",
James Hogan99d58512013-02-13 11:20:27 +0000102 msg, addr, skb->len, sz, skb->head, skb->data,
Joe Perchese005d192012-05-16 19:58:40 +0000103 (unsigned long)skb->tail, (unsigned long)skb->end,
104 skb->dev ? skb->dev->name : "<NULL>");
Linus Torvalds1da177e2005-04-16 15:20:36 -0700105 BUG();
106}
107
Jean Sacrenf05de732013-02-11 13:30:38 +0000108static void skb_over_panic(struct sk_buff *skb, unsigned int sz, void *addr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700109{
Jean Sacrenf05de732013-02-11 13:30:38 +0000110 skb_panic(skb, sz, addr, __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700111}
112
Jean Sacrenf05de732013-02-11 13:30:38 +0000113static void skb_under_panic(struct sk_buff *skb, unsigned int sz, void *addr)
114{
115 skb_panic(skb, sz, addr, __func__);
116}
Mel Gormanc93bdd02012-07-31 16:44:19 -0700117
118/*
119 * kmalloc_reserve is a wrapper around kmalloc_node_track_caller that tells
120 * the caller if emergency pfmemalloc reserves are being used. If it is and
121 * the socket is later found to be SOCK_MEMALLOC then PFMEMALLOC reserves
122 * may be used. Otherwise, the packet data may be discarded until enough
123 * memory is free
124 */
125#define kmalloc_reserve(size, gfp, node, pfmemalloc) \
126 __kmalloc_reserve(size, gfp, node, _RET_IP_, pfmemalloc)
stephen hemminger61c5e882012-12-28 18:24:28 +0000127
128static void *__kmalloc_reserve(size_t size, gfp_t flags, int node,
129 unsigned long ip, bool *pfmemalloc)
Mel Gormanc93bdd02012-07-31 16:44:19 -0700130{
131 void *obj;
132 bool ret_pfmemalloc = false;
133
134 /*
135 * Try a regular allocation, when that fails and we're not entitled
136 * to the reserves, fail.
137 */
138 obj = kmalloc_node_track_caller(size,
139 flags | __GFP_NOMEMALLOC | __GFP_NOWARN,
140 node);
141 if (obj || !(gfp_pfmemalloc_allowed(flags)))
142 goto out;
143
144 /* Try again but now we are using pfmemalloc reserves */
145 ret_pfmemalloc = true;
146 obj = kmalloc_node_track_caller(size, flags, node);
147
148out:
149 if (pfmemalloc)
150 *pfmemalloc = ret_pfmemalloc;
151
152 return obj;
153}
154
Linus Torvalds1da177e2005-04-16 15:20:36 -0700155/* Allocate a new skbuff. We do this ourselves so we can fill in a few
156 * 'private' fields and also do memory statistics to find all the
157 * [BEEP] leaks.
158 *
159 */
160
Patrick McHardy0ebd0ac2013-04-17 06:46:58 +0000161struct sk_buff *__alloc_skb_head(gfp_t gfp_mask, int node)
162{
163 struct sk_buff *skb;
164
165 /* Get the HEAD */
166 skb = kmem_cache_alloc_node(skbuff_head_cache,
167 gfp_mask & ~__GFP_DMA, node);
168 if (!skb)
169 goto out;
170
171 /*
172 * Only clear those fields we need to clear, not those that we will
173 * actually initialise below. Hence, don't put any more fields after
174 * the tail pointer in struct sk_buff!
175 */
176 memset(skb, 0, offsetof(struct sk_buff, tail));
Pablo Neira5e71d9d2013-06-03 09:28:43 +0000177 skb->head = NULL;
Patrick McHardy0ebd0ac2013-04-17 06:46:58 +0000178 skb->truesize = sizeof(struct sk_buff);
179 atomic_set(&skb->users, 1);
180
Cong Wang35d04612013-05-29 15:16:05 +0800181 skb->mac_header = (typeof(skb->mac_header))~0U;
Patrick McHardy0ebd0ac2013-04-17 06:46:58 +0000182out:
183 return skb;
184}
185
Linus Torvalds1da177e2005-04-16 15:20:36 -0700186/**
David S. Millerd179cd12005-08-17 14:57:30 -0700187 * __alloc_skb - allocate a network buffer
Linus Torvalds1da177e2005-04-16 15:20:36 -0700188 * @size: size to allocate
189 * @gfp_mask: allocation mask
Mel Gormanc93bdd02012-07-31 16:44:19 -0700190 * @flags: If SKB_ALLOC_FCLONE is set, allocate from fclone cache
191 * instead of head cache and allocate a cloned (child) skb.
192 * If SKB_ALLOC_RX is set, __GFP_MEMALLOC will be used for
193 * allocations in case the data is required for writeback
Christoph Hellwigb30973f2006-12-06 20:32:36 -0800194 * @node: numa node to allocate memory on
Linus Torvalds1da177e2005-04-16 15:20:36 -0700195 *
196 * Allocate a new &sk_buff. The returned buffer has no headroom and a
Ben Hutchings94b60422012-06-06 15:23:37 +0000197 * tail room of at least size bytes. The object has a reference count
198 * of one. The return is the buffer. On a failure the return is %NULL.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700199 *
200 * Buffers may only be allocated from interrupts using a @gfp_mask of
201 * %GFP_ATOMIC.
202 */
Al Virodd0fc662005-10-07 07:46:04 +0100203struct sk_buff *__alloc_skb(unsigned int size, gfp_t gfp_mask,
Mel Gormanc93bdd02012-07-31 16:44:19 -0700204 int flags, int node)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700205{
Christoph Lametere18b8902006-12-06 20:33:20 -0800206 struct kmem_cache *cache;
Benjamin LaHaise4947d3e2006-01-03 14:06:50 -0800207 struct skb_shared_info *shinfo;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700208 struct sk_buff *skb;
209 u8 *data;
Mel Gormanc93bdd02012-07-31 16:44:19 -0700210 bool pfmemalloc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700211
Mel Gormanc93bdd02012-07-31 16:44:19 -0700212 cache = (flags & SKB_ALLOC_FCLONE)
213 ? skbuff_fclone_cache : skbuff_head_cache;
214
215 if (sk_memalloc_socks() && (flags & SKB_ALLOC_RX))
216 gfp_mask |= __GFP_MEMALLOC;
Herbert Xu8798b3f2006-01-23 16:32:45 -0800217
Linus Torvalds1da177e2005-04-16 15:20:36 -0700218 /* Get the HEAD */
Christoph Hellwigb30973f2006-12-06 20:32:36 -0800219 skb = kmem_cache_alloc_node(cache, gfp_mask & ~__GFP_DMA, node);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700220 if (!skb)
221 goto out;
Eric Dumazetec7d2f22010-05-05 01:07:37 -0700222 prefetchw(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700223
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000224 /* We do our best to align skb_shared_info on a separate cache
225 * line. It usually works because kmalloc(X > SMP_CACHE_BYTES) gives
226 * aligned memory blocks, unless SLUB/SLAB debug is enabled.
227 * Both skb->head and skb_shared_info are cache line aligned.
228 */
Tony Lindgrenbc417e32011-11-02 13:40:28 +0000229 size = SKB_DATA_ALIGN(size);
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000230 size += SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
Mel Gormanc93bdd02012-07-31 16:44:19 -0700231 data = kmalloc_reserve(size, gfp_mask, node, &pfmemalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700232 if (!data)
233 goto nodata;
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000234 /* kmalloc(size) might give us more room than requested.
235 * Put skb_shared_info exactly at the end of allocated zone,
236 * to allow max possible filling before reallocation.
237 */
238 size = SKB_WITH_OVERHEAD(ksize(data));
Eric Dumazetec7d2f22010-05-05 01:07:37 -0700239 prefetchw(data + size);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700240
Arnaldo Carvalho de Meloca0605a2007-03-19 10:48:59 -0300241 /*
Johannes Bergc8005782008-05-03 20:56:42 -0700242 * Only clear those fields we need to clear, not those that we will
243 * actually initialise below. Hence, don't put any more fields after
244 * the tail pointer in struct sk_buff!
Arnaldo Carvalho de Meloca0605a2007-03-19 10:48:59 -0300245 */
246 memset(skb, 0, offsetof(struct sk_buff, tail));
Eric Dumazet87fb4b72011-10-13 07:28:54 +0000247 /* Account for allocated memory : skb + skb->head */
248 skb->truesize = SKB_TRUESIZE(size);
Mel Gormanc93bdd02012-07-31 16:44:19 -0700249 skb->pfmemalloc = pfmemalloc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700250 atomic_set(&skb->users, 1);
251 skb->head = data;
252 skb->data = data;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700253 skb_reset_tail_pointer(skb);
Arnaldo Carvalho de Melo4305b542007-04-19 20:43:29 -0700254 skb->end = skb->tail + size;
Cong Wang35d04612013-05-29 15:16:05 +0800255 skb->mac_header = (typeof(skb->mac_header))~0U;
256 skb->transport_header = (typeof(skb->transport_header))~0U;
Stephen Hemminger19633e12009-06-17 05:23:27 +0000257
Benjamin LaHaise4947d3e2006-01-03 14:06:50 -0800258 /* make sure we initialize shinfo sequentially */
259 shinfo = skb_shinfo(skb);
Eric Dumazetec7d2f22010-05-05 01:07:37 -0700260 memset(shinfo, 0, offsetof(struct skb_shared_info, dataref));
Benjamin LaHaise4947d3e2006-01-03 14:06:50 -0800261 atomic_set(&shinfo->dataref, 1);
Eric Dumazetc2aa3662011-01-25 23:18:38 +0000262 kmemcheck_annotate_variable(shinfo->destructor_arg);
Benjamin LaHaise4947d3e2006-01-03 14:06:50 -0800263
Mel Gormanc93bdd02012-07-31 16:44:19 -0700264 if (flags & SKB_ALLOC_FCLONE) {
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700265 struct sk_buff_fclones *fclones;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700266
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700267 fclones = container_of(skb, struct sk_buff_fclones, skb1);
268
269 kmemcheck_annotate_bitfield(&fclones->skb2, flags1);
David S. Millerd179cd12005-08-17 14:57:30 -0700270 skb->fclone = SKB_FCLONE_ORIG;
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700271 atomic_set(&fclones->fclone_ref, 1);
David S. Millerd179cd12005-08-17 14:57:30 -0700272
Eric Dumazet6ffe75e2014-12-03 17:04:39 -0800273 fclones->skb2.fclone = SKB_FCLONE_CLONE;
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700274 fclones->skb2.pfmemalloc = pfmemalloc;
David S. Millerd179cd12005-08-17 14:57:30 -0700275 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700276out:
277 return skb;
278nodata:
Herbert Xu8798b3f2006-01-23 16:32:45 -0800279 kmem_cache_free(cache, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700280 skb = NULL;
281 goto out;
282}
David S. Millerb4ac530fc2009-02-10 02:09:24 -0800283EXPORT_SYMBOL(__alloc_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700284
285/**
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700286 * __build_skb - build a network buffer
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000287 * @data: data buffer provided by caller
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700288 * @frag_size: size of data, or 0 if head was kmalloced
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000289 *
290 * Allocate a new &sk_buff. Caller provides space holding head and
Florian Fainellideceb4c2013-07-23 20:22:39 +0100291 * skb_shared_info. @data must have been allocated by kmalloc() only if
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700292 * @frag_size is 0, otherwise data should come from the page allocator
293 * or vmalloc()
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000294 * The return is the new skb buffer.
295 * On a failure the return is %NULL, and @data is not freed.
296 * Notes :
297 * Before IO, driver allocates only data buffer where NIC put incoming frame
298 * Driver should add room at head (NET_SKB_PAD) and
299 * MUST add room at tail (SKB_DATA_ALIGN(skb_shared_info))
300 * After IO, driver calls build_skb(), to allocate sk_buff and populate it
301 * before giving packet to stack.
302 * RX rings only contains data buffers, not full skbs.
303 */
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700304struct sk_buff *__build_skb(void *data, unsigned int frag_size)
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000305{
306 struct skb_shared_info *shinfo;
307 struct sk_buff *skb;
Eric Dumazetd3836f22012-04-27 00:33:38 +0000308 unsigned int size = frag_size ? : ksize(data);
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000309
310 skb = kmem_cache_alloc(skbuff_head_cache, GFP_ATOMIC);
311 if (!skb)
312 return NULL;
313
Eric Dumazetd3836f22012-04-27 00:33:38 +0000314 size -= SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000315
316 memset(skb, 0, offsetof(struct sk_buff, tail));
317 skb->truesize = SKB_TRUESIZE(size);
318 atomic_set(&skb->users, 1);
319 skb->head = data;
320 skb->data = data;
321 skb_reset_tail_pointer(skb);
322 skb->end = skb->tail + size;
Cong Wang35d04612013-05-29 15:16:05 +0800323 skb->mac_header = (typeof(skb->mac_header))~0U;
324 skb->transport_header = (typeof(skb->transport_header))~0U;
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000325
326 /* make sure we initialize shinfo sequentially */
327 shinfo = skb_shinfo(skb);
328 memset(shinfo, 0, offsetof(struct skb_shared_info, dataref));
329 atomic_set(&shinfo->dataref, 1);
330 kmemcheck_annotate_variable(shinfo->destructor_arg);
331
332 return skb;
333}
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700334
335/* build_skb() is wrapper over __build_skb(), that specifically
336 * takes care of skb->head and skb->pfmemalloc
337 * This means that if @frag_size is not zero, then @data must be backed
338 * by a page fragment, not kmalloc() or vmalloc()
339 */
340struct sk_buff *build_skb(void *data, unsigned int frag_size)
341{
342 struct sk_buff *skb = __build_skb(data, frag_size);
343
344 if (skb && frag_size) {
345 skb->head_frag = 1;
Michal Hocko2f064f32015-08-21 14:11:51 -0700346 if (page_is_pfmemalloc(virt_to_head_page(data)))
Eric Dumazet2ea2f622015-04-24 16:05:01 -0700347 skb->pfmemalloc = 1;
348 }
349 return skb;
350}
Eric Dumazetb2b5ce92011-11-14 06:03:34 +0000351EXPORT_SYMBOL(build_skb);
352
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100353#define NAPI_SKB_CACHE_SIZE 64
354
355struct napi_alloc_cache {
356 struct page_frag_cache page;
357 size_t skb_count;
358 void *skb_cache[NAPI_SKB_CACHE_SIZE];
359};
360
Alexander Duyckb63ae8c2015-05-06 21:11:57 -0700361static DEFINE_PER_CPU(struct page_frag_cache, netdev_alloc_cache);
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100362static DEFINE_PER_CPU(struct napi_alloc_cache, napi_alloc_cache);
Alexander Duyckffde7322014-12-09 19:40:42 -0800363
364static void *__netdev_alloc_frag(unsigned int fragsz, gfp_t gfp_mask)
365{
Alexander Duyckb63ae8c2015-05-06 21:11:57 -0700366 struct page_frag_cache *nc;
Alexander Duyckffde7322014-12-09 19:40:42 -0800367 unsigned long flags;
368 void *data;
369
370 local_irq_save(flags);
Alexander Duyck94519802015-05-06 21:11:40 -0700371 nc = this_cpu_ptr(&netdev_alloc_cache);
372 data = __alloc_page_frag(nc, fragsz, gfp_mask);
Eric Dumazet6f532612012-05-18 05:12:12 +0000373 local_irq_restore(flags);
374 return data;
375}
Mel Gormanc93bdd02012-07-31 16:44:19 -0700376
377/**
378 * netdev_alloc_frag - allocate a page fragment
379 * @fragsz: fragment size
380 *
381 * Allocates a frag from a page for receive buffer.
382 * Uses GFP_ATOMIC allocations.
383 */
384void *netdev_alloc_frag(unsigned int fragsz)
385{
Alexander Duyck8b4ceed2019-02-15 14:44:18 -0800386 fragsz = SKB_DATA_ALIGN(fragsz);
387
Mel Gormanc93bdd02012-07-31 16:44:19 -0700388 return __netdev_alloc_frag(fragsz, GFP_ATOMIC | __GFP_COLD);
389}
Eric Dumazet6f532612012-05-18 05:12:12 +0000390EXPORT_SYMBOL(netdev_alloc_frag);
391
Alexander Duyckffde7322014-12-09 19:40:42 -0800392static void *__napi_alloc_frag(unsigned int fragsz, gfp_t gfp_mask)
393{
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100394 struct napi_alloc_cache *nc = this_cpu_ptr(&napi_alloc_cache);
Alexander Duyck94519802015-05-06 21:11:40 -0700395
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100396 return __alloc_page_frag(&nc->page, fragsz, gfp_mask);
Alexander Duyckffde7322014-12-09 19:40:42 -0800397}
398
399void *napi_alloc_frag(unsigned int fragsz)
400{
Alexander Duyck8b4ceed2019-02-15 14:44:18 -0800401 fragsz = SKB_DATA_ALIGN(fragsz);
402
Alexander Duyckffde7322014-12-09 19:40:42 -0800403 return __napi_alloc_frag(fragsz, GFP_ATOMIC | __GFP_COLD);
404}
405EXPORT_SYMBOL(napi_alloc_frag);
406
Eric Dumazet6f532612012-05-18 05:12:12 +0000407/**
Alexander Duyckfd11a832014-12-09 19:40:49 -0800408 * __netdev_alloc_skb - allocate an skbuff for rx on a specific device
409 * @dev: network device to receive on
Masanari Iidad7499162015-08-24 22:56:54 +0900410 * @len: length to allocate
Alexander Duyckfd11a832014-12-09 19:40:49 -0800411 * @gfp_mask: get_free_pages mask, passed to alloc_skb
412 *
413 * Allocate a new &sk_buff and assign it a usage count of one. The
414 * buffer has NET_SKB_PAD headroom built in. Users should allocate
415 * the headroom they think they need without accounting for the
416 * built in space. The built in space is used for optimisations.
417 *
418 * %NULL is returned if there is no free memory.
419 */
Alexander Duyck94519802015-05-06 21:11:40 -0700420struct sk_buff *__netdev_alloc_skb(struct net_device *dev, unsigned int len,
421 gfp_t gfp_mask)
Alexander Duyckfd11a832014-12-09 19:40:49 -0800422{
Alexander Duyckb63ae8c2015-05-06 21:11:57 -0700423 struct page_frag_cache *nc;
Alexander Duyck94519802015-05-06 21:11:40 -0700424 unsigned long flags;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800425 struct sk_buff *skb;
Alexander Duyck94519802015-05-06 21:11:40 -0700426 bool pfmemalloc;
427 void *data;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800428
Alexander Duyck94519802015-05-06 21:11:40 -0700429 len += NET_SKB_PAD;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800430
Alexander Duyck94519802015-05-06 21:11:40 -0700431 if ((len > SKB_WITH_OVERHEAD(PAGE_SIZE)) ||
Mel Gormand0164ad2015-11-06 16:28:21 -0800432 (gfp_mask & (__GFP_DIRECT_RECLAIM | GFP_DMA))) {
Alexander Duycka080e7b2015-05-13 13:34:13 -0700433 skb = __alloc_skb(len, gfp_mask, SKB_ALLOC_RX, NUMA_NO_NODE);
434 if (!skb)
435 goto skb_fail;
436 goto skb_success;
437 }
Alexander Duyck94519802015-05-06 21:11:40 -0700438
439 len += SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
440 len = SKB_DATA_ALIGN(len);
441
442 if (sk_memalloc_socks())
443 gfp_mask |= __GFP_MEMALLOC;
444
445 local_irq_save(flags);
446
447 nc = this_cpu_ptr(&netdev_alloc_cache);
448 data = __alloc_page_frag(nc, len, gfp_mask);
449 pfmemalloc = nc->pfmemalloc;
450
451 local_irq_restore(flags);
452
453 if (unlikely(!data))
454 return NULL;
455
456 skb = __build_skb(data, len);
457 if (unlikely(!skb)) {
Alexander Duyck181edb22015-05-06 21:12:03 -0700458 skb_free_frag(data);
Alexander Duyck94519802015-05-06 21:11:40 -0700459 return NULL;
Christoph Hellwig7b2e4972006-08-07 16:09:04 -0700460 }
Alexander Duyckfd11a832014-12-09 19:40:49 -0800461
Alexander Duyck94519802015-05-06 21:11:40 -0700462 /* use OR instead of assignment to avoid clearing of bits in mask */
463 if (pfmemalloc)
464 skb->pfmemalloc = 1;
465 skb->head_frag = 1;
466
Alexander Duycka080e7b2015-05-13 13:34:13 -0700467skb_success:
Alexander Duyck94519802015-05-06 21:11:40 -0700468 skb_reserve(skb, NET_SKB_PAD);
469 skb->dev = dev;
470
Alexander Duycka080e7b2015-05-13 13:34:13 -0700471skb_fail:
Christoph Hellwig8af27452006-07-31 22:35:23 -0700472 return skb;
473}
David S. Millerb4ac530fc2009-02-10 02:09:24 -0800474EXPORT_SYMBOL(__netdev_alloc_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700475
Alexander Duyckfd11a832014-12-09 19:40:49 -0800476/**
477 * __napi_alloc_skb - allocate skbuff for rx in a specific NAPI instance
478 * @napi: napi instance this buffer was allocated for
Masanari Iidad7499162015-08-24 22:56:54 +0900479 * @len: length to allocate
Alexander Duyckfd11a832014-12-09 19:40:49 -0800480 * @gfp_mask: get_free_pages mask, passed to alloc_skb and alloc_pages
481 *
482 * Allocate a new sk_buff for use in NAPI receive. This buffer will
483 * attempt to allocate the head from a special reserved region used
484 * only for NAPI Rx allocation. By doing this we can save several
485 * CPU cycles by avoiding having to disable and re-enable IRQs.
486 *
487 * %NULL is returned if there is no free memory.
488 */
Alexander Duyck94519802015-05-06 21:11:40 -0700489struct sk_buff *__napi_alloc_skb(struct napi_struct *napi, unsigned int len,
490 gfp_t gfp_mask)
Alexander Duyckfd11a832014-12-09 19:40:49 -0800491{
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100492 struct napi_alloc_cache *nc = this_cpu_ptr(&napi_alloc_cache);
Alexander Duyckfd11a832014-12-09 19:40:49 -0800493 struct sk_buff *skb;
Alexander Duyck94519802015-05-06 21:11:40 -0700494 void *data;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800495
Alexander Duyck94519802015-05-06 21:11:40 -0700496 len += NET_SKB_PAD + NET_IP_ALIGN;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800497
Alexander Duyck94519802015-05-06 21:11:40 -0700498 if ((len > SKB_WITH_OVERHEAD(PAGE_SIZE)) ||
Mel Gormand0164ad2015-11-06 16:28:21 -0800499 (gfp_mask & (__GFP_DIRECT_RECLAIM | GFP_DMA))) {
Alexander Duycka080e7b2015-05-13 13:34:13 -0700500 skb = __alloc_skb(len, gfp_mask, SKB_ALLOC_RX, NUMA_NO_NODE);
501 if (!skb)
502 goto skb_fail;
503 goto skb_success;
504 }
Alexander Duyck94519802015-05-06 21:11:40 -0700505
506 len += SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
507 len = SKB_DATA_ALIGN(len);
508
509 if (sk_memalloc_socks())
510 gfp_mask |= __GFP_MEMALLOC;
511
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100512 data = __alloc_page_frag(&nc->page, len, gfp_mask);
Alexander Duyck94519802015-05-06 21:11:40 -0700513 if (unlikely(!data))
514 return NULL;
515
516 skb = __build_skb(data, len);
517 if (unlikely(!skb)) {
Alexander Duyck181edb22015-05-06 21:12:03 -0700518 skb_free_frag(data);
Alexander Duyck94519802015-05-06 21:11:40 -0700519 return NULL;
Alexander Duyckfd11a832014-12-09 19:40:49 -0800520 }
521
Alexander Duyck94519802015-05-06 21:11:40 -0700522 /* use OR instead of assignment to avoid clearing of bits in mask */
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100523 if (nc->page.pfmemalloc)
Alexander Duyck94519802015-05-06 21:11:40 -0700524 skb->pfmemalloc = 1;
525 skb->head_frag = 1;
526
Alexander Duycka080e7b2015-05-13 13:34:13 -0700527skb_success:
Alexander Duyck94519802015-05-06 21:11:40 -0700528 skb_reserve(skb, NET_SKB_PAD + NET_IP_ALIGN);
529 skb->dev = napi->dev;
530
Alexander Duycka080e7b2015-05-13 13:34:13 -0700531skb_fail:
Alexander Duyckfd11a832014-12-09 19:40:49 -0800532 return skb;
533}
534EXPORT_SYMBOL(__napi_alloc_skb);
535
Peter Zijlstra654bed12008-10-07 14:22:33 -0700536void skb_add_rx_frag(struct sk_buff *skb, int i, struct page *page, int off,
Eric Dumazet50269e12012-03-23 23:59:33 +0000537 int size, unsigned int truesize)
Peter Zijlstra654bed12008-10-07 14:22:33 -0700538{
539 skb_fill_page_desc(skb, i, page, off, size);
540 skb->len += size;
541 skb->data_len += size;
Eric Dumazet50269e12012-03-23 23:59:33 +0000542 skb->truesize += truesize;
Peter Zijlstra654bed12008-10-07 14:22:33 -0700543}
544EXPORT_SYMBOL(skb_add_rx_frag);
545
Jason Wangf8e617e2013-11-01 14:07:47 +0800546void skb_coalesce_rx_frag(struct sk_buff *skb, int i, int size,
547 unsigned int truesize)
548{
549 skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
550
551 skb_frag_size_add(frag, size);
552 skb->len += size;
553 skb->data_len += size;
554 skb->truesize += truesize;
555}
556EXPORT_SYMBOL(skb_coalesce_rx_frag);
557
Herbert Xu27b437c2006-07-13 19:26:39 -0700558static void skb_drop_list(struct sk_buff **listp)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700559{
Eric Dumazetbd8a7032013-06-24 06:26:00 -0700560 kfree_skb_list(*listp);
Herbert Xu27b437c2006-07-13 19:26:39 -0700561 *listp = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700562}
563
Herbert Xu27b437c2006-07-13 19:26:39 -0700564static inline void skb_drop_fraglist(struct sk_buff *skb)
565{
566 skb_drop_list(&skb_shinfo(skb)->frag_list);
567}
568
Linus Torvalds1da177e2005-04-16 15:20:36 -0700569static void skb_clone_fraglist(struct sk_buff *skb)
570{
571 struct sk_buff *list;
572
David S. Millerfbb398a2009-06-09 00:18:59 -0700573 skb_walk_frags(skb, list)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700574 skb_get(list);
575}
576
Eric Dumazetd3836f22012-04-27 00:33:38 +0000577static void skb_free_head(struct sk_buff *skb)
578{
Alexander Duyck181edb22015-05-06 21:12:03 -0700579 unsigned char *head = skb->head;
580
Eric Dumazetd3836f22012-04-27 00:33:38 +0000581 if (skb->head_frag)
Alexander Duyck181edb22015-05-06 21:12:03 -0700582 skb_free_frag(head);
Eric Dumazetd3836f22012-04-27 00:33:38 +0000583 else
Alexander Duyck181edb22015-05-06 21:12:03 -0700584 kfree(head);
Eric Dumazetd3836f22012-04-27 00:33:38 +0000585}
586
Adrian Bunk5bba1712006-06-29 13:02:35 -0700587static void skb_release_data(struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700588{
Eric Dumazetff04a772014-09-23 18:39:30 -0700589 struct skb_shared_info *shinfo = skb_shinfo(skb);
590 int i;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700591
Eric Dumazetff04a772014-09-23 18:39:30 -0700592 if (skb->cloned &&
593 atomic_sub_return(skb->nohdr ? (1 << SKB_DATAREF_SHIFT) + 1 : 1,
594 &shinfo->dataref))
595 return;
Shirley Maa6686f22011-07-06 12:22:12 +0000596
Eric Dumazetff04a772014-09-23 18:39:30 -0700597 for (i = 0; i < shinfo->nr_frags; i++)
598 __skb_frag_unref(&shinfo->frags[i]);
Shirley Maa6686f22011-07-06 12:22:12 +0000599
Eric Dumazetff04a772014-09-23 18:39:30 -0700600 /*
601 * If skb buf is from userspace, we need to notify the caller
602 * the lower device DMA has done;
603 */
604 if (shinfo->tx_flags & SKBTX_DEV_ZEROCOPY) {
605 struct ubuf_info *uarg;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700606
Eric Dumazetff04a772014-09-23 18:39:30 -0700607 uarg = shinfo->destructor_arg;
608 if (uarg->callback)
609 uarg->callback(uarg, true);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700610 }
Eric Dumazetff04a772014-09-23 18:39:30 -0700611
612 if (shinfo->frag_list)
613 kfree_skb_list(shinfo->frag_list);
614
615 skb_free_head(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700616}
617
618/*
619 * Free an skbuff by memory without cleaning the state.
620 */
Herbert Xu2d4baff2007-11-26 23:11:19 +0800621static void kfree_skbmem(struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700622{
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700623 struct sk_buff_fclones *fclones;
David S. Millerd179cd12005-08-17 14:57:30 -0700624
David S. Millerd179cd12005-08-17 14:57:30 -0700625 switch (skb->fclone) {
626 case SKB_FCLONE_UNAVAILABLE:
627 kmem_cache_free(skbuff_head_cache, skb);
Eric Dumazet6ffe75e2014-12-03 17:04:39 -0800628 return;
David S. Millerd179cd12005-08-17 14:57:30 -0700629
630 case SKB_FCLONE_ORIG:
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700631 fclones = container_of(skb, struct sk_buff_fclones, skb1);
Eric Dumazet6ffe75e2014-12-03 17:04:39 -0800632
633 /* We usually free the clone (TX completion) before original skb
634 * This test would have no chance to be true for the clone,
635 * while here, branch prediction will be good.
636 */
637 if (atomic_read(&fclones->fclone_ref) == 1)
638 goto fastpath;
David S. Millerd179cd12005-08-17 14:57:30 -0700639 break;
640
Eric Dumazet6ffe75e2014-12-03 17:04:39 -0800641 default: /* SKB_FCLONE_CLONE */
Eric Dumazetd0bf4a92014-09-29 13:29:15 -0700642 fclones = container_of(skb, struct sk_buff_fclones, skb2);
David S. Millerd179cd12005-08-17 14:57:30 -0700643 break;
Stephen Hemminger3ff50b72007-04-20 17:09:22 -0700644 }
Eric Dumazet6ffe75e2014-12-03 17:04:39 -0800645 if (!atomic_dec_and_test(&fclones->fclone_ref))
646 return;
647fastpath:
648 kmem_cache_free(skbuff_fclone_cache, fclones);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700649}
650
Lennert Buytenhek04a4bb52008-10-01 02:33:12 -0700651static void skb_release_head_state(struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700652{
Eric Dumazetadf30902009-06-02 05:19:30 +0000653 skb_dst_drop(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700654#ifdef CONFIG_XFRM
655 secpath_put(skb->sp);
656#endif
Stephen Hemminger9c2b3322005-04-19 22:39:42 -0700657 if (skb->destructor) {
658 WARN_ON(in_irq());
Linus Torvalds1da177e2005-04-16 15:20:36 -0700659 skb->destructor(skb);
660 }
Igor Maravića3bf7ae2011-12-12 02:58:22 +0000661#if IS_ENABLED(CONFIG_NF_CONNTRACK)
Yasuyuki Kozakai5f79e0f2007-03-23 11:17:07 -0700662 nf_conntrack_put(skb->nfct);
KOVACS Krisztian2fc72c72011-01-12 20:25:08 +0100663#endif
Pablo Neira Ayuso1109a902014-10-01 11:19:17 +0200664#if IS_ENABLED(CONFIG_BRIDGE_NETFILTER)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700665 nf_bridge_put(skb->nf_bridge);
666#endif
Lennert Buytenhek04a4bb52008-10-01 02:33:12 -0700667}
668
669/* Free everything but the sk_buff shell. */
670static void skb_release_all(struct sk_buff *skb)
671{
672 skb_release_head_state(skb);
Pablo Neira5e71d9d2013-06-03 09:28:43 +0000673 if (likely(skb->head))
Patrick McHardy0ebd0ac2013-04-17 06:46:58 +0000674 skb_release_data(skb);
Herbert Xu2d4baff2007-11-26 23:11:19 +0800675}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700676
Herbert Xu2d4baff2007-11-26 23:11:19 +0800677/**
678 * __kfree_skb - private function
679 * @skb: buffer
680 *
681 * Free an sk_buff. Release anything attached to the buffer.
682 * Clean the state. This is an internal helper function. Users should
683 * always call kfree_skb
684 */
685
686void __kfree_skb(struct sk_buff *skb)
687{
688 skb_release_all(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700689 kfree_skbmem(skb);
690}
David S. Millerb4ac530fc2009-02-10 02:09:24 -0800691EXPORT_SYMBOL(__kfree_skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700692
693/**
Jörn Engel231d06a2006-03-20 21:28:35 -0800694 * kfree_skb - free an sk_buff
695 * @skb: buffer to free
696 *
697 * Drop a reference to the buffer and free it if the usage count has
698 * hit zero.
699 */
700void kfree_skb(struct sk_buff *skb)
701{
702 if (unlikely(!skb))
703 return;
704 if (likely(atomic_read(&skb->users) == 1))
705 smp_rmb();
706 else if (likely(!atomic_dec_and_test(&skb->users)))
707 return;
Neil Hormanead2ceb2009-03-11 09:49:55 +0000708 trace_kfree_skb(skb, __builtin_return_address(0));
Jörn Engel231d06a2006-03-20 21:28:35 -0800709 __kfree_skb(skb);
710}
David S. Millerb4ac530fc2009-02-10 02:09:24 -0800711EXPORT_SYMBOL(kfree_skb);
Jörn Engel231d06a2006-03-20 21:28:35 -0800712
Eric Dumazetbd8a7032013-06-24 06:26:00 -0700713void kfree_skb_list(struct sk_buff *segs)
714{
715 while (segs) {
716 struct sk_buff *next = segs->next;
717
718 kfree_skb(segs);
719 segs = next;
720 }
721}
722EXPORT_SYMBOL(kfree_skb_list);
723
Stephen Hemmingerd1a203e2008-11-01 21:01:09 -0700724/**
Michael S. Tsirkin25121172012-11-01 09:16:28 +0000725 * skb_tx_error - report an sk_buff xmit error
726 * @skb: buffer that triggered an error
727 *
728 * Report xmit error if a device callback is tracking this skb.
729 * skb must be freed afterwards.
730 */
731void skb_tx_error(struct sk_buff *skb)
732{
733 if (skb_shinfo(skb)->tx_flags & SKBTX_DEV_ZEROCOPY) {
734 struct ubuf_info *uarg;
735
736 uarg = skb_shinfo(skb)->destructor_arg;
737 if (uarg->callback)
738 uarg->callback(uarg, false);
739 skb_shinfo(skb)->tx_flags &= ~SKBTX_DEV_ZEROCOPY;
740 }
741}
742EXPORT_SYMBOL(skb_tx_error);
743
744/**
Neil Hormanead2ceb2009-03-11 09:49:55 +0000745 * consume_skb - free an skbuff
746 * @skb: buffer to free
747 *
748 * Drop a ref to the buffer and free it if the usage count has hit zero
749 * Functions identically to kfree_skb, but kfree_skb assumes that the frame
750 * is being dropped after a failure and notes that
751 */
752void consume_skb(struct sk_buff *skb)
753{
754 if (unlikely(!skb))
755 return;
756 if (likely(atomic_read(&skb->users) == 1))
757 smp_rmb();
758 else if (likely(!atomic_dec_and_test(&skb->users)))
759 return;
Koki Sanagi07dc22e2010-08-23 18:46:12 +0900760 trace_consume_skb(skb);
Neil Hormanead2ceb2009-03-11 09:49:55 +0000761 __kfree_skb(skb);
762}
763EXPORT_SYMBOL(consume_skb);
764
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100765void __kfree_skb_flush(void)
766{
767 struct napi_alloc_cache *nc = this_cpu_ptr(&napi_alloc_cache);
768
769 /* flush skb_cache if containing objects */
770 if (nc->skb_count) {
771 kmem_cache_free_bulk(skbuff_head_cache, nc->skb_count,
772 nc->skb_cache);
773 nc->skb_count = 0;
774 }
775}
776
Jesper Dangaard Brouer15fad712016-02-08 13:15:04 +0100777static inline void _kfree_skb_defer(struct sk_buff *skb)
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100778{
779 struct napi_alloc_cache *nc = this_cpu_ptr(&napi_alloc_cache);
780
781 /* drop skb->head and call any destructors for packet */
782 skb_release_all(skb);
783
784 /* record skb to CPU local list */
785 nc->skb_cache[nc->skb_count++] = skb;
786
787#ifdef CONFIG_SLUB
788 /* SLUB writes into objects when freeing */
789 prefetchw(skb);
790#endif
791
792 /* flush skb_cache if it is filled */
793 if (unlikely(nc->skb_count == NAPI_SKB_CACHE_SIZE)) {
794 kmem_cache_free_bulk(skbuff_head_cache, NAPI_SKB_CACHE_SIZE,
795 nc->skb_cache);
796 nc->skb_count = 0;
797 }
798}
Jesper Dangaard Brouer15fad712016-02-08 13:15:04 +0100799void __kfree_skb_defer(struct sk_buff *skb)
800{
801 _kfree_skb_defer(skb);
802}
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100803
804void napi_consume_skb(struct sk_buff *skb, int budget)
805{
806 if (unlikely(!skb))
807 return;
808
Jesper Dangaard Brouer885eb0a2016-03-11 09:43:58 +0100809 /* Zero budget indicate non-NAPI context called us, like netpoll */
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100810 if (unlikely(!budget)) {
Jesper Dangaard Brouer885eb0a2016-03-11 09:43:58 +0100811 dev_consume_skb_any(skb);
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100812 return;
813 }
814
815 if (likely(atomic_read(&skb->users) == 1))
816 smp_rmb();
817 else if (likely(!atomic_dec_and_test(&skb->users)))
818 return;
819 /* if reaching here SKB is ready to free */
820 trace_consume_skb(skb);
821
822 /* if SKB is a clone, don't handle this case */
Eric Dumazetabbdb5a2016-03-20 11:27:47 -0700823 if (skb->fclone != SKB_FCLONE_UNAVAILABLE) {
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100824 __kfree_skb(skb);
825 return;
826 }
827
Jesper Dangaard Brouer15fad712016-02-08 13:15:04 +0100828 _kfree_skb_defer(skb);
Jesper Dangaard Brouer795bb1c2016-02-08 13:14:59 +0100829}
830EXPORT_SYMBOL(napi_consume_skb);
831
Eric Dumazetb1937222014-09-28 22:18:47 -0700832/* Make sure a field is enclosed inside headers_start/headers_end section */
833#define CHECK_SKB_FIELD(field) \
834 BUILD_BUG_ON(offsetof(struct sk_buff, field) < \
835 offsetof(struct sk_buff, headers_start)); \
836 BUILD_BUG_ON(offsetof(struct sk_buff, field) > \
837 offsetof(struct sk_buff, headers_end)); \
838
Herbert Xudec18812007-10-14 00:37:30 -0700839static void __copy_skb_header(struct sk_buff *new, const struct sk_buff *old)
840{
841 new->tstamp = old->tstamp;
Eric Dumazetb1937222014-09-28 22:18:47 -0700842 /* We do not copy old->sk */
Herbert Xudec18812007-10-14 00:37:30 -0700843 new->dev = old->dev;
Eric Dumazetb1937222014-09-28 22:18:47 -0700844 memcpy(new->cb, old->cb, sizeof(old->cb));
Eric Dumazet7fee2262010-05-11 23:19:48 +0000845 skb_dst_copy(new, old);
Alexey Dobriyandef8b4f2008-10-28 13:24:06 -0700846#ifdef CONFIG_XFRM
Herbert Xudec18812007-10-14 00:37:30 -0700847 new->sp = secpath_get(old->sp);
848#endif
Eric Dumazetb1937222014-09-28 22:18:47 -0700849 __nf_copy(new, old, false);
Patrick McHardy6aa895b02008-07-14 22:49:06 -0700850
Eric Dumazetb1937222014-09-28 22:18:47 -0700851 /* Note : this field could be in headers_start/headers_end section
852 * It is not yet because we do not want to have a 16 bit hole
853 */
854 new->queue_mapping = old->queue_mapping;
Eliezer Tamir06021292013-06-10 11:39:50 +0300855
Eric Dumazetb1937222014-09-28 22:18:47 -0700856 memcpy(&new->headers_start, &old->headers_start,
857 offsetof(struct sk_buff, headers_end) -
858 offsetof(struct sk_buff, headers_start));
859 CHECK_SKB_FIELD(protocol);
860 CHECK_SKB_FIELD(csum);
861 CHECK_SKB_FIELD(hash);
862 CHECK_SKB_FIELD(priority);
863 CHECK_SKB_FIELD(skb_iif);
864 CHECK_SKB_FIELD(vlan_proto);
865 CHECK_SKB_FIELD(vlan_tci);
866 CHECK_SKB_FIELD(transport_header);
867 CHECK_SKB_FIELD(network_header);
868 CHECK_SKB_FIELD(mac_header);
869 CHECK_SKB_FIELD(inner_protocol);
870 CHECK_SKB_FIELD(inner_transport_header);
871 CHECK_SKB_FIELD(inner_network_header);
872 CHECK_SKB_FIELD(inner_mac_header);
873 CHECK_SKB_FIELD(mark);
874#ifdef CONFIG_NETWORK_SECMARK
875 CHECK_SKB_FIELD(secmark);
876#endif
Cong Wange0d10952013-08-01 11:10:25 +0800877#ifdef CONFIG_NET_RX_BUSY_POLL
Eric Dumazetb1937222014-09-28 22:18:47 -0700878 CHECK_SKB_FIELD(napi_id);
Eliezer Tamir06021292013-06-10 11:39:50 +0300879#endif
Eric Dumazet2bd82482015-02-03 23:48:24 -0800880#ifdef CONFIG_XPS
881 CHECK_SKB_FIELD(sender_cpu);
882#endif
Eric Dumazetb1937222014-09-28 22:18:47 -0700883#ifdef CONFIG_NET_SCHED
884 CHECK_SKB_FIELD(tc_index);
885#ifdef CONFIG_NET_CLS_ACT
886 CHECK_SKB_FIELD(tc_verd);
887#endif
888#endif
889
Herbert Xudec18812007-10-14 00:37:30 -0700890}
891
Herbert Xu82c49a32009-05-22 22:11:37 +0000892/*
893 * You should not add any new code to this function. Add it to
894 * __copy_skb_header above instead.
895 */
Herbert Xue0053ec2007-10-14 00:37:52 -0700896static struct sk_buff *__skb_clone(struct sk_buff *n, struct sk_buff *skb)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700897{
Linus Torvalds1da177e2005-04-16 15:20:36 -0700898#define C(x) n->x = skb->x
899
900 n->next = n->prev = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700901 n->sk = NULL;
Herbert Xudec18812007-10-14 00:37:30 -0700902 __copy_skb_header(n, skb);
903
Linus Torvalds1da177e2005-04-16 15:20:36 -0700904 C(len);
905 C(data_len);
Alexey Dobriyan3e6b3b22007-03-16 15:00:46 -0700906 C(mac_len);
Patrick McHardy334a8132007-06-25 04:35:20 -0700907 n->hdr_len = skb->nohdr ? skb_headroom(skb) : skb->hdr_len;
Paul Moore02f1c892008-01-07 21:56:41 -0800908 n->cloned = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700909 n->nohdr = 0;
Eric Dumazetec986182018-04-07 13:42:39 -0700910 n->peeked = 0;
Stefano Brivioad375ea2018-07-13 13:21:07 +0200911 C(pfmemalloc);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700912 n->destructor = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700913 C(tail);
914 C(end);
Paul Moore02f1c892008-01-07 21:56:41 -0800915 C(head);
Eric Dumazetd3836f22012-04-27 00:33:38 +0000916 C(head_frag);
Paul Moore02f1c892008-01-07 21:56:41 -0800917 C(data);
918 C(truesize);
919 atomic_set(&n->users, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700920
921 atomic_inc(&(skb_shinfo(skb)->dataref));
922 skb->cloned = 1;
923
924 return n;
Herbert Xue0053ec2007-10-14 00:37:52 -0700925#undef C
926}
927
928/**
929 * skb_morph - morph one skb into another
930 * @dst: the skb to receive the contents
931 * @src: the skb to supply the contents
932 *
933 * This is identical to skb_clone except that the target skb is
934 * supplied by the user.
935 *
936 * The target skb is returned upon exit.
937 */
938struct sk_buff *skb_morph(struct sk_buff *dst, struct sk_buff *src)
939{
Herbert Xu2d4baff2007-11-26 23:11:19 +0800940 skb_release_all(dst);
Herbert Xue0053ec2007-10-14 00:37:52 -0700941 return __skb_clone(dst, src);
942}
943EXPORT_SYMBOL_GPL(skb_morph);
944
Ben Hutchings2c530402012-07-10 10:55:09 +0000945/**
946 * skb_copy_ubufs - copy userspace skb frags buffers to kernel
Michael S. Tsirkin48c83012011-08-31 08:03:29 +0000947 * @skb: the skb to modify
948 * @gfp_mask: allocation priority
949 *
950 * This must be called on SKBTX_DEV_ZEROCOPY skb.
951 * It will copy all frags into kernel and drop the reference
952 * to userspace pages.
953 *
954 * If this function is called from an interrupt gfp_mask() must be
955 * %GFP_ATOMIC.
956 *
957 * Returns 0 on success or a negative error code on failure
958 * to allocate kernel memory to copy to.
959 */
960int skb_copy_ubufs(struct sk_buff *skb, gfp_t gfp_mask)
Shirley Maa6686f22011-07-06 12:22:12 +0000961{
962 int i;
963 int num_frags = skb_shinfo(skb)->nr_frags;
964 struct page *page, *head = NULL;
965 struct ubuf_info *uarg = skb_shinfo(skb)->destructor_arg;
966
967 for (i = 0; i < num_frags; i++) {
968 u8 *vaddr;
969 skb_frag_t *f = &skb_shinfo(skb)->frags[i];
970
Krishna Kumar02756ed2012-07-17 02:05:29 +0000971 page = alloc_page(gfp_mask);
Shirley Maa6686f22011-07-06 12:22:12 +0000972 if (!page) {
973 while (head) {
Sunghan Suh40dadff2013-07-12 16:17:23 +0900974 struct page *next = (struct page *)page_private(head);
Shirley Maa6686f22011-07-06 12:22:12 +0000975 put_page(head);
976 head = next;
977 }
978 return -ENOMEM;
979 }
Eric Dumazet51c56b02012-04-05 11:35:15 +0200980 vaddr = kmap_atomic(skb_frag_page(f));
Shirley Maa6686f22011-07-06 12:22:12 +0000981 memcpy(page_address(page),
Eric Dumazet9e903e02011-10-18 21:00:24 +0000982 vaddr + f->page_offset, skb_frag_size(f));
Eric Dumazet51c56b02012-04-05 11:35:15 +0200983 kunmap_atomic(vaddr);
Sunghan Suh40dadff2013-07-12 16:17:23 +0900984 set_page_private(page, (unsigned long)head);
Shirley Maa6686f22011-07-06 12:22:12 +0000985 head = page;
986 }
987
988 /* skb frags release userspace buffers */
Krishna Kumar02756ed2012-07-17 02:05:29 +0000989 for (i = 0; i < num_frags; i++)
Ian Campbella8605c62011-10-19 23:01:49 +0000990 skb_frag_unref(skb, i);
Shirley Maa6686f22011-07-06 12:22:12 +0000991
Michael S. Tsirkine19d6762012-11-01 09:16:22 +0000992 uarg->callback(uarg, false);
Shirley Maa6686f22011-07-06 12:22:12 +0000993
994 /* skb frags point to kernel buffers */
Krishna Kumar02756ed2012-07-17 02:05:29 +0000995 for (i = num_frags - 1; i >= 0; i--) {
996 __skb_fill_page_desc(skb, i, head, 0,
997 skb_shinfo(skb)->frags[i].size);
Sunghan Suh40dadff2013-07-12 16:17:23 +0900998 head = (struct page *)page_private(head);
Shirley Maa6686f22011-07-06 12:22:12 +0000999 }
Michael S. Tsirkin48c83012011-08-31 08:03:29 +00001000
1001 skb_shinfo(skb)->tx_flags &= ~SKBTX_DEV_ZEROCOPY;
Shirley Maa6686f22011-07-06 12:22:12 +00001002 return 0;
1003}
Michael S. Tsirkindcc0fb72012-07-20 09:23:20 +00001004EXPORT_SYMBOL_GPL(skb_copy_ubufs);
Shirley Maa6686f22011-07-06 12:22:12 +00001005
Herbert Xue0053ec2007-10-14 00:37:52 -07001006/**
1007 * skb_clone - duplicate an sk_buff
1008 * @skb: buffer to clone
1009 * @gfp_mask: allocation priority
1010 *
1011 * Duplicate an &sk_buff. The new one is not owned by a socket. Both
1012 * copies share the same packet data but not structure. The new
1013 * buffer has a reference count of 1. If the allocation fails the
1014 * function returns %NULL otherwise the new buffer is returned.
1015 *
1016 * If this function is called from an interrupt gfp_mask() must be
1017 * %GFP_ATOMIC.
1018 */
1019
1020struct sk_buff *skb_clone(struct sk_buff *skb, gfp_t gfp_mask)
1021{
Eric Dumazetd0bf4a92014-09-29 13:29:15 -07001022 struct sk_buff_fclones *fclones = container_of(skb,
1023 struct sk_buff_fclones,
1024 skb1);
Eric Dumazet6ffe75e2014-12-03 17:04:39 -08001025 struct sk_buff *n;
Herbert Xue0053ec2007-10-14 00:37:52 -07001026
Michael S. Tsirkin70008aa2012-07-20 09:23:10 +00001027 if (skb_orphan_frags(skb, gfp_mask))
1028 return NULL;
Shirley Maa6686f22011-07-06 12:22:12 +00001029
Herbert Xue0053ec2007-10-14 00:37:52 -07001030 if (skb->fclone == SKB_FCLONE_ORIG &&
Eric Dumazet6ffe75e2014-12-03 17:04:39 -08001031 atomic_read(&fclones->fclone_ref) == 1) {
1032 n = &fclones->skb2;
1033 atomic_set(&fclones->fclone_ref, 2);
Herbert Xue0053ec2007-10-14 00:37:52 -07001034 } else {
Mel Gormanc93bdd02012-07-31 16:44:19 -07001035 if (skb_pfmemalloc(skb))
1036 gfp_mask |= __GFP_MEMALLOC;
1037
Herbert Xue0053ec2007-10-14 00:37:52 -07001038 n = kmem_cache_alloc(skbuff_head_cache, gfp_mask);
1039 if (!n)
1040 return NULL;
Vegard Nossumfe55f6d2008-08-30 12:16:35 +02001041
1042 kmemcheck_annotate_bitfield(n, flags1);
Herbert Xue0053ec2007-10-14 00:37:52 -07001043 n->fclone = SKB_FCLONE_UNAVAILABLE;
1044 }
1045
1046 return __skb_clone(n, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001047}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001048EXPORT_SYMBOL(skb_clone);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001049
Pravin B Shelarf5b17292013-03-07 13:21:40 +00001050static void skb_headers_offset_update(struct sk_buff *skb, int off)
1051{
Eric Dumazet030737b2013-10-19 11:42:54 -07001052 /* Only adjust this if it actually is csum_start rather than csum */
1053 if (skb->ip_summed == CHECKSUM_PARTIAL)
1054 skb->csum_start += off;
Pravin B Shelarf5b17292013-03-07 13:21:40 +00001055 /* {transport,network,mac}_header and tail are relative to skb->head */
1056 skb->transport_header += off;
1057 skb->network_header += off;
1058 if (skb_mac_header_was_set(skb))
1059 skb->mac_header += off;
1060 skb->inner_transport_header += off;
1061 skb->inner_network_header += off;
Pravin B Shelaraefbd2b2013-03-07 13:21:46 +00001062 skb->inner_mac_header += off;
Pravin B Shelarf5b17292013-03-07 13:21:40 +00001063}
1064
Linus Torvalds1da177e2005-04-16 15:20:36 -07001065static void copy_skb_header(struct sk_buff *new, const struct sk_buff *old)
1066{
Herbert Xudec18812007-10-14 00:37:30 -07001067 __copy_skb_header(new, old);
1068
Herbert Xu79671682006-06-22 02:40:14 -07001069 skb_shinfo(new)->gso_size = skb_shinfo(old)->gso_size;
1070 skb_shinfo(new)->gso_segs = skb_shinfo(old)->gso_segs;
1071 skb_shinfo(new)->gso_type = skb_shinfo(old)->gso_type;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001072}
1073
Mel Gormanc93bdd02012-07-31 16:44:19 -07001074static inline int skb_alloc_rx_flag(const struct sk_buff *skb)
1075{
1076 if (skb_pfmemalloc(skb))
1077 return SKB_ALLOC_RX;
1078 return 0;
1079}
1080
Linus Torvalds1da177e2005-04-16 15:20:36 -07001081/**
1082 * skb_copy - create private copy of an sk_buff
1083 * @skb: buffer to copy
1084 * @gfp_mask: allocation priority
1085 *
1086 * Make a copy of both an &sk_buff and its data. This is used when the
1087 * caller wishes to modify the data and needs a private copy of the
1088 * data to alter. Returns %NULL on failure or the pointer to the buffer
1089 * on success. The returned buffer has a reference count of 1.
1090 *
1091 * As by-product this function converts non-linear &sk_buff to linear
1092 * one, so that &sk_buff becomes completely private and caller is allowed
1093 * to modify all the data of returned buffer. This means that this
1094 * function is not recommended for use in circumstances when only
1095 * header is going to be modified. Use pskb_copy() instead.
1096 */
1097
Al Virodd0fc662005-10-07 07:46:04 +01001098struct sk_buff *skb_copy(const struct sk_buff *skb, gfp_t gfp_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001099{
Eric Dumazet6602ceb2010-09-01 05:25:10 +00001100 int headerlen = skb_headroom(skb);
Alexander Duyckec47ea82012-05-04 14:26:56 +00001101 unsigned int size = skb_end_offset(skb) + skb->data_len;
Mel Gormanc93bdd02012-07-31 16:44:19 -07001102 struct sk_buff *n = __alloc_skb(size, gfp_mask,
1103 skb_alloc_rx_flag(skb), NUMA_NO_NODE);
Eric Dumazet6602ceb2010-09-01 05:25:10 +00001104
Linus Torvalds1da177e2005-04-16 15:20:36 -07001105 if (!n)
1106 return NULL;
1107
1108 /* Set the data pointer */
1109 skb_reserve(n, headerlen);
1110 /* Set the tail pointer and length */
1111 skb_put(n, skb->len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001112
1113 if (skb_copy_bits(skb, -headerlen, n->head, headerlen + skb->len))
1114 BUG();
1115
1116 copy_skb_header(n, skb);
1117 return n;
1118}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001119EXPORT_SYMBOL(skb_copy);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001120
1121/**
Octavian Purdilabad93e92014-06-12 01:36:26 +03001122 * __pskb_copy_fclone - create copy of an sk_buff with private head.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001123 * @skb: buffer to copy
Eric Dumazet117632e2011-12-03 21:39:53 +00001124 * @headroom: headroom of new skb
Linus Torvalds1da177e2005-04-16 15:20:36 -07001125 * @gfp_mask: allocation priority
Octavian Purdilabad93e92014-06-12 01:36:26 +03001126 * @fclone: if true allocate the copy of the skb from the fclone
1127 * cache instead of the head cache; it is recommended to set this
1128 * to true for the cases where the copy will likely be cloned
Linus Torvalds1da177e2005-04-16 15:20:36 -07001129 *
1130 * Make a copy of both an &sk_buff and part of its data, located
1131 * in header. Fragmented data remain shared. This is used when
1132 * the caller wishes to modify only header of &sk_buff and needs
1133 * private copy of the header to alter. Returns %NULL on failure
1134 * or the pointer to the buffer on success.
1135 * The returned buffer has a reference count of 1.
1136 */
1137
Octavian Purdilabad93e92014-06-12 01:36:26 +03001138struct sk_buff *__pskb_copy_fclone(struct sk_buff *skb, int headroom,
1139 gfp_t gfp_mask, bool fclone)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001140{
Eric Dumazet117632e2011-12-03 21:39:53 +00001141 unsigned int size = skb_headlen(skb) + headroom;
Octavian Purdilabad93e92014-06-12 01:36:26 +03001142 int flags = skb_alloc_rx_flag(skb) | (fclone ? SKB_ALLOC_FCLONE : 0);
1143 struct sk_buff *n = __alloc_skb(size, gfp_mask, flags, NUMA_NO_NODE);
Eric Dumazet6602ceb2010-09-01 05:25:10 +00001144
Linus Torvalds1da177e2005-04-16 15:20:36 -07001145 if (!n)
1146 goto out;
1147
1148 /* Set the data pointer */
Eric Dumazet117632e2011-12-03 21:39:53 +00001149 skb_reserve(n, headroom);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001150 /* Set the tail pointer and length */
1151 skb_put(n, skb_headlen(skb));
1152 /* Copy the bytes */
Arnaldo Carvalho de Melod626f622007-03-27 18:55:52 -03001153 skb_copy_from_linear_data(skb, n->data, n->len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001154
Herbert Xu25f484a2006-11-07 14:57:15 -08001155 n->truesize += skb->data_len;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001156 n->data_len = skb->data_len;
1157 n->len = skb->len;
1158
1159 if (skb_shinfo(skb)->nr_frags) {
1160 int i;
1161
Michael S. Tsirkin70008aa2012-07-20 09:23:10 +00001162 if (skb_orphan_frags(skb, gfp_mask)) {
1163 kfree_skb(n);
1164 n = NULL;
1165 goto out;
Shirley Maa6686f22011-07-06 12:22:12 +00001166 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001167 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
1168 skb_shinfo(n)->frags[i] = skb_shinfo(skb)->frags[i];
Ian Campbellea2ab692011-08-22 23:44:58 +00001169 skb_frag_ref(skb, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001170 }
1171 skb_shinfo(n)->nr_frags = i;
1172 }
1173
David S. Miller21dc3302010-08-23 00:13:46 -07001174 if (skb_has_frag_list(skb)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001175 skb_shinfo(n)->frag_list = skb_shinfo(skb)->frag_list;
1176 skb_clone_fraglist(n);
1177 }
1178
1179 copy_skb_header(n, skb);
1180out:
1181 return n;
1182}
Octavian Purdilabad93e92014-06-12 01:36:26 +03001183EXPORT_SYMBOL(__pskb_copy_fclone);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001184
1185/**
1186 * pskb_expand_head - reallocate header of &sk_buff
1187 * @skb: buffer to reallocate
1188 * @nhead: room to add at head
1189 * @ntail: room to add at tail
1190 * @gfp_mask: allocation priority
1191 *
Mathias Krausebc323832013-11-07 14:18:26 +01001192 * Expands (or creates identical copy, if @nhead and @ntail are zero)
1193 * header of @skb. &sk_buff itself is not changed. &sk_buff MUST have
Linus Torvalds1da177e2005-04-16 15:20:36 -07001194 * reference count of 1. Returns zero in the case of success or error,
1195 * if expansion failed. In the last case, &sk_buff is not changed.
1196 *
1197 * All the pointers pointing into skb header may change and must be
1198 * reloaded after call to this function.
1199 */
1200
Victor Fusco86a76ca2005-07-08 14:57:47 -07001201int pskb_expand_head(struct sk_buff *skb, int nhead, int ntail,
Al Virodd0fc662005-10-07 07:46:04 +01001202 gfp_t gfp_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001203{
1204 int i;
1205 u8 *data;
Alexander Duyckec47ea82012-05-04 14:26:56 +00001206 int size = nhead + skb_end_offset(skb) + ntail;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001207 long off;
1208
Herbert Xu4edd87a2008-10-01 07:09:38 -07001209 BUG_ON(nhead < 0);
1210
Linus Torvalds1da177e2005-04-16 15:20:36 -07001211 if (skb_shared(skb))
1212 BUG();
1213
1214 size = SKB_DATA_ALIGN(size);
1215
Mel Gormanc93bdd02012-07-31 16:44:19 -07001216 if (skb_pfmemalloc(skb))
1217 gfp_mask |= __GFP_MEMALLOC;
1218 data = kmalloc_reserve(size + SKB_DATA_ALIGN(sizeof(struct skb_shared_info)),
1219 gfp_mask, NUMA_NO_NODE, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001220 if (!data)
1221 goto nodata;
Eric Dumazet87151b82012-04-10 20:08:39 +00001222 size = SKB_WITH_OVERHEAD(ksize(data));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001223
1224 /* Copy only real data... and, alas, header. This should be
Eric Dumazet6602ceb2010-09-01 05:25:10 +00001225 * optimized for the cases when header is void.
1226 */
1227 memcpy(data + nhead, skb->head, skb_tail_pointer(skb) - skb->head);
1228
1229 memcpy((struct skb_shared_info *)(data + size),
1230 skb_shinfo(skb),
Eric Dumazetfed66382010-07-22 19:09:08 +00001231 offsetof(struct skb_shared_info, frags[skb_shinfo(skb)->nr_frags]));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001232
Alexander Duyck3e245912012-05-04 14:26:51 +00001233 /*
1234 * if shinfo is shared we must drop the old head gracefully, but if it
1235 * is not we can just drop the old head and let the existing refcount
1236 * be since all we did is relocate the values
1237 */
1238 if (skb_cloned(skb)) {
Shirley Maa6686f22011-07-06 12:22:12 +00001239 /* copy this zero copy skb frags */
Michael S. Tsirkin70008aa2012-07-20 09:23:10 +00001240 if (skb_orphan_frags(skb, gfp_mask))
1241 goto nofrags;
Eric Dumazet1fd63042010-09-02 23:09:32 +00001242 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++)
Ian Campbellea2ab692011-08-22 23:44:58 +00001243 skb_frag_ref(skb, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001244
Eric Dumazet1fd63042010-09-02 23:09:32 +00001245 if (skb_has_frag_list(skb))
1246 skb_clone_fraglist(skb);
1247
1248 skb_release_data(skb);
Alexander Duyck3e245912012-05-04 14:26:51 +00001249 } else {
1250 skb_free_head(skb);
Eric Dumazet1fd63042010-09-02 23:09:32 +00001251 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001252 off = (data + nhead) - skb->head;
1253
1254 skb->head = data;
Eric Dumazetd3836f22012-04-27 00:33:38 +00001255 skb->head_frag = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001256 skb->data += off;
Arnaldo Carvalho de Melo4305b542007-04-19 20:43:29 -07001257#ifdef NET_SKBUFF_DATA_USES_OFFSET
1258 skb->end = size;
Patrick McHardy56eb8882007-04-09 11:45:04 -07001259 off = nhead;
Arnaldo Carvalho de Melo4305b542007-04-19 20:43:29 -07001260#else
1261 skb->end = skb->head + size;
Patrick McHardy56eb8882007-04-09 11:45:04 -07001262#endif
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001263 skb->tail += off;
Peter Pan(潘卫平)b41abb42013-06-06 21:27:21 +08001264 skb_headers_offset_update(skb, nhead);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001265 skb->cloned = 0;
Patrick McHardy334a8132007-06-25 04:35:20 -07001266 skb->hdr_len = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001267 skb->nohdr = 0;
1268 atomic_set(&skb_shinfo(skb)->dataref, 1);
1269 return 0;
1270
Shirley Maa6686f22011-07-06 12:22:12 +00001271nofrags:
1272 kfree(data);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001273nodata:
1274 return -ENOMEM;
1275}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001276EXPORT_SYMBOL(pskb_expand_head);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001277
1278/* Make private copy of skb with writable head and some headroom */
1279
1280struct sk_buff *skb_realloc_headroom(struct sk_buff *skb, unsigned int headroom)
1281{
1282 struct sk_buff *skb2;
1283 int delta = headroom - skb_headroom(skb);
1284
1285 if (delta <= 0)
1286 skb2 = pskb_copy(skb, GFP_ATOMIC);
1287 else {
1288 skb2 = skb_clone(skb, GFP_ATOMIC);
1289 if (skb2 && pskb_expand_head(skb2, SKB_DATA_ALIGN(delta), 0,
1290 GFP_ATOMIC)) {
1291 kfree_skb(skb2);
1292 skb2 = NULL;
1293 }
1294 }
1295 return skb2;
1296}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001297EXPORT_SYMBOL(skb_realloc_headroom);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001298
1299/**
1300 * skb_copy_expand - copy and expand sk_buff
1301 * @skb: buffer to copy
1302 * @newheadroom: new free bytes at head
1303 * @newtailroom: new free bytes at tail
1304 * @gfp_mask: allocation priority
1305 *
1306 * Make a copy of both an &sk_buff and its data and while doing so
1307 * allocate additional space.
1308 *
1309 * This is used when the caller wishes to modify the data and needs a
1310 * private copy of the data to alter as well as more space for new fields.
1311 * Returns %NULL on failure or the pointer to the buffer
1312 * on success. The returned buffer has a reference count of 1.
1313 *
1314 * You must pass %GFP_ATOMIC as the allocation priority if this function
1315 * is called from an interrupt.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001316 */
1317struct sk_buff *skb_copy_expand(const struct sk_buff *skb,
Victor Fusco86a76ca2005-07-08 14:57:47 -07001318 int newheadroom, int newtailroom,
Al Virodd0fc662005-10-07 07:46:04 +01001319 gfp_t gfp_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001320{
1321 /*
1322 * Allocate the copy buffer
1323 */
Mel Gormanc93bdd02012-07-31 16:44:19 -07001324 struct sk_buff *n = __alloc_skb(newheadroom + skb->len + newtailroom,
1325 gfp_mask, skb_alloc_rx_flag(skb),
1326 NUMA_NO_NODE);
Patrick McHardyefd1e8d2007-04-10 18:30:09 -07001327 int oldheadroom = skb_headroom(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001328 int head_copy_len, head_copy_off;
1329
1330 if (!n)
1331 return NULL;
1332
1333 skb_reserve(n, newheadroom);
1334
1335 /* Set the tail pointer and length */
1336 skb_put(n, skb->len);
1337
Patrick McHardyefd1e8d2007-04-10 18:30:09 -07001338 head_copy_len = oldheadroom;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001339 head_copy_off = 0;
1340 if (newheadroom <= head_copy_len)
1341 head_copy_len = newheadroom;
1342 else
1343 head_copy_off = newheadroom - head_copy_len;
1344
1345 /* Copy the linear header and data. */
1346 if (skb_copy_bits(skb, -head_copy_len, n->head + head_copy_off,
1347 skb->len + head_copy_len))
1348 BUG();
1349
1350 copy_skb_header(n, skb);
1351
Eric Dumazet030737b2013-10-19 11:42:54 -07001352 skb_headers_offset_update(n, newheadroom - oldheadroom);
Patrick McHardyefd1e8d2007-04-10 18:30:09 -07001353
Linus Torvalds1da177e2005-04-16 15:20:36 -07001354 return n;
1355}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001356EXPORT_SYMBOL(skb_copy_expand);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001357
1358/**
1359 * skb_pad - zero pad the tail of an skb
1360 * @skb: buffer to pad
1361 * @pad: space to pad
1362 *
1363 * Ensure that a buffer is followed by a padding area that is zero
1364 * filled. Used by network drivers which may DMA or transfer data
1365 * beyond the buffer end onto the wire.
1366 *
Herbert Xu5b057c62006-06-23 02:06:41 -07001367 * May return error in out of memory cases. The skb is freed on error.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001368 */
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001369
Herbert Xu5b057c62006-06-23 02:06:41 -07001370int skb_pad(struct sk_buff *skb, int pad)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001371{
Herbert Xu5b057c62006-06-23 02:06:41 -07001372 int err;
1373 int ntail;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001374
Linus Torvalds1da177e2005-04-16 15:20:36 -07001375 /* If the skbuff is non linear tailroom is always zero.. */
Herbert Xu5b057c62006-06-23 02:06:41 -07001376 if (!skb_cloned(skb) && skb_tailroom(skb) >= pad) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001377 memset(skb->data+skb->len, 0, pad);
Herbert Xu5b057c62006-06-23 02:06:41 -07001378 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001379 }
Herbert Xu5b057c62006-06-23 02:06:41 -07001380
Arnaldo Carvalho de Melo4305b542007-04-19 20:43:29 -07001381 ntail = skb->data_len + pad - (skb->end - skb->tail);
Herbert Xu5b057c62006-06-23 02:06:41 -07001382 if (likely(skb_cloned(skb) || ntail > 0)) {
1383 err = pskb_expand_head(skb, 0, ntail, GFP_ATOMIC);
1384 if (unlikely(err))
1385 goto free_skb;
1386 }
1387
1388 /* FIXME: The use of this function with non-linear skb's really needs
1389 * to be audited.
1390 */
1391 err = skb_linearize(skb);
1392 if (unlikely(err))
1393 goto free_skb;
1394
1395 memset(skb->data + skb->len, 0, pad);
1396 return 0;
1397
1398free_skb:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001399 kfree_skb(skb);
Herbert Xu5b057c62006-06-23 02:06:41 -07001400 return err;
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001401}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001402EXPORT_SYMBOL(skb_pad);
YOSHIFUJI Hideaki4ec93ed2007-02-09 23:24:36 +09001403
Ilpo Järvinen0dde3e12008-03-27 17:43:41 -07001404/**
Mathias Krause0c7ddf32013-11-07 14:18:24 +01001405 * pskb_put - add data to the tail of a potentially fragmented buffer
1406 * @skb: start of the buffer to use
1407 * @tail: tail fragment of the buffer to use
1408 * @len: amount of data to add
1409 *
1410 * This function extends the used data area of the potentially
1411 * fragmented buffer. @tail must be the last fragment of @skb -- or
1412 * @skb itself. If this would exceed the total buffer size the kernel
1413 * will panic. A pointer to the first byte of the extra data is
1414 * returned.
1415 */
1416
1417unsigned char *pskb_put(struct sk_buff *skb, struct sk_buff *tail, int len)
1418{
1419 if (tail != skb) {
1420 skb->data_len += len;
1421 skb->len += len;
1422 }
1423 return skb_put(tail, len);
1424}
1425EXPORT_SYMBOL_GPL(pskb_put);
1426
1427/**
Ilpo Järvinen0dde3e12008-03-27 17:43:41 -07001428 * skb_put - add data to a buffer
1429 * @skb: buffer to use
1430 * @len: amount of data to add
1431 *
1432 * This function extends the used data area of the buffer. If this would
1433 * exceed the total buffer size the kernel will panic. A pointer to the
1434 * first byte of the extra data is returned.
1435 */
1436unsigned char *skb_put(struct sk_buff *skb, unsigned int len)
1437{
1438 unsigned char *tmp = skb_tail_pointer(skb);
1439 SKB_LINEAR_ASSERT(skb);
1440 skb->tail += len;
1441 skb->len += len;
1442 if (unlikely(skb->tail > skb->end))
1443 skb_over_panic(skb, len, __builtin_return_address(0));
1444 return tmp;
1445}
1446EXPORT_SYMBOL(skb_put);
1447
Ilpo Järvinen6be8ac22008-03-27 17:47:24 -07001448/**
Ilpo Järvinenc2aa2702008-03-27 17:52:40 -07001449 * skb_push - add data to the start of a buffer
1450 * @skb: buffer to use
1451 * @len: amount of data to add
1452 *
1453 * This function extends the used data area of the buffer at the buffer
1454 * start. If this would exceed the total buffer headroom the kernel will
1455 * panic. A pointer to the first byte of the extra data is returned.
1456 */
1457unsigned char *skb_push(struct sk_buff *skb, unsigned int len)
1458{
1459 skb->data -= len;
1460 skb->len += len;
1461 if (unlikely(skb->data<skb->head))
1462 skb_under_panic(skb, len, __builtin_return_address(0));
1463 return skb->data;
1464}
1465EXPORT_SYMBOL(skb_push);
1466
1467/**
Ilpo Järvinen6be8ac22008-03-27 17:47:24 -07001468 * skb_pull - remove data from the start of a buffer
1469 * @skb: buffer to use
1470 * @len: amount of data to remove
1471 *
1472 * This function removes data from the start of a buffer, returning
1473 * the memory to the headroom. A pointer to the next data in the buffer
1474 * is returned. Once the data has been pulled future pushes will overwrite
1475 * the old data.
1476 */
1477unsigned char *skb_pull(struct sk_buff *skb, unsigned int len)
1478{
David S. Miller47d29642010-05-02 02:21:44 -07001479 return skb_pull_inline(skb, len);
Ilpo Järvinen6be8ac22008-03-27 17:47:24 -07001480}
1481EXPORT_SYMBOL(skb_pull);
1482
Ilpo Järvinen419ae742008-03-27 17:54:01 -07001483/**
1484 * skb_trim - remove end from a buffer
1485 * @skb: buffer to alter
1486 * @len: new length
1487 *
1488 * Cut the length of a buffer down by removing data from the tail. If
1489 * the buffer is already under the length specified it is not modified.
1490 * The skb must be linear.
1491 */
1492void skb_trim(struct sk_buff *skb, unsigned int len)
1493{
1494 if (skb->len > len)
1495 __skb_trim(skb, len);
1496}
1497EXPORT_SYMBOL(skb_trim);
1498
Herbert Xu3cc0e872006-06-09 16:13:38 -07001499/* Trims skb to length len. It can change skb pointers.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001500 */
1501
Herbert Xu3cc0e872006-06-09 16:13:38 -07001502int ___pskb_trim(struct sk_buff *skb, unsigned int len)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001503{
Herbert Xu27b437c2006-07-13 19:26:39 -07001504 struct sk_buff **fragp;
1505 struct sk_buff *frag;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001506 int offset = skb_headlen(skb);
1507 int nfrags = skb_shinfo(skb)->nr_frags;
1508 int i;
Herbert Xu27b437c2006-07-13 19:26:39 -07001509 int err;
1510
1511 if (skb_cloned(skb) &&
1512 unlikely((err = pskb_expand_head(skb, 0, 0, GFP_ATOMIC))))
1513 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001514
Herbert Xuf4d26fb2006-07-30 20:20:28 -07001515 i = 0;
1516 if (offset >= len)
1517 goto drop_pages;
1518
1519 for (; i < nfrags; i++) {
Eric Dumazet9e903e02011-10-18 21:00:24 +00001520 int end = offset + skb_frag_size(&skb_shinfo(skb)->frags[i]);
Herbert Xu27b437c2006-07-13 19:26:39 -07001521
1522 if (end < len) {
1523 offset = end;
1524 continue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001525 }
Herbert Xu27b437c2006-07-13 19:26:39 -07001526
Eric Dumazet9e903e02011-10-18 21:00:24 +00001527 skb_frag_size_set(&skb_shinfo(skb)->frags[i++], len - offset);
Herbert Xu27b437c2006-07-13 19:26:39 -07001528
Herbert Xuf4d26fb2006-07-30 20:20:28 -07001529drop_pages:
Herbert Xu27b437c2006-07-13 19:26:39 -07001530 skb_shinfo(skb)->nr_frags = i;
1531
1532 for (; i < nfrags; i++)
Ian Campbellea2ab692011-08-22 23:44:58 +00001533 skb_frag_unref(skb, i);
Herbert Xu27b437c2006-07-13 19:26:39 -07001534
David S. Miller21dc3302010-08-23 00:13:46 -07001535 if (skb_has_frag_list(skb))
Herbert Xu27b437c2006-07-13 19:26:39 -07001536 skb_drop_fraglist(skb);
Herbert Xuf4d26fb2006-07-30 20:20:28 -07001537 goto done;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001538 }
1539
Herbert Xu27b437c2006-07-13 19:26:39 -07001540 for (fragp = &skb_shinfo(skb)->frag_list; (frag = *fragp);
1541 fragp = &frag->next) {
1542 int end = offset + frag->len;
1543
1544 if (skb_shared(frag)) {
1545 struct sk_buff *nfrag;
1546
1547 nfrag = skb_clone(frag, GFP_ATOMIC);
1548 if (unlikely(!nfrag))
1549 return -ENOMEM;
1550
1551 nfrag->next = frag->next;
Eric Dumazet85bb2a62012-04-19 02:24:53 +00001552 consume_skb(frag);
Herbert Xu27b437c2006-07-13 19:26:39 -07001553 frag = nfrag;
1554 *fragp = frag;
1555 }
1556
1557 if (end < len) {
1558 offset = end;
1559 continue;
1560 }
1561
1562 if (end > len &&
1563 unlikely((err = pskb_trim(frag, len - offset))))
1564 return err;
1565
1566 if (frag->next)
1567 skb_drop_list(&frag->next);
1568 break;
1569 }
1570
Herbert Xuf4d26fb2006-07-30 20:20:28 -07001571done:
Herbert Xu27b437c2006-07-13 19:26:39 -07001572 if (len > skb_headlen(skb)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001573 skb->data_len -= skb->len - len;
1574 skb->len = len;
1575 } else {
Herbert Xu27b437c2006-07-13 19:26:39 -07001576 skb->len = len;
1577 skb->data_len = 0;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001578 skb_set_tail_pointer(skb, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001579 }
1580
1581 return 0;
1582}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001583EXPORT_SYMBOL(___pskb_trim);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001584
Eric Dumazet791521e2018-10-10 12:30:11 -07001585/* Note : use pskb_trim_rcsum() instead of calling this directly
1586 */
1587int pskb_trim_rcsum_slow(struct sk_buff *skb, unsigned int len)
1588{
1589 if (skb->ip_summed == CHECKSUM_COMPLETE) {
1590 int delta = skb->len - len;
1591
Dimitris Michailidis0c49b5e2018-10-19 17:07:13 -07001592 skb->csum = csum_block_sub(skb->csum,
1593 skb_checksum(skb, len, delta, 0),
1594 len);
Eric Dumazet791521e2018-10-10 12:30:11 -07001595 }
1596 return __pskb_trim(skb, len);
1597}
1598EXPORT_SYMBOL(pskb_trim_rcsum_slow);
1599
Linus Torvalds1da177e2005-04-16 15:20:36 -07001600/**
1601 * __pskb_pull_tail - advance tail of skb header
1602 * @skb: buffer to reallocate
1603 * @delta: number of bytes to advance tail
1604 *
1605 * The function makes a sense only on a fragmented &sk_buff,
1606 * it expands header moving its tail forward and copying necessary
1607 * data from fragmented part.
1608 *
1609 * &sk_buff MUST have reference count of 1.
1610 *
1611 * Returns %NULL (and &sk_buff does not change) if pull failed
1612 * or value of new tail of skb in the case of success.
1613 *
1614 * All the pointers pointing into skb header may change and must be
1615 * reloaded after call to this function.
1616 */
1617
1618/* Moves tail of skb head forward, copying data from fragmented part,
1619 * when it is necessary.
1620 * 1. It may fail due to malloc failure.
1621 * 2. It may change skb pointers.
1622 *
1623 * It is pretty complicated. Luckily, it is called only in exceptional cases.
1624 */
1625unsigned char *__pskb_pull_tail(struct sk_buff *skb, int delta)
1626{
1627 /* If skb has not enough free space at tail, get new one
1628 * plus 128 bytes for future expansions. If we have enough
1629 * room at tail, reallocate without expansion only if skb is cloned.
1630 */
Arnaldo Carvalho de Melo4305b542007-04-19 20:43:29 -07001631 int i, k, eat = (skb->tail + delta) - skb->end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001632
1633 if (eat > 0 || skb_cloned(skb)) {
1634 if (pskb_expand_head(skb, 0, eat > 0 ? eat + 128 : 0,
1635 GFP_ATOMIC))
1636 return NULL;
1637 }
1638
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001639 if (skb_copy_bits(skb, skb_headlen(skb), skb_tail_pointer(skb), delta))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001640 BUG();
1641
1642 /* Optimization: no fragments, no reasons to preestimate
1643 * size of pulled pages. Superb.
1644 */
David S. Miller21dc3302010-08-23 00:13:46 -07001645 if (!skb_has_frag_list(skb))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001646 goto pull_pages;
1647
1648 /* Estimate size of pulled pages. */
1649 eat = delta;
1650 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
Eric Dumazet9e903e02011-10-18 21:00:24 +00001651 int size = skb_frag_size(&skb_shinfo(skb)->frags[i]);
1652
1653 if (size >= eat)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001654 goto pull_pages;
Eric Dumazet9e903e02011-10-18 21:00:24 +00001655 eat -= size;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001656 }
1657
1658 /* If we need update frag list, we are in troubles.
1659 * Certainly, it possible to add an offset to skb data,
1660 * but taking into account that pulling is expected to
1661 * be very rare operation, it is worth to fight against
1662 * further bloating skb head and crucify ourselves here instead.
1663 * Pure masohism, indeed. 8)8)
1664 */
1665 if (eat) {
1666 struct sk_buff *list = skb_shinfo(skb)->frag_list;
1667 struct sk_buff *clone = NULL;
1668 struct sk_buff *insp = NULL;
1669
1670 do {
Kris Katterjohn09a62662006-01-08 22:24:28 -08001671 BUG_ON(!list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001672
1673 if (list->len <= eat) {
1674 /* Eaten as whole. */
1675 eat -= list->len;
1676 list = list->next;
1677 insp = list;
1678 } else {
1679 /* Eaten partially. */
1680
1681 if (skb_shared(list)) {
1682 /* Sucks! We need to fork list. :-( */
1683 clone = skb_clone(list, GFP_ATOMIC);
1684 if (!clone)
1685 return NULL;
1686 insp = list->next;
1687 list = clone;
1688 } else {
1689 /* This may be pulled without
1690 * problems. */
1691 insp = list;
1692 }
1693 if (!pskb_pull(list, eat)) {
Wei Yongjunf3fbbe02009-02-25 00:37:32 +00001694 kfree_skb(clone);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001695 return NULL;
1696 }
1697 break;
1698 }
1699 } while (eat);
1700
1701 /* Free pulled out fragments. */
1702 while ((list = skb_shinfo(skb)->frag_list) != insp) {
1703 skb_shinfo(skb)->frag_list = list->next;
1704 kfree_skb(list);
1705 }
1706 /* And insert new clone at head. */
1707 if (clone) {
1708 clone->next = list;
1709 skb_shinfo(skb)->frag_list = clone;
1710 }
1711 }
1712 /* Success! Now we may commit changes to skb data. */
1713
1714pull_pages:
1715 eat = delta;
1716 k = 0;
1717 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
Eric Dumazet9e903e02011-10-18 21:00:24 +00001718 int size = skb_frag_size(&skb_shinfo(skb)->frags[i]);
1719
1720 if (size <= eat) {
Ian Campbellea2ab692011-08-22 23:44:58 +00001721 skb_frag_unref(skb, i);
Eric Dumazet9e903e02011-10-18 21:00:24 +00001722 eat -= size;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001723 } else {
1724 skb_shinfo(skb)->frags[k] = skb_shinfo(skb)->frags[i];
1725 if (eat) {
1726 skb_shinfo(skb)->frags[k].page_offset += eat;
Eric Dumazet9e903e02011-10-18 21:00:24 +00001727 skb_frag_size_sub(&skb_shinfo(skb)->frags[k], eat);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001728 eat = 0;
1729 }
1730 k++;
1731 }
1732 }
1733 skb_shinfo(skb)->nr_frags = k;
1734
1735 skb->tail += delta;
1736 skb->data_len -= delta;
1737
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001738 return skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001739}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001740EXPORT_SYMBOL(__pskb_pull_tail);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001741
Eric Dumazet22019b12011-07-29 18:37:31 +00001742/**
1743 * skb_copy_bits - copy bits from skb to kernel buffer
1744 * @skb: source skb
1745 * @offset: offset in source
1746 * @to: destination buffer
1747 * @len: number of bytes to copy
1748 *
1749 * Copy the specified number of bytes from the source skb to the
1750 * destination buffer.
1751 *
1752 * CAUTION ! :
1753 * If its prototype is ever changed,
1754 * check arch/{*}/net/{*}.S files,
1755 * since it is called from BPF assembly code.
1756 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001757int skb_copy_bits(const struct sk_buff *skb, int offset, void *to, int len)
1758{
David S. Miller1a028e52007-04-27 15:21:23 -07001759 int start = skb_headlen(skb);
David S. Millerfbb398a2009-06-09 00:18:59 -07001760 struct sk_buff *frag_iter;
1761 int i, copy;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001762
1763 if (offset > (int)skb->len - len)
1764 goto fault;
1765
1766 /* Copy header. */
David S. Miller1a028e52007-04-27 15:21:23 -07001767 if ((copy = start - offset) > 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001768 if (copy > len)
1769 copy = len;
Arnaldo Carvalho de Melod626f622007-03-27 18:55:52 -03001770 skb_copy_from_linear_data_offset(skb, offset, to, copy);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001771 if ((len -= copy) == 0)
1772 return 0;
1773 offset += copy;
1774 to += copy;
1775 }
1776
1777 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
David S. Miller1a028e52007-04-27 15:21:23 -07001778 int end;
Eric Dumazet51c56b02012-04-05 11:35:15 +02001779 skb_frag_t *f = &skb_shinfo(skb)->frags[i];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001780
Ilpo Järvinen547b7922008-07-25 21:43:18 -07001781 WARN_ON(start > offset + len);
David S. Miller1a028e52007-04-27 15:21:23 -07001782
Eric Dumazet51c56b02012-04-05 11:35:15 +02001783 end = start + skb_frag_size(f);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001784 if ((copy = end - offset) > 0) {
1785 u8 *vaddr;
1786
1787 if (copy > len)
1788 copy = len;
1789
Eric Dumazet51c56b02012-04-05 11:35:15 +02001790 vaddr = kmap_atomic(skb_frag_page(f));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001791 memcpy(to,
Eric Dumazet51c56b02012-04-05 11:35:15 +02001792 vaddr + f->page_offset + offset - start,
1793 copy);
1794 kunmap_atomic(vaddr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001795
1796 if ((len -= copy) == 0)
1797 return 0;
1798 offset += copy;
1799 to += copy;
1800 }
David S. Miller1a028e52007-04-27 15:21:23 -07001801 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001802 }
1803
David S. Millerfbb398a2009-06-09 00:18:59 -07001804 skb_walk_frags(skb, frag_iter) {
1805 int end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001806
David S. Millerfbb398a2009-06-09 00:18:59 -07001807 WARN_ON(start > offset + len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001808
David S. Millerfbb398a2009-06-09 00:18:59 -07001809 end = start + frag_iter->len;
1810 if ((copy = end - offset) > 0) {
1811 if (copy > len)
1812 copy = len;
1813 if (skb_copy_bits(frag_iter, offset - start, to, copy))
1814 goto fault;
1815 if ((len -= copy) == 0)
1816 return 0;
1817 offset += copy;
1818 to += copy;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001819 }
David S. Millerfbb398a2009-06-09 00:18:59 -07001820 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001821 }
Shirley Maa6686f22011-07-06 12:22:12 +00001822
Linus Torvalds1da177e2005-04-16 15:20:36 -07001823 if (!len)
1824 return 0;
1825
1826fault:
1827 return -EFAULT;
1828}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08001829EXPORT_SYMBOL(skb_copy_bits);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001830
Jens Axboe9c55e012007-11-06 23:30:13 -08001831/*
1832 * Callback from splice_to_pipe(), if we need to release some pages
1833 * at the end of the spd in case we error'ed out in filling the pipe.
1834 */
1835static void sock_spd_release(struct splice_pipe_desc *spd, unsigned int i)
1836{
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001837 put_page(spd->pages[i]);
1838}
Jens Axboe9c55e012007-11-06 23:30:13 -08001839
David S. Millera108d5f2012-04-23 23:06:11 -04001840static struct page *linear_to_page(struct page *page, unsigned int *len,
1841 unsigned int *offset,
Eric Dumazet18aafc62013-01-11 14:46:37 +00001842 struct sock *sk)
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001843{
Eric Dumazet5640f762012-09-23 23:04:42 +00001844 struct page_frag *pfrag = sk_page_frag(sk);
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001845
Eric Dumazet5640f762012-09-23 23:04:42 +00001846 if (!sk_page_frag_refill(sk, pfrag))
1847 return NULL;
Jarek Poplawski4fb66992009-02-01 00:41:42 -08001848
Eric Dumazet5640f762012-09-23 23:04:42 +00001849 *len = min_t(unsigned int, *len, pfrag->size - pfrag->offset);
Jarek Poplawski4fb66992009-02-01 00:41:42 -08001850
Eric Dumazet5640f762012-09-23 23:04:42 +00001851 memcpy(page_address(pfrag->page) + pfrag->offset,
1852 page_address(page) + *offset, *len);
1853 *offset = pfrag->offset;
1854 pfrag->offset += *len;
Jarek Poplawski4fb66992009-02-01 00:41:42 -08001855
Eric Dumazet5640f762012-09-23 23:04:42 +00001856 return pfrag->page;
Jens Axboe9c55e012007-11-06 23:30:13 -08001857}
1858
Eric Dumazet41c73a02012-04-22 12:26:16 +00001859static bool spd_can_coalesce(const struct splice_pipe_desc *spd,
1860 struct page *page,
1861 unsigned int offset)
1862{
1863 return spd->nr_pages &&
1864 spd->pages[spd->nr_pages - 1] == page &&
1865 (spd->partial[spd->nr_pages - 1].offset +
1866 spd->partial[spd->nr_pages - 1].len == offset);
1867}
1868
Jens Axboe9c55e012007-11-06 23:30:13 -08001869/*
1870 * Fill page/offset/length into spd, if it can hold more pages.
1871 */
David S. Millera108d5f2012-04-23 23:06:11 -04001872static bool spd_fill_page(struct splice_pipe_desc *spd,
1873 struct pipe_inode_info *pipe, struct page *page,
1874 unsigned int *len, unsigned int offset,
Eric Dumazet18aafc62013-01-11 14:46:37 +00001875 bool linear,
David S. Millera108d5f2012-04-23 23:06:11 -04001876 struct sock *sk)
Jens Axboe9c55e012007-11-06 23:30:13 -08001877{
Eric Dumazet41c73a02012-04-22 12:26:16 +00001878 if (unlikely(spd->nr_pages == MAX_SKB_FRAGS))
David S. Millera108d5f2012-04-23 23:06:11 -04001879 return true;
Jens Axboe9c55e012007-11-06 23:30:13 -08001880
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001881 if (linear) {
Eric Dumazet18aafc62013-01-11 14:46:37 +00001882 page = linear_to_page(page, len, &offset, sk);
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001883 if (!page)
David S. Millera108d5f2012-04-23 23:06:11 -04001884 return true;
Eric Dumazet41c73a02012-04-22 12:26:16 +00001885 }
1886 if (spd_can_coalesce(spd, page, offset)) {
1887 spd->partial[spd->nr_pages - 1].len += *len;
David S. Millera108d5f2012-04-23 23:06:11 -04001888 return false;
Eric Dumazet41c73a02012-04-22 12:26:16 +00001889 }
1890 get_page(page);
Jens Axboe9c55e012007-11-06 23:30:13 -08001891 spd->pages[spd->nr_pages] = page;
Jarek Poplawski4fb66992009-02-01 00:41:42 -08001892 spd->partial[spd->nr_pages].len = *len;
Jens Axboe9c55e012007-11-06 23:30:13 -08001893 spd->partial[spd->nr_pages].offset = offset;
Jens Axboe9c55e012007-11-06 23:30:13 -08001894 spd->nr_pages++;
Jarek Poplawski8b9d3722009-01-19 17:03:56 -08001895
David S. Millera108d5f2012-04-23 23:06:11 -04001896 return false;
Jens Axboe9c55e012007-11-06 23:30:13 -08001897}
1898
David S. Millera108d5f2012-04-23 23:06:11 -04001899static bool __splice_segment(struct page *page, unsigned int poff,
1900 unsigned int plen, unsigned int *off,
Eric Dumazet18aafc62013-01-11 14:46:37 +00001901 unsigned int *len,
Eric Dumazetd7ccf7c2012-04-23 23:35:04 -04001902 struct splice_pipe_desc *spd, bool linear,
David S. Millera108d5f2012-04-23 23:06:11 -04001903 struct sock *sk,
1904 struct pipe_inode_info *pipe)
Octavian Purdila2870c432008-07-15 00:49:11 -07001905{
1906 if (!*len)
David S. Millera108d5f2012-04-23 23:06:11 -04001907 return true;
Octavian Purdila2870c432008-07-15 00:49:11 -07001908
1909 /* skip this segment if already processed */
1910 if (*off >= plen) {
1911 *off -= plen;
David S. Millera108d5f2012-04-23 23:06:11 -04001912 return false;
Octavian Purdiladb43a282008-06-27 17:27:21 -07001913 }
Jens Axboe9c55e012007-11-06 23:30:13 -08001914
Octavian Purdila2870c432008-07-15 00:49:11 -07001915 /* ignore any bits we already processed */
Eric Dumazet9ca1b222013-01-05 21:31:18 +00001916 poff += *off;
1917 plen -= *off;
1918 *off = 0;
Octavian Purdila2870c432008-07-15 00:49:11 -07001919
Eric Dumazet18aafc62013-01-11 14:46:37 +00001920 do {
1921 unsigned int flen = min(*len, plen);
Octavian Purdila2870c432008-07-15 00:49:11 -07001922
Eric Dumazet18aafc62013-01-11 14:46:37 +00001923 if (spd_fill_page(spd, pipe, page, &flen, poff,
1924 linear, sk))
1925 return true;
1926 poff += flen;
1927 plen -= flen;
1928 *len -= flen;
1929 } while (*len && plen);
Octavian Purdila2870c432008-07-15 00:49:11 -07001930
David S. Millera108d5f2012-04-23 23:06:11 -04001931 return false;
Octavian Purdila2870c432008-07-15 00:49:11 -07001932}
1933
1934/*
David S. Millera108d5f2012-04-23 23:06:11 -04001935 * Map linear and fragment data from the skb to spd. It reports true if the
Octavian Purdila2870c432008-07-15 00:49:11 -07001936 * pipe is full or if we already spliced the requested length.
1937 */
David S. Millera108d5f2012-04-23 23:06:11 -04001938static bool __skb_splice_bits(struct sk_buff *skb, struct pipe_inode_info *pipe,
1939 unsigned int *offset, unsigned int *len,
1940 struct splice_pipe_desc *spd, struct sock *sk)
Octavian Purdila2870c432008-07-15 00:49:11 -07001941{
1942 int seg;
Tom Herbertfa9835e2016-03-07 14:11:04 -08001943 struct sk_buff *iter;
Octavian Purdila2870c432008-07-15 00:49:11 -07001944
Eric Dumazet1d0c0b32012-04-27 02:10:03 +00001945 /* map the linear part :
Alexander Duyck2996d312012-05-02 18:18:42 +00001946 * If skb->head_frag is set, this 'linear' part is backed by a
1947 * fragment, and if the head is not shared with any clones then
1948 * we can avoid a copy since we own the head portion of this page.
Jens Axboe9c55e012007-11-06 23:30:13 -08001949 */
Octavian Purdila2870c432008-07-15 00:49:11 -07001950 if (__splice_segment(virt_to_page(skb->data),
1951 (unsigned long) skb->data & (PAGE_SIZE - 1),
1952 skb_headlen(skb),
Eric Dumazet18aafc62013-01-11 14:46:37 +00001953 offset, len, spd,
Alexander Duyck3a7c1ee42012-05-03 01:09:42 +00001954 skb_head_is_locked(skb),
Eric Dumazet1d0c0b32012-04-27 02:10:03 +00001955 sk, pipe))
David S. Millera108d5f2012-04-23 23:06:11 -04001956 return true;
Jens Axboe9c55e012007-11-06 23:30:13 -08001957
1958 /*
1959 * then map the fragments
1960 */
Jens Axboe9c55e012007-11-06 23:30:13 -08001961 for (seg = 0; seg < skb_shinfo(skb)->nr_frags; seg++) {
1962 const skb_frag_t *f = &skb_shinfo(skb)->frags[seg];
1963
Ian Campbellea2ab692011-08-22 23:44:58 +00001964 if (__splice_segment(skb_frag_page(f),
Eric Dumazet9e903e02011-10-18 21:00:24 +00001965 f->page_offset, skb_frag_size(f),
Eric Dumazet18aafc62013-01-11 14:46:37 +00001966 offset, len, spd, false, sk, pipe))
David S. Millera108d5f2012-04-23 23:06:11 -04001967 return true;
Jens Axboe9c55e012007-11-06 23:30:13 -08001968 }
1969
Tom Herbertfa9835e2016-03-07 14:11:04 -08001970 skb_walk_frags(skb, iter) {
1971 if (*offset >= iter->len) {
1972 *offset -= iter->len;
1973 continue;
1974 }
1975 /* __skb_splice_bits() only fails if the output has no room
1976 * left, so no point in going over the frag_list for the error
1977 * case.
1978 */
1979 if (__skb_splice_bits(iter, pipe, offset, len, spd, sk))
1980 return true;
1981 }
1982
David S. Millera108d5f2012-04-23 23:06:11 -04001983 return false;
Jens Axboe9c55e012007-11-06 23:30:13 -08001984}
1985
1986/*
1987 * Map data from the skb to a pipe. Should handle both the linear part,
Tom Herbertfa9835e2016-03-07 14:11:04 -08001988 * the fragments, and the frag list.
Jens Axboe9c55e012007-11-06 23:30:13 -08001989 */
Hannes Frederic Sowaa60e3cc2015-05-21 17:00:00 +02001990int skb_splice_bits(struct sk_buff *skb, struct sock *sk, unsigned int offset,
Jens Axboe9c55e012007-11-06 23:30:13 -08001991 struct pipe_inode_info *pipe, unsigned int tlen,
Al Viro25869262016-09-17 21:02:10 -04001992 unsigned int flags)
Jens Axboe9c55e012007-11-06 23:30:13 -08001993{
Eric Dumazet41c73a02012-04-22 12:26:16 +00001994 struct partial_page partial[MAX_SKB_FRAGS];
1995 struct page *pages[MAX_SKB_FRAGS];
Jens Axboe9c55e012007-11-06 23:30:13 -08001996 struct splice_pipe_desc spd = {
1997 .pages = pages,
1998 .partial = partial,
Eric Dumazet047fe362012-06-12 15:24:40 +02001999 .nr_pages_max = MAX_SKB_FRAGS,
Jens Axboe9c55e012007-11-06 23:30:13 -08002000 .flags = flags,
Miklos Szeredi28a625c2014-01-22 19:36:57 +01002001 .ops = &nosteal_pipe_buf_ops,
Jens Axboe9c55e012007-11-06 23:30:13 -08002002 .spd_release = sock_spd_release,
2003 };
Jens Axboe35f3d142010-05-20 10:43:18 +02002004 int ret = 0;
2005
Tom Herbertfa9835e2016-03-07 14:11:04 -08002006 __skb_splice_bits(skb, pipe, &offset, &tlen, &spd, sk);
Jens Axboe9c55e012007-11-06 23:30:13 -08002007
Hannes Frederic Sowaa60e3cc2015-05-21 17:00:00 +02002008 if (spd.nr_pages)
Al Viro25869262016-09-17 21:02:10 -04002009 ret = splice_to_pipe(pipe, &spd);
Jens Axboe9c55e012007-11-06 23:30:13 -08002010
Jens Axboe35f3d142010-05-20 10:43:18 +02002011 return ret;
Jens Axboe9c55e012007-11-06 23:30:13 -08002012}
Hannes Frederic Sowa2b514572015-05-21 17:00:01 +02002013EXPORT_SYMBOL_GPL(skb_splice_bits);
Jens Axboe9c55e012007-11-06 23:30:13 -08002014
Herbert Xu357b40a2005-04-19 22:30:14 -07002015/**
2016 * skb_store_bits - store bits from kernel buffer to skb
2017 * @skb: destination buffer
2018 * @offset: offset in destination
2019 * @from: source buffer
2020 * @len: number of bytes to copy
2021 *
2022 * Copy the specified number of bytes from the source buffer to the
2023 * destination skb. This function handles all the messy bits of
2024 * traversing fragment lists and such.
2025 */
2026
Stephen Hemminger0c6fcc82007-04-20 16:40:01 -07002027int skb_store_bits(struct sk_buff *skb, int offset, const void *from, int len)
Herbert Xu357b40a2005-04-19 22:30:14 -07002028{
David S. Miller1a028e52007-04-27 15:21:23 -07002029 int start = skb_headlen(skb);
David S. Millerfbb398a2009-06-09 00:18:59 -07002030 struct sk_buff *frag_iter;
2031 int i, copy;
Herbert Xu357b40a2005-04-19 22:30:14 -07002032
2033 if (offset > (int)skb->len - len)
2034 goto fault;
2035
David S. Miller1a028e52007-04-27 15:21:23 -07002036 if ((copy = start - offset) > 0) {
Herbert Xu357b40a2005-04-19 22:30:14 -07002037 if (copy > len)
2038 copy = len;
Arnaldo Carvalho de Melo27d7ff42007-03-31 11:55:19 -03002039 skb_copy_to_linear_data_offset(skb, offset, from, copy);
Herbert Xu357b40a2005-04-19 22:30:14 -07002040 if ((len -= copy) == 0)
2041 return 0;
2042 offset += copy;
2043 from += copy;
2044 }
2045
2046 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
2047 skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
David S. Miller1a028e52007-04-27 15:21:23 -07002048 int end;
Herbert Xu357b40a2005-04-19 22:30:14 -07002049
Ilpo Järvinen547b7922008-07-25 21:43:18 -07002050 WARN_ON(start > offset + len);
David S. Miller1a028e52007-04-27 15:21:23 -07002051
Eric Dumazet9e903e02011-10-18 21:00:24 +00002052 end = start + skb_frag_size(frag);
Herbert Xu357b40a2005-04-19 22:30:14 -07002053 if ((copy = end - offset) > 0) {
2054 u8 *vaddr;
2055
2056 if (copy > len)
2057 copy = len;
2058
Eric Dumazet51c56b02012-04-05 11:35:15 +02002059 vaddr = kmap_atomic(skb_frag_page(frag));
David S. Miller1a028e52007-04-27 15:21:23 -07002060 memcpy(vaddr + frag->page_offset + offset - start,
2061 from, copy);
Eric Dumazet51c56b02012-04-05 11:35:15 +02002062 kunmap_atomic(vaddr);
Herbert Xu357b40a2005-04-19 22:30:14 -07002063
2064 if ((len -= copy) == 0)
2065 return 0;
2066 offset += copy;
2067 from += copy;
2068 }
David S. Miller1a028e52007-04-27 15:21:23 -07002069 start = end;
Herbert Xu357b40a2005-04-19 22:30:14 -07002070 }
2071
David S. Millerfbb398a2009-06-09 00:18:59 -07002072 skb_walk_frags(skb, frag_iter) {
2073 int end;
Herbert Xu357b40a2005-04-19 22:30:14 -07002074
David S. Millerfbb398a2009-06-09 00:18:59 -07002075 WARN_ON(start > offset + len);
Herbert Xu357b40a2005-04-19 22:30:14 -07002076
David S. Millerfbb398a2009-06-09 00:18:59 -07002077 end = start + frag_iter->len;
2078 if ((copy = end - offset) > 0) {
2079 if (copy > len)
2080 copy = len;
2081 if (skb_store_bits(frag_iter, offset - start,
2082 from, copy))
2083 goto fault;
2084 if ((len -= copy) == 0)
2085 return 0;
2086 offset += copy;
2087 from += copy;
Herbert Xu357b40a2005-04-19 22:30:14 -07002088 }
David S. Millerfbb398a2009-06-09 00:18:59 -07002089 start = end;
Herbert Xu357b40a2005-04-19 22:30:14 -07002090 }
2091 if (!len)
2092 return 0;
2093
2094fault:
2095 return -EFAULT;
2096}
Herbert Xu357b40a2005-04-19 22:30:14 -07002097EXPORT_SYMBOL(skb_store_bits);
2098
Linus Torvalds1da177e2005-04-16 15:20:36 -07002099/* Checksum skb data. */
Daniel Borkmann2817a332013-10-30 11:50:51 +01002100__wsum __skb_checksum(const struct sk_buff *skb, int offset, int len,
2101 __wsum csum, const struct skb_checksum_ops *ops)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002102{
David S. Miller1a028e52007-04-27 15:21:23 -07002103 int start = skb_headlen(skb);
2104 int i, copy = start - offset;
David S. Millerfbb398a2009-06-09 00:18:59 -07002105 struct sk_buff *frag_iter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002106 int pos = 0;
2107
2108 /* Checksum header. */
2109 if (copy > 0) {
2110 if (copy > len)
2111 copy = len;
Daniel Borkmann2817a332013-10-30 11:50:51 +01002112 csum = ops->update(skb->data + offset, copy, csum);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002113 if ((len -= copy) == 0)
2114 return csum;
2115 offset += copy;
2116 pos = copy;
2117 }
2118
2119 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
David S. Miller1a028e52007-04-27 15:21:23 -07002120 int end;
Eric Dumazet51c56b02012-04-05 11:35:15 +02002121 skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
Linus Torvalds1da177e2005-04-16 15:20:36 -07002122
Ilpo Järvinen547b7922008-07-25 21:43:18 -07002123 WARN_ON(start > offset + len);
David S. Miller1a028e52007-04-27 15:21:23 -07002124
Eric Dumazet51c56b02012-04-05 11:35:15 +02002125 end = start + skb_frag_size(frag);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002126 if ((copy = end - offset) > 0) {
Al Viro44bb9362006-11-14 21:36:14 -08002127 __wsum csum2;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002128 u8 *vaddr;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002129
2130 if (copy > len)
2131 copy = len;
Eric Dumazet51c56b02012-04-05 11:35:15 +02002132 vaddr = kmap_atomic(skb_frag_page(frag));
Daniel Borkmann2817a332013-10-30 11:50:51 +01002133 csum2 = ops->update(vaddr + frag->page_offset +
2134 offset - start, copy, 0);
Eric Dumazet51c56b02012-04-05 11:35:15 +02002135 kunmap_atomic(vaddr);
Daniel Borkmann2817a332013-10-30 11:50:51 +01002136 csum = ops->combine(csum, csum2, pos, copy);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002137 if (!(len -= copy))
2138 return csum;
2139 offset += copy;
2140 pos += copy;
2141 }
David S. Miller1a028e52007-04-27 15:21:23 -07002142 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002143 }
2144
David S. Millerfbb398a2009-06-09 00:18:59 -07002145 skb_walk_frags(skb, frag_iter) {
2146 int end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002147
David S. Millerfbb398a2009-06-09 00:18:59 -07002148 WARN_ON(start > offset + len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002149
David S. Millerfbb398a2009-06-09 00:18:59 -07002150 end = start + frag_iter->len;
2151 if ((copy = end - offset) > 0) {
2152 __wsum csum2;
2153 if (copy > len)
2154 copy = len;
Daniel Borkmann2817a332013-10-30 11:50:51 +01002155 csum2 = __skb_checksum(frag_iter, offset - start,
2156 copy, 0, ops);
2157 csum = ops->combine(csum, csum2, pos, copy);
David S. Millerfbb398a2009-06-09 00:18:59 -07002158 if ((len -= copy) == 0)
2159 return csum;
2160 offset += copy;
2161 pos += copy;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002162 }
David S. Millerfbb398a2009-06-09 00:18:59 -07002163 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002164 }
Kris Katterjohn09a62662006-01-08 22:24:28 -08002165 BUG_ON(len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002166
2167 return csum;
2168}
Daniel Borkmann2817a332013-10-30 11:50:51 +01002169EXPORT_SYMBOL(__skb_checksum);
2170
2171__wsum skb_checksum(const struct sk_buff *skb, int offset,
2172 int len, __wsum csum)
2173{
2174 const struct skb_checksum_ops ops = {
Daniel Borkmanncea80ea2013-11-04 17:10:25 +01002175 .update = csum_partial_ext,
Daniel Borkmann2817a332013-10-30 11:50:51 +01002176 .combine = csum_block_add_ext,
2177 };
2178
2179 return __skb_checksum(skb, offset, len, csum, &ops);
2180}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002181EXPORT_SYMBOL(skb_checksum);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002182
2183/* Both of above in one bottle. */
2184
Al Viro81d77662006-11-14 21:37:33 -08002185__wsum skb_copy_and_csum_bits(const struct sk_buff *skb, int offset,
2186 u8 *to, int len, __wsum csum)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002187{
David S. Miller1a028e52007-04-27 15:21:23 -07002188 int start = skb_headlen(skb);
2189 int i, copy = start - offset;
David S. Millerfbb398a2009-06-09 00:18:59 -07002190 struct sk_buff *frag_iter;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002191 int pos = 0;
2192
2193 /* Copy header. */
2194 if (copy > 0) {
2195 if (copy > len)
2196 copy = len;
2197 csum = csum_partial_copy_nocheck(skb->data + offset, to,
2198 copy, csum);
2199 if ((len -= copy) == 0)
2200 return csum;
2201 offset += copy;
2202 to += copy;
2203 pos = copy;
2204 }
2205
2206 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
David S. Miller1a028e52007-04-27 15:21:23 -07002207 int end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002208
Ilpo Järvinen547b7922008-07-25 21:43:18 -07002209 WARN_ON(start > offset + len);
David S. Miller1a028e52007-04-27 15:21:23 -07002210
Eric Dumazet9e903e02011-10-18 21:00:24 +00002211 end = start + skb_frag_size(&skb_shinfo(skb)->frags[i]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002212 if ((copy = end - offset) > 0) {
Al Viro50842052006-11-14 21:36:34 -08002213 __wsum csum2;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002214 u8 *vaddr;
2215 skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
2216
2217 if (copy > len)
2218 copy = len;
Eric Dumazet51c56b02012-04-05 11:35:15 +02002219 vaddr = kmap_atomic(skb_frag_page(frag));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002220 csum2 = csum_partial_copy_nocheck(vaddr +
David S. Miller1a028e52007-04-27 15:21:23 -07002221 frag->page_offset +
2222 offset - start, to,
2223 copy, 0);
Eric Dumazet51c56b02012-04-05 11:35:15 +02002224 kunmap_atomic(vaddr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002225 csum = csum_block_add(csum, csum2, pos);
2226 if (!(len -= copy))
2227 return csum;
2228 offset += copy;
2229 to += copy;
2230 pos += copy;
2231 }
David S. Miller1a028e52007-04-27 15:21:23 -07002232 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002233 }
2234
David S. Millerfbb398a2009-06-09 00:18:59 -07002235 skb_walk_frags(skb, frag_iter) {
2236 __wsum csum2;
2237 int end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002238
David S. Millerfbb398a2009-06-09 00:18:59 -07002239 WARN_ON(start > offset + len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002240
David S. Millerfbb398a2009-06-09 00:18:59 -07002241 end = start + frag_iter->len;
2242 if ((copy = end - offset) > 0) {
2243 if (copy > len)
2244 copy = len;
2245 csum2 = skb_copy_and_csum_bits(frag_iter,
2246 offset - start,
2247 to, copy, 0);
2248 csum = csum_block_add(csum, csum2, pos);
2249 if ((len -= copy) == 0)
2250 return csum;
2251 offset += copy;
2252 to += copy;
2253 pos += copy;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002254 }
David S. Millerfbb398a2009-06-09 00:18:59 -07002255 start = end;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002256 }
Kris Katterjohn09a62662006-01-08 22:24:28 -08002257 BUG_ON(len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002258 return csum;
2259}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002260EXPORT_SYMBOL(skb_copy_and_csum_bits);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002261
Thomas Grafaf2806f2013-12-13 15:22:17 +01002262 /**
2263 * skb_zerocopy_headlen - Calculate headroom needed for skb_zerocopy()
2264 * @from: source buffer
2265 *
2266 * Calculates the amount of linear headroom needed in the 'to' skb passed
2267 * into skb_zerocopy().
2268 */
2269unsigned int
2270skb_zerocopy_headlen(const struct sk_buff *from)
2271{
2272 unsigned int hlen = 0;
2273
2274 if (!from->head_frag ||
2275 skb_headlen(from) < L1_CACHE_BYTES ||
2276 skb_shinfo(from)->nr_frags >= MAX_SKB_FRAGS)
2277 hlen = skb_headlen(from);
2278
2279 if (skb_has_frag_list(from))
2280 hlen = from->len;
2281
2282 return hlen;
2283}
2284EXPORT_SYMBOL_GPL(skb_zerocopy_headlen);
2285
2286/**
2287 * skb_zerocopy - Zero copy skb to skb
2288 * @to: destination buffer
Masanari Iida7fceb4d2014-01-29 01:05:28 +09002289 * @from: source buffer
Thomas Grafaf2806f2013-12-13 15:22:17 +01002290 * @len: number of bytes to copy from source buffer
2291 * @hlen: size of linear headroom in destination buffer
2292 *
2293 * Copies up to `len` bytes from `from` to `to` by creating references
2294 * to the frags in the source buffer.
2295 *
2296 * The `hlen` as calculated by skb_zerocopy_headlen() specifies the
2297 * headroom in the `to` buffer.
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002298 *
2299 * Return value:
2300 * 0: everything is OK
2301 * -ENOMEM: couldn't orphan frags of @from due to lack of memory
2302 * -EFAULT: skb_copy_bits() found some problem with skb geometry
Thomas Grafaf2806f2013-12-13 15:22:17 +01002303 */
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002304int
2305skb_zerocopy(struct sk_buff *to, struct sk_buff *from, int len, int hlen)
Thomas Grafaf2806f2013-12-13 15:22:17 +01002306{
2307 int i, j = 0;
2308 int plen = 0; /* length of skb->head fragment */
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002309 int ret;
Thomas Grafaf2806f2013-12-13 15:22:17 +01002310 struct page *page;
2311 unsigned int offset;
2312
2313 BUG_ON(!from->head_frag && !hlen);
2314
2315 /* dont bother with small payloads */
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002316 if (len <= skb_tailroom(to))
2317 return skb_copy_bits(from, 0, skb_put(to, len), len);
Thomas Grafaf2806f2013-12-13 15:22:17 +01002318
2319 if (hlen) {
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002320 ret = skb_copy_bits(from, 0, skb_put(to, hlen), hlen);
2321 if (unlikely(ret))
2322 return ret;
Thomas Grafaf2806f2013-12-13 15:22:17 +01002323 len -= hlen;
2324 } else {
2325 plen = min_t(int, skb_headlen(from), len);
2326 if (plen) {
2327 page = virt_to_head_page(from->head);
2328 offset = from->data - (unsigned char *)page_address(page);
2329 __skb_fill_page_desc(to, 0, page, offset, plen);
2330 get_page(page);
2331 j = 1;
2332 len -= plen;
2333 }
2334 }
2335
2336 to->truesize += len + plen;
2337 to->len += len + plen;
2338 to->data_len += len + plen;
2339
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002340 if (unlikely(skb_orphan_frags(from, GFP_ATOMIC))) {
2341 skb_tx_error(from);
2342 return -ENOMEM;
2343 }
2344
Thomas Grafaf2806f2013-12-13 15:22:17 +01002345 for (i = 0; i < skb_shinfo(from)->nr_frags; i++) {
2346 if (!len)
2347 break;
2348 skb_shinfo(to)->frags[j] = skb_shinfo(from)->frags[i];
2349 skb_shinfo(to)->frags[j].size = min_t(int, skb_shinfo(to)->frags[j].size, len);
2350 len -= skb_shinfo(to)->frags[j].size;
2351 skb_frag_ref(to, j);
2352 j++;
2353 }
2354 skb_shinfo(to)->nr_frags = j;
Zoltan Kiss36d5fe62014-03-26 22:37:45 +00002355
2356 return 0;
Thomas Grafaf2806f2013-12-13 15:22:17 +01002357}
2358EXPORT_SYMBOL_GPL(skb_zerocopy);
2359
Linus Torvalds1da177e2005-04-16 15:20:36 -07002360void skb_copy_and_csum_dev(const struct sk_buff *skb, u8 *to)
2361{
Al Virod3bc23e2006-11-14 21:24:49 -08002362 __wsum csum;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002363 long csstart;
2364
Patrick McHardy84fa7932006-08-29 16:44:56 -07002365 if (skb->ip_summed == CHECKSUM_PARTIAL)
Michał Mirosław55508d62010-12-14 15:24:08 +00002366 csstart = skb_checksum_start_offset(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002367 else
2368 csstart = skb_headlen(skb);
2369
Kris Katterjohn09a62662006-01-08 22:24:28 -08002370 BUG_ON(csstart > skb_headlen(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002371
Arnaldo Carvalho de Melod626f622007-03-27 18:55:52 -03002372 skb_copy_from_linear_data(skb, to, csstart);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002373
2374 csum = 0;
2375 if (csstart != skb->len)
2376 csum = skb_copy_and_csum_bits(skb, csstart, to + csstart,
2377 skb->len - csstart, 0);
2378
Patrick McHardy84fa7932006-08-29 16:44:56 -07002379 if (skb->ip_summed == CHECKSUM_PARTIAL) {
Al Viroff1dcad2006-11-20 18:07:29 -08002380 long csstuff = csstart + skb->csum_offset;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002381
Al Virod3bc23e2006-11-14 21:24:49 -08002382 *((__sum16 *)(to + csstuff)) = csum_fold(csum);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002383 }
2384}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002385EXPORT_SYMBOL(skb_copy_and_csum_dev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002386
2387/**
2388 * skb_dequeue - remove from the head of the queue
2389 * @list: list to dequeue from
2390 *
2391 * Remove the head of the list. The list lock is taken so the function
2392 * may be used safely with other locking list functions. The head item is
2393 * returned or %NULL if the list is empty.
2394 */
2395
2396struct sk_buff *skb_dequeue(struct sk_buff_head *list)
2397{
2398 unsigned long flags;
2399 struct sk_buff *result;
2400
2401 spin_lock_irqsave(&list->lock, flags);
2402 result = __skb_dequeue(list);
2403 spin_unlock_irqrestore(&list->lock, flags);
2404 return result;
2405}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002406EXPORT_SYMBOL(skb_dequeue);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002407
2408/**
2409 * skb_dequeue_tail - remove from the tail of the queue
2410 * @list: list to dequeue from
2411 *
2412 * Remove the tail of the list. The list lock is taken so the function
2413 * may be used safely with other locking list functions. The tail item is
2414 * returned or %NULL if the list is empty.
2415 */
2416struct sk_buff *skb_dequeue_tail(struct sk_buff_head *list)
2417{
2418 unsigned long flags;
2419 struct sk_buff *result;
2420
2421 spin_lock_irqsave(&list->lock, flags);
2422 result = __skb_dequeue_tail(list);
2423 spin_unlock_irqrestore(&list->lock, flags);
2424 return result;
2425}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002426EXPORT_SYMBOL(skb_dequeue_tail);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002427
2428/**
2429 * skb_queue_purge - empty a list
2430 * @list: list to empty
2431 *
2432 * Delete all buffers on an &sk_buff list. Each buffer is removed from
2433 * the list and one reference dropped. This function takes the list
2434 * lock and is atomic with respect to other list locking functions.
2435 */
2436void skb_queue_purge(struct sk_buff_head *list)
2437{
2438 struct sk_buff *skb;
2439 while ((skb = skb_dequeue(list)) != NULL)
2440 kfree_skb(skb);
2441}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002442EXPORT_SYMBOL(skb_queue_purge);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002443
2444/**
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002445 * skb_rbtree_purge - empty a skb rbtree
2446 * @root: root of the rbtree to empty
Peter Oskolkov87169592018-10-10 12:30:09 -07002447 * Return value: the sum of truesizes of all purged skbs.
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002448 *
2449 * Delete all buffers on an &sk_buff rbtree. Each buffer is removed from
2450 * the list and one reference dropped. This function does not take
2451 * any lock. Synchronization should be handled by the caller (e.g., TCP
2452 * out-of-order queue is protected by the socket lock).
2453 */
Peter Oskolkov87169592018-10-10 12:30:09 -07002454unsigned int skb_rbtree_purge(struct rb_root *root)
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002455{
Eric Dumazetf5d17b52018-10-10 12:30:08 -07002456 struct rb_node *p = rb_first(root);
Peter Oskolkov87169592018-10-10 12:30:09 -07002457 unsigned int sum = 0;
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002458
Eric Dumazetf5d17b52018-10-10 12:30:08 -07002459 while (p) {
2460 struct sk_buff *skb = rb_entry(p, struct sk_buff, rbnode);
2461
2462 p = rb_next(p);
2463 rb_erase(&skb->rbnode, root);
Peter Oskolkov87169592018-10-10 12:30:09 -07002464 sum += skb->truesize;
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002465 kfree_skb(skb);
Eric Dumazetf5d17b52018-10-10 12:30:08 -07002466 }
Peter Oskolkov87169592018-10-10 12:30:09 -07002467 return sum;
Yaogong Wang9f5afea2016-09-07 14:49:28 -07002468}
2469
2470/**
Linus Torvalds1da177e2005-04-16 15:20:36 -07002471 * skb_queue_head - queue a buffer at the list head
2472 * @list: list to use
2473 * @newsk: buffer to queue
2474 *
2475 * Queue a buffer at the start of the list. This function takes the
2476 * list lock and can be used safely with other locking &sk_buff functions
2477 * safely.
2478 *
2479 * A buffer cannot be placed on two lists at the same time.
2480 */
2481void skb_queue_head(struct sk_buff_head *list, struct sk_buff *newsk)
2482{
2483 unsigned long flags;
2484
2485 spin_lock_irqsave(&list->lock, flags);
2486 __skb_queue_head(list, newsk);
2487 spin_unlock_irqrestore(&list->lock, flags);
2488}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002489EXPORT_SYMBOL(skb_queue_head);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002490
2491/**
2492 * skb_queue_tail - queue a buffer at the list tail
2493 * @list: list to use
2494 * @newsk: buffer to queue
2495 *
2496 * Queue a buffer at the tail of the list. This function takes the
2497 * list lock and can be used safely with other locking &sk_buff functions
2498 * safely.
2499 *
2500 * A buffer cannot be placed on two lists at the same time.
2501 */
2502void skb_queue_tail(struct sk_buff_head *list, struct sk_buff *newsk)
2503{
2504 unsigned long flags;
2505
2506 spin_lock_irqsave(&list->lock, flags);
2507 __skb_queue_tail(list, newsk);
2508 spin_unlock_irqrestore(&list->lock, flags);
2509}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002510EXPORT_SYMBOL(skb_queue_tail);
David S. Miller8728b832005-08-09 19:25:21 -07002511
Linus Torvalds1da177e2005-04-16 15:20:36 -07002512/**
2513 * skb_unlink - remove a buffer from a list
2514 * @skb: buffer to remove
David S. Miller8728b832005-08-09 19:25:21 -07002515 * @list: list to use
Linus Torvalds1da177e2005-04-16 15:20:36 -07002516 *
David S. Miller8728b832005-08-09 19:25:21 -07002517 * Remove a packet from a list. The list locks are taken and this
2518 * function is atomic with respect to other list locked calls
Linus Torvalds1da177e2005-04-16 15:20:36 -07002519 *
David S. Miller8728b832005-08-09 19:25:21 -07002520 * You must know what list the SKB is on.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002521 */
David S. Miller8728b832005-08-09 19:25:21 -07002522void skb_unlink(struct sk_buff *skb, struct sk_buff_head *list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002523{
David S. Miller8728b832005-08-09 19:25:21 -07002524 unsigned long flags;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002525
David S. Miller8728b832005-08-09 19:25:21 -07002526 spin_lock_irqsave(&list->lock, flags);
2527 __skb_unlink(skb, list);
2528 spin_unlock_irqrestore(&list->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002529}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002530EXPORT_SYMBOL(skb_unlink);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002531
Linus Torvalds1da177e2005-04-16 15:20:36 -07002532/**
2533 * skb_append - append a buffer
2534 * @old: buffer to insert after
2535 * @newsk: buffer to insert
David S. Miller8728b832005-08-09 19:25:21 -07002536 * @list: list to use
Linus Torvalds1da177e2005-04-16 15:20:36 -07002537 *
2538 * Place a packet after a given packet in a list. The list locks are taken
2539 * and this function is atomic with respect to other list locked calls.
2540 * A buffer cannot be placed on two lists at the same time.
2541 */
David S. Miller8728b832005-08-09 19:25:21 -07002542void skb_append(struct sk_buff *old, struct sk_buff *newsk, struct sk_buff_head *list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002543{
2544 unsigned long flags;
2545
David S. Miller8728b832005-08-09 19:25:21 -07002546 spin_lock_irqsave(&list->lock, flags);
Gerrit Renker7de6c032008-04-14 00:05:09 -07002547 __skb_queue_after(list, old, newsk);
David S. Miller8728b832005-08-09 19:25:21 -07002548 spin_unlock_irqrestore(&list->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002549}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002550EXPORT_SYMBOL(skb_append);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002551
2552/**
2553 * skb_insert - insert a buffer
2554 * @old: buffer to insert before
2555 * @newsk: buffer to insert
David S. Miller8728b832005-08-09 19:25:21 -07002556 * @list: list to use
Linus Torvalds1da177e2005-04-16 15:20:36 -07002557 *
David S. Miller8728b832005-08-09 19:25:21 -07002558 * Place a packet before a given packet in a list. The list locks are
2559 * taken and this function is atomic with respect to other list locked
2560 * calls.
2561 *
Linus Torvalds1da177e2005-04-16 15:20:36 -07002562 * A buffer cannot be placed on two lists at the same time.
2563 */
David S. Miller8728b832005-08-09 19:25:21 -07002564void skb_insert(struct sk_buff *old, struct sk_buff *newsk, struct sk_buff_head *list)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002565{
2566 unsigned long flags;
2567
David S. Miller8728b832005-08-09 19:25:21 -07002568 spin_lock_irqsave(&list->lock, flags);
2569 __skb_insert(newsk, old->prev, old, list);
2570 spin_unlock_irqrestore(&list->lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002571}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002572EXPORT_SYMBOL(skb_insert);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002573
Linus Torvalds1da177e2005-04-16 15:20:36 -07002574static inline void skb_split_inside_header(struct sk_buff *skb,
2575 struct sk_buff* skb1,
2576 const u32 len, const int pos)
2577{
2578 int i;
2579
Arnaldo Carvalho de Melod626f622007-03-27 18:55:52 -03002580 skb_copy_from_linear_data_offset(skb, len, skb_put(skb1, pos - len),
2581 pos - len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002582 /* And move data appendix as is. */
2583 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++)
2584 skb_shinfo(skb1)->frags[i] = skb_shinfo(skb)->frags[i];
2585
2586 skb_shinfo(skb1)->nr_frags = skb_shinfo(skb)->nr_frags;
2587 skb_shinfo(skb)->nr_frags = 0;
2588 skb1->data_len = skb->data_len;
2589 skb1->len += skb1->data_len;
2590 skb->data_len = 0;
2591 skb->len = len;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07002592 skb_set_tail_pointer(skb, len);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002593}
2594
2595static inline void skb_split_no_header(struct sk_buff *skb,
2596 struct sk_buff* skb1,
2597 const u32 len, int pos)
2598{
2599 int i, k = 0;
2600 const int nfrags = skb_shinfo(skb)->nr_frags;
2601
2602 skb_shinfo(skb)->nr_frags = 0;
2603 skb1->len = skb1->data_len = skb->len - len;
2604 skb->len = len;
2605 skb->data_len = len - pos;
2606
2607 for (i = 0; i < nfrags; i++) {
Eric Dumazet9e903e02011-10-18 21:00:24 +00002608 int size = skb_frag_size(&skb_shinfo(skb)->frags[i]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002609
2610 if (pos + size > len) {
2611 skb_shinfo(skb1)->frags[k] = skb_shinfo(skb)->frags[i];
2612
2613 if (pos < len) {
2614 /* Split frag.
2615 * We have two variants in this case:
2616 * 1. Move all the frag to the second
2617 * part, if it is possible. F.e.
2618 * this approach is mandatory for TUX,
2619 * where splitting is expensive.
2620 * 2. Split is accurately. We make this.
2621 */
Ian Campbellea2ab692011-08-22 23:44:58 +00002622 skb_frag_ref(skb, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002623 skb_shinfo(skb1)->frags[0].page_offset += len - pos;
Eric Dumazet9e903e02011-10-18 21:00:24 +00002624 skb_frag_size_sub(&skb_shinfo(skb1)->frags[0], len - pos);
2625 skb_frag_size_set(&skb_shinfo(skb)->frags[i], len - pos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002626 skb_shinfo(skb)->nr_frags++;
2627 }
2628 k++;
2629 } else
2630 skb_shinfo(skb)->nr_frags++;
2631 pos += size;
2632 }
2633 skb_shinfo(skb1)->nr_frags = k;
2634}
2635
2636/**
2637 * skb_split - Split fragmented skb to two parts at length len.
2638 * @skb: the buffer to split
2639 * @skb1: the buffer to receive the second part
2640 * @len: new length for skb
2641 */
2642void skb_split(struct sk_buff *skb, struct sk_buff *skb1, const u32 len)
2643{
2644 int pos = skb_headlen(skb);
2645
Willem de Bruijnfcfb53c2017-06-08 11:35:03 -04002646 skb_shinfo(skb1)->tx_flags |= skb_shinfo(skb)->tx_flags &
2647 SKBTX_SHARED_FRAG;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002648 if (len < pos) /* Split line is inside header. */
2649 skb_split_inside_header(skb, skb1, len, pos);
2650 else /* Second chunk has no header, nothing to copy. */
2651 skb_split_no_header(skb, skb1, len, pos);
2652}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002653EXPORT_SYMBOL(skb_split);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002654
Ilpo Järvinen9f782db2008-11-25 13:57:01 -08002655/* Shifting from/to a cloned skb is a no-go.
2656 *
2657 * Caller cannot keep skb_shinfo related pointers past calling here!
2658 */
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002659static int skb_prepare_for_shift(struct sk_buff *skb)
2660{
Ilpo Järvinen0ace2852008-11-24 21:30:21 -08002661 return skb_cloned(skb) && pskb_expand_head(skb, 0, 0, GFP_ATOMIC);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002662}
2663
2664/**
2665 * skb_shift - Shifts paged data partially from skb to another
2666 * @tgt: buffer into which tail data gets added
2667 * @skb: buffer from which the paged data comes from
2668 * @shiftlen: shift up to this many bytes
2669 *
2670 * Attempts to shift up to shiftlen worth of bytes, which may be less than
Feng King20e994a2011-11-21 01:47:11 +00002671 * the length of the skb, from skb to tgt. Returns number bytes shifted.
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002672 * It's up to caller to free skb if everything was shifted.
2673 *
2674 * If @tgt runs out of frags, the whole operation is aborted.
2675 *
2676 * Skb cannot include anything else but paged data while tgt is allowed
2677 * to have non-paged data as well.
2678 *
2679 * TODO: full sized shift could be optimized but that would need
2680 * specialized skb free'er to handle frags without up-to-date nr_frags.
2681 */
2682int skb_shift(struct sk_buff *tgt, struct sk_buff *skb, int shiftlen)
2683{
2684 int from, to, merge, todo;
2685 struct skb_frag_struct *fragfrom, *fragto;
2686
2687 BUG_ON(shiftlen > skb->len);
2688 BUG_ON(skb_headlen(skb)); /* Would corrupt stream */
2689
2690 todo = shiftlen;
2691 from = 0;
2692 to = skb_shinfo(tgt)->nr_frags;
2693 fragfrom = &skb_shinfo(skb)->frags[from];
2694
2695 /* Actual merge is delayed until the point when we know we can
2696 * commit all, so that we don't have to undo partial changes
2697 */
2698 if (!to ||
Ian Campbellea2ab692011-08-22 23:44:58 +00002699 !skb_can_coalesce(tgt, to, skb_frag_page(fragfrom),
2700 fragfrom->page_offset)) {
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002701 merge = -1;
2702 } else {
2703 merge = to - 1;
2704
Eric Dumazet9e903e02011-10-18 21:00:24 +00002705 todo -= skb_frag_size(fragfrom);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002706 if (todo < 0) {
2707 if (skb_prepare_for_shift(skb) ||
2708 skb_prepare_for_shift(tgt))
2709 return 0;
2710
Ilpo Järvinen9f782db2008-11-25 13:57:01 -08002711 /* All previous frag pointers might be stale! */
2712 fragfrom = &skb_shinfo(skb)->frags[from];
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002713 fragto = &skb_shinfo(tgt)->frags[merge];
2714
Eric Dumazet9e903e02011-10-18 21:00:24 +00002715 skb_frag_size_add(fragto, shiftlen);
2716 skb_frag_size_sub(fragfrom, shiftlen);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002717 fragfrom->page_offset += shiftlen;
2718
2719 goto onlymerged;
2720 }
2721
2722 from++;
2723 }
2724
2725 /* Skip full, not-fitting skb to avoid expensive operations */
2726 if ((shiftlen == skb->len) &&
2727 (skb_shinfo(skb)->nr_frags - from) > (MAX_SKB_FRAGS - to))
2728 return 0;
2729
2730 if (skb_prepare_for_shift(skb) || skb_prepare_for_shift(tgt))
2731 return 0;
2732
2733 while ((todo > 0) && (from < skb_shinfo(skb)->nr_frags)) {
2734 if (to == MAX_SKB_FRAGS)
2735 return 0;
2736
2737 fragfrom = &skb_shinfo(skb)->frags[from];
2738 fragto = &skb_shinfo(tgt)->frags[to];
2739
Eric Dumazet9e903e02011-10-18 21:00:24 +00002740 if (todo >= skb_frag_size(fragfrom)) {
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002741 *fragto = *fragfrom;
Eric Dumazet9e903e02011-10-18 21:00:24 +00002742 todo -= skb_frag_size(fragfrom);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002743 from++;
2744 to++;
2745
2746 } else {
Ian Campbellea2ab692011-08-22 23:44:58 +00002747 __skb_frag_ref(fragfrom);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002748 fragto->page = fragfrom->page;
2749 fragto->page_offset = fragfrom->page_offset;
Eric Dumazet9e903e02011-10-18 21:00:24 +00002750 skb_frag_size_set(fragto, todo);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002751
2752 fragfrom->page_offset += todo;
Eric Dumazet9e903e02011-10-18 21:00:24 +00002753 skb_frag_size_sub(fragfrom, todo);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002754 todo = 0;
2755
2756 to++;
2757 break;
2758 }
2759 }
2760
2761 /* Ready to "commit" this state change to tgt */
2762 skb_shinfo(tgt)->nr_frags = to;
2763
2764 if (merge >= 0) {
2765 fragfrom = &skb_shinfo(skb)->frags[0];
2766 fragto = &skb_shinfo(tgt)->frags[merge];
2767
Eric Dumazet9e903e02011-10-18 21:00:24 +00002768 skb_frag_size_add(fragto, skb_frag_size(fragfrom));
Ian Campbellea2ab692011-08-22 23:44:58 +00002769 __skb_frag_unref(fragfrom);
Ilpo Järvinen832d11c2008-11-24 21:20:15 -08002770 }
2771
2772 /* Reposition in the original skb */
2773 to = 0;
2774 while (from < skb_shinfo(skb)->nr_frags)
2775 skb_shinfo(skb)->frags[to++] = skb_shinfo(skb)->frags[from++];
2776 skb_shinfo(skb)->nr_frags = to;
2777
2778 BUG_ON(todo > 0 && !skb_shinfo(skb)->nr_frags);
2779
2780onlymerged:
2781 /* Most likely the tgt won't ever need its checksum anymore, skb on
2782 * the other hand might need it if it needs to be resent
2783 */
2784 tgt->ip_summed = CHECKSUM_PARTIAL;
2785 skb->ip_summed = CHECKSUM_PARTIAL;
2786
2787 /* Yak, is it really working this way? Some helper please? */
2788 skb->len -= shiftlen;
2789 skb->data_len -= shiftlen;
2790 skb->truesize -= shiftlen;
2791 tgt->len += shiftlen;
2792 tgt->data_len += shiftlen;
2793 tgt->truesize += shiftlen;
2794
2795 return shiftlen;
2796}
2797
Thomas Graf677e90e2005-06-23 20:59:51 -07002798/**
2799 * skb_prepare_seq_read - Prepare a sequential read of skb data
2800 * @skb: the buffer to read
2801 * @from: lower offset of data to be read
2802 * @to: upper offset of data to be read
2803 * @st: state variable
2804 *
2805 * Initializes the specified state variable. Must be called before
2806 * invoking skb_seq_read() for the first time.
2807 */
2808void skb_prepare_seq_read(struct sk_buff *skb, unsigned int from,
2809 unsigned int to, struct skb_seq_state *st)
2810{
2811 st->lower_offset = from;
2812 st->upper_offset = to;
2813 st->root_skb = st->cur_skb = skb;
2814 st->frag_idx = st->stepped_offset = 0;
2815 st->frag_data = NULL;
2816}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002817EXPORT_SYMBOL(skb_prepare_seq_read);
Thomas Graf677e90e2005-06-23 20:59:51 -07002818
2819/**
2820 * skb_seq_read - Sequentially read skb data
2821 * @consumed: number of bytes consumed by the caller so far
2822 * @data: destination pointer for data to be returned
2823 * @st: state variable
2824 *
Mathias Krausebc323832013-11-07 14:18:26 +01002825 * Reads a block of skb data at @consumed relative to the
Thomas Graf677e90e2005-06-23 20:59:51 -07002826 * lower offset specified to skb_prepare_seq_read(). Assigns
Mathias Krausebc323832013-11-07 14:18:26 +01002827 * the head of the data block to @data and returns the length
Thomas Graf677e90e2005-06-23 20:59:51 -07002828 * of the block or 0 if the end of the skb data or the upper
2829 * offset has been reached.
2830 *
2831 * The caller is not required to consume all of the data
Mathias Krausebc323832013-11-07 14:18:26 +01002832 * returned, i.e. @consumed is typically set to the number
Thomas Graf677e90e2005-06-23 20:59:51 -07002833 * of bytes already consumed and the next call to
2834 * skb_seq_read() will return the remaining part of the block.
2835 *
Lucas De Marchi25985ed2011-03-30 22:57:33 -03002836 * Note 1: The size of each block of data returned can be arbitrary,
Masanari Iidae793c0f2014-09-04 23:44:36 +09002837 * this limitation is the cost for zerocopy sequential
Thomas Graf677e90e2005-06-23 20:59:51 -07002838 * reads of potentially non linear data.
2839 *
Randy Dunlapbc2cda12008-02-13 15:03:25 -08002840 * Note 2: Fragment lists within fragments are not implemented
Thomas Graf677e90e2005-06-23 20:59:51 -07002841 * at the moment, state->root_skb could be replaced with
2842 * a stack for this purpose.
2843 */
2844unsigned int skb_seq_read(unsigned int consumed, const u8 **data,
2845 struct skb_seq_state *st)
2846{
2847 unsigned int block_limit, abs_offset = consumed + st->lower_offset;
2848 skb_frag_t *frag;
2849
Wedson Almeida Filhoaeb193e2013-06-23 23:33:48 -07002850 if (unlikely(abs_offset >= st->upper_offset)) {
2851 if (st->frag_data) {
2852 kunmap_atomic(st->frag_data);
2853 st->frag_data = NULL;
2854 }
Thomas Graf677e90e2005-06-23 20:59:51 -07002855 return 0;
Wedson Almeida Filhoaeb193e2013-06-23 23:33:48 -07002856 }
Thomas Graf677e90e2005-06-23 20:59:51 -07002857
2858next_skb:
Herbert Xu95e3b242009-01-29 16:07:52 -08002859 block_limit = skb_headlen(st->cur_skb) + st->stepped_offset;
Thomas Graf677e90e2005-06-23 20:59:51 -07002860
Thomas Chenault995b3372009-05-18 21:43:27 -07002861 if (abs_offset < block_limit && !st->frag_data) {
Herbert Xu95e3b242009-01-29 16:07:52 -08002862 *data = st->cur_skb->data + (abs_offset - st->stepped_offset);
Thomas Graf677e90e2005-06-23 20:59:51 -07002863 return block_limit - abs_offset;
2864 }
2865
2866 if (st->frag_idx == 0 && !st->frag_data)
2867 st->stepped_offset += skb_headlen(st->cur_skb);
2868
2869 while (st->frag_idx < skb_shinfo(st->cur_skb)->nr_frags) {
2870 frag = &skb_shinfo(st->cur_skb)->frags[st->frag_idx];
Eric Dumazet9e903e02011-10-18 21:00:24 +00002871 block_limit = skb_frag_size(frag) + st->stepped_offset;
Thomas Graf677e90e2005-06-23 20:59:51 -07002872
2873 if (abs_offset < block_limit) {
2874 if (!st->frag_data)
Eric Dumazet51c56b02012-04-05 11:35:15 +02002875 st->frag_data = kmap_atomic(skb_frag_page(frag));
Thomas Graf677e90e2005-06-23 20:59:51 -07002876
2877 *data = (u8 *) st->frag_data + frag->page_offset +
2878 (abs_offset - st->stepped_offset);
2879
2880 return block_limit - abs_offset;
2881 }
2882
2883 if (st->frag_data) {
Eric Dumazet51c56b02012-04-05 11:35:15 +02002884 kunmap_atomic(st->frag_data);
Thomas Graf677e90e2005-06-23 20:59:51 -07002885 st->frag_data = NULL;
2886 }
2887
2888 st->frag_idx++;
Eric Dumazet9e903e02011-10-18 21:00:24 +00002889 st->stepped_offset += skb_frag_size(frag);
Thomas Graf677e90e2005-06-23 20:59:51 -07002890 }
2891
Olaf Kirch5b5a60d2007-06-23 23:11:52 -07002892 if (st->frag_data) {
Eric Dumazet51c56b02012-04-05 11:35:15 +02002893 kunmap_atomic(st->frag_data);
Olaf Kirch5b5a60d2007-06-23 23:11:52 -07002894 st->frag_data = NULL;
2895 }
2896
David S. Miller21dc3302010-08-23 00:13:46 -07002897 if (st->root_skb == st->cur_skb && skb_has_frag_list(st->root_skb)) {
Shyam Iyer71b33462009-01-29 16:12:42 -08002898 st->cur_skb = skb_shinfo(st->root_skb)->frag_list;
Thomas Graf677e90e2005-06-23 20:59:51 -07002899 st->frag_idx = 0;
2900 goto next_skb;
Shyam Iyer71b33462009-01-29 16:12:42 -08002901 } else if (st->cur_skb->next) {
2902 st->cur_skb = st->cur_skb->next;
Herbert Xu95e3b242009-01-29 16:07:52 -08002903 st->frag_idx = 0;
Thomas Graf677e90e2005-06-23 20:59:51 -07002904 goto next_skb;
2905 }
2906
2907 return 0;
2908}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002909EXPORT_SYMBOL(skb_seq_read);
Thomas Graf677e90e2005-06-23 20:59:51 -07002910
2911/**
2912 * skb_abort_seq_read - Abort a sequential read of skb data
2913 * @st: state variable
2914 *
2915 * Must be called if skb_seq_read() was not called until it
2916 * returned 0.
2917 */
2918void skb_abort_seq_read(struct skb_seq_state *st)
2919{
2920 if (st->frag_data)
Eric Dumazet51c56b02012-04-05 11:35:15 +02002921 kunmap_atomic(st->frag_data);
Thomas Graf677e90e2005-06-23 20:59:51 -07002922}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002923EXPORT_SYMBOL(skb_abort_seq_read);
Thomas Graf677e90e2005-06-23 20:59:51 -07002924
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002925#define TS_SKB_CB(state) ((struct skb_seq_state *) &((state)->cb))
2926
2927static unsigned int skb_ts_get_next_block(unsigned int offset, const u8 **text,
2928 struct ts_config *conf,
2929 struct ts_state *state)
2930{
2931 return skb_seq_read(offset, text, TS_SKB_CB(state));
2932}
2933
2934static void skb_ts_finish(struct ts_config *conf, struct ts_state *state)
2935{
2936 skb_abort_seq_read(TS_SKB_CB(state));
2937}
2938
2939/**
2940 * skb_find_text - Find a text pattern in skb data
2941 * @skb: the buffer to look in
2942 * @from: search offset
2943 * @to: search limit
2944 * @config: textsearch configuration
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002945 *
2946 * Finds a pattern in the skb data according to the specified
2947 * textsearch configuration. Use textsearch_next() to retrieve
2948 * subsequent occurrences of the pattern. Returns the offset
2949 * to the first occurrence or UINT_MAX if no match was found.
2950 */
2951unsigned int skb_find_text(struct sk_buff *skb, unsigned int from,
Bojan Prtvar059a2442015-02-22 11:46:35 +01002952 unsigned int to, struct ts_config *config)
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002953{
Bojan Prtvar059a2442015-02-22 11:46:35 +01002954 struct ts_state state;
Phil Oesterf72b9482006-06-26 00:00:57 -07002955 unsigned int ret;
2956
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002957 config->get_next_block = skb_ts_get_next_block;
2958 config->finish = skb_ts_finish;
2959
Bojan Prtvar059a2442015-02-22 11:46:35 +01002960 skb_prepare_seq_read(skb, from, to, TS_SKB_CB(&state));
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002961
Bojan Prtvar059a2442015-02-22 11:46:35 +01002962 ret = textsearch_find(config, &state);
Phil Oesterf72b9482006-06-26 00:00:57 -07002963 return (ret <= to - from ? ret : UINT_MAX);
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002964}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08002965EXPORT_SYMBOL(skb_find_text);
Thomas Graf3fc7e8a2005-06-23 21:00:17 -07002966
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002967/**
Ben Hutchings2c530402012-07-10 10:55:09 +00002968 * skb_append_datato_frags - append the user data to a skb
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002969 * @sk: sock structure
Masanari Iidae793c0f2014-09-04 23:44:36 +09002970 * @skb: skb structure to be appended with user data.
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002971 * @getfrag: call back function to be used for getting the user data
2972 * @from: pointer to user message iov
2973 * @length: length of the iov message
2974 *
2975 * Description: This procedure append the user data in the fragment part
2976 * of the skb if any page alloc fails user this procedure returns -ENOMEM
2977 */
2978int skb_append_datato_frags(struct sock *sk, struct sk_buff *skb,
Martin Waitzdab96302005-12-05 13:40:12 -08002979 int (*getfrag)(void *from, char *to, int offset,
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002980 int len, int odd, struct sk_buff *skb),
2981 void *from, int length)
2982{
Eric Dumazetb2111722012-12-28 06:06:37 +00002983 int frg_cnt = skb_shinfo(skb)->nr_frags;
2984 int copy;
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002985 int offset = 0;
2986 int ret;
Eric Dumazetb2111722012-12-28 06:06:37 +00002987 struct page_frag *pfrag = &current->task_frag;
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002988
2989 do {
2990 /* Return error if we don't have space for new frag */
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002991 if (frg_cnt >= MAX_SKB_FRAGS)
Eric Dumazetb2111722012-12-28 06:06:37 +00002992 return -EMSGSIZE;
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002993
Eric Dumazetb2111722012-12-28 06:06:37 +00002994 if (!sk_page_frag_refill(sk, pfrag))
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002995 return -ENOMEM;
2996
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002997 /* copy the user data to page */
Eric Dumazetb2111722012-12-28 06:06:37 +00002998 copy = min_t(int, length, pfrag->size - pfrag->offset);
Ananda Rajue89e9cf2005-10-18 15:46:41 -07002999
Eric Dumazetb2111722012-12-28 06:06:37 +00003000 ret = getfrag(from, page_address(pfrag->page) + pfrag->offset,
3001 offset, copy, 0, skb);
Ananda Rajue89e9cf2005-10-18 15:46:41 -07003002 if (ret < 0)
3003 return -EFAULT;
3004
3005 /* copy was successful so update the size parameters */
Eric Dumazetb2111722012-12-28 06:06:37 +00003006 skb_fill_page_desc(skb, frg_cnt, pfrag->page, pfrag->offset,
3007 copy);
3008 frg_cnt++;
3009 pfrag->offset += copy;
3010 get_page(pfrag->page);
3011
3012 skb->truesize += copy;
3013 atomic_add(copy, &sk->sk_wmem_alloc);
Ananda Rajue89e9cf2005-10-18 15:46:41 -07003014 skb->len += copy;
3015 skb->data_len += copy;
3016 offset += copy;
3017 length -= copy;
3018
3019 } while (length > 0);
3020
3021 return 0;
3022}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08003023EXPORT_SYMBOL(skb_append_datato_frags);
Ananda Rajue89e9cf2005-10-18 15:46:41 -07003024
Hannes Frederic Sowabe12a1f2015-05-21 16:59:58 +02003025int skb_append_pagefrags(struct sk_buff *skb, struct page *page,
3026 int offset, size_t size)
3027{
3028 int i = skb_shinfo(skb)->nr_frags;
3029
3030 if (skb_can_coalesce(skb, i, page, offset)) {
3031 skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], size);
3032 } else if (i < MAX_SKB_FRAGS) {
3033 get_page(page);
3034 skb_fill_page_desc(skb, i, page, offset, size);
3035 } else {
3036 return -EMSGSIZE;
3037 }
3038
3039 return 0;
3040}
3041EXPORT_SYMBOL_GPL(skb_append_pagefrags);
3042
Herbert Xucbb042f2006-03-20 22:43:56 -08003043/**
3044 * skb_pull_rcsum - pull skb and update receive checksum
3045 * @skb: buffer to update
Herbert Xucbb042f2006-03-20 22:43:56 -08003046 * @len: length of data pulled
3047 *
3048 * This function performs an skb_pull on the packet and updates
Urs Thuermannfee54fa2008-02-12 22:03:25 -08003049 * the CHECKSUM_COMPLETE checksum. It should be used on
Patrick McHardy84fa7932006-08-29 16:44:56 -07003050 * receive path processing instead of skb_pull unless you know
3051 * that the checksum difference is zero (e.g., a valid IP header)
3052 * or you are setting ip_summed to CHECKSUM_NONE.
Herbert Xucbb042f2006-03-20 22:43:56 -08003053 */
3054unsigned char *skb_pull_rcsum(struct sk_buff *skb, unsigned int len)
3055{
Pravin B Shelar31b33df2015-09-28 17:24:25 -07003056 unsigned char *data = skb->data;
3057
Herbert Xucbb042f2006-03-20 22:43:56 -08003058 BUG_ON(len > skb->len);
Pravin B Shelar31b33df2015-09-28 17:24:25 -07003059 __skb_pull(skb, len);
3060 skb_postpull_rcsum(skb, data, len);
3061 return skb->data;
Herbert Xucbb042f2006-03-20 22:43:56 -08003062}
Arnaldo Carvalho de Melof94691a2006-03-20 22:47:55 -08003063EXPORT_SYMBOL_GPL(skb_pull_rcsum);
3064
Herbert Xuf4c50d92006-06-22 03:02:40 -07003065/**
3066 * skb_segment - Perform protocol segmentation on skb.
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003067 * @head_skb: buffer to segment
Herbert Xu576a30e2006-06-27 13:22:38 -07003068 * @features: features for the output path (see dev->features)
Herbert Xuf4c50d92006-06-22 03:02:40 -07003069 *
3070 * This function performs segmentation on the given skb. It returns
Ben Hutchings4c821d72008-04-13 21:52:48 -07003071 * a pointer to the first in a list of new skbs for the segments.
3072 * In case of error it returns ERR_PTR(err).
Herbert Xuf4c50d92006-06-22 03:02:40 -07003073 */
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003074struct sk_buff *skb_segment(struct sk_buff *head_skb,
3075 netdev_features_t features)
Herbert Xuf4c50d92006-06-22 03:02:40 -07003076{
3077 struct sk_buff *segs = NULL;
3078 struct sk_buff *tail = NULL;
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003079 struct sk_buff *list_skb = skb_shinfo(head_skb)->frag_list;
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003080 skb_frag_t *frag = skb_shinfo(head_skb)->frags;
3081 unsigned int mss = skb_shinfo(head_skb)->gso_size;
3082 unsigned int doffset = head_skb->data - skb_mac_header(head_skb);
Michael S. Tsirkin1fd819e2014-03-10 19:28:08 +02003083 struct sk_buff *frag_skb = head_skb;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003084 unsigned int offset = doffset;
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003085 unsigned int tnl_hlen = skb_tnl_header_len(head_skb);
Alexander Duyck802ab552016-04-10 21:45:03 -04003086 unsigned int partial_segs = 0;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003087 unsigned int headroom;
Alexander Duyck802ab552016-04-10 21:45:03 -04003088 unsigned int len = head_skb->len;
Pravin B Shelarec5f0612013-03-07 09:28:01 +00003089 __be16 proto;
Alexander Duyck36c98382016-05-02 09:38:18 -07003090 bool csum, sg;
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003091 int nfrags = skb_shinfo(head_skb)->nr_frags;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003092 int err = -ENOMEM;
3093 int i = 0;
3094 int pos;
Vlad Yasevich53d64712014-03-27 17:26:18 -04003095 int dummy;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003096
Shmulik Ladkani162a5a82019-09-06 12:23:50 +03003097 if (list_skb && !list_skb->head_frag && skb_headlen(list_skb) &&
3098 (skb_shinfo(head_skb)->gso_type & SKB_GSO_DODGY)) {
3099 /* gso_size is untrusted, and we have a frag_list with a linear
3100 * non head_frag head.
3101 *
3102 * (we assume checking the first list_skb member suffices;
3103 * i.e if either of the list_skb members have non head_frag
3104 * head, then the first one has too).
3105 *
3106 * If head_skb's headlen does not fit requested gso_size, it
3107 * means that the frag_list members do NOT terminate on exact
3108 * gso_size boundaries. Hence we cannot perform skb_frag_t page
3109 * sharing. Therefore we must fallback to copying the frag_list
3110 * skbs; we do so by disabling SG.
3111 */
3112 if (mss != GSO_BY_FRAGS && mss != skb_headlen(head_skb))
3113 features &= ~NETIF_F_SG;
3114 }
3115
Wei-Chun Chao5882a072014-06-08 23:48:54 -07003116 __skb_push(head_skb, doffset);
Vlad Yasevich53d64712014-03-27 17:26:18 -04003117 proto = skb_network_protocol(head_skb, &dummy);
Pravin B Shelarec5f0612013-03-07 09:28:01 +00003118 if (unlikely(!proto))
3119 return ERR_PTR(-EINVAL);
3120
Alexander Duyck36c98382016-05-02 09:38:18 -07003121 sg = !!(features & NETIF_F_SG);
Alexander Duyckf245d072016-02-05 15:28:26 -08003122 csum = !!can_checksum_protocol(features, proto);
Tom Herbert7e2b10c2014-06-04 17:20:02 -07003123
Steffen Klassert07b26c92016-09-19 12:58:47 +02003124 if (sg && csum && (mss != GSO_BY_FRAGS)) {
3125 if (!(features & NETIF_F_GSO_PARTIAL)) {
3126 struct sk_buff *iter;
Ilan Tayari7ab89b1762017-04-19 21:26:07 +03003127 unsigned int frag_len;
Steffen Klassert07b26c92016-09-19 12:58:47 +02003128
3129 if (!list_skb ||
3130 !net_gso_ok(features, skb_shinfo(head_skb)->gso_type))
3131 goto normal;
3132
Ilan Tayari7ab89b1762017-04-19 21:26:07 +03003133 /* If we get here then all the required
3134 * GSO features except frag_list are supported.
3135 * Try to split the SKB to multiple GSO SKBs
3136 * with no frag_list.
3137 * Currently we can do that only when the buffers don't
3138 * have a linear part and all the buffers except
3139 * the last are of the same length.
Steffen Klassert07b26c92016-09-19 12:58:47 +02003140 */
Ilan Tayari7ab89b1762017-04-19 21:26:07 +03003141 frag_len = list_skb->len;
Steffen Klassert07b26c92016-09-19 12:58:47 +02003142 skb_walk_frags(head_skb, iter) {
Ilan Tayari7ab89b1762017-04-19 21:26:07 +03003143 if (frag_len != iter->len && iter->next)
3144 goto normal;
Steffen Klassert07b26c92016-09-19 12:58:47 +02003145 if (skb_headlen(iter))
3146 goto normal;
3147
3148 len -= iter->len;
3149 }
Ilan Tayari7ab89b1762017-04-19 21:26:07 +03003150
3151 if (len != frag_len)
3152 goto normal;
Steffen Klassert07b26c92016-09-19 12:58:47 +02003153 }
3154
3155 /* GSO partial only requires that we trim off any excess that
3156 * doesn't fit into an MSS sized block, so take care of that
3157 * now.
3158 */
Alexander Duyck802ab552016-04-10 21:45:03 -04003159 partial_segs = len / mss;
Alexander Duyckd7fb5a82016-05-02 09:38:12 -07003160 if (partial_segs > 1)
3161 mss *= partial_segs;
3162 else
3163 partial_segs = 0;
Alexander Duyck802ab552016-04-10 21:45:03 -04003164 }
3165
Steffen Klassert07b26c92016-09-19 12:58:47 +02003166normal:
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003167 headroom = skb_headroom(head_skb);
3168 pos = skb_headlen(head_skb);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003169
3170 do {
3171 struct sk_buff *nskb;
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003172 skb_frag_t *nskb_frag;
Herbert Xuc8884ed2006-10-29 15:59:41 -08003173 int hsize;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003174 int size;
3175
Marcelo Ricardo Leitner3953c462016-06-02 15:05:40 -03003176 if (unlikely(mss == GSO_BY_FRAGS)) {
3177 len = list_skb->len;
3178 } else {
3179 len = head_skb->len - offset;
3180 if (len > mss)
3181 len = mss;
3182 }
Herbert Xuf4c50d92006-06-22 03:02:40 -07003183
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003184 hsize = skb_headlen(head_skb) - offset;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003185 if (hsize < 0)
3186 hsize = 0;
Herbert Xuc8884ed2006-10-29 15:59:41 -08003187 if (hsize > len || !sg)
3188 hsize = len;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003189
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003190 if (!hsize && i >= nfrags && skb_headlen(list_skb) &&
3191 (skb_headlen(list_skb) == len || sg)) {
3192 BUG_ON(skb_headlen(list_skb) > len);
Herbert Xu89319d382008-12-15 23:26:06 -08003193
Herbert Xu9d8506c2013-11-21 11:10:04 -08003194 i = 0;
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003195 nfrags = skb_shinfo(list_skb)->nr_frags;
3196 frag = skb_shinfo(list_skb)->frags;
Michael S. Tsirkin1fd819e2014-03-10 19:28:08 +02003197 frag_skb = list_skb;
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003198 pos += skb_headlen(list_skb);
Herbert Xu9d8506c2013-11-21 11:10:04 -08003199
3200 while (pos < offset + len) {
3201 BUG_ON(i >= nfrags);
3202
Michael S. Tsirkin4e1beba2014-03-10 18:29:14 +02003203 size = skb_frag_size(frag);
Herbert Xu9d8506c2013-11-21 11:10:04 -08003204 if (pos + size > offset + len)
3205 break;
3206
3207 i++;
3208 pos += size;
Michael S. Tsirkin4e1beba2014-03-10 18:29:14 +02003209 frag++;
Herbert Xu9d8506c2013-11-21 11:10:04 -08003210 }
3211
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003212 nskb = skb_clone(list_skb, GFP_ATOMIC);
3213 list_skb = list_skb->next;
Herbert Xu89319d382008-12-15 23:26:06 -08003214
3215 if (unlikely(!nskb))
3216 goto err;
3217
Herbert Xu9d8506c2013-11-21 11:10:04 -08003218 if (unlikely(pskb_trim(nskb, len))) {
3219 kfree_skb(nskb);
3220 goto err;
3221 }
3222
Alexander Duyckec47ea82012-05-04 14:26:56 +00003223 hsize = skb_end_offset(nskb);
Herbert Xu89319d382008-12-15 23:26:06 -08003224 if (skb_cow_head(nskb, doffset + headroom)) {
3225 kfree_skb(nskb);
3226 goto err;
3227 }
3228
Alexander Duyckec47ea82012-05-04 14:26:56 +00003229 nskb->truesize += skb_end_offset(nskb) - hsize;
Herbert Xu89319d382008-12-15 23:26:06 -08003230 skb_release_head_state(nskb);
3231 __skb_push(nskb, doffset);
3232 } else {
Mel Gormanc93bdd02012-07-31 16:44:19 -07003233 nskb = __alloc_skb(hsize + doffset + headroom,
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003234 GFP_ATOMIC, skb_alloc_rx_flag(head_skb),
Mel Gormanc93bdd02012-07-31 16:44:19 -07003235 NUMA_NO_NODE);
Herbert Xu89319d382008-12-15 23:26:06 -08003236
3237 if (unlikely(!nskb))
3238 goto err;
3239
3240 skb_reserve(nskb, headroom);
3241 __skb_put(nskb, doffset);
3242 }
Herbert Xuf4c50d92006-06-22 03:02:40 -07003243
3244 if (segs)
3245 tail->next = nskb;
3246 else
3247 segs = nskb;
3248 tail = nskb;
3249
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003250 __copy_skb_header(nskb, head_skb);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003251
Eric Dumazet030737b2013-10-19 11:42:54 -07003252 skb_headers_offset_update(nskb, skb_headroom(nskb) - headroom);
Vlad Yasevichfcdfe3a2014-07-31 10:33:06 -04003253 skb_reset_mac_len(nskb);
Pravin B Shelar68c33162013-02-14 14:02:41 +00003254
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003255 skb_copy_from_linear_data_offset(head_skb, -tnl_hlen,
Pravin B Shelar68c33162013-02-14 14:02:41 +00003256 nskb->data - tnl_hlen,
3257 doffset + tnl_hlen);
Herbert Xu89319d382008-12-15 23:26:06 -08003258
Herbert Xu9d8506c2013-11-21 11:10:04 -08003259 if (nskb->len == len + doffset)
Simon Horman1cdbcb72013-05-19 15:46:49 +00003260 goto perform_csum_check;
Herbert Xu89319d382008-12-15 23:26:06 -08003261
Alexander Duyck7fbeffe2016-02-05 15:27:43 -08003262 if (!sg) {
3263 if (!nskb->remcsum_offload)
3264 nskb->ip_summed = CHECKSUM_NONE;
Alexander Duyck76443452016-02-05 15:27:37 -08003265 SKB_GSO_CB(nskb)->csum =
3266 skb_copy_and_csum_bits(head_skb, offset,
3267 skb_put(nskb, len),
3268 len, 0);
Tom Herbert7e2b10c2014-06-04 17:20:02 -07003269 SKB_GSO_CB(nskb)->csum_start =
Alexander Duyck76443452016-02-05 15:27:37 -08003270 skb_headroom(nskb) + doffset;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003271 continue;
3272 }
3273
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003274 nskb_frag = skb_shinfo(nskb)->frags;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003275
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003276 skb_copy_from_linear_data_offset(head_skb, offset,
Arnaldo Carvalho de Melod626f622007-03-27 18:55:52 -03003277 skb_put(nskb, hsize), hsize);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003278
Willem de Bruijnfcfb53c2017-06-08 11:35:03 -04003279 skb_shinfo(nskb)->tx_flags |= skb_shinfo(head_skb)->tx_flags &
3280 SKBTX_SHARED_FRAG;
Eric Dumazetcef401d2013-01-25 20:34:37 +00003281
Herbert Xu9d8506c2013-11-21 11:10:04 -08003282 while (pos < offset + len) {
3283 if (i >= nfrags) {
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003284 BUG_ON(skb_headlen(list_skb));
Herbert Xu9d8506c2013-11-21 11:10:04 -08003285
3286 i = 0;
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003287 nfrags = skb_shinfo(list_skb)->nr_frags;
3288 frag = skb_shinfo(list_skb)->frags;
Michael S. Tsirkin1fd819e2014-03-10 19:28:08 +02003289 frag_skb = list_skb;
Herbert Xu9d8506c2013-11-21 11:10:04 -08003290
3291 BUG_ON(!nfrags);
3292
Michael S. Tsirkin1a4ceda2014-03-10 19:27:59 +02003293 list_skb = list_skb->next;
Herbert Xu9d8506c2013-11-21 11:10:04 -08003294 }
3295
3296 if (unlikely(skb_shinfo(nskb)->nr_frags >=
3297 MAX_SKB_FRAGS)) {
3298 net_warn_ratelimited(
3299 "skb_segment: too many frags: %u %u\n",
3300 pos, mss);
Eric Dumazete2ffdd62018-07-19 16:04:38 -07003301 err = -EINVAL;
Herbert Xu9d8506c2013-11-21 11:10:04 -08003302 goto err;
3303 }
3304
Michael S. Tsirkin1fd819e2014-03-10 19:28:08 +02003305 if (unlikely(skb_orphan_frags(frag_skb, GFP_ATOMIC)))
3306 goto err;
3307
Michael S. Tsirkin4e1beba2014-03-10 18:29:14 +02003308 *nskb_frag = *frag;
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003309 __skb_frag_ref(nskb_frag);
3310 size = skb_frag_size(nskb_frag);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003311
3312 if (pos < offset) {
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003313 nskb_frag->page_offset += offset - pos;
3314 skb_frag_size_sub(nskb_frag, offset - pos);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003315 }
3316
Herbert Xu89319d382008-12-15 23:26:06 -08003317 skb_shinfo(nskb)->nr_frags++;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003318
3319 if (pos + size <= offset + len) {
3320 i++;
Michael S. Tsirkin4e1beba2014-03-10 18:29:14 +02003321 frag++;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003322 pos += size;
3323 } else {
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003324 skb_frag_size_sub(nskb_frag, pos + size - (offset + len));
Herbert Xu89319d382008-12-15 23:26:06 -08003325 goto skip_fraglist;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003326 }
3327
Michael S. Tsirkin8cb19902014-03-10 18:29:04 +02003328 nskb_frag++;
Herbert Xuf4c50d92006-06-22 03:02:40 -07003329 }
3330
Herbert Xu89319d382008-12-15 23:26:06 -08003331skip_fraglist:
Herbert Xuf4c50d92006-06-22 03:02:40 -07003332 nskb->data_len = len - hsize;
3333 nskb->len += nskb->data_len;
3334 nskb->truesize += nskb->data_len;
Pravin B Shelarec5f0612013-03-07 09:28:01 +00003335
Simon Horman1cdbcb72013-05-19 15:46:49 +00003336perform_csum_check:
Alexander Duyck7fbeffe2016-02-05 15:27:43 -08003337 if (!csum) {
Eric Dumazete2ffdd62018-07-19 16:04:38 -07003338 if (skb_has_shared_frag(nskb) &&
3339 __skb_linearize(nskb))
3340 goto err;
3341
Alexander Duyck7fbeffe2016-02-05 15:27:43 -08003342 if (!nskb->remcsum_offload)
3343 nskb->ip_summed = CHECKSUM_NONE;
Alexander Duyck76443452016-02-05 15:27:37 -08003344 SKB_GSO_CB(nskb)->csum =
3345 skb_checksum(nskb, doffset,
3346 nskb->len - doffset, 0);
Tom Herbert7e2b10c2014-06-04 17:20:02 -07003347 SKB_GSO_CB(nskb)->csum_start =
Alexander Duyck76443452016-02-05 15:27:37 -08003348 skb_headroom(nskb) + doffset;
Pravin B Shelarec5f0612013-03-07 09:28:01 +00003349 }
Michael S. Tsirkindf5771f2014-03-10 18:29:19 +02003350 } while ((offset += len) < head_skb->len);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003351
Eric Dumazetbec3cfd2014-10-03 20:59:19 -07003352 /* Some callers want to get the end of the list.
3353 * Put it in segs->prev to avoid walking the list.
3354 * (see validate_xmit_skb_list() for example)
3355 */
3356 segs->prev = tail;
Toshiaki Makita432c8562014-10-27 10:30:51 -07003357
Alexander Duyck802ab552016-04-10 21:45:03 -04003358 if (partial_segs) {
Steffen Klassert07b26c92016-09-19 12:58:47 +02003359 struct sk_buff *iter;
Alexander Duyck802ab552016-04-10 21:45:03 -04003360 int type = skb_shinfo(head_skb)->gso_type;
Steffen Klassert07b26c92016-09-19 12:58:47 +02003361 unsigned short gso_size = skb_shinfo(head_skb)->gso_size;
Alexander Duyck802ab552016-04-10 21:45:03 -04003362
3363 /* Update type to add partial and then remove dodgy if set */
Steffen Klassert07b26c92016-09-19 12:58:47 +02003364 type |= (features & NETIF_F_GSO_PARTIAL) / NETIF_F_GSO_PARTIAL * SKB_GSO_PARTIAL;
Alexander Duyck802ab552016-04-10 21:45:03 -04003365 type &= ~SKB_GSO_DODGY;
3366
3367 /* Update GSO info and prepare to start updating headers on
3368 * our way back down the stack of protocols.
3369 */
Steffen Klassert07b26c92016-09-19 12:58:47 +02003370 for (iter = segs; iter; iter = iter->next) {
3371 skb_shinfo(iter)->gso_size = gso_size;
3372 skb_shinfo(iter)->gso_segs = partial_segs;
3373 skb_shinfo(iter)->gso_type = type;
3374 SKB_GSO_CB(iter)->data_offset = skb_headroom(iter) + doffset;
3375 }
3376
3377 if (tail->len - doffset <= gso_size)
3378 skb_shinfo(tail)->gso_size = 0;
3379 else if (tail != segs)
3380 skb_shinfo(tail)->gso_segs = DIV_ROUND_UP(tail->len - doffset, gso_size);
Alexander Duyck802ab552016-04-10 21:45:03 -04003381 }
3382
Toshiaki Makita432c8562014-10-27 10:30:51 -07003383 /* Following permits correct backpressure, for protocols
3384 * using skb_set_owner_w().
3385 * Idea is to tranfert ownership from head_skb to last segment.
3386 */
3387 if (head_skb->destructor == sock_wfree) {
3388 swap(tail->truesize, head_skb->truesize);
3389 swap(tail->destructor, head_skb->destructor);
3390 swap(tail->sk, head_skb->sk);
3391 }
Herbert Xuf4c50d92006-06-22 03:02:40 -07003392 return segs;
3393
3394err:
Eric Dumazet289dccb2013-12-20 14:29:08 -08003395 kfree_skb_list(segs);
Herbert Xuf4c50d92006-06-22 03:02:40 -07003396 return ERR_PTR(err);
3397}
Herbert Xuf4c50d92006-06-22 03:02:40 -07003398EXPORT_SYMBOL_GPL(skb_segment);
3399
Herbert Xu71d93b32008-12-15 23:42:33 -08003400int skb_gro_receive(struct sk_buff **head, struct sk_buff *skb)
3401{
Eric Dumazet8a291112013-10-08 09:02:23 -07003402 struct skb_shared_info *pinfo, *skbinfo = skb_shinfo(skb);
Herbert Xu67147ba2009-05-26 18:50:22 +00003403 unsigned int offset = skb_gro_offset(skb);
3404 unsigned int headlen = skb_headlen(skb);
Eric Dumazet8a291112013-10-08 09:02:23 -07003405 unsigned int len = skb_gro_len(skb);
Eric Dumazet58025e42015-03-05 13:47:48 -08003406 struct sk_buff *lp, *p = *head;
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003407 unsigned int delta_truesize;
Herbert Xu71d93b32008-12-15 23:42:33 -08003408
Eric Dumazet8a291112013-10-08 09:02:23 -07003409 if (unlikely(p->len + len >= 65536))
Herbert Xu71d93b32008-12-15 23:42:33 -08003410 return -E2BIG;
3411
Eric Dumazet29e98242014-05-16 11:34:37 -07003412 lp = NAPI_GRO_CB(p)->last;
Eric Dumazet8a291112013-10-08 09:02:23 -07003413 pinfo = skb_shinfo(lp);
3414
3415 if (headlen <= offset) {
Herbert Xu42da6992009-05-26 18:50:19 +00003416 skb_frag_t *frag;
Herbert Xu66e92fc2009-05-26 18:50:32 +00003417 skb_frag_t *frag2;
Herbert Xu9aaa1562009-05-26 18:50:33 +00003418 int i = skbinfo->nr_frags;
3419 int nr_frags = pinfo->nr_frags + i;
Herbert Xu42da6992009-05-26 18:50:19 +00003420
Herbert Xu66e92fc2009-05-26 18:50:32 +00003421 if (nr_frags > MAX_SKB_FRAGS)
Eric Dumazet8a291112013-10-08 09:02:23 -07003422 goto merge;
Herbert Xu81705ad2009-01-29 14:19:51 +00003423
Eric Dumazet8a291112013-10-08 09:02:23 -07003424 offset -= headlen;
Herbert Xu9aaa1562009-05-26 18:50:33 +00003425 pinfo->nr_frags = nr_frags;
3426 skbinfo->nr_frags = 0;
Herbert Xuf5572062009-01-14 20:40:03 -08003427
Herbert Xu9aaa1562009-05-26 18:50:33 +00003428 frag = pinfo->frags + nr_frags;
3429 frag2 = skbinfo->frags + i;
Herbert Xu66e92fc2009-05-26 18:50:32 +00003430 do {
3431 *--frag = *--frag2;
3432 } while (--i);
3433
3434 frag->page_offset += offset;
Eric Dumazet9e903e02011-10-18 21:00:24 +00003435 skb_frag_size_sub(frag, offset);
Herbert Xu66e92fc2009-05-26 18:50:32 +00003436
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003437 /* all fragments truesize : remove (head size + sk_buff) */
Alexander Duyckec47ea82012-05-04 14:26:56 +00003438 delta_truesize = skb->truesize -
3439 SKB_TRUESIZE(skb_end_offset(skb));
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003440
Herbert Xuf5572062009-01-14 20:40:03 -08003441 skb->truesize -= skb->data_len;
3442 skb->len -= skb->data_len;
3443 skb->data_len = 0;
3444
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003445 NAPI_GRO_CB(skb)->free = NAPI_GRO_FREE;
Herbert Xu5d38a072009-01-04 16:13:40 -08003446 goto done;
Eric Dumazetd7e88832012-04-30 08:10:34 +00003447 } else if (skb->head_frag) {
3448 int nr_frags = pinfo->nr_frags;
3449 skb_frag_t *frag = pinfo->frags + nr_frags;
3450 struct page *page = virt_to_head_page(skb->head);
3451 unsigned int first_size = headlen - offset;
3452 unsigned int first_offset;
3453
3454 if (nr_frags + 1 + skbinfo->nr_frags > MAX_SKB_FRAGS)
Eric Dumazet8a291112013-10-08 09:02:23 -07003455 goto merge;
Eric Dumazetd7e88832012-04-30 08:10:34 +00003456
3457 first_offset = skb->data -
3458 (unsigned char *)page_address(page) +
3459 offset;
3460
3461 pinfo->nr_frags = nr_frags + 1 + skbinfo->nr_frags;
3462
3463 frag->page.p = page;
3464 frag->page_offset = first_offset;
3465 skb_frag_size_set(frag, first_size);
3466
3467 memcpy(frag + 1, skbinfo->frags, sizeof(*frag) * skbinfo->nr_frags);
3468 /* We dont need to clear skbinfo->nr_frags here */
3469
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003470 delta_truesize = skb->truesize - SKB_DATA_ALIGN(sizeof(struct sk_buff));
Eric Dumazetd7e88832012-04-30 08:10:34 +00003471 NAPI_GRO_CB(skb)->free = NAPI_GRO_FREE_STOLEN_HEAD;
3472 goto done;
Eric Dumazet8a291112013-10-08 09:02:23 -07003473 }
Herbert Xu71d93b32008-12-15 23:42:33 -08003474
3475merge:
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003476 delta_truesize = skb->truesize;
Herbert Xu67147ba2009-05-26 18:50:22 +00003477 if (offset > headlen) {
Michal Schmidtd1dc7ab2011-01-24 12:08:48 +00003478 unsigned int eat = offset - headlen;
3479
3480 skbinfo->frags[0].page_offset += eat;
Eric Dumazet9e903e02011-10-18 21:00:24 +00003481 skb_frag_size_sub(&skbinfo->frags[0], eat);
Michal Schmidtd1dc7ab2011-01-24 12:08:48 +00003482 skb->data_len -= eat;
3483 skb->len -= eat;
Herbert Xu67147ba2009-05-26 18:50:22 +00003484 offset = headlen;
Herbert Xu56035022009-02-05 21:26:52 -08003485 }
3486
Herbert Xu67147ba2009-05-26 18:50:22 +00003487 __skb_pull(skb, offset);
Herbert Xu56035022009-02-05 21:26:52 -08003488
Eric Dumazet29e98242014-05-16 11:34:37 -07003489 if (NAPI_GRO_CB(p)->last == p)
Eric Dumazet8a291112013-10-08 09:02:23 -07003490 skb_shinfo(p)->frag_list = skb;
3491 else
3492 NAPI_GRO_CB(p)->last->next = skb;
Eric Dumazetc3c7c252012-12-06 13:54:59 +00003493 NAPI_GRO_CB(p)->last = skb;
Eric Dumazetf4a775d2014-09-22 16:29:32 -07003494 __skb_header_release(skb);
Eric Dumazet8a291112013-10-08 09:02:23 -07003495 lp = p;
Herbert Xu71d93b32008-12-15 23:42:33 -08003496
Herbert Xu5d38a072009-01-04 16:13:40 -08003497done:
3498 NAPI_GRO_CB(p)->count++;
Herbert Xu37fe4732009-01-17 19:48:13 +00003499 p->data_len += len;
Eric Dumazet715dc1f2012-05-02 23:33:21 +00003500 p->truesize += delta_truesize;
Herbert Xu37fe4732009-01-17 19:48:13 +00003501 p->len += len;
Eric Dumazet8a291112013-10-08 09:02:23 -07003502 if (lp != p) {
3503 lp->data_len += len;
3504 lp->truesize += delta_truesize;
3505 lp->len += len;
3506 }
Herbert Xu71d93b32008-12-15 23:42:33 -08003507 NAPI_GRO_CB(skb)->same_flow = 1;
3508 return 0;
3509}
Marcelo Ricardo Leitner57c056502016-06-02 15:05:39 -03003510EXPORT_SYMBOL_GPL(skb_gro_receive);
Herbert Xu71d93b32008-12-15 23:42:33 -08003511
Linus Torvalds1da177e2005-04-16 15:20:36 -07003512void __init skb_init(void)
3513{
3514 skbuff_head_cache = kmem_cache_create("skbuff_head_cache",
3515 sizeof(struct sk_buff),
3516 0,
Alexey Dobriyane5d679f332006-08-26 19:25:52 -07003517 SLAB_HWCACHE_ALIGN|SLAB_PANIC,
Paul Mundt20c2df82007-07-20 10:11:58 +09003518 NULL);
David S. Millerd179cd12005-08-17 14:57:30 -07003519 skbuff_fclone_cache = kmem_cache_create("skbuff_fclone_cache",
Eric Dumazetd0bf4a92014-09-29 13:29:15 -07003520 sizeof(struct sk_buff_fclones),
David S. Millerd179cd12005-08-17 14:57:30 -07003521 0,
Alexey Dobriyane5d679f332006-08-26 19:25:52 -07003522 SLAB_HWCACHE_ALIGN|SLAB_PANIC,
Paul Mundt20c2df82007-07-20 10:11:58 +09003523 NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003524}
3525
David S. Miller51c739d2007-10-30 21:29:29 -07003526static int
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003527__skb_to_sgvec(struct sk_buff *skb, struct scatterlist *sg, int offset, int len,
3528 unsigned int recursion_level)
David Howells716ea3a2007-04-02 20:19:53 -07003529{
David S. Miller1a028e52007-04-27 15:21:23 -07003530 int start = skb_headlen(skb);
3531 int i, copy = start - offset;
David S. Millerfbb398a2009-06-09 00:18:59 -07003532 struct sk_buff *frag_iter;
David Howells716ea3a2007-04-02 20:19:53 -07003533 int elt = 0;
3534
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003535 if (unlikely(recursion_level >= 24))
3536 return -EMSGSIZE;
3537
David Howells716ea3a2007-04-02 20:19:53 -07003538 if (copy > 0) {
3539 if (copy > len)
3540 copy = len;
Jens Axboe642f1492007-10-24 11:20:47 +02003541 sg_set_buf(sg, skb->data + offset, copy);
David Howells716ea3a2007-04-02 20:19:53 -07003542 elt++;
3543 if ((len -= copy) == 0)
3544 return elt;
3545 offset += copy;
3546 }
3547
3548 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
David S. Miller1a028e52007-04-27 15:21:23 -07003549 int end;
David Howells716ea3a2007-04-02 20:19:53 -07003550
Ilpo Järvinen547b7922008-07-25 21:43:18 -07003551 WARN_ON(start > offset + len);
David S. Miller1a028e52007-04-27 15:21:23 -07003552
Eric Dumazet9e903e02011-10-18 21:00:24 +00003553 end = start + skb_frag_size(&skb_shinfo(skb)->frags[i]);
David Howells716ea3a2007-04-02 20:19:53 -07003554 if ((copy = end - offset) > 0) {
3555 skb_frag_t *frag = &skb_shinfo(skb)->frags[i];
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003556 if (unlikely(elt && sg_is_last(&sg[elt - 1])))
3557 return -EMSGSIZE;
David Howells716ea3a2007-04-02 20:19:53 -07003558
3559 if (copy > len)
3560 copy = len;
Ian Campbellea2ab692011-08-22 23:44:58 +00003561 sg_set_page(&sg[elt], skb_frag_page(frag), copy,
Jens Axboe642f1492007-10-24 11:20:47 +02003562 frag->page_offset+offset-start);
David Howells716ea3a2007-04-02 20:19:53 -07003563 elt++;
3564 if (!(len -= copy))
3565 return elt;
3566 offset += copy;
3567 }
David S. Miller1a028e52007-04-27 15:21:23 -07003568 start = end;
David Howells716ea3a2007-04-02 20:19:53 -07003569 }
3570
David S. Millerfbb398a2009-06-09 00:18:59 -07003571 skb_walk_frags(skb, frag_iter) {
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003572 int end, ret;
David Howells716ea3a2007-04-02 20:19:53 -07003573
David S. Millerfbb398a2009-06-09 00:18:59 -07003574 WARN_ON(start > offset + len);
David Howells716ea3a2007-04-02 20:19:53 -07003575
David S. Millerfbb398a2009-06-09 00:18:59 -07003576 end = start + frag_iter->len;
3577 if ((copy = end - offset) > 0) {
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003578 if (unlikely(elt && sg_is_last(&sg[elt - 1])))
3579 return -EMSGSIZE;
3580
David S. Millerfbb398a2009-06-09 00:18:59 -07003581 if (copy > len)
3582 copy = len;
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003583 ret = __skb_to_sgvec(frag_iter, sg+elt, offset - start,
3584 copy, recursion_level + 1);
3585 if (unlikely(ret < 0))
3586 return ret;
3587 elt += ret;
David S. Millerfbb398a2009-06-09 00:18:59 -07003588 if ((len -= copy) == 0)
3589 return elt;
3590 offset += copy;
David Howells716ea3a2007-04-02 20:19:53 -07003591 }
David S. Millerfbb398a2009-06-09 00:18:59 -07003592 start = end;
David Howells716ea3a2007-04-02 20:19:53 -07003593 }
3594 BUG_ON(len);
3595 return elt;
3596}
3597
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003598/**
3599 * skb_to_sgvec - Fill a scatter-gather list from a socket buffer
3600 * @skb: Socket buffer containing the buffers to be mapped
3601 * @sg: The scatter-gather list to map into
3602 * @offset: The offset into the buffer's contents to start mapping
3603 * @len: Length of buffer space to be mapped
3604 *
3605 * Fill the specified scatter-gather list with mappings/pointers into a
3606 * region of the buffer space attached to a socket buffer. Returns either
3607 * the number of scatterlist items used, or -EMSGSIZE if the contents
3608 * could not fit.
3609 */
3610int skb_to_sgvec(struct sk_buff *skb, struct scatterlist *sg, int offset, int len)
3611{
3612 int nsg = __skb_to_sgvec(skb, sg, offset, len, 0);
3613
3614 if (nsg <= 0)
3615 return nsg;
3616
3617 sg_mark_end(&sg[nsg - 1]);
3618
3619 return nsg;
3620}
3621EXPORT_SYMBOL_GPL(skb_to_sgvec);
3622
Fan Du25a91d82014-01-18 09:54:23 +08003623/* As compared with skb_to_sgvec, skb_to_sgvec_nomark only map skb to given
3624 * sglist without mark the sg which contain last skb data as the end.
3625 * So the caller can mannipulate sg list as will when padding new data after
3626 * the first call without calling sg_unmark_end to expend sg list.
3627 *
3628 * Scenario to use skb_to_sgvec_nomark:
3629 * 1. sg_init_table
3630 * 2. skb_to_sgvec_nomark(payload1)
3631 * 3. skb_to_sgvec_nomark(payload2)
3632 *
3633 * This is equivalent to:
3634 * 1. sg_init_table
3635 * 2. skb_to_sgvec(payload1)
3636 * 3. sg_unmark_end
3637 * 4. skb_to_sgvec(payload2)
3638 *
3639 * When mapping mutilple payload conditionally, skb_to_sgvec_nomark
3640 * is more preferable.
3641 */
3642int skb_to_sgvec_nomark(struct sk_buff *skb, struct scatterlist *sg,
3643 int offset, int len)
3644{
Jason A. Donenfeld2cd3aa52017-06-04 04:16:22 +02003645 return __skb_to_sgvec(skb, sg, offset, len, 0);
Fan Du25a91d82014-01-18 09:54:23 +08003646}
3647EXPORT_SYMBOL_GPL(skb_to_sgvec_nomark);
3648
David S. Miller51c739d2007-10-30 21:29:29 -07003649
David S. Miller51c739d2007-10-30 21:29:29 -07003650
David Howells716ea3a2007-04-02 20:19:53 -07003651/**
3652 * skb_cow_data - Check that a socket buffer's data buffers are writable
3653 * @skb: The socket buffer to check.
3654 * @tailbits: Amount of trailing space to be added
3655 * @trailer: Returned pointer to the skb where the @tailbits space begins
3656 *
3657 * Make sure that the data buffers attached to a socket buffer are
3658 * writable. If they are not, private copies are made of the data buffers
3659 * and the socket buffer is set to use these instead.
3660 *
3661 * If @tailbits is given, make sure that there is space to write @tailbits
3662 * bytes of data beyond current end of socket buffer. @trailer will be
3663 * set to point to the skb in which this space begins.
3664 *
3665 * The number of scatterlist elements required to completely map the
3666 * COW'd and extended socket buffer will be returned.
3667 */
3668int skb_cow_data(struct sk_buff *skb, int tailbits, struct sk_buff **trailer)
3669{
3670 int copyflag;
3671 int elt;
3672 struct sk_buff *skb1, **skb_p;
3673
3674 /* If skb is cloned or its head is paged, reallocate
3675 * head pulling out all the pages (pages are considered not writable
3676 * at the moment even if they are anonymous).
3677 */
3678 if ((skb_cloned(skb) || skb_shinfo(skb)->nr_frags) &&
3679 __pskb_pull_tail(skb, skb_pagelen(skb)-skb_headlen(skb)) == NULL)
3680 return -ENOMEM;
3681
3682 /* Easy case. Most of packets will go this way. */
David S. Miller21dc3302010-08-23 00:13:46 -07003683 if (!skb_has_frag_list(skb)) {
David Howells716ea3a2007-04-02 20:19:53 -07003684 /* A little of trouble, not enough of space for trailer.
3685 * This should not happen, when stack is tuned to generate
3686 * good frames. OK, on miss we reallocate and reserve even more
3687 * space, 128 bytes is fair. */
3688
3689 if (skb_tailroom(skb) < tailbits &&
3690 pskb_expand_head(skb, 0, tailbits-skb_tailroom(skb)+128, GFP_ATOMIC))
3691 return -ENOMEM;
3692
3693 /* Voila! */
3694 *trailer = skb;
3695 return 1;
3696 }
3697
3698 /* Misery. We are in troubles, going to mincer fragments... */
3699
3700 elt = 1;
3701 skb_p = &skb_shinfo(skb)->frag_list;
3702 copyflag = 0;
3703
3704 while ((skb1 = *skb_p) != NULL) {
3705 int ntail = 0;
3706
3707 /* The fragment is partially pulled by someone,
3708 * this can happen on input. Copy it and everything
3709 * after it. */
3710
3711 if (skb_shared(skb1))
3712 copyflag = 1;
3713
3714 /* If the skb is the last, worry about trailer. */
3715
3716 if (skb1->next == NULL && tailbits) {
3717 if (skb_shinfo(skb1)->nr_frags ||
David S. Miller21dc3302010-08-23 00:13:46 -07003718 skb_has_frag_list(skb1) ||
David Howells716ea3a2007-04-02 20:19:53 -07003719 skb_tailroom(skb1) < tailbits)
3720 ntail = tailbits + 128;
3721 }
3722
3723 if (copyflag ||
3724 skb_cloned(skb1) ||
3725 ntail ||
3726 skb_shinfo(skb1)->nr_frags ||
David S. Miller21dc3302010-08-23 00:13:46 -07003727 skb_has_frag_list(skb1)) {
David Howells716ea3a2007-04-02 20:19:53 -07003728 struct sk_buff *skb2;
3729
3730 /* Fuck, we are miserable poor guys... */
3731 if (ntail == 0)
3732 skb2 = skb_copy(skb1, GFP_ATOMIC);
3733 else
3734 skb2 = skb_copy_expand(skb1,
3735 skb_headroom(skb1),
3736 ntail,
3737 GFP_ATOMIC);
3738 if (unlikely(skb2 == NULL))
3739 return -ENOMEM;
3740
3741 if (skb1->sk)
3742 skb_set_owner_w(skb2, skb1->sk);
3743
3744 /* Looking around. Are we still alive?
3745 * OK, link new skb, drop old one */
3746
3747 skb2->next = skb1->next;
3748 *skb_p = skb2;
3749 kfree_skb(skb1);
3750 skb1 = skb2;
3751 }
3752 elt++;
3753 *trailer = skb1;
3754 skb_p = &skb1->next;
3755 }
3756
3757 return elt;
3758}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08003759EXPORT_SYMBOL_GPL(skb_cow_data);
David Howells716ea3a2007-04-02 20:19:53 -07003760
Eric Dumazetb1faf562010-05-31 23:44:05 -07003761static void sock_rmem_free(struct sk_buff *skb)
3762{
3763 struct sock *sk = skb->sk;
3764
3765 atomic_sub(skb->truesize, &sk->sk_rmem_alloc);
3766}
3767
3768/*
3769 * Note: We dont mem charge error packets (no sk_forward_alloc changes)
3770 */
3771int sock_queue_err_skb(struct sock *sk, struct sk_buff *skb)
3772{
3773 if (atomic_read(&sk->sk_rmem_alloc) + skb->truesize >=
Eric Dumazet95c96172012-04-15 05:58:06 +00003774 (unsigned int)sk->sk_rcvbuf)
Eric Dumazetb1faf562010-05-31 23:44:05 -07003775 return -ENOMEM;
3776
3777 skb_orphan(skb);
3778 skb->sk = sk;
3779 skb->destructor = sock_rmem_free;
3780 atomic_add(skb->truesize, &sk->sk_rmem_alloc);
3781
Eric Dumazetabb57ea2011-05-18 02:21:31 -04003782 /* before exiting rcu section, make sure dst is refcounted */
3783 skb_dst_force(skb);
3784
Eric Dumazetb1faf562010-05-31 23:44:05 -07003785 skb_queue_tail(&sk->sk_error_queue, skb);
3786 if (!sock_flag(sk, SOCK_DEAD))
Vinicius Costa Gomesd5862b02018-03-14 13:32:09 -07003787 sk->sk_error_report(sk);
Eric Dumazetb1faf562010-05-31 23:44:05 -07003788 return 0;
3789}
3790EXPORT_SYMBOL(sock_queue_err_skb);
3791
Willem de Bruijn364a9e92014-08-31 21:30:27 -04003792struct sk_buff *sock_dequeue_err_skb(struct sock *sk)
3793{
3794 struct sk_buff_head *q = &sk->sk_error_queue;
3795 struct sk_buff *skb, *skb_next;
Eric Dumazet997d5c32015-02-18 05:47:55 -08003796 unsigned long flags;
Willem de Bruijn364a9e92014-08-31 21:30:27 -04003797 int err = 0;
3798
Eric Dumazet997d5c32015-02-18 05:47:55 -08003799 spin_lock_irqsave(&q->lock, flags);
Willem de Bruijn364a9e92014-08-31 21:30:27 -04003800 skb = __skb_dequeue(q);
3801 if (skb && (skb_next = skb_peek(q)))
3802 err = SKB_EXT_ERR(skb_next)->ee.ee_errno;
Eric Dumazet997d5c32015-02-18 05:47:55 -08003803 spin_unlock_irqrestore(&q->lock, flags);
Willem de Bruijn364a9e92014-08-31 21:30:27 -04003804
3805 sk->sk_err = err;
3806 if (err)
3807 sk->sk_error_report(sk);
3808
3809 return skb;
3810}
3811EXPORT_SYMBOL(sock_dequeue_err_skb);
3812
Alexander Duyckcab41c42014-09-10 18:05:26 -04003813/**
3814 * skb_clone_sk - create clone of skb, and take reference to socket
3815 * @skb: the skb to clone
3816 *
3817 * This function creates a clone of a buffer that holds a reference on
3818 * sk_refcnt. Buffers created via this function are meant to be
3819 * returned using sock_queue_err_skb, or free via kfree_skb.
3820 *
3821 * When passing buffers allocated with this function to sock_queue_err_skb
3822 * it is necessary to wrap the call with sock_hold/sock_put in order to
3823 * prevent the socket from being released prior to being enqueued on
3824 * the sk_error_queue.
3825 */
Alexander Duyck62bccb82014-09-04 13:31:35 -04003826struct sk_buff *skb_clone_sk(struct sk_buff *skb)
3827{
3828 struct sock *sk = skb->sk;
3829 struct sk_buff *clone;
3830
3831 if (!sk || !atomic_inc_not_zero(&sk->sk_refcnt))
3832 return NULL;
3833
3834 clone = skb_clone(skb, GFP_ATOMIC);
3835 if (!clone) {
3836 sock_put(sk);
3837 return NULL;
3838 }
3839
3840 clone->sk = sk;
3841 clone->destructor = sock_efree;
3842
3843 return clone;
3844}
3845EXPORT_SYMBOL(skb_clone_sk);
3846
Alexander Duyck37846ef2014-09-04 13:31:10 -04003847static void __skb_complete_tx_timestamp(struct sk_buff *skb,
3848 struct sock *sk,
3849 int tstype)
Patrick Ohlyac45f602009-02-12 05:03:37 +00003850{
Patrick Ohlyac45f602009-02-12 05:03:37 +00003851 struct sock_exterr_skb *serr;
Patrick Ohlyac45f602009-02-12 05:03:37 +00003852 int err;
3853
Patrick Ohlyac45f602009-02-12 05:03:37 +00003854 serr = SKB_EXT_ERR(skb);
3855 memset(serr, 0, sizeof(*serr));
3856 serr->ee.ee_errno = ENOMSG;
3857 serr->ee.ee_origin = SO_EE_ORIGIN_TIMESTAMPING;
Willem de Bruijne7fd2882014-08-04 22:11:48 -04003858 serr->ee.ee_info = tstype;
Willem de Bruijn8c04e2a2017-04-12 19:24:35 -04003859 serr->header.h4.iif = skb->dev ? skb->dev->ifindex : 0;
Willem de Bruijn4ed2d762014-08-04 22:11:49 -04003860 if (sk->sk_tsflags & SOF_TIMESTAMPING_OPT_ID) {
Willem de Bruijn09c2d252014-08-04 22:11:47 -04003861 serr->ee.ee_data = skb_shinfo(skb)->tskey;
WANG Congac5cc972015-12-16 23:39:04 -08003862 if (sk->sk_protocol == IPPROTO_TCP &&
3863 sk->sk_type == SOCK_STREAM)
Willem de Bruijn4ed2d762014-08-04 22:11:49 -04003864 serr->ee.ee_data -= sk->sk_tskey;
3865 }
Eric Dumazet29030372010-05-29 00:20:48 -07003866
Patrick Ohlyac45f602009-02-12 05:03:37 +00003867 err = sock_queue_err_skb(sk, skb);
Eric Dumazet29030372010-05-29 00:20:48 -07003868
Patrick Ohlyac45f602009-02-12 05:03:37 +00003869 if (err)
3870 kfree_skb(skb);
3871}
Alexander Duyck37846ef2014-09-04 13:31:10 -04003872
Willem de Bruijnb245be12015-01-30 13:29:32 -05003873static bool skb_may_tx_timestamp(struct sock *sk, bool tsonly)
3874{
3875 bool ret;
3876
3877 if (likely(sysctl_tstamp_allow_data || tsonly))
3878 return true;
3879
3880 read_lock_bh(&sk->sk_callback_lock);
3881 ret = sk->sk_socket && sk->sk_socket->file &&
3882 file_ns_capable(sk->sk_socket->file, &init_user_ns, CAP_NET_RAW);
3883 read_unlock_bh(&sk->sk_callback_lock);
3884 return ret;
3885}
3886
Alexander Duyck37846ef2014-09-04 13:31:10 -04003887void skb_complete_tx_timestamp(struct sk_buff *skb,
3888 struct skb_shared_hwtstamps *hwtstamps)
3889{
3890 struct sock *sk = skb->sk;
3891
Willem de Bruijnb245be12015-01-30 13:29:32 -05003892 if (!skb_may_tx_timestamp(sk, false))
Willem de Bruijn58f6ebb2017-12-13 14:41:06 -05003893 goto err;
Willem de Bruijnb245be12015-01-30 13:29:32 -05003894
Eric Dumazetf157cc12017-03-03 21:01:03 -08003895 /* Take a reference to prevent skb_orphan() from freeing the socket,
3896 * but only if the socket refcount is not zero.
3897 */
3898 if (likely(atomic_inc_not_zero(&sk->sk_refcnt))) {
3899 *skb_hwtstamps(skb) = *hwtstamps;
3900 __skb_complete_tx_timestamp(skb, sk, SCM_TSTAMP_SND);
3901 sock_put(sk);
Willem de Bruijn58f6ebb2017-12-13 14:41:06 -05003902 return;
Eric Dumazetf157cc12017-03-03 21:01:03 -08003903 }
Willem de Bruijn58f6ebb2017-12-13 14:41:06 -05003904
3905err:
3906 kfree_skb(skb);
Alexander Duyck37846ef2014-09-04 13:31:10 -04003907}
3908EXPORT_SYMBOL_GPL(skb_complete_tx_timestamp);
3909
3910void __skb_tstamp_tx(struct sk_buff *orig_skb,
3911 struct skb_shared_hwtstamps *hwtstamps,
3912 struct sock *sk, int tstype)
3913{
3914 struct sk_buff *skb;
Willem de Bruijn3a8dd972015-03-11 15:43:55 -04003915 bool tsonly;
Alexander Duyck37846ef2014-09-04 13:31:10 -04003916
Willem de Bruijn3a8dd972015-03-11 15:43:55 -04003917 if (!sk)
3918 return;
3919
3920 tsonly = sk->sk_tsflags & SOF_TIMESTAMPING_OPT_TSONLY;
3921 if (!skb_may_tx_timestamp(sk, tsonly))
Alexander Duyck37846ef2014-09-04 13:31:10 -04003922 return;
3923
Willem de Bruijn49ca0d82015-01-30 13:29:31 -05003924 if (tsonly)
3925 skb = alloc_skb(0, GFP_ATOMIC);
Alexander Duyck37846ef2014-09-04 13:31:10 -04003926 else
Willem de Bruijn49ca0d82015-01-30 13:29:31 -05003927 skb = skb_clone(orig_skb, GFP_ATOMIC);
Alexander Duyck37846ef2014-09-04 13:31:10 -04003928 if (!skb)
3929 return;
3930
Willem de Bruijn49ca0d82015-01-30 13:29:31 -05003931 if (tsonly) {
Willem de Bruijnfcfb53c2017-06-08 11:35:03 -04003932 skb_shinfo(skb)->tx_flags |= skb_shinfo(orig_skb)->tx_flags &
3933 SKBTX_ANY_TSTAMP;
Willem de Bruijn49ca0d82015-01-30 13:29:31 -05003934 skb_shinfo(skb)->tskey = skb_shinfo(orig_skb)->tskey;
3935 }
3936
3937 if (hwtstamps)
3938 *skb_hwtstamps(skb) = *hwtstamps;
3939 else
3940 skb->tstamp = ktime_get_real();
3941
Alexander Duyck37846ef2014-09-04 13:31:10 -04003942 __skb_complete_tx_timestamp(skb, sk, tstype);
3943}
Willem de Bruijne7fd2882014-08-04 22:11:48 -04003944EXPORT_SYMBOL_GPL(__skb_tstamp_tx);
3945
3946void skb_tstamp_tx(struct sk_buff *orig_skb,
3947 struct skb_shared_hwtstamps *hwtstamps)
3948{
3949 return __skb_tstamp_tx(orig_skb, hwtstamps, orig_skb->sk,
3950 SCM_TSTAMP_SND);
3951}
Patrick Ohlyac45f602009-02-12 05:03:37 +00003952EXPORT_SYMBOL_GPL(skb_tstamp_tx);
3953
Johannes Berg6e3e9392011-11-09 10:15:42 +01003954void skb_complete_wifi_ack(struct sk_buff *skb, bool acked)
3955{
3956 struct sock *sk = skb->sk;
3957 struct sock_exterr_skb *serr;
Eric Dumazet98fa3d22017-03-03 21:01:02 -08003958 int err = 1;
Johannes Berg6e3e9392011-11-09 10:15:42 +01003959
3960 skb->wifi_acked_valid = 1;
3961 skb->wifi_acked = acked;
3962
3963 serr = SKB_EXT_ERR(skb);
3964 memset(serr, 0, sizeof(*serr));
3965 serr->ee.ee_errno = ENOMSG;
3966 serr->ee.ee_origin = SO_EE_ORIGIN_TXSTATUS;
3967
Eric Dumazet98fa3d22017-03-03 21:01:02 -08003968 /* Take a reference to prevent skb_orphan() from freeing the socket,
3969 * but only if the socket refcount is not zero.
3970 */
3971 if (likely(atomic_inc_not_zero(&sk->sk_refcnt))) {
3972 err = sock_queue_err_skb(sk, skb);
3973 sock_put(sk);
3974 }
Johannes Berg6e3e9392011-11-09 10:15:42 +01003975 if (err)
3976 kfree_skb(skb);
3977}
3978EXPORT_SYMBOL_GPL(skb_complete_wifi_ack);
3979
Rusty Russellf35d9d82008-02-04 23:49:54 -05003980/**
3981 * skb_partial_csum_set - set up and verify partial csum values for packet
3982 * @skb: the skb to set
3983 * @start: the number of bytes after skb->data to start checksumming.
3984 * @off: the offset from start to place the checksum.
3985 *
3986 * For untrusted partially-checksummed packets, we need to make sure the values
3987 * for skb->csum_start and skb->csum_offset are valid so we don't oops.
3988 *
3989 * This function checks and sets those values and skb->ip_summed: if this
3990 * returns false you should drop the packet.
3991 */
3992bool skb_partial_csum_set(struct sk_buff *skb, u16 start, u16 off)
3993{
Herbert Xu5ff8dda2009-06-04 01:22:01 +00003994 if (unlikely(start > skb_headlen(skb)) ||
3995 unlikely((int)start + off > skb_headlen(skb) - 2)) {
Joe Perchese87cc472012-05-13 21:56:26 +00003996 net_warn_ratelimited("bad partial csum: csum=%u/%u len=%u\n",
3997 start, off, skb_headlen(skb));
Rusty Russellf35d9d82008-02-04 23:49:54 -05003998 return false;
3999 }
4000 skb->ip_summed = CHECKSUM_PARTIAL;
4001 skb->csum_start = skb_headroom(skb) + start;
4002 skb->csum_offset = off;
Jason Wange5d5dec2013-03-26 23:11:20 +00004003 skb_set_transport_header(skb, start);
Rusty Russellf35d9d82008-02-04 23:49:54 -05004004 return true;
4005}
David S. Millerb4ac530fc2009-02-10 02:09:24 -08004006EXPORT_SYMBOL_GPL(skb_partial_csum_set);
Rusty Russellf35d9d82008-02-04 23:49:54 -05004007
Paul Durranted1f50c2014-01-09 10:02:46 +00004008static int skb_maybe_pull_tail(struct sk_buff *skb, unsigned int len,
4009 unsigned int max)
4010{
4011 if (skb_headlen(skb) >= len)
4012 return 0;
4013
4014 /* If we need to pullup then pullup to the max, so we
4015 * won't need to do it again.
4016 */
4017 if (max > skb->len)
4018 max = skb->len;
4019
4020 if (__pskb_pull_tail(skb, max - skb_headlen(skb)) == NULL)
4021 return -ENOMEM;
4022
4023 if (skb_headlen(skb) < len)
4024 return -EPROTO;
4025
4026 return 0;
4027}
4028
Jan Beulichf9708b42014-03-11 13:56:05 +00004029#define MAX_TCP_HDR_LEN (15 * 4)
4030
4031static __sum16 *skb_checksum_setup_ip(struct sk_buff *skb,
4032 typeof(IPPROTO_IP) proto,
4033 unsigned int off)
4034{
4035 switch (proto) {
4036 int err;
4037
4038 case IPPROTO_TCP:
4039 err = skb_maybe_pull_tail(skb, off + sizeof(struct tcphdr),
4040 off + MAX_TCP_HDR_LEN);
4041 if (!err && !skb_partial_csum_set(skb, off,
4042 offsetof(struct tcphdr,
4043 check)))
4044 err = -EPROTO;
4045 return err ? ERR_PTR(err) : &tcp_hdr(skb)->check;
4046
4047 case IPPROTO_UDP:
4048 err = skb_maybe_pull_tail(skb, off + sizeof(struct udphdr),
4049 off + sizeof(struct udphdr));
4050 if (!err && !skb_partial_csum_set(skb, off,
4051 offsetof(struct udphdr,
4052 check)))
4053 err = -EPROTO;
4054 return err ? ERR_PTR(err) : &udp_hdr(skb)->check;
4055 }
4056
4057 return ERR_PTR(-EPROTO);
4058}
4059
Paul Durranted1f50c2014-01-09 10:02:46 +00004060/* This value should be large enough to cover a tagged ethernet header plus
4061 * maximally sized IP and TCP or UDP headers.
4062 */
4063#define MAX_IP_HDR_LEN 128
4064
Jan Beulichf9708b42014-03-11 13:56:05 +00004065static int skb_checksum_setup_ipv4(struct sk_buff *skb, bool recalculate)
Paul Durranted1f50c2014-01-09 10:02:46 +00004066{
4067 unsigned int off;
4068 bool fragment;
Jan Beulichf9708b42014-03-11 13:56:05 +00004069 __sum16 *csum;
Paul Durranted1f50c2014-01-09 10:02:46 +00004070 int err;
4071
4072 fragment = false;
4073
4074 err = skb_maybe_pull_tail(skb,
4075 sizeof(struct iphdr),
4076 MAX_IP_HDR_LEN);
4077 if (err < 0)
4078 goto out;
4079
4080 if (ip_hdr(skb)->frag_off & htons(IP_OFFSET | IP_MF))
4081 fragment = true;
4082
4083 off = ip_hdrlen(skb);
4084
4085 err = -EPROTO;
4086
4087 if (fragment)
4088 goto out;
4089
Jan Beulichf9708b42014-03-11 13:56:05 +00004090 csum = skb_checksum_setup_ip(skb, ip_hdr(skb)->protocol, off);
4091 if (IS_ERR(csum))
4092 return PTR_ERR(csum);
Paul Durranted1f50c2014-01-09 10:02:46 +00004093
Jan Beulichf9708b42014-03-11 13:56:05 +00004094 if (recalculate)
4095 *csum = ~csum_tcpudp_magic(ip_hdr(skb)->saddr,
4096 ip_hdr(skb)->daddr,
4097 skb->len - off,
4098 ip_hdr(skb)->protocol, 0);
Paul Durranted1f50c2014-01-09 10:02:46 +00004099 err = 0;
4100
4101out:
4102 return err;
4103}
4104
4105/* This value should be large enough to cover a tagged ethernet header plus
4106 * an IPv6 header, all options, and a maximal TCP or UDP header.
4107 */
4108#define MAX_IPV6_HDR_LEN 256
4109
4110#define OPT_HDR(type, skb, off) \
4111 (type *)(skb_network_header(skb) + (off))
4112
4113static int skb_checksum_setup_ipv6(struct sk_buff *skb, bool recalculate)
4114{
4115 int err;
4116 u8 nexthdr;
4117 unsigned int off;
4118 unsigned int len;
4119 bool fragment;
4120 bool done;
Jan Beulichf9708b42014-03-11 13:56:05 +00004121 __sum16 *csum;
Paul Durranted1f50c2014-01-09 10:02:46 +00004122
4123 fragment = false;
4124 done = false;
4125
4126 off = sizeof(struct ipv6hdr);
4127
4128 err = skb_maybe_pull_tail(skb, off, MAX_IPV6_HDR_LEN);
4129 if (err < 0)
4130 goto out;
4131
4132 nexthdr = ipv6_hdr(skb)->nexthdr;
4133
4134 len = sizeof(struct ipv6hdr) + ntohs(ipv6_hdr(skb)->payload_len);
4135 while (off <= len && !done) {
4136 switch (nexthdr) {
4137 case IPPROTO_DSTOPTS:
4138 case IPPROTO_HOPOPTS:
4139 case IPPROTO_ROUTING: {
4140 struct ipv6_opt_hdr *hp;
4141
4142 err = skb_maybe_pull_tail(skb,
4143 off +
4144 sizeof(struct ipv6_opt_hdr),
4145 MAX_IPV6_HDR_LEN);
4146 if (err < 0)
4147 goto out;
4148
4149 hp = OPT_HDR(struct ipv6_opt_hdr, skb, off);
4150 nexthdr = hp->nexthdr;
4151 off += ipv6_optlen(hp);
4152 break;
4153 }
4154 case IPPROTO_AH: {
4155 struct ip_auth_hdr *hp;
4156
4157 err = skb_maybe_pull_tail(skb,
4158 off +
4159 sizeof(struct ip_auth_hdr),
4160 MAX_IPV6_HDR_LEN);
4161 if (err < 0)
4162 goto out;
4163
4164 hp = OPT_HDR(struct ip_auth_hdr, skb, off);
4165 nexthdr = hp->nexthdr;
4166 off += ipv6_authlen(hp);
4167 break;
4168 }
4169 case IPPROTO_FRAGMENT: {
4170 struct frag_hdr *hp;
4171
4172 err = skb_maybe_pull_tail(skb,
4173 off +
4174 sizeof(struct frag_hdr),
4175 MAX_IPV6_HDR_LEN);
4176 if (err < 0)
4177 goto out;
4178
4179 hp = OPT_HDR(struct frag_hdr, skb, off);
4180
4181 if (hp->frag_off & htons(IP6_OFFSET | IP6_MF))
4182 fragment = true;
4183
4184 nexthdr = hp->nexthdr;
4185 off += sizeof(struct frag_hdr);
4186 break;
4187 }
4188 default:
4189 done = true;
4190 break;
4191 }
4192 }
4193
4194 err = -EPROTO;
4195
4196 if (!done || fragment)
4197 goto out;
4198
Jan Beulichf9708b42014-03-11 13:56:05 +00004199 csum = skb_checksum_setup_ip(skb, nexthdr, off);
4200 if (IS_ERR(csum))
4201 return PTR_ERR(csum);
Paul Durranted1f50c2014-01-09 10:02:46 +00004202
Jan Beulichf9708b42014-03-11 13:56:05 +00004203 if (recalculate)
4204 *csum = ~csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
4205 &ipv6_hdr(skb)->daddr,
4206 skb->len - off, nexthdr, 0);
Paul Durranted1f50c2014-01-09 10:02:46 +00004207 err = 0;
4208
4209out:
4210 return err;
4211}
4212
4213/**
4214 * skb_checksum_setup - set up partial checksum offset
4215 * @skb: the skb to set up
4216 * @recalculate: if true the pseudo-header checksum will be recalculated
4217 */
4218int skb_checksum_setup(struct sk_buff *skb, bool recalculate)
4219{
4220 int err;
4221
4222 switch (skb->protocol) {
4223 case htons(ETH_P_IP):
Jan Beulichf9708b42014-03-11 13:56:05 +00004224 err = skb_checksum_setup_ipv4(skb, recalculate);
Paul Durranted1f50c2014-01-09 10:02:46 +00004225 break;
4226
4227 case htons(ETH_P_IPV6):
4228 err = skb_checksum_setup_ipv6(skb, recalculate);
4229 break;
4230
4231 default:
4232 err = -EPROTO;
4233 break;
4234 }
4235
4236 return err;
4237}
4238EXPORT_SYMBOL(skb_checksum_setup);
4239
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004240/**
4241 * skb_checksum_maybe_trim - maybe trims the given skb
4242 * @skb: the skb to check
4243 * @transport_len: the data length beyond the network header
4244 *
4245 * Checks whether the given skb has data beyond the given transport length.
4246 * If so, returns a cloned skb trimmed to this transport length.
4247 * Otherwise returns the provided skb. Returns NULL in error cases
4248 * (e.g. transport_len exceeds skb length or out-of-memory).
4249 *
Linus Lüssinga5169932015-08-13 05:54:07 +02004250 * Caller needs to set the skb transport header and free any returned skb if it
4251 * differs from the provided skb.
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004252 */
4253static struct sk_buff *skb_checksum_maybe_trim(struct sk_buff *skb,
4254 unsigned int transport_len)
4255{
4256 struct sk_buff *skb_chk;
4257 unsigned int len = skb_transport_offset(skb) + transport_len;
4258 int ret;
4259
Linus Lüssinga5169932015-08-13 05:54:07 +02004260 if (skb->len < len)
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004261 return NULL;
Linus Lüssinga5169932015-08-13 05:54:07 +02004262 else if (skb->len == len)
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004263 return skb;
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004264
4265 skb_chk = skb_clone(skb, GFP_ATOMIC);
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004266 if (!skb_chk)
4267 return NULL;
4268
4269 ret = pskb_trim_rcsum(skb_chk, len);
4270 if (ret) {
4271 kfree_skb(skb_chk);
4272 return NULL;
4273 }
4274
4275 return skb_chk;
4276}
4277
4278/**
4279 * skb_checksum_trimmed - validate checksum of an skb
4280 * @skb: the skb to check
4281 * @transport_len: the data length beyond the network header
4282 * @skb_chkf: checksum function to use
4283 *
4284 * Applies the given checksum function skb_chkf to the provided skb.
4285 * Returns a checked and maybe trimmed skb. Returns NULL on error.
4286 *
4287 * If the skb has data beyond the given transport length, then a
4288 * trimmed & cloned skb is checked and returned.
4289 *
Linus Lüssinga5169932015-08-13 05:54:07 +02004290 * Caller needs to set the skb transport header and free any returned skb if it
4291 * differs from the provided skb.
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004292 */
4293struct sk_buff *skb_checksum_trimmed(struct sk_buff *skb,
4294 unsigned int transport_len,
4295 __sum16(*skb_chkf)(struct sk_buff *skb))
4296{
4297 struct sk_buff *skb_chk;
4298 unsigned int offset = skb_transport_offset(skb);
Linus Lüssingfcba67c2015-05-05 00:19:35 +02004299 __sum16 ret;
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004300
4301 skb_chk = skb_checksum_maybe_trim(skb, transport_len);
4302 if (!skb_chk)
Linus Lüssinga5169932015-08-13 05:54:07 +02004303 goto err;
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004304
Linus Lüssinga5169932015-08-13 05:54:07 +02004305 if (!pskb_may_pull(skb_chk, offset))
4306 goto err;
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004307
Linus Lüssing9b368812016-02-24 04:21:42 +01004308 skb_pull_rcsum(skb_chk, offset);
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004309 ret = skb_chkf(skb_chk);
Linus Lüssing9b368812016-02-24 04:21:42 +01004310 skb_push_rcsum(skb_chk, offset);
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004311
Linus Lüssinga5169932015-08-13 05:54:07 +02004312 if (ret)
4313 goto err;
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004314
4315 return skb_chk;
Linus Lüssinga5169932015-08-13 05:54:07 +02004316
4317err:
4318 if (skb_chk && skb_chk != skb)
4319 kfree_skb(skb_chk);
4320
4321 return NULL;
4322
Linus Lüssing9afd85c2015-05-02 14:01:07 +02004323}
4324EXPORT_SYMBOL(skb_checksum_trimmed);
4325
Ben Hutchings4497b072008-06-19 16:22:28 -07004326void __skb_warn_lro_forwarding(const struct sk_buff *skb)
4327{
Joe Perchese87cc472012-05-13 21:56:26 +00004328 net_warn_ratelimited("%s: received packets cannot be forwarded while LRO is enabled\n",
4329 skb->dev->name);
Ben Hutchings4497b072008-06-19 16:22:28 -07004330}
Ben Hutchings4497b072008-06-19 16:22:28 -07004331EXPORT_SYMBOL(__skb_warn_lro_forwarding);
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004332
4333void kfree_skb_partial(struct sk_buff *skb, bool head_stolen)
4334{
Eric Dumazet3d861f62012-10-22 09:03:40 +00004335 if (head_stolen) {
4336 skb_release_head_state(skb);
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004337 kmem_cache_free(skbuff_head_cache, skb);
Eric Dumazet3d861f62012-10-22 09:03:40 +00004338 } else {
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004339 __kfree_skb(skb);
Eric Dumazet3d861f62012-10-22 09:03:40 +00004340 }
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004341}
4342EXPORT_SYMBOL(kfree_skb_partial);
4343
4344/**
4345 * skb_try_coalesce - try to merge skb to prior one
4346 * @to: prior buffer
4347 * @from: buffer to add
4348 * @fragstolen: pointer to boolean
Randy Dunlapc6c4b972012-06-08 14:01:44 +00004349 * @delta_truesize: how much more was allocated than was requested
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004350 */
4351bool skb_try_coalesce(struct sk_buff *to, struct sk_buff *from,
4352 bool *fragstolen, int *delta_truesize)
4353{
4354 int i, delta, len = from->len;
4355
4356 *fragstolen = false;
4357
4358 if (skb_cloned(to))
4359 return false;
4360
4361 if (len <= skb_tailroom(to)) {
Eric Dumazete93a0432014-09-15 04:19:52 -07004362 if (len)
4363 BUG_ON(skb_copy_bits(from, 0, skb_put(to, len), len));
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004364 *delta_truesize = 0;
4365 return true;
4366 }
4367
4368 if (skb_has_frag_list(to) || skb_has_frag_list(from))
4369 return false;
4370
4371 if (skb_headlen(from) != 0) {
4372 struct page *page;
4373 unsigned int offset;
4374
4375 if (skb_shinfo(to)->nr_frags +
4376 skb_shinfo(from)->nr_frags >= MAX_SKB_FRAGS)
4377 return false;
4378
4379 if (skb_head_is_locked(from))
4380 return false;
4381
4382 delta = from->truesize - SKB_DATA_ALIGN(sizeof(struct sk_buff));
4383
4384 page = virt_to_head_page(from->head);
4385 offset = from->data - (unsigned char *)page_address(page);
4386
4387 skb_fill_page_desc(to, skb_shinfo(to)->nr_frags,
4388 page, offset, skb_headlen(from));
4389 *fragstolen = true;
4390 } else {
4391 if (skb_shinfo(to)->nr_frags +
4392 skb_shinfo(from)->nr_frags > MAX_SKB_FRAGS)
4393 return false;
4394
Weiping Panf4b549a2012-09-28 20:15:30 +00004395 delta = from->truesize - SKB_TRUESIZE(skb_end_offset(from));
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004396 }
4397
4398 WARN_ON_ONCE(delta < len);
4399
4400 memcpy(skb_shinfo(to)->frags + skb_shinfo(to)->nr_frags,
4401 skb_shinfo(from)->frags,
4402 skb_shinfo(from)->nr_frags * sizeof(skb_frag_t));
4403 skb_shinfo(to)->nr_frags += skb_shinfo(from)->nr_frags;
4404
4405 if (!skb_cloned(from))
4406 skb_shinfo(from)->nr_frags = 0;
4407
Li RongQing8ea853f2012-09-18 16:53:21 +00004408 /* if the skb is not cloned this does nothing
4409 * since we set nr_frags to 0.
4410 */
Eric Dumazetbad43ca2012-05-19 03:02:02 +00004411 for (i = 0; i < skb_shinfo(from)->nr_frags; i++)
4412 skb_frag_ref(from, i);
4413
4414 to->truesize += delta;
4415 to->len += len;
4416 to->data_len += len;
4417
4418 *delta_truesize = delta;
4419 return true;
4420}
4421EXPORT_SYMBOL(skb_try_coalesce);
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004422
4423/**
Nicolas Dichtel8b27f272013-09-02 15:34:56 +02004424 * skb_scrub_packet - scrub an skb
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004425 *
4426 * @skb: buffer to clean
Nicolas Dichtel8b27f272013-09-02 15:34:56 +02004427 * @xnet: packet is crossing netns
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004428 *
Nicolas Dichtel8b27f272013-09-02 15:34:56 +02004429 * skb_scrub_packet can be used after encapsulating or decapsulting a packet
4430 * into/from a tunnel. Some information have to be cleared during these
4431 * operations.
4432 * skb_scrub_packet can also be used to clean a skb before injecting it in
4433 * another namespace (@xnet == true). We have to clear all information in the
4434 * skb that could impact namespace isolation.
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004435 */
Nicolas Dichtel8b27f272013-09-02 15:34:56 +02004436void skb_scrub_packet(struct sk_buff *skb, bool xnet)
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004437{
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004438 skb->tstamp.tv64 = 0;
4439 skb->pkt_type = PACKET_HOST;
4440 skb->skb_iif = 0;
WANG Cong60ff7462014-05-04 16:39:18 -07004441 skb->ignore_df = 0;
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004442 skb_dst_drop(skb);
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004443 secpath_reset(skb);
4444 nf_reset(skb);
4445 nf_reset_trace(skb);
Herbert Xu213dd742015-04-16 09:03:27 +08004446
Petr Machatad39ebd12018-11-20 11:39:56 +00004447#ifdef CONFIG_NET_SWITCHDEV
4448 skb->offload_fwd_mark = 0;
4449#endif
4450
Herbert Xu213dd742015-04-16 09:03:27 +08004451 if (!xnet)
4452 return;
4453
Ye Yinafd9fa62017-10-26 16:57:05 +08004454 ipvs_reset(skb);
Herbert Xu213dd742015-04-16 09:03:27 +08004455 skb_orphan(skb);
4456 skb->mark = 0;
Nicolas Dichtel621e84d2013-06-26 16:11:27 +02004457}
4458EXPORT_SYMBOL_GPL(skb_scrub_packet);
Florian Westphalde960aa2014-01-26 10:58:16 +01004459
4460/**
4461 * skb_gso_transport_seglen - Return length of individual segments of a gso packet
4462 *
4463 * @skb: GSO skb
4464 *
4465 * skb_gso_transport_seglen is used to determine the real size of the
4466 * individual segments, including Layer4 headers (TCP/UDP).
4467 *
4468 * The MAC/L2 or network (IP, IPv6) headers are not accounted for.
4469 */
4470unsigned int skb_gso_transport_seglen(const struct sk_buff *skb)
4471{
4472 const struct skb_shared_info *shinfo = skb_shinfo(skb);
Florian Westphalf993bc22014-10-20 13:49:18 +02004473 unsigned int thlen = 0;
Florian Westphalde960aa2014-01-26 10:58:16 +01004474
Florian Westphalf993bc22014-10-20 13:49:18 +02004475 if (skb->encapsulation) {
4476 thlen = skb_inner_transport_header(skb) -
4477 skb_transport_header(skb);
Florian Westphal6d39d582014-04-09 10:28:50 +02004478
Florian Westphalf993bc22014-10-20 13:49:18 +02004479 if (likely(shinfo->gso_type & (SKB_GSO_TCPV4 | SKB_GSO_TCPV6)))
4480 thlen += inner_tcp_hdrlen(skb);
4481 } else if (likely(shinfo->gso_type & (SKB_GSO_TCPV4 | SKB_GSO_TCPV6))) {
4482 thlen = tcp_hdrlen(skb);
Marcelo Ricardo Leitner90017ac2016-06-02 15:05:43 -03004483 } else if (unlikely(shinfo->gso_type & SKB_GSO_SCTP)) {
4484 thlen = sizeof(struct sctphdr);
Florian Westphalf993bc22014-10-20 13:49:18 +02004485 }
Florian Westphal6d39d582014-04-09 10:28:50 +02004486 /* UFO sets gso_size to the size of the fragmentation
4487 * payload, i.e. the size of the L4 (UDP) header is already
4488 * accounted for.
4489 */
Florian Westphalf993bc22014-10-20 13:49:18 +02004490 return thlen + shinfo->gso_size;
Florian Westphalde960aa2014-01-26 10:58:16 +01004491}
4492EXPORT_SYMBOL_GPL(skb_gso_transport_seglen);
Vlad Yasevich0d5501c2014-08-08 14:42:13 -04004493
Marcelo Ricardo Leitnerae7ef812016-06-02 15:05:41 -03004494/**
Daniel Axtens84d8c3a2019-02-14 11:31:17 +01004495 * skb_gso_size_check - check the skb size, considering GSO_BY_FRAGS
4496 *
4497 * There are a couple of instances where we have a GSO skb, and we
4498 * want to determine what size it would be after it is segmented.
4499 *
4500 * We might want to check:
4501 * - L3+L4+payload size (e.g. IP forwarding)
4502 * - L2+L3+L4+payload size (e.g. sanity check before passing to driver)
4503 *
4504 * This is a helper to do that correctly considering GSO_BY_FRAGS.
4505 *
4506 * @seg_len: The segmented length (from skb_gso_*_seglen). In the
4507 * GSO_BY_FRAGS case this will be [header sizes + GSO_BY_FRAGS].
4508 *
4509 * @max_len: The maximum permissible length.
4510 *
4511 * Returns true if the segmented length <= max length.
4512 */
4513static inline bool skb_gso_size_check(const struct sk_buff *skb,
4514 unsigned int seg_len,
4515 unsigned int max_len) {
4516 const struct skb_shared_info *shinfo = skb_shinfo(skb);
4517 const struct sk_buff *iter;
4518
4519 if (shinfo->gso_size != GSO_BY_FRAGS)
4520 return seg_len <= max_len;
4521
4522 /* Undo this so we can re-use header sizes */
4523 seg_len -= GSO_BY_FRAGS;
4524
4525 skb_walk_frags(skb, iter) {
4526 if (seg_len + skb_headlen(iter) > max_len)
4527 return false;
4528 }
4529
4530 return true;
4531}
4532
4533/**
Marcelo Ricardo Leitnerae7ef812016-06-02 15:05:41 -03004534 * skb_gso_validate_mtu - Return in case such skb fits a given MTU
4535 *
4536 * @skb: GSO skb
David S. Miller76f21b92016-06-03 22:56:28 -07004537 * @mtu: MTU to validate against
Marcelo Ricardo Leitnerae7ef812016-06-02 15:05:41 -03004538 *
4539 * skb_gso_validate_mtu validates if a given skb will fit a wanted MTU
4540 * once split.
4541 */
4542bool skb_gso_validate_mtu(const struct sk_buff *skb, unsigned int mtu)
4543{
Daniel Axtens84d8c3a2019-02-14 11:31:17 +01004544 return skb_gso_size_check(skb, skb_gso_network_seglen(skb), mtu);
Marcelo Ricardo Leitnerae7ef812016-06-02 15:05:41 -03004545}
4546EXPORT_SYMBOL_GPL(skb_gso_validate_mtu);
4547
Daniel Axtens84d8c3a2019-02-14 11:31:17 +01004548/**
4549 * skb_gso_validate_mac_len - Will a split GSO skb fit in a given length?
4550 *
4551 * @skb: GSO skb
4552 * @len: length to validate against
4553 *
4554 * skb_gso_validate_mac_len validates if a given skb will fit a wanted
4555 * length once split, including L2, L3 and L4 headers and the payload.
4556 */
4557bool skb_gso_validate_mac_len(const struct sk_buff *skb, unsigned int len)
4558{
4559 return skb_gso_size_check(skb, skb_gso_mac_seglen(skb), len);
4560}
4561EXPORT_SYMBOL_GPL(skb_gso_validate_mac_len);
4562
Vlad Yasevich0d5501c2014-08-08 14:42:13 -04004563static struct sk_buff *skb_reorder_vlan_header(struct sk_buff *skb)
4564{
Toshiaki Makita58df2892018-03-13 14:51:27 +09004565 int mac_len;
4566
Vlad Yasevich0d5501c2014-08-08 14:42:13 -04004567 if (skb_cow(skb, skb_headroom(skb)) < 0) {
4568 kfree_skb(skb);
4569 return NULL;
4570 }
4571
Toshiaki Makita58df2892018-03-13 14:51:27 +09004572 mac_len = skb->data - skb_mac_header(skb);
Toshiaki Makitab389e042018-03-29 19:05:29 +09004573 if (likely(mac_len > VLAN_HLEN + ETH_TLEN)) {
4574 memmove(skb_mac_header(skb) + VLAN_HLEN, skb_mac_header(skb),
4575 mac_len - VLAN_HLEN - ETH_TLEN);
4576 }
Vlad Yasevich0d5501c2014-08-08 14:42:13 -04004577 skb->mac_header += VLAN_HLEN;
4578 return skb;
4579}
4580
4581struct sk_buff *skb_vlan_untag(struct sk_buff *skb)
4582{
4583 struct vlan_hdr *vhdr;
4584 u16 vlan_tci;
4585
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01004586 if (unlikely(skb_vlan_tag_present(skb))) {
Vlad Yasevich0d5501c2014-08-08 14:42:13 -04004587 /* vlan_tci is already set-up so leave this for another time */
4588 return skb;
4589 }
4590
4591 skb = skb_share_check(skb, GFP_ATOMIC);
4592 if (unlikely(!skb))
4593 goto err_free;
4594
4595 if (unlikely(!pskb_may_pull(skb, VLAN_HLEN)))
4596 goto err_free;
4597
4598 vhdr = (struct vlan_hdr *)skb->data;
4599 vlan_tci = ntohs(vhdr->h_vlan_TCI);
4600 __vlan_hwaccel_put_tag(skb, skb->protocol, vlan_tci);
4601
4602 skb_pull_rcsum(skb, VLAN_HLEN);
4603 vlan_set_encap_proto(skb, vhdr);
4604
4605 skb = skb_reorder_vlan_header(skb);
4606 if (unlikely(!skb))
4607 goto err_free;
4608
4609 skb_reset_network_header(skb);
4610 skb_reset_transport_header(skb);
4611 skb_reset_mac_len(skb);
4612
4613 return skb;
4614
4615err_free:
4616 kfree_skb(skb);
4617 return NULL;
4618}
4619EXPORT_SYMBOL(skb_vlan_untag);
Eric Dumazet2e4e4412014-09-17 04:49:49 -07004620
Jiri Pirkoe2195122014-11-19 14:05:01 +01004621int skb_ensure_writable(struct sk_buff *skb, int write_len)
4622{
4623 if (!pskb_may_pull(skb, write_len))
4624 return -ENOMEM;
4625
4626 if (!skb_cloned(skb) || skb_clone_writable(skb, write_len))
4627 return 0;
4628
4629 return pskb_expand_head(skb, 0, 0, GFP_ATOMIC);
4630}
4631EXPORT_SYMBOL(skb_ensure_writable);
4632
Shmulik Ladkanibfca4c52016-09-19 19:11:09 +03004633/* remove VLAN header from packet and update csum accordingly.
4634 * expects a non skb_vlan_tag_present skb with a vlan tag payload
4635 */
4636int __skb_vlan_pop(struct sk_buff *skb, u16 *vlan_tci)
Jiri Pirko93515d52014-11-19 14:05:02 +01004637{
4638 struct vlan_hdr *vhdr;
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004639 int offset = skb->data - skb_mac_header(skb);
Jiri Pirko93515d52014-11-19 14:05:02 +01004640 int err;
4641
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004642 if (WARN_ONCE(offset,
4643 "__skb_vlan_pop got skb with skb->data not at mac header (offset %d)\n",
4644 offset)) {
4645 return -EINVAL;
4646 }
4647
Jiri Pirko93515d52014-11-19 14:05:02 +01004648 err = skb_ensure_writable(skb, VLAN_ETH_HLEN);
4649 if (unlikely(err))
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004650 return err;
Jiri Pirko93515d52014-11-19 14:05:02 +01004651
4652 skb_postpull_rcsum(skb, skb->data + (2 * ETH_ALEN), VLAN_HLEN);
4653
4654 vhdr = (struct vlan_hdr *)(skb->data + ETH_HLEN);
4655 *vlan_tci = ntohs(vhdr->h_vlan_TCI);
4656
4657 memmove(skb->data + VLAN_HLEN, skb->data, 2 * ETH_ALEN);
4658 __skb_pull(skb, VLAN_HLEN);
4659
4660 vlan_set_encap_proto(skb, vhdr);
4661 skb->mac_header += VLAN_HLEN;
4662
4663 if (skb_network_offset(skb) < ETH_HLEN)
4664 skb_set_network_header(skb, ETH_HLEN);
4665
4666 skb_reset_mac_len(skb);
Jiri Pirko93515d52014-11-19 14:05:02 +01004667
4668 return err;
4669}
Shmulik Ladkanibfca4c52016-09-19 19:11:09 +03004670EXPORT_SYMBOL(__skb_vlan_pop);
Jiri Pirko93515d52014-11-19 14:05:02 +01004671
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004672/* Pop a vlan tag either from hwaccel or from payload.
4673 * Expects skb->data at mac header.
4674 */
Jiri Pirko93515d52014-11-19 14:05:02 +01004675int skb_vlan_pop(struct sk_buff *skb)
4676{
4677 u16 vlan_tci;
4678 __be16 vlan_proto;
4679 int err;
4680
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01004681 if (likely(skb_vlan_tag_present(skb))) {
Jiri Pirko93515d52014-11-19 14:05:02 +01004682 skb->vlan_tci = 0;
4683 } else {
Shmulik Ladkaniecf4ee42016-09-20 12:48:37 +03004684 if (unlikely(!eth_type_vlan(skb->protocol)))
Jiri Pirko93515d52014-11-19 14:05:02 +01004685 return 0;
4686
4687 err = __skb_vlan_pop(skb, &vlan_tci);
4688 if (err)
4689 return err;
4690 }
4691 /* move next vlan tag to hw accel tag */
Shmulik Ladkaniecf4ee42016-09-20 12:48:37 +03004692 if (likely(!eth_type_vlan(skb->protocol)))
Jiri Pirko93515d52014-11-19 14:05:02 +01004693 return 0;
4694
4695 vlan_proto = skb->protocol;
4696 err = __skb_vlan_pop(skb, &vlan_tci);
4697 if (unlikely(err))
4698 return err;
4699
4700 __vlan_hwaccel_put_tag(skb, vlan_proto, vlan_tci);
4701 return 0;
4702}
4703EXPORT_SYMBOL(skb_vlan_pop);
4704
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004705/* Push a vlan tag either into hwaccel or into payload (if hwaccel tag present).
4706 * Expects skb->data at mac header.
4707 */
Jiri Pirko93515d52014-11-19 14:05:02 +01004708int skb_vlan_push(struct sk_buff *skb, __be16 vlan_proto, u16 vlan_tci)
4709{
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01004710 if (skb_vlan_tag_present(skb)) {
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004711 int offset = skb->data - skb_mac_header(skb);
Jiri Pirko93515d52014-11-19 14:05:02 +01004712 int err;
4713
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004714 if (WARN_ONCE(offset,
4715 "skb_vlan_push got skb with skb->data not at mac header (offset %d)\n",
4716 offset)) {
4717 return -EINVAL;
4718 }
4719
Jiri Pirko93515d52014-11-19 14:05:02 +01004720 err = __vlan_insert_tag(skb, skb->vlan_proto,
Jiri Pirkodf8a39d2015-01-13 17:13:44 +01004721 skb_vlan_tag_get(skb));
Shmulik Ladkanib6a79202016-09-29 12:10:41 +03004722 if (err)
Jiri Pirko93515d52014-11-19 14:05:02 +01004723 return err;
Daniel Borkmann9241e2d2016-04-16 02:27:58 +02004724
Jiri Pirko93515d52014-11-19 14:05:02 +01004725 skb->protocol = skb->vlan_proto;
4726 skb->mac_len += VLAN_HLEN;
Jiri Pirko93515d52014-11-19 14:05:02 +01004727
Daniel Borkmann6b83d282016-02-20 00:29:30 +01004728 skb_postpush_rcsum(skb, skb->data + (2 * ETH_ALEN), VLAN_HLEN);
Jiri Pirko93515d52014-11-19 14:05:02 +01004729 }
4730 __vlan_hwaccel_put_tag(skb, vlan_proto, vlan_tci);
4731 return 0;
4732}
4733EXPORT_SYMBOL(skb_vlan_push);
4734
Eric Dumazet2e4e4412014-09-17 04:49:49 -07004735/**
4736 * alloc_skb_with_frags - allocate skb with page frags
4737 *
Masanari Iidade3f0d02014-10-09 12:58:08 +09004738 * @header_len: size of linear part
4739 * @data_len: needed length in frags
4740 * @max_page_order: max page order desired.
4741 * @errcode: pointer to error code if any
4742 * @gfp_mask: allocation mask
Eric Dumazet2e4e4412014-09-17 04:49:49 -07004743 *
4744 * This can be used to allocate a paged skb, given a maximal order for frags.
4745 */
4746struct sk_buff *alloc_skb_with_frags(unsigned long header_len,
4747 unsigned long data_len,
4748 int max_page_order,
4749 int *errcode,
4750 gfp_t gfp_mask)
4751{
4752 int npages = (data_len + (PAGE_SIZE - 1)) >> PAGE_SHIFT;
4753 unsigned long chunk;
4754 struct sk_buff *skb;
4755 struct page *page;
4756 gfp_t gfp_head;
4757 int i;
4758
4759 *errcode = -EMSGSIZE;
4760 /* Note this test could be relaxed, if we succeed to allocate
4761 * high order pages...
4762 */
4763 if (npages > MAX_SKB_FRAGS)
4764 return NULL;
4765
4766 gfp_head = gfp_mask;
Mel Gormand0164ad2015-11-06 16:28:21 -08004767 if (gfp_head & __GFP_DIRECT_RECLAIM)
Eric Dumazet2e4e4412014-09-17 04:49:49 -07004768 gfp_head |= __GFP_REPEAT;
4769
4770 *errcode = -ENOBUFS;
4771 skb = alloc_skb(header_len, gfp_head);
4772 if (!skb)
4773 return NULL;
4774
4775 skb->truesize += npages << PAGE_SHIFT;
4776
4777 for (i = 0; npages > 0; i++) {
4778 int order = max_page_order;
4779
4780 while (order) {
4781 if (npages >= 1 << order) {
Mel Gormand0164ad2015-11-06 16:28:21 -08004782 page = alloc_pages((gfp_mask & ~__GFP_DIRECT_RECLAIM) |
Eric Dumazet2e4e4412014-09-17 04:49:49 -07004783 __GFP_COMP |
4784 __GFP_NOWARN |
4785 __GFP_NORETRY,
4786 order);
4787 if (page)
4788 goto fill_page;
4789 /* Do not retry other high order allocations */
4790 order = 1;
4791 max_page_order = 0;
4792 }
4793 order--;
4794 }
4795 page = alloc_page(gfp_mask);
4796 if (!page)
4797 goto failure;
4798fill_page:
4799 chunk = min_t(unsigned long, data_len,
4800 PAGE_SIZE << order);
4801 skb_fill_page_desc(skb, i, page, 0, chunk);
4802 data_len -= chunk;
4803 npages -= 1 << order;
4804 }
4805 return skb;
4806
4807failure:
4808 kfree_skb(skb);
4809 return NULL;
4810}
4811EXPORT_SYMBOL(alloc_skb_with_frags);
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004812
4813/* carve out the first off bytes from skb when off < headlen */
4814static int pskb_carve_inside_header(struct sk_buff *skb, const u32 off,
4815 const int headlen, gfp_t gfp_mask)
4816{
4817 int i;
4818 int size = skb_end_offset(skb);
4819 int new_hlen = headlen - off;
4820 u8 *data;
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004821
4822 size = SKB_DATA_ALIGN(size);
4823
4824 if (skb_pfmemalloc(skb))
4825 gfp_mask |= __GFP_MEMALLOC;
4826 data = kmalloc_reserve(size +
4827 SKB_DATA_ALIGN(sizeof(struct skb_shared_info)),
4828 gfp_mask, NUMA_NO_NODE, NULL);
4829 if (!data)
4830 return -ENOMEM;
4831
4832 size = SKB_WITH_OVERHEAD(ksize(data));
4833
4834 /* Copy real data, and all frags */
4835 skb_copy_from_linear_data_offset(skb, off, data, new_hlen);
4836 skb->len -= off;
4837
4838 memcpy((struct skb_shared_info *)(data + size),
4839 skb_shinfo(skb),
4840 offsetof(struct skb_shared_info,
4841 frags[skb_shinfo(skb)->nr_frags]));
4842 if (skb_cloned(skb)) {
4843 /* drop the old head gracefully */
4844 if (skb_orphan_frags(skb, gfp_mask)) {
4845 kfree(data);
4846 return -ENOMEM;
4847 }
4848 for (i = 0; i < skb_shinfo(skb)->nr_frags; i++)
4849 skb_frag_ref(skb, i);
4850 if (skb_has_frag_list(skb))
4851 skb_clone_fraglist(skb);
4852 skb_release_data(skb);
4853 } else {
4854 /* we can reuse existing recount- all we did was
4855 * relocate values
4856 */
4857 skb_free_head(skb);
4858 }
4859
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004860 skb->head = data;
4861 skb->data = data;
4862 skb->head_frag = 0;
4863#ifdef NET_SKBUFF_DATA_USES_OFFSET
4864 skb->end = size;
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004865#else
4866 skb->end = skb->head + size;
4867#endif
4868 skb_set_tail_pointer(skb, skb_headlen(skb));
4869 skb_headers_offset_update(skb, 0);
4870 skb->cloned = 0;
4871 skb->hdr_len = 0;
4872 skb->nohdr = 0;
4873 atomic_set(&skb_shinfo(skb)->dataref, 1);
4874
4875 return 0;
4876}
4877
4878static int pskb_carve(struct sk_buff *skb, const u32 off, gfp_t gfp);
4879
4880/* carve out the first eat bytes from skb's frag_list. May recurse into
4881 * pskb_carve()
4882 */
4883static int pskb_carve_frag_list(struct sk_buff *skb,
4884 struct skb_shared_info *shinfo, int eat,
4885 gfp_t gfp_mask)
4886{
4887 struct sk_buff *list = shinfo->frag_list;
4888 struct sk_buff *clone = NULL;
4889 struct sk_buff *insp = NULL;
4890
4891 do {
4892 if (!list) {
4893 pr_err("Not enough bytes to eat. Want %d\n", eat);
4894 return -EFAULT;
4895 }
4896 if (list->len <= eat) {
4897 /* Eaten as whole. */
4898 eat -= list->len;
4899 list = list->next;
4900 insp = list;
4901 } else {
4902 /* Eaten partially. */
4903 if (skb_shared(list)) {
4904 clone = skb_clone(list, gfp_mask);
4905 if (!clone)
4906 return -ENOMEM;
4907 insp = list->next;
4908 list = clone;
4909 } else {
4910 /* This may be pulled without problems. */
4911 insp = list;
4912 }
4913 if (pskb_carve(list, eat, gfp_mask) < 0) {
4914 kfree_skb(clone);
4915 return -ENOMEM;
4916 }
4917 break;
4918 }
4919 } while (eat);
4920
4921 /* Free pulled out fragments. */
4922 while ((list = shinfo->frag_list) != insp) {
4923 shinfo->frag_list = list->next;
4924 kfree_skb(list);
4925 }
4926 /* And insert new clone at head. */
4927 if (clone) {
4928 clone->next = list;
4929 shinfo->frag_list = clone;
4930 }
4931 return 0;
4932}
4933
4934/* carve off first len bytes from skb. Split line (off) is in the
4935 * non-linear part of skb
4936 */
4937static int pskb_carve_inside_nonlinear(struct sk_buff *skb, const u32 off,
4938 int pos, gfp_t gfp_mask)
4939{
4940 int i, k = 0;
4941 int size = skb_end_offset(skb);
4942 u8 *data;
4943 const int nfrags = skb_shinfo(skb)->nr_frags;
4944 struct skb_shared_info *shinfo;
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004945
4946 size = SKB_DATA_ALIGN(size);
4947
4948 if (skb_pfmemalloc(skb))
4949 gfp_mask |= __GFP_MEMALLOC;
4950 data = kmalloc_reserve(size +
4951 SKB_DATA_ALIGN(sizeof(struct skb_shared_info)),
4952 gfp_mask, NUMA_NO_NODE, NULL);
4953 if (!data)
4954 return -ENOMEM;
4955
4956 size = SKB_WITH_OVERHEAD(ksize(data));
4957
4958 memcpy((struct skb_shared_info *)(data + size),
4959 skb_shinfo(skb), offsetof(struct skb_shared_info,
4960 frags[skb_shinfo(skb)->nr_frags]));
4961 if (skb_orphan_frags(skb, gfp_mask)) {
4962 kfree(data);
4963 return -ENOMEM;
4964 }
4965 shinfo = (struct skb_shared_info *)(data + size);
4966 for (i = 0; i < nfrags; i++) {
4967 int fsize = skb_frag_size(&skb_shinfo(skb)->frags[i]);
4968
4969 if (pos + fsize > off) {
4970 shinfo->frags[k] = skb_shinfo(skb)->frags[i];
4971
4972 if (pos < off) {
4973 /* Split frag.
4974 * We have two variants in this case:
4975 * 1. Move all the frag to the second
4976 * part, if it is possible. F.e.
4977 * this approach is mandatory for TUX,
4978 * where splitting is expensive.
4979 * 2. Split is accurately. We make this.
4980 */
4981 shinfo->frags[0].page_offset += off - pos;
4982 skb_frag_size_sub(&shinfo->frags[0], off - pos);
4983 }
4984 skb_frag_ref(skb, i);
4985 k++;
4986 }
4987 pos += fsize;
4988 }
4989 shinfo->nr_frags = k;
4990 if (skb_has_frag_list(skb))
4991 skb_clone_fraglist(skb);
4992
4993 if (k == 0) {
4994 /* split line is in frag list */
4995 pskb_carve_frag_list(skb, shinfo, off - pos, gfp_mask);
4996 }
4997 skb_release_data(skb);
4998
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07004999 skb->head = data;
5000 skb->head_frag = 0;
5001 skb->data = data;
5002#ifdef NET_SKBUFF_DATA_USES_OFFSET
5003 skb->end = size;
Sowmini Varadhan6fa01cc2016-04-22 18:36:35 -07005004#else
5005 skb->end = skb->head + size;
5006#endif
5007 skb_reset_tail_pointer(skb);
5008 skb_headers_offset_update(skb, 0);
5009 skb->cloned = 0;
5010 skb->hdr_len = 0;
5011 skb->nohdr = 0;
5012 skb->len -= off;
5013 skb->data_len = skb->len;
5014 atomic_set(&skb_shinfo(skb)->dataref, 1);
5015 return 0;
5016}
5017
5018/* remove len bytes from the beginning of the skb */
5019static int pskb_carve(struct sk_buff *skb, const u32 len, gfp_t gfp)
5020{
5021 int headlen = skb_headlen(skb);
5022
5023 if (len < headlen)
5024 return pskb_carve_inside_header(skb, len, headlen, gfp);
5025 else
5026 return pskb_carve_inside_nonlinear(skb, len, headlen, gfp);
5027}
5028
5029/* Extract to_copy bytes starting at off from skb, and return this in
5030 * a new skb
5031 */
5032struct sk_buff *pskb_extract(struct sk_buff *skb, int off,
5033 int to_copy, gfp_t gfp)
5034{
5035 struct sk_buff *clone = skb_clone(skb, gfp);
5036
5037 if (!clone)
5038 return NULL;
5039
5040 if (pskb_carve(clone, off, gfp) < 0 ||
5041 pskb_trim(clone, to_copy)) {
5042 kfree_skb(clone);
5043 return NULL;
5044 }
5045 return clone;
5046}
5047EXPORT_SYMBOL(pskb_extract);