RegenerateUserEnvironment.cpp File Reference

#include "shelltest.h"
#include <userenv.h>
#include <debug.h>

Include dependency graph for RegenerateUserEnvironment.cpp:

Go to the source code of this file.

Macros
#define	NDEBUG

Typedefs
typedef BOOL(WINAPI *	PFN_REGENERATE_USER_ENVIRONMENT) (PVOID *pPrevEnv, BOOL bSetCurrentEnv)

Functions
static BOOL	IsValidEnvBlock (PVOID pEnv)
static void	Test_BasicCall (void)
static void	Test_PreviousEnvironmentReturned (void)
static void	Test_RegistryVariablePickedUp (void)
	START_TEST (RegenerateUserEnvironment)

Variables
static PFN_REGENERATE_USER_ENVIRONMENT	pRegenerateUserEnvironment = NULL

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file RegenerateUserEnvironment.cpp.

Typedef Documentation

PFN_REGENERATE_USER_ENVIRONMENT

typedef BOOL(WINAPI * PFN_REGENERATE_USER_ENVIRONMENT) (PVOID *pPrevEnv, BOOL bSetCurrentEnv)

Definition at line 14 of file RegenerateUserEnvironment.cpp.

Function Documentation

IsValidEnvBlock()

static BOOL IsValidEnvBlock ( PVOID  pEnv )

static

Definition at line 17 of file RegenerateUserEnvironment.cpp.

{
    LPWCH p = (LPWCH)pEnv;
 
    if (!p || *p == L'\0')
        return FALSE;
 
    /* Walk the block but bail out early if it looks corrupted */
    while (*p != L'\0')
    {
        SIZE_T len = wcslen(p);
        if (len == 0)
            return FALSE;
        p += len + 1;
    }
 
    return TRUE;
}

Referenced by Test_PreviousEnvironmentReturned().

START_TEST()

START_TEST

(

RegenerateUserEnvironment

)

Definition at line 135 of file RegenerateUserEnvironment.cpp.

{
    HMODULE hShell32 = GetModuleHandleW(L"shell32.dll");
 
    if (!hShell32)
        hShell32 = LoadLibraryW(L"shell32.dll");
 
    pRegenerateUserEnvironment =
        (PFN_REGENERATE_USER_ENVIRONMENT)GetProcAddress(hShell32, "RegenerateUserEnvironment");
 
    if (!pRegenerateUserEnvironment)
    {
        skip("RegenerateUserEnvironment not found in shell32.dll\n");
        return;
    }
 
    Test_BasicCall();
    Test_PreviousEnvironmentReturned();
    Test_RegistryVariablePickedUp();
}

Test_BasicCall()

static void Test_BasicCall ( void  )

static

Definition at line 36 of file RegenerateUserEnvironment.cpp.

{
    PVOID pEnv = NULL;
    BOOL ret;
 
    ret = pRegenerateUserEnvironment(&pEnv, TRUE);
    ok(ret != FALSE,
       "RegenerateUserEnvironment(&pEnv, TRUE) failed with last error %lu\n",
       GetLastError());
 
    pEnv = NULL;
    ret = pRegenerateUserEnvironment(&pEnv, FALSE);
    ok(ret != FALSE,
       "RegenerateUserEnvironment(&pEnv, FALSE) failed with last error %lu\n",
       GetLastError());
    if (pEnv)
        DestroyEnvironmentBlock(pEnv);
}

Referenced by START_TEST().

Test_PreviousEnvironmentReturned()

static void Test_PreviousEnvironmentReturned ( void  )

static

Definition at line 55 of file RegenerateUserEnvironment.cpp.

{
    PVOID pPrev = NULL;
    BOOL ret;
 
    ret = pRegenerateUserEnvironment(&pPrev, FALSE);
    ok(ret != FALSE,
       "RegenerateUserEnvironment(&pPrev, FALSE) failed with last error %lu\n",
       GetLastError());
 
    if (!ret)
    {
        skip("RegenerateUserEnvironment failed; skipping pPrev checks.\n");
        return;
    }
 
    ok(pPrev != NULL,
       "pPrev must not be NULL when bSetCurrentEnv is FALSE\n");
 
    if (!pPrev)
        return;
 
    ok(IsValidEnvBlock(pPrev),
       "pPrev does not point to a valid environment block\n");
 
    DestroyEnvironmentBlock(pPrev);
}

Referenced by START_TEST().

Test_RegistryVariablePickedUp()

static void Test_RegistryVariablePickedUp ( void  )

static

Definition at line 83 of file RegenerateUserEnvironment.cpp.

{
    static const WCHAR szKey[]     = L"Environment";
    static const WCHAR szVarName[] = L"__REGEN_TEST_VAR__";
    static const WCHAR szVarValue[]= L"ReactOS_RegenerateTest";
 
    HKEY hKey;
    LONG lRet;
    BOOL ret;
    WCHAR szBuf[64];
 
    lRet = RegOpenKeyExW(HKEY_CURRENT_USER, szKey, 0, KEY_SET_VALUE, &hKey);
    if (lRet != ERROR_SUCCESS)
    {
        skip("Cannot open HKCU\\Environment (error %ld); skipping registry test.\n", lRet);
        return;
    }
 
    /* Write canary value */
    lRet = RegSetValueExW(hKey, szVarName, 0, REG_SZ,
                          (const BYTE *)szVarValue,
                          (DWORD)((wcslen(szVarValue) + 1) * sizeof(WCHAR)));
    ok(lRet == ERROR_SUCCESS,
       "RegSetValueExW failed: %ld\n", lRet);
 
    if (lRet != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    /* Regenerate and apply to current process */
    PVOID pEnv = NULL;
    ret = pRegenerateUserEnvironment(&pEnv, TRUE);
    ok(ret != FALSE,
       "RegenerateUserEnvironment failed: %lu\n", GetLastError());
 
    /* The canary should be visible with GetEnvironmentVariableW */
    szBuf[0] = L'\0';
    GetEnvironmentVariableW(szVarName, szBuf, _countof(szBuf));
    ok(wcscmp(szBuf, szVarValue) == 0,
       "Expected env var '%ls' = '%ls', got '%ls'\n",
       szVarName, szVarValue, szBuf);
 
    /* Remove canary from registry */
    RegDeleteValueW(hKey, szVarName);
    RegCloseKey(hKey);
 
    PVOID pEnv2 = NULL;
    pRegenerateUserEnvironment(&pEnv2, TRUE);
}

Referenced by START_TEST().

Variable Documentation

pRegenerateUserEnvironment

PFN_REGENERATE_USER_ENVIRONMENT pRegenerateUserEnvironment = NULL

static

Definition at line 15 of file RegenerateUserEnvironment.cpp.

Referenced by START_TEST(), Test_BasicCall(), Test_PreviousEnvironmentReturned(), and Test_RegistryVariablePickedUp().