vmdk2raw: convert VMware disk images to raw images

git-svn-id: svn://svn.savannah.nongnu.org/qemu/trunk@813 c046a42c-6fe2-441c-8c8c-71466251a162

3 files changed, 239 insertions, 2 deletions

diff --git a/Makefile b/Makefile
index 0f64aae..4137e9b 100644
--- a/Makefile
+++ b/Makefile
@@ -8,7 +8,7 @@ LDFLAGS=-g
 LIBS=
 DEFINES+=-D_GNU_SOURCE
 ifndef CONFIG_WIN32
-TOOLS=qemu-mkcow
+TOOLS=qemu-mkcow vmdk2raw
 endif
 ifdef CONFIG_STATIC
 LDFLAGS+=-static
@@ -22,6 +22,9 @@ all: dyngen$(EXESUF) $(TOOLS) qemu-doc.html qemu-tech.html qemu.1
 qemu-mkcow: qemu-mkcow.c
 	$(CC) $(CFLAGS) $(LDFLAGS) $(DEFINES) -o $@ $^ $(LIBS)
 
+vmdk2raw: vmdk2raw.c
+	$(CC) $(CFLAGS) $(LDFLAGS) $(DEFINES) -o $@ $^ $(LIBS)
+
 dyngen$(EXESUF): dyngen.c
 	$(HOST_CC) $(CFLAGS) $(DEFINES) -o $@ $^
 
@@ -91,7 +94,7 @@ tarbin:
         $(bindir)/qemu-arm \
         $(bindir)/qemu-sparc \
         $(bindir)/qemu-ppc \
-        $(bindir)/qemu-mkcow \
+        $(bindir)/qemu-mkcow $(bindir)/vmdk2raw \
 	$(datadir)/bios.bin \
 	$(datadir)/vgabios.bin \
 	$(datadir)/linux_boot.bin \
