mirror of https://github.com/lianthony/NT4.0
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.
149 lines
2.2 KiB
149 lines
2.2 KiB
/*++
|
|
|
|
Copyright (c) 1990 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
MEMORY.C
|
|
|
|
Abstract:
|
|
|
|
This file contains the routines that deal with memory management.
|
|
|
|
Author:
|
|
|
|
Rajen Shah (rajens) 12-Jul-1991
|
|
|
|
[Environment:]
|
|
|
|
User Mode - Win32, except for NTSTATUS returned by some functions.
|
|
|
|
Revision History:
|
|
|
|
|
|
--*/
|
|
|
|
//
|
|
// INCLUDES
|
|
//
|
|
|
|
#include <eventp.h>
|
|
|
|
//
|
|
// Implement my own tail-checking, since the system code requires a
|
|
// debugging build
|
|
//
|
|
|
|
//#define TAIL_CHECKING
|
|
#ifdef TAIL_CHECKING
|
|
#define CHECK_HEAP_TAIL_SIZE 16
|
|
#define CHECK_HEAP_TAIL_FILL 0xAB
|
|
#endif
|
|
|
|
|
|
PVOID
|
|
ElfpAllocateBuffer (
|
|
ULONG Size
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Allocate a buffer of the given size, and return the pointer in BufPtr.
|
|
|
|
|
|
Arguments:
|
|
|
|
|
|
|
|
Return Value:
|
|
|
|
Pointer to allocated buffer (or NULL).
|
|
|
|
Note:
|
|
|
|
|
|
--*/
|
|
{
|
|
PVOID BufPtr;
|
|
|
|
#ifdef TAIL_CHECKING
|
|
//
|
|
// Keep the offset of the pattern (so we don't have to have internal
|
|
// knowledge about the granularity of the heap block) and copy a
|
|
// known pattern after the end of the user's block
|
|
//
|
|
BufPtr = (PVOID *) MIDL_user_allocate ( Size
|
|
+ CHECK_HEAP_TAIL_SIZE + sizeof(DWORD));
|
|
*((PDWORD)BufPtr) = Size + sizeof(DWORD);
|
|
(PBYTE) BufPtr += sizeof(DWORD);
|
|
RtlFillMemory((PBYTE)BufPtr + Size,
|
|
CHECK_HEAP_TAIL_SIZE,
|
|
CHECK_HEAP_TAIL_FILL);
|
|
#else
|
|
|
|
BufPtr = (PVOID *) MIDL_user_allocate ( Size );
|
|
|
|
#endif
|
|
|
|
return (BufPtr);
|
|
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
ElfpFreeBuffer (
|
|
PVOID BufPtr)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Frees a buffer previously allocated by AllocateBuffer.
|
|
|
|
|
|
Arguments:
|
|
|
|
Pointer to buffer.
|
|
|
|
Return Value:
|
|
|
|
NOTHING
|
|
|
|
Note:
|
|
|
|
|
|
--*/
|
|
{
|
|
|
|
#ifdef TAIL_CHECKING
|
|
{
|
|
DWORD i;
|
|
PBYTE pb;
|
|
|
|
//
|
|
// Back up to real start of block
|
|
//
|
|
|
|
(PBYTE)BufPtr -= sizeof(DWORD);
|
|
i = *((PDWORD)BufPtr);
|
|
pb = (PBYTE)BufPtr + i;
|
|
|
|
for (i = 0; i < CHECK_HEAP_TAIL_SIZE ; i++, pb++) {
|
|
if (*pb != CHECK_HEAP_TAIL_FILL) {
|
|
ElfDbgPrint(("[ELF] Heap has been corrupted at 0x%x\n",
|
|
BufPtr));
|
|
// Make it access violate
|
|
pb = (PBYTE) 0;
|
|
*pb = 1;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
MIDL_user_free ( BufPtr );
|
|
return;
|
|
|
|
}
|