File: leak_dumper.h

package info (click to toggle)
glest 3.2.2-2
  • links: PTS, VCS
  • area: contrib
  • in suites: squeeze
  • size: 2,800 kB
  • ctags: 6,581
  • sloc: cpp: 32,575; sh: 8,341; makefile: 63
file content (121 lines) | stat: -rw-r--r-- 3,353 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// ==============================================================
//	This file is part of Glest Shared Library (www.glest.org)
//
//	Copyright (C) 2001-2008 Martio Figueroa
//
//	You can redistribute this code and/or modify it under 
//	the terms of the GNU General Public License as published 
//	by the Free Software Foundation; either version 2 of the 
//	License, or (at your option) any later version
// ==============================================================

#ifndef _LEAKDUMPER_H_
#define _LEAKDUMPER_H_

//#define SL_LEAK_DUMP
//SL_LEAK_DUMP controls if leak dumping is enabled or not

#ifdef SL_LEAK_DUMP

#include <cstdlib>
#include <cstdio>

//including this header in any file of a project will cause all
//leaks to be dumped into leak_dump.txt, but only allocations that 
//ocurred in a file where this header is included will have
//file and line number

struct AllocInfo{
	int line;
	const char *file;
	size_t bytes;
	void *ptr;
	bool free;
	bool array;

	AllocInfo();
	AllocInfo(void* ptr, const char* file, int line, size_t bytes, bool array);
};

// =====================================================
//	class AllocRegistry
// =====================================================

class AllocRegistry{
private:
	static const unsigned maxAllocs= 40000;

private:
	AllocRegistry();

private:
	AllocInfo allocs[maxAllocs];	//array to store allocation info
	int allocCount;					//allocations
	size_t allocBytes;					//bytes allocated
	int nonMonitoredCount;			
	size_t nonMonitoredBytes;

public:
	~AllocRegistry();

	static AllocRegistry &getInstance();

	void allocate(AllocInfo info);
	void deallocate(void* ptr, bool array);
	void reset();
	void dump(const char *path);
};

//if an allocation ocurrs in a file where "leaks_dumper.h" is not included
//this operator new is called and file and line will be unknown
inline void * operator new (size_t bytes){
	void *ptr= malloc(bytes);
	AllocRegistry::getInstance().allocate(AllocInfo(ptr, "unknown", 0, bytes, false));
	return ptr;
}

inline void operator delete(void *ptr){
	AllocRegistry::getInstance().deallocate(ptr, false);
	free(ptr);
}

inline void * operator new[](size_t bytes){
	void *ptr= malloc(bytes);
	AllocRegistry::getInstance().allocate(AllocInfo(ptr, "unknown", 0, bytes, true));
	return ptr;
}

inline void operator delete [](void *ptr){
	AllocRegistry::getInstance().deallocate(ptr, true);
	free(ptr);
}

//if an allocation ocurrs in a file where "leaks_dumper.h" is included 
//this operator new is called and file and line will be known
inline void * operator new (size_t bytes, char* file, int line){
	void *ptr= malloc(bytes);
	AllocRegistry::getInstance().allocate(AllocInfo(ptr, file, line, bytes, false));
	return ptr;
}

inline void operator delete(void *ptr, char* file, int line){
	AllocRegistry::getInstance().deallocate(ptr, false);
	free(ptr);
}

inline void * operator new[](size_t bytes, char* file, int line){
	void *ptr= malloc(bytes);
	AllocRegistry::getInstance().allocate(AllocInfo(ptr, file, line, bytes, true));
	return ptr;
}

inline void operator delete [](void *ptr, char* file, int line){
	AllocRegistry::getInstance().deallocate(ptr, true);
	free(ptr);
}

#define new new(__FILE__, __LINE__)

#endif

#endif