Changeset 8230 in vbox

Timestamp:

Apr 21, 2008 2:13:27 PM (17 years ago)

Author:

vboxsync

Message:

Move raw disk/partition code to regular internalcommands.

Location:

trunk/src/VBox/Frontends/VBoxManage

Files:

• Makefile.kmk (modified) (1 diff)
• VBoxInternalManage.cpp (modified) (3 diffs)

trunk/src/VBox/Frontends/VBoxManage/Makefile.kmk

@@ -36 +36 @@
         VBoxInternalManage.cpp \
         VBoxManageSVN.cpp
-ifndef VBOX_OSE
- VBoxManage_SOURCES   += \
-        VBoxInternalManageVmdk.cpp
-endif
 VBoxManage_LIBS       += $(LIB_DDU)
 

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

@@ -44 +44 @@
 
 #include <VBox/VBoxHDD.h>
+#include <VBox/VBoxHDD-new.h>
 #include <VBox/sup.h>
 
 #include "VBoxManage.h"
 
-#ifndef VBOX_OSE
-HRESULT CmdListPartitions(int argc, char **argv, ComPtr<IVirtualBox> aVirtualBox, ComPtr<ISession> aSession);
-HRESULT CmdCreateRawVMDK(int argc, char **argv, ComPtr<IVirtualBox> aVirtualBox, ComPtr<ISession> aSession);
-#endif /* !VBOX_OSE */
+/* Includes for the raw disk stuff. */
+#ifdef RT_OS_WINDOWS
+#include <windows.h>
+#include <winioctl.h>
+#elif RT_OS_LINUX
+#include <errno.h>
+#include <sys/ioctl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <linux/hdreg.h>
+#include <linux/fs.h>
+#endif /* !RT_OS_WINDOWS && !RT_OS_LINUX */
 
 using namespace com;
+
+
+/** Macro for checking whether a partition is of extended type or not. */
+#define PARTTYPE_IS_EXTENDED(x) ((x) == 0x05 || (x) == 0x0f || (x) == 0x85)
+
+/* Maximum number of partitions we can deal with. Ridiculously large number,
+ * but the memory consumption is rather low so who cares about never using
+ * most entries. */
+#define HOSTPARTITION_MAX 100
+
+
+typedef struct HOSTPARTITION
+{
+    unsigned        uIndex;
+    unsigned        uType;
+    unsigned        uStartCylinder;
+    unsigned        uStartHead;
+    unsigned        uStartSector;
+    unsigned        uEndCylinder;
+    unsigned        uEndHead;
+    unsigned        uEndSector;
+    uint64_t        uStart;
+    uint64_t        uSize;
+    uint64_t        uPartDataStart;
+    uint64_t        cPartDataSectors;
+} HOSTPARTITION, *PHOSTPARTITION;
+
+typedef struct HOSTPARTITIONS
+{
+    unsigned        cPartitions;
+    HOSTPARTITION   aPartitions[HOSTPARTITION_MAX];
+} HOSTPARTITIONS, *PHOSTPARTITIONS;
 
 /** flag whether we're in internal mode */
@@ -414 +456 @@
 }
 
