1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
//===-- interception_linux_foreign_test.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 ThreadSanitizer/AddressSanitizer runtime.
//
// Tests that foreign interceptors work.
//
//===----------------------------------------------------------------------===//
// Do not declare functions in ctype.h.
#define __NO_CTYPE
#include "gtest/gtest.h"
#include "sanitizer_common/sanitizer_asm.h"
#include "sanitizer_common/sanitizer_internal_defs.h"
#if SANITIZER_LINUX
extern "C" int isalnum(int d);
extern "C" int __interceptor_isalpha(int d);
extern "C" int ___interceptor_isalnum(int d); // the sanitizer interceptor
extern "C" int ___interceptor_islower(int d); // the sanitizer interceptor
namespace __interception {
extern int isalpha_called;
extern int isalnum_called;
extern int islower_called;
} // namespace __interception
using namespace __interception;
// Direct foreign interceptor. This is the "normal" protocol that other
// interceptors should follow.
extern "C" int isalpha(int d) {
// Use non-commutative arithmetic to verify order of calls.
isalpha_called = isalpha_called * 10 + 1;
return __interceptor_isalpha(d);
}
#if ASM_INTERCEPTOR_TRAMPOLINE_SUPPORT
// Indirect foreign interceptor. This pattern should only be used to co-exist
// with direct foreign interceptors and sanitizer interceptors.
extern "C" int __interceptor_isalnum(int d) {
isalnum_called = isalnum_called * 10 + 1;
return ___interceptor_isalnum(d);
}
extern "C" int __interceptor_islower(int d) {
islower_called = islower_called * 10 + 2;
return ___interceptor_islower(d);
}
extern "C" int islower(int d) {
islower_called = islower_called * 10 + 1;
return __interceptor_islower(d);
}
#endif // ASM_INTERCEPTOR_TRAMPOLINE_SUPPORT
namespace __interception {
TEST(ForeignInterception, ForeignOverrideDirect) {
isalpha_called = 0;
EXPECT_NE(0, isalpha('a'));
EXPECT_EQ(13, isalpha_called);
isalpha_called = 0;
EXPECT_EQ(0, isalpha('_'));
EXPECT_EQ(13, isalpha_called);
}
#if ASM_INTERCEPTOR_TRAMPOLINE_SUPPORT
TEST(ForeignInterception, ForeignOverrideIndirect) {
isalnum_called = 0;
EXPECT_NE(0, isalnum('a'));
EXPECT_EQ(13, isalnum_called);
isalnum_called = 0;
EXPECT_EQ(0, isalnum('_'));
EXPECT_EQ(13, isalnum_called);
}
TEST(ForeignInterception, ForeignOverrideThree) {
islower_called = 0;
EXPECT_NE(0, islower('a'));
EXPECT_EQ(123, islower_called);
islower_called = 0;
EXPECT_EQ(0, islower('_'));
EXPECT_EQ(123, islower_called);
}
#endif // ASM_INTERCEPTOR_TRAMPOLINE_SUPPORT
} // namespace __interception
#endif // SANITIZER_LINUX
|
