mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.2 KiB
77 lines
2.2 KiB
//====================================================================
|
|
// Debugging memory problems.
|
|
//====================================================================
|
|
|
|
#undef LocalAlloc
|
|
#undef LocalReAlloc
|
|
#undef LocalFree
|
|
|
|
#define CBALLOCEXTRA (sizeof(LPARAM)+sizeof(UINT))
|
|
#define HMEM2PTR(hMem,i) (((BYTE*)hMem)+i)
|
|
|
|
void _StoreSigniture(HLOCAL hMem, UINT uByte, LPARAM lParam)
|
|
{
|
|
if (hMem)
|
|
{
|
|
SIZE_T uSize = LocalSize(hMem);
|
|
ASSERT(uSize>=uByte+CBALLOCEXTRA);
|
|
*(UINT*)HMEM2PTR(hMem, uSize-sizeof(UINT)) = uByte;
|
|
*(LPARAM*)HMEM2PTR(hMem, uByte) = lParam;
|
|
}
|
|
}
|
|
|
|
UINT _ValidateLocalMem(HLOCAL hMem, LPARAM lParam, LPCSTR pszText)
|
|
{
|
|
UINT uByte = 0;
|
|
if (hMem)
|
|
{
|
|
SIZE_T uSize = LocalSize(hMem);
|
|
if (uSize)
|
|
{
|
|
LPARAM lParamStored;
|
|
uByte = *(UINT*)HMEM2PTR(hMem, uSize-sizeof(UINT));
|
|
AssertMsg(uByte+CBALLOCEXTRA <= uSize,
|
|
TEXT("cm ASSERT! Bogus uByte %d (%s for %x)"),
|
|
uByte, pszText, hMem);
|
|
lParamStored = *(LPARAM*)HMEM2PTR(hMem, uByte);
|
|
AssertMsg(lParamStored==lParam, TEXT("cm ASSERT! Bad Signiture %x (%s for %x)"),
|
|
lParamStored, pszText, hMem);
|
|
}
|
|
else
|
|
{
|
|
AssertMsg(0, TEXT("cm ASSERT! LocalSize is zero (%s for %x)"),
|
|
pszText, hMem);
|
|
}
|
|
}
|
|
return uByte;
|
|
}
|
|
|
|
HLOCAL WINAPI DebugLocalAlloc(UINT uFlags, UINT uBytes)
|
|
{
|
|
HLOCAL hMem = LocalAlloc(uFlags, uBytes+CBALLOCEXTRA);
|
|
_StoreSigniture(hMem, uBytes, (LPARAM)hMem);
|
|
return hMem;
|
|
}
|
|
|
|
HLOCAL WINAPI DebugLocalReAlloc(HLOCAL hMem, UINT uBytes, UINT uFlags)
|
|
{
|
|
HLOCAL hNew;
|
|
_ValidateLocalMem(hMem, (LPARAM)hMem, "LocalReAlloc");
|
|
hNew = LocalReAlloc(hMem, uBytes+CBALLOCEXTRA, uFlags);
|
|
_StoreSigniture(hNew, uBytes, (LPARAM)hNew);
|
|
return hNew;
|
|
}
|
|
|
|
HLOCAL WINAPI DebugLocalFree( HLOCAL hMem )
|
|
{
|
|
UINT uBytes = _ValidateLocalMem(hMem, (LPARAM)hMem, "LocalFree");
|
|
if (uBytes)
|
|
{
|
|
_StoreSigniture(hMem, uBytes, (LPARAM)0xDDDDDDDDL);
|
|
}
|
|
else
|
|
{
|
|
AssertMsg(0, TEXT("cm ASSERT! LocalFree _ValidateLocalMem returned 0 for %x"), hMem);
|
|
}
|
|
return LocalFree(hMem);
|
|
}
|