From 7f642e51670bc38a4ef782a363936850bc2b0ba9 Mon Sep 17 00:00:00 2001 From: Rob Mensching Date: Thu, 22 Apr 2021 06:38:23 -0700 Subject: Move dutil into libs/dutil --- src/libs/dutil/WixToolset.DUtil/proc3utl.cpp | 129 +++++++++++++++++++++++++++ 1 file changed, 129 insertions(+) create mode 100644 src/libs/dutil/WixToolset.DUtil/proc3utl.cpp (limited to 'src/libs/dutil/WixToolset.DUtil/proc3utl.cpp') diff --git a/src/libs/dutil/WixToolset.DUtil/proc3utl.cpp b/src/libs/dutil/WixToolset.DUtil/proc3utl.cpp new file mode 100644 index 00000000..6d3cbc67 --- /dev/null +++ b/src/libs/dutil/WixToolset.DUtil/proc3utl.cpp @@ -0,0 +1,129 @@ +// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information. + +#include "precomp.h" + + +// Exit macros +#define ProcExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_PROCUTIL, x, s, __VA_ARGS__) +#define ProcExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_PROCUTIL, p, x, e, s, __VA_ARGS__) +#define ProcExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_PROCUTIL, p, x, s, __VA_ARGS__) +#define ProcExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_PROCUTIL, p, x, e, s, __VA_ARGS__) +#define ProcExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_PROCUTIL, p, x, s, __VA_ARGS__) +#define ProcExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_PROCUTIL, e, x, s, __VA_ARGS__) +#define ProcExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_PROCUTIL, g, x, s, __VA_ARGS__) + +static HRESULT GetActiveSessionUserToken( + __out HANDLE *phToken + ); + + +/******************************************************************** + ProcExecuteAsInteractiveUser() - runs process as currently logged in + user. +*******************************************************************/ +extern "C" HRESULT DAPI ProcExecuteAsInteractiveUser( + __in_z LPCWSTR wzExecutablePath, + __in_z LPCWSTR wzCommandLine, + __out HANDLE *phProcess + ) +{ + HRESULT hr = S_OK; + HANDLE hToken = NULL; + LPVOID pEnvironment = NULL; + LPWSTR sczFullCommandLine = NULL; + STARTUPINFOW si = { }; + PROCESS_INFORMATION pi = { }; + + hr = GetActiveSessionUserToken(&hToken); + ProcExitOnFailure(hr, "Failed to get active session user token."); + + if (!::CreateEnvironmentBlock(&pEnvironment, hToken, FALSE)) + { + ProcExitWithLastError(hr, "Failed to create environment block for UI process."); + } + + hr = StrAllocFormatted(&sczFullCommandLine, L"\"%ls\" %ls", wzExecutablePath, wzCommandLine); + ProcExitOnFailure(hr, "Failed to allocate full command-line."); + + si.cb = sizeof(si); + if (!::CreateProcessAsUserW(hToken, wzExecutablePath, sczFullCommandLine, NULL, NULL, FALSE, CREATE_UNICODE_ENVIRONMENT, pEnvironment, NULL, &si, &pi)) + { + ProcExitWithLastError(hr, "Failed to create UI process: %ls", sczFullCommandLine); + } + + *phProcess = pi.hProcess; + pi.hProcess = NULL; + +LExit: + ReleaseHandle(pi.hThread); + ReleaseHandle(pi.hProcess); + ReleaseStr(sczFullCommandLine); + + if (pEnvironment) + { + ::DestroyEnvironmentBlock(pEnvironment); + } + + ReleaseHandle(hToken); + + return hr; +} + + +static HRESULT GetActiveSessionUserToken( + __out HANDLE *phToken + ) +{ + HRESULT hr = S_OK; + PWTS_SESSION_INFO pSessionInfo = NULL; + DWORD cSessions = 0; + DWORD dwSessionId = 0; + BOOL fSessionFound = FALSE; + HANDLE hToken = NULL; + + // Loop through the sessions looking for the active one. + if (!::WTSEnumerateSessions(WTS_CURRENT_SERVER_HANDLE, 0, 1, &pSessionInfo, &cSessions)) + { + ProcExitWithLastError(hr, "Failed to enumerate sessions."); + } + + for (DWORD i = 0; i < cSessions; ++i) + { + if (WTSActive == pSessionInfo[i].State) + { + dwSessionId = pSessionInfo[i].SessionId; + fSessionFound = TRUE; + + break; + } + } + + if (!fSessionFound) + { + ExitFunction1(hr = E_NOTFOUND); + } + + // Get the user token from the active session. + if (!::WTSQueryUserToken(dwSessionId, &hToken)) + { + ProcExitWithLastError(hr, "Failed to get active session user token."); + } + + *phToken = hToken; + hToken = NULL; + +LExit: + ReleaseHandle(hToken); + + if (pSessionInfo) + { + ::WTSFreeMemory(pSessionInfo); + } + + return hr; +} -- cgit v1.2.3-55-g6feb