00001 /* _______ __ 00002 / ___/ / ___ __ _ / / ___ 00003 / /__/ _ \/ _ \/ ' \/ _ \/ _ \ 00004 \___/_//_/\___/_/_/_/_.__/\___/ 00005 */ 00006 // 00007 // This software is copyright (C) by the Lawrence Berkeley 00008 // National Laboratory. Permission is granted to reproduce 00009 // this software for non-commercial purposes provided that 00010 // this notice is left intact. 00011 // 00012 // It is acknowledged that the U.S. Government has rights to 00013 // this software under Contract DE-AC03-765F00098 between 00014 // the U.S. Department of Energy and the University of 00015 // California. 00016 // 00017 // This software is provided as a professional and academic 00018 // contribution for joint exchange. Thus it is experimental, 00019 // is provided ``as is'', with no warranties of any kind 00020 // whatsoever, no support, no promise of updates, or printed 00021 // documentation. By using this software, you acknowledge 00022 // that the Lawrence Berkeley National Laboratory and 00023 // Regents of the University of California shall have no 00024 // liability with respect to the infringement of other 00025 // copyrights by any part of this software. 00026 // 00027 00028 #ifndef MEMTRACK_H 00029 #define MEMTRACK_H 00030 00031 #ifdef ENABLE_MEMORY_TRACKING 00032 00033 #include <iostream> 00034 using std::ostream; 00035 00036 // user functions 00037 00038 void ReportUnfreedMemory(ostream& os); 00039 void ReportMemoryUsage(ostream& os); 00040 void UnfreedMemory(); // send output to pout() 00041 void MemoryUsage(); // sned output to pout() 00042 00043 void memTrackingOn(); 00044 void memtrackingOff(); 00045 00046 // =========== end user functions.=================================== 00047 00048 00049 void AddTrack(void* addr, size_t asize, const char *fname, 00050 int lnum, bool malloc); 00051 void RemoveTrack(void* addr, bool malloc); 00052 00053 00054 // void * operator new(size_t size, char const *file, int line); 00055 00056 // void operator delete(void *p) throw(); 00057 00058 // void * operator new[](size_t size, char const *file, int line); 00059 00060 // void operator delete[](void *p) throw(); 00061 00062 00063 // these are ordered in this way so that we don't double track malloc 00064 // functions. 00065 void* mallocp(size_t size, const char* file, int line); 00066 void* reallocp(void* p, size_t size, const char* file, int line); 00067 void* callocp(size_t nelem, size_t elsize, const char* file, int line); 00068 void freep(void* p); 00069 00070 #define calloc(a,b) callocp(a,b,__FILE__, __LINE__) 00071 #define realloc(a, b) reallocp(a, b, __FILE__, __LINE__) 00072 #define malloc(a) mallocp(a, __FILE__, __LINE__) 00073 #define free(a) freep(a) 00074 00075 #endif 00076 00077 #endif 00078