glibc/elf/neededtest4.c

159 lines
3.4 KiB
C

#include <dlfcn.h>
#include <libintl.h>
#include <link.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAPS ((struct link_map *) _r_debug.r_map)
static int
check_loaded_objects (const char **loaded)
{
struct link_map *lm;
int n;
int *found = NULL;
int errors = 0;
for (n = 0; loaded[n]; n++)
/* NOTHING */;
if (n)
{
found = (int *) alloca (sizeof (int) * n);
memset (found, 0, sizeof (int) * n);
}
printf(" Name\n");
printf(" --------------------------------------------------------\n");
for (lm = MAPS; lm; lm = lm->l_next)
{
if (lm->l_name && lm->l_name[0])
printf(" %s, count = %d\n", lm->l_name, (int) lm->l_direct_opencount);
if (lm->l_type == lt_loaded && lm->l_name)
{
int match = 0;
for (n = 0; loaded[n] != NULL; n++)
{
if (strcmp (basename (loaded[n]), basename (lm->l_name)) == 0)
{
found[n] = 1;
match = 1;
break;
}
}
if (match == 0)
{
++errors;
printf ("ERRORS: %s is not unloaded\n", lm->l_name);
}
}
}
for (n = 0; loaded[n] != NULL; n++)
{
if (found[n] == 0)
{
++errors;
printf ("ERRORS: %s is not loaded\n", loaded[n]);
}
}
return errors;
}
extern void c_function (void);
extern char *dirname (__const char *__filename);
int
main (int argc, char **argv)
{
void *obj;
const char *loaded[] = { NULL, NULL, NULL};
int errors = 0;
void (*f) (void);
const char *dir = dirname (argv [0]);
char *oldfilename;
char *newfilename;
c_function ();
printf ("\nThis is what is in memory now:\n");
errors += check_loaded_objects (loaded);
printf( "Loading shared object neededobj6.so\n");
obj = dlopen( "neededobj6.so", RTLD_LAZY);
if (obj == NULL)
{
printf ("%s\n", dlerror ());
exit (1);
}
f = dlsym (obj, "a2_function");
if (f == NULL)
{
printf ("%s\n", dlerror ());
exit (1);
}
f ();
loaded[0] = "neededobj5.so";
loaded[1] = "neededobj6.so";
errors += check_loaded_objects (loaded);
printf ("Closing neededobj6.so\n");
dlclose (obj);
loaded[0] = NULL;
errors += check_loaded_objects (loaded);
printf ("Rename neededobj5.so\n");
oldfilename = alloca (strlen (dir) + 1 + sizeof ("neededobj5.so"));
strcpy (oldfilename, dir);
strcat (oldfilename, "/");
strcat (oldfilename, "neededobj5.so");
newfilename = alloca (strlen (oldfilename) + sizeof (".renamed"));
strcpy (newfilename, oldfilename);
strcat (newfilename, ".renamed");
if (rename (oldfilename, newfilename))
{
perror ("rename");
exit (1);
}
printf( "Loading shared object neededobj6.so\n");
obj = dlopen( "neededobj6.so", RTLD_LAZY);
if (obj == NULL)
printf ("%s\n", dlerror ());
else
{
printf ("neededobj6.so should fail to load\n");
exit (1);
}
printf( "Loading shared object neededobj1.so\n");
obj = dlopen( "neededobj1.so", RTLD_LAZY);
if (obj == NULL)
{
printf ("%s\n", dlerror ());
exit (1);
}
errors += check_loaded_objects (loaded);
f = dlsym (obj, "c_function");
if (f == NULL)
{
printf ("%s\n", dlerror ());
exit (1);
}
f ();
printf ("Restore neededobj5.so\n");
if (rename (newfilename, oldfilename))
{
perror ("rename");
exit (1);
}
if (errors != 0)
printf ("%d errors found\n", errors);
return errors;
}