diff options
Diffstat (limited to 'lib/interception/interception_win.h')
-rw-r--r-- | lib/interception/interception_win.h | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/lib/interception/interception_win.h b/lib/interception/interception_win.h new file mode 100644 index 0000000..9d1586e --- /dev/null +++ b/lib/interception/interception_win.h @@ -0,0 +1,42 @@ +//===-- 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, void **func_addr); + +// returns true if the old function existed, false on failure. +bool OverrideFunction(void *old_func, void *new_func, void **orig_old_func); +} // namespace __interception + +#if defined(_DLL) +# define INTERCEPT_FUNCTION_WIN(func) \ + ::__interception::GetRealFunctionAddress(#func, (void**)&REAL(func)) +#else +# define INTERCEPT_FUNCTION_WIN(func) \ + ::__interception::OverrideFunction((void*)func, (void*)WRAP(func), \ + (void**)&REAL(func)) +#endif + +#endif // INTERCEPTION_WIN_H +#endif // _WIN32 |