// TODO: make it work
private static void TryElevate()
{
if (UACHelper.IsRunAsAdmin())
{
return;
}
var process = new Process()
{
StartInfo = new ProcessStartInfo
{
UseShellExecute = true,
WorkingDirectory = Environment.CurrentDirectory,
FileName = Globals.ExecutablePath,
Arguments = Environment.CommandLine,
Verb = "runas",
}
};
try
{
process.Start();
process.WaitForExit();
Environment.Exit(process.ExitCode);
}
catch
{
LogMuxer.Instance.Error("Elevation failed.");
Environment.Exit(1);
}
}
// ReSharper disable once UnusedMember.Global
public void OnExecute()
{
AppDomain.CurrentDomain.UnhandledException += CurrentDomainUnhandledException;
AppDomain.CurrentDomain.ProcessExit += OnProcessExit;
var currentDirectory = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
if (currentDirectory != null)
{
Directory.SetCurrentDirectory(currentDirectory);
}
// If this is a helper process
if (IsImpersonatedProcess)
{
LoadConfig();
LogMuxer.Instance.Debug("Executing impersonation helper routine");
ExecConfig execConfig;
switch (ExecConfigLaunchType)
{
case "ExecStart":
execConfig = Globals.Config.ExecStart[ExecConfigIndex];
break;
default:
throw new ArgumentException();
}
Wrapper = new ManagedProgramWrapper(execConfig.ProgramPath, execConfig.Arguments);
var hasExited = false;
Wrapper.ProgramExited += (sender, eventArgs) => { hasExited = true; };
Wrapper.Start();
while (!hasExited)
{
Thread.Sleep(1000);
}
return;
}
if (Environment.UserInteractive)
{
var configOk = true;
try
{
LoadConfig();
}
catch
{
configOk = false;
}
LogMuxer.Instance.Debug($"IsElevated: {UACHelper.IsRunAsAdmin()}");
LogMuxer.Instance.Debug($"Cmdline: {Environment.CommandLine}");
if (!UACHelper.IsRunAsAdmin())
{
LogMuxer.Instance.Warning("Warning: you may not have sufficient privilege to install services.");
}
//if (!DoNotElevate && !UACHelper.IsRunAsAdmin())
//{
// TryElevate();
//}
if (Install && Uninstall)
{
LogMuxer.Instance.Fatal("Self-contradictory arguments?");
Environment.Exit(1);
}
else if (Install)
{
if (!configOk)
{
LogMuxer.Instance.Fatal("Cannot read config");
Environment.Exit(1);
}
InstallService();
}
else if (Uninstall)
{
if (!configOk)
{
LogMuxer.Instance.Fatal("Cannot read config");
Environment.Exit(1);
}
UninstallService();
}
else if (RunOnly)
{
if (!configOk)
{
LogMuxer.Instance.Fatal("Cannot read config");
Environment.Exit(1);
}
var s = new Supervisor();
s.Start();
s.WaitForExit();
}
else
{
LogMuxer.Instance.Info("Searching for unit files in program directory...");
// Let's install every service in this folder
try
{
var txtFiles = Directory.EnumerateFiles(Globals.ExecutableDirectory, "*.service", SearchOption.TopDirectoryOnly);
foreach (string currentFile in txtFiles)
{
var proceed = Prompt.GetYesNo($"Do you want to register {Path.GetFileName(currentFile)}?", defaultAnswer: true);
if (proceed)
{
RegisterService(currentFile);
}
}
}
catch (Exception e)
{
LogMuxer.Instance.Error(e.Message);
}
}
}
else
{
LoadConfig();
ServiceBase.Run(new Service());
}
}
