From c0b20d5a1379c1c942246b71e7a0d7defae4bafa Mon Sep 17 00:00:00 2001 From: Tau Date: Wed, 7 Nov 2018 16:44:11 -0500 Subject: [PATCH] util/setupapi.c: Add hook for SETUPAPI lib --- util/meson.build | 2 + util/setupapi.c | 321 +++++++++++++++++++++++++++++++++++++++++++++++ util/setupapi.h | 7 ++ 3 files changed, 330 insertions(+) create mode 100644 util/setupapi.c create mode 100644 util/setupapi.h diff --git a/util/meson.build b/util/meson.build index 0cedae5..9eab6d3 100644 --- a/util/meson.build +++ b/util/meson.build @@ -9,5 +9,7 @@ util_lib = static_library( sources : [ 'dprintf.c', 'dprintf.h', + 'setupapi.c', + 'setupapi.h', ], ) diff --git a/util/setupapi.c b/util/setupapi.c new file mode 100644 index 0000000..e7c2b46 --- /dev/null +++ b/util/setupapi.c @@ -0,0 +1,321 @@ +#include +#include + +#include +#include +#include +#include + +#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); +} diff --git a/util/setupapi.h b/util/setupapi.h new file mode 100644 index 0000000..a09520c --- /dev/null +++ b/util/setupapi.h @@ -0,0 +1,7 @@ +#pragma once + +#include + +#include + +HRESULT setupapi_add_phantom_dev(const GUID *iface_class, const wchar_t *path);