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 __FBSDID("$FreeBSD$");
34
35 #ifndef DSO_LIB
36 #include <atf-c.h>
37 #endif
38
39 #include <crt.h>
40
41 typedef void (*func_ptr)(void);
42
43 extern volatile int jcr_run;
44 extern const func_ptr *jcr_ptr;
45 extern const void *jcr_func_ptr;
46 extern volatile int ctors_run;
47 extern volatile int preinit_array_run;
48 extern volatile int preinit_array_state;
49 extern volatile int init_array_run;
50 extern volatile int init_array_state;
51
52 #ifndef DSO_BASE
53 volatile int jcr_run;
54 const func_ptr *jcr_ptr;
55 volatile int ctors_run;
56 volatile int preinit_array_run;
57 volatile int preinit_array_state = -1;
58 volatile int init_array_run;
59 volatile int init_array_state = -1;
60
61 void _Jv_RegisterClasses(const func_ptr *);
62
63 __section(".jcr") __used static func_ptr jcr_func = (func_ptr)1;
64 const void *jcr_func_ptr = &jcr_func;
65
66 void
_Jv_RegisterClasses(const func_ptr * jcr)67 _Jv_RegisterClasses(const func_ptr *jcr)
68 {
69
70 jcr_run = 1;
71 jcr_ptr = jcr;
72 }
73 #endif
74
75 #ifndef DSO_LIB
76 ATF_TC_WITHOUT_HEAD(jcr_test);
ATF_TC_BODY(jcr_test,tc)77 ATF_TC_BODY(jcr_test, tc)
78 {
79
80 ATF_REQUIRE_MSG(jcr_run == 1, ".jcr not run");
81 ATF_REQUIRE_MSG(jcr_ptr == jcr_func_ptr,
82 "Incorrect pointer passed to _Jv_RegisterClasses");
83 }
84 #endif
85
86 #ifndef DSO_BASE
87 static void
ctors_handler(void)88 ctors_handler(void)
89 {
90
91 ctors_run = 1;
92 }
93 __section(".ctors") __used static func_ptr ctors_func =
94 &ctors_handler;
95 #endif
96
97 #ifndef DSO_LIB
98 ATF_TC_WITHOUT_HEAD(ctors_test);
ATF_TC_BODY(ctors_test,tc)99 ATF_TC_BODY(ctors_test, tc)
100 {
101
102 #ifdef HAVE_CTORS
103 ATF_REQUIRE_MSG(ctors_run == 1, ".ctors not run");
104 #else
105 ATF_REQUIRE_MSG(ctors_run == 0, ".ctors run");
106 #endif
107 }
108 #endif
109
110 #if !defined(DSO_BASE) && !defined(DSO_LIB)
111 static void
preinit_array_handler(void)112 preinit_array_handler(void)
113 {
114
115 preinit_array_run = 1;
116 preinit_array_state = init_array_run;
117 }
118 __section(".preinit_array") __used static func_ptr preinit_array_func =
119 &preinit_array_handler;
120 #endif
121
122 #ifndef DSO_LIB
123 ATF_TC_WITHOUT_HEAD(preinit_array_test);
ATF_TC_BODY(preinit_array_test,tc)124 ATF_TC_BODY(preinit_array_test, tc)
125 {
126
127 #ifdef DSO_BASE
128 /* Check .preinit_array wasn't run in a DSO */
129 ATF_REQUIRE_MSG(preinit_array_run == 0, ".preinit_array run in DSO");
130 #else
131 ATF_REQUIRE_MSG(preinit_array_run == 1, ".preinit_array not run");
132 ATF_REQUIRE_MSG(preinit_array_state == 0,
133 ".preinit_array was not run before .init_array");
134 #endif
135 }
136 #endif
137
138 #ifndef DSO_BASE
139 static void
init_array_handler(void)140 init_array_handler(void)
141 {
142
143 init_array_run = 1;
144 init_array_state = preinit_array_run;
145 }
146 __section(".init_array") __used static func_ptr init_array_func =
147 &init_array_handler;
148 #endif
149
150 #ifndef DSO_LIB
151 ATF_TC_WITHOUT_HEAD(init_array_test);
ATF_TC_BODY(init_array_test,tc)152 ATF_TC_BODY(init_array_test, tc)
153 {
154
155 ATF_REQUIRE_MSG(init_array_run == 1, ".init_array not run");
156 #ifndef DSO_BASE
157 ATF_REQUIRE_MSG(init_array_state == 1,
158 ".init_array was not run after .preinit_array");
159 #endif
160 }
161
ATF_TP_ADD_TCS(tp)162 ATF_TP_ADD_TCS(tp)
163 {
164
165 ATF_TP_ADD_TC(tp, jcr_test);
166 ATF_TP_ADD_TC(tp, ctors_test);
167 ATF_TP_ADD_TC(tp, preinit_array_test);
168 ATF_TP_ADD_TC(tp, init_array_test);
169
170 return (atf_no_error());
171 }
172 #endif
173