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
      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
      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  }