Merge git://git.kernel.org/pub/scm/linux/kernel/git/gregkh/staging-2.6
[linux-2.6] / include / linux / slab_def.h
1 #ifndef _LINUX_SLAB_DEF_H
2 #define _LINUX_SLAB_DEF_H
3
4 /*
5  * Definitions unique to the original Linux SLAB allocator.
6  *
7  * What we provide here is a way to optimize the frequent kmalloc
8  * calls in the kernel by selecting the appropriate general cache
9  * if kmalloc was called with a size that can be established at
10  * compile time.
11  */
12
13 #include <linux/init.h>
14 #include <asm/page.h>           /* kmalloc_sizes.h needs PAGE_SIZE */
15 #include <asm/cache.h>          /* kmalloc_sizes.h needs L1_CACHE_BYTES */
16 #include <linux/compiler.h>
17 #include <trace/kmemtrace.h>
18
19 /* Size description struct for general caches. */
20 struct cache_sizes {
21         size_t                  cs_size;
22         struct kmem_cache       *cs_cachep;
23 #ifdef CONFIG_ZONE_DMA
24         struct kmem_cache       *cs_dmacachep;
25 #endif
26 };
27 extern struct cache_sizes malloc_sizes[];
28
29 void *kmem_cache_alloc(struct kmem_cache *, gfp_t);
30 void *__kmalloc(size_t size, gfp_t flags);
31
32 #ifdef CONFIG_KMEMTRACE
33 extern void *kmem_cache_alloc_notrace(struct kmem_cache *cachep, gfp_t flags);
34 extern size_t slab_buffer_size(struct kmem_cache *cachep);
35 #else
36 static __always_inline void *
37 kmem_cache_alloc_notrace(struct kmem_cache *cachep, gfp_t flags)
38 {
39         return kmem_cache_alloc(cachep, flags);
40 }
41 static inline size_t slab_buffer_size(struct kmem_cache *cachep)
42 {
43         return 0;
44 }
45 #endif
46
47 static __always_inline void *kmalloc(size_t size, gfp_t flags)
48 {
49         struct kmem_cache *cachep;
50         void *ret;
51
52         if (__builtin_constant_p(size)) {
53                 int i = 0;
54
55                 if (!size)
56                         return ZERO_SIZE_PTR;
57
58 #define CACHE(x) \
59                 if (size <= x) \
60                         goto found; \
61                 else \
62                         i++;
63 #include <linux/kmalloc_sizes.h>
64 #undef CACHE
65                 return NULL;
66 found:
67 #ifdef CONFIG_ZONE_DMA
68                 if (flags & GFP_DMA)
69                         cachep = malloc_sizes[i].cs_dmacachep;
70                 else
71 #endif
72                         cachep = malloc_sizes[i].cs_cachep;
73
74                 ret = kmem_cache_alloc_notrace(cachep, flags);
75
76                 kmemtrace_mark_alloc(KMEMTRACE_TYPE_KMALLOC, _THIS_IP_, ret,
77                                      size, slab_buffer_size(cachep), flags);
78
79                 return ret;
80         }
81         return __kmalloc(size, flags);
82 }
83
84 #ifdef CONFIG_NUMA
85 extern void *__kmalloc_node(size_t size, gfp_t flags, int node);
86 extern void *kmem_cache_alloc_node(struct kmem_cache *, gfp_t flags, int node);
87
88 #ifdef CONFIG_KMEMTRACE
89 extern void *kmem_cache_alloc_node_notrace(struct kmem_cache *cachep,
90                                            gfp_t flags,
91                                            int nodeid);
92 #else
93 static __always_inline void *
94 kmem_cache_alloc_node_notrace(struct kmem_cache *cachep,
95                               gfp_t flags,
96                               int nodeid)
97 {
98         return kmem_cache_alloc_node(cachep, flags, nodeid);
99 }
100 #endif
101
102 static __always_inline void *kmalloc_node(size_t size, gfp_t flags, int node)
103 {
104         struct kmem_cache *cachep;
105         void *ret;
106
107         if (__builtin_constant_p(size)) {
108                 int i = 0;
109
110                 if (!size)
111                         return ZERO_SIZE_PTR;
112
113 #define CACHE(x) \
114                 if (size <= x) \
115                         goto found; \
116                 else \
117                         i++;
118 #include <linux/kmalloc_sizes.h>
119 #undef CACHE
120                 return NULL;
121 found:
122 #ifdef CONFIG_ZONE_DMA
123                 if (flags & GFP_DMA)
124                         cachep = malloc_sizes[i].cs_dmacachep;
125                 else
126 #endif
127                         cachep = malloc_sizes[i].cs_cachep;
128
129                 ret = kmem_cache_alloc_node_notrace(cachep, flags, node);
130
131                 kmemtrace_mark_alloc_node(KMEMTRACE_TYPE_KMALLOC, _THIS_IP_,
132                                           ret, size, slab_buffer_size(cachep),
133                                           flags, node);
134
135                 return ret;
136         }
137         return __kmalloc_node(size, flags, node);
138 }
139
140 #endif  /* CONFIG_NUMA */
141
142 #endif  /* _LINUX_SLAB_DEF_H */