Home | History | Annotate | Download | only in interception
      1 //===-- interception_linux.h ------------------------------------*- C++ -*-===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // This file is a part of AddressSanitizer, an address sanity checker.
     11 //
     12 // Windows-specific interception methods.
     13 //===----------------------------------------------------------------------===//
     14 
     15 #ifdef _WIN32
     16 
     17 #if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
     18 # error "interception_win.h should be included from interception library only"
     19 #endif
     20 
     21 #ifndef INTERCEPTION_WIN_H
     22 #define INTERCEPTION_WIN_H
     23 
     24 namespace __interception {
     25 // returns true if a function with the given name was found.
     26 bool GetRealFunctionAddress(const char *func_name, uptr *func_addr);
     27 
     28 // returns true if the old function existed, false on failure.
     29 bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func);
     30 }  // namespace __interception
     31 
     32 #if defined(_DLL)
     33 # define INTERCEPT_FUNCTION_WIN(func) \
     34     ::__interception::GetRealFunctionAddress( \
     35         #func, (::__interception::uptr*)&REAL(func))
     36 #else
     37 # define INTERCEPT_FUNCTION_WIN(func) \
     38     ::__interception::OverrideFunction( \
     39         (::__interception::uptr)func, \
     40         (::__interception::uptr)WRAP(func), \
     41         (::__interception::uptr*)&REAL(func))
     42 #endif
     43 
     44 #endif  // INTERCEPTION_WIN_H
     45 #endif  // _WIN32
     46