mem_dbg.c 19.6 KB
Newer Older
1
/* crypto/mem_dbg.c */
R
Richard Levitte 已提交
2 3
/* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
 * All rights reserved.
4
 *
R
Richard Levitte 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
 * 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.
 * 
23 24 25
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
R
Richard Levitte 已提交
26 27
 * 1. Redistributions of source code must retain the copyright
 *    notice, this list of conditions and the following disclaimer.
28
 * 2. Redistributions in binary form must reproduce the above copyright
R
Richard Levitte 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
 *    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.]
57 58 59 60 61 62 63 64 65 66 67
 */

#include <stdio.h>
#include <stdlib.h>
#include <time.h>	
#include <openssl/crypto.h>
#include <openssl/buffer.h>
#include <openssl/bio.h>
#include <openssl/lhash.h>
#include "cryptlib.h"

68 69 70 71 72 73
static int mh_mode=CRYPTO_MEM_CHECK_OFF;
/* The state changes to CRYPTO_MEM_CHECK_ON | CRYPTO_MEM_CHECK_ENABLE
 * when the application asks for it (usually after library initialisation
 * for which no book-keeping is desired).
 *
 * State CRYPTO_MEM_CHECK_ON exists only temporarily when the library
74 75 76 77 78 79 80 81 82
 * thinks that certain allocations should not be checked (e.g. the data
 * structures used for memory checking).  It is not suitable as an initial
 * state: the library will unexpectedly enable memory checking when it
 * executes one of those sections that want to disable checking
 * temporarily.
 *
 * State CRYPTO_MEM_CHECK_ENABLE without ..._ON makes no sense whatsoever.
 */

83
static unsigned long order = 0; /* number of memory requests */
84 85
static LHASH *mh=NULL; /* hash-table of memory requests (address as key);
                        * access requires MALLOC2 lock */
86 87 88


typedef struct app_mem_info_st
89 90 91
/* For application-defined information (static C-string `info')
 * to be displayed in memory leak list.
 * Each thread has its own stack.  For applications, there is
B
Rename  
Bodo Möller 已提交
92 93 94
 *   CRYPTO_push_info("...")     to push an entry,
 *   CRYPTO_pop_info()           to pop an entry,
 *   CRYPTO_remove_all_info()    to pop all entries.
95
 */
96 97 98 99 100
	{	
	unsigned long thread;
	const char *file;
	int line;
	const char *info;
101
	struct app_mem_info_st *next; /* tail of thread's stack */
102 103 104
	int references;
	} APP_INFO;

105 106
static void app_info_free(APP_INFO *);

107 108
static LHASH *amih=NULL; /* hash-table with those app_mem_info_st's
                          * that are at the top of their thread's stack
109 110
                          * (with `thread' as key);
                          * access requires MALLOC2 lock */
111 112

typedef struct mem_st
113
/* memory-block description */
114
	{
115
	void *addr;
116 117 118 119 120 121 122 123 124
	int num;
	const char *file;
	int line;
	unsigned long thread;
	unsigned long order;
	time_t time;
	APP_INFO *app_info;
	} MEM;

125 126 127
static long options =             /* extra information to be recorded */
#if defined(CRYPTO_MDEBUG_TIME) || defined(CRYPTO_MDEBUG_ALL)
	V_CRYPTO_MDEBUG_TIME |
128
#endif
129 130
#if defined(CRYPTO_MDEBUG_THREAD) || defined(CRYPTO_MDEBUG_ALL)
	V_CRYPTO_MDEBUG_THREAD |
131
#endif
132
	0;
133

134

135 136 137 138 139 140 141 142 143
static unsigned int num_disable = 0; /* num_disable > 0
                                      *     iff
                                      * mh_mode == CRYPTO_MEM_CHECK_ON (w/o ..._ENABLE)
                                      */
static unsigned long disabling_thread = 0; /* Valid iff num_disable > 0.
                                            * CRYPTO_LOCK_MALLOC2 is locked
                                            * exactly in this case (by the
                                            * thread named in disabling_thread).
                                            */
144

145 146 147 148 149 150 151 152 153 154 155 156
static void app_info_free(APP_INFO *inf)
	{
	if (--(inf->references) <= 0)
		{
		if (inf->next != NULL)
			{
			app_info_free(inf->next);
			}
		OPENSSL_free(inf);
		}
	}

