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 6 /* Need to keep consistent with definition in include/linux/bpf.h */ 7 struct bpf_dummy_ops_state { 8 int val; 9 }; 10 11 static void test_dummy_st_ops_attach(void) 12 { 13 struct dummy_st_ops *skel; 14 struct bpf_link *link; 15 16 skel = dummy_st_ops__open_and_load(); 17 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 18 return; 19 20 link = bpf_map__attach_struct_ops(skel->maps.dummy_1); 21 ASSERT_EQ(libbpf_get_error(link), -EOPNOTSUPP, "dummy_st_ops_attach"); 22 23 dummy_st_ops__destroy(skel); 24 } 25 26 static void test_dummy_init_ret_value(void) 27 { 28 __u64 args[1] = {0}; 29 struct bpf_prog_test_run_attr attr = { 30 .ctx_size_in = sizeof(args), 31 .ctx_in = args, 32 }; 33 struct dummy_st_ops *skel; 34 int fd, err; 35 36 skel = dummy_st_ops__open_and_load(); 37 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 38 return; 39 40 fd = bpf_program__fd(skel->progs.test_1); 41 attr.prog_fd = fd; 42 err = bpf_prog_test_run_xattr(&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 struct bpf_prog_test_run_attr attr = { 57 .ctx_size_in = sizeof(args), 58 .ctx_in = args, 59 }; 60 struct dummy_st_ops *skel; 61 int fd, err; 62 63 skel = dummy_st_ops__open_and_load(); 64 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 65 return; 66 67 fd = bpf_program__fd(skel->progs.test_1); 68 attr.prog_fd = fd; 69 err = bpf_prog_test_run_xattr(&attr); 70 ASSERT_OK(err, "test_run"); 71 ASSERT_EQ(in_state.val, 0x5a, "test_ptr_ret"); 72 ASSERT_EQ(attr.retval, exp_retval, "test_ret"); 73 74 dummy_st_ops__destroy(skel); 75 } 76 77 static void test_dummy_multiple_args(void) 78 { 79 __u64 args[5] = {0, -100, 0x8a5f, 'c', 0x1234567887654321ULL}; 80 struct bpf_prog_test_run_attr attr = { 81 .ctx_size_in = sizeof(args), 82 .ctx_in = args, 83 }; 84 struct dummy_st_ops *skel; 85 int fd, err; 86 size_t i; 87 char name[8]; 88 89 skel = dummy_st_ops__open_and_load(); 90 if (!ASSERT_OK_PTR(skel, "dummy_st_ops_load")) 91 return; 92 93 fd = bpf_program__fd(skel->progs.test_2); 94 attr.prog_fd = fd; 95 err = bpf_prog_test_run_xattr(&attr); 96 ASSERT_OK(err, "test_run"); 97 for (i = 0; i < ARRAY_SIZE(args); i++) { 98 snprintf(name, sizeof(name), "arg %zu", i); 99 ASSERT_EQ(skel->bss->test_2_args[i], args[i], name); 100 } 101 102 dummy_st_ops__destroy(skel); 103 } 104 105 void test_dummy_st_ops(void) 106 { 107 if (test__start_subtest("dummy_st_ops_attach")) 108 test_dummy_st_ops_attach(); 109 if (test__start_subtest("dummy_init_ret_value")) 110 test_dummy_init_ret_value(); 111 if (test__start_subtest("dummy_init_ptr_arg")) 112 test_dummy_init_ptr_arg(); 113 if (test__start_subtest("dummy_multiple_args")) 114 test_dummy_multiple_args(); 115 } 116