/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2024 Oxide Computer Company
 */

/*
 * Basic test of libjedec temperature features.
 */

#include <stdlib.h>
#include <limits.h>
#include <err.h>
#include <sys/sysmacros.h>
#include <libjedec.h>

typedef struct {
	uint32_t ltt_temp;
	boolean_t ltt_res;
	int32_t ltt_min;
	int32_t ltt_max;
	const char *ltt_desc;
} libjedec_temp_test_t;

static const libjedec_temp_test_t temp_tests[] = {
	{ JEDEC_TEMP_CASE_A2T, B_TRUE, -40, 105, "Operating temperature A2T" },
	{ JEDEC_TEMP_CASE_RT, B_TRUE, 0, 45, "Operating temperature RT" },
	{ JEDEC_TEMP_AMB_CT, B_TRUE, 0, 70, "Ambient temperature CT" },
	{ JEDEC_TEMP_AMB_IOT, B_TRUE, -40, 85, "Ambient temperature IOT" },
	{ JEDEC_TEMP_AMB_AO1T, B_TRUE,  -40, 125, "Ambient temperature A01T" },
	{ JEDEC_TEMP_STOR_ST, B_TRUE, -40, 85, "Storage temperature ST" },
	{ JEDEC_TEMP_JNCT_A90, B_TRUE, -40, 90, "Junction temperature A90" },
	{ JEDEC_TEMP_JNCT_LT115, B_TRUE, -40, 115,
	    "Junction temperature LT115" },
	{ JEDEC_TEMP_JNCT_ET, B_TRUE, -25, 105, "Junction temperature ET" },
	{ JEDEC_TEMP_JNCT_XT, B_TRUE, 0, 95, "Junction temperature XT" },
	{ 142, B_FALSE, 0, 0, "invalid temperature (142)" },
	{ INT32_MAX, B_FALSE, 0, 0, "invalid temperature (INT32_MAX)" },
	{ UINT32_MAX, B_FALSE, 0, 0, "invalid temperature (UINT32_MAX)" }
};

static boolean_t
libjedec_temp_run_single(const libjedec_temp_test_t *test)
{
	int32_t min = INT32_MIN, max = INT32_MAX;
	boolean_t res;

	res = libjedec_temp_range(test->ltt_temp, &min, &max);
	if (res != test->ltt_res) {
		if (test->ltt_res) {
			warnx("libjedec_temp_range() succeeded, but we "
			    "expected failure!");
		} else {
			warnx("libjedec_temp_range() failed, but we expected "
			    "success!");
		}
		return (B_FALSE);
	}

	if (!res) {
		return (B_TRUE);
	}

	if (min != test->ltt_min) {
		warnx("received incorrect minimum temperature: expected %d, "
		    "found %d\n", test->ltt_min, min);
	}

	if (max != test->ltt_max) {
		warnx("received incorrect maximum temperature: expected %d, "
		    "found %d\n", test->ltt_max, max);
	}

	return (B_TRUE);
}

int
main(void)
{
	int ret = EXIT_SUCCESS;

	for (size_t i = 0; i < ARRAY_SIZE(temp_tests); i++) {
		const libjedec_temp_test_t *test = &temp_tests[i];

		if (libjedec_temp_run_single(test)) {
			(void) printf("TEST PASSED: %s\n", test->ltt_desc);
		} else {
			(void) fprintf(stderr, "TEST FAILED: %s\n",
			    test->ltt_desc);
			ret = EXIT_FAILURE;
		}
	}

	if (ret == EXIT_SUCCESS) {
		(void) printf("All tests passed successfully!\n");
	}

	return (ret);
}