1 #include <dlfcn.h>
2 #include <libintl.h>
3 #include <link.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <string.h>
7
8 #define MAPS ((struct link_map *) _r_debug.r_map)
9
10 static int
check_loaded_objects(const char ** loaded)11 check_loaded_objects (const char **loaded)
12 {
13 struct link_map *lm;
14 int n;
15 int *found = NULL;
16 int errors = 0;
17
18 for (n = 0; loaded[n]; n++)
19 /* NOTHING */;
20
21 if (n)
22 {
23 found = (int *) alloca (sizeof (int) * n);
24 memset (found, 0, sizeof (int) * n);
25 }
26
27 printf(" Name\n");
28 printf(" --------------------------------------------------------\n");
29 for (lm = MAPS; lm; lm = lm->l_next)
30 {
31 if (lm->l_name && lm->l_name[0])
32 printf(" %s, count = %d\n", lm->l_name, (int) lm->l_direct_opencount);
33 if (lm->l_type == lt_loaded && lm->l_name)
34 {
35 int match = 0;
36 for (n = 0; loaded[n] != NULL; n++)
37 {
38 if (strcmp (basename (loaded[n]), basename (lm->l_name)) == 0)
39 {
40 found[n] = 1;
41 match = 1;
42 break;
43 }
44 }
45
46 if (match == 0)
47 {
48 ++errors;
49 printf ("ERRORS: %s is not unloaded\n", lm->l_name);
50 }
51 }
52 }
53
54 for (n = 0; loaded[n] != NULL; n++)
55 {
56 if (found[n] == 0)
57 {
58 ++errors;
59 printf ("ERRORS: %s is not loaded\n", loaded[n]);
60 }
61 }
62
63 return errors;
64 }
65
66 extern void c_function (void);
67 extern char *dirname (const char *__filename);
68
69 int
main(int argc,char ** argv)70 main (int argc, char **argv)
71 {
72 void *obj;
73 const char *loaded[] = { NULL, NULL, NULL};
74 int errors = 0;
75 void (*f) (void);
76 const char *dir = dirname (argv [0]);
77 char *oldfilename;
78 char *newfilename;
79
80 c_function ();
81
82 printf ("\nThis is what is in memory now:\n");
83 errors += check_loaded_objects (loaded);
84
85 printf( "Loading shared object neededobj6.so\n");
86 obj = dlopen( "neededobj6.so", RTLD_LAZY);
87 if (obj == NULL)
88 {
89 printf ("%s\n", dlerror ());
90 exit (1);
91 }
92 f = dlsym (obj, "a2_function");
93 if (f == NULL)
94 {
95 printf ("%s\n", dlerror ());
96 exit (1);
97 }
98 f ();
99 loaded[0] = "neededobj5.so";
100 loaded[1] = "neededobj6.so";
101 errors += check_loaded_objects (loaded);
102
103 printf ("Closing neededobj6.so\n");
104 dlclose (obj);
105 loaded[0] = NULL;
106 errors += check_loaded_objects (loaded);
107
108 printf ("Rename neededobj5.so\n");
109 oldfilename = alloca (strlen (dir) + 1 + sizeof ("neededobj5.so"));
110 strcpy (oldfilename, dir);
111 strcat (oldfilename, "/");
112 strcat (oldfilename, "neededobj5.so");
113 newfilename = alloca (strlen (oldfilename) + sizeof (".renamed"));
114 strcpy (newfilename, oldfilename);
115 strcat (newfilename, ".renamed");
116 if (rename (oldfilename, newfilename))
117 {
118 perror ("rename");
119 exit (1);
120 }
121
122 printf( "Loading shared object neededobj6.so\n");
123 obj = dlopen( "neededobj6.so", RTLD_LAZY);
124 if (obj == NULL)
125 printf ("%s\n", dlerror ());
126 else
127 {
128 printf ("neededobj6.so should fail to load\n");
129 exit (1);
130 }
131
132 printf( "Loading shared object neededobj1.so\n");
133 obj = dlopen( "neededobj1.so", RTLD_LAZY);
134 if (obj == NULL)
135 {
136 printf ("%s\n", dlerror ());
137 exit (1);
138 }
139 errors += check_loaded_objects (loaded);
140 f = dlsym (obj, "c_function");
141 if (f == NULL)
142 {
143 printf ("%s\n", dlerror ());
144 exit (1);
145 }
146 f ();
147
148 printf ("Restore neededobj5.so\n");
149 if (rename (newfilename, oldfilename))
150 {
151 perror ("rename");
152 exit (1);
153 }
154
155 if (errors != 0)
156 printf ("%d errors found\n", errors);
157 return errors;
158 }
159