1 /*
2 * Copyright (C) 2019 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <gtest/gtest.h>
18
19 #include <sys/auxv.h>
20 #if defined(__BIONIC__)
21 #include <sys/ifunc.h>
22 #endif
23
24 typedef int (*fn_ptr_t)();
25
ret42()26 int ret42() {
27 return 42;
28 }
29
resolver()30 extern "C" fn_ptr_t resolver() {
31 return ret42;
32 }
33
34 int ifunc() __attribute__((ifunc("resolver")));
35
TEST(ifunc,function)36 TEST(ifunc, function) {
37 ASSERT_EQ(42, ifunc());
38 }
39
40 #if defined(__BIONIC__)
41
42 #if defined(__aarch64__)
43
44 static uint64_t g_hwcap;
45 static __ifunc_arg_t g_arg;
46
hwcap_resolver(uint64_t hwcap,__ifunc_arg_t * arg)47 extern "C" fn_ptr_t hwcap_resolver(uint64_t hwcap, __ifunc_arg_t* arg)
48 __attribute__((no_sanitize("hwaddress"))) {
49 g_hwcap = hwcap;
50 g_arg = *arg;
51 return ret42;
52 }
53
54 #elif defined(__arm__)
55
56 static unsigned long g_hwcap;
57
hwcap_resolver(unsigned long hwcap)58 extern "C" fn_ptr_t hwcap_resolver(unsigned long hwcap) {
59 g_hwcap = hwcap;
60 return ret42;
61 }
62
63 #else
64
hwcap_resolver()65 extern "C" fn_ptr_t hwcap_resolver() {
66 return ret42;
67 }
68
69 #endif
70
71 int hwcap() __attribute__((ifunc("hwcap_resolver")));
72
TEST(ifunc,hwcap)73 TEST(ifunc, hwcap) {
74 ASSERT_EQ(42, hwcap());
75
76 #if defined(__aarch64__)
77 EXPECT_EQ(getauxval(AT_HWCAP) | _IFUNC_ARG_HWCAP, g_hwcap);
78
79 EXPECT_EQ(sizeof(__ifunc_arg_t), g_arg._size);
80 EXPECT_EQ(getauxval(AT_HWCAP), g_arg._hwcap);
81 EXPECT_EQ(getauxval(AT_HWCAP2), g_arg._hwcap2);
82 #elif defined(__arm__)
83 EXPECT_EQ(getauxval(AT_HWCAP), g_hwcap);
84 #endif
85 }
86
87 #endif // defined(__BIONIC__)
88