1 /* Test for dlinfo (RTLD_DI_PHDR).
2 Copyright (C) 2022 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19 #include <dlfcn.h>
20 #include <link.h>
21 #include <stdbool.h>
22 #include <stdio.h>
23 #include <string.h>
24 #include <sys/auxv.h>
25
26 #include <support/check.h>
27 #include <support/xdlfcn.h>
28
29 /* Used to verify that the program header array appears as expected
30 among the dl_iterate_phdr callback invocations. */
31
32 struct dlip_callback_args
33 {
34 struct link_map *l; /* l->l_addr is used to find the object. */
35 const ElfW(Phdr) *phdr; /* Expected program header pointed. */
36 int phnum; /* Expected program header count. */
37 bool found; /* True if l->l_addr has been found. */
38 };
39
40 static int
dlip_callback(struct dl_phdr_info * dlpi,size_t size,void * closure)41 dlip_callback (struct dl_phdr_info *dlpi, size_t size, void *closure)
42 {
43 TEST_COMPARE (sizeof (*dlpi), size);
44 struct dlip_callback_args *args = closure;
45
46 if (dlpi->dlpi_addr == args->l->l_addr)
47 {
48 TEST_VERIFY (!args->found);
49 args->found = true;
50 TEST_VERIFY (args->phdr == dlpi->dlpi_phdr);
51 TEST_COMPARE (args->phnum, dlpi->dlpi_phnum);
52 }
53
54 return 0;
55 }
56
57 static int
do_test(void)58 do_test (void)
59 {
60 /* Avoid a copy relocation. */
61 struct r_debug *debug = xdlsym (RTLD_DEFAULT, "_r_debug");
62 struct link_map *l = (struct link_map *) debug->r_map;
63 TEST_VERIFY_EXIT (l != NULL);
64
65 do
66 {
67 printf ("info: checking link map %p (%p) for \"%s\"\n",
68 l, l->l_phdr, l->l_name);
69
70 /* Cause dlerror () to return an error message. */
71 dlsym (RTLD_DEFAULT, "does-not-exist");
72
73 /* Use the extension that link maps are valid dlopen handles. */
74 const ElfW(Phdr) *phdr;
75 int phnum = dlinfo (l, RTLD_DI_PHDR, &phdr);
76 TEST_VERIFY (phnum >= 0);
77 /* Verify that the error message has been cleared. */
78 TEST_COMPARE_STRING (dlerror (), NULL);
79
80 TEST_VERIFY (phdr == l->l_phdr);
81 TEST_COMPARE (phnum, l->l_phnum);
82
83 /* Check that we can find PT_DYNAMIC among the array. */
84 {
85 bool dynamic_found = false;
86 for (int i = 0; i < phnum; ++i)
87 if (phdr[i].p_type == PT_DYNAMIC)
88 {
89 dynamic_found = true;
90 TEST_COMPARE ((ElfW(Addr)) l->l_ld, l->l_addr + phdr[i].p_vaddr);
91 }
92 TEST_VERIFY (dynamic_found);
93 }
94
95 /* Check that dl_iterate_phdr finds the link map with the same
96 program headers. */
97 {
98 struct dlip_callback_args args =
99 {
100 .l = l,
101 .phdr = phdr,
102 .phnum = phnum,
103 .found = false,
104 };
105 TEST_COMPARE (dl_iterate_phdr (dlip_callback, &args), 0);
106 TEST_VERIFY (args.found);
107 }
108
109 if (l->l_prev == NULL)
110 {
111 /* This is the executable, so the information is also
112 available via getauxval. */
113 TEST_COMPARE_STRING (l->l_name, "");
114 TEST_VERIFY (phdr == (const ElfW(Phdr) *) getauxval (AT_PHDR));
115 TEST_COMPARE (phnum, getauxval (AT_PHNUM));
116 }
117
118 l = l->l_next;
119 }
120 while (l != NULL);
121
122 return 0;
123 }
124
125 #include <support/test-driver.c>
126