MiQueryVirtualMemory(): created because ZwQueryVirtualMemory() didn't work for me...
[reactos.git] / reactos / ntoskrnl / mm / virtual.c
index 45df45d..3970987 100644 (file)
-/* $Id: virtual.c,v 1.32 2000/07/07 10:30:56 dwelch Exp $
+/*
+ *  ReactOS kernel
+ *  Copyright (C) 1998, 1999, 2000, 2001 ReactOS Team
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+/* $Id: virtual.c,v 1.86 2004/12/22 05:17:44 royce Exp $
  *
- * COPYRIGHT:   See COPYING in the top directory
  * PROJECT:     ReactOS kernel
  * FILE:        ntoskrnl/mm/virtual.c
- * PURPOSE:     implementing the Virtualxxx section of the win32 api
+ * PURPOSE:     Implementing operations on virtual memory.
  * PROGRAMMER:  David Welch
- * UPDATE HISTORY:
- *              09/4/98: Created
- *              10/6/98: Corrections from Fatahi (i_fatahi@hotmail.com)
- *              30/9/98: Implemented ZwxxxVirtualMemory functions
  */
+
 /* INCLUDE *****************************************************************/
 
-#include <ddk/ntddk.h>
-#include <internal/mm.h>
-#include <internal/mmhal.h>
-#include <internal/ob.h>
-#include <internal/io.h>
-#include <internal/ps.h>
-#include <string.h>
-#include <internal/string.h>
+#include <ntoskrnl.h>
 
 #define NDEBUG
 #include <internal/debug.h>
 
-/* FUNCTIONS ****************************************************************/
+/* FUNCTIONS *****************************************************************/
 
-NTSTATUS MmWritePageVirtualMemory(PMADDRESS_SPACE AddressSpace,
-                                 PMEMORY_AREA MArea,
-                                 PVOID Address)
-{
-   SWAPENTRY se;
-   ULONG Flags;
-   PHYSICAL_ADDRESS PhysicalAddress;
-   PMDL Mdl;
-   NTSTATUS Status;
-   
-   /*
-    * FIXME: What should we do if an i/o operation is pending on
-    * this page
-    */
-   
-   /*
-    * If the memory area is readonly then there is nothing to do
-    */
-   if (MArea->Attributes & PAGE_READONLY ||
-       MArea->Attributes & PAGE_EXECUTE_READ)
-     {
-       return(STATUS_SUCCESS);
-     }
-   /*
-    * Set the page to readonly. This ensures the current contents aren't
-    * modified while we are writing it to swap.
-    */
-   MmSetPageProtect(AddressSpace->Process,
-                   Address,
-                   PAGE_READONLY);
-   /*
-    * If the page isn't dirty then there is nothing to do.
-    */
-   if (!MmIsPageDirty(AddressSpace->Process, Address))
-     {
-       MmSetPageProtect(AddressSpace->Process,
-                        Address,
-                        MArea->Attributes);
-       return(STATUS_SUCCESS); 
-     }
-   PhysicalAddress = MmGetPhysicalAddress(Address);
-   /*
-    * If we haven't already allocated a swap entry for this page
-    * then allocate one
-    */
-   if ((se = MmGetSavedSwapEntryPage((PVOID)PhysicalAddress.u.LowPart)) != 0)
-     {
-       se = MmAllocSwapPage();
-       if (se == 0)
-         {
-            MmSetPageProtect(AddressSpace->Process,
-                             Address,
-                             MArea->Attributes);
-            return(STATUS_UNSUCCESSFUL);
-         }
-       MmSetSavedSwapEntryPage((PVOID)PhysicalAddress.u.LowPart, se);
-     }
-   /*
-    * Set the flags so other threads will know what we are doing
-    */
-   Flags = MmGetFlagsPage((PVOID)PhysicalAddress.u.LowPart);
-   Flags = Flags | MM_PHYSICAL_PAGE_MPW_PENDING;
-   MmSetFlagsPage((PVOID)PhysicalAddress.u.LowPart, Flags);
-   /*
-    * Build an mdl to hold the page for writeout
-    */
-   Mdl = MmCreateMdl(NULL, NULL, PAGESIZE);
-   MmBuildMdlFromPages(Mdl, (PULONG)&PhysicalAddress.u.LowPart);
-   /*
-    * Unlock the address space and write out the page to swap.
-    */
-   MmUnlockAddressSpace(AddressSpace);
-   Status = MmWriteToSwapPage(se, Mdl);
-   /*
-    * Cleanup 
-    */
-   MmLockAddressSpace(AddressSpace);
-   Flags = MmGetFlagsPage((PVOID)PhysicalAddress.u.LowPart);
-   Flags = Flags & (~MM_PHYSICAL_PAGE_MPW_PENDING);
-   MmSetFlagsPage((PVOID)PhysicalAddress.u.LowPart,Flags);
-   /*
-    * If we successfully wrote the page then reset the dirty bit
-    */
-   if (NT_SUCCESS(Status))
-     {
-       MmSetCleanPage(AddressSpace->Process, Address);
-     }
-   return(Status);
-}
-
-
-ULONG MmPageOutVirtualMemory(PMADDRESS_SPACE AddressSpace,
-                            PMEMORY_AREA MemoryArea,
-                            PVOID Address,
-                            PBOOLEAN Ul)
-{
-   PHYSICAL_ADDRESS PhysicalAddress;
-   
-   if ((MemoryArea->Attributes & PAGE_READONLY) ||
-       (MemoryArea->Attributes & PAGE_EXECUTE_READ) ||
-       !MmIsPageDirty(PsGetCurrentProcess(), Address))
-     {
-       PhysicalAddress = MmGetPhysicalAddress(Address);
-       
-       MmRemovePageFromWorkingSet(AddressSpace->Process,
-                                  Address);
-       MmSetPage(PsGetCurrentProcess(),
-                 Address,
-                 0,
-                 0);
-       MmDereferencePage((PVOID)PhysicalAddress.u.LowPart);
-       *Ul = TRUE;
-       return(1);
-     }
-   *Ul = FALSE;
-   return(0);     
-}
-
-NTSTATUS MmNotPresentFaultVirtualMemory(PMADDRESS_SPACE AddressSpace,
-                                       MEMORY_AREA* MemoryArea, 
-                                       PVOID Address)
+NTSTATUS STDCALL
+NtFlushVirtualMemory(IN HANDLE ProcessHandle,
+                     IN PVOID BaseAddress,
+                     IN ULONG NumberOfBytesToFlush,
+                     OUT PULONG NumberOfBytesFlushed OPTIONAL)
 /*
- * FUNCTION: Move data into memory to satisfy a page not present fault
+ * FUNCTION: Flushes virtual memory to file
  * ARGUMENTS:
- *      AddressSpace = Address space within which the fault occurred
- *      MemoryArea = The memory area within which the fault occurred
- *      Address = The absolute address of fault
+ *        ProcessHandle = Points to the process that allocated the virtual
+ *                        memory
+ *        BaseAddress = Points to the memory address
+ *        NumberOfBytesToFlush = Limits the range to flush,
+ *        NumberOfBytesFlushed = Actual number of bytes flushed
  * RETURNS: Status
- * NOTES: This function is called with the address space lock held.
  */
 {
-   PVOID Page;
-   NTSTATUS Status;
-   
-   if (MmIsPagePresent(NULL, Address))
-     { 
-       return(STATUS_SUCCESS);
-     }
-   
-   Page = MmAllocPage(0);
-   while (Page == NULL)
-     {
-       MmUnlockAddressSpace(AddressSpace);
-       MmWaitForFreePages();
-       MmLockAddressSpace(AddressSpace);
-       if (MmIsPagePresent(NULL, Address))
-         {
-            return(STATUS_SUCCESS);
-         }
-       Page = MmAllocPage(0);
-     }
-   Status = MmCreatePageTable(Address);
-   while (!NT_SUCCESS(Status))
-     {
-       MmUnlockAddressSpace(AddressSpace);
-       MmWaitForFreePages();
-       MmLockAddressSpace(AddressSpace);
-       if (MmIsPagePresent(NULL, Address))
-         {
-            MmDereferencePage(Page);
-            return(STATUS_SUCCESS);
-         }
-       Status = MmCreatePageTable(Address);     
-     }
-   MmAddPageToWorkingSet(PsGetCurrentProcess(), Address);
-   MmSetPage(PsGetCurrentProcess(),
-            Address,
-            MemoryArea->Attributes,
-            (ULONG)Page);
-   
-   return(STATUS_SUCCESS);
+   UNIMPLEMENTED;
+   return(STATUS_NOT_IMPLEMENTED);
 }
 
