1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2018 Andrew Turner
5 *
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.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
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.
18 *
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
29 * SUCH DAMAGE.
30 */
31
32 #include <sys/types.h>
33 #include <sys/wait.h>
34
35 #include <errno.h>
36 #include <stdbool.h>
37 #include <stdlib.h>
38 #include <unistd.h>
39
40 #include <atf-c.h>
41
42 #include <crt.h>
43
44 extern bool run_dtors_test;
45 extern bool run_fini_array_test;
46 void dso_handle_check(void);
47
48
49 #ifndef DSO_BASE
50 typedef void (*func_ptr)(void);
51
52 bool run_dtors_test = false;
53 bool run_fini_array_test = false;
54
55 static void
dtors_handler(void)56 dtors_handler(void)
57 {
58
59 if (run_dtors_test)
60 _exit(1);
61 }
62 __section(".dtors") __used static func_ptr dtors_func =
63 &dtors_handler;
64 #endif
65
66 #ifndef DSO_LIB
67 ATF_TC_WITHOUT_HEAD(dtors_test);
ATF_TC_BODY(dtors_test,tc)68 ATF_TC_BODY(dtors_test, tc)
69 {
70 pid_t pid, wpid;
71 int status;
72
73 pid = fork();
74 switch(pid) {
75 case -1:
76 break;
77 case 0:
78 run_dtors_test = true;
79 exit(0);
80 default:
81 while ((wpid = waitpid(pid, &status, 0)) == -1 &&
82 errno == EINTR)
83 ;
84 #ifdef HAVE_CTORS
85 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1,
86 ".dtors failed to run");
87 #else
88 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 0,
89 ".dtors incorrectly ran");
90 #endif
91 break;
92 }
93 }
94 #endif
95
96 #ifndef DSO_BASE
97 static void
fini_array_handler(void)98 fini_array_handler(void)
99 {
100
101 if (run_fini_array_test)
102 _exit(1);
103 }
104 __section(".fini_array") __used static func_ptr fini_array_func =
105 &fini_array_handler;
106 #endif
107
108 #ifndef DSO_LIB
109 ATF_TC_WITHOUT_HEAD(fini_array_test);
ATF_TC_BODY(fini_array_test,tc)110 ATF_TC_BODY(fini_array_test, tc)
111 {
112 pid_t pid, wpid;
113 int status;
114
115 pid = fork();
116 switch(pid) {
117 case -1:
118 break;
119 case 0:
120 run_fini_array_test = true;
121 exit(0);
122 default:
123 while ((wpid = waitpid(pid, &status, 0)) == -1 &&
124 errno == EINTR)
125 ;
126 ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1,
127 ".fini_array failed to run");
128 break;
129 }
130 }
131 #endif
132
133 #ifndef DSO_BASE
134 extern void *__dso_handle;
135
136 void
dso_handle_check(void)137 dso_handle_check(void)
138 {
139 void *dso = __dso_handle;
140
141 #if defined(DSO_LIB) || defined(__PIE__)
142 ATF_REQUIRE_MSG(dso != NULL,
143 "Null __dso_handle in DSO/PIE");
144 #else
145 ATF_REQUIRE_MSG(dso == NULL,
146 "Invalid __dso_handle in non-DSO");
147 #endif
148 }
149 #endif
150
151 #ifndef DSO_LIB
152 ATF_TC_WITHOUT_HEAD(dso_handle_test);
ATF_TC_BODY(dso_handle_test,tc)153 ATF_TC_BODY(dso_handle_test, tc)
154 {
155
156 dso_handle_check();
157 }
158
ATF_TP_ADD_TCS(tp)159 ATF_TP_ADD_TCS(tp)
160 {
161
162 ATF_TP_ADD_TC(tp, dtors_test);
163 ATF_TP_ADD_TC(tp, fini_array_test);
164 ATF_TP_ADD_TC(tp, dso_handle_test);
165
166 return (atf_no_error());
167 }
168 #endif
169