+
+static DECLCALLBACK(void) handleVDError(void *pvUser, int rc, RT_SRC_POS_DECL, const char *pszFormat, va_list va)
+{
+    RTPrintf("ERROR: ");
+    RTPrintfV(pszFormat, va);
+    RTPrintf("\n");
+    RTPrintf("Error code %Vrc at %s(%u) in function %s\n", rc, RT_SRC_POS_ARGS);
+}
+
+static int partRead(RTFILE File, PHOSTPARTITIONS pPart)
+{
+    uint8_t aBuffer[512];
+    int rc;
+
+    pPart->cPartitions = 0;
+    memset(pPart->aPartitions, '\0', sizeof(pPart->aPartitions));
+    rc = RTFileReadAt(File, 0, &aBuffer, sizeof(aBuffer), NULL);
+    if (VBOX_FAILURE(rc))
+        return rc;
+    if (aBuffer[510] != 0x55 || aBuffer[511] != 0xaa)
+        return VERR_INVALID_PARAMETER;
+
+    unsigned uExtended = (unsigned)-1;
+
+    for (unsigned i = 0; i < 4; i++)
+    {
+        uint8_t *p = &aBuffer[0x1be + i * 16];
+        if (p[4] == 0)
+            continue;
+        PHOSTPARTITION pCP = &pPart->aPartitions[pPart->cPartitions++];
+        pCP->uIndex = i + 1;
+        pCP->uType = p[4];
+        pCP->uStartCylinder = (uint32_t)p[3] + ((uint32_t)(p[2] & 0xc0) << 2);
+        pCP->uStartHead = p[1];
+        pCP->uStartSector = p[2] & 0x3f;
+        pCP->uEndCylinder = (uint32_t)p[7] + ((uint32_t)(p[6] & 0xc0) << 2);
+        pCP->uEndHead = p[5];
+        pCP->uEndSector = p[6] & 0x3f;
+        pCP->uStart = RT_MAKE_U32_FROM_U8(p[8], p[9], p[10], p[11]);
+        pCP->uSize = RT_MAKE_U32_FROM_U8(p[12], p[13], p[14], p[15]);
+        pCP->uPartDataStart = 0;    /* will be filled out later properly. */
+        pCP->cPartDataSectors = 0;
+
+        if (PARTTYPE_IS_EXTENDED(p[4]))
+        {
+            if (uExtended == (unsigned)-1)
+                uExtended = pCP - pPart->aPartitions;
+            else
+            {
+                RTPrintf("More than one extended partition. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+        }
+    }
+
+    if (uExtended != (unsigned)-1)
+    {
+        unsigned uIndex = 5;
+        uint64_t uStart = pPart->aPartitions[uExtended].uStart;
+        uint64_t uOffset = 0;
+        if (!uStart)
+        {
+            RTPrintf("Inconsistency for logical partition start. Aborting\n");
+            return VERR_INVALID_PARAMETER;
+        }
+
+        do
+        {
+            rc = RTFileReadAt(File, (uStart + uOffset) * 512, &aBuffer, sizeof(aBuffer), NULL);
+            if (VBOX_FAILURE(rc))
+                return rc;
+
+            if (aBuffer[510] != 0x55 || aBuffer[511] != 0xaa)
+            {
+                RTPrintf("Logical partition without magic. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+            uint8_t *p = &aBuffer[0x1be];
+
+            if (p[4] == 0)
+            {
+                RTPrintf("Logical partition with type 0 encountered. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+
+            PHOSTPARTITION pCP = &pPart->aPartitions[pPart->cPartitions++];
+            pCP->uIndex = uIndex;
+            pCP->uType = p[4];
+            pCP->uStartCylinder = (uint32_t)p[3] + ((uint32_t)(p[2] & 0xc0) << 2);
+            pCP->uStartHead = p[1];
+            pCP->uStartSector = p[2] & 0x3f;
+            pCP->uEndCylinder = (uint32_t)p[7] + ((uint32_t)(p[6] & 0xc0) << 2);
+            pCP->uEndHead = p[5];
+            pCP->uEndSector = p[6] & 0x3f;
+            uint32_t uStartOffset = RT_MAKE_U32_FROM_U8(p[8], p[9], p[10], p[11]);
+            pCP->uStart = uStart + uOffset + uStartOffset;
+            pCP->uSize = RT_MAKE_U32_FROM_U8(p[12], p[13], p[14], p[15]);
+            /* Fill out partitioning location info for EBR. */
+            pCP->uPartDataStart = uStart + uOffset;
+            pCP->cPartDataSectors = RT_MIN(uStartOffset, 63);
+            p += 16;
+            if (p[4] == 0)
+                uExtended = (unsigned)-1;
+            else if (PARTTYPE_IS_EXTENDED(p[4]))
+            {
+                uExtended = uIndex++;
+                uOffset = RT_MAKE_U32_FROM_U8(p[8], p[9], p[10], p[11]);
+            }
+            else
+            {
+                RTPrintf("Logical partition chain broken. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+        } while (uExtended != (unsigned)-1);
+    }
+
+    /* Sort partitions in ascending order of start sector. Also do a lot of
+     * consistency checking. */
+    uint64_t uPrevEnd = 0;
+    for (unsigned i = 0; i < pPart->cPartitions-1; i++)
+    {
+        unsigned uMinIdx = i;
+        uint64_t uMinVal = pPart->aPartitions[i].uStart;
+        for (unsigned j = i + 1; j < pPart->cPartitions; j++)
+        {
+            if (pPart->aPartitions[j].uStart < uMinVal)
+            {
+                uMinIdx = j;
+                uMinVal = pPart->aPartitions[j].uStart;
+            }
+            else if (pPart->aPartitions[j].uStart == uMinVal)
+            {
+                RTPrintf("Two partitions start at the same place. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            } else if (pPart->aPartitions[j].uStart == 0)
+            {
+                RTPrintf("Partition starts at sector 0. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+        }
+        if (uMinIdx != i)
+        {
+            /* Swap entries at index i and uMinIdx. */
+            memcpy(&pPart->aPartitions[pPart->cPartitions],
+                   &pPart->aPartitions[i], sizeof(HOSTPARTITION));
+            memcpy(&pPart->aPartitions[i],
+                   &pPart->aPartitions[uMinIdx], sizeof(HOSTPARTITION));
+            memcpy(&pPart->aPartitions[uMinIdx],
+                   &pPart->aPartitions[pPart->cPartitions], sizeof(HOSTPARTITION));
+        }
+        if (pPart->aPartitions[i].cPartDataSectors)
+        {
+            if (pPart->aPartitions[i].uPartDataStart < uPrevEnd)
+            {
+                RTPrintf("Overlapping partition description areas. Aborting\n");
+                return VERR_INVALID_PARAMETER;
+            }
+            uPrevEnd = pPart->aPartitions[i].uPartDataStart + pPart->aPartitions[i].cPartDataSectors;
+        }
+        if (pPart->aPartitions[i].uStart < uPrevEnd)
+        {
+            RTPrintf("Overlapping partitions. Aborting\n");
+            return VERR_INVALID_PARAMETER;
+        }
+        if (!PARTTYPE_IS_EXTENDED(pPart->aPartitions[i].uType))
+            uPrevEnd = pPart->aPartitions[i].uStart + pPart->aPartitions[i].uSize;
+    }
+
+    /* Fill out partitioning location info for MBR. */
+    pPart->aPartitions[0].uPartDataStart = 0;
+    pPart->aPartitions[0].cPartDataSectors = RT_MIN(pPart->aPartitions[0].uStart, 63);
+
+    return VINF_SUCCESS;
+}
+
+HRESULT CmdListPartitions(int argc, char **argv, ComPtr<IVirtualBox> aVirtualBox, ComPtr<ISession> aSession)
+{
+    Utf8Str rawdisk;
+
+    /* let's have a closer look at the arguments */
+    for (int i = 0; i < argc; i++)
+    {
+        if (strcmp(argv[i], "-rawdisk") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            rawdisk = argv[i];
+        }
+        else
+        {
+            return errorSyntax(USAGE_LISTPARTITIONS, "Invalid parameter '%s'", Utf8Str(argv[i]).raw());
+        }
+    }
+
+    if (rawdisk.isEmpty())
+        return errorSyntax(USAGE_LISTPARTITIONS, "Mandatory parameter -rawdisk missing");
+
+    RTFILE RawFile;
+    int vrc = RTFileOpen(&RawFile, rawdisk.raw(), RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_WRITE);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error opening the raw disk: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    HOSTPARTITIONS partitions;
+    vrc = partRead(RawFile, &partitions);
+    if (VBOX_FAILURE(vrc))
+        return vrc;
+
+    RTPrintf("Number  Type   StartCHS       EndCHS      Size (MiB)  Start (Sect)\n");
+    for (unsigned i = 0; i < partitions.cPartitions; i++)
+    {
+        /* Suppress printing the extended partition. Otherwise people
+         * might add it to the list of partitions for raw partition
+         * access (which is not good). */
+        if (PARTTYPE_IS_EXTENDED(partitions.aPartitions[i].uType))
+            continue;
+
+        RTPrintf("%-7u %#04x  %-4u/%-3u/%-2u  %-4u/%-3u/%-2u    %10llu   %10llu\n",
+                 partitions.aPartitions[i].uIndex,
+                 partitions.aPartitions[i].uType,
+                 partitions.aPartitions[i].uStartCylinder,
+                 partitions.aPartitions[i].uStartHead,
+                 partitions.aPartitions[i].uStartSector,
+                 partitions.aPartitions[i].uEndCylinder,
+                 partitions.aPartitions[i].uEndHead,
+                 partitions.aPartitions[i].uEndSector,
+                 partitions.aPartitions[i].uSize / 2048,
+                 partitions.aPartitions[i].uStart);
+    }
+
+    return 0;
+}
+
+HRESULT CmdCreateRawVMDK(int argc, char **argv, ComPtr<IVirtualBox> aVirtualBox, ComPtr<ISession> aSession)
+{
+    HRESULT rc = S_OK;
+    Bstr filename;
+    const char *pszMBRFilename = NULL;
+    Utf8Str rawdisk;
+    const char *pszPartitions = NULL;
+    bool fRegister = false;
+    bool fRelative = false;
+
+    /* let's have a closer look at the arguments */
+    for (int i = 0; i < argc; i++)
+    {
+        if (strcmp(argv[i], "-filename") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            filename = argv[i];
+        }
+        else if (strcmp(argv[i], "-mbr") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            pszMBRFilename = argv[i];
+        }
+        else if (strcmp(argv[i], "-rawdisk") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            rawdisk = argv[i];
+        }
+        else if (strcmp(argv[i], "-partitions") == 0)
+        {
+            if (argc <= i + 1)
+            {
+                return errorArgument("Missing argument to '%s'", argv[i]);
+            }
+            i++;
+            pszPartitions = argv[i];
+        }
+        else if (strcmp(argv[i], "-register") == 0)
+        {
+            fRegister = true;
+        }
+#ifdef RT_OS_LINUX
+        else if (strcmp(argv[i], "-relative") == 0)
+        {
+            fRelative = true;
+        }
+#endif /* RT_OS_LINUX */
+        else
+        {
+            return errorSyntax(USAGE_CREATERAWVMDK, "Invalid parameter '%s'", Utf8Str(argv[i]).raw());
+        }
+    }
+
+    if (filename.isEmpty())
+        return errorSyntax(USAGE_CREATERAWVMDK, "Mandatory parameter -filename missing");
+    if (rawdisk.isEmpty())
+        return errorSyntax(USAGE_CREATERAWVMDK, "Mandatory parameter -rawdisk missing");
+    if (!pszPartitions && pszMBRFilename)
+        return errorSyntax(USAGE_CREATERAWVMDK, "The parameter -mbr is only valid when the parameter -partitions is also present");
+
+    RTFILE RawFile;
+    int vrc = RTFileOpen(&RawFile, rawdisk.raw(), RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_WRITE);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error opening the raw disk: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    uint64_t cbSize = 0;
+#ifdef RT_OS_WINDOWS
+    DISK_GEOMETRY DriveGeo;
+    DWORD cbDriveGeo;
+    /* Windows NT has no IOCTL_DISK_GET_LENGTH_INFORMATION ioctl. This was
+     * added to Windows XP, so use the available info from DriveGeo. */
+    if (DeviceIoControl((HANDLE)RawFile,
+                        IOCTL_DISK_GET_DRIVE_GEOMETRY, NULL, 0,
+                        &DriveGeo, sizeof(DriveGeo), &cbDriveGeo, NULL))
+    {
+        if (DriveGeo.MediaType == FixedMedia)
+        {
+            cbSize =     DriveGeo.Cylinders.QuadPart
+                     *   DriveGeo.TracksPerCylinder
+                     *   DriveGeo.SectorsPerTrack
+                     *   DriveGeo.BytesPerSector;
+        }
+        else
+            return VERR_MEDIA_NOT_RECOGNIZED;
+    }
+    else
+        rc = RTErrConvertFromWin32(GetLastError());
+#elif defined(RT_OS_LINUX)
+    struct stat DevStat;
+    if (!fstat(RawFile, &DevStat) && S_ISBLK(DevStat.st_mode))
+    {
+        long cBlocks;
+        if (!ioctl(RawFile, BLKGETSIZE, &cBlocks))
+            cbSize = (uint64_t)cBlocks * 512;
+        else
+            return RTErrConvertFromErrno(errno);
+    }
+    else
+    {
+        RTPrintf("File '%s' is no disk\n", rawdisk.raw());
+        return VERR_INVALID_PARAMETER;
+    }
+#else /* !RT_OS_WINDOWS && !RT_OS_LINUX */
+    /* Hopefully this works on all other hosts. If it doesn't, it'll just fail
+     * creating the VMDK, so no real harm done. */
+    vrc = RTFileGetSize(RawFile, &cbSize);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error getting the size of the raw disk: %Vrc\n", vrc);
+        return vrc;
+    }
+#endif /* !RT_OS_WINDOWS && !RT_OS_LINUX */
+
+    PVBOXHDD pDisk = NULL;
+    VBOXHDDRAW RawDescriptor;
+    HOSTPARTITIONS partitions;
+    uint32_t uPartitions = 0;
+
+    RawDescriptor.szSignature[0] = 'R';
+    RawDescriptor.szSignature[1] = 'A';
+    RawDescriptor.szSignature[2] = 'W';
+    RawDescriptor.szSignature[3] = '\0';
+    if (!pszPartitions)
+    {
+        RawDescriptor.fRawDisk = true;
+        RawDescriptor.pszRawDisk = rawdisk.raw();
+    }
+    else
+    {
+        RawDescriptor.fRawDisk = false;
+        RawDescriptor.pszRawDisk = NULL;
+        RawDescriptor.cPartitions = 0;
+
+        const char *p = pszPartitions;
+        char *pszNext;
+        uint32_t u32;
+        while (*p != '\0')
+        {
+            vrc = RTStrToUInt32Ex(p, &pszNext, 0, &u32);
+            if (VBOX_FAILURE(vrc))
+            {
+                RTPrintf("Incorrect value in partitions parameter\n");
+                return vrc;
+            }
+            uPartitions |= RT_BIT(u32);
+            p = pszNext;
+            if (*p == ',')
+                p++;
+            else if (*p != '\0')
+            {
+                RTPrintf("Incorrect separator in partitions parameter\n");
+                return VERR_INVALID_PARAMETER;
+            }
+        }
+
+        vrc = partRead(RawFile, &partitions);
+        if (VBOX_FAILURE(vrc))
+        {
+            RTPrintf("Error reading the partition information from '%s'\n", rawdisk.raw());
+            return vrc;
+        }
+
+        for (unsigned i = 0; i < partitions.cPartitions; i++)
+        {
+            if (    uPartitions & RT_BIT(partitions.aPartitions[i].uIndex)
+                &&  PARTTYPE_IS_EXTENDED(partitions.aPartitions[i].uType))
+            {
+                /* Some ignorant user specified an extended partition.
+                 * Bad idea, as this would trigger an overlapping
+                 * partitions error later during VMDK creation. So warn
+                 * here and ignore what the user requested. */
+                RTPrintf("Warning: it is not possible (and necessary) to explicitly give access to the\n"
+                         "         extended partition %u. If required, enable access to all logical\n"
+                         "         partitions inside this extended partition.\n", partitions.aPartitions[i].uIndex);
+                uPartitions &= ~RT_BIT(partitions.aPartitions[i].uIndex);
+            }
+        }
+
+        RawDescriptor.cPartitions = partitions.cPartitions;
+        RawDescriptor.pPartitions = (PVBOXHDDRAWPART)RTMemAllocZ(partitions.cPartitions * sizeof(VBOXHDDRAWPART));
+        if (!RawDescriptor.pPartitions)
+            return VERR_NO_MEMORY;
+        for (unsigned i = 0; i < partitions.cPartitions; i++)
+        {
+            if (uPartitions & RT_BIT(partitions.aPartitions[i].uIndex))
+            {
+                if (fRelative)
+                {
+#ifdef RT_OS_LINUX
+                    /* Refer to the correct partition and use offset 0. */
+                    char *pszRawName;
+                    vrc = RTStrAPrintf(&pszRawName, "%s%u", rawdisk.raw(),
+                                       partitions.aPartitions[i].uIndex);
+                    if (VBOX_FAILURE(vrc))
+                    {
+                        RTPrintf("Error creating reference to individual partition %u, rc=%Vrc\n",
+                                 partitions.aPartitions[i].uIndex, vrc);
+                        return vrc;
+                    }
+                    RawDescriptor.pPartitions[i].pszRawDevice = pszRawName;
+                    RawDescriptor.pPartitions[i].uPartitionStartOffset = 0;
+                    RawDescriptor.pPartitions[i].uPartitionStart = partitions.aPartitions[i].uStart * 512;
+#else
+                    /** @todo not implemented yet for Windows host. Treat just
+                     * like not specified (this code is actually never reached). */
+                    RawDescriptor.pPartitions[i].pszRawDevice = rawdisk.raw();
+                    RawDescriptor.pPartitions[i].uPartitionStartOffset = partitions.aPartitions[i].uStart * 512;
+                    RawDescriptor.pPartitions[i].uPartitionStart = partitions.aPartitions[i].uStart * 512;
+#endif
+                }
+                else
+                {
+                    /* This is the "everything refers to the base raw device"
+                     * variant. This requires opening the base device in RW
+                     * mode even for creation. */
+                    RawDescriptor.pPartitions[i].pszRawDevice = rawdisk.raw();
+                    RawDescriptor.pPartitions[i].uPartitionStartOffset = partitions.aPartitions[i].uStart * 512;
+                    RawDescriptor.pPartitions[i].uPartitionStart = partitions.aPartitions[i].uStart * 512;
+                }
+            }
+            else
+            {
+                /* Suppress access to this partition. */
+                RawDescriptor.pPartitions[i].pszRawDevice = NULL;
+                RawDescriptor.pPartitions[i].uPartitionStartOffset = 0;
+                /* This is used in the plausibility check in the creation
+                 * code. In theory it's a dummy, but I don't want to make
+                 * the VMDK creatiion any more complicated than what it needs
+                 * to be. */
+                RawDescriptor.pPartitions[i].uPartitionStart = partitions.aPartitions[i].uStart * 512;
+            }
+            if (PARTTYPE_IS_EXTENDED(partitions.aPartitions[i].uType))
+            {
+                /* Suppress exporting the actual extended partition. Only
+                 * logical partitions should be processed. However completely
+                 * ignoring it leads to leaving out the MBR data. */
+                RawDescriptor.pPartitions[i].cbPartition = 0;
+            }
+            else
+                RawDescriptor.pPartitions[i].cbPartition =  partitions.aPartitions[i].uSize * 512;
+            RawDescriptor.pPartitions[i].uPartitionDataStart = partitions.aPartitions[i].uPartDataStart * 512;
+            RawDescriptor.pPartitions[i].cbPartitionData = partitions.aPartitions[i].cPartDataSectors * 512;
+            if (RawDescriptor.pPartitions[i].cbPartitionData)
+            {
+                Assert (RawDescriptor.pPartitions[i].cbPartitionData -
+                        (size_t)RawDescriptor.pPartitions[i].cbPartitionData == 0);
+                void *pPartData = RTMemAlloc((size_t)RawDescriptor.pPartitions[i].cbPartitionData);
+                if (!pPartData)
+                    return VERR_NO_MEMORY;
+                vrc = RTFileReadAt(RawFile, partitions.aPartitions[i].uPartDataStart * 512, pPartData, (size_t)RawDescriptor.pPartitions[i].cbPartitionData, NULL);
+                if (VBOX_FAILURE(vrc))
+                {
+                    RTPrintf("Cannot read partition data from raw device '%s': %Vrc\n", rawdisk.raw(), vrc);
+                    return vrc;
+                }
+                /* Splice in the replacement MBR code if specified. */
+                if (    partitions.aPartitions[i].uPartDataStart == 0
+                    &&  pszMBRFilename)
+                {
+                    RTFILE MBRFile;
+                    vrc = RTFileOpen(&MBRFile, pszMBRFilename, RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_WRITE);
+                    if (VBOX_FAILURE(vrc))
+                    {
+                        RTPrintf("Cannot open replacement MBR file '%s' specified with -mbr: %Vrc\n", pszMBRFilename, vrc);
+                        return vrc;
+                    }
+                    vrc = RTFileReadAt(MBRFile, 0, pPartData, 0x1be, NULL);
+                    RTFileClose(MBRFile);
+                    if (VBOX_FAILURE(vrc))
+                    {
+                        RTPrintf("Cannot read replacement MBR file '%s': %Vrc\n", pszMBRFilename, vrc);
+                        return vrc;
+                    }
+                }
+                RawDescriptor.pPartitions[i].pvPartitionData = pPartData;
+            }
+        }
+    }
+
+    RTFileClose(RawFile);
+
+    vrc = VDCreate(handleVDError, NULL, &pDisk);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error while creating the virtual disk container: %Vrc\n", vrc);
+        return vrc;
+    }
+
+    Assert(RT_MIN(cbSize / 512 / 16 / 63, 16383) -
+           (unsigned int)RT_MIN(cbSize / 512 / 16 / 63, 16383) == 0);
+    PDMMEDIAGEOMETRY PCHS, LCHS;
+    PCHS.cCylinders = (unsigned int)RT_MIN(cbSize / 512 / 16 / 63, 16383);
+    PCHS.cHeads = 16;
+    PCHS.cSectors = 63;
+    LCHS.cCylinders = 0;
+    LCHS.cHeads = 0;
+    LCHS.cSectors = 0;
+    vrc = VDCreateBase(pDisk, "VMDK", Utf8Str(filename).raw(),
+                       VD_IMAGE_TYPE_FIXED, cbSize,
+                       VD_VMDK_IMAGE_FLAGS_RAWDISK, (char *)&RawDescriptor,
+                       &PCHS, &LCHS, VD_OPEN_FLAGS_NORMAL, NULL, NULL);
+    if (VBOX_FAILURE(vrc))
+    {
+        RTPrintf("Error while creating the raw disk VMDK: %Vrc\n", vrc);
+        return vrc;
+    }
+    RTPrintf("RAW host disk access VMDK file %s created successfully.\n", Utf8Str(filename).raw());
+
+    VDCloseAll(pDisk);
+
+    /* Clean up allocated memory etc. */
+    if (pszPartitions)
+    {
+        for (unsigned i = 0; i < partitions.cPartitions; i++)
+        {
+            if (uPartitions & RT_BIT(partitions.aPartitions[i].uIndex))
+            {
+                if (fRelative)
+                {
+#ifdef RT_OS_LINUX
+                    /* Free memory allocated above. */
+                    RTStrFree((char *)(void *)RawDescriptor.pPartitions[i].pszRawDevice);
+#endif /* RT_OS_LINUX */
+                }
+            }
+        }
+    }
+
+    if (fRegister)
+    {
+        ComPtr<IHardDisk> hardDisk;
+        CHECK_ERROR(aVirtualBox, OpenHardDisk(filename, hardDisk.asOutParam()));
+
+        if (SUCCEEDED(rc) && hardDisk)
+        {
+            CHECK_ERROR(aVirtualBox, RegisterHardDisk(hardDisk));
+        }
+    }
+
+    return SUCCEEDED(rc) ? 0 : 1;
+}
+
 /**
  * Unloads the neccessary driver.
@@ -470 +1107 @@
     if (!strcmp(pszCmd, "setvdiuuid"))
         return handleSetVDIUUID(argc - 1, &argv[1], aVirtualBox, aSession);
-#ifndef VBOX_OSE
     if (!strcmp(pszCmd, "listpartitions"))
         return CmdListPartitions(argc - 1, &argv[1], aVirtualBox, aSession);
     if (!strcmp(pszCmd, "createrawvmdk"))
         return CmdCreateRawVMDK(argc - 1, &argv[1], aVirtualBox, aSession);
-#endif /* !VBOX_OSE */
 
     if (!strcmp(pszCmd, "modinstall"))