8b36610bfa
Summary: Hopefully, this will enable cleanup/removal of GetRealFunctionAddress in follow-up commits. Subscribers: #sanitizers, llvm-commits Tags: #sanitizers, #llvm Differential Revision: https://reviews.llvm.org/D61110 llvm-svn: 359213
73 lines
2.0 KiB
C++
73 lines
2.0 KiB
C++
//===-- interception_linux_test.cc ----------------------------------------===//
|
|
//
|
|
// 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 ThreadSanitizer/AddressSanitizer runtime.
|
|
// Tests for interception_linux.h.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// Do not declare isdigit in ctype.h.
|
|
#define __NO_CTYPE
|
|
|
|
#include "interception/interception.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
// Too slow for debug build
|
|
#if !SANITIZER_DEBUG
|
|
#if SANITIZER_LINUX
|
|
|
|
static int InterceptorFunctionCalled;
|
|
|
|
DECLARE_REAL(int, isdigit, int);
|
|
|
|
INTERCEPTOR(int, isdigit, int d) {
|
|
++InterceptorFunctionCalled;
|
|
return d >= '0' && d <= '9';
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
TEST(Interception, Basic) {
|
|
ASSERT_TRUE(INTERCEPT_FUNCTION(isdigit));
|
|
|
|
// After interception, the counter should be incremented.
|
|
InterceptorFunctionCalled = 0;
|
|
EXPECT_NE(0, isdigit('1'));
|
|
EXPECT_EQ(1, InterceptorFunctionCalled);
|
|
EXPECT_EQ(0, isdigit('a'));
|
|
EXPECT_EQ(2, InterceptorFunctionCalled);
|
|
|
|
// Calling the REAL function should not affect the counter.
|
|
InterceptorFunctionCalled = 0;
|
|
EXPECT_NE(0, REAL(isdigit)('1'));
|
|
EXPECT_EQ(0, REAL(isdigit)('a'));
|
|
EXPECT_EQ(0, InterceptorFunctionCalled);
|
|
}
|
|
|
|
} // namespace __interception
|
|
|
|
#endif // SANITIZER_LINUX
|
|
#endif // #if !SANITIZER_DEBUG
|