1 /*	$NetBSD: ch_malloc.c,v 1.3 2021/08/14 16:14:58 christos Exp $	*/
2 
3 /* ch_malloc.c - malloc routines that test returns from malloc and friends */
4 /* $OpenLDAP$ */
5 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
6  *
7  * Copyright 1998-2021 The OpenLDAP Foundation.
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted only as authorized by the OpenLDAP
12  * Public License.
13  *
14  * A copy of this license is available in the file LICENSE in the
15  * top-level directory of the distribution or, alternatively, at
16  * <http://www.OpenLDAP.org/license.html>.
17  */
18 /* Portions Copyright (c) 1995 Regents of the University of Michigan.
19  * All rights reserved.
20  *
21  * Redistribution and use in source and binary forms are permitted
22  * provided that this notice is preserved and that due credit is given
23  * to the University of Michigan at Ann Arbor. The name of the University
24  * may not be used to endorse or promote products derived from this
25  * software without specific prior written permission. This software
26  * is provided ``as is'' without express or implied warranty.
27  */
28 
29 #define CH_FREE 1
30 
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: ch_malloc.c,v 1.3 2021/08/14 16:14:58 christos Exp $");
33 
34 #include "portable.h"
35 
36 #include <stdio.h>
37 
38 #include <ac/stdlib.h>
39 
40 #include <ac/string.h>
41 #include <ac/socket.h>
42 
43 #include "slap.h"
44 
45 BerMemoryFunctions ch_mfuncs = {
46 	(BER_MEMALLOC_FN *)ch_malloc,
47 	(BER_MEMCALLOC_FN *)ch_calloc,
48 	(BER_MEMREALLOC_FN *)ch_realloc,
49 	(BER_MEMFREE_FN *)ch_free
50 };
51 
52 void *
ch_malloc(ber_len_t size)53 ch_malloc(
54     ber_len_t	size
55 )
56 {
57 	void	*new;
58 
59 	if ( (new = (void *) ber_memalloc_x( size, NULL )) == NULL ) {
60 		Debug( LDAP_DEBUG_ANY, "ch_malloc of %lu bytes failed\n",
61 			(long) size );
62 		assert( 0 );
63 		exit( EXIT_FAILURE );
64 	}
65 
66 	return( new );
67 }
68 
69 void *
ch_realloc(void * block,ber_len_t size)70 ch_realloc(
71     void		*block,
72     ber_len_t	size
73 )
74 {
75 	void	*new, *ctx;
76 
77 	if ( block == NULL ) {
78 		return( ch_malloc( size ) );
79 	}
80 
81 	if( size == 0 ) {
82 		ch_free( block );
83 		return NULL;
84 	}
85 
86 	ctx = slap_sl_context( block );
87 	if ( ctx ) {
88 		return slap_sl_realloc( block, size, ctx );
89 	}
90 
91 	if ( (new = (void *) ber_memrealloc_x( block, size, NULL )) == NULL ) {
92 		Debug( LDAP_DEBUG_ANY, "ch_realloc of %lu bytes failed\n",
93 			(long) size );
94 		assert( 0 );
95 		exit( EXIT_FAILURE );
96 	}
97 
98 	return( new );
99 }
100 
101 void *
ch_calloc(ber_len_t nelem,ber_len_t size)102 ch_calloc(
103     ber_len_t	nelem,
104     ber_len_t	size
105 )
106 {
107 	void	*new;
108 
109 	if ( (new = (void *) ber_memcalloc_x( nelem, size, NULL )) == NULL ) {
110 		Debug( LDAP_DEBUG_ANY, "ch_calloc of %lu elems of %lu bytes failed\n",
111 		  (long) nelem, (long) size );
112 		assert( 0 );
113 		exit( EXIT_FAILURE );
114 	}
115 
116 	return( new );
117 }
118 
119 char *
ch_strdup(const char * string)120 ch_strdup(
121     const char *string
122 )
123 {
124 	char	*new;
125 
126 	if ( (new = ber_strdup_x( string, NULL )) == NULL ) {
127 		Debug( LDAP_DEBUG_ANY, "ch_strdup(%s) failed\n", string );
128 		assert( 0 );
129 		exit( EXIT_FAILURE );
130 	}
131 
132 	return( new );
133 }
134 
135 void
ch_free(void * ptr)136 ch_free( void *ptr )
137 {
138 	void *ctx;
139 
140 	ctx = slap_sl_context( ptr );
141 	if (ctx) {
142 		slap_sl_free( ptr, ctx );
143 	} else {
144 		ber_memfree_x( ptr, NULL );
145 	}
146 }
147 
148