mirror of
https://github.com/reactos/reactos.git
synced 2024-08-26 05:08:39 +00:00
227 lines
7 KiB
C
227 lines
7 KiB
C
/*
|
|
* PROJECT: appshim_apitest
|
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
|
* PURPOSE: Tests for ForceDxSetupSuccess shim
|
|
* COPYRIGHT: Copyright 2019 Mark Jansen (mark.jansen@reactos.org)
|
|
*/
|
|
|
|
#include <ntstatus.h>
|
|
#define WIN32_NO_STATUS
|
|
#include <windows.h>
|
|
#include "wine/test.h"
|
|
|
|
#include "appshim_apitest.h"
|
|
|
|
static HMODULE g_hShimDll;
|
|
static HMODULE g_hSentinelModule = (HMODULE)&g_hShimDll; /* Not a valid hmodule, so a nice sentinel */
|
|
static tGETHOOKAPIS pGetHookAPIs;
|
|
|
|
static INT (WINAPI *DirectXSetupGetVersion)(DWORD *lpdwVersion, DWORD *lpdwMinorVersion);
|
|
|
|
|
|
typedef HMODULE(WINAPI* LOADLIBRARYAPROC)(PCSTR);
|
|
typedef HMODULE(WINAPI* LOADLIBRARYWPROC)(PCWSTR);
|
|
typedef FARPROC(WINAPI* GETPROCADDRESSPROC)(HMODULE, PCSTR);
|
|
typedef BOOL(WINAPI* FREELIBRARYPROC)(HMODULE);
|
|
|
|
|
|
static HMODULE WINAPI my_LoadLibraryA(PCSTR Name)
|
|
{
|
|
return g_hSentinelModule;
|
|
}
|
|
|
|
static void test_LoadLibraryA(PHOOKAPI hook)
|
|
{
|
|
LOADLIBRARYAPROC proc;
|
|
|
|
ok_str(hook->LibraryName, "KERNEL32.DLL");
|
|
hook->OriginalFunction = my_LoadLibraryA;
|
|
proc = hook->ReplacementFunction;
|
|
|
|
/* Original function is not called */
|
|
ok_ptr(proc("dsetup"), g_hShimDll);
|
|
ok_ptr(proc("dsetup.dll"), g_hShimDll);
|
|
ok_ptr(proc("DSETUP.DLL"), g_hShimDll);
|
|
ok_ptr(proc("DSeTuP.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc("dsetup32"), g_hShimDll);
|
|
ok_ptr(proc("dsetup32.dll"), g_hShimDll);
|
|
ok_ptr(proc("DSETUP32.DLL"), g_hShimDll);
|
|
ok_ptr(proc("DSeTuP32.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc("c:\\bogus/dsetup"), g_hShimDll);
|
|
ok_ptr(proc("c:\\bogus\\dsetup.dll"), g_hShimDll);
|
|
ok_ptr(proc("c:/bogus/DSETUP.DLL"), g_hShimDll);
|
|
ok_ptr(proc("c:\\bogus\\DSeTuP.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc("////////////////////dsetup32"), g_hShimDll);
|
|
ok_ptr(proc("\\\\\\\\\\\\\\dsetup32.dll"), g_hShimDll);
|
|
ok_ptr(proc("xxxxxxxxxxx\\DSETUP32.DLL"), g_hShimDll);
|
|
ok_ptr(proc("xxxxxxxxxxx//DSeTuP32.DlL"), g_hShimDll);
|
|
|
|
/* Original function is called */
|
|
ok_ptr(proc(NULL), g_hSentinelModule);
|
|
ok_ptr(proc("dsetup\\something"), g_hSentinelModule);
|
|
ok_ptr(proc("dsetup.dl"), g_hSentinelModule);
|
|
ok_ptr(proc("c:\\DSETUP.DLL."), g_hSentinelModule);
|
|
ok_ptr(proc("DSeTuP.D"), g_hSentinelModule);
|
|
}
|
|
|
|
static HMODULE WINAPI my_LoadLibraryW(PCWSTR Name)
|
|
{
|
|
return g_hSentinelModule;
|
|
}
|
|
|
|
static void test_LoadLibraryW(PHOOKAPI hook)
|
|
{
|
|
LOADLIBRARYWPROC proc;
|
|
|
|
ok_str(hook->LibraryName, "KERNEL32.DLL");
|
|
|
|
hook->OriginalFunction = my_LoadLibraryW;
|
|
proc = hook->ReplacementFunction;
|
|
|
|
/* Original function is not called */
|
|
ok_ptr(proc(L"dsetup"), g_hShimDll);
|
|
ok_ptr(proc(L"dsetup.dll"), g_hShimDll);
|
|
ok_ptr(proc(L"DSETUP.DLL"), g_hShimDll);
|
|
ok_ptr(proc(L"DSeTuP.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc(L"dsetup32"), g_hShimDll);
|
|
ok_ptr(proc(L"dsetup32.dll"), g_hShimDll);
|
|
ok_ptr(proc(L"DSETUP32.DLL"), g_hShimDll);
|
|
ok_ptr(proc(L"DSeTuP32.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc(L"c:\\bogus/dsetup"), g_hShimDll);
|
|
ok_ptr(proc(L"c:\\bogus\\dsetup.dll"), g_hShimDll);
|
|
ok_ptr(proc(L"c:/bogus/DSETUP.DLL"), g_hShimDll);
|
|
ok_ptr(proc(L"c:\\bogus\\DSeTuP.DlL"), g_hShimDll);
|
|
|
|
ok_ptr(proc(L"////////////////////dsetup32"), g_hShimDll);
|
|
ok_ptr(proc(L"\\\\\\\\\\\\\\dsetup32.dll"), g_hShimDll);
|
|
ok_ptr(proc(L"xxxxxxxxxxx\\DSETUP32.DLL"), g_hShimDll);
|
|
ok_ptr(proc(L"xxxxxxxxxxx//DSeTuP32.DlL"), g_hShimDll);
|
|
|
|
/* Original function is called */
|
|
ok_ptr(proc(NULL), g_hSentinelModule);
|
|
ok_ptr(proc(L"dsetup\\something"), g_hSentinelModule);
|
|
ok_ptr(proc(L"dsetup.dl"), g_hSentinelModule);
|
|
ok_ptr(proc(L"c:\\DSETUP.DLL."), g_hSentinelModule);
|
|
ok_ptr(proc(L"DSeTuP.D"), g_hSentinelModule);
|
|
}
|
|
|
|
static void test_GetProcAddress(PHOOKAPI hook)
|
|
{
|
|
GETPROCADDRESSPROC proc;
|
|
DWORD n;
|
|
PCSTR Functions[] = {
|
|
"DirectXSetup",
|
|
"DirectXSetupA",
|
|
"DirectXSetupW",
|
|
"DirectXSetupGetVersion",
|
|
/* And not case sensitive? */
|
|
"directxsetup",
|
|
"DIRECTXSETUPA",
|
|
"DiReCtXsEtUpW",
|
|
NULL
|
|
};
|
|
HMODULE mod = GetModuleHandleA("kernel32.dll");
|
|
|
|
ok_str(hook->LibraryName, "KERNEL32.DLL");
|
|
hook->OriginalFunction = GetProcAddress; /* Some versions seem to call GetProcAddress regardless of what is put here.. */
|
|
proc = hook->ReplacementFunction;
|
|
|
|
ok_ptr(proc(mod, "CreateFileA"), GetProcAddress(mod, "CreateFileA"));
|
|
ok_ptr(proc(g_hShimDll, "CreateFileA"), NULL);
|
|
|
|
for (n = 0; Functions[n]; ++n)
|
|
{
|
|
FARPROC fn = proc(g_hShimDll, Functions[n]);
|
|
ok(fn != NULL, "Got NULL for %s\n", Functions[n]);
|
|
/* And yet, no export! */
|
|
ok_ptr(GetProcAddress(g_hShimDll, Functions[n]), NULL);
|
|
}
|
|
|
|
/* Not every function is shimmed */
|
|
ok_ptr(proc(g_hShimDll, "DirectXSetupShowEULA"), NULL);
|
|
/* Retrieve a pointer for our next test */
|
|
DirectXSetupGetVersion = (PVOID)proc(g_hShimDll, "DirectXSetupGetVersion");
|
|
}
|
|
|
|
static int g_NumShim = 0;
|
|
static int g_NumSentinel = 0;
|
|
static int g_NumCalls = 0;
|
|
BOOL WINAPI my_FreeLibrary(HMODULE module)
|
|
{
|
|
if (module == g_hSentinelModule)
|
|
++g_NumSentinel;
|
|
else if (module == g_hShimDll)
|
|
++g_NumShim;
|
|
++g_NumCalls;
|
|
return 33;
|
|
}
|
|
|
|
static void test_FreeLibrary(PHOOKAPI hook)
|
|
{
|
|
FREELIBRARYPROC proc;
|
|
|
|
ok_str(hook->LibraryName, "KERNEL32.DLL");
|
|
hook->OriginalFunction = my_FreeLibrary;
|
|
proc = hook->ReplacementFunction;
|
|
|
|
ok_int(proc(NULL), 33);
|
|
ok_int(proc(NULL), 33);
|
|
ok_int(proc(g_hSentinelModule), 33);
|
|
ok_int(proc(g_hSentinelModule), 33);
|
|
ok_int(proc(g_hShimDll), TRUE);
|
|
ok_int(proc(g_hShimDll), TRUE);
|
|
|
|
ok_int(g_NumShim, 0);
|
|
ok_int(g_NumSentinel, 2);
|
|
ok_int(g_NumCalls, 4);
|
|
}
|
|
|
|
|
|
START_TEST(forcedxsetup)
|
|
{
|
|
DWORD num_shims = 0, n;
|
|
PHOOKAPI hook;
|
|
|
|
if (!LoadShimDLL(L"aclayers.dll", &g_hShimDll, &pGetHookAPIs))
|
|
return;
|
|
|
|
hook = pGetHookAPIs("", L"ForceDxSetupSuccess", &num_shims);
|
|
|
|
ok(hook != NULL, "Expected hook to be a valid pointer\n");
|
|
ok(num_shims == 6, "Expected num_shims to be 6, was: %u\n", num_shims);
|
|
|
|
if (!hook || !num_shims)
|
|
return;
|
|
|
|
for (n = 0; n < num_shims; ++n)
|
|
{
|
|
if (!_stricmp(hook[n].FunctionName, "LoadLibraryA"))
|
|
test_LoadLibraryA(hook + n);
|
|
else if (!_stricmp(hook[n].FunctionName, "LoadLibraryW"))
|
|
test_LoadLibraryW(hook + n);
|
|
else if (!_stricmp(hook[n].FunctionName, "GetProcAddress"))
|
|
test_GetProcAddress(hook + n);
|
|
else if (!_stricmp(hook[n].FunctionName, "FreeLibrary"))
|
|
test_FreeLibrary(hook + n);
|
|
}
|
|
|
|
ok(DirectXSetupGetVersion != NULL, "No DirectXSetupGetVersion\n");
|
|
if (DirectXSetupGetVersion)
|
|
{
|
|
DWORD dwVersion = 0xdeadbeef;
|
|
DWORD dwMinorVersion = 0xbeefdead;
|
|
INT Res = DirectXSetupGetVersion(&dwVersion, &dwMinorVersion);
|
|
ok_int(Res, 1);
|
|
ok_hex(dwVersion, MAKELONG(7, 4)); // DirectX 7.0
|
|
ok_hex(dwMinorVersion, MAKELONG(1792, 0));
|
|
|
|
Res = DirectXSetupGetVersion(NULL, NULL);
|
|
ok_int(Res, 1);
|
|
}
|
|
}
|