1 /* sl_malloc.c - malloc routines using a per-thread slab */
4 * Copyright 2003 The OpenLDAP Foundation, All Rights Reserved.
5 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
11 #include <ac/string.h>
27 struct slab_heap *sh = data;
29 ber_memfree_x(sh->h_base, NULL);
30 ber_memfree_x(sh, NULL);
33 BER_MEMALLOC_FN sl_malloc;
34 BER_MEMCALLOC_FN sl_calloc;
35 BER_MEMREALLOC_FN sl_realloc;
36 BER_MEMFREE_FN sl_free;
39 BerMemoryFunctions sl_mfuncs =
40 { sl_malloc, sl_calloc, sl_realloc, sl_free };
45 ber_set_option( NULL, LBER_OPT_MEMORY_FNS, &sl_mfuncs );
54 struct slab_heap *sh = NULL;
55 int pad = 2*sizeof(int)-1;
57 ldap_pvt_thread_pool_getkey( ctx, sl_mem_init, (void **)&sh, NULL );
59 /* round up to doubleword boundary */
64 sh = ch_malloc( sizeof(struct slab_heap) );
65 sh->h_base = ch_malloc( size );
66 ldap_pvt_thread_pool_setkey( ctx, sl_mem_init, (void *)sh, sl_mem_destroy );
67 } else if ( size > (char *) sh->h_end - (char *) sh->h_base ) {
68 sh->h_base = ch_realloc( sh->h_base, size );
70 sh->h_last = sh->h_base;
71 sh->h_end = (char *) sh->h_base + size;
81 struct slab_heap *sh = memctx;
82 int size = (char *) sh->h_end - (char *) sh->h_base;
84 sh->h_base = ch_realloc( sh->h_base, size );
86 ldap_pvt_thread_pool_setkey( ctx, sl_mem_init, NULL, NULL );
95 struct slab_heap *sh = ctx;
96 int pad = 2*sizeof(int)-1;
99 /* ber_set_option calls us like this */
100 if (!ctx) return ber_memalloc_x( size, NULL );
102 /* round up to doubleword boundary */
103 size += pad + sizeof( ber_len_t );
106 if ((char *) sh->h_last + size >= (char *) sh->h_end ) {
108 LDAP_LOG( OPERATION, INFO,
109 "sl_malloc of %lu bytes failed, using ch_malloc\n", (long)size, 0,0 );
111 Debug( LDAP_DEBUG_TRACE,
112 "sl_malloc of %lu bytes failed, using ch_malloc\n", (long)size, 0,0 );
114 return ch_malloc( size );
117 *new++ = size - sizeof(ber_len_t);
118 sh->h_last = (char *) sh->h_last + size;
120 return( (void *)new );
124 sl_calloc( ber_len_t n, ber_len_t size, void *ctx )
128 new = sl_malloc( n*size, ctx );
130 memset( new, 0, n*size );
136 sl_realloc( void *ptr, ber_len_t size, void *ctx )
138 struct slab_heap *sh = ctx;
139 int pad = 2*sizeof(int)-1;
140 ber_len_t *p = (ber_len_t *)ptr;
143 if ( ptr == NULL ) return sl_malloc( size, ctx );
145 /* Not our memory? */
146 if ( !sh || ptr < sh->h_base || ptr >= sh->h_end ) {
147 /* duplicate of ch_realloc behavior, oh well */
148 new = ber_memrealloc_x( ptr, size, NULL );
153 LDAP_LOG( OPERATION, ERR,
154 "ch_realloc: reallocation of %lu bytes failed\n", (long)size, 0,0 );
156 Debug( LDAP_DEBUG_ANY, "ch_realloc of %lu bytes failed\n",
160 exit( EXIT_FAILURE );
168 /* round up to doubleword boundary */
169 size += pad + sizeof( ber_len_t );
172 /* Never shrink blocks */
176 /* If reallocing the last block, we can grow it */
177 } else if ( (char *)ptr + p[-1] == sh->h_last ) {
179 sh->h_last = (char *) sh->h_last + size - p[-1];
182 /* Nowhere to grow, need to alloc and copy */
184 new = sl_malloc( size, ctx );
185 AC_MEMCPY( new, ptr, p[-1] );
191 sl_free( void *ptr, void *ctx )
193 struct slab_heap *sh = ctx;
194 ber_len_t *p = (ber_len_t *)ptr;
196 if ( !sh || ptr < sh->h_base || ptr >= sh->h_end ) {
197 ber_memfree_x( ptr, NULL );
198 } else if ( (char *)ptr + p[-1] == sh->h_last ) {
205 sl_release( void *ptr, void *ctx )
207 struct slab_heap *sh = ctx;
209 if ( sh && ptr >= sh->h_base && ptr <= sh->h_end ) {
217 struct slab_heap *sh = ctx;
220 if (sh) ret = sh->h_last;
226 sl_context( void *ptr )
228 struct slab_heap *sh = NULL;
231 ctx = ldap_pvt_thread_pool_context();
233 ldap_pvt_thread_pool_getkey( ctx, sl_mem_init, (void **)&sh, NULL );
235 if ( sh && ptr >= sh->h_base && ptr <= sh->h_end ) {