157 158 159 160 161 162 163
int CRYPTO_mem_ctrl(int mode)
	{
	int ret=mh_mode;

	CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
	switch (mode)
		{
164 165
	/* for applications (not to be called while multiple threads
	 * use the library): */
166 167
	case CRYPTO_MEM_CHECK_ON: /* aka MemCheck_start() */
		mh_mode = CRYPTO_MEM_CHECK_ON|CRYPTO_MEM_CHECK_ENABLE;
168
		num_disable = 0;
169 170 171
		break;
	case CRYPTO_MEM_CHECK_OFF: /* aka MemCheck_stop() */
		mh_mode = 0;
172 173
		num_disable = 0; /* should be true *before* MemCheck_stop is used,
		                    or there'll be a lot of confusion */
174 175 176 177 178 179
		break;

	/* switch off temporarily (for library-internal use): */
	case CRYPTO_MEM_CHECK_DISABLE: /* aka MemCheck_off() */
		if (mh_mode & CRYPTO_MEM_CHECK_ON)
			{
180
			if (!num_disable || (disabling_thread != CRYPTO_thread_id())) /* otherwise we already have the MALLOC2 lock */
181 182 183
				{
				/* Long-time lock CRYPTO_LOCK_MALLOC2 must not be claimed while
				 * we're holding CRYPTO_LOCK_MALLOC, or we'll deadlock if
184 185
				 * somebody else holds CRYPTO_LOCK_MALLOC2 (and cannot release
				 * it because we block entry to this function).
186 187 188 189 190 191 192 193 194 195 196
				 * Give them a chance, first, and then claim the locks in
				 * appropriate order (long-time lock first).
				 */
				CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
				/* Note that after we have waited for CRYPTO_LOCK_MALLOC2
				 * and CRYPTO_LOCK_MALLOC, we'll still be in the right
				 * "case" and "if" branch because MemCheck_start and
				 * MemCheck_stop may never be used while there are multiple
				 * OpenSSL threads. */
				CRYPTO_w_lock(CRYPTO_LOCK_MALLOC2);
				CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
197
				mh_mode &= ~CRYPTO_MEM_CHECK_ENABLE;
198 199
				disabling_thread=CRYPTO_thread_id();
				}
200
			num_disable++;
201 202 203 204 205
			}
		break;
	case CRYPTO_MEM_CHECK_ENABLE: /* aka MemCheck_on() */
		if (mh_mode & CRYPTO_MEM_CHECK_ON)
			{
206
			if (num_disable) /* always true, or something is going wrong */
207
				{
208 209 210 211 212 213
				num_disable--;
				if (num_disable == 0)
					{
					mh_mode|=CRYPTO_MEM_CHECK_ENABLE;
					CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC2);
					}
214 215 216 217 218 219 220 221 222 223 224
				}
			}
		break;

	default:
		break;
		}
	CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
	return(ret);
	}

225
int CRYPTO_is_mem_check_on(void)
226 227 228 229 230
	{
	int ret = 0;

	if (mh_mode & CRYPTO_MEM_CHECK_ON)
		{
231
		CRYPTO_r_lock(CRYPTO_LOCK_MALLOC);
232 233

		ret = (mh_mode & CRYPTO_MEM_CHECK_ENABLE)
234
			|| (disabling_thread != CRYPTO_thread_id());
235

236
		CRYPTO_r_unlock(CRYPTO_LOCK_MALLOC);
237 238 239 240 241
		}
	return(ret);
	}	


242
void CRYPTO_dbg_set_options(long bits)
243 244 245 246
	{
	options = bits;
	}

247
long CRYPTO_dbg_get_options(void)
248 249 250 251
	{
	return options;
	}

252
/* static int mem_cmp(MEM *a, MEM *b) */
253
static int mem_cmp(const void *a_void, const void *b_void)
254
	{
B
Ben Laurie 已提交
255 256
	return((const char *)((const MEM *)a_void)->addr
		- (const char *)((const MEM *)b_void)->addr);
257 258
	}

259
/* static unsigned long mem_hash(MEM *a) */
260
static unsigned long mem_hash(const void *a_void)
261 262 263
	{
	unsigned long ret;

264
	ret=(unsigned long)((const MEM *)a_void)->addr;
265 266 267 268 269

	ret=ret*17851+(ret>>14)*7+(ret>>4)*251;
	return(ret);
	}

270
/* static int app_info_cmp(APP_INFO *a, APP_INFO *b) */
271
static int app_info_cmp(const void *a_void, const void *b_void)
272
	{
273 274
	return(((const APP_INFO *)a_void)->thread
		!= ((const APP_INFO *)b_void)->thread);
275 276
	}

