mirror of
https://github.com/luckfox-eng29/kvm.git
synced 2026-01-18 03:28:19 +01:00
120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
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 {
|
|
switch mode {
|
|
case "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)
|
|
}
|
|
case "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)
|
|
}
|
|
case "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)
|
|
}
|
|
case "kernel-activity":
|
|
err := os.WriteFile(ledConfigPath+"/trigger", []byte("activity"), 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to set LED trigger: %v", err)
|
|
}
|
|
default:
|
|
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)
|
|
}
|