xref: /src/tests/kernel/kqueue/t_timer.c

/* $NetBSD: t_timer.c,v 1.3 2021/10/22 13:53:20 thorpej Exp $ */

/*-
 * Copyright (c) 2021 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
__RCSID("$NetBSD: t_timer.c,v 1.3 2021/10/22 13:53:20 thorpej Exp $");

#include <sys/types.h>
#include <sys/event.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <unistd.h>

#include <atf-c.h>

ATF_TC(basic_timer);
ATF_TC_HEAD(basic_timer, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "tests basic EVFILT_TIMER functionality");
}

#define	TIME1		1000		/* 1000ms -> 1s */
#define	TIME1_COUNT	5
#define	TIME2		6000		/* 6000ms -> 6s */

#define	TIME1_TOTAL_SEC	((TIME1 * TIME1_COUNT) / 1000)
#define	TIME2_TOTAL_SEC	(TIME2 / 1000)

ATF_TC_BODY(basic_timer, tc)
{
	struct kevent event[2];
	int ntimer1 = 0, ntimer2 = 0;
	struct timespec ots, ts;
	int kq;

	ATF_REQUIRE((kq = kqueue()) >= 0);

	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, TIME1, NULL);
	EV_SET(&event[1], 2, EVFILT_TIMER, EV_ADD | EV_ONESHOT, 0, TIME2, NULL);

	ATF_REQUIRE(kevent(kq, event, 2, NULL, 0, NULL) == 0);
	ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC, &ots) == 0);

	for (;;) {
		ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, NULL) == 1);
		ATF_REQUIRE(event[0].filter == EVFILT_TIMER);
		ATF_REQUIRE(event[0].ident == 1 ||
			    event[0].ident == 2);
		if (event[0].ident == 1) {
			ATF_REQUIRE(ntimer1 < TIME1_COUNT);
			if (++ntimer1 == TIME1_COUNT) {
				/*
				 * Make sure TIME1_TOTAL_SEC seconds have
				 * elapsed, allowing for a little slop.
				 */
				ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC,
				    &ts) == 0);
				timespecsub(&ts, &ots, &ts);
				ATF_REQUIRE(ts.tv_sec ==
					    (TIME1_TOTAL_SEC - 1) ||
				    ts.tv_sec == TIME1_TOTAL_SEC);
				if (ts.tv_sec == TIME1_TOTAL_SEC - 1) {
					ATF_REQUIRE(ts.tv_nsec >=
					    900000000);
				}
				EV_SET(&event[0], 1, EVFILT_TIMER, EV_DELETE,
				    0, 0, NULL);
				ATF_REQUIRE(kevent(kq, event, 1, NULL, 0,
				    NULL) == 0);
			}
		} else {
			ATF_REQUIRE(ntimer1 == TIME1_COUNT);
			ATF_REQUIRE(ntimer2 == 0);
			ntimer2++;
			/*
			 * Make sure TIME2_TOTAL_SEC seconds have
			 * elapsed, allowing for a little slop.
			 */
			ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC,
			    &ts) == 0);
			timespecsub(&ts, &ots, &ts);
			ATF_REQUIRE(ts.tv_sec ==
				    (TIME2_TOTAL_SEC - 1) ||
			    ts.tv_sec == TIME2_TOTAL_SEC);
			if (ts.tv_sec == TIME2_TOTAL_SEC - 1) {
				ATF_REQUIRE(ts.tv_nsec >= 900000000);
			}
			EV_SET(&event[0], 2, EVFILT_TIMER, EV_DELETE,
			    0, 0, NULL);
			ATF_REQUIRE_ERRNO(ENOENT,
			    kevent(kq, event, 1, NULL, 0, NULL) == -1);
			break;
		}
	}

	/*
	 * Now block in kqueue for TIME2_TOTAL_SEC, and ensure we
	 * don't receive any new events.
	 */
	ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC, &ots) == 0);
	ts.tv_sec = TIME2_TOTAL_SEC;
	ts.tv_nsec = 0;
	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 0);
	ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC, &ts) == 0);
	timespecsub(&ts, &ots, &ts);
	ATF_REQUIRE(ts.tv_sec == (TIME2_TOTAL_SEC - 1) ||
	    ts.tv_sec == TIME2_TOTAL_SEC ||
	    ts.tv_sec == (TIME2_TOTAL_SEC + 1));
	if (ts.tv_sec == TIME2_TOTAL_SEC - 1) {
		ATF_REQUIRE(ts.tv_nsec >= 900000000);
	} else if (ts.tv_sec == TIME2_TOTAL_SEC + 1) {
		ATF_REQUIRE(ts.tv_nsec < 500000000);
	}
}