277
/* static unsigned long app_info_hash(APP_INFO *a) */
278
static unsigned long app_info_hash(const void *a_void)
279 280 281
	{
	unsigned long ret;

282
	ret=(unsigned long)((const APP_INFO *)a_void)->thread;
283 284 285 286 287

	ret=ret*17851+(ret>>14)*7+(ret>>4)*251;
	return(ret);
	}

U
Ulf Möller 已提交
288
static APP_INFO *pop_info(void)
289 290 291 292 293 294 295
	{
	APP_INFO tmp;
	APP_INFO *ret = NULL;

	if (amih != NULL)
		{
		tmp.thread=CRYPTO_thread_id();
296
		if ((ret=(APP_INFO *)lh_delete(amih,&tmp)) != NULL)
297 298 299 300 301 302 303 304
			{
			APP_INFO *next=ret->next;

			if (next != NULL)
				{
				next->references++;
				lh_insert(amih,(char *)next);
				}
305
#ifdef LEVITTE_DEBUG_MEM
306 307
			if (ret->thread != tmp.thread)
				{
B
Rename  
Bodo Möller 已提交
308
				fprintf(stderr, "pop_info(): deleted info has other thread ID (%lu) than the current thread (%lu)!!!!\n",
309 310 311 312 313 314 315 316 317
					ret->thread, tmp.thread);
				abort();
				}
#endif
			if (--(ret->references) <= 0)
				{
				ret->next = NULL;
				if (next != NULL)
					next->references--;
318
				OPENSSL_free(ret);
319 320 321 322 323 324
				}
			}
		}
	return(ret);
	}

B
Rename  
Bodo Möller 已提交
325
int CRYPTO_push_info_(const char *info, const char *file, int line)
326 327 328 329 330 331
	{
	APP_INFO *ami, *amim;
	int ret=0;

	if (is_MemCheck_on())
		{
332
		MemCheck_off(); /* obtain MALLOC2 lock */
333

334
		if ((ami = (APP_INFO *)OPENSSL_malloc(sizeof(APP_INFO))) == NULL)
335 336 337 338 339 340
			{
			ret=0;
			goto err;
			}
		if (amih == NULL)
			{
341
			if ((amih=lh_new(app_info_hash, app_info_cmp)) == NULL)
342
				{
343
				OPENSSL_free(ami);
344 345 346 347 348 349 350 351 352 353 354 355 356 357
				ret=0;
				goto err;
				}
			}

		ami->thread=CRYPTO_thread_id();
		ami->file=file;
		ami->line=line;
		ami->info=info;
		ami->references=1;
		ami->next=NULL;

		if ((amim=(APP_INFO *)lh_insert(amih,(char *)ami)) != NULL)
			{
358
#ifdef LEVITTE_DEBUG_MEM
359 360
			if (ami->thread != amim->thread)
				{
B
Rename  
Bodo Möller 已提交
361
				fprintf(stderr, "CRYPTO_push_info(): previous info has other thread ID (%lu) than the current thread (%lu)!!!!\n",
362 363 364 365 366 367 368
					amim->thread, ami->thread);
				abort();
				}
#endif
			ami->next=amim;
			}
 err:
369
		MemCheck_on(); /* release MALLOC2 lock */
370 371 372 373 374
		}

	return(ret);
	}

B
Rename  
Bodo Möller 已提交
375
int CRYPTO_pop_info(void)
376 377 378
	{
	int ret=0;

379
	if (is_MemCheck_on()) /* _must_ be true, or something went severely wrong */
380
		{
381
		MemCheck_off(); /* obtain MALLOC2 lock */
382

B
Rename  
Bodo Möller 已提交
383
		ret=(pop_info() != NULL);
384

385
		MemCheck_on(); /* release MALLOC2 lock */
386 387 388 389 390 391 392 393
		}
	return(ret);
	}

int CRYPTO_remove_all_info(void)
	{
	int ret=0;

394
	if (is_MemCheck_on()) /* _must_ be true */
395
		{
396
		MemCheck_off(); /* obtain MALLOC2 lock */
397

B
Rename  
Bodo Möller 已提交
398
		while(pop_info() != NULL)
399 400
			ret++;

401
		MemCheck_on(); /* release MALLOC2 lock */
402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423
		}
	return(ret);
	}