-NTSTATUS STDCALL NtAllocateVirtualMemory(IN    HANDLE  ProcessHandle,
-                                        IN OUT PVOID   * BaseAddress,
-                                        IN     ULONG   ZeroBits,
-                                        IN OUT PULONG  RegionSize,
-                                        IN     ULONG   AllocationType, 
-                                        IN     ULONG   Protect)
-/*
- * FUNCTION: Allocates a block of virtual memory in the process address space
- * ARGUMENTS:
- *      ProcessHandle = The handle of the process which owns the virtual memory
- *      BaseAddress   = A pointer to the virtual memory allocated. If you 
- *                      supply a non zero value the system will try to 
- *                      allocate the memory at the address supplied. It round 
- *                      it down to a multiple  of the page size.
- *      ZeroBits  = (OPTIONAL) You can specify the number of high order bits 
- *                      that must be zero, ensuring that the memory will be 
- *                      allocated at a address below a certain value.
- *      RegionSize = The number of bytes to allocate
- *      AllocationType = Indicates the type of virtual memory you like to 
- *                       allocated, can be one of the values : MEM_COMMIT, 
- *                       MEM_RESERVE, MEM_RESET, MEM_TOP_DOWN
- *      Protect = Indicates the protection type of the pages allocated, can be
- *                a combination of PAGE_READONLY, PAGE_READWRITE, 
- *                PAGE_EXECUTE_READ, PAGE_EXECUTE_READWRITE, PAGE_GUARD, 
- *                PAGE_NOACCESS
- * REMARKS:
- *       This function maps to the win32 VirtualAllocEx. Virtual memory is 
- *       process based so the  protocol starts with a ProcessHandle. I 
- *       splitted the functionality of obtaining the actual address and 
- *       specifying the start address in two parameters ( BaseAddress and 
- *       StartAddress ) The NumberOfBytesAllocated specify the range and the 
- *       AllocationType and ProctectionType map to the other two parameters.
- * RETURNS: Status
- */
+
+NTSTATUS STDCALL
+MiLockVirtualMemory(HANDLE ProcessHandle,
+  PVOID BaseAddress,
+  ULONG NumberOfBytesToLock,
+  PULONG NumberOfBytesLocked,
+  PObReferenceObjectByHandle pObReferenceObjectByHandle,
+  PMmCreateMdl pMmCreateMdl,
+  PObDereferenceObject pObDereferenceObject,
+  PMmProbeAndLockPages pMmProbeAndLockPages,
+  PExFreePool pExFreePool)
 {
-   PEPROCESS Process;
-   MEMORY_AREA* MemoryArea;
-   ULONG Type;
-   NTSTATUS Status;
-   PMADDRESS_SPACE AddressSpace;
-   
-   DPRINT("NtAllocateVirtualMemory(ProcessHandle %x, *BaseAddress %x, "
-         "ZeroBits %d, *RegionSize %x, AllocationType %x, Protect %x)\n",
-         ProcessHandle,*BaseAddress,ZeroBits,*RegionSize,AllocationType,
-         Protect);
-   
-   Status = ObReferenceObjectByHandle(ProcessHandle,
-                                     PROCESS_VM_OPERATION,
-                                     NULL,
-                                     UserMode,
-                                     (PVOID*)(&Process),
-                                     NULL);
-   if (Status != STATUS_SUCCESS)
-     {
-       DPRINT("NtAllocateVirtualMemory() = %x\n",Status);
-       return(Status);
-     }
-   
-   if (AllocationType & MEM_RESERVE)
-     {
-       Type = MEMORY_AREA_RESERVE;
-     }
-   else
-     {
-       Type = MEMORY_AREA_COMMIT;
-     }
-   
-   AddressSpace = &Process->AddressSpace;
-   MmLockAddressSpace(AddressSpace);
-   
-   if ((*BaseAddress) != 0)
-     {
-       MemoryArea = MmOpenMemoryAreaByAddress(&Process->AddressSpace,
-                                              *BaseAddress);
-       
-       if (MemoryArea != NULL)
-         {
-            if (MemoryArea->BaseAddress == (*BaseAddress) &&
-                MemoryArea->Length == *RegionSize)
-              {
-                 MemoryArea->Type = Type;
-                 MemoryArea->Attributes = Protect;
-                 DPRINT("*BaseAddress %x\n",*BaseAddress);
-                 MmUnlockAddressSpace(AddressSpace);
-                 ObDereferenceObject(Process);
-                 return(STATUS_SUCCESS);
-              }
-            
-            MemoryArea = MmSplitMemoryArea(Process,
-                                           &Process->AddressSpace,
-                                           MemoryArea,
-                                           *BaseAddress,
-                                           *RegionSize,
-                                           Type,
-                                           Protect);
-            DPRINT("*BaseAddress %x\n",*BaseAddress);
-            /* FIXME: Reserve/dereserve swap pages */
-            MmUnlockAddressSpace(AddressSpace);
-            ObDereferenceObject(Process);
-            return(STATUS_SUCCESS);
-         }
-     }
-   
-   Status = MmCreateMemoryArea(Process,
-                              &Process->AddressSpace,
-                              Type,
-                              BaseAddress,
-                              *RegionSize,
-                              Protect,
-                              &MemoryArea);
-   
-   if (Status != STATUS_SUCCESS)
-     {
-       DPRINT("NtAllocateVirtualMemory() = %x\n",Status);
-       MmUnlockAddressSpace(AddressSpace);     
-       ObDereferenceObject(Process);
-       return(Status);
-     }
-   
-   DPRINT("*BaseAddress %x\n",*BaseAddress);
-   if ((AllocationType & MEM_COMMIT) &&
-       ((Protect & PAGE_READWRITE) ||
-       (Protect & PAGE_EXECUTE_READWRITE)))
-     {
-       MmReserveSwapPages(PAGE_ROUND_UP((*RegionSize)));
-     }
-   MmUnlockAddressSpace(AddressSpace);
-   ObDereferenceObject(Process);
-   return(STATUS_SUCCESS);
+  PEPROCESS Process;
+  NTSTATUS Status;
+  PMDL Mdl;
+
+  Status = pObReferenceObjectByHandle(ProcessHandle,
+    PROCESS_VM_WRITE,
+    NULL,
+    UserMode,
+    (PVOID*)(&Process),
+    NULL);
+  if (!NT_SUCCESS(Status))
+    return(Status);
+
+  Mdl = pMmCreateMdl(NULL,
+    BaseAddress,
+    NumberOfBytesToLock);
+  if (Mdl == NULL)
+    {
+      pObDereferenceObject(Process);
+      return(STATUS_NO_MEMORY);
+    }
+
+  pMmProbeAndLockPages(Mdl,
+    UserMode,
+    IoWriteAccess);
+
+  pExFreePool(Mdl);
+
+  pObDereferenceObject(Process);
+
+  *NumberOfBytesLocked = NumberOfBytesToLock;
+  return(STATUS_SUCCESS);
 }
 
 
