feat: add support for dynamic directpath i/o

Adding support for vSphere Dynamic DirectPath I/O.
kshivakumar-nvidia · Mar 3, 2024 · ab98a98 · ab98a98
1 parent 30d53c6
commit ab98a98
Show file tree

Hide file tree

Showing 10 changed files with 188 additions and 39 deletions.
diff --git a/.web-docs/components/builder/vsphere-clone/README.md b/.web-docs/components/builder/vsphere-clone/README.md
@@ -738,7 +738,10 @@ wget http://{{ .HTTPIP }}:{{ .HTTPPort }}/foo/bar/preseed.cfg
 - `displays` (int32) - Number of video displays. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-789C3913-1053-4850-A0F0-E29C3D32B6DA.html)
   for supported maximums. Defaults to 1.
 
-- `vgpu_profile` (string) - vGPU profile for accelerated graphics. See [NVIDIA GRID vGPU documentation](https://docs.nvidia.com/grid/latest/grid-vgpu-user-guide/index.html#configure-vmware-vsphere-vm-with-vgpu)
+- `pci_passthrough_allowed_device` ([]PCIPassthroughAllowedDevice) - Configure Dynamic DirectPath I/O PCI Passthrough for virtual machine. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-5B3CAB26-5D06-4A99-92A0-3A04C69CE64B.html)
+
+- `vgpu_profile` (string) - vGPU profile for accelerated graphics.
+  vGPU profile for accelerated graphics. See [NVIDIA GRID vGPU documentation](https://docs.nvidia.com/grid/latest/grid-vgpu-user-guide/index.html#configure-vmware-vsphere-vm-with-vgpu)
   for examples of profile names. Defaults to none.
 
 - `NestedHV` (bool) - Enable nested hardware virtualization for VM. Defaults to `false`.

diff --git a/.web-docs/components/builder/vsphere-iso/README.md b/.web-docs/components/builder/vsphere-iso/README.md
@@ -395,7 +395,10 @@ wget http://{{ .HTTPIP }}:{{ .HTTPPort }}/foo/bar/preseed.cfg
 - `displays` (int32) - Number of video displays. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-789C3913-1053-4850-A0F0-E29C3D32B6DA.html)
   for supported maximums. Defaults to 1.
 
