// SPDX-License-Identifier: GPL-2.0
#include <test_progs.h>
#include <testing_helpers.h>

#include "kfunc_module_order.skel.h"

static int test_run_prog(const struct bpf_program *prog,
			 struct bpf_test_run_opts *opts)
{
	int err;

	err = bpf_prog_test_run_opts(bpf_program__fd(prog), opts);
	if (!ASSERT_OK(err, "bpf_prog_test_run_opts"))
		return err;

	if (!ASSERT_EQ((int)opts->retval, 0, bpf_program__name(prog)))
		return -EINVAL;

	return 0;
}

void test_kfunc_module_order(void)
{
	struct kfunc_module_order *skel;
	char pkt_data[64] = {};
	int err = 0;

	DECLARE_LIBBPF_OPTS(bpf_test_run_opts, test_opts, .data_in = pkt_data,
			    .data_size_in = sizeof(pkt_data));

	err = load_module("bpf_test_modorder_x.ko",
			  env_verbosity > VERBOSE_NONE);
	if (!ASSERT_OK(err, "load bpf_test_modorder_x.ko"))
		return;

	err = load_module("bpf_test_modorder_y.ko",
			  env_verbosity > VERBOSE_NONE);
	if (!ASSERT_OK(err, "load bpf_test_modorder_y.ko"))
		goto exit_modx;

	skel = kfunc_module_order__open_and_load();
	if (!ASSERT_OK_PTR(skel, "kfunc_module_order__open_and_load()")) {
		err = -EINVAL;
		goto exit_mods;
	}

	test_run_prog(skel->progs.call_kfunc_xy, &test_opts);
	test_run_prog(skel->progs.call_kfunc_yx, &test_opts);

	kfunc_module_order__destroy(skel);
exit_mods:
	unload_module("bpf_test_modorder_y", env_verbosity > VERBOSE_NONE);
exit_modx:
	unload_module("bpf_test_modorder_x", env_verbosity > VERBOSE_NONE);
}