561e804ffb
This way, callees don't need to worry about whether or not their reference to __riscv_hwprobe() has been resolved before their ifunc resolver is called. This matches the current glibc proposal from rivos. Test: treehugger Change-Id: I0d5244aa837d0d1f0e6bd7d22091dfedb8a55bdb
117 lines
2.8 KiB
C++
117 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2019 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <sys/auxv.h>
|
|
#if defined(__BIONIC__)
|
|
#include <sys/ifunc.h>
|
|
#endif
|
|
|
|
typedef int (*fn_ptr_t)();
|
|
|
|
int ret42() {
|
|
return 42;
|
|
}
|
|
|
|
extern "C" fn_ptr_t resolver() {
|
|
return ret42;
|
|
}
|
|
|
|
int ifunc() __attribute__((ifunc("resolver")));
|
|
|
|
TEST(ifunc, function) {
|
|
ASSERT_EQ(42, ifunc());
|
|
}
|
|
|
|
#if defined(__BIONIC__)
|
|
|
|
#if defined(__aarch64__)
|
|
|
|
static uint64_t g_hwcap;
|
|
static __ifunc_arg_t g_arg;
|
|
|
|
extern "C" fn_ptr_t hwcap_resolver(uint64_t hwcap, __ifunc_arg_t* arg)
|
|
__attribute__((no_sanitize("hwaddress"))) {
|
|
g_hwcap = hwcap;
|
|
g_arg = *arg;
|
|
return ret42;
|
|
}
|
|
|
|
#elif defined(__arm__)
|
|
|
|
static unsigned long g_hwcap;
|
|
|
|
extern "C" fn_ptr_t hwcap_resolver(unsigned long hwcap) {
|
|
g_hwcap = hwcap;
|
|
return ret42;
|
|
}
|
|
|
|
#elif defined(__riscv)
|
|
|
|
#include <sys/hwprobe.h>
|
|
|
|
static uint64_t g_hwcap;
|
|
static __riscv_hwprobe_t g_hwprobe_ptr;
|
|
static void* g_null;
|
|
|
|
static riscv_hwprobe g_hwprobes[] = {{.key = RISCV_HWPROBE_KEY_IMA_EXT_0}};
|
|
|
|
extern "C" fn_ptr_t hwcap_resolver(uint64_t hwcap, __riscv_hwprobe_t hwprobe_ptr, void* null) {
|
|
g_hwcap = hwcap;
|
|
g_hwprobe_ptr = hwprobe_ptr;
|
|
g_null = null;
|
|
|
|
// Ensure that __riscv_hwprobe() can be called from an ifunc.
|
|
if ((*hwprobe_ptr)(g_hwprobes, 1, 0, nullptr, 0) != 0) return nullptr;
|
|
return ret42;
|
|
}
|
|
|
|
#else
|
|
|
|
extern "C" fn_ptr_t hwcap_resolver() {
|
|
return ret42;
|
|
}
|
|
|
|
#endif
|
|
|
|
int hwcap() __attribute__((ifunc("hwcap_resolver")));
|
|
|
|
TEST(ifunc, hwcap) {
|
|
ASSERT_EQ(42, hwcap());
|
|
|
|
#if defined(__aarch64__)
|
|
EXPECT_EQ(getauxval(AT_HWCAP) | _IFUNC_ARG_HWCAP, g_hwcap);
|
|
|
|
EXPECT_EQ(sizeof(__ifunc_arg_t), g_arg._size);
|
|
EXPECT_EQ(getauxval(AT_HWCAP), g_arg._hwcap);
|
|
EXPECT_EQ(getauxval(AT_HWCAP2), g_arg._hwcap2);
|
|
#elif defined(__arm__)
|
|
EXPECT_EQ(getauxval(AT_HWCAP), g_hwcap);
|
|
#elif defined(__riscv)
|
|
printf("hwcap=%lx hwprobe_ptr=%p (__riscv_hwprobe=%p) null=%p\n", g_hwcap, g_hwprobe_ptr,
|
|
__riscv_hwprobe, g_null);
|
|
|
|
EXPECT_EQ(getauxval(AT_HWCAP), g_hwcap);
|
|
EXPECT_EQ(nullptr, g_null);
|
|
|
|
riscv_hwprobe probes[] = {{.key = RISCV_HWPROBE_KEY_IMA_EXT_0}};
|
|
ASSERT_EQ(0, __riscv_hwprobe(probes, 1, 0, nullptr, 0));
|
|
EXPECT_EQ(probes[0].value, g_hwprobes[0].value);
|
|
#endif
|
|
}
|
|
|
|
#endif // defined(__BIONIC__)
|