/*	$OpenBSD: microtime.c,v 1.2 2022/07/23 22:58:51 cheloha Exp $ */
/*
 * Copyright (c) 2000 Poul-Henning Kamp <phk@FreeBSD.org>
 * Copyright (c) 2020 Paul Irofti <paul@irofti.net>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>
#include <sys/atomic.h>
#include <sys/timetc.h>

#include <time.h>

/*
 * Return the difference between the timehands' counter value now and what
 * was when we copied it to the timehands' offset_count.
 */
static inline int
tc_delta(struct timekeep *tk, u_int *delta)
{
	u_int tc;

	if (_tc_get_timecount(tk, &tc))
		return -1;
	*delta = (tc - tk->tk_offset_count) & tk->tk_counter_mask;
	return 0;
}

static int
binuptime(struct bintime *bt, struct timekeep *tk)
{
	u_int gen, delta;

	do {
		gen = tk->tk_generation;
		membar_consumer();
		if (tc_delta(tk, &delta))
			return -1;
		TIMECOUNT_TO_BINTIME(delta, tk->tk_scale, bt);
		bintimeadd(bt, &tk->tk_offset, bt);
		membar_consumer();
	} while (gen == 0 || gen != tk->tk_generation);

	return 0;
}

static int
binruntime(struct bintime *bt, struct timekeep *tk)
{
	u_int gen, delta;

	do {
		gen = tk->tk_generation;
		membar_consumer();
		if (tc_delta(tk, &delta))
			return -1;
		TIMECOUNT_TO_BINTIME(delta, tk->tk_scale, bt);
		bintimeadd(bt, &tk->tk_offset, bt);
		bintimesub(bt, &tk->tk_naptime, bt);
		membar_consumer();
	} while (gen == 0 || gen != tk->tk_generation);

	return 0;
}

static int
bintime(struct bintime *bt, struct timekeep *tk)
{
	u_int gen, delta;

	do {
		gen = tk->tk_generation;
		membar_consumer();
		if (tc_delta(tk, &delta))
			return -1;
		TIMECOUNT_TO_BINTIME(delta, tk->tk_scale, bt);
		bintimeadd(bt, &tk->tk_offset, bt);
		bintimeadd(bt, &tk->tk_boottime, bt);
		membar_consumer();
	} while (gen == 0 || gen != tk->tk_generation);

	return 0;
}

int
_microtime(struct timeval *tvp, struct timekeep *tk)
{
	struct bintime bt;

	if (bintime(&bt, tk))
		return -1;
	BINTIME_TO_TIMEVAL(&bt, tvp);
	return 0;
}

int
_nanotime(struct timespec *tsp, struct timekeep *tk)
{
	struct bintime bt;

	if (bintime(&bt, tk))
		return -1;
	BINTIME_TO_TIMESPEC(&bt, tsp);
	return 0;
}

int
_nanoruntime(struct timespec *ts, struct timekeep *tk)
{
	struct bintime bt;

	if (binruntime(&bt, tk))
		return -1;
	BINTIME_TO_TIMESPEC(&bt, ts);
	return 0;
}


int
_nanouptime(struct timespec *tsp, struct timekeep *tk)
{
	struct bintime bt;

	if (binuptime(&bt, tk))
		return -1;
	BINTIME_TO_TIMESPEC(&bt, tsp);
	return 0;
}