platform_bionic/tests/ifunc_test.cpp
Peter Collingbourne e949195f64 Adopt GNU calling convention for ifunc resolvers.
In order for an ifunc resolver to detect the presence of certain CPU features,
access to getauxval(AT_HWCAP) or getauxval(AT_HWCAP2) may be required. In order
for getauxval() to work, it needs to access the pointer to the auxiliary vector
stored by the linker in the libc shared globals data structure. Accessing the
shared globals requires libc to call the __libc_shared_globals() function
exported by the linker. However, in order to call this function, libc must
be fully relocated, which is not guaranteed to be the case at the point when
ifunc resolvers are called.

glibc solves this problem by passing the values of getauxval(AT_HWCAP)
(and getauxval(AT_HWCAP2) on aarch64) as arguments to the ifunc resolver.
Since this seems to be not only the most straightforward way to solve the
problem but also improves our compatibility with glibc, we adopt their
calling convention.

This change is ABI compatible with old resolvers because the arguments are
passed in registers, so the old resolvers will simply ignore the new arguments.

Bug: 135772972
Change-Id: Ie65bd6e7067f0c878df3d348c815fda61dc12de2
2019-10-28 20:20:29 -07:00

86 lines
1.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) {
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;
}
#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);
#endif
}
#endif // defined(__BIONIC__)