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 26static 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 41static 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 134out: 135 btf__free(btf); 136 test_global_func_ctx_args__destroy(skel); 137} 138 139void 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} 163