static unsigned long break_order_num=0;
void CRYPTO_dbg_malloc(void *addr, int num, const char *file, int line,
	int before_p)
	{
	MEM *m,*mm;
	APP_INFO tmp,*amim;

	switch(before_p & 127)
		{
	case 0:
		break;
	case 1:
		if (addr == NULL)
			break;

		if (is_MemCheck_on())
			{
424
			MemCheck_off(); /* make sure we hold MALLOC2 lock */
425
			if ((m=(MEM *)OPENSSL_malloc(sizeof(MEM))) == NULL)
426
				{
427
				OPENSSL_free(addr);
428 429
				MemCheck_on(); /* release MALLOC2 lock
				                * if num_disabled drops to 0 */
430 431 432 433
				return;
				}
			if (mh == NULL)
				{
434
				if ((mh=lh_new(mem_hash, mem_cmp)) == NULL)
435
					{
436 437
					OPENSSL_free(addr);
					OPENSSL_free(m);
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457
					addr=NULL;
					goto err;
					}
				}

			m->addr=addr;
			m->file=file;
			m->line=line;
			m->num=num;
			if (options & V_CRYPTO_MDEBUG_THREAD)
				m->thread=CRYPTO_thread_id();
			else
				m->thread=0;

			if (order == break_order_num)
				{
				/* BREAK HERE */
				m->order=order;
				}
			m->order=order++;
458 459
#ifdef LEVITTE_DEBUG_MEM
			fprintf(stderr, "LEVITTE_DEBUG_MEM: [%5d] %c 0x%p (%d)\n",
460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484
				m->order,
				(before_p & 128) ? '*' : '+',
				m->addr, m->num);
#endif
			if (options & V_CRYPTO_MDEBUG_TIME)
				m->time=time(NULL);
			else
				m->time=0;

			tmp.thread=CRYPTO_thread_id();
			m->app_info=NULL;
			if (amih != NULL
				&& (amim=(APP_INFO *)lh_retrieve(amih,(char *)&tmp)) != NULL)
				{
				m->app_info = amim;
				amim->references++;
				}

			if ((mm=(MEM *)lh_insert(mh,(char *)m)) != NULL)
				{
				/* Not good, but don't sweat it */
				if (mm->app_info != NULL)
					{
					mm->app_info->references--;
					}
485
				OPENSSL_free(mm);
486 487
				}
		err:
488 489
			MemCheck_on(); /* release MALLOC2 lock
			                * if num_disabled drops to 0 */
490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507
			}
		break;
		}
	return;
	}

void CRYPTO_dbg_free(void *addr, int before_p)
	{
	MEM m,*mp;

	switch(before_p)
		{
	case 0:
		if (addr == NULL)
			break;

		if (is_MemCheck_on() && (mh != NULL))
			{
508
			MemCheck_off(); /* make sure we hold MALLOC2 lock */
509 510 511 512 513

			m.addr=addr;
			mp=(MEM *)lh_delete(mh,(char *)&m);
			if (mp != NULL)
				{
514 515
#ifdef LEVITTE_DEBUG_MEM
			fprintf(stderr, "LEVITTE_DEBUG_MEM: [%5d] - 0x%p (%d)\n",
516 517 518
				mp->order, mp->addr, mp->num);
#endif
				if (mp->app_info != NULL)
519
					app_info_free(mp->app_info);
520
				OPENSSL_free(mp);
521 522
				}

523 524
			MemCheck_on(); /* release MALLOC2 lock
			                * if num_disabled drops to 0 */
525 526 527 528 529 530 531 532 533 534 535 536
			}
		break;
	case 1:
		break;
		}
	}

