Paul Gofman : ntdll: Fill NumberOfPhysicalPages field in user shared data area.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 9689d0774027bffec13cd77d10baa7bbda8f3d91
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=9689d0774027bffec13cd77d1...

Author: Paul Gofman 
Date:   Thu Feb 13 14:11:24 2020 +0300

ntdll: Fill NumberOfPhysicalPages field in user shared data area.

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=48387
Signed-off-by: Paul Gofman 
Signed-off-by: Alexandre Julliard

---

dlls/ntdll/tests/virtual.c | 14 +++++++++++++-
 dlls/ntdll/thread.c        |  4 ++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/dlls/ntdll/tests/virtual.c b/dlls/ntdll/tests/virtual.c
index 31eb66c143..683a5e7bdf 100644
--- a/dlls/ntdll/tests/virtual.c
+++ b/dlls/ntdll/tests/virtual.c
@@ -25,6 +25,7 @@
 #include "windef.h"
 #include "winternl.h"
 #include "wine/test.h"
+#include "ddk/wdm.h"

static unsigned int page_size;

@@ -33,6 +34,8 @@ static NTSTATUS (WINAPI *pRtlFreeUserStack)(void *);
 static BOOL (WINAPI *pIsWow64Process)(HANDLE, PBOOL);
 static const BOOL is_win64 = sizeof(void*) != sizeof(int);

+static SYSTEM_BASIC_INFORMATION sbi;
+
 static HANDLE create_target_process(const char *arg)
 {
     char **argv;
@@ -513,9 +516,17 @@ static void test_NtMapViewOfSection(void)
     CloseHandle(process);
 }

+static void test_user_shared_data(void)
+{
+    const KSHARED_USER_DATA *user_shared_data = (void *)0x7ffe0000;
+
+    ok(user_shared_data->NumberOfPhysicalPages == sbi.MmNumberOfPhysicalPages,
+            "Got number of physical pages %#x, expected %#x.\n",
+            user_shared_data->NumberOfPhysicalPages, sbi.MmNumberOfPhysicalPages);
+}
+
 START_TEST(virtual)
 {
-    SYSTEM_BASIC_INFORMATION sbi;
     HMODULE mod;

int argc;
@@ -546,4 +557,5 @@ START_TEST(virtual)
     test_NtAllocateVirtualMemory();
     test_RtlCreateUserStack();
     test_NtMapViewOfSection();
+    test_user_shared_data();
 }
diff --git a/dlls/ntdll/thread.c b/dlls/ntdll/thread.c
index f993dd314c..9c9e5f6055 100644
--- a/dlls/ntdll/thread.c
+++ b/dlls/ntdll/thread.c
@@ -222,6 +222,7 @@ static void set_process_name( int argc, char *argv[] )
  */
 TEB *thread_init(void)
 {
+    SYSTEM_BASIC_INFORMATION sbi;
     TEB *teb;
     void *addr;
     SIZE_T size;
@@ -310,6 +311,9 @@ TEB *thread_init(void)
     user_shared_data->TickCountMultiplier = 1 << 24;
     fill_cpu_info();

+    virtual_get_system_info( &sbi );
+    user_shared_data->NumberOfPhysicalPages = sbi.MmNumberOfPhysicalPages;
+
     return teb;
 }