Skip to content

Commit

Permalink
Improved SCSI controller handling
Browse files Browse the repository at this point in the history
Govmomi tries to use the 7th slot in a scsi controller, which is not
allowed. This patch will appropriately select the slot to attach a disk
to as well as determine if a scsi controller is full.
  • Loading branch information
dkalleg committed Aug 2, 2016
1 parent 79f2f22 commit 6431f75
Show file tree
Hide file tree
Showing 2 changed files with 126 additions and 0 deletions.
56 changes: 56 additions & 0 deletions builtin/providers/vsphere/resource_vsphere_virtual_machine.go
Original file line number Diff line number Diff line change
Expand Up @@ -1198,6 +1198,12 @@ func addHardDisk(vm *object.VirtualMachine, size, iops int64, diskType string, d
}

if err != nil || controller == nil {
// Check if max number of scsi controller are already used
diskControllers := getSCSIControllers(devices)
if len(diskControllers) >= 4 {
return fmt.Errorf("[ERROR] Maximum number of SCSI controllers created")
}

log.Printf("[DEBUG] Couldn't find a %v controller. Creating one..", controller_type)

var c types.BaseVirtualDevice
Expand Down Expand Up @@ -1268,6 +1274,14 @@ func addHardDisk(vm *object.VirtualMachine, size, iops int64, diskType string, d
log.Printf("[DEBUG] addHardDisk - diskPath: %v", diskPath)
disk := devices.CreateDisk(controller, datastore.Reference(), diskPath)

if strings.Contains(controller_type, "scsi") {
unitNumber, err := getNextUnitNumber(devices, controller)
if err != nil {
return err
}
*disk.UnitNumber = unitNumber
}

existing := devices.SelectByBackingInfo(disk.Backing)
log.Printf("[DEBUG] disk: %#v\n", disk)

Expand Down Expand Up @@ -1300,6 +1314,44 @@ func addHardDisk(vm *object.VirtualMachine, size, iops int64, diskType string, d
}
}

func getSCSIControllers(vmDevices object.VirtualDeviceList) []*types.VirtualController {
// get virtual scsi controllers of all supported types
var scsiControllers []*types.VirtualController
for _, device := range vmDevices {
devType := vmDevices.Type(device)
switch devType {
case "scsi", "lsilogic", "buslogic", "pvscsi", "lsilogic-sas":
if c, ok := device.(types.BaseVirtualController); ok {
scsiControllers = append(scsiControllers, c.GetVirtualController())
}
}
}
return scsiControllers
}

func getNextUnitNumber(devices object.VirtualDeviceList, c types.BaseVirtualController) (int32, error) {
key := c.GetVirtualController().Key

var unitNumbers [16]bool
unitNumbers[7] = true

for _, device := range devices {
d := device.GetVirtualDevice()

if d.ControllerKey == key {
if d.UnitNumber != nil {
unitNumbers[*d.UnitNumber] = true
}
}
}
for i, taken := range unitNumbers {
if !taken {
return int32(i), nil
}
}
return -1, fmt.Errorf("[ERROR] getNextUnitNumber - controller is full")
}

// addCdrom adds a new virtual cdrom drive to the VirtualMachine and attaches an image (ISO) to it from a datastore path.
func addCdrom(vm *object.VirtualMachine, datastore, path string) error {
devices, err := vm.Device(context.TODO())
Expand Down Expand Up @@ -1902,6 +1954,10 @@ func (vm *virtualMachine) setupVirtualMachine(c *govmomi.Client) error {

err = addHardDisk(newVM, vm.hardDisks[i].size, vm.hardDisks[i].iops, vm.hardDisks[i].initType, datastore, diskPath, vm.hardDisks[i].controller)
if err != nil {
err2 := addHardDisk(newVM, vm.hardDisks[i].size, vm.hardDisks[i].iops, vm.hardDisks[i].initType, datastore, diskPath, vm.hardDisks[i].controller)
if err2 != nil {
return err2
}
return err
}
}
Expand Down
70 changes: 70 additions & 0 deletions builtin/providers/vsphere/resource_vsphere_virtual_machine_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -328,6 +328,76 @@ func TestAccVSphereVirtualMachine_client_debug(t *testing.T) {
})
}

const testAccCheckVSphereVirtualMachineConfig_diskSCSICapacity = `
resource "vsphere_virtual_machine" "scsiCapacity" {
name = "terraform-test"
` + testAccTemplateBasicBody + `
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "one"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "two"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "three"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "four"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "five"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "six"
}
disk {
size = 1
controller_type = "scsi-paravirtual"
name = "seven"
}
}
`

func TestAccVSphereVirtualMachine_diskSCSICapacity(t *testing.T) {
var vm virtualMachine
basic_vars := setupTemplateBasicBodyVars()
config := basic_vars.testSprintfTemplateBody(testAccCheckVSphereVirtualMachineConfig_diskSCSICapacity)

vmName := "vsphere_virtual_machine.scsiCapacity"

test_exists, test_name, test_cpu, test_uuid, test_mem, test_num_disk, test_num_of_nic, test_nic_label :=
TestFuncData{vm: vm, label: basic_vars.label, vmName: vmName, numDisks: "8"}.testCheckFuncBasic()

log.Printf("[DEBUG] template= %s", testAccCheckVSphereVirtualMachineConfig_diskSCSICapacity)
log.Printf("[DEBUG] template config= %s", config)

resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccCheckVSphereVirtualMachineDestroy,
Steps: []resource.TestStep{
resource.TestStep{
Config: config,
Check: resource.ComposeTestCheckFunc(
test_exists, test_name, test_cpu, test_uuid, test_mem, test_num_disk, test_num_of_nic, test_nic_label,
),
},
},
})
}

const testAccCheckVSphereVirtualMachineConfig_initType = `
resource "vsphere_virtual_machine" "thin" {
name = "terraform-test"
Expand Down

0 comments on commit 6431f75

Please sign in to comment.