DDrawCompat/DDrawCompat/CompatGdiPaintHandlers.cpp

#include "CompatGdi.h"
#include "CompatGdiDc.h"
#include "CompatGdiPaintHandlers.h"
#include "CompatGdiScrollBar.h"
#include "CompatGdiTitleBar.h"

#include <detours.h>

namespace
{
	LRESULT WINAPI defWindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam,
		WNDPROC origDefWindowProc, const char* funcName);
	LRESULT WINAPI eraseBackgroundProc(
		HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam, WNDPROC origWndProc, const char* wndProcName);
	LRESULT onEraseBackground(HWND hwnd, HDC dc, WNDPROC origWndProc);
	LRESULT onMenuPaint(HWND hwnd, WNDPROC origWndProc);
	LRESULT onNcPaint(HWND hwnd, WPARAM wParam, WNDPROC origWndProc);
	LRESULT onPaint(HWND hwnd, WNDPROC origWndProc);
	LRESULT onPrint(HWND hwnd, UINT msg, HDC dc, LONG flags, WNDPROC origWndProc);

	WNDPROC g_origEditWndProc = nullptr;
	WNDPROC g_origListBoxWndProc = nullptr;
	WNDPROC g_origMenuWndProc = nullptr;
	WNDPROC g_origScrollBarWndProc = nullptr;

	LRESULT WINAPI defDlgProcA(HWND hdlg, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return defWindowProc(hdlg, msg, wParam, lParam, CALL_ORIG_GDI(DefDlgProcA), "defDlgProcA");
	}

	LRESULT WINAPI defDlgProcW(HWND hdlg, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return defWindowProc(hdlg, msg, wParam, lParam, CALL_ORIG_GDI(DefDlgProcW), "defDlgProcW");
	}

	LRESULT WINAPI defWindowProc(
		HWND hwnd,
		UINT msg,
		WPARAM wParam,
		LPARAM lParam,
		WNDPROC origDefWindowProc,
		const char* funcName)
	{
		Compat::LogEnter(funcName, hwnd, msg, wParam, lParam);
		LRESULT result = 0;

		switch (msg)
		{
		case WM_ERASEBKGND:
			result = onEraseBackground(hwnd, reinterpret_cast<HDC>(wParam), origDefWindowProc);
			break;

		case WM_NCPAINT:
			result = onNcPaint(hwnd, wParam, origDefWindowProc);
			break;

		case WM_PRINT:
		case WM_PRINTCLIENT:
			result = onPrint(hwnd, msg, reinterpret_cast<HDC>(wParam), lParam, origDefWindowProc);
			break;

		default:
			result = origDefWindowProc(hwnd, msg, wParam, lParam);
			break;
		}

		Compat::LogLeave(funcName, hwnd, msg, wParam, lParam) << result;
		return result;
	}

	LRESULT WINAPI defWindowProcA(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return defWindowProc(hwnd, msg, wParam, lParam, CALL_ORIG_GDI(DefWindowProcA), "defWindowProcA");
	}

	LRESULT WINAPI defWindowProcW(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return defWindowProc(hwnd, msg, wParam, lParam, CALL_ORIG_GDI(DefWindowProcW), "defWindowProcW");
	}

	LRESULT WINAPI editWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return eraseBackgroundProc(hwnd, msg, wParam, lParam, g_origEditWndProc, "editWndProc");
	}

	LRESULT WINAPI eraseBackgroundProc(
		HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam, WNDPROC origWndProc, const char* wndProcName)
	{
		Compat::LogEnter(wndProcName, hwnd, msg, wParam, lParam);

		LPARAM result = 0;
		if (WM_ERASEBKGND == msg)
		{
			result = onEraseBackground(hwnd, reinterpret_cast<HDC>(wParam), origWndProc);
		}
		else
		{
			result = origWndProc(hwnd, msg, wParam, lParam);
		}

		Compat::LogLeave(wndProcName, hwnd, msg, wParam, lParam) << result;
		return result;
	}

	LRESULT WINAPI listBoxWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		return eraseBackgroundProc(hwnd, msg, wParam, lParam, g_origListBoxWndProc, "listBoxWndProc");
	}

	LRESULT WINAPI menuWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		Compat::LogEnter("menuWndProc", hwnd, msg, wParam, lParam);
		LRESULT result = 0;

		switch (msg)
		{
		case WM_PAINT:
			result = onMenuPaint(hwnd, g_origMenuWndProc);
			break;

		default:
			result = g_origMenuWndProc(hwnd, msg, wParam, lParam);
			break;
		}

		Compat::LogLeave("menuWndProc", hwnd, msg, wParam, lParam) << result;
		return result;
	}

	LRESULT onEraseBackground(HWND hwnd, HDC dc, WNDPROC origWndProc)
	{
		if (!hwnd || !CompatGdi::beginGdiRendering())
		{
			return origWndProc(hwnd, WM_ERASEBKGND, reinterpret_cast<WPARAM>(dc) , 0);
		}

		LRESULT result = 0;
		HDC compatDc = CompatGdiDc::getDc(dc);
		if (compatDc)
		{
			result = origWndProc(hwnd, WM_ERASEBKGND, reinterpret_cast<WPARAM>(compatDc), 0);
			CompatGdiDc::releaseDc(dc);
		}
		else
		{
			result = origWndProc(hwnd, WM_ERASEBKGND, reinterpret_cast<WPARAM>(dc), 0);
		}

		CompatGdi::endGdiRendering();
		return result;
	}

	LRESULT onNcPaint(HWND hwnd, WPARAM wParam, WNDPROC origWndProc)
	{
		if (!hwnd || !CompatGdi::beginGdiRendering())
		{
			return origWndProc(hwnd, WM_NCPAINT, wParam, 0);
		}

		HDC windowDc = GetWindowDC(hwnd);
		HDC compatDc = CompatGdiDc::getDc(windowDc);

		if (compatDc)
		{
			CompatGdi::TitleBar titleBar(hwnd, compatDc);
			titleBar.drawAll();
			titleBar.excludeFromClipRegion();

			CompatGdi::ScrollBar scrollBar(hwnd, compatDc);
			scrollBar.drawAll();
			scrollBar.excludeFromClipRegion();

			SendMessage(hwnd, WM_PRINT, reinterpret_cast<WPARAM>(compatDc), PRF_NONCLIENT);

			CompatGdiDc::releaseDc(windowDc);
		}

		ReleaseDC(hwnd, windowDc);
		CompatGdi::endGdiRendering();
		return 0;
	}

	LRESULT onMenuPaint(HWND hwnd, WNDPROC origWndProc)
	{
		if (!hwnd || !CompatGdi::beginGdiRendering())
		{
			return origWndProc(hwnd, WM_PAINT, 0, 0);
		}

		HDC dc = GetWindowDC(hwnd);
		HDC compatDc = CompatGdiDc::getDc(dc);
		if (compatDc)
		{
			origWndProc(hwnd, WM_PRINT, reinterpret_cast<WPARAM>(compatDc),
				PRF_NONCLIENT | PRF_ERASEBKGND | PRF_CLIENT);
			ValidateRect(hwnd, nullptr);
			CompatGdiDc::releaseDc(dc);
		}
		else
		{
			origWndProc(hwnd, WM_PAINT, 0, 0);
		}

		ReleaseDC(hwnd, dc);
		CompatGdi::endGdiRendering();
		return 0;
	}

	LRESULT onPaint(HWND hwnd, WNDPROC origWndProc)
	{
		if (!hwnd || !CompatGdi::beginGdiRendering())
		{
			return origWndProc(hwnd, WM_PAINT, 0, 0);
		}

		PAINTSTRUCT paint = {};
		HDC dc = BeginPaint(hwnd, &paint);
		HDC compatDc = CompatGdiDc::getDc(dc);

		if (compatDc)
		{
			origWndProc(hwnd, WM_PRINTCLIENT, reinterpret_cast<WPARAM>(compatDc), PRF_CLIENT);
			CompatGdiDc::releaseDc(dc);
		}
		else
		{
			origWndProc(hwnd, WM_PRINTCLIENT, reinterpret_cast<WPARAM>(dc), PRF_CLIENT);
		}

		EndPaint(hwnd, &paint);

		CompatGdi::endGdiRendering();
		return 0;
	}

	LRESULT onPrint(HWND hwnd, UINT msg, HDC dc, LONG flags, WNDPROC origWndProc)
	{
		if (!CompatGdi::beginGdiRendering())
		{
			return origWndProc(hwnd, msg, reinterpret_cast<WPARAM>(dc), flags);
		}

		LRESULT result = 0;
		HDC compatDc = CompatGdiDc::getDc(dc);
		if (compatDc)
		{
			result = origWndProc(hwnd, msg, reinterpret_cast<WPARAM>(compatDc), flags);
			CompatGdiDc::releaseDc(dc);
		}
		else
		{
			result = origWndProc(hwnd, msg, reinterpret_cast<WPARAM>(dc), flags);
		}

		CompatGdi::endGdiRendering();
		return result;
	}

	LRESULT WINAPI scrollBarWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
	{
		Compat::LogEnter("scrollBarWndProc", hwnd, msg, wParam, lParam);
		LRESULT result = 0;

		switch (msg)
		{
		case WM_PAINT:
			result = onPaint(hwnd, g_origScrollBarWndProc);
			break;

		case WM_SETCURSOR:
			if (GetWindowLong(hwnd, GWL_STYLE) & (SBS_SIZEBOX | SBS_SIZEGRIP))
			{
				SetCursor(LoadCursor(nullptr, IDC_SIZENWSE));
			}
			result = TRUE;
			break;

		default:
			result = g_origScrollBarWndProc(hwnd, msg, wParam, lParam);
			break;
		}

		Compat::LogLeave("scrollBarWndProc", hwnd, msg, wParam, lParam) << result;
		return result;
	}
}

namespace CompatGdiPaintHandlers
{
	void installHooks()
	{
		CompatGdi::hookWndProc("Edit", g_origEditWndProc, &editWndProc);
		CompatGdi::hookWndProc("ListBox", g_origListBoxWndProc, &listBoxWndProc);
		CompatGdi::hookWndProc("#32768", g_origMenuWndProc, &menuWndProc);
		CompatGdi::hookWndProc("ScrollBar", g_origScrollBarWndProc, &scrollBarWndProc);

		DetourTransactionBegin();
		HOOK_GDI_FUNCTION(user32, DefWindowProcA, defWindowProcA);
		HOOK_GDI_FUNCTION(user32, DefWindowProcW, defWindowProcW);
		HOOK_GDI_FUNCTION(user32, DefDlgProcA, defDlgProcA);
		HOOK_GDI_FUNCTION(user32, DefDlgProcW, defDlgProcW);
		DetourTransactionCommit();
	}
}