1 /* Test STT_GNU_IFUNC resolver with second argument.
2    Copyright (C) 2019-2021 Free Software Foundation, Inc.
3    This file is part of the GNU C Library.
4 
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9 
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14 
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, see
17    <https://www.gnu.org/licenses/>.  */
18 
19 #include <stdint.h>
20 #include <sys/auxv.h>
21 #include <sys/ifunc.h>
22 #include <support/check.h>
23 
24 static int
one(void)25 one (void)
26 {
27   return 1;
28 }
29 
30 static uint64_t saved_arg1;
31 static __ifunc_arg_t saved_arg2;
32 
33 /* extern visible ifunc symbol.  */
34 int
35 foo (void);
36 
37 void *
38 foo_ifunc (uint64_t, const __ifunc_arg_t *) __asm__ ("foo");
39 __asm__(".type foo, %gnu_indirect_function");
40 
41 void *
42 inhibit_stack_protector
foo_ifunc(uint64_t arg1,const __ifunc_arg_t * arg2)43 foo_ifunc (uint64_t arg1, const __ifunc_arg_t *arg2)
44 {
45   saved_arg1 = arg1;
46   if (arg1 & _IFUNC_ARG_HWCAP)
47       saved_arg2 = *arg2;
48   return (void *) one;
49 }
50 
51 static int
do_test(void)52 do_test (void)
53 {
54   TEST_VERIFY (foo () == 1);
55   TEST_VERIFY (saved_arg1 & _IFUNC_ARG_HWCAP);
56   TEST_COMPARE ((uint32_t)saved_arg1, (uint32_t)getauxval (AT_HWCAP));
57   TEST_COMPARE (saved_arg2._size, sizeof (__ifunc_arg_t));
58   TEST_COMPARE (saved_arg2._hwcap, getauxval (AT_HWCAP));
59   TEST_COMPARE (saved_arg2._hwcap2, getauxval (AT_HWCAP2));
60   return 0;
61 }
62 
63 #include <support/test-driver.c>
64