2 * SPDX-License-Identifier: BSD-2-Clause
4 * Copyright (c) 2018 Andrew Turner
6 * This software was developed by SRI International and the University of
7 * Cambridge Computer Laboratory under DARPA/AFRL contract FA8750-10-C-0237
8 * ("CTSRD"), as part of the DARPA CRASH research programme.
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 #include <sys/cdefs.h>
33 __FBSDID("$FreeBSD$");
35 #include <sys/types.h>
47 extern bool run_dtors_test;
48 extern bool run_fini_array_test;
49 void dso_handle_check(void);
53 typedef void (*func_ptr)(void);
55 bool run_dtors_test = false;
56 bool run_fini_array_test = false;
65 __section(".dtors") __used static func_ptr dtors_func =
70 ATF_TC_WITHOUT_HEAD(dtors_test);
71 ATF_TC_BODY(dtors_test, tc)
81 run_dtors_test = true;
84 while ((wpid = waitpid(pid, &status, 0)) == -1 &&
88 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1,
89 ".dtors failed to run");
91 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 0,
92 ".dtors incorrectly ran");
101 fini_array_handler(void)
104 if (run_fini_array_test)
107 __section(".fini_array") __used static func_ptr fini_array_func =
112 ATF_TC_WITHOUT_HEAD(fini_array_test);
113 ATF_TC_BODY(fini_array_test, tc)
123 run_fini_array_test = true;
126 while ((wpid = waitpid(pid, &status, 0)) == -1 &&
129 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1,
130 ".fini_array failed to run");
137 extern void *__dso_handle;
140 dso_handle_check(void)
142 void *dso = __dso_handle;
145 ATF_REQUIRE_MSG(dso != NULL,
146 "Null __dso_handle in DSO");
148 ATF_REQUIRE_MSG(dso == NULL,
149 "Invalid __dso_handle in non-DSO");
155 ATF_TC_WITHOUT_HEAD(dso_handle_test);
156 ATF_TC_BODY(dso_handle_test, tc)
165 ATF_TP_ADD_TC(tp, dtors_test);
166 ATF_TP_ADD_TC(tp, fini_array_test);
167 ATF_TP_ADD_TC(tp, dso_handle_test);
169 return (atf_no_error());