Adding function to modify VM settings

pkg/virtualization/core/service/virtualmachine.go

@@ -272,3 +272,48 @@ func (vmms *VirtualSystemManagementService) SetMemoryMB(vm *virtualsystem.Virtua
 	err = vmms.ModifyVirtualSystemResourceEx(mem.WmiInstance, -1)
 	return
 }
+
+func (vmms *VirtualSystemManagementService) ModifyVirtualMachineSystemSettings(settings *virtualsystem.VirtualSystemSettingData) (err error) {
+
+	method, err := vmms.GetWmiMethod("ModifySystemSettings")
+	if err != nil {
+		return
+	}
+	defer method.Close()
+
+	embeddedInstance, err := settings.EmbeddedXMLInstance()
+	if err != nil {
+		return
+	}
+
+	inparams := wmi.WmiMethodParamCollection{}
+	inparams = append(inparams, wmi.NewWmiMethodParam("SystemSettings", embeddedInstance))
+	outparams := wmi.WmiMethodParamCollection{wmi.NewWmiMethodParam("Job", nil)}
+
+	result, err := method.Execute(inparams, outparams)
+	if err != nil {
+		return
+	}
+
+	if !(result.ReturnValue == 4096 || result.ReturnValue == 0) {
+		err = errors.Wrapf(errors.Failed, "Method failed with [%d]", result.ReturnValue)
+		return
+	}
+
+	if result.ReturnValue == 0 {
+		return
+	}
+
+	val, ok := result.OutMethodParams["Job"]
+	if !ok || val.Value == nil {
+		err = errors.Wrapf(errors.NotFound, "Job")
+		return
+	}
+	job, err := instance.GetWmiJob(vmms.GetWmiHost(), string(constant.Virtualization), val.Value.(string))
+	if err != nil {
+		return
+	}
+	defer job.Close()
+	err = job.WaitForJobCompletion(result.ReturnValue, -1)
+	return
+}

pkg/virtualization/core/service/virtualmachinemanagementservice_test.go

@@ -18,8 +18,8 @@ import (
 	"github.com/microsoft/wmi/pkg/virtualization/core/virtualsystem"
 	vswitchservice "github.com/microsoft/wmi/pkg/virtualization/network/service"
 	"github.com/microsoft/wmi/pkg/virtualization/network/virtualswitch"
-	//v2 "github.com/microsoft/wmi/server2019/root/virtualization/v2"
 	"github.com/nwoodmsft/iso9660"
+	//v2 "github.com/microsoft/wmi/server2019/root/virtualization/v2"
 )
 
 var (
@@ -684,7 +684,7 @@ func TestAddRemoveVirtualHardDisk(t *testing.T) {
 		defer os.RemoveAll(path)
 		t.Logf("Created vhd [%s]", path)
 
-		vhd, vhddrive, err := vmms.AttachVirtualHardDisk(vm, path)
+		vhd, vhddrive, err := vmms.AttachVirtualHardDisk(vm, path, virtualsystem.VirtualHardDiskType_DATADISK_VIRTUALHARDDISK)
 		if err != nil {
 			t.Fatalf("Failed [%+v]", err)
 		}
@@ -753,7 +753,7 @@ func TestAddRemoveVirtualHardDiskGen1(t *testing.T) {
 	defer os.RemoveAll(path)
 	t.Logf("Created vhd [%s]", path)
 
-	vhd, vhddrive, err := vmms.AttachVirtualHardDisk(vm, path)
+	vhd, vhddrive, err := vmms.AttachVirtualHardDisk(vm, path, virtualsystem.VirtualHardDiskType_DATADISK_VIRTUALHARDDISK)
 	if err != nil {
 		t.Fatalf("Failed [%+v]", err)
 	}
@@ -1266,3 +1266,105 @@ func TestAddRemoveGpuDdaGen1(t *testing.T) {
 	}
 	t.Logf("Detached GPU-DDA from [%s] VMs", "testGen1")
 }
+
+func TestModifyVirtualMachineSystemSettings(t *testing.T) {
+	vmms, err := GetVirtualSystemManagementService(whost)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	vm, err := vmms.GetVirtualMachineByName("test")
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	defer vm.Close()
+	t.Logf("Found [%s] VMs", "test")
+
+	// Get the original value
+	setting, err := vm.GetVirtualSystemSettingData()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	valBefore, err := setting.GetPropertyGuestControlledCacheTypes()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	t.Logf("Settings before change: %v", valBefore)
+
+	// Invert the setting
+	err = setting.SetPropertyGuestControlledCacheTypes(!valBefore)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	// Modify the VM settings
+	err = vmms.ModifyVirtualMachineSystemSettings(setting)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+
+	// Get the new value
+	setting, err = vm.GetVirtualSystemSettingData()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	valAfter, err := setting.GetPropertyGuestControlledCacheTypes()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	t.Logf("Settings after change: %v", valAfter)
+
+	if valBefore == valAfter {
+		t.Fatalf("Settings did not change after ModifyVirtualMachineSystemSettings(): before:%v after: %v", valBefore, valAfter)
+	}
+	t.Logf("Successfully changes vm setting from %v to %v", valBefore, valAfter)
+}
+
+func TestModifyVirtualMachineSystemSettingsGen1(t *testing.T) {
+	vmms, err := GetVirtualSystemManagementService(whost)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	vm, err := vmms.GetVirtualMachineByName("testGen1")
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	defer vm.Close()
+	t.Logf("Found [%s] VMs", "testGen1")
+
+	// Get the original value
+	setting, err := vm.GetVirtualSystemSettingData()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	valBefore, err := setting.GetPropertyGuestControlledCacheTypes()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	t.Logf("Settings before change: %v", valBefore)
+
+	// Invert the setting
+	err = setting.SetPropertyGuestControlledCacheTypes(!valBefore)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	// Modify the VM settings
+	err = vmms.ModifyVirtualMachineSystemSettings(setting)
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+
+	// Get the new value
+	setting, err = vm.GetVirtualSystemSettingData()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	valAfter, err := setting.GetPropertyGuestControlledCacheTypes()
+	if err != nil {
+		t.Fatalf("Failed [%+v]", err)
+	}
+	t.Logf("Settings after change: %v", valAfter)
+
+	if valBefore == valAfter {
+		t.Fatalf("Settings did not change after ModifyVirtualMachineSystemSettings(): before:%v after: %v", valBefore, valAfter)
+	}
+	t.Logf("Successfully changes vm setting from %v to %v", valBefore, valAfter)
+}