mirror of
https://github.com/reactos/reactos.git
synced 2025-04-06 05:34:22 +00:00
[APPSHIM_APITEST] Add test for IgnoreLoadLibrary
CORE-15845
This commit is contained in:
Mark Jansen
parent
0fdfdcb1ea
commit
996f559fd7
3 changed files with 304 additions and 0 deletions
|
|
@ -6,6 +6,7 @@ list(APPEND SOURCE
|
|||
forcedxsetup.c
|
||||
genral_hooks.c
|
||||
ignorefreelib.c
|
||||
ignoreloadlib.c
|
||||
layer_hooks.c
|
||||
versionlie.c
|
||||
testlist.c
|
||||
|
|
|
|||
301
modules/rostests/apitests/appshim/ignoreloadlib.c
Normal file
301
modules/rostests/apitests/appshim/ignoreloadlib.c
Normal file
|
|
@ -0,0 +1,301 @@
|
|||
/*
|
||||
* PROJECT: appshim_apitest
|
||||
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
||||
* PURPOSE: Tests for IgnoreLoadLibrary shim
|
||||
* COPYRIGHT: Copyright 2019 Mark Jansen (mark.jansen@reactos.org)
|
||||
*/
|
||||
|
||||
#include <ntstatus.h>
|
||||
#define WIN32_NO_STATUS
|
||||
#include <windows.h>
|
||||
#include <ntndk.h>
|
||||
#include "wine/test.h"
|
||||
|
||||
#include "appshim_apitest.h"
|
||||
|
||||
static DWORD g_WinVersion;
|
||||
static tGETHOOKAPIS pGetHookAPIs;
|
||||
static HMODULE g_hSentinelModule = (HMODULE)&pGetHookAPIs; /* Not a valid hmodule, so a nice sentinel */
|
||||
static HMODULE g_h123 = (HMODULE)123;
|
||||
static HMODULE g_h111 = (HMODULE)111;
|
||||
static HMODULE g_h0 = (HMODULE)0;
|
||||
|
||||
typedef HMODULE(WINAPI* LOADLIBRARYAPROC)(LPCSTR lpLibFileName);
|
||||
typedef HMODULE(WINAPI* LOADLIBRARYWPROC)(LPCWSTR lpLibFileName);
|
||||
typedef HMODULE(WINAPI* LOADLIBRARYEXAPROC)(LPCSTR lpLibFileName, HANDLE hFile, DWORD dwFlags);
|
||||
typedef HMODULE(WINAPI* LOADLIBRARYEXWPROC)(LPCWSTR lpLibFileName, HANDLE hFile, DWORD dwFlags);
|
||||
|
||||
|
||||
UINT
|
||||
WINAPI
|
||||
GetErrorMode(VOID)
|
||||
{
|
||||
NTSTATUS Status;
|
||||
UINT ErrMode;
|
||||
|
||||
/* Query the current setting */
|
||||
Status = NtQueryInformationProcess(NtCurrentProcess(),
|
||||
ProcessDefaultHardErrorMode,
|
||||
&ErrMode,
|
||||
sizeof(ErrMode),
|
||||
NULL);
|
||||
if (!NT_SUCCESS(Status))
|
||||
{
|
||||
/* Fail if we couldn't query */
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Check if NOT failing critical errors was requested */
|
||||
if (ErrMode & SEM_FAILCRITICALERRORS)
|
||||
{
|
||||
/* Mask it out, since the native API works differently */
|
||||
ErrMode &= ~SEM_FAILCRITICALERRORS;
|
||||
}
|
||||
else
|
||||
{
|
||||
/* OR it if the caller didn't, due to different native semantics */
|
||||
ErrMode |= SEM_FAILCRITICALERRORS;
|
||||
}
|
||||
|
||||
/* Return the mode */
|
||||
return ErrMode;
|
||||
}
|
||||
|
||||
static HMODULE WINAPI my_LoadLibraryA(PCSTR Name)
|
||||
{
|
||||
DWORD dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX),
|
||||
"Unexpected error mode: 0x%x\n", dwErrorMode);
|
||||
return g_hSentinelModule;
|
||||
}
|
||||
|
||||
static HMODULE WINAPI my_LoadLibraryExA(PCSTR Name, HANDLE hFile, DWORD dwFlags)
|
||||
{
|
||||
DWORD dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX),
|
||||
"Unexpected error mode: 0x%x\n", dwErrorMode);
|
||||
return g_hSentinelModule;
|
||||
}
|
||||
|
||||
static HMODULE WINAPI my_LoadLibraryW(PCWSTR Name)
|
||||
{
|
||||
DWORD dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX),
|
||||
"Unexpected error mode: 0x%x\n", dwErrorMode);
|
||||
return g_hSentinelModule;
|
||||
}
|
||||
|
||||
static HMODULE WINAPI my_LoadLibraryExW(PCWSTR Name, HANDLE hFile, DWORD dwFlags)
|
||||
{
|
||||
DWORD dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == (SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX | SEM_NOOPENFILEERRORBOX),
|
||||
"Unexpected error mode: 0x%x\n", dwErrorMode);
|
||||
return g_hSentinelModule;
|
||||
}
|
||||
|
||||
|
||||
static void test_LoadLibraryA(PHOOKAPI hook)
|
||||
{
|
||||
LOADLIBRARYAPROC proc;
|
||||
DWORD dwErrorMode, dwOldErrorMode;
|
||||
|
||||
hook->OriginalFunction = my_LoadLibraryA;
|
||||
proc = hook->ReplacementFunction;
|
||||
|
||||
dwOldErrorMode = GetErrorMode();
|
||||
|
||||
/* Exact names return what is specified */
|
||||
ok_ptr(proc("test123.dll"), g_h123);
|
||||
ok_ptr(proc("test111"), g_h111);
|
||||
/* Extension is not added */
|
||||
ok_ptr(proc("test111.dll"), g_hSentinelModule);
|
||||
/* Zero can be specified */
|
||||
ok_ptr(proc("Something.mark"), g_h0);
|
||||
/* Or default returned */
|
||||
ok_ptr(proc("empty"), g_h0);
|
||||
|
||||
/* Paths, do not have to be valid */
|
||||
ok_ptr(proc("\\test123.dll"), g_h123);
|
||||
ok_ptr(proc("/test123.dll"), g_h123);
|
||||
ok_ptr(proc("\\\\\\\\test123.dll"), g_h123);
|
||||
ok_ptr(proc("////test123.dll"), g_h123);
|
||||
ok_ptr(proc("mypath:something\\does\\not\\matter\\test123.dll"), g_h123);
|
||||
ok_ptr(proc("/put/whatever/you/want/here/test123.dll"), g_h123);
|
||||
|
||||
/* path separator is checked, not just any point in the path */
|
||||
ok_ptr(proc("-test123.dll"), g_hSentinelModule);
|
||||
ok_ptr(proc("test123.dll-"), g_hSentinelModule);
|
||||
|
||||
dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == dwOldErrorMode, "ErrorMode changed, was 0x%x, is 0x%x\n", dwOldErrorMode, dwErrorMode);
|
||||
}
|
||||
|
||||
static void test_LoadLibraryW(PHOOKAPI hook)
|
||||
{
|
||||
LOADLIBRARYWPROC proc;
|
||||
DWORD dwErrorMode, dwOldErrorMode;
|
||||
|
||||
hook->OriginalFunction = my_LoadLibraryW;
|
||||
proc = hook->ReplacementFunction;
|
||||
|
||||
dwOldErrorMode = GetErrorMode();
|
||||
|
||||
/* Exact names return what is specified */
|
||||
ok_ptr(proc(L"test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"test111"), g_h111);
|
||||
/* Extension is not added */
|
||||
ok_ptr(proc(L"test111.dll"), g_hSentinelModule);
|
||||
/* Zero can be specified */
|
||||
ok_ptr(proc(L"Something.mark"), g_h0);
|
||||
/* Or default returned */
|
||||
ok_ptr(proc(L"empty"), g_h0);
|
||||
|
||||
/* Paths, do not have to be valid */
|
||||
ok_ptr(proc(L"\\test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"/test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"\\\\\\\\test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"////test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"mypath:something\\does\\not\\matter\\test123.dll"), g_h123);
|
||||
ok_ptr(proc(L"/put/whatever/you/want/here/test123.dll"), g_h123);
|
||||
|
||||
/* path separator is checked, not just any point in the path */
|
||||
ok_ptr(proc(L"-test123.dll"), g_hSentinelModule);
|
||||
ok_ptr(proc(L"test123.dll-"), g_hSentinelModule);
|
||||
|
||||
dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == dwOldErrorMode, "ErrorMode changed, was 0x%x, is 0x%x\n", dwOldErrorMode, dwErrorMode);
|
||||
}
|
||||
|
||||
static void test_LoadLibraryExA(PHOOKAPI hook)
|
||||
{
|
||||
LOADLIBRARYEXAPROC proc;
|
||||
DWORD dwErrorMode, dwOldErrorMode;
|
||||
|
||||
hook->OriginalFunction = my_LoadLibraryExA;
|
||||
proc = hook->ReplacementFunction;
|
||||
|
||||
dwOldErrorMode = GetErrorMode();
|
||||
|
||||
/* Exact names return what is specified */
|
||||
ok_ptr(proc("test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("test111", INVALID_HANDLE_VALUE, 0), g_h111);
|
||||
/* Extension is not added */
|
||||
ok_ptr(proc("test111.dll", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
/* Zero can be specified */
|
||||
ok_ptr(proc("Something.mark", INVALID_HANDLE_VALUE, 0), g_h0);
|
||||
/* Or default returned */
|
||||
ok_ptr(proc("empty", INVALID_HANDLE_VALUE, 0), g_h0);
|
||||
|
||||
/* Paths, do not have to be valid */
|
||||
ok_ptr(proc("\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("/test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("\\\\\\\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("////test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("mypath:something\\does\\not\\matter\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc("/put/whatever/you/want/here/test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
|
||||
/* path separator is checked, not just any point in the path */
|
||||
ok_ptr(proc("-test123.dll", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
ok_ptr(proc("test123.dll-", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
|
||||
dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == dwOldErrorMode, "ErrorMode changed, was 0x%x, is 0x%x\n", dwOldErrorMode, dwErrorMode);
|
||||
}
|
||||
|
||||
static void test_LoadLibraryExW(PHOOKAPI hook)
|
||||
{
|
||||
LOADLIBRARYEXWPROC proc;
|
||||
DWORD dwErrorMode, dwOldErrorMode;
|
||||
|
||||
hook->OriginalFunction = my_LoadLibraryExW;
|
||||
proc = hook->ReplacementFunction;
|
||||
|
||||
dwOldErrorMode = GetErrorMode();
|
||||
|
||||
/* Exact names return what is specified */
|
||||
ok_ptr(proc(L"test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"test111", INVALID_HANDLE_VALUE, 0), g_h111);
|
||||
/* Extension is not added */
|
||||
ok_ptr(proc(L"test111.dll", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
/* Zero can be specified */
|
||||
ok_ptr(proc(L"Something.mark", INVALID_HANDLE_VALUE, 0), g_h0);
|
||||
/* Or default returned */
|
||||
ok_ptr(proc(L"empty", INVALID_HANDLE_VALUE, 0), g_h0);
|
||||
|
||||
/* Paths, do not have to be valid */
|
||||
ok_ptr(proc(L"\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"/test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"\\\\\\\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"////test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"mypath:something\\does\\not\\matter\\test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
ok_ptr(proc(L"/put/whatever/you/want/here/test123.dll", INVALID_HANDLE_VALUE, 0), g_h123);
|
||||
|
||||
/* path separator is checked, not just any point in the path */
|
||||
ok_ptr(proc(L"-test123.dll", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
ok_ptr(proc(L"test123.dll-", INVALID_HANDLE_VALUE, 0), g_hSentinelModule);
|
||||
|
||||
dwErrorMode = GetErrorMode();
|
||||
ok(dwErrorMode == dwOldErrorMode, "ErrorMode changed, was 0x%x, is 0x%x\n", dwOldErrorMode, dwErrorMode);
|
||||
}
|
||||
|
||||
/* versionlie.c */
|
||||
DWORD get_host_winver(void);
|
||||
|
||||
START_TEST(ignoreloadlib)
|
||||
{
|
||||
DWORD num_shims = 0, n, dwErrorMode;
|
||||
PHOOKAPI hook;
|
||||
|
||||
g_WinVersion = get_host_winver();
|
||||
|
||||
if (g_WinVersion < _WIN32_WINNT_WIN8)
|
||||
pGetHookAPIs = LoadShimDLL2(L"aclayers.dll");
|
||||
else
|
||||
pGetHookAPIs = LoadShimDLL2(L"acgenral.dll");
|
||||
|
||||
if (!pGetHookAPIs)
|
||||
{
|
||||
skip("GetHookAPIs not found\n");
|
||||
return;
|
||||
}
|
||||
|
||||
hook = pGetHookAPIs("test123.dll:123;test111:111;Something.mark:0;empty", L"IgnoreLoadLibrary", &num_shims);
|
||||
|
||||
ok(hook != NULL, "Expected hook to be a valid pointer\n");
|
||||
ok(num_shims == 4, "Expected num_shims to be 0, was: %u\n", num_shims);
|
||||
|
||||
if (!hook || num_shims != 4)
|
||||
return;
|
||||
|
||||
dwErrorMode = GetErrorMode();
|
||||
trace("Error mode: 0x%x\n", dwErrorMode);
|
||||
|
||||
for (n = 0; n < num_shims; ++n)
|
||||
{
|
||||
ok_str(hook[n].LibraryName, "KERNEL32.DLL");
|
||||
if (!_stricmp(hook[n].FunctionName, "LoadLibraryA"))
|
||||
{
|
||||
ok_int(n, 0);
|
||||
test_LoadLibraryA(hook + n);
|
||||
}
|
||||
else if (!_stricmp(hook[n].FunctionName, "LoadLibraryExA"))
|
||||
{
|
||||
ok_int(n, 1);
|
||||
test_LoadLibraryExA(hook + n);
|
||||
}
|
||||
else if (!_stricmp(hook[n].FunctionName, "LoadLibraryW"))
|
||||
{
|
||||
ok_int(n, 2);
|
||||
test_LoadLibraryW(hook + n);
|
||||
}
|
||||
else if (!_stricmp(hook[n].FunctionName, "LoadLibraryExW"))
|
||||
{
|
||||
ok_int(n, 3);
|
||||
test_LoadLibraryExW(hook + n);
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(0, "Unknown function %s\n", hook[n].FunctionName);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -7,6 +7,7 @@ extern void func_dispmode(void);
|
|||
extern void func_forcedxsetup(void);
|
||||
extern void func_genral_hooks(void);
|
||||
extern void func_ignorefreelib(void);
|
||||
extern void func_ignoreloadlib(void);
|
||||
extern void func_layer_hooks(void);
|
||||
extern void func_versionlie(void);
|
||||
|
||||
|
|
@ -16,6 +17,7 @@ const struct test winetest_testlist[] =
|
|||
{ "forcedxsetup", func_forcedxsetup },
|
||||
{ "genral_hooks", func_genral_hooks },
|
||||
{ "ignorefreelib", func_ignorefreelib },
|
||||
{ "ignoreloadlib", func_ignoreloadlib },
|
||||
{ "layer_hooks", func_layer_hooks },
|
||||
{ "versionlie", func_versionlie },
|
||||
{ 0, 0 }
|
||||
|
|
|
|||
Loading…
Reference in a new issue