tools/testing/selftests/bpf/prog_tests/test_global_funcs.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2020 Facebook */
   3 #include <test_progs.h>
   4 #include "test_global_func1.skel.h"
   5 #include "test_global_func2.skel.h"
   6 #include "test_global_func3.skel.h"
   7 #include "test_global_func4.skel.h"
   8 #include "test_global_func5.skel.h"
   9 #include "test_global_func6.skel.h"
  10 #include "test_global_func7.skel.h"
  11 #include "test_global_func8.skel.h"
  12 #include "test_global_func9.skel.h"
  13 #include "test_global_func10.skel.h"
  14 #include "test_global_func11.skel.h"
  15 #include "test_global_func12.skel.h"
  16 #include "test_global_func13.skel.h"
  17 #include "test_global_func14.skel.h"
  18 #include "test_global_func15.skel.h"
  19 #include "test_global_func16.skel.h"
  20 #include "test_global_func17.skel.h"
  21 #include "test_global_func_ctx_args.skel.h"
  22
  23 #include "bpf/libbpf_internal.h"
  24 #include "btf_helpers.h"
  25
  26 static void check_ctx_arg_type(const struct btf *btf, const struct btf_param *p)
  27 {
  28         const struct btf_type *t;
  29         const char *s;
  30
  31         t = btf__type_by_id(btf, p->type);
  32         if (!ASSERT_EQ(btf_kind(t), BTF_KIND_PTR, "ptr_t"))
  33                 return;
  34
  35         s = btf_type_raw_dump(btf, t->type);
  36         if (!ASSERT_HAS_SUBSTR(s, "STRUCT 'bpf_perf_event_data' size=0 vlen=0",
  37                                "ctx_struct_t"))
  38                 return;
  39 }
  40
  41 static void subtest_ctx_arg_rewrite(void)
  42 {
  43         struct test_global_func_ctx_args *skel = NULL;
  44         struct bpf_prog_info info;
  45         char func_info_buf[1024] __attribute__((aligned(8)));
  46         struct bpf_func_info_min *rec;
  47         struct btf *btf = NULL;
  48         __u32 info_len = sizeof(info);
  49         int err, fd, i;
  50         struct btf *kern_btf = NULL;
  51
  52         kern_btf = btf__load_vmlinux_btf();
  53         if (!ASSERT_OK_PTR(kern_btf, "kern_btf_load"))
  54                 return;
  55
  56         /* simple detection of kernel native arg:ctx tag support */
  57         if (btf__find_by_name_kind(kern_btf, "bpf_subprog_arg_info", BTF_KIND_STRUCT) > 0) {
  58                 test__skip();
  59                 btf__free(kern_btf);
  60                 return;
  61         }
  62         btf__free(kern_btf);
  63
  64         skel = test_global_func_ctx_args__open();
  65         if (!ASSERT_OK_PTR(skel, "skel_open"))
  66                 return;
  67
  68         bpf_program__set_autoload(skel->progs.arg_tag_ctx_perf, true);
  69
  70         err = test_global_func_ctx_args__load(skel);
  71         if (!ASSERT_OK(err, "skel_load"))
  72                 goto out;
  73
  74         memset(&info, 0, sizeof(info));
  75         info.func_info = ptr_to_u64(&func_info_buf);
  76         info.nr_func_info = 3;
  77         info.func_info_rec_size = sizeof(struct bpf_func_info_min);
  78
  79         fd = bpf_program__fd(skel->progs.arg_tag_ctx_perf);
  80         err = bpf_prog_get_info_by_fd(fd, &info, &info_len);
  81         if (!ASSERT_OK(err, "prog_info"))
  82                 goto out;
  83
  84         if (!ASSERT_EQ(info.nr_func_info, 3, "nr_func_info"))
  85                 goto out;
  86
  87         btf = btf__load_from_kernel_by_id(info.btf_id);
  88         if (!ASSERT_OK_PTR(btf, "obj_kern_btf"))
  89                 goto out;
  90
  91         rec = (struct bpf_func_info_min *)func_info_buf;
  92         for (i = 0; i < info.nr_func_info; i++, rec = (void *)rec + info.func_info_rec_size) {
  93                 const struct btf_type *fn_t, *proto_t;
  94                 const char *name;
  95
  96                 if (rec->insn_off == 0)
  97                         continue; /* main prog, skip */
  98
  99                 fn_t = btf__type_by_id(btf, rec->type_id);
 100                 if (!ASSERT_OK_PTR(fn_t, "fn_type"))
 101                         goto out;
 102                 if (!ASSERT_EQ(btf_kind(fn_t), BTF_KIND_FUNC, "fn_type_kind"))
 103                         goto out;
 104                 proto_t = btf__type_by_id(btf, fn_t->type);
 105                 if (!ASSERT_OK_PTR(proto_t, "proto_type"))
 106                         goto out;
 107
 108                 name = btf__name_by_offset(btf, fn_t->name_off);
 109                 if (strcmp(name, "subprog_ctx_tag") == 0) {
 110                         /* int subprog_ctx_tag(void *ctx __arg_ctx) */
 111                         if (!ASSERT_EQ(btf_vlen(proto_t), 1, "arg_cnt"))
 112                                 goto out;
 113
 114                         /* arg 0 is PTR -> STRUCT bpf_perf_event_data */
 115                         check_ctx_arg_type(btf, &btf_params(proto_t)[0]);
 116                 } else if (strcmp(name, "subprog_multi_ctx_tags") == 0) {
 117                         /* int subprog_multi_ctx_tags(void *ctx1 __arg_ctx,
 118                          *                            struct my_struct *mem,
 119                          *                            void *ctx2 __arg_ctx)
 120                          */
 121                         if (!ASSERT_EQ(btf_vlen(proto_t), 3, "arg_cnt"))
 122                                 goto out;
 123
 124                         /* arg 0 is PTR -> STRUCT bpf_perf_event_data */
 125                         check_ctx_arg_type(btf, &btf_params(proto_t)[0]);
 126                         /* arg 2 is PTR -> STRUCT bpf_perf_event_data */
 127                         check_ctx_arg_type(btf, &btf_params(proto_t)[2]);
 128                 } else {
 129                         ASSERT_FAIL("unexpected subprog %s", name);
 130                         goto out;
 131                 }
 132         }
 133
 134 out:
 135         btf__free(btf);
 136         test_global_func_ctx_args__destroy(skel);
 137 }
 138
 139 void test_test_global_funcs(void)
 140 {
 141         RUN_TESTS(test_global_func1);
 142         RUN_TESTS(test_global_func2);
 143         RUN_TESTS(test_global_func3);
 144         RUN_TESTS(test_global_func4);
 145         RUN_TESTS(test_global_func5);
 146         RUN_TESTS(test_global_func6);
 147         RUN_TESTS(test_global_func7);
 148         RUN_TESTS(test_global_func8);
 149         RUN_TESTS(test_global_func9);
 150         RUN_TESTS(test_global_func10);
 151         RUN_TESTS(test_global_func11);
 152         RUN_TESTS(test_global_func12);
 153         RUN_TESTS(test_global_func13);
 154         RUN_TESTS(test_global_func14);
 155         RUN_TESTS(test_global_func15);
 156         RUN_TESTS(test_global_func16);
 157         RUN_TESTS(test_global_func17);
 158         RUN_TESTS(test_global_func_ctx_args);
 159
 160         if (test__start_subtest("ctx_arg_rewrite"))
 161                 subtest_ctx_arg_rewrite();
 162 }