mirror of
https://github.com/baldurk/renderdoc.git
synced 2026-09-08 14:55:41 +00:00
125 lines
3.4 KiB
C++
125 lines
3.4 KiB
C++
/******************************************************************************
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2015-2016 Baldur Karlsson
|
|
* 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(RENDERDOC_PLATFORM_WIN32)
|
|
|
|
#include "os/win32/win32_hook.h"
|
|
|
|
template<typename FuncType>
|
|
class Hook
|
|
{
|
|
public:
|
|
Hook()
|
|
{
|
|
orig_funcptr = NULL;
|
|
}
|
|
~Hook()
|
|
{
|
|
}
|
|
|
|
FuncType operator()()
|
|
{
|
|
return (FuncType)orig_funcptr;
|
|
}
|
|
|
|
void SetFuncPtr(void *ptr)
|
|
{
|
|
orig_funcptr = ptr;
|
|
}
|
|
|
|
bool Initialize(const char *function, const char *module_name, void *destination_function_ptr)
|
|
{
|
|
orig_funcptr = Process::GetFunctionAddress(Process::LoadModule(module_name), function);
|
|
|
|
return Win32_IAT_Hook(&orig_funcptr, module_name, function, destination_function_ptr);
|
|
}
|
|
|
|
private:
|
|
void *orig_funcptr;
|
|
};
|
|
|
|
#define HOOKS_BEGIN() Win32_IAT_BeginHooks()
|
|
#define HOOKS_END() Win32_IAT_EndHooks()
|
|
#define HOOKS_REMOVE() Win32_IAT_RemoveHooks()
|
|
|
|
#elif defined(RENDERDOC_PLATFORM_POSIX)
|
|
|
|
#include "os/posix/posix_hook.h"
|
|
|
|
// just need this for dlsym
|
|
#include <dlfcn.h>
|
|
|
|
#define HOOKS_BEGIN() LinuxHookInit()
|
|
#define HOOKS_END()
|
|
#define HOOKS_REMOVE()
|
|
|
|
#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;
|
|
virtual void OptionsUpdated(const char *libName) = 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:
|
|
LibraryHooks() : m_HooksRemoved(false) {}
|
|
static LibraryHooks &GetInstance();
|
|
void RegisterHook(const char *libName, LibraryHook *hook);
|
|
void CreateHooks();
|
|
void OptionsUpdated();
|
|
void EnableHooks(bool enable);
|
|
void RemoveHooks();
|
|
|
|
private:
|
|
typedef map<const char *, LibraryHook *> HookMap;
|
|
|
|
bool m_HooksRemoved;
|
|
|
|
HookMap m_Hooks;
|
|
};
|