void CRYPTO_dbg_realloc(void *addr1, void *addr2, int num,
	const char *file, int line, int before_p)
	{
	MEM m,*mp;

537 538
#ifdef LEVITTE_DEBUG_MEM
	fprintf(stderr, "LEVITTE_DEBUG_MEM: --> CRYPTO_dbg_malloc(addr1 = %p, addr2 = %p, num = %d, file = \"%s\", line = %d, before_p = %d)\n",
539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
		addr1, addr2, num, file, line, before_p);
#endif

	switch(before_p)
		{
	case 0:
		break;
	case 1:
		if (addr2 == NULL)
			break;

		if (addr1 == NULL)
			{
			CRYPTO_dbg_malloc(addr2, num, file, line, 128 | before_p);
			break;
			}

		if (is_MemCheck_on())
			{
558
			MemCheck_off(); /* make sure we hold MALLOC2 lock */
559 560 561 562 563

			m.addr=addr1;
			mp=(MEM *)lh_delete(mh,(char *)&m);
			if (mp != NULL)
				{
564 565
#ifdef LEVITTE_DEBUG_MEM
				fprintf(stderr, "LEVITTE_DEBUG_MEM: [%5d] * 0x%p (%d) -> 0x%p (%d)\n",
566 567 568 569 570 571 572 573 574
					mp->order,
					mp->addr, mp->num,
					addr2, num);
#endif
				mp->addr=addr2;
				mp->num=num;
				lh_insert(mh,(char *)mp);
				}

575 576
			MemCheck_on(); /* release MALLOC2 lock
			                * if num_disabled drops to 0 */
577 578 579 580 581 582 583 584 585 586 587 588 589 590
			}
		break;
		}
	return;
	}


typedef struct mem_leak_st
	{
	BIO *bio;
	int chunks;
	long bytes;
	} MEM_LEAK;

B
Ben Laurie 已提交
591
static void print_leak(const MEM *m, MEM_LEAK *l)
592 593 594 595 596 597 598 599
	{
	char buf[1024];
	char *bufp = buf;
	APP_INFO *amip;
	int ami_cnt;
	struct tm *lcl = NULL;
	unsigned long ti;

600 601
#define BUF_REMAIN (sizeof buf - (size_t)(bufp - buf))

602 603 604 605 606 607 608
	if(m->addr == (char *)l->bio)
	    return;

	if (options & V_CRYPTO_MDEBUG_TIME)
		{
		lcl = localtime(&m->time);
	
609
		BIO_snprintf(bufp, BUF_REMAIN, "[%02d:%02d:%02d] ",
610 611 612 613
			lcl->tm_hour,lcl->tm_min,lcl->tm_sec);
		bufp += strlen(bufp);
		}

614
	BIO_snprintf(bufp, BUF_REMAIN, "%5lu file=%s, line=%d, ",
615 616 617 618 619
		m->order,m->file,m->line);
	bufp += strlen(bufp);

	if (options & V_CRYPTO_MDEBUG_THREAD)
		{
620
		BIO_snprintf(bufp, BUF_REMAIN, "thread=%lu, ", m->thread);
621 622 623
		bufp += strlen(bufp);
		}

624
	BIO_snprintf(bufp, BUF_REMAIN, "number=%d, address=%08lX\n",
625 626 627 628 629 630 631 632 633 634
		m->num,(unsigned long)m->addr);
	bufp += strlen(bufp);

	BIO_puts(l->bio,buf);
	
	l->chunks++;
	l->bytes+=m->num;

	amip=m->app_info;
	ami_cnt=0;
635 636 637 638 639
	if (!amip)
		return;
	ti=amip->thread;
	
	do
640 641 642 643 644 645
		{
		int buf_len;
		int info_len;

		ami_cnt++;
		memset(buf,'>',ami_cnt);
646
		BIO_snprintf(buf + ami_cnt, sizeof buf - ami_cnt,
647
			" thread=%lu, file=%s, line=%d, info=\"",
648 649 650 651 652 653 654 655 656 657
			amip->thread, amip->file, amip->line);
		buf_len=strlen(buf);
		info_len=strlen(amip->info);
		if (128 - buf_len - 3 < info_len)
			{
			memcpy(buf + buf_len, amip->info, 128 - buf_len - 3);
			buf_len = 128 - 3;
			}
		else
			{
658 659
			BUF_strlcpy(buf + buf_len, amip->info,
				    sizeof buf - buf_len);
660 661
			buf_len = strlen(buf);
			}
662
		BIO_snprintf(buf + buf_len, sizeof buf - buf_len, "\"\n");
663 664 665 666 667
		
		BIO_puts(l->bio,buf);

		amip = amip->next;
		}
668 669
	while(amip && amip->thread == ti);
		
670
#ifdef LEVITTE_DEBUG_MEM
671 672
	if (amip)
		{
R
Typo...  
Richard Levitte 已提交
673
		fprintf(stderr, "Thread switch detected in backtrace!!!!\n");
674 675 676 677 678
		abort();
		}
#endif
	}

B
Ben Laurie 已提交
679
static IMPLEMENT_LHASH_DOALL_ARG_FN(print_leak, const MEM *, MEM_LEAK *)
680

