The-Homebrew-Cloud/cemu-DS4Windows
2
0
Fork 0
mirror of https://github.com/cemu-project/DS4Windows.git synced 2024-12-25 08:01:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Move elevation code to ControlService class

Browse Source
This commit is contained in:
Travis Nickles 2020-02-13 17:01:14 -06:00
parent 4444ce94c8
commit 20962895b0
2 changed files with 53 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
DS4Windows/DS4Control/ControlService.cs
View File

@ -176,6 +176,31 @@ namespace DS4Windows
}
outputslotMan = new OutputSlotManager();
DS4Devices.RequestElevation += DS4Devices_RequestElevation;
}
private void DS4Devices_RequestElevation(RequestElevationArgs args)
{
// Launches an elevated child process to re-enable device
string exeName = Global.exelocation;
ProcessStartInfo startInfo = new ProcessStartInfo(exeName);
startInfo.Verb = "runas";
startInfo.Arguments = "re-enabledevice " + args.InstanceId;
try
{
Process child = Process.Start(startInfo);
if (!child.WaitForExit(30000))
{
child.Kill();
}
else
{
args.StatusCode = child.ExitCode;
}
child.Dispose();
}
catch { }
}
public void LaunchHidGuardHelper()

40
DS4Windows/DS4Library/DS4Devices.cs
View File

@ -20,6 +20,27 @@ namespace DS4Windows
}
}
public class RequestElevationArgs : EventArgs
{
public const int STATUS_SUCCESS = 0;
public const int STATUS_INIT_FAILURE = -1;
private int statusCode = STATUS_INIT_FAILURE;
private string instanceId;
public int StatusCode
{
get => statusCode;
set => statusCode = value;
}
public string InstanceId { get => instanceId; }
public RequestElevationArgs(string instanceId)
{
this.instanceId = instanceId;
}
}
public delegate void RequestElevationDelegate(RequestElevationArgs args);
public class DS4Devices
{
// (HID device path, DS4Device)
@ -29,6 +50,7 @@ namespace DS4Windows
// Keep instance of opened exclusive mode devices not in use (Charging while using BT connection)
private static List<HidDevice> DisabledDevices = new List<HidDevice>();
private static Stopwatch sw = new Stopwatch();
public static event RequestElevationDelegate RequestElevation;
public static bool isExclusiveMode = false;
internal const int SONY_VID = 0x054C;
internal const int RAZER_VID = 0x1532;
@ -120,24 +142,18 @@ namespace DS4Windows
{
try
{
// Check if running with elevated permissions
WindowsIdentity identity = WindowsIdentity.GetCurrent();
WindowsPrincipal principal = new WindowsPrincipal(identity);
bool elevated = principal.IsInRole(WindowsBuiltInRole.Administrator);
if (!elevated)
{
// Launches an elevated child process to re-enable device
string exeName = Process.GetCurrentProcess().MainModule.FileName;
ProcessStartInfo startInfo = new ProcessStartInfo(exeName);
startInfo.Verb = "runas";
startInfo.Arguments = "re-enabledevice " + devicePathToInstanceId(hDevice.DevicePath);
Process child = Process.Start(startInfo);
if (!child.WaitForExit(30000))
{
child.Kill();
}
else if (child.ExitCode == 0)
// Tell the client to launch routine to re-enable a device
RequestElevationArgs eleArgs =
new RequestElevationArgs(devicePathToInstanceId(hDevice.DevicePath));
RequestElevation?.Invoke(eleArgs);
if (eleArgs.StatusCode == RequestElevationArgs.STATUS_SUCCESS)
{
hDevice.OpenDevice(isExclusiveMode);
}