// SPDX-License-Identifier: GPL-2.0
#include <stdio.h>
#include <string.h>

#include <ynl.h>

#include <kselftest_harness.h>

#include "devlink-user.h"

FIXTURE(devlink)
{
	struct ynl_sock *ys;
};

FIXTURE_SETUP(devlink)
{
	self->ys = ynl_sock_create(&ynl_devlink_family, NULL);
	ASSERT_NE(NULL, self->ys)
		TH_LOG("failed to create devlink socket");
}

FIXTURE_TEARDOWN(devlink)
{
	ynl_sock_destroy(self->ys);
}

TEST_F(devlink, dump)
{
	struct devlink_get_list *devs;

	devs = devlink_get_dump(self->ys);
	ASSERT_NE(NULL, devs) {
		TH_LOG("dump failed: %s", self->ys->err.msg);
	}

	if (ynl_dump_empty(devs)) {
		devlink_get_list_free(devs);
		SKIP(return, "no entries in dump");
	}

	ynl_dump_foreach(devs, d) {
		EXPECT_TRUE((bool)d->_len.bus_name);
		EXPECT_TRUE((bool)d->_len.dev_name);
		ksft_print_msg("%s/%s\n", d->bus_name, d->dev_name);
	}

	devlink_get_list_free(devs);
}

TEST_F(devlink, info)
{
	struct devlink_get_list *devs;

	devs = devlink_get_dump(self->ys);
	ASSERT_NE(NULL, devs) {
		TH_LOG("dump failed: %s", self->ys->err.msg);
	}

	if (ynl_dump_empty(devs)) {
		devlink_get_list_free(devs);
		SKIP(return, "no devices to query");
	}

	ynl_dump_foreach(devs, d) {
		struct devlink_info_get_req *info_req;
		struct devlink_info_get_rsp *info_rsp;
		unsigned int i;

		EXPECT_TRUE((bool)d->_len.bus_name);
		EXPECT_TRUE((bool)d->_len.dev_name);
		ksft_print_msg("%s/%s:\n", d->bus_name, d->dev_name);

		info_req = devlink_info_get_req_alloc();
		ASSERT_NE(NULL, info_req);
		devlink_info_get_req_set_bus_name(info_req, d->bus_name);
		devlink_info_get_req_set_dev_name(info_req, d->dev_name);

		info_rsp = devlink_info_get(self->ys, info_req);
		devlink_info_get_req_free(info_req);
		ASSERT_NE(NULL, info_rsp) {
			devlink_get_list_free(devs);
			TH_LOG("info_get failed: %s", self->ys->err.msg);
		}

		EXPECT_TRUE((bool)info_rsp->_len.info_driver_name);
		if (info_rsp->_len.info_driver_name)
			ksft_print_msg("  driver: %s\n",
				       info_rsp->info_driver_name);
		if (info_rsp->_count.info_version_running)
			ksft_print_msg("  running fw:\n");
		for (i = 0; i < info_rsp->_count.info_version_running; i++)
			ksft_print_msg("    %s: %s\n",
				       info_rsp->info_version_running[i].info_version_name,
				       info_rsp->info_version_running[i].info_version_value);
		devlink_info_get_rsp_free(info_rsp);
	}
	devlink_get_list_free(devs);
}

TEST_HARNESS_MAIN