mirror of
https://github.com/cemu-project/DS4Windows.git
synced 2024-11-30 04:54:20 +01:00
558 lines
20 KiB
C#
558 lines
20 KiB
C#
using System;
|
|
using System.Runtime.InteropServices;
|
|
using System.Threading;
|
|
using System.IO;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Win32.SafeHandles;
|
|
namespace DS4Windows
|
|
{
|
|
public class HidDevice : IDisposable
|
|
{
|
|
public enum ReadStatus
|
|
{
|
|
Success = 0,
|
|
WaitTimedOut = 1,
|
|
WaitFail = 2,
|
|
NoDataRead = 3,
|
|
ReadError = 4,
|
|
NotConnected = 5
|
|
}
|
|
|
|
private readonly string _description;
|
|
private readonly string _devicePath;
|
|
private readonly HidDeviceAttributes _deviceAttributes;
|
|
|
|
private readonly HidDeviceCapabilities _deviceCapabilities;
|
|
private bool _monitorDeviceEvents;
|
|
private string serial = null;
|
|
private const string BLANK_SERIAL = "00:00:00:00:00:00";
|
|
|
|
internal HidDevice(string devicePath, string description = null)
|
|
{
|
|
_devicePath = devicePath;
|
|
_description = description;
|
|
|
|
try
|
|
{
|
|
var hidHandle = OpenHandle(_devicePath, false);
|
|
|
|
_deviceAttributes = GetDeviceAttributes(hidHandle);
|
|
_deviceCapabilities = GetDeviceCapabilities(hidHandle);
|
|
|
|
hidHandle.Close();
|
|
}
|
|
catch (Exception exception)
|
|
{
|
|
Console.WriteLine(exception.Message);
|
|
throw new Exception(string.Format("Error querying HID device '{0}'.", devicePath), exception);
|
|
}
|
|
}
|
|
|
|
public SafeFileHandle safeReadHandle { get; private set; }
|
|
public FileStream fileStream { get; private set; }
|
|
public bool IsOpen { get; private set; }
|
|
public bool IsExclusive { get; private set; }
|
|
public bool IsConnected { get { return HidDevices.IsConnected(_devicePath); } }
|
|
public string Description { get { return _description; } }
|
|
public HidDeviceCapabilities Capabilities { get { return _deviceCapabilities; } }
|
|
public HidDeviceAttributes Attributes { get { return _deviceAttributes; } }
|
|
public string DevicePath { get { return _devicePath; } }
|
|
|
|
public override string ToString()
|
|
{
|
|
return string.Format("VendorID={0}, ProductID={1}, Version={2}, DevicePath={3}",
|
|
_deviceAttributes.VendorHexId,
|
|
_deviceAttributes.ProductHexId,
|
|
_deviceAttributes.Version,
|
|
_devicePath);
|
|
}
|
|
|
|
public void OpenDevice(bool isExclusive)
|
|
{
|
|
if (IsOpen) return;
|
|
try
|
|
{
|
|
if (safeReadHandle == null || safeReadHandle.IsInvalid)
|
|
safeReadHandle = OpenHandle(_devicePath, isExclusive);
|
|
}
|
|
catch (Exception exception)
|
|
{
|
|
IsOpen = false;
|
|
throw new Exception("Error opening HID device.", exception);
|
|
}
|
|
|
|
IsOpen = !safeReadHandle.IsInvalid;
|
|
IsExclusive = isExclusive;
|
|
}
|
|
|
|
public void OpenFileStream(int reportSize)
|
|
{
|
|
if (fileStream == null && !safeReadHandle.IsInvalid)
|
|
{
|
|
fileStream = new FileStream(safeReadHandle, FileAccess.ReadWrite, reportSize, true);
|
|
}
|
|
}
|
|
|
|
public bool IsFileStreamOpen()
|
|
{
|
|
bool result = false;
|
|
if (fileStream != null)
|
|
{
|
|
result = !fileStream.SafeFileHandle.IsInvalid && !fileStream.SafeFileHandle.IsClosed;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
public void CloseDevice()
|
|
{
|
|
if (!IsOpen) return;
|
|
closeFileStreamIO();
|
|
|
|
IsOpen = false;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
CancelIO();
|
|
CloseDevice();
|
|
}
|
|
|
|
public void CancelIO()
|
|
{
|
|
if (IsOpen)
|
|
NativeMethods.CancelIoEx(safeReadHandle.DangerousGetHandle(), IntPtr.Zero);
|
|
}
|
|
|
|
public bool ReadInputReport(byte[] data)
|
|
{
|
|
if (safeReadHandle == null)
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
return NativeMethods.HidD_GetInputReport(safeReadHandle, data, data.Length);
|
|
}
|
|
|
|
public bool WriteFeatureReport(byte[] data)
|
|
{
|
|
bool result = false;
|
|
if (IsOpen && safeReadHandle != null)
|
|
{
|
|
result = NativeMethods.HidD_SetFeature(safeReadHandle, data, data.Length);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
private static HidDeviceAttributes GetDeviceAttributes(SafeFileHandle hidHandle)
|
|
{
|
|
var deviceAttributes = default(NativeMethods.HIDD_ATTRIBUTES);
|
|
deviceAttributes.Size = Marshal.SizeOf(deviceAttributes);
|
|
NativeMethods.HidD_GetAttributes(hidHandle.DangerousGetHandle(), ref deviceAttributes);
|
|
return new HidDeviceAttributes(deviceAttributes);
|
|
}
|
|
|
|
private static HidDeviceCapabilities GetDeviceCapabilities(SafeFileHandle hidHandle)
|
|
{
|
|
var capabilities = default(NativeMethods.HIDP_CAPS);
|
|
var preparsedDataPointer = default(IntPtr);
|
|
|
|
if (NativeMethods.HidD_GetPreparsedData(hidHandle.DangerousGetHandle(), ref preparsedDataPointer))
|
|
{
|
|
NativeMethods.HidP_GetCaps(preparsedDataPointer, ref capabilities);
|
|
NativeMethods.HidD_FreePreparsedData(preparsedDataPointer);
|
|
}
|
|
|
|
return new HidDeviceCapabilities(capabilities);
|
|
}
|
|
|
|
private void closeFileStreamIO()
|
|
{
|
|
if (fileStream != null)
|
|
{
|
|
try
|
|
{
|
|
fileStream.Close();
|
|
}
|
|
catch (IOException) { }
|
|
catch (OperationCanceledException) { }
|
|
}
|
|
|
|
fileStream = null;
|
|
Console.WriteLine("Close fs");
|
|
if (safeReadHandle != null && !safeReadHandle.IsInvalid)
|
|
{
|
|
try
|
|
{
|
|
if (!safeReadHandle.IsClosed)
|
|
{
|
|
safeReadHandle.Close();
|
|
Console.WriteLine("Close sh");
|
|
}
|
|
}
|
|
catch (IOException) { }
|
|
}
|
|
|
|
safeReadHandle = null;
|
|
}
|
|
|
|
public void flush_Queue()
|
|
{
|
|
if (safeReadHandle != null)
|
|
{
|
|
NativeMethods.HidD_FlushQueue(safeReadHandle);
|
|
}
|
|
}
|
|
|
|
private ReadStatus ReadWithFileStreamTask(byte[] inputBuffer)
|
|
{
|
|
try
|
|
{
|
|
if (fileStream.Read(inputBuffer, 0, inputBuffer.Length) > 0)
|
|
{
|
|
return ReadStatus.Success;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.NoDataRead;
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
}
|
|
|
|
public ReadStatus ReadFile(byte[] inputBuffer)
|
|
{
|
|
if (safeReadHandle == null)
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
try
|
|
{
|
|
uint bytesRead;
|
|
if (NativeMethods.ReadFile(safeReadHandle.DangerousGetHandle(), inputBuffer, (uint)inputBuffer.Length, out bytesRead, IntPtr.Zero))
|
|
{
|
|
return ReadStatus.Success;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.NoDataRead;
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
}
|
|
|
|
public ReadStatus ReadWithFileStream(byte[] inputBuffer)
|
|
{
|
|
try
|
|
{
|
|
if (fileStream.Read(inputBuffer, 0, inputBuffer.Length) > 0)
|
|
{
|
|
return ReadStatus.Success;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.NoDataRead;
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
}
|
|
|
|
public ReadStatus ReadWithFileStream(byte[] inputBuffer, int timeout)
|
|
{
|
|
try
|
|
{
|
|
if (safeReadHandle == null)
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
if (fileStream == null && !safeReadHandle.IsInvalid)
|
|
fileStream = new FileStream(safeReadHandle, FileAccess.ReadWrite, inputBuffer.Length, true);
|
|
|
|
if (!safeReadHandle.IsInvalid && fileStream.CanRead)
|
|
{
|
|
Task<ReadStatus> readFileTask = new Task<ReadStatus>(() => ReadWithFileStreamTask(inputBuffer));
|
|
readFileTask.Start();
|
|
bool success = readFileTask.Wait(timeout);
|
|
if (success)
|
|
{
|
|
if (readFileTask.Result == ReadStatus.Success)
|
|
{
|
|
return ReadStatus.Success;
|
|
}
|
|
else if (readFileTask.Result == ReadStatus.ReadError)
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
else if (readFileTask.Result == ReadStatus.NoDataRead)
|
|
{
|
|
return ReadStatus.NoDataRead;
|
|
}
|
|
}
|
|
else
|
|
return ReadStatus.WaitTimedOut;
|
|
}
|
|
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
if (e is AggregateException)
|
|
{
|
|
Console.WriteLine(e.Message);
|
|
return ReadStatus.WaitFail;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
}
|
|
|
|
return ReadStatus.ReadError;
|
|
}
|
|
|
|
public ReadStatus ReadAsyncWithFileStream(byte[] inputBuffer, int timeout)
|
|
{
|
|
try
|
|
{
|
|
if (safeReadHandle == null)
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
if (fileStream == null && !safeReadHandle.IsInvalid)
|
|
fileStream = new FileStream(safeReadHandle, FileAccess.ReadWrite, inputBuffer.Length, true);
|
|
|
|
if (!safeReadHandle.IsInvalid && fileStream.CanRead)
|
|
{
|
|
Task<int> readTask = fileStream.ReadAsync(inputBuffer, 0, inputBuffer.Length);
|
|
bool success = readTask.Wait(timeout);
|
|
if (success)
|
|
{
|
|
if (readTask.Result > 0)
|
|
{
|
|
return ReadStatus.Success;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.NoDataRead;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.WaitTimedOut;
|
|
}
|
|
}
|
|
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
if (e is AggregateException)
|
|
{
|
|
Console.WriteLine(e.Message);
|
|
return ReadStatus.WaitFail;
|
|
}
|
|
else
|
|
{
|
|
return ReadStatus.ReadError;
|
|
}
|
|
}
|
|
|
|
return ReadStatus.ReadError;
|
|
}
|
|
|
|
public bool WriteOutputReportViaControl(byte[] outputBuffer)
|
|
{
|
|
if (safeReadHandle == null)
|
|
{
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
}
|
|
|
|
if (NativeMethods.HidD_SetOutputReport(safeReadHandle, outputBuffer, outputBuffer.Length))
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
private bool WriteOutputReportViaInterruptTask(byte[] outputBuffer)
|
|
{
|
|
try
|
|
{
|
|
fileStream.Write(outputBuffer, 0, outputBuffer.Length);
|
|
return true;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Console.WriteLine(e.Message);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public bool WriteOutputReportViaInterrupt(byte[] outputBuffer, int timeout)
|
|
{
|
|
try
|
|
{
|
|
if (safeReadHandle == null)
|
|
{
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
}
|
|
if (fileStream == null && !safeReadHandle.IsInvalid)
|
|
{
|
|
fileStream = new FileStream(safeReadHandle, FileAccess.ReadWrite, outputBuffer.Length, true);
|
|
}
|
|
if (fileStream != null && fileStream.CanWrite && !safeReadHandle.IsInvalid)
|
|
{
|
|
fileStream.Write(outputBuffer, 0, outputBuffer.Length);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
public bool WriteAsyncOutputReportViaInterrupt(byte[] outputBuffer)
|
|
{
|
|
try
|
|
{
|
|
if (safeReadHandle == null)
|
|
{
|
|
safeReadHandle = OpenHandle(_devicePath, true);
|
|
}
|
|
if (fileStream == null && !safeReadHandle.IsInvalid)
|
|
{
|
|
fileStream = new FileStream(safeReadHandle, FileAccess.ReadWrite, outputBuffer.Length, true);
|
|
}
|
|
if (fileStream != null && fileStream.CanWrite && !safeReadHandle.IsInvalid)
|
|
{
|
|
Task writeTask = fileStream.WriteAsync(outputBuffer, 0, outputBuffer.Length);
|
|
//fileStream.Write(outputBuffer, 0, outputBuffer.Length);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
catch (Exception)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
private SafeFileHandle OpenHandle(String devicePathName, Boolean isExclusive)
|
|
{
|
|
SafeFileHandle hidHandle;
|
|
|
|
if (isExclusive)
|
|
{
|
|
hidHandle = NativeMethods.CreateFile(devicePathName, NativeMethods.GENERIC_READ | NativeMethods.GENERIC_WRITE, 0, IntPtr.Zero, NativeMethods.OpenExisting, 0x20000000 | 0x80000000 | 0x100 | NativeMethods.FILE_FLAG_OVERLAPPED, 0);
|
|
}
|
|
else
|
|
{
|
|
hidHandle = NativeMethods.CreateFile(devicePathName, NativeMethods.GENERIC_READ | NativeMethods.GENERIC_WRITE, NativeMethods.FILE_SHARE_READ | NativeMethods.FILE_SHARE_WRITE, IntPtr.Zero, NativeMethods.OpenExisting, 0x20000000 | 0x80000000 | 0x100 | NativeMethods.FILE_FLAG_OVERLAPPED, 0);
|
|
}
|
|
|
|
return hidHandle;
|
|
}
|
|
|
|
public bool readFeatureData(byte[] inputBuffer)
|
|
{
|
|
return NativeMethods.HidD_GetFeature(safeReadHandle.DangerousGetHandle(), inputBuffer, inputBuffer.Length);
|
|
}
|
|
|
|
public void resetSerial()
|
|
{
|
|
serial = null;
|
|
}
|
|
|
|
public string readSerial()
|
|
{
|
|
if (serial != null)
|
|
return serial;
|
|
|
|
// Some devices don't have MAC address (especially gamepads with USB only suports in PC). If the serial number reading fails
|
|
// then use dummy zero MAC address, because there is a good chance the gamepad stll works in DS4Windows app (the code would throw
|
|
// an index out of bounds exception anyway without IF-THEN-ELSE checks after trying to read a serial number).
|
|
|
|
if (Capabilities.InputReportByteLength == 64)
|
|
{
|
|
byte[] buffer = new byte[16];
|
|
buffer[0] = 18;
|
|
if (readFeatureData(buffer))
|
|
serial = String.Format("{0:X02}:{1:X02}:{2:X02}:{3:X02}:{4:X02}:{5:X02}",
|
|
buffer[6], buffer[5], buffer[4], buffer[3], buffer[2], buffer[1]);
|
|
}
|
|
else
|
|
{
|
|
byte[] buffer = new byte[126];
|
|
#if WIN64
|
|
ulong bufferLen = 126;
|
|
#else
|
|
uint bufferLen = 126;
|
|
#endif
|
|
if (NativeMethods.HidD_GetSerialNumberString(safeReadHandle.DangerousGetHandle(), buffer, bufferLen))
|
|
{
|
|
string MACAddr = System.Text.Encoding.Unicode.GetString(buffer).Replace("\0", string.Empty).ToUpper();
|
|
MACAddr = $"{MACAddr[0]}{MACAddr[1]}:{MACAddr[2]}{MACAddr[3]}:{MACAddr[4]}{MACAddr[5]}:{MACAddr[6]}{MACAddr[7]}:{MACAddr[8]}{MACAddr[9]}:{MACAddr[10]}{MACAddr[11]}";
|
|
serial = MACAddr;
|
|
}
|
|
}
|
|
|
|
// If serial# reading failed then generate a dummy MAC address based on HID device path (WinOS generated runtime unique value based on connected usb port and hub or BT channel).
|
|
// The device path remains the same as long the gamepad is always connected to the same usb/BT port, but may be different in other usb ports. Therefore this value is unique
|
|
// as long the same device is always connected to the same usb port.
|
|
if (serial == null)
|
|
{
|
|
string MACAddr = string.Empty;
|
|
|
|
AppLogger.LogToGui($"WARNING: Failed to read serial# from a gamepad ({this._deviceAttributes.VendorHexId}/{this._deviceAttributes.ProductHexId}). Generating MAC address from a device path. From now on you should connect this gamepad always into the same USB port or BT pairing host to keep the same device path.", true);
|
|
|
|
try
|
|
{
|
|
// Substring: \\?\hid#vid_054c&pid_09cc&mi_03#7&1f882A25&0&0001#{4d1e55b2-f16f-11cf-88cb-001111000030} -> \\?\hid#vid_054c&pid_09cc&mi_03#7&1f882A25&0&0001#
|
|
int endPos = this.DevicePath.LastIndexOf('{');
|
|
if (endPos < 0)
|
|
endPos = this.DevicePath.Length;
|
|
|
|
// String array: \\?\hid#vid_054c&pid_09cc&mi_03#7&1f882A25&0&0001# -> [0]=\\?\hidvid_054c, [1]=pid_09cc, [2]=mi_037, [3]=1f882A25, [4]=0, [5]=0001
|
|
string[] devPathItems = this.DevicePath.Substring(0, endPos).Replace("#", "").Replace("-", "").Replace("{", "").Replace("}", "").Split('&');
|
|
|
|
if (devPathItems.Length >= 3)
|
|
MACAddr = devPathItems[devPathItems.Length - 3].ToUpper() // 1f882A25
|
|
+ devPathItems[devPathItems.Length - 2].ToUpper() // 0
|
|
+ devPathItems[devPathItems.Length - 1].TrimStart('0').ToUpper(); // 0001 -> 1
|
|
else if (devPathItems.Length >= 1)
|
|
// Device and usb hub and port identifiers missing in devicePath string. Fallback to use vendor and product ID values and
|
|
// take a number from the last part of the devicePath. Hopefully the last part is a usb port number as it usually should be.
|
|
MACAddr = this._deviceAttributes.VendorId.ToString("X4")
|
|
+ this._deviceAttributes.ProductId.ToString("X4")
|
|
+ devPathItems[devPathItems.Length - 1].TrimStart('0').ToUpper();
|
|
|
|
if (!string.IsNullOrEmpty(MACAddr))
|
|
{
|
|
MACAddr = MACAddr.PadRight(12, '0');
|
|
serial = $"{MACAddr[0]}{MACAddr[1]}:{MACAddr[2]}{MACAddr[3]}:{MACAddr[4]}{MACAddr[5]}:{MACAddr[6]}{MACAddr[7]}:{MACAddr[8]}{MACAddr[9]}:{MACAddr[10]}{MACAddr[11]}";
|
|
}
|
|
else
|
|
// Hmm... Shold never come here. Strange format in devicePath because all identifier items of devicePath string are missing.
|
|
serial = BLANK_SERIAL;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
AppLogger.LogToGui($"ERROR: Failed to generate runtime MAC address from device path {this.DevicePath}. {e.Message}", true);
|
|
serial = BLANK_SERIAL;
|
|
}
|
|
}
|
|
|
|
return serial;
|
|
}
|
|
}
|
|
}
|