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
|
||
|
|
}
|