681 682 683 684
void CRYPTO_mem_leaks(BIO *b)
	{
	MEM_LEAK ml;

685 686
	if (mh == NULL && amih == NULL)
		return;
687 688 689

	MemCheck_off(); /* obtain MALLOC2 lock */

690 691 692
	ml.bio=b;
	ml.bytes=0;
	ml.chunks=0;
693
	if (mh != NULL)
694
		lh_doall_arg(mh, LHASH_DOALL_ARG_FN(print_leak),
695
				(char *)&ml);
696 697
	if (ml.chunks != 0)
		{
698 699
		BIO_printf(b,"%ld bytes leaked in %d chunks\n",
			   ml.bytes,ml.chunks);
700
		}
701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717
	else
		{
		/* Make sure that, if we found no leaks, memory-leak debugging itself
		 * does not introduce memory leaks (which might irritate
		 * external debugging tools).
		 * (When someone enables leak checking, but does not call
		 * this function, we declare it to be their fault.)
		 *
		 * XXX    This should be in CRYPTO_mem_leaks_cb,
		 * and CRYPTO_mem_leaks should be implemented by
		 * using CRYPTO_mem_leaks_cb.
		 * (Also their should be a variant of lh_doall_arg
		 * that takes a function pointer instead of a void *;
		 * this would obviate the ugly and illegal
		 * void_fn_to_char kludge in CRYPTO_mem_leaks_cb.
		 * Otherwise the code police will come and get us.)
		 */
B
Bodo Möller 已提交
718 719
		int old_mh_mode;

720
		CRYPTO_w_lock(CRYPTO_LOCK_MALLOC);
B
Bodo Möller 已提交
721 722 723 724 725 726

		/* avoid deadlock when lh_free() uses CRYPTO_dbg_free(),
		 * which uses CRYPTO_is_mem_check_on */
		old_mh_mode = mh_mode;
		mh_mode = CRYPTO_MEM_CHECK_OFF;

727 728 729 730 731 732 733 734 735 736 737 738 739
		if (mh != NULL)
			{
			lh_free(mh);
			mh = NULL;
			}
		if (amih != NULL)
			{
			if (lh_num_items(amih) == 0) 
				{
				lh_free(amih);
				amih = NULL;
				}
			}
B
Bodo Möller 已提交
740 741

		mh_mode = old_mh_mode;
742 743
		CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC);
		}
744
	MemCheck_on(); /* release MALLOC2 lock */
745 746
	}

747
#ifndef OPENSSL_NO_FP_API
748 749 750 751 752
void CRYPTO_mem_leaks_fp(FILE *fp)
	{
	BIO *b;

	if (mh == NULL) return;
753 754 755 756
	/* Need to turn off memory checking when allocated BIOs ... especially
	 * as we're creating them at a time when we're trying to check we've not
	 * left anything un-free()'d!! */
	MemCheck_off();
757
	b = BIO_new(BIO_s_file());
758
	MemCheck_on();
759
	if(!b) return;
760 761 762 763 764 765
	BIO_set_fp(b,fp,BIO_NOCLOSE);
	CRYPTO_mem_leaks(b);
	BIO_free(b);
	}
#endif

766 767 768 769 770


/* FIXME: We really don't allow much to the callback.  For example, it has
   no chance of reaching the info stack for the item it processes.  Should
   it really be this way?  -- Richard Levitte */
771 772 773
/* NB: The prototypes have been typedef'd to CRYPTO_MEM_LEAK_CB inside crypto.h
 * If this code is restructured, remove the callback type if it is no longer
 * needed. -- Geoff Thorpe */
B
Ben Laurie 已提交
774
static void cb_leak(const MEM *m, CRYPTO_MEM_LEAK_CB **cb)
775 776 777 778
	{
	(**cb)(m->order,m->file,m->line,m->num,m->addr);
	}

B
Ben Laurie 已提交
779
static IMPLEMENT_LHASH_DOALL_ARG_FN(cb_leak, const MEM *, CRYPTO_MEM_LEAK_CB **)
780

781
void CRYPTO_mem_leaks_cb(CRYPTO_MEM_LEAK_CB *cb)
782 783 784
	{
	if (mh == NULL) return;
	CRYPTO_w_lock(CRYPTO_LOCK_MALLOC2);
785
	lh_doall_arg(mh, LHASH_DOALL_ARG_FN(cb_leak), &cb);
786 787
	CRYPTO_w_unlock(CRYPTO_LOCK_MALLOC2);
	}