/* $NetBSD: t_clock_gettime.c,v 1.10 2025/03/25 19:51:32 riastradh Exp $ */

/*-
 * Copyright (c) 2008 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Frank Kardel.
 *
 * 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.
 */

/*-
 * Copyright (c) 2006 Frank Kardel
 * 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>
__COPYRIGHT("@(#) Copyright (c) 2008\
 The NetBSD Foundation, inc. All rights reserved.");
__RCSID("$NetBSD: t_clock_gettime.c,v 1.10 2025/03/25 19:51:32 riastradh Exp $");

#include <sys/param.h>

#include <sys/ioctl.h>
#include <sys/sysctl.h>

#include <atf-c.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include "h_macros.h"

#define MINPOSDIFF	15000000	/* 15 ms for now */
#define TIMEOUT		5

#define TC_HARDWARE	"kern.timecounter.hardware"
#define TC_CHOICE	"kern.timecounter.choice"

static void
check_timecounter(void)
{
	struct timespec tsa, tsb, tsl, res;
	long long mindiff = INTMAX_MAX;
	time_t endlimit;

#define CL(x) \
	do { \
		if ((x) != -1) \
			break; \
		atf_tc_fail_nonfatal("%s: %s", #x, strerror(errno)); \
		return; \
	} while (0)

	CL(clock_gettime(CLOCK_REALTIME, &tsa));
	tsl = tsa;

	CL(time(&endlimit));
	endlimit += TIMEOUT + 1;

	while ((time_t)tsa.tv_sec < endlimit) {
		long long diff;

		CL(clock_gettime(CLOCK_REALTIME, &tsb));
		diff = 1000000000LL * (tsb.tv_sec - tsa.tv_sec)
		    + tsb.tv_nsec - tsa.tv_nsec;

		if (diff > 0 && mindiff > diff)
			mindiff = diff;

		if (diff < 0 || diff > MINPOSDIFF) {
			long long elapsed;
			(void)printf("%stime"
			    " TSA: 0x%jx.%08jx, TSB: 0x%jx.%08jx, "
			    "diff = %lld nsec, ", (diff < 0) ? "BAD " : "",
			    (uintmax_t)tsa.tv_sec, (uintmax_t)tsa.tv_nsec,
			    (uintmax_t)tsb.tv_sec, (uintmax_t)tsb.tv_nsec,
			    diff);

			elapsed = 1000000000LL * (tsb.tv_sec - tsl.tv_sec)
			    + tsb.tv_nsec - tsl.tv_nsec;


			(void)printf("%lld nsec\n", elapsed);
			tsl = tsb;

			ATF_CHECK(diff >= 0);
			if (diff < 0)
				return;
		}

		tsa.tv_sec = tsb.tv_sec;
		tsa.tv_nsec = tsb.tv_nsec;
	}

	if (clock_getres(CLOCK_REALTIME, &res) == 0) {
		long long r = res.tv_sec * 1000000000 + res.tv_nsec;

		(void)printf("Claimed resolution: %lld nsec (%f Hz) or "
		    "better\n", r, 1.0 / r * 1e9);
		(void)printf("Observed minimum non zero delta: %lld "
		    "nsec\n", mindiff);
	}

#undef CL
}

ATF_TC(clock_gettime_real);
ATF_TC_HEAD(clock_gettime_real, tc)
{
	atf_tc_set_md_var(tc, "require.user", "root");
	atf_tc_set_md_var(tc, "descr",
	    "Checks the monotonicity of the CLOCK_REALTIME implementation");
	atf_tc_set_md_var(tc, "timeout", "300");
}

ATF_TC_BODY(clock_gettime_real, tc)
{
	char name[128], cbuf[512], ctrbuf[10240];
	size_t cbufsiz = sizeof(cbuf);
	size_t ctrbufsiz = sizeof(ctrbuf);
	const char *p;
	char *save;
	int quality, n;

	if (sysctlbyname(TC_HARDWARE, cbuf, &cbufsiz, NULL, 0) != 0) {
		(void)printf("\nChecking legacy time implementation "
		    "for %d seconds\n", TIMEOUT);
		check_timecounter();
		return;
		/* NOTREACHED */
	}
	(void)printf("%s = %s\n", TC_HARDWARE, cbuf);
	REQUIRE_LIBC(save = strdup(cbuf), NULL);

	RL(sysctlbyname(TC_CHOICE, ctrbuf, &ctrbufsiz, NULL, 0));
	(void)printf("%s = %s\n", TC_CHOICE, ctrbuf);

	for (p = ctrbuf, n = 0; sscanf(p, "%127[^(](q=%d, f=%*u Hz)%*[ ]%n",
	    name, &quality, &n) == 2; p += n) {
		struct timespec ts;
		int ret;

		if (quality < 0)
			continue;

		(void)printf("\nChecking %s for %d seconds\n", name, TIMEOUT);
		CHECK_LIBC(ret = sysctlbyname(TC_HARDWARE, NULL, 0,
		    name, strlen(name)), -1);
		if (ret == -1)
			continue;

		/* wait a bit to select new counter in clockinterrupt */
		ts.tv_sec = 0;
		ts.tv_nsec = 100000000;
		(void)nanosleep(&ts, NULL);

		check_timecounter();
	}

	RL(sysctlbyname(TC_HARDWARE, NULL, 0, save, strlen(save)));
}

static void
waste_user_time(void)
{
	static char buf[4*4096];

	arc4random_buf(buf, sizeof(buf));
}

static void __unused
waste_system_time(void)
{
	static char buf[4*4096];
	int fd[2];
	int i, n;

	RL(pipe2(fd, O_NONBLOCK));
	RL(n = ioctl(fd[1], FIONSPACE));
	n = MIN((unsigned)MAX(0, n), sizeof(buf));
	for (i = 0; i < 16; i++) {
		RL(write(fd[1], buf, n));
		RL(read(fd[0], buf, n));
	}
	RL(close(fd[0]));
	RL(close(fd[1]));
}

static void
check_monotonicity(const char *clockname, clockid_t clockid,
    void (*waste_time)(void))
{
	static const struct timespec maxtime = {5, 0};
	struct timespec mono_t0, t0, mono_d;

	RL(clock_gettime(CLOCK_MONOTONIC, &mono_t0));
	RL(clock_gettime(clockid, &t0));

	do {
		struct timespec t1, mono_t1;

		(*waste_time)();

		RL(clock_gettime(clockid, &t1));
		ATF_CHECK_MSG(timespeccmp(&t0, &t1, <=),
		    "clock %s=0x%jx went backwards t0=%jd.%09ld t1=%jd.%09ld",
		    clockname, (uintmax_t)clockid,
		    (intmax_t)t0.tv_sec, t0.tv_nsec,
		    (intmax_t)t1.tv_sec, t1.tv_nsec);

		t0 = t1;

		RL(clock_gettime(CLOCK_MONOTONIC, &mono_t1));
		timespecsub(&mono_t1, &mono_t0, &mono_d);
	} while (timespeccmp(&mono_d, &maxtime, <));
}

ATF_TC(clock_gettime_process_cputime_is_monotonic);
ATF_TC_HEAD(clock_gettime_process_cputime_is_monotonic, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "Checks that CLOCK_PROCESS_CPUTIME_ID is monotonic");
}
ATF_TC_BODY(clock_gettime_process_cputime_is_monotonic, tc)
{
	check_monotonicity("CLOCK_PROCESS_CPUTIME_ID",
	    CLOCK_PROCESS_CPUTIME_ID, &waste_user_time);
}

