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
41static bool	fail;
42static void	handler(int);
43
44static void
45handler(int signo)
46{
47
48	if (signo == SIGALRM)
49		fail = false;
50}
51
52ATF_TC(alarm_basic);
53ATF_TC_HEAD(alarm_basic, tc)
54{
55	atf_tc_set_md_var(tc, "descr", "A basic test of alarm(3)");
56}
57
58ATF_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
72ATF_TC(alarm_fork);
73ATF_TC_HEAD(alarm_fork, tc)
74{
75	atf_tc_set_md_var(tc, "descr", "Does fork(2) clear a pending alarm?");
76}
77
78ATF_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
110ATF_TC(alarm_previous);
111ATF_TC_HEAD(alarm_previous, tc)
112{
113	atf_tc_set_md_var(tc, "descr", "Test return value from alarm(3)");
114}
115
116ATF_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
138fail:
139	atf_tc_fail("invalid return value from alarm(3)");
140}
141
142ATF_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