Fix KB5015882

ALTaleX531 · Jul 22, 2022 · 93734ef · 93734ef
1 parent 6df2dd2
commit 93734ef
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/TranslucentFlyouts/Win32HookHelper.cpp b/TranslucentFlyouts/Win32HookHelper.cpp
@@ -344,8 +344,7 @@ HRESULT WINAPI TranslucentFlyoutsLib::MyDrawThemeTextEx(
 	            !(pOptions->dwFlags & DTT_CALCRECT) and
 	            !(pOptions->dwFlags & DTT_COMPOSITED)
 	        )
-	    ) and
-		!VerifyCaller(g_hModule)
+	    )
 	)
 	{
 		DTTOPTS Options = *pOptions;
@@ -456,7 +455,7 @@ HRESULT WINAPI TranslucentFlyoutsLib::MyDrawThemeText(
 
 int WINAPI TranslucentFlyoutsLib::MyDrawTextW(
     HDC     hdc,
-    LPCTSTR lpchText,
+    LPCWSTR lpchText,
     int     cchText,
     LPRECT  lprc,
     UINT    format
@@ -494,12 +493,13 @@ int WINAPI TranslucentFlyoutsLib::MyDrawTextW(
 	return nResult;
 Default:
 	nResult =
-	    DrawTextWHook.OldFunction<decltype(MyDrawTextW)>(
+	    DrawTextExWHook.OldFunction<decltype(MyDrawTextExW)>(
 	        hdc,
-	        lpchText,
+	        (LPWSTR)lpchText,
 	        cchText,
 	        lprc,
-	        format
+	        format,
+	        nullptr
 	    );
 	return nResult;
 }
@@ -523,7 +523,8 @@ int WINAPI TranslucentFlyoutsLib::MyDrawTextExW(
 	    lpdtp or
 	    !(GetCurrentFlyoutPolicy() != Null) or
 	    VerifyCaller(TEXT("Uxtheme")) or
-	    VerifyCaller(TEXT("MToolExtend"))
+	    VerifyCaller(TEXT("MToolExtend")) or
+	    VerifyCaller(g_hModule)
 	)
 	{
 		goto Default;

diff --git a/TranslucentFlyouts/Win32HookHelper.h b/TranslucentFlyouts/Win32HookHelper.h
@@ -36,7 +36,7 @@ namespace TranslucentFlyoutsLib
 	);
 	extern int WINAPI MyDrawTextW(
 	    HDC     hdc,
-	    LPCTSTR lpchText,
+		LPCWSTR lpchText,
 	    int     cchText,
 	    LPRECT  lprc,
 	    UINT    format