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 do \
77 { \
78 for (map = MAPS; map != NULL; map = map->l_next) \
79 if (map->l_type == lt_loaded) \
80 printf ("name = \"%s\", direct_opencount = %d\n", \
81 map->l_name, (int) map->l_direct_opencount); \
82 fflush (stdout); \
83 } \
84 while (0)
85
86
87 int
88 main (int argc, char *argv[])
89 {
90 int debug = argc > 1 && argv[1][0] != '\0';
91 int count = TEST_ROUNDS;
92 int result = 0;
93 struct link_map *map;
94
95 mtrace ();
96
97 /* Just a seed. */
98 srandom (TEST_ROUNDS);
99
100 if (debug)
101 {
102 puts ("in the beginning");
103 OUT;
104 }
105
106 while (count--)
107 {
108 int nr = random () % NTESTS;
109 int index = tests[nr].index;
110
111 printf ("%4d: %4d: ", count + 1, nr);
112 fflush (stdout);
113
114 if (testobjs[index].handle == NULL)
115 {
116 int (*fct) (int);
117
118 /* Load the object. */
119 testobjs[index].handle = dlopen (testobjs[index].name,
120 tests[nr].options);
121 if (testobjs[index].handle == NULL)
122 error (EXIT_FAILURE, 0, "cannot load `%s': %s",
123 testobjs[index].name, dlerror ());
124
125 /* Test the function call. */
126 fct = dlsym (testobjs[index].handle, tests[nr].fname);
127 if (fct == NULL)
128 error (EXIT_FAILURE, 0,
129 "cannot get function `%s' from shared object `%s': %s",
130 tests[nr].fname, testobjs[index].name, dlerror ());
131
132 fct (10);
133
134 printf ("successfully loaded `%s', handle %p\n",
135 testobjs[index].name, testobjs[index].handle);
136 }
137 else
138 {
139 if (dlclose (testobjs[index].handle) != 0)
140 {
141 printf ("failed to close %s\n", testobjs[index].name);
142 result = 1;
143 }
144 else
145 printf ("successfully unloaded `%s', handle %p\n",
146 testobjs[index].name, testobjs[index].handle);
147
148 testobjs[index].handle = NULL;
149
150 if (testobjs[0].handle == NULL
151 && testobjs[1].handle == NULL
152 && testobjs[5].handle == NULL)
153 {
154 /* In this case none of the objects above should be
155 present. */
156 for (map = MAPS; map != NULL; map = map->l_next)
157 if (map->l_type == lt_loaded
158 && (strstr (map->l_name, testobjs[0].name) != NULL
159 || strstr (map->l_name, testobjs[1].name) != NULL
160 || strstr (map->l_name, testobjs[5].name) != NULL))
161 {
162 printf ("`%s' is still loaded\n", map->l_name);
163 result = 1;
164 }
165 }
166 }
167
168 if (debug)
169 OUT;
170 }
171
172 /* Unload all loaded modules. */
173 for (count = 0; count < (int) NOBJS; ++count)
174 if (testobjs[count].handle != NULL)
175 {
176 printf ("\nclose: %s: l_initfini = %p, l_versions = %p\n",
177 testobjs[count].name,
178 ((struct link_map *) testobjs[count].handle)->l_initfini,
179 ((struct link_map *) testobjs[count].handle)->l_versions);
180
181 if (dlclose (testobjs[count].handle) != 0)
182 {
183 printf ("failed to close %s\n", testobjs[count].name);
184 result = 1;
185 }
186 }
187
188 /* Check whether all files are unloaded. */
189 for (map = MAPS; map != NULL; map = map->l_next)
190 if (map->l_type == lt_loaded)
191 {
192 printf ("name = \"%s\", direct_opencount = %d\n",
193 map->l_name, (int) map->l_direct_opencount);
194 result = 1;
195 }
196
197 return result;
198 }
199
200
201 extern int foo (int a);
202 int
203 foo (int a)
204 {
205 return a - 1;
206 }