#ifdef _DEBUG #define MEM_LEAK_TEST #ifdef MEM_LEAK_TEST #ifndef __MEM_TEST__H__ #define __MEM_TEST__H__ #pragma warning(disable:4786) #pragma warning(disable:4291) #ifndef _WIN32_WINNT #define _WIN32_WINNT 0x0500 #endif #include #include #include #include #define new new(__FILE__, __LINE__) #include #include #undef new /* #include #include #include #include #include #include #include */ typedef struct { char File[300]; int line; int size; int order; } DEBUG_LOCATION; typedef std::map MEMLIST; typedef MEMLIST::iterator MEMLIST_ITER; class CMemList { MEMLIST MemList; HANDLE hSemaphone; DWORD ThreadID; HMODULE m_hModule; enum INIT { Initialized = 123, NotInitialized }; INIT State; int order; bool LogAllocate; void Insert(void *res, DEBUG_LOCATION &info); void Remove(void *ptr); public: CMemList(); ~CMemList(); void * AddItem ( size_t size, char * filename, int line ); void * ReAllocItem ( void * ptr, size_t size, const char * filename, int line ); void removeItem ( void * ptr, bool bFree ); void RecordAddItem ( void * ptr, size_t size, const char * filename, int line ); }; void* operator new (size_t size ); void* operator new (size_t size, char* filename, int line); void* operator new [] (size_t size ); void* operator new [] (size_t size, char* filename, int line); void operator delete ( void* ptr); void operator delete[](void* ptr); void* MemTest_malloc(size_t size, char* filename, int line); void* MemTest_realloc(void* ptr, size_t size, char* filename, int line); void MemTest_AddLeak(char* Comment); LPVOID MemTest_VirtualAlloc(LPVOID lpAddress, SIZE_T dwSize, DWORD flAllocationType, DWORD flProtect, LPCSTR filename, int line); BOOL MemTest_VirtualFree( LPVOID lpAddress, SIZE_T dwSize, DWORD dwFreeType ); #endif #undef new #undef malloc #undef free #define new new(__FILE__, __LINE__) #define malloc(x) (MemTest_malloc((x),__FILE__, __LINE__)) #define realloc(mem,x) (MemTest_realloc((mem),(x),__FILE__, __LINE__)) #define free(x) (delete (x)) #define VirtualAlloc(lpAddress, dwSize, flAllocationType, flProtect ) \ (MemTest_VirtualAlloc((lpAddress), (dwSize), (flAllocationType), (flProtect),__FILE__, __LINE__)) #define VirtualFree(lpAddress, dwSize, dwFreeType ) \ (MemTest_VirtualFree((lpAddress), (dwSize), (dwFreeType))) #endif #endif