RunAsTrustedInstaller/RunAsTI.cpp at main · mbcdev/RunAsTrustedInstaller · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include <iostream>
#include <Shlwapi.h>

#include "Utils.h"

BOOL ExecuteCommand(std::wstring& appName, std::wstring& cmdArgs);

int main()
{
	int nArgs;
	int ret = 1;
	LPWSTR* lpArgs = CommandLineToArgvW(GetCommandLineW(), &nArgs);

	if (lpArgs != NULL)
	{
		if (nArgs == 2 && !_wcsicmp(lpArgs[1], L"/install")) {
			ShellExecute(NULL, L"open", L"regsvr32.exe", L"/i RunAsTIShellExtension.dll", NULL, SW_SHOWNORMAL);
		}
		else if (nArgs == 2 && !_wcsicmp(lpArgs[1], L"/uninstall")) {
			ShellExecute(NULL, L"open", L"regsvr32.exe", L"/u RunAsTIShellExtension.dll", NULL, SW_SHOWNORMAL);
		}
		else if (nArgs > 1)
		{
			std::wstring cmdApp;
			std::wstring cmdArgs;
			LPWSTR* lpCommandLine = (lpArgs + 1);
			cmdApp = *lpCommandLine;
			lpCommandLine++;
			while (*lpCommandLine != NULL) {
				if (!cmdArgs.empty()) cmdArgs += L" ";
				cmdArgs += *lpCommandLine;
				lpCommandLine++;
			}

			if (nArgs > 2) {
				std::wcout << "Executing '" << cmdApp.c_str() << "' with args '" << cmdArgs.c_str() << "'..." << std::endl;
			}
			else {
				std::wcout << "Executing '" << cmdApp.c_str() << "'..." << std::endl;
			}

			if (ExecuteCommand(cmdApp, cmdArgs))
			{
				std::wcout << "SUCCESS!" << std::endl;
				ret = 0;
			}
			else {
				std::wcout << "FAILED!" << std::endl;
			}
		}
		else {
			std::wstring exeName = *lpArgs;
			PathStripPath(&exeName[0]);
			std::wcout << "Usage: " << exeName.c_str() << " <cmdline>    Executes <cmdline> with TrustedInstaller privileges." << std::endl;
			std::wcout << "       " << exeName.c_str() << " /install     Installs Shell Extension Handler." << std::endl;
			std::wcout << "       " << exeName.c_str() << " /uninstall   Uninstalls Shell Extension Handler." << std::endl;
		}
		LocalFree(lpArgs);
	}

	return ret;
}

BOOL ExecuteCommand(std::wstring& appName, std::wstring& cmdArgs)
{
	BOOL success = FALSE;

	if (!Utils::ImpersonateAsSystem())
	{
		return FALSE;
	}

	DWORD dwPID = Utils::StartTIServiceAndGetPID();
	if (dwPID == 0)
	{
		return FALSE;
	}

	HANDLE hToken = Utils::CreateAccessTokenFromPID(dwPID);
	if (hToken != INVALID_HANDLE_VALUE)
	{
		STARTUPINFO si;
		PROCESS_INFORMATION pi;

		GetStartupInfo(&si);
		success = CreateProcessWithTokenW(hToken, LOGON_WITH_PROFILE, (LPWSTR)appName.c_str(), (LPWSTR)cmdArgs.c_str(), CREATE_UNICODE_ENVIRONMENT, NULL, NULL, &si, &pi);
		if (!success) {
			std::cout << "CreateProcessWithTokenW: " << std::system_category().message(GetLastError()) << std::endl;
		}
	}
	CloseHandle(hToken);

	return success;
}