92 lines · c
1//===-- interception_linux.h ------------------------------------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file is a part of AddressSanitizer, an address sanity checker.10//11// Windows-specific interception methods.12//===----------------------------------------------------------------------===//13 14#if SANITIZER_WINDOWS15 16#if !defined(INCLUDED_FROM_INTERCEPTION_LIB)17# error "interception_win.h should be included from interception library only"18#endif19 20#ifndef INTERCEPTION_WIN_H21#define INTERCEPTION_WIN_H22 23namespace __interception {24// All the functions in the OverrideFunction() family return true on success,25// false on failure (including "couldn't find the function").26 27// Overrides a function by its address.28bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);29 30// Overrides a function in a system DLL or DLL CRT by its exported name.31bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0);32 33// Windows-only replacement for GetProcAddress. Useful for some sanitizers.34uptr InternalGetProcAddress(void *module, const char *func_name);35 36// Overrides a function only when it is called from a specific DLL. For example,37// this is used to override calls to HeapAlloc/HeapFree from ucrtbase without38// affecting other third party libraries.39bool OverrideImportedFunction(const char *module_to_patch,40 const char *imported_module,41 const char *function_name, uptr new_function,42 uptr *orig_old_func);43 44// Sets a callback to be used for reporting errors by interception_win. The45// callback will be called with printf-like arguments. Intended to be used with46// __sanitizer::Report. Pass nullptr to disable error reporting (default).47void SetErrorReportCallback(void (*callback)(const char *format, ...));48 49#if !SANITIZER_WINDOWS6450// Exposed for unittests51bool OverrideFunctionWithDetour(52 uptr old_func, uptr new_func, uptr *orig_old_func);53#endif54 55// Exposed for unittests56bool OverrideFunctionWithRedirectJump(57 uptr old_func, uptr new_func, uptr *orig_old_func);58bool OverrideFunctionWithHotPatch(59 uptr old_func, uptr new_func, uptr *orig_old_func);60bool OverrideFunctionWithTrampoline(61 uptr old_func, uptr new_func, uptr *orig_old_func);62 63// Exposed for unittests64void TestOnlyReleaseTrampolineRegions();65 66// Exposed for unittests67SIZE_T TestOnlyGetInstructionSize(uptr address, SIZE_T *rel_offset);68 69} // namespace __interception70 71#if defined(INTERCEPTION_DYNAMIC_CRT)72#define INTERCEPT_FUNCTION_WIN(func) \73 ::__interception::OverrideFunction(#func, \74 (::__interception::uptr)WRAP(func), \75 (::__interception::uptr *)&REAL(func))76#else77#define INTERCEPT_FUNCTION_WIN(func) \78 ::__interception::OverrideFunction((::__interception::uptr)func, \79 (::__interception::uptr)WRAP(func), \80 (::__interception::uptr *)&REAL(func))81#endif82 83#define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)84 85#define INTERCEPT_FUNCTION_DLLIMPORT(user_dll, provider_dll, func) \86 ::__interception::OverrideImportedFunction( \87 user_dll, provider_dll, #func, (::__interception::uptr)WRAP(func), \88 (::__interception::uptr *)&REAL(func))89 90#endif // INTERCEPTION_WIN_H91#endif // SANITIZER_WINDOWS92