Add support for Luckfox PicoKVM

Signed-off-by: luckfox-eng29 <eng29@luckfox.com>

io.go

@@ -0,0 +1,118 @@
+package kvm
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+)
+
+const (
+	gpioBasePath  = "/sys/class/gpio"
+	ledGreenPath  = "/sys/class/leds/led:g"
+	ledYellowPath = "/sys/class/leds/led:y"
+)
+
+func exportGPIO(pin int) error {
+	exportFile := gpioBasePath + "/export"
+	return os.WriteFile(exportFile, []byte(strconv.Itoa(pin)), 0644)
+}
+
+func unexportGPIO(pin int) error {
+	unexportFile := gpioBasePath + "/unexport"
+	return os.WriteFile(unexportFile, []byte(strconv.Itoa(pin)), 0644)
+}
+
+func isGPIOExported(pin int) bool {
+	gpioPath := fmt.Sprintf("%s/gpio%d", gpioBasePath, pin)
+	_, err := os.Stat(gpioPath)
+	return err == nil
+}
+
+func setGPIODirection(pin int, direction string) error {
+	if !isGPIOExported(pin) {
+		if err := exportGPIO(pin); err != nil {
+			return fmt.Errorf("failed to export GPIO: %v", err)
+		}
+	}
+	directionFile := fmt.Sprintf("%s/gpio%d/direction", gpioBasePath, pin)
+	return os.WriteFile(directionFile, []byte(direction), 0644)
+}
+
+func setGPIOValue(pin int, status bool) error {
+	var value int
+	if status {
+		value = 1
+	} else {
+		value = 0
+	}
+	if !isGPIOExported(pin) {
+		if err := exportGPIO(pin); err != nil {
+			return fmt.Errorf("failed to export GPIO: %v", err)
+		}
+		if err := setGPIODirection(pin, "out"); err != nil {
+			return fmt.Errorf("failed to set GPIO direction: %v", err)
+		}
+	}
+	valueFile := fmt.Sprintf("%s/gpio%d/value", gpioBasePath, pin)
+	return os.WriteFile(valueFile, []byte(strconv.Itoa(value)), 0644)
+}
+
+func setLedMode(ledConfigPath string, mode string) error {
+	if mode == "network-link" {
+		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED trigger: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED device name: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/link", []byte("1"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED link: %v", err)
+		}
+	} else if mode == "network-tx" {
+		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED trigger: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED device name: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/tx", []byte("1"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED tx: %v", err)
+		}
+	} else if mode == "network-rx" {
+		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED trigger: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED device name: %v", err)
+		}
+		err = os.WriteFile(ledConfigPath+"/rx", []byte("1"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED rx: %v", err)
+		}
+	} else if mode == "kernel-activity" {
+		err := os.WriteFile(ledConfigPath+"/trigger", []byte("activity"), 0644)
+		if err != nil {
+			return fmt.Errorf("failed to set LED trigger: %v", err)
+		}
+	} else {
+		return fmt.Errorf("invalid LED mode: %s", mode)
+	}
+	return nil
+}
+
+func initGPIO() {
+	LoadConfig()
+	// IO0: GPIO58 IO1: GPIO59
+	_ = setGPIOValue(58, config.IO0Status)
+	_ = setGPIOValue(59, config.IO1Status)
+	_ = setLedMode(ledYellowPath, config.LEDYellowMode)
+	_ = setLedMode(ledGreenPath, config.LEDGreenMode)
+}