3434 lines · 84.9 KB · C
RawDownload
/* *	Routines having to do with the 'struct sk_buff' memory handlers. * *	Authors:	Alan Cox <alan@lxorguk.ukuu.org.uk> *			Florian La Roche <rzsfl@rz.uni-sb.de> * *	Fixes: *		Alan Cox	:	Fixed the worst of the load *					balancer bugs. *		Dave Platt	:	Interrupt stacking fix. *	Richard Kooijman	:	Timestamp fixes. *		Alan Cox	:	Changed buffer format. *		Alan Cox	:	destructor hook for AF_UNIX etc. *		Linus Torvalds	:	Better skb_clone. *		Alan Cox	:	Added skb_copy. *		Alan Cox	:	Added all the changed routines Linus *					only put in the headers *		Ray VanTassle	:	Fixed --skb->lock in free *		Alan Cox	:	skb_copy copy arp field *		Andi Kleen	:	slabified it. *		Robert Olsson	:	Removed skb_head_pool * *	NOTE: *		The __skb_ routines should be called with interrupts *	disabled, or you better be *real* sure that the operation is atomic *	with respect to whatever list is being frobbed (e.g. via lock_sock() *	or via disabling bottom half handlers, etc). * *	This program is free software; you can redistribute it and/or *	modify it under the terms of the GNU General Public License *	as published by the Free Software Foundation; either version *	2 of the License, or (at your option) any later version. */ /* *	The functions in this file will not compile correctly with gcc 2.4.x */ #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt #include <linux/module.h>#include <linux/types.h>#include <linux/kernel.h>#include <linux/kmemcheck.h>#include <linux/mm.h>#include <linux/interrupt.h>#include <linux/in.h>#include <linux/inet.h>#include <linux/slab.h>#include <linux/netdevice.h>#ifdef CONFIG_NET_CLS_ACT#include <net/pkt_sched.h>#endif#include <linux/string.h>#include <linux/skbuff.h>#include <linux/splice.h>#include <linux/cache.h>#include <linux/rtnetlink.h>#include <linux/init.h>#include <linux/scatterlist.h>#include <linux/errqueue.h>#include <linux/prefetch.h> #include <net/protocol.h>#include <net/dst.h>#include <net/sock.h>#include <net/checksum.h>#include <net/xfrm.h> #include <asm/uaccess.h>#include <trace/events/skb.h>#include <linux/highmem.h> struct kmem_cache *skbuff_head_cache __read_mostly;static struct kmem_cache *skbuff_fclone_cache __read_mostly; static void sock_pipe_buf_release(struct pipe_inode_info *pipe,				  struct pipe_buffer *buf){	put_page(buf->page);} static void sock_pipe_buf_get(struct pipe_inode_info *pipe,				struct pipe_buffer *buf){	get_page(buf->page);} static int sock_pipe_buf_steal(struct pipe_inode_info *pipe,			       struct pipe_buffer *buf){	return 1;}  /* Pipe buffer operations for a socket. */static const struct pipe_buf_operations sock_pipe_buf_ops = {	.can_merge = 0,	.map = generic_pipe_buf_map,	.unmap = generic_pipe_buf_unmap,	.confirm = generic_pipe_buf_confirm,	.release = sock_pipe_buf_release,	.steal = sock_pipe_buf_steal,	.get = sock_pipe_buf_get,}; /* *	Keep out-of-line to prevent kernel bloat. *	__builtin_return_address is not used because it is not always *	reliable. */ /** *	skb_over_panic	- 	private function *	@skb: buffer *	@sz: size *	@here: address * *	Out of line support code for skb_put(). Not user callable. */static void skb_over_panic(struct sk_buff *skb, int sz, void *here){	pr_emerg("%s: text:%p len:%d put:%d head:%p data:%p tail:%#lx end:%#lx dev:%s\n",		 __func__, here, skb->len, sz, skb->head, skb->data,		 (unsigned long)skb->tail, (unsigned long)skb->end,		 skb->dev ? skb->dev->name : "<NULL>");	BUG();} /** *	skb_under_panic	- 	private function *	@skb: buffer *	@sz: size *	@here: address * *	Out of line support code for skb_push(). Not user callable. */ static void skb_under_panic(struct sk_buff *skb, int sz, void *here){	pr_emerg("%s: text:%p len:%d put:%d head:%p data:%p tail:%#lx end:%#lx dev:%s\n",		 __func__, here, skb->len, sz, skb->head, skb->data,		 (unsigned long)skb->tail, (unsigned long)skb->end,		 skb->dev ? skb->dev->name : "<NULL>");	BUG();} /* 	Allocate a new skbuff. We do this ourselves so we can fill in a few *	'private' fields and also do memory statistics to find all the *	[BEEP] leaks. * */ /** *	__alloc_skb	-	allocate a network buffer *	@size: size to allocate *	@gfp_mask: allocation mask *	@fclone: allocate from fclone cache instead of head cache *		and allocate a cloned (child) skb *	@node: numa node to allocate memory on * *	Allocate a new &sk_buff. The returned buffer has no headroom and a *	tail room of size bytes. The object has a reference count of one. *	The return is the buffer. On a failure the return is %NULL. * *	Buffers may only be allocated from interrupts using a @gfp_mask of *	%GFP_ATOMIC. */struct sk_buff *__alloc_skb(unsigned int size, gfp_t gfp_mask,			    int fclone, int node){	struct kmem_cache *cache;	struct skb_shared_info *shinfo;	struct sk_buff *skb;	u8 *data; 	cache = fclone ? skbuff_fclone_cache : skbuff_head_cache; 	/* Get the HEAD */	skb = kmem_cache_alloc_node(cache, gfp_mask & ~__GFP_DMA, node);	if (!skb)		goto out;	prefetchw(skb); 	/* We do our best to align skb_shared_info on a separate cache	 * line. It usually works because kmalloc(X > SMP_CACHE_BYTES) gives	 * aligned memory blocks, unless SLUB/SLAB debug is enabled.	 * Both skb->head and skb_shared_info are cache line aligned.	 */	size = SKB_DATA_ALIGN(size);	size += SKB_DATA_ALIGN(sizeof(struct skb_shared_info));	data = kmalloc_node_track_caller(size, gfp_mask, node);	if (!data)		goto nodata;	/* kmalloc(size) might give us more room than requested.	 * Put skb_shared_info exactly at the end of allocated zone,	 * to allow max possible filling before reallocation.	 */	size = SKB_WITH_OVERHEAD(ksize(data));	prefetchw(data + size); 	/*	 * Only clear those fields we need to clear, not those that we will	 * actually initialise below. Hence, don't put any more fields after	 * the tail pointer in struct sk_buff!	 */	memset(skb, 0, offsetof(struct sk_buff, tail));	/* Account for allocated memory : skb + skb->head */	skb->truesize = SKB_TRUESIZE(size);	atomic_set(&skb->users, 1);	skb->head = data;	skb->data = data;	skb_reset_tail_pointer(skb);	skb->end = skb->tail + size;#ifdef NET_SKBUFF_DATA_USES_OFFSET	skb->mac_header = ~0U;#endif 	/* make sure we initialize shinfo sequentially */	shinfo = skb_shinfo(skb);	memset(shinfo, 0, offsetof(struct skb_shared_info, dataref));	atomic_set(&shinfo->dataref, 1);	kmemcheck_annotate_variable(shinfo->destructor_arg); 	if (fclone) {		struct sk_buff *child = skb + 1;		atomic_t *fclone_ref = (atomic_t *) (child + 1); 		kmemcheck_annotate_bitfield(child, flags1);		kmemcheck_annotate_bitfield(child, flags2);		skb->fclone = SKB_FCLONE_ORIG;		atomic_set(fclone_ref, 1); 		child->fclone = SKB_FCLONE_UNAVAILABLE;	}out:	return skb;nodata:	kmem_cache_free(cache, skb);	skb = NULL;	goto out;}EXPORT_SYMBOL(__alloc_skb); /** * build_skb - build a network buffer * @data: data buffer provided by caller * @frag_size: size of fragment, or 0 if head was kmalloced * * Allocate a new &sk_buff. Caller provides space holding head and * skb_shared_info. @data must have been allocated by kmalloc() * The return is the new skb buffer. * On a failure the return is %NULL, and @data is not freed. * Notes : *  Before IO, driver allocates only data buffer where NIC put incoming frame *  Driver should add room at head (NET_SKB_PAD) and *  MUST add room at tail (SKB_DATA_ALIGN(skb_shared_info)) *  After IO, driver calls build_skb(), to allocate sk_buff and populate it *  before giving packet to stack. *  RX rings only contains data buffers, not full skbs. */struct sk_buff *build_skb(void *data, unsigned int frag_size){	struct skb_shared_info *shinfo;	struct sk_buff *skb;	unsigned int size = frag_size ? : ksize(data); 	skb = kmem_cache_alloc(skbuff_head_cache, GFP_ATOMIC);	if (!skb)		return NULL; 	size -= SKB_DATA_ALIGN(sizeof(struct skb_shared_info)); 	memset(skb, 0, offsetof(struct sk_buff, tail));	skb->truesize = SKB_TRUESIZE(size);	skb->head_frag = frag_size != 0;	atomic_set(&skb->users, 1);	skb->head = data;	skb->data = data;	skb_reset_tail_pointer(skb);	skb->end = skb->tail + size;#ifdef NET_SKBUFF_DATA_USES_OFFSET	skb->mac_header = ~0U;#endif 	/* make sure we initialize shinfo sequentially */	shinfo = skb_shinfo(skb);	memset(shinfo, 0, offsetof(struct skb_shared_info, dataref));	atomic_set(&shinfo->dataref, 1);	kmemcheck_annotate_variable(shinfo->destructor_arg); 	return skb;}EXPORT_SYMBOL(build_skb); struct netdev_alloc_cache {	struct page *page;	unsigned int offset;};static DEFINE_PER_CPU(struct netdev_alloc_cache, netdev_alloc_cache); /** * netdev_alloc_frag - allocate a page fragment * @fragsz: fragment size * * Allocates a frag from a page for receive buffer. * Uses GFP_ATOMIC allocations. */void *netdev_alloc_frag(unsigned int fragsz){	struct netdev_alloc_cache *nc;	void *data = NULL;	unsigned long flags; 	local_irq_save(flags);	nc = &__get_cpu_var(netdev_alloc_cache);	if (unlikely(!nc->page)) {refill:		nc->page = alloc_page(GFP_ATOMIC | __GFP_COLD);		nc->offset = 0;	}	if (likely(nc->page)) {		if (nc->offset + fragsz > PAGE_SIZE) {			put_page(nc->page);			goto refill;		}		data = page_address(nc->page) + nc->offset;		nc->offset += fragsz;		get_page(nc->page);	}	local_irq_restore(flags);	return data;}EXPORT_SYMBOL(netdev_alloc_frag); /** *	__netdev_alloc_skb - allocate an skbuff for rx on a specific device *	@dev: network device to receive on *	@length: length to allocate *	@gfp_mask: get_free_pages mask, passed to alloc_skb * *	Allocate a new &sk_buff and assign it a usage count of one. The *	buffer has unspecified headroom built in. Users should allocate *	the headroom they think they need without accounting for the *	built in space. The built in space is used for optimisations. * *	%NULL is returned if there is no free memory. */struct sk_buff *__netdev_alloc_skb(struct net_device *dev,				   unsigned int length, gfp_t gfp_mask){	struct sk_buff *skb = NULL;	unsigned int fragsz = SKB_DATA_ALIGN(length + NET_SKB_PAD) +			      SKB_DATA_ALIGN(sizeof(struct skb_shared_info)); 	if (fragsz <= PAGE_SIZE && !(gfp_mask & __GFP_WAIT)) {		void *data = netdev_alloc_frag(fragsz); 		if (likely(data)) {			skb = build_skb(data, fragsz);			if (unlikely(!skb))				put_page(virt_to_head_page(data));		}	} else {		skb = __alloc_skb(length + NET_SKB_PAD, gfp_mask, 0, NUMA_NO_NODE);	}	if (likely(skb)) {		skb_reserve(skb, NET_SKB_PAD);		skb->dev = dev;	}	return skb;}EXPORT_SYMBOL(__netdev_alloc_skb); void skb_add_rx_frag(struct sk_buff *skb, int i, struct page *page, int off,		     int size, unsigned int truesize){	skb_fill_page_desc(skb, i, page, off, size);	skb->len += size;	skb->data_len += size;	skb->truesize += truesize;}EXPORT_SYMBOL(skb_add_rx_frag); static void skb_drop_list(struct sk_buff **listp){	struct sk_buff *list = *listp; 	*listp = NULL; 	do {		struct sk_buff *this = list;		list = list->next;		kfree_skb(this);	} while (list);} static inline void skb_drop_fraglist(struct sk_buff *skb){	skb_drop_list(&skb_shinfo(skb)->frag_list);} static void skb_clone_fraglist(struct sk_buff *skb){	struct sk_buff *list; 	skb_walk_frags(skb, list)		skb_get(list);} static void skb_free_head(struct sk_buff *skb){	if (skb->head_frag)		put_page(virt_to_head_page(skb->head));	else		kfree(skb->head);} static void skb_release_data(struct sk_buff *skb){	if (!skb->cloned ||	    !atomic_sub_return(skb->nohdr ? (1 << SKB_DATAREF_SHIFT) + 1 : 1,			       &skb_shinfo(skb)->dataref)) {		if (skb_shinfo(skb)->nr_frags) {			int i;			for (i = 0; i < skb_shinfo(skb)->nr_frags; i++)				skb_frag_unref(skb, i);		} 		/*		 * If skb buf is from userspace, we need to notify the caller		 * the lower device DMA has done;		 */		if (skb_shinfo(skb)->tx_flags & SKBTX_DEV_ZEROCOPY) {			struct ubuf_info *uarg; 			uarg = skb_shinfo(skb)->destructor_arg;			if (uarg->callback)				uarg->callback(uarg);		} 		if (skb_has_frag_list(skb))			skb_drop_fraglist(skb); 		skb_free_head(skb);	}} /* *	Free an skbuff by memory without cleaning the state. */static void kfree_skbmem(struct sk_buff *skb){	struct sk_buff *other;	atomic_t *fclone_ref; 	switch (skb->fclone) {	case SKB_FCLONE_UNAVAILABLE:		kmem_cache_free(skbuff_head_cache, skb);		break; 	case SKB_FCLONE_ORIG:		fclone_ref = (atomic_t *) (skb + 2);		if (atomic_dec_and_test(fclone_ref))			kmem_cache_free(skbuff_fclone_cache, skb);		break; 	case SKB_FCLONE_CLONE:		fclone_ref = (atomic_t *) (skb + 1);		other = skb - 1; 		/* The clone portion is available for		 * fast-cloning again.		 */		skb->fclone = SKB_FCLONE_UNAVAILABLE; 		if (atomic_dec_and_test(fclone_ref))			kmem_cache_free(skbuff_fclone_cache, other);		break;	}} static void skb_release_head_state(struct sk_buff *skb){	skb_dst_drop(skb);#ifdef CONFIG_XFRM	secpath_put(skb->sp);#endif	if (skb->destructor) {		WARN_ON(in_irq());		skb->destructor(skb);	}#if IS_ENABLED(CONFIG_NF_CONNTRACK)	nf_conntrack_put(skb->nfct);#endif#ifdef NET_SKBUFF_NF_DEFRAG_NEEDED	nf_conntrack_put_reasm(skb->nfct_reasm);#endif#ifdef CONFIG_BRIDGE_NETFILTER	nf_bridge_put(skb->nf_bridge);

Lines 1–500 of 3434.