]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/memory.c
Merge remote branch 'origin/mdb.master'
[openldap] / libraries / liblber / memory.c
index 5b4089fec2c1cd756cdafb48438e7f3e5f66bff2..c56e0508140b3e9f86db58ebe376a2f95746e397 100644 (file)
@@ -1,7 +1,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 1998-2005 The OpenLDAP Foundation.
+ * Copyright 1998-2013 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
 #include "lber-int.h"
 
 #ifdef LDAP_MEMORY_TRACE
-# ifndef LDAP_MEMORY_DEBUG
-#  define LDAP_MEMORY_DEBUG 1
-# endif
 #include <stdio.h>
 #endif
 
-#if LDAP_MEMORY_DEBUG
+#ifdef LDAP_MEMORY_DEBUG
 /*
  * LDAP_MEMORY_DEBUG should only be enabled for the purposes of
  * debugging memory management within OpenLDAP libraries and slapd.
- * It should only be enabled by an experienced developer as it
- * causes the inclusion of numerous assert()'s, many of which may
- * be triggered by a prefectly valid program.
+ *
+ * It should only be enabled by an experienced developer as it causes
+ * the inclusion of numerous assert()'s, many of which may be triggered
+ * by a prefectly valid program.  If LDAP_MEMORY_DEBUG & 2 is true,
+ * that includes asserts known to break both slapd and current clients.
  *
  * The code behind this macro is subject to change as needed to
  * support this testing.
@@ -63,17 +62,18 @@ struct ber_mem_hdr {
 };
 
 /* Pattern at top of allocated space */
-#define LBER_MEM_JUNK 0xdeaddadaU
+#define LBER_MEM_JUNK ((ber_int_t) 0xdeaddada)
 
-static const struct ber_mem_hdr ber_int_mem_hdr = { LBER_MEM_JUNK, 0, 0 };
+static const struct ber_mem_hdr ber_int_mem_hdr = { LBER_MEM_JUNK };
 
-/* Note sequence and ber_int_options.lbu_meminuse are counters, but are not
+/* Note sequence and ber_int_meminuse are counters, but are not
  * thread safe.  If you want to use these values for multithreaded applications,
  * you must put mutexes around them, otherwise they will have incorrect values.
  * When debugging, if you sort the debug output, the sequence number will 
  * put allocations/frees together.  It is then a simple matter to write a script
  * to find any allocations that don't have a buffer free function.
  */
+long ber_int_meminuse = 0;
 #ifdef LDAP_MEMORY_TRACE
 static ber_int_t sequence = 0;
 #endif
@@ -125,8 +125,6 @@ BerMemoryFunctions *ber_int_memory_fns = NULL;
 void
 ber_memfree_x( void *p, void *ctx )
 {
-    ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( p == NULL ) {
                return;
        }
@@ -140,11 +138,12 @@ ber_memfree_x( void *p, void *ctx )
                assert( mh->bm_top == LBER_MEM_JUNK);
                assert( testdatatop( mh));
                assert( testend( (char *)&mh[1] + mh->bm_length) );
-               ber_int_options.lbo_meminuse -= mh->bm_length;
+               ber_int_meminuse -= mh->bm_length;
 
 #ifdef LDAP_MEMORY_TRACE
-               fprintf(stderr, "0x%08x 0x%08x -f- %d ber_memfree %d\n",
-                       mh->bm_sequence, mh, mh->bm_length, ber_int_options.lbo_meminuse);
+               fprintf(stderr, "0x%08lx 0x%08lx -f- %ld ber_memfree %ld\n",
+                       (long)mh->bm_sequence, (long)mh, (long)mh->bm_length,
+                       ber_int_meminuse);
 #endif
                /* Fill the free space with poison */
                memset( mh, 0xff, mh->bm_length + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t));
@@ -155,7 +154,7 @@ ber_memfree_x( void *p, void *ctx )
                return;
        }
 
-       assert( ber_int_memory_fns->bmf_free );
+       assert( ber_int_memory_fns->bmf_free != 0 );
 
        (*ber_int_memory_fns->bmf_free)( p, ctx );
 }
