forked from OSchip/llvm-project
46 lines
1.5 KiB
C++
46 lines
1.5 KiB
C++
//===-- interception_linux.h ------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of AddressSanitizer, an address sanity checker.
|
|
//
|
|
// Windows-specific interception methods.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifdef _WIN32
|
|
|
|
#if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
|
|
# error "interception_win.h should be included from interception library only"
|
|
#endif
|
|
|
|
#ifndef INTERCEPTION_WIN_H
|
|
#define INTERCEPTION_WIN_H
|
|
|
|
namespace __interception {
|
|
// returns true if a function with the given name was found.
|
|
bool GetRealFunctionAddress(const char *func_name, uptr *func_addr);
|
|
|
|
// returns true if the old function existed, false on failure.
|
|
bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func);
|
|
} // namespace __interception
|
|
|
|
#if defined(_DLL)
|
|
# define INTERCEPT_FUNCTION_WIN(func) \
|
|
::__interception::GetRealFunctionAddress( \
|
|
#func, (::__interception::uptr*)&REAL(func))
|
|
#else
|
|
# define INTERCEPT_FUNCTION_WIN(func) \
|
|
::__interception::OverrideFunction( \
|
|
(::__interception::uptr)func, \
|
|
(::__interception::uptr)WRAP(func), \
|
|
(::__interception::uptr*)&REAL(func))
|
|
#endif
|
|
|
|
#endif // INTERCEPTION_WIN_H
|
|
#endif // _WIN32
|