* pt.c (tsubst_default_argument): Use push_to/pop_from_top_level.
[official-gcc.git] / libsanitizer / interception / interception_win.h
blobf092d18d28f85919238a6775ba156b4d09a535fd
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 //===----------------------------------------------------------------------===//
13 #ifdef _WIN32
15 #if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
16 # error "interception_win.h should be included from interception library only"
17 #endif
19 #ifndef INTERCEPTION_WIN_H
20 #define INTERCEPTION_WIN_H
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").
26 // Overrides a function by its address.
27 bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);
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);
32 // Windows-only replacement for GetProcAddress. Useful for some sanitizers.
33 uptr InternalGetProcAddress(void *module, const char *func_name);
35 // Overrides a function only when it is called from a specific DLL. For example,
36 // this is used to override calls to HeapAlloc/HeapFree from ucrtbase without
37 // affecting other third party libraries.
38 bool OverrideImportedFunction(const char *module_to_patch,
39 const char *imported_module,
40 const char *function_name, uptr new_function,
41 uptr *orig_old_func);
43 #if !SANITIZER_WINDOWS64
44 // Exposed for unittests
45 bool OverrideFunctionWithDetour(
46 uptr old_func, uptr new_func, uptr *orig_old_func);
47 #endif
49 // Exposed for unittests
50 bool OverrideFunctionWithRedirectJump(
51 uptr old_func, uptr new_func, uptr *orig_old_func);
52 bool OverrideFunctionWithHotPatch(
53 uptr old_func, uptr new_func, uptr *orig_old_func);
54 bool OverrideFunctionWithTrampoline(
55 uptr old_func, uptr new_func, uptr *orig_old_func);
57 // Exposed for unittests
58 void TestOnlyReleaseTrampolineRegions();
60 } // namespace __interception
62 #if defined(INTERCEPTION_DYNAMIC_CRT)
63 #define INTERCEPT_FUNCTION_WIN(func) \
64 ::__interception::OverrideFunction(#func, \
65 (::__interception::uptr)WRAP(func), \
66 (::__interception::uptr *)&REAL(func))
67 #else
68 #define INTERCEPT_FUNCTION_WIN(func) \
69 ::__interception::OverrideFunction((::__interception::uptr)func, \
70 (::__interception::uptr)WRAP(func), \
71 (::__interception::uptr *)&REAL(func))
72 #endif
74 #define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)
76 #define INTERCEPT_FUNCTION_DLLIMPORT(user_dll, provider_dll, func) \
77 ::__interception::OverrideImportedFunction( \
78 user_dll, provider_dll, #func, (::__interception::uptr)WRAP(func), \
79 (::__interception::uptr *)&REAL(func))
81 #endif // INTERCEPTION_WIN_H
82 #endif // _WIN32