windows-nt/Source/XPSP1/NT/windows/winstate/cobra/utils/main/basemem.c

982 lines
22 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
basemem.c
Abstract:
Implements macros and declares functions for basic allocation functions.
Consolidated into this file from debug.c and main.c
Author:
Marc R. Whitten (marcw) 09-Sep-1999
Revision History:
--*/
#include "pch.h"
//
// Includes
//
#include "utilsp.h"
//
// Constants
//
#ifdef DEBUG
#define TRAIL_SIG 0x708aa210
#define TRACK_SIGNATURE 0x30405060
#endif
//
// Macros
//
#define REUSE_SIZE_PTR(ptr) ((PDWORD) ((PBYTE) ptr - sizeof (DWORD)))
#define REUSE_TAG_PTR(ptr) ((PDWORD) ((PBYTE) ptr + (*REUSE_SIZE_PTR(ptr))))
//
// Types
//
#ifdef DEBUG
typedef struct _tagTRACKSTRUCT {
DWORD Signature;
PCSTR File;
DWORD Line;
SIZE_T Size;
PSTR Comment;
PCSTR CallerFile;
DWORD CallerLine;
BOOL Allocated;
struct _tagTRACKSTRUCT *PrevAlloc;
struct _tagTRACKSTRUCT *NextAlloc;
} TRACKSTRUCT, *PTRACKSTRUCT;
#endif
//
// Globals
//
#ifdef DEBUG
PTRACKSTRUCT g_TrackHead = NULL;
#endif
//
// Heap debug statistics
//
static SIZE_T g_TotalBytesAllocated = 0;
static SIZE_T g_MaxBytesInUse = 0;
static SIZE_T g_HeapAllocs = 0;
static SIZE_T g_HeapReAllocs = 0;
static SIZE_T g_HeapFrees = 0;
static SIZE_T g_HeapAllocFails = 0;
static SIZE_T g_HeapReAllocFails = 0;
static SIZE_T g_HeapFreeFails = 0;
//
// Out of memory string -- loaded at initialization
//
PCSTR g_OutOfMemoryString = NULL;
PCSTR g_OutOfMemoryRetry = NULL;
HWND g_OutOfMemoryParentWnd;
//
// Macro expansion list
//
// None
//
// Private function prototypes
//
#ifdef DEBUG
SIZE_T
pDebugHeapValidatePtrUnlocked (
HANDLE hHeap,
PCVOID CallerPtr,
PCSTR File,
DWORD Line
);
VOID
pTrackInsert (
PCSTR File,
DWORD Line,
SIZE_T Size,
PTRACKSTRUCT p
);
VOID
pTrackDelete (
PTRACKSTRUCT p
);
#endif
//
// Macro expansion definition
//
// None
//
// Code
//
void
HeapCallFailed (
PCSTR Msg,
PCSTR File,
DWORD Line
)
{
CHAR Msg2[2048];
wsprintfA (Msg2, "Error in %s line %u\n\n", File, Line);
if (_tcslen(Msg) + _tcslen(Msg2) < 2025) {
strcat (Msg2, Msg);
}
strcat (Msg2, "\n\nBreak execution now?");
if (IDYES == MessageBoxA (GetFocus(), Msg2, "Heap Call Failed", MB_YESNO|MB_APPLMODAL)) {
DebugBreak ();
}
}
#ifdef DEBUG
SIZE_T
DebugHeapValidatePtr (
HANDLE hHeap,
PCVOID CallerPtr,
PCSTR File,
DWORD Line
)
{
SIZE_T rc;
EnterCriticalSection (&g_MemAllocCs);
rc = pDebugHeapValidatePtrUnlocked (hHeap, CallerPtr, File, Line);
LeaveCriticalSection (&g_MemAllocCs);
return rc;
}
SIZE_T
pDebugHeapValidatePtrUnlocked (
HANDLE hHeap,
PCVOID CallerPtr,
PCSTR File,
DWORD Line
)
{
SIZE_T size;
PCVOID RealPtr;
SIZE_T SizeAdjust;
SizeAdjust = sizeof (TRACKSTRUCT);
RealPtr = (PCVOID) ((PBYTE) CallerPtr - SizeAdjust);
if (IsBadWritePtr ((PBYTE) RealPtr - 8, 8)) {
CHAR BadPtrMsg[256];
//lint --e(572)
wsprintfA (
BadPtrMsg,
"Attempt to free memory at 0x%08x%08x. This address is not valid.",
(DWORD)((UBINT)CallerPtr >> 32),
(DWORD)(UBINT)CallerPtr
);
HeapCallFailed (BadPtrMsg, File, Line);
return (SIZE_T)INVALID_PTR;
}
size = HeapSize (hHeap, 0, RealPtr);
if (size == (SIZE_T)-1) {
CHAR BadPtrMsg[256];
//lint --e(572)
wsprintfA (
BadPtrMsg,
"Attempt to free memory at 0x%08x%08x. "
"This address is not the start of a memory block.",
(DWORD)((UBINT)CallerPtr >> 32),
(DWORD)(UBINT)CallerPtr
);
HeapCallFailed (BadPtrMsg, File, Line);
return (SIZE_T)INVALID_PTR;
}
return size;
}
PVOID
DebugHeapAlloc (
PCSTR File,
DWORD Line,
HANDLE hHeap,
DWORD Flags,
SIZE_T BytesToAlloc
)
{
PVOID RealPtr;
PVOID ReturnPtr = NULL;
DWORD SizeAdjust;
DWORD TrackStructSize;
DWORD OrgError;
EnterCriticalSection (&g_MemAllocCs);
__try {
OrgError = GetLastError();
SizeAdjust = sizeof (TRACKSTRUCT) + sizeof (DWORD);
TrackStructSize = sizeof (TRACKSTRUCT);
if (!HeapValidate (hHeap, 0, NULL)) {
HeapCallFailed ("Heap is corrupt!", File, Line);
// we want to go on, most likely we will AV shortly
}
RealPtr = SafeHeapAlloc(hHeap, Flags, BytesToAlloc + SizeAdjust);
if (RealPtr) {
g_HeapAllocs++;
g_TotalBytesAllocated += HeapSize (hHeap, 0, RealPtr);
g_MaxBytesInUse = max (g_MaxBytesInUse, g_TotalBytesAllocated);
pTrackInsert (File, Line, BytesToAlloc, (PTRACKSTRUCT) RealPtr);
*((PDWORD) ((PBYTE) RealPtr + TrackStructSize + BytesToAlloc)) = TRAIL_SIG;
}
else {
g_HeapAllocFails++;
}
if (RealPtr) {
ReturnPtr = (PVOID) ((PBYTE) RealPtr + TrackStructSize);
}
if (ReturnPtr && !(Flags & HEAP_ZERO_MEMORY)) {
FillMemory (ReturnPtr, BytesToAlloc, 0xAA);
}
if (RealPtr) {
SetLastError(OrgError);
}
}
__finally {
LeaveCriticalSection (&g_MemAllocCs);
}
return ReturnPtr;
}
PVOID
DebugHeapReAlloc (
PCSTR File,
DWORD Line,
HANDLE hHeap,
DWORD Flags,
PCVOID CallerPtr,
SIZE_T BytesToAlloc
)
{
UBINT lastSize;
PVOID NewRealPtr;
PCVOID RealPtr;
PVOID ReturnPtr = NULL;
DWORD SizeAdjust;
DWORD OrgError;
DWORD TrackStructSize;
SIZE_T OrgSize;
PTRACKSTRUCT pts = NULL;
EnterCriticalSection (&g_MemAllocCs);
__try {
OrgError = GetLastError();
SizeAdjust = sizeof (TRACKSTRUCT) + sizeof (DWORD);
TrackStructSize = sizeof (TRACKSTRUCT);
RealPtr = (PCVOID) ((PBYTE) CallerPtr - TrackStructSize);
pts = (PTRACKSTRUCT) RealPtr;
OrgSize = pts->Size;
if (!HeapValidate (hHeap, 0, NULL)) {
HeapCallFailed ("Heap is corrupt!", File, Line);
// we want to go on, most likely we will AV shortly
}
lastSize = pDebugHeapValidatePtrUnlocked (hHeap, CallerPtr, File, Line);
if (lastSize == (UBINT)INVALID_PTR) {
// we want to go on, most likely we will AV shortly
}
pTrackDelete (pts);
NewRealPtr = SafeHeapReAlloc (hHeap, Flags, (PVOID) RealPtr, BytesToAlloc + SizeAdjust);
if (NewRealPtr) {
g_HeapReAllocs++;
g_TotalBytesAllocated -= lastSize;
g_TotalBytesAllocated += HeapSize (hHeap, 0, NewRealPtr);
g_MaxBytesInUse = max (g_MaxBytesInUse, g_TotalBytesAllocated);
pTrackInsert (File, Line, BytesToAlloc, (PTRACKSTRUCT) NewRealPtr);
*((PDWORD) ((PBYTE) NewRealPtr + TrackStructSize + BytesToAlloc)) = TRAIL_SIG;
}
else {
g_HeapReAllocFails++;
// Put original address back in
pTrackInsert (
pts->File,
pts->Line,
pts->Size,
pts
);
}
if (NewRealPtr) {
ReturnPtr = (PVOID) ((PBYTE) NewRealPtr + TrackStructSize);
}
if (ReturnPtr && BytesToAlloc > OrgSize && !(Flags & HEAP_ZERO_MEMORY)) {
FillMemory ((PBYTE) ReturnPtr + OrgSize, BytesToAlloc - OrgSize, 0xAA);
}
if (ReturnPtr) {
SetLastError (OrgError);
}
}
__finally {
LeaveCriticalSection (&g_MemAllocCs);
}
return ReturnPtr;
}
BOOL
DebugHeapFree (
PCSTR File,
DWORD Line,
HANDLE hHeap,
DWORD Flags,
PCVOID CallerPtr
)
{
UBINT size;
PCVOID RealPtr;
DWORD SizeAdjust;
DWORD OrgError;
BOOL Result = FALSE;
PTRACKSTRUCT pts = NULL;
EnterCriticalSection (&g_MemAllocCs);
__try {
OrgError = GetLastError();
SizeAdjust = sizeof (TRACKSTRUCT);
RealPtr = (PCVOID) ((PBYTE) CallerPtr - SizeAdjust);
pts = (PTRACKSTRUCT) RealPtr;
if (*((PDWORD) ((PBYTE) CallerPtr + pts->Size)) != TRAIL_SIG) {
HeapCallFailed ("Heap tag was overwritten!", File, Line);
__leave;
}
if (!HeapValidate (hHeap, 0, NULL)) {
HeapCallFailed ("Heap is corrupt!", File, Line);
g_HeapFreeFails++;
__leave;
}
size = pDebugHeapValidatePtrUnlocked (hHeap, CallerPtr, File, Line);
if (size == (UBINT)INVALID_PTR) {
g_HeapFreeFails++;
__leave;
}
pTrackDelete ((PTRACKSTRUCT) RealPtr);
if (!HeapFree (hHeap, Flags, (PVOID) RealPtr)) {
CHAR BadPtrMsg[256];
wsprintfA (
BadPtrMsg,
"Attempt to free memory at 0x%08x with flags 0x%08x. "
"HeapFree() failed.",
CallerPtr,
Flags
);
HeapCallFailed (BadPtrMsg, File, Line);
g_HeapFreeFails++;
__leave;
}
g_HeapFrees++;
if (g_TotalBytesAllocated < size) {
DEBUGMSG ((DBG_WARNING, "Total bytes allocated is less than amount being freed. "
"This suggests memory corruption."));
g_TotalBytesAllocated = 0;
} else {
g_TotalBytesAllocated -= size;
}
SetLastError (OrgError);
Result = TRUE;
}
__finally {
LeaveCriticalSection (&g_MemAllocCs);
}
return Result;
}
VOID
DumpHeapStats (
VOID
)
{
CHAR OutputMsg[4096];
wsprintfA (OutputMsg,
"Bytes currently allocated: %u\n"
"Peak bytes allocated: %u\n"
"Allocation count: %u\n"
"Reallocation count: %u\n"
"Free count: %u\n",
g_TotalBytesAllocated,
g_MaxBytesInUse,
g_HeapAllocs,
g_HeapReAllocs,
g_HeapFrees
);
if (g_HeapAllocFails) {
wsprintfA (strchr (OutputMsg, 0),
"***Allocation failures: %u\n",
g_HeapAllocFails);
}
if (g_HeapReAllocFails) {
wsprintfA (strchr (OutputMsg, 0),
"***Reallocation failures: %u\n",
g_HeapReAllocFails);
}
if (g_HeapFreeFails) {
wsprintfA (strchr (OutputMsg, 0),
"***Free failures: %u\n",
g_HeapFreeFails);
}
DEBUGMSG ((DBG_STATS, "%s", OutputMsg));
#ifdef CONSOLE
printf ("%s", OutputMsg);
#endif // #ifndef CONSOLE
}
void
DebugHeapCheck (
PCSTR File,
DWORD Line,
HANDLE hHeap
)
{
EnterCriticalSection (&g_MemAllocCs);
if (!HeapValidate (hHeap, 0, NULL)) {
HeapCallFailed ("HeapCheck failed: Heap is corrupt!", File, Line);
}
LeaveCriticalSection (&g_MemAllocCs);
}
#endif
PVOID
ReuseAlloc (
HANDLE Heap,
PVOID OldPtr,
DWORD SizeNeeded
)
{
DWORD CurrentSize;
PVOID Ptr = NULL;
UINT AllocAdjustment = sizeof(DWORD);
//
// HeapSize is not very good, so while it may look good, don't
// use it.
//
#ifdef DEBUG
AllocAdjustment += sizeof (DWORD);
#endif
if (!OldPtr) {
Ptr = MemAlloc (Heap, 0, SizeNeeded + AllocAdjustment);
} else {
CurrentSize = *REUSE_SIZE_PTR(OldPtr);
#ifdef DEBUG
if (*REUSE_TAG_PTR(OldPtr) != 0x10a28a70) {
DEBUGMSG ((DBG_WHOOPS, "MemReuse detected corruption!"));
Ptr = MemAlloc (Heap, 0, SizeNeeded + AllocAdjustment);
} else
#endif
if (SizeNeeded > CurrentSize) {
SizeNeeded += 1024 - (SizeNeeded & 1023);
Ptr = MemReAlloc (Heap, 0, REUSE_SIZE_PTR(OldPtr), SizeNeeded + AllocAdjustment);
OldPtr = NULL;
}
}
if (Ptr) {
*((PDWORD) Ptr) = SizeNeeded;
Ptr = (PVOID) ((PBYTE) Ptr + sizeof (DWORD));
#ifdef DEBUG
*REUSE_TAG_PTR(Ptr) = 0x10a28a70;
#endif
}
return Ptr ? Ptr : OldPtr;
}
VOID
ReuseFree (
HANDLE Heap,
PVOID Ptr
)
{
if (Ptr) {
MemFree (Heap, 0, REUSE_SIZE_PTR(Ptr));
}
}
VOID
SetOutOfMemoryParent (
HWND hwnd
)
{
g_OutOfMemoryParentWnd = hwnd;
}
VOID
OutOfMemory_Terminate (
VOID
)
{
if (!g_OutOfMemoryString || !g_OutOfMemoryString[0]) {
return;
}
MessageBoxA (
g_OutOfMemoryParentWnd,
g_OutOfMemoryString,
NULL,
MB_OK|MB_ICONHAND|MB_SYSTEMMODAL|MB_SETFOREGROUND|MB_TOPMOST
);
ExitProcess (0);
//
// Not needed, will never get here
//
// TerminateProcess (GetModuleHandle (NULL), 0);
}
VOID
pValidateBlock (
PVOID Block,
SIZE_T Size
)
/*++
Routine Description:
pValidateBlock makes sure Block is non-NULL. If it is NULL, then the user
is given a popup, unless the request size is bogus.
There are two cases for the popup.
- If g_OutOfMemoryParentWnd was set with SetOutOfMemoryParent,
then the user is asked to close other programs, and is given a retry
option.
- If there is no out of memory parent, then the user is told they
need to get more memory.
In either case, Setup is terminated. In GUI mode, Setup will be
stuck and the machine will be unbootable.
Arguments:
Block - Specifies the block to validate.
Size - Specifies the request size
Return Value:
none
--*/
{
LONG rc;
if (!Block && Size < 0x2000000) {
if (g_OutOfMemoryParentWnd) {
rc = MessageBoxA (
g_OutOfMemoryParentWnd,
g_OutOfMemoryRetry,
NULL,
MB_RETRYCANCEL|MB_ICONHAND|MB_SYSTEMMODAL|MB_SETFOREGROUND|MB_TOPMOST
);
if (rc == IDCANCEL) {
OutOfMemory_Terminate();
}
} else {
OutOfMemory_Terminate();
}
} else {
if (!Block) {
// this is serious. We want to break now and give Dr. Watson a
// chance to get our stack.
DebugBreak ();
}
}
}
PVOID
SafeHeapAlloc (
HANDLE Heap,
DWORD Flags,
SIZE_T Size
)
{
PVOID Block;
do {
Block = HeapAlloc (Heap, Flags, Size);
pValidateBlock (Block, Size);
} while (!Block);
return Block;
}
PVOID
SafeHeapReAlloc (
HANDLE Heap,
DWORD Flags,
PVOID OldBlock,
SIZE_T Size
)
{
PVOID Block;
do {
Block = HeapReAlloc (Heap, Flags, OldBlock, Size);
pValidateBlock (Block, Size);
} while (!Block);
return Block;
}
#ifdef DEBUG
VOID
pTrackInsert (
PCSTR File,
DWORD Line,
SIZE_T Size,
PTRACKSTRUCT p
)
{
p->Signature = TRACK_SIGNATURE;
if (g_TrackAlloc) {
p->File = SafeHeapAlloc (g_hHeap, 0, SizeOfStringA (File));
if (p->File) {
StringCopyA ((PSTR)p->File, File);
}
} else {
p->File = File;
p->Allocated = FALSE;
}
p->Line = Line;
p->Size = Size;
p->Comment = g_TrackComment ? SafeHeapAlloc (g_hHeap, 0, SizeOfStringA (g_TrackComment)) : NULL;
p->PrevAlloc = NULL;
p->NextAlloc = g_TrackHead;
p->CallerFile = g_TrackFile;
p->CallerLine = g_TrackLine;
if (p->Comment) {
StringCopyA (p->Comment, g_TrackComment);
}
if (g_TrackHead) {
g_TrackHead->PrevAlloc = p;
}
g_TrackHead = p;
}
VOID
pTrackDelete (
PTRACKSTRUCT p
)
{
if (p->Signature != TRACK_SIGNATURE) {
DEBUGMSG ((DBG_WARNING, "A tracking signature is invalid. "
"This suggests memory corruption."));
return;
}
if (p->PrevAlloc) {
p->PrevAlloc->NextAlloc = p->NextAlloc;
} else {
g_TrackHead = p->NextAlloc;
}
if (p->NextAlloc) {
p->NextAlloc->PrevAlloc = p->PrevAlloc;
}
if (p->Allocated) {
if (p->File) {
HeapFree (g_hHeap, 0, (PSTR)p->File);
}
}
}
VOID
DumpHeapLeaks (
VOID
)
{
CHAR LineBuf[4096];
PTRACKSTRUCT p;
BOOL BadMem = FALSE;
if (g_TrackHead) {
__try {
for (p = g_TrackHead ; p ; p = p->NextAlloc) {
__try {
if (p->Comment) {
if (p->CallerFile) {
wsprintfA (
LineBuf,
"%s line %u\n"
" %s\n"
" Caller: %s line %u",
p->File,
p->Line,
p->Comment,
p->CallerFile,
p->CallerLine
);
} else {
wsprintfA (LineBuf, "%s line %u\n %s", p->File, p->Line, p->Comment);
}
} else {
if (p->CallerFile) {
wsprintfA (
LineBuf,
"%s line %u\n"
" Caller: %s line %u\n",
p->File,
p->Line,
p->CallerFile,
p->CallerLine
);
} else {
wsprintfA (LineBuf, "(direct alloc) %s line %u", p->File, p->Line);
}
}
}
__except (TRUE) {
wsprintfA (LineBuf, "Address %Xh was freed, but not by MemFree!!", p);
BadMem = TRUE;
}
DEBUGMSG (("Leaks", "%s", LineBuf));
#ifdef CONSOLE
printf ("%s", LineBuf);
#endif
//lint --e(774)
if (BadMem) {
break;
}
}
}
__except (TRUE) {
}
}
}
/*
VOID
DumpHeapLeaks (
VOID
)
{
HANDLE File;
CHAR LineBuf[2048];
PTRACKSTRUCT p;
DWORD DontCare;
DWORD Count;
BOOL BadMem = FALSE;
CHAR TempPath[MAX_TCHAR_PATH];
CHAR memtrackLogPath[] = "?:\\memtrack.log";
GetSystemDirectory(TempPath, MAX_TCHAR_PATH);
memtrackLogPath[0] = TempPath[0];
File = CreateFileA (memtrackLogPath, GENERIC_WRITE, 0, NULL,
OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL
);
if (File != INVALID_HANDLE_VALUE) {
SetFilePointer (File, 0, NULL, FILE_END);
if (g_TrackHead) {
Count = 0;
__try {
for (p = g_TrackHead ; p ; p = p->NextAlloc) {
Count++;
__try {
if (p->Comment) {
if (p->CallerFile) {
wsprintfA (
LineBuf,
"%s line %u\r\n"
" %s\r\n"
" Caller: %s line %u\r\n"
"\r\n",
p->File,
p->Line,
p->Comment,
p->CallerFile,
p->CallerLine
);
} else {
wsprintfA (LineBuf, "%s line %u\r\n %s\r\n\r\n", p->File, p->Line, p->Comment);
}
} else {
if (p->CallerFile) {
wsprintfA (
LineBuf,
"%s line %u\r\n"
" Caller: %s line %u\r\n"
"\r\n",
p->File,
p->Line,
p->CallerFile,
p->CallerLine
);
} else {
wsprintfA (LineBuf, "(direct alloc) %s line %u\r\n\r\n", p->File, p->Line);
}
}
}
__except (TRUE) {
wsprintfA (LineBuf, "Address %Xh was freed, but not by MemFree!!\r\n", p);
BadMem = TRUE;
}
WriteFile (File, LineBuf, (DWORD)ByteCountA (LineBuf), &DontCare, NULL);
//lint --e(774)
if (BadMem) {
break;
}
}
}
__except (TRUE) {
}
wsprintfA (LineBuf, "\r\n%i item%s allocated but not freed.\r\n\r\n", Count, Count == 1 ? "":"s");
WriteFile (File, LineBuf, (DWORD)ByteCountA (LineBuf), &DontCare, NULL);
}
wsprintfA (
LineBuf,
"Bytes currently allocated: %u\r\n"
"Peak bytes allocated: %u\r\n"
"Allocation count: %u\r\n"
"Reallocation count: %u\r\n"
"Free count: %u\r\n\r\n\r\n\r\n\r\n",
g_TotalBytesAllocated,
g_MaxBytesInUse,
g_HeapAllocs,
g_HeapReAllocs,
g_HeapFrees
);
WriteFile (File, LineBuf, (DWORD)ByteCountA (LineBuf), &DontCare, NULL);
CloseHandle (File);
}
}
*/
#endif