]> git.sur5r.net Git - openldap/blobdiff - libraries/liblber/memory.c
hex escape everything requires escape (uncomment #define PRETTY_ESCAPE to revert...
[openldap] / libraries / liblber / memory.c
index 24a4eddfb72c76f6c3a78f3355b5b6d8a12853f8..d836dcc6eab3ad0f55b38c7cc14bde3e0b01369e 100644 (file)
 
 #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
 /*
  * LDAP_MEMORY_DEBUG should only be enabled for the purposes of
  */
 
 struct ber_mem_hdr {
-       union bmu_align_u {
+       ber_int_t       bm_top; /* Pattern to detect buf overrun from prev buffer */
+       ber_int_t       bm_length; /* Length of user allocated area */
+#ifdef LDAP_MEMORY_TRACE
+       ber_int_t       bm_sequence; /* Allocation sequence number */
+#endif
+       union bmu_align_u {     /* Force alignment, pattern to detect back clobber */
                ber_len_t       bmu_len_t;
                ber_tag_t       bmu_tag_t;
                ber_int_t       bmu_int_t;
@@ -33,13 +45,61 @@ struct ber_mem_hdr {
                double  bmu_double;
                long    bmu_long;
                long    (*bmu_funcp)( double );
-               char    bmu_char[4];
+               unsigned char   bmu_char[4];
        } ber_align;
 #define bm_junk        ber_align.bmu_len_t
 #define bm_data        ber_align.bmu_char[1]
+#define bm_char        ber_align.bmu_char
 };
-#define BER_MEM_JUNK 0xdeaddadaU
-static const struct ber_mem_hdr ber_int_mem_hdr = { BER_MEM_JUNK };
+
+/* Pattern at top of allocated space */
+#define LLBER_MEM_JUNK 0xdeaddadaU
+
+static const struct ber_mem_hdr ber_int_mem_hdr = { LBER_MEM_JUNK, 0, 0 };
+
+/* Note sequence and ber_int_options.lbu_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.
+ */
+#ifdef LDAP_MEMORY_TRACE
+static ber_int_t sequence = 0;
+#endif
+
+/* Pattern placed just before user data */
+static unsigned char toppattern[4] = { 0xde, 0xad, 0xba, 0xde };
+/* Pattern placed just after user data */
+static unsigned char endpattern[4] = { 0xd1, 0xed, 0xde, 0xca };
+
+#define mbu_len sizeof(ber_int_mem_hdr.ber_align)
+
+/* Test if pattern placed just before user data is good */
+#define testdatatop(val) ( \
+       *(val->bm_char+mbu_len-4)==toppattern[0] && \
+       *(val->bm_char+mbu_len-3)==toppattern[1] && \
+       *(val->bm_char+mbu_len-2)==toppattern[2] && \
+       *(val->bm_char+mbu_len-1)==toppattern[3] )
+
+/* Place pattern just before user data */
+#define setdatatop(val)        *(val->bm_char+mbu_len-4)=toppattern[0]; \
+       *(val->bm_char+mbu_len-3)=toppattern[1]; \
+       *(val->bm_char+mbu_len-2)=toppattern[2]; \
+       *(val->bm_char+mbu_len-1)=toppattern[3];
+
+/* Test if pattern placed just after user data is good */
+#define testend(val) (         *((unsigned char *)val+0)==endpattern[0] && \
+       *((unsigned char *)val+1)==endpattern[1] && \
+       *((unsigned char *)val+2)==endpattern[2] && \
+       *((unsigned char *)val+3)==endpattern[3] )
+
+/* Place pattern just after user data */
+#define setend(val)    *((unsigned char *)val+0)=endpattern[0]; \
+       *((unsigned char *)val+1)=endpattern[1]; \
+       *((unsigned char *)val+2)=endpattern[2]; \
+       *((unsigned char *)val+3)=endpattern[3];
+
 #define BER_MEM_BADADDR        ((void *) &ber_int_mem_hdr.bm_data)
 #define BER_MEM_VALID(p)       do { \
                assert( (p) != BER_MEM_BADADDR );       \
@@ -80,9 +140,17 @@ ber_memfree( void *p )
 #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 );                          
-               mh->bm_junk = ~BER_MEM_JUNK;
+               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;
+
+#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);
+#endif
+               /* Fill the free space with poison */
+               memset( mh, 0xff, mh->bm_length + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t));
                free( mh );
 #else
                free( p );
@@ -133,11 +201,23 @@ ber_memalloc( ber_len_t s )
 
        if( ber_int_memory_fns == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
-               struct ber_mem_hdr *mh = malloc(s + sizeof(struct ber_mem_hdr));
-
+               struct ber_mem_hdr *mh = malloc(s + sizeof(struct ber_mem_hdr) + sizeof( ber_int_t));
                if( mh == NULL ) return NULL;
 
-               mh->bm_junk = BER_MEM_JUNK;
+               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 */
+
+#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);
+#endif
+               /* poison new memory */
+               memset( (char *)&mh[1], 0xff, s);
 
                BER_MEM_VALID( &mh[1] );
                new = &mh[1];
@@ -173,10 +253,21 @@ ber_memcalloc( ber_len_t n, ber_len_t s )
        if( ber_int_memory_fns == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
                struct ber_mem_hdr *mh = calloc(1,
-                       (n * s) + sizeof(struct ber_mem_hdr) );
+                       (n * s) + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t) );
+               if( mh == NULL ) return NULL;
 
-               mh->bm_junk = BER_MEM_JUNK;
+               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;
+
+#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);
+#endif
                BER_MEM_VALID( &mh[1] );
                new = &mh[1];
 #else
