Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-05-20 20:31:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

[SHLWAPI_APITEST] Add SHGetRestriction testcase (#6557)

Browse source 
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.
This commit is contained in:
Katayama Hirofumi MZ 2024-02-29 20:28:10 +09:00 committed by GitHub
parent b944563c7b
commit ac870f986d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 146 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
modules/rostests/apitests/shlwapi/CMakeLists.txt
View file

@ -12,6 +12,7 @@ list(APPEND SOURCE
PathUnExpandEnvStrings.c
PathUnExpandEnvStringsForUser.c
SHAreIconsEqual.c
SHGetRestriction.c
SHLoadIndirectString.c
SHLoadRegUIString.c
SHPropertyBag.cpp

143
modules/rostests/apitests/shlwapi/SHGetRestriction.c Normal file
View file

@ -0,0 +1,143 @@
/*
* 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);
}

2
modules/rostests/apitests/shlwapi/testlist.c
View file

@ -8,6 +8,7 @@ extern void func_isuncpathservershare(void);
extern void func_PathUnExpandEnvStrings(void);
extern void func_PathUnExpandEnvStringsForUser(void);
extern void func_SHAreIconsEqual(void);
extern void func_SHGetRestriction(void);
extern void func_SHLoadIndirectString(void);
extern void func_SHLoadRegUIString(void);
extern void func_SHPropertyBag(void);
@ -22,6 +23,7 @@ const struct test winetest_testlist[] =
{ "PathUnExpandEnvStrings", func_PathUnExpandEnvStrings },
{ "PathUnExpandEnvStringsForUser", func_PathUnExpandEnvStringsForUser },
{ "SHAreIconsEqual", func_SHAreIconsEqual },
{ "SHGetRestriction", func_SHGetRestriction },
{ "SHLoadIndirectString", func_SHLoadIndirectString },
{ "SHLoadRegUIString", func_SHLoadRegUIString },
{ "SHPropertyBag", func_SHPropertyBag },