]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/memory.c
Move schema needed by slapd to core.schema so that only one file
[openldap] / libraries / liblber / memory.c
index 5dcc1ec38864a68125075bd41b9a87aeba0a7870..24a4eddfb72c76f6c3a78f3355b5b6d8a12853f8 100644 (file)
@@ -1,5 +1,6 @@
+/* $OpenLDAP$ */
 /*
- * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
+ * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
  */
 #include "portable.h"
@@ -7,44 +8,81 @@
 #include <ac/stdlib.h>
 #include <ac/string.h>
 
-#undef LDAP_F_PRE
-#define LDAP_F_PRE LDAP_F_EXPORT
-
 #include "lber-int.h"
 
-#ifdef LDAP_MEMORY_DEBUG
+#if 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.
+ *
+ * The code behind this macro is subject to change as needed to
+ * support this testing.
+ */
+
 struct ber_mem_hdr {
        union bmu_align_u {
+               ber_len_t       bmu_len_t;
+               ber_tag_t       bmu_tag_t;
+               ber_int_t       bmu_int_t;
+
                size_t  bmu_size_t;
                void *  bmu_voidp;
                double  bmu_double;
                long    bmu_long;
+               long    (*bmu_funcp)( double );
+               char    bmu_char[4];
        } ber_align;
-#define bm_junk        ber_align.bmu_size_t
+#define bm_junk        ber_align.bmu_len_t
+#define bm_data        ber_align.bmu_char[1]
 };
-#define BER_MEM_JUNK 0xddeeddeeU
+#define BER_MEM_JUNK 0xdeaddadaU
+static const struct ber_mem_hdr ber_int_mem_hdr = { BER_MEM_JUNK };
+#define BER_MEM_BADADDR        ((void *) &ber_int_mem_hdr.bm_data)
+#define BER_MEM_VALID(p)       do { \
+               assert( (p) != BER_MEM_BADADDR );       \
+               assert( (p) != (void *) &ber_int_mem_hdr );     \
+       } while(0)
+
+#else
+#define BER_MEM_VALID(p)       /* no-op */
 #endif
 
 BerMemoryFunctions *ber_int_memory_fns = NULL;
 
+#if 0 && defined( LDAP_MEMORY_DEBUG )
+void
+ber_int_memfree( void **p )
+{
+       assert( p != NULL );
+       BER_MEM_VALID( *p );
+
+       ber_memfree( p );
+
+       *p = BER_MEM_BADADDR;
+}
+#endif
+
 void
 ber_memfree( void *p )
 {
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
-       /* catch p == NULL when debugging */
-       assert( p != NULL );
-
-       /* ignore p == NULL when not debugging */
        if( p == NULL ) {
                return;
        }
 
+       BER_MEM_VALID( p );
+
        if( ber_int_memory_fns == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
                struct ber_mem_hdr *mh = (struct ber_mem_hdr *)
                        ((char *)p - sizeof(struct ber_mem_hdr));
-               assert( mh->bm_junk == BER_MEM_JUNK );
+
+               assert( mh->bm_junk == BER_MEM_JUNK );                          
+               mh->bm_junk = ~BER_MEM_JUNK;
                free( mh );
 #else
                free( p );
@@ -54,7 +92,6 @@ ber_memfree( void *p )
 
        assert( ber_int_memory_fns->bmf_free );
 
-               
        (*ber_int_memory_fns->bmf_free)( p );
 }
 
@@ -64,7 +101,13 @@ ber_memvfree( void **vec )
 {
        int     i;
 
-       assert(vec != NULL);    /* vec damn better point to something */
+    ber_int_options.lbo_valid = LBER_INITIALIZED;
+
+       if( vec == NULL ) {
+               return;
+       }
+
+       BER_MEM_VALID( vec );
 
        for ( i = 0; vec[i] != NULL; i++ ) {
                LBER_FREE( vec[i] );
@@ -75,14 +118,15 @@ ber_memvfree( void **vec )
 
 
 void *
-ber_memalloc( size_t s )
+ber_memalloc( ber_len_t s )
 {
+       void *new;
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
-       /* catch s == 0 when debugging */
-       assert( s );
+#ifdef LDAP_MEMORY_DEBUG
+       assert( s != 0 );
+#endif
 
-       /* ignore s == 0 when not debugging */
        if( s == 0 ) {
                return NULL;
        }
@@ -95,27 +139,33 @@ ber_memalloc( size_t s )
 
                mh->bm_junk = BER_MEM_JUNK;
 
-               return &mh[1];
+               BER_MEM_VALID( &mh[1] );
+               new = &mh[1];
 #else
-               return malloc( s );
+               new = malloc( s );
 #endif
+       } else {
+               new = (*ber_int_memory_fns->bmf_malloc)( s );
        }
 
-       assert( ber_int_memory_fns->bmf_malloc );
+       if( new == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+       }
 
-       return (*ber_int_memory_fns->bmf_malloc)( s );
+       return new;
 }
 
 
 void *
-ber_memcalloc( size_t n, size_t s )
+ber_memcalloc( ber_len_t n, ber_len_t s )
 {
+       void *new;
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
-       /* catch s,n == 0 when debugging */
-       assert( n && s );
+#ifdef LDAP_MEMORY_DEBUG
+       assert( n != 0 && s != 0);
+#endif
 
-       /* ignore s,n == 0 when not debugging */
        if( n == 0 || s == 0 ) {
                return NULL;
        }
@@ -126,21 +176,29 @@ ber_memcalloc( size_t n, size_t s )
                        (n * s) + sizeof(struct ber_mem_hdr) );
 
                mh->bm_junk = BER_MEM_JUNK;
-               return &mh[1];
+
+               BER_MEM_VALID( &mh[1] );
+               new = &mh[1];
 #else
-               return calloc( n, s );
+               new = calloc( n, s );
 #endif
+
+       } else {
+               new = (*ber_int_memory_fns->bmf_calloc)( n, s );
        }
 
-       assert( ber_int_memory_fns->bmf_calloc );
+       if( new == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+       }
 
-       return (*ber_int_memory_fns->bmf_calloc)( n, s );
+       return new;
 }
 
 
 void *
