1 /* $NetBSD: t_dlinfo.c,v 1.5 2017/01/13 21:30:42 christos 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_CHECK_EQ(rv, 0);
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(rv, -1);
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_CHECK(handle);
80
81 rv = dlinfo(handle, RTLD_DI_LINKMAP, &map);
82 ATF_CHECK_EQ(rv, 0);
83 ATF_CHECK((strstr(map->l_name, "libutil.so") != NULL));
84 dlclose(handle);
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_CHECK(handle);
99
100 RZ(dlinfo(RTLD_SELF, RTLD_DI_LINKMAP, &map));
101
102 for (; map->l_next; map = map->l_next)
103 continue;
104 for (; map; map = map->l_prev)
105 if (strstr(map->l_name, "libutil.so") != NULL)
106 break;
107
108 ATF_REQUIRE_MSG(map, "dlopen()d object not found from linkmap");
109 ATF_REQUIRE_MSG(dlopen(map->l_name, RTLD_LAZY) != NULL,
110 "could not dlopen() name in linkmap");
111 }
112
ATF_TP_ADD_TCS(tp)113 ATF_TP_ADD_TCS(tp)
114 {
115 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_self);
116 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_inval);
117 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_dlopen);
118 ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_dlopen_iter);
119 return 0;
120 }
121