Changeset 11066 in vbox for trunk/src/VBox/Frontends

Timestamp:

Aug 1, 2008 2:47:33 PM (17 years ago)

Author:

vboxsync

Message:

VBoxManage: Implement converting arbitrary image formats supported by VBoxHDD-new (i.e. VDI, VMDK, VHD) to raw.

Location:

trunk/src/VBox/Frontends/VBoxManage

Files:

• VBoxInternalManage.cpp (modified) (4 diffs)
• VBoxManage.h (modified) (1 diff)

trunk/src/VBox/Frontends/VBoxManage/VBoxInternalManage.cpp

@@ -121 +121 @@
              "Commands:\n"
              "\n"
-             "%s%s%s%s%s%s%s"
+             "%s%s%s%s%s%s%s%s"
              "WARNING: This is a development tool and shall only be used to analyse\n"
              "         problems. It is completely unsupported and will change in\n"
@@ -171 +171 @@
                  "\n"
                  : "",
+             (u64Cmd & USAGE_CONVERTTORAW) ?
+                 "  converttoraw [-format <fileformat>] <filename> <outputfile>"
+#ifdef ENABLE_CONVERT_RAW_TO_STDOUT
+                 "|stdout"
+#endif /* ENABLE_CONVERT_RAW_TO_STDOUT */
+                 "\n"
+                 "       Convert image to raw, writing to file"
+#ifdef ENABLE_CONVERT_RAW_TO_STDOUT
+                 " or stdout"
+#endif /* ENABLE_CONVERT_RAW_TO_STDOUT */
+                 ".\n"
+                 "\n"
+                 : "",
 #ifdef RT_OS_WINDOWS
             (u64Cmd & USAGE_MODINSTALL) ?
@@ -1238 +1251 @@
 }
 
