详解Userenv

一、Userenv 简介

Userenv 是 Windows 系统中的一个用于处理用户环境的核心组件, 它主要负责以下三个方面事情：

1、用户配置文件的加载

Userenv 会根据配置文件（如 ntuser.dat、usrclass.dat、default user.dat、all user.dat 等）创建当前用户的环境配置。

2、组策略的应用

当用户登录或计算机启动时，Userenv 会读取组策略的配置信息，并将其应用于计算机或用户的环境。

3、用户已有配置文件的重定向

如果用户的配置文件存储在网络共享文件夹中，Userenv 会根据管理员设置为该用户的配置文件重定向到网络位置，以提高系统性能。

二、Userenv 的使用场景

在我们的实际开发中，Userenv 组件主要用于以下几个方面：

1、实现 Windows C++ 程序自启动

可以通过注册表将 Windows C++ 程序加入自启动项，使系统启动时自动运行对应程序。

LONG SetAutoRun(const WCHAR* lpszAppName, const WCHAR* lpszPath)
{
    HKEY hKey = nullptr;
    WCHAR szKeyName[MAX_PATH];
    LONG lRet = RegCreateKeyEx(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", 0, nullptr,
                               REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS | KEY_WOW64_64KEY, nullptr, &hKey, nullptr);
    if (lRet != ERROR_SUCCESS)
    {
        return lRet;
    }
    lRet = RegSetValueEx(hKey, lpszAppName, 0, REG_SZ, (BYTE*)lpszPath, (DWORD)(wcslen(lpszPath) * sizeof(WCHAR)));
    RegCloseKey(hKey);
    return lRet;
}

2、设置 Windows 系统环境变量

可以通过 Userenv 的接口将环境变量写入系统的环境变量中，从而影响系统的运作。

BOOL SetEnvironmentVariableValue(LPCTSTR pszVariable, LPCTSTR pszValue)
{
    BOOL bResult = FALSE;
    LPTSTR pszOldValue = NULL;
    DWORD cbOldValue = 0;
    HKEY hKey;
    DWORD dwDisp;

    if (ERROR_SUCCESS ==
        RegCreateKeyEx(HKEY_LOCAL_MACHINE, _T("System\\CurrentControlSet\\Control\\Session Manager\\Environment"), 0, NULL,
            REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hKey, &dwDisp))
    {
        // First, get the value
        DWORD dwCount = 0;
        if (ERROR_SUCCESS != ::RegQueryValueEx(hKey, pszVariable, NULL, NULL, NULL, &dwCount))
            dwCount = 0;

        if (dwCount > 0)
        {
            cbOldValue = dwCount;
            pszOldValue = new TCHAR[dwCount];
            ::RegQueryValueEx(hKey, pszVariable, NULL, NULL, (LPBYTE)pszOldValue, &cbOldValue);
            pszOldValue[cbOldValue / sizeof(TCHAR) - 1] = 0;
        }
        if (pszValue != NULL && *pszValue != _T('\0'))
        {
            // Set the new value
            if (ERROR_SUCCESS == ::RegSetValueEx(hKey, pszVariable, NULL, REG_SZ, (LPBYTE)pszValue, (DWORD)(_tcslen(pszValue) + 1) * sizeof(TCHAR)))
                bResult = TRUE;
        }   
        RegCloseKey(hKey);
    }
    if (pszOldValue != NULL)
        delete[] pszOldValue;
    return bResult;
}

3、获取 Windows 系统环境变量

Userenv 的接口还可以获取当前系统环境变量的值，方便后续程序的开发。

std::wstring GetEnvironmentVariable(LPCTSTR key)
{
    std::wstring value;

    WCHAR* pValue = nullptr;
    if (::GetEnvironmentVariableW(key, pValue, 0))
    {
        pValue = new (std::nothrow) WCHAR[::GetEnvironmentVariableW(key, pValue, 0)];
        if (pValue && ::GetEnvironmentVariableW(key, pValue, ::GetEnvironmentVariableW(key, pValue, 0)))
        {
            value = std::wstring(pValue);
        }
        delete[]pValue;
    }
    return value;
}

三、Userenv 的注意事项

在使用 Userenv 的过程中，我们也需要注意以下几个方面：

1、Userenv 接口需要以管理员权限运行

在使用 Userenv 的接口时，需要程序以管理员权限运行，否则无法获得系统环境变量的写入权限。

2、Userenv 接口可能会被 AntiVirus 软件拦截

由于 Userenv 接口具有强大的操作权限，因此在某些情况下可能会被 AntiVirus 软件拦截或误报。

3、Userenv 接口已被弃用

在 Windows10 中，Userenv 组件已经被弃用，推荐使用新的设置模块 UserEnvironment。

四、总结

Userenv 是 Windows 系统的一个重要组件，通过其方便的接口可以实现对系统环境变量的添加、更改及读取等功能。但同时也需要注意其需要管理员权限、易被拦截等问题，在使用时需要谨慎处理。