Changeset 21644 in vbox for trunk/src/VBox/Additions/WINNT

Timestamp:

Jul 16, 2009 11:54:31 AM (16 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

50176

Message:

Register guest allocated memory (locked & executable) for patching purposes.

Location:

trunk/src/VBox/Additions/WINNT/VBoxGuest

Files:

• VBoxGuest.cpp (modified) (2 diffs)
• VBoxGuest_Internal.h (modified) (2 diffs)

trunk/src/VBox/Additions/WINNT/VBoxGuest/VBoxGuest.cpp

@@ -1740 +1740 @@
     }
 
+    /* Allocate locked executable memory that can be used for patching guest code. */
+    {
+        VMMDevReqPatchMemory *req = NULL;
+        int rc = VbglGRAlloc ((VMMDevRequestHeader **)&req, sizeof (VMMDevReqPatchMemory), VMMDevReq_SetPatchMemory);
+        if (RT_SUCCESS(rc))
+        {
+            req->cbPatchMem = VMMDEV_GUEST_DEFAULT_PATCHMEM_SIZE;
+
+            rc = RTR0MemObjAllocPage(&pDevExt->PatchMemObj, req->cbPatchMem, true /* executable. */);
+            if (RT_SUCCESS(rc))
+            {
+                req->pPatchMem = (RTGCPTR)RTR0MemObjAddress(pDevExt->PatchMemObj);
+
+                rc = VbglGRPerform (&req->header);
+                if (RT_FAILURE(rc) || RT_FAILURE(req->header.rc))
+                {
+                    dprintf(("VBoxGuest::reserveHypervisorMemory: VMMDevReq_SetPatchMemory error!"
+                                "rc = %d, VMMDev rc = %Rrc\n", rc, req->header.rc));
+                    RTR0MemObjFree(pDevExt->PatchMemObj, true);
+                    pDevExt->PatchMemObj = NULL;
+                }
+            }
+            else
+            {
+                dprintf(("VBoxGuest::reserveHypervisorMemory: RTR0MemObjAllocPage failed with rc %d\n", rc));
+            }
+            VbglGRFree (&req->header);
+        }
+    }
     return;
 }
@@ -1750 +1779 @@
 VOID unreserveHypervisorMemory(PVBOXGUESTDEVEXT pDevExt)
 {
+    /* Remove the locked executable memory range that can be used for patching guest code. */
+    if (pDevExt->PatchMemObj)
+    {
+        VMMDevReqPatchMemory *req = NULL;
+        int rc = VbglGRAlloc ((VMMDevRequestHeader **)&req, sizeof (VMMDevReqPatchMemory), VMMDevReq_ClearPatchMemory);
+        if (RT_SUCCESS(rc))
+        {
+            req->cbPatchMem = (uint32_t)RTR0MemObjSize(pDevExt->PatchMemObj);
+            req->pPatchMem  = (RTGCPTR)RTR0MemObjAddress(pDevExt->PatchMemObj);
+
+            rc = VbglGRPerform (&req->header);
+            if (RT_FAILURE(rc) || RT_FAILURE(req->header.rc))
+            {
+                dprintf(("VBoxGuest::reserveHypervisorMemory: VMMDevReq_ClearPatchMemory error!"
+                            "rc = %d, VMMDev rc = %Rrc\n", rc, req->header.rc));
+                /* We intentially leak the memory object here as there still could 
+                 * be references to it!!!
+                 */
+            }
+            else
+            {
+                RTR0MemObjFree(pDevExt->PatchMemObj, true);
+            }
+        }
+    }
+
     VMMDevReqHypervisorInfo *req = NULL;
 

trunk/src/VBox/Additions/WINNT/VBoxGuest/VBoxGuest_Internal.h

@@ -53 +53 @@
 
 #include <iprt/spinlock.h>
+#include <iprt/memobj.h>
 
 #include <VBox/VMMDev.h>
@@ -181 +182 @@
     ULONG hypervisorMappingSize;
 
+    /* Patch memory object. */
+    RTR0MEMOBJ PatchMemObj;
+
     /* Physical address and length of VMMDev memory */
     PHYSICAL_ADDRESS memoryAddress;