xref: /minix3/external/bsd/bind/dist/lib/isc/mem.c (revision 00b67f09dd46474d133c95011a48590a8e8f94c7)
1*00b67f09SDavid van Moolenbroek /*	$NetBSD: mem.c,v 1.12 2015/07/08 17:28:59 christos Exp $	*/
2*00b67f09SDavid van Moolenbroek 
3*00b67f09SDavid van Moolenbroek /*
4*00b67f09SDavid van Moolenbroek  * Copyright (C) 2004-2010, 2012-2015  Internet Systems Consortium, Inc. ("ISC")
5*00b67f09SDavid van Moolenbroek  * Copyright (C) 1997-2003  Internet Software Consortium.
6*00b67f09SDavid van Moolenbroek  *
7*00b67f09SDavid van Moolenbroek  * Permission to use, copy, modify, and/or distribute this software for any
8*00b67f09SDavid van Moolenbroek  * purpose with or without fee is hereby granted, provided that the above
9*00b67f09SDavid van Moolenbroek  * copyright notice and this permission notice appear in all copies.
10*00b67f09SDavid van Moolenbroek  *
11*00b67f09SDavid van Moolenbroek  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
12*00b67f09SDavid van Moolenbroek  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
13*00b67f09SDavid van Moolenbroek  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
14*00b67f09SDavid van Moolenbroek  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
15*00b67f09SDavid van Moolenbroek  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
16*00b67f09SDavid van Moolenbroek  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
17*00b67f09SDavid van Moolenbroek  * PERFORMANCE OF THIS SOFTWARE.
18*00b67f09SDavid van Moolenbroek  */
19*00b67f09SDavid van Moolenbroek 
20*00b67f09SDavid van Moolenbroek /* Id */
21*00b67f09SDavid van Moolenbroek 
22*00b67f09SDavid van Moolenbroek /*! \file */
23*00b67f09SDavid van Moolenbroek 
24*00b67f09SDavid van Moolenbroek #include <config.h>
25*00b67f09SDavid van Moolenbroek 
26*00b67f09SDavid van Moolenbroek #include <stdio.h>
27*00b67f09SDavid van Moolenbroek #include <stdlib.h>
28*00b67f09SDavid van Moolenbroek #include <stddef.h>
29*00b67f09SDavid van Moolenbroek 
30*00b67f09SDavid van Moolenbroek #include <limits.h>
31*00b67f09SDavid van Moolenbroek 
32*00b67f09SDavid van Moolenbroek #include <isc/bind9.h>
33*00b67f09SDavid van Moolenbroek #include <isc/json.h>
34*00b67f09SDavid van Moolenbroek #include <isc/magic.h>
35*00b67f09SDavid van Moolenbroek #include <isc/mem.h>
36*00b67f09SDavid van Moolenbroek #include <isc/msgs.h>
37*00b67f09SDavid van Moolenbroek #include <isc/once.h>
38*00b67f09SDavid van Moolenbroek #include <isc/ondestroy.h>
39*00b67f09SDavid van Moolenbroek #include <isc/string.h>
40*00b67f09SDavid van Moolenbroek #include <isc/mutex.h>
41*00b67f09SDavid van Moolenbroek #include <isc/print.h>
42*00b67f09SDavid van Moolenbroek #include <isc/util.h>
43*00b67f09SDavid van Moolenbroek #include <isc/xml.h>
44*00b67f09SDavid van Moolenbroek 
45*00b67f09SDavid van Moolenbroek #define MCTXLOCK(m, l) if (((m)->flags & ISC_MEMFLAG_NOLOCK) == 0) LOCK(l)
46*00b67f09SDavid van Moolenbroek #define MCTXUNLOCK(m, l) if (((m)->flags & ISC_MEMFLAG_NOLOCK) == 0) UNLOCK(l)
47*00b67f09SDavid van Moolenbroek 
48*00b67f09SDavid van Moolenbroek #ifndef ISC_MEM_DEBUGGING
49*00b67f09SDavid van Moolenbroek #define ISC_MEM_DEBUGGING 0
50*00b67f09SDavid van Moolenbroek #endif
51*00b67f09SDavid van Moolenbroek LIBISC_EXTERNAL_DATA unsigned int isc_mem_debugging = ISC_MEM_DEBUGGING;
52*00b67f09SDavid van Moolenbroek 
53*00b67f09SDavid van Moolenbroek /*
54*00b67f09SDavid van Moolenbroek  * Constants.
55*00b67f09SDavid van Moolenbroek  */
56*00b67f09SDavid van Moolenbroek 
57*00b67f09SDavid van Moolenbroek #define DEF_MAX_SIZE		1100
58*00b67f09SDavid van Moolenbroek #define DEF_MEM_TARGET		4096
59*00b67f09SDavid van Moolenbroek #define ALIGNMENT_SIZE		8U		/*%< must be a power of 2 */
60*00b67f09SDavid van Moolenbroek #define NUM_BASIC_BLOCKS	64		/*%< must be > 1 */
61*00b67f09SDavid van Moolenbroek #define TABLE_INCREMENT		1024
62*00b67f09SDavid van Moolenbroek #define DEBUGLIST_COUNT		1024
63*00b67f09SDavid van Moolenbroek 
64*00b67f09SDavid van Moolenbroek /*
65*00b67f09SDavid van Moolenbroek  * Types.
66*00b67f09SDavid van Moolenbroek  */
67*00b67f09SDavid van Moolenbroek typedef struct isc__mem isc__mem_t;
68*00b67f09SDavid van Moolenbroek typedef struct isc__mempool isc__mempool_t;
69*00b67f09SDavid van Moolenbroek 
70*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
71*00b67f09SDavid van Moolenbroek typedef struct debuglink debuglink_t;
72*00b67f09SDavid van Moolenbroek struct debuglink {
73*00b67f09SDavid van Moolenbroek 	ISC_LINK(debuglink_t)	link;
74*00b67f09SDavid van Moolenbroek 	const void	       *ptr[DEBUGLIST_COUNT];
75*00b67f09SDavid van Moolenbroek 	size_t			size[DEBUGLIST_COUNT];
76*00b67f09SDavid van Moolenbroek 	const char	       *file[DEBUGLIST_COUNT];
77*00b67f09SDavid van Moolenbroek 	unsigned int		line[DEBUGLIST_COUNT];
78*00b67f09SDavid van Moolenbroek 	unsigned int		count;
79*00b67f09SDavid van Moolenbroek };
80*00b67f09SDavid van Moolenbroek 
81*00b67f09SDavid van Moolenbroek #define FLARG_PASS	, file, line
82*00b67f09SDavid van Moolenbroek #define FLARG		, const char *file, unsigned int line
83*00b67f09SDavid van Moolenbroek #else
84*00b67f09SDavid van Moolenbroek #define FLARG_PASS
85*00b67f09SDavid van Moolenbroek #define FLARG
86*00b67f09SDavid van Moolenbroek #endif
87*00b67f09SDavid van Moolenbroek 
88*00b67f09SDavid van Moolenbroek typedef struct element element;
89*00b67f09SDavid van Moolenbroek struct element {
90*00b67f09SDavid van Moolenbroek 	element *		next;
91*00b67f09SDavid van Moolenbroek };
92*00b67f09SDavid van Moolenbroek 
93*00b67f09SDavid van Moolenbroek typedef struct {
94*00b67f09SDavid van Moolenbroek 	/*!
95*00b67f09SDavid van Moolenbroek 	 * This structure must be ALIGNMENT_SIZE bytes.
96*00b67f09SDavid van Moolenbroek 	 */
97*00b67f09SDavid van Moolenbroek 	union {
98*00b67f09SDavid van Moolenbroek 		size_t		size;
99*00b67f09SDavid van Moolenbroek 		isc__mem_t	*ctx;
100*00b67f09SDavid van Moolenbroek 		char		bytes[ALIGNMENT_SIZE];
101*00b67f09SDavid van Moolenbroek 	} u;
102*00b67f09SDavid van Moolenbroek } size_info;
103*00b67f09SDavid van Moolenbroek 
104*00b67f09SDavid van Moolenbroek struct stats {
105*00b67f09SDavid van Moolenbroek 	unsigned long		gets;
106*00b67f09SDavid van Moolenbroek 	unsigned long		totalgets;
107*00b67f09SDavid van Moolenbroek 	unsigned long		blocks;
108*00b67f09SDavid van Moolenbroek 	unsigned long		freefrags;
109*00b67f09SDavid van Moolenbroek };
110*00b67f09SDavid van Moolenbroek 
111*00b67f09SDavid van Moolenbroek #define MEM_MAGIC		ISC_MAGIC('M', 'e', 'm', 'C')
112*00b67f09SDavid van Moolenbroek #define VALID_CONTEXT(c)	ISC_MAGIC_VALID(c, MEM_MAGIC)
113*00b67f09SDavid van Moolenbroek 
114*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
115*00b67f09SDavid van Moolenbroek typedef ISC_LIST(debuglink_t)	debuglist_t;
116*00b67f09SDavid van Moolenbroek #endif
117*00b67f09SDavid van Moolenbroek 
118*00b67f09SDavid van Moolenbroek /* List of all active memory contexts. */
119*00b67f09SDavid van Moolenbroek 
120*00b67f09SDavid van Moolenbroek static ISC_LIST(isc__mem_t)	contexts;
121*00b67f09SDavid van Moolenbroek 
122*00b67f09SDavid van Moolenbroek static isc_once_t		once = ISC_ONCE_INIT;
123*00b67f09SDavid van Moolenbroek static isc_mutex_t		contextslock;
124*00b67f09SDavid van Moolenbroek static isc_mutex_t 		createlock;
125*00b67f09SDavid van Moolenbroek 
126*00b67f09SDavid van Moolenbroek /*%
127*00b67f09SDavid van Moolenbroek  * Total size of lost memory due to a bug of external library.
128*00b67f09SDavid van Moolenbroek  * Locked by the global lock.
129*00b67f09SDavid van Moolenbroek  */
130*00b67f09SDavid van Moolenbroek static isc_uint64_t		totallost;
131*00b67f09SDavid van Moolenbroek 
132*00b67f09SDavid van Moolenbroek struct isc__mem {
133*00b67f09SDavid van Moolenbroek 	isc_mem_t		common;
134*00b67f09SDavid van Moolenbroek 	isc_ondestroy_t		ondestroy;
135*00b67f09SDavid van Moolenbroek 	unsigned int		flags;
136*00b67f09SDavid van Moolenbroek 	isc_mutex_t		lock;
137*00b67f09SDavid van Moolenbroek 	isc_memalloc_t		memalloc;
138*00b67f09SDavid van Moolenbroek 	isc_memfree_t		memfree;
139*00b67f09SDavid van Moolenbroek 	void *			arg;
140*00b67f09SDavid van Moolenbroek 	size_t			max_size;
141*00b67f09SDavid van Moolenbroek 	isc_boolean_t		checkfree;
142*00b67f09SDavid van Moolenbroek 	struct stats *		stats;
143*00b67f09SDavid van Moolenbroek 	unsigned int		references;
144*00b67f09SDavid van Moolenbroek 	char			name[16];
145*00b67f09SDavid van Moolenbroek 	void *			tag;
146*00b67f09SDavid van Moolenbroek 	size_t			quota;
147*00b67f09SDavid van Moolenbroek 	size_t			total;
148*00b67f09SDavid van Moolenbroek 	size_t			inuse;
149*00b67f09SDavid van Moolenbroek 	size_t			maxinuse;
150*00b67f09SDavid van Moolenbroek 	size_t			hi_water;
151*00b67f09SDavid van Moolenbroek 	size_t			lo_water;
152*00b67f09SDavid van Moolenbroek 	isc_boolean_t		hi_called;
153*00b67f09SDavid van Moolenbroek 	isc_boolean_t		is_overmem;
154*00b67f09SDavid van Moolenbroek 	isc_mem_water_t		water;
155*00b67f09SDavid van Moolenbroek 	void *			water_arg;
156*00b67f09SDavid van Moolenbroek 	ISC_LIST(isc__mempool_t) pools;
157*00b67f09SDavid van Moolenbroek 	unsigned int		poolcnt;
158*00b67f09SDavid van Moolenbroek 
159*00b67f09SDavid van Moolenbroek 	/*  ISC_MEMFLAG_INTERNAL */
160*00b67f09SDavid van Moolenbroek 	size_t			mem_target;
161*00b67f09SDavid van Moolenbroek 	element **		freelists;
162*00b67f09SDavid van Moolenbroek 	element *		basic_blocks;
163*00b67f09SDavid van Moolenbroek 	unsigned char **	basic_table;
164*00b67f09SDavid van Moolenbroek 	unsigned int		basic_table_count;
165*00b67f09SDavid van Moolenbroek 	unsigned int		basic_table_size;
166*00b67f09SDavid van Moolenbroek 	unsigned char *		lowest;
167*00b67f09SDavid van Moolenbroek 	unsigned char *		highest;
168*00b67f09SDavid van Moolenbroek 
169*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
170*00b67f09SDavid van Moolenbroek 	debuglist_t *	 	debuglist;
171*00b67f09SDavid van Moolenbroek 	unsigned int		debuglistcnt;
172*00b67f09SDavid van Moolenbroek #endif
173*00b67f09SDavid van Moolenbroek 
174*00b67f09SDavid van Moolenbroek 	unsigned int		memalloc_failures;
175*00b67f09SDavid van Moolenbroek 	ISC_LINK(isc__mem_t)	link;
176*00b67f09SDavid van Moolenbroek };
177*00b67f09SDavid van Moolenbroek 
178*00b67f09SDavid van Moolenbroek #define MEMPOOL_MAGIC		ISC_MAGIC('M', 'E', 'M', 'p')
179*00b67f09SDavid van Moolenbroek #define VALID_MEMPOOL(c)	ISC_MAGIC_VALID(c, MEMPOOL_MAGIC)
180*00b67f09SDavid van Moolenbroek 
181*00b67f09SDavid van Moolenbroek struct isc__mempool {
182*00b67f09SDavid van Moolenbroek 	/* always unlocked */
183*00b67f09SDavid van Moolenbroek 	isc_mempool_t	common;		/*%< common header of mempool's */
184*00b67f09SDavid van Moolenbroek 	isc_mutex_t    *lock;		/*%< optional lock */
185*00b67f09SDavid van Moolenbroek 	isc__mem_t      *mctx;		/*%< our memory context */
186*00b67f09SDavid van Moolenbroek 	/*%< locked via the memory context's lock */
187*00b67f09SDavid van Moolenbroek 	ISC_LINK(isc__mempool_t)	link;	/*%< next pool in this mem context */
188*00b67f09SDavid van Moolenbroek 	/*%< optionally locked from here down */
189*00b67f09SDavid van Moolenbroek 	element	       *items;		/*%< low water item list */
190*00b67f09SDavid van Moolenbroek 	size_t		size;		/*%< size of each item on this pool */
191*00b67f09SDavid van Moolenbroek 	unsigned int	maxalloc;	/*%< max number of items allowed */
192*00b67f09SDavid van Moolenbroek 	unsigned int	allocated;	/*%< # of items currently given out */
193*00b67f09SDavid van Moolenbroek 	unsigned int	freecount;	/*%< # of items on reserved list */
194*00b67f09SDavid van Moolenbroek 	unsigned int	freemax;	/*%< # of items allowed on free list */
195*00b67f09SDavid van Moolenbroek 	unsigned int	fillcount;	/*%< # of items to fetch on each fill */
196*00b67f09SDavid van Moolenbroek 	/*%< Stats only. */
197*00b67f09SDavid van Moolenbroek 	unsigned int	gets;		/*%< # of requests to this pool */
198*00b67f09SDavid van Moolenbroek 	/*%< Debugging only. */
199*00b67f09SDavid van Moolenbroek #if ISC_MEMPOOL_NAMES
200*00b67f09SDavid van Moolenbroek 	char		name[16];	/*%< printed name in stats reports */
201*00b67f09SDavid van Moolenbroek #endif
202*00b67f09SDavid van Moolenbroek };
203*00b67f09SDavid van Moolenbroek 
204*00b67f09SDavid van Moolenbroek /*
205*00b67f09SDavid van Moolenbroek  * Private Inline-able.
206*00b67f09SDavid van Moolenbroek  */
207*00b67f09SDavid van Moolenbroek 
208*00b67f09SDavid van Moolenbroek #if ! ISC_MEM_TRACKLINES
209*00b67f09SDavid van Moolenbroek #define ADD_TRACE(a, b, c, d, e)
210*00b67f09SDavid van Moolenbroek #define DELETE_TRACE(a, b, c, d, e)
211*00b67f09SDavid van Moolenbroek #define ISC_MEMFUNC_SCOPE
212*00b67f09SDavid van Moolenbroek #else
213*00b67f09SDavid van Moolenbroek #define ADD_TRACE(a, b, c, d, e) \
214*00b67f09SDavid van Moolenbroek 	do { \
215*00b67f09SDavid van Moolenbroek 		if ((isc_mem_debugging & (ISC_MEM_DEBUGTRACE | \
216*00b67f09SDavid van Moolenbroek 					  ISC_MEM_DEBUGRECORD)) != 0 && \
217*00b67f09SDavid van Moolenbroek 		     b != NULL) \
218*00b67f09SDavid van Moolenbroek 			 add_trace_entry(a, b, c, d, e); \
219*00b67f09SDavid van Moolenbroek 	} while (/*CONSTCOND*/0)
220*00b67f09SDavid van Moolenbroek #define DELETE_TRACE(a, b, c, d, e)	delete_trace_entry(a, b, c, d, e)
221*00b67f09SDavid van Moolenbroek 
222*00b67f09SDavid van Moolenbroek static void
223*00b67f09SDavid van Moolenbroek print_active(isc__mem_t *ctx, FILE *out);
224*00b67f09SDavid van Moolenbroek 
225*00b67f09SDavid van Moolenbroek /*%
226*00b67f09SDavid van Moolenbroek  * The following are intended for internal use (indicated by "isc__"
227*00b67f09SDavid van Moolenbroek  * prefix) but are not declared as static, allowing direct access
228*00b67f09SDavid van Moolenbroek  * from unit tests, etc.
229*00b67f09SDavid van Moolenbroek  */
230*00b67f09SDavid van Moolenbroek 
231*00b67f09SDavid van Moolenbroek isc_result_t
232*00b67f09SDavid van Moolenbroek isc__mem_create2(size_t init_max_size, size_t target_size,
233*00b67f09SDavid van Moolenbroek 		 isc_mem_t **ctxp, unsigned int flags);
234*00b67f09SDavid van Moolenbroek void
235*00b67f09SDavid van Moolenbroek isc__mem_attach(isc_mem_t *source, isc_mem_t **targetp);
236*00b67f09SDavid van Moolenbroek void
237*00b67f09SDavid van Moolenbroek isc__mem_detach(isc_mem_t **ctxp);
238*00b67f09SDavid van Moolenbroek void
239*00b67f09SDavid van Moolenbroek isc___mem_putanddetach(isc_mem_t **ctxp, void *ptr, size_t size FLARG);
240*00b67f09SDavid van Moolenbroek void
241*00b67f09SDavid van Moolenbroek isc__mem_destroy(isc_mem_t **ctxp);
242*00b67f09SDavid van Moolenbroek isc_result_t
243*00b67f09SDavid van Moolenbroek isc__mem_ondestroy(isc_mem_t *ctx, isc_task_t *task, isc_event_t **event);
244*00b67f09SDavid van Moolenbroek void *
245*00b67f09SDavid van Moolenbroek isc___mem_get(isc_mem_t *ctx, size_t size FLARG);
246*00b67f09SDavid van Moolenbroek void
247*00b67f09SDavid van Moolenbroek isc___mem_put(isc_mem_t *ctx, void *ptr, size_t size FLARG);
248*00b67f09SDavid van Moolenbroek void
249*00b67f09SDavid van Moolenbroek isc__mem_stats(isc_mem_t *ctx, FILE *out);
250*00b67f09SDavid van Moolenbroek void *
251*00b67f09SDavid van Moolenbroek isc___mem_allocate(isc_mem_t *ctx, size_t size FLARG);
252*00b67f09SDavid van Moolenbroek void *
253*00b67f09SDavid van Moolenbroek isc___mem_reallocate(isc_mem_t *ctx, void *ptr, size_t size FLARG);
254*00b67f09SDavid van Moolenbroek void
255*00b67f09SDavid van Moolenbroek isc___mem_free(isc_mem_t *ctx, void *ptr FLARG);
256*00b67f09SDavid van Moolenbroek char *
257*00b67f09SDavid van Moolenbroek isc___mem_strdup(isc_mem_t *mctx, const char *s FLARG);
258*00b67f09SDavid van Moolenbroek void
259*00b67f09SDavid van Moolenbroek isc__mem_setdestroycheck(isc_mem_t *ctx, isc_boolean_t flag);
260*00b67f09SDavid van Moolenbroek void
261*00b67f09SDavid van Moolenbroek isc__mem_setquota(isc_mem_t *ctx, size_t quota);
262*00b67f09SDavid van Moolenbroek size_t
263*00b67f09SDavid van Moolenbroek isc__mem_getquota(isc_mem_t *ctx);
264*00b67f09SDavid van Moolenbroek size_t
265*00b67f09SDavid van Moolenbroek isc__mem_inuse(isc_mem_t *ctx);
266*00b67f09SDavid van Moolenbroek size_t
267*00b67f09SDavid van Moolenbroek isc__mem_maxinuse(isc_mem_t *ctx);
268*00b67f09SDavid van Moolenbroek size_t
269*00b67f09SDavid van Moolenbroek isc__mem_total(isc_mem_t *ctx);
270*00b67f09SDavid van Moolenbroek isc_boolean_t
271*00b67f09SDavid van Moolenbroek isc__mem_isovermem(isc_mem_t *ctx);
272*00b67f09SDavid van Moolenbroek void
273*00b67f09SDavid van Moolenbroek isc__mem_setwater(isc_mem_t *ctx, isc_mem_water_t water, void *water_arg,
274*00b67f09SDavid van Moolenbroek 		  size_t hiwater, size_t lowater);
275*00b67f09SDavid van Moolenbroek void
276*00b67f09SDavid van Moolenbroek isc__mem_waterack(isc_mem_t *ctx0, int flag);
277*00b67f09SDavid van Moolenbroek void
278*00b67f09SDavid van Moolenbroek isc__mem_setname(isc_mem_t *ctx, const char *name, void *tag);
279*00b67f09SDavid van Moolenbroek const char *
280*00b67f09SDavid van Moolenbroek isc__mem_getname(isc_mem_t *ctx);
281*00b67f09SDavid van Moolenbroek void *
282*00b67f09SDavid van Moolenbroek isc__mem_gettag(isc_mem_t *ctx);
283*00b67f09SDavid van Moolenbroek isc_result_t
284*00b67f09SDavid van Moolenbroek isc__mempool_create(isc_mem_t *mctx, size_t size, isc_mempool_t **mpctxp);
285*00b67f09SDavid van Moolenbroek void
286*00b67f09SDavid van Moolenbroek isc__mempool_setname(isc_mempool_t *mpctx, const char *name);
287*00b67f09SDavid van Moolenbroek void
288*00b67f09SDavid van Moolenbroek isc__mempool_destroy(isc_mempool_t **mpctxp);
289*00b67f09SDavid van Moolenbroek void
290*00b67f09SDavid van Moolenbroek isc__mempool_associatelock(isc_mempool_t *mpctx, isc_mutex_t *lock);
291*00b67f09SDavid van Moolenbroek void *
292*00b67f09SDavid van Moolenbroek isc___mempool_get(isc_mempool_t *mpctx FLARG);
293*00b67f09SDavid van Moolenbroek void
294*00b67f09SDavid van Moolenbroek isc___mempool_put(isc_mempool_t *mpctx, void *mem FLARG);
295*00b67f09SDavid van Moolenbroek void
296*00b67f09SDavid van Moolenbroek isc__mempool_setfreemax(isc_mempool_t *mpctx, unsigned int limit);
297*00b67f09SDavid van Moolenbroek unsigned int
298*00b67f09SDavid van Moolenbroek isc__mempool_getfreemax(isc_mempool_t *mpctx);
299*00b67f09SDavid van Moolenbroek unsigned int
300*00b67f09SDavid van Moolenbroek isc__mempool_getfreecount(isc_mempool_t *mpctx);
301*00b67f09SDavid van Moolenbroek void
302*00b67f09SDavid van Moolenbroek isc__mempool_setmaxalloc(isc_mempool_t *mpctx, unsigned int limit);
303*00b67f09SDavid van Moolenbroek unsigned int
304*00b67f09SDavid van Moolenbroek isc__mempool_getmaxalloc(isc_mempool_t *mpctx);
305*00b67f09SDavid van Moolenbroek unsigned int
306*00b67f09SDavid van Moolenbroek isc__mempool_getallocated(isc_mempool_t *mpctx);
307*00b67f09SDavid van Moolenbroek void
308*00b67f09SDavid van Moolenbroek isc__mempool_setfillcount(isc_mempool_t *mpctx, unsigned int limit);
309*00b67f09SDavid van Moolenbroek unsigned int
310*00b67f09SDavid van Moolenbroek isc__mempool_getfillcount(isc_mempool_t *mpctx);
311*00b67f09SDavid van Moolenbroek void
312*00b67f09SDavid van Moolenbroek isc__mem_printactive(isc_mem_t *ctx0, FILE *file);
313*00b67f09SDavid van Moolenbroek void
314*00b67f09SDavid van Moolenbroek isc__mem_printallactive(FILE *file);
315*00b67f09SDavid van Moolenbroek unsigned int
316*00b67f09SDavid van Moolenbroek isc__mem_references(isc_mem_t *ctx0);
317*00b67f09SDavid van Moolenbroek #endif /* ISC_MEM_TRACKLINES */
318*00b67f09SDavid van Moolenbroek 
319*00b67f09SDavid van Moolenbroek static struct isc__memmethods {
320*00b67f09SDavid van Moolenbroek 	isc_memmethods_t methods;
321*00b67f09SDavid van Moolenbroek 
322*00b67f09SDavid van Moolenbroek 	/*%
323*00b67f09SDavid van Moolenbroek 	 * The following are defined just for avoiding unused static functions.
324*00b67f09SDavid van Moolenbroek 	 */
325*00b67f09SDavid van Moolenbroek 	void *createx, *create, *create2, *ondestroy, *stats,
326*00b67f09SDavid van Moolenbroek 	     *setquota, *getquota, *setname, *getname, *gettag;
327*00b67f09SDavid van Moolenbroek } memmethods = {
328*00b67f09SDavid van Moolenbroek 	{
329*00b67f09SDavid van Moolenbroek 		isc__mem_attach,
330*00b67f09SDavid van Moolenbroek 		isc__mem_detach,
331*00b67f09SDavid van Moolenbroek 		isc__mem_destroy,
332*00b67f09SDavid van Moolenbroek 		isc___mem_get,
333*00b67f09SDavid van Moolenbroek 		isc___mem_put,
334*00b67f09SDavid van Moolenbroek 		isc___mem_putanddetach,
335*00b67f09SDavid van Moolenbroek 		isc___mem_allocate,
336*00b67f09SDavid van Moolenbroek 		isc___mem_reallocate,
337*00b67f09SDavid van Moolenbroek 		isc___mem_strdup,
338*00b67f09SDavid van Moolenbroek 		isc___mem_free,
339*00b67f09SDavid van Moolenbroek 		isc__mem_setdestroycheck,
340*00b67f09SDavid van Moolenbroek 		isc__mem_setwater,
341*00b67f09SDavid van Moolenbroek 		isc__mem_waterack,
342*00b67f09SDavid van Moolenbroek 		isc__mem_inuse,
343*00b67f09SDavid van Moolenbroek 		isc__mem_maxinuse,
344*00b67f09SDavid van Moolenbroek 		isc__mem_total,
345*00b67f09SDavid van Moolenbroek 		isc__mem_isovermem,
346*00b67f09SDavid van Moolenbroek 		isc__mempool_create
347*00b67f09SDavid van Moolenbroek 	},
348*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_createx,
349*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_create,
350*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_create2,
351*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_ondestroy,
352*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_stats,
353*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_setquota,
354*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_getquota,
355*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_setname,
356*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_getname,
357*00b67f09SDavid van Moolenbroek 	(void *)isc_mem_gettag
358*00b67f09SDavid van Moolenbroek };
359*00b67f09SDavid van Moolenbroek 
360*00b67f09SDavid van Moolenbroek static struct isc__mempoolmethods {
361*00b67f09SDavid van Moolenbroek 	isc_mempoolmethods_t methods;
362*00b67f09SDavid van Moolenbroek 
363*00b67f09SDavid van Moolenbroek 	/*%
364*00b67f09SDavid van Moolenbroek 	 * The following are defined just for avoiding unused static functions.
365*00b67f09SDavid van Moolenbroek 	 */
366*00b67f09SDavid van Moolenbroek 	void *getfreemax, *getfreecount, *getmaxalloc, *getfillcount;
367*00b67f09SDavid van Moolenbroek } mempoolmethods = {
368*00b67f09SDavid van Moolenbroek 	{
369*00b67f09SDavid van Moolenbroek 		isc__mempool_destroy,
370*00b67f09SDavid van Moolenbroek 		isc___mempool_get,
371*00b67f09SDavid van Moolenbroek 		isc___mempool_put,
372*00b67f09SDavid van Moolenbroek 		isc__mempool_getallocated,
373*00b67f09SDavid van Moolenbroek 		isc__mempool_setmaxalloc,
374*00b67f09SDavid van Moolenbroek 		isc__mempool_setfreemax,
375*00b67f09SDavid van Moolenbroek 		isc__mempool_setname,
376*00b67f09SDavid van Moolenbroek 		isc__mempool_associatelock,
377*00b67f09SDavid van Moolenbroek 		isc__mempool_setfillcount
378*00b67f09SDavid van Moolenbroek 	},
379*00b67f09SDavid van Moolenbroek 	(void *)isc_mempool_getfreemax,
380*00b67f09SDavid van Moolenbroek 	(void *)isc_mempool_getfreecount,
381*00b67f09SDavid van Moolenbroek 	(void *)isc_mempool_getmaxalloc,
382*00b67f09SDavid van Moolenbroek 	(void *)isc_mempool_getfillcount
383*00b67f09SDavid van Moolenbroek };
384*00b67f09SDavid van Moolenbroek 
385*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
386*00b67f09SDavid van Moolenbroek /*!
387*00b67f09SDavid van Moolenbroek  * mctx must be locked.
388*00b67f09SDavid van Moolenbroek  */
389*00b67f09SDavid van Moolenbroek static inline void
add_trace_entry(isc__mem_t * mctx,const void * ptr,size_t size FLARG)390*00b67f09SDavid van Moolenbroek add_trace_entry(isc__mem_t *mctx, const void *ptr, size_t size FLARG) {
391*00b67f09SDavid van Moolenbroek 	debuglink_t *dl;
392*00b67f09SDavid van Moolenbroek 	unsigned int i;
393*00b67f09SDavid van Moolenbroek 	size_t mysize = size;
394*00b67f09SDavid van Moolenbroek 
395*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGTRACE) != 0)
396*00b67f09SDavid van Moolenbroek 		fprintf(stderr, isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
397*00b67f09SDavid van Moolenbroek 					       ISC_MSG_ADDTRACE,
398*00b67f09SDavid van Moolenbroek 					       "add %p size %u "
399*00b67f09SDavid van Moolenbroek 					       "file %s line %u mctx %p\n"),
400*00b67f09SDavid van Moolenbroek 			ptr, size, file, line, mctx);
401*00b67f09SDavid van Moolenbroek 
402*00b67f09SDavid van Moolenbroek 	if (mctx->debuglist == NULL)
403*00b67f09SDavid van Moolenbroek 		return;
404*00b67f09SDavid van Moolenbroek 
405*00b67f09SDavid van Moolenbroek 	if (mysize > mctx->max_size)
406*00b67f09SDavid van Moolenbroek 		mysize = mctx->max_size;
407*00b67f09SDavid van Moolenbroek 
408*00b67f09SDavid van Moolenbroek 	dl = ISC_LIST_HEAD(mctx->debuglist[mysize]);
409*00b67f09SDavid van Moolenbroek 	while (dl != NULL) {
410*00b67f09SDavid van Moolenbroek 		if (dl->count == DEBUGLIST_COUNT)
411*00b67f09SDavid van Moolenbroek 			goto next;
412*00b67f09SDavid van Moolenbroek 		for (i = 0; i < DEBUGLIST_COUNT; i++) {
413*00b67f09SDavid van Moolenbroek 			if (dl->ptr[i] == NULL) {
414*00b67f09SDavid van Moolenbroek 				dl->ptr[i] = ptr;
415*00b67f09SDavid van Moolenbroek 				dl->size[i] = size;
416*00b67f09SDavid van Moolenbroek 				dl->file[i] = file;
417*00b67f09SDavid van Moolenbroek 				dl->line[i] = line;
418*00b67f09SDavid van Moolenbroek 				dl->count++;
419*00b67f09SDavid van Moolenbroek 				return;
420*00b67f09SDavid van Moolenbroek 			}
421*00b67f09SDavid van Moolenbroek 		}
422*00b67f09SDavid van Moolenbroek 	next:
423*00b67f09SDavid van Moolenbroek 		dl = ISC_LIST_NEXT(dl, link);
424*00b67f09SDavid van Moolenbroek 	}
425*00b67f09SDavid van Moolenbroek 
426*00b67f09SDavid van Moolenbroek 	dl = malloc(sizeof(debuglink_t));
427*00b67f09SDavid van Moolenbroek 	INSIST(dl != NULL);
428*00b67f09SDavid van Moolenbroek 
429*00b67f09SDavid van Moolenbroek 	ISC_LINK_INIT(dl, link);
430*00b67f09SDavid van Moolenbroek 	for (i = 1; i < DEBUGLIST_COUNT; i++) {
431*00b67f09SDavid van Moolenbroek 		dl->ptr[i] = NULL;
432*00b67f09SDavid van Moolenbroek 		dl->size[i] = 0;
433*00b67f09SDavid van Moolenbroek 		dl->file[i] = NULL;
434*00b67f09SDavid van Moolenbroek 		dl->line[i] = 0;
435*00b67f09SDavid van Moolenbroek 	}
436*00b67f09SDavid van Moolenbroek 
437*00b67f09SDavid van Moolenbroek 	dl->ptr[0] = ptr;
438*00b67f09SDavid van Moolenbroek 	dl->size[0] = size;
439*00b67f09SDavid van Moolenbroek 	dl->file[0] = file;
440*00b67f09SDavid van Moolenbroek 	dl->line[0] = line;
441*00b67f09SDavid van Moolenbroek 	dl->count = 1;
442*00b67f09SDavid van Moolenbroek 
443*00b67f09SDavid van Moolenbroek 	ISC_LIST_PREPEND(mctx->debuglist[mysize], dl, link);
444*00b67f09SDavid van Moolenbroek 	mctx->debuglistcnt++;
445*00b67f09SDavid van Moolenbroek }
446*00b67f09SDavid van Moolenbroek 
447*00b67f09SDavid van Moolenbroek static inline void
delete_trace_entry(isc__mem_t * mctx,const void * ptr,size_t size,const char * file,unsigned int line)448*00b67f09SDavid van Moolenbroek delete_trace_entry(isc__mem_t *mctx, const void *ptr, size_t size,
449*00b67f09SDavid van Moolenbroek 		   const char *file, unsigned int line)
450*00b67f09SDavid van Moolenbroek {
451*00b67f09SDavid van Moolenbroek 	debuglink_t *dl;
452*00b67f09SDavid van Moolenbroek 	unsigned int i;
453*00b67f09SDavid van Moolenbroek 
454*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGTRACE) != 0)
455*00b67f09SDavid van Moolenbroek 		fprintf(stderr, isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
456*00b67f09SDavid van Moolenbroek 					       ISC_MSG_DELTRACE,
457*00b67f09SDavid van Moolenbroek 					       "del %p size %u "
458*00b67f09SDavid van Moolenbroek 					       "file %s line %u mctx %p\n"),
459*00b67f09SDavid van Moolenbroek 			ptr, size, file, line, mctx);
460*00b67f09SDavid van Moolenbroek 
461*00b67f09SDavid van Moolenbroek 	if (mctx->debuglist == NULL)
462*00b67f09SDavid van Moolenbroek 		return;
463*00b67f09SDavid van Moolenbroek 
464*00b67f09SDavid van Moolenbroek 	if (size > mctx->max_size)
465*00b67f09SDavid van Moolenbroek 		size = mctx->max_size;
466*00b67f09SDavid van Moolenbroek 
467*00b67f09SDavid van Moolenbroek 	dl = ISC_LIST_HEAD(mctx->debuglist[size]);
468*00b67f09SDavid van Moolenbroek 	while (dl != NULL) {
469*00b67f09SDavid van Moolenbroek 		for (i = 0; i < DEBUGLIST_COUNT; i++) {
470*00b67f09SDavid van Moolenbroek 			if (dl->ptr[i] == ptr) {
471*00b67f09SDavid van Moolenbroek 				dl->ptr[i] = NULL;
472*00b67f09SDavid van Moolenbroek 				dl->size[i] = 0;
473*00b67f09SDavid van Moolenbroek 				dl->file[i] = NULL;
474*00b67f09SDavid van Moolenbroek 				dl->line[i] = 0;
475*00b67f09SDavid van Moolenbroek 
476*00b67f09SDavid van Moolenbroek 				INSIST(dl->count > 0);
477*00b67f09SDavid van Moolenbroek 				dl->count--;
478*00b67f09SDavid van Moolenbroek 				if (dl->count == 0) {
479*00b67f09SDavid van Moolenbroek 					ISC_LIST_UNLINK(mctx->debuglist[size],
480*00b67f09SDavid van Moolenbroek 							dl, link);
481*00b67f09SDavid van Moolenbroek 					free(dl);
482*00b67f09SDavid van Moolenbroek 				}
483*00b67f09SDavid van Moolenbroek 				return;
484*00b67f09SDavid van Moolenbroek 			}
485*00b67f09SDavid van Moolenbroek 		}
486*00b67f09SDavid van Moolenbroek 		dl = ISC_LIST_NEXT(dl, link);
487*00b67f09SDavid van Moolenbroek 	}
488*00b67f09SDavid van Moolenbroek 
489*00b67f09SDavid van Moolenbroek 	/*
490*00b67f09SDavid van Moolenbroek 	 * If we get here, we didn't find the item on the list.  We're
491*00b67f09SDavid van Moolenbroek 	 * screwed.
492*00b67f09SDavid van Moolenbroek 	 */
493*00b67f09SDavid van Moolenbroek 	INSIST(dl != NULL);
494*00b67f09SDavid van Moolenbroek }
495*00b67f09SDavid van Moolenbroek #endif /* ISC_MEM_TRACKLINES */
496*00b67f09SDavid van Moolenbroek 
497*00b67f09SDavid van Moolenbroek static inline size_t
rmsize(size_t size)498*00b67f09SDavid van Moolenbroek rmsize(size_t size) {
499*00b67f09SDavid van Moolenbroek 	/*
500*00b67f09SDavid van Moolenbroek 	 * round down to ALIGNMENT_SIZE
501*00b67f09SDavid van Moolenbroek 	 */
502*00b67f09SDavid van Moolenbroek 	return (size & (~(ALIGNMENT_SIZE - 1)));
503*00b67f09SDavid van Moolenbroek }
504*00b67f09SDavid van Moolenbroek 
505*00b67f09SDavid van Moolenbroek static inline size_t
quantize(size_t size)506*00b67f09SDavid van Moolenbroek quantize(size_t size) {
507*00b67f09SDavid van Moolenbroek 	/*!
508*00b67f09SDavid van Moolenbroek 	 * Round up the result in order to get a size big
509*00b67f09SDavid van Moolenbroek 	 * enough to satisfy the request and be aligned on ALIGNMENT_SIZE
510*00b67f09SDavid van Moolenbroek 	 * byte boundaries.
511*00b67f09SDavid van Moolenbroek 	 */
512*00b67f09SDavid van Moolenbroek 
513*00b67f09SDavid van Moolenbroek 	if (size == 0U)
514*00b67f09SDavid van Moolenbroek 		return (ALIGNMENT_SIZE);
515*00b67f09SDavid van Moolenbroek 	return ((size + ALIGNMENT_SIZE - 1) & (~(ALIGNMENT_SIZE - 1)));
516*00b67f09SDavid van Moolenbroek }
517*00b67f09SDavid van Moolenbroek 
518*00b67f09SDavid van Moolenbroek static inline isc_boolean_t
more_basic_blocks(isc__mem_t * ctx)519*00b67f09SDavid van Moolenbroek more_basic_blocks(isc__mem_t *ctx) {
520*00b67f09SDavid van Moolenbroek 	void *new;
521*00b67f09SDavid van Moolenbroek 	unsigned char *curr, *next;
522*00b67f09SDavid van Moolenbroek 	unsigned char *first, *last;
523*00b67f09SDavid van Moolenbroek 	unsigned char **table;
524*00b67f09SDavid van Moolenbroek 	unsigned int table_size;
525*00b67f09SDavid van Moolenbroek 	size_t increment;
526*00b67f09SDavid van Moolenbroek 	int i;
527*00b67f09SDavid van Moolenbroek 
528*00b67f09SDavid van Moolenbroek 	/* Require: we hold the context lock. */
529*00b67f09SDavid van Moolenbroek 
530*00b67f09SDavid van Moolenbroek 	/*
531*00b67f09SDavid van Moolenbroek 	 * Did we hit the quota for this context?
532*00b67f09SDavid van Moolenbroek 	 */
533*00b67f09SDavid van Moolenbroek 	increment = NUM_BASIC_BLOCKS * ctx->mem_target;
534*00b67f09SDavid van Moolenbroek 	if (ctx->quota != 0U && ctx->total + increment > ctx->quota)
535*00b67f09SDavid van Moolenbroek 		return (ISC_FALSE);
536*00b67f09SDavid van Moolenbroek 
537*00b67f09SDavid van Moolenbroek 	INSIST(ctx->basic_table_count <= ctx->basic_table_size);
538*00b67f09SDavid van Moolenbroek 	if (ctx->basic_table_count == ctx->basic_table_size) {
539*00b67f09SDavid van Moolenbroek 		table_size = ctx->basic_table_size + TABLE_INCREMENT;
540*00b67f09SDavid van Moolenbroek 		table = (ctx->memalloc)(ctx->arg,
541*00b67f09SDavid van Moolenbroek 					table_size * sizeof(unsigned char *));
542*00b67f09SDavid van Moolenbroek 		if (table == NULL) {
543*00b67f09SDavid van Moolenbroek 			ctx->memalloc_failures++;
544*00b67f09SDavid van Moolenbroek 			return (ISC_FALSE);
545*00b67f09SDavid van Moolenbroek 		}
546*00b67f09SDavid van Moolenbroek 		if (ctx->basic_table_size != 0) {
547*00b67f09SDavid van Moolenbroek 			memmove(table, ctx->basic_table,
548*00b67f09SDavid van Moolenbroek 				ctx->basic_table_size *
549*00b67f09SDavid van Moolenbroek 				  sizeof(unsigned char *));
550*00b67f09SDavid van Moolenbroek 			(ctx->memfree)(ctx->arg, ctx->basic_table);
551*00b67f09SDavid van Moolenbroek 		}
552*00b67f09SDavid van Moolenbroek 		ctx->basic_table = table;
553*00b67f09SDavid van Moolenbroek 		ctx->basic_table_size = table_size;
554*00b67f09SDavid van Moolenbroek 	}
555*00b67f09SDavid van Moolenbroek 
556*00b67f09SDavid van Moolenbroek 	new = (ctx->memalloc)(ctx->arg, NUM_BASIC_BLOCKS * ctx->mem_target);
557*00b67f09SDavid van Moolenbroek 	if (new == NULL) {
558*00b67f09SDavid van Moolenbroek 		ctx->memalloc_failures++;
559*00b67f09SDavid van Moolenbroek 		return (ISC_FALSE);
560*00b67f09SDavid van Moolenbroek 	}
561*00b67f09SDavid van Moolenbroek 	ctx->total += increment;
562*00b67f09SDavid van Moolenbroek 	ctx->basic_table[ctx->basic_table_count] = new;
563*00b67f09SDavid van Moolenbroek 	ctx->basic_table_count++;
564*00b67f09SDavid van Moolenbroek 
565*00b67f09SDavid van Moolenbroek 	curr = new;
566*00b67f09SDavid van Moolenbroek 	next = curr + ctx->mem_target;
567*00b67f09SDavid van Moolenbroek 	for (i = 0; i < (NUM_BASIC_BLOCKS - 1); i++) {
568*00b67f09SDavid van Moolenbroek 		((element *)curr)->next = (element *)next;
569*00b67f09SDavid van Moolenbroek 		curr = next;
570*00b67f09SDavid van Moolenbroek 		next += ctx->mem_target;
571*00b67f09SDavid van Moolenbroek 	}
572*00b67f09SDavid van Moolenbroek 	/*
573*00b67f09SDavid van Moolenbroek 	 * curr is now pointing at the last block in the
574*00b67f09SDavid van Moolenbroek 	 * array.
575*00b67f09SDavid van Moolenbroek 	 */
576*00b67f09SDavid van Moolenbroek 	((element *)curr)->next = NULL;
577*00b67f09SDavid van Moolenbroek 	first = new;
578*00b67f09SDavid van Moolenbroek 	last = first + NUM_BASIC_BLOCKS * ctx->mem_target - 1;
579*00b67f09SDavid van Moolenbroek 	if (first < ctx->lowest || ctx->lowest == NULL)
580*00b67f09SDavid van Moolenbroek 		ctx->lowest = first;
581*00b67f09SDavid van Moolenbroek 	if (last > ctx->highest)
582*00b67f09SDavid van Moolenbroek 		ctx->highest = last;
583*00b67f09SDavid van Moolenbroek 	ctx->basic_blocks = new;
584*00b67f09SDavid van Moolenbroek 
585*00b67f09SDavid van Moolenbroek 	return (ISC_TRUE);
586*00b67f09SDavid van Moolenbroek }
587*00b67f09SDavid van Moolenbroek 
588*00b67f09SDavid van Moolenbroek static inline isc_boolean_t
more_frags(isc__mem_t * ctx,size_t new_size)589*00b67f09SDavid van Moolenbroek more_frags(isc__mem_t *ctx, size_t new_size) {
590*00b67f09SDavid van Moolenbroek 	int i, frags;
591*00b67f09SDavid van Moolenbroek 	size_t total_size;
592*00b67f09SDavid van Moolenbroek 	void *new;
593*00b67f09SDavid van Moolenbroek 	unsigned char *curr, *next;
594*00b67f09SDavid van Moolenbroek 
595*00b67f09SDavid van Moolenbroek 	/*!
596*00b67f09SDavid van Moolenbroek 	 * Try to get more fragments by chopping up a basic block.
597*00b67f09SDavid van Moolenbroek 	 */
598*00b67f09SDavid van Moolenbroek 
599*00b67f09SDavid van Moolenbroek 	if (ctx->basic_blocks == NULL) {
600*00b67f09SDavid van Moolenbroek 		if (!more_basic_blocks(ctx)) {
601*00b67f09SDavid van Moolenbroek 			/*
602*00b67f09SDavid van Moolenbroek 			 * We can't get more memory from the OS, or we've
603*00b67f09SDavid van Moolenbroek 			 * hit the quota for this context.
604*00b67f09SDavid van Moolenbroek 			 */
605*00b67f09SDavid van Moolenbroek 			/*
606*00b67f09SDavid van Moolenbroek 			 * XXXRTH  "At quota" notification here.
607*00b67f09SDavid van Moolenbroek 			 */
608*00b67f09SDavid van Moolenbroek 			return (ISC_FALSE);
609*00b67f09SDavid van Moolenbroek 		}
610*00b67f09SDavid van Moolenbroek 	}
611*00b67f09SDavid van Moolenbroek 
612*00b67f09SDavid van Moolenbroek 	total_size = ctx->mem_target;
613*00b67f09SDavid van Moolenbroek 	new = ctx->basic_blocks;
614*00b67f09SDavid van Moolenbroek 	ctx->basic_blocks = ctx->basic_blocks->next;
615*00b67f09SDavid van Moolenbroek 	frags = (int)(total_size / new_size);
616*00b67f09SDavid van Moolenbroek 	ctx->stats[new_size].blocks++;
617*00b67f09SDavid van Moolenbroek 	ctx->stats[new_size].freefrags += frags;
618*00b67f09SDavid van Moolenbroek 	/*
619*00b67f09SDavid van Moolenbroek 	 * Set up a linked-list of blocks of size
620*00b67f09SDavid van Moolenbroek 	 * "new_size".
621*00b67f09SDavid van Moolenbroek 	 */
622*00b67f09SDavid van Moolenbroek 	curr = new;
623*00b67f09SDavid van Moolenbroek 	next = curr + new_size;
624*00b67f09SDavid van Moolenbroek 	total_size -= new_size;
625*00b67f09SDavid van Moolenbroek 	for (i = 0; i < (frags - 1); i++) {
626*00b67f09SDavid van Moolenbroek 		((element *)curr)->next = (element *)next;
627*00b67f09SDavid van Moolenbroek 		curr = next;
628*00b67f09SDavid van Moolenbroek 		next += new_size;
629*00b67f09SDavid van Moolenbroek 		total_size -= new_size;
630*00b67f09SDavid van Moolenbroek 	}
631*00b67f09SDavid van Moolenbroek 	/*
632*00b67f09SDavid van Moolenbroek 	 * Add the remaining fragment of the basic block to a free list.
633*00b67f09SDavid van Moolenbroek 	 */
634*00b67f09SDavid van Moolenbroek 	total_size = rmsize(total_size);
635*00b67f09SDavid van Moolenbroek 	if (total_size > 0U) {
636*00b67f09SDavid van Moolenbroek 		((element *)next)->next = ctx->freelists[total_size];
637*00b67f09SDavid van Moolenbroek 		ctx->freelists[total_size] = (element *)next;
638*00b67f09SDavid van Moolenbroek 		ctx->stats[total_size].freefrags++;
639*00b67f09SDavid van Moolenbroek 	}
640*00b67f09SDavid van Moolenbroek 	/*
641*00b67f09SDavid van Moolenbroek 	 * curr is now pointing at the last block in the
642*00b67f09SDavid van Moolenbroek 	 * array.
643*00b67f09SDavid van Moolenbroek 	 */
644*00b67f09SDavid van Moolenbroek 	((element *)curr)->next = NULL;
645*00b67f09SDavid van Moolenbroek 	ctx->freelists[new_size] = new;
646*00b67f09SDavid van Moolenbroek 
647*00b67f09SDavid van Moolenbroek 	return (ISC_TRUE);
648*00b67f09SDavid van Moolenbroek }
649*00b67f09SDavid van Moolenbroek 
650*00b67f09SDavid van Moolenbroek static inline void *
mem_getunlocked(isc__mem_t * ctx,size_t size)651*00b67f09SDavid van Moolenbroek mem_getunlocked(isc__mem_t *ctx, size_t size) {
652*00b67f09SDavid van Moolenbroek 	size_t new_size = quantize(size);
653*00b67f09SDavid van Moolenbroek 	void *ret;
654*00b67f09SDavid van Moolenbroek 
655*00b67f09SDavid van Moolenbroek 	if (size >= ctx->max_size || new_size >= ctx->max_size) {
656*00b67f09SDavid van Moolenbroek 		/*
657*00b67f09SDavid van Moolenbroek 		 * memget() was called on something beyond our upper limit.
658*00b67f09SDavid van Moolenbroek 		 */
659*00b67f09SDavid van Moolenbroek 		if (ctx->quota != 0U && ctx->total + size > ctx->quota) {
660*00b67f09SDavid van Moolenbroek 			ret = NULL;
661*00b67f09SDavid van Moolenbroek 			goto done;
662*00b67f09SDavid van Moolenbroek 		}
663*00b67f09SDavid van Moolenbroek 		ret = (ctx->memalloc)(ctx->arg, size);
664*00b67f09SDavid van Moolenbroek 		if (ret == NULL) {
665*00b67f09SDavid van Moolenbroek 			ctx->memalloc_failures++;
666*00b67f09SDavid van Moolenbroek 			goto done;
667*00b67f09SDavid van Moolenbroek 		}
668*00b67f09SDavid van Moolenbroek 		ctx->total += size;
669*00b67f09SDavid van Moolenbroek 		ctx->inuse += size;
670*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].gets++;
671*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].totalgets++;
672*00b67f09SDavid van Moolenbroek 		/*
673*00b67f09SDavid van Moolenbroek 		 * If we don't set new_size to size, then the
674*00b67f09SDavid van Moolenbroek 		 * ISC_MEM_FILL code might write over bytes we
675*00b67f09SDavid van Moolenbroek 		 * don't own.
676*00b67f09SDavid van Moolenbroek 		 */
677*00b67f09SDavid van Moolenbroek 		new_size = size;
678*00b67f09SDavid van Moolenbroek 		goto done;
679*00b67f09SDavid van Moolenbroek 	}
680*00b67f09SDavid van Moolenbroek 
681*00b67f09SDavid van Moolenbroek 	/*
682*00b67f09SDavid van Moolenbroek 	 * If there are no blocks in the free list for this size, get a chunk
683*00b67f09SDavid van Moolenbroek 	 * of memory and then break it up into "new_size"-sized blocks, adding
684*00b67f09SDavid van Moolenbroek 	 * them to the free list.
685*00b67f09SDavid van Moolenbroek 	 */
686*00b67f09SDavid van Moolenbroek 	if (ctx->freelists[new_size] == NULL && !more_frags(ctx, new_size))
687*00b67f09SDavid van Moolenbroek 		return (NULL);
688*00b67f09SDavid van Moolenbroek 
689*00b67f09SDavid van Moolenbroek 	/*
690*00b67f09SDavid van Moolenbroek 	 * The free list uses the "rounded-up" size "new_size".
691*00b67f09SDavid van Moolenbroek 	 */
692*00b67f09SDavid van Moolenbroek 	ret = ctx->freelists[new_size];
693*00b67f09SDavid van Moolenbroek 	ctx->freelists[new_size] = ctx->freelists[new_size]->next;
694*00b67f09SDavid van Moolenbroek 
695*00b67f09SDavid van Moolenbroek 	/*
696*00b67f09SDavid van Moolenbroek 	 * The stats[] uses the _actual_ "size" requested by the
697*00b67f09SDavid van Moolenbroek 	 * caller, with the caveat (in the code above) that "size" >= the
698*00b67f09SDavid van Moolenbroek 	 * max. size (max_size) ends up getting recorded as a call to
699*00b67f09SDavid van Moolenbroek 	 * max_size.
700*00b67f09SDavid van Moolenbroek 	 */
701*00b67f09SDavid van Moolenbroek 	ctx->stats[size].gets++;
702*00b67f09SDavid van Moolenbroek 	ctx->stats[size].totalgets++;
703*00b67f09SDavid van Moolenbroek 	ctx->stats[new_size].freefrags--;
704*00b67f09SDavid van Moolenbroek 	ctx->inuse += new_size;
705*00b67f09SDavid van Moolenbroek 
706*00b67f09SDavid van Moolenbroek  done:
707*00b67f09SDavid van Moolenbroek 
708*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL
709*00b67f09SDavid van Moolenbroek 	if (ret != NULL)
710*00b67f09SDavid van Moolenbroek 		memset(ret, 0xbe, new_size); /* Mnemonic for "beef". */
711*00b67f09SDavid van Moolenbroek #endif
712*00b67f09SDavid van Moolenbroek 
713*00b67f09SDavid van Moolenbroek 	return (ret);
714*00b67f09SDavid van Moolenbroek }
715*00b67f09SDavid van Moolenbroek 
716*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL && ISC_MEM_CHECKOVERRUN
717*00b67f09SDavid van Moolenbroek static inline void
check_overrun(void * mem,size_t size,size_t new_size)718*00b67f09SDavid van Moolenbroek check_overrun(void *mem, size_t size, size_t new_size) {
719*00b67f09SDavid van Moolenbroek 	unsigned char *cp;
720*00b67f09SDavid van Moolenbroek 
721*00b67f09SDavid van Moolenbroek 	cp = (unsigned char *)mem;
722*00b67f09SDavid van Moolenbroek 	cp += size;
723*00b67f09SDavid van Moolenbroek 	while (size < new_size) {
724*00b67f09SDavid van Moolenbroek 		INSIST(*cp == 0xbe);
725*00b67f09SDavid van Moolenbroek 		cp++;
726*00b67f09SDavid van Moolenbroek 		size++;
727*00b67f09SDavid van Moolenbroek 	}
728*00b67f09SDavid van Moolenbroek }
729*00b67f09SDavid van Moolenbroek #endif
730*00b67f09SDavid van Moolenbroek 
731*00b67f09SDavid van Moolenbroek /* coverity[+free : arg-1] */
732*00b67f09SDavid van Moolenbroek static inline void
mem_putunlocked(isc__mem_t * ctx,void * mem,size_t size)733*00b67f09SDavid van Moolenbroek mem_putunlocked(isc__mem_t *ctx, void *mem, size_t size) {
734*00b67f09SDavid van Moolenbroek 	size_t new_size = quantize(size);
735*00b67f09SDavid van Moolenbroek 
736*00b67f09SDavid van Moolenbroek 	if (size == ctx->max_size || new_size >= ctx->max_size) {
737*00b67f09SDavid van Moolenbroek 		/*
738*00b67f09SDavid van Moolenbroek 		 * memput() called on something beyond our upper limit.
739*00b67f09SDavid van Moolenbroek 		 */
740*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL
741*00b67f09SDavid van Moolenbroek 		memset(mem, 0xde, size); /* Mnemonic for "dead". */
742*00b67f09SDavid van Moolenbroek #endif
743*00b67f09SDavid van Moolenbroek 		(ctx->memfree)(ctx->arg, mem);
744*00b67f09SDavid van Moolenbroek 		INSIST(ctx->stats[ctx->max_size].gets != 0U);
745*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].gets--;
746*00b67f09SDavid van Moolenbroek 		INSIST(size <= ctx->inuse);
747*00b67f09SDavid van Moolenbroek 		ctx->inuse -= size;
748*00b67f09SDavid van Moolenbroek 		return;
749*00b67f09SDavid van Moolenbroek 	}
750*00b67f09SDavid van Moolenbroek 
751*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL
752*00b67f09SDavid van Moolenbroek #if ISC_MEM_CHECKOVERRUN
753*00b67f09SDavid van Moolenbroek 	check_overrun(mem, size, new_size);
754*00b67f09SDavid van Moolenbroek #endif
755*00b67f09SDavid van Moolenbroek 	memset(mem, 0xde, new_size); /* Mnemonic for "dead". */
756*00b67f09SDavid van Moolenbroek #endif
757*00b67f09SDavid van Moolenbroek 
758*00b67f09SDavid van Moolenbroek 	/*
759*00b67f09SDavid van Moolenbroek 	 * The free list uses the "rounded-up" size "new_size".
760*00b67f09SDavid van Moolenbroek 	 */
761*00b67f09SDavid van Moolenbroek 	((element *)mem)->next = ctx->freelists[new_size];
762*00b67f09SDavid van Moolenbroek 	ctx->freelists[new_size] = (element *)mem;
763*00b67f09SDavid van Moolenbroek 
764*00b67f09SDavid van Moolenbroek 	/*
765*00b67f09SDavid van Moolenbroek 	 * The stats[] uses the _actual_ "size" requested by the
766*00b67f09SDavid van Moolenbroek 	 * caller, with the caveat (in the code above) that "size" >= the
767*00b67f09SDavid van Moolenbroek 	 * max. size (max_size) ends up getting recorded as a call to
768*00b67f09SDavid van Moolenbroek 	 * max_size.
769*00b67f09SDavid van Moolenbroek 	 */
770*00b67f09SDavid van Moolenbroek 	INSIST(ctx->stats[size].gets != 0U);
771*00b67f09SDavid van Moolenbroek 	ctx->stats[size].gets--;
772*00b67f09SDavid van Moolenbroek 	ctx->stats[new_size].freefrags++;
773*00b67f09SDavid van Moolenbroek 	ctx->inuse -= new_size;
774*00b67f09SDavid van Moolenbroek }
775*00b67f09SDavid van Moolenbroek 
776*00b67f09SDavid van Moolenbroek /*!
777*00b67f09SDavid van Moolenbroek  * Perform a malloc, doing memory filling and overrun detection as necessary.
778*00b67f09SDavid van Moolenbroek  */
779*00b67f09SDavid van Moolenbroek static inline void *
mem_get(isc__mem_t * ctx,size_t size)780*00b67f09SDavid van Moolenbroek mem_get(isc__mem_t *ctx, size_t size) {
781*00b67f09SDavid van Moolenbroek 	char *ret;
782*00b67f09SDavid van Moolenbroek 
783*00b67f09SDavid van Moolenbroek #if ISC_MEM_CHECKOVERRUN
784*00b67f09SDavid van Moolenbroek 	size += 1;
785*00b67f09SDavid van Moolenbroek #endif
786*00b67f09SDavid van Moolenbroek 
787*00b67f09SDavid van Moolenbroek 	ret = (ctx->memalloc)(ctx->arg, size);
788*00b67f09SDavid van Moolenbroek 	if (ret == NULL)
789*00b67f09SDavid van Moolenbroek 		ctx->memalloc_failures++;
790*00b67f09SDavid van Moolenbroek 
791*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL
792*00b67f09SDavid van Moolenbroek 	if (ret != NULL)
793*00b67f09SDavid van Moolenbroek 		memset(ret, 0xbe, size); /* Mnemonic for "beef". */
794*00b67f09SDavid van Moolenbroek #else
795*00b67f09SDavid van Moolenbroek #  if ISC_MEM_CHECKOVERRUN
796*00b67f09SDavid van Moolenbroek 	if (ret != NULL)
797*00b67f09SDavid van Moolenbroek 		ret[size-1] = 0xbe;
798*00b67f09SDavid van Moolenbroek #  endif
799*00b67f09SDavid van Moolenbroek #endif
800*00b67f09SDavid van Moolenbroek 
801*00b67f09SDavid van Moolenbroek 	return (ret);
802*00b67f09SDavid van Moolenbroek }
803*00b67f09SDavid van Moolenbroek 
804*00b67f09SDavid van Moolenbroek /*!
805*00b67f09SDavid van Moolenbroek  * Perform a free, doing memory filling and overrun detection as necessary.
806*00b67f09SDavid van Moolenbroek  */
807*00b67f09SDavid van Moolenbroek /* coverity[+free : arg-1] */
808*00b67f09SDavid van Moolenbroek static inline void
mem_put(isc__mem_t * ctx,void * mem,size_t size)809*00b67f09SDavid van Moolenbroek mem_put(isc__mem_t *ctx, void *mem, size_t size) {
810*00b67f09SDavid van Moolenbroek #if ISC_MEM_CHECKOVERRUN
811*00b67f09SDavid van Moolenbroek 	INSIST(((unsigned char *)mem)[size] == 0xbe);
812*00b67f09SDavid van Moolenbroek #endif
813*00b67f09SDavid van Moolenbroek #if ISC_MEM_FILL
814*00b67f09SDavid van Moolenbroek 	memset(mem, 0xde, size); /* Mnemonic for "dead". */
815*00b67f09SDavid van Moolenbroek #else
816*00b67f09SDavid van Moolenbroek 	UNUSED(size);
817*00b67f09SDavid van Moolenbroek #endif
818*00b67f09SDavid van Moolenbroek 	(ctx->memfree)(ctx->arg, mem);
819*00b67f09SDavid van Moolenbroek }
820*00b67f09SDavid van Moolenbroek 
821*00b67f09SDavid van Moolenbroek /*!
822*00b67f09SDavid van Moolenbroek  * Update internal counters after a memory get.
823*00b67f09SDavid van Moolenbroek  */
824*00b67f09SDavid van Moolenbroek static inline void
mem_getstats(isc__mem_t * ctx,size_t size)825*00b67f09SDavid van Moolenbroek mem_getstats(isc__mem_t *ctx, size_t size) {
826*00b67f09SDavid van Moolenbroek 	ctx->total += size;
827*00b67f09SDavid van Moolenbroek 	ctx->inuse += size;
828*00b67f09SDavid van Moolenbroek 
829*00b67f09SDavid van Moolenbroek 	if (size > ctx->max_size) {
830*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].gets++;
831*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].totalgets++;
832*00b67f09SDavid van Moolenbroek 	} else {
833*00b67f09SDavid van Moolenbroek 		ctx->stats[size].gets++;
834*00b67f09SDavid van Moolenbroek 		ctx->stats[size].totalgets++;
835*00b67f09SDavid van Moolenbroek 	}
836*00b67f09SDavid van Moolenbroek }
837*00b67f09SDavid van Moolenbroek 
838*00b67f09SDavid van Moolenbroek /*!
839*00b67f09SDavid van Moolenbroek  * Update internal counters after a memory put.
840*00b67f09SDavid van Moolenbroek  */
841*00b67f09SDavid van Moolenbroek static inline void
mem_putstats(isc__mem_t * ctx,void * ptr,size_t size)842*00b67f09SDavid van Moolenbroek mem_putstats(isc__mem_t *ctx, void *ptr, size_t size) {
843*00b67f09SDavid van Moolenbroek 	UNUSED(ptr);
844*00b67f09SDavid van Moolenbroek 
845*00b67f09SDavid van Moolenbroek 	INSIST(ctx->inuse >= size);
846*00b67f09SDavid van Moolenbroek 	ctx->inuse -= size;
847*00b67f09SDavid van Moolenbroek 
848*00b67f09SDavid van Moolenbroek 	if (size > ctx->max_size) {
849*00b67f09SDavid van Moolenbroek 		INSIST(ctx->stats[ctx->max_size].gets > 0U);
850*00b67f09SDavid van Moolenbroek 		ctx->stats[ctx->max_size].gets--;
851*00b67f09SDavid van Moolenbroek 	} else {
852*00b67f09SDavid van Moolenbroek 		INSIST(ctx->stats[size].gets > 0U);
853*00b67f09SDavid van Moolenbroek 		ctx->stats[size].gets--;
854*00b67f09SDavid van Moolenbroek 	}
855*00b67f09SDavid van Moolenbroek }
856*00b67f09SDavid van Moolenbroek 
857*00b67f09SDavid van Moolenbroek /*
858*00b67f09SDavid van Moolenbroek  * Private.
859*00b67f09SDavid van Moolenbroek  */
860*00b67f09SDavid van Moolenbroek 
861*00b67f09SDavid van Moolenbroek static void *
default_memalloc(void * arg,size_t size)862*00b67f09SDavid van Moolenbroek default_memalloc(void *arg, size_t size) {
863*00b67f09SDavid van Moolenbroek 	UNUSED(arg);
864*00b67f09SDavid van Moolenbroek 	if (size == 0U)
865*00b67f09SDavid van Moolenbroek 		size = 1;
866*00b67f09SDavid van Moolenbroek 	return (malloc(size));
867*00b67f09SDavid van Moolenbroek }
868*00b67f09SDavid van Moolenbroek 
869*00b67f09SDavid van Moolenbroek static void
default_memfree(void * arg,void * ptr)870*00b67f09SDavid van Moolenbroek default_memfree(void *arg, void *ptr) {
871*00b67f09SDavid van Moolenbroek 	UNUSED(arg);
872*00b67f09SDavid van Moolenbroek 	free(ptr);
873*00b67f09SDavid van Moolenbroek }
874*00b67f09SDavid van Moolenbroek 
875*00b67f09SDavid van Moolenbroek static void
initialize_action(void)876*00b67f09SDavid van Moolenbroek initialize_action(void) {
877*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_mutex_init(&createlock) == ISC_R_SUCCESS);
878*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_mutex_init(&contextslock) == ISC_R_SUCCESS);
879*00b67f09SDavid van Moolenbroek 	ISC_LIST_INIT(contexts);
880*00b67f09SDavid van Moolenbroek 	totallost = 0;
881*00b67f09SDavid van Moolenbroek }
882*00b67f09SDavid van Moolenbroek 
883*00b67f09SDavid van Moolenbroek /*
884*00b67f09SDavid van Moolenbroek  * Public.
885*00b67f09SDavid van Moolenbroek  */
886*00b67f09SDavid van Moolenbroek 
887*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_createx(size_t init_max_size,size_t target_size,isc_memalloc_t memalloc,isc_memfree_t memfree,void * arg,isc_mem_t ** ctxp)888*00b67f09SDavid van Moolenbroek isc_mem_createx(size_t init_max_size, size_t target_size,
889*00b67f09SDavid van Moolenbroek 		 isc_memalloc_t memalloc, isc_memfree_t memfree, void *arg,
890*00b67f09SDavid van Moolenbroek 		 isc_mem_t **ctxp)
891*00b67f09SDavid van Moolenbroek {
892*00b67f09SDavid van Moolenbroek 	return (isc_mem_createx2(init_max_size, target_size, memalloc, memfree,
893*00b67f09SDavid van Moolenbroek 				 arg, ctxp, ISC_MEMFLAG_DEFAULT));
894*00b67f09SDavid van Moolenbroek 
895*00b67f09SDavid van Moolenbroek }
896*00b67f09SDavid van Moolenbroek 
897*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_createx2(size_t init_max_size,size_t target_size,isc_memalloc_t memalloc,isc_memfree_t memfree,void * arg,isc_mem_t ** ctxp,unsigned int flags)898*00b67f09SDavid van Moolenbroek isc_mem_createx2(size_t init_max_size, size_t target_size,
899*00b67f09SDavid van Moolenbroek 		  isc_memalloc_t memalloc, isc_memfree_t memfree, void *arg,
900*00b67f09SDavid van Moolenbroek 		  isc_mem_t **ctxp, unsigned int flags)
901*00b67f09SDavid van Moolenbroek {
902*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
903*00b67f09SDavid van Moolenbroek 	isc_result_t result;
904*00b67f09SDavid van Moolenbroek 
905*00b67f09SDavid van Moolenbroek 	REQUIRE(ctxp != NULL && *ctxp == NULL);
906*00b67f09SDavid van Moolenbroek 	REQUIRE(memalloc != NULL);
907*00b67f09SDavid van Moolenbroek 	REQUIRE(memfree != NULL);
908*00b67f09SDavid van Moolenbroek 
909*00b67f09SDavid van Moolenbroek 	INSIST((ALIGNMENT_SIZE & (ALIGNMENT_SIZE - 1)) == 0);
910*00b67f09SDavid van Moolenbroek 
911*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
912*00b67f09SDavid van Moolenbroek 
913*00b67f09SDavid van Moolenbroek 	ctx = (memalloc)(arg, sizeof(*ctx));
914*00b67f09SDavid van Moolenbroek 	if (ctx == NULL)
915*00b67f09SDavid van Moolenbroek 		return (ISC_R_NOMEMORY);
916*00b67f09SDavid van Moolenbroek 
917*00b67f09SDavid van Moolenbroek 	if ((flags & ISC_MEMFLAG_NOLOCK) == 0) {
918*00b67f09SDavid van Moolenbroek 		result = isc_mutex_init(&ctx->lock);
919*00b67f09SDavid van Moolenbroek 		if (result != ISC_R_SUCCESS) {
920*00b67f09SDavid van Moolenbroek 			(memfree)(arg, ctx);
921*00b67f09SDavid van Moolenbroek 			return (result);
922*00b67f09SDavid van Moolenbroek 		}
923*00b67f09SDavid van Moolenbroek 	}
924*00b67f09SDavid van Moolenbroek 
925*00b67f09SDavid van Moolenbroek 	if (init_max_size == 0U)
926*00b67f09SDavid van Moolenbroek 		ctx->max_size = DEF_MAX_SIZE;
927*00b67f09SDavid van Moolenbroek 	else
928*00b67f09SDavid van Moolenbroek 		ctx->max_size = init_max_size;
929*00b67f09SDavid van Moolenbroek 	ctx->flags = flags;
930*00b67f09SDavid van Moolenbroek 	ctx->references = 1;
931*00b67f09SDavid van Moolenbroek 	memset(ctx->name, 0, sizeof(ctx->name));
932*00b67f09SDavid van Moolenbroek 	ctx->tag = NULL;
933*00b67f09SDavid van Moolenbroek 	ctx->quota = 0;
934*00b67f09SDavid van Moolenbroek 	ctx->total = 0;
935*00b67f09SDavid van Moolenbroek 	ctx->inuse = 0;
936*00b67f09SDavid van Moolenbroek 	ctx->maxinuse = 0;
937*00b67f09SDavid van Moolenbroek 	ctx->hi_water = 0;
938*00b67f09SDavid van Moolenbroek 	ctx->lo_water = 0;
939*00b67f09SDavid van Moolenbroek 	ctx->hi_called = ISC_FALSE;
940*00b67f09SDavid van Moolenbroek 	ctx->is_overmem = ISC_FALSE;
941*00b67f09SDavid van Moolenbroek 	ctx->water = NULL;
942*00b67f09SDavid van Moolenbroek 	ctx->water_arg = NULL;
943*00b67f09SDavid van Moolenbroek 	ctx->common.impmagic = MEM_MAGIC;
944*00b67f09SDavid van Moolenbroek 	ctx->common.magic = ISCAPI_MCTX_MAGIC;
945*00b67f09SDavid van Moolenbroek 	ctx->common.methods = (isc_memmethods_t *)&memmethods;
946*00b67f09SDavid van Moolenbroek 	isc_ondestroy_init(&ctx->ondestroy);
947*00b67f09SDavid van Moolenbroek 	ctx->memalloc = memalloc;
948*00b67f09SDavid van Moolenbroek 	ctx->memfree = memfree;
949*00b67f09SDavid van Moolenbroek 	ctx->arg = arg;
950*00b67f09SDavid van Moolenbroek 	ctx->stats = NULL;
951*00b67f09SDavid van Moolenbroek 	ctx->checkfree = ISC_TRUE;
952*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
953*00b67f09SDavid van Moolenbroek 	ctx->debuglist = NULL;
954*00b67f09SDavid van Moolenbroek 	ctx->debuglistcnt = 0;
955*00b67f09SDavid van Moolenbroek #endif
956*00b67f09SDavid van Moolenbroek 	ISC_LIST_INIT(ctx->pools);
957*00b67f09SDavid van Moolenbroek 	ctx->poolcnt = 0;
958*00b67f09SDavid van Moolenbroek 	ctx->freelists = NULL;
959*00b67f09SDavid van Moolenbroek 	ctx->basic_blocks = NULL;
960*00b67f09SDavid van Moolenbroek 	ctx->basic_table = NULL;
961*00b67f09SDavid van Moolenbroek 	ctx->basic_table_count = 0;
962*00b67f09SDavid van Moolenbroek 	ctx->basic_table_size = 0;
963*00b67f09SDavid van Moolenbroek 	ctx->lowest = NULL;
964*00b67f09SDavid van Moolenbroek 	ctx->highest = NULL;
965*00b67f09SDavid van Moolenbroek 
966*00b67f09SDavid van Moolenbroek 	ctx->stats = (memalloc)(arg,
967*00b67f09SDavid van Moolenbroek 				(ctx->max_size+1) * sizeof(struct stats));
968*00b67f09SDavid van Moolenbroek 	if (ctx->stats == NULL) {
969*00b67f09SDavid van Moolenbroek 		result = ISC_R_NOMEMORY;
970*00b67f09SDavid van Moolenbroek 		goto error;
971*00b67f09SDavid van Moolenbroek 	}
972*00b67f09SDavid van Moolenbroek 	memset(ctx->stats, 0, (ctx->max_size + 1) * sizeof(struct stats));
973*00b67f09SDavid van Moolenbroek 
974*00b67f09SDavid van Moolenbroek 	if ((flags & ISC_MEMFLAG_INTERNAL) != 0) {
975*00b67f09SDavid van Moolenbroek 		if (target_size == 0U)
976*00b67f09SDavid van Moolenbroek 			ctx->mem_target = DEF_MEM_TARGET;
977*00b67f09SDavid van Moolenbroek 		else
978*00b67f09SDavid van Moolenbroek 			ctx->mem_target = target_size;
979*00b67f09SDavid van Moolenbroek 		ctx->freelists = (memalloc)(arg, ctx->max_size *
980*00b67f09SDavid van Moolenbroek 						 sizeof(element *));
981*00b67f09SDavid van Moolenbroek 		if (ctx->freelists == NULL) {
982*00b67f09SDavid van Moolenbroek 			result = ISC_R_NOMEMORY;
983*00b67f09SDavid van Moolenbroek 			goto error;
984*00b67f09SDavid van Moolenbroek 		}
985*00b67f09SDavid van Moolenbroek 		memset(ctx->freelists, 0,
986*00b67f09SDavid van Moolenbroek 		       ctx->max_size * sizeof(element *));
987*00b67f09SDavid van Moolenbroek 	}
988*00b67f09SDavid van Moolenbroek 
989*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
990*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGRECORD) != 0) {
991*00b67f09SDavid van Moolenbroek 		unsigned int i;
992*00b67f09SDavid van Moolenbroek 
993*00b67f09SDavid van Moolenbroek 		ctx->debuglist = (memalloc)(arg,
994*00b67f09SDavid van Moolenbroek 				      (ctx->max_size+1) * sizeof(debuglist_t));
995*00b67f09SDavid van Moolenbroek 		if (ctx->debuglist == NULL) {
996*00b67f09SDavid van Moolenbroek 			result = ISC_R_NOMEMORY;
997*00b67f09SDavid van Moolenbroek 			goto error;
998*00b67f09SDavid van Moolenbroek 		}
999*00b67f09SDavid van Moolenbroek 		for (i = 0; i <= ctx->max_size; i++)
1000*00b67f09SDavid van Moolenbroek 			ISC_LIST_INIT(ctx->debuglist[i]);
1001*00b67f09SDavid van Moolenbroek 	}
1002*00b67f09SDavid van Moolenbroek #endif
1003*00b67f09SDavid van Moolenbroek 
1004*00b67f09SDavid van Moolenbroek 	ctx->memalloc_failures = 0;
1005*00b67f09SDavid van Moolenbroek 
1006*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
1007*00b67f09SDavid van Moolenbroek 	ISC_LIST_INITANDAPPEND(contexts, ctx, link);
1008*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
1009*00b67f09SDavid van Moolenbroek 
1010*00b67f09SDavid van Moolenbroek 	*ctxp = (isc_mem_t *)ctx;
1011*00b67f09SDavid van Moolenbroek 	return (ISC_R_SUCCESS);
1012*00b67f09SDavid van Moolenbroek 
1013*00b67f09SDavid van Moolenbroek   error:
1014*00b67f09SDavid van Moolenbroek 	if (ctx != NULL) {
1015*00b67f09SDavid van Moolenbroek 		if (ctx->stats != NULL)
1016*00b67f09SDavid van Moolenbroek 			(memfree)(arg, ctx->stats);
1017*00b67f09SDavid van Moolenbroek 		if (ctx->freelists != NULL)
1018*00b67f09SDavid van Moolenbroek 			(memfree)(arg, ctx->freelists);
1019*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1020*00b67f09SDavid van Moolenbroek 		if (ctx->debuglist != NULL)
1021*00b67f09SDavid van Moolenbroek 			(ctx->memfree)(ctx->arg, ctx->debuglist);
1022*00b67f09SDavid van Moolenbroek #endif /* ISC_MEM_TRACKLINES */
1023*00b67f09SDavid van Moolenbroek 		if ((ctx->flags & ISC_MEMFLAG_NOLOCK) == 0)
1024*00b67f09SDavid van Moolenbroek 			DESTROYLOCK(&ctx->lock);
1025*00b67f09SDavid van Moolenbroek 		(memfree)(arg, ctx);
1026*00b67f09SDavid van Moolenbroek 	}
1027*00b67f09SDavid van Moolenbroek 
1028*00b67f09SDavid van Moolenbroek 	return (result);
1029*00b67f09SDavid van Moolenbroek }
1030*00b67f09SDavid van Moolenbroek 
1031*00b67f09SDavid van Moolenbroek static void
destroy(isc__mem_t * ctx)1032*00b67f09SDavid van Moolenbroek destroy(isc__mem_t *ctx) {
1033*00b67f09SDavid van Moolenbroek 	unsigned int i;
1034*00b67f09SDavid van Moolenbroek 	isc_ondestroy_t ondest;
1035*00b67f09SDavid van Moolenbroek 
1036*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
1037*00b67f09SDavid van Moolenbroek 	ISC_LIST_UNLINK(contexts, ctx, link);
1038*00b67f09SDavid van Moolenbroek 	totallost += ctx->inuse;
1039*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
1040*00b67f09SDavid van Moolenbroek 
1041*00b67f09SDavid van Moolenbroek 	ctx->common.impmagic = 0;
1042*00b67f09SDavid van Moolenbroek 	ctx->common.magic = 0;
1043*00b67f09SDavid van Moolenbroek 
1044*00b67f09SDavid van Moolenbroek 	INSIST(ISC_LIST_EMPTY(ctx->pools));
1045*00b67f09SDavid van Moolenbroek 
1046*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1047*00b67f09SDavid van Moolenbroek 	if (ctx->debuglist != NULL) {
1048*00b67f09SDavid van Moolenbroek 		if (ctx->checkfree) {
1049*00b67f09SDavid van Moolenbroek 			for (i = 0; i <= ctx->max_size; i++) {
1050*00b67f09SDavid van Moolenbroek 				if (!ISC_LIST_EMPTY(ctx->debuglist[i]))
1051*00b67f09SDavid van Moolenbroek 					print_active(ctx, stderr);
1052*00b67f09SDavid van Moolenbroek 				INSIST(ISC_LIST_EMPTY(ctx->debuglist[i]));
1053*00b67f09SDavid van Moolenbroek 			}
1054*00b67f09SDavid van Moolenbroek 		} else {
1055*00b67f09SDavid van Moolenbroek 			debuglink_t *dl;
1056*00b67f09SDavid van Moolenbroek 
1057*00b67f09SDavid van Moolenbroek 			for (i = 0; i <= ctx->max_size; i++)
1058*00b67f09SDavid van Moolenbroek 				for (dl = ISC_LIST_HEAD(ctx->debuglist[i]);
1059*00b67f09SDavid van Moolenbroek 				     dl != NULL;
1060*00b67f09SDavid van Moolenbroek 				     dl = ISC_LIST_HEAD(ctx->debuglist[i])) {
1061*00b67f09SDavid van Moolenbroek 					ISC_LIST_UNLINK(ctx->debuglist[i],
1062*00b67f09SDavid van Moolenbroek 							dl, link);
1063*00b67f09SDavid van Moolenbroek 					free(dl);
1064*00b67f09SDavid van Moolenbroek 				}
1065*00b67f09SDavid van Moolenbroek 		}
1066*00b67f09SDavid van Moolenbroek 		(ctx->memfree)(ctx->arg, ctx->debuglist);
1067*00b67f09SDavid van Moolenbroek 	}
1068*00b67f09SDavid van Moolenbroek #endif
1069*00b67f09SDavid van Moolenbroek 	INSIST(ctx->references == 0);
1070*00b67f09SDavid van Moolenbroek 
1071*00b67f09SDavid van Moolenbroek 	if (ctx->checkfree) {
1072*00b67f09SDavid van Moolenbroek 		for (i = 0; i <= ctx->max_size; i++) {
1073*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1074*00b67f09SDavid van Moolenbroek 			if (ctx->stats[i].gets != 0U)
1075*00b67f09SDavid van Moolenbroek 				print_active(ctx, stderr);
1076*00b67f09SDavid van Moolenbroek #endif
1077*00b67f09SDavid van Moolenbroek 			INSIST(ctx->stats[i].gets == 0U);
1078*00b67f09SDavid van Moolenbroek 		}
1079*00b67f09SDavid van Moolenbroek 	}
1080*00b67f09SDavid van Moolenbroek 
1081*00b67f09SDavid van Moolenbroek 	(ctx->memfree)(ctx->arg, ctx->stats);
1082*00b67f09SDavid van Moolenbroek 
1083*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1084*00b67f09SDavid van Moolenbroek 		for (i = 0; i < ctx->basic_table_count; i++)
1085*00b67f09SDavid van Moolenbroek 			(ctx->memfree)(ctx->arg, ctx->basic_table[i]);
1086*00b67f09SDavid van Moolenbroek 		(ctx->memfree)(ctx->arg, ctx->freelists);
1087*00b67f09SDavid van Moolenbroek 		if (ctx->basic_table != NULL)
1088*00b67f09SDavid van Moolenbroek 			(ctx->memfree)(ctx->arg, ctx->basic_table);
1089*00b67f09SDavid van Moolenbroek 	}
1090*00b67f09SDavid van Moolenbroek 
1091*00b67f09SDavid van Moolenbroek 	ondest = ctx->ondestroy;
1092*00b67f09SDavid van Moolenbroek 
1093*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_NOLOCK) == 0)
1094*00b67f09SDavid van Moolenbroek 		DESTROYLOCK(&ctx->lock);
1095*00b67f09SDavid van Moolenbroek 	(ctx->memfree)(ctx->arg, ctx);
1096*00b67f09SDavid van Moolenbroek 
1097*00b67f09SDavid van Moolenbroek 	isc_ondestroy_notify(&ondest, ctx);
1098*00b67f09SDavid van Moolenbroek }
1099*00b67f09SDavid van Moolenbroek 
1100*00b67f09SDavid van Moolenbroek void
isc__mem_attach(isc_mem_t * source0,isc_mem_t ** targetp)1101*00b67f09SDavid van Moolenbroek isc__mem_attach(isc_mem_t *source0, isc_mem_t **targetp) {
1102*00b67f09SDavid van Moolenbroek 	isc__mem_t *source = (isc__mem_t *)source0;
1103*00b67f09SDavid van Moolenbroek 
1104*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(source));
1105*00b67f09SDavid van Moolenbroek 	REQUIRE(targetp != NULL && *targetp == NULL);
1106*00b67f09SDavid van Moolenbroek 
1107*00b67f09SDavid van Moolenbroek 	MCTXLOCK(source, &source->lock);
1108*00b67f09SDavid van Moolenbroek 	source->references++;
1109*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(source, &source->lock);
1110*00b67f09SDavid van Moolenbroek 
1111*00b67f09SDavid van Moolenbroek 	*targetp = (isc_mem_t *)source;
1112*00b67f09SDavid van Moolenbroek }
1113*00b67f09SDavid van Moolenbroek 
1114*00b67f09SDavid van Moolenbroek void
isc__mem_detach(isc_mem_t ** ctxp)1115*00b67f09SDavid van Moolenbroek isc__mem_detach(isc_mem_t **ctxp) {
1116*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
1117*00b67f09SDavid van Moolenbroek 	isc_boolean_t want_destroy = ISC_FALSE;
1118*00b67f09SDavid van Moolenbroek 
1119*00b67f09SDavid van Moolenbroek 	REQUIRE(ctxp != NULL);
1120*00b67f09SDavid van Moolenbroek 	ctx = (isc__mem_t *)*ctxp;
1121*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1122*00b67f09SDavid van Moolenbroek 
1123*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1124*00b67f09SDavid van Moolenbroek 	INSIST(ctx->references > 0);
1125*00b67f09SDavid van Moolenbroek 	ctx->references--;
1126*00b67f09SDavid van Moolenbroek 	if (ctx->references == 0)
1127*00b67f09SDavid van Moolenbroek 		want_destroy = ISC_TRUE;
1128*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1129*00b67f09SDavid van Moolenbroek 
1130*00b67f09SDavid van Moolenbroek 	if (want_destroy)
1131*00b67f09SDavid van Moolenbroek 		destroy(ctx);
1132*00b67f09SDavid van Moolenbroek 
1133*00b67f09SDavid van Moolenbroek 	*ctxp = NULL;
1134*00b67f09SDavid van Moolenbroek }
1135*00b67f09SDavid van Moolenbroek 
1136*00b67f09SDavid van Moolenbroek /*
1137*00b67f09SDavid van Moolenbroek  * isc_mem_putanddetach() is the equivalent of:
1138*00b67f09SDavid van Moolenbroek  *
1139*00b67f09SDavid van Moolenbroek  * mctx = NULL;
1140*00b67f09SDavid van Moolenbroek  * isc_mem_attach(ptr->mctx, &mctx);
1141*00b67f09SDavid van Moolenbroek  * isc_mem_detach(&ptr->mctx);
1142*00b67f09SDavid van Moolenbroek  * isc_mem_put(mctx, ptr, sizeof(*ptr);
1143*00b67f09SDavid van Moolenbroek  * isc_mem_detach(&mctx);
1144*00b67f09SDavid van Moolenbroek  */
1145*00b67f09SDavid van Moolenbroek 
1146*00b67f09SDavid van Moolenbroek void
isc___mem_putanddetach(isc_mem_t ** ctxp,void * ptr,size_t size FLARG)1147*00b67f09SDavid van Moolenbroek isc___mem_putanddetach(isc_mem_t **ctxp, void *ptr, size_t size FLARG) {
1148*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
1149*00b67f09SDavid van Moolenbroek 	isc_boolean_t want_destroy = ISC_FALSE;
1150*00b67f09SDavid van Moolenbroek 	size_info *si;
1151*00b67f09SDavid van Moolenbroek 	size_t oldsize;
1152*00b67f09SDavid van Moolenbroek 
1153*00b67f09SDavid van Moolenbroek 	REQUIRE(ctxp != NULL);
1154*00b67f09SDavid van Moolenbroek 	ctx = (isc__mem_t *)*ctxp;
1155*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1156*00b67f09SDavid van Moolenbroek 	REQUIRE(ptr != NULL);
1157*00b67f09SDavid van Moolenbroek 
1158*00b67f09SDavid van Moolenbroek 	/*
1159*00b67f09SDavid van Moolenbroek 	 * Must be before mem_putunlocked() as ctxp is usually within
1160*00b67f09SDavid van Moolenbroek 	 * [ptr..ptr+size).
1161*00b67f09SDavid van Moolenbroek 	 */
1162*00b67f09SDavid van Moolenbroek 	*ctxp = NULL;
1163*00b67f09SDavid van Moolenbroek 
1164*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & (ISC_MEM_DEBUGSIZE|ISC_MEM_DEBUGCTX)) != 0) {
1165*00b67f09SDavid van Moolenbroek 		if ((isc_mem_debugging & ISC_MEM_DEBUGSIZE) != 0) {
1166*00b67f09SDavid van Moolenbroek 			si = &(((size_info *)ptr)[-1]);
1167*00b67f09SDavid van Moolenbroek 			oldsize = si->u.size - ALIGNMENT_SIZE;
1168*00b67f09SDavid van Moolenbroek 			if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0)
1169*00b67f09SDavid van Moolenbroek 				oldsize -= ALIGNMENT_SIZE;
1170*00b67f09SDavid van Moolenbroek 			INSIST(oldsize == size);
1171*00b67f09SDavid van Moolenbroek 		}
1172*00b67f09SDavid van Moolenbroek 		isc__mem_free((isc_mem_t *)ctx, ptr FLARG_PASS);
1173*00b67f09SDavid van Moolenbroek 
1174*00b67f09SDavid van Moolenbroek 		MCTXLOCK(ctx, &ctx->lock);
1175*00b67f09SDavid van Moolenbroek 		ctx->references--;
1176*00b67f09SDavid van Moolenbroek 		if (ctx->references == 0)
1177*00b67f09SDavid van Moolenbroek 			want_destroy = ISC_TRUE;
1178*00b67f09SDavid van Moolenbroek 		MCTXUNLOCK(ctx, &ctx->lock);
1179*00b67f09SDavid van Moolenbroek 		if (want_destroy)
1180*00b67f09SDavid van Moolenbroek 			destroy(ctx);
1181*00b67f09SDavid van Moolenbroek 
1182*00b67f09SDavid van Moolenbroek 		return;
1183*00b67f09SDavid van Moolenbroek 	}
1184*00b67f09SDavid van Moolenbroek 
1185*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1186*00b67f09SDavid van Moolenbroek 
1187*00b67f09SDavid van Moolenbroek 	DELETE_TRACE(ctx, ptr, size, file, line);
1188*00b67f09SDavid van Moolenbroek 
1189*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1190*00b67f09SDavid van Moolenbroek 		mem_putunlocked(ctx, ptr, size);
1191*00b67f09SDavid van Moolenbroek 	} else {
1192*00b67f09SDavid van Moolenbroek 		mem_putstats(ctx, ptr, size);
1193*00b67f09SDavid van Moolenbroek 		mem_put(ctx, ptr, size);
1194*00b67f09SDavid van Moolenbroek 	}
1195*00b67f09SDavid van Moolenbroek 
1196*00b67f09SDavid van Moolenbroek 	INSIST(ctx->references > 0);
1197*00b67f09SDavid van Moolenbroek 	ctx->references--;
1198*00b67f09SDavid van Moolenbroek 	if (ctx->references == 0)
1199*00b67f09SDavid van Moolenbroek 		want_destroy = ISC_TRUE;
1200*00b67f09SDavid van Moolenbroek 
1201*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1202*00b67f09SDavid van Moolenbroek 
1203*00b67f09SDavid van Moolenbroek 	if (want_destroy)
1204*00b67f09SDavid van Moolenbroek 		destroy(ctx);
1205*00b67f09SDavid van Moolenbroek }
1206*00b67f09SDavid van Moolenbroek 
1207*00b67f09SDavid van Moolenbroek void
isc__mem_destroy(isc_mem_t ** ctxp)1208*00b67f09SDavid van Moolenbroek isc__mem_destroy(isc_mem_t **ctxp) {
1209*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
1210*00b67f09SDavid van Moolenbroek 
1211*00b67f09SDavid van Moolenbroek 	/*
1212*00b67f09SDavid van Moolenbroek 	 * This routine provides legacy support for callers who use mctxs
1213*00b67f09SDavid van Moolenbroek 	 * without attaching/detaching.
1214*00b67f09SDavid van Moolenbroek 	 */
1215*00b67f09SDavid van Moolenbroek 
1216*00b67f09SDavid van Moolenbroek 	REQUIRE(ctxp != NULL);
1217*00b67f09SDavid van Moolenbroek 	ctx = (isc__mem_t *)*ctxp;
1218*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1219*00b67f09SDavid van Moolenbroek 
1220*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1221*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1222*00b67f09SDavid van Moolenbroek 	if (ctx->references != 1)
1223*00b67f09SDavid van Moolenbroek 		print_active(ctx, stderr);
1224*00b67f09SDavid van Moolenbroek #endif
1225*00b67f09SDavid van Moolenbroek 	REQUIRE(ctx->references == 1);
1226*00b67f09SDavid van Moolenbroek 	ctx->references--;
1227*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1228*00b67f09SDavid van Moolenbroek 
1229*00b67f09SDavid van Moolenbroek 	destroy(ctx);
1230*00b67f09SDavid van Moolenbroek 
1231*00b67f09SDavid van Moolenbroek 	*ctxp = NULL;
1232*00b67f09SDavid van Moolenbroek }
1233*00b67f09SDavid van Moolenbroek 
1234*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_ondestroy(isc_mem_t * ctx0,isc_task_t * task,isc_event_t ** event)1235*00b67f09SDavid van Moolenbroek isc_mem_ondestroy(isc_mem_t *ctx0, isc_task_t *task, isc_event_t **event) {
1236*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1237*00b67f09SDavid van Moolenbroek 	isc_result_t res;
1238*00b67f09SDavid van Moolenbroek 
1239*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1240*00b67f09SDavid van Moolenbroek 	res = isc_ondestroy_register(&ctx->ondestroy, task, event);
1241*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1242*00b67f09SDavid van Moolenbroek 
1243*00b67f09SDavid van Moolenbroek 	return (res);
1244*00b67f09SDavid van Moolenbroek }
1245*00b67f09SDavid van Moolenbroek 
1246*00b67f09SDavid van Moolenbroek void *
isc___mem_get(isc_mem_t * ctx0,size_t size FLARG)1247*00b67f09SDavid van Moolenbroek isc___mem_get(isc_mem_t *ctx0, size_t size FLARG) {
1248*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1249*00b67f09SDavid van Moolenbroek 	void *ptr;
1250*00b67f09SDavid van Moolenbroek 	isc_boolean_t call_water = ISC_FALSE;
1251*00b67f09SDavid van Moolenbroek 
1252*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1253*00b67f09SDavid van Moolenbroek 
1254*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & (ISC_MEM_DEBUGSIZE|ISC_MEM_DEBUGCTX)) != 0)
1255*00b67f09SDavid van Moolenbroek 		return (isc__mem_allocate(ctx0, size FLARG_PASS));
1256*00b67f09SDavid van Moolenbroek 
1257*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1258*00b67f09SDavid van Moolenbroek 		MCTXLOCK(ctx, &ctx->lock);
1259*00b67f09SDavid van Moolenbroek 		ptr = mem_getunlocked(ctx, size);
1260*00b67f09SDavid van Moolenbroek 	} else {
1261*00b67f09SDavid van Moolenbroek 		ptr = mem_get(ctx, size);
1262*00b67f09SDavid van Moolenbroek 		MCTXLOCK(ctx, &ctx->lock);
1263*00b67f09SDavid van Moolenbroek 		if (ptr != NULL)
1264*00b67f09SDavid van Moolenbroek 			mem_getstats(ctx, size);
1265*00b67f09SDavid van Moolenbroek 	}
1266*00b67f09SDavid van Moolenbroek 
1267*00b67f09SDavid van Moolenbroek 	ADD_TRACE(ctx, ptr, size, file, line);
1268*00b67f09SDavid van Moolenbroek 	if (ctx->hi_water != 0U && ctx->inuse > ctx->hi_water &&
1269*00b67f09SDavid van Moolenbroek 	    !ctx->is_overmem) {
1270*00b67f09SDavid van Moolenbroek 		ctx->is_overmem = ISC_TRUE;
1271*00b67f09SDavid van Moolenbroek 	}
1272*00b67f09SDavid van Moolenbroek 	if (ctx->hi_water != 0U && !ctx->hi_called &&
1273*00b67f09SDavid van Moolenbroek 	    ctx->inuse > ctx->hi_water) {
1274*00b67f09SDavid van Moolenbroek 		call_water = ISC_TRUE;
1275*00b67f09SDavid van Moolenbroek 	}
1276*00b67f09SDavid van Moolenbroek 	if (ctx->inuse > ctx->maxinuse) {
1277*00b67f09SDavid van Moolenbroek 		ctx->maxinuse = ctx->inuse;
1278*00b67f09SDavid van Moolenbroek 		if (ctx->hi_water != 0U && ctx->inuse > ctx->hi_water &&
1279*00b67f09SDavid van Moolenbroek 		    (isc_mem_debugging & ISC_MEM_DEBUGUSAGE) != 0)
1280*00b67f09SDavid van Moolenbroek 			fprintf(stderr, "maxinuse = %lu\n",
1281*00b67f09SDavid van Moolenbroek 				(unsigned long)ctx->inuse);
1282*00b67f09SDavid van Moolenbroek 	}
1283*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1284*00b67f09SDavid van Moolenbroek 
1285*00b67f09SDavid van Moolenbroek 	if (call_water)
1286*00b67f09SDavid van Moolenbroek 		(ctx->water)(ctx->water_arg, ISC_MEM_HIWATER);
1287*00b67f09SDavid van Moolenbroek 
1288*00b67f09SDavid van Moolenbroek 	return (ptr);
1289*00b67f09SDavid van Moolenbroek }
1290*00b67f09SDavid van Moolenbroek 
1291*00b67f09SDavid van Moolenbroek void
isc___mem_put(isc_mem_t * ctx0,void * ptr,size_t size FLARG)1292*00b67f09SDavid van Moolenbroek isc___mem_put(isc_mem_t *ctx0, void *ptr, size_t size FLARG) {
1293*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1294*00b67f09SDavid van Moolenbroek 	isc_boolean_t call_water = ISC_FALSE;
1295*00b67f09SDavid van Moolenbroek 	size_info *si;
1296*00b67f09SDavid van Moolenbroek 	size_t oldsize;
1297*00b67f09SDavid van Moolenbroek 
1298*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1299*00b67f09SDavid van Moolenbroek 	REQUIRE(ptr != NULL);
1300*00b67f09SDavid van Moolenbroek 
1301*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & (ISC_MEM_DEBUGSIZE|ISC_MEM_DEBUGCTX)) != 0) {
1302*00b67f09SDavid van Moolenbroek 		if ((isc_mem_debugging & ISC_MEM_DEBUGSIZE) != 0) {
1303*00b67f09SDavid van Moolenbroek 			si = &(((size_info *)ptr)[-1]);
1304*00b67f09SDavid van Moolenbroek 			oldsize = si->u.size - ALIGNMENT_SIZE;
1305*00b67f09SDavid van Moolenbroek 			if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0)
1306*00b67f09SDavid van Moolenbroek 				oldsize -= ALIGNMENT_SIZE;
1307*00b67f09SDavid van Moolenbroek 			INSIST(oldsize == size);
1308*00b67f09SDavid van Moolenbroek 		}
1309*00b67f09SDavid van Moolenbroek 		isc__mem_free((isc_mem_t *)ctx, ptr FLARG_PASS);
1310*00b67f09SDavid van Moolenbroek 		return;
1311*00b67f09SDavid van Moolenbroek 	}
1312*00b67f09SDavid van Moolenbroek 
1313*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1314*00b67f09SDavid van Moolenbroek 
1315*00b67f09SDavid van Moolenbroek 	DELETE_TRACE(ctx, ptr, size, file, line);
1316*00b67f09SDavid van Moolenbroek 
1317*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1318*00b67f09SDavid van Moolenbroek 		mem_putunlocked(ctx, ptr, size);
1319*00b67f09SDavid van Moolenbroek 	} else {
1320*00b67f09SDavid van Moolenbroek 		mem_putstats(ctx, ptr, size);
1321*00b67f09SDavid van Moolenbroek 		mem_put(ctx, ptr, size);
1322*00b67f09SDavid van Moolenbroek 	}
1323*00b67f09SDavid van Moolenbroek 
1324*00b67f09SDavid van Moolenbroek 	/*
1325*00b67f09SDavid van Moolenbroek 	 * The check against ctx->lo_water == 0 is for the condition
1326*00b67f09SDavid van Moolenbroek 	 * when the context was pushed over hi_water but then had
1327*00b67f09SDavid van Moolenbroek 	 * isc_mem_setwater() called with 0 for hi_water and lo_water.
1328*00b67f09SDavid van Moolenbroek 	 */
1329*00b67f09SDavid van Moolenbroek 	if (ctx->is_overmem &&
1330*00b67f09SDavid van Moolenbroek 	    (ctx->inuse < ctx->lo_water || ctx->lo_water == 0U)) {
1331*00b67f09SDavid van Moolenbroek 		ctx->is_overmem = ISC_FALSE;
1332*00b67f09SDavid van Moolenbroek 	}
1333*00b67f09SDavid van Moolenbroek 	if (ctx->hi_called &&
1334*00b67f09SDavid van Moolenbroek 	    (ctx->inuse < ctx->lo_water || ctx->lo_water == 0U)) {
1335*00b67f09SDavid van Moolenbroek 		if (ctx->water != NULL)
1336*00b67f09SDavid van Moolenbroek 			call_water = ISC_TRUE;
1337*00b67f09SDavid van Moolenbroek 	}
1338*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1339*00b67f09SDavid van Moolenbroek 
1340*00b67f09SDavid van Moolenbroek 	if (call_water)
1341*00b67f09SDavid van Moolenbroek 		(ctx->water)(ctx->water_arg, ISC_MEM_LOWATER);
1342*00b67f09SDavid van Moolenbroek }
1343*00b67f09SDavid van Moolenbroek 
1344*00b67f09SDavid van Moolenbroek void
isc__mem_waterack(isc_mem_t * ctx0,int flag)1345*00b67f09SDavid van Moolenbroek isc__mem_waterack(isc_mem_t *ctx0, int flag) {
1346*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1347*00b67f09SDavid van Moolenbroek 
1348*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1349*00b67f09SDavid van Moolenbroek 
1350*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1351*00b67f09SDavid van Moolenbroek 	if (flag == ISC_MEM_LOWATER)
1352*00b67f09SDavid van Moolenbroek 		ctx->hi_called = ISC_FALSE;
1353*00b67f09SDavid van Moolenbroek 	else if (flag == ISC_MEM_HIWATER)
1354*00b67f09SDavid van Moolenbroek 		ctx->hi_called = ISC_TRUE;
1355*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1356*00b67f09SDavid van Moolenbroek }
1357*00b67f09SDavid van Moolenbroek 
1358*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1359*00b67f09SDavid van Moolenbroek static void
print_active(isc__mem_t * mctx,FILE * out)1360*00b67f09SDavid van Moolenbroek print_active(isc__mem_t *mctx, FILE *out) {
1361*00b67f09SDavid van Moolenbroek 	if (mctx->debuglist != NULL) {
1362*00b67f09SDavid van Moolenbroek 		debuglink_t *dl;
1363*00b67f09SDavid van Moolenbroek 		unsigned int i, j;
1364*00b67f09SDavid van Moolenbroek 		const char *format;
1365*00b67f09SDavid van Moolenbroek 		isc_boolean_t found;
1366*00b67f09SDavid van Moolenbroek 
1367*00b67f09SDavid van Moolenbroek 		fprintf(out, "%s", isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1368*00b67f09SDavid van Moolenbroek 					    ISC_MSG_DUMPALLOC,
1369*00b67f09SDavid van Moolenbroek 					    "Dump of all outstanding "
1370*00b67f09SDavid van Moolenbroek 					    "memory allocations:\n"));
1371*00b67f09SDavid van Moolenbroek 		found = ISC_FALSE;
1372*00b67f09SDavid van Moolenbroek 		format = isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1373*00b67f09SDavid van Moolenbroek 					ISC_MSG_PTRFILELINE,
1374*00b67f09SDavid van Moolenbroek 					"\tptr %p size %u file %s line %u\n");
1375*00b67f09SDavid van Moolenbroek 		for (i = 0; i <= mctx->max_size; i++) {
1376*00b67f09SDavid van Moolenbroek 			dl = ISC_LIST_HEAD(mctx->debuglist[i]);
1377*00b67f09SDavid van Moolenbroek 
1378*00b67f09SDavid van Moolenbroek 			if (dl != NULL)
1379*00b67f09SDavid van Moolenbroek 				found = ISC_TRUE;
1380*00b67f09SDavid van Moolenbroek 
1381*00b67f09SDavid van Moolenbroek 			while (dl != NULL) {
1382*00b67f09SDavid van Moolenbroek 				for (j = 0; j < DEBUGLIST_COUNT; j++)
1383*00b67f09SDavid van Moolenbroek 					if (dl->ptr[j] != NULL)
1384*00b67f09SDavid van Moolenbroek 						fprintf(out, format,
1385*00b67f09SDavid van Moolenbroek 							dl->ptr[j],
1386*00b67f09SDavid van Moolenbroek 							dl->size[j],
1387*00b67f09SDavid van Moolenbroek 							dl->file[j],
1388*00b67f09SDavid van Moolenbroek 							dl->line[j]);
1389*00b67f09SDavid van Moolenbroek 				dl = ISC_LIST_NEXT(dl, link);
1390*00b67f09SDavid van Moolenbroek 			}
1391*00b67f09SDavid van Moolenbroek 		}
1392*00b67f09SDavid van Moolenbroek 		if (!found)
1393*00b67f09SDavid van Moolenbroek 			fprintf(out, "%s", isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1394*00b67f09SDavid van Moolenbroek 						    ISC_MSG_NONE, "\tNone.\n"));
1395*00b67f09SDavid van Moolenbroek 	}
1396*00b67f09SDavid van Moolenbroek }
1397*00b67f09SDavid van Moolenbroek #endif
1398*00b67f09SDavid van Moolenbroek 
1399*00b67f09SDavid van Moolenbroek /*
1400*00b67f09SDavid van Moolenbroek  * Print the stats[] on the stream "out" with suitable formatting.
1401*00b67f09SDavid van Moolenbroek  */
1402*00b67f09SDavid van Moolenbroek void
isc_mem_stats(isc_mem_t * ctx0,FILE * out)1403*00b67f09SDavid van Moolenbroek isc_mem_stats(isc_mem_t *ctx0, FILE *out) {
1404*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1405*00b67f09SDavid van Moolenbroek 	size_t i;
1406*00b67f09SDavid van Moolenbroek 	const struct stats *s;
1407*00b67f09SDavid van Moolenbroek 	const isc__mempool_t *pool;
1408*00b67f09SDavid van Moolenbroek 
1409*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1410*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1411*00b67f09SDavid van Moolenbroek 
1412*00b67f09SDavid van Moolenbroek 	for (i = 0; i <= ctx->max_size; i++) {
1413*00b67f09SDavid van Moolenbroek 		s = &ctx->stats[i];
1414*00b67f09SDavid van Moolenbroek 
1415*00b67f09SDavid van Moolenbroek 		if (s->totalgets == 0U && s->gets == 0U)
1416*00b67f09SDavid van Moolenbroek 			continue;
1417*00b67f09SDavid van Moolenbroek 		fprintf(out, "%s%5lu: %11lu gets, %11lu rem",
1418*00b67f09SDavid van Moolenbroek 			(i == ctx->max_size) ? ">=" : "  ",
1419*00b67f09SDavid van Moolenbroek 			(unsigned long) i, s->totalgets, s->gets);
1420*00b67f09SDavid van Moolenbroek 		if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0 &&
1421*00b67f09SDavid van Moolenbroek 		    (s->blocks != 0U || s->freefrags != 0U))
1422*00b67f09SDavid van Moolenbroek 			fprintf(out, " (%lu bl, %lu ff)",
1423*00b67f09SDavid van Moolenbroek 				s->blocks, s->freefrags);
1424*00b67f09SDavid van Moolenbroek 		fputc('\n', out);
1425*00b67f09SDavid van Moolenbroek 	}
1426*00b67f09SDavid van Moolenbroek 
1427*00b67f09SDavid van Moolenbroek 	/*
1428*00b67f09SDavid van Moolenbroek 	 * Note that since a pool can be locked now, these stats might be
1429*00b67f09SDavid van Moolenbroek 	 * somewhat off if the pool is in active use at the time the stats
1430*00b67f09SDavid van Moolenbroek 	 * are dumped.  The link fields are protected by the isc_mem_t's
1431*00b67f09SDavid van Moolenbroek 	 * lock, however, so walking this list and extracting integers from
1432*00b67f09SDavid van Moolenbroek 	 * stats fields is always safe.
1433*00b67f09SDavid van Moolenbroek 	 */
1434*00b67f09SDavid van Moolenbroek 	pool = ISC_LIST_HEAD(ctx->pools);
1435*00b67f09SDavid van Moolenbroek 	if (pool != NULL) {
1436*00b67f09SDavid van Moolenbroek 		fprintf(out, "%s", isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1437*00b67f09SDavid van Moolenbroek 					    ISC_MSG_POOLSTATS,
1438*00b67f09SDavid van Moolenbroek 					    "[Pool statistics]\n"));
1439*00b67f09SDavid van Moolenbroek 		fprintf(out, "%15s %10s %10s %10s %10s %10s %10s %10s %1s\n",
1440*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1441*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLNAME, "name"),
1442*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1443*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLSIZE, "size"),
1444*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1445*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLMAXALLOC, "maxalloc"),
1446*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1447*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLALLOCATED, "allocated"),
1448*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1449*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLFREECOUNT, "freecount"),
1450*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1451*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLFREEMAX, "freemax"),
1452*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1453*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLFILLCOUNT, "fillcount"),
1454*00b67f09SDavid van Moolenbroek 			isc_msgcat_get(isc_msgcat, ISC_MSGSET_MEM,
1455*00b67f09SDavid van Moolenbroek 				       ISC_MSG_POOLGETS, "gets"),
1456*00b67f09SDavid van Moolenbroek 			"L");
1457*00b67f09SDavid van Moolenbroek 	}
1458*00b67f09SDavid van Moolenbroek 	while (pool != NULL) {
1459*00b67f09SDavid van Moolenbroek 		fprintf(out, "%15s %10lu %10u %10u %10u %10u %10u %10u %s\n",
1460*00b67f09SDavid van Moolenbroek #if ISC_MEMPOOL_NAMES
1461*00b67f09SDavid van Moolenbroek 			pool->name,
1462*00b67f09SDavid van Moolenbroek #else
1463*00b67f09SDavid van Moolenbroek 			"(not tracked)",
1464*00b67f09SDavid van Moolenbroek #endif
1465*00b67f09SDavid van Moolenbroek 			(unsigned long) pool->size, pool->maxalloc,
1466*00b67f09SDavid van Moolenbroek 			pool->allocated, pool->freecount, pool->freemax,
1467*00b67f09SDavid van Moolenbroek 			pool->fillcount, pool->gets,
1468*00b67f09SDavid van Moolenbroek 			(pool->lock == NULL ? "N" : "Y"));
1469*00b67f09SDavid van Moolenbroek 		pool = ISC_LIST_NEXT(pool, link);
1470*00b67f09SDavid van Moolenbroek 	}
1471*00b67f09SDavid van Moolenbroek 
1472*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1473*00b67f09SDavid van Moolenbroek 	print_active(ctx, out);
1474*00b67f09SDavid van Moolenbroek #endif
1475*00b67f09SDavid van Moolenbroek 
1476*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1477*00b67f09SDavid van Moolenbroek }
1478*00b67f09SDavid van Moolenbroek 
1479*00b67f09SDavid van Moolenbroek /*
1480*00b67f09SDavid van Moolenbroek  * Replacements for malloc() and free() -- they implicitly remember the
1481*00b67f09SDavid van Moolenbroek  * size of the object allocated (with some additional overhead).
1482*00b67f09SDavid van Moolenbroek  */
1483*00b67f09SDavid van Moolenbroek 
1484*00b67f09SDavid van Moolenbroek static void *
mem_allocateunlocked(isc_mem_t * ctx0,size_t size)1485*00b67f09SDavid van Moolenbroek mem_allocateunlocked(isc_mem_t *ctx0, size_t size) {
1486*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1487*00b67f09SDavid van Moolenbroek 	size_info *si;
1488*00b67f09SDavid van Moolenbroek 
1489*00b67f09SDavid van Moolenbroek 	size += ALIGNMENT_SIZE;
1490*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0)
1491*00b67f09SDavid van Moolenbroek 		size += ALIGNMENT_SIZE;
1492*00b67f09SDavid van Moolenbroek 
1493*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0)
1494*00b67f09SDavid van Moolenbroek 		si = mem_getunlocked(ctx, size);
1495*00b67f09SDavid van Moolenbroek 	else
1496*00b67f09SDavid van Moolenbroek 		si = mem_get(ctx, size);
1497*00b67f09SDavid van Moolenbroek 
1498*00b67f09SDavid van Moolenbroek 	if (si == NULL)
1499*00b67f09SDavid van Moolenbroek 		return (NULL);
1500*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0) {
1501*00b67f09SDavid van Moolenbroek 		si->u.ctx = ctx;
1502*00b67f09SDavid van Moolenbroek 		si++;
1503*00b67f09SDavid van Moolenbroek 	}
1504*00b67f09SDavid van Moolenbroek 	si->u.size = size;
1505*00b67f09SDavid van Moolenbroek 	return (&si[1]);
1506*00b67f09SDavid van Moolenbroek }
1507*00b67f09SDavid van Moolenbroek 
1508*00b67f09SDavid van Moolenbroek void *
isc___mem_allocate(isc_mem_t * ctx0,size_t size FLARG)1509*00b67f09SDavid van Moolenbroek isc___mem_allocate(isc_mem_t *ctx0, size_t size FLARG) {
1510*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1511*00b67f09SDavid van Moolenbroek 	size_info *si;
1512*00b67f09SDavid van Moolenbroek 	isc_boolean_t call_water = ISC_FALSE;
1513*00b67f09SDavid van Moolenbroek 
1514*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1515*00b67f09SDavid van Moolenbroek 
1516*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1517*00b67f09SDavid van Moolenbroek 		MCTXLOCK(ctx, &ctx->lock);
1518*00b67f09SDavid van Moolenbroek 		si = mem_allocateunlocked((isc_mem_t *)ctx, size);
1519*00b67f09SDavid van Moolenbroek 	} else {
1520*00b67f09SDavid van Moolenbroek 		si = mem_allocateunlocked((isc_mem_t *)ctx, size);
1521*00b67f09SDavid van Moolenbroek 		MCTXLOCK(ctx, &ctx->lock);
1522*00b67f09SDavid van Moolenbroek 		if (si != NULL)
1523*00b67f09SDavid van Moolenbroek 			mem_getstats(ctx, si[-1].u.size);
1524*00b67f09SDavid van Moolenbroek 	}
1525*00b67f09SDavid van Moolenbroek 
1526*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
1527*00b67f09SDavid van Moolenbroek 	ADD_TRACE(ctx, si, si[-1].u.size, file, line);
1528*00b67f09SDavid van Moolenbroek #endif
1529*00b67f09SDavid van Moolenbroek 	if (ctx->hi_water != 0U && ctx->inuse > ctx->hi_water &&
1530*00b67f09SDavid van Moolenbroek 	    !ctx->is_overmem) {
1531*00b67f09SDavid van Moolenbroek 		ctx->is_overmem = ISC_TRUE;
1532*00b67f09SDavid van Moolenbroek 	}
1533*00b67f09SDavid van Moolenbroek 
1534*00b67f09SDavid van Moolenbroek 	if (ctx->hi_water != 0U && !ctx->hi_called &&
1535*00b67f09SDavid van Moolenbroek 	    ctx->inuse > ctx->hi_water) {
1536*00b67f09SDavid van Moolenbroek 		ctx->hi_called = ISC_TRUE;
1537*00b67f09SDavid van Moolenbroek 		call_water = ISC_TRUE;
1538*00b67f09SDavid van Moolenbroek 	}
1539*00b67f09SDavid van Moolenbroek 	if (ctx->inuse > ctx->maxinuse) {
1540*00b67f09SDavid van Moolenbroek 		ctx->maxinuse = ctx->inuse;
1541*00b67f09SDavid van Moolenbroek 		if (ctx->hi_water != 0U && ctx->inuse > ctx->hi_water &&
1542*00b67f09SDavid van Moolenbroek 		    (isc_mem_debugging & ISC_MEM_DEBUGUSAGE) != 0)
1543*00b67f09SDavid van Moolenbroek 			fprintf(stderr, "maxinuse = %lu\n",
1544*00b67f09SDavid van Moolenbroek 				(unsigned long)ctx->inuse);
1545*00b67f09SDavid van Moolenbroek 	}
1546*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1547*00b67f09SDavid van Moolenbroek 
1548*00b67f09SDavid van Moolenbroek 	if (call_water)
1549*00b67f09SDavid van Moolenbroek 		(ctx->water)(ctx->water_arg, ISC_MEM_HIWATER);
1550*00b67f09SDavid van Moolenbroek 
1551*00b67f09SDavid van Moolenbroek 	return (si);
1552*00b67f09SDavid van Moolenbroek }
1553*00b67f09SDavid van Moolenbroek 
1554*00b67f09SDavid van Moolenbroek void *
isc___mem_reallocate(isc_mem_t * ctx0,void * ptr,size_t size FLARG)1555*00b67f09SDavid van Moolenbroek isc___mem_reallocate(isc_mem_t *ctx0, void *ptr, size_t size FLARG) {
1556*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1557*00b67f09SDavid van Moolenbroek 	void *new_ptr = NULL;
1558*00b67f09SDavid van Moolenbroek 	size_t oldsize, copysize;
1559*00b67f09SDavid van Moolenbroek 
1560*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1561*00b67f09SDavid van Moolenbroek 
1562*00b67f09SDavid van Moolenbroek 	/*
1563*00b67f09SDavid van Moolenbroek 	 * This function emulates the realloc(3) standard library function:
1564*00b67f09SDavid van Moolenbroek 	 * - if size > 0, allocate new memory; and if ptr is non NULL, copy
1565*00b67f09SDavid van Moolenbroek 	 *   as much of the old contents to the new buffer and free the old one.
1566*00b67f09SDavid van Moolenbroek 	 *   Note that when allocation fails the original pointer is intact;
1567*00b67f09SDavid van Moolenbroek 	 *   the caller must free it.
1568*00b67f09SDavid van Moolenbroek 	 * - if size is 0 and ptr is non NULL, simply free the given ptr.
1569*00b67f09SDavid van Moolenbroek 	 * - this function returns:
1570*00b67f09SDavid van Moolenbroek 	 *     pointer to the newly allocated memory, or
1571*00b67f09SDavid van Moolenbroek 	 *     NULL if allocation fails or doesn't happen.
1572*00b67f09SDavid van Moolenbroek 	 */
1573*00b67f09SDavid van Moolenbroek 	if (size > 0U) {
1574*00b67f09SDavid van Moolenbroek 		new_ptr = isc__mem_allocate(ctx0, size FLARG_PASS);
1575*00b67f09SDavid van Moolenbroek 		if (new_ptr != NULL && ptr != NULL) {
1576*00b67f09SDavid van Moolenbroek 			oldsize = (((size_info *)ptr)[-1]).u.size;
1577*00b67f09SDavid van Moolenbroek 			INSIST(oldsize >= ALIGNMENT_SIZE);
1578*00b67f09SDavid van Moolenbroek 			oldsize -= ALIGNMENT_SIZE;
1579*00b67f09SDavid van Moolenbroek 			if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0) {
1580*00b67f09SDavid van Moolenbroek 				INSIST(oldsize >= ALIGNMENT_SIZE);
1581*00b67f09SDavid van Moolenbroek 				oldsize -= ALIGNMENT_SIZE;
1582*00b67f09SDavid van Moolenbroek 			}
1583*00b67f09SDavid van Moolenbroek 			copysize = (oldsize > size) ? size : oldsize;
1584*00b67f09SDavid van Moolenbroek 			memmove(new_ptr, ptr, copysize);
1585*00b67f09SDavid van Moolenbroek 			isc__mem_free(ctx0, ptr FLARG_PASS);
1586*00b67f09SDavid van Moolenbroek 		}
1587*00b67f09SDavid van Moolenbroek 	} else if (ptr != NULL)
1588*00b67f09SDavid van Moolenbroek 		isc__mem_free(ctx0, ptr FLARG_PASS);
1589*00b67f09SDavid van Moolenbroek 
1590*00b67f09SDavid van Moolenbroek 	return (new_ptr);
1591*00b67f09SDavid van Moolenbroek }
1592*00b67f09SDavid van Moolenbroek 
1593*00b67f09SDavid van Moolenbroek void
isc___mem_free(isc_mem_t * ctx0,void * ptr FLARG)1594*00b67f09SDavid van Moolenbroek isc___mem_free(isc_mem_t *ctx0, void *ptr FLARG) {
1595*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1596*00b67f09SDavid van Moolenbroek 	size_info *si;
1597*00b67f09SDavid van Moolenbroek 	size_t size;
1598*00b67f09SDavid van Moolenbroek 	isc_boolean_t call_water= ISC_FALSE;
1599*00b67f09SDavid van Moolenbroek 
1600*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1601*00b67f09SDavid van Moolenbroek 	REQUIRE(ptr != NULL);
1602*00b67f09SDavid van Moolenbroek 
1603*00b67f09SDavid van Moolenbroek 	if ((isc_mem_debugging & ISC_MEM_DEBUGCTX) != 0) {
1604*00b67f09SDavid van Moolenbroek 		si = &(((size_info *)ptr)[-2]);
1605*00b67f09SDavid van Moolenbroek 		REQUIRE(si->u.ctx == ctx);
1606*00b67f09SDavid van Moolenbroek 		size = si[1].u.size;
1607*00b67f09SDavid van Moolenbroek 	} else {
1608*00b67f09SDavid van Moolenbroek 		si = &(((size_info *)ptr)[-1]);
1609*00b67f09SDavid van Moolenbroek 		size = si->u.size;
1610*00b67f09SDavid van Moolenbroek 	}
1611*00b67f09SDavid van Moolenbroek 
1612*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1613*00b67f09SDavid van Moolenbroek 
1614*00b67f09SDavid van Moolenbroek 	DELETE_TRACE(ctx, ptr, size, file, line);
1615*00b67f09SDavid van Moolenbroek 
1616*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1617*00b67f09SDavid van Moolenbroek 		mem_putunlocked(ctx, si, size);
1618*00b67f09SDavid van Moolenbroek 	} else {
1619*00b67f09SDavid van Moolenbroek 		mem_putstats(ctx, si, size);
1620*00b67f09SDavid van Moolenbroek 		mem_put(ctx, si, size);
1621*00b67f09SDavid van Moolenbroek 	}
1622*00b67f09SDavid van Moolenbroek 
1623*00b67f09SDavid van Moolenbroek 	/*
1624*00b67f09SDavid van Moolenbroek 	 * The check against ctx->lo_water == 0 is for the condition
1625*00b67f09SDavid van Moolenbroek 	 * when the context was pushed over hi_water but then had
1626*00b67f09SDavid van Moolenbroek 	 * isc_mem_setwater() called with 0 for hi_water and lo_water.
1627*00b67f09SDavid van Moolenbroek 	 */
1628*00b67f09SDavid van Moolenbroek 	if (ctx->is_overmem &&
1629*00b67f09SDavid van Moolenbroek 	    (ctx->inuse < ctx->lo_water || ctx->lo_water == 0U)) {
1630*00b67f09SDavid van Moolenbroek 		ctx->is_overmem = ISC_FALSE;
1631*00b67f09SDavid van Moolenbroek 	}
1632*00b67f09SDavid van Moolenbroek 
1633*00b67f09SDavid van Moolenbroek 	if (ctx->hi_called &&
1634*00b67f09SDavid van Moolenbroek 	    (ctx->inuse < ctx->lo_water || ctx->lo_water == 0U)) {
1635*00b67f09SDavid van Moolenbroek 		ctx->hi_called = ISC_FALSE;
1636*00b67f09SDavid van Moolenbroek 
1637*00b67f09SDavid van Moolenbroek 		if (ctx->water != NULL)
1638*00b67f09SDavid van Moolenbroek 			call_water = ISC_TRUE;
1639*00b67f09SDavid van Moolenbroek 	}
1640*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1641*00b67f09SDavid van Moolenbroek 
1642*00b67f09SDavid van Moolenbroek 	if (call_water)
1643*00b67f09SDavid van Moolenbroek 		(ctx->water)(ctx->water_arg, ISC_MEM_LOWATER);
1644*00b67f09SDavid van Moolenbroek }
1645*00b67f09SDavid van Moolenbroek 
1646*00b67f09SDavid van Moolenbroek 
1647*00b67f09SDavid van Moolenbroek /*
1648*00b67f09SDavid van Moolenbroek  * Other useful things.
1649*00b67f09SDavid van Moolenbroek  */
1650*00b67f09SDavid van Moolenbroek 
1651*00b67f09SDavid van Moolenbroek char *
isc___mem_strdup(isc_mem_t * mctx0,const char * s FLARG)1652*00b67f09SDavid van Moolenbroek isc___mem_strdup(isc_mem_t *mctx0, const char *s FLARG) {
1653*00b67f09SDavid van Moolenbroek 	isc__mem_t *mctx = (isc__mem_t *)mctx0;
1654*00b67f09SDavid van Moolenbroek 	size_t len;
1655*00b67f09SDavid van Moolenbroek 	char *ns;
1656*00b67f09SDavid van Moolenbroek 
1657*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(mctx));
1658*00b67f09SDavid van Moolenbroek 	REQUIRE(s != NULL);
1659*00b67f09SDavid van Moolenbroek 
1660*00b67f09SDavid van Moolenbroek 	len = strlen(s);
1661*00b67f09SDavid van Moolenbroek 
1662*00b67f09SDavid van Moolenbroek 	ns = isc__mem_allocate((isc_mem_t *)mctx, len + 1 FLARG_PASS);
1663*00b67f09SDavid van Moolenbroek 
1664*00b67f09SDavid van Moolenbroek 	if (ns != NULL)
1665*00b67f09SDavid van Moolenbroek 		strncpy(ns, s, len + 1);
1666*00b67f09SDavid van Moolenbroek 
1667*00b67f09SDavid van Moolenbroek 	return (ns);
1668*00b67f09SDavid van Moolenbroek }
1669*00b67f09SDavid van Moolenbroek 
1670*00b67f09SDavid van Moolenbroek void
isc__mem_setdestroycheck(isc_mem_t * ctx0,isc_boolean_t flag)1671*00b67f09SDavid van Moolenbroek isc__mem_setdestroycheck(isc_mem_t *ctx0, isc_boolean_t flag) {
1672*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1673*00b67f09SDavid van Moolenbroek 
1674*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1675*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1676*00b67f09SDavid van Moolenbroek 
1677*00b67f09SDavid van Moolenbroek 	ctx->checkfree = flag;
1678*00b67f09SDavid van Moolenbroek 
1679*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1680*00b67f09SDavid van Moolenbroek }
1681*00b67f09SDavid van Moolenbroek 
1682*00b67f09SDavid van Moolenbroek /*
1683*00b67f09SDavid van Moolenbroek  * Quotas
1684*00b67f09SDavid van Moolenbroek  */
1685*00b67f09SDavid van Moolenbroek 
1686*00b67f09SDavid van Moolenbroek void
isc_mem_setquota(isc_mem_t * ctx0,size_t quota)1687*00b67f09SDavid van Moolenbroek isc_mem_setquota(isc_mem_t *ctx0, size_t quota) {
1688*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1689*00b67f09SDavid van Moolenbroek 
1690*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1691*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1692*00b67f09SDavid van Moolenbroek 
1693*00b67f09SDavid van Moolenbroek 	ctx->quota = quota;
1694*00b67f09SDavid van Moolenbroek 
1695*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1696*00b67f09SDavid van Moolenbroek }
1697*00b67f09SDavid van Moolenbroek 
1698*00b67f09SDavid van Moolenbroek size_t
isc_mem_getquota(isc_mem_t * ctx0)1699*00b67f09SDavid van Moolenbroek isc_mem_getquota(isc_mem_t *ctx0) {
1700*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1701*00b67f09SDavid van Moolenbroek 	size_t quota;
1702*00b67f09SDavid van Moolenbroek 
1703*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1704*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1705*00b67f09SDavid van Moolenbroek 
1706*00b67f09SDavid van Moolenbroek 	quota = ctx->quota;
1707*00b67f09SDavid van Moolenbroek 
1708*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1709*00b67f09SDavid van Moolenbroek 
1710*00b67f09SDavid van Moolenbroek 	return (quota);
1711*00b67f09SDavid van Moolenbroek }
1712*00b67f09SDavid van Moolenbroek 
1713*00b67f09SDavid van Moolenbroek size_t
isc__mem_inuse(isc_mem_t * ctx0)1714*00b67f09SDavid van Moolenbroek isc__mem_inuse(isc_mem_t *ctx0) {
1715*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1716*00b67f09SDavid van Moolenbroek 	size_t inuse;
1717*00b67f09SDavid van Moolenbroek 
1718*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1719*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1720*00b67f09SDavid van Moolenbroek 
1721*00b67f09SDavid van Moolenbroek 	inuse = ctx->inuse;
1722*00b67f09SDavid van Moolenbroek 
1723*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1724*00b67f09SDavid van Moolenbroek 
1725*00b67f09SDavid van Moolenbroek 	return (inuse);
1726*00b67f09SDavid van Moolenbroek }
1727*00b67f09SDavid van Moolenbroek 
1728*00b67f09SDavid van Moolenbroek size_t
isc__mem_maxinuse(isc_mem_t * ctx0)1729*00b67f09SDavid van Moolenbroek isc__mem_maxinuse(isc_mem_t *ctx0) {
1730*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1731*00b67f09SDavid van Moolenbroek 	size_t maxinuse;
1732*00b67f09SDavid van Moolenbroek 
1733*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1734*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1735*00b67f09SDavid van Moolenbroek 
1736*00b67f09SDavid van Moolenbroek 	maxinuse = ctx->maxinuse;
1737*00b67f09SDavid van Moolenbroek 
1738*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1739*00b67f09SDavid van Moolenbroek 
1740*00b67f09SDavid van Moolenbroek 	return (maxinuse);
1741*00b67f09SDavid van Moolenbroek }
1742*00b67f09SDavid van Moolenbroek 
1743*00b67f09SDavid van Moolenbroek size_t
isc__mem_total(isc_mem_t * ctx0)1744*00b67f09SDavid van Moolenbroek isc__mem_total(isc_mem_t *ctx0) {
1745*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1746*00b67f09SDavid van Moolenbroek 	size_t total;
1747*00b67f09SDavid van Moolenbroek 
1748*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1749*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1750*00b67f09SDavid van Moolenbroek 
1751*00b67f09SDavid van Moolenbroek 	total = ctx->total;
1752*00b67f09SDavid van Moolenbroek 
1753*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1754*00b67f09SDavid van Moolenbroek 
1755*00b67f09SDavid van Moolenbroek 	return (total);
1756*00b67f09SDavid van Moolenbroek }
1757*00b67f09SDavid van Moolenbroek 
1758*00b67f09SDavid van Moolenbroek void
isc__mem_setwater(isc_mem_t * ctx0,isc_mem_water_t water,void * water_arg,size_t hiwater,size_t lowater)1759*00b67f09SDavid van Moolenbroek isc__mem_setwater(isc_mem_t *ctx0, isc_mem_water_t water, void *water_arg,
1760*00b67f09SDavid van Moolenbroek 		  size_t hiwater, size_t lowater)
1761*00b67f09SDavid van Moolenbroek {
1762*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1763*00b67f09SDavid van Moolenbroek 	isc_boolean_t callwater = ISC_FALSE;
1764*00b67f09SDavid van Moolenbroek 	isc_mem_water_t oldwater;
1765*00b67f09SDavid van Moolenbroek 	void *oldwater_arg;
1766*00b67f09SDavid van Moolenbroek 
1767*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1768*00b67f09SDavid van Moolenbroek 	REQUIRE(hiwater >= lowater);
1769*00b67f09SDavid van Moolenbroek 
1770*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
1771*00b67f09SDavid van Moolenbroek 	oldwater = ctx->water;
1772*00b67f09SDavid van Moolenbroek 	oldwater_arg = ctx->water_arg;
1773*00b67f09SDavid van Moolenbroek 	if (water == NULL) {
1774*00b67f09SDavid van Moolenbroek 		callwater = ctx->hi_called;
1775*00b67f09SDavid van Moolenbroek 		ctx->water = NULL;
1776*00b67f09SDavid van Moolenbroek 		ctx->water_arg = NULL;
1777*00b67f09SDavid van Moolenbroek 		ctx->hi_water = 0;
1778*00b67f09SDavid van Moolenbroek 		ctx->lo_water = 0;
1779*00b67f09SDavid van Moolenbroek 		ctx->hi_called = ISC_FALSE;
1780*00b67f09SDavid van Moolenbroek 	} else {
1781*00b67f09SDavid van Moolenbroek 		if (ctx->hi_called &&
1782*00b67f09SDavid van Moolenbroek 		    (ctx->water != water || ctx->water_arg != water_arg ||
1783*00b67f09SDavid van Moolenbroek 		     ctx->inuse < lowater || lowater == 0U))
1784*00b67f09SDavid van Moolenbroek 			callwater = ISC_TRUE;
1785*00b67f09SDavid van Moolenbroek 		ctx->water = water;
1786*00b67f09SDavid van Moolenbroek 		ctx->water_arg = water_arg;
1787*00b67f09SDavid van Moolenbroek 		ctx->hi_water = hiwater;
1788*00b67f09SDavid van Moolenbroek 		ctx->lo_water = lowater;
1789*00b67f09SDavid van Moolenbroek 	}
1790*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
1791*00b67f09SDavid van Moolenbroek 
1792*00b67f09SDavid van Moolenbroek 	if (callwater && oldwater != NULL)
1793*00b67f09SDavid van Moolenbroek 		(oldwater)(oldwater_arg, ISC_MEM_LOWATER);
1794*00b67f09SDavid van Moolenbroek }
1795*00b67f09SDavid van Moolenbroek 
1796*00b67f09SDavid van Moolenbroek isc_boolean_t
isc__mem_isovermem(isc_mem_t * ctx0)1797*00b67f09SDavid van Moolenbroek isc__mem_isovermem(isc_mem_t *ctx0) {
1798*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1799*00b67f09SDavid van Moolenbroek 
1800*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1801*00b67f09SDavid van Moolenbroek 
1802*00b67f09SDavid van Moolenbroek 	/*
1803*00b67f09SDavid van Moolenbroek 	 * We don't bother to lock the context because 100% accuracy isn't
1804*00b67f09SDavid van Moolenbroek 	 * necessary (and even if we locked the context the returned value
1805*00b67f09SDavid van Moolenbroek 	 * could be different from the actual state when it's used anyway)
1806*00b67f09SDavid van Moolenbroek 	 */
1807*00b67f09SDavid van Moolenbroek 	return (ctx->is_overmem);
1808*00b67f09SDavid van Moolenbroek }
1809*00b67f09SDavid van Moolenbroek 
1810*00b67f09SDavid van Moolenbroek void
isc_mem_setname(isc_mem_t * ctx0,const char * name,void * tag)1811*00b67f09SDavid van Moolenbroek isc_mem_setname(isc_mem_t *ctx0, const char *name, void *tag) {
1812*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1813*00b67f09SDavid van Moolenbroek 
1814*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1815*00b67f09SDavid van Moolenbroek 
1816*00b67f09SDavid van Moolenbroek 	LOCK(&ctx->lock);
1817*00b67f09SDavid van Moolenbroek 	memset(ctx->name, 0, sizeof(ctx->name));
1818*00b67f09SDavid van Moolenbroek 	strncpy(ctx->name, name, sizeof(ctx->name) - 1);
1819*00b67f09SDavid van Moolenbroek 	ctx->tag = tag;
1820*00b67f09SDavid van Moolenbroek 	UNLOCK(&ctx->lock);
1821*00b67f09SDavid van Moolenbroek }
1822*00b67f09SDavid van Moolenbroek 
1823*00b67f09SDavid van Moolenbroek const char *
isc_mem_getname(isc_mem_t * ctx0)1824*00b67f09SDavid van Moolenbroek isc_mem_getname(isc_mem_t *ctx0) {
1825*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1826*00b67f09SDavid van Moolenbroek 
1827*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1828*00b67f09SDavid van Moolenbroek 
1829*00b67f09SDavid van Moolenbroek 	if (ctx->name[0] == 0)
1830*00b67f09SDavid van Moolenbroek 		return ("");
1831*00b67f09SDavid van Moolenbroek 
1832*00b67f09SDavid van Moolenbroek 	return (ctx->name);
1833*00b67f09SDavid van Moolenbroek }
1834*00b67f09SDavid van Moolenbroek 
1835*00b67f09SDavid van Moolenbroek void *
isc_mem_gettag(isc_mem_t * ctx0)1836*00b67f09SDavid van Moolenbroek isc_mem_gettag(isc_mem_t *ctx0) {
1837*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
1838*00b67f09SDavid van Moolenbroek 
1839*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
1840*00b67f09SDavid van Moolenbroek 
1841*00b67f09SDavid van Moolenbroek 	return (ctx->tag);
1842*00b67f09SDavid van Moolenbroek }
1843*00b67f09SDavid van Moolenbroek 
1844*00b67f09SDavid van Moolenbroek /*
1845*00b67f09SDavid van Moolenbroek  * Memory pool stuff
1846*00b67f09SDavid van Moolenbroek  */
1847*00b67f09SDavid van Moolenbroek 
1848*00b67f09SDavid van Moolenbroek isc_result_t
isc__mempool_create(isc_mem_t * mctx0,size_t size,isc_mempool_t ** mpctxp)1849*00b67f09SDavid van Moolenbroek isc__mempool_create(isc_mem_t *mctx0, size_t size, isc_mempool_t **mpctxp) {
1850*00b67f09SDavid van Moolenbroek 	isc__mem_t *mctx = (isc__mem_t *)mctx0;
1851*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx;
1852*00b67f09SDavid van Moolenbroek 
1853*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(mctx));
1854*00b67f09SDavid van Moolenbroek 	REQUIRE(size > 0U);
1855*00b67f09SDavid van Moolenbroek 	REQUIRE(mpctxp != NULL && *mpctxp == NULL);
1856*00b67f09SDavid van Moolenbroek 
1857*00b67f09SDavid van Moolenbroek 	/*
1858*00b67f09SDavid van Moolenbroek 	 * Allocate space for this pool, initialize values, and if all works
1859*00b67f09SDavid van Moolenbroek 	 * well, attach to the memory context.
1860*00b67f09SDavid van Moolenbroek 	 */
1861*00b67f09SDavid van Moolenbroek 	mpctx = isc_mem_get((isc_mem_t *)mctx, sizeof(isc__mempool_t));
1862*00b67f09SDavid van Moolenbroek 	if (mpctx == NULL)
1863*00b67f09SDavid van Moolenbroek 		return (ISC_R_NOMEMORY);
1864*00b67f09SDavid van Moolenbroek 
1865*00b67f09SDavid van Moolenbroek 	mpctx->common.methods = (isc_mempoolmethods_t *)&mempoolmethods;
1866*00b67f09SDavid van Moolenbroek 	mpctx->common.impmagic = MEMPOOL_MAGIC;
1867*00b67f09SDavid van Moolenbroek 	mpctx->common.magic = ISCAPI_MPOOL_MAGIC;
1868*00b67f09SDavid van Moolenbroek 	mpctx->lock = NULL;
1869*00b67f09SDavid van Moolenbroek 	mpctx->mctx = mctx;
1870*00b67f09SDavid van Moolenbroek 	mpctx->size = size;
1871*00b67f09SDavid van Moolenbroek 	mpctx->maxalloc = UINT_MAX;
1872*00b67f09SDavid van Moolenbroek 	mpctx->allocated = 0;
1873*00b67f09SDavid van Moolenbroek 	mpctx->freecount = 0;
1874*00b67f09SDavid van Moolenbroek 	mpctx->freemax = 1;
1875*00b67f09SDavid van Moolenbroek 	mpctx->fillcount = 1;
1876*00b67f09SDavid van Moolenbroek 	mpctx->gets = 0;
1877*00b67f09SDavid van Moolenbroek #if ISC_MEMPOOL_NAMES
1878*00b67f09SDavid van Moolenbroek 	mpctx->name[0] = 0;
1879*00b67f09SDavid van Moolenbroek #endif
1880*00b67f09SDavid van Moolenbroek 	mpctx->items = NULL;
1881*00b67f09SDavid van Moolenbroek 
1882*00b67f09SDavid van Moolenbroek 	*mpctxp = (isc_mempool_t *)mpctx;
1883*00b67f09SDavid van Moolenbroek 
1884*00b67f09SDavid van Moolenbroek 	MCTXLOCK(mctx, &mctx->lock);
1885*00b67f09SDavid van Moolenbroek 	ISC_LIST_INITANDAPPEND(mctx->pools, mpctx, link);
1886*00b67f09SDavid van Moolenbroek 	mctx->poolcnt++;
1887*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(mctx, &mctx->lock);
1888*00b67f09SDavid van Moolenbroek 
1889*00b67f09SDavid van Moolenbroek 	return (ISC_R_SUCCESS);
1890*00b67f09SDavid van Moolenbroek }
1891*00b67f09SDavid van Moolenbroek 
1892*00b67f09SDavid van Moolenbroek void
isc__mempool_setname(isc_mempool_t * mpctx0,const char * name)1893*00b67f09SDavid van Moolenbroek isc__mempool_setname(isc_mempool_t *mpctx0, const char *name) {
1894*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
1895*00b67f09SDavid van Moolenbroek 
1896*00b67f09SDavid van Moolenbroek 	REQUIRE(name != NULL);
1897*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
1898*00b67f09SDavid van Moolenbroek 
1899*00b67f09SDavid van Moolenbroek #if ISC_MEMPOOL_NAMES
1900*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
1901*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
1902*00b67f09SDavid van Moolenbroek 
1903*00b67f09SDavid van Moolenbroek 	strncpy(mpctx->name, name, sizeof(mpctx->name) - 1);
1904*00b67f09SDavid van Moolenbroek 	mpctx->name[sizeof(mpctx->name) - 1] = '\0';
1905*00b67f09SDavid van Moolenbroek 
1906*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
1907*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
1908*00b67f09SDavid van Moolenbroek #else
1909*00b67f09SDavid van Moolenbroek 	UNUSED(mpctx);
1910*00b67f09SDavid van Moolenbroek 	UNUSED(name);
1911*00b67f09SDavid van Moolenbroek #endif
1912*00b67f09SDavid van Moolenbroek }
1913*00b67f09SDavid van Moolenbroek 
1914*00b67f09SDavid van Moolenbroek void
isc__mempool_destroy(isc_mempool_t ** mpctxp)1915*00b67f09SDavid van Moolenbroek isc__mempool_destroy(isc_mempool_t **mpctxp) {
1916*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx;
1917*00b67f09SDavid van Moolenbroek 	isc__mem_t *mctx;
1918*00b67f09SDavid van Moolenbroek 	isc_mutex_t *lock;
1919*00b67f09SDavid van Moolenbroek 	element *item;
1920*00b67f09SDavid van Moolenbroek 
1921*00b67f09SDavid van Moolenbroek 	REQUIRE(mpctxp != NULL);
1922*00b67f09SDavid van Moolenbroek 	mpctx = (isc__mempool_t *)*mpctxp;
1923*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
1924*00b67f09SDavid van Moolenbroek #if ISC_MEMPOOL_NAMES
1925*00b67f09SDavid van Moolenbroek 	if (mpctx->allocated > 0)
1926*00b67f09SDavid van Moolenbroek 		UNEXPECTED_ERROR(__FILE__, __LINE__,
1927*00b67f09SDavid van Moolenbroek 				 "isc__mempool_destroy(): mempool %s "
1928*00b67f09SDavid van Moolenbroek 				 "leaked memory",
1929*00b67f09SDavid van Moolenbroek 				 mpctx->name);
1930*00b67f09SDavid van Moolenbroek #endif
1931*00b67f09SDavid van Moolenbroek 	REQUIRE(mpctx->allocated == 0);
1932*00b67f09SDavid van Moolenbroek 
1933*00b67f09SDavid van Moolenbroek 	mctx = mpctx->mctx;
1934*00b67f09SDavid van Moolenbroek 
1935*00b67f09SDavid van Moolenbroek 	lock = mpctx->lock;
1936*00b67f09SDavid van Moolenbroek 
1937*00b67f09SDavid van Moolenbroek 	if (lock != NULL)
1938*00b67f09SDavid van Moolenbroek 		LOCK(lock);
1939*00b67f09SDavid van Moolenbroek 
1940*00b67f09SDavid van Moolenbroek 	/*
1941*00b67f09SDavid van Moolenbroek 	 * Return any items on the free list
1942*00b67f09SDavid van Moolenbroek 	 */
1943*00b67f09SDavid van Moolenbroek 	MCTXLOCK(mctx, &mctx->lock);
1944*00b67f09SDavid van Moolenbroek 	while (mpctx->items != NULL) {
1945*00b67f09SDavid van Moolenbroek 		INSIST(mpctx->freecount > 0);
1946*00b67f09SDavid van Moolenbroek 		mpctx->freecount--;
1947*00b67f09SDavid van Moolenbroek 		item = mpctx->items;
1948*00b67f09SDavid van Moolenbroek 		mpctx->items = item->next;
1949*00b67f09SDavid van Moolenbroek 
1950*00b67f09SDavid van Moolenbroek 		if ((mctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
1951*00b67f09SDavid van Moolenbroek 			mem_putunlocked(mctx, item, mpctx->size);
1952*00b67f09SDavid van Moolenbroek 		} else {
1953*00b67f09SDavid van Moolenbroek 			mem_putstats(mctx, item, mpctx->size);
1954*00b67f09SDavid van Moolenbroek 			mem_put(mctx, item, mpctx->size);
1955*00b67f09SDavid van Moolenbroek 		}
1956*00b67f09SDavid van Moolenbroek 	}
1957*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(mctx, &mctx->lock);
1958*00b67f09SDavid van Moolenbroek 
1959*00b67f09SDavid van Moolenbroek 	/*
1960*00b67f09SDavid van Moolenbroek 	 * Remove our linked list entry from the memory context.
1961*00b67f09SDavid van Moolenbroek 	 */
1962*00b67f09SDavid van Moolenbroek 	MCTXLOCK(mctx, &mctx->lock);
1963*00b67f09SDavid van Moolenbroek 	ISC_LIST_UNLINK(mctx->pools, mpctx, link);
1964*00b67f09SDavid van Moolenbroek 	mctx->poolcnt--;
1965*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(mctx, &mctx->lock);
1966*00b67f09SDavid van Moolenbroek 
1967*00b67f09SDavid van Moolenbroek 	mpctx->common.impmagic = 0;
1968*00b67f09SDavid van Moolenbroek 	mpctx->common.magic = 0;
1969*00b67f09SDavid van Moolenbroek 
1970*00b67f09SDavid van Moolenbroek 	isc_mem_put((isc_mem_t *)mpctx->mctx, mpctx, sizeof(isc__mempool_t));
1971*00b67f09SDavid van Moolenbroek 
1972*00b67f09SDavid van Moolenbroek 	if (lock != NULL)
1973*00b67f09SDavid van Moolenbroek 		UNLOCK(lock);
1974*00b67f09SDavid van Moolenbroek 
1975*00b67f09SDavid van Moolenbroek 	*mpctxp = NULL;
1976*00b67f09SDavid van Moolenbroek }
1977*00b67f09SDavid van Moolenbroek 
1978*00b67f09SDavid van Moolenbroek void
isc__mempool_associatelock(isc_mempool_t * mpctx0,isc_mutex_t * lock)1979*00b67f09SDavid van Moolenbroek isc__mempool_associatelock(isc_mempool_t *mpctx0, isc_mutex_t *lock) {
1980*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
1981*00b67f09SDavid van Moolenbroek 
1982*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
1983*00b67f09SDavid van Moolenbroek 	REQUIRE(mpctx->lock == NULL);
1984*00b67f09SDavid van Moolenbroek 	REQUIRE(lock != NULL);
1985*00b67f09SDavid van Moolenbroek 
1986*00b67f09SDavid van Moolenbroek 	mpctx->lock = lock;
1987*00b67f09SDavid van Moolenbroek }
1988*00b67f09SDavid van Moolenbroek 
1989*00b67f09SDavid van Moolenbroek void *
isc___mempool_get(isc_mempool_t * mpctx0 FLARG)1990*00b67f09SDavid van Moolenbroek isc___mempool_get(isc_mempool_t *mpctx0 FLARG) {
1991*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
1992*00b67f09SDavid van Moolenbroek 	element *item;
1993*00b67f09SDavid van Moolenbroek 	isc__mem_t *mctx;
1994*00b67f09SDavid van Moolenbroek 	unsigned int i;
1995*00b67f09SDavid van Moolenbroek 
1996*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
1997*00b67f09SDavid van Moolenbroek 
1998*00b67f09SDavid van Moolenbroek 	mctx = mpctx->mctx;
1999*00b67f09SDavid van Moolenbroek 
2000*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2001*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2002*00b67f09SDavid van Moolenbroek 
2003*00b67f09SDavid van Moolenbroek 	/*
2004*00b67f09SDavid van Moolenbroek 	 * Don't let the caller go over quota
2005*00b67f09SDavid van Moolenbroek 	 */
2006*00b67f09SDavid van Moolenbroek 	if (mpctx->allocated >= mpctx->maxalloc) {
2007*00b67f09SDavid van Moolenbroek 		item = NULL;
2008*00b67f09SDavid van Moolenbroek 		goto out;
2009*00b67f09SDavid van Moolenbroek 	}
2010*00b67f09SDavid van Moolenbroek 
2011*00b67f09SDavid van Moolenbroek 	/*
2012*00b67f09SDavid van Moolenbroek 	 * if we have a free list item, return the first here
2013*00b67f09SDavid van Moolenbroek 	 */
2014*00b67f09SDavid van Moolenbroek 	item = mpctx->items;
2015*00b67f09SDavid van Moolenbroek 	if (item != NULL) {
2016*00b67f09SDavid van Moolenbroek 		mpctx->items = item->next;
2017*00b67f09SDavid van Moolenbroek 		INSIST(mpctx->freecount > 0);
2018*00b67f09SDavid van Moolenbroek 		mpctx->freecount--;
2019*00b67f09SDavid van Moolenbroek 		mpctx->gets++;
2020*00b67f09SDavid van Moolenbroek 		mpctx->allocated++;
2021*00b67f09SDavid van Moolenbroek 		goto out;
2022*00b67f09SDavid van Moolenbroek 	}
2023*00b67f09SDavid van Moolenbroek 
2024*00b67f09SDavid van Moolenbroek 	/*
2025*00b67f09SDavid van Moolenbroek 	 * We need to dip into the well.  Lock the memory context here and
2026*00b67f09SDavid van Moolenbroek 	 * fill up our free list.
2027*00b67f09SDavid van Moolenbroek 	 */
2028*00b67f09SDavid van Moolenbroek 	MCTXLOCK(mctx, &mctx->lock);
2029*00b67f09SDavid van Moolenbroek 	for (i = 0; i < mpctx->fillcount; i++) {
2030*00b67f09SDavid van Moolenbroek 		if ((mctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
2031*00b67f09SDavid van Moolenbroek 			item = mem_getunlocked(mctx, mpctx->size);
2032*00b67f09SDavid van Moolenbroek 		} else {
2033*00b67f09SDavid van Moolenbroek 			item = mem_get(mctx, mpctx->size);
2034*00b67f09SDavid van Moolenbroek 			if (item != NULL)
2035*00b67f09SDavid van Moolenbroek 				mem_getstats(mctx, mpctx->size);
2036*00b67f09SDavid van Moolenbroek 		}
2037*00b67f09SDavid van Moolenbroek 		if (item == NULL)
2038*00b67f09SDavid van Moolenbroek 			break;
2039*00b67f09SDavid van Moolenbroek 		item->next = mpctx->items;
2040*00b67f09SDavid van Moolenbroek 		mpctx->items = item;
2041*00b67f09SDavid van Moolenbroek 		mpctx->freecount++;
2042*00b67f09SDavid van Moolenbroek 	}
2043*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(mctx, &mctx->lock);
2044*00b67f09SDavid van Moolenbroek 
2045*00b67f09SDavid van Moolenbroek 	/*
2046*00b67f09SDavid van Moolenbroek 	 * If we didn't get any items, return NULL.
2047*00b67f09SDavid van Moolenbroek 	 */
2048*00b67f09SDavid van Moolenbroek 	item = mpctx->items;
2049*00b67f09SDavid van Moolenbroek 	if (item == NULL)
2050*00b67f09SDavid van Moolenbroek 		goto out;
2051*00b67f09SDavid van Moolenbroek 
2052*00b67f09SDavid van Moolenbroek 	mpctx->items = item->next;
2053*00b67f09SDavid van Moolenbroek 	mpctx->freecount--;
2054*00b67f09SDavid van Moolenbroek 	mpctx->gets++;
2055*00b67f09SDavid van Moolenbroek 	mpctx->allocated++;
2056*00b67f09SDavid van Moolenbroek 
2057*00b67f09SDavid van Moolenbroek  out:
2058*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2059*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2060*00b67f09SDavid van Moolenbroek 
2061*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2062*00b67f09SDavid van Moolenbroek 	if (item != NULL) {
2063*00b67f09SDavid van Moolenbroek 		MCTXLOCK(mctx, &mctx->lock);
2064*00b67f09SDavid van Moolenbroek 		ADD_TRACE(mctx, item, mpctx->size, file, line);
2065*00b67f09SDavid van Moolenbroek 		MCTXUNLOCK(mctx, &mctx->lock);
2066*00b67f09SDavid van Moolenbroek 	}
2067*00b67f09SDavid van Moolenbroek #endif /* ISC_MEM_TRACKLINES */
2068*00b67f09SDavid van Moolenbroek 
2069*00b67f09SDavid van Moolenbroek 	return (item);
2070*00b67f09SDavid van Moolenbroek }
2071*00b67f09SDavid van Moolenbroek 
2072*00b67f09SDavid van Moolenbroek /* coverity[+free : arg-1] */
2073*00b67f09SDavid van Moolenbroek void
isc___mempool_put(isc_mempool_t * mpctx0,void * mem FLARG)2074*00b67f09SDavid van Moolenbroek isc___mempool_put(isc_mempool_t *mpctx0, void *mem FLARG) {
2075*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2076*00b67f09SDavid van Moolenbroek 	isc__mem_t *mctx;
2077*00b67f09SDavid van Moolenbroek 	element *item;
2078*00b67f09SDavid van Moolenbroek 
2079*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2080*00b67f09SDavid van Moolenbroek 	REQUIRE(mem != NULL);
2081*00b67f09SDavid van Moolenbroek 
2082*00b67f09SDavid van Moolenbroek 	mctx = mpctx->mctx;
2083*00b67f09SDavid van Moolenbroek 
2084*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2085*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2086*00b67f09SDavid van Moolenbroek 
2087*00b67f09SDavid van Moolenbroek 	INSIST(mpctx->allocated > 0);
2088*00b67f09SDavid van Moolenbroek 	mpctx->allocated--;
2089*00b67f09SDavid van Moolenbroek 
2090*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2091*00b67f09SDavid van Moolenbroek 	MCTXLOCK(mctx, &mctx->lock);
2092*00b67f09SDavid van Moolenbroek 	DELETE_TRACE(mctx, mem, mpctx->size, file, line);
2093*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(mctx, &mctx->lock);
2094*00b67f09SDavid van Moolenbroek #endif /* ISC_MEM_TRACKLINES */
2095*00b67f09SDavid van Moolenbroek 
2096*00b67f09SDavid van Moolenbroek 	/*
2097*00b67f09SDavid van Moolenbroek 	 * If our free list is full, return this to the mctx directly.
2098*00b67f09SDavid van Moolenbroek 	 */
2099*00b67f09SDavid van Moolenbroek 	if (mpctx->freecount >= mpctx->freemax) {
2100*00b67f09SDavid van Moolenbroek 		MCTXLOCK(mctx, &mctx->lock);
2101*00b67f09SDavid van Moolenbroek 		if ((mctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
2102*00b67f09SDavid van Moolenbroek 			mem_putunlocked(mctx, mem, mpctx->size);
2103*00b67f09SDavid van Moolenbroek 		} else {
2104*00b67f09SDavid van Moolenbroek 			mem_putstats(mctx, mem, mpctx->size);
2105*00b67f09SDavid van Moolenbroek 			mem_put(mctx, mem, mpctx->size);
2106*00b67f09SDavid van Moolenbroek 		}
2107*00b67f09SDavid van Moolenbroek 		MCTXUNLOCK(mctx, &mctx->lock);
2108*00b67f09SDavid van Moolenbroek 		if (mpctx->lock != NULL)
2109*00b67f09SDavid van Moolenbroek 			UNLOCK(mpctx->lock);
2110*00b67f09SDavid van Moolenbroek 		return;
2111*00b67f09SDavid van Moolenbroek 	}
2112*00b67f09SDavid van Moolenbroek 
2113*00b67f09SDavid van Moolenbroek 	/*
2114*00b67f09SDavid van Moolenbroek 	 * Otherwise, attach it to our free list and bump the counter.
2115*00b67f09SDavid van Moolenbroek 	 */
2116*00b67f09SDavid van Moolenbroek 	mpctx->freecount++;
2117*00b67f09SDavid van Moolenbroek 	item = (element *)mem;
2118*00b67f09SDavid van Moolenbroek 	item->next = mpctx->items;
2119*00b67f09SDavid van Moolenbroek 	mpctx->items = item;
2120*00b67f09SDavid van Moolenbroek 
2121*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2122*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2123*00b67f09SDavid van Moolenbroek }
2124*00b67f09SDavid van Moolenbroek 
2125*00b67f09SDavid van Moolenbroek /*
2126*00b67f09SDavid van Moolenbroek  * Quotas
2127*00b67f09SDavid van Moolenbroek  */
2128*00b67f09SDavid van Moolenbroek 
2129*00b67f09SDavid van Moolenbroek void
isc__mempool_setfreemax(isc_mempool_t * mpctx0,unsigned int limit)2130*00b67f09SDavid van Moolenbroek isc__mempool_setfreemax(isc_mempool_t *mpctx0, unsigned int limit) {
2131*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2132*00b67f09SDavid van Moolenbroek 
2133*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2134*00b67f09SDavid van Moolenbroek 
2135*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2136*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2137*00b67f09SDavid van Moolenbroek 
2138*00b67f09SDavid van Moolenbroek 	mpctx->freemax = limit;
2139*00b67f09SDavid van Moolenbroek 
2140*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2141*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2142*00b67f09SDavid van Moolenbroek }
2143*00b67f09SDavid van Moolenbroek 
2144*00b67f09SDavid van Moolenbroek unsigned int
isc_mempool_getfreemax(isc_mempool_t * mpctx0)2145*00b67f09SDavid van Moolenbroek isc_mempool_getfreemax(isc_mempool_t *mpctx0) {
2146*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2147*00b67f09SDavid van Moolenbroek 	unsigned int freemax;
2148*00b67f09SDavid van Moolenbroek 
2149*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2150*00b67f09SDavid van Moolenbroek 
2151*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2152*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2153*00b67f09SDavid van Moolenbroek 
2154*00b67f09SDavid van Moolenbroek 	freemax = mpctx->freemax;
2155*00b67f09SDavid van Moolenbroek 
2156*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2157*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2158*00b67f09SDavid van Moolenbroek 
2159*00b67f09SDavid van Moolenbroek 	return (freemax);
2160*00b67f09SDavid van Moolenbroek }
2161*00b67f09SDavid van Moolenbroek 
2162*00b67f09SDavid van Moolenbroek unsigned int
isc_mempool_getfreecount(isc_mempool_t * mpctx0)2163*00b67f09SDavid van Moolenbroek isc_mempool_getfreecount(isc_mempool_t *mpctx0) {
2164*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2165*00b67f09SDavid van Moolenbroek 	unsigned int freecount;
2166*00b67f09SDavid van Moolenbroek 
2167*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2168*00b67f09SDavid van Moolenbroek 
2169*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2170*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2171*00b67f09SDavid van Moolenbroek 
2172*00b67f09SDavid van Moolenbroek 	freecount = mpctx->freecount;
2173*00b67f09SDavid van Moolenbroek 
2174*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2175*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2176*00b67f09SDavid van Moolenbroek 
2177*00b67f09SDavid van Moolenbroek 	return (freecount);
2178*00b67f09SDavid van Moolenbroek }
2179*00b67f09SDavid van Moolenbroek 
2180*00b67f09SDavid van Moolenbroek void
isc__mempool_setmaxalloc(isc_mempool_t * mpctx0,unsigned int limit)2181*00b67f09SDavid van Moolenbroek isc__mempool_setmaxalloc(isc_mempool_t *mpctx0, unsigned int limit) {
2182*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2183*00b67f09SDavid van Moolenbroek 
2184*00b67f09SDavid van Moolenbroek 	REQUIRE(limit > 0);
2185*00b67f09SDavid van Moolenbroek 
2186*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2187*00b67f09SDavid van Moolenbroek 
2188*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2189*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2190*00b67f09SDavid van Moolenbroek 
2191*00b67f09SDavid van Moolenbroek 	mpctx->maxalloc = limit;
2192*00b67f09SDavid van Moolenbroek 
2193*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2194*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2195*00b67f09SDavid van Moolenbroek }
2196*00b67f09SDavid van Moolenbroek 
2197*00b67f09SDavid van Moolenbroek unsigned int
isc_mempool_getmaxalloc(isc_mempool_t * mpctx0)2198*00b67f09SDavid van Moolenbroek isc_mempool_getmaxalloc(isc_mempool_t *mpctx0) {
2199*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2200*00b67f09SDavid van Moolenbroek 	unsigned int maxalloc;
2201*00b67f09SDavid van Moolenbroek 
2202*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2203*00b67f09SDavid van Moolenbroek 
2204*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2205*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2206*00b67f09SDavid van Moolenbroek 
2207*00b67f09SDavid van Moolenbroek 	maxalloc = mpctx->maxalloc;
2208*00b67f09SDavid van Moolenbroek 
2209*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2210*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2211*00b67f09SDavid van Moolenbroek 
2212*00b67f09SDavid van Moolenbroek 	return (maxalloc);
2213*00b67f09SDavid van Moolenbroek }
2214*00b67f09SDavid van Moolenbroek 
2215*00b67f09SDavid van Moolenbroek unsigned int
isc__mempool_getallocated(isc_mempool_t * mpctx0)2216*00b67f09SDavid van Moolenbroek isc__mempool_getallocated(isc_mempool_t *mpctx0) {
2217*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2218*00b67f09SDavid van Moolenbroek 	unsigned int allocated;
2219*00b67f09SDavid van Moolenbroek 
2220*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2221*00b67f09SDavid van Moolenbroek 
2222*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2223*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2224*00b67f09SDavid van Moolenbroek 
2225*00b67f09SDavid van Moolenbroek 	allocated = mpctx->allocated;
2226*00b67f09SDavid van Moolenbroek 
2227*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2228*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2229*00b67f09SDavid van Moolenbroek 
2230*00b67f09SDavid van Moolenbroek 	return (allocated);
2231*00b67f09SDavid van Moolenbroek }
2232*00b67f09SDavid van Moolenbroek 
2233*00b67f09SDavid van Moolenbroek void
isc__mempool_setfillcount(isc_mempool_t * mpctx0,unsigned int limit)2234*00b67f09SDavid van Moolenbroek isc__mempool_setfillcount(isc_mempool_t *mpctx0, unsigned int limit) {
2235*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2236*00b67f09SDavid van Moolenbroek 
2237*00b67f09SDavid van Moolenbroek 	REQUIRE(limit > 0);
2238*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2239*00b67f09SDavid van Moolenbroek 
2240*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2241*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2242*00b67f09SDavid van Moolenbroek 
2243*00b67f09SDavid van Moolenbroek 	mpctx->fillcount = limit;
2244*00b67f09SDavid van Moolenbroek 
2245*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2246*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2247*00b67f09SDavid van Moolenbroek }
2248*00b67f09SDavid van Moolenbroek 
2249*00b67f09SDavid van Moolenbroek unsigned int
isc_mempool_getfillcount(isc_mempool_t * mpctx0)2250*00b67f09SDavid van Moolenbroek isc_mempool_getfillcount(isc_mempool_t *mpctx0) {
2251*00b67f09SDavid van Moolenbroek 	isc__mempool_t *mpctx = (isc__mempool_t *)mpctx0;
2252*00b67f09SDavid van Moolenbroek 
2253*00b67f09SDavid van Moolenbroek 	unsigned int fillcount;
2254*00b67f09SDavid van Moolenbroek 
2255*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_MEMPOOL(mpctx));
2256*00b67f09SDavid van Moolenbroek 
2257*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2258*00b67f09SDavid van Moolenbroek 		LOCK(mpctx->lock);
2259*00b67f09SDavid van Moolenbroek 
2260*00b67f09SDavid van Moolenbroek 	fillcount = mpctx->fillcount;
2261*00b67f09SDavid van Moolenbroek 
2262*00b67f09SDavid van Moolenbroek 	if (mpctx->lock != NULL)
2263*00b67f09SDavid van Moolenbroek 		UNLOCK(mpctx->lock);
2264*00b67f09SDavid van Moolenbroek 
2265*00b67f09SDavid van Moolenbroek 	return (fillcount);
2266*00b67f09SDavid van Moolenbroek }
2267*00b67f09SDavid van Moolenbroek 
2268*00b67f09SDavid van Moolenbroek isc_result_t
isc__mem_register(void)2269*00b67f09SDavid van Moolenbroek isc__mem_register(void) {
2270*00b67f09SDavid van Moolenbroek 	return (isc_mem_register(isc_mem_create2));
2271*00b67f09SDavid van Moolenbroek }
2272*00b67f09SDavid van Moolenbroek 
2273*00b67f09SDavid van Moolenbroek void
isc__mem_printactive(isc_mem_t * ctx0,FILE * file)2274*00b67f09SDavid van Moolenbroek isc__mem_printactive(isc_mem_t *ctx0, FILE *file) {
2275*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2276*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
2277*00b67f09SDavid van Moolenbroek 
2278*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
2279*00b67f09SDavid van Moolenbroek 	REQUIRE(file != NULL);
2280*00b67f09SDavid van Moolenbroek 
2281*00b67f09SDavid van Moolenbroek 	print_active(ctx, file);
2282*00b67f09SDavid van Moolenbroek #else
2283*00b67f09SDavid van Moolenbroek 	UNUSED(ctx0);
2284*00b67f09SDavid van Moolenbroek 	UNUSED(file);
2285*00b67f09SDavid van Moolenbroek #endif
2286*00b67f09SDavid van Moolenbroek }
2287*00b67f09SDavid van Moolenbroek 
2288*00b67f09SDavid van Moolenbroek void
isc_mem_printallactive(FILE * file)2289*00b67f09SDavid van Moolenbroek isc_mem_printallactive(FILE *file) {
2290*00b67f09SDavid van Moolenbroek #if !ISC_MEM_TRACKLINES
2291*00b67f09SDavid van Moolenbroek 	UNUSED(file);
2292*00b67f09SDavid van Moolenbroek #else
2293*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
2294*00b67f09SDavid van Moolenbroek 
2295*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
2296*00b67f09SDavid van Moolenbroek 
2297*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
2298*00b67f09SDavid van Moolenbroek 	for (ctx = ISC_LIST_HEAD(contexts);
2299*00b67f09SDavid van Moolenbroek 	     ctx != NULL;
2300*00b67f09SDavid van Moolenbroek 	     ctx = ISC_LIST_NEXT(ctx, link)) {
2301*00b67f09SDavid van Moolenbroek 		fprintf(file, "context: %p\n", ctx);
2302*00b67f09SDavid van Moolenbroek 		print_active(ctx, file);
2303*00b67f09SDavid van Moolenbroek 	}
2304*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
2305*00b67f09SDavid van Moolenbroek #endif
2306*00b67f09SDavid van Moolenbroek }
2307*00b67f09SDavid van Moolenbroek 
2308*00b67f09SDavid van Moolenbroek void
isc_mem_checkdestroyed(FILE * file)2309*00b67f09SDavid van Moolenbroek isc_mem_checkdestroyed(FILE *file) {
2310*00b67f09SDavid van Moolenbroek #if !ISC_MEM_TRACKLINES
2311*00b67f09SDavid van Moolenbroek 	UNUSED(file);
2312*00b67f09SDavid van Moolenbroek #endif
2313*00b67f09SDavid van Moolenbroek 
2314*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
2315*00b67f09SDavid van Moolenbroek 
2316*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
2317*00b67f09SDavid van Moolenbroek 	if (!ISC_LIST_EMPTY(contexts))  {
2318*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2319*00b67f09SDavid van Moolenbroek 		isc__mem_t *ctx;
2320*00b67f09SDavid van Moolenbroek 
2321*00b67f09SDavid van Moolenbroek 		for (ctx = ISC_LIST_HEAD(contexts);
2322*00b67f09SDavid van Moolenbroek 		     ctx != NULL;
2323*00b67f09SDavid van Moolenbroek 		     ctx = ISC_LIST_NEXT(ctx, link)) {
2324*00b67f09SDavid van Moolenbroek 			fprintf(file, "context: %p\n", ctx);
2325*00b67f09SDavid van Moolenbroek 			print_active(ctx, file);
2326*00b67f09SDavid van Moolenbroek 		}
2327*00b67f09SDavid van Moolenbroek 		fflush(file);
2328*00b67f09SDavid van Moolenbroek #endif
2329*00b67f09SDavid van Moolenbroek 		INSIST(0);
2330*00b67f09SDavid van Moolenbroek 	}
2331*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
2332*00b67f09SDavid van Moolenbroek }
2333*00b67f09SDavid van Moolenbroek 
2334*00b67f09SDavid van Moolenbroek unsigned int
isc_mem_references(isc_mem_t * ctx0)2335*00b67f09SDavid van Moolenbroek isc_mem_references(isc_mem_t *ctx0) {
2336*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx = (isc__mem_t *)ctx0;
2337*00b67f09SDavid van Moolenbroek 	unsigned int references;
2338*00b67f09SDavid van Moolenbroek 
2339*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
2340*00b67f09SDavid van Moolenbroek 
2341*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
2342*00b67f09SDavid van Moolenbroek 	references = ctx->references;
2343*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
2344*00b67f09SDavid van Moolenbroek 
2345*00b67f09SDavid van Moolenbroek 	return (references);
2346*00b67f09SDavid van Moolenbroek }
2347*00b67f09SDavid van Moolenbroek 
2348*00b67f09SDavid van Moolenbroek #if defined(HAVE_LIBXML2) || defined(HAVE_JSON)
2349*00b67f09SDavid van Moolenbroek typedef struct summarystat {
2350*00b67f09SDavid van Moolenbroek 	isc_uint64_t	total;
2351*00b67f09SDavid van Moolenbroek 	isc_uint64_t	inuse;
2352*00b67f09SDavid van Moolenbroek 	isc_uint64_t	blocksize;
2353*00b67f09SDavid van Moolenbroek 	isc_uint64_t	contextsize;
2354*00b67f09SDavid van Moolenbroek } summarystat_t;
2355*00b67f09SDavid van Moolenbroek #endif
2356*00b67f09SDavid van Moolenbroek 
2357*00b67f09SDavid van Moolenbroek #ifdef HAVE_LIBXML2
2358*00b67f09SDavid van Moolenbroek #define TRY0(a) do { xmlrc = (a); if (xmlrc < 0) goto error; } while(/*CONSTCOND*/0)
2359*00b67f09SDavid van Moolenbroek static int
xml_renderctx(isc__mem_t * ctx,summarystat_t * summary,xmlTextWriterPtr writer)2360*00b67f09SDavid van Moolenbroek xml_renderctx(isc__mem_t *ctx, summarystat_t *summary,
2361*00b67f09SDavid van Moolenbroek 	      xmlTextWriterPtr writer)
2362*00b67f09SDavid van Moolenbroek {
2363*00b67f09SDavid van Moolenbroek 	int xmlrc;
2364*00b67f09SDavid van Moolenbroek 
2365*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
2366*00b67f09SDavid van Moolenbroek 
2367*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
2368*00b67f09SDavid van Moolenbroek 
2369*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "context"));
2370*00b67f09SDavid van Moolenbroek 
2371*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "id"));
2372*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer, "%p", ctx));
2373*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* id */
2374*00b67f09SDavid van Moolenbroek 
2375*00b67f09SDavid van Moolenbroek 	if (ctx->name[0] != 0) {
2376*00b67f09SDavid van Moolenbroek 		TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
2377*00b67f09SDavid van Moolenbroek 		TRY0(xmlTextWriterWriteFormatString(writer, "%s", ctx->name));
2378*00b67f09SDavid van Moolenbroek 		TRY0(xmlTextWriterEndElement(writer)); /* name */
2379*00b67f09SDavid van Moolenbroek 	}
2380*00b67f09SDavid van Moolenbroek 
2381*00b67f09SDavid van Moolenbroek 	summary->contextsize += sizeof(*ctx) +
2382*00b67f09SDavid van Moolenbroek 		(ctx->max_size + 1) * sizeof(struct stats) +
2383*00b67f09SDavid van Moolenbroek 		ctx->max_size * sizeof(element *) +
2384*00b67f09SDavid van Moolenbroek 		ctx->basic_table_count * sizeof(char *);
2385*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2386*00b67f09SDavid van Moolenbroek 	if (ctx->debuglist != NULL) {
2387*00b67f09SDavid van Moolenbroek 		summary->contextsize +=
2388*00b67f09SDavid van Moolenbroek 			(ctx->max_size + 1) * sizeof(debuglist_t) +
2389*00b67f09SDavid van Moolenbroek 			ctx->debuglistcnt * sizeof(debuglink_t);
2390*00b67f09SDavid van Moolenbroek 	}
2391*00b67f09SDavid van Moolenbroek #endif
2392*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "references"));
2393*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer, "%d", ctx->references));
2394*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* references */
2395*00b67f09SDavid van Moolenbroek 
2396*00b67f09SDavid van Moolenbroek 	summary->total += ctx->total;
2397*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "total"));
2398*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2399*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2400*00b67f09SDavid van Moolenbroek 					    (isc_uint64_t)ctx->total));
2401*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* total */
2402*00b67f09SDavid van Moolenbroek 
2403*00b67f09SDavid van Moolenbroek 	summary->inuse += ctx->inuse;
2404*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "inuse"));
2405*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2406*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2407*00b67f09SDavid van Moolenbroek 					    (isc_uint64_t)ctx->inuse));
2408*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* inuse */
2409*00b67f09SDavid van Moolenbroek 
2410*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "maxinuse"));
2411*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2412*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2413*00b67f09SDavid van Moolenbroek 					    (isc_uint64_t)ctx->maxinuse));
2414*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* maxinuse */
2415*00b67f09SDavid van Moolenbroek 
2416*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "blocksize"));
2417*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
2418*00b67f09SDavid van Moolenbroek 		summary->blocksize += ctx->basic_table_count *
2419*00b67f09SDavid van Moolenbroek 			NUM_BASIC_BLOCKS * ctx->mem_target;
2420*00b67f09SDavid van Moolenbroek 		TRY0(xmlTextWriterWriteFormatString(writer,
2421*00b67f09SDavid van Moolenbroek 					       "%" ISC_PRINT_QUADFORMAT "u",
2422*00b67f09SDavid van Moolenbroek 					       (isc_uint64_t)
2423*00b67f09SDavid van Moolenbroek 					       ctx->basic_table_count *
2424*00b67f09SDavid van Moolenbroek 					       NUM_BASIC_BLOCKS *
2425*00b67f09SDavid van Moolenbroek 					       ctx->mem_target));
2426*00b67f09SDavid van Moolenbroek 	} else
2427*00b67f09SDavid van Moolenbroek 		TRY0(xmlTextWriterWriteFormatString(writer, "%s", "-"));
2428*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* blocksize */
2429*00b67f09SDavid van Moolenbroek 
2430*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "pools"));
2431*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer, "%u", ctx->poolcnt));
2432*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* pools */
2433*00b67f09SDavid van Moolenbroek 	summary->contextsize += ctx->poolcnt * sizeof(isc_mempool_t);
2434*00b67f09SDavid van Moolenbroek 
2435*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "hiwater"));
2436*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2437*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2438*00b67f09SDavid van Moolenbroek 					    (isc_uint64_t)ctx->hi_water));
2439*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* hiwater */
2440*00b67f09SDavid van Moolenbroek 
2441*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "lowater"));
2442*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2443*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2444*00b67f09SDavid van Moolenbroek 					    (isc_uint64_t)ctx->lo_water));
2445*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* lowater */
2446*00b67f09SDavid van Moolenbroek 
2447*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* context */
2448*00b67f09SDavid van Moolenbroek 
2449*00b67f09SDavid van Moolenbroek  error:
2450*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
2451*00b67f09SDavid van Moolenbroek 
2452*00b67f09SDavid van Moolenbroek 	return (xmlrc);
2453*00b67f09SDavid van Moolenbroek }
2454*00b67f09SDavid van Moolenbroek 
2455*00b67f09SDavid van Moolenbroek int
isc_mem_renderxml(xmlTextWriterPtr writer)2456*00b67f09SDavid van Moolenbroek isc_mem_renderxml(xmlTextWriterPtr writer) {
2457*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
2458*00b67f09SDavid van Moolenbroek 	summarystat_t summary;
2459*00b67f09SDavid van Moolenbroek 	isc_uint64_t lost;
2460*00b67f09SDavid van Moolenbroek 	int xmlrc;
2461*00b67f09SDavid van Moolenbroek 
2462*00b67f09SDavid van Moolenbroek 	memset(&summary, 0, sizeof(summary));
2463*00b67f09SDavid van Moolenbroek 
2464*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "contexts"));
2465*00b67f09SDavid van Moolenbroek 
2466*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
2467*00b67f09SDavid van Moolenbroek 
2468*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
2469*00b67f09SDavid van Moolenbroek 	lost = totallost;
2470*00b67f09SDavid van Moolenbroek 	for (ctx = ISC_LIST_HEAD(contexts);
2471*00b67f09SDavid van Moolenbroek 	     ctx != NULL;
2472*00b67f09SDavid van Moolenbroek 	     ctx = ISC_LIST_NEXT(ctx, link)) {
2473*00b67f09SDavid van Moolenbroek 		xmlrc = xml_renderctx(ctx, &summary, writer);
2474*00b67f09SDavid van Moolenbroek 		if (xmlrc < 0) {
2475*00b67f09SDavid van Moolenbroek 			UNLOCK(&contextslock);
2476*00b67f09SDavid van Moolenbroek 			goto error;
2477*00b67f09SDavid van Moolenbroek 		}
2478*00b67f09SDavid van Moolenbroek 	}
2479*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
2480*00b67f09SDavid van Moolenbroek 
2481*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* contexts */
2482*00b67f09SDavid van Moolenbroek 
2483*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "summary"));
2484*00b67f09SDavid van Moolenbroek 
2485*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "TotalUse"));
2486*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2487*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2488*00b67f09SDavid van Moolenbroek 					    summary.total));
2489*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* TotalUse */
2490*00b67f09SDavid van Moolenbroek 
2491*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "InUse"));
2492*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2493*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2494*00b67f09SDavid van Moolenbroek 					    summary.inuse));
2495*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* InUse */
2496*00b67f09SDavid van Moolenbroek 
2497*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "BlockSize"));
2498*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2499*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2500*00b67f09SDavid van Moolenbroek 					    summary.blocksize));
2501*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* BlockSize */
2502*00b67f09SDavid van Moolenbroek 
2503*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "ContextSize"));
2504*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2505*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2506*00b67f09SDavid van Moolenbroek 					    summary.contextsize));
2507*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* ContextSize */
2508*00b67f09SDavid van Moolenbroek 
2509*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "Lost"));
2510*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterWriteFormatString(writer,
2511*00b67f09SDavid van Moolenbroek 					    "%" ISC_PRINT_QUADFORMAT "u",
2512*00b67f09SDavid van Moolenbroek 					    lost));
2513*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* Lost */
2514*00b67f09SDavid van Moolenbroek 
2515*00b67f09SDavid van Moolenbroek 	TRY0(xmlTextWriterEndElement(writer)); /* summary */
2516*00b67f09SDavid van Moolenbroek  error:
2517*00b67f09SDavid van Moolenbroek 	return (xmlrc);
2518*00b67f09SDavid van Moolenbroek }
2519*00b67f09SDavid van Moolenbroek 
2520*00b67f09SDavid van Moolenbroek #endif /* HAVE_LIBXML2 */
2521*00b67f09SDavid van Moolenbroek 
2522*00b67f09SDavid van Moolenbroek #ifdef HAVE_JSON
2523*00b67f09SDavid van Moolenbroek #define CHECKMEM(m) do { \
2524*00b67f09SDavid van Moolenbroek 	if (m == NULL) { \
2525*00b67f09SDavid van Moolenbroek 		result = ISC_R_NOMEMORY;\
2526*00b67f09SDavid van Moolenbroek 		goto error;\
2527*00b67f09SDavid van Moolenbroek 	} \
2528*00b67f09SDavid van Moolenbroek } while(/*CONSTCOND*/0)
2529*00b67f09SDavid van Moolenbroek 
2530*00b67f09SDavid van Moolenbroek static isc_result_t
json_renderctx(isc__mem_t * ctx,summarystat_t * summary,json_object * array)2531*00b67f09SDavid van Moolenbroek json_renderctx(isc__mem_t *ctx, summarystat_t *summary, json_object *array) {
2532*00b67f09SDavid van Moolenbroek 	isc_result_t result = ISC_R_FAILURE;
2533*00b67f09SDavid van Moolenbroek 	json_object *ctxobj, *obj;
2534*00b67f09SDavid van Moolenbroek 	char buf[1024];
2535*00b67f09SDavid van Moolenbroek 
2536*00b67f09SDavid van Moolenbroek 	REQUIRE(VALID_CONTEXT(ctx));
2537*00b67f09SDavid van Moolenbroek 	REQUIRE(summary != NULL);
2538*00b67f09SDavid van Moolenbroek 	REQUIRE(array != NULL);
2539*00b67f09SDavid van Moolenbroek 
2540*00b67f09SDavid van Moolenbroek 	MCTXLOCK(ctx, &ctx->lock);
2541*00b67f09SDavid van Moolenbroek 
2542*00b67f09SDavid van Moolenbroek 	summary->contextsize += sizeof(*ctx) +
2543*00b67f09SDavid van Moolenbroek 		(ctx->max_size + 1) * sizeof(struct stats) +
2544*00b67f09SDavid van Moolenbroek 		ctx->max_size * sizeof(element *) +
2545*00b67f09SDavid van Moolenbroek 		ctx->basic_table_count * sizeof(char *);
2546*00b67f09SDavid van Moolenbroek 	summary->total += ctx->total;
2547*00b67f09SDavid van Moolenbroek 	summary->inuse += ctx->inuse;
2548*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0)
2549*00b67f09SDavid van Moolenbroek 		summary->blocksize += ctx->basic_table_count *
2550*00b67f09SDavid van Moolenbroek 			NUM_BASIC_BLOCKS * ctx->mem_target;
2551*00b67f09SDavid van Moolenbroek #if ISC_MEM_TRACKLINES
2552*00b67f09SDavid van Moolenbroek 	if (ctx->debuglist != NULL) {
2553*00b67f09SDavid van Moolenbroek 		summary->contextsize +=
2554*00b67f09SDavid van Moolenbroek 			(ctx->max_size + 1) * sizeof(debuglist_t) +
2555*00b67f09SDavid van Moolenbroek 			ctx->debuglistcnt * sizeof(debuglink_t);
2556*00b67f09SDavid van Moolenbroek 	}
2557*00b67f09SDavid van Moolenbroek #endif
2558*00b67f09SDavid van Moolenbroek 
2559*00b67f09SDavid van Moolenbroek 	ctxobj = json_object_new_object();
2560*00b67f09SDavid van Moolenbroek 	CHECKMEM(ctxobj);
2561*00b67f09SDavid van Moolenbroek 
2562*00b67f09SDavid van Moolenbroek 	sprintf(buf, "%p", ctx);
2563*00b67f09SDavid van Moolenbroek 	obj = json_object_new_string(buf);
2564*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2565*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "id", obj);
2566*00b67f09SDavid van Moolenbroek 
2567*00b67f09SDavid van Moolenbroek 	if (ctx->name[0] != 0) {
2568*00b67f09SDavid van Moolenbroek 		obj = json_object_new_string(ctx->name);
2569*00b67f09SDavid van Moolenbroek 		CHECKMEM(obj);
2570*00b67f09SDavid van Moolenbroek 		json_object_object_add(ctxobj, "name", obj);
2571*00b67f09SDavid van Moolenbroek 	}
2572*00b67f09SDavid van Moolenbroek 
2573*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->references);
2574*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2575*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "references", obj);
2576*00b67f09SDavid van Moolenbroek 
2577*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->total);
2578*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2579*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "total", obj);
2580*00b67f09SDavid van Moolenbroek 
2581*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->inuse);
2582*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2583*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "inuse", obj);
2584*00b67f09SDavid van Moolenbroek 
2585*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->maxinuse);
2586*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2587*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "maxinuse", obj);
2588*00b67f09SDavid van Moolenbroek 
2589*00b67f09SDavid van Moolenbroek 	if ((ctx->flags & ISC_MEMFLAG_INTERNAL) != 0) {
2590*00b67f09SDavid van Moolenbroek 		isc_uint64_t blocksize;
2591*00b67f09SDavid van Moolenbroek 		blocksize = ctx->basic_table_count * NUM_BASIC_BLOCKS *
2592*00b67f09SDavid van Moolenbroek 			ctx->mem_target;
2593*00b67f09SDavid van Moolenbroek 		obj = json_object_new_int64(blocksize);
2594*00b67f09SDavid van Moolenbroek 		CHECKMEM(obj);
2595*00b67f09SDavid van Moolenbroek 		json_object_object_add(ctxobj, "blocksize", obj);
2596*00b67f09SDavid van Moolenbroek 	}
2597*00b67f09SDavid van Moolenbroek 
2598*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->poolcnt);
2599*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2600*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "pools", obj);
2601*00b67f09SDavid van Moolenbroek 
2602*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->hi_water);
2603*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2604*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "hiwater", obj);
2605*00b67f09SDavid van Moolenbroek 
2606*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(ctx->lo_water);
2607*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2608*00b67f09SDavid van Moolenbroek 	json_object_object_add(ctxobj, "lowater", obj);
2609*00b67f09SDavid van Moolenbroek 
2610*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
2611*00b67f09SDavid van Moolenbroek 	json_object_array_add(array, ctxobj);
2612*00b67f09SDavid van Moolenbroek 	return (ISC_R_SUCCESS);
2613*00b67f09SDavid van Moolenbroek 
2614*00b67f09SDavid van Moolenbroek  error:
2615*00b67f09SDavid van Moolenbroek 	MCTXUNLOCK(ctx, &ctx->lock);
2616*00b67f09SDavid van Moolenbroek 	if (ctxobj != NULL)
2617*00b67f09SDavid van Moolenbroek 		json_object_put(ctxobj);
2618*00b67f09SDavid van Moolenbroek 	return (result);
2619*00b67f09SDavid van Moolenbroek }
2620*00b67f09SDavid van Moolenbroek 
2621*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_renderjson(json_object * memobj)2622*00b67f09SDavid van Moolenbroek isc_mem_renderjson(json_object *memobj) {
2623*00b67f09SDavid van Moolenbroek 	isc_result_t result = ISC_R_SUCCESS;
2624*00b67f09SDavid van Moolenbroek 	isc__mem_t *ctx;
2625*00b67f09SDavid van Moolenbroek 	summarystat_t summary;
2626*00b67f09SDavid van Moolenbroek 	isc_uint64_t lost;
2627*00b67f09SDavid van Moolenbroek 	json_object *ctxarray, *obj;
2628*00b67f09SDavid van Moolenbroek 
2629*00b67f09SDavid van Moolenbroek 	memset(&summary, 0, sizeof(summary));
2630*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
2631*00b67f09SDavid van Moolenbroek 
2632*00b67f09SDavid van Moolenbroek 	ctxarray = json_object_new_array();
2633*00b67f09SDavid van Moolenbroek 	CHECKMEM(ctxarray);
2634*00b67f09SDavid van Moolenbroek 
2635*00b67f09SDavid van Moolenbroek 	LOCK(&contextslock);
2636*00b67f09SDavid van Moolenbroek 	lost = totallost;
2637*00b67f09SDavid van Moolenbroek 	for (ctx = ISC_LIST_HEAD(contexts);
2638*00b67f09SDavid van Moolenbroek 	     ctx != NULL;
2639*00b67f09SDavid van Moolenbroek 	     ctx = ISC_LIST_NEXT(ctx, link)) {
2640*00b67f09SDavid van Moolenbroek 		result = json_renderctx(ctx, &summary, ctxarray);
2641*00b67f09SDavid van Moolenbroek 		if (result != ISC_R_SUCCESS) {
2642*00b67f09SDavid van Moolenbroek 			UNLOCK(&contextslock);
2643*00b67f09SDavid van Moolenbroek 			goto error;
2644*00b67f09SDavid van Moolenbroek 		}
2645*00b67f09SDavid van Moolenbroek 	}
2646*00b67f09SDavid van Moolenbroek 	UNLOCK(&contextslock);
2647*00b67f09SDavid van Moolenbroek 
2648*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(summary.total);
2649*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2650*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "TotalUse", obj);
2651*00b67f09SDavid van Moolenbroek 
2652*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(summary.inuse);
2653*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2654*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "InUse", obj);
2655*00b67f09SDavid van Moolenbroek 
2656*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(summary.blocksize);
2657*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2658*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "BlockSize", obj);
2659*00b67f09SDavid van Moolenbroek 
2660*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(summary.contextsize);
2661*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2662*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "ContextSize", obj);
2663*00b67f09SDavid van Moolenbroek 
2664*00b67f09SDavid van Moolenbroek 	obj = json_object_new_int64(lost);
2665*00b67f09SDavid van Moolenbroek 	CHECKMEM(obj);
2666*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "Lost", obj);
2667*00b67f09SDavid van Moolenbroek 
2668*00b67f09SDavid van Moolenbroek 	json_object_object_add(memobj, "contexts", ctxarray);
2669*00b67f09SDavid van Moolenbroek 	return (ISC_R_SUCCESS);
2670*00b67f09SDavid van Moolenbroek 
2671*00b67f09SDavid van Moolenbroek  error:
2672*00b67f09SDavid van Moolenbroek 	if (ctxarray != NULL)
2673*00b67f09SDavid van Moolenbroek 		json_object_put(ctxarray);
2674*00b67f09SDavid van Moolenbroek 	return (result);
2675*00b67f09SDavid van Moolenbroek }
2676*00b67f09SDavid van Moolenbroek #endif /* HAVE_JSON */
2677*00b67f09SDavid van Moolenbroek 
2678*00b67f09SDavid van Moolenbroek static isc_memcreatefunc_t mem_createfunc = NULL;
2679*00b67f09SDavid van Moolenbroek 
2680*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_register(isc_memcreatefunc_t createfunc)2681*00b67f09SDavid van Moolenbroek isc_mem_register(isc_memcreatefunc_t createfunc) {
2682*00b67f09SDavid van Moolenbroek 	isc_result_t result = ISC_R_SUCCESS;
2683*00b67f09SDavid van Moolenbroek 
2684*00b67f09SDavid van Moolenbroek 	RUNTIME_CHECK(isc_once_do(&once, initialize_action) == ISC_R_SUCCESS);
2685*00b67f09SDavid van Moolenbroek 
2686*00b67f09SDavid van Moolenbroek 	LOCK(&createlock);
2687*00b67f09SDavid van Moolenbroek 	if (mem_createfunc == NULL)
2688*00b67f09SDavid van Moolenbroek 		mem_createfunc = createfunc;
2689*00b67f09SDavid van Moolenbroek 	else
2690*00b67f09SDavid van Moolenbroek 		result = ISC_R_EXISTS;
2691*00b67f09SDavid van Moolenbroek 	UNLOCK(&createlock);
2692*00b67f09SDavid van Moolenbroek 
2693*00b67f09SDavid van Moolenbroek 	return (result);
2694*00b67f09SDavid van Moolenbroek }
2695*00b67f09SDavid van Moolenbroek 
2696*00b67f09SDavid van Moolenbroek 
2697*00b67f09SDavid van Moolenbroek isc_result_t
isc__mem_create2(size_t init_max_size,size_t target_size,isc_mem_t ** mctxp,unsigned int flags)2698*00b67f09SDavid van Moolenbroek isc__mem_create2(size_t init_max_size, size_t target_size, isc_mem_t **mctxp,
2699*00b67f09SDavid van Moolenbroek 		 unsigned int flags)
2700*00b67f09SDavid van Moolenbroek {
2701*00b67f09SDavid van Moolenbroek 	isc_result_t result;
2702*00b67f09SDavid van Moolenbroek 
2703*00b67f09SDavid van Moolenbroek 	LOCK(&createlock);
2704*00b67f09SDavid van Moolenbroek 
2705*00b67f09SDavid van Moolenbroek 	REQUIRE(mem_createfunc != NULL);
2706*00b67f09SDavid van Moolenbroek 	result = (*mem_createfunc)(init_max_size, target_size, mctxp, flags);
2707*00b67f09SDavid van Moolenbroek 
2708*00b67f09SDavid van Moolenbroek 	UNLOCK(&createlock);
2709*00b67f09SDavid van Moolenbroek 
2710*00b67f09SDavid van Moolenbroek 	return (result);
2711*00b67f09SDavid van Moolenbroek }
2712*00b67f09SDavid van Moolenbroek 
2713*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_create(size_t init_max_size,size_t target_size,isc_mem_t ** mctxp)2714*00b67f09SDavid van Moolenbroek isc_mem_create(size_t init_max_size, size_t target_size, isc_mem_t **mctxp) {
2715*00b67f09SDavid van Moolenbroek 	isc_result_t result;
2716*00b67f09SDavid van Moolenbroek 
2717*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2718*00b67f09SDavid van Moolenbroek 		return (isc_mem_createx2(init_max_size, target_size,
2719*00b67f09SDavid van Moolenbroek 					 default_memalloc, default_memfree,
2720*00b67f09SDavid van Moolenbroek 					 NULL, mctxp, ISC_MEMFLAG_DEFAULT));
2721*00b67f09SDavid van Moolenbroek 	LOCK(&createlock);
2722*00b67f09SDavid van Moolenbroek 
2723*00b67f09SDavid van Moolenbroek 	REQUIRE(mem_createfunc != NULL);
2724*00b67f09SDavid van Moolenbroek 	result = (*mem_createfunc)(init_max_size, target_size, mctxp,
2725*00b67f09SDavid van Moolenbroek 				   ISC_MEMFLAG_DEFAULT);
2726*00b67f09SDavid van Moolenbroek 
2727*00b67f09SDavid van Moolenbroek 	UNLOCK(&createlock);
2728*00b67f09SDavid van Moolenbroek 
2729*00b67f09SDavid van Moolenbroek 	return (result);
2730*00b67f09SDavid van Moolenbroek }
2731*00b67f09SDavid van Moolenbroek 
2732*00b67f09SDavid van Moolenbroek isc_result_t
isc_mem_create2(size_t init_max_size,size_t target_size,isc_mem_t ** mctxp,unsigned int flags)2733*00b67f09SDavid van Moolenbroek isc_mem_create2(size_t init_max_size, size_t target_size, isc_mem_t **mctxp,
2734*00b67f09SDavid van Moolenbroek 		 unsigned int flags)
2735*00b67f09SDavid van Moolenbroek {
2736*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2737*00b67f09SDavid van Moolenbroek 		return (isc_mem_createx2(init_max_size, target_size,
2738*00b67f09SDavid van Moolenbroek 					 default_memalloc, default_memfree,
2739*00b67f09SDavid van Moolenbroek 					 NULL, mctxp, flags));
2740*00b67f09SDavid van Moolenbroek 
2741*00b67f09SDavid van Moolenbroek 	return (isc_mem_createx2(init_max_size, target_size,
2742*00b67f09SDavid van Moolenbroek 				 default_memalloc, default_memfree,
2743*00b67f09SDavid van Moolenbroek 				 NULL, mctxp, flags));
2744*00b67f09SDavid van Moolenbroek }
2745*00b67f09SDavid van Moolenbroek 
2746*00b67f09SDavid van Moolenbroek void
isc_mem_attach(isc_mem_t * source,isc_mem_t ** targetp)2747*00b67f09SDavid van Moolenbroek isc_mem_attach(isc_mem_t *source, isc_mem_t **targetp) {
2748*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(source));
2749*00b67f09SDavid van Moolenbroek 	REQUIRE(targetp != NULL && *targetp == NULL);
2750*00b67f09SDavid van Moolenbroek 
2751*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2752*00b67f09SDavid van Moolenbroek 		isc__mem_attach(source, targetp);
2753*00b67f09SDavid van Moolenbroek 	else
2754*00b67f09SDavid van Moolenbroek 		source->methods->attach(source, targetp);
2755*00b67f09SDavid van Moolenbroek 
2756*00b67f09SDavid van Moolenbroek 	ENSURE(*targetp == source);
2757*00b67f09SDavid van Moolenbroek }
2758*00b67f09SDavid van Moolenbroek 
2759*00b67f09SDavid van Moolenbroek void
isc_mem_detach(isc_mem_t ** mctxp)2760*00b67f09SDavid van Moolenbroek isc_mem_detach(isc_mem_t **mctxp) {
2761*00b67f09SDavid van Moolenbroek 	REQUIRE(mctxp != NULL && ISCAPI_MCTX_VALID(*mctxp));
2762*00b67f09SDavid van Moolenbroek 
2763*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2764*00b67f09SDavid van Moolenbroek 		isc__mem_detach(mctxp);
2765*00b67f09SDavid van Moolenbroek 	else
2766*00b67f09SDavid van Moolenbroek 		(*mctxp)->methods->detach(mctxp);
2767*00b67f09SDavid van Moolenbroek 
2768*00b67f09SDavid van Moolenbroek 	ENSURE(*mctxp == NULL);
2769*00b67f09SDavid van Moolenbroek }
2770*00b67f09SDavid van Moolenbroek 
2771*00b67f09SDavid van Moolenbroek void
isc_mem_destroy(isc_mem_t ** mctxp)2772*00b67f09SDavid van Moolenbroek isc_mem_destroy(isc_mem_t **mctxp) {
2773*00b67f09SDavid van Moolenbroek 	REQUIRE(mctxp != NULL && ISCAPI_MCTX_VALID(*mctxp));
2774*00b67f09SDavid van Moolenbroek 
2775*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2776*00b67f09SDavid van Moolenbroek 		isc__mem_destroy(mctxp);
2777*00b67f09SDavid van Moolenbroek 	else
2778*00b67f09SDavid van Moolenbroek 		(*mctxp)->methods->destroy(mctxp);
2779*00b67f09SDavid van Moolenbroek 
2780*00b67f09SDavid van Moolenbroek 	ENSURE(*mctxp == NULL);
2781*00b67f09SDavid van Moolenbroek }
2782*00b67f09SDavid van Moolenbroek 
2783*00b67f09SDavid van Moolenbroek void
isc_mem_setdestroycheck(isc_mem_t * mctx,isc_boolean_t flag)2784*00b67f09SDavid van Moolenbroek isc_mem_setdestroycheck(isc_mem_t *mctx, isc_boolean_t flag) {
2785*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2786*00b67f09SDavid van Moolenbroek 
2787*00b67f09SDavid van Moolenbroek 	mctx->methods->setdestroycheck(mctx, flag);
2788*00b67f09SDavid van Moolenbroek }
2789*00b67f09SDavid van Moolenbroek 
2790*00b67f09SDavid van Moolenbroek void
isc_mem_setwater(isc_mem_t * ctx,isc_mem_water_t water,void * water_arg,size_t hiwater,size_t lowater)2791*00b67f09SDavid van Moolenbroek isc_mem_setwater(isc_mem_t *ctx, isc_mem_water_t water, void *water_arg,
2792*00b67f09SDavid van Moolenbroek 		 size_t hiwater, size_t lowater)
2793*00b67f09SDavid van Moolenbroek {
2794*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(ctx));
2795*00b67f09SDavid van Moolenbroek 
2796*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2797*00b67f09SDavid van Moolenbroek 		isc__mem_setwater(ctx, water, water_arg, hiwater, lowater);
2798*00b67f09SDavid van Moolenbroek 	else
2799*00b67f09SDavid van Moolenbroek 		ctx->methods->setwater(ctx, water, water_arg, hiwater, lowater);
2800*00b67f09SDavid van Moolenbroek }
2801*00b67f09SDavid van Moolenbroek 
2802*00b67f09SDavid van Moolenbroek void
isc_mem_waterack(isc_mem_t * ctx,int flag)2803*00b67f09SDavid van Moolenbroek isc_mem_waterack(isc_mem_t *ctx, int flag) {
2804*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(ctx));
2805*00b67f09SDavid van Moolenbroek 
2806*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2807*00b67f09SDavid van Moolenbroek 		isc__mem_waterack(ctx, flag);
2808*00b67f09SDavid van Moolenbroek 	else
2809*00b67f09SDavid van Moolenbroek 		ctx->methods->waterack(ctx, flag);
2810*00b67f09SDavid van Moolenbroek }
2811*00b67f09SDavid van Moolenbroek 
2812*00b67f09SDavid van Moolenbroek size_t
isc_mem_inuse(isc_mem_t * mctx)2813*00b67f09SDavid van Moolenbroek isc_mem_inuse(isc_mem_t *mctx) {
2814*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2815*00b67f09SDavid van Moolenbroek 
2816*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2817*00b67f09SDavid van Moolenbroek 		return (isc__mem_inuse(mctx));
2818*00b67f09SDavid van Moolenbroek 
2819*00b67f09SDavid van Moolenbroek 	return (mctx->methods->inuse(mctx));
2820*00b67f09SDavid van Moolenbroek }
2821*00b67f09SDavid van Moolenbroek 
2822*00b67f09SDavid van Moolenbroek size_t
isc_mem_maxinuse(isc_mem_t * mctx)2823*00b67f09SDavid van Moolenbroek isc_mem_maxinuse(isc_mem_t *mctx) {
2824*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2825*00b67f09SDavid van Moolenbroek 
2826*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2827*00b67f09SDavid van Moolenbroek 		return (isc__mem_maxinuse(mctx));
2828*00b67f09SDavid van Moolenbroek 
2829*00b67f09SDavid van Moolenbroek 	return (mctx->methods->maxinuse(mctx));
2830*00b67f09SDavid van Moolenbroek }
2831*00b67f09SDavid van Moolenbroek 
2832*00b67f09SDavid van Moolenbroek size_t
isc_mem_total(isc_mem_t * mctx)2833*00b67f09SDavid van Moolenbroek isc_mem_total(isc_mem_t *mctx) {
2834*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2835*00b67f09SDavid van Moolenbroek 
2836*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2837*00b67f09SDavid van Moolenbroek 		return (isc__mem_total(mctx));
2838*00b67f09SDavid van Moolenbroek 
2839*00b67f09SDavid van Moolenbroek 	return (mctx->methods->total(mctx));
2840*00b67f09SDavid van Moolenbroek }
2841*00b67f09SDavid van Moolenbroek 
2842*00b67f09SDavid van Moolenbroek isc_boolean_t
isc_mem_isovermem(isc_mem_t * mctx)2843*00b67f09SDavid van Moolenbroek isc_mem_isovermem(isc_mem_t *mctx) {
2844*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2845*00b67f09SDavid van Moolenbroek 
2846*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2847*00b67f09SDavid van Moolenbroek 		return (isc__mem_isovermem(mctx));
2848*00b67f09SDavid van Moolenbroek 
2849*00b67f09SDavid van Moolenbroek 	return (mctx->methods->isovermem(mctx));
2850*00b67f09SDavid van Moolenbroek }
2851*00b67f09SDavid van Moolenbroek 
2852*00b67f09SDavid van Moolenbroek 
2853*00b67f09SDavid van Moolenbroek isc_result_t
isc_mempool_create(isc_mem_t * mctx,size_t size,isc_mempool_t ** mpctxp)2854*00b67f09SDavid van Moolenbroek isc_mempool_create(isc_mem_t *mctx, size_t size, isc_mempool_t **mpctxp) {
2855*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2856*00b67f09SDavid van Moolenbroek 
2857*00b67f09SDavid van Moolenbroek 	return (mctx->methods->mpcreate(mctx, size, mpctxp));
2858*00b67f09SDavid van Moolenbroek }
2859*00b67f09SDavid van Moolenbroek 
2860*00b67f09SDavid van Moolenbroek void
isc_mempool_destroy(isc_mempool_t ** mpctxp)2861*00b67f09SDavid van Moolenbroek isc_mempool_destroy(isc_mempool_t **mpctxp) {
2862*00b67f09SDavid van Moolenbroek 	REQUIRE(mpctxp != NULL && ISCAPI_MPOOL_VALID(*mpctxp));
2863*00b67f09SDavid van Moolenbroek 
2864*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2865*00b67f09SDavid van Moolenbroek 		isc__mempool_destroy(mpctxp);
2866*00b67f09SDavid van Moolenbroek 	else
2867*00b67f09SDavid van Moolenbroek 		(*mpctxp)->methods->destroy(mpctxp);
2868*00b67f09SDavid van Moolenbroek 
2869*00b67f09SDavid van Moolenbroek 	ENSURE(*mpctxp == NULL);
2870*00b67f09SDavid van Moolenbroek }
2871*00b67f09SDavid van Moolenbroek 
2872*00b67f09SDavid van Moolenbroek unsigned int
isc_mempool_getallocated(isc_mempool_t * mpctx)2873*00b67f09SDavid van Moolenbroek isc_mempool_getallocated(isc_mempool_t *mpctx) {
2874*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2875*00b67f09SDavid van Moolenbroek 
2876*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2877*00b67f09SDavid van Moolenbroek 		return (isc__mempool_getallocated(mpctx));
2878*00b67f09SDavid van Moolenbroek 
2879*00b67f09SDavid van Moolenbroek 	return (mpctx->methods->getallocated(mpctx));
2880*00b67f09SDavid van Moolenbroek }
2881*00b67f09SDavid van Moolenbroek 
2882*00b67f09SDavid van Moolenbroek void
isc_mempool_setmaxalloc(isc_mempool_t * mpctx,unsigned int limit)2883*00b67f09SDavid van Moolenbroek isc_mempool_setmaxalloc(isc_mempool_t *mpctx, unsigned int limit) {
2884*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2885*00b67f09SDavid van Moolenbroek 
2886*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2887*00b67f09SDavid van Moolenbroek 		isc__mempool_setmaxalloc(mpctx, limit);
2888*00b67f09SDavid van Moolenbroek 	else
2889*00b67f09SDavid van Moolenbroek 		mpctx->methods->setmaxalloc(mpctx, limit);
2890*00b67f09SDavid van Moolenbroek }
2891*00b67f09SDavid van Moolenbroek 
2892*00b67f09SDavid van Moolenbroek void
isc_mempool_setfreemax(isc_mempool_t * mpctx,unsigned int limit)2893*00b67f09SDavid van Moolenbroek isc_mempool_setfreemax(isc_mempool_t *mpctx, unsigned int limit) {
2894*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2895*00b67f09SDavid van Moolenbroek 
2896*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2897*00b67f09SDavid van Moolenbroek 		isc__mempool_setfreemax(mpctx, limit);
2898*00b67f09SDavid van Moolenbroek 	else
2899*00b67f09SDavid van Moolenbroek 		mpctx->methods->setfreemax(mpctx, limit);
2900*00b67f09SDavid van Moolenbroek }
2901*00b67f09SDavid van Moolenbroek 
2902*00b67f09SDavid van Moolenbroek void
isc_mempool_setname(isc_mempool_t * mpctx,const char * name)2903*00b67f09SDavid van Moolenbroek isc_mempool_setname(isc_mempool_t *mpctx, const char *name) {
2904*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2905*00b67f09SDavid van Moolenbroek 
2906*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2907*00b67f09SDavid van Moolenbroek 		isc__mempool_setname(mpctx, name);
2908*00b67f09SDavid van Moolenbroek 	else
2909*00b67f09SDavid van Moolenbroek 		mpctx->methods->setname(mpctx, name);
2910*00b67f09SDavid van Moolenbroek }
2911*00b67f09SDavid van Moolenbroek 
2912*00b67f09SDavid van Moolenbroek void
isc_mempool_associatelock(isc_mempool_t * mpctx,isc_mutex_t * lock)2913*00b67f09SDavid van Moolenbroek isc_mempool_associatelock(isc_mempool_t *mpctx, isc_mutex_t *lock) {
2914*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2915*00b67f09SDavid van Moolenbroek 
2916*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2917*00b67f09SDavid van Moolenbroek 		isc__mempool_associatelock(mpctx, lock);
2918*00b67f09SDavid van Moolenbroek 	else
2919*00b67f09SDavid van Moolenbroek 		mpctx->methods->associatelock(mpctx, lock);
2920*00b67f09SDavid van Moolenbroek }
2921*00b67f09SDavid van Moolenbroek 
2922*00b67f09SDavid van Moolenbroek void
isc_mempool_setfillcount(isc_mempool_t * mpctx,unsigned int limit)2923*00b67f09SDavid van Moolenbroek isc_mempool_setfillcount(isc_mempool_t *mpctx, unsigned int limit) {
2924*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
2925*00b67f09SDavid van Moolenbroek 
2926*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2927*00b67f09SDavid van Moolenbroek 		isc__mempool_setfillcount(mpctx, limit);
2928*00b67f09SDavid van Moolenbroek 	else
2929*00b67f09SDavid van Moolenbroek 		mpctx->methods->setfillcount(mpctx, limit);
2930*00b67f09SDavid van Moolenbroek }
2931*00b67f09SDavid van Moolenbroek 
2932*00b67f09SDavid van Moolenbroek void *
isc__mem_get(isc_mem_t * mctx,size_t size FLARG)2933*00b67f09SDavid van Moolenbroek isc__mem_get(isc_mem_t *mctx, size_t size FLARG) {
2934*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2935*00b67f09SDavid van Moolenbroek 
2936*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2937*00b67f09SDavid van Moolenbroek 		return (isc___mem_get(mctx, size FLARG_PASS));
2938*00b67f09SDavid van Moolenbroek 
2939*00b67f09SDavid van Moolenbroek 	return (mctx->methods->memget(mctx, size FLARG_PASS));
2940*00b67f09SDavid van Moolenbroek 
2941*00b67f09SDavid van Moolenbroek }
2942*00b67f09SDavid van Moolenbroek 
2943*00b67f09SDavid van Moolenbroek void
isc__mem_put(isc_mem_t * mctx,void * ptr,size_t size FLARG)2944*00b67f09SDavid van Moolenbroek isc__mem_put(isc_mem_t *mctx, void *ptr, size_t size FLARG) {
2945*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2946*00b67f09SDavid van Moolenbroek 
2947*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2948*00b67f09SDavid van Moolenbroek 		isc___mem_put(mctx, ptr, size FLARG_PASS);
2949*00b67f09SDavid van Moolenbroek 	else
2950*00b67f09SDavid van Moolenbroek 		mctx->methods->memput(mctx, ptr, size FLARG_PASS);
2951*00b67f09SDavid van Moolenbroek }
2952*00b67f09SDavid van Moolenbroek 
2953*00b67f09SDavid van Moolenbroek void
isc__mem_putanddetach(isc_mem_t ** mctxp,void * ptr,size_t size FLARG)2954*00b67f09SDavid van Moolenbroek isc__mem_putanddetach(isc_mem_t **mctxp, void *ptr, size_t size FLARG) {
2955*00b67f09SDavid van Moolenbroek 	REQUIRE(mctxp != NULL && ISCAPI_MCTX_VALID(*mctxp));
2956*00b67f09SDavid van Moolenbroek 
2957*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2958*00b67f09SDavid van Moolenbroek 		isc___mem_putanddetach(mctxp, ptr, size FLARG_PASS);
2959*00b67f09SDavid van Moolenbroek 	else
2960*00b67f09SDavid van Moolenbroek 		(*mctxp)->methods->memputanddetach(mctxp, ptr, size FLARG_PASS);
2961*00b67f09SDavid van Moolenbroek 
2962*00b67f09SDavid van Moolenbroek 	/*
2963*00b67f09SDavid van Moolenbroek 	 * XXX: We cannot always ensure *mctxp == NULL here
2964*00b67f09SDavid van Moolenbroek 	 * (see lib/isc/mem.c).
2965*00b67f09SDavid van Moolenbroek 	 */
2966*00b67f09SDavid van Moolenbroek }
2967*00b67f09SDavid van Moolenbroek 
2968*00b67f09SDavid van Moolenbroek void *
isc__mem_allocate(isc_mem_t * mctx,size_t size FLARG)2969*00b67f09SDavid van Moolenbroek isc__mem_allocate(isc_mem_t *mctx, size_t size FLARG) {
2970*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2971*00b67f09SDavid van Moolenbroek 
2972*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2973*00b67f09SDavid van Moolenbroek 		return (isc___mem_allocate(mctx, size FLARG_PASS));
2974*00b67f09SDavid van Moolenbroek 
2975*00b67f09SDavid van Moolenbroek 	return (mctx->methods->memallocate(mctx, size FLARG_PASS));
2976*00b67f09SDavid van Moolenbroek }
2977*00b67f09SDavid van Moolenbroek 
2978*00b67f09SDavid van Moolenbroek void *
isc__mem_reallocate(isc_mem_t * mctx,void * ptr,size_t size FLARG)2979*00b67f09SDavid van Moolenbroek isc__mem_reallocate(isc_mem_t *mctx, void *ptr, size_t size FLARG) {
2980*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2981*00b67f09SDavid van Moolenbroek 
2982*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2983*00b67f09SDavid van Moolenbroek 		return (isc___mem_reallocate(mctx, ptr, size FLARG_PASS));
2984*00b67f09SDavid van Moolenbroek 
2985*00b67f09SDavid van Moolenbroek 	return (mctx->methods->memreallocate(mctx, ptr, size FLARG_PASS));
2986*00b67f09SDavid van Moolenbroek }
2987*00b67f09SDavid van Moolenbroek 
2988*00b67f09SDavid van Moolenbroek char *
isc__mem_strdup(isc_mem_t * mctx,const char * s FLARG)2989*00b67f09SDavid van Moolenbroek isc__mem_strdup(isc_mem_t *mctx, const char *s FLARG) {
2990*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
2991*00b67f09SDavid van Moolenbroek 
2992*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
2993*00b67f09SDavid van Moolenbroek 		return (isc___mem_strdup(mctx, s FLARG_PASS));
2994*00b67f09SDavid van Moolenbroek 
2995*00b67f09SDavid van Moolenbroek 	return (mctx->methods->memstrdup(mctx, s FLARG_PASS));
2996*00b67f09SDavid van Moolenbroek }
2997*00b67f09SDavid van Moolenbroek 
2998*00b67f09SDavid van Moolenbroek void
isc__mem_free(isc_mem_t * mctx,void * ptr FLARG)2999*00b67f09SDavid van Moolenbroek isc__mem_free(isc_mem_t *mctx, void *ptr FLARG) {
3000*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MCTX_VALID(mctx));
3001*00b67f09SDavid van Moolenbroek 
3002*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
3003*00b67f09SDavid van Moolenbroek 		isc___mem_free(mctx, ptr FLARG_PASS);
3004*00b67f09SDavid van Moolenbroek 	else
3005*00b67f09SDavid van Moolenbroek 		mctx->methods->memfree(mctx, ptr FLARG_PASS);
3006*00b67f09SDavid van Moolenbroek }
3007*00b67f09SDavid van Moolenbroek 
3008*00b67f09SDavid van Moolenbroek void *
isc__mempool_get(isc_mempool_t * mpctx FLARG)3009*00b67f09SDavid van Moolenbroek isc__mempool_get(isc_mempool_t *mpctx FLARG) {
3010*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
3011*00b67f09SDavid van Moolenbroek 
3012*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
3013*00b67f09SDavid van Moolenbroek 		return (isc___mempool_get(mpctx FLARG_PASS));
3014*00b67f09SDavid van Moolenbroek 
3015*00b67f09SDavid van Moolenbroek 	return (mpctx->methods->get(mpctx FLARG_PASS));
3016*00b67f09SDavid van Moolenbroek }
3017*00b67f09SDavid van Moolenbroek 
3018*00b67f09SDavid van Moolenbroek void
isc__mempool_put(isc_mempool_t * mpctx,void * mem FLARG)3019*00b67f09SDavid van Moolenbroek isc__mempool_put(isc_mempool_t *mpctx, void *mem FLARG) {
3020*00b67f09SDavid van Moolenbroek 	REQUIRE(ISCAPI_MPOOL_VALID(mpctx));
3021*00b67f09SDavid van Moolenbroek 
3022*00b67f09SDavid van Moolenbroek 	if (isc_bind9)
3023*00b67f09SDavid van Moolenbroek 		isc___mempool_put(mpctx, mem FLARG_PASS);
3024*00b67f09SDavid van Moolenbroek 	else
3025*00b67f09SDavid van Moolenbroek 		mpctx->methods->put(mpctx, mem FLARG_PASS);
3026*00b67f09SDavid van Moolenbroek }
3027