mirror of
https://github.com/djhackersdev/bemanitools.git
synced 2026-04-12 20:55:49 -05:00
93e0a090bc
Boils down to: - Include headers - Reduce boiler plate with helpers - Swap out explicit usages with core API layer and ensure the right API is configured beforehand
104 lines
2.2 KiB
C
104 lines
2.2 KiB
C
#include <windows.h>
|
|
|
|
#define DIRECTINPUT_VERSION 0x0800
|
|
#include <dinput.h>
|
|
|
|
#include <stdbool.h>
|
|
|
|
#include "core/log.h"
|
|
|
|
#include "hook/com-proxy.h"
|
|
#include "hook/pe.h"
|
|
#include "hook/table.h"
|
|
|
|
#include "util/defs.h"
|
|
|
|
static HRESULT STDCALL my_DirectInput8Create(
|
|
HINSTANCE hinst,
|
|
DWORD dwVersion,
|
|
REFIID riidltf,
|
|
LPVOID *ppvOut,
|
|
LPVOID punkOuter);
|
|
|
|
static HRESULT(STDCALL *real_DirectInput8Create)(
|
|
HINSTANCE hinst,
|
|
DWORD dwVersion,
|
|
REFIID riidltf,
|
|
LPVOID *ppvOut,
|
|
LPVOID punkOuter);
|
|
|
|
static const struct hook_symbol dinput_syms[] = {
|
|
{
|
|
.name = "DirectInput8Create",
|
|
.patch = my_DirectInput8Create,
|
|
.link = (void **) &real_DirectInput8Create,
|
|
},
|
|
};
|
|
|
|
static HRESULT STDCALL my_CreateDevice(
|
|
IDirectInput8W *self,
|
|
REFGUID rguid,
|
|
LPDIRECTINPUTDEVICE8W *lplpDirectInputDevice,
|
|
LPUNKNOWN pUnkOuter)
|
|
{
|
|
log_misc("IDirectInput8::CreateDevice hook hit");
|
|
return DIERR_NOINTERFACE;
|
|
}
|
|
|
|
static HRESULT STDCALL my_EnumDevices(
|
|
IDirectInput8W *self,
|
|
DWORD dwDevType,
|
|
LPDIENUMDEVICESCALLBACKW lpCallback,
|
|
LPVOID pvRef,
|
|
DWORD dwFlags)
|
|
{
|
|
log_misc("IDirectInput8::EnumDevices hook hit");
|
|
|
|
return DI_OK;
|
|
}
|
|
|
|
static HRESULT STDCALL my_DirectInput8Create(
|
|
HINSTANCE hinst,
|
|
DWORD dwVersion,
|
|
REFIID riidltf,
|
|
LPVOID *ppvOut,
|
|
LPVOID punkOuter)
|
|
{
|
|
IDirectInput8W *api;
|
|
IDirectInput8WVtbl *api_vtbl;
|
|
struct com_proxy *api_proxy;
|
|
HRESULT res;
|
|
|
|
log_info("DirectInput8Create hook hit");
|
|
|
|
res = real_DirectInput8Create(
|
|
hinst, dwVersion, riidltf, (LPVOID *) &api, punkOuter);
|
|
|
|
if (res != DI_OK) {
|
|
return res;
|
|
}
|
|
|
|
res = com_proxy_wrap(&api_proxy, api, sizeof(*api->lpVtbl));
|
|
|
|
if (res != S_OK) {
|
|
log_warning("Wrapping com proxy failed: %08lx", res);
|
|
return res;
|
|
}
|
|
|
|
api_vtbl = api_proxy->vptr;
|
|
|
|
api_vtbl->EnumDevices = my_EnumDevices;
|
|
api_vtbl->CreateDevice = my_CreateDevice;
|
|
|
|
*(IDirectInput8W **) ppvOut = (IDirectInput8W *) api_proxy;
|
|
|
|
return res;
|
|
}
|
|
|
|
void dinput_init(HMODULE target)
|
|
{
|
|
hook_table_apply(target, "dinput8.dll", dinput_syms, lengthof(dinput_syms));
|
|
|
|
log_info("Inserted dinput hooks into %p", target);
|
|
}
|