ATF_TC(clock_gettime_thread_cputime_is_monotonic);
ATF_TC_HEAD(clock_gettime_thread_cputime_is_monotonic, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "Checks that CLOCK_THREAD_CPUTIME_ID is monotonic");
}
ATF_TC_BODY(clock_gettime_thread_cputime_is_monotonic, tc)
{
	check_monotonicity("CLOCK_THREAD_CPUTIME_ID",
	    CLOCK_THREAD_CPUTIME_ID, &waste_user_time);
}

static void
check_resolution(const char *clockname, clockid_t clockid)
{
	struct timespec ts;
	int rv;

	RLF(rv = clock_getres(clockid, &ts), "%s", clockname);
	if (rv != -1) {
		ATF_CHECK_MSG(ts.tv_sec == 0,
		    "The resolution of the clock %s is reported as %jd.%09ld"
		    " which is lower than a second; most likely a wrong value",
		    clockname, ts.tv_sec, ts.tv_nsec);
	}
}

ATF_TC(clock_getres);
ATF_TC_HEAD(clock_getres, tc)
{
	atf_tc_set_md_var(tc, "descr",
	    "Checks that clock_getres(2) returns some reasonable resolution"
	    " for all supported clocks");
}
ATF_TC_BODY(clock_getres, tc)
{
	check_resolution("CLOCK_REALTIME", CLOCK_REALTIME);
	check_resolution("CLOCK_MONOTONIC", CLOCK_MONOTONIC);
	atf_tc_expect_fail("These clocks aren't supported but are documented"
	    " in clock_gettime(2) for some reason");
	check_resolution("CLOCK_VIRTUAL", CLOCK_VIRTUAL);
	check_resolution("CLOCK_PROF", CLOCK_PROF);
	atf_tc_expect_pass();
	check_resolution("CLOCK_PROCESS_CPUTIME_ID", CLOCK_PROCESS_CPUTIME_ID);
	check_resolution("CLOCK_THREAD_CPUTIME_ID", CLOCK_THREAD_CPUTIME_ID);
}

ATF_TP_ADD_TCS(tp)
{

	ATF_TP_ADD_TC(tp, clock_gettime_real);
	ATF_TP_ADD_TC(tp, clock_gettime_process_cputime_is_monotonic);
	ATF_TP_ADD_TC(tp, clock_gettime_thread_cputime_is_monotonic);
	ATF_TP_ADD_TC(tp, clock_getres);

	return atf_no_error();
}