| 1 | /*
 | 
|---|
| 2 |  * MemDebug.hpp
 | 
|---|
| 3 |  *
 | 
|---|
| 4 |  *  Created on: Apr 28, 2010
 | 
|---|
| 5 |  *      Author: crueger
 | 
|---|
| 6 |  */
 | 
|---|
| 7 | 
 | 
|---|
| 8 | #ifndef MEMDEBUG_HPP_
 | 
|---|
| 9 | #define MEMDEBUG_HPP_
 | 
|---|
| 10 | 
 | 
|---|
| 11 | /**
 | 
|---|
| 12 |  * @file
 | 
|---|
| 13 |  * This module allows easy automatic memory tracking. Link this module to replace the
 | 
|---|
| 14 |  * operators new, new[], delete and delete[] with custom versions that add tracking
 | 
|---|
| 15 |  * information to allocated blocks.
 | 
|---|
| 16 |  *
 | 
|---|
| 17 |  * All tracking is done in O(1) for new and delete operators. Summaries for the
 | 
|---|
| 18 |  * used memory take O(N), where N is the number of currently allocated memory chunks.
 | 
|---|
| 19 |  *
 | 
|---|
| 20 |  * To use full tracking including file name and line number include this file in
 | 
|---|
| 21 |  * your sourcefiles.
 | 
|---|
| 22 |  */
 | 
|---|
| 23 | #ifndef NDEBUG
 | 
|---|
| 24 | #ifndef NO_MEMDEBUG
 | 
|---|
| 25 | 
 | 
|---|
| 26 | #ifndef MEMDEBUG
 | 
|---|
| 27 | #define MEMDEBUG
 | 
|---|
| 28 | #endif
 | 
|---|
| 29 | 
 | 
|---|
| 30 | #include <new>
 | 
|---|
| 31 | 
 | 
|---|
| 32 | // some light header files, that do weird new stuff and therefore need
 | 
|---|
| 33 | // to be loaded before the define
 | 
|---|
| 34 | #include <string>
 | 
|---|
| 35 | #include <boost/optional.hpp>
 | 
|---|
| 36 | #include <boost/shared_ptr.hpp>
 | 
|---|
| 37 | #include <boost/function.hpp>
 | 
|---|
| 38 | #include <boost/program_options.hpp>
 | 
|---|
| 39 | 
 | 
|---|
| 40 | 
 | 
|---|
| 41 | namespace Memory {
 | 
|---|
| 42 | 
 | 
|---|
| 43 |   /**
 | 
|---|
| 44 |    * Displays a short summary of the memory state.
 | 
|---|
| 45 |    */
 | 
|---|
| 46 |   void getState();
 | 
|---|
| 47 | 
 | 
|---|
| 48 |   void _ignore(void*);
 | 
|---|
| 49 | 
 | 
|---|
| 50 |   /**
 | 
|---|
| 51 |    * Use this to disable memory for a certain pointer on the heap.
 | 
|---|
| 52 |    * This is useful for pointers which should live over the run of the
 | 
|---|
| 53 |    * program, and which are deleted automatically at the end. Usually these
 | 
|---|
| 54 |    * pointers should be wrapped inside some kind of smart pointer to
 | 
|---|
| 55 |    * ensure destruction at the end.
 | 
|---|
| 56 |    */
 | 
|---|
| 57 |   template <typename T>
 | 
|---|
| 58 |   T *ignore(T* ptr){
 | 
|---|
| 59 |     _ignore((void*)ptr);
 | 
|---|
| 60 |     return ptr;
 | 
|---|
| 61 |   }
 | 
|---|
| 62 | }
 | 
|---|
| 63 | #ifdef __GNUC__
 | 
|---|
| 64 | void *operator new   (size_t nbytes,const char* file, int line, const char* func) throw(std::bad_alloc);
 | 
|---|
| 65 | void *operator new[] (size_t nbytes,const char* file, int line, const char* func) throw(std::bad_alloc);
 | 
|---|
| 66 | #else
 | 
|---|
| 67 | void *operator new   (size_t nbytes,const char* file, int line) throw(std::bad_alloc);
 | 
|---|
| 68 | void *operator new[] (size_t nbytes,const char* file, int line) throw(std::bad_alloc);
 | 
|---|
| 69 | #endif
 | 
|---|
| 70 | void operator delete   (void *ptr,const char*, int) throw();
 | 
|---|
| 71 | void operator delete[] (void *ptr,const char*, int) throw();
 | 
|---|
| 72 | 
 | 
|---|
| 73 | 
 | 
|---|
| 74 | 
 | 
|---|
| 75 | /**
 | 
|---|
| 76 |  * This macro replaces all occurences of the keyword new with a replaced
 | 
|---|
| 77 |  * version that allows tracking.
 | 
|---|
| 78 |  */
 | 
|---|
| 79 | #ifdef __GNUC__
 | 
|---|
| 80 | #define new new(__FILE__,__LINE__,__PRETTY_FUNCTION__)
 | 
|---|
| 81 | #else
 | 
|---|
| 82 | #define new new(__FILE__,__LINE__)
 | 
|---|
| 83 | #endif
 | 
|---|
| 84 | 
 | 
|---|
| 85 | #endif
 | 
|---|
| 86 | #endif
 | 
|---|
| 87 | 
 | 
|---|
| 88 | 
 | 
|---|
| 89 | #ifdef NDEBUG
 | 
|---|
| 90 | #undef MEMDEBUG
 | 
|---|
| 91 | #endif
 | 
|---|
| 92 | 
 | 
|---|
| 93 | #ifndef MEMDEBUG
 | 
|---|
| 94 | // memory debugging was disabled
 | 
|---|
| 95 | 
 | 
|---|
| 96 | namespace Memory {
 | 
|---|
| 97 |   inline void getState(){}
 | 
|---|
| 98 | 
 | 
|---|
| 99 |   template <typename T>
 | 
|---|
| 100 |   inline T *ignore(T* ptr){
 | 
|---|
| 101 |     return ptr;
 | 
|---|
| 102 |   }
 | 
|---|
| 103 | }
 | 
|---|
| 104 | 
 | 
|---|
| 105 | #endif
 | 
|---|
| 106 | #endif /* MEMDEBUG_HPP_ */
 | 
|---|