update macOS example

Signed-off-by: Luca Stocchi <lstocchi@redhat.com>

Author: lstocchi

configuration.go

@@ -174,6 +174,20 @@ func (v *VirtualMachineConfiguration) SetStorageDevicesVirtualMachineConfigurati
 	C.setStorageDevicesVZVirtualMachineConfiguration(objc.Ptr(v), objc.Ptr(array))
 }
 
+// StorageDevices return the list of storage device configuration configured in this virtual machine configuration.
+// Return an empty array if no storage device configuration is set.
+func (v *VirtualMachineConfiguration) StorageDevices() []StorageDeviceConfiguration {
+	nsArray := objc.NewNSArray(
+		C.storageDevicesVZVirtualMachineConfiguration(objc.Ptr(v)),
+	)
+	ptrs := nsArray.ToPointerSlice()
+	storageDevices := make([]StorageDeviceConfiguration, len(ptrs))
+	for i, ptr := range ptrs {
+		storageDevices[i] = newVirtioBlockDeviceConfiguration(ptr)
+	}
+	return storageDevices
+}
+
 // SetDirectorySharingDevicesVirtualMachineConfiguration sets list of directory sharing devices. Empty by default.
 //
 // This is only supported on macOS 12 and newer. Older versions do nothing.

example/macOS/main.go

@@ -78,6 +78,9 @@ func runVM(ctx context.Context) error {
 		}
 	}()
 
+	// it start listening to the NBD server, if any
+	listenNetworkBlockDevice(config)
+
 	// cleanup is this function is useful when finished graphic application.
 	cleanup := func() {
 		for i := 1; vm.CanRequestStop(); i++ {
@@ -331,3 +334,24 @@ func setupVMConfiguration(platformConfig vz.PlatformConfiguration) (*vz.VirtualM
 
 	return config, nil
 }
+
+func listenNetworkBlockDevice(vm *vz.VirtualMachineConfiguration) error {
+	storages := vm.StorageDevices()
+	for _, storage := range storages {
+		attachment := storage.Attachment()
+		if nbdAttachment, isNbdAttachment := attachment.(*vz.NetworkBlockDeviceStorageDeviceAttachment); isNbdAttachment {
+			nbdAttachmentStatusCh := make(chan vz.NetworkBlockDeviceStorageDeviceAttachmentStatus)
+			nbdAttachment.Listen(nbdAttachmentStatusCh)
+			go func() {
+				for status := range nbdAttachmentStatusCh {
+					if status.IsConnected() {
+						log.Println("Successfully connected to NBD server")
+					} else {
+						log.Printf("Disconnected from NBD server. Error %v\n", status.Error().Error())
+					}
+				}
+			}()
+		}
+	}
+	return nil
+}

storage.go

@@ -210,6 +210,12 @@ func NewVirtioBlockDeviceConfiguration(attachment StorageDeviceAttachment) (*Vir
 	return config, nil
 }
 
+func newVirtioBlockDeviceConfiguration(ptr unsafe.Pointer) *VirtioBlockDeviceConfiguration {
+	return &VirtioBlockDeviceConfiguration{
+		pointer: objc.NewPointer(ptr),
+	}
+}
+
 // BlockDeviceIdentifier returns the device identifier is a string identifying the Virtio block device.
 // Empty string by default.
 //

virtualization_11.h

@@ -59,6 +59,7 @@ void setSocketDevicesVZVirtualMachineConfiguration(void *config,
 void *socketDevicesVZVirtualMachineConfiguration(void *config);
 void setStorageDevicesVZVirtualMachineConfiguration(void *config,
     void *storageDevices);
+void *storageDevicesVZVirtualMachineConfiguration(void *config);
 
 /* Configurations */
 void *newVZFileHandleSerialPortAttachment(int readFileDescriptor, int writeFileDescriptor);

virtualization_11.m

@@ -327,6 +327,18 @@ void setStorageDevicesVZVirtualMachineConfiguration(void *config,
     RAISE_UNSUPPORTED_MACOS_EXCEPTION();
 }
 
+/*!
+ @abstract Return the list of storage devices configurations for this VZVirtualMachineConfiguration. Return an empty array if no storage device configuration is set.
+ */
+void *storageDevicesVZVirtualMachineConfiguration(void *config)
+{
+    if (@available(macOS 11, *)) {
+        return [(VZVirtualMachineConfiguration *)config storageDevices]; // NSArray<VZStorageDeviceConfiguration *>
+    }
+
+    RAISE_UNSUPPORTED_MACOS_EXCEPTION();
+}
+
 /*!
  @abstract Intialize the VZFileHandleSerialPortAttachment from file descriptors.
  @param readFileDescriptor File descriptor for reading from the file.