TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-02-18 02:48:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

util/winutil: add GetRegUserString/SetRegUserString accessors for storage and retrieval of string values in HKEY_CURRENT_USER

Browse Source 
Fixes #13187

Signed-off-by: Aaron Klotz <aaron@tailscale.com>
This commit is contained in:
Aaron Klotz 2024-08-13 15:38:00 -06:00
parent 2105773874
commit 8f6a2353d8
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
util/winutil/winutil_windows.go
View File

@ -52,7 +52,7 @@ func GetDesktopPID() (uint32, error) {
}
func getPolicyString(name string) (string, error) {
s, err := getRegStringInternal(regPolicyBase, name)
s, err := getRegStringInternal(registry.LOCAL_MACHINE, regPolicyBase, name)
if err != nil {
// Fall back to the legacy path
return getRegString(name)
@ -65,7 +65,7 @@ func getPolicyStringArray(name string) ([]string, error) {
}
func getRegString(name string) (string, error) {
s, err := getRegStringInternal(regBase, name)
s, err := getRegStringInternal(registry.LOCAL_MACHINE, regBase, name)
if err != nil {
return "", err
}
@ -89,8 +89,8 @@ func getRegInteger(name string) (uint64, error) {
return i, err
}
func getRegStringInternal(subKey, name string) (string, error) {
key, err := registry.OpenKey(registry.LOCAL_MACHINE, subKey, registry.READ)
func getRegStringInternal(key registry.Key, subKey, name string) (string, error) {
key, err := registry.OpenKey(key, subKey, registry.READ)
if err != nil {
if err != ErrNoValue {
log.Printf("registry.OpenKey(%v): %v", subKey, err)
@ -109,6 +109,24 @@ func getRegStringInternal(subKey, name string) (string, error) {
return val, nil
}
// GetRegUserString looks up a registry path in the current user key, or returns
// an empty string and error.
func GetRegUserString(name string) (string, error) {
return getRegStringInternal(registry.CURRENT_USER, regBase, name)
}
// SetRegUserString sets a SZ value identified by name in the current user key
// to the string specified by value.
func SetRegUserString(name, value string) error {
key, _, err := registry.CreateKey(registry.CURRENT_USER, regBase, registry.SET_VALUE)
if err != nil {
log.Printf("registry.CreateKey(%v): %v", regBase, err)
}
defer key.Close()
return key.SetStringValue(name, value)
}
// GetRegStrings looks up a registry value in the local machine path, or returns
// the given default if it can't.
func GetRegStrings(name string, defval []string) []string {