t_dlinfo.c revision 314817 
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
38ATF_TC(rtld_dlinfo_linkmap_self);
39ATF_TC_HEAD(rtld_dlinfo_linkmap_self, tc)
40{
41	atf_tc_set_md_var(tc, "descr", "dlinfo with RTLD_SELF handle works");
42}
43ATF_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
53ATF_TC(rtld_dlinfo_linkmap_inval);
54ATF_TC_HEAD(rtld_dlinfo_linkmap_inval, tc)
55{
56	atf_tc_set_md_var(tc, "descr", "dlinfo with invalid handle fails");
57}
58ATF_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
67ATF_TC(rtld_dlinfo_linkmap_dlopen);
68ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen, tc)
69{
70	atf_tc_set_md_var(tc, "descr", "dlinfo dlopen'd handle works");
71}
72ATF_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
87ATF_TC(rtld_dlinfo_linkmap_dlopen_iter);
88ATF_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}
92ATF_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
113ATF_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