-ber_memrealloc( void* p, size_t s )
+ber_memrealloc( void* p, ber_len_t s )
 {
+       void *new;
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        /* realloc(NULL,s) -> malloc(s) */
@@ -154,37 +212,50 @@ ber_memrealloc( void* p, size_t s )
                return NULL;
        }
 
+       BER_MEM_VALID( p );
+
        if( ber_int_memory_fns == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
                struct ber_mem_hdr *mh = (struct ber_mem_hdr *)
                        ((char *)p - sizeof(struct ber_mem_hdr));
                assert( mh->bm_junk == BER_MEM_JUNK );
 
-               p = realloc( mh, s );
+               p = realloc( mh, s + sizeof(struct ber_mem_hdr) );
 
-               if( p == NULL ) return NULL;
+               if( p != NULL ) {
+                       mh = p;
 
-               mh = p;
+                       assert( mh->bm_junk == BER_MEM_JUNK );
 
-               return &mh[1];
+                       BER_MEM_VALID( &mh[1] );
+                       new = &mh[1];
+               }
 #else
-               return realloc( p, s );
+               new = realloc( p, s );
 #endif
+       } else {
+               new = (*ber_int_memory_fns->bmf_realloc)( p, s );
        }
 
-       assert( ber_int_memory_fns->bmf_realloc );
+       if( new == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+       }
 
