145 lines
3.1 KiB
C++
145 lines
3.1 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 2001-2002 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
BaseBall2000.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
If you use a video card that supports more than 10 texture formats the
|
||
|
|
app will AV writing passed the end of their SURFACEDESC array.
|
||
|
|
|
||
|
|
History:
|
||
|
|
|
||
|
|
01/04/2001 maonis Created
|
||
|
|
03/07/2002 robkenny Security changes
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
#include "d3d.h"
|
||
|
|
|
||
|
|
IMPLEMENT_SHIM_BEGIN(BaseBall2000)
|
||
|
|
#include "ShimHookMacro.h"
|
||
|
|
|
||
|
|
APIHOOK_ENUM_BEGIN
|
||
|
|
APIHOOK_ENUM_ENTRY_DIRECTX_COMSERVER()
|
||
|
|
APIHOOK_ENUM_END
|
||
|
|
|
||
|
|
IMPLEMENT_DIRECTX_COMSERVER_HOOKS()
|
||
|
|
|
||
|
|
typedef HRESULT (*_pfn_IDirect3D3_CreateDevice)(PVOID pThis, REFCLSID rclsid, LPDIRECTDRAWSURFACE4, LPDIRECT3DDEVICE3*, LPUNKNOWN);
|
||
|
|
typedef HRESULT (*_pfn_IDirect3DDevice3_EnumTextureFormats)(PVOID pThis, LPD3DENUMPIXELFORMATSCALLBACK, LPVOID);
|
||
|
|
|
||
|
|
typedef HRESULT (*_pfn_EnumPixelFormatsCallback)(LPDDPIXELFORMAT lpDDPixFmt, LPVOID lpContext);
|
||
|
|
|
||
|
|
_pfn_EnumPixelFormatsCallback g_pfnEnumPixelFormatsCallback = NULL;
|
||
|
|
int g_cD3DEnumPixelFormatsCallbacks = 0;
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Hook this call so we can make sure that calling methods on the
|
||
|
|
IDirect3DDevice3 interface is hooked.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
COMHOOK(IDirect3D3, CreateDevice)(
|
||
|
|
PVOID pThis,
|
||
|
|
REFCLSID rclsid,
|
||
|
|
LPDIRECTDRAWSURFACE4 lpDDS,
|
||
|
|
LPDIRECT3DDEVICE3* lplpD3DDevice,
|
||
|
|
LPUNKNOWN /*lpUnkOuter*/
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hReturn;
|
||
|
|
|
||
|
|
_pfn_IDirect3D3_CreateDevice pfnOld =
|
||
|
|
ORIGINAL_COM(IDirect3D3, CreateDevice, pThis);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hReturn = (*pfnOld)(
|
||
|
|
pThis,
|
||
|
|
rclsid,
|
||
|
|
lpDDS,
|
||
|
|
lplpD3DDevice,
|
||
|
|
NULL)))
|
||
|
|
{
|
||
|
|
HookObject(
|
||
|
|
NULL,
|
||
|
|
IID_IDirect3DDevice3,
|
||
|
|
(PVOID*)lplpD3DDevice,
|
||
|
|
NULL,
|
||
|
|
FALSE);
|
||
|
|
}
|
||
|
|
|
||
|
|
return hReturn;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Restrict to returning at most 10 texture formats.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CALLBACK
|
||
|
|
EnumPixelFormatsCallback(
|
||
|
|
LPDDPIXELFORMAT lpDDPixFmt,
|
||
|
|
LPVOID lpContext
|
||
|
|
)
|
||
|
|
{
|
||
|
|
// The app only supports up to 10 texture formats.
|
||
|
|
if (++g_cD3DEnumPixelFormatsCallbacks >= 11)
|
||
|
|
{
|
||
|
|
return D3DENUMRET_CANCEL;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return g_pfnEnumPixelFormatsCallback(lpDDPixFmt, lpContext);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Call our private callback instead.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
COMHOOK(IDirect3DDevice3, EnumTextureFormats)(
|
||
|
|
PVOID pThis,
|
||
|
|
LPD3DENUMPIXELFORMATSCALLBACK lpd3dEnumPixelProc,
|
||
|
|
LPVOID lpArg
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DPFN( eDbgLevelError, "it IS getting called");
|
||
|
|
|
||
|
|
g_pfnEnumPixelFormatsCallback = lpd3dEnumPixelProc;
|
||
|
|
|
||
|
|
_pfn_IDirect3DDevice3_EnumTextureFormats EnumTextureFormats = ORIGINAL_COM(IDirect3DDevice3, EnumTextureFormats, pThis);
|
||
|
|
|
||
|
|
return EnumTextureFormats(pThis, EnumPixelFormatsCallback, lpArg);
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Register hooked functions
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
HOOK_BEGIN
|
||
|
|
|
||
|
|
APIHOOK_ENTRY_DIRECTX_COMSERVER()
|
||
|
|
|
||
|
|
COMHOOK_ENTRY(DirectDraw, IDirect3D3, CreateDevice, 8)
|
||
|
|
|
||
|
|
COMHOOK_ENTRY(DirectDraw, IDirect3DDevice3, EnumTextureFormats, 8)
|
||
|
|
|
||
|
|
HOOK_END
|
||
|
|
|
||
|
|
|
||
|
|
IMPLEMENT_SHIM_END
|
||
|
|
|