platform_bionic/tests/ifunc_test.cpp
Elliott Hughes 620a72252e riscv64: fix ifuncs, improve the ifunc tests.
Talking futher to the person doing the glibc risc-v ifunc work, they
clarified that glibc _is_ passing hwcap as the first argument, and the
null pointer is actually the second argument.

https://sourceware.org/pipermail/libc-alpha/2023-August/150967.html

So since our whole purpose here was source compatibility, let's do what
they're actually doing, and let's add some tests. I've also added a test
that __riscv_hwprobe() works from an ifunc resolver because that's one
place where it might well be used. That said, one other thing that came
out of the discussion is that I actually went away and looked at a
sample of top apps to see how many are using ifuncs currently. The
result? Zero. So although this _might_ be interesting long term
(especially if clang gets riscv64 FMV), I think we've done more than we
need to with riscv64 ifuncs for now!

Test: ran locally, both dynamic and static tests
Change-Id: Ie2044d9f4e47c32c00ad381f045c537f4df38b08
2023-08-21 16:17:35 -07:00

113 lines
2.6 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 g_hwprobes[] = {{.key = RISCV_HWPROBE_KEY_IMA_EXT_0}};
extern "C" fn_ptr_t hwcap_resolver(uint64_t hwcap, void* null) {
// Check hwcap like arm32/arm64.
g_hwcap = hwcap;
// For now, the pointer argument is reserved for future expansion.
if (null != NULL) abort();
// Ensure that __riscv_hwprobe() can be called from an ifunc.
if (__riscv_hwprobe(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)
EXPECT_EQ(getauxval(AT_HWCAP), g_hwcap);
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__)