[PATCH v6 1/2] ntdll: Add environment to override cpuid for debugging and test.

[Bernhard Übelacker]

From: Bernhard Übelacker

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=58335
---
 dlls/ntdll/tests/virtual.c | 48 ++++++++++++++++++++++++++++++++++++++
 dlls/ntdll/unix/system.c   | 32 ++++++++++++++++++++++++-
 2 files changed, 79 insertions(+), 1 deletion(-)

diff --git a/dlls/ntdll/tests/virtual.c b/dlls/ntdll/tests/virtual.c
index c48652f0f65..2bfb800de74 100644
--- a/dlls/ntdll/tests/virtual.c
+++ b/dlls/ntdll/tests/virtual.c
@@ -3209,6 +3209,48 @@ static void test_exec_memory_writes(void)
     RtlRemoveVectoredExceptionHandler( handler );
 }

+static void test_cpuid_child(void)
+{
+#if defined(__x86_64__) && !defined(__arm64ec__)
+    HMODULE module = GetModuleHandleW( L"ntdll.dll" );
+    NTSTATUS (WINAPI *pNtClose)(HANDLE);
+    int val[] = {0xabcdef00, 0x12345678, 0xfedcba00, 0x87654321};
+
+    __asm__ __volatile__ ( "movaps %0,%%xmm6" : : "m" (val) );
+    /* just some call to get through __wine_syscall_dispatcher */
+    pNtClose = (void *)GetProcAddress( module, "NtClose" );
+    pNtClose(INVALID_HANDLE_VALUE);
+    __asm__ __volatile__ ( "movaps %%xmm6,%0" : "=m" (val) );
+
+    ok(val[0] == 0xabcdef00 && val[1] == 0x12345678 && val[2] == 0xfedcba00 && val[3] == 0x87654321,
+       "register xmm6 has unexpected values 0x%08x, 0x%08x, 0x%08x, 0x%08x\n", val[0], val[1], val[2], val[3]);
+#endif
+}
+
+static void test_cpuid(void)
+{
+#if defined(__x86_64__) && !defined(__aarch64__)
+    PROCESS_INFORMATION info;
+    STARTUPINFOA startup;
+    char **argv;
+    char path_name[MAX_PATH];
+    DWORD ret;
+
+    winetest_get_mainargs(&argv);
+    memset(&startup, 0, sizeof(startup));
+    startup.cb = sizeof(startup);
+    sprintf(path_name, "%s %s cpuid", argv[0], argv[1]);
+
+    ret = CreateProcessA(NULL, path_name, NULL, NULL, FALSE, 0, NULL, NULL, &startup, &info);
+    ok(ret, "Failed to create target process.\n");
+
+    wait_child_process(info.hProcess);
+
+    CloseHandle(info.hProcess);
+    CloseHandle(info.hThread);
+#endif
+}
+
 START_TEST(virtual)
 {
     HMODULE mod;
@@ -3224,6 +3266,11 @@ START_TEST(virtual)
             Sleep(5000); /* spawned process runs for at most 5 seconds */
             return;
         }
+        if (!strcmp(argv[2], "cpuid"))
+        {
+            test_cpuid_child();
+            return;
+        }
         return;
     }

@@ -3262,4 +3309,5 @@ START_TEST(virtual)
     test_query_region_information();
     test_query_image_information();
     test_exec_memory_writes();
+    test_cpuid();
 }
diff --git a/dlls/ntdll/unix/system.c b/dlls/ntdll/unix/system.c
index e486da40691..431d9b808b1 100644
--- a/dlls/ntdll/unix/system.c
+++ b/dlls/ntdll/unix/system.c
@@ -79,6 +79,9 @@
 #include "wine/debug.h"

WINE_DEFAULT_DEBUG_CHANNEL(ntdll);
+#if defined(__i386__) || defined(__x86_64__)
+WINE_DECLARE_DEBUG_CHANNEL(cpuid);
+#endif

#pragma pack(push,1)

@@ -321,11 +324,34 @@ void copy_xstate( XSAVE_AREA_HEADER *dst, XSAVE_AREA_HEADER *src, UINT64 mask )
     }
 }

-static inline void do_cpuid( unsigned int ax, unsigned int cx, unsigned int *p )
+static inline void do_cpuid_( unsigned int ax, unsigned int cx, unsigned int *p )
 {
     __asm__ ( "cpuid" : "=a" (p[0]), "=b" (p[1]), "=c" (p[2]), "=d" (p[3]) : "a" (ax), "c" (cx) );
 }

+static inline void do_cpuid( unsigned int ax, unsigned int cx, unsigned int *p )
+{
+    char buf[30];
+    char *override;
+
+    memset(p, 0, sizeof(*p)*4);
+    do_cpuid_(ax, cx, p);
+    TRACE_(cpuid)("cpuid returned WINE_CPUID_%08x_%08x=%08x_%08x_%08x_%08x\n",
+          ax, cx, p[0], p[1], p[2], p[3]);
+
+    snprintf(buf, sizeof(buf), "WINE_CPUID_%08x_%08x", ax, cx);
+    override = getenv(buf);
+    if (override && strlen(override) == 35)
+    {
+        sscanf(override, "%08x", &p[0]);
+        sscanf(override+9, "%08x", &p[1]);
+        sscanf(override+18, "%08x", &p[2]);
+        sscanf(override+27, "%08x", &p[3]);
+        ERR_(cpuid)("overriding with WINE_CPUID_%08x_%08x=%08x_%08x_%08x_%08x\n",
+              ax, cx, p[0], p[1], p[2], p[3]);
+    }
+}
+
 static inline UINT64 do_xgetbv( unsigned int cx )
 {
     UINT low, high;
@@ -432,6 +458,8 @@ static void init_xstate_features( XSTATE_CONFIGURATION *xstate )
     ULONG64 supported_mask;
     unsigned int i, off, regs[4];

+    TRACE_(cpuid)("init_xstate_features reached.\n");
+
     do_cpuid( 0x0000000d, 0, regs );
     TRACE( "XSAVE details %#x, %#x, %#x, %#x.\n", regs[0], regs[1], regs[2], regs[3] );
     supported_mask = ((ULONG64)regs[3] << 32) | regs[0];
@@ -479,6 +507,8 @@ void init_shared_data_cpuinfo( KUSER_SHARED_DATA *data )
     BOOLEAN *features = data->ProcessorFeatures;
     unsigned int regs[4];

+    TRACE_(cpuid)("init_shared_data_cpuinfo reached.\n");
+
     features[PF_FASTFAIL_AVAILABLE]      = TRUE;
     features[PF_COMPARE_EXCHANGE_DOUBLE] = TRUE;