Hello,阳光柠檬! 养成记录笔记的好习惯

c++ 提升应用程序的权限

2018-08-15
C++

BOOL IsRunAsAdmin()
{
	BOOL fIsRunAsAdmin = FALSE;
	DWORD dwError = ERROR_SUCCESS;
	PSID pAdministratorsGroup = NULL;

	// Allocate and initialize a SID of the administrators group.  
	SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
	if (!AllocateAndInitializeSid(
		&NtAuthority,
		2,
		SECURITY_BUILTIN_DOMAIN_RID,
		DOMAIN_ALIAS_RID_ADMINS,
		0, 0, 0, 0, 0, 0,
		&pAdministratorsGroup))
	{
		dwError = GetLastError();
		goto Cleanup;
	}

	// Determine whether the SID of administrators group is enabled in   
	// the primary access token of the process.  
	if (!CheckTokenMembership(NULL, pAdministratorsGroup, &fIsRunAsAdmin)) 
	{
		dwError = GetLastError();
		goto Cleanup;
	}

Cleanup:
	// Centralized cleanup for all allocated resources.  
	if (pAdministratorsGroup) 
	{
		FreeSid(pAdministratorsGroup);
		pAdministratorsGroup = NULL;
	}

	// Throw the error if something failed in the function.  
	if (ERROR_SUCCESS != dwError)
	{
		throw dwError;
	}

	return fIsRunAsAdmin;
}
BOOL ElevateCurrentProcess(LPCTSTR sCmdLine)
{
	USES_CONVERSION;
	TCHAR szPath[MAX_PATH] = { 0 };
	if (::GetModuleFileName(NULL, szPath, MAX_PATH)) 
	{
		// Launch itself as administrator.  
		SHELLEXECUTEINFO sei = { sizeof(SHELLEXECUTEINFO) };
		sei.lpVerb = L"runas";
		sei.lpFile = szPath;
		sei.lpParameters = sCmdLine;
		//  sei.hwnd = hWnd;  
		sei.nShow = SW_SHOWNORMAL;

		if (!ShellExecuteEx(&sei))
		{
			DWORD dwStatus = GetLastError();
			if (dwStatus == ERROR_CANCELLED) 
			{
				return FALSE;
			}
			else if (dwStatus == ERROR_FILE_NOT_FOUND)
			{
				return FALSE;
			}
			return FALSE;
		}
		return TRUE;
	}
	return FALSE;
}

使用

if (!IsRunAsAdmin())
{
    QString cmdline = " /action=add";
	ElevateCurrentProcess(cmdline.utf16());
	emit exitApp();
	return;
}

Similar Posts

Comments