Home | History | Annotate | Line # | Download | only in ld.elf_so
t_dlinfo.c revision 1.8 
      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);
     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 }
     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);
     54 ATF_TC_HEAD(rtld_dlinfo_linkmap_inval, tc)
     55 {
     56 	atf_tc_set_md_var(tc, "descr", "dlinfo with invalid handle fails");
     57 }
     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);
     68 ATF_TC_HEAD(rtld_dlinfo_linkmap_dlopen, tc)
     69 {
     70 	atf_tc_set_md_var(tc, "descr", "dlinfo dlopen'd handle works");
     71 }
     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);
     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 }
     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 
    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