summaryrefslogtreecommitdiff
path: root/elf/ifuncmain3.c
blob: 1574dd5cbe461efbb6852bc3b89d743d56dfdce3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* Test STT_GNU_IFUNC symbols with dlopen:

   1. Direct function call.
   2. Function pointer.
   3. Visibility with override.
 */

#include <dlfcn.h>
#include <stdlib.h>
#include <stdio.h>

typedef int (*foo_p) (void);

int
__attribute__ ((noinline))
foo (void)
{
  return -30;
}

int
__attribute__ ((noinline))
foo_hidden (void)
{
  return -20;
}

int
__attribute__ ((noinline))
foo_protected (void)
{
  return -40;
}

int
main (void)
{
  foo_p p;
  foo_p (*f) (void);
  int *ret;

  void *h = dlopen ("ifuncmod3.so", RTLD_LAZY);
  if (h == NULL)
    {
      printf ("cannot load: %s\n", dlerror ());
      return 1;
    }

  p = dlsym (h, "foo");
  if (p == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }
  if ((*p) () != -1)
    abort ();

  f = dlsym (h, "get_foo_p");
  if (f == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  ret = dlsym (h, "ret_foo");
  if (ret == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  p = (*f) ();
  if (p != foo)
    abort ();
  if (foo () != -30)
    abort ();
  if (*ret != -30 || (*p) () != *ret)
    abort ();

  f = dlsym (h, "get_foo_hidden_p");
  if (f == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  ret = dlsym (h, "ret_foo_hidden");
  if (ret == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  p = (*f) ();
  if (foo_hidden () != -20)
    abort ();
  if (*ret != 1 || (*p) () != *ret)
    abort ();

  f = dlsym (h, "get_foo_protected_p");
  if (f == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  ret = dlsym (h, "ret_foo_protected");
  if (ret == NULL)
    {
      printf ("symbol not found: %s\n", dlerror ());
      return 1;
    }

  p = (*f) ();
  if (p == foo_protected)
    abort ();
  if (foo_protected () != -40)
    abort ();
  if (*ret != 0 || (*p) () != *ret)
    abort ();

  if (dlclose (h) != 0)
    {
      printf ("cannot close: %s\n", dlerror ());
      return 1;
    }

  return 0;
}