[ros-diffs] [tkreuzer] 34651: implement HackSecureVirtualMemory as a possible workaround for missing MmSecureVirtualMemory

[tkreuzer at svn.reactos.org]

Author: tkreuzer
Date: Mon Jul 21 18:56:26 2008
New Revision: 34651

URL: http://svn.reactos.org/svn/reactos?rev=34651&view=rev
Log:
implement HackSecureVirtualMemory as a possible workaround for missing MmSecureVirtualMemory

Modified:
    trunk/reactos/subsystems/win32/win32k/eng/mem.c

Modified: trunk/reactos/subsystems/win32/win32k/eng/mem.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/subsystems/win32/win32k/eng/mem.c?rev=34651&r1=34650&r2=34651&view=diff
==============================================================================
--- trunk/reactos/subsystems/win32/win32k/eng/mem.c [iso-8859-1] (original)
+++ trunk/reactos/subsystems/win32/win32k/eng/mem.c [iso-8859-1] Mon Jul 21 18:56:26 2008
@@ -105,6 +105,69 @@
   ZwFreeVirtualMemory(NtCurrentProcess(), (PVOID *) &Header, &MemSize, MEM_RELEASE);
 }
 
+
+
+PVOID
+NTAPI
+HackSecureVirtualMemory(
+	IN PVOID Address,
+	IN SIZE_T Size,
+	IN ULONG ProbeMode,
+	OUT PVOID *SafeAddress)
+{
+	NTSTATUS Status = STATUS_SUCCESS;
+	PMDL mdl;
+	LOCK_OPERATION Operation;
+
+	if (ProbeMode == PAGE_READONLY) Operation = IoReadAccess;
+	else if (ProbeMode == PAGE_READWRITE) Operation = IoModifyAccess;
+	else return NULL;
+
+	mdl = IoAllocateMdl(Address, Size, FALSE, TRUE, NULL);
+	if (mdl == NULL)
+	{
+		return NULL;
+	}
+
+	_SEH_TRY
+	{
+		MmProbeAndLockPages(mdl, UserMode, Operation);
+	}
+	_SEH_HANDLE
+	{
+		Status = _SEH_GetExceptionCode();
+	}
+	_SEH_END
+
+	if (!NT_SUCCESS(Status))
+	{
+		IoFreeMdl(mdl);
+		return NULL;
+	}
+
+	*SafeAddress = MmGetSystemAddressForMdlSafe(mdl, NormalPagePriority);
+
+	if(!*SafeAddress)
+	{
+		MmUnlockPages(mdl);
+		IoFreeMdl(mdl);
+		return NULL;           
+	}
+
+	return mdl;
+}
+
+VOID
+NTAPI
+HackUnsecureVirtualMemory(
+	IN PVOID  SecureHandle)
+{
+	PMDL mdl = (PMDL)SecureHandle;
+
+	MmUnlockPages(mdl);
+	IoFreeMdl(mdl);  
+}
+
 /*
  * @implemented
  */