1 files changed, 54 insertions, 0 deletions
diff --git a/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/FilterGenericDiskCdromDevices.java b/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/FilterGenericDiskCdromDevices.java
new file mode 100644
index 00000000..a17f3a8b
--- /dev/null
+++ b/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/FilterGenericDiskCdromDevices.java
@@ -0,0 +1,54 @@
+package org.openslx.runvirt.plugin.qemu.configuration;
+
+import java.util.ArrayList;
+
+import org.openslx.libvirt.domain.Domain;
+import org.openslx.libvirt.domain.device.Disk.StorageType;
+import org.openslx.libvirt.domain.device.DiskCdrom;
+import org.openslx.runvirt.configuration.FilterException;
+import org.openslx.runvirt.configuration.FilterGeneric;
+import org.openslx.runvirt.plugin.qemu.cmdln.CommandLineArgs;
+import org.openslx.vm.QemuMetaData;
+import org.openslx.vm.QemuMetaDataUtils;
+
+public class FilterGenericDiskCdromDevices extends FilterGeneric<Domain, CommandLineArgs>
+{
+	private static final String FILTER_NAME = "Disk CDROM devices";
+
+	public FilterGenericDiskCdromDevices()
+	{
+		super( FilterGenericDiskCdromDevices.FILTER_NAME );
+	}
+
+	private void filterDiskCdromDevice( Domain config, String fileName, int index ) throws FilterException
+	{
+		final ArrayList<DiskCdrom> devices = config.getDiskCdromDevices();
+		final DiskCdrom disk = QemuMetaDataUtils.getArrayIndex( devices, index );
+
+		if ( disk != null ) {
+			if ( fileName == null ) {
+				// do not remove disk CDROM drive, but set local physical drive as input source
+				disk.setStorage( StorageType.BLOCK, QemuMetaData.CDROM_DEFAULT_PHYSICAL_DRIVE );
+			} else if ( fileName.equals( "" ) ) {
+				// remove storage source if empty string is specified to emulate an empty CDROM drive
+				disk.removeStorage();
+			} else {
+				// set disk image file as storage source of the disk CDROM drive
+				disk.setStorage( StorageType.FILE, fileName );
+			}
+		}
+	}
+
+	@Override
+	public void filter( Domain config, CommandLineArgs args ) throws FilterException
+	{
+		this.filterDiskCdromDevice( config, args.getVmDiskFileNameCdrom0(), 0 );
+		this.filterDiskCdromDevice( config, args.getVmDiskFileNameCdrom1(), 1 );
+
+		// remove all additional disk CDROM devices
+		final ArrayList<DiskCdrom> devices = config.getDiskCdromDevices();
+		for ( int i = 2; i < devices.size(); i++ ) {
+			devices.get( i ).remove();
+		}
+	}
+}