Rust 设置 Windows 环境变量，告别重启！

windows

2024-03-09 06:28:48

使用 Rust 设置 Windows 环境变量，无需重启

简介

环境变量在 Windows 操作系统中扮演着至关重要的角色，它们允许程序访问系统范围内的配置和设置。通过使用 Rust 语言，您可以动态地更新这些变量，从而增强您的应用程序的可配置性和灵活性。

更新注册表

要修改环境变量，您需要编辑 Windows 注册表中的特定项。Rust 的 RegKey 库提供了便捷的方法来操作注册表。以下是更新注册表项的步骤：

使用 RegKey::predef 创建对 HKEY_LOCAL_MACHINE 项的引用。
创建或打开子项 System\\CurrentControlSet\\Control\\Session Manager\\Environment。
使用 set_value 方法设置或更新环境变量的值。

广播 WM_SETTINGCHANGE 消息

更新注册表后，您需要广播 WM_SETTINGCHANGE 消息来通知系统有关更改。Rust 的 winapi crate 为此提供了 SendMessageTimeoutW 函数。以下是如何广播消息：

构建一个指向环境子项的宽字符串消息。
使用 SendMessageTimeoutW 函数广播消息，等待 1 秒钟的响应。

示例代码

以下 Rust 代码展示了如何设置环境变量：

use winreg::{RegKey, HKEY_LOCAL_MACHINE};

fn set_environment_variable(name: &str, value: &str) -> Result<(), winreg::Error> {
    let hklm = RegKey::predef(HKEY_LOCAL_MACHINE);
    let mut environment = hklm.create_subkey("System\\CurrentControlSet\\Control\\Session Manager\\Environment")?;

    environment.set_value(name, value)?;

    let message = w!("System\\CurrentControlSet\\Control\\Session Manager\\Environment");
    let mut result = 0;
    let success = unsafe {
        SendMessageTimeoutW(
            HWND_BROADCAST,
            WM_SETTINGCHANGE,
            0,
            message as _,
            0,
            1000,
            &mut result,
        )
    };

    if success == 0 {
        Err(winreg::Error::from_raw_os_error(GetLastError()))
    } else {
        Ok(())
    }
}