Display a list of all leaked memory blocks when terminating

svn path=/trunk/; revision=29910
This commit is contained in:
Thomas Bluemel 2007-10-27 04:00:25 +00:00
parent 283b505889
commit 9c105be99b

View file

@ -9,12 +9,14 @@
typedef struct typedef struct
{ {
size_t size; size_t size;
LIST_ENTRY list_entry;
const char *file; const char *file;
int line; int line;
} alloc_info, *palloc_info; } alloc_info, *palloc_info;
static size_t allocations = 0; static size_t allocations = 0;
static size_t allocated_memory = 0; static size_t allocated_memory = 0;
static LIST_ENTRY alloc_list_head = {&alloc_list_head, &alloc_list_head};
static void * static void *
get_base_ptr(void *ptr) get_base_ptr(void *ptr)
@ -22,6 +24,12 @@ get_base_ptr(void *ptr)
return (void *)((UINT_PTR)ptr - REDZONE_SIZE - sizeof(alloc_info)); return (void *)((UINT_PTR)ptr - REDZONE_SIZE - sizeof(alloc_info));
} }
static void *
get_ptr_from_base(palloc_info info)
{
return (void*)((size_t)(info + 1) + REDZONE_SIZE);
}
static void * static void *
write_redzone(void *ptr, size_t size, const char *file, int line) write_redzone(void *ptr, size_t size, const char *file, int line)
{ {
@ -88,6 +96,50 @@ calculate_size_with_redzone(size_t size)
return sizeof(alloc_info) + size + (2 * REDZONE_SIZE); return sizeof(alloc_info) + size + (2 * REDZONE_SIZE);
} }
static void
add_mem_to_list(void *ptr)
{
palloc_info info = (palloc_info)ptr;
InsertTailList(&alloc_list_head, &info->list_entry);
}
static void
del_mem_from_list(void *ptr)
{
palloc_info info = (palloc_info)ptr;
RemoveEntryList(&info->list_entry);
}
static void
dump_mem_list(void)
{
palloc_info info;
PLIST_ENTRY entry;
void *ptr;
entry = alloc_list_head.Flink;
while (entry != &alloc_list_head)
{
info = CONTAINING_RECORD(entry, alloc_info, list_entry);
DbgPrint(" * Block: 0x%p Size: %lu allocated from %s:%d\n", get_ptr_from_base(info), info->size, info->file, info->line);
ptr = (void *)(info + 1);
if (!check_redzone_region(ptr, REDZONE_LEFT, &ptr))
{
DbgPrint(" !!! Detected buffer underflow !!!\n");
}
ptr = (void *)((UINT_PTR)ptr + info->size);
if (!check_redzone_region(ptr, REDZONE_RIGHT, NULL))
{
DbgPrint(" !!! Detected buffer overflow !!!\n");
}
entry = entry->Flink;
}
}
void * void *
cmd_alloc_dbg(size_t size, const char *file, int line) cmd_alloc_dbg(size_t size, const char *file, int line)
{ {
@ -98,6 +150,7 @@ cmd_alloc_dbg(size_t size, const char *file, int line)
{ {
allocations++; allocations++;
allocated_memory += size; allocated_memory += size;
add_mem_to_list(newptr);
newptr = write_redzone(newptr, size, file, line); newptr = write_redzone(newptr, size, file, line);
} }
@ -122,12 +175,16 @@ cmd_realloc_dbg(void *ptr, size_t size, const char *file, int line)
prev_size = ((palloc_info)ptr)->size; prev_size = ((palloc_info)ptr)->size;
check_redzone(ptr, file, line); check_redzone(ptr, file, line);
del_mem_from_list(ptr);
newptr = realloc(ptr, calculate_size_with_redzone(size)); newptr = realloc(ptr, calculate_size_with_redzone(size));
if (newptr != NULL) if (newptr != NULL)
{ {
allocated_memory += size - prev_size; allocated_memory += size - prev_size;
add_mem_to_list(newptr);
newptr = write_redzone(newptr, size, file, line); newptr = write_redzone(newptr, size, file, line);
} }
else
add_mem_to_list(ptr);
return newptr; return newptr;
} }
@ -141,6 +198,7 @@ cmd_free_dbg(void *ptr, const char *file, int line)
check_redzone(ptr, file, line); check_redzone(ptr, file, line);
allocations--; allocations--;
allocated_memory -= ((palloc_info)ptr)->size; allocated_memory -= ((palloc_info)ptr)->size;
del_mem_from_list(ptr);
} }
free(ptr); free(ptr);
@ -179,6 +237,8 @@ cmd_exit(int code)
if (allocations != 0 || allocated_memory != 0) if (allocations != 0 || allocated_memory != 0)
{ {
DbgPrint("CMD: Leaking %lu bytes of memory in %lu blocks! Exit code: %d\n", allocated_memory, allocations, code); DbgPrint("CMD: Leaking %lu bytes of memory in %lu blocks! Exit code: %d\n", allocated_memory, allocations, code);
if (allocations != 0)
dump_mem_list();
} }
ExitProcess(code); ExitProcess(code);