1272343Sngie/*	$NetBSD: t_timer_create.c,v 1.4 2012/03/18 07:00:52 jruoho Exp $ */
2272343Sngie
3272343Sngie/*-
4272343Sngie * Copyright (c) 2010 The NetBSD Foundation, Inc.
5272343Sngie * All rights reserved.
6272343Sngie *
7272343Sngie * Redistribution and use in source and binary forms, with or without
8272343Sngie * modification, are permitted provided that the following conditions
9272343Sngie * are met:
10272343Sngie * 1. Redistributions of source code must retain the above copyright
11272343Sngie *    notice, this list of conditions and the following disclaimer.
12272343Sngie * 2. Redistributions in binary form must reproduce the above copyright
13272343Sngie *    notice, this list of conditions and the following disclaimer in the
14272343Sngie *    documentation and/or other materials provided with the distribution.
15272343Sngie *
16272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19272343Sngie * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26272343Sngie * POSSIBILITY OF SUCH DAMAGE.
27272343Sngie */
28272343Sngie
29272343Sngie#include <atf-c.h>
30272343Sngie#include <errno.h>
31272343Sngie#include <stdio.h>
32272343Sngie#include <signal.h>
33272343Sngie#include <string.h>
34272343Sngie#include <time.h>
35272343Sngie#include <unistd.h>
36272343Sngie
37272343Sngiestatic timer_t t;
38272343Sngiestatic bool fail = true;
39272343Sngie
40272343Sngiestatic void
41276478Sngie#ifdef __FreeBSD__
42276478Sngietimer_signal_handler(int signo, siginfo_t *si, void *osi __unused)
43276478Sngie#else
44272343Sngietimer_signal_handler(int signo, siginfo_t *si, void *osi)
45276478Sngie#endif
46272343Sngie{
47272343Sngie	timer_t *tp;
48272343Sngie
49272343Sngie	tp = si->si_value.sival_ptr;
50272343Sngie
51272343Sngie	if (*tp == t && signo == SIGALRM)
52272343Sngie		fail = false;
53272343Sngie
54272343Sngie	(void)fprintf(stderr, "%s: %s\n", __func__, strsignal(signo));
55272343Sngie}
56272343Sngie
57272343Sngiestatic void
58272343Sngietimer_signal_create(clockid_t cid, bool expire)
59272343Sngie{
60272343Sngie	struct itimerspec tim;
61272343Sngie	struct sigaction act;
62272343Sngie	struct sigevent evt;
63272343Sngie	sigset_t set;
64272343Sngie
65272343Sngie	t = 0;
66272343Sngie	fail = true;
67272343Sngie
68272343Sngie	(void)memset(&evt, 0, sizeof(struct sigevent));
69272343Sngie	(void)memset(&act, 0, sizeof(struct sigaction));
70272343Sngie	(void)memset(&tim, 0, sizeof(struct itimerspec));
71272343Sngie
72272343Sngie	/*
73272343Sngie	 * Set handler.
74272343Sngie	 */
75272343Sngie	act.sa_flags = SA_SIGINFO;
76272343Sngie	act.sa_sigaction = timer_signal_handler;
77272343Sngie
78272343Sngie	ATF_REQUIRE(sigemptyset(&set) == 0);
79272343Sngie	ATF_REQUIRE(sigemptyset(&act.sa_mask) == 0);
80272343Sngie
81272343Sngie	/*
82272343Sngie	 * Block SIGALRM while configuring the timer.
83272343Sngie	 */
84272343Sngie	ATF_REQUIRE(sigaction(SIGALRM, &act, NULL) == 0);
85272343Sngie	ATF_REQUIRE(sigaddset(&set, SIGALRM) == 0);
86272343Sngie	ATF_REQUIRE(sigprocmask(SIG_SETMASK, &set, NULL) == 0);
87272343Sngie
88272343Sngie	/*
89272343Sngie	 * Create the timer (SIGEV_SIGNAL).
90272343Sngie	 */
91272343Sngie	evt.sigev_signo = SIGALRM;
92272343Sngie	evt.sigev_value.sival_ptr = &t;
93272343Sngie	evt.sigev_notify = SIGEV_SIGNAL;
94272343Sngie
95272343Sngie	ATF_REQUIRE(timer_create(cid, &evt, &t) == 0);
96272343Sngie
97272343Sngie	/*
98272343Sngie	 * Start the timer. After this, unblock the signal.
99272343Sngie	 */
100272343Sngie	tim.it_value.tv_sec = expire ? 5 : 1;
101272343Sngie	tim.it_value.tv_nsec = 0;
102272343Sngie
103272343Sngie	ATF_REQUIRE(timer_settime(t, 0, &tim, NULL) == 0);
104272343Sngie
105272343Sngie	(void)sigprocmask(SIG_UNBLOCK, &set, NULL);
106272343Sngie	(void)sleep(2);
107272343Sngie
108272343Sngie	if (expire) {
109272343Sngie		if (!fail)
110272343Sngie			atf_tc_fail("timer fired too soon");
111272343Sngie	} else {
112272343Sngie		if (fail)
113272343Sngie			atf_tc_fail("timer failed to fire");
114272343Sngie	}
115272343Sngie
116272343Sngie	ATF_REQUIRE(timer_delete(t) == 0);
117272343Sngie}
118272343Sngie
119272343SngieATF_TC(timer_create_err);
120272343SngieATF_TC_HEAD(timer_create_err, tc)
121272343Sngie{
122272343Sngie	atf_tc_set_md_var(tc, "descr",
123272343Sngie	    "Check errors from timer_create(2) (PR lib/42434");
124272343Sngie}
125272343Sngie
126272343SngieATF_TC_BODY(timer_create_err, tc)
127272343Sngie{
128272343Sngie	struct sigevent ev;
129272343Sngie
130272343Sngie	(void)memset(&ev, 0, sizeof(struct sigevent));
131272343Sngie
132272343Sngie	errno = 0;
133272343Sngie	ev.sigev_signo = -1;
134272343Sngie	ev.sigev_notify = SIGEV_SIGNAL;
135272343Sngie
136272343Sngie	ATF_REQUIRE_ERRNO(EINVAL, timer_create(CLOCK_REALTIME, &ev, &t) == -1);
137272343Sngie
138272343Sngie	errno = 0;
139272343Sngie	ev.sigev_signo = SIGUSR1;
140272343Sngie	ev.sigev_notify = SIGEV_THREAD + 100;
141272343Sngie
142272343Sngie	ATF_REQUIRE_ERRNO(EINVAL, timer_create(CLOCK_REALTIME, &ev, &t) == -1);
143272343Sngie}
144272343Sngie
145272343SngieATF_TC(timer_create_real);
146272343SngieATF_TC_HEAD(timer_create_real, tc)
147272343Sngie{
148272343Sngie
149272343Sngie	atf_tc_set_md_var(tc, "descr",
150272343Sngie	    "Checks timer_create(2) with CLOCK_REALTIME and sigevent(3), "
151272343Sngie	    "SIGEV_SIGNAL");
152272343Sngie}
153272343Sngie
154272343SngieATF_TC_BODY(timer_create_real, tc)
155272343Sngie{
156272343Sngie	timer_signal_create(CLOCK_REALTIME, false);
157272343Sngie}
158272343Sngie
159272343SngieATF_TC(timer_create_mono);
160272343SngieATF_TC_HEAD(timer_create_mono, tc)
161272343Sngie{
162272343Sngie
163272343Sngie	atf_tc_set_md_var(tc, "descr",
164272343Sngie	    "Checks timer_create(2) with CLOCK_MONOTONIC and sigevent(3), "
165272343Sngie	    "SIGEV_SIGNAL");
166272343Sngie}
167272343Sngie
168272343SngieATF_TC_BODY(timer_create_mono, tc)
169272343Sngie{
170272343Sngie	timer_signal_create(CLOCK_MONOTONIC, false);
171272343Sngie}
172272343Sngie
173272343SngieATF_TC(timer_create_real_expire);
174272343SngieATF_TC_HEAD(timer_create_real_expire, tc)
175272343Sngie{
176272343Sngie
177272343Sngie	atf_tc_set_md_var(tc, "descr",
178272343Sngie	    "Checks timer_create(2) with CLOCK_REALTIME and sigevent(3), "
179272343Sngie	    "SIGEV_SIGNAL, with expiration");
180272343Sngie}
181272343Sngie
182272343SngieATF_TC_BODY(timer_create_real_expire, tc)
183272343Sngie{
184272343Sngie	timer_signal_create(CLOCK_REALTIME, true);
185272343Sngie}
186272343Sngie
187272343SngieATF_TC(timer_create_mono_expire);
188272343SngieATF_TC_HEAD(timer_create_mono_expire, tc)
189272343Sngie{
190272343Sngie
191272343Sngie	atf_tc_set_md_var(tc, "descr",
192272343Sngie	    "Checks timer_create(2) with CLOCK_MONOTONIC and sigevent(3), "
193272343Sngie	    "SIGEV_SIGNAL, with expiration");
194272343Sngie}
195272343Sngie
196272343SngieATF_TC_BODY(timer_create_mono_expire, tc)
197272343Sngie{
198272343Sngie	timer_signal_create(CLOCK_MONOTONIC, true);
199272343Sngie}
200272343Sngie
201272343SngieATF_TP_ADD_TCS(tp)
202272343Sngie{
203272343Sngie
204272343Sngie	ATF_TP_ADD_TC(tp, timer_create_err);
205272343Sngie	ATF_TP_ADD_TC(tp, timer_create_real);
206272343Sngie	ATF_TP_ADD_TC(tp, timer_create_mono);
207272343Sngie	ATF_TP_ADD_TC(tp, timer_create_real_expire);
208272343Sngie	ATF_TP_ADD_TC(tp, timer_create_mono_expire);
209272343Sngie
210272343Sngie	return atf_no_error();
211272343Sngie}
212