-NTSTATUS STDCALL NtFlushVirtualMemory(IN       HANDLE  ProcessHandle,
-                                     IN        PVOID   BaseAddress,
-                                     IN        ULONG   NumberOfBytesToFlush,
-                                     OUT PULONG NumberOfBytesFlushed OPTIONAL)
-/*
- * FUNCTION: Flushes virtual memory to file
- * ARGUMENTS:
- *        ProcessHandle = Points to the process that allocated the virtual 
- *                        memory
- *        BaseAddress = Points to the memory address
- *        NumberOfBytesToFlush = Limits the range to flush,
- *        NumberOfBytesFlushed = Actual number of bytes flushed
- * RETURNS: Status 
- */
+NTSTATUS STDCALL
+NtLockVirtualMemory(HANDLE ProcessHandle,
+  PVOID BaseAddress,
+  ULONG NumberOfBytesToLock,
+  PULONG NumberOfBytesLocked)
 {
-       UNIMPLEMENTED;
+  DPRINT("NtLockVirtualMemory(ProcessHandle %x, BaseAddress %x, "
+    "NumberOfBytesToLock %d, NumberOfBytesLocked %x)\n",
+    ProcessHandle,
+    BaseAddress,
+    NumberOfBytesToLock,
+    NumberOfBytesLocked);
+
+  return MiLockVirtualMemory(ProcessHandle,
+    BaseAddress,
+    NumberOfBytesToLock,
+    NumberOfBytesLocked,
+    ObReferenceObjectByHandle,
+    MmCreateMdl,
+    ObfDereferenceObject,
+    MmProbeAndLockPages,
+    ExFreePool);
 }
 
 
