vbox的更動 39557 路徑 trunk/src/VBox/NetworkServices/NetLib/VBoxNetBaseService.cpp

時間撮記:

2011-12-8 上午05:55:10 (13 年 以前)

作者:

vboxsync

訊息:

NAT/service: registering port-forawarding from command line.

檔案:

• trunk/src/VBox/NetworkServices/NetLib/VBoxNetBaseService.cpp (修改) (7 筆差異)

trunk/src/VBox/NetworkServices/NetLib/VBoxNetBaseService.cpp

@@ -39 +39 @@
 #include <iprt/string.h>
 #include <iprt/time.h>
+#include <iprt/mem.h>
 
 #include <VBox/sup.h>
@@ -52 +53 @@
 #include "VBoxNetLib.h"
 #include "VBoxNetBaseService.h"
-#include "VBoxNetLib.h"
-#include "VBoxNetBaseService.h"
 
 #ifdef RT_OS_WINDOWS /* WinMain */
@@ -64 +63 @@
 *   Structures and Typedefs                                                    *
 *******************************************************************************/
+static RTGETOPTDEF g_aGetOptDef[] =
+{
+    { "--name",           'N',   RTGETOPT_REQ_STRING },
+    { "--network",        'n',   RTGETOPT_REQ_STRING },
+    { "--trunk-name",     't',   RTGETOPT_REQ_STRING },
+    { "--trunk-type",     'T',   RTGETOPT_REQ_STRING },
+    { "--mac-address",    'a',   RTGETOPT_REQ_MACADDR },
+    { "--ip-address",     'i',   RTGETOPT_REQ_IPV4ADDR },
+    { "--verbose",        'v',   RTGETOPT_REQ_NOTHING },
+};
 VBoxNetBaseService::VBoxNetBaseService()
 {
-    /* numbers from DrvIntNet */
-    m_cbSendBuf             =  36 * _1K;
-    m_cbRecvBuf             = 218 * _1K;
-    m_hIf                   = INTNET_HANDLE_INVALID;
-    m_pIfBuf                = NULL;
-
-    m_cVerbosity            = 0;
-    m_Name                  = "VBoxNetNAT";
-    m_Network               = "intnet";
 }
 VBoxNetBaseService::~VBoxNetBaseService()
@@ -99 +99 @@
     }
 }
+
+int VBoxNetBaseService::init()
+{
+    /* numbers from DrvIntNet */
+    m_cbSendBuf             =  36 * _1K;
+    m_cbRecvBuf             = 218 * _1K;
+    m_hIf                   = INTNET_HANDLE_INVALID;
+    m_pIfBuf                = NULL;
+
+    m_cVerbosity            = 0;
+    m_Name                  = "VBoxNetNAT";
+    m_Network               = "intnet";
+    for(unsigned int i = 0; i < RT_ELEMENTS(g_aGetOptDef); ++i)
+        m_vecOptionDefs.push_back(&g_aGetOptDef[i]);
+    return VINF_SUCCESS;
+}
 /**
  * Parse the arguments.
@@ -109 +125 @@
 int VBoxNetBaseService::parseArgs(int argc, char **argv)
 {
-    static const RTGETOPTDEF s_aOptionDefs[] =
-    {
-        { "--name",           'N',   RTGETOPT_REQ_STRING },
-        { "--network",        'n',   RTGETOPT_REQ_STRING },
-        { "--trunk-name",     't',   RTGETOPT_REQ_STRING },
-        { "--trunk-type",     'T',   RTGETOPT_REQ_STRING },
-        { "--mac-address",    'a',   RTGETOPT_REQ_MACADDR },
-        { "--ip-address",     'i',   RTGETOPT_REQ_IPV4ADDR },
-        { "--verbose",        'v',   RTGETOPT_REQ_NOTHING },
-    };
 
     RTGETOPTSTATE State;
-    int rc = RTGetOptInit(&State, argc, argv, &s_aOptionDefs[0], RT_ELEMENTS(s_aOptionDefs), 0, 0 /*fFlags*/);
+    PRTGETOPTDEF paOptionArray = getOptionsPtr();
+    int rc = RTGetOptInit(&State, argc, argv, paOptionArray, m_vecOptionDefs.size(), 0, 0 /*fFlags*/);
     AssertRCReturn(rc, 49);
     Log2(("BaseService: parseArgs enter\n"));
@@ -183 +190 @@
                          "Options:\n",
                          RTBldCfgVersion());
-                for (size_t i = 0; i < RT_ELEMENTS(s_aOptionDefs); i++)
-                    RTPrintf("    -%c, %s\n", s_aOptionDefs[i].iShort, s_aOptionDefs[i].pszLong);
+                for (unsigned int i = 0; i < m_vecOptionDefs.size(); i++)
+                    RTPrintf("    -%c, %s\n", m_vecOptionDefs[i]->iShort, m_vecOptionDefs[i]->pszLong);
                 usage(); /* to print Service Specific usage */
                 return 1;
 
             default:
-                rc = RTGetOptPrintError(rc, &Val);
-                RTPrintf("Use --help for more information.\n");
-                return rc;
+                int rc1 = parseOpt(rc, Val);
+                if (RT_FAILURE(rc1))
+                {
+                    rc = RTGetOptPrintError(rc, &Val);
+                    RTPrintf("Use --help for more information.\n");
+                    return rc;
+                }
         }
     }
 
+    RTMemFree(paOptionArray);
     return rc;
 }
@@ -343 +355 @@
 }
 
+PRTGETOPTDEF VBoxNetBaseService::getOptionsPtr()
+{
+    PRTGETOPTDEF pOptArray = NULL;
+    pOptArray = (PRTGETOPTDEF)RTMemAlloc(sizeof(RTGETOPTDEF) * m_vecOptionDefs.size());
+    if (!pOptArray)
+        return NULL;
+    for (unsigned int i = 0; i < m_vecOptionDefs.size(); ++i)
+    {
+        PRTGETOPTDEF pOpt = m_vecOptionDefs[i];
+        memcpy(&pOptArray[i], m_vecOptionDefs[i], sizeof(RTGETOPTDEF));
+    }
+    return pOptArray;
+}