diff --git a/vmdk.h b/vmdk.h
new file mode 100644
index 0000000..a847f8b
--- /dev/null
+++ b/vmdk.h
@@ -0,0 +1,52 @@
+/*
+   Copyright (C) Matthew Chapman 2003
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#define SECTOR_BITS	9
+#define SECTOR_SIZE	(1 << SECTOR_BITS)
+#define SECTOR_MASK	(SECTOR_SIZE - 1)
+
+#define L1_BITS		(SECTOR_BITS - 3)
+#define L1_SIZE		(1 << L1_BITS)
+#define L1_MASK		(L1_SIZE - 1)
+
+#define L2_BITS		SECTOR_BITS
+#define L2_SIZE		(1 << L2_BITS)
+#define L2_MASK		(L2_SIZE - 1)
+
+#define MIN(x,y)	(((x) < (y)) ? (x) : (y))
+
+struct cowdisk_header
+{
+    char magic[4]; /* COWD */
+    uint32_t version;
+    uint32_t flags;
+    uint32_t disk_sectors;
+    uint32_t granularity;
+    uint32_t l1dir_sector;
+    uint32_t l1dir_size;
+    uint32_t file_sectors;
+    uint32_t cylinders;
+    uint32_t heads;
+    uint32_t sectors_per_track;
+};
+
+struct cowdisk_header2
+{
+   uint32_t parent_ts;
+   uint32_t timestamp;
+};
diff --git a/vmdk2raw.c b/vmdk2raw.c
new file mode 100644
index 0000000..8bf5eea
--- /dev/null
+++ b/vmdk2raw.c
@@ -0,0 +1,182 @@
+/*
+   vmdk2raw: convert vmware images to raw disk images
+   Copyright (C) Net Integration Technologies 2004
+   Copyright (C) Matthew Chapman 2003
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <inttypes.h>
+#include <sys/types.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include "vmdk.h"
+#include "config-host.h"
+
+struct cowdisk_header header;
+struct cowdisk_header2 header2;
+off_t disk_base, disk_limit;
+unsigned int granule_size;
+uint32_t l1dir[L1_SIZE];
+
+unsigned int cached_l2dir;
+uint32_t l2dir[L2_SIZE];
+
+size_t read_physical(int fd, off64_t offset, size_t length, void *buffer)
+{
+    size_t n;
+
+    if (lseek64(fd, offset, SEEK_SET) == -1)
+    {
+        perror("lseek");
+        return -1;
+    }
+
+    n = read(fd, buffer, length);
+    if (n == -1)
+    {
+        perror("read from disk");
+        return -1;
+    }
+
+    return n;
+}
+
+size_t copy_virtual(int in_fd, int out_fd, off64_t offset, void *buffer, size_t length)
+{
+    unsigned int granule_index, granule_offset;
+    unsigned int l1index, l2index;
+
+    granule_index = offset / granule_size;
+    granule_offset = offset % granule_size;
+    length = MIN(length, granule_size - granule_offset);
+    length = MIN(length, disk_limit - offset);
+
+    l1index = (granule_index >> L2_BITS) & L1_MASK;
+    l2index = granule_index & L2_MASK;
+
+    if (l1dir[l1index] == 0)
+        goto zero_fill;
+
+    if (l1index != cached_l2dir)
+    {
+        if (read_physical(in_fd, (l1dir[l1index] << SECTOR_BITS), sizeof(l2dir), (char *)l2dir) != sizeof(l2dir))
+            return 0;
+
+        cached_l2dir = l1index;
+    }
+
+    if (l2dir[l2index] == 0)
+        goto zero_fill;
+
+    if (read_physical(in_fd, (l2dir[l2index] << SECTOR_BITS) + granule_offset, length, buffer) != length)
+        return 0;
+
+    write(out_fd, buffer, length);
+    return length;
+
+zero_fill:
+    /* the last chunk of the file can not be sparse
+     * or the file will be truncated */
+    if (offset + length < disk_limit) {
+        memset(buffer, 0, length);
+        write(out_fd, buffer, length);
+    } else {
+        if (lseek(out_fd, length, SEEK_CUR) == (off_t)-1)
+            perror("lseek");
+    }
+    return length;
+}
+
+
+int open_vmdk(const char *filename)
+{
+    int fd = open(filename, O_RDONLY);
+    if (fd == -1)
+    {
+        perror(filename);
+        return -1;
+    }
+
+    if (read(fd, &header, sizeof(header)) != sizeof(header))
+    {
+        perror("read from disk");
+        return -1;
+    }
+
+    if (memcmp(header.magic, "COWD", 4) != 0)
+    {
+        fprintf(stderr, "%s is not a VMware virtual disk image\n", filename);
+        return -1;
+    }
+
+    granule_size = header.granularity << SECTOR_BITS;
+    if (read_physical(fd, header.l1dir_sector << SECTOR_BITS, sizeof(l1dir), (char *)l1dir) != sizeof(l1dir))
+        return -1;
+
+    disk_limit = header.disk_sectors << SECTOR_BITS;
+
+    cached_l2dir = -1;
+    return fd;
+}
+
+void help(void)
+{
+    printf("vmdk2raw\n"
+           "usage: vmdk2raw vmware_image output_image\n"
+           "\n"
+           "vmware_image   a vmware 2.x/3.x cow image\n"
+           "output_image   the created disk image\n" 
+          );
+    exit(1);
+}
+
+#define BUF_SIZE granule_size
+void copy_disk(in_fd, out_fd)
+{
+    char buf[BUF_SIZE];
+    off64_t i = 0;
+    while (i < disk_limit) {
+        i += copy_virtual(in_fd, out_fd, i, buf, sizeof(buf));
+    }
+}
+
+int main(int argc, char **argv)
+{
+    int out_fd, in_fd;
+
+    if (argc < 3)
+        help();
+
+    in_fd = open_vmdk(argv[1]);
+    if (in_fd < 0) {
+        return -1;
+    }
+
+    out_fd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR);
+    if (out_fd < 0) {
+        perror(argv[2]);
+        return -1;
+    }
+
+    copy_disk(in_fd, out_fd);
+    close(out_fd);
+    return 0;
+}