Windows2000/private/shell/inc/deballoc.c
2020-09-30 17:12:32 +02:00

78 lines
2.1 KiB
C

// 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);
}