1 files changed, 67 insertions, 0 deletions
diff --git a/dwm/dxhk.cpp b/dwm/dxhk.cpp
new file mode 100644
index 0000000..4de113f
--- /dev/null
+++ b/dwm/dxhk.cpp
@@ -0,0 +1,67 @@
+#include "dxhk.h"
+#include "inter.h"
+
+HOOK_DATA* hk;
+D3D11_RENDERDATA* dx11;
+
+U64 __fastcall hk_multiplane_draw( 
+  void* dxdev, 
+  IDXGISwapChain* swpchain, 
+  U32 a3, 
+  U32 a4, 
+  U32 a5, 
+  U64 a6, 
+  U64 a7, 
+  U32 a8 
+) {
+  static decltype( &hk_multiplane_draw ) original = (decltype( &hk_multiplane_draw ))hk->multiplane_o;
+  
+  U64 ret = original( dxdev, swpchain, a3, a4, a5, a6, a7, a8 );
+  return ret;
+}
+
+U64 __fastcall hk_present( 
+  void* dxdev, 
+  IDXGISwapChain* swpchain, 
+  U32 a3, 
+  U32 a4, 
+  const RECT* rect, 
+  U64 a6, 
+  void* a7, 
+  U32 a8, 
+  void* rsrc, 
+  U32 a10 
+) {
+  static decltype( &hk_present ) original = (decltype( &hk_present ))hk->present_o;
+
+  U64 ret = original( dxdev, swpchain, a3, a4, rect, a6, a7, a8, rsrc, a10 );
+  return ret;
+}
+
+U64 __stdcall hk_wndproc( HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam ) {
+  U64 ret = CallWindowProcA( (WNDPROC)hk->wndproc_o, hwnd, msg, wparam, lparam );
+
+
+  return ret;
+}
+
+HOOK_DATA* dxhk_init() {
+  HOOK_DATA* hkd = new HOOK_DATA;
+
+  U64 dxdev = u_find_pattern( (U64)GetModuleHandleA( "d2d1.dll" ), "48 8D 05 ? ? ? ? 33 ED 48 8D 71 08" );
+  if( !dxdev ) {
+    delete hkd;
+    return 0;
+  }
+
+  dxdev = dxdev + *(U64*)( dxdev + 3 ) + 7;
+  VMT vmt( *(U64*)dxdev );
+
+  hkd->multiplane_o = vmt.set( 7, (U64)&hk_multiplane_draw );
+  hkd->present_o = vmt.set( 6, (U64)&hk_present );
+  hkd->wndproc_o = SetWindowLongPtrA( FindWindowA( "Progman", "Program Manager" ), GWLP_WNDPROC, (LONG_PTR)&hk_wndproc );
+
+  dx11 = d3d11_init();
+
+  return hkd;
+}
+\ No newline at end of file