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