From 2d4742aa934934d58bb6960bb74d71d9e08d8a24 Mon Sep 17 00:00:00 2001
From: Eric Pouech <epouech@codeweavers.com>
Date: Wed, 17 May 2023 11:15:19 +0200
Subject: [PATCH] kernel32/tests: Harden some wow64 module tests.

Showing some cases where loader should unredirect some
module paths from syswow64 to system32.

Signed-off-by: Eric Pouech <epouech@codeweavers.com>
---
 dlls/kernel32/tests/loader.c | 48 +++++++++++++++++++++++++++++++++---
 1 file changed, 44 insertions(+), 4 deletions(-)

diff --git a/dlls/kernel32/tests/loader.c b/dlls/kernel32/tests/loader.c
index 19f112efcd8..3f826537a62 100644
--- a/dlls/kernel32/tests/loader.c
+++ b/dlls/kernel32/tests/loader.c
@@ -53,6 +53,8 @@ static WORD cb_count, cb_count_sys;
 static DWORD page_size;
 static BOOL is_win64 = sizeof(void *) > sizeof(int);
 static BOOL is_wow64;
+static char system_dir[MAX_PATH];
+static char syswow_dir[MAX_PATH]; /* only available if is_wow64 */
 
 static NTSTATUS (WINAPI *pNtCreateSection)(HANDLE *, ACCESS_MASK, const OBJECT_ATTRIBUTES *,
                                            const LARGE_INTEGER *, ULONG, ULONG, HANDLE );
@@ -4047,10 +4049,18 @@ static void test_InMemoryOrderModuleList(void)
     ok(entry2 == mark2, "expected entry2 == mark2, got %p and %p\n", entry2, mark2);
 }
 
-static void test_wow64_redirection_for_dll(const char *libname)
+static BOOL is_path_made_of(const char *filename, const char *pfx, const char *sfx)
+{
+    const size_t len = strlen(pfx);
+    return !strncasecmp(filename, pfx, len) && filename[len] == '\\' &&
+        !strcasecmp(filename + len + 1, sfx);
+}
+
+static void test_wow64_redirection_for_dll(const char *libname, BOOL will_fail)
 {
     HMODULE lib;
     char buf[256];
+    const char *modname;
 
     if (!GetModuleHandleA(libname))
     {
@@ -4066,6 +4076,12 @@ static void test_wow64_redirection_for_dll(const char *libname)
             ok(lib != NULL, "Loading %s from full path should succeed with WOW64 redirection disabled\n", libname);
             if (lib)
                 FreeLibrary(lib);
+            modname = strrchr(libname, '\\');
+            modname = modname ? modname + 1 : libname;
+            todo_wine_if(will_fail)
+            ok(is_path_made_of(buf, system_dir, modname) ||
+               /* Win7 report from syswow64 */ broken(is_path_made_of(buf, syswow_dir, modname)),
+               "Unexpected loaded DLL name %s for %s\n", buf, libname);
         }
     }
     else
@@ -4077,6 +4093,9 @@ static void test_wow64_redirection_for_dll(const char *libname)
 static void test_wow64_redirection(void)
 {
     void *OldValue;
+    char buffer[MAX_PATH];
+    static const char *dlls[] = {"wlanapi.dll", "dxgi.dll", "dwrite.dll"};
+    unsigned i;
 
     if (!is_wow64)
         return;
@@ -4085,10 +4104,22 @@ static void test_wow64_redirection(void)
      * already be loaded in this process).
      */
     ok(pWow64DisableWow64FsRedirection(&OldValue), "Disabling FS redirection failed\n");
-    test_wow64_redirection_for_dll("wlanapi.dll");
-    test_wow64_redirection_for_dll("dxgi.dll");
-    test_wow64_redirection_for_dll("dwrite.dll");
+    for (i = 0; i < ARRAY_SIZE(dlls); i++)
+    {
+        test_wow64_redirection_for_dll(dlls[i], FALSE);
+        /* even absolute paths to syswow64 are loaded with path to system32 */
+        snprintf(buffer, ARRAY_SIZE(buffer), "%s\\%s", syswow_dir, dlls[i]);
+        test_wow64_redirection_for_dll(buffer, TRUE);
+    }
+
     ok(pWow64RevertWow64FsRedirection(OldValue), "Re-enabling FS redirection failed\n");
+    /* and results don't depend whether redirection is enabled or not */
+    for (i = 0; i < ARRAY_SIZE(dlls); i++)
+    {
+        test_wow64_redirection_for_dll(dlls[i], FALSE);
+        snprintf(buffer, ARRAY_SIZE(buffer), "%s\\%s", syswow_dir, dlls[i]);
+        test_wow64_redirection_for_dll(buffer, TRUE);
+    }
 }
 
 static void test_dll_file( const char *name )
@@ -4182,6 +4213,7 @@ START_TEST(loader)
     char **argv;
     HANDLE ntdll, mapping, kernel32;
     SYSTEM_INFO si;
+    DWORD len;
 
     ntdll = GetModuleHandleA("ntdll.dll");
     kernel32 = GetModuleHandleA("kernel32.dll");
@@ -4239,6 +4271,14 @@ START_TEST(loader)
         return;
     }
 
+    len = GetSystemDirectoryA(system_dir, ARRAY_SIZE(system_dir));
+    ok(len && len < ARRAY_SIZE(system_dir), "Couldn't get system directory: %lu\n", GetLastError());
+    if (is_wow64)
+    {
+        len = GetSystemWow64DirectoryA(syswow_dir, ARRAY_SIZE(syswow_dir));
+        ok(len && len < ARRAY_SIZE(syswow_dir), "Couldn't get wow directory: %lu\n", GetLastError());
+    }
+
     test_filenames();
     test_ResolveDelayLoadedAPI();
     test_ImportDescriptors();