backend config for media control

Author: tpretz

server/proto/vm.go

@@ -57,6 +57,7 @@ type DeleteScriptReq struct {
 
 type GetVirtualDeviceRsp struct {
 	Network bool `json:"network"`
+	Media   bool `json:"media"`
 	Disk    bool `json:"disk"`
 }
 

server/service/vm/virtual-device.go

@@ -14,6 +14,7 @@ import (
 
 const (
 	virtualNetwork = "/boot/usb.rndis0"
+	virtualMedia   = "/boot/usb.disk0"
 	virtualDisk    = "/boot/usb.disk1"
 )
 
@@ -31,6 +32,19 @@ var (
 		"/etc/init.d/S03usbdev start",
 	}
 
+	mountMediaCommands = []string{
+		"touch /boot/usb.disk0",
+		"/etc/init.d/S03usbdev stop",
+		"/etc/init.d/S03usbdev start",
+	}
+
+	unmountMediaCommands = []string{
+		"/etc/init.d/S03usbdev stop",
+		"rm -rf /sys/kernel/config/usb_gadget/g0/configs/c.1/mass_storage.disk0",
+		"rm /boot/usb.disk0",
+		"/etc/init.d/S03usbdev start",
+	}
+
 	mountDiskCommands = []string{
 		"touch /boot/usb.disk1",
 		"/etc/init.d/S03usbdev stop",
@@ -49,10 +63,12 @@ func (s *Service) GetVirtualDevice(c *gin.Context) {
 	var rsp proto.Response
 
 	network, _ := isDeviceExist(virtualNetwork)
+	media, _ := isDeviceExist(virtualMedia)
 	disk, _ := isDeviceExist(virtualDisk)
 
 	rsp.OkRspWithData(c, &proto.GetVirtualDeviceRsp{
 		Network: network,
+		Media:   media,
 		Disk:    disk,
 	})
 	log.Debugf("get virtual device success")
@@ -80,6 +96,15 @@ func (s *Service) UpdateVirtualDevice(c *gin.Context) {
 		} else {
 			commands = unmountNetworkCommands
 		}
+	case "media":
+		device = virtualMedia
+
+		exist, _ := isDeviceExist(device)
+		if !exist {
+			commands = mountMediaCommands
+		} else {
+			commands = unmountMediaCommands
+		}
 	case "disk":
 		device = virtualDisk