testing: Use curl instead of soup plugin in libipsec/rw-suite-b scenario
[strongswan.git] / src / libstrongswan / utils / leak_detective.c
1 /*
2 * Copyright (C) 2013-2014 Tobias Brunner
3 * Copyright (C) 2006-2013 Martin Willi
4 * Hochschule fuer Technik Rapperswil
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; either version 2 of the License, or (at your
9 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
10 *
11 * This program is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * for more details.
15 */
16
17 #define _GNU_SOURCE
18 #include <stddef.h>
19 #include <string.h>
20 #include <stdio.h>
21 #include <signal.h>
22 #include <unistd.h>
23 #include <locale.h>
24 #ifdef HAVE_DLADDR
25 #include <dlfcn.h>
26 #endif
27 #include <time.h>
28 #include <errno.h>
29
30 #ifdef __APPLE__
31 #include <sys/mman.h>
32 #include <malloc/malloc.h>
33 /* overload some of our types clashing with mach */
34 #define host_t strongswan_host_t
35 #define processor_t strongswan_processor_t
36 #define thread_t strongswan_thread_t
37 #endif /* __APPLE__ */
38
39 #include "leak_detective.h"
40
41 #include <library.h>
42 #include <utils/utils.h>
43 #include <utils/debug.h>
44 #include <utils/backtrace.h>
45 #include <collections/hashtable.h>
46 #include <threading/thread_value.h>
47 #include <threading/spinlock.h>
48
49 typedef struct private_leak_detective_t private_leak_detective_t;
50
51 /**
52 * private data of leak_detective
53 */
54 struct private_leak_detective_t {
55
56 /**
57 * public functions
58 */
59 leak_detective_t public;
60
61 /**
62 * Registered report() function
63 */
64 leak_detective_report_cb_t report_cb;
65
66 /**
67 * Registered report() summary function
68 */
69 leak_detective_summary_cb_t report_scb;
70
71 /**
72 * Registered user data for callbacks
73 */
74 void *report_data;
75 };
76
77 /**
78 * Magic value which helps to detect memory corruption. Yummy!
79 */
80 #define MEMORY_HEADER_MAGIC 0x7ac0be11
81
82 /**
83 * Magic written to tail of allocation
84 */
85 #define MEMORY_TAIL_MAGIC 0xcafebabe
86
87 /**
88 * Pattern which is filled in memory before freeing it
89 */
90 #define MEMORY_FREE_PATTERN 0xFF
91
92 /**
93 * Pattern which is filled in newly allocated memory
94 */
95 #define MEMORY_ALLOC_PATTERN 0xEE
96
97 typedef struct memory_header_t memory_header_t;
98 typedef struct memory_tail_t memory_tail_t;
99
100 /**
101 * Header which is prepended to each allocated memory block
102 */
103 struct memory_header_t {
104
105 /**
106 * Pointer to previous entry in linked list
107 */
108 memory_header_t *previous;
109
110 /**
111 * Pointer to next entry in linked list
112 */
113 memory_header_t *next;
114
115 /**
116 * backtrace taken during (re-)allocation
117 */
118 backtrace_t *backtrace;
119
120 /**
121 * Padding to make sizeof(memory_header_t) == 32
122 */
123 uint32_t padding[sizeof(void*) == sizeof(uint32_t) ? 3 : 0];
124
125 /**
126 * Number of bytes following after the header
127 */
128 uint32_t bytes;
129
130 /**
131 * magic bytes to detect bad free or heap underflow, MEMORY_HEADER_MAGIC
132 */
133 uint32_t magic;
134
135 }__attribute__((__packed__));
136
137 /**
138 * tail appended to each allocated memory block
139 */
140 struct memory_tail_t {
141
142 /**
143 * Magic bytes to detect heap overflow, MEMORY_TAIL_MAGIC
144 */
145 uint32_t magic;
146
147 }__attribute__((__packed__));
148
149 /**
150 * first mem header is just a dummy to chain
151 * the others on it...
152 */
153 static memory_header_t first_header = {
154 .magic = MEMORY_HEADER_MAGIC,
155 };
156
157 /**
158 * Spinlock to access header linked list
159 */
160 static spinlock_t *lock;
161
162 /**
163 * Is leak detection currently enabled?
164 */
165 static bool enabled = FALSE;
166
167 /**
168 * Is leak detection disabled for the current thread?
169 */
170 static thread_value_t *thread_disabled;
171
172 /**
173 * Installs the malloc hooks, enables leak detection
174 */
175 static void enable_leak_detective()
176 {
177 enabled = TRUE;
178 }
179
180 /**
181 * Uninstalls the malloc hooks, disables leak detection
182 */
183 static void disable_leak_detective()
184 {
185 enabled = FALSE;
186 }
187
188 /**
189 * Enable/Disable leak detective for the current thread
190 *
191 * @return Previous value
192 */
193 static bool enable_thread(bool enable)
194 {
195 bool before;
196
197 before = thread_disabled->get(thread_disabled) == NULL;
198 thread_disabled->set(thread_disabled, enable ? NULL : (void*)TRUE);
199 return before;
200 }
201
202 /**
203 * Add a header to the beginning of the list
204 */
205 static void add_hdr(memory_header_t *hdr)
206 {
207 lock->lock(lock);
208 hdr->next = first_header.next;
209 if (hdr->next)
210 {
211 hdr->next->previous = hdr;
212 }
213 hdr->previous = &first_header;
214 first_header.next = hdr;
215 lock->unlock(lock);
216 }
217
218 /**
219 * Remove a header from the list
220 */
221 static void remove_hdr(memory_header_t *hdr)
222 {
223 lock->lock(lock);
224 if (hdr->next)
225 {
226 hdr->next->previous = hdr->previous;
227 }
228 hdr->previous->next = hdr->next;
229 lock->unlock(lock);
230 }
231
232 /**
233 * Check if a header is in the list
234 */
235 static bool has_hdr(memory_header_t *hdr)
236 {
237 memory_header_t *current;
238 bool found = FALSE;
239
240 lock->lock(lock);
241 for (current = &first_header; current != NULL; current = current->next)
242 {
243 if (current == hdr)
244 {
245 found = TRUE;
246 break;
247 }
248 }
249 lock->unlock(lock);
250
251 return found;
252 }
253
254 #ifdef __APPLE__
255
256 /**
257 * Copy of original default zone, with functions we call in hooks
258 */
259 static malloc_zone_t original;
260
261 /**
262 * Call original malloc()
263 */
264 static void* real_malloc(size_t size)
265 {
266 return original.malloc(malloc_default_zone(), size);
267 }
268
269 /**
270 * Call original free()
271 */
272 static void real_free(void *ptr)
273 {
274 original.free(malloc_default_zone(), ptr);
275 }
276
277 /**
278 * Call original realloc()
279 */
280 static void* real_realloc(void *ptr, size_t size)
281 {
282 return original.realloc(malloc_default_zone(), ptr, size);
283 }
284
285 /**
286 * Hook definition: static function with _hook suffix, takes additional zone
287 */
288 #define HOOK(ret, name, ...) \
289 static ret name ## _hook(malloc_zone_t *_z, __VA_ARGS__)
290
291 /**
292 * forward declaration of hooks
293 */
294 HOOK(void*, malloc, size_t bytes);
295 HOOK(void*, calloc, size_t nmemb, size_t size);
296 HOOK(void*, valloc, size_t size);
297 HOOK(void, free, void *ptr);
298 HOOK(void*, realloc, void *old, size_t bytes);
299
300 /**
301 * malloc zone size(), must consider the memory header prepended
302 */
303 HOOK(size_t, size, const void *ptr)
304 {
305 bool before;
306 size_t size;
307
308 if (enabled)
309 {
310 before = enable_thread(FALSE);
311 if (before)
312 {
313 ptr -= sizeof(memory_header_t);
314 }
315 }
316 size = original.size(malloc_default_zone(), ptr);
317 if (enabled)
318 {
319 enable_thread(before);
320 }
321 return size;
322 }
323
324 /**
325 * Version of malloc zones we currently support
326 */
327 #define MALLOC_ZONE_VERSION 8 /* Snow Leopard */
328
329 /**
330 * Hook-in our malloc functions into the default zone
331 */
332 static bool register_hooks()
333 {
334 static bool once = FALSE;
335 malloc_zone_t *zone;
336 void *page;
337
338 if (once)
339 {
340 return TRUE;
341 }
342 once = TRUE;
343
344 zone = malloc_default_zone();
345 if (zone->version != MALLOC_ZONE_VERSION)
346 {
347 DBG1(DBG_CFG, "malloc zone version %d unsupported (requiring %d)",
348 zone->version, MALLOC_ZONE_VERSION);
349 return FALSE;
350 }
351
352 original = *zone;
353
354 page = (void*)((uintptr_t)zone / getpagesize() * getpagesize());
355 if (mprotect(page, getpagesize(), PROT_WRITE | PROT_READ) != 0)
356 {
357 DBG1(DBG_CFG, "malloc zone unprotection failed: %s", strerror(errno));
358 return FALSE;
359 }
360
361 zone->size = size_hook;
362 zone->malloc = malloc_hook;
363 zone->calloc = calloc_hook;
364 zone->valloc = valloc_hook;
365 zone->free = free_hook;
366 zone->realloc = realloc_hook;
367
368 /* those other functions can be NULLed out to not use them */
369 zone->batch_malloc = NULL;
370 zone->batch_free = NULL;
371 zone->memalign = NULL;
372 zone->free_definite_size = NULL;
373
374 return TRUE;
375 }
376
377 #else /* !__APPLE__ */
378
379 /**
380 * dlsym() might do a malloc(), but we can't do one before we get the malloc()
381 * function pointer. Use this minimalistic malloc implementation instead.
382 */
383 static void* malloc_for_dlsym(size_t size)
384 {
385 static char buf[1024] = {};
386 static size_t used = 0;
387 char *ptr;
388
389 /* roundup to a multiple of 32 */
390 size = (size - 1) / 32 * 32 + 32;
391
392 if (used + size > sizeof(buf))
393 {
394 return NULL;
395 }
396 ptr = buf + used;
397 used += size;
398 return ptr;
399 }
400
401 /**
402 * Lookup a malloc function, while disabling wrappers
403 */
404 static void* get_malloc_fn(char *name)
405 {
406 bool before = FALSE;
407 void *fn;
408
409 if (enabled)
410 {
411 before = enable_thread(FALSE);
412 }
413 fn = dlsym(RTLD_NEXT, name);
414 if (enabled)
415 {
416 enable_thread(before);
417 }
418 return fn;
419 }
420
421 /**
422 * Call original malloc()
423 */
424 static void* real_malloc(size_t size)
425 {
426 static void* (*fn)(size_t size);
427 static int recursive = 0;
428
429 if (!fn)
430 {
431 /* checking recursiveness should actually be thread-specific. But as
432 * it is very likely that the first allocation is done before we go
433 * multi-threaded, we keep it simple. */
434 if (recursive)
435 {
436 return malloc_for_dlsym(size);
437 }
438 recursive++;
439 fn = get_malloc_fn("malloc");
440 recursive--;
441 }
442 return fn(size);
443 }
444
445 /**
446 * Call original free()
447 */
448 static void real_free(void *ptr)
449 {
450 static void (*fn)(void *ptr);
451
452 if (!fn)
453 {
454 fn = get_malloc_fn("free");
455 }
456 return fn(ptr);
457 }
458
459 /**
460 * Call original realloc()
461 */
462 static void* real_realloc(void *ptr, size_t size)
463 {
464 static void* (*fn)(void *ptr, size_t size);
465
466 if (!fn)
467 {
468 fn = get_malloc_fn("realloc");
469 }
470 return fn(ptr, size);
471 }
472
473 /**
474 * Hook definition: plain function overloading existing malloc calls
475 */
476 #define HOOK(ret, name, ...) ret name(__VA_ARGS__)
477
478 /**
479 * Hook initialization when not using hooks, resolve functions.
480 */
481 static bool register_hooks()
482 {
483 void *buf = real_malloc(8);
484 buf = real_realloc(buf, 16);
485 real_free(buf);
486 return TRUE;
487 }
488
489 #endif /* !__APPLE__ */
490
491 /**
492 * Leak report white list
493 *
494 * List of functions using static allocation buffers or should be suppressed
495 * otherwise on leak report.
496 */
497 char *whitelist[] = {
498 /* backtraces, including own */
499 "backtrace_create",
500 "strerror_safe",
501 /* pthread stuff */
502 "pthread_create",
503 "pthread_setspecific",
504 "__pthread_setspecific",
505 /* glibc functions */
506 "inet_ntoa",
507 "strerror",
508 "getprotobyname",
509 "getprotobynumber",
510 "getservbyport",
511 "getservbyname",
512 "gethostbyname",
513 "gethostbyname2",
514 "gethostbyname_r",
515 "gethostbyname2_r",
516 "getnetbyname",
517 "getpwnam_r",
518 "getgrnam_r",
519 "register_printf_function",
520 "register_printf_specifier",
521 "syslog",
522 "vsyslog",
523 "__syslog_chk",
524 "__vsyslog_chk",
525 "__fprintf_chk",
526 "getaddrinfo",
527 "setlocale",
528 "getpass",
529 "getpwent_r",
530 "setpwent",
531 "endpwent",
532 "getspnam_r",
533 "getpwuid_r",
534 "initgroups",
535 "tzset",
536 "_IO_file_doallocate",
537 /* ignore dlopen, as we do not dlclose to get proper leak reports */
538 "dlopen",
539 "dlerror",
540 "dlclose",
541 "dlsym",
542 /* mysql functions */
543 "mysql_init_character_set",
544 "init_client_errs",
545 "my_thread_init",
546 /* fastcgi library */
547 "FCGX_Init",
548 /* libxml */
549 "xmlInitCharEncodingHandlers",
550 "xmlInitParser",
551 "xmlInitParserCtxt",
552 /* libcurl */
553 "Curl_client_write",
554 /* libldap */
555 "ldap_int_initialize",
556 /* ClearSilver */
557 "nerr_init",
558 /* libgcrypt */
559 "gcrypt_plugin_create",
560 "gcry_control",
561 "gcry_check_version",
562 "gcry_randomize",
563 "gcry_create_nonce",
564 /* OpenSSL: These are needed for unit-tests only, the openssl plugin
565 * does properly clean up any memory during destroy(). */
566 "ECDSA_do_sign_ex",
567 "ECDSA_verify",
568 "RSA_new_method",
569 /* OpenSSL 1.1.0 does not cleanup anymore until the library is unloaded */
570 "OPENSSL_init_crypto",
571 "CRYPTO_THREAD_lock_new",
572 "ERR_add_error_data",
573 /* OpenSSL libssl */
574 "SSL_COMP_get_compression_methods",
575 /* NSPR */
576 "PR_CallOnce",
577 /* libapr */
578 "apr_pool_create_ex",
579 /* glib */
580 "g_type_init_with_debug_flags",
581 "g_type_register_static",
582 "g_type_class_ref",
583 "g_type_create_instance",
584 "g_type_add_interface_static",
585 "g_type_interface_add_prerequisite",
586 "g_socket_connection_factory_lookup_type",
587 /* libgpg */
588 "gpg_err_init",
589 /* gnutls */
590 "gnutls_global_init",
591 };
592
593 /**
594 * Some functions are hard to whitelist, as they don't use a symbol directly.
595 * Use some static initialization to suppress them on leak reports
596 */
597 static void init_static_allocations()
598 {
599 struct tm tm;
600 time_t t = 0;
601
602 tzset();
603 gmtime_r(&t, &tm);
604 localtime_r(&t, &tm);
605 }
606
607 /**
608 * Hashtable hash function
609 */
610 static u_int hash(backtrace_t *key)
611 {
612 enumerator_t *enumerator;
613 void *addr;
614 u_int hash = 0;
615
616 enumerator = key->create_frame_enumerator(key);
617 while (enumerator->enumerate(enumerator, &addr))
618 {
619 hash = chunk_hash_inc(chunk_from_thing(addr), hash);
620 }
621 enumerator->destroy(enumerator);
622
623 return hash;
624 }
625
626 /**
627 * Hashtable equals function
628 */
629 static bool equals(backtrace_t *a, backtrace_t *b)
630 {
631 return a->equals(a, b);
632 }
633
634 /**
635 * Summarize and print backtraces
636 */
637 static int print_traces(private_leak_detective_t *this,
638 leak_detective_report_cb_t cb, void *user,
639 int thresh, int thresh_count,
640 bool detailed, int *whitelisted, size_t *sum)
641 {
642 int leaks = 0;
643 memory_header_t *hdr;
644 enumerator_t *enumerator;
645 hashtable_t *entries;
646 struct {
647 /** associated backtrace */
648 backtrace_t *backtrace;
649 /** total size of all allocations */
650 size_t bytes;
651 /** number of allocations */
652 u_int count;
653 } *entry;
654 bool before;
655
656 before = enable_thread(FALSE);
657
658 entries = hashtable_create((hashtable_hash_t)hash,
659 (hashtable_equals_t)equals, 1024);
660 lock->lock(lock);
661 for (hdr = first_header.next; hdr != NULL; hdr = hdr->next)
662 {
663 if (whitelisted &&
664 hdr->backtrace->contains_function(hdr->backtrace,
665 whitelist, countof(whitelist)))
666 {
667 (*whitelisted)++;
668 continue;
669 }
670 entry = entries->get(entries, hdr->backtrace);
671 if (entry)
672 {
673 entry->bytes += hdr->bytes;
674 entry->count++;
675 }
676 else
677 {
678 INIT(entry,
679 .backtrace = hdr->backtrace->clone(hdr->backtrace),
680 .bytes = hdr->bytes,
681 .count = 1,
682 );
683 entries->put(entries, entry->backtrace, entry);
684 }
685 if (sum)
686 {
687 *sum += hdr->bytes;
688 }
689 leaks++;
690 }
691 lock->unlock(lock);
692
693 enumerator = entries->create_enumerator(entries);
694 while (enumerator->enumerate(enumerator, NULL, &entry))
695 {
696 if (cb)
697 {
698 if (!thresh || entry->bytes >= thresh)
699 {
700 if (!thresh_count || entry->count >= thresh_count)
701 {
702 cb(user, entry->count, entry->bytes, entry->backtrace,
703 detailed);
704 }
705 }
706 }
707 entry->backtrace->destroy(entry->backtrace);
708 free(entry);
709 }
710 enumerator->destroy(enumerator);
711 entries->destroy(entries);
712
713 enable_thread(before);
714 return leaks;
715 }
716
717 METHOD(leak_detective_t, report, void,
718 private_leak_detective_t *this, bool detailed)
719 {
720 if (lib->leak_detective)
721 {
722 int leaks, whitelisted = 0;
723 size_t sum = 0;
724
725 leaks = print_traces(this, this->report_cb, this->report_data,
726 0, 0, detailed, &whitelisted, &sum);
727 if (this->report_scb)
728 {
729 this->report_scb(this->report_data, leaks, sum, whitelisted);
730 }
731 }
732 }
733
734 METHOD(leak_detective_t, set_report_cb, void,
735 private_leak_detective_t *this, leak_detective_report_cb_t cb,
736 leak_detective_summary_cb_t scb, void *user)
737 {
738 this->report_cb = cb;
739 this->report_scb = scb;
740 this->report_data = user;
741 }
742
743 METHOD(leak_detective_t, leaks, int,
744 private_leak_detective_t *this)
745 {
746 int whitelisted = 0;
747
748 return print_traces(this, NULL, NULL, 0, 0, FALSE, &whitelisted, NULL);
749 }
750
751 METHOD(leak_detective_t, set_state, bool,
752 private_leak_detective_t *this, bool enable)
753 {
754 return enable_thread(enable);
755 }
756
757 METHOD(leak_detective_t, usage, void,
758 private_leak_detective_t *this, leak_detective_report_cb_t cb,
759 leak_detective_summary_cb_t scb, void *user)
760 {
761 bool detailed;
762 int thresh, thresh_count, leaks, whitelisted = 0;
763 size_t sum = 0;
764
765 thresh = lib->settings->get_int(lib->settings,
766 "%s.leak_detective.usage_threshold", 10240, lib->ns);
767 thresh_count = lib->settings->get_int(lib->settings,
768 "%s.leak_detective.usage_threshold_count", 0, lib->ns);
769 detailed = lib->settings->get_bool(lib->settings,
770 "%s.leak_detective.detailed", TRUE, lib->ns);
771
772 leaks = print_traces(this, cb, user, thresh, thresh_count,
773 detailed, &whitelisted, &sum);
774 if (scb)
775 {
776 scb(user, leaks, sum, whitelisted);
777 }
778 }
779
780 /**
781 * Wrapped malloc() function
782 */
783 HOOK(void*, malloc, size_t bytes)
784 {
785 memory_header_t *hdr;
786 memory_tail_t *tail;
787 bool before;
788
789 if (!enabled || thread_disabled->get(thread_disabled))
790 {
791 return real_malloc(bytes);
792 }
793
794 hdr = real_malloc(sizeof(memory_header_t) + bytes + sizeof(memory_tail_t));
795 tail = ((void*)hdr) + bytes + sizeof(memory_header_t);
796 /* set to something which causes crashes */
797 memset(hdr, MEMORY_ALLOC_PATTERN,
798 sizeof(memory_header_t) + bytes + sizeof(memory_tail_t));
799
800 before = enable_thread(FALSE);
801 hdr->backtrace = backtrace_create(2);
802 enable_thread(before);
803
804 hdr->magic = MEMORY_HEADER_MAGIC;
805 hdr->bytes = bytes;
806 tail->magic = MEMORY_TAIL_MAGIC;
807
808 add_hdr(hdr);
809
810 return hdr + 1;
811 }
812
813 /**
814 * Wrapped calloc() function
815 */
816 HOOK(void*, calloc, size_t nmemb, size_t size)
817 {
818 void *ptr;
819 volatile size_t total;
820
821 total = nmemb * size;
822 ptr = malloc(total);
823 memset(ptr, 0, total);
824
825 return ptr;
826 }
827
828 /**
829 * Wrapped valloc(), TODO: currently not supported
830 */
831 HOOK(void*, valloc, size_t size)
832 {
833 DBG1(DBG_LIB, "valloc() used, but leak-detective hook missing");
834 return NULL;
835 }
836
837 /**
838 * Wrapped free() function
839 */
840 HOOK(void, free, void *ptr)
841 {
842 memory_header_t *hdr;
843 memory_tail_t *tail;
844 backtrace_t *backtrace;
845 bool before;
846
847 if (!enabled || thread_disabled->get(thread_disabled))
848 {
849 /* after deinitialization we might have to free stuff we allocated
850 * while we were enabled */
851 if (!first_header.magic && ptr)
852 {
853 hdr = ptr - sizeof(memory_header_t);
854 tail = ptr + hdr->bytes;
855 if (hdr->magic == MEMORY_HEADER_MAGIC &&
856 tail->magic == MEMORY_TAIL_MAGIC)
857 {
858 ptr = hdr;
859 }
860 }
861 real_free(ptr);
862 return;
863 }
864 /* allow freeing of NULL */
865 if (ptr == NULL)
866 {
867 return;
868 }
869 hdr = ptr - sizeof(memory_header_t);
870 tail = ptr + hdr->bytes;
871
872 before = enable_thread(FALSE);
873 if (hdr->magic != MEMORY_HEADER_MAGIC ||
874 tail->magic != MEMORY_TAIL_MAGIC)
875 {
876 if (has_hdr(hdr))
877 {
878 /* memory was allocated by our hooks but is corrupted */
879 fprintf(stderr, "freeing corrupted memory (%p): "
880 "header magic 0x%x, tail magic 0x%x:\n",
881 ptr, hdr->magic, tail->magic);
882 }
883 else
884 {
885 /* memory was not allocated by our hooks */
886 fprintf(stderr, "freeing invalid memory (%p)\n", ptr);
887 }
888 backtrace = backtrace_create(2);
889 backtrace->log(backtrace, stderr, TRUE);
890 backtrace->destroy(backtrace);
891 }
892 else
893 {
894 remove_hdr(hdr);
895
896 hdr->backtrace->destroy(hdr->backtrace);
897
898 /* clear MAGIC, set mem to something remarkable */
899 memset(hdr, MEMORY_FREE_PATTERN,
900 sizeof(memory_header_t) + hdr->bytes + sizeof(memory_tail_t));
901
902 real_free(hdr);
903 }
904 enable_thread(before);
905 }
906
907 /**
908 * Wrapped realloc() function
909 */
910 HOOK(void*, realloc, void *old, size_t bytes)
911 {
912 memory_header_t *hdr;
913 memory_tail_t *tail;
914 backtrace_t *backtrace;
915 bool before;
916
917 if (!enabled || thread_disabled->get(thread_disabled))
918 {
919 return real_realloc(old, bytes);
920 }
921 /* allow reallocation of NULL */
922 if (old == NULL)
923 {
924 return malloc(bytes);
925 }
926 /* handle zero size as a free() */
927 if (bytes == 0)
928 {
929 free(old);
930 return NULL;
931 }
932
933 hdr = old - sizeof(memory_header_t);
934 tail = old + hdr->bytes;
935
936 remove_hdr(hdr);
937
938 if (hdr->magic != MEMORY_HEADER_MAGIC ||
939 tail->magic != MEMORY_TAIL_MAGIC)
940 {
941 fprintf(stderr, "reallocating invalid memory (%p):\n"
942 "header magic 0x%x:\n", old, hdr->magic);
943 backtrace = backtrace_create(2);
944 backtrace->log(backtrace, stderr, TRUE);
945 backtrace->destroy(backtrace);
946 }
947 else
948 {
949 /* clear tail magic, allocate, set tail magic */
950 memset(&tail->magic, MEMORY_ALLOC_PATTERN, sizeof(tail->magic));
951 }
952 hdr = real_realloc(hdr,
953 sizeof(memory_header_t) + bytes + sizeof(memory_tail_t));
954 tail = ((void*)hdr) + bytes + sizeof(memory_header_t);
955 tail->magic = MEMORY_TAIL_MAGIC;
956
957 /* update statistics */
958 hdr->bytes = bytes;
959
960 before = enable_thread(FALSE);
961 hdr->backtrace->destroy(hdr->backtrace);
962 hdr->backtrace = backtrace_create(2);
963 enable_thread(before);
964
965 add_hdr(hdr);
966
967 return hdr + 1;
968 }
969
970 METHOD(leak_detective_t, destroy, void,
971 private_leak_detective_t *this)
972 {
973 disable_leak_detective();
974 lock->destroy(lock);
975 thread_disabled->destroy(thread_disabled);
976 free(this);
977 first_header.magic = 0;
978 first_header.next = NULL;
979 }
980
981 /*
982 * see header file
983 */
984 leak_detective_t *leak_detective_create()
985 {
986 private_leak_detective_t *this;
987
988 INIT(this,
989 .public = {
990 .report = _report,
991 .set_report_cb = _set_report_cb,
992 .usage = _usage,
993 .leaks = _leaks,
994 .set_state = _set_state,
995 .destroy = _destroy,
996 },
997 );
998
999 if (getenv("LEAK_DETECTIVE_DISABLE") != NULL)
1000 {
1001 free(this);
1002 return NULL;
1003 }
1004
1005 lock = spinlock_create();
1006 thread_disabled = thread_value_create(NULL);
1007
1008 init_static_allocations();
1009
1010 if (register_hooks())
1011 {
1012 enable_leak_detective();
1013 }
1014 return &this->public;
1015 }