Changeset 32656 in vbox for trunk/src/VBox/Runtime/r3/win

Timestamp:

Sep 21, 2010 8:47:18 AM (14 years ago)

Author:

vboxsync

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

66067

Message:

IPRT/process-win: Added dynamic loading of Userenv-APIs.

File:

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

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

@@ -89 +89 @@
 typedef DWORD FNGETMODULEBASENAME(HANDLE, HMODULE, LPTSTR, DWORD);
 typedef FNGETMODULEBASENAME *PFNGETMODULEBASENAME;
+
+typedef BOOL WINAPI FNCREATEENVIRONMENTBLOCK(LPVOID *, HANDLE, BOOL);
+typedef FNCREATEENVIRONMENTBLOCK *PFNCREATEENVIRONMENTBLOCK;
+
+typedef BOOL WINAPI FNPFNDESTROYENVIRONMENTBLOCK(LPVOID);
+typedef FNPFNDESTROYENVIRONMENTBLOCK *PFNPFNDESTROYENVIRONMENTBLOCK;
+
+typedef BOOL WINAPI FNLOADUSERPROFILEW(HANDLE, LPPROFILEINFOW);
+typedef FNLOADUSERPROFILEW *PFNLOADUSERPROFILEW;
+
+typedef BOOL WINAPI FNUNLOADUSERPROFILE(HANDLE, HANDLE);
+typedef FNUNLOADUSERPROFILE *PFNUNLOADUSERPROFILE;
 
 
@@ -590 +602 @@
 static int rtProcEnvironmentCreateFromToken(HANDLE hToken, RTENV hEnv, PRTUTF16 *ppwszBlock)
 {
-    int rc = VINF_SUCCESS;
-    LPVOID pEnvBlockProfile = NULL;
-    if (CreateEnvironmentBlock(&pEnvBlockProfile, hToken, FALSE /* Don't inherit from parent. */))
-    {
-        rc = rtProcEnvironmentCreateInternal(pEnvBlockProfile, hEnv, ppwszBlock);
-        DestroyEnvironmentBlock(pEnvBlockProfile);
-    }
-    else
-        rc = RTErrConvertFromWin32(GetLastError());
+    RTLDRMOD hUserenv;
+    int rc = RTLdrLoad("Userenv.dll", &hUserenv);
+    if (RT_SUCCESS(rc))
+    {
+        PFNCREATEENVIRONMENTBLOCK pfnCreateEnvironmentBlock;
+        rc = RTLdrGetSymbol(hUserenv, "CreateEnvironmentBlock", (void**)&pfnCreateEnvironmentBlock);
+        if (RT_SUCCESS(rc))
+        {
+            PFNPFNDESTROYENVIRONMENTBLOCK pfnDestroyEnvironmentBlock;
+            rc = RTLdrGetSymbol(hUserenv, "DestroyEnvironmentBlock", (void**)&pfnDestroyEnvironmentBlock);
+            if (RT_SUCCESS(rc))
+            {
+                LPVOID pEnvBlockProfile = NULL;
+                if (pfnCreateEnvironmentBlock(&pEnvBlockProfile, hToken, FALSE /* Don't inherit from parent. */))
+                {
+                    rc = rtProcEnvironmentCreateInternal(pEnvBlockProfile, hEnv, ppwszBlock);
+                    pfnDestroyEnvironmentBlock(pEnvBlockProfile);
+                }
+                else
+                    rc = RTErrConvertFromWin32(GetLastError());
+            }
+        }
+        RTLdrClose(hUserenv);
+    }
     return rc;
 }
@@ -807 +834 @@
             phToken = fFound ? &hTokenUserDesktop : &hTokenLogon;
 
