1 /* $NetBSD: t_dlinfo.c,v 1.8 2023/11/24 17:40:20 riastradh Exp $ */
2
3 /*
4 * Copyright (c) 2009 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
30 #include <sys/types.h>
31
32 #include <atf-c.h>
33 #include <dlfcn.h>
34 #include <link_elf.h>
35
36 #include "h_macros.h"
37
38 ATF_TC(rtld_dlinfo_linkmap_self);
ATF_TC_HEAD(rtld_dlinfo_linkmap_self,tc)39 ATF_TC_HEAD(rtld_dlinfo_linkmap_self, tc)
40 {
41 atf_tc_set_md_var(tc, "descr", "dlinfo with RTLD_SELF handle works");
42 }
ATF_TC_BODY(rtld_dlinfo_linkmap_self,tc)43 ATF_TC_BODY(rtld_dlinfo_linkmap_self, tc)
44 {
45 struct link_map *map;
46 int rv;
47
48 rv = dlinfo(RTLD_SELF, RTLD_DI_LINKMAP, &map);
49 ATF_REQUIRE_EQ_MSG(rv, 0, "dlinfo: %s", dlerror());
50 ATF_CHECK((strstr(map->l_name, "t_dlinfo") != NULL));
51 }
52
53 ATF_TC(rtld_dlinfo_linkmap_inval);
ATF_TC_HEAD(rtld_dlinfo_linkmap_inval,tc)54 ATF_TC_HEAD(rtld_dlinfo_linkmap_inval, tc)
55 {
56 atf_tc_set_md_var(tc, "descr", "dlinfo with invalid handle fails");
57 }
ATF_TC_BODY(rtld_dlinfo_linkmap_inval,tc)58 ATF_TC_BODY(rtld_dlinfo_linkmap_inval, tc)
59 {
60 void *v;
61 int rv;
62
63 rv = dlinfo(NULL, RTLD_DI_LINKMAP, &v);
64 ATF_CHECK_EQ_MSG(rv, -1, "rv=%d", rv);
65 }
66
67 ATF_TC(rtld_dlinfo_linkmap_dlopen);
ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen,tc)68 ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen, tc)
69 {
70 atf_tc_set_md_var(tc, "descr", "dlinfo dlopen'd handle works");
71 }
ATF_TC_BODY(rtld_dlinfo_linkmap_dlopen,tc)72 ATF_TC_BODY(rtld_dlinfo_linkmap_dlopen, tc)
73 {
74 struct link_map *map;
75 void *handle;
76 int rv;
77
78 handle = dlopen("libutil.so", RTLD_LAZY);
79 ATF_REQUIRE_MSG(handle, "dlopen: %s", dlerror());
80
81 rv = dlinfo(handle, RTLD_DI_LINKMAP, &map);
82 ATF_REQUIRE_EQ_MSG(rv, 0, "dlinfo: %s", dlerror());
83 ATF_CHECK((strstr(map->l_name, "libutil.so") != NULL));
84 ATF_CHECK_EQ_MSG(dlclose(handle), 0, "dlclose: %s", dlerror());
85 }
86
87 ATF_TC(rtld_dlinfo_linkmap_dlopen_iter);
ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen_iter,tc)88 ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen_iter, tc)
89 {
90 atf_tc_set_md_var(tc, "descr", "dlopen'd dso's show up in the list");
91 }
ATF_TC_BODY(rtld_dlinfo_linkmap_dlopen_iter,tc)92 ATF_TC_BODY(rtld_dlinfo_linkmap_dlopen_iter, tc)
93 {
94 struct link_map *map;
95 void *handle;
96
97 handle = dlopen("libutil.so", RTLD_LAZY);
98 ATF_REQUIRE_MSG(handle, "dlopen: %s", dlerror());
99
100 ATF_REQUIRE_EQ_MSG(dlinfo(RTLD_SELF, RTLD_DI_LINKMAP, &map), 0,
101 "dlinfo: %s", dlerror());
102
103 for (; map->l_next; map = map->l_next)
104 continue;
105 for (; map; map = map->l_prev)
106 if (strstr(map->l_name, "libutil.so") != NULL)
107 break;
108
109 ATF_REQUIRE_MSG(map, "dlopen()d object not found from linkmap");
110 ATF_REQUIRE_MSG(dlopen(map->l_name, RTLD_LAZY) != NULL,
111 "could not dlopen() name in linkmap: %s", dlerror());
112 }
113
ATF_TP_ADD_TCS(tp)114 ATF_TP_ADD_TCS(tp)
115 {
116 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_self);
117 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_inval);
118 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_dlopen);
119 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_dlopen_iter);
120 return atf_no_error();
121 }
122