ATF_TC(count_expirations);
ATF_TC_HEAD(count_expirations, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "tests counting timer expirations");
}

ATF_TC_BODY(count_expirations, tc)
{
	struct kevent event[1];
	struct timespec ts = { 0, 0 };
	struct timespec sleepts;
	int kq;

	ATF_REQUIRE((kq = kqueue()) >= 0);

	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, TIME1, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, NULL, 0, NULL) == 0);

	/* Sleep a little longer to mitigate timing jitter. */
	sleepts.tv_sec = TIME1_TOTAL_SEC;
	sleepts.tv_nsec = 500000000;
	ATF_REQUIRE(nanosleep(&sleepts, NULL) == 0);

	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 1);
	ATF_REQUIRE(event[0].ident == 1);
	ATF_REQUIRE(event[0].data == TIME1_COUNT ||
		    event[0].data == TIME1_COUNT + 1);
}

ATF_TC(modify);
ATF_TC_HEAD(modify, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "tests modifying a timer");
}

ATF_TC_BODY(modify, tc)
{
	struct kevent event[1];
	struct timespec ts = { 0, 0 };
	struct timespec sleepts;
	int kq;

	ATF_REQUIRE((kq = kqueue()) >= 0);

	/*
	 * Start a 500ms timer, sleep for 5 seconds, and check
	 * the total count.
	 */
	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, 500, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, NULL, 0, NULL) == 0);

	sleepts.tv_sec = 5;
	sleepts.tv_nsec = 0;
	ATF_REQUIRE(nanosleep(&sleepts, NULL) == 0);

	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 1);
	ATF_REQUIRE(event[0].ident == 1);
	ATF_REQUIRE(event[0].data >= 9 && event[0].data <= 11);

	/*
	 * Modify to a 4 second timer, sleep for 5 seconds, and check
	 * the total count.
	 */
	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, 4000, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, NULL, 0, NULL) == 0);

	/*
	 * Before we sleep, verify that the knote for this timer is
	 * no longer activated.
	 */
	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 0);

	sleepts.tv_sec = 5;
	sleepts.tv_nsec = 0;
	ATF_REQUIRE(nanosleep(&sleepts, NULL) == 0);

	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 1);
	ATF_REQUIRE(event[0].ident == 1);
	ATF_REQUIRE(event[0].data == 1);

	/*
	 * Start a 500ms timer, sleep for 2 seconds.
	 */
	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, 500, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, NULL, 0, NULL) == 0);

	sleepts.tv_sec = 2;
	sleepts.tv_nsec = 0;
	ATF_REQUIRE(nanosleep(&sleepts, NULL) == 0);

	/*
	 * Set the SAME timer, sleep for 2 seconds.
	 */
	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD, 0, 500, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, NULL, 0, NULL) == 0);

	sleepts.tv_sec = 2;
	sleepts.tv_nsec = 0;
	ATF_REQUIRE(nanosleep(&sleepts, NULL) == 0);

	/*
	 * The kernel should have reset the count when modifying the
	 * timer, so we should only expect to see the expiration count
	 * for the second sleep.
	 */
	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 1);
	ATF_REQUIRE(event[0].ident == 1);
	ATF_REQUIRE(event[0].data >= 3 && event[0].data <= 5);
}

