SHGetRestriction.c File Reference

#include <apitest.h>
#include <shlwapi.h>

Include dependency graph for SHGetRestriction.c:

Go to the source code of this file.

Classes
struct	tagTEST_ENTRY

Macros
#define	REGKEY_POLICIES   L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies"
#define	REGKEY_POLICIES_EXPLORER   REGKEY_POLICIES L"\\Explorer"
#define	DELETE_VALUE(hBaseKey)    SHDeleteValueW((hBaseKey), REGKEY_POLICIES_EXPLORER, L"NoRun")
#define	SET_VALUE(hBaseKey, value)

Typedefs
typedef DWORD(WINAPI *	FN_SHGetRestriction) (LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue)
typedef BOOL(WINAPI *	FN_SHSettingsChanged) (LPCVOID unused, LPCVOID inpRegKey)
typedef struct tagTEST_ENTRY	TEST_ENTRY
typedef struct tagTEST_ENTRY *	PTEST_ENTRY

Functions
static DWORD	Candidate_SHGetRestriction (LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue)
static void	TEST_DoEntry (const TEST_ENTRY *entry, FN_SHGetRestriction fnSHGetRestriction)
static void	TEST_SHGetRestriction_Stage (INT iStage, FN_SHGetRestriction fnSHGetRestriction)
	START_TEST (SHGetRestriction)

Variables
static const TEST_ENTRY	s_Entries []

Macro Definition Documentation

DELETE_VALUE

#define DELETE_VALUE

(

hBaseKey

)

SHDeleteValueW((hBaseKey), REGKEY_POLICIES_EXPLORER, L"NoRun")

Definition at line 66 of file SHGetRestriction.c.

REGKEY_POLICIES

#define REGKEY_POLICIES   L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies"

Definition at line 11 of file SHGetRestriction.c.

REGKEY_POLICIES_EXPLORER

#define REGKEY_POLICIES_EXPLORER   REGKEY_POLICIES L"\\Explorer"

Definition at line 12 of file SHGetRestriction.c.

SET_VALUE

#define SET_VALUE	(		hBaseKey,
			value
	)

Value:

    do { \
    dwValue = (value); \
    SHSetValueW((hBaseKey), REGKEY_POLICIES_EXPLORER, L"NoRun", \
                REG_DWORD, &dwValue, sizeof(dwValue)); \
} while (0)

Definition at line 69 of file SHGetRestriction.c.

Typedef Documentation

FN_SHGetRestriction

typedef DWORD(WINAPI * FN_SHGetRestriction) (LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue)

Definition at line 14 of file SHGetRestriction.c.

FN_SHSettingsChanged

typedef BOOL(WINAPI * FN_SHSettingsChanged) (LPCVOID unused, LPCVOID inpRegKey)

Definition at line 15 of file SHGetRestriction.c.

PTEST_ENTRY

typedef struct tagTEST_ENTRY * PTEST_ENTRY

TEST_ENTRY

typedef struct tagTEST_ENTRY TEST_ENTRY

Function Documentation

Candidate_SHGetRestriction()

static DWORD Candidate_SHGetRestriction ( LPCWSTR  lpSubKey, LPCWSTR  lpSubName, LPCWSTR  lpValue  )

static

Definition at line 18 of file SHGetRestriction.c.

{
    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;
}

Referenced by TEST_DoEntry().

START_TEST()

START_TEST

(

SHGetRestriction

)

Definition at line 123 of file SHGetRestriction.c.

{
    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);
}

TEST_DoEntry()

static void TEST_DoEntry ( const TEST_ENTRY *  entry, FN_SHGetRestriction  fnSHGetRestriction  )

static

Definition at line 58 of file SHGetRestriction.c.

{
    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);
}

Referenced by TEST_SHGetRestriction_Stage().

TEST_SHGetRestriction_Stage()

static void TEST_SHGetRestriction_Stage ( INT  iStage, FN_SHGetRestriction  fnSHGetRestriction  )

static

Definition at line 76 of file SHGetRestriction.c.

{
    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);
    }
}

Referenced by START_TEST().

Variable Documentation

s_Entries

const TEST_ENTRY s_Entries[]

static

Initial value:

=
{
    { 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" },
}

Definition at line 43 of file SHGetRestriction.c.

Referenced by TEST_SHGetRestriction_Stage().