-NTSTATUS STDCALL NtFreeVirtualMemory(IN        HANDLE  ProcessHandle,
-                                    IN PVOID   * BaseAddress,  
-                                    IN PULONG  RegionSize,     
-                                    IN ULONG   FreeType)
-/*
- * FUNCTION: Frees a range of virtual memory
- * ARGUMENTS:
- *        ProcessHandle = Points to the process that allocated the virtual 
- *                        memory
- *        BaseAddress = Points to the memory address, rounded down to a 
- *                      multiple of the pagesize
- *        RegionSize = Limits the range to free, rounded up to a multiple of 
- *                     the paging size
- *        FreeType = Can be one of the values:  MEM_DECOMMIT, or MEM_RELEASE
- * RETURNS: Status 
- */
+NTSTATUS FASTCALL
+MiQueryVirtualMemory (IN HANDLE ProcessHandle,
+                      IN PVOID Address,
+                      IN CINT VirtualMemoryInformationClass,
+                      OUT PVOID VirtualMemoryInformation,
+                      IN ULONG Length,
+                      OUT PULONG ResultLength)
 {
-   MEMORY_AREA* MemoryArea;
    NTSTATUS Status;
    PEPROCESS Process;
+   MEMORY_AREA* MemoryArea;
    PMADDRESS_SPACE AddressSpace;
-   ULONG i;
-   
-   DPRINT("NtFreeVirtualMemory(ProcessHandle %x, *BaseAddress %x, "
-         "*RegionSize %x, FreeType %x)\n",ProcessHandle,*BaseAddress,
-         *RegionSize,FreeType);
-                                
-   
-   Status = ObReferenceObjectByHandle(ProcessHandle,
-                                     PROCESS_VM_OPERATION,
-                                     PsProcessType,
-                                     UserMode,
-                                     (PVOID*)(&Process),
-                                     NULL);
-   if (Status != STATUS_SUCCESS)
-     {
-       return(Status);
-     }
-   
-   AddressSpace = &Process->AddressSpace;
-   
+
+   if (Address < (PVOID)KERNEL_BASE)
+   {
+      Status = ObReferenceObjectByHandle(ProcessHandle,
+                                         PROCESS_QUERY_INFORMATION,
+                                         NULL,
+                                         UserMode,
+                                         (PVOID*)(&Process),
+                                         NULL);
+
+      if (!NT_SUCCESS(Status))
+      {
+         DPRINT("NtQueryVirtualMemory() = %x\n",Status);
+         return(Status);
+      }
+      AddressSpace = &Process->AddressSpace;
+   }
+   else
+   {
+      AddressSpace = MmGetKernelAddressSpace();
+   }
    MmLockAddressSpace(AddressSpace);
    MemoryArea = MmOpenMemoryAreaByAddress(AddressSpace,
-                                         *BaseAddress);
-   if (MemoryArea == NULL)
-     {
-       MmUnlockAddressSpace(AddressSpace);
-       ObDereferenceObject(Process);
-       return(STATUS_UNSUCCESSFUL);
-     }
-   
-   switch (FreeType)
-     {
-      case MEM_RELEASE:
-       if (MemoryArea->BaseAddress != (*BaseAddress))
-         {
-            MmUnlockAddressSpace(AddressSpace);
-            ObDereferenceObject(Process);
-            return(STATUS_UNSUCCESSFUL);
-         }
-       if ((MemoryArea->Type == MEMORY_AREA_COMMIT) &&
-           ((MemoryArea->Attributes & PAGE_READWRITE) ||
-            (MemoryArea->Attributes & PAGE_EXECUTE_READWRITE)))
-         {
-            MmDereserveSwapPages(PAGE_ROUND_UP(MemoryArea->Length));
-         }
-       
-       for (i=0; i<=(MemoryArea->Length/PAGESIZE); i++)
-         {
-            LARGE_INTEGER PhysicalAddr;
-            
-            PhysicalAddr = MmGetPhysicalAddress(MemoryArea->BaseAddress + 
-                                                (i*PAGESIZE));
-            if (PhysicalAddr.u.LowPart != 0)
+                                          Address);
+   switch(VirtualMemoryInformationClass)
+   {
+      case MemoryBasicInformation:
+         {
+           PMEMORY_BASIC_INFORMATION Info =
+               (PMEMORY_BASIC_INFORMATION)VirtualMemoryInformation;
+            if (Length != sizeof(MEMORY_BASIC_INFORMATION))
+            {
+               MmUnlockAddressSpace(AddressSpace);
+               ObDereferenceObject(Process);
+               return(STATUS_INFO_LENGTH_MISMATCH);
+            }
+
+            if (MemoryArea == NULL)
+            {
+              Info->Type = 0;
+               Info->State = MEM_FREE;
+              Info->Protect = PAGE_NOACCESS;
+              Info->AllocationProtect = 0;
+               Info->BaseAddress = (PVOID)PAGE_ROUND_DOWN(Address);
+              Info->AllocationBase = NULL;
+              Info->RegionSize = MmFindGapAtAddress(AddressSpace, Info->BaseAddress);
+               Status = STATUS_SUCCESS;
+               *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
+           }
+            else
+           {
+              switch(MemoryArea->Type)
               {
-                 MmRemovePageFromWorkingSet(AddressSpace->Process,
-                                            MemoryArea->BaseAddress +
-                                            (i*PAGESIZE));
-                 MmDereferencePage((PVOID)(ULONG)(PhysicalAddr.u.LowPart));
+                 case MEMORY_AREA_VIRTUAL_MEMORY:
+                     Status = MmQueryAnonMem(MemoryArea, Address, Info,
+                                             ResultLength);
+                    break;
+                 case MEMORY_AREA_SECTION_VIEW:
+                     Status = MmQuerySectionView(MemoryArea, Address, Info,
+                                                 ResultLength);
+                     break;
+                 case MEMORY_AREA_NO_ACCESS:
+                    Info->Type = 0;
+                     Info->State = MEM_FREE;
+                    Info->Protect = MemoryArea->Attributes;
+                    Info->AllocationProtect = MemoryArea->Attributes;
+                     Info->BaseAddress = MemoryArea->BaseAddress;
+                    Info->AllocationBase = MemoryArea->BaseAddress;
+                    Info->RegionSize = MemoryArea->Length;
+                     Status = STATUS_SUCCESS;
+                     *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
+                    break;
+                 case MEMORY_AREA_SHARED_DATA:
+                    Info->Type = 0;
+                     Info->State = MEM_COMMIT;
+                    Info->Protect = MemoryArea->Attributes;
+                    Info->AllocationProtect = MemoryArea->Attributes;
+                     Info->BaseAddress = MemoryArea->BaseAddress;
+                    Info->AllocationBase = MemoryArea->BaseAddress;
+                    Info->RegionSize = MemoryArea->Length;
+                     Status = STATUS_SUCCESS;
+                     *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
+                    break;
+                 case MEMORY_AREA_SYSTEM:
+                    {
+                       static int warned = 0;
+                       if ( !warned )
+                       {
+                         DPRINT1("FIXME: MEMORY_AREA_SYSTEM case incomplete (or possibly wrong) for NtQueryVirtualMemory()\n");
+                         warned = 1;
+                       }
+                    }
+                    /* FIXME - don't have a clue if this is right, but it's better than nothing */
+                    Info->Type = 0;
+                     Info->State = MEM_COMMIT;
+                    Info->Protect = MemoryArea->Attributes;
+                    Info->AllocationProtect = MemoryArea->Attributes;
+                     Info->BaseAddress = MemoryArea->BaseAddress;
+                    Info->AllocationBase = MemoryArea->BaseAddress;
+                    Info->RegionSize = MemoryArea->Length;
+                     Status = STATUS_SUCCESS;
+                     *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
+                    break;
+                 case MEMORY_AREA_KERNEL_STACK:
+                    {
+                       static int warned = 0;
+                       if ( !warned )
+                       {
+                         DPRINT1("FIXME: MEMORY_AREA_KERNEL_STACK case incomplete (or possibly wrong) for NtQueryVirtualMemory()\n");
+                         warned = 1;
+                       }
+                    }
+                    /* FIXME - don't have a clue if this is right, but it's better than nothing */
+                    Info->Type = 0;
+                     Info->State = MEM_COMMIT;
+                    Info->Protect = MemoryArea->Attributes;
+                    Info->AllocationProtect = MemoryArea->Attributes;
+                     Info->BaseAddress = MemoryArea->BaseAddress;
+                    Info->AllocationBase = MemoryArea->BaseAddress;
+                    Info->RegionSize = MemoryArea->Length;
+                     Status = STATUS_SUCCESS;
+                     *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
+                    break;
+                 default:
+                    DPRINT1("unhandled memory area type: 0x%x\n", MemoryArea->Type);
+                    Status = STATUS_UNSUCCESSFUL;
+                     *ResultLength = 0;
               }
-         }
-       
-       MmFreeMemoryArea(&Process->AddressSpace,
-                        BaseAddress,
-                        0,
-                        FALSE);
-       MmUnlockAddressSpace(AddressSpace);
-       ObDereferenceObject(Process);
-       return(STATUS_SUCCESS);
-       
-      case MEM_DECOMMIT:       
-       if ((MemoryArea->Type == MEMORY_AREA_COMMIT) &&
-           ((MemoryArea->Attributes & PAGE_READWRITE) ||
-            (MemoryArea->Attributes & PAGE_EXECUTE_READWRITE)))
-         {
-            MmDereserveSwapPages(PAGE_ROUND_UP((*RegionSize)));
-         }
-       MmSplitMemoryArea(Process,
-                         &Process->AddressSpace,
-                         MemoryArea,
-                         *BaseAddress,
-                         *RegionSize,
-                         MEMORY_AREA_RESERVE,
-                         MemoryArea->Attributes);
-       MmUnlockAddressSpace(AddressSpace);     
-       ObDereferenceObject(Process);
-       return(STATUS_SUCCESS);
-     }
+           }
+            break;
+         }
+
+      default:
+         {
+            Status = STATUS_INVALID_INFO_CLASS;
+            *ResultLength = 0;
+            break;
+         }
+   }
+
    MmUnlockAddressSpace(AddressSpace);
