Release 202412292127

hw.go

@@ -0,0 +1,79 @@
+package kvm
+
+import (
+	"fmt"
+	"os"
+	"regexp"
+	"sync"
+	"time"
+)
+
+func extractSerialNumber() (string, error) {
+	content, err := os.ReadFile("/proc/cpuinfo")
+	if err != nil {
+		return "", err
+	}
+
+	r, err := regexp.Compile("Serial\\s*:\\s*(\\S+)")
+	if err != nil {
+		return "", fmt.Errorf("failed to compile regex: %w", err)
+	}
+
+	matches := r.FindStringSubmatch(string(content))
+	if len(matches) < 2 {
+		return "", fmt.Errorf("no serial found")
+	}
+
+	return matches[1], nil
+}
+
+func readOtpEntropy() ([]byte, error) {
+	content, err := os.ReadFile("/sys/bus/nvmem/devices/rockchip-otp0/nvmem")
+	if err != nil {
+		return nil, err
+	}
+	return content[0x17:0x1C], nil
+}
+
+var deviceID string
+var deviceIDOnce sync.Once
+
+func GetDeviceID() string {
+	deviceIDOnce.Do(func() {
+		serial, err := extractSerialNumber()
+		if err != nil {
+			logger.Warn("unknown serial number, the program likely not running on RV1106")
+			deviceID = "unknown_device_id"
+		} else {
+			deviceID = serial
+		}
+	})
+	return deviceID
+}
+
+func runWatchdog() {
+	file, err := os.OpenFile("/dev/watchdog", os.O_WRONLY, 0)
+	if err != nil {
+		logger.Warnf("unable to open /dev/watchdog: %v, skipping watchdog reset", err)
+		return
+	}
+	defer file.Close()
+	ticker := time.NewTicker(10 * time.Second)
+	defer ticker.Stop()
+	for {
+		select {
+		case <-ticker.C:
+			_, err = file.Write([]byte{0})
+			if err != nil {
+				logger.Errorf("error writing to /dev/watchdog, system may reboot: %v", err)
+			}
+		case <-appCtx.Done():
+			//disarm watchdog with magic value
+			_, err := file.Write([]byte("V"))
+			if err != nil {
+				logger.Errorf("failed to disarm watchdog, system may reboot: %v", err)
+			}
+			return
+		}
+	}
+}