97 lines
2.1 KiB
C#
97 lines
2.1 KiB
C#
using System;
|
|
using System.ComponentModel;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Text;
|
|
|
|
namespace OpenRA.Launcher
|
|
{
|
|
class UtilityProgramResponse
|
|
{
|
|
public bool IsError
|
|
{
|
|
get { return response.StartsWith("Error:"); }
|
|
}
|
|
|
|
string response;
|
|
|
|
public string Response
|
|
{
|
|
get
|
|
{
|
|
if (IsError)
|
|
return response.Remove(0, 7);
|
|
else
|
|
return response;
|
|
}
|
|
}
|
|
|
|
public string[] ResponseLines
|
|
{
|
|
get
|
|
{
|
|
string s = response;
|
|
if (IsError)
|
|
s = response.Remove(0, 7);
|
|
return s.Split(new string[] { "\r\n" }, StringSplitOptions.RemoveEmptyEntries);
|
|
}
|
|
}
|
|
|
|
public UtilityProgramResponse(string response)
|
|
{
|
|
this.response = response.Trim('\r', '\n');
|
|
}
|
|
}
|
|
|
|
static class UtilityProgram
|
|
{
|
|
static string BuildArgs(string command, string[] args)
|
|
{
|
|
StringBuilder arguments = new StringBuilder();
|
|
arguments.Append("\"");
|
|
arguments.Append(command + "=");
|
|
arguments.Append(string.Join(",", args));
|
|
arguments.Append("\"");
|
|
return arguments.ToString();
|
|
}
|
|
|
|
public static UtilityProgramResponse Call(string command, params string[] args)
|
|
{
|
|
Process p = new Process();
|
|
p.StartInfo.FileName = "OpenRA.Utility.exe";
|
|
p.StartInfo.Arguments = BuildArgs(command, args);
|
|
p.StartInfo.RedirectStandardOutput = true;
|
|
p.StartInfo.UseShellExecute = false;
|
|
p.StartInfo.CreateNoWindow = true;
|
|
|
|
p.Start();
|
|
|
|
return new UtilityProgramResponse(p.StandardOutput.ReadToEnd());
|
|
}
|
|
|
|
public static UtilityProgramResponse CallWithAdmin(string command, params string[] args)
|
|
{
|
|
Process p = new Process();
|
|
p.StartInfo.FileName = "OpenRA.Utility.exe";
|
|
p.StartInfo.Arguments = BuildArgs(command, args);
|
|
p.StartInfo.CreateNoWindow = true;
|
|
p.StartInfo.Verb = "runas";
|
|
|
|
try
|
|
{
|
|
p.Start();
|
|
}
|
|
catch (Win32Exception e)
|
|
{
|
|
if (e.NativeErrorCode == 1223) //ERROR_CANCELLED
|
|
return new UtilityProgramResponse("Error: User cancelled elevation prompt.");
|
|
throw e;
|
|
}
|
|
|
|
p.WaitForExit();
|
|
|
|
return new UtilityProgramResponse(File.ReadAllText("output.txt"));
|
|
}
|
|
}
|
|
}
|