Home | History | Annotate | Line # | Download | only in gen 
      1 /* $NetBSD: t_assert.c,v 1.3 2017/01/10 15:17:57 christos Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Jukka Ruohonen.
      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 NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 #include <sys/cdefs.h>
     32 __RCSID("$NetBSD: t_assert.c,v 1.3 2017/01/10 15:17:57 christos Exp $");
     33 
     34 #include <sys/types.h>
     35 #include <sys/resource.h>
     36 #include <sys/time.h>
     37 #include <sys/wait.h>
     38 
     39 #include <assert.h>
     40 #include <atf-c.h>
     41 #include <signal.h>
     42 #include <stdlib.h>
     43 #include <string.h>
     44 #include <unistd.h>
     45 
     46 static void
     47 disable_corefile(void)
     48 {
     49 	struct rlimit limits;
     50 
     51 	limits.rlim_cur = 0;
     52 	limits.rlim_max = 0;
     53 
     54 	ATF_REQUIRE(setrlimit(RLIMIT_CORE, &limits) == 0);
     55 }
     56 
     57 static void		handler(int);
     58 
     59 static void
     60 handler(int signo)
     61 {
     62 	/* Nothing. */
     63 }
     64 
     65 ATF_TC(assert_false);
     66 ATF_TC_HEAD(assert_false, tc)
     67 {
     68 	atf_tc_set_md_var(tc, "descr", "Test that assert(3) works, #1");
     69 }
     70 
     71 ATF_TC_BODY(assert_false, tc)
     72 {
     73 	struct sigaction sa;
     74 	pid_t pid;
     75 	int sta;
     76 
     77 	pid = fork();
     78 	ATF_REQUIRE(pid >= 0);
     79 
     80 	if (pid == 0) {
     81 
     82 		disable_corefile();
     83 		(void)closefrom(0);
     84 		(void)memset(&sa, 0, sizeof(struct sigaction));
     85 
     86 		sa.sa_flags = 0;
     87 		sa.sa_handler = handler;
     88 
     89 		(void)sigemptyset(&sa.sa_mask);
     90 		(void)sigaction(SIGABRT, &sa, 0);
     91 
     92 		assert(1 == 1);
     93 
     94 		_exit(EXIT_SUCCESS);
     95 	}
     96 
     97 	(void)wait(&sta);
     98 
     99 	if (WIFSIGNALED(sta) != 0 || WIFEXITED(sta) == 0)
    100 		atf_tc_fail("assert(3) fired haphazardly");
    101 }
    102 
    103 ATF_TC(assert_true);
    104 ATF_TC_HEAD(assert_true, tc)
    105 {
    106 	atf_tc_set_md_var(tc, "descr", "Test that assert(3) works, #2");
    107 }
    108 
    109 ATF_TC_BODY(assert_true, tc)
    110 {
    111 	struct sigaction sa;
    112 	pid_t pid;
    113 	int sta;
    114 
    115 	pid = fork();
    116 	ATF_REQUIRE(pid >= 0);
    117 
    118 	if (pid == 0) {
    119 
    120 		disable_corefile();
    121 		(void)closefrom(0);
    122 		(void)memset(&sa, 0, sizeof(struct sigaction));
    123 
    124 		sa.sa_flags = 0;
    125 		sa.sa_handler = handler;
    126 
    127 		(void)sigemptyset(&sa.sa_mask);
    128 		(void)sigaction(SIGABRT, &sa, 0);
    129 
    130 		assert(1 == 2);
    131 
    132 		_exit(EXIT_SUCCESS);
    133 	}
    134 
    135 	(void)wait(&sta);
    136 
    137 	if (WIFSIGNALED(sta) == 0 || WTERMSIG(sta) != SIGABRT)
    138 		atf_tc_fail("assert(3) did not fire");
    139 }
    140 
    141 ATF_TP_ADD_TCS(tp)
    142 {
    143 
    144 	ATF_TP_ADD_TC(tp, assert_false);
    145 	ATF_TP_ADD_TC(tp, assert_true);
    146 
    147 	return atf_no_error();
    148 }
    149