Changeset 16601 in vbox for trunk/src/VBox/Frontends/VBoxManage

Timestamp:

Feb 9, 2009 4:33:28 PM (16 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

42567

Message:

OVF: change IVirtualSystemDescription::setFinalValues method; remove IVirtualSystemDescription::disableItem; API docs; implement lots of VBoxManage import command line handling

File:

• trunk/src/VBox/Frontends/VBoxManage/VBoxManageImport.cpp (modified) (8 diffs)

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

@@ -37 +37 @@
 
 #include <list>
+#include <map>
 #endif /* !VBOX_ONLY_DOCS */
 
@@ -49 +50 @@
 // funcs
 ///////////////////////////////////////////////////////////////////////////////
+
+typedef std::map<Utf8Str, Utf8Str> ArgsMap;         // pairs of strings like "-vmname" => "newvmname"
+typedef std::map<uint32_t, ArgsMap> ArgsMapsMap;   // map of maps, one for each virtual system, sorted by index
+
+static bool findArgValue(Utf8Str &strOut,
+                         const ArgsMap *pmapArgs,
+                         const Utf8Str &strKey)
+{
+    if (pmapArgs)
+    {
+        ArgsMap::const_iterator it;
+        it = pmapArgs->find(strKey);
+        if (it != pmapArgs->end())
+        {
+            strOut = it->second;
+            return true;
+        }
+    }
+
+    return false;
+}
 
 int handleImportAppliance(HandlerArg *a)
@@ -57 +79 @@
     bool fExecute = false;                  // if true, then we actually do the import (-exec argument)
 
-    for (int i = 0; i < a->argc; i++)
+    uint32_t ulCurVsys = (uint32_t)-1;
+
+    // for each -vsys X command, maintain a map of command line items
+    // (we'll parse them later after interpreting the OVF, when we can
+    // actually check whether they make sense semantically)
+    ArgsMapsMap mapArgsMapsPerVsys;
+
+    for (int i = 0;
+         i < a->argc;
+         ++i)
     {
-        if (!strcmp(a->argv[i], "-exec"))
+        Utf8Str strThisArg(a->argv[i]);
+        if (strThisArg == "-exec")
             fExecute = true;
+        else if (strThisArg == "-vsys")
+        {
+            if (++i < a->argc)
+            {
+                uint32_t ulVsys;
+                if (VINF_SUCCESS == (rc = Utf8Str(a->argv[i]).toInt(ulVsys)))       // don't use SUCCESS() macro, fail even on warnings
+                    ulCurVsys = ulVsys;
+                else
+                    return errorSyntax(USAGE_IMPORTAPPLIANCE, "Argument to -vsys option must be a non-negative number.");
+            }
+            else
+                return errorSyntax(USAGE_IMPORTAPPLIANCE, "Missing argument to -vsys option.");
+        }
+        else if (    (strThisArg == "-ostype")
+                  || (strThisArg == "-vmname")
+                  || (strThisArg == "-memory")
+                  || (strThisArg == "-ignore")
+                  || (strThisArg.substr(0, 5) == "-type")
+                  || (strThisArg.substr(0, 11) == "-controller")
+                )
+        {
+            if (ulCurVsys == (uint32_t)-1)
+                return errorSyntax(USAGE_IMPORTAPPLIANCE, "Option \"%s\" requires preceding -vsys argument.", strThisArg.c_str());
+
+            // store both this arg and the next one in the strings map for later parsing
+            if (++i < a->argc)
+                mapArgsMapsPerVsys[ulCurVsys][strThisArg] = Utf8Str(a->argv[i]);
+            else
+                return errorSyntax(USAGE_IMPORTAPPLIANCE, "Missing argument to \"%s\" option.", strThisArg.c_str());
+        }
+        else if (strThisArg[0] == '-')
+            return errorSyntax(USAGE_IMPORTAPPLIANCE, "Unknown option \"%s\".", strThisArg.c_str());
         else if (!strOvfFilename)
-            strOvfFilename = a->argv[i];
+            strOvfFilename = strThisArg;
         else
             return errorSyntax(USAGE_IMPORTAPPLIANCE, "Too many arguments for \"import\" command.");
@@ -76 +140 @@
         CHECK_ERROR_BREAK(a->virtualBox, OpenAppliance(bstrOvfFilename, appliance.asOutParam()));
 
-        RTPrintf("Interpreting...\n");
+        RTPrintf("Interpreting %s... ", strOvfFilename.c_str());
         CHECK_ERROR_BREAK(appliance, Interpret());
         RTPrintf("OK.\n");
@@ -96 +160 @@
         CHECK_ERROR_BREAK(appliance,
                           COMGETTER(VirtualSystemDescriptions)(ComSafeArrayAsOutParam(aVirtualSystemDescriptions)));
-        if (aVirtualSystemDescriptions.size() > 0)
-        {
-            for (unsigned i = 0; i < aVirtualSystemDescriptions.size(); ++i)
+
+        uint32_t cVirtualSystemDescriptions = aVirtualSystemDescriptions.size();
+
+        // match command line arguments with virtual system descriptions;
+        // this is only to sort out invalid indices at this time
+        ArgsMapsMap::const_iterator it;
+        for (it = mapArgsMapsPerVsys.begin();
+             it != mapArgsMapsPerVsys.end();
+             ++it)
+        {
+            uint32_t ulVsys = it->first;
+            if (ulVsys >= cVirtualSystemDescriptions)
+                return errorSyntax(USAGE_IMPORTAPPLIANCE,
+                                   "Invalid index %RI32 with -vsys option; the OVF contains only %RI32 virtual system(s).",
+                                   ulVsys, cVirtualSystemDescriptions);
+        }
+
+        // dump virtual system descriptions and match command-line arguments
+        if (cVirtualSystemDescriptions > 0)
+        {
+            for (unsigned i = 0; i < cVirtualSystemDescriptions; ++i)
             {
                 com::SafeArray<VirtualSystemDescriptionType_T> retTypes;
@@ -113 +195 @@
 
                 RTPrintf("Virtual system %i:\n", i);
+
+                // look up the corresponding command line options, if any
+                const ArgsMap *pmapArgs = NULL;
+                ArgsMapsMap::const_iterator itm = mapArgsMapsPerVsys.find(i);
+                if (itm != mapArgsMapsPerVsys.end())
+                    pmapArgs = &itm->second;
+
+//                     ArgsMap::const_iterator it3;
+//                     for (it3 = pmapArgs->begin();
+//                          it3 != pmapArgs->end();
+//                          ++it3)
+//                     {
+//                         RTPrintf("%s -> %s\n", it3->first.c_str(), it3->second.c_str());
+//                     }
+//                 }
+
+//                 Bstr bstrVMName;
+//                 Bstr bstrOSType;
+
+                // this collects the final values for setFinalValues()
+                com::SafeArray<BOOL> aEnabled(retTypes.size());
+                com::SafeArray<BSTR> aFinalValues(retTypes.size());
+
                 for (unsigned a = 0; a < retTypes.size(); ++a)
                 {
                     VirtualSystemDescriptionType_T t = retTypes[a];
 
-                    Bstr bstrVMname;
-                    Bstr bstrOstype;
-                    uint32_t ulMemMB;
-                    bool fUSB = false;
+                    Utf8Str strOverride;
+
+                    Bstr bstrFinalValue = aConfigValues[a];
 
                     switch (t)
                     {
                         case VirtualSystemDescriptionType_Name:
-                            bstrVMname = aConfigValues[a];
-                            RTPrintf("%2d: Suggested VM name \"%ls\""
-                                     "\n    (change with \"-vsys %d -vmname <name>\")\n",
-                                     a, bstrVMname.raw(), i);
+                            if (findArgValue(strOverride, pmapArgs, "-vmname"))
+                            {
+                                bstrFinalValue = strOverride;
+                                RTPrintf("%2d: VM name specified with -vmname: \"%ls\"\n",
+                                        a, bstrFinalValue.raw());
+                            }
+                            else
+                                RTPrintf("%2d: Suggested VM name \"%ls\""
+                                        "\n    (change with \"-vsys %d -vmname <name>\")\n",
+                                        a, bstrFinalValue.raw(), i);
                         break;
 
                         case VirtualSystemDescriptionType_OS:
-                            bstrOstype = aConfigValues[a];
-                            RTPrintf("%2d: Suggested OS type: \"%ls\""
-                                     "\n    (change with \"-vsys %d -ostype <type>\"; use \"list ostypes\" to list all)\n",
-                                     a, bstrOstype.raw(), i);
+                            if (findArgValue(strOverride, pmapArgs, "-ostype"))
+                            {
+                                bstrFinalValue = strOverride;
+                                RTPrintf("%2d: OS type specified with -ostype: \"%ls\"\n",
+                                        a, bstrFinalValue.raw());
+                            }
+                            else
+                                RTPrintf("%2d: Suggested OS type: \"%ls\""
+                                        "\n    (change with \"-vsys %d -ostype <type>\"; use \"list ostypes\" to list all)\n",
+                                        a, bstrFinalValue.raw(), i);
                         break;
 
@@ -144 +260 @@
 
                         case VirtualSystemDescriptionType_Memory:
-                            Utf8Str(Bstr(aConfigValues[a])).toInt(ulMemMB);
-                            RTPrintf("%2d: Guest memory: %u MB\n    (change with \"-vsys %d -memory <MB>\")\n",
-                                     a, ulMemMB, i);
+                        {
+                            if (findArgValue(strOverride, pmapArgs, "-memory"))
+                            {
+                                uint32_t ulMemMB;
+                                if (VINF_SUCCESS == strOverride.toInt(ulMemMB))
+                                {
+                                    bstrFinalValue = strOverride;
+                                    RTPrintf("%2d: Guest memory specified with -memory: %ls MB\n",
+                                             a, bstrFinalValue.raw());
+                                }
+                                else
+                                    return errorSyntax(USAGE_IMPORTAPPLIANCE,
+                                                       "Argument to -memory option must be a non-negative number.");
+                            }
+                            else
+                                RTPrintf("%2d: Guest memory: %ls MB\n    (change with \"-vsys %d -memory <MB>\")\n",
+                                         a, bstrFinalValue.raw(), i);
+                        }
                         break;
 
@@ -220 +351 @@
                         break;
                     }
+
+                    bstrFinalValue.detachTo(&aFinalValues[a]);
                 }
-            }
+
+                if (fExecute)
+                    CHECK_ERROR_BREAK(aVirtualSystemDescriptions[i],
+                                      SetFinalValues(ComSafeArrayAsInParam(aEnabled),
+                                                     ComSafeArrayAsInParam(aFinalValues)));
+
+            } // for (unsigned i = 0; i < cVirtualSystemDescriptions; ++i)
 
             if (fExecute)