-            PROFILEINFOW profileInfo;
-            ZeroMemory(&profileInfo, sizeof(profileInfo));
-            profileInfo.dwSize = sizeof(profileInfo);
-            profileInfo.lpUserName = pwszUser;
-            profileInfo.dwFlags = PI_NOUI; /* Prevents the display of profile error messages. */
-            if (LoadUserProfileW(*phToken, &profileInfo))
-            {
-                PRTUTF16 pwszzBlock;
-                rc = rtProcEnvironmentCreateFromToken(*phToken, hEnv, &pwszzBlock);
+            RTLDRMOD hUserenv;
+            int rc = RTLdrLoad("Userenv.dll", &hUserenv);
+            if (RT_SUCCESS(rc))
+            {
+                PFNLOADUSERPROFILEW pfnLoadUserProfileW;
+                rc = RTLdrGetSymbol(hUserenv, "LoadUserProfileW", (void**)&pfnLoadUserProfileW);
                 if (RT_SUCCESS(rc))
                 {
-                    /*
-                     * Useful KB articles:
-                     *      http://support.microsoft.com/kb/165194/
-                     *      http://support.microsoft.com/kb/184802/
-                     *      http://support.microsoft.com/kb/327618/
-                     */
-                    fRc = CreateProcessAsUserW(*phToken,
-                                               pwszExec,
-                                               pwszCmdLine,
-                                               NULL,         /* pProcessAttributes */
-                                               NULL,         /* pThreadAttributes */
-                                               TRUE,         /* fInheritHandles */
-                                               dwCreationFlags,
-                                               pwszzBlock,
-                                               NULL,         /* pCurrentDirectory */
-                                               pStartupInfo,
-                                               pProcInfo);
-                    if (fRc)
-                        dwErr = NO_ERROR;
-                    else
-                        dwErr = GetLastError(); /* CreateProcessAsUserW() failed. */
-                    rtProcEnvironmentDestroy(pwszzBlock);
+                    PFNUNLOADUSERPROFILE pfnUnloadUserProfile;
+                    rc = RTLdrGetSymbol(hUserenv, "UnloadUserProfile", (void**)&pfnUnloadUserProfile);
+                    if (RT_SUCCESS(rc))
+                    {
+                        PROFILEINFOW profileInfo;
+                        ZeroMemory(&profileInfo, sizeof(profileInfo));
+                        profileInfo.dwSize = sizeof(profileInfo);
+                        profileInfo.lpUserName = pwszUser;
+                        profileInfo.dwFlags = PI_NOUI; /* Prevents the display of profile error messages. */
+
+                        if (pfnLoadUserProfileW(*phToken, &profileInfo))
+                        {
+                            PRTUTF16 pwszzBlock;
+                            rc = rtProcEnvironmentCreateFromToken(*phToken, hEnv, &pwszzBlock);
+                            if (RT_SUCCESS(rc))
+                            {
+                                /*
+                                 * Useful KB articles:
+                                 *      http://support.microsoft.com/kb/165194/
+                                 *      http://support.microsoft.com/kb/184802/
+                                 *      http://support.microsoft.com/kb/327618/
+                                 */
+                                fRc = CreateProcessAsUserW(*phToken,
+                                                           pwszExec,
+                                                           pwszCmdLine,
+                                                           NULL,         /* pProcessAttributes */
+                                                           NULL,         /* pThreadAttributes */
+                                                           TRUE,         /* fInheritHandles */
+                                                           dwCreationFlags,
+                                                           pwszzBlock,
+                                                           NULL,         /* pCurrentDirectory */
+                                                           pStartupInfo,
+                                                           pProcInfo);
+                                if (fRc)
+                                    dwErr = NO_ERROR;
+                                else
+                                    dwErr = GetLastError(); /* CreateProcessAsUserW() failed. */
+                                rtProcEnvironmentDestroy(pwszzBlock);
+                            }
+                            else
+                                dwErr = rc;
+                            pfnUnloadUserProfile(*phToken, profileInfo.hProfile);
+                        }
+                        else
+                            dwErr = GetLastError(); /* LoadUserProfileW() failed. */
+                    }
                 }
-                else
-                    dwErr = rc;
-                UnloadUserProfile(*phToken, profileInfo.hProfile);
-            }
-            else
-                dwErr = GetLastError(); /* LoadUserProfileW() failed. */
+                RTLdrClose(hUserenv);
+            }
             if (hTokenUserDesktop != INVALID_HANDLE_VALUE)
                 CloseHandle(hTokenUserDesktop);