-- `vgpu_profile` (string) - vGPU profile for accelerated graphics. See [NVIDIA GRID vGPU documentation](https://docs.nvidia.com/grid/latest/grid-vgpu-user-guide/index.html#configure-vmware-vsphere-vm-with-vgpu)
+- `pci_passthrough_allowed_device` ([]PCIPassthroughAllowedDevice) - Configure Dynamic DirectPath I/O PCI Passthrough for virtual machine. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-5B3CAB26-5D06-4A99-92A0-3A04C69CE64B.html)
+
+- `vgpu_profile` (string) - vGPU profile for accelerated graphics.
+  vGPU profile for accelerated graphics. See [NVIDIA GRID vGPU documentation](https://docs.nvidia.com/grid/latest/grid-vgpu-user-guide/index.html#configure-vmware-vsphere-vm-with-vgpu)
   for examples of profile names. Defaults to none.
 
 - `NestedHV` (bool) - Enable nested hardware virtualization for VM. Defaults to `false`.

diff --git a/builder/vsphere/clone/config.hcl2spec.go b/builder/vsphere/clone/config.hcl2spec.go
diff --git a/builder/vsphere/common/step_hardware.go b/builder/vsphere/common/step_hardware.go
@@ -2,19 +2,27 @@
 // SPDX-License-Identifier: MPL-2.0
 
 //go:generate packer-sdc struct-markdown
-//go:generate packer-sdc mapstructure-to-hcl2 -type HardwareConfig
+//go:generate packer-sdc mapstructure-to-hcl2 -type HardwareConfig,PCIPassthroughAllowedDevice
 
 package common
 
 import (
 	"context"
 	"fmt"
+	"reflect"
 
 	"github.com/hashicorp/packer-plugin-sdk/multistep"
 	packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
 	"github.com/hashicorp/packer-plugin-vsphere/builder/vsphere/driver"
 )
 
+type PCIPassthroughAllowedDevice struct {
+	DeviceId    string `mapstructure:"device_id"`
+	VendorId    string `mapstructure:"vendor_id"`
+	SubVendorId string `mapstructure:"sub_vendor_id"`
+	SubDeviceId string `mapstructure:"sub_device_id"`
+}
+
 type HardwareConfig struct {
 	// Number of CPU cores.
 	CPUs int32 `mapstructure:"CPUs"`
@@ -41,6 +49,9 @@ type HardwareConfig struct {
 	// Number of video displays. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-789C3913-1053-4850-A0F0-E29C3D32B6DA.html)
 	// for supported maximums. Defaults to 1.
 	Displays int32 `mapstructure:"displays"`
+	// Configure Dynamic DirectPath I/O PCI Passthrough for virtual machine. See [vSphere documentation](https://docs.vmware.com/en/VMware-vSphere/7.0/com.vmware.vsphere.vm_admin.doc/GUID-5B3CAB26-5D06-4A99-92A0-3A04C69CE64B.html)
+	AllowedDevices []PCIPassthroughAllowedDevice `mapstructure:"pci_passthrough_allowed_device"`
+	// vGPU profile for accelerated graphics.
 	// vGPU profile for accelerated graphics. See [NVIDIA GRID vGPU documentation](https://docs.nvidia.com/grid/latest/grid-vgpu-user-guide/index.html#configure-vmware-vsphere-vm-with-vgpu)
 	// for examples of profile names. Defaults to none.
 	VGPUProfile string `mapstructure:"vgpu_profile"`
@@ -86,9 +97,14 @@ func (s *StepConfigureHardware) Run(_ context.Context, state multistep.StateBag)
 	ui := state.Get("ui").(packersdk.Ui)
 	vm := state.Get("vm").(driver.VirtualMachine)
 
-	if *s.Config != (HardwareConfig{}) {
+	if !reflect.DeepEqual(*s.Config, HardwareConfig{}) {
 		ui.Say("Customizing hardware...")
 
+		var allowedDevices []driver.PCIPassthroughAllowedDevice
+		for _, device := range s.Config.AllowedDevices {
+			allowedDevices = append(allowedDevices, driver.PCIPassthroughAllowedDevice(device))
+		}
+
 		err := vm.Configure(&driver.HardwareConfig{
 			CPUs:                  s.Config.CPUs,
 			CpuCores:              s.Config.CpuCores,
@@ -102,6 +118,7 @@ func (s *StepConfigureHardware) Run(_ context.Context, state multistep.StateBag)
 			MemoryHotAddEnabled:   s.Config.MemoryHotAddEnabled,
 			VideoRAM:              s.Config.VideoRAM,
 			Displays:              s.Config.Displays,
+			AllowedDevices:        allowedDevices,
 			VGPUProfile:           s.Config.VGPUProfile,
 			Firmware:              s.Config.Firmware,
 			ForceBIOSSetup:        s.Config.ForceBIOSSetup,

diff --git a/builder/vsphere/common/step_hardware.hcl2spec.go b/builder/vsphere/common/step_hardware.hcl2spec.go
diff --git a/builder/vsphere/common/step_hardware_test.go b/builder/vsphere/common/step_hardware_test.go
@@ -213,6 +213,12 @@ func basicStepConfigureHardware() *StepConfigureHardware {
 }
 
 func driverHardwareConfigFromConfig(config *HardwareConfig) *driver.HardwareConfig {
+
+	var allowedDevices []driver.PCIPassthroughAllowedDevice
+	for _, device := range config.AllowedDevices {
+		allowedDevices = append(allowedDevices, driver.PCIPassthroughAllowedDevice(device))
+	}
+
 	return &driver.HardwareConfig{
 		CPUs:                config.CPUs,
 		CpuCores:            config.CpuCores,
@@ -225,6 +231,7 @@ func driverHardwareConfigFromConfig(config *HardwareConfig) *driver.HardwareConf
 		CpuHotAddEnabled:    config.CpuHotAddEnabled,
 		MemoryHotAddEnabled: config.MemoryHotAddEnabled,
 		VideoRAM:            config.VideoRAM,
+		AllowedDevices:      allowedDevices,
 		VGPUProfile:         config.VGPUProfile,
 		Firmware:            config.Firmware,
 		ForceBIOSSetup:      config.ForceBIOSSetup,