segatools/idzhook/zinput.c

180 lines
3.9 KiB
C
Raw Normal View History

#include <windows.h>
#include <dinput.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
2019-11-03 21:12:58 +00:00
#include "idzhook/config.h"
#include "idzhook/zinput.h"
#include "hook/table.h"
#include "util/dprintf.h"
2019-11-03 21:12:58 +00:00
HRESULT WINAPI hook_DirectInput8Create(
HINSTANCE hinst,
DWORD dwVersion,
REFIID riidltf,
LPVOID *ppvOut,
LPUNKNOWN punkOuter);
static unsigned long WINAPI my_AddRef(IUnknown *self);
static unsigned long WINAPI my_Release(IUnknown *self);
static HRESULT WINAPI my_CreateDevice(
IDirectInput8W *self,
REFGUID rguid,
LPDIRECTINPUTDEVICE8W * lplpDirectInputDevice,
LPUNKNOWN pUnkOuter);
static HRESULT WINAPI my_EnumDevices(
IDirectInput8W *self,
DWORD dwDevType,
LPDIENUMDEVICESCALLBACKW lpCallback,
LPVOID pvRef,
DWORD dwFlags);
static HRESULT WINAPI my_SetDataFormat(
IDirectInputDevice8W *self,
LPCDIDATAFORMAT lpdf);
static HRESULT WINAPI my_SetCooperativeLevel(
IDirectInputDevice8W *self,
HWND hwnd,
DWORD flags);
static HRESULT WINAPI my_Acquire(IDirectInputDevice8W *self);
static HRESULT WINAPI my_GetDeviceState(
IDirectInputDevice8W *self,
DWORD cbData,
LPVOID lpvData);
static const IDirectInput8WVtbl api_vtbl = {
.AddRef = (void *) my_AddRef,
.Release = (void *) my_Release,
.CreateDevice = my_CreateDevice,
.EnumDevices = my_EnumDevices,
};
static const IDirectInput8W api = { (void *) &api_vtbl };
static const IDirectInputDevice8WVtbl dev_vtbl = {
.AddRef = (void *) my_AddRef,
.Release = (void *) my_Release,
.SetDataFormat = my_SetDataFormat,
.SetCooperativeLevel= my_SetCooperativeLevel,
.Acquire = my_Acquire,
.Unacquire = my_Acquire, // not a c&p error
.GetDeviceState = my_GetDeviceState,
};
static const IDirectInputDevice8W dev = { (void *) &dev_vtbl };
2019-11-03 21:12:58 +00:00
static const struct hook_symbol zinput_hook_syms[] = {
{
.name = "DirectInput8Create",
.patch = hook_DirectInput8Create,
}
};
HRESULT zinput_hook_init(struct zinput_config *cfg)
{
assert(cfg != NULL);
if (!cfg->enable) {
return S_FALSE;
}
hook_table_apply(
NULL,
"dinput8.dll",
zinput_hook_syms,
_countof(zinput_hook_syms));
return S_OK;
}
HRESULT WINAPI hook_DirectInput8Create(
HINSTANCE hinst,
DWORD dwVersion,
REFIID riidltf,
LPVOID *ppvOut,
LPUNKNOWN punkOuter)
{
2019-11-03 21:12:58 +00:00
dprintf("ZInput: Blocking built-in DirectInput support\n");
*ppvOut = (void *) &api;
return S_OK;
}
static unsigned long WINAPI my_AddRef(IUnknown *self)
{
return 1;
}
static unsigned long WINAPI my_Release(IUnknown *self)
{
return 1;
}
static HRESULT WINAPI my_CreateDevice(
IDirectInput8W *self,
REFGUID rguid,
LPDIRECTINPUTDEVICE8W *lplpDirectInputDevice,
LPUNKNOWN pUnkOuter)
{
dprintf("%s\n", __func__);
*lplpDirectInputDevice = (void *) &dev;
return S_OK;
}
static HRESULT WINAPI my_EnumDevices(
IDirectInput8W *self,
DWORD dwDevType,
LPDIENUMDEVICESCALLBACKW lpCallback,
LPVOID pvRef,
DWORD dwFlags)
{
dprintf("%s\n", __func__);
return S_OK;
}
static HRESULT WINAPI my_SetDataFormat(
IDirectInputDevice8W *self,
LPCDIDATAFORMAT lpdf)
{
dprintf("%s\n", __func__);
return S_OK;
}
static HRESULT WINAPI my_SetCooperativeLevel(
IDirectInputDevice8W *self,
HWND hwnd,
DWORD flags)
{
dprintf("%s\n", __func__);
return S_OK;
}
static HRESULT WINAPI my_Acquire(IDirectInputDevice8W *self)
{
return S_OK;
}
static HRESULT WINAPI my_GetDeviceState(
IDirectInputDevice8W *self,
DWORD cbData,
LPVOID lpvData)
{
memset(lpvData, 0, cbData);
return S_OK;
}