xref: /dragonfly/sys/sys/malloc.h (revision 88ed2a5c)
1 /*
2  * Copyright (c) 1987, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  *	@(#)malloc.h	8.5 (Berkeley) 5/3/95
30  * $FreeBSD: src/sys/sys/malloc.h,v 1.48.2.2 2002/03/16 02:19:16 archie Exp $
31  */
32 
33 #ifndef _SYS_MALLOC_H_
34 #define	_SYS_MALLOC_H_
35 
36 #ifndef _SYS_TYPES_H_
37 #include <sys/types.h>
38 #endif
39 #ifndef _MACHINE_TYPES_H_
40 #include <machine/types.h>	/* vm_paddr_t and __* types */
41 #endif
42 
43 #ifndef _MACHINE_PARAM_H_
44 #include <machine/param.h>	/* for SMP_MAXCPU */
45 #endif
46 
47 /*
48  * flags to malloc.
49  */
50 #define	M_RNOWAIT	0x0001	/* do not block */
51 #define	M_WAITOK	0x0002	/* wait for resources / alloc from cache */
52 #define	M_ZERO		0x0100	/* bzero() the allocation */
53 #define	M_USE_RESERVE	0x0200	/* can eat into free list reserve */
54 #define	M_NULLOK	0x0400	/* ok to return NULL */
55 #define	M_PASSIVE_ZERO	0x0800	/* (internal to the slab code only) */
56 #define	M_USE_INTERRUPT_RESERVE \
57 			0x1000	/* can exhaust free list entirely */
58 #define	M_POWEROF2	0x2000	/* roundup size to the nearest power of 2 */
59 #define	M_CACHEALIGN	0x4000	/* force CPU cache line alignment */
60 
61 /*
62  * M_NOWAIT has to be a set of flags for equivalence to prior use.
63  *
64  * M_SYSALLOC should be used for any critical infrastructure allocations
65  * made by the kernel proper.
66  *
67  * M_INTNOWAIT should be used for any critical infrastructure allocations
68  * made by interrupts.  Such allocations can still fail but will not fail
69  * as often as M_NOWAIT.
70  *
71  * NOTE ON DRAGONFLY USE OF M_NOWAIT.  In FreeBSD M_NOWAIT allocations
72  * almost always succeed.  In DragonFly, however, there is a good chance
73  * that an allocation will fail.  M_NOWAIT should only be used when
74  * allocations can fail without any serious detriment to the system.
75  *
76  * Note that allocations made from (preempted) interrupts will attempt to
77  * use pages from the VM PAGE CACHE (PQ_CACHE) (i.e. those associated with
78  * objects).  This is automatic.
79  */
80 
81 #define	M_INTNOWAIT	(M_RNOWAIT | M_NULLOK | 			\
82 			 M_USE_RESERVE | M_USE_INTERRUPT_RESERVE)
83 #define	M_SYSNOWAIT	(M_RNOWAIT | M_NULLOK | M_USE_RESERVE)
84 #define	M_INTWAIT	(M_WAITOK | M_USE_RESERVE | M_USE_INTERRUPT_RESERVE)
85 #define	M_SYSWAIT	(M_WAITOK | M_USE_RESERVE)
86 
87 #define	M_NOWAIT	(M_RNOWAIT | M_NULLOK | M_USE_RESERVE)
88 #define	M_SYSALLOC	M_SYSWAIT
89 
90 #define	M_MAGIC		877983977	/* time when first defined :-) */
91 
92 /*
93  * The malloc tracking structure.  Note that per-cpu entries must be
94  * aggregated for accurate statistics, they do not actually break the
95  * stats down by cpu (e.g. the cpu freeing memory will subtract from
96  * its slot, not the originating cpu's slot).
97  *
98  * SMP_MAXCPU is used so modules which use malloc remain compatible
99  * between UP and SMP.
100  */
101 struct malloc_use {
102 	size_t	memuse;
103 	size_t	inuse;
104 	__int64_t calls;	/* total packets of this type ever allocated */
105 
106 	/*
107 	 * This value will be added to ks_loosememuse and resetted,
108 	 * once it goes above certain threshold (ZoneSize).  This
109 	 * is intended to reduce frequency of ks_loosememuse (global)
110 	 * updates.
111 	 */
112 	size_t	loosememuse;
113 } __cachealign;
114 
115 struct malloc_type {
116 	struct malloc_type *ks_next;	/* next in list */
117 	size_t	ks_loosememuse;		/* (inaccurate) aggregate memuse */
118 	size_t	ks_limit;	/* most that are allowed to exist */
119 	struct malloc_use  ks_use[SMP_MAXCPU];
120 	__uint32_t ks_magic;	/* if it's not magic, don't touch it */
121 	const char *ks_shortdesc;	/* short description */
122 	long	ks_reserved[4];	/* future use (module compatibility) */
123 };
124 
125 typedef struct malloc_type	*malloc_type_t;
126 
127 #if defined(_KERNEL) || defined(_KERNEL_STRUCTURES)
128 #define	MALLOC_DEFINE(type, shortdesc, longdesc)			\
129 	struct malloc_type type[1] = {					\
130 	    { NULL, 0, 0, { { 0, 0, 0, 0 } }, M_MAGIC, shortdesc,	\
131 		{ 0 } }							\
132 	};								\
133 	SYSINIT(type##_init, SI_BOOT1_KMALLOC, SI_ORDER_ANY,		\
134 	    malloc_init, type);						\
135 	SYSUNINIT(type##_uninit, SI_BOOT1_KMALLOC, SI_ORDER_ANY,	\
136 	    malloc_uninit, type)
137 #else
138 #define	MALLOC_DEFINE(type, shortdesc, longdesc)			\
139 	struct malloc_type type[1] = {					\
140 	    { NULL, 0, 0, { { 0, 0, 0, 0 } }, M_MAGIC, shortdesc,	\
141 	        { 0 } }							\
142 	}
143 #endif
144 
145 #define	MALLOC_DECLARE(type) \
146 	extern struct malloc_type type[1]
147 
148 #ifdef _KERNEL
149 
150 MALLOC_DECLARE(M_CACHE);
151 MALLOC_DECLARE(M_DEVBUF);
152 MALLOC_DECLARE(M_TEMP);
153 
154 MALLOC_DECLARE(M_IP6OPT); /* for INET6 */
155 MALLOC_DECLARE(M_IP6NDP); /* for INET6 */
156 
157 #endif /* _KERNEL */
158 
159 #ifdef _KERNEL
160 
161 #define	MINALLOCSIZE	sizeof(void *)
162 
163 /*
164  * XXX this should be declared in <sys/uio.h>, but that tends to fail
165  * because <sys/uio.h> is included in a header before the source file
166  * has a chance to include <sys/malloc.h> to get MALLOC_DECLARE() defined.
167  */
168 MALLOC_DECLARE(M_IOV);
169 
170 /* XXX struct malloc_type is unused for contig*(). */
171 size_t  kmem_lim_size(void);
172 void	contigfree(void *addr, unsigned long size, struct malloc_type *type)
173 	    __nonnull(1);
174 void	*contigmalloc(unsigned long size, struct malloc_type *type, int flags,
175 		      vm_paddr_t low, vm_paddr_t high, unsigned long alignment,
176 		      unsigned long boundary) __malloclike __heedresult
177 		      __alloc_size(1) __alloc_align(6);
178 void	malloc_init(void *);
179 void	malloc_uninit(void *);
180 void	kmalloc_raise_limit(struct malloc_type *type, size_t bytes);
181 void	kmalloc_set_unlimited(struct malloc_type *type);
182 void	kmalloc_create(struct malloc_type **typep, const char *descr);
183 void	kmalloc_destroy(struct malloc_type **typep);
184 
185 /*
186  * Debug and non-debug kmalloc() prototypes.
187  *
188  * The kmalloc() macro allows M_ZERO to be optimized external to
189  * the kmalloc() function.  When combined with the use a builtin
190  * for bzero() this can get rid of a considerable amount of overhead
191  * for M_ZERO based kmalloc() calls.
192  */
193 #ifdef SLAB_DEBUG
194 void	*kmalloc_debug(unsigned long size, struct malloc_type *type, int flags,
195 			const char *file, int line) __malloclike __heedresult
196 			__alloc_size(1);
197 void	*krealloc_debug(void *addr, unsigned long size,
198 			struct malloc_type *type, int flags,
199 			const char *file, int line) __heedresult __alloc_size(2);
200 char	*kstrdup_debug(const char *, struct malloc_type *,
201 			const char *file, int line) __malloclike __heedresult;
202 char	*kstrndup_debug(const char *, size_t maxlen, struct malloc_type *,
203 			const char *file, int line) __malloclike __heedresult;
204 #if 1
205 #define kmalloc(size, type, flags) ({					\
206 	void *_malloc_item;						\
207 	size_t _size = (size);						\
208 									\
209 	if (__builtin_constant_p(size) &&				\
210 	    __builtin_constant_p(flags) &&				\
211 	    ((flags) & M_ZERO)) {					\
212 		_malloc_item = kmalloc_debug(_size, type,		\
213 					    (flags) & ~M_ZERO,		\
214 					    __FILE__, __LINE__);	\
215 		if (((flags) & (M_WAITOK|M_NULLOK)) == M_WAITOK ||	\
216 		    __predict_true(_malloc_item != NULL)) {		\
217 			bzero(_malloc_item, _size);			\
218 		}							\
219 	} else {							\
220 	    _malloc_item = kmalloc_debug(_size, type, flags,		\
221 				   __FILE__, __LINE__);			\
222 	}								\
223 	_malloc_item;							\
224 })
225 #endif
226 #define krealloc(addr, size, type, flags)	\
227 	krealloc_debug(addr, size, type, flags, __FILE__, __LINE__)
228 #define kstrdup(str, type)			\
229 	kstrdup_debug(str, type, __FILE__, __LINE__)
230 #define kstrndup(str, maxlen, type)			\
231 	kstrndup_debug(str, maxlen, type, __FILE__, __LINE__)
232 
233 #else	/* !SLAB_DEBUG */
234 
235 void	*kmalloc(unsigned long size, struct malloc_type *type, int flags)
236 		 __malloclike __heedresult __alloc_size(1);
237 #if 1
238 #define kmalloc(size, type, flags) ({					\
239 	void *_malloc_item;						\
240 	size_t _size = (size);						\
241 									\
242 	if (__builtin_constant_p(size) &&				\
243 	    __builtin_constant_p(flags) &&				\
244 	    ((flags) & M_ZERO)) {					\
245 		_malloc_item = kmalloc(_size, type, (flags) & ~M_ZERO);	\
246 		if (((flags) & (M_WAITOK|M_NULLOK)) == M_WAITOK ||	\
247 		    __predict_true(_malloc_item != NULL)) {		\
248 			bzero(_malloc_item, _size);			\
249 		}							\
250 	} else {							\
251 	    _malloc_item = kmalloc(_size, type, flags);			\
252 	}								\
253 	_malloc_item;							\
254 })
255 #endif
256 void	*krealloc(void *addr, unsigned long size, struct malloc_type *type,
257 		  int flags) __heedresult __alloc_size(2);
258 char	*kstrdup(const char *, struct malloc_type *)
259 		 __malloclike __heedresult;
260 char	*kstrndup(const char *, size_t maxlen, struct malloc_type *)
261 		  __malloclike __heedresult;
262 #define kmalloc_debug(size, type, flags, file, line)		\
263 	kmalloc(size, type, flags)
264 #define krealloc_debug(addr, size, type, flags, file, line)	\
265 	krealloc(addr, size, type, flags)
266 #define kstrdup_debug(str, type, file, line)			\
267 	kstrdup(str, type)
268 #define kstrndup_debug(str, maxlen, type, file, line)		\
269 	kstrndup(str, maxlen, type)
270 #endif
271 void	kfree(void *addr, struct malloc_type *type) __nonnull(2);
272 long	kmalloc_limit(struct malloc_type *type);
273 void	slab_cleanup(void);
274 
275 #endif /* _KERNEL */
276 
277 #endif /* !_SYS_MALLOC_H_ */
278