Change the string passing for rootdisk from disk number to diskname +…

… partition

Signed-off-by: Bella Khizgiyaev <[email protected]>

pkg/controller/plan/adapter/vsphere/BUILD.bazel

@@ -39,6 +39,7 @@ go_library(
         "//vendor/github.com/vmware/govmomi/vim25/types",
         "//vendor/k8s.io/api/core/v1:core",
         "//vendor/k8s.io/apimachinery/pkg/api/resource",
+        "//vendor/k8s.io/utils/ptr",
         "//vendor/kubevirt.io/api/core/v1:core",
         "//vendor/kubevirt.io/containerized-data-importer-api/pkg/apis/core/v1beta1",
         "//vendor/sigs.k8s.io/controller-runtime/pkg/client",

pkg/controller/plan/adapter/vsphere/builder.go

@@ -9,8 +9,7 @@ import (
 	"regexp"
 	"sort"
 	"strings"
-	"strconv"
-
+	"unicode"
 
 	api "github.com/konveyor/forklift-controller/pkg/apis/forklift/v1beta1"
 	"github.com/konveyor/forklift-controller/pkg/apis/forklift/v1beta1/plan"
@@ -32,6 +31,7 @@ import (
 	"github.com/vmware/govmomi/vim25/types"
 	core "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
+	"k8s.io/utils/ptr"
 	cnv "kubevirt.io/api/core/v1"
 	cdi "kubevirt.io/containerized-data-importer-api/pkg/apis/core/v1beta1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -660,15 +660,10 @@ func (r *Builder) mapDisks(vm *model.VM, vmRef ref.Ref, persistentVolumeClaims [
 	var bootDisk int
 	for _, vmConf := range r.Migration.Status.VMs {
 		if vmConf.ID == vmRef.ID {
-			var err error
-			bootDisk, err = strconv.Atoi(vmConf.RootDisk)
-			if err != nil {
-				bootDisk = 1
-			}
+			bootDisk = getDeviceNumber(vmConf.RootDisk)
 			break
 		}
 	}
-	bootOrder := func(order uint) *uint { return &order }
 
 	for i, disk := range disks {
 		pvc := pvcMap[r.baseVolume(disk.File)]
@@ -692,8 +687,8 @@ func (r *Builder) mapDisks(vm *model.VM, vmRef ref.Ref, persistentVolumeClaims [
 			},
 		}
 		if bootDisk == i+1 {
-				kubevirtDisk.BootOrder = bootOrder(1)
-			}
+			kubevirtDisk.BootOrder = ptr.To(uint(1))
+		}
 		kVolumes = append(kVolumes, volume)
 		kDisks = append(kDisks, kubevirtDisk)
 	}
@@ -945,3 +940,18 @@ func (r *Builder) GetPopulatorTaskName(pvc *core.PersistentVolumeClaim) (taskNam
 	err = planbase.VolumePopulatorNotSupportedError
 	return
 }
+
+func getDeviceNumber(deviceString string) int {
+	if !strings.HasPrefix(deviceString, "/dev/sd") || len(deviceString) < 8 {
+		// In case we encounter an issue detecting the root disk order,
+		// we will return zero to avoid failing the migration due to boot orde
+		return 0
+	}
+
+	for i := 7; i < len(deviceString); i++ {
+		if unicode.IsLetter(rune(deviceString[i])) {
+			return int(deviceString[i] - 'a' + 1)
+		}
+	}
+	return 0
+}

pkg/controller/plan/adapter/vsphere/vsphere_suite_test.go

@@ -11,3 +11,26 @@ func TestVsphere(t *testing.T) {
 	RegisterFailHandler(Fail)
 	RunSpecs(t, "vSphere Suite")
 }
+
+func TestGetDeviceNumber(t *testing.T) {
+	tests := []struct {
+		input    string
+		expected int
+	}{
+		{"/dev/sda", 1},
+		{"/dev/sdb", 2},
+		{"/dev/sdz", 26},
+		{"/dev/sda1", 1},
+		{"/dev/sda5", 1},
+		{"/dev/sdb2", 2},
+		{"/dev/sdza", 26},
+		{"/dev/sdzb", 26},
+	}
+
+	for _, test := range tests {
+		result := getDeviceNumber(test.input)
+		if result != test.expected {
+			t.Errorf("For input '%s', expected %d, but got %d", test.input, test.expected, result)
+		}
+	}
+}

pkg/controller/plan/kubevirt.go

@@ -1597,19 +1597,11 @@ func (r *KubeVirt) guestConversionPod(vm *plan.VMStatus, vmVolumes []cnv.Volume,
 	}
 
 	if vm.RootDisk != "" {
-		diskNum, errDisk := strconv.Atoi(vm.RootDisk)
-		if errDisk != nil {
-			return
-		}
-
-		if diskNum > 0 {
-
-			environment = append(environment,
-				core.EnvVar{
-					Name:  "V2V_RootDisk",
-					Value: vm.RootDisk,
-				})
-		}
+		environment = append(environment,
+			core.EnvVar{
+				Name:  "V2V_RootDisk",
+				Value: vm.RootDisk,
+			})
 	}
 	// pod annotations
 	annotations := map[string]string{}

pkg/controller/provider/container/vsphere/model.go

@@ -701,7 +701,6 @@ func (v *VmAdapter) Apply(u types.ObjectUpdate) {
 // Update virtual disk devices.
 func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 	disks := []model.Disk{}
-	//var diskNum int32 = 1
 	for _, dev := range devArray.VirtualDevice {
 		switch dev.(type) {
 		case *types.VirtualDisk:
@@ -713,7 +712,6 @@ func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 					File:     backing.FileName,
 					Capacity: disk.CapacityInBytes,
 					Mode:     backing.DiskMode,
-					//Number:   diskNum,
 				}
 				if backing.Datastore != nil {
 					datastoreId, _ := sanitize(backing.Datastore.Value)
@@ -723,15 +721,13 @@ func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 					}
 				}
 				disks = append(disks, md)
-				//diskNum++
 			case *types.VirtualDiskFlatVer2BackingInfo:
 				md := model.Disk{
 					Key:      disk.Key,
 					File:     backing.FileName,
 					Capacity: disk.CapacityInBytes,
 					Shared:   backing.Sharing != "sharingNone",
 					Mode:     backing.DiskMode,
-					//Number:   diskNum,
 				}
 				if backing.Datastore != nil {
 					datastoreId, _ := sanitize(backing.Datastore.Value)
@@ -741,7 +737,6 @@ func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 					}
 				}
 				disks = append(disks, md)
-				//diskNum++
 			case *types.VirtualDiskRawDiskMappingVer1BackingInfo:
 				md := model.Disk{
 					Key:      disk.Key,
@@ -750,7 +745,6 @@ func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 					Shared:   backing.Sharing != "sharingNone",
 					Mode:     backing.DiskMode,
 					RDM:      true,
-					//Number:   diskNum,
 				}
 				if backing.Datastore != nil {
 					datastoreId, _ := sanitize(backing.Datastore.Value)
@@ -760,18 +754,15 @@ func (v *VmAdapter) updateDisks(devArray *types.ArrayOfVirtualDevice) {
 					}
 				}
 				disks = append(disks, md)
-				//diskNum++
 			case *types.VirtualDiskRawDiskVer2BackingInfo:
 				md := model.Disk{
 					Key:      disk.Key,
 					File:     backing.DescriptorFileName,
 					Capacity: disk.CapacityInBytes,
 					Shared:   backing.Sharing != "sharingNone",
 					RDM:      true,
-					//Number:   diskNum,
 				}
 				disks = append(disks, md)
-				//diskNum++
 			}
 		}
 	}

