Restructure vm meta data parser

1 files changed, 85 insertions, 0 deletions

diff --git a/src/main/java/org/openslx/util/vm/DiskImage.java b/src/main/java/org/openslx/util/vm/DiskImage.java
new file mode 100644
index 0000000..92138a1
--- /dev/null
+++ b/src/main/java/org/openslx/util/vm/DiskImage.java
@@ -0,0 +1,85 @@
+package org.openslx.util.vm;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.RandomAccessFile;
+
+import org.openslx.bwlp.thrift.iface.Virtualizer;
+
+public class DiskImage
+{
+
+	/**
+	 * Big endian representation of the 4 bytes 'VMDK'
+	 */
+	private static final int VMDK_MAGIC = 0x4b444d56;
+
+	public enum ImageFormat
+	{
+		VMDK( "vmdk" ),
+		QCOW2( "qcow2" ),
+		VDI( "vdi" );
+
+		public final String extension;
+
+		private ImageFormat( String extension )
+		{
+			this.extension = extension;
+		}
+
+		public ImageFormat defaultForVirtualizer( Virtualizer virt )
+		{
+			if ( virt == null )
+				return null;
+			if ( virt.virtId.equals( "vmware" ) )
+				return VMDK;
+			if ( virt.virtId.equals( "virtualbox" ) )
+				return VDI;
+			return null;
+		}
+	}
+
+	public final boolean isStandalone;
+	public final boolean isCompressed;
+	public final ImageFormat format;
+
+	public DiskImage( File disk ) throws FileNotFoundException, IOException, UnknownImageFormatException
+	{
+		// For now we only support VMDK...
+		try ( RandomAccessFile file = new RandomAccessFile( disk, "r" ) ) {
+			if ( file.read() != VMDK_MAGIC )
+				throw new UnknownImageFormatException();
+			file.seek( 512 );
+			byte[] buffer = new byte[ 2048 ];
+			file.readFully( buffer );
+			VmwareConfig config = new VmwareConfig( buffer, findNull( buffer ) );
+			String ct = config.get( "createType" );
+			this.isStandalone = isStandaloneCreateType( ct );
+			this.isCompressed = ct != null && ct.equalsIgnoreCase( "streamOptimized" );
+			this.format = ImageFormat.VMDK;
+		}
+	}
+
+	private int findNull( byte[] buffer )
+	{
+		for ( int i = 0; i < buffer.length; ++i ) {
+			if ( buffer[i] == 0 )
+				return i;
+		}
+		return buffer.length;
+	}
+
+	private boolean isStandaloneCreateType( String type )
+	{
+		if ( type == null )
+			return false;
+		return type.equalsIgnoreCase( "streamOptimized" ) || type.equalsIgnoreCase( "monolithicSparse" );
+	}
+
+	public static class UnknownImageFormatException extends Exception
+	{
+		private static final long serialVersionUID = -6647935235475007171L;
+	}
+
+}