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

#include <ynl.h>

#include <arpa/inet.h>
#include <net/if.h>

#include <kselftest_harness.h>

#include "rt-link-user.h"

static void rt_link_print(struct __test_metadata *_metadata,
			  struct rt_link_getlink_rsp *r)
{
	unsigned int i;

	EXPECT_TRUE((bool)r->_hdr.ifi_index);
	ksft_print_msg("%3d: ", r->_hdr.ifi_index);

	EXPECT_TRUE((bool)r->_len.ifname);
	if (r->_len.ifname)
		printf("%6s: ", r->ifname);

	if (r->_present.mtu)
		printf("mtu %5d  ", r->mtu);

	if (r->linkinfo._len.kind)
		printf("kind %-8s  ", r->linkinfo.kind);
	else
		printf("     %8s  ", "");

	if (r->prop_list._count.alt_ifname) {
		printf("altname ");
		for (i = 0; i < r->prop_list._count.alt_ifname; i++)
			printf("%s ", r->prop_list.alt_ifname[i]->str);
		printf(" ");
	}

	if (r->linkinfo._present.data && r->linkinfo.data._present.netkit) {
		struct rt_link_linkinfo_netkit_attrs *netkit;
		const char *name;

		netkit = &r->linkinfo.data.netkit;
		printf("primary %d  ", netkit->primary);

		name = NULL;
		if (netkit->_present.policy)
			name = rt_link_netkit_policy_str(netkit->policy);
		if (name)
			printf("policy %s  ", name);
	}

	printf("\n");
}

static int netkit_create(struct ynl_sock *ys)
{
	struct rt_link_getlink_ntf *ntf_gl;
	struct rt_link_newlink_req *req;
	struct ynl_ntf_base_type *ntf;
	int ret;

	req = rt_link_newlink_req_alloc();
	if (!req)
		return -1;

	rt_link_newlink_req_set_nlflags(req, NLM_F_CREATE | NLM_F_ECHO);
	rt_link_newlink_req_set_linkinfo_kind(req, "netkit");
	rt_link_newlink_req_set_linkinfo_data_netkit_policy(req, NETKIT_DROP);

	ret = rt_link_newlink(ys, req);
	rt_link_newlink_req_free(req);
	if (ret)
		return -1;

	if (!ynl_has_ntf(ys))
		return 0;

	ntf = ynl_ntf_dequeue(ys);
	if (!ntf || ntf->cmd != RTM_NEWLINK) {
		ynl_ntf_free(ntf);
		return 0;
	}
	ntf_gl = (void *)ntf;
	ret = ntf_gl->obj._hdr.ifi_index;
	ynl_ntf_free(ntf);

	return ret;
}

static void netkit_delete(struct __test_metadata *_metadata,
			  struct ynl_sock *ys, int ifindex)
{
	struct rt_link_dellink_req *req;

	req = rt_link_dellink_req_alloc();
	ASSERT_NE(NULL, req);

	req->_hdr.ifi_index = ifindex;
	EXPECT_EQ(0, rt_link_dellink(ys, req));
	rt_link_dellink_req_free(req);
}

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

FIXTURE_SETUP(rt_link)
{
	struct ynl_error yerr;

	self->ys = ynl_sock_create(&ynl_rt_link_family, &yerr);
	ASSERT_NE(NULL, self->ys) {
		TH_LOG("failed to create rt-link socket: %s", yerr.msg);
	}
}

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

TEST_F(rt_link, dump)
{
	struct rt_link_getlink_req_dump *req;
	struct rt_link_getlink_list *rsp;

	req = rt_link_getlink_req_dump_alloc();
	ASSERT_NE(NULL, req);
	rsp = rt_link_getlink_dump(self->ys, req);
	rt_link_getlink_req_dump_free(req);
	ASSERT_NE(NULL, rsp) {
		TH_LOG("dump failed: %s", self->ys->err.msg);
	}
	ASSERT_FALSE(ynl_dump_empty(rsp));

	ynl_dump_foreach(rsp, link)
		rt_link_print(_metadata, link);

	rt_link_getlink_list_free(rsp);
}

TEST_F(rt_link, netkit)
{
	struct rt_link_getlink_req_dump *dreq;
	struct rt_link_getlink_list *rsp;
	bool found = false;
	int netkit_ifindex;

	/* Create netkit with valid policy */
	netkit_ifindex = netkit_create(self->ys);
	ASSERT_GT(netkit_ifindex, 0)
		TH_LOG("failed to create netkit: %s", self->ys->err.msg);

	/* Verify it appears in a dump */
	dreq = rt_link_getlink_req_dump_alloc();
	ASSERT_NE(NULL, dreq);
	rsp = rt_link_getlink_dump(self->ys, dreq);
	rt_link_getlink_req_dump_free(dreq);
	ASSERT_NE(NULL, rsp) {
		TH_LOG("dump failed: %s", self->ys->err.msg);
	}

	ynl_dump_foreach(rsp, link) {
		if (link->_hdr.ifi_index == netkit_ifindex) {
			rt_link_print(_metadata, link);
			found = true;
		}
	}
	rt_link_getlink_list_free(rsp);
	EXPECT_TRUE(found);

	netkit_delete(_metadata, self->ys, netkit_ifindex);
}

TEST_F(rt_link, netkit_err_msg)
{
	struct rt_link_newlink_req *req;
	int ret;

	/* Test creating netkit with bad policy - should fail */
	req = rt_link_newlink_req_alloc();
	ASSERT_NE(NULL, req);
	rt_link_newlink_req_set_nlflags(req, NLM_F_CREATE);
	rt_link_newlink_req_set_linkinfo_kind(req, "netkit");
	rt_link_newlink_req_set_linkinfo_data_netkit_policy(req, 10);

	ret = rt_link_newlink(self->ys, req);
	rt_link_newlink_req_free(req);
	EXPECT_NE(0, ret) {
		TH_LOG("creating netkit with bad policy should fail");
	}

	/* Expect:
	 * Kernel error: 'Provided default xmit policy not supported' (bad attribute: .linkinfo.data(netkit).policy)
	 */
	EXPECT_NE(NULL, strstr(self->ys->err.msg, "bad attribute: .linkinfo.data(netkit).policy")) {
		TH_LOG("expected extack msg not found: %s",
		       self->ys->err.msg);
	}
}

TEST_HARNESS_MAIN