elf/ifuncmain3.c

   1 /* Test STT_GNU_IFUNC symbols with dlopen:
   2
   3    1. Direct function call.
   4    2. Function pointer.
   5    3. Visibility with override.
   6  */
   7
   8 #include <dlfcn.h>
   9 #include <stdlib.h>
  10 #include <stdio.h>
  11
  12 typedef int (*foo_p) (void);
  13
  14 int
  15 __attribute__ ((noinline))
  16 foo (void)
  17 {
  18   return -30;
  19 }
  20
  21 int
  22 __attribute__ ((noinline))
  23 foo_hidden (void)
  24 {
  25   return -20;
  26 }
  27
  28 int
  29 __attribute__ ((noinline))
  30 foo_protected (void)
  31 {
  32   return -40;
  33 }
  34
  35 int
  36 main (void)
  37 {
  38   foo_p p;
  39   foo_p (*f) (void);
  40   int *ret;
  41
  42   void *h = dlopen ("ifuncmod3.so", RTLD_LAZY);
  43   if (h == NULL)
  44     {
  45       printf ("cannot load: %s\n", dlerror ());
  46       return 1;
  47     }
  48
  49   f = dlsym (h, "get_foo_p");
  50   if (f == NULL)
  51     {
  52       printf ("symbol not found: %s\n", dlerror ());
  53       return 1;
  54     }
  55
  56   ret = dlsym (h, "ret_foo");
  57   if (ret == NULL)
  58     {
  59       printf ("symbol not found: %s\n", dlerror ());
  60       return 1;
  61     }
  62
  63   p = (*f) ();
  64   if (p != foo)
  65     abort ();
  66   if (foo () != -30)
  67     abort ();
  68   if (*ret != -30 || (*p) () != *ret)
  69     abort ();
  70
  71   f = dlsym (h, "get_foo_hidden_p");
  72   if (f == NULL)
  73     {
  74       printf ("symbol not found: %s\n", dlerror ());
  75       return 1;
  76     }
  77
  78   ret = dlsym (h, "ret_foo_hidden");
  79   if (ret == NULL)
  80     {
  81       printf ("symbol not found: %s\n", dlerror ());
  82       return 1;
  83     }
  84
  85   p = (*f) ();
  86   if (foo_hidden () != -20)
  87     abort ();
  88   if (*ret != 1 || (*p) () != *ret)
  89     abort ();
  90
  91   f = dlsym (h, "get_foo_protected_p");
  92   if (f == NULL)
  93     {
  94       printf ("symbol not found: %s\n", dlerror ());
  95       return 1;
  96     }
  97
  98   ret = dlsym (h, "ret_foo_protected");
  99   if (ret == NULL)
 100     {
 101       printf ("symbol not found: %s\n", dlerror ());
 102       return 1;
 103     }
 104
 105   p = (*f) ();
 106   if (p == foo_protected)
 107     abort ();
 108   if (foo_protected () != -40)
 109     abort ();
 110   if (*ret != 0 || (*p) () != *ret)
 111     abort ();
 112
 113   if (dlclose (h) != 0)
 114     {
 115       printf ("cannot close: %s\n", dlerror ());
 116       return 1;
 117     }
 118
 119   return 0;
 120 }