mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-18 19:16:43 +00:00
Revert r359325 "[NFC][Sanitizer] Change "return type" of INTERCEPT_FUNCTION to void"
Changing INTERCEPT_FUNCTION to return void is not functionally correct. IMO the best way to communicate failure or success of interception is with a return value, not some external address comparison. This change was also creating link errors for _except_handler4_common, which is exported from ucrtbase.dll in 32-bit Windows. Also revert dependent changes r359362 and r359466. llvm-svn: 359611
This commit is contained in:
parent
f3ee97731e
commit
5642c3feb0
@ -122,14 +122,12 @@ DECLARE_REAL(char*, strstr, const char *s1, const char *s2)
|
||||
#if !SANITIZER_MAC
|
||||
#define ASAN_INTERCEPT_FUNC(name) \
|
||||
do { \
|
||||
INTERCEPT_FUNCTION(name); \
|
||||
if (&(name) != &WRAP(name) || !REAL(name)) \
|
||||
if ((!INTERCEPT_FUNCTION(name) || !REAL(name))) \
|
||||
VReport(1, "AddressSanitizer: failed to intercept '" #name "'\n"); \
|
||||
} while (0)
|
||||
#define ASAN_INTERCEPT_FUNC_VER(name, ver) \
|
||||
do { \
|
||||
INTERCEPT_FUNCTION_VER(name, ver); \
|
||||
if (&(name) != &WRAP(name) || !REAL(name)) \
|
||||
if ((!INTERCEPT_FUNCTION_VER(name, ver) || !REAL(name))) \
|
||||
VReport( \
|
||||
1, "AddressSanitizer: failed to intercept '" #name "@@" #ver "'\n"); \
|
||||
} while (0)
|
||||
|
@ -33,6 +33,12 @@ static int StrCmp(const char *s1, const char *s2) {
|
||||
}
|
||||
#endif
|
||||
|
||||
bool GetRealFunctionAddress(const char *func_name, uptr *func_addr,
|
||||
uptr real, uptr wrapper) {
|
||||
*func_addr = (uptr)GetFuncAddr(func_name);
|
||||
return real == wrapper;
|
||||
}
|
||||
|
||||
void *GetFuncAddr(const char *name) {
|
||||
#if SANITIZER_NETBSD
|
||||
// FIXME: Find a better way to handle renames
|
||||
|
@ -22,18 +22,24 @@
|
||||
#define INTERCEPTION_LINUX_H
|
||||
|
||||
namespace __interception {
|
||||
// returns true if a function with the given name was found.
|
||||
bool GetRealFunctionAddress(const char *func_name, uptr *func_addr,
|
||||
uptr real, uptr wrapper);
|
||||
void *GetFuncAddr(const char *name);
|
||||
void *GetFuncAddrVer(const char *name, const char *ver);
|
||||
} // namespace __interception
|
||||
|
||||
#define INTERCEPT_FUNCTION_LINUX_OR_FREEBSD(func) \
|
||||
(REAL(func) = (FUNC_TYPE(func)) ::__interception::GetFuncAddr(#func))
|
||||
::__interception::GetRealFunctionAddress( \
|
||||
#func, (::__interception::uptr *)&__interception::PTR_TO_REAL(func), \
|
||||
(::__interception::uptr) & (func), \
|
||||
(::__interception::uptr) & WRAP(func))
|
||||
|
||||
// Android, Solaris and OpenBSD do not have dlvsym
|
||||
#if !SANITIZER_ANDROID && !SANITIZER_SOLARIS && !SANITIZER_OPENBSD
|
||||
#define INTERCEPT_FUNCTION_VER_LINUX_OR_FREEBSD(func, symver) \
|
||||
(REAL(func) = \
|
||||
(FUNC_TYPE(func)) ::__interception::GetFuncAddrVer(#func, symver))
|
||||
#define INTERCEPT_FUNCTION_VER_LINUX_OR_FREEBSD(func, symver) \
|
||||
(::__interception::real_##func = (func##_type)( \
|
||||
unsigned long)::__interception::GetFuncAddrVer(#func, symver))
|
||||
#else
|
||||
#define INTERCEPT_FUNCTION_VER_LINUX_OR_FREEBSD(func, symver) \
|
||||
INTERCEPT_FUNCTION_LINUX_OR_FREEBSD(func)
|
||||
|
@ -33,6 +33,17 @@ INTERCEPTOR(int, isdigit, int d) {
|
||||
|
||||
namespace __interception {
|
||||
|
||||
TEST(Interception, GetRealFunctionAddress) {
|
||||
uptr malloc_address = 0;
|
||||
EXPECT_TRUE(GetRealFunctionAddress("malloc", &malloc_address, 0, 0));
|
||||
EXPECT_NE(0U, malloc_address);
|
||||
|
||||
uptr dummy_address = 0;
|
||||
EXPECT_TRUE(
|
||||
GetRealFunctionAddress("dummy_doesnt_exist__", &dummy_address, 0, 0));
|
||||
EXPECT_EQ(0U, dummy_address);
|
||||
}
|
||||
|
||||
TEST(Interception, GetFuncAddr) {
|
||||
EXPECT_NE(GetFuncAddr("malloc"), nullptr);
|
||||
EXPECT_EQ(GetFuncAddr("does_not_exist"), nullptr);
|
||||
|
@ -1243,16 +1243,13 @@ int OnExit() {
|
||||
|
||||
#define MSAN_INTERCEPT_FUNC(name) \
|
||||
do { \
|
||||
INTERCEPT_FUNCTION(name); \
|
||||
if (&(name) != &WRAP(name) || !REAL(name)) \
|
||||
if ((!INTERCEPT_FUNCTION(name) || !REAL(name))) \
|
||||
VReport(1, "MemorySanitizer: failed to intercept '" #name "'\n"); \
|
||||
} while (0)
|
||||
|
||||
#define MSAN_INTERCEPT_FUNC_VER(name, ver) \
|
||||
do { \
|
||||
INTERCEPT_FUNCTION_VER(name, ver); \
|
||||
name##_type ptr = (::__interception::real_##name); \
|
||||
if (&(name) != &WRAP(name) || !REAL(name)) \
|
||||
if ((!INTERCEPT_FUNCTION_VER(name, ver) || !REAL(name))) \
|
||||
VReport( \
|
||||
1, "MemorySanitizer: failed to intercept '" #name "@@" #ver "'\n"); \
|
||||
} while (0)
|
||||
|
@ -573,8 +573,12 @@ TSAN_INTERCEPTOR(int, sigsetjmp, void *env);
|
||||
#endif
|
||||
|
||||
#define TSAN_INTERCEPTOR_SETJMP_(x) __interceptor_ ## x
|
||||
#define TSAN_INTERCEPTOR_SETJMP TSAN_INTERCEPTOR_SETJMP_(setjmp_symname)
|
||||
#define TSAN_INTERCEPTOR_SIGSETJMP TSAN_INTERCEPTOR_SETJMP_(sigsetjmp_symname)
|
||||
#define TSAN_INTERCEPTOR_SETJMP__(x) TSAN_INTERCEPTOR_SETJMP_(x)
|
||||
#define TSAN_INTERCEPTOR_SETJMP TSAN_INTERCEPTOR_SETJMP__(setjmp_symname)
|
||||
#define TSAN_INTERCEPTOR_SIGSETJMP TSAN_INTERCEPTOR_SETJMP__(sigsetjmp_symname)
|
||||
|
||||
#define TSAN_STRING_SETJMP SANITIZER_STRINGIFY(setjmp_symname)
|
||||
#define TSAN_STRING_SIGSETJMP SANITIZER_STRINGIFY(sigsetjmp_symname)
|
||||
|
||||
// Not called. Merely to satisfy TSAN_INTERCEPT().
|
||||
extern "C" SANITIZER_INTERFACE_ATTRIBUTE
|
||||
@ -2639,12 +2643,18 @@ void InitializeInterceptors() {
|
||||
InitializeSignalInterceptors();
|
||||
InitializeLibdispatchInterceptors();
|
||||
|
||||
TSAN_INTERCEPT(setjmp_symname);
|
||||
TSAN_INTERCEPT(_setjmp);
|
||||
TSAN_INTERCEPT(sigsetjmp_symname);
|
||||
|
||||
#if !SANITIZER_MAC
|
||||
// We can not use TSAN_INTERCEPT to get setjmp addr,
|
||||
// because it does &setjmp and setjmp is not present in some versions of libc.
|
||||
using __interception::GetRealFunctionAddress;
|
||||
GetRealFunctionAddress(TSAN_STRING_SETJMP,
|
||||
(uptr*)&REAL(setjmp_symname), 0, 0);
|
||||
GetRealFunctionAddress("_setjmp", (uptr*)&REAL(_setjmp), 0, 0);
|
||||
GetRealFunctionAddress(TSAN_STRING_SIGSETJMP,
|
||||
(uptr*)&REAL(sigsetjmp_symname), 0, 0);
|
||||
#if !SANITIZER_NETBSD
|
||||
TSAN_INTERCEPT(__sigsetjmp);
|
||||
GetRealFunctionAddress("__sigsetjmp", (uptr*)&REAL(__sigsetjmp), 0, 0);
|
||||
#endif
|
||||
#endif
|
||||
|
||||
TSAN_INTERCEPT(longjmp_symname);
|
||||
|
Loading…
x
Reference in New Issue
Block a user