Module: wine Branch: master Commit: 21c3cfe541c53e36334d16ad0a92c4bfd4155fdd URL: http://source.winehq.org/git/wine.git/?a=commit;h=21c3cfe541c53e36334d16ad0a...
Author: André Hentschel nerv@dawncrow.de Date: Wed Sep 11 22:31:01 2013 +0200
kernel32/tests: Add tests for ResolveDelayLoadedAPI.
---
dlls/kernel32/tests/loader.c | 106 ++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 106 insertions(+), 0 deletions(-)
diff --git a/dlls/kernel32/tests/loader.c b/dlls/kernel32/tests/loader.c index 3b72cc8..96d57f4 100644 --- a/dlls/kernel32/tests/loader.c +++ b/dlls/kernel32/tests/loader.c @@ -28,6 +28,7 @@ #include "winbase.h" #include "winternl.h" #include "wine/test.h" +#include "delayloadhandler.h"
#define ALIGN_SIZE(size, alignment) (((size) + (alignment - 1)) & ~((alignment - 1)))
@@ -43,6 +44,7 @@ struct PROCESS_BASIC_INFORMATION_PRIVATE
static BOOL is_child; static LONG *child_failures; +static WORD cb_count;
static NTSTATUS (WINAPI *pNtMapViewOfSection)(HANDLE, HANDLE, PVOID *, ULONG, SIZE_T, const LARGE_INTEGER *, SIZE_T *, ULONG, ULONG, ULONG); static NTSTATUS (WINAPI *pNtUnmapViewOfSection)(HANDLE, PVOID); @@ -57,6 +59,9 @@ static NTSTATUS (WINAPI *pLdrLockLoaderLock)(ULONG, ULONG *, ULONG *); static NTSTATUS (WINAPI *pLdrUnlockLoaderLock)(ULONG, ULONG); static void (WINAPI *pRtlAcquirePebLock)(void); static void (WINAPI *pRtlReleasePebLock)(void); +static PVOID (WINAPI *pResolveDelayLoadedAPI)(PVOID, PCIMAGE_DELAYLOAD_DESCRIPTOR, + PDELAYLOAD_FAILURE_DLL_CALLBACK, PVOID, + PIMAGE_THUNK_DATA ThunkAddress,ULONG);
static PVOID RVAToAddr(DWORD_PTR rva, HMODULE module) { @@ -2184,6 +2189,105 @@ if (0) ok(ret, "DeleteFile error %d\n", GetLastError()); }
+static PVOID WINAPI failuredllhook(ULONG ul, DELAYLOAD_INFO* pd) +{ + cb_count++; + return NULL; +} + +static void test_ResolveDelayLoadedAPI(void) +{ + HMODULE hlib; + int i; + static const char* td[] = + { + "advapi32.dll", + "comdlg32.dll", + }; + + if (!pResolveDelayLoadedAPI) + { + todo_wine win_skip("ResolveDelayLoadedAPI is not available\n"); + return; + } + + if (0) /* crashes on native */ + { + SetLastError(0xdeadbeef); + ok(!pResolveDelayLoadedAPI(NULL, NULL, NULL, NULL, NULL, 0), + "ResolveDelayLoadedAPI succeeded\n"); + ok(GetLastError() == 0xdeadbeef, "GetLastError changed to %x\n", GetLastError()); + + cb_count = 0; + SetLastError(0xdeadbeef); + ok(!pResolveDelayLoadedAPI(NULL, NULL, failuredllhook, NULL, NULL, 0), + "ResolveDelayLoadedAPI succeeded\n"); + ok(GetLastError() == 0xdeadbeef, "GetLastError changed to %x\n", GetLastError()); + ok(cb_count == 1, "Wrong callback count: %d\n", cb_count); + } + + for (i = 0; i < sizeof(td)/sizeof(td[0]); i++) + { + IMAGE_DELAYLOAD_DESCRIPTOR *delaydir; + ULONG size; + + SetLastError(0xdeadbeef); + hlib = LoadLibrary(td[i]); + ok(hlib != NULL, "LoadLibrary error %u\n", GetLastError()); + if (!hlib) + { + skip("couldn't load %s.\n", td[i]); + continue; + } + + delaydir = RtlImageDirectoryEntryToData(hlib, TRUE, IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT, &size); + if (!delaydir) + { + skip("haven't found section for delay import directory in %s.\n", td[i]); + FreeLibrary(hlib); + continue; + } + + for (;;) + { + const IMAGE_THUNK_DATA *itdn; + IMAGE_THUNK_DATA *itda; + HMODULE htarget; + int j; + + if (!delaydir->DllNameRVA || + !delaydir->ImportAddressTableRVA || + !delaydir->ImportNameTableRVA) break; + + itdn = RVAToAddr(delaydir->ImportNameTableRVA, hlib); + itda = RVAToAddr(delaydir->ImportAddressTableRVA, hlib); + htarget = LoadLibrary(RVAToAddr(delaydir->DllNameRVA, hlib)); + for (j = 0; itdn[j].u1.Ordinal; j++) + { + void *ret, *load; + + if (IMAGE_SNAP_BY_ORDINAL(itdn[j].u1.Ordinal)) + load = (void *)GetProcAddress(htarget, (LPSTR)IMAGE_ORDINAL(itdn[j].u1.Ordinal)); + else + { + const IMAGE_IMPORT_BY_NAME* iibn = RVAToAddr(itdn[j].u1.AddressOfData, hlib); + load = (void *)GetProcAddress(htarget, (char*)iibn->Name); + } + + cb_count = 0; + ret = pResolveDelayLoadedAPI(hlib, delaydir, failuredllhook, NULL, &itda[j], 0); + ok(ret != NULL, "ResolveDelayLoadedAPI failed\n"); + ok(ret == load, "expected %p, got %p\n", ret, load); + ok(ret == (void*)itda[j].u1.AddressOfData, "expected %p, got %p\n", + ret, (void*)itda[j].u1.AddressOfData); + ok(!cb_count, "Wrong callback count: %d\n", cb_count); + } + delaydir++; + } + FreeLibrary(hlib); + } +} + START_TEST(loader) { int argc; @@ -2203,6 +2307,7 @@ START_TEST(loader) pLdrUnlockLoaderLock = (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "LdrUnlockLoaderLock"); pRtlAcquirePebLock = (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "RtlAcquirePebLock"); pRtlReleasePebLock = (void *)GetProcAddress(GetModuleHandle("ntdll.dll"), "RtlReleasePebLock"); + pResolveDelayLoadedAPI = (void *)GetProcAddress(GetModuleHandle("kernel32.dll"), "ResolveDelayLoadedAPI");
mapping = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, 4096, "winetest_loader"); ok(mapping != 0, "CreateFileMapping failed\n"); @@ -2227,4 +2332,5 @@ START_TEST(loader) test_ImportDescriptors(); test_section_access(); test_ExitProcess(); + test_ResolveDelayLoadedAPI(); }
-
Alexandre Julliard