9c72a9db7b
If built as part of the main llvm build, via ENABLE_LLVM_PROJECTS=compiler-rt, the code gets built with more warning options than if built standalone. Some of these trigger warnings like: warning: extra ';' outside of a function is incompatible with C++98 [-Wc++98-compat-extra-semi]
113 lines
4.3 KiB
C++
113 lines
4.3 KiB
C++
//===-- asan_win_common_runtime_thunk.cpp --------------------------- -----===//
|
|
//
|
|
// 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.
|
|
//
|
|
// This file defines things that need to be present in the application modules
|
|
// to interact with the ASan DLL runtime correctly and can't be implemented
|
|
// using the default "import library" generated when linking the DLL.
|
|
//
|
|
// This includes:
|
|
// - Cloning shadow memory dynamic address from ASAN DLL
|
|
// - Creating weak aliases to default implementation imported from asan dll
|
|
// - Forwarding the detect_stack_use_after_return runtime option
|
|
// - installing a custom SEH handler
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if defined(SANITIZER_DYNAMIC_RUNTIME_THUNK) || \
|
|
defined(SANITIZER_STATIC_RUNTIME_THUNK)
|
|
# define SANITIZER_IMPORT_INTERFACE 1
|
|
# define WIN32_LEAN_AND_MEAN
|
|
# include "asan_win_common_runtime_thunk.h"
|
|
|
|
# include <windows.h>
|
|
|
|
# include "sanitizer_common/sanitizer_win_defs.h"
|
|
# include "sanitizer_common/sanitizer_win_thunk_interception.h"
|
|
|
|
// Define weak alias for all weak functions imported from asan dll.
|
|
# define INTERFACE_FUNCTION(Name)
|
|
# define INTERFACE_WEAK_FUNCTION(Name) REGISTER_WEAK_FUNCTION(Name)
|
|
# include "asan_interface.inc"
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// Define a copy of __asan_option_detect_stack_use_after_return that should be
|
|
// used when linking an MD runtime with a set of object files on Windows.
|
|
//
|
|
// The ASan MD runtime dllexports '__asan_option_detect_stack_use_after_return',
|
|
// so normally we would just dllimport it. Unfortunately, the dllimport
|
|
// attribute adds __imp_ prefix to the symbol name of a variable.
|
|
// Since in general we don't know if a given TU is going to be used
|
|
// with a MT or MD runtime and we don't want to use ugly __imp_ names on Windows
|
|
// just to work around this issue, let's clone the variable that is constant
|
|
// after initialization anyways.
|
|
|
|
extern "C" {
|
|
__declspec(dllimport) int __asan_should_detect_stack_use_after_return();
|
|
int __asan_option_detect_stack_use_after_return;
|
|
|
|
__declspec(dllimport) void *__asan_get_shadow_memory_dynamic_address();
|
|
void *__asan_shadow_memory_dynamic_address;
|
|
|
|
static void __asan_initialize_cloned_variables() {
|
|
__asan_option_detect_stack_use_after_return =
|
|
__asan_should_detect_stack_use_after_return();
|
|
__asan_shadow_memory_dynamic_address =
|
|
__asan_get_shadow_memory_dynamic_address();
|
|
}
|
|
}
|
|
|
|
static int asan_thunk_init() {
|
|
__asan_initialize_cloned_variables();
|
|
|
|
# ifdef SANITIZER_STATIC_RUNTIME_THUNK
|
|
__asan_initialize_static_thunk();
|
|
# endif
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void WINAPI asan_thread_init(void *mod, unsigned long reason,
|
|
void *reserved) {
|
|
if (reason == DLL_PROCESS_ATTACH) {
|
|
asan_thunk_init();
|
|
}
|
|
}
|
|
|
|
// Our cloned variables must be initialized before C/C++ constructors. If TLS
|
|
// is used, our .CRT$XLAB initializer will run first. If not, our .CRT$XIB
|
|
// initializer is needed as a backup.
|
|
extern "C" __declspec(allocate(".CRT$XIB")) int (*__asan_thunk_init)() =
|
|
asan_thunk_init;
|
|
WIN_FORCE_LINK(__asan_thunk_init)
|
|
|
|
extern "C" __declspec(allocate(".CRT$XLAB")) void(WINAPI *__asan_tls_init)(
|
|
void *, unsigned long, void *) = asan_thread_init;
|
|
WIN_FORCE_LINK(__asan_tls_init)
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// ASan SEH handling.
|
|
// We need to set the ASan-specific SEH handler at the end of CRT initialization
|
|
// of each module (see also asan_win.cpp).
|
|
extern "C" {
|
|
__declspec(dllimport) int __asan_set_seh_filter();
|
|
static int SetSEHFilter() { return __asan_set_seh_filter(); }
|
|
|
|
// Unfortunately, putting a pointer to __asan_set_seh_filter into
|
|
// __asan_intercept_seh gets optimized out, so we have to use an extra function.
|
|
extern "C" __declspec(allocate(".CRT$XCAB")) int (*__asan_seh_interceptor)() =
|
|
SetSEHFilter;
|
|
WIN_FORCE_LINK(__asan_seh_interceptor)
|
|
}
|
|
|
|
WIN_FORCE_LINK(__asan_dso_reg_hook)
|
|
|
|
#endif // defined(SANITIZER_DYNAMIC_RUNTIME_THUNK) ||
|
|
// defined(SANITIZER_STATIC_RUNTIME_THUNK)
|