jsonrpc.go

@@ -479,15 +479,17 @@ func rpcSetUsbEmulationState(enabled bool) error {
 }
 
 func rpcSetUsbConfig(usbConfig UsbConfig) error {
+	log.Printf("[jsonrpc.go:rpcSetUsbConfig] called")
+
 	LoadConfig()
 	config.UsbConfig = usbConfig
 
-	err := UpdateGadgetConfig()
+	err2 := UpdateGadgetConfig()
-	if err != nil {
+	if err2 != nil {
-		return fmt.Errorf("failed to write gadget config: %w", err)
+		return fmt.Errorf("failed to write gadget config: %w", err2)
 	}
 
-	err = SaveConfig()
+	err := SaveConfig()
 	if err != nil {
 		return fmt.Errorf("failed to save usb config: %w", err)
 	}

ui/src/components/sidebar/settings.tsx

@@ -28,11 +28,11 @@ import { useRevalidator } from "react-router-dom";
 import { ShieldCheckIcon } from "@heroicons/react/20/solid";
 
 export function SettingsItem({
-  title,
+                               title,
-  description,
+                               description,
-  children,
+                               children,
-  className,
+                               className,
-}: {
+                             }: {
   title: string;
   description: string | React.ReactNode;
   children?: React.ReactNode;
@@ -103,12 +103,6 @@ export default function SettingsSidebar() {
   } | null>(null);
 
   const [usbEmulationEnabled, setUsbEmulationEnabled] = useState(false);
-  const getUsbEmulationState = useCallback(() => {
-    send("getUsbEmulationState", {}, resp => {
-      if ("error" in resp) return;
-      setUsbEmulationEnabled(resp.result as boolean);
-    });
-  }, [send]);
 
   const [usbConfig, setUsbConfig] = useState({
     usb_product_id: '',
@@ -118,6 +112,13 @@ export default function SettingsSidebar() {
     usb_name: '',
   })
 
+  const getUsbEmulationState = useCallback(() => {
+    send("getUsbEmulationState", {}, resp => {
+      if ("error" in resp) return;
+      setUsbEmulationEnabled(resp.result as boolean);
+    });
+  }, [send]);
+
   const handleUsbEmulationToggle = useCallback(
       (enabled: boolean) => {
         send("setUsbEmulationState", { enabled: enabled }, resp => {
@@ -224,7 +225,7 @@ export default function SettingsSidebar() {
         return;
       }
     });
-  }, [send]);
+  }, [send, usbConfig]);
 
   const handleSSHKeyChange = (newKey: string) => {
     setSSHKey(newKey);

usb.go

@@ -3,6 +3,7 @@ package kvm
 import (
 	"errors"
 	"fmt"
+	gadget "github.com/openstadia/go-usb-gadget"
 	"log"
 	"os"
 	"os/exec"
@@ -11,8 +12,6 @@ import (
 	"strings"
 	"sync"
 	"time"
-
-	gadget "github.com/openstadia/go-usb-gadget"
 )
 
 const configFSPath = "/sys/kernel/config"
@@ -20,19 +19,6 @@ const gadgetPath = "/sys/kernel/config/usb_gadget"
 const kvmGadgetPath = "/sys/kernel/config/usb_gadget/jetkvm"
 const configC1Path = "/sys/kernel/config/usb_gadget/jetkvm/configs/c.1"
 
-func mountConfigFS() error {
-	_, err := os.Stat(gadgetPath)
-	if os.IsNotExist(err) {
-		err = exec.Command("mount", "-t", "configfs", "none", configFSPath).Run()
-		if err != nil {
-			return fmt.Errorf("failed to mount configfs: %w", err)
-		}
-	} else {
-		return fmt.Errorf("unable to access usb gadget path: %w", err)
-	}
-	return nil
-}
-
 func init() {
 	_ = os.MkdirAll(imagesFolder, 0755)
 	udcs := gadget.GetUdcs()
@@ -86,7 +72,7 @@ func UpdateGadgetConfig() error {
 		return err
 	}
 
-	log.Printf("Successfully updated usb string attributes: %s", strAttrs)
+	log.Printf("Successfully updated string attributes: %s", strAttrs)
 
 	err = rebindUsb()
 	if err != nil {
@@ -96,6 +82,22 @@ func UpdateGadgetConfig() error {
 	return nil
 }
 
+func mountConfigFS() error {
+	logger.Infof("Checking for gadgetPath: %s ...", gadgetPath)
+	_, err := os.Stat(gadgetPath)
+	if os.IsNotExist(err) {
+		logger.Infof("running mount command...")
+		err = exec.Command("mount", "-t", "configfs", "none", configFSPath).Run()
+		if err != nil {
+			return fmt.Errorf("failed to mount configfs: %w", err)
+		}
+	} else {
+		return fmt.Errorf("unable to access usb gadget path: %w", err)
+	}
+	logger.Infof("Successfully mounted usb gadget at %s", gadgetPath)
+	return nil
+}
+
 func writeGadgetAttrs(basePath string, attrs [][]string) error {
 	for _, item := range attrs {
 		filePath := filepath.Join(basePath, item[0])
@@ -117,30 +119,35 @@ func writeGadgetConfig() error {
 		return err
 	}
 
-	err = writeGadgetAttrs(kvmGadgetPath, [][]string{
+	LoadConfig()
-		{"bcdUSB", "0x0200"},   //USB 2.0
+	gadgetAttrs := [][]string{
-		{"idVendor", "0x1d6b"}, //The Linux Foundation
+		{"bcdUSB", "0x0200"}, //USB 2.0
-		{"idProduct", "0104"},  //Multifunction Composite Gadget¬
+		{"idVendor", config.UsbConfig.UsbVendorId},
+		{"idProduct", config.UsbConfig.UsbProductId},
 		{"bcdDevice", "0100"},
-	})
+	}
+	err = writeGadgetAttrs(kvmGadgetPath, gadgetAttrs)
 	if err != nil {
 		return err
 	}
 
+	logger.Infof("Successfully wrote gadget attributes: %s", gadgetAttrs)
 	gadgetStringsPath := filepath.Join(kvmGadgetPath, "strings", "0x409")
 	err = os.MkdirAll(gadgetStringsPath, 0755)
 	if err != nil {
 		return err
 	}
 
-	err = writeGadgetAttrs(gadgetStringsPath, [][]string{
+	strAttrs := [][]string{
-		{"serialnumber", GetDeviceID()},
+		{"serialnumber", config.UsbConfig.UsbSerialNumber},
-		{"manufacturer", "JetKVM"},
+		{"manufacturer", config.UsbConfig.UsbManufacturer},
-		{"product", "JetKVM USB Emulation Device"},
+		{"product", config.UsbConfig.UsbName},
-	})
+	}
+	err = writeGadgetAttrs(gadgetStringsPath, strAttrs)
 	if err != nil {
 		return err
 	}
+	logger.Infof("Successfully wrote string attributes: %s", strAttrs)
 
 	configC1StringsPath := path.Join(configC1Path, "strings", "0x409")
 	err = os.MkdirAll(configC1StringsPath, 0755)
@@ -254,11 +261,27 @@ func writeGadgetConfig() error {
 }
 
 func rebindUsb() error {
+	unbindErr := unbindUsb()
+	if unbindErr != nil {
+		return unbindErr
+	}
+	bindErr := bindUsb()
+	if bindErr != nil {
+		return bindErr
+	}
+	return nil
+}
+
+func unbindUsb() error {
 	err := os.WriteFile("/sys/bus/platform/drivers/dwc3/unbind", []byte(udc), 0644)
 	if err != nil {
 		return err
 	}
-	err = os.WriteFile("/sys/bus/platform/drivers/dwc3/bind", []byte(udc), 0644)
+	return nil
+}
+
+func bindUsb() error {
+	err := os.WriteFile("/sys/bus/platform/drivers/dwc3/bind", []byte(udc), 0644)
 	if err != nil {
 		return err
 	}