+static int CmdConvertToRaw(int argc, char **argv, ComPtr<IVirtualBox> aVirtualBox, ComPtr<ISession> aSession)
+{
+    Bstr srcformat;
+    Bstr src;
+    Bstr dst;
+    bool fWriteToStdOut = false;
+
+    /* Parse the arguments. */
+    for (int i = 0; i < argc; i++)
+    {
+        if (strcmp(argv[i], "-format") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            srcformat = argv[i];
+        }
+        else if (src.isEmpty())
+        {
+            src = argv[i];
+        }
+        else if (dst.isEmpty())
+        {
+            dst = argv[i];
+#ifdef ENABLE_CONVERT_RAW_TO_STDOUT
+            if (!strcmp(argv[i], "stdout"))
+                fWriteToStdOut = true;
+#endif /* ENABLE_CONVERT_RAW_TO_STDOUT */
+        }
+        else
+        {
+            return errorSyntax(USAGE_CONVERTTORAW, "Invalid parameter '%s'", Utf8Str(argv[i]).raw());
+        }
+    }
+
+    if (src.isEmpty())
+        return errorSyntax(USAGE_CONVERTTORAW, "Mandatory filename parameter missing");
+    if (dst.isEmpty())
+        return errorSyntax(USAGE_CONVERTTORAW, "Mandatory outputfile parameter missing");
+
+    PVBOXHDD pDisk = NULL;
+
+    VDINTERFACE      vdInterfaceError;
+    VDINTERFACEERROR vdInterfaceErrorCallbacks;
+    vdInterfaceErrorCallbacks.cbSize       = sizeof(VDINTERFACEERROR);
+    vdInterfaceErrorCallbacks.enmInterface = VDINTERFACETYPE_ERROR;
+    vdInterfaceErrorCallbacks.pfnError     = handleVDError;
+
+    int vrc = VDInterfaceCreate(&vdInterfaceError, "VBoxManage_IError", VDINTERFACETYPE_ERROR,
+                                &vdInterfaceErrorCallbacks, NULL, NULL);
+    AssertRC(vrc);
+
+    vrc = VDCreate(&vdInterfaceError, &pDisk);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error while creating the virtual disk container: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    /* Open raw output file. */
+    RTFILE outFile;
+    vrc = VINF_SUCCESS;
+    if (fWriteToStdOut)
+        outFile = 1;
+    else
+        vrc = RTFileOpen(&outFile, Utf8Str(dst).raw(), RTFILE_O_OPEN | RTFILE_O_CREATE | RTFILE_O_WRITE | RTFILE_O_DENY_ALL);
+    if (VBOX_FAILURE(vrc))
+    {
+        VDCloseAll(pDisk);
+        RTPrintf("Error while creating destination file \"%s\": %Vrc\n", Utf8Str(dst).raw(), vrc);
+        return vrc;
+    }
+
+    if (srcformat.isEmpty())
+    {
+        char *pszFormat = NULL;
+        vrc = VDGetFormat(Utf8Str(src).raw(), &pszFormat);
+        if (VBOX_FAILURE(vrc))
+        {
+            VDCloseAll(pDisk);
+            if (!fWriteToStdOut)
+            {
+                RTFileClose(outFile);
+                RTFileDelete(Utf8Str(dst).raw());
+            }
+            RTPrintf("No file format specified and autodetect failed - please specify format: %Vrc\n", vrc);
+            return vrc;
+        }
+        srcformat = pszFormat;
+        RTStrFree(pszFormat);
+    }
+    vrc = VDOpen(pDisk, Utf8Str(srcformat).raw(), Utf8Str(src).raw(), VD_OPEN_FLAGS_READONLY);
+    if (VBOX_FAILURE(vrc))
+    {
+        VDCloseAll(pDisk);
+        if (!fWriteToStdOut)
+        {
+            RTFileClose(outFile);
+            RTFileDelete(Utf8Str(dst).raw());
+        }
+        RTPrintf("Error while opening the source image: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    uint64_t cbSize = VDGetSize(pDisk, VD_LAST_IMAGE);
+    uint64_t offFile = 0;
+#define RAW_BUFFER_SIZE _128K
+    uint64_t cbBuf = RAW_BUFFER_SIZE;
+    void *pvBuf = RTMemAlloc(cbBuf);
+    if (pvBuf)
+    {
+        RTPrintf("Converting image \"%s\" with size %RU64 bytes (%RU64MB) to raw...\n", Utf8Str(src).raw(), cbSize, (cbSize + _1M - 1) / _1M);
+        while (offFile < cbSize)
+        {
+            size_t cb = cbSize - offFile >= (uint64_t)cbBuf ? cbBuf : (size_t)(cbSize - offFile);
+            vrc = VDRead(pDisk, offFile, pvBuf, cb);
+            if (VBOX_FAILURE(vrc))
+                break;
+            vrc = RTFileWrite(outFile, pvBuf, cb, NULL);
+            if (VBOX_FAILURE(vrc))
+                break;
+            offFile += cb;
+        }
+        if (VBOX_FAILURE(vrc))
+        {
+            VDCloseAll(pDisk);
+            if (!fWriteToStdOut)
+            {
+                RTFileClose(outFile);
+                RTFileDelete(Utf8Str(dst).raw());
+            }
+            RTPrintf("Error copying image data: %Vrc\n", vrc);
+            return vrc;
+        }
+    }
+    else
+    {
+        vrc = VERR_NO_MEMORY;
+        VDCloseAll(pDisk);
+        if (!fWriteToStdOut)
+        {
+            RTFileClose(outFile);
+            RTFileDelete(Utf8Str(dst).raw());
+        }
+        RTPrintf("Error allocating read buffer: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    if (!fWriteToStdOut)
+        RTFileClose(outFile);
+    VDCloseAll(pDisk);
+    return vrc;
+}
+
 /**
  * Unloads the neccessary driver.
@@ -1300 +1469 @@
     if (!strcmp(pszCmd, "renamevmdk"))
         return CmdRenameVMDK(argc - 1, &argv[1], aVirtualBox, aSession);
+    if (!strcmp(pszCmd, "converttoraw"))
+        return CmdConvertToRaw(argc - 1, &argv[1], aVirtualBox, aSession);
 
     if (!strcmp(pszCmd, "modinstall"))

trunk/src/VBox/Frontends/VBoxManage/VBoxManage.h

@@ -73 +73 @@
 #define USAGE_GUESTPROPERTY         RT_BIT_64(40)
 #endif  /* VBOX_WITH_GUEST_PROPS defined */
+#define USAGE_CONVERTTORAW          RT_BIT_64(41)
 #define USAGE_ALL                   (~(uint64_t)0)
 /** @} */