| //===-- 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 { |
| // All the functions in the OverrideFunction() family return true on success, |
| // false on failure (including "couldn't find the function"). |
| |
| // Overrides a function by its address. |
| bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0); |
| |
| // Overrides a function in a system DLL or DLL CRT by its exported name. |
| bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0); |
| |
| // Windows-only replacement for GetProcAddress. Useful for some sanitizers. |
| uptr InternalGetProcAddress(void *module, const char *func_name); |
| |
| } // namespace __interception |
| |
| #if defined(INTERCEPTION_DYNAMIC_CRT) |
| #define INTERCEPT_FUNCTION_WIN(func) \ |
| ::__interception::OverrideFunction(#func, \ |
| (::__interception::uptr)WRAP(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 |
| |
| #define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func) |
| |
| #endif // INTERCEPTION_WIN_H |
| #endif // _WIN32 |