mirror of
https://github.com/baldurk/renderdoc.git
synced 2026-05-25 03:10:47 +00:00
Baldur Karlsson
12a30a7b97
* Needs vigorous testing in as many situations as I can think of, not least 32bit, 64bit, D3D11, D3D9, OpenGL, managed apps, different windows SDKs, delay loading dlls (not sure if these will work), statically linked libs, LoadLibrary + GetProcAddress * Credit goes to jslutter for originally bringing it to my attention & suggesting it. * References: http://msdn.microsoft.com/en-us/magazine/bb985992.aspx http://msdn.microsoft.com/en-us/magazine/cc301808.aspx http://www.codeproject.com/Articles/6265/Process-wide-API-spying-an-ultimate-hack
135 lines
3.4 KiB
C++
135 lines
3.4 KiB
C++
/******************************************************************************
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2014 Crytek
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
******************************************************************************/
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
#include <map>
|
|
using std::map;
|
|
|
|
#include "os/os_specific.h"
|
|
|
|
// include os-specific hooking mechanisms here
|
|
|
|
#if defined(WIN32)
|
|
|
|
#define USE_MHOOK 0
|
|
#define USE_IAT_HOOK 1
|
|
|
|
#if USE_MHOOK
|
|
#include "3rdparty/mhook/mhook-lib/mhook.h"
|
|
#elif USE_IAT_HOOK
|
|
#include "os/win32/win32_hook.h"
|
|
#else
|
|
#error "No hook method enabled"
|
|
#endif
|
|
|
|
template<typename FuncType>
|
|
class Hook
|
|
{
|
|
public:
|
|
Hook()
|
|
{
|
|
orig_funcptr = NULL;
|
|
}
|
|
~Hook()
|
|
{
|
|
#if USE_MHOOK
|
|
Mhook_Unhook(&orig_funcptr);
|
|
#endif
|
|
}
|
|
|
|
FuncType operator()()
|
|
{
|
|
return (FuncType)orig_funcptr;
|
|
}
|
|
|
|
bool Initialize(const char *function, const char *module_name, void *destination_function_ptr)
|
|
{
|
|
orig_funcptr = Process::GetFunctionAddress(module_name, function);
|
|
|
|
#if USE_MHOOK
|
|
if(orig_funcptr == NULL)
|
|
return false;
|
|
|
|
return Mhook_SetHook(&orig_funcptr, destination_function_ptr);
|
|
#elif USE_IAT_HOOK
|
|
return Win32_IAT_Hook(&orig_funcptr, module_name, function, destination_function_ptr);
|
|
#else
|
|
#error "No hook method enabled"
|
|
#endif
|
|
}
|
|
|
|
private:
|
|
void *orig_funcptr;
|
|
};
|
|
|
|
#if USE_MHOOK
|
|
#define HOOKS_BEGIN() Mhook_SuspendOtherThreads()
|
|
#define HOOKS_END() Mhook_ResumeOtherThreads()
|
|
#else
|
|
#define HOOKS_BEGIN() Win32_IAT_BeginHooks()
|
|
#define HOOKS_END() Win32_IAT_EndHooks()
|
|
#endif
|
|
|
|
#elif defined(LINUX)
|
|
|
|
// just need this for dlsym
|
|
#include <dlfcn.h>
|
|
|
|
#define HOOKS_BEGIN()
|
|
#define HOOKS_END()
|
|
|
|
#else
|
|
|
|
#error "undefined platform"
|
|
|
|
#endif
|
|
|
|
// defines the interface that a library hooking class will implement.
|
|
// the libName is the name they used when registering
|
|
struct LibraryHook
|
|
{
|
|
virtual bool CreateHooks(const char *libName) = 0;
|
|
virtual void EnableHooks(const char *libName, bool enable) = 0;
|
|
};
|
|
|
|
// this singleton allows you to compile in code that defines a hook for a given library
|
|
// (and it will be registered). Then when the renderdoc library is initialised in the target
|
|
// program CreateHooks() will be called to set up the hooks.
|
|
class LibraryHooks
|
|
{
|
|
public:
|
|
static LibraryHooks &GetInstance();
|
|
void RegisterHook(const char *libName, LibraryHook *hook);
|
|
void CreateHooks();
|
|
void EnableHooks(bool enable);
|
|
|
|
private:
|
|
typedef map<const char *, LibraryHook *> HookMap;
|
|
|
|
HookMap m_Hooks;
|
|
};
|