mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-27 22:26:05 +00:00
2946cd7010
to reflect the new license. We understand that people may be surprised that we're moving the header entirely to discuss the new license. We checked this carefully with the Foundation's lawyer and we believe this is the correct approach. Essentially, all code in the project is now made available by the LLVM project under our new license, so you will see that the license headers include that license only. Some of our contributors have contributed code under our old license, and accordingly, we have retained a copy of our old license notice in the top-level files in each project and repository. llvm-svn: 351636
84 lines
3.4 KiB
C++
84 lines
3.4 KiB
C++
//===-- interception_linux.h ------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file is a part of AddressSanitizer, an address sanity checker.
|
|
//
|
|
// Windows-specific interception methods.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if SANITIZER_WINDOWS
|
|
|
|
#if !defined(INCLUDED_FROM_INTERCEPTION_LIB)
|
|
# error "interception_win.h should be included from interception library only"
|
|
#endif
|
|
|
|
#ifndef INTERCEPTION_WIN_H
|
|
#define INTERCEPTION_WIN_H
|
|
|
|
namespace __interception {
|
|
// All the functions in the OverrideFunction() family return true on success,
|
|
// false on failure (including "couldn't find the function").
|
|
|
|
// Overrides a function by its address.
|
|
bool OverrideFunction(uptr old_func, uptr new_func, uptr *orig_old_func = 0);
|
|
|
|
// Overrides a function in a system DLL or DLL CRT by its exported name.
|
|
bool OverrideFunction(const char *name, uptr new_func, uptr *orig_old_func = 0);
|
|
|
|
// Windows-only replacement for GetProcAddress. Useful for some sanitizers.
|
|
uptr InternalGetProcAddress(void *module, const char *func_name);
|
|
|
|
// Overrides a function only when it is called from a specific DLL. For example,
|
|
// this is used to override calls to HeapAlloc/HeapFree from ucrtbase without
|
|
// affecting other third party libraries.
|
|
bool OverrideImportedFunction(const char *module_to_patch,
|
|
const char *imported_module,
|
|
const char *function_name, uptr new_function,
|
|
uptr *orig_old_func);
|
|
|
|
#if !SANITIZER_WINDOWS64
|
|
// Exposed for unittests
|
|
bool OverrideFunctionWithDetour(
|
|
uptr old_func, uptr new_func, uptr *orig_old_func);
|
|
#endif
|
|
|
|
// Exposed for unittests
|
|
bool OverrideFunctionWithRedirectJump(
|
|
uptr old_func, uptr new_func, uptr *orig_old_func);
|
|
bool OverrideFunctionWithHotPatch(
|
|
uptr old_func, uptr new_func, uptr *orig_old_func);
|
|
bool OverrideFunctionWithTrampoline(
|
|
uptr old_func, uptr new_func, uptr *orig_old_func);
|
|
|
|
// Exposed for unittests
|
|
void TestOnlyReleaseTrampolineRegions();
|
|
|
|
} // namespace __interception
|
|
|
|
#if defined(INTERCEPTION_DYNAMIC_CRT)
|
|
#define INTERCEPT_FUNCTION_WIN(func) \
|
|
::__interception::OverrideFunction(#func, \
|
|
(::__interception::uptr)WRAP(func), \
|
|
(::__interception::uptr *)&REAL(func))
|
|
#else
|
|
#define INTERCEPT_FUNCTION_WIN(func) \
|
|
::__interception::OverrideFunction((::__interception::uptr)func, \
|
|
(::__interception::uptr)WRAP(func), \
|
|
(::__interception::uptr *)&REAL(func))
|
|
#endif
|
|
|
|
#define INTERCEPT_FUNCTION_VER_WIN(func, symver) INTERCEPT_FUNCTION_WIN(func)
|
|
|
|
#define INTERCEPT_FUNCTION_DLLIMPORT(user_dll, provider_dll, func) \
|
|
::__interception::OverrideImportedFunction( \
|
|
user_dll, provider_dll, #func, (::__interception::uptr)WRAP(func), \
|
|
(::__interception::uptr *)&REAL(func))
|
|
|
|
#endif // INTERCEPTION_WIN_H
|
|
#endif // SANITIZER_WINDOWS
|