199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome#ifndef VBOX_ONLY_DOCS

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/com/com.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/com/array.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/com/ErrorInfo.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/com/errorprint2.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/com/VirtualBox.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/asm.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/file.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/stream.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/string.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/ctype.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <iprt/getopt.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/log.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include <VBox/VBoxHDD.h>

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome#include "VBoxManage.h"

199767f8919635c4928607450d9e0abb932109ceToomas Soomeusing namespace com;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic DECLCALLBACK(void) handleVDError(void *pvUser, int rc, RT_SRC_POS_DECL, const char *pszFormat, va_list va)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("ERROR: ");

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintfV(pszFormat, va);

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("\n");

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Error code %Rrc at %s(%u) in function %s\n", rc, RT_SRC_POS_ARGS);

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int parseDiskVariant(const char *psz, HardDiskVariant_T *pDiskVariant)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome int rc = VINF_SUCCESS;

199767f8919635c4928607450d9e0abb932109ceToomas Soome unsigned DiskVariant = (unsigned)(*pDiskVariant);

199767f8919635c4928607450d9e0abb932109ceToomas Soome while (psz && *psz && RT_SUCCESS(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome size_t len;

199767f8919635c4928607450d9e0abb932109ceToomas Soome const char *pszComma = strchr(psz, ',');

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (pszComma)

199767f8919635c4928607450d9e0abb932109ceToomas Soome len = pszComma - psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome len = strlen(psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (len > 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome // Parsing is intentionally inconsistent: "standard" resets the

199767f8919635c4928607450d9e0abb932109ceToomas Soome // variant, whereas the other flags are cumulative.

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!RTStrNICmp(psz, "standard", len))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant = HardDiskVariant_Standard;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if ( !RTStrNICmp(psz, "fixed", len)

199767f8919635c4928607450d9e0abb932109ceToomas Soome || !RTStrNICmp(psz, "static", len))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant |= HardDiskVariant_Fixed;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (!RTStrNICmp(psz, "Diff", len))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant |= HardDiskVariant_Diff;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (!RTStrNICmp(psz, "split2g", len))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant |= HardDiskVariant_VmdkSplit2G;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if ( !RTStrNICmp(psz, "stream", len)

199767f8919635c4928607450d9e0abb932109ceToomas Soome || !RTStrNICmp(psz, "streamoptimized", len))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant |= HardDiskVariant_VmdkStreamOptimized;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = VERR_PARSE_ERROR;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (pszComma)

199767f8919635c4928607450d9e0abb932109ceToomas Soome psz += len + 1;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome psz += len;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_SUCCESS(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome *pDiskVariant = (HardDiskVariant_T)DiskVariant;

199767f8919635c4928607450d9e0abb932109ceToomas Soome return rc;

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic int parseDiskType(const char *psz, HardDiskType_T *pDiskType)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome int rc = VINF_SUCCESS;

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskType_T DiskType = HardDiskType_Normal;

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!RTStrICmp(psz, "normal"))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskType = HardDiskType_Normal;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (RTStrICmp(psz, "immutable"))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskType = HardDiskType_Immutable;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (RTStrICmp(psz, "writethrough"))

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskType = HardDiskType_Writethrough;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = VERR_PARSE_ERROR;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_SUCCESS(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome *pDiskType = DiskType;

199767f8919635c4928607450d9e0abb932109ceToomas Soome return rc;

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic const RTGETOPTDEF g_aCreateHardDiskOptions[] =

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--filename", 'f', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-filename", 'f', RTGETOPT_REQ_STRING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--size", 's', RTGETOPT_REQ_UINT64 },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-size", 's', RTGETOPT_REQ_UINT64 }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--format", 'o', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-format", 'o', RTGETOPT_REQ_STRING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--static", 'F', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-static", 'F', RTGETOPT_REQ_NOTHING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--variant", 'm', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-variant", 'm', RTGETOPT_REQ_STRING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--type", 't', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-type", 't', RTGETOPT_REQ_STRING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--comment", 'c', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-comment", 'c', RTGETOPT_REQ_STRING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--remember", 'r', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-remember", 'r', RTGETOPT_REQ_NOTHING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--register", 'r', RTGETOPT_REQ_NOTHING }, // deprecated (inofficial)

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-register", 'r', RTGETOPT_REQ_NOTHING }, // deprecated

199767f8919635c4928607450d9e0abb932109ceToomas Soome};

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomeint handleCreateHardDisk(HandlerArg *a)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome HRESULT rc;

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr filename;

199767f8919635c4928607450d9e0abb932109ceToomas Soome uint64_t sizeMB = 0;

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr format = "VDI";

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskVariant_T DiskVariant = HardDiskVariant_Standard;

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr comment;

199767f8919635c4928607450d9e0abb932109ceToomas Soome bool fRemember = false;

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskType_T DiskType = HardDiskType_Normal;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome int c;

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGETOPTUNION ValueUnion;

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGETOPTSTATE GetState;

199767f8919635c4928607450d9e0abb932109ceToomas Soome // start at 0 because main() has hacked both the argc and argv given to us

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGetOptInit(&GetState, a->argc, a->argv, g_aCreateHardDiskOptions, RT_ELEMENTS(g_aCreateHardDiskOptions), 0, 0 /* fFlags */);

199767f8919635c4928607450d9e0abb932109ceToomas Soome while ((c = RTGetOpt(&GetState, &ValueUnion)))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome switch (c)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'f': // --filename

199767f8919635c4928607450d9e0abb932109ceToomas Soome filename = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 's': // --size

199767f8919635c4928607450d9e0abb932109ceToomas Soome sizeMB = ValueUnion.u64;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'o': // --format

199767f8919635c4928607450d9e0abb932109ceToomas Soome format = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'F': // --static ("fixed"/"flat")

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome unsigned uDiskVariant = (unsigned)DiskVariant;

199767f8919635c4928607450d9e0abb932109ceToomas Soome uDiskVariant |= HardDiskVariant_Fixed;

199767f8919635c4928607450d9e0abb932109ceToomas Soome DiskVariant = (HardDiskVariant_T)uDiskVariant;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'm': // --variant

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = parseDiskVariant(ValueUnion.psz, &DiskVariant);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_FAILURE(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk variant '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'c': // --comment

199767f8919635c4928607450d9e0abb932109ceToomas Soome comment = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'r': // --remember

199767f8919635c4928607450d9e0abb932109ceToomas Soome fRemember = true;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 't': // --type

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = parseDiskType(ValueUnion.psz, &DiskType);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if ( RT_FAILURE(rc)

199767f8919635c4928607450d9e0abb932109ceToomas Soome || (DiskType != HardDiskType_Normal && DiskType != HardDiskType_Writethrough))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk type '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case VINF_GETOPT_NOT_OPTION:

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "Invalid parameter '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome default:

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (c > 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_C_IS_PRINT(c))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "Invalid option -%c", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "Invalid option case %i", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (ValueUnion.pDef)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "%s: %Rrs", ValueUnion.pDef->pszLong, c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "error: %Rrs", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* check the outcome */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!filename || (sizeMB == 0))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CREATEHD, "Parameters -filename and -size are required");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome ComPtr<IHardDisk> hardDisk;

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(a->virtualBox, CreateHardDisk(format, filename, hardDisk.asOutParam()));

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (SUCCEEDED(rc) && hardDisk)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* we will close the hard disk after the storage has been successfully

199767f8919635c4928607450d9e0abb932109ceToomas Soome bool doClose = false;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!comment.isNull())

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk,COMSETTER(Description)(comment));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome ComPtr<IProgress> progress;

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, CreateBaseStorage(sizeMB, DiskVariant, progress.asOutParam()));

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (SUCCEEDED(rc) && progress)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome showProgress(progress);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (SUCCEEDED(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome progress->COMGETTER(ResultCode)(&rc);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (FAILED(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome com::ProgressErrorInfo info(progress);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (info.isBasicAvailable())

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Error: failed to create hard disk. Error message: %lS\n", info.getText().raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Error: failed to create hard disk. No error message available!\n");

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome doClose = !fRemember;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome Guid uuid;

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMGETTER(Id)(uuid.asOutParam()));

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (DiskType == HardDiskType_Writethrough)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Writethrough));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Disk image created. UUID: %s\n", uuid.toString().raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (doClose)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, Close());

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome return SUCCEEDED(rc) ? 0 : 1;

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome#if 0 /* disabled until disk shrinking is implemented based on VBoxHDD */

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic DECLCALLBACK(int) hardDiskProgressCallback(PVM pVM, unsigned uPercent, void *pvUser)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome unsigned *pPercent = (unsigned *)pvUser;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (*pPercent != uPercent)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome *pPercent = uPercent;

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf(".");

