mirror of
https://github.com/reactos/reactos.git
synced 2024-09-18 16:52:54 +00:00
b944563c7b
I have a plan to implement shell32!SHRestricted correctly. JIRA issue: CORE-11515 - Get SHRestricted and SHSettingsChanged procedures from shell32.dll. - Use them and check the results.
109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
/*
|
|
* PROJECT: ReactOS API tests
|
|
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
|
|
* PURPOSE: Test for SHRestricted
|
|
* COPYRIGHT: Copyright 2024 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
|
|
*/
|
|
|
|
#include "shelltest.h"
|
|
#include <versionhelpers.h>
|
|
|
|
#define REGKEY_POLICIES L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies"
|
|
#define REGKEY_POLICIES_EXPLORER REGKEY_POLICIES L"\\Explorer"
|
|
|
|
typedef DWORD (WINAPI *FN_SHRestricted)(RESTRICTIONS rest);
|
|
typedef BOOL (WINAPI *FN_SHSettingsChanged)(LPCVOID unused, LPCVOID inpRegKey);
|
|
|
|
#define DELETE_VALUE(hBaseKey) \
|
|
SHDeleteValueW((hBaseKey), REGKEY_POLICIES_EXPLORER, L"NoRun")
|
|
|
|
#define SET_VALUE(hBaseKey, value) do { \
|
|
dwValue = (value); \
|
|
SHSetValueW((hBaseKey), REGKEY_POLICIES_EXPLORER, L"NoRun", \
|
|
REG_DWORD, &dwValue, sizeof(dwValue)); \
|
|
} while (0)
|
|
|
|
static VOID
|
|
TEST_SHRestricted(FN_SHRestricted fnGetValue, FN_SHSettingsChanged fnRefresh)
|
|
{
|
|
DWORD dwValue;
|
|
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
|
|
SET_VALUE(HKEY_CURRENT_USER, 0);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
|
|
SET_VALUE(HKEY_CURRENT_USER, 1);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 0);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 1);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
|
|
SET_VALUE(HKEY_CURRENT_USER, 2);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 1);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
|
|
ok_long(fnGetValue(REST_NORUN), 1);
|
|
fnRefresh(NULL, NULL);
|
|
ok_long(fnGetValue(REST_NORUN), 0);
|
|
}
|
|
|
|
START_TEST(SHRestricted)
|
|
{
|
|
if (IsWindowsVistaOrGreater())
|
|
{
|
|
skip("Vista+");
|
|
return;
|
|
}
|
|
|
|
HMODULE hSHELL32 = LoadLibraryW(L"shell32.dll");
|
|
FN_SHRestricted fnGetValue;
|
|
FN_SHSettingsChanged fnRefresh;
|
|
|
|
fnGetValue = (FN_SHRestricted)GetProcAddress(hSHELL32, MAKEINTRESOURCEA(100));
|
|
fnRefresh = (FN_SHSettingsChanged)GetProcAddress(hSHELL32, MAKEINTRESOURCEA(244));
|
|
|
|
if (fnGetValue && fnRefresh)
|
|
{
|
|
TEST_SHRestricted(fnGetValue, fnRefresh);
|
|
}
|
|
else
|
|
{
|
|
if (!fnGetValue)
|
|
skip("SHRestricted not found\n");
|
|
if (!fnRefresh)
|
|
skip("SHSettingsChanged not found\n");
|
|
}
|
|
|
|
FreeLibrary(hSHELL32);
|
|
}
|