]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/sl_malloc.c
Fix corrupted CSN issue
[openldap] / servers / slapd / sl_malloc.c
index 5b591e9eda2db575301340798765498a9fa2a804..89614e55a8b943f7faa6daea5e6b4a055fc7b840 100644 (file)
@@ -2,7 +2,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2003-2005 The OpenLDAP Foundation.
+ * Copyright 2003-2009 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -22,7 +22,9 @@
 #include "slap.h"
 
 static struct slab_object * slap_replenish_sopool(struct slab_heap* sh);
+#ifdef SLAPD_UNUSED
 static void print_slheap(int level, void *ctx);
+#endif
 
 void
 slap_sl_mem_destroy(
@@ -73,8 +75,6 @@ slap_sl_mem_destroy(
                ber_memfree_x(sh->sh_base, NULL);
                ber_memfree_x(sh, NULL);
        }
-
-       ber_set_option( NULL, LBER_OPT_MEMORY_FNS, NULL );
 }
 
 BerMemoryFunctions slap_sl_mfuncs =
@@ -94,7 +94,8 @@ void *
 slap_sl_mem_create(
        ber_len_t size,
        int stack,
-       void *ctx
+       void *ctx,
+       int new
 )
 {
        struct slab_heap *sh;
@@ -113,6 +114,9 @@ slap_sl_mem_create(
        sh = sh_tmp;
 #endif
 
+       if ( !new )
+               return sh;
+
        /* round up to doubleword boundary */
        size += pad;
        size &= ~pad;
@@ -125,10 +129,14 @@ slap_sl_mem_create(
                        slheap = sh;
 #else
                        ldap_pvt_thread_pool_setkey(ctx, (void *)slap_sl_mem_init,
-                               (void *)sh, slap_sl_mem_destroy);
+                               (void *)sh, slap_sl_mem_destroy, NULL, NULL);
 #endif
                } else if ( size > (char *)sh->sh_end - (char *)sh->sh_base ) {
-                       sh->sh_base = ch_realloc(sh->sh_base, size);
+                       void    *newptr;
+
+                       newptr = ch_realloc( sh->sh_base, size );
+                       if ( newptr == NULL ) return NULL;
+                       sh->sh_base = newptr;
                }
                sh->sh_last = sh->sh_base;
                sh->sh_end = (char *) sh->sh_base + size;
@@ -154,7 +162,7 @@ slap_sl_mem_create(
                        slheap = sh;
 #else
                        ldap_pvt_thread_pool_setkey(ctx, (void *)slap_sl_mem_init,
-                               (void *)sh, slap_sl_mem_destroy);
+                               (void *)sh, slap_sl_mem_destroy, NULL, NULL);
 #endif
                } else {
                        for (i = 0; i <= sh->sh_maxorder - order_start; i++) {
@@ -185,7 +193,11 @@ slap_sl_mem_create(
                        }
 
                        if (size > (char *)sh->sh_end - (char *)sh->sh_base) {
-                               sh->sh_base = realloc(sh->sh_base, size);
+                               void    *newptr;
+
+                               newptr = realloc( sh->sh_base, size );
+                               if ( newptr == NULL ) return NULL;
+                               sh->sh_base = newptr;
                        }
                }
                sh->sh_end = (char *)sh->sh_base + size;
@@ -234,7 +246,8 @@ slap_sl_mem_detach(
        slheap = NULL;
 #else
        /* separate from context */
-       ldap_pvt_thread_pool_setkey( ctx, (void *)slap_sl_mem_init, NULL, NULL );
+       ldap_pvt_thread_pool_setkey( ctx, (void *)slap_sl_mem_init,
+               NULL, 0, NULL, NULL );
 #endif
 }
 
@@ -249,10 +262,14 @@ slap_sl_malloc(
        int pad = 2*sizeof(int)-1, pad_shift;
        int order = -1, order_start = -1;
        struct slab_object *so_new, *so_left, *so_right;
-       ber_len_t *ptr, *new;
+       ber_len_t *ptr, *newptr;
        unsigned long diff;
        int i, j;
 
+#ifdef SLAP_NO_SL_MALLOC
+       return ber_memalloc_x( size, NULL );
+#endif
+
        /* ber_set_option calls us like this */
        if (!ctx) return ber_memalloc_x(size, NULL);
 
@@ -267,10 +284,10 @@ slap_sl_malloc(
                                (long)size, 0, 0);
                        return ch_malloc(size);
                }
-               new = sh->sh_last;
-               *new++ = size - sizeof(ber_len_t);
+               newptr = sh->sh_last;
+               *newptr++ = size - sizeof(ber_len_t);
                sh->sh_last = (char *) sh->sh_last + size;
-               return( (void *)new );
+               return( (void *)newptr );
        } else {
                size_shift = size - 1;
                do {
@@ -338,13 +355,13 @@ slap_sl_malloc(
 void *
 slap_sl_calloc( ber_len_t n, ber_len_t size, void *ctx )
 {
-       void *new;
+       void *newptr;
 
-       new = slap_sl_malloc( n*size, ctx );
-       if ( new ) {
-               memset( new, 0, n*size );
+       newptr = slap_sl_malloc( n*size, ctx );
+       if ( newptr ) {
+               memset( newptr, 0, n*size );
        }
-       return new;
+       return newptr;
 }
 
 void *
@@ -352,17 +369,21 @@ slap_sl_realloc(void *ptr, ber_len_t size, void *ctx)
 {
        struct slab_heap *sh = ctx;
        int pad = 2*sizeof(int) -1;
-       ber_len_t *p = (ber_len_t *)ptr, *new;
+       ber_len_t *p = (ber_len_t *)ptr, *newptr;
 
        if (ptr == NULL)
                return slap_sl_malloc(size, ctx);
 
+#ifdef SLAP_NO_SL_MALLOC
+       return ber_memrealloc_x( ptr, size, NULL );
+#endif
+
        /* Not our memory? */
        if (!sh || ptr < sh->sh_base || ptr >= sh->sh_end) {
                /* duplicate of realloc behavior, oh well */
-               new = ber_memrealloc_x(ptr, size, NULL);
-               if (new) {
-                       return new;
+               newptr = ber_memrealloc_x(ptr, size, NULL);
+               if (newptr) {
+                       return newptr;
                }
                Debug(LDAP_DEBUG_ANY, "ch_realloc of %lu bytes failed\n",
                                (long) size, 0, 0);
@@ -382,31 +403,32 @@ slap_sl_realloc(void *ptr, ber_len_t size, void *ctx)
 
                /* Never shrink blocks */
                if (size <= p[-1]) {
-                       new = p;
+                       newptr = p;
        
                /* If reallocing the last block, we can grow it */
                } else if ((char *)ptr + p[-1] == sh->sh_last &&
                        (char *)ptr + size < (char *)sh->sh_end ) {
-                       new = p;
+                       newptr = p;
                        sh->sh_last = (char *)sh->sh_last + size - p[-1];
                        p[-1] = size;
        
                /* Nowhere to grow, need to alloc and copy */
                } else {
-                       new = slap_sl_malloc(size, ctx);
-                       AC_MEMCPY(new, ptr, p[-1]);
+                       newptr = slap_sl_malloc(size, ctx);
+                       AC_MEMCPY(newptr, ptr, p[-1]);
                }
-               return new;
+               return newptr;
        } else {
-               void *newptr;
-               newptr = slap_sl_malloc(size, ctx);
+               void *newptr2;
+
+               newptr2 = slap_sl_malloc(size, ctx);
                if (size < p[-1]) {
-                       AC_MEMCPY(newptr, ptr, size);
+                       AC_MEMCPY(newptr2, ptr, size);
                } else {
-                       AC_MEMCPY(newptr, ptr, p[-1]);
+                       AC_MEMCPY(newptr2, ptr, p[-1]);
                }
                slap_sl_free(ptr, ctx);
-               return newptr;
+               return newptr2;
        }
 }
 
@@ -422,6 +444,14 @@ slap_sl_free(void *ptr, void *ctx)
        unsigned long diff;
        int i, inserted = 0;
 
+       if (!ptr)
+               return;
+
+#ifdef SLAP_NO_SL_MALLOC
+       ber_memfree_x( ptr, NULL );
+       return;
+#endif
+
        if (!sh || ptr < sh->sh_base || ptr >= sh->sh_end) {
                ber_memfree_x(ptr, NULL);
        } else if (sh->sh_stack && (char *)ptr + p[-1] == sh->sh_last) {
@@ -600,6 +630,7 @@ slap_replenish_sopool(
     return so_block;
 }
 
+#ifdef SLAPD_UNUSED
 static void
 print_slheap(int level, void *ctx)
 {
@@ -640,3 +671,4 @@ print_slheap(int level, void *ctx)
                }
        }
 }
+#endif