1 /* $NetBSD: t_dlinfo.c,v 1.4 2011/03/25 14:47:31 pooka 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_CHECK_EQ(rv, 0); 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(rv, -1); 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_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); 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_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 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