199767f8919635c4928607450d9e0abb932109ceToomas Soome if ((uPercent % 10) == 0 && uPercent)

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("%d%%", uPercent);

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTStrmFlush(g_pStdOut);

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome return VINF_SUCCESS;

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomeint handleModifyHardDisk(HandlerArg *a)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome HRESULT rc;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* The uuid/filename and a command */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (a->argc < 2)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_MODIFYHD, "Incorrect number of parameters");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome ComPtr<IHardDisk> hardDisk;

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr filepath;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* first guess is that it's a UUID */

199767f8919635c4928607450d9e0abb932109ceToomas Soome Guid uuid(a->argv[0]);

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = a->virtualBox->GetHardDisk(uuid, hardDisk.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* no? then it must be a filename */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!hardDisk)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome filepath = a->argv[0];

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(a->virtualBox, FindHardDisk(filepath, hardDisk.asOutParam()));

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (FAILED(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return 1;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* let's find out which command */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (strcmp(a->argv[1], "settype") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* hard disk must be registered */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (SUCCEEDED(rc) && hardDisk)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome char *type = NULL;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (a->argc <= 2)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Missing argument for settype");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome type = a->argv[2];

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskType_T hddType;

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMGETTER(Type)(&hddType));

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (strcmp(type, "normal") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (hddType != HardDiskType_Normal)

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Normal));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (strcmp(type, "writethrough") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (hddType != HardDiskType_Writethrough)

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Writethrough));

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (strcmp(type, "immutable") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (hddType != HardDiskType_Immutable)

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Immutable));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk type '%s' specified", Utf8Str(type).raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Hard disk image not registered");

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (strcmp(a->argv[1], "autoreset") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome char *onOff = NULL;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (a->argc <= 2)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Missing argument for autoreset");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome onOff = a->argv[2];

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (strcmp(onOff, "on") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(AutoReset)(TRUE));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (strcmp(onOff, "off") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(hardDisk, COMSETTER(AutoReset)(FALSE));

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid autoreset argument '%s' specified",

199767f8919635c4928607450d9e0abb932109ceToomas Soome Utf8Str(onOff).raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (strcmp(a->argv[1], "compact") == 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome#if 1

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Error: Shrink hard disk operation is not implemented!\n");

199767f8919635c4928607450d9e0abb932109ceToomas Soome return 1;

199767f8919635c4928607450d9e0abb932109ceToomas Soome#else

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* the hard disk image might not be registered */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!hardDisk)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome a->virtualBox->OpenHardDisk(Bstr(a->argv[0]), AccessMode_ReadWrite, hardDisk.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!hardDisk)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Hard disk image not found");

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr format;

199767f8919635c4928607450d9e0abb932109ceToomas Soome hardDisk->COMGETTER(Format)(format.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (format != "VDI")

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk type. The command only works on VDI files\n");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr fileName;

199767f8919635c4928607450d9e0abb932109ceToomas Soome hardDisk->COMGETTER(Location)(fileName.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* make sure the object reference is released */

199767f8919635c4928607450d9e0abb932109ceToomas Soome hardDisk = NULL;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome unsigned uProcent;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Shrinking '%lS': 0%%", fileName.raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome int vrc = VDIShrinkImage(Utf8Str(fileName).raw(), hardDiskProgressCallback, &uProcent);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_FAILURE(vrc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTPrintf("Error while shrinking hard disk image: %Rrc\n", vrc);

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = E_FAIL;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome#endif

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_MODIFYHD, "Invalid parameter '%s'", Utf8Str(a->argv[1]).raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome return SUCCEEDED(rc) ? 0 : 1;

199767f8919635c4928607450d9e0abb932109ceToomas Soome}

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomestatic const RTGETOPTDEF g_aCloneHardDiskOptions[] =

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--format", 'o', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-format", 'o', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--static", 'F', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-static", 'F', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--variant", 'm', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-variant", 'm', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--type", 't', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-type", 't', RTGETOPT_REQ_STRING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--remember", 'r', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-remember", 'r', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "--register", 'r', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome { "-register", 'r', RTGETOPT_REQ_NOTHING },

199767f8919635c4928607450d9e0abb932109ceToomas Soome};

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soomeint handleCloneHardDisk(HandlerArg *a)

199767f8919635c4928607450d9e0abb932109ceToomas Soome{

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr src, dst;

199767f8919635c4928607450d9e0abb932109ceToomas Soome Bstr format;

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskVariant_T DiskVariant = HardDiskVariant_Standard;

199767f8919635c4928607450d9e0abb932109ceToomas Soome bool fRemember = false;

199767f8919635c4928607450d9e0abb932109ceToomas Soome HardDiskType_T DiskType = HardDiskType_Normal;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome HRESULT rc;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome int c;

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGETOPTUNION ValueUnion;

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGETOPTSTATE GetState;

199767f8919635c4928607450d9e0abb932109ceToomas Soome // start at 0 because main() has hacked both the argc and argv given to us

199767f8919635c4928607450d9e0abb932109ceToomas Soome RTGetOptInit(&GetState, a->argc, a->argv, g_aCloneHardDiskOptions, RT_ELEMENTS(g_aCloneHardDiskOptions), 0, 0 /* fFlags */);

199767f8919635c4928607450d9e0abb932109ceToomas Soome while ((c = RTGetOpt(&GetState, &ValueUnion)))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome switch (c)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'o': // --format

199767f8919635c4928607450d9e0abb932109ceToomas Soome format = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'm': // --variant

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = parseDiskVariant(ValueUnion.psz, &DiskVariant);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_FAILURE(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk variant '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 'r': // --remember

199767f8919635c4928607450d9e0abb932109ceToomas Soome fRemember = true;

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case 't': // --type

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = parseDiskType(ValueUnion.psz, &DiskType);

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_FAILURE(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorArgument("Invalid hard disk type '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome case VINF_GETOPT_NOT_OPTION:

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (src.isEmpty())

199767f8919635c4928607450d9e0abb932109ceToomas Soome src = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (dst.isEmpty())

199767f8919635c4928607450d9e0abb932109ceToomas Soome dst = ValueUnion.psz;

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "Invalid parameter '%s'", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome default:

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (c > 0)

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (RT_C_IS_GRAPH(c))

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "unhandled option: -%c", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "unhandled option: %i", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (c == VERR_GETOPT_UNKNOWN_OPTION)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "unknown option: %s", ValueUnion.psz);

199767f8919635c4928607450d9e0abb932109ceToomas Soome else if (ValueUnion.pDef)

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "%s: %Rrs", ValueUnion.pDef->pszLong, c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome else

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "error: %Rrs", c);

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (src.isEmpty())

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "Mandatory UUID or input file parameter missing");

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (dst.isEmpty())

199767f8919635c4928607450d9e0abb932109ceToomas Soome return errorSyntax(USAGE_CLONEHD, "Mandatory output file parameter missing");

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome ComPtr<IHardDisk> srcDisk;

199767f8919635c4928607450d9e0abb932109ceToomas Soome ComPtr<IHardDisk> dstDisk;

199767f8919635c4928607450d9e0abb932109ceToomas Soome bool unknown = false;

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* first guess is that it's a UUID */

199767f8919635c4928607450d9e0abb932109ceToomas Soome Guid uuid(Utf8Str(src).raw());

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = a->virtualBox->GetHardDisk(uuid, srcDisk.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* no? then it must be a filename */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (FAILED (rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome rc = a->virtualBox->FindHardDisk(src, srcDisk.asOutParam());

199767f8919635c4928607450d9e0abb932109ceToomas Soome /* no? well, then it's an unkwnown image */

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (FAILED (rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome CHECK_ERROR(a->virtualBox, OpenHardDisk(src, AccessMode_ReadWrite, srcDisk.asOutParam()));

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (SUCCEEDED (rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome unknown = true;

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome }

199767f8919635c4928607450d9e0abb932109ceToomas Soome

199767f8919635c4928607450d9e0abb932109ceToomas Soome do

199767f8919635c4928607450d9e0abb932109ceToomas Soome {

199767f8919635c4928607450d9e0abb932109ceToomas Soome if (!SUCCEEDED(rc))

199767f8919635c4928607450d9e0abb932109ceToomas Soome break;

if (format.isEmpty())

{

/* get the format of the source hard disk */

CHECK_ERROR_BREAK(srcDisk, COMGETTER(Format) (format.asOutParam()));

}

CHECK_ERROR_BREAK(a->virtualBox, CreateHardDisk(format, dst, dstDisk.asOutParam()));

ComPtr<IProgress> progress;

CHECK_ERROR_BREAK(srcDisk, CloneTo(dstDisk, DiskVariant, NULL, progress.asOutParam()));

showProgress(progress);

progress->COMGETTER(ResultCode)(&rc);

if (FAILED(rc))

{

com::ProgressErrorInfo info(progress);

if (info.isBasicAvailable())

RTPrintf("Error: failed to clone hard disk. Error message: %lS\n", info.getText().raw());

else

RTPrintf("Error: failed to clone hard disk. No error message available!\n");

break;

}

CHECK_ERROR_BREAK(dstDisk, COMGETTER(Id)(uuid.asOutParam()));

RTPrintf("Clone hard disk created in format '%ls'. UUID: %s\n",

format.raw(), uuid.toString().raw());

}

while (0);

if (!fRemember && !dstDisk.isNull())

{

/* forget the created clone */

dstDisk->Close();

}

if (unknown)

{

/* close the unknown hard disk to forget it again */

srcDisk->Close();

}

return SUCCEEDED(rc) ? 0 : 1;

}

static const RTGETOPTDEF g_aConvertFromRawHardDiskOptions[] =

{

{ "--format", 'o', RTGETOPT_REQ_STRING },

{ "-format", 'o', RTGETOPT_REQ_STRING },

{ "--static", 'F', RTGETOPT_REQ_NOTHING },

{ "-static", 'F', RTGETOPT_REQ_NOTHING },

{ "--variant", 'm', RTGETOPT_REQ_STRING },

{ "-variant", 'm', RTGETOPT_REQ_STRING },

};

int handleConvertFromRaw(int argc, char *argv[])

{

int rc = VINF_SUCCESS;

bool fReadFromStdIn = false;

const char *format = "VDI";

const char *srcfilename = NULL;

const char *dstfilename = NULL;

const char *filesize = NULL;

unsigned uImageFlags = VD_IMAGE_FLAGS_NONE;

void *pvBuf = NULL;

int c;

RTGETOPTUNION ValueUnion;

RTGETOPTSTATE GetState;

// start at 0 because main() has hacked both the argc and argv given to us

RTGetOptInit(&GetState, argc, argv, g_aConvertFromRawHardDiskOptions, RT_ELEMENTS(g_aConvertFromRawHardDiskOptions), 0, 0 /* fFlags */);

while ((c = RTGetOpt(&GetState, &ValueUnion)))

{

switch (c)

{

case 'o': // --format

format = ValueUnion.psz;

break;

case 'm': // --variant

HardDiskVariant_T DiskVariant;

rc = parseDiskVariant(ValueUnion.psz, &DiskVariant);

if (RT_FAILURE(rc))

return errorArgument("Invalid hard disk variant '%s'", ValueUnion.psz);

/// @todo cleaner solution than assuming 1:1 mapping?

uImageFlags = (unsigned)DiskVariant;

break;

case VINF_GETOPT_NOT_OPTION:

if (!srcfilename)

{

srcfilename = ValueUnion.psz;

#if defined(RT_OS_LINUX) || defined(RT_OS_DARWIN) || defined(RT_OS_SOLARIS)

fReadFromStdIn = !strcmp(srcfilename, "stdin");

#endif

}

else if (!dstfilename)

dstfilename = ValueUnion.psz;

else if (fReadFromStdIn && !filesize)

filesize = ValueUnion.psz;

else

return errorSyntax(USAGE_CONVERTFROMRAW, "Invalid parameter '%s'", ValueUnion.psz);

break;

default:

if (c > 0)

{

if (RT_C_IS_PRINT(c))

return errorSyntax(USAGE_CONVERTFROMRAW, "Invalid option -%c", c);

else

return errorSyntax(USAGE_CONVERTFROMRAW, "Invalid option case %i", c);

}

else if (ValueUnion.pDef)

return errorSyntax(USAGE_CONVERTFROMRAW, "%s: %Rrs", ValueUnion.pDef->pszLong, c);

else

return errorSyntax(USAGE_CONVERTFROMRAW, "error: %Rrs", c);

}

}

if (!srcfilename || !dstfilename || (fReadFromStdIn && !filesize))

return errorSyntax(USAGE_CONVERTFROMRAW, "Incorrect number of parameters");

RTPrintf("Converting from raw image file=\"%s\" to file=\"%s\"...\n",

srcfilename, dstfilename);

PVBOXHDD pDisk = NULL;

PVDINTERFACE pVDIfs = NULL;

VDINTERFACE vdInterfaceError;

VDINTERFACEERROR vdInterfaceErrorCallbacks;

vdInterfaceErrorCallbacks.cbSize = sizeof(VDINTERFACEERROR);

vdInterfaceErrorCallbacks.enmInterface = VDINTERFACETYPE_ERROR;

vdInterfaceErrorCallbacks.pfnError = handleVDError;

rc = VDInterfaceAdd(&vdInterfaceError, "VBoxManage_IError", VDINTERFACETYPE_ERROR,

&vdInterfaceErrorCallbacks, NULL, &pVDIfs);

AssertRC(rc);

/* open raw image file. */

RTFILE File;

if (fReadFromStdIn)

File = 0;

else

rc = RTFileOpen(&File, srcfilename, RTFILE_O_OPEN | RTFILE_O_READ | RTFILE_O_DENY_WRITE);

if (RT_FAILURE(rc))

{

RTPrintf("File=\"%s\" open error: %Rrf\n", srcfilename, rc);

goto out;

}

uint64_t cbFile;

/* get image size. */

if (fReadFromStdIn)

cbFile = RTStrToUInt64(filesize);

else

rc = RTFileGetSize(File, &cbFile);

if (RT_FAILURE(rc))

{

RTPrintf("Error getting image size for file \"%s\": %Rrc\n", srcfilename, rc);

goto out;

}

RTPrintf("Creating %s image with size %RU64 bytes (%RU64MB)...\n", (uImageFlags & VD_IMAGE_FLAGS_FIXED) ? "fixed" : "dynamic", cbFile, (cbFile + _1M - 1) / _1M);

char pszComment[256];

RTStrPrintf(pszComment, sizeof(pszComment), "Converted image from %s", srcfilename);

rc = VDCreate(pVDIfs, &pDisk);

if (RT_FAILURE(rc))

{

RTPrintf("Error while creating the virtual disk container: %Rrc\n", rc);

goto out;

}

Assert(RT_MIN(cbFile / 512 / 16 / 63, 16383) -

(unsigned int)RT_MIN(cbFile / 512 / 16 / 63, 16383) == 0);

PDMMEDIAGEOMETRY PCHS, LCHS;

PCHS.cCylinders = (unsigned int)RT_MIN(cbFile / 512 / 16 / 63, 16383);

PCHS.cHeads = 16;

PCHS.cSectors = 63;

LCHS.cCylinders = 0;

LCHS.cHeads = 0;

LCHS.cSectors = 0;

rc = VDCreateBase(pDisk, format, dstfilename, cbFile,

uImageFlags, pszComment, &PCHS, &LCHS, NULL,

VD_OPEN_FLAGS_NORMAL, NULL, NULL);

if (RT_FAILURE(rc))

{

RTPrintf("Error while creating the disk image \"%s\": %Rrc\n", dstfilename, rc);

goto out;

}

size_t cbBuffer;

cbBuffer = _1M;

pvBuf = RTMemAlloc(cbBuffer);

if (!pvBuf)

{

rc = VERR_NO_MEMORY;

RTPrintf("Not enough memory allocating buffers for image \"%s\": %Rrc\n", dstfilename, rc);

goto out;

}

uint64_t offFile;

offFile = 0;

while (offFile < cbFile)

{

size_t cbRead;

size_t cbToRead;

cbRead = 0;

cbToRead = cbFile - offFile >= (uint64_t)cbBuffer ?

cbBuffer : (size_t) (cbFile - offFile);

rc = RTFileRead(File, pvBuf, cbToRead, &cbRead);

if (RT_FAILURE(rc) || !cbRead)

break;

rc = VDWrite(pDisk, offFile, pvBuf, cbRead);

if (RT_FAILURE(rc))

{

RTPrintf("Failed to write to disk image \"%s\": %Rrc\n", dstfilename, rc);

goto out;

}

offFile += cbRead;

}

out:

if (pvBuf)

RTMemFree(pvBuf);

if (pDisk)

VDClose(pDisk, RT_FAILURE(rc));

if (File != NIL_RTFILE)

RTFileClose(File);

return RT_FAILURE(rc);

}

int handleAddiSCSIDisk(HandlerArg *a)

{

HRESULT rc;

Bstr server;

Bstr target;

Bstr port;

Bstr lun;

Bstr username;

Bstr password;

Bstr comment;

bool fIntNet = false;

/* at least server and target */

if (a->argc < 4)

return errorSyntax(USAGE_ADDISCSIDISK, "Not enough parameters");

/* let's have a closer look at the arguments */

for (int i = 0; i < a->argc; i++)

{

if (strcmp(a->argv[i], "-server") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

server = a->argv[i];

}

else if (strcmp(a->argv[i], "-target") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

target = a->argv[i];

}

else if (strcmp(a->argv[i], "-port") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

port = a->argv[i];

}

else if (strcmp(a->argv[i], "-lun") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

lun = a->argv[i];

}

else if (strcmp(a->argv[i], "-encodedlun") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

lun = BstrFmt("enc%s", a->argv[i]);

}

else if (strcmp(a->argv[i], "-username") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

username = a->argv[i];

}

else if (strcmp(a->argv[i], "-password") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

password = a->argv[i];

}

else if (strcmp(a->argv[i], "-comment") == 0)

{

if (a->argc <= i + 1)

return errorArgument("Missing argument to '%s'", a->argv[i]);

i++;

comment = a->argv[i];

}

else if (strcmp(a->argv[i], "-intnet") == 0)

{

i++;

fIntNet = true;

}

else

return errorSyntax(USAGE_ADDISCSIDISK, "Invalid parameter '%s'", Utf8Str(a->argv[i]).raw());

}

/* check for required options */

if (!server || !target)

return errorSyntax(USAGE_ADDISCSIDISK, "Parameters -server and -target are required");

do

{

ComPtr<IHardDisk> hardDisk;

/** @todo move the location stuff to Main, which can use pfnComposeName

CHECK_ERROR_BREAK (a->virtualBox,

CreateHardDisk(Bstr ("iSCSI"),

BstrFmt ("%ls/%ls/%ls", server.raw(), target.raw(), lun.raw()),

hardDisk.asOutParam()));

CheckComRCBreakRC (rc);

if (!comment.isNull())

CHECK_ERROR_BREAK(hardDisk, COMSETTER(Description)(comment));

if (!port.isNull())

server = BstrFmt ("%ls:%ls", server.raw(), port.raw());

com::SafeArray <BSTR> names;

com::SafeArray <BSTR> values;

Bstr ("TargetAddress").detachTo (names.appendedRaw());

server.detachTo (values.appendedRaw());

Bstr ("TargetName").detachTo (names.appendedRaw());

target.detachTo (values.appendedRaw());

if (!lun.isNull())

{

Bstr ("LUN").detachTo (names.appendedRaw());

lun.detachTo (values.appendedRaw());

}

if (!username.isNull())

{

Bstr ("InitiatorUsername").detachTo (names.appendedRaw());

username.detachTo (values.appendedRaw());

}

if (!password.isNull())

{

Bstr ("InitiatorSecret").detachTo (names.appendedRaw());

password.detachTo (values.appendedRaw());

}

/// @todo add -initiator option

Bstr ("InitiatorName").detachTo (names.appendedRaw());

Bstr ("iqn.2008-04.com.sun.virtualbox.initiator").detachTo (values.appendedRaw());

/// @todo add -targetName and -targetPassword options

if (fIntNet)

{

Bstr ("HostIPStack").detachTo (names.appendedRaw());

Bstr ("0").detachTo (values.appendedRaw());

}

CHECK_ERROR_BREAK (hardDisk,

SetProperties (ComSafeArrayAsInParam (names),

ComSafeArrayAsInParam (values)));

Guid guid;

CHECK_ERROR(hardDisk, COMGETTER(Id)(guid.asOutParam()));

RTPrintf("iSCSI disk created. UUID: %s\n", guid.toString().raw());

}

while (0);

return SUCCEEDED(rc) ? 0 : 1;

}

int handleShowHardDiskInfo(HandlerArg *a)

{

HRESULT rc;

if (a->argc != 1)

return errorSyntax(USAGE_SHOWHDINFO, "Incorrect number of parameters");

ComPtr<IHardDisk> hardDisk;

Bstr filepath;

bool unknown = false;

/* first guess is that it's a UUID */

Guid uuid(a->argv[0]);

rc = a->virtualBox->GetHardDisk(uuid, hardDisk.asOutParam());

/* no? then it must be a filename */

if (FAILED (rc))

{

filepath = a->argv[0];

rc = a->virtualBox->FindHardDisk(filepath, hardDisk.asOutParam());

/* no? well, then it's an unkwnown image */

if (FAILED (rc))

{

CHECK_ERROR(a->virtualBox, OpenHardDisk(filepath, AccessMode_ReadWrite, hardDisk.asOutParam()));

if (SUCCEEDED (rc))

{

unknown = true;

}

}

}

do

{

if (!SUCCEEDED(rc))

break;

hardDisk->COMGETTER(Id)(uuid.asOutParam());

RTPrintf("UUID: %s\n", uuid.toString().raw());

/* check for accessibility */

/// @todo NEWMEDIA check accessibility of all parents

/// @todo NEWMEDIA print the full state value

MediaState_T state;

CHECK_ERROR_BREAK (hardDisk, COMGETTER(State)(&state));

RTPrintf("Accessible: %s\n", state != MediaState_Inaccessible ? "yes" : "no");

if (state == MediaState_Inaccessible)

{

Bstr err;

CHECK_ERROR_BREAK (hardDisk, COMGETTER(LastAccessError)(err.asOutParam()));

RTPrintf("Access Error: %lS\n", err.raw());

}

Bstr description;

hardDisk->COMGETTER(Description)(description.asOutParam());

if (description)

{

RTPrintf("Description: %lS\n", description.raw());

}

ULONG64 logicalSize;

hardDisk->COMGETTER(LogicalSize)(&logicalSize);

RTPrintf("Logical size: %llu MBytes\n", logicalSize);

ULONG64 actualSize;

hardDisk->COMGETTER(Size)(&actualSize);

RTPrintf("Current size on disk: %llu MBytes\n", actualSize >> 20);

ComPtr <IHardDisk> parent;

hardDisk->COMGETTER(Parent) (parent.asOutParam());

HardDiskType_T type;

hardDisk->COMGETTER(Type)(&type);

const char *typeStr = "unknown";

switch (type)

{

case HardDiskType_Normal:

if (!parent.isNull())

typeStr = "normal (differencing)";

else

typeStr = "normal (base)";

break;

case HardDiskType_Immutable:

typeStr = "immutable";

break;

case HardDiskType_Writethrough:

typeStr = "writethrough";

break;

}

RTPrintf("Type: %s\n", typeStr);

Bstr format;

hardDisk->COMGETTER(Format)(format.asOutParam());

RTPrintf("Storage format: %lS\n", format.raw());

if (!unknown)

{

com::SafeGUIDArray machineIds;

hardDisk->COMGETTER(MachineIds)(ComSafeArrayAsOutParam(machineIds));

for (size_t j = 0; j < machineIds.size(); ++ j)

{

ComPtr<IMachine> machine;

CHECK_ERROR(a->virtualBox, GetMachine(machineIds[j], machine.asOutParam()));

ASSERT(machine);

Bstr name;

machine->COMGETTER(Name)(name.asOutParam());

machine->COMGETTER(Id)(uuid.asOutParam());

RTPrintf("%s%lS (UUID: %RTuuid)\n",

j == 0 ? "In use by VMs: " : " ",

name.raw(), &machineIds[j]);

}

/// @todo NEWMEDIA check usage in snapshots too

/// @todo NEWMEDIA also list children

}

Bstr loc;

hardDisk->COMGETTER(Location)(loc.asOutParam());

RTPrintf("Location: %lS\n", loc.raw());

/* print out information specific for differencing hard disks */

if (!parent.isNull())

{

BOOL autoReset = FALSE;

hardDisk->COMGETTER(AutoReset)(&autoReset);

RTPrintf("Auto-Reset: %s\n", autoReset ? "on" : "off");

}

}

while (0);

if (unknown)

{

/* close the unknown hard disk to forget it again */

hardDisk->Close();

}

return SUCCEEDED(rc) ? 0 : 1;

}

int handleOpenMedium(HandlerArg *a)

{

HRESULT rc;

if (a->argc < 2)

return errorSyntax(USAGE_REGISTERIMAGE, "Not enough parameters");

Bstr filepath(a->argv[1]);

if (strcmp(a->argv[0], "disk") == 0)

{

const char *type = NULL;

/* there can be a type parameter */

if ((a->argc > 2) && (a->argc != 4))

return errorSyntax(USAGE_REGISTERIMAGE, "Incorrect number of parameters");

if (a->argc == 4)

{

if (strcmp(a->argv[2], "-type") != 0)

return errorSyntax(USAGE_REGISTERIMAGE, "Invalid parameter '%s'", Utf8Str(a->argv[2]).raw());

if ( (strcmp(a->argv[3], "normal") != 0)

&& (strcmp(a->argv[3], "immutable") != 0)

&& (strcmp(a->argv[3], "writethrough") != 0))

return errorArgument("Invalid hard disk type '%s' specified", Utf8Str(a->argv[3]).raw());

type = a->argv[3];

}

ComPtr<IHardDisk> hardDisk;

CHECK_ERROR(a->virtualBox, OpenHardDisk(filepath, AccessMode_ReadWrite, hardDisk.asOutParam()));

if (SUCCEEDED(rc) && hardDisk)

{

/* change the type if requested */

if (type)

{

if (strcmp(type, "normal") == 0)

CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Normal));

else if (strcmp(type, "immutable") == 0)

CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Immutable));

else if (strcmp(type, "writethrough") == 0)

CHECK_ERROR(hardDisk, COMSETTER(Type)(HardDiskType_Writethrough));

}

}

}

else if (strcmp(a->argv[0], "dvd") == 0)

{

ComPtr<IDVDImage> dvdImage;

CHECK_ERROR(a->virtualBox, OpenDVDImage(filepath, Guid(), dvdImage.asOutParam()));

}

else if (strcmp(a->argv[0], "floppy") == 0)

{

ComPtr<IFloppyImage> floppyImage;

CHECK_ERROR(a->virtualBox, OpenFloppyImage(filepath, Guid(), floppyImage.asOutParam()));

}

else

return errorSyntax(USAGE_REGISTERIMAGE, "Invalid parameter '%s'", Utf8Str(a->argv[1]).raw());

return SUCCEEDED(rc) ? 0 : 1;

}

int handleCloseMedium(HandlerArg *a)

{

HRESULT rc;

if (a->argc != 2)

return errorSyntax(USAGE_UNREGISTERIMAGE, "Incorrect number of parameters");

/* first guess is that it's a UUID */

Guid uuid(a->argv[1]);

if (strcmp(a->argv[0], "disk") == 0)

{

ComPtr<IHardDisk> hardDisk;

rc = a->virtualBox->GetHardDisk(uuid, hardDisk.asOutParam());

/* not a UUID or not registered? Then it must be a filename */

if (!hardDisk)

{

CHECK_ERROR(a->virtualBox, FindHardDisk(Bstr(a->argv[1]), hardDisk.asOutParam()));

}

if (SUCCEEDED(rc) && hardDisk)

{

CHECK_ERROR(hardDisk, Close());

}

}

else

if (strcmp(a->argv[0], "dvd") == 0)

{

ComPtr<IDVDImage> dvdImage;

rc = a->virtualBox->GetDVDImage(uuid, dvdImage.asOutParam());

/* not a UUID or not registered? Then it must be a filename */

if (!dvdImage)

{

CHECK_ERROR(a->virtualBox, FindDVDImage(Bstr(a->argv[1]), dvdImage.asOutParam()));

}

if (SUCCEEDED(rc) && dvdImage)

{

CHECK_ERROR(dvdImage, Close());

}

}

else

if (strcmp(a->argv[0], "floppy") == 0)

{

ComPtr<IFloppyImage> floppyImage;

rc = a->virtualBox->GetFloppyImage(uuid, floppyImage.asOutParam());

/* not a UUID or not registered? Then it must be a filename */

if (!floppyImage)

{

CHECK_ERROR(a->virtualBox, FindFloppyImage(Bstr(a->argv[1]), floppyImage.asOutParam()));

}

if (SUCCEEDED(rc) && floppyImage)

{

CHECK_ERROR(floppyImage, Close());

}

}

else

return errorSyntax(USAGE_UNREGISTERIMAGE, "Invalid parameter '%s'", Utf8Str(a->argv[1]).raw());

return SUCCEEDED(rc) ? 0 : 1;

}

#endif /* !VBOX_ONLY_DOCS */