1  /* Copyright (C) 2004-2023 Free Software Foundation, Inc.
       2     This file is part of the GNU C Library.
       3  
       4     The GNU C Library is free software; you can redistribute it and/or
       5     modify it under the terms of the GNU Lesser General Public
       6     License as published by the Free Software Foundation; either
       7     version 2.1 of the License, or (at your option) any later version.
       8  
       9     The GNU C Library is distributed in the hope that it will be useful,
      10     but WITHOUT ANY WARRANTY; without even the implied warranty of
      11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      12     Lesser General Public License for more details.
      13  
      14     You should have received a copy of the GNU Lesser General Public
      15     License along with the GNU C Library; if not, see
      16     <https://www.gnu.org/licenses/>.  */
      17  
      18  #include <link.h>
      19  #include <stddef.h>
      20  #include <stdio.h>
      21  #include <stdlib.h>
      22  
      23  #define SET	0
      24  #define ADD	1
      25  #define REMOVE	2
      26  
      27  #define leq(l,r)	(((r) - (l)) <= ~0ULL / 2)
      28  
      29  static int
      30  callback (struct dl_phdr_info *info, size_t size, void *ptr)
      31  {
      32    static int last_adds = 0, last_subs = 0;
      33    intptr_t cmd = (intptr_t) ptr;
      34  
      35    printf ("  size = %zu\n", size);
      36    if (size < (offsetof (struct dl_phdr_info, dlpi_subs)
      37  	      + sizeof (info->dlpi_subs)))
      38      {
      39        fprintf (stderr, "dl_iterate_phdr failed to pass dlpi_adds/dlpi_subs\n");
      40        exit (5);
      41      }
      42  
      43    printf ("  dlpi_adds = %Lu dlpi_subs = %Lu\n",
      44  	  info->dlpi_adds, info->dlpi_subs);
      45  
      46    switch (cmd)
      47      {
      48      case SET:
      49        break;
      50  
      51      case ADD:
      52        if (leq (info->dlpi_adds, last_adds))
      53  	{
      54  	  fprintf (stderr, "dlpi_adds failed to get incremented!\n");
      55  	  exit (3);
      56  	}
      57        break;
      58  
      59      case REMOVE:
      60        if (leq (info->dlpi_subs, last_subs))
      61  	{
      62  	  fprintf (stderr, "dlpi_subs failed to get incremented!\n");
      63  	  exit (4);
      64  	}
      65        break;
      66      }
      67    last_adds = info->dlpi_adds;
      68    last_subs = info->dlpi_subs;
      69    return -1;
      70  }
      71  
      72  static void *
      73  load (const char *path)
      74  {
      75    void *handle;
      76  
      77    printf ("loading `%s'\n", path);
      78    handle = dlopen (path, RTLD_LAZY);
      79    if (!handle)
      80      exit (1);
      81    dl_iterate_phdr (callback, (void *)(intptr_t) ADD);
      82    return handle;
      83  }
      84  
      85  static void
      86  unload (const char *path, void *handle)
      87  {
      88    printf ("unloading `%s'\n", path);
      89    if (dlclose (handle) < 0)
      90      exit (2);
      91    dl_iterate_phdr (callback, (void *)(intptr_t) REMOVE);
      92  }
      93  
      94  static int
      95  do_test (void)
      96  {
      97    void *handle1, *handle2;
      98  
      99    dl_iterate_phdr (callback, (void *)(intptr_t) SET);
     100    handle1 = load ("firstobj.so");
     101    handle2 = load ("globalmod1.so");
     102    unload ("firstobj.so", handle1);
     103    unload ("globalmod1.so", handle2);
     104    return 0;
     105  }
     106  
     107  #include <support/test-driver.c>