2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
5 * This package is an SSL implementation written
6 * by Eric Young (eay@cryptsoft.com).
7 * The implementation was written so as to conform with Netscapes SSL.
9 * This library is free for commercial and non-commercial use as long as
10 * the following conditions are aheared to. The following conditions
11 * apply to all code found in this distribution, be it the RC4, RSA,
12 * lhash, DES, etc., code; not just the SSL code. The SSL documentation
13 * included with this distribution is covered by the same copyright terms
14 * except that the holder is Tim Hudson (tjh@cryptsoft.com).
16 * Copyright remains Eric Young's, and as such any Copyright notices in
17 * the code are not to be removed.
18 * If this package is used in a product, Eric Young should be given attribution
19 * as the author of the parts of the library used.
20 * This can be in the form of a textual message at program startup or
21 * in documentation (online or textual) provided with the package.
23 * Redistribution and use in source and binary forms, with or without
24 * modification, are permitted provided that the following conditions
26 * 1. Redistributions of source code must retain the copyright
27 * notice, this list of conditions and the following disclaimer.
28 * 2. Redistributions in binary form must reproduce the above copyright
29 * notice, this list of conditions and the following disclaimer in the
30 * documentation and/or other materials provided with the distribution.
31 * 3. All advertising materials mentioning features or use of this software
32 * must display the following acknowledgement:
33 * "This product includes cryptographic software written by
34 * Eric Young (eay@cryptsoft.com)"
35 * The word 'cryptographic' can be left out if the rouines from the library
36 * being used are not cryptographic related :-).
37 * 4. If you include any Windows specific code (or a derivative thereof) from
38 * the apps directory (application code) you must include an acknowledgement:
39 * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
41 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
53 * The licence and distribution terms for any publically available version or
54 * derivative of this code cannot be changed. i.e. this code cannot simply be
55 * copied and put under another distribution licence
56 * [including the GNU Public Licence.]
61 #include <openssl/crypto.h>
62 #ifdef CRYPTO_MDEBUG_TIME
65 #include <openssl/buffer.h>
66 #include <openssl/bio.h>
67 #include <openssl/lhash.h>
70 /* #ifdef CRYPTO_MDEBUG */
71 /* static int mh_mode=CRYPTO_MEM_CHECK_ON; */
73 static int mh_mode=CRYPTO_MEM_CHECK_OFF;
75 /* State CRYPTO_MEM_CHECK_ON exists only temporarily when the library
76 * thinks that certain allocations should not be checked (e.g. the data
77 * structures used for memory checking). It is not suitable as an initial
78 * state: the library will unexpectedly enable memory checking when it
79 * executes one of those sections that want to disable checking
82 * State CRYPTO_MEM_CHECK_ENABLE without ..._ON makes no sense whatsoever.
85 static unsigned long order=0;
87 static LHASH *mh=NULL;
95 #ifdef CRYPTO_MDEBUG_THREAD
99 #ifdef CRYPTO_MDEBUG_TIME
104 int CRYPTO_mem_ctrl(int mode)
108 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
111 /* for applications: */
112 case CRYPTO_MEM_CHECK_ON: /* aka MemCheck_start() */
113 mh_mode = CRYPTO_MEM_CHECK_ON|CRYPTO_MEM_CHECK_ENABLE;
115 case CRYPTO_MEM_CHECK_OFF: /* aka MemCheck_stop() */
119 /* switch off temporarily (for library-internal use): */
120 case CRYPTO_MEM_CHECK_DISABLE: /* aka MemCheck_off() */
121 mh_mode&= ~CRYPTO_MEM_CHECK_ENABLE;
123 case CRYPTO_MEM_CHECK_ENABLE: /* aka MemCheck_on() */
124 if (mh_mode&CRYPTO_MEM_CHECK_ON)
125 mh_mode|=CRYPTO_MEM_CHECK_ENABLE;
131 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
135 static int mem_cmp(MEM *a, MEM *b)
137 return(a->addr - b->addr);
140 static unsigned long mem_hash(MEM *a)
144 ret=(unsigned long)a->addr;
146 ret=ret*17851+(ret>>14)*7+(ret>>4)*251;
150 static char *(*malloc_locked_func)()=(char *(*)())malloc;
151 static void (*free_locked_func)()=(void (*)())free;
152 static char *(*malloc_func)()= (char *(*)())malloc;
153 static char *(*realloc_func)()= (char *(*)())realloc;
154 static void (*free_func)()= (void (*)())free;
156 void CRYPTO_set_mem_functions(char *(*m)(), char *(*r)(), void (*f)())
158 if ((m == NULL) || (r == NULL) || (f == NULL)) return;
162 malloc_locked_func=m;
166 void CRYPTO_set_locked_mem_functions(char *(*m)(), void (*f)())
168 if ((m == NULL) || (f == NULL)) return;
169 malloc_locked_func=m;
173 void CRYPTO_get_mem_functions(char *(**m)(), char *(**r)(), void (**f)())
175 if (m != NULL) *m=malloc_func;
176 if (r != NULL) *r=realloc_func;
177 if (f != NULL) *f=free_func;
180 void CRYPTO_get_locked_mem_functions(char *(**m)(), void (**f)())
182 if (m != NULL) *m=malloc_locked_func;
183 if (f != NULL) *f=free_locked_func;
186 void *CRYPTO_malloc_locked(int num)
188 return(malloc_locked_func(num));
191 void CRYPTO_free_locked(void *str)
193 free_locked_func(str);
196 void *CRYPTO_malloc(int num)
198 return(malloc_func(num));
201 void *CRYPTO_realloc(void *str, int num)
203 return(realloc_func(str,num));
206 void CRYPTO_free(void *str)
211 static unsigned long break_order_num=0;
212 void *CRYPTO_dbg_malloc(int num, const char *file, int line)
217 if ((ret=malloc_func(num)) == NULL)
220 if (mh_mode & CRYPTO_MEM_CHECK_ENABLE)
223 if ((m=(MEM *)Malloc(sizeof(MEM))) == NULL)
229 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
232 if ((mh=lh_new(mem_hash,mem_cmp)) == NULL)
245 #ifdef CRYPTO_MDEBUG_THREAD
246 m->thread=CRYPTO_thread_id();
248 if (order == break_order_num)
254 #ifdef CRYPTO_MDEBUG_TIME
257 if ((mm=(MEM *)lh_insert(mh,(char *)m)) != NULL)
259 /* Not good, but don't sweat it */
263 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
269 void CRYPTO_dbg_free(void *addr)
273 if ((mh_mode & CRYPTO_MEM_CHECK_ENABLE) && (mh != NULL))
276 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
278 mp=(MEM *)lh_delete(mh,(char *)&m);
281 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
287 void *CRYPTO_dbg_realloc(void *addr, int num, const char *file, int line)
292 ret=realloc_func(addr,num);
293 if (ret == addr) return(ret);
295 if (mh_mode & CRYPTO_MEM_CHECK_ENABLE)
298 if (ret == NULL) return(NULL);
300 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
301 mp=(MEM *)lh_delete(mh,(char *)&m);
305 lh_insert(mh,(char *)mp);
307 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
313 void *CRYPTO_remalloc(void *a, int n)
315 if (a != NULL) Free(a);
320 void *CRYPTO_dbg_remalloc(void *a, int n, const char *file, int line)
322 if (a != NULL) CRYPTO_dbg_free(a);
323 a=(char *)CRYPTO_dbg_malloc(n,file,line);
328 typedef struct mem_leak_st
335 static void print_leak(MEM *m, MEM_LEAK *l)
338 #ifdef CRYPTO_MDEBUG_TIME
342 if(m->addr == (char *)l->bio)
345 #ifdef CRYPTO_MDEBUG_TIME
346 lcl = localtime(&m->time);
350 #ifdef CRYPTO_MDEBUG_TIME
353 "%5lu file=%s, line=%d, "
354 #ifdef CRYPTO_MDEBUG_THREAD
357 "number=%d, address=%08lX\n",
358 #ifdef CRYPTO_MDEBUG_TIME
359 lcl->tm_hour,lcl->tm_min,lcl->tm_sec,
361 m->order,m->file,m->line,
362 #ifdef CRYPTO_MDEBUG_THREAD
365 m->num,(unsigned long)m->addr);
367 BIO_puts(l->bio,buf);
372 void CRYPTO_mem_leaks(BIO *b)
377 if (mh == NULL) return;
381 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
382 lh_doall_arg(mh,(void (*)())print_leak,(char *)&ml);
383 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
386 sprintf(buf,"%ld bytes leaked in %d chunks\n",
393 lh_node_stats_bio(mh,b);
394 lh_node_usage_stats_bio(mh,b);
398 static void (*mem_cb)()=NULL;
400 static void cb_leak(MEM *m, char *cb)
402 void (*mem_callback)()=(void (*)())cb;
403 mem_callback(m->order,m->file,m->line,m->num,m->addr);
406 void CRYPTO_mem_leaks_cb(void (*cb)())
408 if (mh == NULL) return;
409 CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
411 lh_doall_arg(mh,(void (*)())cb_leak,(char *)mem_cb);
413 CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
417 void CRYPTO_mem_leaks_fp(FILE *fp)
421 if (mh == NULL) return;
422 if ((b=BIO_new(BIO_s_file())) == NULL)
424 BIO_set_fp(b,fp,BIO_NOCLOSE);