From 798446ec79d8a4fec2373e94d275a39169b7b1ff Mon Sep 17 00:00:00 2001
From: Felix <me@mbilker.us>
Date: Thu, 11 Nov 2021 12:51:44 +0000
Subject: [PATCH] gfxhook/d3d9.c: Add Direct3D 9 Ex hook

---
 gfxhook/d3d9.c | 114 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 107 insertions(+), 7 deletions(-)

diff --git a/gfxhook/d3d9.c b/gfxhook/d3d9.c
index 4c391781..c8de146a 100644
--- a/gfxhook/d3d9.c
+++ b/gfxhook/d3d9.c
@@ -17,8 +17,9 @@
 #include "util/dprintf.h"
 
 typedef IDirect3D9 * (WINAPI *Direct3DCreate9_t)(UINT sdk_ver);
+typedef HRESULT (WINAPI *Direct3DCreate9Ex_t)(UINT sdk_ver, IDirect3D9Ex **d3d9ex);
 
-static HRESULT STDMETHODCALLTYPE my_CreateDevice(
+static HRESULT STDMETHODCALLTYPE my_IDirect3D9_CreateDevice(
         IDirect3D9 *self,
         UINT adapter,
         D3DDEVTYPE type,
@@ -26,15 +27,28 @@ static HRESULT STDMETHODCALLTYPE my_CreateDevice(
         DWORD flags,
         D3DPRESENT_PARAMETERS *pp,
         IDirect3DDevice9 **pdev);
+static HRESULT STDMETHODCALLTYPE my_IDirect3D9Ex_CreateDevice(
+        IDirect3D9Ex *self,
+        UINT adapter,
+        D3DDEVTYPE type,
+        HWND hwnd,
+        DWORD flags,
+        D3DPRESENT_PARAMETERS *pp,
+        IDirect3DDevice9 **pdev);
 
 static struct gfx_config gfx_config;
 static Direct3DCreate9_t next_Direct3DCreate9;
+static Direct3DCreate9Ex_t next_Direct3DCreate9Ex;
 
 static const struct hook_symbol gfx_hooks[] = {
     {
         .name   = "Direct3DCreate9",
         .patch  = Direct3DCreate9,
-        .link   = (void **) &next_Direct3DCreate9
+        .link   = (void **) &next_Direct3DCreate9,
+    }, {
+        .name   = "Direct3DCreate9Ex",
+        .patch  = Direct3DCreate9Ex,
+        .link   = (void **) &next_Direct3DCreate9Ex,
     },
 };
 
@@ -51,7 +65,7 @@ void gfx_d3d9_hook_init(const struct gfx_config *cfg, HINSTANCE self)
     memcpy(&gfx_config, cfg, sizeof(*cfg));
     hook_table_apply(NULL, "d3d9.dll", gfx_hooks, _countof(gfx_hooks));
 
-    if (next_Direct3DCreate9 == NULL) {
+    if (next_Direct3DCreate9 == NULL || next_Direct3DCreate9Ex == NULL) {
         d3d9 = LoadLibraryW(L"d3d9.dll");
 
         if (d3d9 == NULL) {
@@ -60,12 +74,20 @@ void gfx_d3d9_hook_init(const struct gfx_config *cfg, HINSTANCE self)
             goto fail;
         }
 
-        next_Direct3DCreate9 = (Direct3DCreate9_t) GetProcAddress(d3d9, "Direct3DCreate9");
+        if (next_Direct3DCreate9 == NULL) {
+            next_Direct3DCreate9 = (Direct3DCreate9_t) GetProcAddress(d3d9, "Direct3DCreate9");
+        }
+        if (next_Direct3DCreate9Ex == NULL) {
+            next_Direct3DCreate9Ex = (Direct3DCreate9Ex_t) GetProcAddress(d3d9, "Direct3DCreate9Ex");
+        }
 
         if (next_Direct3DCreate9 == NULL) {
             dprintf("Gfx: Direct3DCreate9 not found in loaded d3d9.dll\n");
 
-            FreeLibrary(d3d9);
+            goto fail;
+        }
+        if (next_Direct3DCreate9Ex == NULL) {
+            dprintf("Gfx: Direct3DCreate9Ex not found in loaded d3d9.dll\n");
 
             goto fail;
         }
@@ -75,7 +97,13 @@ void gfx_d3d9_hook_init(const struct gfx_config *cfg, HINSTANCE self)
         dll_hook_push(self, L"d3d9.dll");
     }
 
+    return;
+
 fail:
+    if (d3d9 != NULL) {
+        FreeLibrary(d3d9);
+    }
+
     return;
 }
 
@@ -88,6 +116,8 @@ IDirect3D9 * WINAPI Direct3DCreate9(UINT sdk_ver)
 
     dprintf("Gfx: Direct3DCreate9 hook hit\n");
 
+    api = NULL;
+
     if (next_Direct3DCreate9 == NULL) {
         dprintf("Gfx: next_Direct3DCreate9 == NULL\n");
 
@@ -111,7 +141,7 @@ IDirect3D9 * WINAPI Direct3DCreate9(UINT sdk_ver)
     }
 
     vtbl = proxy->vptr;
-    vtbl->CreateDevice = my_CreateDevice;
+    vtbl->CreateDevice = my_IDirect3D9_CreateDevice;
 
     return (IDirect3D9 *) proxy;
 
@@ -123,7 +153,56 @@ fail:
     return NULL;
 }
 
-static HRESULT STDMETHODCALLTYPE my_CreateDevice(
+HRESULT WINAPI Direct3DCreate9Ex(UINT sdk_ver, IDirect3D9Ex **d3d9ex)
+{
+    struct com_proxy *proxy;
+    IDirect3D9ExVtbl *vtbl;
+    IDirect3D9Ex *api;
+    HRESULT hr;
+
+    dprintf("Gfx: Direct3DCreate9Ex hook hit\n");
+
+    api = NULL;
+
+    if (next_Direct3DCreate9Ex == NULL) {
+        dprintf("Gfx: next_Direct3DCreate9Ex == NULL\n");
+
+        goto fail;
+    }
+
+    hr = next_Direct3DCreate9Ex(sdk_ver, d3d9ex);
+
+    if (FAILED(hr)) {
+        dprintf("Gfx: next_Direct3DCreate9Ex returned %x\n", (int) hr);
+
+        goto fail;
+    }
+
+    api = *d3d9ex;
+    hr = com_proxy_wrap(&proxy, api, sizeof(*api->lpVtbl));
+
+    if (FAILED(hr)) {
+        dprintf("Gfx: com_proxy_wrap returned %x\n", (int) hr);
+
+        goto fail;
+    }
+
+    vtbl = proxy->vptr;
+    vtbl->CreateDevice = my_IDirect3D9Ex_CreateDevice;
+
+    *d3d9ex = (IDirect3D9Ex *) proxy;
+
+    return S_OK;
+
+fail:
+    if (api != NULL) {
+        IDirect3D9Ex_Release(api);
+    }
+
+    return hr;
+}
+
+static HRESULT STDMETHODCALLTYPE my_IDirect3D9_CreateDevice(
         IDirect3D9 *self,
         UINT adapter,
         D3DDEVTYPE type,
@@ -153,3 +232,24 @@ static HRESULT STDMETHODCALLTYPE my_CreateDevice(
 
     return IDirect3D9_CreateDevice(real, gfx_config.monitor, type, hwnd, flags, pp, pdev);
 }
+
+static HRESULT STDMETHODCALLTYPE my_IDirect3D9Ex_CreateDevice(
+        IDirect3D9Ex *self,
+        UINT adapter,
+        D3DDEVTYPE type,
+        HWND hwnd,
+        DWORD flags,
+        D3DPRESENT_PARAMETERS *pp,
+        IDirect3DDevice9 **pdev)
+{
+    dprintf("Gfx: IDirect3D9Ex::CreateDevice hook forwarding to my_IDirect3D9_CreateDevice\n");
+
+    return my_IDirect3D9_CreateDevice(
+            (IDirect3D9 *) self,
+            adapter,
+            type,
+            hwnd,
+            flags,
+            pp,
+            pdev);
+}