// for kernel32.dll
LPVOID VisualLeakDetector::_HeapReAlloc (HANDLE heap, DWORD flags, LPVOID mem, SIZE_T size)
{
PRINT_HOOKED_FUNCTION();
// Reallocate the block.
LPVOID newmem = HeapReAlloc(heap, flags, mem, size);
if ((newmem == NULL) || !g_vld.enabled())
return newmem;
if (!g_DbgHelp.IsLockedByCurrentThread()) { // skip dbghelp.dll calls
CAPTURE_CONTEXT();
CaptureContext cc(HeapReAlloc, size, context_);
cc.Set(heap, mem, newmem, size);
}
return newmem;
}
// HeapAlloc (kernel32.dll) call RtlAllocateHeap (ntdll.dll)
LPVOID VisualLeakDetector::_HeapAlloc (HANDLE heap, DWORD flags, SIZE_T size)
{
PRINT_HOOKED_FUNCTION2();
// Allocate the block.
LPVOID block = HeapAlloc(heap, flags, size);
if ((block == NULL) || !g_vld.enabled())
return block;
if (!g_DbgHelp.IsLockedByCurrentThread()) { // skip dbghelp.dll calls
CAPTURE_CONTEXT();
CaptureContext cc(HeapAlloc, size, context_);
cc.Set(heap, block, NULL, size);
}
return block;
}
// HeapFree (kernel32.dll) call RtlFreeHeap (ntdll.dll)
BOOL VisualLeakDetector::_HeapFree (HANDLE heap, DWORD flags, LPVOID mem)
{
PRINT_HOOKED_FUNCTION2();
BOOL status;
if (!g_DbgHelp.IsLockedByCurrentThread()) // skip dbghelp.dll calls
{
// Record the current frame pointer.
CAPTURE_CONTEXT();
context_.func = reinterpret_cast<UINT_PTR>(m_HeapFree);
// Unmap the block from the specified heap.
g_vld.unmapBlock(heap, mem, context_);
}
status = m_HeapFree(heap, flags, mem);
return status;
}
