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