Windows 11 support

ledoge · Oct 19, 2021 · 10796ed · 10796ed
1 parent 928ffda
commit 10796ed
Showing 1 changed file with 12 additions and 13 deletions.
diff --git a/dwm_lut.c b/dwm_lut.c
@@ -14,21 +14,20 @@
 #define MAX_LUTS 32
 
 #define RELEASE_IF_NOT_NULL(x) { if (x != NULL) { x->lpVtbl->Release(x); } }
-#define STRINGIFY(x) #x
-#define _STRINGIFY(x) STRINGIFY(x)
+#define _STRINGIFY(x) #x
+#define STRINGIFY(x) _STRINGIFY(x)
 
-const unsigned char COverlayContext_Present_bytes[] = {0x48, 0x89, 0x5c, 0x24, 0x08, 0x48, 0x89, 0x74, 0x24, 0x10, 0x57, 0x48, 0x83, 0xec, 0x40, 0x48, 0x8b, 0xb1, 0x20, 0x2c, 0x00, 0x00, 0x45, 0x8b, 0xd0, 0x48, 0x8b, 0xfa, 0x48, 0x8b, 0xd9, 0x48, 0x85, 0xf6, 0x0f, 0x85};
-const int IOverlaySwapChain_IDXGISwapChain_offset = -0x118;
+const unsigned char COverlayContext_Present_bytes[] = {0x48, 0x33, 0xc4, 0x48, 0x89, 0x44, 0x24, 0x50, 0x48, 0x8b, 0xb1, 0xa0, 0x2b, 0x00, 0x00, 0x48, 0x8b, 0xfa, 0x48, 0x8b, 0xd9, 0x48, 0x85, 0xf6};
+const int IOverlaySwapChain_IDXGISwapChain_offset = -0x148;
 
-const unsigned char COverlayContext_IsCandidateDirectFlipCompatbile_bytes[] = {0x48, 0x89, 0x7c, 0x24, 0x20, 0x55, 0x41, 0x54, 0x41, 0x55, 0x41, 0x56, 0x41, 0x57, 0x48, 0x8b, 0xec, 0x48, 0x83, 0xec, 0x40};
-const unsigned char COverlayContext_OverlaysEnabled_bytes[] = {0x75, 0x04, 0x32, 0xc0, 0xc3, 0xcc, 0x83, 0x79, 0x30, 0x01, 0x0f, 0x97, 0xc0, 0xc3};
+const unsigned char COverlayContext_IsCandidateDirectFlipCompatbile_bytes[] = {0x40, 0x55, 0x53, 0x56, 0x57, 0x41, 0x54, 0x41, 0x55, 0x41, 0x56, 0x41, 0x57, 0x48, 0x8b, 0xec, 0x48, 0x83, 0xec, 0x68};
+const unsigned char COverlayContext_OverlaysEnabled_bytes[] = {0x74, 0x09, 0x83, 0x79, 0x2c, 0x01, 0x0f, 0x97, 0xc0, 0xc3, 0xcc, 0x32, 0xc0, 0xc3};
 
-const int COverlayContext_CLegacyRenderTarget_offset = -0x150;
-const int CLegacyRenderTarget_DeviceClipBox_offset = 0x30;
+const int COverlayContext_DeviceClipBox_offset = 0x462c;
 
 #pragma push_macro("bool")
 #undef bool
-char shaders[] = _STRINGIFY(
+char shaders[] = STRINGIFY(
         struct VS_INPUT {
             float2 pos: POSITION;
             float2 tex: TEXCOORD;
@@ -280,9 +279,9 @@ void RemoveLUTActiveTarget(void *address) {
 }
 
 lutData *GetLUTDataFromCOverlayContext(void *context) {
-    struct tagRECT *rect = (struct tagRECT *) ((unsigned char *) context + COverlayContext_CLegacyRenderTarget_offset + CLegacyRenderTarget_DeviceClipBox_offset);
+    float* rect = (float *) ((unsigned char *) context + COverlayContext_DeviceClipBox_offset);
     for (int i = 0; i < numLuts; i++) {
-        if (luts[i].left == rect->left && luts[i].top == rect->top) {
+        if (luts[i].left == (int) rect[0] && luts[i].top == (int) rect[1]) {
             return &luts[i];
         }
     }
@@ -561,13 +560,13 @@ BOOL APIENTRY DllMain(HMODULE hModule, DWORD fdwReason, LPVOID lpReserved) {
             for (int i = 0; i <= moduleInfo.SizeOfImage - sizeof(COverlayContext_Present_bytes); i++) {
                 unsigned char *address = (unsigned char *) dwmcore + i;
                 if (!COverlayContext_Present_orig && !memcmp(address, COverlayContext_Present_bytes, sizeof(COverlayContext_Present_bytes))) {
-                    COverlayContext_Present_orig = (COverlayContext_Present_t *) address;
+                    COverlayContext_Present_orig = (COverlayContext_Present_t *) (address - 0xf);
                     COverlayContext_Present_real_orig = COverlayContext_Present_orig;
                 } else if (!COverlayContext_IsCandidateDirectFlipCompatbile_orig && !memcmp(address, COverlayContext_IsCandidateDirectFlipCompatbile_bytes, sizeof(COverlayContext_IsCandidateDirectFlipCompatbile_bytes))) {
                     static int found = 0;
                     found++;
                     if (found == 2) {
-                        COverlayContext_IsCandidateDirectFlipCompatbile_orig = (COverlayContext_IsCandidateDirectFlipCompatbile_t *) (address - 0xa);
+                        COverlayContext_IsCandidateDirectFlipCompatbile_orig = (COverlayContext_IsCandidateDirectFlipCompatbile_t *) address;
                     }
                 } else if (!COverlayContext_OverlaysEnabled_orig && !memcmp(address, COverlayContext_OverlaysEnabled_bytes, sizeof(COverlayContext_OverlaysEnabled_bytes))) {
                     COverlayContext_OverlaysEnabled_orig = (COverlayContext_OverlaysEnabled_t *) (address - 0x7);