fdwalk should return 0 on an empty directory
[kopensolaris-gnu/glibc.git] / elf / loadtest.c
1 #include <assert.h>
2 #include <dlfcn.h>
3 #include <errno.h>
4 #include <error.h>
5 #include <mcheck.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9
10
11 /* How many load/unload operations do we do.  */
12 #define TEST_ROUNDS     1000
13
14
15 static struct
16 {
17   /* Name of the module.  */
18   const char *name;
19   /* The handle.  */
20   void *handle;
21 } testobjs[] =
22 {
23   { "testobj1.so", NULL },
24   { "testobj2.so", NULL },
25   { "testobj3.so", NULL },
26   { "testobj4.so", NULL },
27   { "testobj5.so", NULL },
28   { "testobj6.so", NULL },
29 };
30 #define NOBJS   (sizeof (testobjs) / sizeof (testobjs[0]))
31
32
33 static const struct
34 {
35   /* Name of a function to call.  */
36   const char *fname;
37   /* Index in status and handle array.  */
38   int index;
39   /* Options while loading the module.  */
40   int options;
41 } tests[] =
42 {
43   { "obj1func2", 0, RTLD_LAZY },
44   { "obj1func1", 0, RTLD_LAZY | RTLD_GLOBAL },
45   { "obj1func1", 0, RTLD_NOW, },
46   { "obj1func2", 0, RTLD_NOW | RTLD_GLOBAL },
47   { "obj2func2", 1, RTLD_LAZY },
48   { "obj2func1", 1, RTLD_LAZY | RTLD_GLOBAL, },
49   { "obj2func1", 1, RTLD_NOW, },
50   { "obj2func2", 1, RTLD_NOW | RTLD_GLOBAL },
51   { "obj3func2", 2, RTLD_LAZY },
52   { "obj3func1", 2, RTLD_LAZY | RTLD_GLOBAL },
53   { "obj3func1", 2, RTLD_NOW },
54   { "obj3func2", 2, RTLD_NOW | RTLD_GLOBAL },
55   { "obj4func2", 3, RTLD_LAZY },
56   { "obj4func1", 3, RTLD_LAZY | RTLD_GLOBAL },
57   { "obj4func1", 3, RTLD_NOW },
58   { "obj4func2", 3, RTLD_NOW | RTLD_GLOBAL },
59   { "obj5func2", 4, RTLD_LAZY },
60   { "obj5func1", 4, RTLD_LAZY | RTLD_GLOBAL },
61   { "obj5func1", 4, RTLD_NOW },
62   { "obj5func2", 4, RTLD_NOW | RTLD_GLOBAL },
63   { "obj6func2", 5, RTLD_LAZY },
64   { "obj6func1", 5, RTLD_LAZY | RTLD_GLOBAL },
65   { "obj6func1", 5, RTLD_NOW },
66   { "obj6func2", 5, RTLD_NOW | RTLD_GLOBAL },
67 };
68 #define NTESTS  (sizeof (tests) / sizeof (tests[0]))
69
70
71 #include <include/link.h>
72
73 #define MAPS ((struct link_map *) _r_debug.r_map)
74
75 #define OUT \
76   for (map = MAPS; map != NULL; map = map->l_next)                    \
77     if (map->l_type == lt_loaded)                                             \
78       printf ("name = \"%s\", direct_opencount = %d\n",                       \
79               map->l_name, (int) map->l_direct_opencount);                    \
80   fflush (stdout)
81
82
83 int
84 main (int argc, char *argv[])
85 {
86   int debug = argc > 1 && argv[1][0] != '\0';
87   int count = TEST_ROUNDS;
88   int result = 0;
89   struct link_map *map;
90
91   mtrace ();
92
93   /* Just a seed.  */
94   srandom (TEST_ROUNDS);
95
96   if (debug)
97     {
98       puts ("in the beginning");
99       OUT;
100     }
101
102   while (count--)
103     {
104       int nr = random () % NTESTS;
105       int index = tests[nr].index;
106
107       printf ("%4d: %4d: ", count + 1, nr);
108       fflush (stdout);
109
110       if (testobjs[index].handle == NULL)
111         {
112           int (*fct) (int);
113
114           /* Load the object.  */
115           testobjs[index].handle = dlopen (testobjs[index].name,
116                                            tests[nr].options);
117           if (testobjs[index].handle == NULL)
118             error (EXIT_FAILURE, 0, "cannot load `%s': %s",
119                    testobjs[index].name, dlerror ());
120
121           /* Test the function call.  */
122           fct = dlsym (testobjs[index].handle, tests[nr].fname);
123           if (fct == NULL)
124             error (EXIT_FAILURE, 0,
125                    "cannot get function `%s' from shared object `%s': %s",
126                    tests[nr].fname, testobjs[index].name, dlerror ());
127
128           fct (10);
129
130           printf ("successfully loaded `%s', handle %p\n",
131                   testobjs[index].name, testobjs[index].handle);
132         }
133       else
134         {
135           if (dlclose (testobjs[index].handle) != 0)
136             {
137               printf ("failed to close %s\n", testobjs[index].name);
138               result = 1;
139             }
140           else
141             printf ("successfully unloaded `%s', handle %p\n",
142                     testobjs[index].name, testobjs[index].handle);
143
144           testobjs[index].handle = NULL;
145
146           if (testobjs[0].handle == NULL
147               && testobjs[1].handle == NULL
148               && testobjs[5].handle == NULL)
149             {
150               /* In this case none of the objects above should be
151                  present.  */
152               for (map = MAPS; map != NULL; map = map->l_next)
153                 if (map->l_type == lt_loaded
154                     && (strstr (map->l_name, testobjs[0].name) != NULL
155                         || strstr (map->l_name, testobjs[1].name) != NULL
156                         || strstr (map->l_name, testobjs[5].name) != NULL))
157                   {
158                     printf ("`%s' is still loaded\n", map->l_name);
159                     result = 1;
160                   }
161             }
162         }
163
164       if (debug)
165         OUT;
166     }
167
168   /* Unload all loaded modules.  */
169   for (count = 0; count < (int) NOBJS; ++count)
170     if (testobjs[count].handle != NULL)
171       {
172         printf ("\nclose: %s: l_initfini = %p, l_versions = %p\n",
173                 testobjs[count].name,
174                 ((struct link_map *) testobjs[count].handle)->l_initfini,
175                 ((struct link_map *) testobjs[count].handle)->l_versions);
176
177         if (dlclose (testobjs[count].handle) != 0)
178           {
179             printf ("failed to close %s\n", testobjs[count].name);
180             result = 1;
181           }
182       }
183
184   /* Check whether all files are unloaded.  */
185   for (map = MAPS; map != NULL; map = map->l_next)
186     if (map->l_type == lt_loaded)
187       {
188         printf ("name = \"%s\", direct_opencount = %d\n",
189                 map->l_name, (int) map->l_direct_opencount);
190         result = 1;
191       }
192
193   return result;
194 }
195
196
197 extern int foo (int a);
198 int
199 foo (int a)
200 {
201   return a - 1;
202 }