1 files changed, 110 insertions, 0 deletions
diff --git a/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskCdromDevices.java b/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskCdromDevices.java
new file mode 100644
index 00000000..643c40ed
--- /dev/null
+++ b/core/modules/qemu/runvirt-plugin-qemu/src/main/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskCdromDevices.java
@@ -0,0 +1,110 @@
+package org.openslx.runvirt.plugin.qemu.configuration;
+
+import java.util.ArrayList;
+
+import org.openslx.libvirt.domain.Domain;
+import org.openslx.libvirt.domain.device.Disk.BusType;
+import org.openslx.libvirt.domain.device.Disk.StorageType;
+import org.openslx.libvirt.domain.device.DiskCdrom;
+import org.openslx.runvirt.plugin.qemu.cmdln.CommandLineArgs;
+import org.openslx.virtualization.configuration.VirtualizationConfigurationQemuUtils;
+import org.openslx.virtualization.configuration.transformation.TransformationException;
+import org.openslx.virtualization.configuration.transformation.TransformationGeneric;
+
+/**
+ * Generic CDROM drive transformation for Libvirt/QEMU virtualization configurations.
+ * 
+ * @author Manuel Bentele
+ * @version 1.0
+ */
+public class TransformationGenericDiskCdromDevices extends TransformationGeneric<Domain, CommandLineArgs>
+{
+	/**
+	 * Name of the configuration transformation.
+	 */
+	private static final String NAME = "Disk CDROM devices";
+
+	/**
+	 * Creates a new generic CDROM drive transformation for Libvirt/QEMU virtualization
+	 * configurations.
+	 */
+	public TransformationGenericDiskCdromDevices()
+	{
+		super( TransformationGenericDiskCdromDevices.NAME );
+	}
+
+	/**
+	 * Validates a virtualization configuration and input arguments for this transformation.
+	 * 
+	 * @param config virtualization configuration for the validation.
+	 * @param args input arguments for the validation.
+	 * @throws TransformationException validation has failed.
+	 */
+	private void validateInputs( Domain config, CommandLineArgs args ) throws TransformationException
+	{
+		if ( config == null || args == null ) {
+			throw new TransformationException( "Virtualization configuration or input arguments are missing!" );
+		}
+	}
+
+	/**
+	 * Transforms a CDROM drive in a virtualization configuration selected by its {@code index}.
+	 * 
+	 * @param config virtualization configuration for the transformation.
+	 * @param fileName name of the image file for the CDROM drive.
+	 * @param index number of the CDROM drive in the virtualization configuration that is selected.
+	 * @throws TransformationException transformation has failed.
+	 */
+	private void transformDiskCdromDevice( Domain config, String fileName, int index ) throws TransformationException
+	{
+		final ArrayList<DiskCdrom> devices = config.getDiskCdromDevices();
+		final DiskCdrom disk = VirtualizationConfigurationQemuUtils.getArrayIndex( devices, index );
+
+		if ( disk == null ) {
+			if ( fileName != null ) {
+				// CDROM drive does not exist, so create new CDROM drive
+				final DiskCdrom newDisk = config.addDiskCdromDevice();
+				newDisk.setBusType( BusType.SATA );
+				String targetDevName = VirtualizationConfigurationQemuUtils.createAlphabeticalDeviceName( "sd", index );
+				newDisk.setTargetDevice( targetDevName );
+
+				if ( fileName.isEmpty() ) {
+					// remove storage source if empty string is specified to emulate an empty CDROM drive
+					newDisk.removeStorage();
+				} else {
+					// set disk image file as storage source of the disk CDROM drive
+					newDisk.setStorage( StorageType.FILE, fileName );
+				}
+			}
+		} else {
+			// CDROM drive exists, so update existing CDROM drive
+			if ( fileName == null ) {
+				// remove disk storage device if disk image file name is not set
+				disk.remove();
+			} else if ( fileName.isEmpty() ) {
+				// 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 transform( Domain config, CommandLineArgs args ) throws TransformationException
+	{
+		// validate configuration and input arguments
+		this.validateInputs( config, args );
+
+		// alter CDROM drives
+		this.transformDiskCdromDevice( config, args.getVmDiskFileNameCdrom0(), 0 );
+		this.transformDiskCdromDevice( 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();
+		}
+	}
+}