Changeset 876 in vbox for trunk/src/VBox/HostDrivers/Support/linux/SUPDrv-linux.c

Timestamp:

Feb 13, 2007 2:53:17 PM (18 years ago)

Author:

vboxsync

Message:

protect locked pages against fork()

File:

• trunk/src/VBox/HostDrivers/Support/linux/SUPDrv-linux.c (modified) (5 diffs)

trunk/src/VBox/HostDrivers/Support/linux/SUPDrv-linux.c

@@ -724 +724 @@
     unsigned    cPages = pMem->cb >> PAGE_SHIFT;
     unsigned long pv = (unsigned long)pMem->pvR3;
+    struct vm_area_struct **papVMAs;
 
     /*
@@ -730 +731 @@
     papPages = vmalloc(cPages * sizeof(*papPages));
     if (!papPages)
+        return SUPDRV_ERR_NO_MEMORY;
+
+    /*
+     * Allocate the VMA pointer array.
+     */
+    papVMAs = vmalloc(cPages * sizeof(*papVMAs));
+    if (!papVMAs)
         return SUPDRV_ERR_NO_MEMORY;
 
@@ -743 +751 @@
                         0,                      /* force. */
                         papPages,               /* Page array. */
-                        NULL);                  /* vmas */
+                        papVMAs);               /* vmas */
     if (rc != cPages)
     {
@@ -759 +767 @@
 
     /*
-     * Get addresses.
+     * Get addresses, protect against fork()
      */
     for (iPage = 0; iPage < cPages; iPage++)
@@ -765 +773 @@
         paPages[iPage].Phys = page_to_phys(papPages[iPage]);
         paPages[iPage].uReserved = 0;
-    }
+        papVMAs[iPage]->vm_flags |= VM_DONTCOPY;
+    }
+
+    vfree(papVMAs);
 
     dprintf2(("supdrvOSLockMemOne: pvR3=%p cb=%d papPages=%p\n",