mirror of
https://github.com/reactos/reactos.git
synced 2024-11-02 04:37:32 +00:00
ac870f986d
I have a plan to implement shell32!SHRestricted correctly. JIRA issue: CORE-11515 In the testcase: - Get procedure SHGetRestriction from shlwapi.dll. - Call SHGetRestriction to test and check the results.
144 lines
4 KiB
C
144 lines
4 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
|
* PURPOSE: Tests for SHGetRestriction
|
|
* COPYRIGHT: Copyright 2024 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
|
|
*/
|
|
|
|
#include <apitest.h>
|
|
#include <shlwapi.h>
|
|
|
|
#define REGKEY_POLICIES L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies"
|
|
#define REGKEY_POLICIES_EXPLORER REGKEY_POLICIES L"\\Explorer"
|
|
|
|
typedef DWORD (WINAPI *FN_SHGetRestriction)(LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue);
|
|
typedef BOOL (WINAPI *FN_SHSettingsChanged)(LPCVOID unused, LPCVOID inpRegKey);
|
|
|
|
static DWORD
|
|
Candidate_SHGetRestriction(LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue)
|
|
{
|
|
WCHAR szPath[MAX_PATH];
|
|
DWORD cbValue, dwValue = 0;
|
|
|
|
if (!lpSubKey)
|
|
lpSubKey = REGKEY_POLICIES;
|
|
|
|
PathCombineW(szPath, lpSubKey, lpSubName);
|
|
|
|
cbValue = sizeof(dwValue);
|
|
if (SHGetValueW(HKEY_LOCAL_MACHINE, szPath, lpValue, NULL, &dwValue, &cbValue) == ERROR_SUCCESS)
|
|
return dwValue;
|
|
|
|
cbValue = sizeof(dwValue);
|
|
SHGetValueW(HKEY_CURRENT_USER, szPath, lpValue, NULL, &dwValue, &cbValue);
|
|
return dwValue;
|
|
}
|
|
|
|
typedef struct tagTEST_ENTRY
|
|
{
|
|
LPCWSTR lpSubName;
|
|
LPCWSTR lpValue;
|
|
} TEST_ENTRY, *PTEST_ENTRY;
|
|
|
|
static const TEST_ENTRY s_Entries[] =
|
|
{
|
|
{ L"Explorer", L"NoRun" },
|
|
{ L"Explorer", L"ForceActiveDesktopOn" },
|
|
{ L"Explorer", L"NoActiveDesktop" },
|
|
{ L"Explorer", L"NoDisconnect" },
|
|
{ L"Explorer", L"NoRecentDocsHistory" },
|
|
{ L"Explorer", L"NoDriveTypeAutoRun" },
|
|
{ L"Explorer", L"NoSimpleStartMenu" },
|
|
{ L"System", L"DontDisplayLastUserName" },
|
|
{ L"System", L"ShutdownWithoutLogon" },
|
|
{ L"System", L"UndockWithoutLogon" },
|
|
};
|
|
|
|
static void
|
|
TEST_DoEntry(const TEST_ENTRY *entry, FN_SHGetRestriction fnSHGetRestriction)
|
|
{
|
|
DWORD value1 = fnSHGetRestriction(NULL, entry->lpSubName, entry->lpValue);
|
|
DWORD value2 = Candidate_SHGetRestriction(NULL, entry->lpSubName, entry->lpValue);
|
|
//trace("%ld vs %ld\n", value1, value2);
|
|
ok_long(value1, value2);
|
|
}
|
|
|
|
#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_SHGetRestriction_Stage(
|
|
INT iStage,
|
|
FN_SHGetRestriction fnSHGetRestriction)
|
|
{
|
|
size_t iItem;
|
|
DWORD dwValue;
|
|
|
|
trace("Stage #%d\n", iStage);
|
|
|
|
switch (iStage)
|
|
{
|
|
case 0:
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
break;
|
|
case 1:
|
|
SET_VALUE(HKEY_CURRENT_USER, 0);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
break;
|
|
case 2:
|
|
SET_VALUE(HKEY_CURRENT_USER, 1);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
break;
|
|
case 3:
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 0);
|
|
break;
|
|
case 4:
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 1);
|
|
break;
|
|
case 5:
|
|
SET_VALUE(HKEY_CURRENT_USER, 0);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 1);
|
|
break;
|
|
case 6:
|
|
SET_VALUE(HKEY_CURRENT_USER, 1);
|
|
SET_VALUE(HKEY_LOCAL_MACHINE, 0);
|
|
break;
|
|
}
|
|
|
|
for (iItem = 0; iItem < _countof(s_Entries); ++iItem)
|
|
{
|
|
TEST_DoEntry(&s_Entries[iItem], fnSHGetRestriction);
|
|
}
|
|
}
|
|
|
|
START_TEST(SHGetRestriction)
|
|
{
|
|
HMODULE hSHLWAPI = LoadLibraryW(L"shlwapi.dll");
|
|
FN_SHGetRestriction fn = (FN_SHGetRestriction)GetProcAddress(hSHLWAPI, MAKEINTRESOURCEA(271));
|
|
INT iStage;
|
|
|
|
if (fn)
|
|
{
|
|
for (iStage = 0; iStage < 7; ++iStage)
|
|
TEST_SHGetRestriction_Stage(iStage, fn);
|
|
|
|
DELETE_VALUE(HKEY_CURRENT_USER);
|
|
DELETE_VALUE(HKEY_LOCAL_MACHINE);
|
|
}
|
|
else
|
|
{
|
|
skip("SHGetRestriction not found\n");
|
|
}
|
|
|
|
FreeLibrary(hSHLWAPI);
|
|
}
|