/* $Id: vbox-img.cpp 57415 2015-08-18 10:58:19Z vboxsync $ */ /** @file * Standalone image manipulation tool */ /* * Copyright (C) 2010-2014 Oracle Corporation * * This file is part of VirtualBox Open Source Edition (OSE), as * available from http://www.virtualbox.org. This file is free software; * you can redistribute it and/or modify it under the terms of the GNU * General Public License (GPL) as published by the Free Software * Foundation, in version 2 as it comes in the "COPYING" file of the * VirtualBox OSE distribution. VirtualBox OSE is distributed in the * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. */ /********************************************************************************************************************************* * Header Files * *********************************************************************************************************************************/ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include static const char *g_pszProgName = ""; static void printUsage(PRTSTREAM pStrm) { RTStrmPrintf(pStrm, "Usage: %s\n" " setuuid --filename \n" " [--format VDI|VMDK|VHD|...]\n" " [--uuid ]\n" " [--parentuuid ]\n" " [--zeroparentuuid]\n" "\n" " geometry --filename \n" " [--format VDI|VMDK|VHD|...]\n" " [--clearchs]\n" " [--cylinders ]\n" " [--heads ]\n" " [--sectors ]\n" "\n" " convert --srcfilename \n" " --dstfilename \n" " [--stdin]|[--stdout]\n" " [--srcformat VDI|VMDK|VHD|RAW|..]\n" " [--dstformat VDI|VMDK|VHD|RAW|..]\n" " [--variant Standard,Fixed,Split2G,Stream,ESX]\n" "\n" " info --filename \n" "\n" " compact --filename \n" " [--filesystemaware]\n" "\n" " createcache --filename \n" " --size \n" "\n" " createbase --filename \n" " --size \n" " [--format VDI|VMDK|VHD] (default: VDI)\n" " [--variant Standard,Fixed,Split2G,Stream,ESX]\n" " [--dataalignment ]\n" "\n" " repair --filename \n" " [--dry-run]\n" " [--format VDI|VMDK|VHD] (default: autodetect)\n" "\n" " clearcomment --filename \n" "\n" " resize --filename \n" " --size \n", g_pszProgName); } static void showLogo(PRTSTREAM pStrm) { static bool s_fShown; /* show only once */ if (!s_fShown) { RTStrmPrintf(pStrm, VBOX_PRODUCT " Disk Utility " VBOX_VERSION_STRING "\n" "(C) 2005-" VBOX_C_YEAR " " VBOX_VENDOR "\n" "All rights reserved.\n" "\n"); s_fShown = true; } } /** command handler argument */ struct HandlerArg { int argc; char **argv; }; static PVDINTERFACE pVDIfs; static DECLCALLBACK(void) handleVDError(void *pvUser, int rc, RT_SRC_POS_DECL, const char *pszFormat, va_list va) { NOREF(pvUser); NOREF(rc); RTMsgErrorV(pszFormat, va); } static DECLCALLBACK(int) handleVDMessage(void *pvUser, const char *pszFormat, va_list va) { NOREF(pvUser); RTPrintfV(pszFormat, va); return VINF_SUCCESS; } /** * Print a usage synopsis and the syntax error message. */ static int errorSyntax(const char *pszFormat, ...) { va_list args; showLogo(g_pStdErr); // show logo even if suppressed va_start(args, pszFormat); RTStrmPrintf(g_pStdErr, "\nSyntax error: %N\n", pszFormat, &args); va_end(args); printUsage(g_pStdErr); return 1; } static int errorRuntime(const char *pszFormat, ...) { va_list args; va_start(args, pszFormat); RTMsgErrorV(pszFormat, args); va_end(args); return 1; } static int parseDiskVariant(const char *psz, unsigned *puImageFlags) { int rc = VINF_SUCCESS; unsigned uImageFlags = *puImageFlags; while (psz && *psz && RT_SUCCESS(rc)) { size_t len; const char *pszComma = strchr(psz, ','); if (pszComma) len = pszComma - psz; else len = strlen(psz); if (len > 0) { /* * Parsing is intentionally inconsistent: "standard" resets the * variant, whereas the other flags are cumulative. */ if (!RTStrNICmp(psz, "standard", len)) uImageFlags = VD_IMAGE_FLAGS_NONE; else if ( !RTStrNICmp(psz, "fixed", len) || !RTStrNICmp(psz, "static", len)) uImageFlags |= VD_IMAGE_FLAGS_FIXED; else if (!RTStrNICmp(psz, "Diff", len)) uImageFlags |= VD_IMAGE_FLAGS_DIFF; else if (!RTStrNICmp(psz, "split2g", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_SPLIT_2G; else if ( !RTStrNICmp(psz, "stream", len) || !RTStrNICmp(psz, "streamoptimized", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_STREAM_OPTIMIZED; else if (!RTStrNICmp(psz, "esx", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_ESX; else rc = VERR_PARSE_ERROR; } if (pszComma) psz += len + 1; else psz += len; } if (RT_SUCCESS(rc)) *puImageFlags = uImageFlags; return rc; } static int handleSetUUID(HandlerArg *a) { const char *pszFilename = NULL; char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; RTUUID imageUuid; RTUUID parentUuid; bool fSetImageUuid = false; bool fSetParentUuid = false; RTUuidClear(&imageUuid); RTUuidClear(&parentUuid); int rc; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--format", 'o', RTGETOPT_REQ_STRING }, { "--uuid", 'u', RTGETOPT_REQ_UUID }, { "--parentuuid", 'p', RTGETOPT_REQ_UUID }, { "--zeroparentuuid", 'P', RTGETOPT_REQ_NOTHING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 'o': // --format pszFormat = RTStrDup(ValueUnion.psz); break; case 'u': // --uuid imageUuid = ValueUnion.Uuid; fSetImageUuid = true; break; case 'p': // --parentuuid parentUuid = ValueUnion.Uuid; fSetParentUuid = true; break; case 'P': // --zeroparentuuid RTUuidClear(&parentUuid); fSetParentUuid = true; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* Check for consistency of optional parameters. */ if (fSetImageUuid && RTUuidIsNull(&imageUuid)) return errorSyntax("Invalid parameter to --uuid option\n"); /* Autodetect image format. */ if (!pszFormat) { /* Don't pass error interface, as that would triggers error messages * because some backends fail to open the image. */ rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorRuntime("Format autodetect failed: %Rrc\n", rc); } PVBOXHDD pVD = NULL; rc = VDCreate(pVDIfs, enmType, &pVD); if (RT_FAILURE(rc)) return errorRuntime("Cannot create the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open in info mode to be able to open diff images without their parent. */ rc = VDOpen(pVD, pszFormat, pszFilename, VD_OPEN_FLAGS_INFO, NULL); if (RT_FAILURE(rc)) return errorRuntime("Cannot open the virtual disk image \"%s\": %Rrf (%Rrc)\n", pszFilename, rc, rc); RTUUID oldImageUuid; rc = VDGetUuid(pVD, VD_LAST_IMAGE, &oldImageUuid); if (RT_FAILURE(rc)) return errorRuntime("Cannot get UUID of virtual disk image \"%s\": %Rrc\n", pszFilename, rc); RTPrintf("Old image UUID: %RTuuid\n", &oldImageUuid); RTUUID oldParentUuid; rc = VDGetParentUuid(pVD, VD_LAST_IMAGE, &oldParentUuid); if (RT_FAILURE(rc)) return errorRuntime("Cannot get parent UUID of virtual disk image \"%s\": %Rrc\n", pszFilename, rc); RTPrintf("Old parent UUID: %RTuuid\n", &oldParentUuid); if (fSetImageUuid) { RTPrintf("New image UUID: %RTuuid\n", &imageUuid); rc = VDSetUuid(pVD, VD_LAST_IMAGE, &imageUuid); if (RT_FAILURE(rc)) return errorRuntime("Cannot set UUID of virtual disk image \"%s\": %Rrf (%Rrc)\n", pszFilename, rc, rc); } if (fSetParentUuid) { RTPrintf("New parent UUID: %RTuuid\n", &parentUuid); rc = VDSetParentUuid(pVD, VD_LAST_IMAGE, &parentUuid); if (RT_FAILURE(rc)) return errorRuntime("Cannot set parent UUID of virtual disk image \"%s\": %Rrf (%Rrc)\n", pszFilename, rc, rc); } VDDestroy(pVD); if (pszFormat) { RTStrFree(pszFormat); pszFormat = NULL; } return 0; } static int handleGeometry(HandlerArg *a) { const char *pszFilename = NULL; char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; uint16_t cCylinders = 0; uint8_t cHeads = 0; uint8_t cSectors = 0; bool fCylinders = false; bool fHeads = false; bool fSectors = false; int rc; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--format", 'o', RTGETOPT_REQ_STRING }, { "--clearchs", 'C', RTGETOPT_REQ_NOTHING }, { "--cylinders", 'c', RTGETOPT_REQ_UINT16 }, { "--heads", 'e', RTGETOPT_REQ_UINT8 }, { "--sectors", 's', RTGETOPT_REQ_UINT8 } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 'o': // --format pszFormat = RTStrDup(ValueUnion.psz); break; case 'C': // --clearchs cCylinders = 0; cHeads = 0; cSectors = 0; fCylinders = true; fHeads = true; fSectors = true; break; case 'c': // --cylinders cCylinders = ValueUnion.u16; fCylinders = true; break; case 'e': // --heads cHeads = ValueUnion.u8; fHeads = true; break; case 's': // --sectors cSectors = ValueUnion.u8; fSectors = true; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* Autodetect image format. */ if (!pszFormat) { /* Don't pass error interface, as that would triggers error messages * because some backends fail to open the image. */ rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorRuntime("Format autodetect failed: %Rrc\n", rc); } PVBOXHDD pVD = NULL; rc = VDCreate(pVDIfs, enmType, &pVD); if (RT_FAILURE(rc)) return errorRuntime("Cannot create the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open in info mode to be able to open diff images without their parent. */ rc = VDOpen(pVD, pszFormat, pszFilename, VD_OPEN_FLAGS_INFO, NULL); if (RT_FAILURE(rc)) return errorRuntime("Cannot open the virtual disk image \"%s\": %Rrf (%Rrc)\n", pszFilename, rc, rc); VDGEOMETRY oldLCHSGeometry; rc = VDGetLCHSGeometry(pVD, VD_LAST_IMAGE, &oldLCHSGeometry); if (rc == VERR_VD_GEOMETRY_NOT_SET) { memset(&oldLCHSGeometry, 0, sizeof(oldLCHSGeometry)); rc = VINF_SUCCESS; } if (RT_FAILURE(rc)) return errorRuntime("Cannot get LCHS geometry of virtual disk image \"%s\": %Rrc\n", pszFilename, rc); VDGEOMETRY newLCHSGeometry = oldLCHSGeometry; if (fCylinders) newLCHSGeometry.cCylinders = cCylinders; if (fHeads) newLCHSGeometry.cHeads = cHeads; if (fSectors) newLCHSGeometry.cSectors = cSectors; if (fCylinders || fHeads || fSectors) { RTPrintf("Old image LCHS: %u/%u/%u\n", oldLCHSGeometry.cCylinders, oldLCHSGeometry.cHeads, oldLCHSGeometry.cSectors); RTPrintf("New image LCHS: %u/%u/%u\n", newLCHSGeometry.cCylinders, newLCHSGeometry.cHeads, newLCHSGeometry.cSectors); rc = VDSetLCHSGeometry(pVD, VD_LAST_IMAGE, &newLCHSGeometry); if (RT_FAILURE(rc)) return errorRuntime("Cannot set LCHS geometry of virtual disk image \"%s\": %Rrf (%Rrc)\n", pszFilename, rc, rc); } else RTPrintf("Current image LCHS: %u/%u/%u\n", oldLCHSGeometry.cCylinders, oldLCHSGeometry.cHeads, oldLCHSGeometry.cSectors); VDDestroy(pVD); if (pszFormat) { RTStrFree(pszFormat); pszFormat = NULL; } return 0; } typedef struct FILEIOSTATE { RTFILE file; /** Offset in the file. */ uint64_t off; /** Offset where the buffer contents start. UINT64_MAX=buffer invalid. */ uint64_t offBuffer; /** Size of valid data in the buffer. */ uint32_t cbBuffer; /** Buffer for efficient I/O */ uint8_t abBuffer[16 *_1M]; } FILEIOSTATE, *PFILEIOSTATE; static DECLCALLBACK(int) convInOpen(void *pvUser, const char *pszLocation, uint32_t fOpen, PFNVDCOMPLETED pfnCompleted, void **ppStorage) { NOREF(pvUser); /* Validate input. */ AssertPtrReturn(ppStorage, VERR_INVALID_POINTER); AssertPtrNullReturn(pfnCompleted, VERR_INVALID_PARAMETER); AssertReturn((fOpen & RTFILE_O_ACCESS_MASK) == RTFILE_O_READ, VERR_INVALID_PARAMETER); RTFILE file; int rc = RTFileFromNative(&file, RTFILE_NATIVE_STDIN); if (RT_FAILURE(rc)) return rc; /* No need to clear the buffer, the data will be read from disk. */ PFILEIOSTATE pFS = (PFILEIOSTATE)RTMemAlloc(sizeof(FILEIOSTATE)); if (!pFS) return VERR_NO_MEMORY; pFS->file = file; pFS->off = 0; pFS->offBuffer = UINT64_MAX; pFS->cbBuffer = 0; *ppStorage = pFS; return VINF_SUCCESS; } static DECLCALLBACK(int) convInClose(void *pvUser, void *pStorage) { NOREF(pvUser); AssertPtrReturn(pStorage, VERR_INVALID_POINTER); PFILEIOSTATE pFS = (PFILEIOSTATE)pStorage; RTMemFree(pFS); return VINF_SUCCESS; } static DECLCALLBACK(int) convInDelete(void *pvUser, const char *pcszFilename) { NOREF(pvUser); NOREF(pcszFilename); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInMove(void *pvUser, const char *pcszSrc, const char *pcszDst, unsigned fMove) { NOREF(pvUser); NOREF(pcszSrc); NOREF(pcszDst); NOREF(fMove); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInGetFreeSpace(void *pvUser, const char *pcszFilename, int64_t *pcbFreeSpace) { NOREF(pvUser); NOREF(pcszFilename); AssertPtrReturn(pcbFreeSpace, VERR_INVALID_POINTER); *pcbFreeSpace = 0; return VINF_SUCCESS; } static DECLCALLBACK(int) convInGetModificationTime(void *pvUser, const char *pcszFilename, PRTTIMESPEC pModificationTime) { NOREF(pvUser); NOREF(pcszFilename); AssertPtrReturn(pModificationTime, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInGetSize(void *pvUser, void *pStorage, uint64_t *pcbSize) { NOREF(pvUser); NOREF(pStorage); AssertPtrReturn(pcbSize, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInSetSize(void *pvUser, void *pStorage, uint64_t cbSize) { NOREF(pvUser); NOREF(pStorage); NOREF(cbSize); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInRead(void *pvUser, void *pStorage, uint64_t uOffset, void *pvBuffer, size_t cbBuffer, size_t *pcbRead) { NOREF(pvUser); AssertPtrReturn(pStorage, VERR_INVALID_POINTER); AssertPtrReturn(pvBuffer, VERR_INVALID_POINTER); PFILEIOSTATE pFS = (PFILEIOSTATE)pStorage; AssertReturn(uOffset >= pFS->off, VERR_INVALID_PARAMETER); int rc; /* Fill buffer if it is empty. */ if (pFS->offBuffer == UINT64_MAX) { /* Repeat reading until buffer is full or EOF. */ size_t cbRead; size_t cbSumRead = 0; uint8_t *pbTmp = (uint8_t *)&pFS->abBuffer[0]; size_t cbTmp = sizeof(pFS->abBuffer); do { rc = RTFileRead(pFS->file, pbTmp, cbTmp, &cbRead); if (RT_FAILURE(rc)) return rc; pbTmp += cbRead; cbTmp -= cbRead; cbSumRead += cbRead; } while (cbTmp && cbRead); pFS->offBuffer = 0; pFS->cbBuffer = (uint32_t)cbSumRead; if (!cbSumRead && !pcbRead) /* Caller can't handle partial reads. */ return VERR_EOF; } /* Read several blocks and assemble the result if necessary */ size_t cbTotalRead = 0; do { /* Skip over areas no one wants to read. */ while (uOffset > pFS->offBuffer + pFS->cbBuffer - 1) { if (pFS->cbBuffer < sizeof(pFS->abBuffer)) { if (pcbRead) *pcbRead = cbTotalRead; return VERR_EOF; } /* Repeat reading until buffer is full or EOF. */ size_t cbRead; size_t cbSumRead = 0; uint8_t *pbTmp = (uint8_t *)&pFS->abBuffer[0]; size_t cbTmp = sizeof(pFS->abBuffer); do { rc = RTFileRead(pFS->file, pbTmp, cbTmp, &cbRead); if (RT_FAILURE(rc)) return rc; pbTmp += cbRead; cbTmp -= cbRead; cbSumRead += cbRead; } while (cbTmp && cbRead); pFS->offBuffer += pFS->cbBuffer; pFS->cbBuffer = (uint32_t)cbSumRead; } uint32_t cbThisRead = (uint32_t)RT_MIN(cbBuffer, pFS->cbBuffer - uOffset % sizeof(pFS->abBuffer)); memcpy(pvBuffer, &pFS->abBuffer[uOffset % sizeof(pFS->abBuffer)], cbThisRead); uOffset += cbThisRead; pvBuffer = (uint8_t *)pvBuffer + cbThisRead; cbBuffer -= cbThisRead; cbTotalRead += cbThisRead; if (!cbTotalRead && !pcbRead) /* Caller can't handle partial reads. */ return VERR_EOF; } while (cbBuffer > 0); if (pcbRead) *pcbRead = cbTotalRead; pFS->off = uOffset; return VINF_SUCCESS; } static DECLCALLBACK(int) convInWrite(void *pvUser, void *pStorage, uint64_t uOffset, const void *pvBuffer, size_t cbBuffer, size_t *pcbWritten) { NOREF(pvUser); NOREF(pStorage); NOREF(uOffset); NOREF(cbBuffer); NOREF(pcbWritten); AssertPtrReturn(pvBuffer, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convInFlush(void *pvUser, void *pStorage) { NOREF(pvUser); NOREF(pStorage); return VINF_SUCCESS; } static DECLCALLBACK(int) convOutOpen(void *pvUser, const char *pszLocation, uint32_t fOpen, PFNVDCOMPLETED pfnCompleted, void **ppStorage) { NOREF(pvUser); /* Validate input. */ AssertPtrReturn(ppStorage, VERR_INVALID_POINTER); AssertPtrNullReturn(pfnCompleted, VERR_INVALID_PARAMETER); AssertReturn((fOpen & RTFILE_O_ACCESS_MASK) == RTFILE_O_WRITE, VERR_INVALID_PARAMETER); RTFILE file; int rc = RTFileFromNative(&file, RTFILE_NATIVE_STDOUT); if (RT_FAILURE(rc)) return rc; /* Must clear buffer, so that skipped over data is initialized properly. */ PFILEIOSTATE pFS = (PFILEIOSTATE)RTMemAllocZ(sizeof(FILEIOSTATE)); if (!pFS) return VERR_NO_MEMORY; pFS->file = file; pFS->off = 0; pFS->offBuffer = 0; pFS->cbBuffer = sizeof(FILEIOSTATE); *ppStorage = pFS; return VINF_SUCCESS; } static DECLCALLBACK(int) convOutClose(void *pvUser, void *pStorage) { NOREF(pvUser); AssertPtrReturn(pStorage, VERR_INVALID_POINTER); PFILEIOSTATE pFS = (PFILEIOSTATE)pStorage; int rc = VINF_SUCCESS; /* Flush any remaining buffer contents. */ if (pFS->cbBuffer) rc = RTFileWrite(pFS->file, &pFS->abBuffer[0], pFS->cbBuffer, NULL); RTMemFree(pFS); return rc; } static DECLCALLBACK(int) convOutDelete(void *pvUser, const char *pcszFilename) { NOREF(pvUser); NOREF(pcszFilename); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutMove(void *pvUser, const char *pcszSrc, const char *pcszDst, unsigned fMove) { NOREF(pvUser); NOREF(pcszSrc); NOREF(pcszDst); NOREF(fMove); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutGetFreeSpace(void *pvUser, const char *pcszFilename, int64_t *pcbFreeSpace) { NOREF(pvUser); NOREF(pcszFilename); AssertPtrReturn(pcbFreeSpace, VERR_INVALID_POINTER); *pcbFreeSpace = INT64_MAX; return VINF_SUCCESS; } static DECLCALLBACK(int) convOutGetModificationTime(void *pvUser, const char *pcszFilename, PRTTIMESPEC pModificationTime) { NOREF(pvUser); NOREF(pcszFilename); AssertPtrReturn(pModificationTime, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutGetSize(void *pvUser, void *pStorage, uint64_t *pcbSize) { NOREF(pvUser); NOREF(pStorage); AssertPtrReturn(pcbSize, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutSetSize(void *pvUser, void *pStorage, uint64_t cbSize) { NOREF(pvUser); NOREF(pStorage); NOREF(cbSize); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutRead(void *pvUser, void *pStorage, uint64_t uOffset, void *pvBuffer, size_t cbBuffer, size_t *pcbRead) { NOREF(pvUser); NOREF(pStorage); NOREF(uOffset); NOREF(cbBuffer); NOREF(pcbRead); AssertPtrReturn(pvBuffer, VERR_INVALID_POINTER); AssertFailedReturn(VERR_NOT_SUPPORTED); } static DECLCALLBACK(int) convOutWrite(void *pvUser, void *pStorage, uint64_t uOffset, const void *pvBuffer, size_t cbBuffer, size_t *pcbWritten) { NOREF(pvUser); AssertPtrReturn(pStorage, VERR_INVALID_POINTER); AssertPtrReturn(pvBuffer, VERR_INVALID_POINTER); PFILEIOSTATE pFS = (PFILEIOSTATE)pStorage; AssertReturn(uOffset >= pFS->off, VERR_INVALID_PARAMETER); int rc; /* Write the data to the buffer, flushing as required. */ size_t cbTotalWritten = 0; do { /* Flush the buffer if we need a new one. */ while (uOffset > pFS->offBuffer + sizeof(pFS->abBuffer) - 1) { rc = RTFileWrite(pFS->file, &pFS->abBuffer[0], sizeof(pFS->abBuffer), NULL); RT_ZERO(pFS->abBuffer); pFS->offBuffer += sizeof(pFS->abBuffer); pFS->cbBuffer = 0; } uint32_t cbThisWrite = (uint32_t)RT_MIN(cbBuffer, sizeof(pFS->abBuffer) - uOffset % sizeof(pFS->abBuffer)); memcpy(&pFS->abBuffer[uOffset % sizeof(pFS->abBuffer)], pvBuffer, cbThisWrite); uOffset += cbThisWrite; pvBuffer = (uint8_t *)pvBuffer + cbThisWrite; cbBuffer -= cbThisWrite; cbTotalWritten += cbThisWrite; } while (cbBuffer > 0); if (pcbWritten) *pcbWritten = cbTotalWritten; pFS->cbBuffer = uOffset % sizeof(pFS->abBuffer); if (!pFS->cbBuffer) pFS->cbBuffer = sizeof(pFS->abBuffer); pFS->off = uOffset; return VINF_SUCCESS; } static DECLCALLBACK(int) convOutFlush(void *pvUser, void *pStorage) { NOREF(pvUser); NOREF(pStorage); return VINF_SUCCESS; } static int handleConvert(HandlerArg *a) { const char *pszSrcFilename = NULL; const char *pszDstFilename = NULL; bool fStdIn = false; bool fStdOut = false; const char *pszSrcFormat = NULL; VDTYPE enmSrcType = VDTYPE_HDD; const char *pszDstFormat = NULL; const char *pszVariant = NULL; PVBOXHDD pSrcDisk = NULL; PVBOXHDD pDstDisk = NULL; unsigned uImageFlags = VD_IMAGE_FLAGS_NONE; PVDINTERFACE pIfsImageInput = NULL; PVDINTERFACE pIfsImageOutput = NULL; VDINTERFACEIO IfsInputIO; VDINTERFACEIO IfsOutputIO; int rc = VINF_SUCCESS; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--srcfilename", 'i', RTGETOPT_REQ_STRING }, { "--dstfilename", 'o', RTGETOPT_REQ_STRING }, { "--stdin", 'p', RTGETOPT_REQ_NOTHING }, { "--stdout", 'P', RTGETOPT_REQ_NOTHING }, { "--srcformat", 's', RTGETOPT_REQ_STRING }, { "--dstformat", 'd', RTGETOPT_REQ_STRING }, { "--variant", 'v', RTGETOPT_REQ_STRING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'i': // --srcfilename pszSrcFilename = ValueUnion.psz; break; case 'o': // --dstfilename pszDstFilename = ValueUnion.psz; break; case 'p': // --stdin fStdIn = true; break; case 'P': // --stdout fStdOut = true; break; case 's': // --srcformat pszSrcFormat = ValueUnion.psz; break; case 'd': // --dstformat pszDstFormat = ValueUnion.psz; break; case 'v': // --variant pszVariant = ValueUnion.psz; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters and handle dummies/defaults. */ if (fStdIn && !pszSrcFormat) return errorSyntax("Mandatory --srcformat option missing\n"); if (!pszDstFormat) pszDstFormat = "VDI"; if (fStdIn && !pszSrcFilename) { /* Complete dummy, will be just passed to various calls to fulfill * the "must be non-NULL" requirement, and is completely ignored * otherwise. It shown in the stderr message below. */ pszSrcFilename = "stdin"; } if (fStdOut && !pszDstFilename) { /* Will be stored in the destination image if it is a streamOptimized * VMDK, but it isn't really relevant - use it for "branding". */ if (!RTStrICmp(pszDstFormat, "VMDK")) pszDstFilename = "VirtualBoxStream.vmdk"; else pszDstFilename = "stdout"; } if (!pszSrcFilename) return errorSyntax("Mandatory --srcfilename option missing\n"); if (!pszDstFilename) return errorSyntax("Mandatory --dstfilename option missing\n"); if (fStdIn) { IfsInputIO.pfnOpen = convInOpen; IfsInputIO.pfnClose = convInClose; IfsInputIO.pfnDelete = convInDelete; IfsInputIO.pfnMove = convInMove; IfsInputIO.pfnGetFreeSpace = convInGetFreeSpace; IfsInputIO.pfnGetModificationTime = convInGetModificationTime; IfsInputIO.pfnGetSize = convInGetSize; IfsInputIO.pfnSetSize = convInSetSize; IfsInputIO.pfnReadSync = convInRead; IfsInputIO.pfnWriteSync = convInWrite; IfsInputIO.pfnFlushSync = convInFlush; VDInterfaceAdd(&IfsInputIO.Core, "stdin", VDINTERFACETYPE_IO, NULL, sizeof(VDINTERFACEIO), &pIfsImageInput); } if (fStdOut) { IfsOutputIO.pfnOpen = convOutOpen; IfsOutputIO.pfnClose = convOutClose; IfsOutputIO.pfnDelete = convOutDelete; IfsOutputIO.pfnMove = convOutMove; IfsOutputIO.pfnGetFreeSpace = convOutGetFreeSpace; IfsOutputIO.pfnGetModificationTime = convOutGetModificationTime; IfsOutputIO.pfnGetSize = convOutGetSize; IfsOutputIO.pfnSetSize = convOutSetSize; IfsOutputIO.pfnReadSync = convOutRead; IfsOutputIO.pfnWriteSync = convOutWrite; IfsOutputIO.pfnFlushSync = convOutFlush; VDInterfaceAdd(&IfsOutputIO.Core, "stdout", VDINTERFACETYPE_IO, NULL, sizeof(VDINTERFACEIO), &pIfsImageOutput); } /* check the variant parameter */ if (pszVariant) { char *psz = (char*)pszVariant; while (psz && *psz && RT_SUCCESS(rc)) { size_t len; const char *pszComma = strchr(psz, ','); if (pszComma) len = pszComma - psz; else len = strlen(psz); if (len > 0) { if (!RTStrNICmp(pszVariant, "standard", len)) uImageFlags |= VD_IMAGE_FLAGS_NONE; else if (!RTStrNICmp(pszVariant, "fixed", len)) uImageFlags |= VD_IMAGE_FLAGS_FIXED; else if (!RTStrNICmp(pszVariant, "split2g", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_SPLIT_2G; else if (!RTStrNICmp(pszVariant, "stream", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_STREAM_OPTIMIZED; else if (!RTStrNICmp(pszVariant, "esx", len)) uImageFlags |= VD_VMDK_IMAGE_FLAGS_ESX; else return errorSyntax("Invalid --variant option\n"); } if (pszComma) psz += len + 1; else psz += len; } } do { /* try to determine input format if not specified */ if (!pszSrcFormat) { char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; rc = VDGetFormat(NULL, NULL, pszSrcFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) { errorSyntax("No file format specified, please specify format: %Rrc\n", rc); break; } pszSrcFormat = pszFormat; enmSrcType = enmType; } rc = VDCreate(pVDIfs, enmSrcType, &pSrcDisk); if (RT_FAILURE(rc)) { errorRuntime("Error while creating source disk container: %Rrf (%Rrc)\n", rc, rc); break; } rc = VDOpen(pSrcDisk, pszSrcFormat, pszSrcFilename, VD_OPEN_FLAGS_READONLY | VD_OPEN_FLAGS_SEQUENTIAL, pIfsImageInput); if (RT_FAILURE(rc)) { errorRuntime("Error while opening source image: %Rrf (%Rrc)\n", rc, rc); break; } rc = VDCreate(pVDIfs, VDTYPE_HDD, &pDstDisk); if (RT_FAILURE(rc)) { errorRuntime("Error while creating the destination disk container: %Rrf (%Rrc)\n", rc, rc); break; } uint64_t cbSize = VDGetSize(pSrcDisk, VD_LAST_IMAGE); RTStrmPrintf(g_pStdErr, "Converting image \"%s\" with size %RU64 bytes (%RU64MB)...\n", pszSrcFilename, cbSize, (cbSize + _1M - 1) / _1M); /* Create the output image */ rc = VDCopy(pSrcDisk, VD_LAST_IMAGE, pDstDisk, pszDstFormat, pszDstFilename, false, 0, uImageFlags, NULL, VD_OPEN_FLAGS_NORMAL | VD_OPEN_FLAGS_SEQUENTIAL, NULL, pIfsImageOutput, NULL); if (RT_FAILURE(rc)) { errorRuntime("Error while copying the image: %Rrf (%Rrc)\n", rc, rc); break; } } while (0); if (pDstDisk) VDDestroy(pDstDisk); if (pSrcDisk) VDDestroy(pSrcDisk); return RT_SUCCESS(rc) ? 0 : 1; } static int handleInfo(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* just try it */ char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorSyntax("Format autodetect failed: %Rrc\n", rc); rc = VDCreate(pVDIfs, enmType, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open the image */ rc = VDOpen(pDisk, pszFormat, pszFilename, VD_OPEN_FLAGS_INFO | VD_OPEN_FLAGS_READONLY, NULL); if (RT_FAILURE(rc)) return errorRuntime("Error while opening the image: %Rrf (%Rrc)\n", rc, rc); VDDumpImages(pDisk); VDDestroy(pDisk); return rc; } static DECLCALLBACK(int) vboximgDvmRead(void *pvUser, uint64_t off, void *pvBuf, size_t cbRead) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = (PVBOXHDD)pvUser; /* Take shortcut if possible. */ if ( off % 512 == 0 && cbRead % 512 == 0) rc = VDRead(pDisk, off, pvBuf, cbRead); else { uint8_t *pbBuf = (uint8_t *)pvBuf; uint8_t abBuf[512]; /* Unaligned access, make it aligned. */ if (off % 512 != 0) { uint64_t offAligned = off & ~(uint64_t)(512 - 1); size_t cbToCopy = 512 - (off - offAligned); rc = VDRead(pDisk, offAligned, abBuf, 512); if (RT_SUCCESS(rc)) { memcpy(pbBuf, &abBuf[off - offAligned], cbToCopy); pbBuf += cbToCopy; off += cbToCopy; cbRead -= cbToCopy; } } if ( RT_SUCCESS(rc) && (cbRead & ~(uint64_t)(512 - 1))) { size_t cbReadAligned = cbRead & ~(uint64_t)(512 - 1); Assert(!(off % 512)); rc = VDRead(pDisk, off, pbBuf, cbReadAligned); if (RT_SUCCESS(rc)) { pbBuf += cbReadAligned; off += cbReadAligned; cbRead -= cbReadAligned; } } if ( RT_SUCCESS(rc) && cbRead) { Assert(cbRead < 512); Assert(!(off % 512)); rc = VDRead(pDisk, off, abBuf, 512); if (RT_SUCCESS(rc)) memcpy(pbBuf, abBuf, cbRead); } } return rc; } static DECLCALLBACK(int) vboximgDvmWrite(void *pvUser, uint64_t off, const void *pvBuf, size_t cbWrite) { PVBOXHDD pDisk = (PVBOXHDD)pvUser; return VDWrite(pDisk, off, pvBuf, cbWrite); } static DECLCALLBACK(int) vboximgQueryBlockStatus(void *pvUser, uint64_t off, uint64_t cb, bool *pfAllocated) { RTVFS hVfs = (RTVFS)pvUser; return RTVfsIsRangeInUse(hVfs, off, cb, pfAllocated); } static DECLCALLBACK(int) vboximgQueryRangeUse(void *pvUser, uint64_t off, uint64_t cb, bool *pfUsed) { RTDVM hVolMgr = (RTDVM)pvUser; return RTDvmMapQueryBlockStatus(hVolMgr, off, cb, pfUsed); } typedef struct VBOXIMGVFS { /** Pointer to the next VFS handle. */ struct VBOXIMGVFS *pNext; /** VFS handle. */ RTVFS hVfs; } VBOXIMGVFS, *PVBOXIMGVFS; static int handleCompact(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; bool fFilesystemAware = false; VDINTERFACEQUERYRANGEUSE VDIfQueryRangeUse; PVDINTERFACE pIfsCompact = NULL; RTDVM hDvm = NIL_RTDVM; PVBOXIMGVFS pVBoxImgVfsHead = NULL; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--filesystemaware", 'a', RTGETOPT_REQ_NOTHING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 'a': fFilesystemAware = true; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* just try it */ char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorSyntax("Format autodetect failed: %Rrc\n", rc); rc = VDCreate(pVDIfs, enmType, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open the image */ rc = VDOpen(pDisk, pszFormat, pszFilename, VD_OPEN_FLAGS_NORMAL, NULL); if (RT_FAILURE(rc)) return errorRuntime("Error while opening the image: %Rrf (%Rrc)\n", rc, rc); if ( RT_SUCCESS(rc) && fFilesystemAware) { uint64_t cbDisk = 0; cbDisk = VDGetSize(pDisk, 0); if (cbDisk > 0) { rc = RTDvmCreate(&hDvm, vboximgDvmRead, vboximgDvmWrite, cbDisk, 512, 0 /* fFlags*/, pDisk); if (RT_SUCCESS(rc)) { rc = RTDvmMapOpen(hDvm); if ( RT_SUCCESS(rc) && RTDvmMapGetValidVolumes(hDvm)) { RTDVMVOLUME hVol; /* Get all volumes and set the block query status callback. */ rc = RTDvmMapQueryFirstVolume(hDvm, &hVol); AssertRC(rc); do { RTVFSFILE hVfsFile; RTVFS hVfs; RTDVMVOLUME hVolNext; rc = RTDvmVolumeCreateVfsFile(hVol, &hVfsFile); if (RT_FAILURE(rc)) break; /* Try to detect the filesystem in this volume. */ rc = RTFilesystemVfsFromFile(hVfsFile, &hVfs); if (rc == VERR_NOT_SUPPORTED) { /* Release the file handle and continue.*/ RTVfsFileRelease(hVfsFile); } else if (RT_FAILURE(rc)) break; else { PVBOXIMGVFS pVBoxImgVfs = (PVBOXIMGVFS)RTMemAllocZ(sizeof(VBOXIMGVFS)); if (!pVBoxImgVfs) rc = VERR_NO_MEMORY; else { pVBoxImgVfs->hVfs = hVfs; pVBoxImgVfs->pNext = pVBoxImgVfsHead; pVBoxImgVfsHead = pVBoxImgVfs; RTDvmVolumeSetQueryBlockStatusCallback(hVol, vboximgQueryBlockStatus, hVfs); } } if (RT_SUCCESS(rc)) rc = RTDvmMapQueryNextVolume(hDvm, hVol, &hVolNext); /* * Release the volume handle, the file handle has a reference * to keep it open. */ RTDvmVolumeRelease(hVol); hVol = hVolNext; } while (RT_SUCCESS(rc)); if (rc == VERR_DVM_MAP_NO_VOLUME) rc = VINF_SUCCESS; if (RT_SUCCESS(rc)) { VDIfQueryRangeUse.pfnQueryRangeUse = vboximgQueryRangeUse; VDInterfaceAdd(&VDIfQueryRangeUse.Core, "QueryRangeUse", VDINTERFACETYPE_QUERYRANGEUSE, hDvm, sizeof(VDINTERFACEQUERYRANGEUSE), &pIfsCompact); } } else if (RT_SUCCESS(rc)) RTPrintf("There are no partitions in the volume map\n"); else if (rc == VERR_NOT_FOUND) { rc = VINF_SUCCESS; RTPrintf("No known volume format on disk found\n"); } else errorRuntime("Error while opening the volume manager: %Rrf (%Rrc)\n", rc, rc); } else errorRuntime("Error creating the volume manager: %Rrf (%Rrc)\n", rc, rc); } else { rc = VERR_INVALID_STATE; errorRuntime("Error while getting the disk size\n"); } } if (RT_SUCCESS(rc)) { rc = VDCompact(pDisk, 0, pIfsCompact); if (RT_FAILURE(rc)) errorRuntime("Error while compacting image: %Rrf (%Rrc)\n", rc, rc); } while (pVBoxImgVfsHead) { PVBOXIMGVFS pVBoxImgVfsFree = pVBoxImgVfsHead; pVBoxImgVfsHead = pVBoxImgVfsHead->pNext; RTVfsRelease(pVBoxImgVfsFree->hVfs); RTMemFree(pVBoxImgVfsFree); } if (hDvm) RTDvmRelease(hDvm); VDDestroy(pDisk); return rc; } static int handleCreateCache(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; uint64_t cbSize = 0; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--size", 's', RTGETOPT_REQ_UINT64 } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 's': // --size cbSize = ValueUnion.u64; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); if (!cbSize) return errorSyntax("Mandatory --size option missing\n"); /* just try it */ rc = VDCreate(pVDIfs, VDTYPE_HDD, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); rc = VDCreateCache(pDisk, "VCI", pszFilename, cbSize, VD_IMAGE_FLAGS_DEFAULT, NULL, NULL, VD_OPEN_FLAGS_NORMAL, NULL, NULL); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk cache: %Rrf (%Rrc)\n", rc, rc); VDDestroy(pDisk); return rc; } static DECLCALLBACK(bool) vdIfCfgCreateBaseAreKeysValid(void *pvUser, const char *pszzValid) { return VINF_SUCCESS; /** @todo: Implement. */ } static DECLCALLBACK(int) vdIfCfgCreateBaseQuerySize(void *pvUser, const char *pszName, size_t *pcbValue) { AssertReturn(VALID_PTR(pcbValue), VERR_INVALID_POINTER); AssertPtrReturn(pvUser, VERR_GENERAL_FAILURE); if (RTStrCmp(pszName, "DataAlignment")) return VERR_CFGM_VALUE_NOT_FOUND; *pcbValue = strlen((const char *)pvUser) + 1 /* include terminator */; return VINF_SUCCESS; } static DECLCALLBACK(int) vdIfCfgCreateBaseQuery(void *pvUser, const char *pszName, char *pszValue, size_t cchValue) { AssertReturn(VALID_PTR(pszValue), VERR_INVALID_POINTER); AssertPtrReturn(pvUser, VERR_GENERAL_FAILURE); if (RTStrCmp(pszName, "DataAlignment")) return VERR_CFGM_VALUE_NOT_FOUND; if (strlen((const char *)pvUser) >= cchValue) return VERR_CFGM_NOT_ENOUGH_SPACE; memcpy(pszValue, pvUser, strlen((const char *)pvUser) + 1); return VINF_SUCCESS; } static int handleCreateBase(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; const char *pszBackend = "VDI"; const char *pszVariant = NULL; unsigned uImageFlags = VD_IMAGE_FLAGS_NONE; uint64_t cbSize = 0; const char *pszDataAlignment = NULL; VDGEOMETRY LCHSGeometry, PCHSGeometry; PVDINTERFACE pVDIfsOperation = NULL; VDINTERFACECONFIG vdIfCfg; memset(&LCHSGeometry, 0, sizeof(LCHSGeometry)); memset(&PCHSGeometry, 0, sizeof(PCHSGeometry)); /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--size", 's', RTGETOPT_REQ_UINT64 }, { "--format", 'b', RTGETOPT_REQ_STRING }, { "--variant", 'v', RTGETOPT_REQ_STRING }, { "--dataalignment", 'a', RTGETOPT_REQ_STRING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 's': // --size cbSize = ValueUnion.u64; break; case 'b': // --format pszBackend = ValueUnion.psz; break; case 'v': // --variant pszVariant = ValueUnion.psz; break; case 'a': // --dataalignment pszDataAlignment = ValueUnion.psz; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); if (!cbSize) return errorSyntax("Mandatory --size option missing\n"); if (pszVariant) { rc = parseDiskVariant(pszVariant, &uImageFlags); if (RT_FAILURE(rc)) return errorSyntax("Invalid variant %s given\n", pszVariant); } /* Setup the config interface if required. */ if (pszDataAlignment) { vdIfCfg.pfnAreKeysValid = vdIfCfgCreateBaseAreKeysValid; vdIfCfg.pfnQuerySize = vdIfCfgCreateBaseQuerySize; vdIfCfg.pfnQuery = vdIfCfgCreateBaseQuery; VDInterfaceAdd(&vdIfCfg.Core, "Config", VDINTERFACETYPE_CONFIG, (void *)pszDataAlignment, sizeof(vdIfCfg), &pVDIfsOperation); } /* just try it */ rc = VDCreate(pVDIfs, VDTYPE_HDD, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); rc = VDCreateBase(pDisk, pszBackend, pszFilename, cbSize, uImageFlags, NULL, &PCHSGeometry, &LCHSGeometry, NULL, VD_OPEN_FLAGS_NORMAL, NULL, pVDIfsOperation); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk: %Rrf (%Rrc)\n", rc, rc); VDDestroy(pDisk); return rc; } static int handleRepair(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; char *pszBackend = NULL; const char *pszFormat = NULL; bool fDryRun = false; VDTYPE enmType = VDTYPE_HDD; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--dry-run", 'd', RTGETOPT_REQ_NOTHING }, { "--format", 'b', RTGETOPT_REQ_STRING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 'd': // --dry-run fDryRun = true; break; case 'b': // --format pszFormat = ValueUnion.psz; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* just try it */ if (!pszFormat) { rc = VDGetFormat(NULL, NULL, pszFilename, &pszBackend, &enmType); if (RT_FAILURE(rc)) return errorSyntax("Format autodetect failed: %Rrc\n", rc); pszFormat = pszBackend; } rc = VDRepair(pVDIfs, NULL, pszFilename, pszFormat, fDryRun ? VD_REPAIR_DRY_RUN : 0); if (RT_FAILURE(rc)) rc = errorRuntime("Error while repairing the virtual disk: %Rrf (%Rrc)\n", rc, rc); if (pszBackend) RTStrFree(pszBackend); return rc; } static int handleClearComment(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; bool fDryRun = false; /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); /* just try it */ char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorSyntax("Format autodetect failed: %Rrc\n", rc); rc = VDCreate(pVDIfs, enmType, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open the image */ rc = VDOpen(pDisk, pszFormat, pszFilename, VD_OPEN_FLAGS_INFO, NULL); if (RT_FAILURE(rc)) return errorRuntime("Error while opening the image: %Rrf (%Rrc)\n", rc, rc); VDSetComment(pDisk, 0, NULL); VDDestroy(pDisk); return rc; } static int handleClearResize(HandlerArg *a) { int rc = VINF_SUCCESS; PVBOXHDD pDisk = NULL; const char *pszFilename = NULL; uint64_t cbNew = 0; bool fDryRun = false; VDGEOMETRY LCHSGeometry, PCHSGeometry; memset(&LCHSGeometry, 0, sizeof(LCHSGeometry)); memset(&PCHSGeometry, 0, sizeof(PCHSGeometry)); /* Parse the command line. */ static const RTGETOPTDEF s_aOptions[] = { { "--filename", 'f', RTGETOPT_REQ_STRING }, { "--size", 's', RTGETOPT_REQ_UINT64 } }; int ch; RTGETOPTUNION ValueUnion; RTGETOPTSTATE GetState; RTGetOptInit(&GetState, a->argc, a->argv, s_aOptions, RT_ELEMENTS(s_aOptions), 0, 0 /* fFlags */); while ((ch = RTGetOpt(&GetState, &ValueUnion))) { switch (ch) { case 'f': // --filename pszFilename = ValueUnion.psz; break; case 's': // --size cbNew = ValueUnion.u64; break; default: ch = RTGetOptPrintError(ch, &ValueUnion); printUsage(g_pStdErr); return ch; } } /* Check for mandatory parameters. */ if (!pszFilename) return errorSyntax("Mandatory --filename option missing\n"); if (!cbNew) return errorSyntax("Mandatory --size option missing or invalid\n"); /* just try it */ char *pszFormat = NULL; VDTYPE enmType = VDTYPE_INVALID; rc = VDGetFormat(NULL, NULL, pszFilename, &pszFormat, &enmType); if (RT_FAILURE(rc)) return errorSyntax("Format autodetect failed: %Rrc\n", rc); rc = VDCreate(pVDIfs, enmType, &pDisk); if (RT_FAILURE(rc)) return errorRuntime("Error while creating the virtual disk container: %Rrf (%Rrc)\n", rc, rc); /* Open the image */ rc = VDOpen(pDisk, pszFormat, pszFilename, VD_OPEN_FLAGS_NORMAL, NULL); if (RT_FAILURE(rc)) return errorRuntime("Error while opening the image: %Rrf (%Rrc)\n", rc, rc); rc = VDResize(pDisk, cbNew, &PCHSGeometry, &LCHSGeometry, NULL); if (RT_FAILURE(rc)) rc = errorRuntime("Error while resizing the virtual disk: %Rrf (%Rrc)\n", rc, rc); VDDestroy(pDisk); return rc; } int main(int argc, char *argv[]) { int exitcode = 0; int rc = RTR3InitExe(argc, &argv, 0); if (RT_FAILURE(rc)) return RTMsgInitFailure(rc); g_pszProgName = RTPathFilename(argv[0]); bool fShowLogo = false; int iCmd = 1; int iCmdArg; /* global options */ for (int i = 1; i < argc || argc <= iCmd; i++) { if ( argc <= iCmd || !strcmp(argv[i], "help") || !strcmp(argv[i], "-?") || !strcmp(argv[i], "-h") || !strcmp(argv[i], "-help") || !strcmp(argv[i], "--help")) { showLogo(g_pStdOut); printUsage(g_pStdOut); return 0; } if ( !strcmp(argv[i], "-v") || !strcmp(argv[i], "-version") || !strcmp(argv[i], "-Version") || !strcmp(argv[i], "--version")) { /* Print version number, and do nothing else. */ RTPrintf("%sr%d\n", VBOX_VERSION_STRING, RTBldCfgRevision()); return 0; } if ( !strcmp(argv[i], "--nologo") || !strcmp(argv[i], "-nologo") || !strcmp(argv[i], "-q")) { /* suppress the logo */ fShowLogo = false; iCmd++; } else { break; } } iCmdArg = iCmd + 1; if (fShowLogo) showLogo(g_pStdOut); /* initialize the VD backend with dummy handlers */ VDINTERFACEERROR vdInterfaceError; vdInterfaceError.pfnError = handleVDError; vdInterfaceError.pfnMessage = handleVDMessage; rc = VDInterfaceAdd(&vdInterfaceError.Core, "VBoxManage_IError", VDINTERFACETYPE_ERROR, NULL, sizeof(VDINTERFACEERROR), &pVDIfs); rc = VDInit(); if (RT_FAILURE(rc)) { errorSyntax("Initializing backends failed! rc=%Rrc\n", rc); return 1; } /* * All registered command handlers */ static const struct { const char *command; int (*handler)(HandlerArg *a); } s_commandHandlers[] = { { "setuuid", handleSetUUID }, { "geometry", handleGeometry }, { "convert", handleConvert }, { "info", handleInfo }, { "compact", handleCompact }, { "createcache", handleCreateCache }, { "createbase", handleCreateBase }, { "repair", handleRepair }, { "clearcomment", handleClearComment }, { "resize", handleClearResize }, { NULL, NULL } }; HandlerArg handlerArg = { 0, NULL }; int commandIndex; for (commandIndex = 0; s_commandHandlers[commandIndex].command != NULL; commandIndex++) { if (!strcmp(s_commandHandlers[commandIndex].command, argv[iCmd])) { handlerArg.argc = argc - iCmdArg; handlerArg.argv = &argv[iCmdArg]; exitcode = s_commandHandlers[commandIndex].handler(&handlerArg); break; } } if (!s_commandHandlers[commandIndex].command) { errorSyntax("Invalid command '%s'", argv[iCmd]); return 1; } rc = VDShutdown(); if (RT_FAILURE(rc)) { errorSyntax("Unloading backends failed! rc=%Rrc\n", rc); return 1; } return exitcode; } /* dummy stub for RuntimeR3 */ #ifndef RT_OS_WINDOWS RTDECL(bool) RTAssertShouldPanic(void) { return true; } #endif