-   ObDereferenceObject(Process);
-   return(STATUS_NOT_IMPLEMENTED);
-}
+   if (Address < (PVOID)KERNEL_BASE)
+   {
+      ObDereferenceObject(Process);
+   }
 
+   return Status;
+}
 
-NTSTATUS STDCALL NtLockVirtualMemory(HANDLE    ProcessHandle,
-                                    PVOID      BaseAddress,
-                                    ULONG      NumberOfBytesToLock,
-                                    PULONG     NumberOfBytesLocked)
+/* (tMk 2004.II.4)
+ * FUNCTION:
+ * Called from VirtualQueryEx (lib\kernel32\mem\virtual.c)
+ *
+ */
+NTSTATUS STDCALL
+NtQueryVirtualMemory (IN HANDLE ProcessHandle,
+                      IN PVOID Address,
+                      IN CINT VirtualMemoryInformationClass,
+                      OUT PVOID VirtualMemoryInformation,
+                      IN ULONG Length,
+                      OUT PULONG UnsafeResultLength)
 {
-       UNIMPLEMENTED;
-}
+   NTSTATUS Status;
+   ULONG ResultLength = 0;
+   KPROCESSOR_MODE PrevMode;
+   union
+   {
+      MEMORY_BASIC_INFORMATION BasicInfo;
+   }
+   VirtualMemoryInfo;
 
+   DPRINT("NtQueryVirtualMemory(ProcessHandle %x, Address %x, "
+          "VirtualMemoryInformationClass %d, VirtualMemoryInformation %x, "
+          "Length %lu ResultLength %x)\n",ProcessHandle,Address,
+          VirtualMemoryInformationClass,VirtualMemoryInformation,
+          Length,ResultLength);
 
-VOID MmChangeAreaProtection(PEPROCESS Process,
-                           PVOID BaseAddress,
-                           ULONG Length,
-                           ULONG Protect)
-{
-   ULONG i;
-   
-   for (i=0; i<(Length/PAGESIZE); i++)
-     {
-       if (MmIsPagePresent(Process, BaseAddress + (i*PAGESIZE)))
-         {
-            MmSetPageProtect(Process,
-                             BaseAddress + (i*PAGESIZE), 
-                             Protect);
-         }
-     }
+   PrevMode =  ExGetPreviousMode();
+
+   if (PrevMode == UserMode && Address >= (PVOID)KERNEL_BASE)
+   {
+      DPRINT1("Invalid parameter\n");
+      return STATUS_INVALID_PARAMETER;
+   }
+
+   Status = MiQueryVirtualMemory ( ProcessHandle,
+       Address,
+       VirtualMemoryInformationClass,
+       &VirtualMemoryInfo,
+       Length,
+       &ResultLength );
+
+   if (NT_SUCCESS(Status) && ResultLength > 0)
+   {
+      Status = MmCopyToCaller(VirtualMemoryInformation, &VirtualMemoryInfo, ResultLength);
+      if (!NT_SUCCESS(Status))
+      {
+         ResultLength = 0;
+      }
+   }
+
+   if (UnsafeResultLength != NULL)
+   {
+      MmCopyToCaller(UnsafeResultLength, &ResultLength, sizeof(ULONG));
+   }
+   return(Status);
 }
 
 
-NTSTATUS STDCALL NtProtectVirtualMemory(IN     HANDLE  ProcessHandle,
-                                       IN      PVOID   BaseAddress,
-                                       IN      ULONG   NumberOfBytesToProtect,
-                                       IN      ULONG   NewAccessProtection,
-                                       OUT     PULONG  OldAccessProtection)
+/* (tMk 2004.II.5)
+ * FUNCTION:
+ * Called from VirtualProtectEx (lib\kernel32\mem\virtual.c)
+ *
+ */
+NTSTATUS STDCALL
+NtProtectVirtualMemory(IN HANDLE ProcessHandle,
+                       IN PVOID *UnsafeBaseAddress,
+                       IN ULONG *UnsafeNumberOfBytesToProtect,
+                       IN ULONG NewAccessProtection,
+                       OUT PULONG UnsafeOldAccessProtection)
 {
    PMEMORY_AREA MemoryArea;
    PEPROCESS Process;
    NTSTATUS Status;
    PMADDRESS_SPACE AddressSpace;
-   
+   ULONG OldAccessProtection;
+   PVOID BaseAddress;
+   ULONG NumberOfBytesToProtect;
+
+   Status = MmCopyFromCaller(&BaseAddress, UnsafeBaseAddress, sizeof(PVOID));
+   if (!NT_SUCCESS(Status))
+      return Status;
+   Status = MmCopyFromCaller(&NumberOfBytesToProtect, UnsafeNumberOfBytesToProtect, sizeof(ULONG));
+   if (!NT_SUCCESS(Status))
+      return Status;
+
+   // (tMk 2004.II.5) in Microsoft SDK I read:
+   // 'if this parameter is NULL or does not point to a valid variable, the function fails'
+   if(UnsafeOldAccessProtection == NULL)
+   {
+      return(STATUS_INVALID_PARAMETER);
+   }
+
+   NumberOfBytesToProtect =
+      PAGE_ROUND_UP(BaseAddress + NumberOfBytesToProtect) -
+      PAGE_ROUND_DOWN(BaseAddress);
+   BaseAddress = (PVOID)PAGE_ROUND_DOWN(BaseAddress);
+
    Status = ObReferenceObjectByHandle(ProcessHandle,
-                                     PROCESS_VM_OPERATION,
-                                     PsProcessType,
-                                     UserMode,
-                                     (PVOID*)(&Process),
-                                     NULL);
-   if (Status != STATUS_SUCCESS)
-     {
-       DPRINT("NtProtectVirtualMemory() = %x\n",Status);
-       return(Status);
-     }
+                                      PROCESS_VM_OPERATION,
+                                      PsProcessType,
+                                      UserMode,
+                                      (PVOID*)(&Process),
+                                      NULL);
+   if (!NT_SUCCESS(Status))
+   {
+      DPRINT("NtProtectVirtualMemory() = %x\n",Status);
+      return(Status);
+   }
 
    AddressSpace = &Process->AddressSpace;
