From 2956814e33d623076f59184e29571277e5aeb956 Mon Sep 17 00:00:00 2001
From: Originalimoc <wzj9912@icloud.com>
Date: Sat, 16 Dec 2023 14:49:03 -0800
Subject: [PATCH] GetStateEx support (#23)

---
 Cargo.toml |  2 +-
 src/lib.rs | 49 ++++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 49 insertions(+), 2 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index ccbf6f1..7faf270 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "rusty-xinput"
-version = "1.2.1"
+version = "1.2.2"
 authors = ["Lokathor <zefria@gmail.com>"]
 repository = "https://github.com/Lokathor/rusty-xinput"
 readme = "README.md"
diff --git a/src/lib.rs b/src/lib.rs
index 40e6659..db7eec9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -42,10 +42,14 @@ use winapi::shared::winerror::{ERROR_DEVICE_NOT_CONNECTED, ERROR_EMPTY, ERROR_SU
 use winapi::um::libloaderapi::{GetProcAddress, LoadLibraryW};
 use winapi::um::xinput::*;
 
+/// GetStateEx can get this in wButton
+pub const XINPUT_GAMEPAD_GUIDE: winapi::shared::minwindef::WORD = 0x0400;
+
 use std::fmt::{self, Debug, Formatter};
 
 type XInputEnableFunc = unsafe extern "system" fn(BOOL);
 type XInputGetStateFunc = unsafe extern "system" fn(DWORD, *mut XINPUT_STATE) -> DWORD;
+type XInputGetStateExFunc = unsafe extern "system" fn(DWORD, *mut XINPUT_STATE) -> DWORD;
 type XInputSetStateFunc = unsafe extern "system" fn(DWORD, *mut XINPUT_VIBRATION) -> DWORD;
 type XInputGetCapabilitiesFunc =
   unsafe extern "system" fn(DWORD, DWORD, *mut XINPUT_CAPABILITIES) -> DWORD;
@@ -69,6 +73,7 @@ pub struct XInputHandle {
   handle: HMODULE,
   xinput_enable: XInputEnableFunc,
   xinput_get_state: XInputGetStateFunc,
+  xinput_get_state_ex: XInputGetStateExFunc,
   xinput_set_state: XInputSetStateFunc,
   xinput_get_capabilities: XInputGetCapabilitiesFunc,
   opt_xinput_get_keystroke: Option<XInputGetKeystrokeFunc>,
@@ -210,6 +215,7 @@ impl XInputHandle {
 
     let mut opt_xinput_enable = None;
     let mut opt_xinput_get_state = None;
+    let mut opt_xinput_get_state_ex = None;
     let mut opt_xinput_set_state = None;
     let mut opt_xinput_get_capabilities = None;
     let mut opt_xinput_get_keystroke = None;
@@ -239,6 +245,17 @@ impl XInputHandle {
       }
     }
 
+    // using transmute is so dodgy we'll put that in its own unsafe block.
+    unsafe {
+      let get_state_ex_ptr = GetProcAddress(xinput_handle, 100_i32 as winapi::um::winnt::LPCSTR);
+      if !get_state_ex_ptr.is_null() {
+        trace!("Found XInputGetStateEx.");
+        opt_xinput_get_state_ex = Some(::std::mem::transmute(get_state_ex_ptr));
+      } else {
+        trace!("Could not find XInputGetStateEx.");
+      }
+    }
+
     // using transmute is so dodgy we'll put that in its own unsafe block.
     unsafe {
       let set_state_ptr = GetProcAddress(xinput_handle, set_state_name.as_ptr() as *mut i8);
@@ -318,7 +335,8 @@ impl XInputHandle {
     // this is safe because no other code can be loading xinput at the same time as us.
     #[allow(clippy::unnecessary_unwrap)]
     if opt_xinput_enable.is_some()
-      && opt_xinput_get_state.is_some()
+    && opt_xinput_get_state.is_some()
+    && opt_xinput_get_state_ex.is_some()
       && opt_xinput_set_state.is_some()
       && opt_xinput_get_capabilities.is_some()
     {
@@ -327,6 +345,7 @@ impl XInputHandle {
         handle: xinput_handle,
         xinput_enable: opt_xinput_enable.unwrap(),
         xinput_get_state: opt_xinput_get_state.unwrap(),
+        xinput_get_state_ex: opt_xinput_get_state_ex.unwrap(),
         xinput_set_state: opt_xinput_set_state.unwrap(),
         xinput_get_capabilities: opt_xinput_get_capabilities.unwrap(),
         opt_xinput_get_keystroke,
@@ -522,6 +541,16 @@ impl XInputState {
     self.raw.Gamepad.wButtons & XINPUT_GAMEPAD_BACK != 0
   }
 
+  /// The "guide" button.
+  ///
+  /// * Nintendo: Home
+  /// * Playstation: PS
+  /// * XBox: Guide
+  #[inline]
+  pub fn guide_button(&self) -> bool {
+    self.raw.Gamepad.wButtons & XINPUT_GAMEPAD_GUIDE != 0
+  }
+
   /// The upper left shoulder button.
   ///
   /// * Nintendo: L
@@ -770,6 +799,24 @@ impl XInputHandle {
       }
     }
   }
+
+  /// see get_state, but XINPUT_GAMEPAD_GUIDE in XInputState
+  pub fn get_state_ex(&self, user_index: u32) -> Result<XInputState, XInputUsageError> {
+    if user_index >= 4 {
+      Err(XInputUsageError::InvalidControllerID)
+    } else {
+      let mut output: XINPUT_STATE = unsafe { ::std::mem::zeroed() };
+      let return_status = unsafe { (self.xinput_get_state_ex)(user_index, &mut output) };
+      match return_status {
+        ERROR_SUCCESS => Ok(XInputState { raw: output }),
+        ERROR_DEVICE_NOT_CONNECTED => Err(XInputUsageError::DeviceNotConnected),
+        s => {
+          trace!("Unexpected error code: {}", s);
+          Err(XInputUsageError::UnknownError(s))
+        }
+      }
+    }
+  }
 }
 
 /// See `XInputHandle::get_state`