1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (C) 2021. Huawei Technologies Co., Ltd */ 3 #include <test_progs.h> 4 #include "dummy_st_ops.skel.h" 5 #include "trace_dummy_st_ops.skel.h" 6 7 /* Need to keep consistent with definition in include/linux/bpf.h */ 8 struct bpf_dummy_ops_state { 9 int val; 10 }; 11 12 static void test_dummy_st_ops_attach(void) 13 { 14 struct dummy_st_ops *skel; 15 struct bpf_link *link; 16 17 skel = dummy_st_ops__open_and_load(); 18 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 19 return; 20 21 link = bpf_map__attach_struct_ops(skel->maps.dummy_1); 22 ASSERT_EQ(libbpf_get_error(link), -EOPNOTSUPP, "dummy_st_ops_attach"); 23 24 dummy_st_ops__destroy(skel); 25 } 26 27 static void test_dummy_init_ret_value(void) 28 { 29 __u64 args[1] = {0}; 30 LIBBPF_OPTS(bpf_test_run_opts, attr, 31 .ctx_in = args, 32 .ctx_size_in = sizeof(args), 33 ); 34 struct dummy_st_ops *skel; 35 int fd, err; 36 37 skel = dummy_st_ops__open_and_load(); 38 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 39 return; 40 41 fd = bpf_program__fd(skel->progs.test_1); 42 err = bpf_prog_test_run_opts(fd, &attr); 43 ASSERT_OK(err, "test_run"); 44 ASSERT_EQ(attr.retval, 0xf2f3f4f5, "test_ret"); 45 46 dummy_st_ops__destroy(skel); 47 } 48 49 static void test_dummy_init_ptr_arg(void) 50 { 51 int exp_retval = 0xbeef; 52 struct bpf_dummy_ops_state in_state = { 53 .val = exp_retval, 54 }; 55 __u64 args[1] = {(unsigned long)&in_state}; 56 LIBBPF_OPTS(bpf_test_run_opts, attr, 57 .ctx_in = args, 58 .ctx_size_in = sizeof(args), 59 ); 60 struct trace_dummy_st_ops *trace_skel; 61 struct dummy_st_ops *skel; 62 int fd, err; 63 64 skel = dummy_st_ops__open_and_load(); 65 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 66 return; 67 68 fd = bpf_program__fd(skel->progs.test_1); 69 70 trace_skel = trace_dummy_st_ops__open(); 71 if (!ASSERT_OK_PTR(trace_skel, "trace_dummy_st_ops__open")) 72 goto done; 73 74 err = bpf_program__set_attach_target(trace_skel->progs.fentry_test_1, 75 fd, "test_1"); 76 if (!ASSERT_OK(err, "set_attach_target(fentry_test_1)")) 77 goto done; 78 79 err = trace_dummy_st_ops__load(trace_skel); 80 if (!ASSERT_OK(err, "load(trace_skel)")) 81 goto done; 82 83 err = trace_dummy_st_ops__attach(trace_skel); 84 if (!ASSERT_OK(err, "attach(trace_skel)")) 85 goto done; 86 87 err = bpf_prog_test_run_opts(fd, &attr); 88 ASSERT_OK(err, "test_run"); 89 ASSERT_EQ(in_state.val, 0x5a, "test_ptr_ret"); 90 ASSERT_EQ(attr.retval, exp_retval, "test_ret"); 91 ASSERT_EQ(trace_skel->bss->val, exp_retval, "fentry_val"); 92 93 done: 94 dummy_st_ops__destroy(skel); 95 trace_dummy_st_ops__destroy(trace_skel); 96 } 97 98 static void test_dummy_multiple_args(void) 99 { 100 __u64 args[5] = {0, -100, 0x8a5f, 'c', 0x1234567887654321ULL}; 101 LIBBPF_OPTS(bpf_test_run_opts, attr, 102 .ctx_in = args, 103 .ctx_size_in = sizeof(args), 104 ); 105 struct dummy_st_ops *skel; 106 int fd, err; 107 size_t i; 108 char name[8]; 109 110 skel = dummy_st_ops__open_and_load(); 111 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 112 return; 113 114 fd = bpf_program__fd(skel->progs.test_2); 115 err = bpf_prog_test_run_opts(fd, &attr); 116 ASSERT_OK(err, "test_run"); 117 for (i = 0; i < ARRAY_SIZE(args); i++) { 118 snprintf(name, sizeof(name), "arg %zu", i); 119 ASSERT_EQ(skel->bss->test_2_args[i], args[i], name); 120 } 121 122 dummy_st_ops__destroy(skel); 123 } 124 125 void test_dummy_st_ops(void) 126 { 127 if (test__start_subtest("dummy_st_ops_attach")) 128 test_dummy_st_ops_attach(); 129 if (test__start_subtest("dummy_init_ret_value")) 130 test_dummy_init_ret_value(); 131 if (test__start_subtest("dummy_init_ptr_arg")) 132 test_dummy_init_ptr_arg(); 133 if (test__start_subtest("dummy_multiple_args")) 134 test_dummy_multiple_args(); 135 } 136