115 lines
3.2 KiB
C
115 lines
3.2 KiB
C
|
//
|
||
|
|
// MEMTRACK.H
|
||
|
|
// Standard NetMeeting memory leak tracking.
|
||
|
|
//
|
||
|
|
// In retail:
|
||
|
|
// new/MemAlloc become LocalAlloc()
|
||
|
|
// MemReAlloc becomes LocalReAlloc()
|
||
|
|
// delete/MemFree become LocalFree()
|
||
|
|
//
|
||
|
|
// In debug:
|
||
|
|
// allocations are tracked, with module/file/line number
|
||
|
|
// leaked blocks are spewed when the module unloads
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// USAGE:
|
||
|
|
// (1) Include this header and link to NMUTIL
|
||
|
|
// (2) If your component requires zero-initialized memory, define
|
||
|
|
// _MEM_ZEROINIT (for both debug and retail) in your SOURCES file
|
||
|
|
// (3) In your DllMain, on DLL_PROCESS_ATTACH call DBG_INIT_MEMORY_TRACKING,
|
||
|
|
// and on DLL_PROCESS_DETACH call DBG_CHECK_MEMORY_TRACKING
|
||
|
|
// (4) In DEBUG, you can make a call to DbgMemTrackDumpCurrent() to dump
|
||
|
|
// the currently allocated memory list from code.
|
||
|
|
//
|
||
|
|
|
||
|
|
|
||
|
|
#ifndef _MEMTRACK_H
|
||
|
|
#define _MEMTRACK_H
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
//
|
||
|
|
// MEMORY ALLOCATIONS/TRACKING
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// GUI message boxes kill us when we hit an assert or error, because they
|
||
|
|
// have a message pump that causes messages to get dispatched, making it
|
||
|
|
// very difficult for us to debug problems when they occur. Therefore
|
||
|
|
// we redefine ERROR_OUT and ASSERT
|
||
|
|
//
|
||
|
|
#ifdef DEBUG
|
||
|
|
|
||
|
|
|
||
|
|
#undef assert
|
||
|
|
#define assert(x) ASSERT(x)
|
||
|
|
|
||
|
|
|
||
|
|
void WINAPI DbgMemTrackDumpCurrent(void);
|
||
|
|
void WINAPI DbgMemTrackFinalCheck(void);
|
||
|
|
|
||
|
|
LPVOID WINAPI DbgMemAlloc(UINT cbSize, LPVOID caller, LPSTR pszFileName, UINT nLineNumber);
|
||
|
|
void WINAPI DbgMemFree(LPVOID ptr);
|
||
|
|
LPVOID WINAPI DbgMemReAlloc(LPVOID ptr, UINT cbSize, UINT uFlags, LPSTR pszFileName, UINT nLineNumber);
|
||
|
|
|
||
|
|
|
||
|
|
#define DBG_CHECK_MEMORY_TRACKING(hInst) DbgMemTrackFinalCheck()
|
||
|
|
|
||
|
|
#define MemAlloc(cbSize) DbgMemAlloc(cbSize, NULL, __FILE__, __LINE__)
|
||
|
|
|
||
|
|
#ifdef _MEM_ZEROINIT
|
||
|
|
#define MemReAlloc(pObj, cbSize) DbgMemReAlloc((pObj), (cbSize), LMEM_MOVEABLE | LMEM_ZEROINIT, __FILE__, __LINE__)
|
||
|
|
#else
|
||
|
|
#define MemReAlloc(pObj, cbSize) DbgMemReAlloc((pObj), (cbSize), LMEM_MOVEABLE, __FILE__, __LINE__)
|
||
|
|
#endif //_MEM_ZEROINIT
|
||
|
|
|
||
|
|
#define MemFree(pObj) DbgMemFree(pObj)
|
||
|
|
|
||
|
|
void WINAPI DbgSaveFileLine(LPSTR pszFileName, UINT nLineNumber);
|
||
|
|
#define DBG_SAVE_FILE_LINE DbgSaveFileLine(__FILE__, __LINE__);
|
||
|
|
|
||
|
|
|
||
|
|
// RETAIL
|
||
|
|
#else
|
||
|
|
|
||
|
|
#define DBG_CHECK_MEMORY_TRACKING(hInst)
|
||
|
|
|
||
|
|
#ifdef _MEM_ZEROINIT
|
||
|
|
#define MemAlloc(cbSize) LocalAlloc(LPTR, (cbSize))
|
||
|
|
#define MemReAlloc(pObj, cbSize) LocalReAlloc((pObj), (cbSize), LMEM_MOVEABLE | LMEM_ZEROINIT)
|
||
|
|
#else
|
||
|
|
#define MemAlloc(cbSize) LocalAlloc(LMEM_FIXED, (cbSize))
|
||
|
|
#define MemReAlloc(pObj, cbSize) LocalReAlloc((pObj), (cbSize), LMEM_MOVEABLE)
|
||
|
|
#endif // _MEM_ZEROINIT
|
||
|
|
|
||
|
|
#define MemFree(pObj) LocalFree(pObj)
|
||
|
|
|
||
|
|
#define DBG_SAVE_FILE_LINE
|
||
|
|
|
||
|
|
#endif // DEBUG
|
||
|
|
|
||
|
|
|
||
|
|
void WINAPI DbgInitMemTrack(HINSTANCE hDllInst, BOOL fZeroOut);
|
||
|
|
#ifdef _MEM_ZEROINIT
|
||
|
|
#define DBG_INIT_MEMORY_TRACKING(hInst) DbgInitMemTrack(hInst, TRUE)
|
||
|
|
#else
|
||
|
|
#define DBG_INIT_MEMORY_TRACKING(hInst) DbgInitMemTrack(hInst, FALSE)
|
||
|
|
#endif //_MEM_ZEROINIT
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#define MEMALLOC(cb) MemAlloc(cb)
|
||
|
|
#define MEMREALLOC(p, cb) MemReAlloc(p, cb)
|
||
|
|
#define MEMFREE(p) MemFree(p)
|
||
|
|
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#endif // #ifndef _MEMTRACK_H
|