Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-05 14:53:40 +00:00
Code Issues Projects Releases Packages Wiki Activity

[APPSHIM_APITEST] Add tests for the DisableThemes shim. CORE-11927

Browse source 
svn path=/trunk/; revision=73422
This commit is contained in:
Mark Jansen 2016-12-04 15:33:28 +00:00
parent d39ff22033
commit f0d4850705
1 changed files with 69 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

96
rostests/apitests/appshim/dispmode.c
View file

@ -46,7 +46,7 @@ static LONG g_Count;
static DEVMODEA g_LastDevmode; static DEVMODEA g_LastDevmode;
static DWORD g_LastFlags; static DWORD g_LastFlags;
LONG (WINAPI *pChangeDisplaySettingsA)(_In_opt_ PDEVMODEA lpDevMode, _In_ DWORD dwflags); static LONG (WINAPI *pChangeDisplaySettingsA)(_In_opt_ PDEVMODEA lpDevMode, _In_ DWORD dwflags);
LONG WINAPI mChangeDisplaySettingsA(_In_opt_ PDEVMODEA lpDevMode, _In_ DWORD dwflags) LONG WINAPI mChangeDisplaySettingsA(_In_opt_ PDEVMODEA lpDevMode, _In_ DWORD dwflags)
{ {
g_Count++; g_Count++;
@ -56,15 +56,33 @@ LONG WINAPI mChangeDisplaySettingsA(_In_opt_ PDEVMODEA lpDevMode, _In_ DWORD dwf
return DISP_CHANGE_FAILED; return DISP_CHANGE_FAILED;
} }
static LONG g_ThemeCount;
static DWORD g_LastThemeFlags;
static void pre_8bit() static void (WINAPI *pSetThemeAppProperties)(DWORD dwFlags);
void WINAPI mSetThemeAppProperties(DWORD dwFlags)
{
g_ThemeCount++;
g_LastThemeFlags = dwFlags;
}
static const WCHAR* shim_dll(const WCHAR* name)
{
static WCHAR buf[MAX_PATH];
pSdbGetAppPatchDir(NULL, buf, MAX_PATH);
StringCchCatW(buf, _countof(buf), name);
return buf;
}
static void pre_8bit(void)
{ {
g_Count = 0; g_Count = 0;
memset(&g_LastDevmode, 0, sizeof(g_LastDevmode)); memset(&g_LastDevmode, 0, sizeof(g_LastDevmode));
g_LastFlags = 0xffffffff; g_LastFlags = 0xffffffff;
} }
static void post_8bit() static void post_8bit(void)
{ {
ok_int(g_Count, 1); ok_int(g_Count, 1);
ok_hex(g_LastDevmode.dmFields & DM_BITSPERPEL, DM_BITSPERPEL); ok_hex(g_LastDevmode.dmFields & DM_BITSPERPEL, DM_BITSPERPEL);
@ -72,14 +90,14 @@ static void post_8bit()
ok_hex(g_LastFlags, CDS_FULLSCREEN); ok_hex(g_LastFlags, CDS_FULLSCREEN);
} }
static void pre_640() static void pre_640(void)
{ {
g_Count = 0; g_Count = 0;
memset(&g_LastDevmode, 0, sizeof(g_LastDevmode)); memset(&g_LastDevmode, 0, sizeof(g_LastDevmode));
g_LastFlags = 0xffffffff; g_LastFlags = 0xffffffff;
} }
static void post_640() static void post_640(void)
{ {
ok_int(g_Count, 1); ok_int(g_Count, 1);
ok_hex(g_LastDevmode.dmFields & (DM_PELSWIDTH | DM_PELSHEIGHT), (DM_PELSWIDTH | DM_PELSHEIGHT)); ok_hex(g_LastDevmode.dmFields & (DM_PELSWIDTH | DM_PELSHEIGHT), (DM_PELSWIDTH | DM_PELSHEIGHT));
@ -88,8 +106,17 @@ static void post_640()
ok_hex(g_LastFlags, CDS_FULLSCREEN); ok_hex(g_LastFlags, CDS_FULLSCREEN);
} }
static void pre_theme(void)
{
g_ThemeCount = 0;
g_LastThemeFlags = 0xffffffff;
}
static void post_theme(void)
{
ok_int(g_ThemeCount, 1);
ok_hex(g_LastThemeFlags, 0);
}
static PIMAGE_IMPORT_DESCRIPTOR FindImportDescriptor(PBYTE DllBase, PCSTR DllName) static PIMAGE_IMPORT_DESCRIPTOR FindImportDescriptor(PBYTE DllBase, PCSTR DllName)
{ {
@ -144,6 +171,15 @@ static BOOL RedirectIat(HMODULE TargetDll, PCSTR DllName, PCSTR FunctionName, UL
return FALSE; return FALSE;
} }
static BOOL hook_disp(HMODULE dll)
{
return RedirectIat(dll, "user32.dll", "ChangeDisplaySettingsA", (ULONG_PTR)mChangeDisplaySettingsA, (ULONG_PTR*)&pChangeDisplaySettingsA);
}
static BOOL hook_theme(HMODULE dll)
{
return RedirectIat(dll, "uxtheme.dll", "SetThemeAppProperties", (ULONG_PTR)mSetThemeAppProperties, (ULONG_PTR*)&pSetThemeAppProperties);
}
static void test_one(LPCSTR shim, DWORD dwReason, void(*pre)(), void(*post)()) static void test_one(LPCSTR shim, DWORD dwReason, void(*pre)(), void(*post)())
@ -178,28 +214,30 @@ static void test_one(LPCSTR shim, DWORD dwReason, void(*pre)(), void(*post)())
static struct test_info static struct test_info
{ {
const char* name; const char* name;
const WCHAR* dll;
DWORD winver; DWORD winver;
DWORD reason; DWORD reason;
void(*pre)(); BOOL(*hook)(HMODULE);
void(*post)(); void(*pre)(void);
void(*post)(void);
} tests[] = } tests[] =
{ {
{ "Force8BitColor", WINVER_ANY, 1, pre_8bit, post_8bit }, { "Force8BitColor", L"\\aclayers.dll", WINVER_ANY, 1, hook_disp, pre_8bit, post_8bit },
{ "Force8BitColor", WINVER_VISTA, 100, pre_8bit, post_8bit }, { "Force8BitColor", L"\\aclayers.dll", WINVER_VISTA, 100, hook_disp, pre_8bit, post_8bit },
{ "Force640x480", WINVER_ANY, 1, pre_640, post_640 }, { "Force640x480", L"\\aclayers.dll", WINVER_ANY, 1, hook_disp, pre_640, post_640 },
{ "Force640x480", WINVER_VISTA, 100, pre_640, post_640 }, { "Force640x480", L"\\aclayers.dll", WINVER_VISTA, 100, hook_disp, pre_640, post_640 },
/* { "DisableThemes" }, AcGenral.dll */ { "DisableThemes", L"\\acgenral.dll", WINVER_ANY, 1, hook_theme, pre_theme, post_theme },
{ "DisableThemes", L"\\acgenral.dll", WINVER_VISTA, 100, hook_theme, pre_theme, post_theme },
}; };
static void run_test(size_t n, WCHAR* buf, BOOL unload) static void run_test(size_t n, BOOL unload)
{ {
BOOL ret; BOOL ret;
HMODULE dll; HMODULE dll;
const WCHAR* buf = shim_dll(tests[n].dll);
trace("Running %d (%s)\n", n, tests[n].name); dll = LoadLibraryW(shim_dll(tests[n].dll));
dll = LoadLibraryW(buf);
pGetHookAPIs = (void*)GetProcAddress(dll, "GetHookAPIs"); pGetHookAPIs = (void*)GetProcAddress(dll, "GetHookAPIs");
pNotifyShims = (void*)GetProcAddress(dll, "NotifyShims"); pNotifyShims = (void*)GetProcAddress(dll, "NotifyShims");
@ -210,14 +248,14 @@ static void run_test(size_t n, WCHAR* buf, BOOL unload)
return; return;
} }
ret = RedirectIat(dll, "user32.dll", "ChangeDisplaySettingsA", (ULONG_PTR)mChangeDisplaySettingsA, (ULONG_PTR*)&pChangeDisplaySettingsA); ret = tests[n].hook(dll);
if (ret) if (ret)
{ {
test_one(tests[n].name, tests[n].reason, tests[n].pre, tests[n].post); test_one(tests[n].name, tests[n].reason, tests[n].pre, tests[n].post);
} }
else else
{ {
ok(0, "Unable to redirect ChangeDisplaySettingsA!\n"); ok(0, "Unable to redirect functions!\n");
} }
FreeLibrary(dll); FreeLibrary(dll);
if (unload) if (unload)
@ -231,8 +269,6 @@ static void run_test(size_t n, WCHAR* buf, BOOL unload)
START_TEST(dispmode) START_TEST(dispmode)
{ {
HMODULE dll = LoadLibraryA("apphelp.dll"); HMODULE dll = LoadLibraryA("apphelp.dll");
WCHAR buf[MAX_PATH];
WCHAR aclayers[] = L"\\aclayers.dll";
size_t n; size_t n;
int argc; int argc;
char **argv; char **argv;
@ -246,26 +282,27 @@ START_TEST(dispmode)
g_WinVersion = get_host_winver(); g_WinVersion = get_host_winver();
pSdbGetAppPatchDir(NULL, buf, MAX_PATH);
StringCchCatW(buf, _countof(buf), aclayers);
argc = winetest_get_mainargs(&argv); argc = winetest_get_mainargs(&argv);
if (argc < 3) if (argc < 3)
{ {
WCHAR path[MAX_PATH]; WCHAR path[MAX_PATH];
GetModuleFileNameW(NULL, path, _countof(path)); GetModuleFileNameW(NULL, path, _countof(path));
dll = GetModuleHandleW(buf); dll = GetModuleHandleW(shim_dll(L"\\aclayers.dll"));
if (!dll)
dll = GetModuleHandleW(shim_dll(L"\\acgenral.dll"));
if (dll != NULL) if (dll != NULL)
trace("Loaded under a shim, running each test in it's own process\n"); trace("Loaded under a shim, running each test in it's own process\n");
for (n = 0; n < _countof(tests); ++n) for (n = 0; n < _countof(tests); ++n)
{ {
LONG failures = winetest_get_failures();
if (g_WinVersion < tests[n].winver) if (g_WinVersion < tests[n].winver)
continue; continue;
if (dll == NULL) if (dll == NULL)
{ {
run_test(n, buf, TRUE); run_test(n, TRUE);
} }
else else
{ {
@ -284,6 +321,11 @@ START_TEST(dispmode)
CloseHandle(pi.hProcess); CloseHandle(pi.hProcess);
} }
} }
if (failures != winetest_get_failures())
{
trace("Failures from %d (%s)\n", n, tests[n].name);
}
} }
} }
else else
@ -291,7 +333,7 @@ START_TEST(dispmode)
n = (size_t)atoi(argv[2]); n = (size_t)atoi(argv[2]);
if (n >= 0 && n < _countof(tests)) if (n >= 0 && n < _countof(tests))
{ {
run_test(n, buf, FALSE); run_test(n, FALSE);
} }
else else
{ {