[PATCH v2 1/6] kernel32/tests: Test loading dll as resource or datafile with wow64 FS redirection disabled.

[Paul Gofman]

From: Paul Gofman

---
 dlls/kernel32/tests/loader.c | 45 ++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/dlls/kernel32/tests/loader.c b/dlls/kernel32/tests/loader.c
index 9ecaf438322..cd621a04ff4 100644
--- a/dlls/kernel32/tests/loader.c
+++ b/dlls/kernel32/tests/loader.c
@@ -4616,10 +4616,27 @@ static void test_wow64_redirection(void)
     char buffer[MAX_PATH];
     static const char *dlls[] = {"wlanapi.dll", "dxgi.dll", "dwrite.dll"};
     unsigned i;
+    HMODULE mod, mod_fixed, kernelbase;
+    IMAGE_NT_HEADERS *nt;
+    WORD machine;

if (!is_wow64)
         return;

+    kernelbase = GetModuleHandleW(L"kernelbase.dll");
+    nt = RtlImageNtHeader(kernelbase);
+    machine = nt->FileHeader.Machine;
+
+    ok(!GetModuleHandleA("rasapi32.dll"), "rasapi32.dll is already loaded.\n");
+
+    mod = LoadLibraryExW(L"c:\windows\system32\rasapi32.dll", 0, LOAD_LIBRARY_AS_IMAGE_RESOURCE);
+    mod_fixed = (HMODULE)((ULONG_PTR)mod & ~(ULONG_PTR)3);
+    ok(!!mod_fixed, "got NULL.\n" );
+    nt = RtlImageNtHeader(mod_fixed);
+    ok(!!nt, "got NULL.\n");
+    ok(nt->FileHeader.Machine == machine, "got wrong machine.\n");
+    FreeLibrary(mod);
+
     /* Disable FS redirection, then test loading system libraries (pick ones that shouldn't
      * already be loaded in this process).
      */
@@ -4632,6 +4649,34 @@ static void test_wow64_redirection(void)
         test_wow64_redirection_for_dll(buffer, TRUE);
     }

+    mod = LoadLibraryExW(L"c:\windows\system32\kernelbase.dll", 0, LOAD_LIBRARY_AS_IMAGE_RESOURCE);
+    ok(!!mod, "got NULL.\n" );
+    ok(mod == kernelbase, "got different modules.\n");
+    FreeLibrary(mod);
+
+    mod = LoadLibraryExW(L"c:\windows\system32\kernelbase.dll", 0, LOAD_LIBRARY_AS_DATAFILE);
+    ok(!!mod, "got NULL.\n" );
+    ok(mod == kernelbase, "got different modules.\n");
+    FreeLibrary(mod);
+
+    ok(!GetModuleHandleA("rasapi32.dll"), "rasapi32.dll is already loaded.\n");
+    mod = LoadLibraryExW(L"c:\windows\system32\rasapi32.dll", 0, LOAD_LIBRARY_AS_IMAGE_RESOURCE);
+    mod_fixed = (HMODULE)((ULONG_PTR)mod & ~(ULONG_PTR)3);
+    ok(!!mod_fixed, "got NULL.\n" );
+    nt = RtlImageNtHeader(mod_fixed);
+    ok(!!nt, "got NULL.\n");
+    ok(nt->FileHeader.Machine != machine, "got 32 bit dll.\n");
+    FreeLibrary(mod);
+
+    ok(!GetModuleHandleA("rasapi32.dll"), "rasapi32.dll is already loaded.\n");
+    mod = LoadLibraryExW(L"c:\windows\system32\rasapi32.dll", 0, LOAD_LIBRARY_AS_DATAFILE);
+    mod_fixed = (HMODULE)((ULONG_PTR)mod & ~(ULONG_PTR)3);
+    ok(!!mod_fixed, "got NULL.\n" );
+    nt = RtlImageNtHeader(mod_fixed);
+    ok(!!nt, "got NULL.\n");
+    ok(nt->FileHeader.Machine != machine, "got 32 bit dll.\n");
+    FreeLibrary(mod);
+
     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++)