Method EnablePrivilege

Bypass_COM_PPL/trustinstaller_utils.cpp:4–28

#include "trustinstaller_utils.h"

void Elevate::EnablePrivilege(std::wstring privilegeName)
{
    HANDLE hToken;
    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES, &hToken))
        throw std::runtime_error("OpenProcessToken failed: " + std::to_string(GetLastError()));

    LUID luid;
    if (!LookupPrivilegeValueW(nullptr, privilegeName.c_str(), &luid))
    {
        CloseHandle(hToken);
        throw std::runtime_error("LookupPrivilegeValue failed: " + std::to_string(GetLastError()));
    }

    TOKEN_PRIVILEGES tp;
    tp.PrivilegeCount = 1;
    tp.Privileges[0].Luid = luid;
    tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
    if (!AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), nullptr, nullptr))
    {
        CloseHandle(hToken);
        throw std::runtime_error("AdjustTokenPrivilege failed: " + std::to_string(GetLastError()));
    }

    CloseHandle(hToken);
}

DWORD Elevate::GetProcessIdByName(std::wstring  processName)
{