/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2010-2014 Intel Corporation
 */

#include <stdio.h>
#include <stdint.h>
#include <unistd.h>
#include <limits.h>
#include <string.h>
#include <stdbool.h>

#include "test.h"

#ifndef RTE_LIBRTE_POWER

static int
test_power(void)
{
	printf("Power management library not supported, skipping test\n");
	return TEST_SKIPPED;
}

#else

#include <rte_power.h>

static int
check_function_ptrs(void)
{
	enum power_management_env env = rte_power_get_env();

	const bool not_null_expected = !(env == PM_ENV_NOT_SET);

	const char *inject_not_string1 = not_null_expected ? " not" : "";
	const char *inject_not_string2 = not_null_expected ? "" : " not";

	if ((rte_power_freqs == NULL) == not_null_expected) {
		printf("rte_power_freqs should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
					inject_not_string2);
		return -1;
	}
	if ((rte_power_get_freq == NULL) == not_null_expected) {
		printf("rte_power_get_freq should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
					inject_not_string2);
		return -1;
	}
	if ((rte_power_set_freq == NULL) == not_null_expected) {
		printf("rte_power_set_freq should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_up == NULL) == not_null_expected) {
		printf("rte_power_freq_up should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_down == NULL) == not_null_expected) {
		printf("rte_power_freq_down should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_max == NULL) == not_null_expected) {
		printf("rte_power_freq_max should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_min == NULL) == not_null_expected) {
		printf("rte_power_freq_min should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_turbo_status == NULL) == not_null_expected) {
		printf("rte_power_turbo_status should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_enable_turbo == NULL) == not_null_expected) {
		printf("rte_power_freq_enable_turbo should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_freq_disable_turbo == NULL) == not_null_expected) {
		printf("rte_power_freq_disable_turbo should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}
	if ((rte_power_get_capabilities == NULL) == not_null_expected) {
		printf("rte_power_get_capabilities should%s be NULL, environment has%s been "
				"initialised\n", inject_not_string1,
				inject_not_string2);
		return -1;
	}

	return 0;
}

static int
test_power(void)
{
	int ret = -1;
	enum power_management_env env;

	/* Test setting an invalid environment */
	ret = rte_power_set_env(PM_ENV_NOT_SET);
	if (ret == 0) {
		printf("Unexpectedly succeeded on setting an invalid environment\n");
		return -1;
	}

	/* Test that the environment has not been set */
	env = rte_power_get_env();
	if (env != PM_ENV_NOT_SET) {
		printf("Unexpectedly got a valid environment configuration\n");
		return -1;
	}

	/* Verify that function pointers are NULL */
	if (check_function_ptrs() < 0)
		goto fail_all;

	rte_power_unset_env();

	/* Perform tests for valid environments.*/
	const enum power_management_env envs[] = {PM_ENV_ACPI_CPUFREQ,
			PM_ENV_KVM_VM,
			PM_ENV_PSTATE_CPUFREQ};

	const int envs_size = sizeof(envs)/sizeof(enum power_management_env);

	int i;
	for (i = 0; i < envs_size; ++i) {

		/* Test setting a valid environment */
		ret = rte_power_set_env(envs[i]);
		if (ret != 0) {
			printf("Unexpectedly unsucceeded on setting a valid environment\n");
			return -1;
		}

		/* Test that the environment has been set */
		env = rte_power_get_env();
		if (env != envs[i]) {
			printf("Not expected environment configuration\n");
			return -1;
		}

		/* Verify that function pointers are NOT NULL */
		if (check_function_ptrs() < 0)
			goto fail_all;

		rte_power_unset_env();

		/* Verify that function pointers are NULL */
		if (check_function_ptrs() < 0)
			goto fail_all;

	}

	return 0;
fail_all:
	rte_power_unset_env();
	return -1;
}
#endif

REGISTER_TEST_COMMAND(power_autotest, test_power);