Changeset 107990 in vbox for trunk/src/VBox/Installer

Timestamp:

Jan 30, 2025 12:29:09 PM (3 months ago)

Author:

vboxsync

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

167253

Message:

Host installer/win: Added ServiceControl() to the VBoxInstallHelper.dll. bugref:10762

Location:

trunk/src/VBox/Installer/win/InstallHelper

Files:

• VBoxInstallHelper.cpp (modified) (1 diff)
• VBoxInstallHelper.def (modified) (1 diff)

trunk/src/VBox/Installer/win/InstallHelper/VBoxInstallHelper.cpp

@@ -1617 +1617 @@
 }
 
+UINT __stdcall ServiceControl(MSIHANDLE hModule)
+{
+    char *pszSvcCtlName;
+    int rc = VBoxMsiQueryPropUtf8(hModule, "VBoxSvcCtlName", &pszSvcCtlName);
+    if (RT_SUCCESS(rc))
+    {
+        char *pszSvcCtlFn;
+        rc = VBoxMsiQueryPropUtf8(hModule, "VBoxSvcCtlFn", &pszSvcCtlFn);
+        if (RT_SUCCESS(rc))
+        {
+            VBOXWINDRVSVCFN enmFn = VBOXWINDRVSVCFN_INVALID; /* Shut up MSVC. */
+            if (!RTStrICmp(pszSvcCtlFn, "start"))
+                enmFn = VBOXWINDRVSVCFN_START;
+            else if (!RTStrICmp(pszSvcCtlFn, "stop"))
+                enmFn = VBOXWINDRVSVCFN_STOP;
+            else if (!RTStrICmp(pszSvcCtlFn, "restart"))
+                enmFn = VBOXWINDRVSVCFN_RESTART;
+            else
+                rc = VERR_INVALID_PARAMETER;
+
+            if (RT_SUCCESS(rc))
+            {
+                RTMSINTERVAL msTimeout = 0; /* Don't wait by default. */
+                rc = VBoxMsiQueryPropInt32(hModule, "VBoxSvcCtlWaitMs", (DWORD *)&msTimeout);
+                if (   RT_SUCCESS(rc)
+                    || rc == VERR_NOT_FOUND) /* VBoxSvcCtlWaitMs is optional. */
+                {
+                    VBOXWINDRVINST hWinDrvInst;
+                    rc = VBoxWinDrvInstCreateEx(&hWinDrvInst, 1 /* uVerbostiy */,
+                                                &vboxWinDrvInstLogCallback, &hModule /* pvUser */);
+                    if (RT_SUCCESS(rc))
+                    {
+                        rc = VBoxWinDrvInstControlServiceEx(hWinDrvInst, pszSvcCtlName, enmFn,
+                                                            msTimeout ? VBOXWINDRVSVCFN_F_WAIT : VBOXWINDRVSVCFN_F_NONE,
+                                                            msTimeout);
+                        VBoxWinDrvInstDestroy(hWinDrvInst);
+                    }
+                }
+            }
+
+            RTStrFree(pszSvcCtlFn);
+        }
+
+        RTStrFree(pszSvcCtlName);
+    }
+
+    logStringF(hModule, "ServiceControl: Handling done (rc=%Rrc)", rc);
+    return RT_SUCCESS(rc) ? ERROR_SUCCESS : ERROR_INVALID_SERVICE_CONTROL;
+}
+
 #if defined(VBOX_WITH_NETFLT) || defined(VBOX_WITH_NETADP)
 

trunk/src/VBox/Installer/win/InstallHelper/VBoxInstallHelper.def

@@ -48 +48 @@
     DriverInstall
     DriverUninstall
+    ServiceControl
     CreateHostOnlyInterface
     StopHostOnlyInterfaces