pkg/controller/provider/model/vsphere/model.go

@@ -279,7 +279,6 @@ type Disk struct {
 	Shared    bool   `json:"shared"`
 	RDM       bool   `json:"rdm"`
 	Mode      string `json:"mode,omitempty"`
-	//Number    int32  `json:"number,omitempty"`
 }
 
 // Virtual Device.

virt-v2v/cold/entrypoint.go

@@ -99,16 +99,11 @@ func buildCommand() []string {
 
 	fmt.Println("Preparing virt-v2v")
 
+	virtV2vArgs = append(virtV2vArgs, "--root")
 	if checkEnvVariablesSet("V2V_RootDisk") {
-		diskNum, err := strconv.Atoi(os.Getenv("V2V_RootDisk"))
-		if err != nil {
-			fmt.Println("Error:", err)
-			os.Exit(1)
-		}
-		rootDisk := fmt.Sprintf("/dev/sd%s", genName(diskNum))
-		virtV2vArgs = append(virtV2vArgs, "--root", rootDisk)
+		virtV2vArgs = append(virtV2vArgs, os.Getenv("V2V_RootDisk"))
 	} else {
-		virtV2vArgs = append(virtV2vArgs, "--root", "first")
+		virtV2vArgs = append(virtV2vArgs, "first")
 	}
 
 	switch source {