From cc00075a65e0a3c1e93418daeb2a22dd6fede377 Mon Sep 17 00:00:00 2001
From: Aric Stewart <aric@codeweavers.com>
Date: Wed, 25 Jan 2017 07:07:04 -0600
Subject: [PATCH] hid: Implement HidP_GetUsagesEx.

Signed-off-by: Aric Stewart <aric@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 dlls/hid/hid.spec   |  2 +-
 dlls/hid/hidp.c     | 87 +++++++++++++++++++++++++++++++++++++++++++++
 include/ddk/hidpi.h |  6 ++++
 3 files changed, 94 insertions(+), 1 deletion(-)

diff --git a/dlls/hid/hid.spec b/dlls/hid/hid.spec
index b1b6a350358..205b6265e9c 100644
--- a/dlls/hid/hid.spec
+++ b/dlls/hid/hid.spec
@@ -29,7 +29,7 @@
 @ stdcall HidP_GetUsageValue(long long long long ptr ptr ptr long)
 @ stub HidP_GetUsageValueArray
 @ stdcall HidP_GetUsages(long long long ptr ptr ptr ptr long)
-@ stub HidP_GetUsagesEx
+@ stdcall HidP_GetUsagesEx(long long ptr ptr ptr ptr long)
 @ stdcall HidP_GetValueCaps(long ptr ptr ptr)
 @ stdcall HidP_InitializeReportForID(long long ptr ptr long)
 @ stub HidP_MaxDataListLength
diff --git a/dlls/hid/hidp.c b/dlls/hid/hidp.c
index 9dcdeb6f52b..d031b67da40 100644
--- a/dlls/hid/hidp.c
+++ b/dlls/hid/hidp.c
@@ -676,3 +676,90 @@ NTSTATUS WINAPI HidP_GetSpecificValueCaps(HIDP_REPORT_TYPE ReportType,
 
     return HIDP_STATUS_SUCCESS;
 }
+
+NTSTATUS WINAPI HidP_GetUsagesEx(HIDP_REPORT_TYPE ReportType, USHORT LinkCollection, USAGE_AND_PAGE *ButtonList,
+    ULONG *UsageLength, PHIDP_PREPARSED_DATA PreparsedData, CHAR *Report, ULONG ReportLength)
+{
+    WINE_HIDP_PREPARSED_DATA *data = (WINE_HIDP_PREPARSED_DATA*)PreparsedData;
+    WINE_HID_REPORT *report = NULL;
+    USHORT b_count = 0, r_count = 0;
+    int i,uCount = 0;
+    NTSTATUS rc;
+
+    TRACE("(%i, %i, %p, %p(%i), %p, %p, %i)\n", ReportType, LinkCollection, ButtonList,
+        UsageLength, *UsageLength, PreparsedData, Report, ReportLength);
+
+    if (data->magic != HID_MAGIC)
+        return HIDP_STATUS_INVALID_PREPARSED_DATA;
+
+    switch(ReportType)
+    {
+        case HidP_Input:
+            b_count = data->caps.NumberInputButtonCaps;
+            r_count = data->dwInputReportCount;
+            report = HID_INPUT_REPORTS(data);
+            break;
+        case HidP_Output:
+            b_count = data->caps.NumberOutputButtonCaps;
+            r_count = data->dwOutputReportCount;
+            report = HID_OUTPUT_REPORTS(data);
+            break;
+        case HidP_Feature:
+            b_count = data->caps.NumberFeatureButtonCaps;
+            r_count = data->dwFeatureReportCount;
+            report = HID_FEATURE_REPORTS(data);
+            break;
+        default:
+            return HIDP_STATUS_INVALID_REPORT_TYPE;
+    }
+
+    if (!r_count || !b_count || !report)
+        return HIDP_STATUS_USAGE_NOT_FOUND;
+
+    for (i = 0; i < r_count; i++)
+    {
+        if (!report->reportID || report->reportID == Report[0])
+            break;
+        report = HID_NEXT_REPORT(data, report);
+    }
+
+    if (i == r_count)
+        return HIDP_STATUS_REPORT_DOES_NOT_EXIST;
+
+    for (i = 0; i < report->elementCount; i++)
+    {
+        if (report->Elements[i].ElementType == ButtonElement)
+        {
+            int k;
+            WINE_HID_ELEMENT *element = &report->Elements[i];
+            for (k=0; k < element->bitCount; k++)
+            {
+                UINT v = 0;
+                NTSTATUS rc = get_report_data((BYTE*)Report, ReportLength,
+                                element->valueStartBit + k, 1, &v);
+                if (rc != HIDP_STATUS_SUCCESS)
+                    return rc;
+                if (v)
+                {
+                    if (uCount < *UsageLength)
+                    {
+                        ButtonList[uCount].Usage = element->caps.button.u.Range.UsageMin + k;
+                        ButtonList[uCount].UsagePage = element->caps.button.UsagePage;
+                    }
+                    uCount++;
+                }
+            }
+        }
+    }
+
+    TRACE("Returning %i usages\n", uCount);
+
+    if (*UsageLength < uCount)
+        rc = HIDP_STATUS_BUFFER_TOO_SMALL;
+    else
+        rc = HIDP_STATUS_SUCCESS;
+
+    *UsageLength = uCount;
+
+    return rc;
+}
diff --git a/include/ddk/hidpi.h b/include/ddk/hidpi.h
index 9f0fd8fc13c..2b0eebdd76b 100644
--- a/include/ddk/hidpi.h
+++ b/include/ddk/hidpi.h
@@ -161,6 +161,11 @@ typedef struct _HIDP_KEYBOARD_MODIFIER_STATE {
     } DUMMYUNIONNAME;
 } HIDP_KEYBOARD_MODIFIER_STATE, *PHIDP_KEYBOARD_MODIFIER_STATE;
 