-   
+
    MmLockAddressSpace(AddressSpace);
    MemoryArea = MmOpenMemoryAreaByAddress(AddressSpace,
-                                         BaseAddress);
+                                          BaseAddress);
    if (MemoryArea == NULL)
-     {
-       DPRINT("NtProtectVirtualMemory() = %x\n",STATUS_UNSUCCESSFUL);
-       MmUnlockAddressSpace(AddressSpace);
-       ObDereferenceObject(Process);
-       return(STATUS_UNSUCCESSFUL);
-     }
-
-   *OldAccessProtection = MemoryArea->Attributes;
-
-   if (MemoryArea->BaseAddress == BaseAddress &&
-       MemoryArea->Length == NumberOfBytesToProtect)
-     {
-       MemoryArea->Attributes = NewAccessProtection;   
-     }
-   else
-     {
-       MemoryArea = MmSplitMemoryArea(Process,
-                                      &Process->AddressSpace,
-                                      MemoryArea,
-                                      BaseAddress,
-                                      NumberOfBytesToProtect,
-                                      MemoryArea->Type,
-                                      NewAccessProtection);
-     }
-   MmChangeAreaProtection(Process,
-                         BaseAddress,
-                         NumberOfBytesToProtect,
-                         NewAccessProtection);
+   {
+      MmUnlockAddressSpace(AddressSpace);
+      ObDereferenceObject(Process);
+      return(STATUS_UNSUCCESSFUL);
+   }
+
+   if (MemoryArea->Type == MEMORY_AREA_VIRTUAL_MEMORY)
+   {
+      Status = MmProtectAnonMem(AddressSpace, MemoryArea, BaseAddress,
+                                NumberOfBytesToProtect, NewAccessProtection,
+                                &OldAccessProtection);
+   }
+   else if (MemoryArea->Type == MEMORY_AREA_SECTION_VIEW)
+   {
+      Status = MmProtectSectionView(AddressSpace, MemoryArea, BaseAddress,
+                                    NumberOfBytesToProtect,
+                                    NewAccessProtection,
+                                    &OldAccessProtection);
+   }
+
    MmUnlockAddressSpace(AddressSpace);
    ObDereferenceObject(Process);
-   return(STATUS_SUCCESS);
+
+   MmCopyToCaller(UnsafeOldAccessProtection, &OldAccessProtection, sizeof(ULONG));
+   MmCopyToCaller(UnsafeBaseAddress, &BaseAddress, sizeof(PVOID));
+   MmCopyToCaller(UnsafeNumberOfBytesToProtect, &NumberOfBytesToProtect, sizeof(ULONG));
+
+   return(Status);
 }
 
 
-NTSTATUS STDCALL NtQueryVirtualMemory (IN HANDLE ProcessHandle,
-                                      IN PVOID Address,
-                                      IN CINT VirtualMemoryInformationClass,
-                                      OUT PVOID VirtualMemoryInformation,
-                                      IN ULONG Length,
-                                      OUT PULONG ResultLength)
+/* (tMk 2004.II.05)
+ * FUNCTION:
+ * Called from ReadProcessMemory (lib\kernel32\mem\procmem.c) and KlInitPeb(lib\kernel32\process\create.c)
+ *
+ * NOTE: This function will be correct if MmProbeAndLockPages() would be fully IMPLEMENTED.
+ */
+NTSTATUS STDCALL
+NtReadVirtualMemory(IN HANDLE ProcessHandle,
+                    IN PVOID BaseAddress,
+                    OUT PVOID Buffer,
+                    IN ULONG NumberOfBytesToRead,
+                    OUT PULONG NumberOfBytesRead OPTIONAL)
 {
    NTSTATUS Status;
-   PEPROCESS Process;
-   MEMORY_AREA* MemoryArea;
+   PMDL Mdl;
+   PVOID SystemAddress;
+   PEPROCESS Process, CurrentProcess;
 
-   DPRINT("NtQueryVirtualMemory(ProcessHandle %x, Address %x, "
-          "VirtualMemoryInformationClass %d, VirtualMemoryInformation %x, "
-          "Length %lu ResultLength %x)\n",ProcessHandle,Address,
-          VirtualMemoryInformationClass,VirtualMemoryInformation,
-          Length,ResultLength);
 
-   switch(VirtualMemoryInformationClass)
-     {
-        case MemoryBasicInformation:
-          {
-             PMEMORY_BASIC_INFORMATION Info =
-                (PMEMORY_BASIC_INFORMATION)VirtualMemoryInformation;
-            PMADDRESS_SPACE AddressSpace;
-            
-             if (Length < sizeof(MEMORY_BASIC_INFORMATION))
-               {
-                  ObDereferenceObject(Process);
-                  return STATUS_INFO_LENGTH_MISMATCH;
-               }
-
-             if (ResultLength)
-               {
-                  *ResultLength = sizeof(MEMORY_BASIC_INFORMATION);
-               }
-
-             Status = ObReferenceObjectByHandle(ProcessHandle,
-                                                PROCESS_QUERY_INFORMATION,
-                                                NULL,
-                                                UserMode,
-                                                (PVOID*)(&Process),
-                                                NULL);
-
-             if (!NT_SUCCESS(Status))
-               {
-                  DPRINT("NtQueryVirtualMemory() = %x\n",Status);
-                  return(Status);
-               }
-
-            AddressSpace = &Process->AddressSpace;
-            MmLockAddressSpace(AddressSpace);
-             MemoryArea = MmOpenMemoryAreaByAddress(AddressSpace,
-                                                    Address);
-
-             if (MemoryArea == NULL)
-               {
-                  Info->State = MEM_FREE;
-                  DPRINT("Virtual memory at %p is free.\n", Address);
-                 MmUnlockAddressSpace(AddressSpace);
-                  ObDereferenceObject(Process);
-                  return (STATUS_SUCCESS);
-               }
-
-             if (MemoryArea->Type == MEMORY_AREA_COMMIT)
-               {
-                  Info->State = MEM_COMMIT;
-               }
-             else
-               {
-                  Info->State = MEM_RESERVE;
-               }
-
-             Info->BaseAddress = MemoryArea->BaseAddress;
-             Info->RegionSize  = MemoryArea->Length;
-
-             DPRINT("BaseAddress %p, RegionSize %x State %x\n",
-                    Info->BaseAddress, Info->RegionSize, Info->State);
-            
-            MmUnlockAddressSpace(AddressSpace);
-             ObDereferenceObject(Process);
-             return STATUS_SUCCESS;
-          }
-          break;
-     }
-
-   return STATUS_INVALID_INFO_CLASS;
-}
+   DPRINT("NtReadVirtualMemory(ProcessHandle %x, BaseAddress %x, "
+          "Buffer %x, NumberOfBytesToRead %d)\n",ProcessHandle,BaseAddress,
+          Buffer,NumberOfBytesToRead);
+
+   Status = ObReferenceObjectByHandle(ProcessHandle,
+                                      PROCESS_VM_WRITE,
+                                      NULL,
+                                      UserMode,
+                                      (PVOID*)(&Process),
+                                      NULL);
+   if (!NT_SUCCESS(Status))
+   {
+      return(Status);
+   }
+
+   CurrentProcess = PsGetCurrentProcess();
+
+   if (Process == CurrentProcess)
+   {
+      memcpy(Buffer, BaseAddress, NumberOfBytesToRead);
+   }
+   else
+   {
+      Mdl = MmCreateMdl(NULL,
+                        Buffer,
+                        NumberOfBytesToRead);
+      if(Mdl == NULL)
+      {
+         ObDereferenceObject(Process);
+         return(STATUS_NO_MEMORY);
+      }
+      MmProbeAndLockPages(Mdl,
+                          UserMode,
+                          IoWriteAccess);
+
+      KeAttachProcess(&Process->Pcb);
+
+      SystemAddress = MmGetSystemAddressForMdl(Mdl);
+      memcpy(SystemAddress, BaseAddress, NumberOfBytesToRead);
+
+      KeDetachProcess();
+
+      if (Mdl->MappedSystemVa != NULL)
+      {
+         MmUnmapLockedPages(Mdl->MappedSystemVa, Mdl);
+      }
+      MmUnlockPages(Mdl);
+      ExFreePool(Mdl);
+   }
+
+   ObDereferenceObject(Process);
 
