mirror of
https://github.com/reactos/reactos.git
synced 2024-09-05 02:07:06 +00:00
235 lines
4.7 KiB
C
235 lines
4.7 KiB
C
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS kernel
|
|
* PURPOSE: GDI Driver Memory Management Functions
|
|
* FILE: win32ss/gdi/eng/mem.c
|
|
* PROGRAMER: Jason Filby
|
|
*/
|
|
|
|
#include <win32k.h>
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
_Must_inspect_result_
|
|
_When_(fl & FL_ZERO_MEMORY, _Ret_opt_bytecount_(cjMemSize))
|
|
_When_(!(fl & FL_ZERO_MEMORY), _Ret_opt_bytecap_(cjMemSize))
|
|
__drv_allocatesMem(Mem)
|
|
ENGAPI
|
|
PVOID
|
|
APIENTRY
|
|
EngAllocMem(
|
|
_In_ ULONG fl,
|
|
_In_ ULONG cjMemSize,
|
|
_In_ ULONG ulTag)
|
|
{
|
|
PVOID pvBaseAddress;
|
|
|
|
pvBaseAddress = ExAllocatePoolWithTag((fl & FL_NONPAGED_MEMORY) ?
|
|
NonPagedPool : PagedPool,
|
|
cjMemSize,
|
|
ulTag);
|
|
|
|
if (pvBaseAddress == NULL)
|
|
return NULL;
|
|
|
|
if (fl & FL_ZERO_MEMORY)
|
|
RtlZeroMemory(pvBaseAddress, cjMemSize);
|
|
|
|
return pvBaseAddress;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
APIENTRY
|
|
EngFreeMem(PVOID pvBaseAddress)
|
|
{
|
|
/* Windows allows to pass NULL */
|
|
if (pvBaseAddress)
|
|
{
|
|
/* Use 0 as tag, which equals a call to ExFreePool */
|
|
ExFreePoolWithTag(pvBaseAddress, 0);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
_Must_inspect_result_
|
|
_Ret_opt_bytecount_(cjMemSize)
|
|
__drv_allocatesMem(UserMem)
|
|
ENGAPI
|
|
PVOID
|
|
APIENTRY
|
|
EngAllocUserMem(
|
|
_In_ SIZE_T cjMemSize,
|
|
_In_ ULONG ulTag)
|
|
{
|
|
PVOID pvBaseAddress = NULL;
|
|
NTSTATUS Status;
|
|
|
|
Status = ZwAllocateVirtualMemory(NtCurrentProcess(),
|
|
&pvBaseAddress,
|
|
0,
|
|
&cjMemSize,
|
|
MEM_COMMIT | MEM_RESERVE,
|
|
PAGE_READWRITE);
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/* TODO: Add allocation info to AVL tree (stored inside W32PROCESS structure) */
|
|
//hSecure = EngSecureMem(pvBaseAddress, cj);
|
|
|
|
return pvBaseAddress;
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID
|
|
APIENTRY
|
|
EngFreeUserMem(PVOID pvBaseAddress)
|
|
{
|
|
SIZE_T cjSize = 0;
|
|
|
|
ZwFreeVirtualMemory(NtCurrentProcess(),
|
|
&pvBaseAddress,
|
|
&cjSize,
|
|
MEM_RELEASE);
|
|
|
|
/* TODO: Remove allocation info from AVL tree */
|
|
}
|
|
|
|
PVOID
|
|
APIENTRY
|
|
HackSecureVirtualMemory(
|
|
IN PVOID Address,
|
|
IN SIZE_T Size,
|
|
IN ULONG ProbeMode,
|
|
OUT PVOID *SafeAddress)
|
|
{
|
|
NTSTATUS Status = STATUS_SUCCESS;
|
|
PMDL pmdl;
|
|
LOCK_OPERATION Operation;
|
|
|
|
if (ProbeMode == PAGE_READONLY) Operation = IoReadAccess;
|
|
else if (ProbeMode == PAGE_READWRITE) Operation = IoModifyAccess;
|
|
else return NULL;
|
|
|
|
pmdl = IoAllocateMdl(Address, (ULONG)Size, FALSE, TRUE, NULL);
|
|
if (pmdl == NULL)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
_SEH2_TRY
|
|
{
|
|
MmProbeAndLockPages(pmdl, UserMode, Operation);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
Status = _SEH2_GetExceptionCode();
|
|
}
|
|
_SEH2_END
|
|
|
|
if (!NT_SUCCESS(Status))
|
|
{
|
|
IoFreeMdl(pmdl);
|
|
return NULL;
|
|
}
|
|
|
|
*SafeAddress = MmGetSystemAddressForMdlSafe(pmdl, NormalPagePriority);
|
|
|
|
if(!*SafeAddress)
|
|
{
|
|
MmUnlockPages(pmdl);
|
|
IoFreeMdl(pmdl);
|
|
return NULL;
|
|
}
|
|
|
|
return pmdl;
|
|
}
|
|
|
|
VOID
|
|
APIENTRY
|
|
HackUnsecureVirtualMemory(
|
|
IN PVOID SecureHandle)
|
|
{
|
|
PMDL pmdl = (PMDL)SecureHandle;
|
|
|
|
MmUnlockPages(pmdl);
|
|
IoFreeMdl(pmdl);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
HANDLE
|
|
APIENTRY
|
|
EngSecureMem(PVOID Address, ULONG Length)
|
|
{
|
|
{// HACK!!!
|
|
_SEH2_TRY
|
|
{
|
|
ProbeForWrite(Address, Length, 1);
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
_SEH2_YIELD(return NULL);
|
|
}
|
|
_SEH2_END;
|
|
return (HANDLE)-1;
|
|
}
|
|
return MmSecureVirtualMemory(Address, Length, PAGE_READWRITE);
|
|
}
|
|
|
|
HANDLE
|
|
APIENTRY
|
|
EngSecureMemForRead(PVOID Address, ULONG Length)
|
|
{
|
|
{// HACK!!!
|
|
ULONG cPages;
|
|
volatile BYTE *pjProbe;
|
|
|
|
_SEH2_TRY
|
|
{
|
|
ProbeForRead(Address, Length, 1);
|
|
cPages = ADDRESS_AND_SIZE_TO_SPAN_PAGES(Address, Length);
|
|
pjProbe = ALIGN_DOWN_POINTER_BY(Address, PAGE_SIZE);
|
|
while(cPages--)
|
|
{
|
|
/* Do a read probe */
|
|
(void)*pjProbe;
|
|
pjProbe += PAGE_SIZE;
|
|
}
|
|
}
|
|
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
|
|
{
|
|
_SEH2_YIELD(return NULL);
|
|
}
|
|
_SEH2_END;
|
|
return (HANDLE)-1;
|
|
}
|
|
return MmSecureVirtualMemory(Address, Length, PAGE_READONLY);
|
|
}
|
|
|
|
/*
|
|
* @implemented
|
|
*/
|
|
VOID APIENTRY
|
|
EngUnsecureMem(HANDLE Mem)
|
|
{
|
|
if (Mem == (HANDLE)-1) return; // HACK!!!
|
|
MmUnsecureVirtualMemory((PVOID) Mem);
|
|
}
|
|
|
|
/* EOF */
|