Changeset 99109 in vbox for trunk/src/VBox/Runtime/r3/win/process-win.cpp

Timestamp:

Mar 22, 2023 10:13:00 AM (23 months ago)

Author:

vboxsync

Message:

IPRT/process: Added RTPROC_FLAGS_CWD to RTProcCreateEx(), to be able to set the current working directory for created processes. bugref:8053

File:

• trunk/src/VBox/Runtime/r3/win/process-win.cpp (modified) (13 diffs)

trunk/src/VBox/Runtime/r3/win/process-win.cpp

@@ -1613 +1613 @@
                                   STARTUPINFOW *pStartupInfo, PROCESS_INFORMATION *pProcInfo,
                                   uint32_t fFlags, const char *pszExec, uint32_t idDesiredSession,
-                                  HANDLE hUserToken)
+                                  HANDLE hUserToken, PRTUTF16 pwszCwd)
 {
     /*
@@ -1804 +1804 @@
                                                                      *        on XP and up. */
                                                                     pwszzBlock,   /* lpEnvironment */
-                                                                    NULL,         /* pCurrentDirectory */
+                                                                    pwszCwd,       /* pCurrentDirectory */
                                                                     pStartupInfo,
                                                                     pProcInfo);
@@ -1942 +1942 @@
                                   RTENV hEnv, DWORD dwCreationFlags,
                                   STARTUPINFOW *pStartupInfo, PROCESS_INFORMATION *pProcInfo,
-                                  uint32_t fFlags, const char *pszExec)
+                                  uint32_t fFlags, const char *pszExec, PRTUTF16 pwszCwd)
 {
     /* The CreateProcessWithLogonW API was introduced with W2K and later.  It uses a service
@@ -2042 +2042 @@
                                                 dwCreationFlags | (fCreatedSuspended ? CREATE_SUSPENDED : 0),
                                                 pwszzBlock,
-                                                NULL,                       /* pCurrentDirectory */
+                                                pwszCwd,                     /* pCurrentDirectory */
                                                 pStartupInfo,
                                                 pProcInfo);
@@ -2093 +2093 @@
                                  STARTUPINFOW *pStartupInfo, PROCESS_INFORMATION *pProcInfo,
                                  uint32_t fFlags, const char *pszExec, uint32_t idDesiredSession,
-                                 HANDLE hUserToken)
+                                 HANDLE hUserToken, PRTUTF16 pwszCwd)
 {
     /*
@@ -2105 +2105 @@
         AssertPtr(pwszUser);
         int rc = rtProcWinCreateAsUser1(pwszUser, pwszPassword, ppwszExec, pwszCmdLine,
-                                        hEnv, dwCreationFlags, pStartupInfo, pProcInfo, fFlags, pszExec);
+                                        hEnv, dwCreationFlags, pStartupInfo, pProcInfo, fFlags, pszExec, pwszCwd);
         if (RT_SUCCESS(rc))
             return rc;
     }
     return rtProcWinCreateAsUser2(pwszUser, pwszPassword, ppwszExec, pwszCmdLine, hEnv, dwCreationFlags,
-                                  pStartupInfo, pProcInfo, fFlags, pszExec, idDesiredSession, hUserToken);
+                                  pStartupInfo, pProcInfo, fFlags, pszExec, idDesiredSession, hUserToken, pwszCwd);
 }
 
@@ -2262 +2262 @@
                                const char *pszPassword, void *pvExtraData, PRTPROCESS phProcess)
 {
+    int rc;
+
     /*
      * Input validation
@@ -2281 +2283 @@
         ==           (RTPROC_FLAGS_DESIRED_SESSION_ID | RTPROC_FLAGS_SERVICE))
     {
+        AssertReturn(!(fFlags & RTPROC_FLAGS_CWD), VERR_INVALID_PARAMETER);
         AssertPtrReturn(pvExtraData, VERR_INVALID_POINTER);
         idDesiredSession = *(uint32_t *)pvExtraData;
@@ -2294 +2297 @@
      * Initialize the globals.
      */
-    int rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL);
+    rc = RTOnce(&g_rtProcWinInitOnce, rtProcWinInitOnce, NULL);
     AssertRCReturn(rc, rc);
     if (   pszAsUser
@@ -2433 +2436 @@
                                        !(fFlags & RTPROC_FLAGS_UNQUOTED_ARGS)
                                        ? RTGETOPTARGV_CNV_QUOTE_MS_CRT : RTGETOPTARGV_CNV_UNQUOTED);
+    PRTUTF16 pwszExec = NULL;
     if (RT_SUCCESS(rc))
-    {
-        PRTUTF16 pwszExec;
         rc = RTPathWinFromUtf8(&pwszExec, pszExec, 0 /*fFlags*/);
-        if (RT_SUCCESS(rc))
-        {
+    PRTUTF16 pwszCwd = NULL;
+    if (RT_SUCCESS(rc) && (fFlags & RTPROC_FLAGS_CWD))
+        rc = RTPathWinFromUtf8(&pwszCwd, (const char *)pvExtraData, 0);
+    if (RT_SUCCESS(rc))
+    {
             /*
              * Get going...
@@ -2470 +2475 @@
                                        dwCreationFlags,
                                        pwszzBlock,
-                                       NULL,         /* pCurrentDirectory */
+                                       pwszCwd,       /* pCurrentDirectory */
                                        &StartupInfo,
                                        &ProcInfo))
@@ -2496 +2501 @@
                         rc = rtProcWinCreateAsUser(pwszUser, pwszPassword, &pwszExec, pwszCmdLine, hEnv, dwCreationFlags,
                                                    &StartupInfo, &ProcInfo, fFlags, pszExec, idDesiredSession,
-                                                   hUserToken);
+                                                   hUserToken, pwszCwd);
 
                         if (pwszPassword && *pwszPassword)
@@ -2524 +2529 @@
                 rc = VINF_SUCCESS;
             }
-            RTPathWinFree(pwszExec);
-        }
-        RTUtf16Free(pwszCmdLine);
-    }
+    }
+    RTPathWinFree(pwszExec);
+    RTPathWinFree(pwszCwd);
+    RTUtf16Free(pwszCmdLine);
 
     if (g_pfnSetHandleInformation)