SHRestricted.cpp File Reference

#include "shelltest.h"
#include <versionhelpers.h>

Include dependency graph for SHRestricted.cpp:

Go to the source code of this file.

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_SHRestricted) (RESTRICTIONS rest)
typedef BOOL(WINAPI *	FN_SHSettingsChanged) (LPCVOID unused, LPCVOID inpRegKey)

Functions
static VOID	TEST_SHRestricted (FN_SHRestricted fnGetValue, FN_SHSettingsChanged fnRefresh)
	START_TEST (SHRestricted)

Macro Definition Documentation

DELETE_VALUE

#define DELETE_VALUE

(

hBaseKey

)

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

Definition at line 17 of file SHRestricted.cpp.

REGKEY_POLICIES

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

Definition at line 11 of file SHRestricted.cpp.

REGKEY_POLICIES_EXPLORER

#define REGKEY_POLICIES_EXPLORER   REGKEY_POLICIES L"\\Explorer"

Definition at line 12 of file SHRestricted.cpp.

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 20 of file SHRestricted.cpp.

Typedef Documentation

FN_SHRestricted

typedef DWORD(WINAPI * FN_SHRestricted) (RESTRICTIONS rest)

Definition at line 14 of file SHRestricted.cpp.

FN_SHSettingsChanged

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

Definition at line 15 of file SHRestricted.cpp.

Function Documentation

START_TEST()

START_TEST

(

SHRestricted

)

Definition at line 80 of file SHRestricted.cpp.

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

TEST_SHRestricted()

static VOID TEST_SHRestricted ( FN_SHRestricted  fnGetValue, FN_SHSettingsChanged  fnRefresh  )

static

Definition at line 27 of file SHRestricted.cpp.

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

Referenced by START_TEST().