+typedef struct _USAGE_AND_PAGE {
+  USAGE Usage;
+  USAGE UsagePage;
+} USAGE_AND_PAGE, *PUSAGE_AND_PAGE;
+
 typedef BOOLEAN (NTAPI *PHIDP_INSERT_SCANCODES) (VOID *Context, CHAR *NewScanCodes, ULONG Length);
 
 
@@ -175,6 +180,7 @@ NTSTATUS WINAPI HidP_GetScaledUsageValue(HIDP_REPORT_TYPE ReportType, USAGE Usag
 NTSTATUS WINAPI HidP_TranslateUsagesToI8042ScanCodes(USAGE *ChangedUsageList, ULONG UsageListLength, HIDP_KEYBOARD_DIRECTION KeyAction, HIDP_KEYBOARD_MODIFIER_STATE *ModifierState, PHIDP_INSERT_SCANCODES InsertCodesProcedure, VOID *InsertCodesContext);
 NTSTATUS WINAPI HidP_GetSpecificButtonCaps(HIDP_REPORT_TYPE ReportType, USAGE UsagePage, USHORT LinkCollection, USAGE Usage, HIDP_BUTTON_CAPS *ButtonCaps, USHORT *ButtonCapsLength, PHIDP_PREPARSED_DATA PreparsedData);
 NTSTATUS WINAPI HidP_GetSpecificValueCaps(HIDP_REPORT_TYPE ReportType, USAGE UsagePage, USHORT LinkCollection, USAGE Usage, HIDP_VALUE_CAPS *ValueCaps, USHORT *ValueCapsLength, PHIDP_PREPARSED_DATA PreparsedData);
+NTSTATUS WINAPI HidP_GetUsagesEx(HIDP_REPORT_TYPE ReportType, USHORT LinkCollection, USAGE_AND_PAGE *ButtonList, ULONG *UsageLength, PHIDP_PREPARSED_DATA PreparsedData, CHAR *Report, ULONG ReportLength);
 
 #ifndef FACILITY_HID_ERROR_CODE
 #define FACILITY_HID_ERROR_CODE 0x11