Changeset 40993 in vbox for trunk/src/VBox/HostDrivers/VBoxNetFlt/win/tools

Timestamp:

Apr 19, 2012 2:25:38 PM (13 years ago)

Author:

vboxsync

Message:

NetCfg/win: disable/enable/update HostOnly

File:

• trunk/src/VBox/HostDrivers/VBoxNetFlt/win/tools/VBoxNetAdpInstall.cpp (modified) (2 diffs)

trunk/src/VBox/HostDrivers/VBoxNetFlt/win/tools/VBoxNetAdpInstall.cpp

@@ -17 +17 @@
 
 #include <VBox/VBoxNetCfg-win.h>
+#include <VBox/VBoxDrvCfg-win.h>
 #include <stdio.h>
+
+#include <devguid.h>
 
 #define VBOX_NETADP_INF L".\\VBoxNetAdp.inf"
@@ -94 +97 @@
 }
 
+static int VBoxNetAdpUninstall()
+{
+    int r = 1;
+    VBoxNetCfgWinSetLogging(winNetCfgLogger);
+
+    printf("uninstalling all Host-Only interfaces..\n");
+
+    HRESULT hr = CoInitialize(NULL);
+    if(hr == S_OK)
+    {
+        hr = VBoxNetCfgWinRemoveAllNetDevicesOfId(L"sun_VBoxNetAdp");
+        if(hr == S_OK)
+        {
+            hr = VBoxDrvCfgInfUninstallAllSetupDi(&GUID_DEVCLASS_NET, L"Net", L"sun_VBoxNetAdp", 0/* could be SUOI_FORCEDELETE */);
+            if(hr == S_OK)
+            {
+                printf("uninstalled successfully\n");
+            }
+            else
+            {
+                printf("uninstalled successfully, but failed to remove infs\n");
+            }
+            r = 0;
+        }
+        else
+        {
+            printf("uninstall failed, hr = 0x%x\n", hr);
+        }
+
+        CoUninitialize();
+    }
+    else
+    {
+        wprintf(L"Error initializing COM (0x%x)\n", hr);
+    }
+
+    VBoxNetCfgWinSetLogging(NULL);
+
+    return r;
+}
+
+static int VBoxNetAdpUpdate()
+{
+    int r = 1;
+    VBoxNetCfgWinSetLogging(winNetCfgLogger);
+
+    printf("uninstalling all Host-Only interfaces..\n");
+
+    HRESULT hr = CoInitialize(NULL);
+    if(hr == S_OK)
+    {
+        BOOL fRebootRequired = FALSE;
+        hr = VBoxNetCfgWinUpdateHostOnlyNetworkInterface(VBOX_NETADP_INF, &fRebootRequired);
+        if(hr == S_OK)
+        {
+            if (fRebootRequired)
+                printf("!!REBOOT REQUIRED!!\n");
+            printf("updated successfully\n");
+            r = 0;
+        }
+        else
+        {
+            printf("update failed, hr = 0x%x\n", hr);
+        }
+
+        CoUninitialize();
+    }
+    else
+    {
+        wprintf(L"Error initializing COM (0x%x)\n", hr);
+    }
+
+    VBoxNetCfgWinSetLogging(NULL);
+
+    return r;
+}
+
+static int VBoxNetAdpDisable()
+{
+    int r = 1;
+    VBoxNetCfgWinSetLogging(winNetCfgLogger);
+
+    printf("disabling all Host-Only interfaces..\n");
+
+    HRESULT hr = CoInitialize(NULL);
+    if(hr == S_OK)
+    {
+        hr = VBoxNetCfgWinPropChangeAllNetDevicesOfId(L"sun_VBoxNetAdp", VBOXNECTFGWINPROPCHANGE_TYPE_DISABLE);
+        if(hr == S_OK)
+        {
+            printf("disable succeeded!\n");
+            r = 0;
+        }
+        else
+        {
+            printf("disable failed, hr = 0x%x\n", hr);
+        }
+
+        CoUninitialize();
+    }
+    else
+    {
+        wprintf(L"Error initializing COM (0x%x)\n", hr);
+    }
+
+    VBoxNetCfgWinSetLogging(NULL);
+
+    return r;
+}
+
+static int VBoxNetAdpEnable()
+{
+    int r = 1;
+    VBoxNetCfgWinSetLogging(winNetCfgLogger);
+
+    printf("enabling all Host-Only interfaces..\n");
+
+    HRESULT hr = CoInitialize(NULL);
+    if(hr == S_OK)
+    {
+        hr = VBoxNetCfgWinPropChangeAllNetDevicesOfId(L"sun_VBoxNetAdp", VBOXNECTFGWINPROPCHANGE_TYPE_ENABLE);
+        if(hr == S_OK)
+        {
+            printf("disable succeeded!\n");
+            r = 0;
+        }
+        else
+        {
+            printf("disable failed, hr = 0x%x\n", hr);
+        }
+
+        CoUninitialize();
+    }
+    else
+    {
+        wprintf(L"Error initializing COM (0x%x)\n", hr);
+    }
+
+    VBoxNetCfgWinSetLogging(NULL);
+
+    return r;
+}
+
+static void printUsage()
+{
+    printf("Host-Only network adapter configuration tool\n"
+            "  Usage: VBoxNetAdpInstall [cmd]\n"
+            "    cmd can be one of the following values:\n"
+            "       i  - install a new host-only interface\n"
+            "       u  - uninstall all host-only interfaces\n"
+            "       a  - update the host-only driver\n"
+            "       d  - disable all host-only interfaces\n"
+            "       e  - enable all host-only interfaces\n"
+            "       h  - print this message\n");
+}
+
 int __cdecl main(int argc, char **argv)
 {
-    return VBoxNetAdpInstall();
-}
+    if (argc < 2)
+        return VBoxNetAdpInstall();
+    if (argc > 2)
+    {
+        printUsage();
+        return 1;
+    }
+
+    __debugbreak();
+
+    if (!strcmp(argv[1], "i"))
+        return VBoxNetAdpInstall();
+    if (!strcmp(argv[1], "u"))
+        return VBoxNetAdpUninstall();
+    if (!strcmp(argv[1], "a"))
+        return VBoxNetAdpUpdate();
+    if (!strcmp(argv[1], "d"))
+        return VBoxNetAdpDisable();
+    if (!strcmp(argv[1], "e"))
+        return VBoxNetAdpEnable();
+
+    printUsage();
+    return !strcmp(argv[1], "h");
+}