feat: usb dynamic config (#248)

jetkvm · Mar 10, 2025 · c5cec99 · c5cec99
2 parents 5c7acca + d1948ad
commit c5cec99
Show file tree

Hide file tree

Showing 17 changed files with 1,224 additions and 502 deletions.
diff --git a/config.go b/config.go
@@ -3,6 +3,7 @@ package kvm
 import (
 	"encoding/json"
 	"fmt"
+	"kvm/internal/usbgadget"
 	"os"
 	"sync"
 )
@@ -12,32 +13,26 @@ type WakeOnLanDevice struct {
 	MacAddress string `json:"macAddress"`
 }
 
-type UsbConfig struct {
-	VendorId     string `json:"vendor_id"`
-	ProductId    string `json:"product_id"`
-	SerialNumber string `json:"serial_number"`
-	Manufacturer string `json:"manufacturer"`
-	Product      string `json:"product"`
-}
-
 type Config struct {
-	CloudURL             string            `json:"cloud_url"`
-	CloudAppURL          string            `json:"cloud_app_url"`
-	CloudToken           string            `json:"cloud_token"`
-	GoogleIdentity       string            `json:"google_identity"`
-	JigglerEnabled       bool              `json:"jiggler_enabled"`
-	AutoUpdateEnabled    bool              `json:"auto_update_enabled"`
-	IncludePreRelease    bool              `json:"include_pre_release"`
-	HashedPassword       string            `json:"hashed_password"`
-	LocalAuthToken       string            `json:"local_auth_token"`
-	LocalAuthMode        string            `json:"localAuthMode"` //TODO: fix it with migration
-	WakeOnLanDevices     []WakeOnLanDevice `json:"wake_on_lan_devices"`
-	EdidString           string            `json:"hdmi_edid_string"`
-	ActiveExtension      string            `json:"active_extension"`
-	DisplayMaxBrightness int               `json:"display_max_brightness"`
-	DisplayDimAfterSec   int               `json:"display_dim_after_sec"`
-	DisplayOffAfterSec   int               `json:"display_off_after_sec"`
-	UsbConfig            *UsbConfig        `json:"usb_config"`
+	CloudURL             string             `json:"cloud_url"`
+	CloudAppURL          string             `json:"cloud_app_url"`
+	CloudToken           string             `json:"cloud_token"`
+	GoogleIdentity       string             `json:"google_identity"`
+	JigglerEnabled       bool               `json:"jiggler_enabled"`
+	AutoUpdateEnabled    bool               `json:"auto_update_enabled"`
+	IncludePreRelease    bool               `json:"include_pre_release"`
+	HashedPassword       string             `json:"hashed_password"`
+	LocalAuthToken       string             `json:"local_auth_token"`
+	LocalAuthMode        string             `json:"localAuthMode"` //TODO: fix it with migration
+	WakeOnLanDevices     []WakeOnLanDevice  `json:"wake_on_lan_devices"`
+	EdidString           string             `json:"hdmi_edid_string"`
+	ActiveExtension      string             `json:"active_extension"`
+	DisplayMaxBrightness int                `json:"display_max_brightness"`
+	DisplayDimAfterSec   int                `json:"display_dim_after_sec"`
+	DisplayOffAfterSec   int                `json:"display_off_after_sec"`
+	TLSMode              string             `json:"tls_mode"`
+	UsbConfig            *usbgadget.Config  `json:"usb_config"`
+	UsbDevices           *usbgadget.Devices `json:"usb_devices"`
 }
 
 const configPath = "/userdata/kvm_config.json"
@@ -50,13 +45,20 @@ var defaultConfig = &Config{
 	DisplayMaxBrightness: 64,
 	DisplayDimAfterSec:   120,  // 2 minutes
 	DisplayOffAfterSec:   1800, // 30 minutes
-	UsbConfig: &UsbConfig{
+	TLSMode:              "",
+	UsbConfig: &usbgadget.Config{
 		VendorId:     "0x1d6b", //The Linux Foundation
 		ProductId:    "0x0104", //Multifunction Composite Gadget
 		SerialNumber: "",
 		Manufacturer: "JetKVM",
 		Product:      "USB Emulation Device",
 	},
+	UsbDevices: &usbgadget.Devices{
+		AbsoluteMouse: true,
+		RelativeMouse: true,
+		Keyboard:      true,
+		MassStorage:   true,
+	},
 }
 
 var (
@@ -95,6 +97,10 @@ func LoadConfig() {
 		loadedConfig.UsbConfig = defaultConfig.UsbConfig
 	}
 
+	if loadedConfig.UsbDevices == nil {
+		loadedConfig.UsbDevices = defaultConfig.UsbDevices
+	}
+
 	config = &loadedConfig
 }