Function RestartAsAdministrator

PresentMon/Privilege.cpp:96–155

}

int RestartAsAdministrator(
    int argc,
    wchar_t** argv)
{
    // Get the exe path
    wchar_t exe_path[MAX_PATH] = {};
    GetModuleFileNameW(NULL, exe_path, _countof(exe_path));

    // Combine arguments into single string and remove --restart_as_admin to
    // prevent an endless loop if the escalation fails.
    std::wstring args;
    for (int i = 1; i < argc; ++i) {
        if (IsRestartAsAdminArg(argv[i])) continue;

        auto addQuotes = argv[i][0] != L'\"' && wcschr(argv[i], L' ') != nullptr;
        if (addQuotes) {
            args += L'\"';
        }

        args += argv[i];

        if (addQuotes) {
            args += L'\"';
        }

        args += L' ';
    }

    // Re-run the process with the runas verb
    DWORD code = 2;

    SHELLEXECUTEINFO info = {};
    info.cbSize       = sizeof(info);
    info.fMask        = SEE_MASK_NOCLOSEPROCESS; // return info.hProcess for explicit wait
    info.lpVerb       = L"runas";
    info.lpFile       = exe_path;
    info.lpParameters = args.c_str();
    info.nShow        = SW_SHOWDEFAULT;
    auto ok = ShellExecuteEx(&info);
    if (ok) {
        WaitForSingleObject(info.hProcess, INFINITE);
        GetExitCodeProcess(info.hProcess, &code);
        CloseHandle(info.hProcess);
    } else {
        PrintError(L"error: failed to elevate privilege: ");
        int e = GetLastError();
        switch (e) {
        case ERROR_FILE_NOT_FOUND:    PrintError(L"file not found.\n"); break;
        case ERROR_PATH_NOT_FOUND:    PrintError(L"path not found.\n"); break;
        case ERROR_DLL_NOT_FOUND:     PrintError(L"dll not found.\n"); break;
        case ERROR_ACCESS_DENIED:     PrintError(L"access denied.\n"); break;
        case ERROR_CANCELLED:         PrintError(L"cancelled.\n"); break;
        case ERROR_NOT_ENOUGH_MEMORY: PrintError(L"out of memory.\n"); break;
        case ERROR_SHARING_VIOLATION: PrintError(L"sharing violation.\n"); break;
        default:                      PrintError(L"error code %u.\n", e); break;
        }
    }