@@ -171,8 +170,6 @@ ber_memvfree_x( void **vec, void *ctx )
 {
        int     i;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( vec == NULL ) {
                return;
        }
@@ -196,38 +193,37 @@ void *
 ber_memalloc_x( ber_len_t s, void *ctx )
 {
        void *new;
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
-#ifdef LDAP_MEMORY_DEBUG
-       assert( s != 0 );
-#endif
 
        if( s == 0 ) {
+               LDAP_MEMORY_DEBUG_ASSERT( s != 0 );
                return NULL;
        }
 
        if( ber_int_memory_fns == NULL || ctx == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
-               struct ber_mem_hdr *mh = malloc(s + sizeof(struct ber_mem_hdr) + sizeof( ber_int_t));
-               if( mh == NULL ) return NULL;
-
+               new = malloc(s + sizeof(struct ber_mem_hdr) + sizeof( ber_int_t));
+               if( new )
+               {
+               struct ber_mem_hdr *mh = new;
                mh->bm_top = LBER_MEM_JUNK;
                mh->bm_length = s;
                setdatatop( mh);
                setend( (char *)&mh[1] + mh->bm_length );
 
-               ber_int_options.lbo_meminuse += mh->bm_length;  /* Count mem inuse */
+               ber_int_meminuse += mh->bm_length;      /* Count mem inuse */
 
 #ifdef LDAP_MEMORY_TRACE
                mh->bm_sequence = sequence++;
-               fprintf(stderr, "0x%08x 0x%08x -a- %d ber_memalloc %d\n",
-                       mh->bm_sequence, mh, mh->bm_length, ber_int_options.lbo_meminuse);
+               fprintf(stderr, "0x%08lx 0x%08lx -a- %ld ber_memalloc %ld\n",
+                       (long)mh->bm_sequence, (long)mh, (long)mh->bm_length,
+                       ber_int_meminuse);
 #endif
                /* poison new memory */
                memset( (char *)&mh[1], 0xff, s);
 
                BER_MEM_VALID( &mh[1] );
                new = &mh[1];
+               }
 #else
                new = malloc( s );
 #endif
@@ -252,36 +248,37 @@ void *
 ber_memcalloc_x( ber_len_t n, ber_len_t s, void *ctx )
 {
        void *new;
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
-#ifdef LDAP_MEMORY_DEBUG
-       assert( n != 0 && s != 0);
-#endif
 
        if( n == 0 || s == 0 ) {
+               LDAP_MEMORY_DEBUG_ASSERT( n != 0 && s != 0);
                return NULL;
        }
 
        if( ber_int_memory_fns == NULL || ctx == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
-               struct ber_mem_hdr *mh = calloc(1,
-                       (n * s) + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t) );
-               if( mh == NULL ) return NULL;
+               new = n < (-sizeof(struct ber_mem_hdr) - sizeof(ber_int_t)) / s
+                       ? calloc(1, n*s + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t))
+                       : NULL;
+               if( new )
+               {
+               struct ber_mem_hdr *mh = new;
 
                mh->bm_top = LBER_MEM_JUNK;
                mh->bm_length = n*s;
                setdatatop( mh);
                setend( (char *)&mh[1] + mh->bm_length );
 
-               ber_int_options.lbo_meminuse += mh->bm_length;
+               ber_int_meminuse += mh->bm_length;
 
 #ifdef LDAP_MEMORY_TRACE
                mh->bm_sequence = sequence++;
-               fprintf(stderr, "0x%08x 0x%08x -a- %d ber_memcalloc %d\n",
-                       mh->bm_sequence, mh, mh->bm_length, ber_int_options.lbo_meminuse);
+               fprintf(stderr, "0x%08lx 0x%08lx -a- %ld ber_memcalloc %ld\n",
+                       (long)mh->bm_sequence, (long)mh, (long)mh->bm_length,
+                       ber_int_meminuse);
 #endif
                BER_MEM_VALID( &mh[1] );
                new = &mh[1];
+               }
 #else
                new = calloc( n, s );
 #endif
@@ -307,7 +304,6 @@ void *
 ber_memrealloc_x( void* p, ber_len_t s, void *ctx )
 {
        void *new = NULL;
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        /* realloc(NULL,s) -> malloc(s) */
        if( p == NULL ) {
@@ -349,10 +345,11 @@ ber_memrealloc_x( void* p, ber_len_t s, void *ctx )
                assert( mh->bm_top == LBER_MEM_JUNK);
                assert( testdatatop( mh));
 
-               ber_int_options.lbo_meminuse += s - oldlen;
+               ber_int_meminuse += s - oldlen;
 #ifdef LDAP_MEMORY_TRACE
-               fprintf(stderr, "0x%08x 0x%08x -a- %d ber_memrealloc %d\n",
-                       mh->bm_sequence, mh, mh->bm_length, ber_int_options.lbo_meminuse);
+               fprintf(stderr, "0x%08lx 0x%08lx -a- %ld ber_memrealloc %ld\n",
+                       (long)mh->bm_sequence, (long)mh, (long)mh->bm_length,
+                       ber_int_meminuse);
 #endif
                        BER_MEM_VALID( &mh[1] );
                return &mh[1];
@@ -379,8 +376,6 @@ ber_memrealloc( void* p, ber_len_t s )
 void
 ber_bvfree_x( struct berval *bv, void *ctx )
 {
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( bv == NULL ) {
                return;
        }
@@ -405,8 +400,6 @@ ber_bvecfree_x( struct berval **bv, void *ctx )
 {
        int     i;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( bv == NULL ) {
                return;
        }
@@ -436,8 +429,6 @@ ber_bvecadd_x( struct berval ***bvec, struct berval *bv, void *ctx )
        ber_len_t i;
        struct berval **new;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( *bvec == NULL ) {
                if( bv == NULL ) {
                        /* nothing to add */
@@ -493,8 +484,6 @@ ber_dupbv_x(
 {
        struct berval *new;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( src == NULL ) {
                ber_errno = LBER_ERROR_PARAM;
                return NULL;
@@ -504,7 +493,6 @@ ber_dupbv_x(
                new = dst;
        } else {
                if(( new = ber_memalloc_x( sizeof(struct berval), ctx )) == NULL ) {
-                       ber_errno = LBER_ERROR_MEMORY;
                        return NULL;
                }
        }
@@ -516,7 +504,6 @@ ber_dupbv_x(
        }
 
        if(( new->bv_val = ber_memalloc_x( src->bv_len + 1, ctx )) == NULL ) {
-               ber_errno = LBER_ERROR_MEMORY;
                if ( !dst )
                        ber_memfree_x( new, ctx );
                return NULL;
@@ -550,8 +537,6 @@ ber_str2bv_x(
 {
        struct berval *new;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( s == NULL ) {
                ber_errno = LBER_ERROR_PARAM;
                return NULL;
@@ -561,7 +546,6 @@ ber_str2bv_x(
                new = bv;
        } else {
                if(( new = ber_memalloc_x( sizeof(struct berval), ctx )) == NULL ) {
-                       ber_errno = LBER_ERROR_MEMORY;
                        return NULL;
                }
        }
@@ -569,7 +553,6 @@ ber_str2bv_x(
        new->bv_len = len ? len : strlen( s );
        if ( dup ) {
                if ( (new->bv_val = ber_memalloc_x( new->bv_len+1, ctx )) == NULL ) {
-                       ber_errno = LBER_ERROR_MEMORY;
                        if ( !bv )
                                ber_memfree_x( new, ctx );
                        return NULL;
@@ -598,8 +581,6 @@ ber_mem2bv_x(
 {
        struct berval *new;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if( s == NULL ) {
                ber_errno = LBER_ERROR_PARAM;
                return NULL;
@@ -609,7 +590,6 @@ ber_mem2bv_x(
                new = bv;
        } else {
                if(( new = ber_memalloc_x( sizeof(struct berval), ctx )) == NULL ) {
-                       ber_errno = LBER_ERROR_MEMORY;
                        return NULL;
                }
        }
@@ -617,7 +597,6 @@ ber_mem2bv_x(
        new->bv_len = len;
        if ( dup ) {
                if ( (new->bv_val = ber_memalloc_x( new->bv_len+1, ctx )) == NULL ) {
-                       ber_errno = LBER_ERROR_MEMORY;
                        if ( !bv ) {
                                ber_memfree_x( new, ctx );
                        }
@@ -646,8 +625,6 @@ ber_strdup_x( LDAP_CONST char *s, void *ctx )
        char    *p;
        size_t  len;
        
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
 #ifdef LDAP_MEMORY_DEBUG
        assert(s != NULL);                      /* bv damn better point to something */
 #endif
@@ -658,13 +635,10 @@ ber_strdup_x( LDAP_CONST char *s, void *ctx )
        }
 
        len = strlen( s ) + 1;
-
-       if ( (p = ber_memalloc_x( len, ctx )) == NULL ) {
-               ber_errno = LBER_ERROR_MEMORY;
-               return NULL;
+       if ( (p = ber_memalloc_x( len, ctx )) != NULL ) {
+               AC_MEMCPY( p, s, len );
        }
 
-       AC_MEMCPY( p, s, len );
        return p;
 }
 
@@ -674,14 +648,22 @@ ber_strdup( LDAP_CONST char *s )
        return ber_strdup_x( s, NULL );
 }
 
+ber_len_t
+ber_strnlen( LDAP_CONST char *s, ber_len_t len )
+{
+       ber_len_t l;
+
+       for ( l = 0; l < len && s[l] != '\0'; l++ ) ;
+
+       return l;
+}
+
 char *
 ber_strndup_x( LDAP_CONST char *s, ber_len_t l, void *ctx )
 {
        char    *p;
        size_t  len;
        
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
 #ifdef LDAP_MEMORY_DEBUG
        assert(s != NULL);                      /* bv damn better point to something */
 #endif
@@ -691,19 +673,12 @@ ber_strndup_x( LDAP_CONST char *s, ber_len_t l, void *ctx )
                return NULL;
        }
 
-       len = strlen( s );
-
-       if ( len > l ) {
-               len = l;
-       }
-
-       if ( (p = ber_memalloc_x( len + 1, ctx )) == NULL ) {
-               ber_errno = LBER_ERROR_MEMORY;
-               return NULL;
+       len = ber_strnlen( s, l );
+       if ( (p = ber_memalloc_x( len + 1, ctx )) != NULL ) {
+               AC_MEMCPY( p, s, len );
+               p[len] = '\0';
        }
 
-       AC_MEMCPY( p, s, len );
-       p[len] = '\0';
        return p;
 }
 
@@ -722,12 +697,14 @@ struct berval *
 ber_bvreplace_x( struct berval *dst, LDAP_CONST struct berval *src, void *ctx )
 {
        assert( dst != NULL );
+       assert( !BER_BVISNULL( src ) );
 
-       if ( dst->bv_len < src->bv_len ) {
+       if ( BER_BVISNULL( dst ) || dst->bv_len < src->bv_len ) {
                dst->bv_val = ber_memrealloc_x( dst->bv_val, src->bv_len + 1, ctx );
        }
 
        AC_MEMCPY( dst->bv_val, src->bv_val, src->bv_len + 1 );
+       dst->bv_len = src->bv_len;
 
        return dst;
 }
@@ -743,8 +720,6 @@ ber_bvarray_free_x( BerVarray a, void *ctx )
 {
        int i;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if (a) {
                BER_MEM_VALID( a );
 
@@ -766,13 +741,38 @@ ber_bvarray_free( BerVarray a )
        ber_bvarray_free_x(a, NULL);
 }
 
+int
+ber_bvarray_dup_x( BerVarray *dst, BerVarray src, void *ctx )
+{
+       int i, j;
+       BerVarray new;
+
+       if ( !src ) {
+               *dst = NULL;
+               return 0;
+       }
+
+       for (i=0; !BER_BVISNULL( &src[i] ); i++) ;
+       new = ber_memalloc_x(( i+1 ) * sizeof(BerValue), ctx );
+       if ( !new )
+               return -1;
+       for (j=0; j<i; j++) {
+               ber_dupbv_x( &new[j], &src[j], ctx );
+               if ( BER_BVISNULL( &new[j] )) {
+                       ber_bvarray_free_x( new, ctx );
+                       return -1;
+               }
+       }
+       BER_BVZERO( &new[j] );
+       *dst = new;
+       return 0;
+}
+
 int
 ber_bvarray_add_x( BerVarray *a, BerValue *bv, void *ctx )
 {
        int     n;
 
-       ber_int_options.lbo_valid = LBER_INITIALIZED;
-
        if ( *a == NULL ) {
                if (bv == NULL) {
                        return 0;
@@ -808,6 +808,7 @@ ber_bvarray_add_x( BerVarray *a, BerValue *bv, void *ctx )
 
        (*a)[n++] = *bv;
        (*a)[n].bv_val = NULL;
+       (*a)[n].bv_len = 0;
 
        return n;
 }