1 //===-- interception_linux.h ------------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file is a part of AddressSanitizer, an address sanity checker.
12 // Windows-specific interception methods.
13 //===----------------------------------------------------------------------===//
17 #if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
18 # error "interception_win.h should be included from interception library only"
21 #ifndef INTERCEPTION_WIN_H
22 #define INTERCEPTION_WIN_H
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").
28 // Overrides a function by its address.
29 bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);
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);
34 // Windows-only replacement for GetProcAddress. Useful for some sanitizers.
35 uptr InternalGetProcAddress(void *module, const char *func_name);
37 // Overrides a function only when it is called from a specific DLL. For example,
38 // this is used to override calls to HeapAlloc/HeapFree from ucrtbase without
39 // affecting other third party libraries.
40 bool OverrideImportedFunction(const char *module_to_patch,
41 const char *imported_module,
42 const char *function_name, uptr new_function,
45 #if !SANITIZER_WINDOWS64
46 // Exposed for unittests
47 bool OverrideFunctionWithDetour(
48 uptr old_func, uptr new_func, uptr *orig_old_func);
51 // Exposed for unittests
52 bool OverrideFunctionWithRedirectJump(
53 uptr old_func, uptr new_func, uptr *orig_old_func);
54 bool OverrideFunctionWithHotPatch(
55 uptr old_func, uptr new_func, uptr *orig_old_func);
56 bool OverrideFunctionWithTrampoline(
57 uptr old_func, uptr new_func, uptr *orig_old_func);
59 // Exposed for unittests
60 void TestOnlyReleaseTrampolineRegions();
62 } // namespace __interception
64 #if defined(INTERCEPTION_DYNAMIC_CRT)
65 #define INTERCEPT_FUNCTION_WIN(func) \
66 ::__interception::OverrideFunction(#func, \
67 (::__interception::uptr)WRAP(func), \
68 (::__interception::uptr *)&REAL(func))
70 #define INTERCEPT_FUNCTION_WIN(func) \
71 ::__interception::OverrideFunction((::__interception::uptr)func, \
72 (::__interception::uptr)WRAP(func), \
73 (::__interception::uptr *)&REAL(func))
76 #define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)
78 #define INTERCEPT_FUNCTION_DLLIMPORT(user_dll, provider_dll, func) \
79 ::__interception::OverrideImportedFunction( \
80 user_dll, provider_dll, #func, (::__interception::uptr)WRAP(func), \
81 (::__interception::uptr *)&REAL(func))
83 #endif // INTERCEPTION_WIN_H
84 #endif // SANITIZER_WINDOWS