reactos/win32ss/gdi/eng/mem.c

235 lines
4.7 KiB
C
Raw Normal View History

/*
* 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 */