1 /* 2 * Copyright 2009, Ingo Weinhold, ingo_weinhold@gmx.de 3 * Distributed under the terms of the MIT License. 4 */ 5 #ifndef _KERNEL_DEBUG_HEAP_H 6 #define _KERNEL_DEBUG_HEAP_H 7 8 #include <debug.h> 9 10 11 struct DebugAllocPool; 12 typedef struct DebugAllocPool debug_alloc_pool; 13 14 15 #ifdef __cplusplus 16 extern "C" { 17 #endif 18 19 debug_alloc_pool* create_debug_alloc_pool(); 20 void delete_debug_alloc_pool(debug_alloc_pool* pool); 21 void* debug_malloc(size_t size); 22 void* debug_calloc(size_t num, size_t size); 23 void debug_free(void* address); 24 void debug_heap_init(); 25 26 #ifdef __cplusplus 27 } 28 #endif 29 30 31 #ifdef __cplusplus 32 33 struct kdebug_alloc_t {}; 34 extern const kdebug_alloc_t kdebug_alloc; 35 36 inline void* 37 operator new(size_t size, const kdebug_alloc_t&) throw() 38 { 39 return debug_malloc(size); 40 } 41 42 namespace DebugAlloc { 43 template<typename Type> 44 inline void 45 destroy(Type* object) 46 { 47 if (object != NULL) { 48 object->~Type(); 49 debug_free(object); 50 // NOTE: Doesn't work for multiple inheritence! 51 } 52 } 53 } 54 55 struct DebugAllocPoolScope { 56 DebugAllocPoolScope() 57 { 58 fPool = create_debug_alloc_pool(); 59 } 60 61 ~DebugAllocPoolScope() 62 { 63 delete_debug_alloc_pool(fPool); 64 } 65 66 private: 67 DebugAllocPool* fPool; 68 }; 69 70 #endif // __cplusplus 71 72 #endif /* _KERNEL_DEBUG_HEAP_H */ 73