tests/libexec/ld.elf_so/t_dlinfo.c

   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 }