openldap/servers/slapd/ch_malloc.c
Howard Chu be28bf247e Added sl_context() to return the context associated with an alloc'd ptr.
Improved ch_malloc/sl_malloc compatibility.
2003-04-12 05:12:40 +00:00

144 lines
2.5 KiB
C

/* ch_malloc.c - malloc routines that test returns from malloc and friends */
/* $OpenLDAP$ */
/*
* Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
* COPYING RESTRICTIONS APPLY, see COPYRIGHT file
*/
#define CH_FREE 1
#include "portable.h"
#include <stdio.h>
#include <ac/stdlib.h>
#include <ac/string.h>
#include <ac/socket.h>
#include "slap.h"
BerMemoryFunctions ch_mfuncs = {
(BER_MEMALLOC_FN *)ch_malloc,
(BER_MEMCALLOC_FN *)ch_calloc,
(BER_MEMREALLOC_FN *)ch_realloc,
(BER_MEMFREE_FN *)ch_free
};
void *
ch_malloc(
ber_len_t size
)
{
void *new;
if ( (new = (void *) ber_memalloc_x( size, NULL )) == NULL ) {
#ifdef NEW_LOGGING
LDAP_LOG( OPERATION, ERR,
"ch_malloc: allocation of %lu bytes failed\n", (long)size, 0,0 );
#else
Debug( LDAP_DEBUG_ANY, "ch_malloc of %lu bytes failed\n",
(long) size, 0, 0 );
#endif
assert( 0 );
exit( EXIT_FAILURE );
}
return( new );
}
void *
ch_realloc(
void *block,
ber_len_t size
)
{
void *new, *ctx;
if ( block == NULL ) {
return( ch_malloc( size ) );
}
if( size == 0 ) {
ch_free( block );
}
ctx = sl_context( block );
if ( ctx ) {
return sl_realloc( block, size, ctx );
}
if ( (new = (void *) ber_memrealloc_x( block, size, NULL )) == NULL ) {
#ifdef NEW_LOGGING
LDAP_LOG( OPERATION, ERR,
"ch_realloc: reallocation of %lu bytes failed\n", (long)size, 0,0 );
#else
Debug( LDAP_DEBUG_ANY, "ch_realloc of %lu bytes failed\n",
(long) size, 0, 0 );
#endif
assert( 0 );
exit( EXIT_FAILURE );
}
return( new );
}
void *
ch_calloc(
ber_len_t nelem,
ber_len_t size
)
{
void *new;
if ( (new = (void *) ber_memcalloc_x( nelem, size, NULL )) == NULL ) {
#ifdef NEW_LOGGING
LDAP_LOG( OPERATION, ERR,
"ch_calloc: allocation of %lu elements of %lu bytes faild\n",
(long)nelem, (long)size, 0 );
#else
Debug( LDAP_DEBUG_ANY, "ch_calloc of %lu elems of %lu bytes failed\n",
(long) nelem, (long) size, 0 );
#endif
assert( 0 );
exit( EXIT_FAILURE );
}
return( new );
}
char *
ch_strdup(
const char *string
)
{
char *new;
if ( (new = ber_strdup_x( string, NULL )) == NULL ) {
#ifdef NEW_LOGGING
LDAP_LOG( OPERATION, ERR,
"chr_strdup: duplication of \"%s\" failed\n", string, 0, 0 );
#else
Debug( LDAP_DEBUG_ANY, "ch_strdup(%s) failed\n", string, 0, 0 );
#endif
assert( 0 );
exit( EXIT_FAILURE );
}
return( new );
}
void
ch_free( void *ptr )
{
void *ctx;
ctx = sl_context( ptr );
if (ctx) {
sl_free( ptr, ctx );
} else {
ber_memfree_x( ptr, NULL );
}
}