util/setupapi.c: Add hook for SETUPAPI lib

util/meson.build

@@ -9,5 +9,7 @@ util_lib = static_library(
     sources : [
         'dprintf.c',
         'dprintf.h',
+        'setupapi.c',
+        'setupapi.h',
     ],
 )

util/setupapi.c

@@ -0,0 +1,321 @@
+#include <windows.h>
+#include <setupapi.h>
+
+#include <stdbool.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "hook/table.h"
+
+#include "util/setupapi.h"
+
+struct setupapi_class {
+    const GUID *guid;
+    const wchar_t *path;
+    HDEVINFO cur_handle;
+};
+
+static void setupapi_hook_init(void);
+
+/* API hooks */
+
+static HDEVINFO WINAPI my_SetupDiGetClassDevsW(
+        const GUID *ClassGuid,
+        wchar_t *Enumerator,
+        HWND hwndParent,
+        DWORD Flags);
+
+static BOOL WINAPI my_SetupDiEnumDeviceInterfaces(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVINFO_DATA *DeviceInfoData,
+        const GUID *InterfaceClassGuid,
+        DWORD MemberIndex,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData);
+
+static BOOL WINAPI my_SetupDiGetDeviceInterfaceDetailW(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData,
+        SP_DEVICE_INTERFACE_DETAIL_DATA_W *DeviceInterfaceDetailData,
+        DWORD DeviceInterfaceDetailDataSize,
+        DWORD *RequiredSize,
+        SP_DEVINFO_DATA *DeviceInfoData);
+
+static BOOL WINAPI my_SetupDiDestroyDeviceInfoList(HDEVINFO DeviceInfoSet);
+
+/* Links */
+
+static HDEVINFO WINAPI (*next_SetupDiGetClassDevsW)(
+        const GUID *ClassGuid,
+        wchar_t *Enumerator,
+        HWND hwndParent,
+        DWORD Flags);
+
+static BOOL WINAPI (*next_SetupDiEnumDeviceInterfaces)(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVINFO_DATA *DeviceInfoData,
+        const GUID *InterfaceClassGuid,
+        DWORD MemberIndex,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData);
+
+static BOOL WINAPI (*next_SetupDiGetDeviceInterfaceDetailW)(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData,
+        SP_DEVICE_INTERFACE_DETAIL_DATA_W *DeviceInterfaceDetailData,
+        DWORD DeviceInterfaceDetailDataSize,
+        DWORD *RequiredSize,
+        SP_DEVINFO_DATA *DeviceInfoData);
+
+static BOOL WINAPI (*next_SetupDiDestroyDeviceInfoList)(HDEVINFO DeviceInfoSet);
+
+/* Hook tbl */
+
+static const struct hook_symbol setupapi_syms[] = {
+    {
+        .name   = "SetupDiGetClassDevsW",
+        .patch  = my_SetupDiGetClassDevsW,
+        .link   = (void *) &next_SetupDiGetClassDevsW,
+    }, {
+        .name   = "SetupDiEnumDeviceInterfaces",
+        .patch  = my_SetupDiEnumDeviceInterfaces,
+        .link   = (void *) &next_SetupDiEnumDeviceInterfaces,
+    }, {
+        .name   = "SetupDiGetDeviceInterfaceDetailW",
+        .patch  = my_SetupDiGetDeviceInterfaceDetailW,
+        .link   = (void *) &next_SetupDiGetDeviceInterfaceDetailW,
+    }, {
+        .name   = "SetupDiDestroyDeviceInfoList",
+        .patch  = my_SetupDiDestroyDeviceInfoList,
+        .link   = (void *) &next_SetupDiDestroyDeviceInfoList,
+    }
+};
+
+static bool setupapi_initted;
+static CRITICAL_SECTION setupapi_lock;
+static struct setupapi_class *setupapi_classes;
+static size_t setupapi_nclasses;
+
+HRESULT setupapi_add_phantom_dev(const GUID *iface_class, const wchar_t *path)
+{
+    struct setupapi_class *class_;
+    struct setupapi_class *new_array;
+    HRESULT hr;
+
+    assert(iface_class != NULL);
+    assert(path != NULL);
+
+    setupapi_hook_init();
+
+    EnterCriticalSection(&setupapi_lock);
+
+    new_array = realloc(
+            setupapi_classes,
+            (setupapi_nclasses + 1) * sizeof(struct setupapi_class));
+
+    if (new_array == NULL) {
+        hr = E_OUTOFMEMORY;
+
+        goto end;
+    }
+
+    setupapi_classes = new_array;
+
+    class_ = &setupapi_classes[setupapi_nclasses++];
+    class_->guid = iface_class;
+    class_->path = path;
+
+end:
+    LeaveCriticalSection(&setupapi_lock);
+
+    return hr;
+}
+
+static void setupapi_hook_init(void)
+{
+    if (setupapi_initted) {
+        return;
+    }
+
+    hook_table_apply(
+            NULL,
+            "setupapi.dll",
+            setupapi_syms,
+            _countof(setupapi_syms));
+
+    InitializeCriticalSection(&setupapi_lock);
+    setupapi_initted = true;
+}
+
+static HDEVINFO WINAPI my_SetupDiGetClassDevsW(
+        const GUID *ClassGuid,
+        wchar_t *Enumerator,
+        HWND hwndParent,
+        DWORD Flags)
+{
+    struct setupapi_class *class_;
+    HDEVINFO result;
+    size_t i;
+
+    result = next_SetupDiGetClassDevsW(
+            ClassGuid,
+            Enumerator,
+            hwndParent,
+            Flags);
+
+    if (result == INVALID_HANDLE_VALUE || ClassGuid == NULL) {
+        return result;
+    }
+
+    EnterCriticalSection(&setupapi_lock);
+
+    for (i = 0 ; i < setupapi_nclasses ; i++) {
+        class_ = &setupapi_classes[i];
+        if (memcmp(ClassGuid, class_->guid, sizeof(*ClassGuid)) == 0) {
+            class_->cur_handle = result;
+        }
+    }
+
+    LeaveCriticalSection(&setupapi_lock);
+
+    return result;
+}
+
+static BOOL WINAPI my_SetupDiEnumDeviceInterfaces(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVINFO_DATA *DeviceInfoData,
+        const GUID *InterfaceClassGuid,
+        DWORD MemberIndex,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData)
+{
+    const struct setupapi_class *class_;
+    size_t i;
+
+    if (    DeviceInfoSet == INVALID_HANDLE_VALUE ||
+            DeviceInterfaceData == NULL ||
+            DeviceInterfaceData->cbSize != sizeof(*DeviceInterfaceData) ||
+            MemberIndex != 0) {
+        goto pass;
+    }
+
+    EnterCriticalSection(&setupapi_lock);
+
+    for (   i = 0, class_ = NULL ;
+            i < setupapi_nclasses && class_ == NULL ;
+            i++) {
+        if (DeviceInfoSet == setupapi_classes[i].cur_handle) {
+            class_ = &setupapi_classes[i];
+
+            memcpy( &DeviceInterfaceData->InterfaceClassGuid,
+                    class_->guid,
+                    sizeof(GUID));
+            DeviceInterfaceData->Flags = SPINT_ACTIVE;
+            DeviceInterfaceData->Reserved = (ULONG_PTR) class_->path;
+        }
+    }
+
+    LeaveCriticalSection(&setupapi_lock);
+
+    if (class_ == NULL) {
+        goto pass;
+    }
+
+    SetLastError(ERROR_SUCCESS);
+
+    return TRUE;
+
+pass:
+    return next_SetupDiEnumDeviceInterfaces(
+            DeviceInfoSet,
+            DeviceInfoData,
+            InterfaceClassGuid,
+            MemberIndex,
+            DeviceInterfaceData);
+}
+
+static BOOL WINAPI my_SetupDiGetDeviceInterfaceDetailW(
+        HDEVINFO DeviceInfoSet,
+        SP_DEVICE_INTERFACE_DATA *DeviceInterfaceData,
+        SP_DEVICE_INTERFACE_DETAIL_DATA_W *DeviceInterfaceDetailData,
+        DWORD DeviceInterfaceDetailDataSize,
+        DWORD *RequiredSize,
+        SP_DEVINFO_DATA *DeviceInfoData)
+{
+    const wchar_t *wstr;
+    size_t nbytes_wstr;
+    size_t nbytes_total;
+    size_t i;
+    bool match;
+
+    if (DeviceInfoSet == INVALID_HANDLE_VALUE || DeviceInterfaceData == NULL) {
+        goto pass;
+    }
+
+    EnterCriticalSection(&setupapi_lock);
+
+    for (   i = 0, match = false ;
+            i < setupapi_nclasses && !match ;
+            i++) {
+        if (DeviceInfoSet == setupapi_classes[i].cur_handle) {
+            match = true;
+        }
+    }
+
+    LeaveCriticalSection(&setupapi_lock);
+
+    if (!match) {
+        goto pass;
+    }
+
+    wstr = (const wchar_t *) DeviceInterfaceData->Reserved;
+    nbytes_wstr = (wcslen(wstr) + 1) * sizeof(wchar_t);
+    nbytes_total  = offsetof(SP_DEVICE_INTERFACE_DETAIL_DATA_W, DevicePath);
+    nbytes_total += nbytes_wstr;
+
+    if (RequiredSize != NULL) {
+        *RequiredSize = (DWORD) nbytes_total;
+    }
+
+    if (    DeviceInterfaceDetailData == NULL &&
+            DeviceInterfaceDetailDataSize < nbytes_total) {
+        SetLastError(ERROR_INSUFFICIENT_BUFFER);
+
+        return FALSE;
+    }
+
+    if (DeviceInterfaceDetailData->cbSize!=sizeof(*DeviceInterfaceDetailData)) {
+        SetLastError(ERROR_INVALID_PARAMETER);
+
+        return FALSE;
+    }
+
+    memcpy(DeviceInterfaceDetailData->DevicePath, wstr, nbytes_wstr);
+    SetLastError(ERROR_SUCCESS);
+
+    return TRUE;
+
+pass:
+    return next_SetupDiGetDeviceInterfaceDetailW(
+            DeviceInfoSet,
+            DeviceInterfaceData,
+            DeviceInterfaceDetailData,
+            DeviceInterfaceDetailDataSize,
+            RequiredSize,
+            DeviceInfoData);
+}
+
+static BOOL WINAPI my_SetupDiDestroyDeviceInfoList(HDEVINFO DeviceInfoSet)
+{
+    size_t i;
+
+    EnterCriticalSection(&setupapi_lock);
+
+    for (i = 0 ; i < setupapi_nclasses ; i++) {
+        if (setupapi_classes[i].cur_handle == DeviceInfoSet) {
+            setupapi_classes[i].cur_handle = NULL;
+        }
+    }
+
+    LeaveCriticalSection(&setupapi_lock);
+
+    return next_SetupDiDestroyDeviceInfoList(DeviceInfoSet);
+}

util/setupapi.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#include <windows.h>
+
+#include <stddef.h>
+
+HRESULT setupapi_add_phantom_dev(const GUID *iface_class, const wchar_t *path);