Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

501 lines
30 KiB

/*++
Copyright (c) 2000-2001 Microsoft Corporation
Module Name:
ShimHookMacro.h
Abstract:
Shim hooking macros for version 2
Notes:
None
History:
10/29/2000 markder Created
12/06/2000 robkenny Converted to use namespaces
09/11/2001 mnikkel Modified DPFN and LOGN to retain LastError
--*/
#pragma once
#ifndef _ShimHookMacro_h_
#define _ShimHookMacro_h_
//
// These are dwReason values that the shim notification functions
// can be called with.
//
//
// This means that all the static linked DLLs have run their init routines.
//
#define SHIM_STATIC_DLLS_INITIALIZED 100
//
// This means that the current process is about to die.
// This gives the shims a chance to do cleanup work while all the modules
// are still loaded.
//
#define SHIM_PROCESS_DYING 101
//
// This notification is sent to notify the shims that a DLL is unloading
//
#define SHIM_DLL_LOADING 102
extern PLDR_DATA_TABLE_ENTRY g_DllLoadingEntry;
#define GETDLLLOADINGHANDLE() (g_DllLoadingEntry)
//
// This debug macro needs to be in this file because it needs access
// to g_szModuleName which is only defined inside the namespace.
//
inline void DPFN(ShimLib::DEBUGLEVEL dwDetail, LPCSTR pszFmt, ...)
{
#if DBG
// This must be the first line of this routine to preserve LastError.
DWORD dwLastError = GetLastError();
extern const CHAR * g_szModuleName; // created by the DECLARE_SHIM macro, inside of the shim's namespace
va_list vaArgList;
va_start(vaArgList, pszFmt);
ShimLib::DebugPrintfList(g_szModuleName, dwDetail, pszFmt, vaArgList);
va_end(vaArgList);
// This must be the last line of this routine to preserve LastError.
SetLastError(dwLastError);
#else
dwDetail;
pszFmt;
#endif
}
inline void LOGN(ShimLib::DEBUGLEVEL dwDetail, LPCSTR pszFmt, ...)
{
// This must be the first line of this routine to preserve LastError.
DWORD dwLastError = GetLastError();
extern const CHAR * g_szModuleName;
if (ShimLib::g_bFileLogEnabled)
{
va_list vaArgList;
va_start(vaArgList, pszFmt);
ShimLib::ShimLogList(g_szModuleName, dwDetail, pszFmt, vaArgList);
va_end(vaArgList);
}
#if DBG
// If logging isn't enabled, dump to the debugger
else
{
va_list vaArgList;
va_start(vaArgList, pszFmt);
ShimLib::DebugPrintfList(g_szModuleName, dwDetail, pszFmt, vaArgList);
va_end(vaArgList);
}
#endif
// This must be the last line of this routine to preserve LastError.
SetLastError(dwLastError);
}
#define APIHOOK(hook) APIHook_##hook
#define COMHOOK(iface, hook) COMHook_##iface##_##hook
#define APIHOOK_ENUM_BEGIN enum {
#define APIHOOK_ENUM_ENTRY(hook) APIHOOK_##hook##,
#define APIHOOK_ENUM_ENTRY_COMSERVER(module) APIHOOK_##module##_DllGetClassObject,
#define APIHOOK_ENUM_ENTRY_DIRECTX_COMSERVER() \
APIHOOK_ENUM_ENTRY_COMSERVER(DDRAW) \
APIHOOK_ENUM_ENTRY_COMSERVER(DSOUND) \
APIHOOK_ENUM_ENTRY_COMSERVER(DPLAYX) \
APIHOOK_ENUM_ENTRY_COMSERVER(DINPUT) \
APIHOOK_ENUM_ENTRY(DirectDrawCreate) \
APIHOOK_ENUM_ENTRY(DirectDrawCreateEx) \
APIHOOK_ENUM_ENTRY(DirectSoundCreate) \
APIHOOK_ENUM_ENTRY(DirectPlayCreate) \
APIHOOK_ENUM_ENTRY(DirectInputCreateA) \
APIHOOK_ENUM_ENTRY(DirectInputCreateW) \
APIHOOK_ENUM_ENTRY(DirectInputCreateEx)
#define APIHOOK_ENUM_END APIHOOK_Count };
#define HOOK_BEGIN \
PHOOKAPI \
InitializeHooksMulti( \
DWORD fdwReason, \
LPSTR pszCmdLine, \
DWORD* pdwHookCount \
) \
{ \
DPFN(eDbgLevelSpew, \
"[InitializeHooksMulti] fdwReason(%d) pszCmdLine(%s)\n", \
fdwReason, pszCmdLine); \
\
if (fdwReason == DLL_PROCESS_ATTACH) { \
lstrcpynA( g_szCommandLine, pszCmdLine, SHIM_COMMAND_LINE_MAX_BUFFER); \
\
g_pAPIHooks = \
(PHOOKAPI) ShimMalloc(sizeof(HOOKAPI)*APIHOOK_Count); \
if (g_pAPIHooks == NULL) { \
return NULL; \
} \
*pdwHookCount = APIHOOK_Count; \
}
#define HOOK_END \
if (fdwReason == DLL_PROCESS_ATTACH) { \
DPFN(eDbgLevelSpew, \
"[InitializeHooksMulti] pdwHookCount(%d)\n", \
pdwHookCount ? *pdwHookCount : 0); \
} \
return g_pAPIHooks; \
}
#define APIHOOK_ENTRY(module, hook) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
g_pAPIHooks[APIHOOK_##hook##].pszModule = #module; \
g_pAPIHooks[APIHOOK_##hook##].pszFunctionName = #hook; \
g_pAPIHooks[APIHOOK_##hook##].pfnNew = APIHOOK(hook); \
}
#define APIHOOK_ENTRY_NAME(module, hook, hookname) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
g_pAPIHooks[APIHOOK_##hook##].pszModule = #module; \
g_pAPIHooks[APIHOOK_##hook##].pszFunctionName = #hookname; \
g_pAPIHooks[APIHOOK_##hook##].pfnNew = APIHOOK(hook); \
}
#define APIHOOK_ENTRY_ORD(module, hook, hookord) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
g_pAPIHooks[APIHOOK_##hook##].pszModule = #module; \
g_pAPIHooks[APIHOOK_##hook##].pszFunctionName = (char *)IntToPtr(hookord); \
g_pAPIHooks[APIHOOK_##hook##].pfnNew = APIHOOK(hook); \
}
#define APIHOOK_ENTRY_COMSERVER(module) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pszModule = #module ".DLL"; \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pszFunctionName = "DllGetClassObject"; \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pfnNew = APIHOOK(module##_DllGetClassObject); \
}
#define APIHOOK_ENTRY_DIRECTX_COMSERVER() \
if (fdwReason == DLL_PROCESS_ATTACH) { \
APIHOOK_ENTRY_COMSERVER(DDRAW) \
APIHOOK_ENTRY_COMSERVER(DSOUND) \
APIHOOK_ENTRY_COMSERVER(DPLAYX) \
APIHOOK_ENTRY_COMSERVER(DINPUT) \
APIHOOK_ENTRY(DDRAW.DLL, DirectDrawCreate) \
APIHOOK_ENTRY(DDRAW.DLL, DirectDrawCreateEx) \
APIHOOK_ENTRY(DSOUND.DLL, DirectSoundCreate) \
APIHOOK_ENTRY(DPLAYX.DLL, DirectPlayCreate) \
APIHOOK_ENTRY(DINPUT.DLL, DirectInputCreateA) \
APIHOOK_ENTRY(DINPUT.DLL, DirectInputCreateW) \
APIHOOK_ENTRY(DINPUT.DLL, DirectInputCreateEx) \
}
#define COMHOOK_ENTRY(obj, iface, hook, vtblndx) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
AddComHook(CLSID_##obj, IID_##iface, COMHOOK(iface, hook), vtblndx); \
}
#define DECLARE_SHIM(shim) \
namespace NS_##shim \
{ \
const CHAR * g_szModuleName = #shim; \
CHAR g_szCommandLine[SHIM_COMMAND_LINE_MAX_BUFFER]; \
PHOOKAPI g_pAPIHooks = NULL; \
BOOL g_bSubshimUsed = FALSE; \
extern PHOOKAPI InitializeHooksMulti(DWORD, LPSTR, DWORD*); \
};
#define MULTISHIM_BEGIN() \
VOID ShimLib::InitializeHooks(DWORD) \
{ \
g_dwShimVersion = 2; \
} \
PHOOKAPI ShimLib::InitializeHooksEx(DWORD fdwReason, LPWSTR pwszShim, LPSTR pszCmdLine, DWORD* pdwHookCount) \
{ \
PHOOKAPI pAPIHooks = NULL; \
g_bMultiShim = TRUE;
#define MULTISHIM_ENTRY(shim) \
if ((fdwReason == DLL_PROCESS_ATTACH && pwszShim != NULL && 0 == _wcsicmp( pwszShim, L#shim )) || \
(fdwReason == DLL_PROCESS_DETACH && NS_##shim::g_bSubshimUsed) || \
(fdwReason == SHIM_PROCESS_DYING && NS_##shim::g_bSubshimUsed) || \
(fdwReason == SHIM_DLL_LOADING && NS_##shim::g_bSubshimUsed) || \
(fdwReason == SHIM_STATIC_DLLS_INITIALIZED && NS_##shim::g_bSubshimUsed)) { \
NS_##shim::g_bSubshimUsed = TRUE; \
pAPIHooks = NS_##shim::InitializeHooksMulti( fdwReason, pszCmdLine, pdwHookCount ); \
}
#define MULTISHIM_END() \
return pAPIHooks; \
}
#define MULTISHIM_NOTIFY_FUNCTION() NotifyFn
#define NOTIFY_FUNCTION NotifyFn
#define CALL_MULTISHIM_NOTIFY_FUNCTION() NotifyFn(fdwReason);
#define CALL_NOTIFY_FUNCTION \
if (FALSE == NotifyFn(fdwReason) && \
fdwReason == DLL_PROCESS_ATTACH) { \
*pdwHookCount = 0; \
DPFN(eDbgLevelSpew, \
"[InitializeHooksMulti] NotifyFn returned FALSE, fail load shim\n", \
g_pAPIHooks); \
return NULL; \
}
#define ORIGINAL_API(hook) \
(*(_pfn_##hook##)(g_pAPIHooks[APIHOOK_##hook##].pfnOld))
#define _ORIGINAL_API(hook, proto) \
(*(_pfn_##proto##)(g_pAPIHooks[APIHOOK_##hook##].pfnOld))
#define ORIGINAL_COM(iface, hook, pThis) \
(*(_pfn_##iface##_##hook##)(LookupOriginalCOMFunction(*((PVOID *) pThis), COMHOOK(iface, hook), TRUE )))
#define COMMAND_LINE \
(g_szCommandLine)
#define IMPLEMENT_COMSERVER_HOOK(module) \
HRESULT \
APIHOOK(##module##_DllGetClassObject)( \
REFCLSID rclsid, \
REFIID riid, \
LPVOID * ppv \
) \
{ \
HRESULT hrReturn = E_FAIL; \
\
hrReturn = _ORIGINAL_API(module##_DllGetClassObject, DllGetClassObject)( \
rclsid, riid, ppv); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(rclsid, riid, ppv, TRUE); \
} \
\
return hrReturn; \
}
#define IMPLEMENT_DIRECTX_COMSERVER_HOOKS() \
IMPLEMENT_COMSERVER_HOOK(DDRAW) \
IMPLEMENT_COMSERVER_HOOK(DSOUND) \
IMPLEMENT_COMSERVER_HOOK(DPLAYX) \
IMPLEMENT_COMSERVER_HOOK(DINPUT) \
\
HRESULT \
APIHOOK(DirectDrawCreate)( \
IN GUID FAR *lpGUID, \
OUT LPVOID *lplpDD, \
OUT IUnknown* pUnkOuter \
) \
{ \
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectDrawCreate)(lpGUID, lplpDD, pUnkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectDraw, IID_IDirectDraw, lplpDD, FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectDrawCreateEx)( \
GUID FAR *lpGUID, \
LPVOID *lplpDD, \
REFIID iid, \
IUnknown* pUnkOuter \
) \
{ \
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectDrawCreateEx)( \
lpGUID, \
lplpDD, \
iid, \
pUnkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectDraw, iid, lplpDD, FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectSoundCreate)( \
LPCGUID lpcGuid, \
LPDIRECTSOUND *ppDS, \
LPUNKNOWN pUnkOuter) \
{ \
\
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectSoundCreate)( \
lpcGuid, \
ppDS, \
pUnkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectSound, \
IID_IDirectSound, \
(LPVOID*) ppDS, \
FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectPlayCreate)( \
LPGUID lpGUIDSP, \
LPDIRECTPLAY FAR *lplpDP, \
IUnknown *lpUnk) \
{ \
\
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectPlayCreate)( \
lpGUIDSP, \
lplpDP, \
lpUnk); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectPlay, \
IID_IDirectPlay, \
(LPVOID*) lplpDP, \
FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectInputCreateA)( \
HINSTANCE hinst, \
DWORD dwVersion, \
LPDIRECTINPUTA * lplpDirectInput, \
LPUNKNOWN punkOuter) \
{ \
\
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectInputCreateA)( \
hinst, \
dwVersion, \
lplpDirectInput, \
punkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectInput, \
IID_IDirectInputA, \
(LPVOID*) lplpDirectInput, \
FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectInputCreateW)( \
HINSTANCE hinst, \
DWORD dwVersion, \
LPDIRECTINPUTW * lplpDirectInput, \
LPUNKNOWN punkOuter) \
{ \
\
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectInputCreateW)( \
hinst, \
dwVersion, \
lplpDirectInput, \
punkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectInput, \
IID_IDirectInputW, \
(LPVOID*) lplpDirectInput, \
FALSE); \
} \
\
return hrReturn; \
} \
\
HRESULT \
APIHOOK(DirectInputCreateEx)( \
HINSTANCE hinst, \
DWORD dwVersion, \
REFIID riidltf, \
LPVOID * ppvOut, \
LPUNKNOWN punkOuter) \
{ \
\
HRESULT hrReturn = E_FAIL; \
\
hrReturn = ORIGINAL_API(DirectInputCreateEx)( \
hinst, \
dwVersion, \
riidltf, \
ppvOut, \
punkOuter); \
\
if (S_OK == hrReturn) { \
HookCOMInterface(CLSID_DirectInput, \
riidltf, \
(LPVOID*) ppvOut, \
FALSE); \
} \
\
return hrReturn; \
}
// Only add this hook to the list if bDeclare is TRUE
// otherwise a blank entry is added.
#define APIHOOK_ENTRY_OR_NOT(bDeclare, module, hook) \
if (bDeclare) { \
APIHOOK_ENTRY(module, hook) \
}
// Only add this hook to the list if bDeclare is TRUE
// otherwise a blank entry is added.
#define APIHOOK_ENTRY_COMSERVER_OR_NOT(bDeclare, module) \
if (bDeclare) { \
APIHOOK_ENTRY_COMSERVER(module) \
} else { \
APIHOOK_ENTRY_COMSERVER_BLANK(module) \
}
#define APIHOOK_ENTRY_COMSERVER_BLANK(module) \
if (fdwReason == DLL_PROCESS_ATTACH) { \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pszModule = ""; \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pszFunctionName = ""; \
g_pAPIHooks[APIHOOK_##module##_DllGetClassObject].pfnNew = NULL; \
}
#endif // _SHIMHOOKMACRO_H_