ATF_TC(abstime);
ATF_TC_HEAD(abstime, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "tests timers with NOTE_ABSTIME");
}

ATF_TC_BODY(abstime, tc)
{
	struct kevent event[1];
	struct timespec ts, ots;
	time_t seconds;
	int kq;

	ATF_REQUIRE((kq = kqueue()) >= 0);

	ATF_REQUIRE(clock_gettime(CLOCK_REALTIME, &ots) == 0);
	ATF_REQUIRE(ots.tv_sec < INTPTR_MAX - TIME1_TOTAL_SEC);

	seconds = ots.tv_sec + TIME1_TOTAL_SEC;
	if (ots.tv_nsec >= 500000000) {
		seconds++;
	}

	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD,
	    NOTE_ABSTIME | NOTE_SECONDS, seconds, NULL);
	ATF_REQUIRE(kevent(kq, event, 1, event, 1, NULL) == 1);

	ATF_REQUIRE(clock_gettime(CLOCK_REALTIME, &ts) == 0);
	timespecsub(&ts, &ots, &ts);

	/*
	 * We're not going for precision here; just verify that it was
	 * delivered anywhere between 4.5-6.whatever seconds later.
	 */
	ATF_REQUIRE(ts.tv_sec >= 4 && ts.tv_sec <= 6);
	if (ts.tv_sec == 4) {
		ATF_REQUIRE(ts.tv_nsec >= 500000000);
	}

	ts.tv_sec = 0;
	ts.tv_nsec = 0;
	ATF_REQUIRE(kevent(kq, NULL, 0, event, 1, &ts) == 1);
}

#define	PREC_TIMEOUT_SEC	2

static void
do_test_timer_units(const char *which, uint32_t fflag, int64_t data)
{
	struct kevent event[1];
	struct timespec ts, ots;
	int kq;

	ATF_REQUIRE((kq = kqueue()) >= 0);

	EV_SET(&event[0], 1, EVFILT_TIMER, EV_ADD | EV_ONESHOT,
	    fflag, data, NULL);

	ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC, &ots) == 0);
	ATF_REQUIRE(kevent(kq, event, 1, event, 1, NULL) == 1);
	ATF_REQUIRE(clock_gettime(CLOCK_MONOTONIC, &ts) == 0);

	timespecsub(&ts, &ots, &ts);
	ATF_REQUIRE_MSG(ts.tv_sec == (PREC_TIMEOUT_SEC - 1) ||
		    ts.tv_sec == PREC_TIMEOUT_SEC,
		    "units '%s' failed [sec]", which);
	if (ts.tv_sec == PREC_TIMEOUT_SEC - 1) {
		ATF_REQUIRE_MSG(ts.tv_nsec >= 900000000,
		"units '%s' failed [nsec]", which);
	}

	(void)close(kq);
}

#define	test_timer_units(fflag, data)				\
	do_test_timer_units(#fflag, fflag, data)

ATF_TC(timer_units);
ATF_TC_HEAD(timer_units, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "tests timers with NOTE_* units modifiers");
}

ATF_TC_BODY(timer_units, tc)
{
	test_timer_units(NOTE_SECONDS,  PREC_TIMEOUT_SEC);
	test_timer_units(NOTE_MSECONDS, PREC_TIMEOUT_SEC * 1000);
	test_timer_units(NOTE_USECONDS, PREC_TIMEOUT_SEC * 1000000);
	test_timer_units(NOTE_NSECONDS, PREC_TIMEOUT_SEC * 1000000000);
}

ATF_TP_ADD_TCS(tp)
{
	ATF_TP_ADD_TC(tp, basic_timer);
	ATF_TP_ADD_TC(tp, count_expirations);
	ATF_TP_ADD_TC(tp, abstime);
	ATF_TP_ADD_TC(tp, timer_units);
	ATF_TP_ADD_TC(tp, modify);

	return atf_no_error();
}