3715b8c5f2
* open settings from runner * Update src/runner/main.cpp * Update src/runner/main.cpp Co-authored-by: Enrico Giordani <enricogior@users.noreply.github.com>
57 lines
1.4 KiB
C++
57 lines
1.4 KiB
C++
#include "pch.h"
|
|
#include "restart_elevated.h"
|
|
|
|
#include <common/utils/elevation.h>
|
|
|
|
enum State
|
|
{
|
|
None,
|
|
RestartAsElevated,
|
|
RestartAsElevatedOpenSettings,
|
|
RestartAsNonElevated
|
|
};
|
|
static State state = None;
|
|
|
|
void schedule_restart_as_elevated(bool openSettings)
|
|
{
|
|
state = openSettings ? RestartAsElevatedOpenSettings : RestartAsElevated;
|
|
}
|
|
|
|
|
|
void schedule_restart_as_non_elevated()
|
|
{
|
|
state = RestartAsNonElevated;
|
|
}
|
|
|
|
bool is_restart_scheduled()
|
|
{
|
|
return state != None;
|
|
}
|
|
|
|
bool restart_if_scheduled()
|
|
{
|
|
// Make sure we have enough room, even for the long (\\?\) paths
|
|
constexpr DWORD exe_path_size = 0xFFFF;
|
|
auto exe_path = std::make_unique<wchar_t[]>(exe_path_size);
|
|
GetModuleFileNameW(nullptr, exe_path.get(), exe_path_size);
|
|
switch (state)
|
|
{
|
|
case RestartAsElevated:
|
|
return run_elevated(exe_path.get(), {});
|
|
case RestartAsElevatedOpenSettings:
|
|
return run_elevated(exe_path.get(), L"--open-settings");
|
|
case RestartAsNonElevated:
|
|
return run_non_elevated(exe_path.get(), L"--dont-elevate", NULL);
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool restart_same_elevation()
|
|
{
|
|
constexpr DWORD exe_path_size = 0xFFFF;
|
|
auto exe_path = std::make_unique<wchar_t[]>(exe_path_size);
|
|
GetModuleFileNameW(nullptr, exe_path.get(), exe_path_size);
|
|
return run_same_elevation(exe_path.get(), L"--dont-elevate", nullptr);
|
|
}
|