t_ptrace_wait.h revision 1.38 1 1.38 riastrad /* $NetBSD: t_ptrace_wait.h,v 1.38 2025/05/02 02:24:44 riastradh Exp $ */
2 1.1 kamil
3 1.1 kamil /*-
4 1.14 kamil * Copyright (c) 2016, 2017, 2018, 2019 The NetBSD Foundation, Inc.
5 1.1 kamil * All rights reserved.
6 1.1 kamil *
7 1.1 kamil * Redistribution and use in source and binary forms, with or without
8 1.1 kamil * modification, are permitted provided that the following conditions
9 1.1 kamil * are met:
10 1.1 kamil * 1. Redistributions of source code must retain the above copyright
11 1.1 kamil * notice, this list of conditions and the following disclaimer.
12 1.1 kamil * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 kamil * notice, this list of conditions and the following disclaimer in the
14 1.1 kamil * documentation and/or other materials provided with the distribution.
15 1.1 kamil *
16 1.1 kamil * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 1.1 kamil * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 1.1 kamil * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 1.1 kamil * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 1.1 kamil * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 1.1 kamil * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 1.1 kamil * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 1.1 kamil * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 1.1 kamil * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 1.1 kamil * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 1.1 kamil * POSSIBILITY OF SUCH DAMAGE.
27 1.1 kamil */
28 1.1 kamil
29 1.1 kamil /* Detect plain wait(2) use-case */
30 1.1 kamil #if !defined(TWAIT_WAITPID) && \
31 1.1 kamil !defined(TWAIT_WAITID) && \
32 1.1 kamil !defined(TWAIT_WAIT3) && \
33 1.1 kamil !defined(TWAIT_WAIT4) && \
34 1.1 kamil !defined(TWAIT_WAIT6)
35 1.1 kamil #define TWAIT_WAIT
36 1.1 kamil #endif
37 1.1 kamil
38 1.1 kamil /*
39 1.1 kamil * There are two classes of wait(2)-like functions:
40 1.1 kamil * - wait4(2)-like accepting pid_t, optional options parameter, struct rusage*
41 1.1 kamil * - wait6(2)-like accepting idtype_t, id_t, struct wrusage, mandatory options
42 1.1 kamil *
43 1.1 kamil * The TWAIT_FNAME value is to be used for convenience in debug messages.
44 1.1 kamil *
45 1.1 kamil * The TWAIT_GENERIC() macro is designed to reuse the same unmodified
46 1.1 kamil * code with as many wait(2)-like functions as possible.
47 1.1 kamil *
48 1.1 kamil * In a common use-case wait4(2) and wait6(2)-like function can work the almost
49 1.1 kamil * the same way, however there are few important differences:
50 1.1 kamil * wait6(2) must specify P_PID for idtype to match wpid from wait4(2).
51 1.1 kamil * To behave like wait4(2), wait6(2) the 'options' to wait must include
52 1.1 kamil * WEXITED|WTRUNCATED.
53 1.1 kamil *
54 1.1 kamil * There are two helper macros (they purpose it to mach more than one
55 1.1 kamil * wait(2)-like function):
56 1.1 kamil * The TWAIT_HAVE_STATUS - specifies whether a function can retrieve
57 1.1 kamil * status (as integer value).
58 1.1 kamil * The TWAIT_HAVE_PID - specifies whether a function can request
59 1.1 kamil * exact process identifier
60 1.1 kamil * The TWAIT_HAVE_RUSAGE - specifies whether a function can request
61 1.1 kamil * the struct rusage value
62 1.1 kamil *
63 1.1 kamil */
64 1.1 kamil
65 1.1 kamil #if defined(TWAIT_WAIT)
66 1.1 kamil # define TWAIT_FNAME "wait"
67 1.1 kamil # define TWAIT_WAIT4TYPE(a,b,c,d) wait((b))
68 1.1 kamil # define TWAIT_GENERIC(a,b,c) wait((b))
69 1.1 kamil # define TWAIT_HAVE_STATUS 1
70 1.1 kamil #elif defined(TWAIT_WAITPID)
71 1.1 kamil # define TWAIT_FNAME "waitpid"
72 1.1 kamil # define TWAIT_WAIT4TYPE(a,b,c,d) waitpid((a),(b),(c))
73 1.1 kamil # define TWAIT_GENERIC(a,b,c) waitpid((a),(b),(c))
74 1.1 kamil # define TWAIT_HAVE_PID 1
75 1.1 kamil # define TWAIT_HAVE_STATUS 1
76 1.19 kamil # define TWAIT_HAVE_OPTIONS 1
77 1.1 kamil #elif defined(TWAIT_WAITID)
78 1.1 kamil # define TWAIT_FNAME "waitid"
79 1.1 kamil # define TWAIT_GENERIC(a,b,c) \
80 1.1 kamil waitid(P_PID,(a),NULL,(c)|WEXITED|WTRAPPED)
81 1.1 kamil # define TWAIT_WAIT6TYPE(a,b,c,d,e,f) waitid((a),(b),(f),(d))
82 1.1 kamil # define TWAIT_HAVE_PID 1
83 1.19 kamil # define TWAIT_HAVE_OPTIONS 1
84 1.1 kamil #elif defined(TWAIT_WAIT3)
85 1.1 kamil # define TWAIT_FNAME "wait3"
86 1.1 kamil # define TWAIT_WAIT4TYPE(a,b,c,d) wait3((b),(c),(d))
87 1.1 kamil # define TWAIT_GENERIC(a,b,c) wait3((b),(c),NULL)
88 1.1 kamil # define TWAIT_HAVE_STATUS 1
89 1.1 kamil # define TWAIT_HAVE_RUSAGE 1
90 1.19 kamil # define TWAIT_HAVE_OPTIONS 1
91 1.1 kamil #elif defined(TWAIT_WAIT4)
92 1.1 kamil # define TWAIT_FNAME "wait4"
93 1.1 kamil # define TWAIT_WAIT4TYPE(a,b,c,d) wait4((a),(b),(c),(d))
94 1.1 kamil # define TWAIT_GENERIC(a,b,c) wait4((a),(b),(c),NULL)
95 1.1 kamil # define TWAIT_HAVE_PID 1
96 1.1 kamil # define TWAIT_HAVE_STATUS 1
97 1.1 kamil # define TWAIT_HAVE_RUSAGE 1
98 1.19 kamil # define TWAIT_HAVE_OPTIONS 1
99 1.1 kamil #elif defined(TWAIT_WAIT6)
100 1.1 kamil # define TWAIT_FNAME "wait6"
101 1.1 kamil # define TWAIT_WAIT6TYPE(a,b,c,d,e,f) wait6((a),(b),(c),(d),(e),(f))
102 1.1 kamil # define TWAIT_GENERIC(a,b,c) \
103 1.1 kamil wait6(P_PID,(a),(b),(c)|WEXITED|WTRAPPED,NULL,NULL)
104 1.1 kamil # define TWAIT_HAVE_PID 1
105 1.1 kamil # define TWAIT_HAVE_STATUS 1
106 1.19 kamil # define TWAIT_HAVE_OPTIONS 1
107 1.1 kamil #endif
108 1.1 kamil
109 1.1 kamil /*
110 1.1 kamil * There are 3 groups of tests:
111 1.1 kamil * - TWAIT_GENERIC() (wait, wait2, waitpid, wait3, wait4, wait6)
112 1.1 kamil * - TWAIT_WAIT4TYPE() (wait2, waitpid, wait3, wait4)
113 1.1 kamil * - TWAIT_WAIT6TYPE() (waitid, wait6)
114 1.1 kamil *
115 1.1 kamil * Tests only in the above categories are allowed. However some tests are not
116 1.1 kamil * possible in the context requested functionality to be verified, therefore
117 1.1 kamil * there are helper macros:
118 1.1 kamil * - TWAIT_HAVE_PID (wait2, waitpid, waitid, wait4, wait6)
119 1.1 kamil * - TWAIT_HAVE_STATUS (wait, wait2, waitpid, wait3, wait4, wait6)
120 1.1 kamil * - TWAIT_HAVE_RUSAGE (wait3, wait4)
121 1.1 kamil * - TWAIT_HAVE_RETPID (wait, wait2, waitpid, wait3, wait4, wait6)
122 1.1 kamil *
123 1.1 kamil * If there is an intention to test e.g. wait6(2) specific features in the
124 1.1 kamil * ptrace(2) context, find the most matching group and with #ifdefs reduce
125 1.1 kamil * functionality of less featured than wait6(2) interface (TWAIT_WAIT6TYPE).
126 1.1 kamil *
127 1.1 kamil * For clarity never use negative preprocessor checks, like:
128 1.1 kamil * #if !defined(TWAIT_WAIT4)
129 1.1 kamil * always refer to checks for positive values.
130 1.1 kamil */
131 1.1 kamil
132 1.1 kamil #define TEST_REQUIRE_EQ(x, y) \
133 1.1 kamil do { \
134 1.1 kamil uintmax_t vx = (x); \
135 1.1 kamil uintmax_t vy = (y); \
136 1.1 kamil int ret = vx == vy; \
137 1.1 kamil if (!ret) \
138 1.1 kamil ATF_REQUIRE_EQ_MSG(vx, vy, "%s(%ju) == %s(%ju)", \
139 1.1 kamil #x, vx, #y, vy); \
140 1.1 kamil } while (/*CONSTCOND*/0)
141 1.1 kamil
142 1.1 kamil /*
143 1.1 kamil * A child process cannot call atf functions and expect them to magically
144 1.1 kamil * work like in the parent.
145 1.1 kamil * The printf(3) messaging from a child will not work out of the box as well
146 1.34 andvar * without establishing a communication protocol with its parent. To not
147 1.1 kamil * overcomplicate the tests - do not log from a child and use err(3)/errx(3)
148 1.1 kamil * wrapped with FORKEE_ASSERT()/FORKEE_ASSERTX() as that is guaranteed to work.
149 1.1 kamil */
150 1.1 kamil #define FORKEE_ASSERT_EQ(x, y) \
151 1.1 kamil do { \
152 1.1 kamil uintmax_t vx = (x); \
153 1.1 kamil uintmax_t vy = (y); \
154 1.1 kamil int ret = vx == vy; \
155 1.1 kamil if (!ret) \
156 1.1 kamil errx(EXIT_FAILURE, "%s:%d %s(): Assertion failed for: " \
157 1.1 kamil "%s(%ju) == %s(%ju)", __FILE__, __LINE__, __func__, \
158 1.1 kamil #x, vx, #y, vy); \
159 1.1 kamil } while (/*CONSTCOND*/0)
160 1.1 kamil
161 1.5 kamil #define FORKEE_ASSERT_NEQ(x, y) \
162 1.5 kamil do { \
163 1.5 kamil uintmax_t vx = (x); \
164 1.5 kamil uintmax_t vy = (y); \
165 1.5 kamil int ret = vx != vy; \
166 1.5 kamil if (!ret) \
167 1.5 kamil errx(EXIT_FAILURE, "%s:%d %s(): Assertion failed for: " \
168 1.5 kamil "%s(%ju) != %s(%ju)", __FILE__, __LINE__, __func__, \
169 1.5 kamil #x, vx, #y, vy); \
170 1.5 kamil } while (/*CONSTCOND*/0)
171 1.5 kamil
172 1.38 riastrad __unused /* used by FORKEE_ASSERT_MEMEQ, otherwise not used */
173 1.38 riastrad static void
174 1.38 riastrad hexdump(const char *title, const void *buf, size_t len)
175 1.38 riastrad {
176 1.38 riastrad const unsigned char *p = buf;
177 1.38 riastrad size_t i;
178 1.38 riastrad
179 1.38 riastrad fprintf(stderr, "%s (%zu bytes)\n", title, len);
180 1.38 riastrad for (i = 0; i < len; i++) {
181 1.38 riastrad if ((i % 8) == 0)
182 1.38 riastrad fprintf(stderr, " ");
183 1.38 riastrad fprintf(stderr, "%02hhx", p[i]);
184 1.38 riastrad if ((i % 16) == 0)
185 1.38 riastrad fprintf(stderr, "\n");
186 1.38 riastrad }
187 1.38 riastrad if (i % 16)
188 1.38 riastrad fprintf(stderr, "\n");
189 1.38 riastrad }
190 1.38 riastrad
191 1.38 riastrad #define FORKEE_ASSERT_MEMEQ(x, y, n) \
192 1.38 riastrad do { \
193 1.38 riastrad const void *const vx = (x); \
194 1.38 riastrad const void *const vy = (y); \
195 1.38 riastrad const size_t vn = (n); \
196 1.38 riastrad \
197 1.38 riastrad if (__predict_true(memcmp(vx, vy, vn) == 0)) \
198 1.38 riastrad break; \
199 1.38 riastrad fprintf(stderr, "%s != %s (%s = %zu bytes)\n", #x, #y, #n, vn); \
200 1.38 riastrad hexdump(#x, vx, vn); \
201 1.38 riastrad hexdump(#y, vy, vn); \
202 1.38 riastrad } while (/*CONSTCOND*/0)
203 1.38 riastrad
204 1.1 kamil #define FORKEE_ASSERTX(x) \
205 1.1 kamil do { \
206 1.1 kamil int ret = (x); \
207 1.1 kamil if (!ret) \
208 1.1 kamil errx(EXIT_FAILURE, "%s:%d %s(): Assertion failed for: %s",\
209 1.1 kamil __FILE__, __LINE__, __func__, #x); \
210 1.1 kamil } while (/*CONSTCOND*/0)
211 1.1 kamil
212 1.1 kamil #define FORKEE_ASSERT(x) \
213 1.1 kamil do { \
214 1.1 kamil int ret = (x); \
215 1.1 kamil if (!ret) \
216 1.1 kamil err(EXIT_FAILURE, "%s:%d %s(): Assertion failed for: %s",\
217 1.1 kamil __FILE__, __LINE__, __func__, #x); \
218 1.1 kamil } while (/*CONSTCOND*/0)
219 1.1 kamil
220 1.38 riastrad #define FORKEE_PTHREAD(x) \
221 1.38 riastrad do { \
222 1.38 riastrad int _forkee_pthread_error = (x); \
223 1.38 riastrad if (_forkee_pthread_error) { \
224 1.38 riastrad errno = _forkee_pthread_error; \
225 1.38 riastrad err(EXIT_FAILURE, "%s:%d %s(): %s", __FILE__, __LINE__, \
226 1.38 riastrad __func__, #x); \
227 1.38 riastrad } \
228 1.38 riastrad } while (/*CONSTCOND*/0)
229 1.38 riastrad
230 1.1 kamil /*
231 1.1 kamil * Simplify logic for functions using general purpose registers add HAVE_GPREGS
232 1.1 kamil *
233 1.1 kamil * For platforms that do not implement all needed calls for simplicity assume
234 1.1 kamil * that they are unsupported at all.
235 1.1 kamil */
236 1.1 kamil #if defined(PT_GETREGS) \
237 1.1 kamil && defined(PT_SETREGS) \
238 1.1 kamil && defined(PTRACE_REG_PC) \
239 1.1 kamil && defined(PTRACE_REG_SET_PC) \
240 1.1 kamil && defined(PTRACE_REG_SP) \
241 1.1 kamil && defined(PTRACE_REG_INTRV)
242 1.1 kamil #define HAVE_GPREGS
243 1.1 kamil #endif
244 1.1 kamil
245 1.1 kamil /* Add guards for floating point registers */
246 1.1 kamil #if defined(PT_GETFPREGS) \
247 1.1 kamil && defined(PT_SETFPREGS)
248 1.1 kamil #define HAVE_FPREGS
249 1.1 kamil #endif
250 1.1 kamil
251 1.1 kamil /* Add guards for cpu debug registers */
252 1.1 kamil #if defined(PT_GETDBREGS) \
253 1.1 kamil && defined(PT_SETDBREGS)
254 1.1 kamil #define HAVE_DBREGS
255 1.1 kamil #endif
256 1.1 kamil
257 1.1 kamil /*
258 1.1 kamil * If waitid(2) returns because one or more processes have a state change to
259 1.1 kamil * report, 0 is returned. If an error is detected, a value of -1 is returned
260 1.1 kamil * and errno is set to indicate the error. If WNOHANG is specified and there
261 1.1 kamil * are no stopped, continued or exited children, 0 is returned.
262 1.1 kamil */
263 1.1 kamil #if defined(TWAIT_WAITID)
264 1.1 kamil #define TWAIT_REQUIRE_SUCCESS(a,b) TEST_REQUIRE_EQ((a), 0)
265 1.1 kamil #define TWAIT_REQUIRE_FAILURE(a,b) ATF_REQUIRE_ERRNO((a),(b) == -1)
266 1.1 kamil #define FORKEE_REQUIRE_SUCCESS(a,b) FORKEE_ASSERT_EQ(a, 0)
267 1.1 kamil #define FORKEE_REQUIRE_FAILURE(a,b) \
268 1.1 kamil FORKEE_ASSERTX(((a) == errno) && ((b) == -1))
269 1.1 kamil #else
270 1.1 kamil #define TWAIT_REQUIRE_SUCCESS(a,b) TEST_REQUIRE_EQ((a), (b))
271 1.1 kamil #define TWAIT_REQUIRE_FAILURE(a,b) ATF_REQUIRE_ERRNO((a),(b) == -1)
272 1.1 kamil #define FORKEE_REQUIRE_SUCCESS(a,b) FORKEE_ASSERT_EQ(a, b)
273 1.1 kamil #define FORKEE_REQUIRE_FAILURE(a,b) \
274 1.1 kamil FORKEE_ASSERTX(((a) == errno) && ((b) == -1))
275 1.1 kamil #endif
276 1.1 kamil
277 1.1 kamil /*
278 1.1 kamil * Helper tools to verify whether status reports exited value
279 1.1 kamil */
280 1.1 kamil #if TWAIT_HAVE_STATUS
281 1.1 kamil static void __used
282 1.1 kamil validate_status_exited(int status, int expected)
283 1.1 kamil {
284 1.1 kamil ATF_REQUIRE_MSG(WIFEXITED(status), "Reported !exited process");
285 1.1 kamil ATF_REQUIRE_MSG(!WIFCONTINUED(status), "Reported continued process");
286 1.1 kamil ATF_REQUIRE_MSG(!WIFSIGNALED(status), "Reported signaled process");
287 1.1 kamil ATF_REQUIRE_MSG(!WIFSTOPPED(status), "Reported stopped process");
288 1.1 kamil
289 1.1 kamil ATF_REQUIRE_EQ_MSG(WEXITSTATUS(status), expected,
290 1.1 kamil "The process has exited with invalid value %d != %d",
291 1.1 kamil WEXITSTATUS(status), expected);
292 1.1 kamil }
293 1.1 kamil
294 1.1 kamil static void __used
295 1.1 kamil forkee_status_exited(int status, int expected)
296 1.1 kamil {
297 1.1 kamil FORKEE_ASSERTX(WIFEXITED(status));
298 1.1 kamil FORKEE_ASSERTX(!WIFCONTINUED(status));
299 1.1 kamil FORKEE_ASSERTX(!WIFSIGNALED(status));
300 1.1 kamil FORKEE_ASSERTX(!WIFSTOPPED(status));
301 1.1 kamil
302 1.1 kamil FORKEE_ASSERT_EQ(WEXITSTATUS(status), expected);
303 1.1 kamil }
304 1.1 kamil
305 1.1 kamil static void __used
306 1.1 kamil validate_status_continued(int status)
307 1.1 kamil {
308 1.1 kamil ATF_REQUIRE_MSG(!WIFEXITED(status), "Reported exited process");
309 1.1 kamil ATF_REQUIRE_MSG(WIFCONTINUED(status), "Reported !continued process");
310 1.1 kamil ATF_REQUIRE_MSG(!WIFSIGNALED(status), "Reported signaled process");
311 1.1 kamil ATF_REQUIRE_MSG(!WIFSTOPPED(status), "Reported stopped process");
312 1.1 kamil }
313 1.1 kamil
314 1.1 kamil static void __used
315 1.1 kamil forkee_status_continued(int status)
316 1.1 kamil {
317 1.1 kamil FORKEE_ASSERTX(!WIFEXITED(status));
318 1.1 kamil FORKEE_ASSERTX(WIFCONTINUED(status));
319 1.1 kamil FORKEE_ASSERTX(!WIFSIGNALED(status));
320 1.1 kamil FORKEE_ASSERTX(!WIFSTOPPED(status));
321 1.1 kamil }
322 1.1 kamil
323 1.1 kamil static void __used
324 1.1 kamil validate_status_signaled(int status, int expected_termsig, int expected_core)
325 1.1 kamil {
326 1.1 kamil ATF_REQUIRE_MSG(!WIFEXITED(status), "Reported exited process");
327 1.1 kamil ATF_REQUIRE_MSG(!WIFCONTINUED(status), "Reported continued process");
328 1.1 kamil ATF_REQUIRE_MSG(WIFSIGNALED(status), "Reported !signaled process");
329 1.1 kamil ATF_REQUIRE_MSG(!WIFSTOPPED(status), "Reported stopped process");
330 1.1 kamil
331 1.1 kamil ATF_REQUIRE_EQ_MSG(WTERMSIG(status), expected_termsig,
332 1.1 kamil "Unexpected signal received");
333 1.1 kamil
334 1.3 kamil ATF_REQUIRE_EQ_MSG(!!WCOREDUMP(status), expected_core,
335 1.1 kamil "Unexpectedly core file %s generated", expected_core ? "not" : "");
336 1.1 kamil }
337 1.1 kamil
338 1.1 kamil static void __used
339 1.1 kamil forkee_status_signaled(int status, int expected_termsig, int expected_core)
340 1.1 kamil {
341 1.1 kamil FORKEE_ASSERTX(!WIFEXITED(status));
342 1.1 kamil FORKEE_ASSERTX(!WIFCONTINUED(status));
343 1.1 kamil FORKEE_ASSERTX(WIFSIGNALED(status));
344 1.1 kamil FORKEE_ASSERTX(!WIFSTOPPED(status));
345 1.1 kamil
346 1.1 kamil FORKEE_ASSERT_EQ(WTERMSIG(status), expected_termsig);
347 1.3 kamil FORKEE_ASSERT_EQ(!!WCOREDUMP(status), expected_core);
348 1.1 kamil }
349 1.1 kamil
350 1.1 kamil static void __used
351 1.1 kamil validate_status_stopped(int status, int expected)
352 1.1 kamil {
353 1.1 kamil ATF_REQUIRE_MSG(!WIFEXITED(status), "Reported exited process");
354 1.1 kamil ATF_REQUIRE_MSG(!WIFCONTINUED(status), "Reported continued process");
355 1.1 kamil ATF_REQUIRE_MSG(!WIFSIGNALED(status), "Reported signaled process");
356 1.1 kamil ATF_REQUIRE_MSG(WIFSTOPPED(status), "Reported !stopped process");
357 1.1 kamil
358 1.1 kamil char st[128], ex[128];
359 1.1 kamil strlcpy(st, strsignal(WSTOPSIG(status)), sizeof(st));
360 1.1 kamil strlcpy(ex, strsignal(expected), sizeof(ex));
361 1.1 kamil
362 1.1 kamil ATF_REQUIRE_EQ_MSG(WSTOPSIG(status), expected,
363 1.1 kamil "Unexpected stop signal received [%s] != [%s]", st, ex);
364 1.1 kamil }
365 1.1 kamil
366 1.1 kamil static void __used
367 1.1 kamil forkee_status_stopped(int status, int expected)
368 1.1 kamil {
369 1.1 kamil FORKEE_ASSERTX(!WIFEXITED(status));
370 1.1 kamil FORKEE_ASSERTX(!WIFCONTINUED(status));
371 1.1 kamil FORKEE_ASSERTX(!WIFSIGNALED(status));
372 1.1 kamil FORKEE_ASSERTX(WIFSTOPPED(status));
373 1.1 kamil
374 1.1 kamil FORKEE_ASSERT_EQ(WSTOPSIG(status), expected);
375 1.1 kamil }
376 1.1 kamil #else
377 1.1 kamil #define validate_status_exited(a,b)
378 1.1 kamil #define forkee_status_exited(a,b)
379 1.1 kamil #define validate_status_continued(a,b)
380 1.1 kamil #define forkee_status_continued(a,b)
381 1.1 kamil #define validate_status_signaled(a,b,c)
382 1.1 kamil #define forkee_status_signaled(a,b,c)
383 1.1 kamil #define validate_status_stopped(a,b)
384 1.1 kamil #define forkee_status_stopped(a,b)
385 1.1 kamil #endif
386 1.1 kamil
387 1.1 kamil /* This function is currently designed to be run in the main/parent process */
388 1.1 kamil static void __used
389 1.2 kamil await_zombie_raw(pid_t process, useconds_t ms)
390 1.1 kamil {
391 1.1 kamil struct kinfo_proc2 p;
392 1.1 kamil size_t len = sizeof(p);
393 1.1 kamil
394 1.1 kamil const int name[] = {
395 1.1 kamil [0] = CTL_KERN,
396 1.1 kamil [1] = KERN_PROC2,
397 1.1 kamil [2] = KERN_PROC_PID,
398 1.1 kamil [3] = process,
399 1.1 kamil [4] = sizeof(p),
400 1.1 kamil [5] = 1
401 1.1 kamil };
402 1.1 kamil
403 1.1 kamil const size_t namelen = __arraycount(name);
404 1.1 kamil
405 1.1 kamil /* Await the process becoming a zombie */
406 1.1 kamil while(1) {
407 1.1 kamil ATF_REQUIRE(sysctl(name, namelen, &p, &len, NULL, 0) == 0);
408 1.1 kamil
409 1.1 kamil if (p.p_stat == LSZOMB)
410 1.1 kamil break;
411 1.1 kamil
412 1.2 kamil if (ms > 0) {
413 1.2 kamil ATF_REQUIRE(usleep(ms) == 0);
414 1.2 kamil }
415 1.1 kamil }
416 1.1 kamil }
417 1.1 kamil
418 1.2 kamil static void __used
419 1.2 kamil await_zombie(pid_t process)
420 1.2 kamil {
421 1.2 kamil
422 1.2 kamil await_zombie_raw(process, 1000);
423 1.2 kamil }
424 1.2 kamil
425 1.12 kamil static void __used
426 1.12 kamil await_stopped(pid_t process)
427 1.12 kamil {
428 1.12 kamil struct kinfo_proc2 p;
429 1.12 kamil size_t len = sizeof(p);
430 1.12 kamil
431 1.12 kamil const int name[] = {
432 1.12 kamil [0] = CTL_KERN,
433 1.12 kamil [1] = KERN_PROC2,
434 1.12 kamil [2] = KERN_PROC_PID,
435 1.12 kamil [3] = process,
436 1.12 kamil [4] = sizeof(p),
437 1.12 kamil [5] = 1
438 1.12 kamil };
439 1.12 kamil
440 1.12 kamil const size_t namelen = __arraycount(name);
441 1.12 kamil
442 1.12 kamil /* Await the process becoming a zombie */
443 1.12 kamil while(1) {
444 1.12 kamil ATF_REQUIRE(sysctl(name, namelen, &p, &len, NULL, 0) == 0);
445 1.12 kamil
446 1.12 kamil if (p.p_stat == LSSTOP)
447 1.12 kamil break;
448 1.12 kamil
449 1.12 kamil ATF_REQUIRE(usleep(1000) == 0);
450 1.12 kamil }
451 1.12 kamil }
452 1.12 kamil
453 1.5 kamil static pid_t __used
454 1.5 kamil await_stopped_child(pid_t process)
455 1.5 kamil {
456 1.5 kamil struct kinfo_proc2 *p = NULL;
457 1.5 kamil size_t i, len;
458 1.5 kamil pid_t child = -1;
459 1.5 kamil
460 1.5 kamil int name[] = {
461 1.5 kamil [0] = CTL_KERN,
462 1.5 kamil [1] = KERN_PROC2,
463 1.5 kamil [2] = KERN_PROC_ALL,
464 1.5 kamil [3] = 0,
465 1.5 kamil [4] = sizeof(struct kinfo_proc2),
466 1.5 kamil [5] = 0
467 1.5 kamil };
468 1.5 kamil
469 1.5 kamil const size_t namelen = __arraycount(name);
470 1.5 kamil
471 1.5 kamil /* Await the process becoming a zombie */
472 1.5 kamil while(1) {
473 1.5 kamil name[5] = 0;
474 1.5 kamil
475 1.5 kamil FORKEE_ASSERT_EQ(sysctl(name, namelen, 0, &len, NULL, 0), 0);
476 1.5 kamil
477 1.5 kamil FORKEE_ASSERT_EQ(reallocarr(&p,
478 1.5 kamil len,
479 1.5 kamil sizeof(struct kinfo_proc2)), 0);
480 1.5 kamil
481 1.5 kamil name[5] = len;
482 1.5 kamil
483 1.5 kamil FORKEE_ASSERT_EQ(sysctl(name, namelen, p, &len, NULL, 0), 0);
484 1.5 kamil
485 1.5 kamil for (i = 0; i < len/sizeof(struct kinfo_proc2); i++) {
486 1.5 kamil if (p[i].p_pid == getpid())
487 1.5 kamil continue;
488 1.5 kamil if (p[i].p_ppid != process)
489 1.5 kamil continue;
490 1.5 kamil if (p[i].p_stat != LSSTOP)
491 1.5 kamil continue;
492 1.5 kamil child = p[i].p_pid;
493 1.5 kamil break;
494 1.5 kamil }
495 1.5 kamil
496 1.5 kamil if (child != -1)
497 1.5 kamil break;
498 1.5 kamil
499 1.5 kamil FORKEE_ASSERT_EQ(usleep(1000), 0);
500 1.5 kamil }
501 1.5 kamil
502 1.5 kamil /* Free the buffer */
503 1.5 kamil FORKEE_ASSERT_EQ(reallocarr(&p, 0, sizeof(struct kinfo_proc2)), 0);
504 1.5 kamil
505 1.5 kamil return child;
506 1.5 kamil }
507 1.5 kamil
508 1.24 kamil static void __used
509 1.24 kamil await_collected(pid_t process)
510 1.24 kamil {
511 1.24 kamil struct kinfo_proc2 p;
512 1.24 kamil size_t len = sizeof(p);
513 1.24 kamil
514 1.24 kamil const int name[] = {
515 1.24 kamil [0] = CTL_KERN,
516 1.24 kamil [1] = KERN_PROC2,
517 1.24 kamil [2] = KERN_PROC_PID,
518 1.24 kamil [3] = process,
519 1.24 kamil [4] = sizeof(p),
520 1.24 kamil [5] = 1
521 1.24 kamil };
522 1.24 kamil
523 1.24 kamil const size_t namelen = __arraycount(name);
524 1.24 kamil
525 1.24 kamil /* Await the process to disappear */
526 1.24 kamil while(1) {
527 1.24 kamil FORKEE_ASSERT_EQ(sysctl(name, namelen, &p, &len, NULL, 0), 0);
528 1.24 kamil if (len == 0)
529 1.24 kamil break;
530 1.24 kamil
531 1.24 kamil ATF_REQUIRE(usleep(1000) == 0);
532 1.24 kamil }
533 1.24 kamil }
534 1.24 kamil
535 1.1 kamil /* Happy number sequence -- this function is used to just consume cpu cycles */
536 1.1 kamil #define HAPPY_NUMBER 1
537 1.1 kamil
538 1.1 kamil /* If n is not happy then its sequence ends in the cycle:
539 1.1 kamil * 4, 16, 37, 58, 89, 145, 42, 20, 4, ... */
540 1.1 kamil #define SAD_NUMBER 4
541 1.1 kamil
542 1.1 kamil /* Calculate the sum of the squares of the digits of n */
543 1.1 kamil static unsigned __used
544 1.1 kamil dsum(unsigned n)
545 1.1 kamil {
546 1.1 kamil unsigned sum, x;
547 1.1 kamil for (sum = 0; n; n /= 10) {
548 1.1 kamil x = n % 10;
549 1.1 kamil sum += x * x;
550 1.1 kamil }
551 1.1 kamil return sum;
552 1.1 kamil }
553 1.1 kamil
554 1.1 kamil /*
555 1.1 kamil * XXX: Disabled optimization is required to make tests for hardware assisted
556 1.1 kamil * traps in .text functional
557 1.1 kamil *
558 1.1 kamil * Tested with GCC 5.4 on NetBSD 7.99.47 amd64
559 1.1 kamil */
560 1.1 kamil static int __used
561 1.1 kamil #ifdef __clang__
562 1.1 kamil __attribute__((__optnone__))
563 1.1 kamil #else
564 1.1 kamil __attribute__((__optimize__("O0")))
565 1.1 kamil #endif
566 1.1 kamil check_happy(unsigned n)
567 1.1 kamil {
568 1.1 kamil for (;;) {
569 1.1 kamil unsigned total = dsum(n);
570 1.1 kamil
571 1.1 kamil if (total == HAPPY_NUMBER)
572 1.1 kamil return 1;
573 1.1 kamil if (total == SAD_NUMBER)
574 1.1 kamil return 0;
575 1.1 kamil
576 1.1 kamil n = total;
577 1.1 kamil }
578 1.1 kamil }
579 1.1 kamil
580 1.14 kamil static void * __used
581 1.13 kamil infinite_thread(void *arg __unused)
582 1.13 kamil {
583 1.13 kamil
584 1.13 kamil while (true)
585 1.13 kamil continue;
586 1.13 kamil
587 1.13 kamil __unreachable();
588 1.13 kamil }
589 1.13 kamil
590 1.15 kamil static int __used
591 1.15 kamil clone_func(void *arg)
592 1.15 kamil {
593 1.15 kamil int ret;
594 1.15 kamil
595 1.15 kamil ret = (int)(intptr_t)arg;
596 1.15 kamil
597 1.15 kamil return ret;
598 1.15 kamil }
599 1.15 kamil
600 1.4 kamil #if defined(HAVE_DBREGS)
601 1.6 kamil static bool __used
602 1.4 kamil can_we_set_dbregs(void)
603 1.4 kamil {
604 1.4 kamil static long euid = -1;
605 1.4 kamil static int user_set_dbregs = -1;
606 1.4 kamil size_t user_set_dbregs_len = sizeof(user_set_dbregs);
607 1.4 kamil
608 1.4 kamil if (euid == -1)
609 1.4 kamil euid = geteuid();
610 1.4 kamil
611 1.4 kamil if (euid == 0)
612 1.4 kamil return true;
613 1.4 kamil
614 1.4 kamil if (user_set_dbregs == -1) {
615 1.4 kamil if (sysctlbyname("security.models.extensions.user_set_dbregs",
616 1.4 kamil &user_set_dbregs, &user_set_dbregs_len, NULL, 0)
617 1.4 kamil == -1) {
618 1.4 kamil return false;
619 1.4 kamil }
620 1.4 kamil }
621 1.4 kamil
622 1.4 kamil if (user_set_dbregs > 0)
623 1.4 kamil return true;
624 1.4 kamil else
625 1.4 kamil return false;
626 1.4 kamil }
627 1.4 kamil #endif
628 1.4 kamil
629 1.8 kamil static bool __used
630 1.17 kamil get_user_va0_disable(void)
631 1.17 kamil {
632 1.17 kamil static int user_va0_disable = -1;
633 1.17 kamil size_t user_va0_disable_len = sizeof(user_va0_disable);
634 1.17 kamil
635 1.17 kamil if (user_va0_disable == -1) {
636 1.17 kamil if (sysctlbyname("vm.user_va0_disable",
637 1.17 kamil &user_va0_disable, &user_va0_disable_len, NULL, 0)
638 1.17 kamil == -1) {
639 1.17 kamil return true;
640 1.17 kamil }
641 1.17 kamil }
642 1.17 kamil
643 1.17 kamil if (user_va0_disable > 0)
644 1.17 kamil return true;
645 1.17 kamil else
646 1.17 kamil return false;
647 1.17 kamil }
648 1.17 kamil
649 1.17 kamil static bool __used
650 1.8 kamil can_we_write_to_text(pid_t pid)
651 1.8 kamil {
652 1.8 kamil int mib[3];
653 1.8 kamil int paxflags;
654 1.8 kamil size_t len = sizeof(int);
655 1.8 kamil
656 1.8 kamil mib[0] = CTL_PROC;
657 1.8 kamil mib[1] = pid;
658 1.8 kamil mib[2] = PROC_PID_PAXFLAGS;
659 1.8 kamil
660 1.8 kamil if (sysctl(mib, 3, &paxflags, &len, NULL, 0) == -1)
661 1.8 kamil return false;
662 1.8 kamil
663 1.8 kamil return !(paxflags & CTL_PROC_PAXFLAGS_MPROTECT);
664 1.8 kamil }
665 1.8 kamil
666 1.6 kamil static void __used
667 1.6 kamil trigger_trap(void)
668 1.6 kamil {
669 1.6 kamil
670 1.6 kamil /* Software breakpoint causes CPU trap, translated to SIGTRAP */
671 1.6 kamil #ifdef PTRACE_BREAKPOINT_ASM
672 1.6 kamil PTRACE_BREAKPOINT_ASM;
673 1.6 kamil #else
674 1.6 kamil /* port me */
675 1.6 kamil #endif
676 1.6 kamil }
677 1.6 kamil
678 1.6 kamil static void __used
679 1.6 kamil trigger_segv(void)
680 1.6 kamil {
681 1.6 kamil static volatile char *ptr = NULL;
682 1.6 kamil
683 1.6 kamil /* Access to unmapped memory causes CPU trap, translated to SIGSEGV */
684 1.6 kamil *ptr = 1;
685 1.6 kamil }
686 1.6 kamil
687 1.6 kamil static void __used
688 1.6 kamil trigger_ill(void)
689 1.6 kamil {
690 1.6 kamil
691 1.6 kamil /* Illegal instruction causes CPU trap, translated to SIGILL */
692 1.6 kamil #ifdef PTRACE_ILLEGAL_ASM
693 1.33 gson #ifndef __mips__ /* To avoid GXemul crash */
694 1.6 kamil PTRACE_ILLEGAL_ASM;
695 1.33 gson #endif
696 1.6 kamil #else
697 1.6 kamil /* port me */
698 1.6 kamil #endif
699 1.6 kamil }
700 1.6 kamil
701 1.25 christos #include <fenv.h>
702 1.25 christos
703 1.25 christos #if (__arm__ && !__SOFTFP__) || __aarch64__
704 1.25 christos #include <ieeefp.h> /* only need for ARM Cortex/Neon hack */
705 1.25 christos
706 1.16 kamil static bool __used
707 1.16 kamil are_fpu_exceptions_supported(void)
708 1.16 kamil {
709 1.16 kamil /*
710 1.16 kamil * Some NEON fpus do not trap on IEEE 754 FP exceptions.
711 1.16 kamil * Skip these tests if running on them and compiled for
712 1.16 kamil * hard float.
713 1.16 kamil */
714 1.16 kamil if (0 == fpsetmask(fpsetmask(FP_X_INV)))
715 1.16 kamil return false;
716 1.16 kamil return true;
717 1.16 kamil }
718 1.36 riastrad #elif defined __riscv__
719 1.36 riastrad #define are_fpu_exceptions_supported() 0
720 1.25 christos #else
721 1.26 kamil #define are_fpu_exceptions_supported() 1
722 1.25 christos #endif
723 1.16 kamil
724 1.35 riastrad volatile double ignore_result;
725 1.35 riastrad
726 1.9 kamil static void __used
727 1.6 kamil trigger_fpe(void)
728 1.6 kamil {
729 1.32 rin #if __i386__ || __x86_64__
730 1.32 rin /*
731 1.32 rin * XXX
732 1.32 rin * Hack for QEMU bug #1668041, by which floating-point division by
733 1.32 rin * zero is not trapped correctly. Also, assertions for si_code in
734 1.32 rin * ptrace_signal_wait.h are commented out. Clean them up after the
735 1.32 rin * bug is fixed.
736 1.32 rin */
737 1.32 rin volatile int a, b;
738 1.32 rin #else
739 1.32 rin volatile double a, b;
740 1.32 rin #endif
741 1.32 rin
742 1.32 rin a = getpid();
743 1.32 rin b = atoi("0");
744 1.6 kamil
745 1.16 kamil #ifdef __HAVE_FENV
746 1.16 kamil feenableexcept(FE_ALL_EXCEPT);
747 1.16 kamil #endif
748 1.16 kamil
749 1.31 rin /* Division by zero causes CPU trap, translated to SIGFPE */
750 1.35 riastrad ignore_result = (int)(a / b);
751 1.6 kamil }
752 1.6 kamil
753 1.6 kamil static void __used
754 1.6 kamil trigger_bus(void)
755 1.6 kamil {
756 1.6 kamil FILE *fp;
757 1.6 kamil char *p;
758 1.6 kamil
759 1.6 kamil /* Open an empty file for writing. */
760 1.6 kamil fp = tmpfile();
761 1.7 kamil FORKEE_ASSERT_NEQ((uintptr_t)fp, (uintptr_t)NULL);
762 1.6 kamil
763 1.11 kamil /*
764 1.11 kamil * Map an empty file with mmap(2) to a pointer.
765 1.11 kamil *
766 1.11 kamil * PROT_READ handles read-modify-write sequences emitted for
767 1.11 kamil * certain combinations of CPUs and compilers (e.g. Alpha AXP).
768 1.11 kamil */
769 1.10 kamil p = mmap(0, 1, PROT_READ|PROT_WRITE, MAP_PRIVATE, fileno(fp), 0);
770 1.7 kamil FORKEE_ASSERT_NEQ((uintptr_t)p, (uintptr_t)MAP_FAILED);
771 1.6 kamil
772 1.6 kamil /* Invalid memory access causes CPU trap, translated to SIGBUS */
773 1.6 kamil *p = 'a';
774 1.6 kamil }
775 1.6 kamil
776 1.18 kamil struct lwp_event_count {
777 1.18 kamil lwpid_t lec_lwp;
778 1.18 kamil int lec_count;
779 1.18 kamil };
780 1.18 kamil
781 1.23 kamil static int * __used
782 1.18 kamil find_event_count(struct lwp_event_count list[], lwpid_t lwp, size_t max_lwps)
783 1.18 kamil {
784 1.18 kamil size_t i;
785 1.18 kamil
786 1.18 kamil for (i = 0; i < max_lwps; i++) {
787 1.18 kamil if (list[i].lec_lwp == 0)
788 1.18 kamil list[i].lec_lwp = lwp;
789 1.18 kamil if (list[i].lec_lwp == lwp)
790 1.18 kamil return &list[i].lec_count;
791 1.18 kamil }
792 1.18 kamil
793 1.18 kamil atf_tc_fail("More LWPs reported than expected");
794 1.18 kamil }
795 1.18 kamil
796 1.18 kamil #define FIND_EVENT_COUNT(list, lwp) \
797 1.18 kamil find_event_count(list, lwp, __arraycount(list))
798 1.18 kamil
799 1.1 kamil #if defined(TWAIT_HAVE_PID)
800 1.1 kamil #define ATF_TP_ADD_TC_HAVE_PID(a,b) ATF_TP_ADD_TC(a,b)
801 1.1 kamil #else
802 1.1 kamil #define ATF_TP_ADD_TC_HAVE_PID(a,b)
803 1.1 kamil #endif
804 1.1 kamil
805 1.28 kamil #if defined(TWAIT_HAVE_STATUS)
806 1.28 kamil #define ATF_TP_ADD_TC_HAVE_STATUS(a,b) ATF_TP_ADD_TC(a,b)
807 1.28 kamil #else
808 1.28 kamil #define ATF_TP_ADD_TC_HAVE_STATUS(a,b)
809 1.28 kamil #endif
810 1.28 kamil
811 1.28 kamil #if defined(TWAIT_HAVE_STATUS) && (defined(__i386__) || defined(__x86_64__))
812 1.28 kamil #define ATF_TP_ADD_TC_HAVE_STATUS_X86(a,b) ATF_TP_ADD_TC(a,b)
813 1.28 kamil #else
814 1.28 kamil #define ATF_TP_ADD_TC_HAVE_STATUS_X86(a,b)
815 1.28 kamil #endif
816 1.28 kamil
817 1.1 kamil #if defined(HAVE_GPREGS)
818 1.1 kamil #define ATF_TP_ADD_TC_HAVE_GPREGS(a,b) ATF_TP_ADD_TC(a,b)
819 1.1 kamil #else
820 1.1 kamil #define ATF_TP_ADD_TC_HAVE_GPREGS(a,b)
821 1.1 kamil #endif
822 1.1 kamil
823 1.1 kamil #if defined(HAVE_FPREGS)
824 1.1 kamil #define ATF_TP_ADD_TC_HAVE_FPREGS(a,b) ATF_TP_ADD_TC(a,b)
825 1.1 kamil #else
826 1.1 kamil #define ATF_TP_ADD_TC_HAVE_FPREGS(a,b)
827 1.1 kamil #endif
828 1.1 kamil
829 1.1 kamil #if defined(HAVE_DBREGS)
830 1.1 kamil #define ATF_TP_ADD_TC_HAVE_DBREGS(a,b) ATF_TP_ADD_TC(a,b)
831 1.1 kamil #else
832 1.1 kamil #define ATF_TP_ADD_TC_HAVE_DBREGS(a,b)
833 1.1 kamil #endif
834 1.1 kamil
835 1.1 kamil #if defined(PT_STEP)
836 1.1 kamil #define ATF_TP_ADD_TC_PT_STEP(a,b) ATF_TP_ADD_TC(a,b)
837 1.1 kamil #else
838 1.1 kamil #define ATF_TP_ADD_TC_PT_STEP(a,b)
839 1.1 kamil #endif
840