Index: head/lib/csu/tests/fini_test.c =================================================================== --- head/lib/csu/tests/fini_test.c (revision 339863) +++ head/lib/csu/tests/fini_test.c (revision 339864) @@ -1,129 +1,140 @@ /*- * SPDX-License-Identifier: BSD-2-Clause * * Copyright (c) 2018 Andrew Turner * * This software was developed by SRI International and the University of * Cambridge Computer Laboratory under DARPA/AFRL contract FA8750-10-C-0237 * ("CTSRD"), as part of the DARPA CRASH research programme. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #include __FBSDID("$FreeBSD$"); #include #include #include #include #include #include #include typedef void (*func_ptr)(void); static bool run_dtors_test = false; static bool run_fini_array_test = false; static void dtors_handler(void) { if (run_dtors_test) _exit(1); } __section(".dtors") __used static func_ptr dtors_func = &dtors_handler; ATF_TC_WITHOUT_HEAD(dtors_test); ATF_TC_BODY(dtors_test, tc) { pid_t pid, wpid; int status; pid = fork(); switch(pid) { case -1: break; case 0: run_dtors_test = true; exit(0); default: while ((wpid = waitpid(pid, &status, 0)) == -1 && errno == EINTR) ; #ifdef HAVE_CTORS ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1, ".dtors failed to run"); #else ATF_REQUIRE_MSG(WEXITSTATUS(status) == 0, ".dtors incorrectly ran"); #endif break; } } static void fini_array_handler(void) { if (run_fini_array_test) _exit(1); } __section(".fini_array") __used static func_ptr fini_array_func = &fini_array_handler; ATF_TC_WITHOUT_HEAD(fini_array_test); ATF_TC_BODY(fini_array_test, tc) { pid_t pid, wpid; int status; pid = fork(); switch(pid) { case -1: break; case 0: run_fini_array_test = true; exit(0); default: while ((wpid = waitpid(pid, &status, 0)) == -1 && errno == EINTR) ; ATF_REQUIRE_MSG(WEXITSTATUS(status) == 1, ".fini_array failed to run"); break; } } +extern void *__dso_handle; + +ATF_TC_WITHOUT_HEAD(dso_handle_test); +ATF_TC_BODY(dso_handle_test, tc) +{ + + ATF_REQUIRE_MSG(__dso_handle == NULL, + "Invalid __dso_handle in non-DSO"); +} + ATF_TP_ADD_TCS(tp) { ATF_TP_ADD_TC(tp, dtors_test); ATF_TP_ADD_TC(tp, fini_array_test); + ATF_TP_ADD_TC(tp, dso_handle_test); return (atf_no_error()); }