@@ -198,7 +289,7 @@ ber_memcalloc( ber_len_t n, ber_len_t s )
 void *
 ber_memrealloc( void* p, ber_len_t s )
 {
-       void *new;
+       void *new = NULL;
     ber_int_options.lbo_valid = LBER_INITIALIZED;
 
        /* realloc(NULL,s) -> malloc(s) */
@@ -216,20 +307,38 @@ ber_memrealloc( void* p, ber_len_t s )
 
        if( ber_int_memory_fns == NULL ) {
 #ifdef LDAP_MEMORY_DEBUG
+               ber_int_t oldlen;
                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 + sizeof(struct ber_mem_hdr) );
+               assert( mh->bm_top == LBER_MEM_JUNK);
+               assert( testdatatop( mh));
+               assert( testend( (char *)&mh[1] + mh->bm_length) );
+               oldlen = mh->bm_length;
+
+               p = realloc( mh, s + sizeof(struct ber_mem_hdr) + sizeof(ber_int_t) );
+               if( p == NULL ) {
+                       ber_errno = LBER_ERROR_MEMORY;
+                       return NULL;
+               }
 
-               if( p != NULL ) {
                        mh = p;
+               mh->bm_length = s;
+               setend( (char *)&mh[1] + mh->bm_length );
+               if( (s - oldlen) > 0 ) {
+                       /* poison any new memory */
+                       memset( (char *)&mh[1] + oldlen, 0xff, s - oldlen);
+               }
 
-                       assert( mh->bm_junk == BER_MEM_JUNK );
+               assert( mh->bm_top == LBER_MEM_JUNK);
+               assert( testdatatop( mh));
 
+               ber_int_options.lbo_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);
+#endif
                        BER_MEM_VALID( &mh[1] );
-                       new = &mh[1];
-               }
+               return &mh[1];
 #else
                new = realloc( p, s );
 #endif
@@ -256,8 +365,9 @@ ber_bvfree( struct berval *bv )
 
        BER_MEM_VALID( bv );
 
-       if ( bv->bv_val != NULL )
+       if ( bv->bv_val != NULL ) {
                LBER_FREE( bv->bv_val );
+       }
 
        LBER_FREE( (char *) bv );
 }
@@ -276,8 +386,9 @@ ber_bvecfree( struct berval **bv )
 
        BER_MEM_VALID( bv );
 
-       for ( i = 0; bv[i] != NULL; i++ )
+       for ( i = 0; bv[i] != NULL; i++ ) {
                ber_bvfree( bv[i] );
+       }
 
        LBER_FREE( (char *) bv );
 }
@@ -290,7 +401,7 @@ ber_bvecadd( struct berval ***bvec, struct berval *bv )
 
        ber_int_options.lbo_valid = LBER_INITIALIZED;
 
-       if( bvec == NULL ) {
+       if( *bvec == NULL ) {
                if( bv == NULL ) {
                        /* nothing to add */
                        return 0;
@@ -311,7 +422,7 @@ ber_bvecadd( struct berval ***bvec, struct berval *bv )
        BER_MEM_VALID( bvec );
 
        /* count entries */
-       for ( i = 0; bvec[i] != NULL; i++ ) {
+       for ( i = 0; (*bvec)[i] != NULL; i++ ) {
                /* EMPTY */;
        }
 
@@ -364,11 +475,11 @@ ber_bvdup(
                return NULL;
        }
 
-       SAFEMEMCPY( new->bv_val, bv->bv_val, bv->bv_len );
+       AC_MEMCPY( new->bv_val, bv->bv_val, bv->bv_len );
        new->bv_val[bv->bv_len] = '\0';
        new->bv_len = bv->bv_len;
 
-       return( new );
+       return new;
 }
 
 struct berval *
@@ -389,12 +500,6 @@ ber_bvstr(
                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 );
 
@@ -428,7 +533,7 @@ ber_bvstrdup(
                LBER_FREE( p );
        }
 
-       return( new );
+       return new;
 }
 
 char *
@@ -445,16 +550,69 @@ ber_strdup( LDAP_CONST char *s )
 
        if( s == NULL ) {
                ber_errno = LBER_ERROR_PARAM;
-               return( NULL );
+               return NULL;
        }
 
        len = strlen( s ) + 1;
 
        if ( (p = LBER_MALLOC( len )) == NULL ) {
                ber_errno = LBER_ERROR_MEMORY;
-               return( NULL );
+               return NULL;
+       }
+
+       AC_MEMCPY( p, s, len );
+       return p;
+}
+
+char *
+ber_strndup( LDAP_CONST char *s, ber_len_t l )
+{
+       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
+
+       if( s == NULL ) {
+               ber_errno = LBER_ERROR_PARAM;
+               return NULL;
+       }
+
+       len = strlen( s );
+
+       if ( len > l ) {
+               len = l;
+       }
+
+       if ( (p = LBER_MALLOC( len + 1 )) == NULL ) {
+               ber_errno = LBER_ERROR_MEMORY;
+               return NULL;
+       }
+
+       AC_MEMCPY( p, s, len );
+       p[ len ] = '\0';
+       return p;
+}
+
+char *
+ber_strndup__( LDAP_CONST char *s, size_t l )
+{
+       char    *p;
+       size_t  len;
+
+       if ( s == NULL ) {
+               return NULL;
+       }
+
+       len = strlen( s );
+       if (( p = LBER_MALLOC( len + 1 ) ) == NULL ) {
+               return NULL;
        }
 
-       SAFEMEMCPY( p, s, len );
-       return( p );
+       AC_MEMCPY( p, s, len );
+       p[ len ] = '\0';
+       return p;
 }