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