tests/libexec/ld.elf_so/t_dlinfo.c

   1 /*      $NetBSD$        */
   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_TP_ADD_TCS(tp)
  88 {
  89         ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_self);
  90         ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_inval);
  91         ATF_TP_ADD_TC(tp, rtld_dlinfo_linkmap_dlopen);
  92 }