|  | /* crypto/mem.c */ | 
|  | /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com) | 
|  | * All rights reserved. | 
|  | * | 
|  | * This package is an SSL implementation written | 
|  | * by Eric Young (eay@cryptsoft.com). | 
|  | * The implementation was written so as to conform with Netscapes SSL. | 
|  | * | 
|  | * This library is free for commercial and non-commercial use as long as | 
|  | * the following conditions are aheared to.  The following conditions | 
|  | * apply to all code found in this distribution, be it the RC4, RSA, | 
|  | * lhash, DES, etc., code; not just the SSL code.  The SSL documentation | 
|  | * included with this distribution is covered by the same copyright terms | 
|  | * except that the holder is Tim Hudson (tjh@cryptsoft.com). | 
|  | * | 
|  | * Copyright remains Eric Young's, and as such any Copyright notices in | 
|  | * the code are not to be removed. | 
|  | * If this package is used in a product, Eric Young should be given attribution | 
|  | * as the author of the parts of the library used. | 
|  | * This can be in the form of a textual message at program startup or | 
|  | * in documentation (online or textual) provided with the package. | 
|  | * | 
|  | * Redistribution and use in source and binary forms, with or without | 
|  | * modification, are permitted provided that the following conditions | 
|  | * are met: | 
|  | * 1. Redistributions of source code must retain the copyright | 
|  | *    notice, this list of conditions and the following disclaimer. | 
|  | * 2. Redistributions in binary form must reproduce the above copyright | 
|  | *    notice, this list of conditions and the following disclaimer in the | 
|  | *    documentation and/or other materials provided with the distribution. | 
|  | * 3. All advertising materials mentioning features or use of this software | 
|  | *    must display the following acknowledgement: | 
|  | *    "This product includes cryptographic software written by | 
|  | *     Eric Young (eay@cryptsoft.com)" | 
|  | *    The word 'cryptographic' can be left out if the rouines from the library | 
|  | *    being used are not cryptographic related :-). | 
|  | * 4. If you include any Windows specific code (or a derivative thereof) from | 
|  | *    the apps directory (application code) you must include an acknowledgement: | 
|  | *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)" | 
|  | * | 
|  | * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND | 
|  | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | 
|  | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | 
|  | * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE | 
|  | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | 
|  | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | 
|  | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | 
|  | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | 
|  | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | 
|  | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | 
|  | * SUCH DAMAGE. | 
|  | * | 
|  | * The licence and distribution terms for any publically available version or | 
|  | * derivative of this code cannot be changed.  i.e. this code cannot simply be | 
|  | * copied and put under another distribution licence | 
|  | * [including the GNU Public Licence.] | 
|  | */ | 
|  |  | 
|  | #include <stdio.h> | 
|  | #include <stdlib.h> | 
|  | #include <openssl/buffer.h> | 
|  | #include <openssl/bio.h> | 
|  | #include <openssl/lhash.h> | 
|  | #include "cryptlib.h" | 
|  |  | 
|  | #ifdef CRYPTO_MDEBUG | 
|  | static int mh_mode=CRYPTO_MEM_CHECK_ON; | 
|  | #else | 
|  | static int mh_mode=CRYPTO_MEM_CHECK_OFF; | 
|  | #endif | 
|  | static unsigned long order=0; | 
|  |  | 
|  | static LHASH *mh=NULL; | 
|  |  | 
|  | typedef struct mem_st | 
|  | { | 
|  | char *addr; | 
|  | int num; | 
|  | const char *file; | 
|  | int line; | 
|  | unsigned long order; | 
|  | } MEM; | 
|  |  | 
|  | int CRYPTO_mem_ctrl(int mode) | 
|  | { | 
|  | int ret=mh_mode; | 
|  |  | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | switch (mode) | 
|  | { | 
|  | case CRYPTO_MEM_CHECK_ON: | 
|  | mh_mode|=CRYPTO_MEM_CHECK_ON; | 
|  | break; | 
|  | case CRYPTO_MEM_CHECK_OFF: | 
|  | mh_mode&= ~CRYPTO_MEM_CHECK_ON; | 
|  | break; | 
|  | case CRYPTO_MEM_CHECK_DISABLE: | 
|  | mh_mode&= ~CRYPTO_MEM_CHECK_ENABLE; | 
|  | break; | 
|  | case CRYPTO_MEM_CHECK_ENABLE: | 
|  | if (mh_mode&CRYPTO_MEM_CHECK_ON) | 
|  | mh_mode|=CRYPTO_MEM_CHECK_ENABLE; | 
|  | break; | 
|  | default: | 
|  | break; | 
|  | } | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | return(ret); | 
|  | } | 
|  |  | 
|  | static int mem_cmp(MEM *a, MEM *b) | 
|  | { | 
|  | return(a->addr - b->addr); | 
|  | } | 
|  |  | 
|  | static unsigned long mem_hash(MEM *a) | 
|  | { | 
|  | unsigned long ret; | 
|  |  | 
|  | ret=(unsigned long)a->addr; | 
|  |  | 
|  | ret=ret*17851+(ret>>14)*7+(ret>>4)*251; | 
|  | return(ret); | 
|  | } | 
|  |  | 
|  | static char *(*malloc_locked_func)()=(char *(*)())malloc; | 
|  | static void (*free_locked_func)()=(void (*)())free; | 
|  | static char *(*malloc_func)()=	(char *(*)())malloc; | 
|  | static char *(*realloc_func)()=	(char *(*)())realloc; | 
|  | static void (*free_func)()=	(void (*)())free; | 
|  |  | 
|  | void CRYPTO_set_mem_functions(char *(*m)(), char *(*r)(), void (*f)()) | 
|  | { | 
|  | if ((m == NULL) || (r == NULL) || (f == NULL)) return; | 
|  | malloc_func=m; | 
|  | realloc_func=r; | 
|  | free_func=f; | 
|  | malloc_locked_func=m; | 
|  | free_locked_func=f; | 
|  | } | 
|  |  | 
|  | void CRYPTO_set_locked_mem_functions(char *(*m)(), void (*f)()) | 
|  | { | 
|  | if ((m == NULL) || (f == NULL)) return; | 
|  | malloc_locked_func=m; | 
|  | free_locked_func=f; | 
|  | } | 
|  |  | 
|  | void CRYPTO_get_mem_functions(char *(**m)(), char *(**r)(), void (**f)()) | 
|  | { | 
|  | if (m != NULL) *m=malloc_func; | 
|  | if (r != NULL) *r=realloc_func; | 
|  | if (f != NULL) *f=free_func; | 
|  | } | 
|  |  | 
|  | void CRYPTO_get_locked_mem_functions(char *(**m)(), void (**f)()) | 
|  | { | 
|  | if (m != NULL) *m=malloc_locked_func; | 
|  | if (f != NULL) *f=free_locked_func; | 
|  | } | 
|  |  | 
|  | void *CRYPTO_malloc_locked(int num) | 
|  | { | 
|  | return(malloc_locked_func(num)); | 
|  | } | 
|  |  | 
|  | void CRYPTO_free_locked(void *str) | 
|  | { | 
|  | free_locked_func(str); | 
|  | } | 
|  |  | 
|  | void *CRYPTO_malloc(int num) | 
|  | { | 
|  | return(malloc_func(num)); | 
|  | } | 
|  |  | 
|  | void *CRYPTO_realloc(void *str, int num) | 
|  | { | 
|  | return(realloc_func(str,num)); | 
|  | } | 
|  |  | 
|  | void CRYPTO_free(void *str) | 
|  | { | 
|  | free_func(str); | 
|  | } | 
|  |  | 
|  | static unsigned long break_order_num=0; | 
|  | void *CRYPTO_dbg_malloc(int num, const char *file, int line) | 
|  | { | 
|  | char *ret; | 
|  | MEM *m,*mm; | 
|  |  | 
|  | if ((ret=malloc_func(num)) == NULL) | 
|  | return(NULL); | 
|  |  | 
|  | if (mh_mode & CRYPTO_MEM_CHECK_ENABLE) | 
|  | { | 
|  | MemCheck_off(); | 
|  | if ((m=(MEM *)Malloc(sizeof(MEM))) == NULL) | 
|  | { | 
|  | Free(ret); | 
|  | MemCheck_on(); | 
|  | return(NULL); | 
|  | } | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | if (mh == NULL) | 
|  | { | 
|  | if ((mh=lh_new(mem_hash,mem_cmp)) == NULL) | 
|  | { | 
|  | Free(ret); | 
|  | Free(m); | 
|  | ret=NULL; | 
|  | goto err; | 
|  | } | 
|  | } | 
|  |  | 
|  | m->addr=ret; | 
|  | m->file=file; | 
|  | m->line=line; | 
|  | m->num=num; | 
|  | if (order == break_order_num) | 
|  | { | 
|  | /* BREAK HERE */ | 
|  | m->order=order; | 
|  | } | 
|  | m->order=order++; | 
|  | if ((mm=(MEM *)lh_insert(mh,(char *)m)) != NULL) | 
|  | { | 
|  | /* Not good, but don't sweat it */ | 
|  | Free(mm); | 
|  | } | 
|  | err: | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | MemCheck_on(); | 
|  | } | 
|  | return(ret); | 
|  | } | 
|  |  | 
|  | void CRYPTO_dbg_free(void *addr) | 
|  | { | 
|  | MEM m,*mp; | 
|  |  | 
|  | if ((mh_mode & CRYPTO_MEM_CHECK_ENABLE) && (mh != NULL)) | 
|  | { | 
|  | MemCheck_off(); | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | m.addr=addr; | 
|  | mp=(MEM *)lh_delete(mh,(char *)&m); | 
|  | if (mp != NULL) | 
|  | Free(mp); | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | MemCheck_on(); | 
|  | } | 
|  | free_func(addr); | 
|  | } | 
|  |  | 
|  | void *CRYPTO_dbg_realloc(void *addr, int num, const char *file, int line) | 
|  | { | 
|  | char *ret; | 
|  | MEM m,*mp; | 
|  |  | 
|  | ret=realloc_func(addr,num); | 
|  | if (ret == addr) return(ret); | 
|  |  | 
|  | if (mh_mode & CRYPTO_MEM_CHECK_ENABLE) | 
|  | { | 
|  | MemCheck_off(); | 
|  | if (ret == NULL) return(NULL); | 
|  | m.addr=addr; | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | mp=(MEM *)lh_delete(mh,(char *)&m); | 
|  | if (mp != NULL) | 
|  | { | 
|  | mp->addr=ret; | 
|  | lh_insert(mh,(char *)mp); | 
|  | } | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | MemCheck_on(); | 
|  | } | 
|  | return(ret); | 
|  | } | 
|  |  | 
|  | void *CRYPTO_remalloc(void *a, int n) | 
|  | { | 
|  | if (a != NULL) Free(a); | 
|  | a=(char *)Malloc(n); | 
|  | return(a); | 
|  | } | 
|  |  | 
|  | void *CRYPTO_dbg_remalloc(void *a, int n, const char *file, int line) | 
|  | { | 
|  | if (a != NULL) CRYPTO_dbg_free(a); | 
|  | a=(char *)CRYPTO_dbg_malloc(n,file,line); | 
|  | return(a); | 
|  | } | 
|  |  | 
|  |  | 
|  | typedef struct mem_leak_st | 
|  | { | 
|  | BIO *bio; | 
|  | int chunks; | 
|  | long bytes; | 
|  | } MEM_LEAK; | 
|  |  | 
|  | static void print_leak(MEM *m, MEM_LEAK *l) | 
|  | { | 
|  | char buf[128]; | 
|  |  | 
|  | if(m->addr == (char *)l->bio) | 
|  | return; | 
|  | sprintf(buf,"%5ld file=%s, line=%d, number=%d, address=%08lX\n", | 
|  | m->order,m->file,m->line,m->num,(long)m->addr); | 
|  | BIO_puts(l->bio,buf); | 
|  | l->chunks++; | 
|  | l->bytes+=m->num; | 
|  | } | 
|  |  | 
|  | void CRYPTO_mem_leaks(BIO *b) | 
|  | { | 
|  | MEM_LEAK ml; | 
|  | char buf[80]; | 
|  |  | 
|  | if (mh == NULL) return; | 
|  | ml.bio=b; | 
|  | ml.bytes=0; | 
|  | ml.chunks=0; | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | lh_doall_arg(mh,(void (*)())print_leak,(char *)&ml); | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | if (ml.chunks != 0) | 
|  | { | 
|  | sprintf(buf,"%ld bytes leaked in %d chunks\n", | 
|  | ml.bytes,ml.chunks); | 
|  | BIO_puts(b,buf); | 
|  | } | 
|  |  | 
|  | #if 0 | 
|  | lh_stats_bio(mh,b); | 
|  | lh_node_stats_bio(mh,b); | 
|  | lh_node_usage_stats_bio(mh,b); | 
|  | #endif | 
|  | } | 
|  |  | 
|  | static void (*mem_cb)()=NULL; | 
|  |  | 
|  | static void cb_leak(MEM *m, char *cb) | 
|  | { | 
|  | void (*mem_callback)()=(void (*)())cb; | 
|  | mem_callback(m->order,m->file,m->line,m->num,m->addr); | 
|  | } | 
|  |  | 
|  | void CRYPTO_mem_leaks_cb(void (*cb)()) | 
|  | { | 
|  | if (mh == NULL) return; | 
|  | CRYPTO_w_lock(CRYPTO_LOCK_MALLOC); | 
|  | mem_cb=cb; | 
|  | lh_doall_arg(mh,(void (*)())cb_leak,(char *)mem_cb); | 
|  | mem_cb=NULL; | 
|  | CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC); | 
|  | } | 
|  |  | 
|  | #ifndef NO_FP_API | 
|  | void CRYPTO_mem_leaks_fp(FILE *fp) | 
|  | { | 
|  | BIO *b; | 
|  |  | 
|  | if (mh == NULL) return; | 
|  | if ((b=BIO_new(BIO_s_file())) == NULL) | 
|  | return; | 
|  | BIO_set_fp(b,fp,BIO_NOCLOSE); | 
|  | CRYPTO_mem_leaks(b); | 
|  | BIO_free(b); | 
|  | } | 
|  | #endif | 
|  |  |