詳解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 系統的一個重要組件，通過其方便的接口可以實現對系統環境變量的添加、更改及讀取等功能。但同時也需要注意其需要管理員權限、易被攔截等問題，在使用時需要謹慎處理。