+   if (NumberOfBytesRead)
+      *NumberOfBytesRead = NumberOfBytesToRead;
+   return(STATUS_SUCCESS);
+}
 
-NTSTATUS STDCALL NtReadVirtualMemory(IN        HANDLE  ProcessHandle,
-                                    IN PVOID   BaseAddress,
-                                    OUT        PVOID   Buffer,
-                                    IN ULONG   NumberOfBytesToRead,
-                                    OUT        PULONG  NumberOfBytesRead)
+/* (tMk 2004.II.05)
+ * FUNCTION:  THIS function doesn't make a sense...
+ * Called from VirtualUnlock (lib\kernel32\mem\virtual.c)
+ */
+NTSTATUS STDCALL
+NtUnlockVirtualMemory(HANDLE ProcessHandle,
+                      PVOID BaseAddress,
+                      ULONG NumberOfBytesToUnlock,
+                      PULONG NumberOfBytesUnlocked OPTIONAL)
 {
+   // AG [08-20-03] : I have *no* idea if this is correct, I just used the
+   // other functions as a template and made a few intelligent guesses...
+
    NTSTATUS Status;
    PMDL Mdl;
-   PVOID SystemAddress;
    PEPROCESS Process;
-   
-   DPRINT("NtReadVirtualMemory(ProcessHandle %x, BaseAddress %x, "
-           "Buffer %x, NumberOfBytesToRead %d)\n",ProcessHandle,BaseAddress,
-           Buffer,NumberOfBytesToRead);
-   
+
+   DPRINT("NtUnlockVirtualMemory(ProcessHandle %x, BaseAddress %x, "
+          "NumberOfBytesToUnlock %d), NumberOfBytesUnlocked %x\n",ProcessHandle,BaseAddress,
+          NumberOfBytesToUnlock, NumberOfBytesUnlocked);
+
    Status = ObReferenceObjectByHandle(ProcessHandle,
-                                     PROCESS_VM_WRITE,
-                                     NULL,
-                                     UserMode,
-                                     (PVOID*)(&Process),
-                                     NULL);
-   if (Status != STATUS_SUCCESS)
-     {
-       return(Status);
-     }
-   
-   Mdl = MmCreateMdl(NULL, 
-                    Buffer,
-                    NumberOfBytesToRead);
-   MmProbeAndLockPages(Mdl,
-                      UserMode,
-                      IoWriteAccess);
-   
-   KeAttachProcess(Process);
-   
-   SystemAddress = MmGetSystemAddressForMdl(Mdl);
-   memcpy(SystemAddress, BaseAddress, NumberOfBytesToRead);
-   
-   KeDetachProcess();
-   
+                                      PROCESS_VM_WRITE,
+                                      NULL,
+                                      UserMode,
+                                      (PVOID*)(&Process),
+                                      NULL);
+   if (!NT_SUCCESS(Status))
+   {
+      return(Status);
+   }
+
+   Mdl = MmCreateMdl(NULL,
+                     BaseAddress,
+                     NumberOfBytesToUnlock);
+   if(Mdl == NULL)
+   {
+      ObDereferenceObject(Process);
+      return(STATUS_NO_MEMORY);
+   }
+
    ObDereferenceObject(Process);
-   
-   *NumberOfBytesRead = NumberOfBytesToRead;
-   return(STATUS_SUCCESS);
-}
 
+   if (Mdl->MappedSystemVa != NULL)
+   {
+      MmUnmapLockedPages(Mdl->MappedSystemVa, Mdl);
+   }
+   MmUnlockPages(Mdl);
+   ExFreePool(Mdl);
 
-NTSTATUS STDCALL NtUnlockVirtualMemory(HANDLE  ProcessHandle,
-                                      PVOID    BaseAddress,
-                                      ULONG    NumberOfBytesToUnlock,
-                                      PULONG NumberOfBytesUnlocked OPTIONAL)
-{
-       UNIMPLEMENTED;
+   *NumberOfBytesUnlocked = NumberOfBytesToUnlock;
+
+   return(STATUS_SUCCESS);
 }
 
 
