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 24namespace __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. 29bool 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. 32bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0); 33 34// Windows-only replacement for GetProcAddress. Useful for some sanitizers. 35uptr InternalGetProcAddress(void *module, const char *func_name); 36 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. 40bool OverrideImportedFunction(const char *module_to_patch, 41 const char *imported_module, 42 const char *function_name, uptr new_function, 43 uptr *orig_old_func); 44 45#if !SANITIZER_WINDOWS64 46// Exposed for unittests 47bool OverrideFunctionWithDetour( 48 uptr old_func, uptr new_func, uptr *orig_old_func); 49#endif 50 51// Exposed for unittests 52bool OverrideFunctionWithRedirectJump( 53 uptr old_func, uptr new_func, uptr *orig_old_func); 54bool OverrideFunctionWithHotPatch( 55 uptr old_func, uptr new_func, uptr *orig_old_func); 56bool OverrideFunctionWithTrampoline( 57 uptr old_func, uptr new_func, uptr *orig_old_func); 58 59// Exposed for unittests 60void TestOnlyReleaseTrampolineRegions(); 61 62} // namespace __interception 63 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)) 69#else 70#define INTERCEPT_FUNCTION_WIN(func) \ 71 ::__interception::OverrideFunction((::__interception::uptr)func, \ 72 (::__interception::uptr)WRAP(func), \ 73 (::__interception::uptr *)&REAL(func)) 74#endif 75 76#define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func) 77 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)) 82 83#endif // INTERCEPTION_WIN_H 84#endif // _WIN32 85