[ros-diffs] [jgardou] 71391: [NTOS/MM] - Implement MmProbeAndLockProcessPages

Author: jgardou
Date: Mon May 23 20:33:03 2016
New Revision: 71391

URL: http://svn.reactos.org/svn/reactos?rev=71391&view=rev
Log:
[NTOS/MM]
 - Implement MmProbeAndLockProcessPages

Modified:
    branches/TransitionPte/ntoskrnl/mm/ARM3/mdlsup.c

Modified: branches/TransitionPte/ntoskrnl/mm/ARM3/mdlsup.c
URL: http://svn.reactos.org/svn/reactos/branches/TransitionPte/ntoskrnl/mm/ARM3/m...
==============================================================================
--- branches/TransitionPte/ntoskrnl/mm/ARM3/mdlsup.c	[iso-8859-1] (original)
+++ branches/TransitionPte/ntoskrnl/mm/ARM3/mdlsup.c	[iso-8859-1] Mon May 23 20:33:03 2016
@@ -1326,16 +1326,56 @@
 }

/*
- * @unimplemented
+ * @implemented
  */
 VOID
 NTAPI
-MmProbeAndLockProcessPages(IN OUT PMDL MemoryDescriptorList,
+MmProbeAndLockProcessPages(IN OUT PMDL Mdl,
                            IN PEPROCESS Process,
                            IN KPROCESSOR_MODE AccessMode,
                            IN LOCK_OPERATION Operation)
 {
-    UNIMPLEMENTED;
+	PVOID Address;
+	NTSTATUS Status = STATUS_SUCCESS;
+
+	BOOLEAN Attach;
+	KAPC_STATE ApcState;
+
+	DPRINT("Probing MDL: %p for process %p\n", Mdl, Process);
+	//
+	// Get address information, to see if we really need to attach
+	//
+	Address = (PVOID)((ULONG_PTR)Mdl->StartVa + Mdl->ByteOffset);
+
+	Attach = ((Process != PsGetCurrentProcess()) && (Address <= (PVOID)MM_USER_PROBE_ADDRESS));
+	if (Attach)
+	{
+		KeStackAttachProcess(&Process->Pcb, &ApcState);
+	}
+
+	_SEH2_TRY
+	{
+		MmProbeAndLockPages(Mdl, AccessMode, Operation);
+	}
+    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+    {
+        //
+        // Oops :(
+        //
+        Status = _SEH2_GetExceptionCode();
+    }
+    _SEH2_END;
+
+	if (Attach)
+	{
+		KeUnstackDetachProcess(&ApcState);
+	}
+
+	if (!NT_SUCCESS(Status))
+	{
+	    /* If the operation failed while we were attached, forward it to the caller */
+	    ExRaiseStatus(Status);
+	}
 }