-NTSTATUS STDCALL NtWriteVirtualMemory(IN       HANDLE  ProcessHandle,
-                                     IN        PVOID   BaseAddress,
-                                     IN        PVOID   Buffer,
-                                     IN        ULONG   NumberOfBytesToWrite,
-                                     OUT       PULONG  NumberOfBytesWritten)
+/* (tMk 2004.II.05)
+ * FUNCTION:
+ * Called from WriteProcessMemory (lib\kernel32\mem\procmem.c) and KlInitPeb(lib\kernel32\process\create.c)
+ *
+ * NOTE: This function will be correct if MmProbeAndLockPages() would be fully IMPLEMENTED.
+ */
+NTSTATUS STDCALL
+NtWriteVirtualMemory(IN HANDLE ProcessHandle,
+                     IN PVOID BaseAddress,
+                     IN PVOID Buffer,
+                     IN ULONG NumberOfBytesToWrite,
+                     OUT PULONG NumberOfBytesWritten)
 {
    NTSTATUS Status;
    PMDL Mdl;
    PVOID SystemAddress;
    PEPROCESS Process;
-   
+
    DPRINT("NtWriteVirtualMemory(ProcessHandle %x, BaseAddress %x, "
-           "Buffer %x, NumberOfBytesToWrite %d)\n",ProcessHandle,BaseAddress,
-           Buffer,NumberOfBytesToWrite);
-   
+          "Buffer %x, NumberOfBytesToWrite %d)\n",ProcessHandle,BaseAddress,
+          Buffer,NumberOfBytesToWrite);
+
    Status = ObReferenceObjectByHandle(ProcessHandle,
-                                     PROCESS_VM_WRITE,
-                                     NULL,
-                                     UserMode,
-                                     (PVOID*)(&Process),
-                                     NULL);
-   if (Status != STATUS_SUCCESS)
-     {
-       return(Status);
-     }
-   
-   Mdl = MmCreateMdl(NULL, 
-                    Buffer,
-                    NumberOfBytesToWrite);
-   MmProbeAndLockPages(Mdl,
-                      UserMode,
-                      IoReadAccess);
-   
-   KeAttachProcess(Process);
-   
-   DPRINT("Attached to process copying memory\n");
-   
-   SystemAddress = MmGetSystemAddressForMdl(Mdl);
-   memcpy(BaseAddress, SystemAddress, NumberOfBytesToWrite);
-   
-   DPRINT("Done copy\n");
-   
-   KeDetachProcess();
-   
+                                      PROCESS_VM_WRITE,
+                                      NULL,
+                                      UserMode,
+                                      (PVOID*)(&Process),
+                                      NULL);
+   if (!NT_SUCCESS(Status))
+   {
+      return(Status);
+   }
+
+   if (Process == PsGetCurrentProcess())
+   {
+      memcpy(BaseAddress, Buffer, NumberOfBytesToWrite);
+   }
+   else
+   {
+      Mdl = MmCreateMdl(NULL,
+                        Buffer,
+                        NumberOfBytesToWrite);
+      MmProbeAndLockPages(Mdl,
+                          UserMode,
+                          IoReadAccess);
+      if(Mdl == NULL)
+      {
+         ObDereferenceObject(Process);
+         return(STATUS_NO_MEMORY);
+      }
+      KeAttachProcess(&Process->Pcb);
+
+      SystemAddress = MmGetSystemAddressForMdl(Mdl);
+      memcpy(BaseAddress, SystemAddress, NumberOfBytesToWrite);
+
+      KeDetachProcess();
+
+      if (Mdl->MappedSystemVa != NULL)
+      {
+         MmUnmapLockedPages(Mdl->MappedSystemVa, Mdl);
+      }
+      MmUnlockPages(Mdl);
+      ExFreePool(Mdl);
+   }
+
    ObDereferenceObject(Process);
-   
+
    *NumberOfBytesWritten = NumberOfBytesToWrite;
-   
-   DPRINT("Finished NtWriteVirtualMemory()\n");
-   
+
    return(STATUS_SUCCESS);
 }
 
+/*
+ * @unimplemented
+ */
 
-DWORD
+PVOID
 STDCALL
-MmSecureVirtualMemory (
-       DWORD   Unknown0,
-       DWORD   Unknown1,
-       DWORD   Unknown2
-       )
+MmGetVirtualForPhysical (
+    IN PHYSICAL_ADDRESS PhysicalAddress
+    )
 {
        UNIMPLEMENTED;
        return 0;
 }
 
+/* FUNCTION:
+ * Called from EngSecureMem (subsys\win32k\eng\mem.c)
+ * @unimplemented
+ */
+PVOID STDCALL
+MmSecureVirtualMemory (PVOID  Address,
+                       SIZE_T Length,
+                       ULONG  Mode)
+{
+   /* Only works for user space */
+   if (MmHighestUserAddress < Address)
+   {
+      return NULL;
+   }
 
-VOID
-STDCALL
-MmUnsecureVirtualMemory (
-       DWORD   Unknown0
-       )
+   UNIMPLEMENTED;
+
+   return 0;
+}
+
+
+/* FUNCTION:
+ * Called from EngUnsecureMem (subsys\win32k\eng\mem.c)
+ * @unimplemented
+ */
+VOID STDCALL
+MmUnsecureVirtualMemory(PVOID SecureMem)
 {
-       UNIMPLEMENTED;
+   if (NULL == SecureMem)
+   {
+      return;
+   }
+
+   UNIMPLEMENTED;
+}
+
+
+/*
+ * @implemented
+ */
+VOID STDCALL
+ProbeForRead (IN PVOID Address,
+              IN ULONG Length,
+              IN ULONG Alignment)
+{
+   ASSERT(Alignment ==1 || Alignment == 2 || Alignment == 4 || Alignment == 8);
+
+   if (Length == 0)
+      return;
+
+   if (((ULONG_PTR)Address & (Alignment - 1)) != 0)
+   {
+      ExRaiseStatus (STATUS_DATATYPE_MISALIGNMENT);
+   }
+   else if ((ULONG_PTR)Address + Length < (ULONG_PTR)Address ||
+            (ULONG_PTR)Address + Length > (ULONG_PTR)MmUserProbeAddress)
+   {
+      ExRaiseStatus (STATUS_ACCESS_VIOLATION);
+   }
+}
+
+
+/*
+ * @implemented
+ */
+VOID STDCALL
+ProbeForWrite (IN PVOID Address,
+               IN ULONG Length,
+               IN ULONG Alignment)
+{
+   PULONG Ptr;
+   ULONG x;
+   ULONG i;
+
+   ASSERT(Alignment ==1 || Alignment == 2 || Alignment == 4 || Alignment == 8);
+
+   if (Length == 0)
+      return;
+
+   if (((ULONG_PTR)Address & (Alignment - 1)) != 0)
+   {
+      ExRaiseStatus (STATUS_DATATYPE_MISALIGNMENT);
+   }
+   else if ((ULONG_PTR)Address + Length < (ULONG_PTR)Address ||
+            (ULONG_PTR)Address + Length > (ULONG_PTR)MmUserProbeAddress)
+   {
+      ExRaiseStatus (STATUS_ACCESS_VIOLATION);
+   }
+
+   /* Check for accessible pages */
+   for (i = 0; i < Length; i += PAGE_SIZE)
+   {
+      Ptr = (PULONG)(((ULONG_PTR)Address & ~(PAGE_SIZE - 1)) + i);
+      x = *Ptr;
+      *Ptr = x;
+   }
 }
 
 /* EOF */