Home | History | Annotate | Line # | Download | only in gen 
      1 /* $NetBSD: t_alarm.c,v 1.2 2011/05/10 06:58:17 jruoho 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_alarm.c,v 1.2 2011/05/10 06:58:17 jruoho Exp $");
     33 
     34 #include <sys/wait.h>
     35 
     36 #include <atf-c.h>
     37 #include <stdlib.h>
     38 #include <signal.h>
     39 #include <unistd.h>
     40 
     41 static bool	fail;
     42 static void	handler(int);
     43 
     44 static void
     45 handler(int signo)
     46 {
     47 
     48 	if (signo == SIGALRM)
     49 		fail = false;
     50 }
     51 
     52 ATF_TC(alarm_basic);
     53 ATF_TC_HEAD(alarm_basic, tc)
     54 {
     55 	atf_tc_set_md_var(tc, "descr", "A basic test of alarm(3)");
     56 }
     57 
     58 ATF_TC_BODY(alarm_basic, tc)
     59 {
     60 
     61 	ATF_REQUIRE(signal(SIGALRM, handler) == 0);
     62 
     63 	fail = true;
     64 
     65 	(void)alarm(1);
     66 	(void)sleep(2);
     67 
     68 	if (fail != false)
     69 		atf_tc_fail("alarm(3) failed to deliver signal");
     70 }
     71 
     72 ATF_TC(alarm_fork);
     73 ATF_TC_HEAD(alarm_fork, tc)
     74 {
     75 	atf_tc_set_md_var(tc, "descr", "Does fork(2) clear a pending alarm?");
     76 }
     77 
     78 ATF_TC_BODY(alarm_fork, tc)
     79 {
     80 	unsigned int rv;
     81 	pid_t pid;
     82 	int sta;
     83 
     84 	/*
     85 	 * Any pending alarms should be
     86 	 * cleared in the child process.
     87 	 */
     88 	(void)alarm(60);
     89 
     90 	pid = fork();
     91 	ATF_REQUIRE(pid >= 0);
     92 
     93 	if (pid == 0) {
     94 
     95 		rv = alarm(0);
     96 
     97 		if (rv != 0)
     98 			_exit(EXIT_FAILURE);
     99 
    100 		_exit(EXIT_SUCCESS);
    101 	}
    102 
    103 	(void)alarm(0);
    104 	(void)wait(&sta);
    105 
    106 	if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != EXIT_SUCCESS)
    107 		atf_tc_fail("pending alarm was not cleared for child");
    108 }
    109 
    110 ATF_TC(alarm_previous);
    111 ATF_TC_HEAD(alarm_previous, tc)
    112 {
    113 	atf_tc_set_md_var(tc, "descr", "Test return value from alarm(3)");
    114 }
    115 
    116 ATF_TC_BODY(alarm_previous, tc)
    117 {
    118 	unsigned int rv;
    119 
    120 	/*
    121 	 * See that alarm(3) returns the amount
    122 	 * left on the timer from the previous call.
    123 	*/
    124 	rv = alarm(60);
    125 
    126 	if (rv != 0)
    127 		goto fail;
    128 
    129 	rv = alarm(0);
    130 
    131 	if (rv < 50)
    132 		goto fail;
    133 
    134 	(void)alarm(0);
    135 
    136 	return;
    137 
    138 fail:
    139 	atf_tc_fail("invalid return value from alarm(3)");
    140 }
    141 
    142 ATF_TP_ADD_TCS(tp)
    143 {
    144 
    145 	ATF_TP_ADD_TC(tp, alarm_basic);
    146 	ATF_TP_ADD_TC(tp, alarm_fork);
    147 	ATF_TP_ADD_TC(tp, alarm_previous);
    148 
    149 	return atf_no_error();
    150 }
    151