xref: /netbsd-src/external/gpl3/gcc.old/dist/libsanitizer/interception/interception_win.h (revision bdc22b2e01993381dcefeff2bc9b56ca75a4235c)
1 //===-- interception_linux.h ------------------------------------*- C++ -*-===//
2 //
3 // This file is distributed under the University of Illinois Open Source
4 // License. See LICENSE.TXT for details.
5 //
6 //===----------------------------------------------------------------------===//
7 //
8 // This file is a part of AddressSanitizer, an address sanity checker.
9 //
10 // Windows-specific interception methods.
11 //===----------------------------------------------------------------------===//
12 
13 #ifdef _WIN32
14 
15 #if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
16 # error "interception_win.h should be included from interception library only"
17 #endif
18 
19 #ifndef INTERCEPTION_WIN_H
20 #define INTERCEPTION_WIN_H
21 
22 namespace __interception {
23 // All the functions in the OverrideFunction() family return true on success,
24 // false on failure (including "couldn't find the function").
25 
26 // Overrides a function by its address.
27 bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);
28 
29 // Overrides a function in a system DLL or DLL CRT by its exported name.
30 bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0);
31 }  // namespace __interception
32 
33 #if defined(INTERCEPTION_DYNAMIC_CRT)
34 #define INTERCEPT_FUNCTION_WIN(func)                                           \
35   ::__interception::OverrideFunction(#func,                                    \
36                                      (::__interception::uptr)WRAP(func),       \
37                                      (::__interception::uptr *)&REAL(func))
38 #else
39 #define INTERCEPT_FUNCTION_WIN(func)                                           \
40   ::__interception::OverrideFunction((::__interception::uptr)func,             \
41                                      (::__interception::uptr)WRAP(func),       \
42                                      (::__interception::uptr *)&REAL(func))
43 #endif
44 
45 #define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)
46 
47 #endif  // INTERCEPTION_WIN_H
48 #endif  // _WIN32
49