• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 // All the functions in the OverrideFunction() family return true on success,
26 // false on failure (including "couldn't find the function").
27 
28 // Overrides a function by its address.
29 bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);
30 
31 // Overrides a function in a system DLL or DLL CRT by its exported name.
32 bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0);
33 }  // namespace __interception
34 
35 #if defined(INTERCEPTION_DYNAMIC_CRT)
36 #define INTERCEPT_FUNCTION_WIN(func)                                           \
37   ::__interception::OverrideFunction(#func,                                    \
38                                      (::__interception::uptr)WRAP(func),       \
39                                      (::__interception::uptr *)&REAL(func))
40 #else
41 #define INTERCEPT_FUNCTION_WIN(func)                                           \
42   ::__interception::OverrideFunction((::__interception::uptr)func,             \
43                                      (::__interception::uptr)WRAP(func),       \
44                                      (::__interception::uptr *)&REAL(func))
45 #endif
46 
47 #define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)
48 
49 #endif  // INTERCEPTION_WIN_H
50 #endif  // _WIN32
51