// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2020 Facebook */
#include <test_progs.h>
#include "test_global_func1.skel.h"
#include "test_global_func2.skel.h"
#include "test_global_func3.skel.h"
#include "test_global_func4.skel.h"
#include "test_global_func5.skel.h"
#include "test_global_func6.skel.h"
#include "test_global_func7.skel.h"
#include "test_global_func8.skel.h"
#include "test_global_func9.skel.h"
#include "test_global_func10.skel.h"
#include "test_global_func11.skel.h"
#include "test_global_func12.skel.h"
#include "test_global_func13.skel.h"
#include "test_global_func14.skel.h"
#include "test_global_func15.skel.h"
#include "test_global_func16.skel.h"
#include "test_global_func17.skel.h"
#include "test_global_func_ctx_args.skel.h"

#include "bpf/libbpf_internal.h"
#include "btf_helpers.h"

static void check_ctx_arg_type(const struct btf *btf, const struct btf_param *p)
{
        const struct btf_type *t;
        const char *s;

        t = btf__type_by_id(btf, p->type);
        if (!ASSERT_EQ(btf_kind(t), BTF_KIND_PTR, "ptr_t"))
                return;

        s = btf_type_raw_dump(btf, t->type);
        if (!ASSERT_HAS_SUBSTR(s, "STRUCT 'bpf_perf_event_data' size=0 vlen=0",
                               "ctx_struct_t"))
                return;
}

static void subtest_ctx_arg_rewrite(void)
{
        struct test_global_func_ctx_args *skel = NULL;
        struct bpf_prog_info info;
        char func_info_buf[1024] __attribute__((aligned(8)));
        struct bpf_func_info_min *rec;
        struct btf *btf = NULL;
        __u32 info_len = sizeof(info);
        int err, fd, i;
        struct btf *kern_btf = NULL;

        kern_btf = btf__load_vmlinux_btf();
        if (!ASSERT_OK_PTR(kern_btf, "kern_btf_load"))
                return;

        /* simple detection of kernel native arg:ctx tag support */
        if (btf__find_by_name_kind(kern_btf, "bpf_subprog_arg_info", BTF_KIND_STRUCT) > 0) {
                test__skip();
                btf__free(kern_btf);
                return;
        }
        btf__free(kern_btf);

        skel = test_global_func_ctx_args__open();
        if (!ASSERT_OK_PTR(skel, "skel_open"))
                return;

        bpf_program__set_autoload(skel->progs.arg_tag_ctx_perf, true);

        err = test_global_func_ctx_args__load(skel);
        if (!ASSERT_OK(err, "skel_load"))
                goto out;

        memset(&info, 0, sizeof(info));
        info.func_info = ptr_to_u64(&func_info_buf);
        info.nr_func_info = 3;
        info.func_info_rec_size = sizeof(struct bpf_func_info_min);

        fd = bpf_program__fd(skel->progs.arg_tag_ctx_perf);
        err = bpf_prog_get_info_by_fd(fd, &info, &info_len);
        if (!ASSERT_OK(err, "prog_info"))
                goto out;

        if (!ASSERT_EQ(info.nr_func_info, 3, "nr_func_info"))
                goto out;

        btf = btf__load_from_kernel_by_id(info.btf_id);
        if (!ASSERT_OK_PTR(btf, "obj_kern_btf"))
                goto out;

        rec = (struct bpf_func_info_min *)func_info_buf;
        for (i = 0; i < info.nr_func_info; i++, rec = (void *)rec + info.func_info_rec_size) {
                const struct btf_type *fn_t, *proto_t;
                const char *name;

                if (rec->insn_off == 0)
                        continue; /* main prog, skip */

                fn_t = btf__type_by_id(btf, rec->type_id);
                if (!ASSERT_OK_PTR(fn_t, "fn_type"))
                        goto out;
                if (!ASSERT_EQ(btf_kind(fn_t), BTF_KIND_FUNC, "fn_type_kind"))
                        goto out;
                proto_t = btf__type_by_id(btf, fn_t->type);
                if (!ASSERT_OK_PTR(proto_t, "proto_type"))
                        goto out;

                name = btf__name_by_offset(btf, fn_t->name_off);
                if (strcmp(name, "subprog_ctx_tag") == 0) {
                        /* int subprog_ctx_tag(void *ctx __arg_ctx) */
                        if (!ASSERT_EQ(btf_vlen(proto_t), 1, "arg_cnt"))
                                goto out;

                        /* arg 0 is PTR -> STRUCT bpf_perf_event_data */
                        check_ctx_arg_type(btf, &btf_params(proto_t)[0]);
                } else if (strcmp(name, "subprog_multi_ctx_tags") == 0) {
                        /* int subprog_multi_ctx_tags(void *ctx1 __arg_ctx,
                         *                            struct my_struct *mem,
                         *                            void *ctx2 __arg_ctx)
                         */
                        if (!ASSERT_EQ(btf_vlen(proto_t), 3, "arg_cnt"))
                                goto out;

                        /* arg 0 is PTR -> STRUCT bpf_perf_event_data */
                        check_ctx_arg_type(btf, &btf_params(proto_t)[0]);
                        /* arg 2 is PTR -> STRUCT bpf_perf_event_data */
                        check_ctx_arg_type(btf, &btf_params(proto_t)[2]);
                } else {
                        ASSERT_FAIL("unexpected subprog %s", name);
                        goto out;
                }
        }

out:
        btf__free(btf);
        test_global_func_ctx_args__destroy(skel);
}

void test_test_global_funcs(void)
{
        RUN_TESTS(test_global_func1);
        RUN_TESTS(test_global_func2);
        RUN_TESTS(test_global_func3);
        RUN_TESTS(test_global_func4);
        RUN_TESTS(test_global_func5);
        RUN_TESTS(test_global_func6);
        RUN_TESTS(test_global_func7);
        RUN_TESTS(test_global_func8);
        RUN_TESTS(test_global_func9);
        RUN_TESTS(test_global_func10);
        RUN_TESTS(test_global_func11);
        RUN_TESTS(test_global_func12);
        RUN_TESTS(test_global_func13);
        RUN_TESTS(test_global_func14);
        RUN_TESTS(test_global_func15);
        RUN_TESTS(test_global_func16);
        RUN_TESTS(test_global_func17);
        RUN_TESTS(test_global_func_ctx_args);

        if (test__start_subtest("ctx_arg_rewrite"))
                subtest_ctx_arg_rewrite();
}