1/*-
2 * Copyright (c) 2016 Maksym Sobolyev
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD$");
29
30#include <sys/stat.h>
31#include <dlfcn.h>
32#include <errno.h>
33#include <fcntl.h>
34#include <signal.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <unistd.h>
38
39#include <atf-c.h>
40
41static const char *funname;
42static char *soname;
43
44static void
45sigsegv_handler(int sig __unused)
46{
47        unlink(soname);
48        free(soname);
49        atf_tc_fail("got SIGSEGV in the %s(3)", funname);
50}
51
52ATF_TC(dlopen_empty_test);
53ATF_TC_HEAD(dlopen_empty_test, tc)
54{
55        atf_tc_set_md_var(tc, "descr", "Tests the dlopen() of an empty file "
56                      "returns an error");
57}
58ATF_TC_BODY(dlopen_empty_test, tc)
59{
60        char tempname[] = "/tmp/temp.XXXXXX";
61        char *fname;
62        int fd;
63        void *dlh;
64        struct sigaction act, oact;
65
66        fname = mktemp(tempname);
67        ATF_REQUIRE_MSG(fname != NULL, "mktemp failed; errno=%d", errno);
68        asprintf(&soname, "%s.so", fname);
69        ATF_REQUIRE_MSG(soname != NULL, "asprintf failed; errno=%d", ENOMEM);
70        fd = open(soname, O_WRONLY | O_CREAT | O_TRUNC, DEFFILEMODE);
71        ATF_REQUIRE_MSG(fd != -1, "open(\"%s\") failed; errno=%d", soname, errno);
72        close(fd);
73
74        act.sa_handler = sigsegv_handler;
75        act.sa_flags = 0;
76        sigemptyset(&act.sa_mask);
77        ATF_CHECK_MSG(sigaction(SIGSEGV, &act, &oact) != -1,
78            "sigaction() failed");
79
80        funname = "dlopen";
81        dlh = dlopen(soname, RTLD_LAZY);
82        if (dlh != NULL) {
83                funname = "dlclose";
84                dlclose(dlh);
85        }
86        ATF_REQUIRE_MSG(dlh == NULL, "dlopen(\"%s\") did not fail", soname);
87        unlink(soname);
88        free(soname);
89}
90
91ATF_TP_ADD_TCS(tp)
92{
93
94        ATF_TP_ADD_TC(tp, dlopen_empty_test);
95
96        return (atf_no_error());
97}
98