starainrt
7e6cc73106
- 新增自启动幂等配置、统一错误语义、进程等待和进程树终止能力 - 增强服务生命周期管理，支持等待状态、重启、幂等创建和配置更新 - 新增 NTFS 卷索引、文件 ID 解析、文件遍历、USN 变更监听和 bookmark 持久化 - 修复 NTFS boot sector、fragment、MFT、USN 解析边界和路径重建问题 - 补充权限、进程、服务、NTFS 解析和工作流回归测试 - 增加 Windows 测试脚本和管理员 NTFS smoke 验证脚本 - 升级 Go 兼容版本到 1.18，并更新 stario、win32api 及相关间接依赖
45 lines
960 B
Go
45 lines
960 B
Go
package wincmd
|
|
|
|
import (
|
|
"golang.org/x/sys/windows/registry"
|
|
)
|
|
|
|
// EnsureAutoRun idempotently enables or disables HKLM Run startup.
|
|
func EnsureAutoRun(key, path string, enable bool) (changed bool, err error) {
|
|
current, exists, err := getAutoRunValue(key)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
if enable {
|
|
if exists && current == path {
|
|
return false, nil
|
|
}
|
|
_, err := AutoRun(key, path)
|
|
return err == nil, err
|
|
}
|
|
|
|
if !exists {
|
|
return false, nil
|
|
}
|
|
_, err = DeleteAutoRun(key)
|
|
return err == nil, err
|
|
}
|
|
|
|
func getAutoRunValue(key string) (value string, exists bool, err error) {
|
|
reg, err := registry.OpenKey(registry.LOCAL_MACHINE, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.ALL_ACCESS)
|
|
if err != nil {
|
|
return "", false, err
|
|
}
|
|
defer reg.Close()
|
|
|
|
value, _, err = reg.GetStringValue(key)
|
|
if err != nil {
|
|
if err == registry.ErrNotExist {
|
|
return "", false, nil
|
|
}
|
|
return "", false, err
|
|
}
|
|
return value, true, nil
|
|
}
|