[VBox] added handling for vdi files

1 files changed, 16 insertions, 1 deletions

diff --git a/src/main/java/org/openslx/util/vm/DiskImage.java b/src/main/java/org/openslx/util/vm/DiskImage.java
index 94d5d2c..cc0be88 100644
--- a/src/main/java/org/openslx/util/vm/DiskImage.java
+++ b/src/main/java/org/openslx/util/vm/DiskImage.java
@@ -4,6 +4,8 @@ import java.io.File;
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.RandomAccessFile;
+import java.nio.charset.StandardCharsets;
+import java.util.Arrays;
 
 import org.openslx.bwlp.thrift.iface.Virtualizer;
 import org.openslx.util.Util;
@@ -15,6 +17,7 @@ public class DiskImage
 	 * Big endian representation of the 4 bytes 'KDMV'
 	 */
 	private static final int VMDK_MAGIC = 0x4b444d56;
+	private static final String VDI_MAGIC = "<<< Oracle VM VirtualBox Disk Image >>>";
 
 	public enum ImageFormat
 	{
@@ -88,7 +91,19 @@ public class DiskImage
 			}
 			// vdi
 			file.seek( 0 );
-			// TODO
+			byte[] buffer = new byte[ VDI_MAGIC.length() ];
+			file.readFully( buffer );
+			String string = new String(  buffer );	
+			if (  VDI_MAGIC.equals( string ) ) 
+			{
+				this.isStandalone = true;
+				this.isCompressed = false;
+				this.isSnapshot = false;
+				this.format = ImageFormat.VDI;
+				this.subFormat = "";
+				this.hwVersion = 0;
+				return;
+			}
 		}
 		throw new UnknownImageFormatException();
 	}