-       return (*ber_int_memory_fns->bmf_realloc)( p, s );
+       return new;
 }
 
 
 void
 ber_bvfree( struct berval *bv )
 {
-       assert(bv != NULL);                     /* bv damn better point to something */
-
        ber_int_options.lbo_valid = LBER_INITIALIZED;
 
+       if( bv == NULL ) {
+               return;
+       }
+
+       BER_MEM_VALID( bv );
+
        if ( bv->bv_val != NULL )
                LBER_FREE( bv->bv_val );
 
@@ -197,16 +268,71 @@ ber_bvecfree( struct berval **bv )
 {
        int     i;
 
-       assert(bv != NULL);                     /* bv damn better point to something */
-
        ber_int_options.lbo_valid = LBER_INITIALIZED;
 
+       if( bv == NULL ) {
+               return;
+       }
+
+       BER_MEM_VALID( bv );
+
        for ( i = 0; bv[i] != NULL; i++ )
                ber_bvfree( bv[i] );
 
        LBER_FREE( (char *) bv );
 }
 
+int
+ber_bvecadd( struct berval ***bvec, struct berval *bv )
+{
+       ber_len_t i;
+       struct berval **new;
+
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
+
+       if( bvec == NULL ) {
+               if( bv == NULL ) {
+                       /* nothing to add */
+                       return 0;
+               }
+
+               *bvec = ber_memalloc( 2 * sizeof(struct berval *) );
+
+               if( *bvec == NULL ) {
+                       return -1;
+               }
+
+               (*bvec)[0] = bv;
+               (*bvec)[1] = NULL;
+
+               return 1;
+       }
+
+       BER_MEM_VALID( bvec );
+
+       /* count entries */
+       for ( i = 0; bvec[i] != NULL; i++ ) {
+               /* EMPTY */;
+       }
+
+       if( bv == NULL ) {
+               return i;
+       }
+
+       new = ber_memrealloc( *bvec, (i+2) * sizeof(struct berval *));
+
+       if( new == NULL ) {
+               return -1;
+       }
+
+       *bvec = new;
+
+       (*bvec)[i++] = bv;
+       (*bvec)[i] = NULL;
+
+       return i;
+}
+
 
 struct berval *
 ber_bvdup(
@@ -214,16 +340,15 @@ ber_bvdup(
 {
        struct berval *new;
 
-       assert( bv != NULL );
-
        ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        if( bv == NULL ) {
+               ber_errno = LBER_ERROR_PARAM;
                return NULL;
        }
 
-
        if(( new = LBER_MALLOC( sizeof(struct berval) )) == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
                return NULL;
        }
 
@@ -234,27 +359,102 @@ ber_bvdup(
        }
 
        if(( new->bv_val = LBER_MALLOC( bv->bv_len + 1 )) == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
                LBER_FREE( new );
                return NULL;
        }
 
-       SAFEMEMCPY( new->bv_val, bv->bv_val, (size_t) bv->bv_len );
+       SAFEMEMCPY( new->bv_val, bv->bv_val, bv->bv_len );
        new->bv_val[bv->bv_len] = '\0';
        new->bv_len = bv->bv_len;
 
        return( new );
 }
 
+struct berval *
+ber_bvstr(
+       LDAP_CONST char *s )
+{
+       struct berval *new;
+
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
+
+       if( s == NULL ) {
+               ber_errno = LBER_ERROR_PARAM;
+               return NULL;
+       }
+
+       if(( new = LBER_MALLOC( sizeof(struct berval) )) == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+               return NULL;
+       }
+
+       if ( *s == '\0' ) {
+               new->bv_val = NULL;
+               new->bv_len = 0;
+               return new;
+       }
+
+       new->bv_val = (char *) s;
+       new->bv_len = strlen( s );
+
+       return( new );
+}
+
+struct berval *
+ber_bvstrdup(
+       LDAP_CONST char *s )
+{
+       struct berval *new;
+       char *p;
+
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
+
+       if( s == NULL ) {
+               ber_errno = LBER_ERROR_PARAM;
+               return NULL;
+       }
+
+       p = LBER_STRDUP( s );
+
+       if( p == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+               return NULL;
+       }
+
+       new = ber_bvstr( p );
+
+       if( new == NULL || *p == '\0' ) {
+               LBER_FREE( p );
+       }
+
+       return( new );
+}
+
 char *
-(ber_strdup)( LDAP_CONST char *s )
+ber_strdup( LDAP_CONST char *s )
 {
        char    *p;
-       size_t  len = strlen( s ) + 1;
+       size_t  len;
+       
+       ber_int_options.lbo_valid = LBER_INITIALIZED;
 
-       if ( (p = (char *) LBER_MALLOC( len )) == NULL ) {
+#ifdef LDAP_MEMORY_DEBUG
+       assert(s != NULL);                      /* bv damn better point to something */
+#endif
+
+       if( s == NULL ) {
+               ber_errno = LBER_ERROR_PARAM;
+               return( NULL );
+       }
+
+       len = strlen( s ) + 1;
+
+       if ( (p = LBER_MALLOC( len )) == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
                return( NULL );
        }
 
        SAFEMEMCPY( p, s, len );
        return( p );
-}
\ No newline at end of file
+}