[PATCH v18 7/8] ntdll: Support mapping pages for views created with SEC_LARGE_PAGES.

[Vibhav Pant]

From: Vibhav Pant

---
 dlls/ntdll/unix/virtual.c | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/dlls/ntdll/unix/virtual.c b/dlls/ntdll/unix/virtual.c
index 4b23d9954df..5188138d28b 100644
--- a/dlls/ntdll/unix/virtual.c
+++ b/dlls/ntdll/unix/virtual.c
@@ -2082,10 +2082,16 @@ static NTSTATUS map_file_into_view( struct file_view *view, int fd, size_t start
         prot |= PROT_EXEC;
     }

+    if (vprot & SEC_LARGE_PAGES)
+    {
+        flags |= MAP_LOCKED;
+    }
+
     /* only try mmap if media is not removable (or if we require write access) */
     if (!removable || (flags & MAP_SHARED))
     {
-        if (mmap( (char *)view->base + start, size, prot, flags, fd, offset ) != MAP_FAILED)
+        ptr = mmap( (char *)view->base + start, size, prot, flags, fd, offset );
+        if (ptr != MAP_FAILED)
             goto done;

switch (errno)
@@ -2118,7 +2124,8 @@ static NTSTATUS map_file_into_view( struct file_view *view, int fd, size_t start
     }

/* Reserve the memory with an anonymous mmap */
-    ptr = anon_mmap_fixed( (char *)view->base + start, size, PROT_READ | PROT_WRITE, 0 );
+    ptr = anon_mmap_fixed( (char *)view->base + start, size, PROT_READ | PROT_WRITE,
+                           vprot & SEC_LARGE_PAGES ? MAP_LOCKED : 0 );
     if (ptr == MAP_FAILED)
     {
         ERR( "anon mmap error %s, range %p-%p\n",
@@ -2130,6 +2137,16 @@ static NTSTATUS map_file_into_view( struct file_view *view, int fd, size_t start
     if (prot != (PROT_READ|PROT_WRITE)) mprotect( ptr, size, prot );  /* Set the right protection */
 done:
     set_page_vprot( (char *)view->base + start, size, vprot );
+    if ( vprot & SEC_LARGE_PAGES && mlock( ptr, size ) != 0 )
+    {
+        switch (errno)
+        {
+            case EPERM:
+                return STATUS_ACCESS_DENIED;
+            default:
+                return STATUS_NO_MEMORY;
+        }
+    }
     return STATUS_SUCCESS;
 }

@@ -3179,6 +3196,16 @@ static unsigned int virtual_map_section( HANDLE handle, PVOID *addr_ptr, ULONG_P
     base = *addr_ptr;
     offset.QuadPart = offset_ptr ? offset_ptr->QuadPart : 0;
     if (offset.QuadPart >= full_size) return STATUS_INVALID_PARAMETER;
+
+    if (sec_flags & SEC_LARGE_PAGES)
+    {
+        if (!*size_ptr || user_shared_data->LargePageMinimum == 0 ||
+            *size_ptr % user_shared_data->LargePageMinimum != 0)
+            return STATUS_INVALID_PARAMETER;
+        if (base && ((UINT_PTR)base % user_shared_data->LargePageMinimum != 0))
+            return STATUS_INVALID_PARAMETER;
+    }
+
     if (*size_ptr)
     {
         size = *size_ptr;
@@ -3204,7 +3231,8 @@ static unsigned int virtual_map_section( HANDLE handle, PVOID *addr_ptr, ULONG_P

server_enter_uninterrupted_section( &virtual_mutex, &sigset );

-    res = map_view( &view, base, size, alloc_type, vprot, limit_low, limit_high, 0 );
+    res = map_view( &view, base, size, alloc_type, vprot, limit_low, limit_high,
+                    sec_flags & SEC_LARGE_PAGES ? ( user_shared_data->LargePageMinimum - 1 ) : 0 );
     if (res) goto done;

TRACE( "handle=%p size=%lx offset=%s\n", handle, size, wine_dbgstr_longlong(offset.QuadPart) );