src/libcompiler_builtins/compiler-rt/lib/interception/tests/interception_linux_test.cc

   1 //===-- interception_linux_test.cc ----------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file is a part of ThreadSanitizer/AddressSanitizer runtime.
  11 // Tests for interception_linux.h.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 // Do not declare isdigit in ctype.h.
  16 #define __NO_CTYPE
  17
  18 #include "interception/interception.h"
  19
  20 #include "gtest/gtest.h"
  21
  22 // Too slow for debug build
  23 #if !SANITIZER_DEBUG
  24 #if SANITIZER_LINUX
  25
  26 static int InterceptorFunctionCalled;
  27
  28 DECLARE_REAL(int, isdigit, int);
  29
  30 INTERCEPTOR(int, isdigit, int d) {
  31   ++InterceptorFunctionCalled;
  32   return d >= '0' && d <= '9';
  33 }
  34
  35 namespace __interception {
  36
  37 TEST(Interception, GetRealFunctionAddress) {
  38   uptr malloc_address = 0;
  39   EXPECT_TRUE(GetRealFunctionAddress("malloc", &malloc_address, 0, 0));
  40   EXPECT_NE(0U, malloc_address);
  41
  42   uptr dummy_address = 0;
  43   EXPECT_TRUE(
  44       GetRealFunctionAddress("dummy_doesnt_exist__", &dummy_address, 0, 0));
  45   EXPECT_EQ(0U, dummy_address);
  46 }
  47
  48 TEST(Interception, Basic) {
  49   ASSERT_TRUE(INTERCEPT_FUNCTION(isdigit));
  50
  51   // After interception, the counter should be incremented.
  52   InterceptorFunctionCalled = 0;
  53   EXPECT_NE(0, isdigit('1'));
  54   EXPECT_EQ(1, InterceptorFunctionCalled);
  55   EXPECT_EQ(0, isdigit('a'));
  56   EXPECT_EQ(2, InterceptorFunctionCalled);
  57
  58   // Calling the REAL function should not affect the counter.
  59   InterceptorFunctionCalled = 0;
  60   EXPECT_NE(0, REAL(isdigit)('1'));
  61   EXPECT_EQ(0, REAL(isdigit)('a'));
  62   EXPECT_EQ(0, InterceptorFunctionCalled);
  63 }
  64
  65 }  // namespace __interception
  66
  67 #endif  // SANITIZER_LINUX
  68 #endif  // #if !SANITIZER_DEBUG