1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2022 Facebook */ 3 4 #include <test_progs.h> 5 #include <network_helpers.h> 6 #include "dynptr_fail.skel.h" 7 #include "dynptr_success.skel.h" 8 9 enum test_setup_type { 10 SETUP_SYSCALL_SLEEP, 11 SETUP_SKB_PROG, 12 SETUP_SKB_PROG_TP, 13 SETUP_XDP_PROG, 14 }; 15 16 static struct { 17 const char *prog_name; 18 enum test_setup_type type; 19 } success_tests[] = { 20 {"test_read_write", SETUP_SYSCALL_SLEEP}, 21 {"test_dynptr_data", SETUP_SYSCALL_SLEEP}, 22 {"test_dynptr_copy", SETUP_SYSCALL_SLEEP}, 23 {"test_dynptr_copy_xdp", SETUP_XDP_PROG}, 24 {"test_ringbuf", SETUP_SYSCALL_SLEEP}, 25 {"test_skb_readonly", SETUP_SKB_PROG}, 26 {"test_dynptr_skb_data", SETUP_SKB_PROG}, 27 {"test_adjust", SETUP_SYSCALL_SLEEP}, 28 {"test_adjust_err", SETUP_SYSCALL_SLEEP}, 29 {"test_zero_size_dynptr", SETUP_SYSCALL_SLEEP}, 30 {"test_dynptr_is_null", SETUP_SYSCALL_SLEEP}, 31 {"test_dynptr_is_rdonly", SETUP_SKB_PROG}, 32 {"test_dynptr_clone", SETUP_SKB_PROG}, 33 {"test_dynptr_skb_no_buff", SETUP_SKB_PROG}, 34 {"test_dynptr_skb_strcmp", SETUP_SKB_PROG}, 35 {"test_dynptr_skb_tp_btf", SETUP_SKB_PROG_TP}, 36 {"test_probe_read_user_dynptr", SETUP_XDP_PROG}, 37 {"test_probe_read_kernel_dynptr", SETUP_XDP_PROG}, 38 {"test_probe_read_user_str_dynptr", SETUP_XDP_PROG}, 39 {"test_probe_read_kernel_str_dynptr", SETUP_XDP_PROG}, 40 {"test_copy_from_user_dynptr", SETUP_SYSCALL_SLEEP}, 41 {"test_copy_from_user_str_dynptr", SETUP_SYSCALL_SLEEP}, 42 {"test_copy_from_user_task_dynptr", SETUP_SYSCALL_SLEEP}, 43 {"test_copy_from_user_task_str_dynptr", SETUP_SYSCALL_SLEEP}, 44 }; 45 46 static void verify_success(const char *prog_name, enum test_setup_type setup_type) 47 { 48 char user_data[384] = {[0 ... 382] = 'a', '\0'}; 49 struct dynptr_success *skel; 50 struct bpf_program *prog; 51 struct bpf_link *link; 52 int err; 53 54 skel = dynptr_success__open(); 55 if (!ASSERT_OK_PTR(skel, "dynptr_success__open")) 56 return; 57 58 skel->bss->pid = getpid(); 59 60 prog = bpf_object__find_program_by_name(skel->obj, prog_name); 61 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name")) 62 goto cleanup; 63 64 bpf_program__set_autoload(prog, true); 65 66 err = dynptr_success__load(skel); 67 if (!ASSERT_OK(err, "dynptr_success__load")) 68 goto cleanup; 69 70 skel->bss->user_ptr = user_data; 71 skel->data->test_len[0] = sizeof(user_data); 72 memcpy(skel->bss->expected_str, user_data, sizeof(user_data)); 73 74 switch (setup_type) { 75 case SETUP_SYSCALL_SLEEP: 76 link = bpf_program__attach(prog); 77 if (!ASSERT_OK_PTR(link, "bpf_program__attach")) 78 goto cleanup; 79 80 usleep(1); 81 82 bpf_link__destroy(link); 83 break; 84 case SETUP_SKB_PROG: 85 { 86 int prog_fd; 87 char buf[64]; 88 89 LIBBPF_OPTS(bpf_test_run_opts, topts, 90 .data_in = &pkt_v4, 91 .data_size_in = sizeof(pkt_v4), 92 .data_out = buf, 93 .data_size_out = sizeof(buf), 94 .repeat = 1, 95 ); 96 97 prog_fd = bpf_program__fd(prog); 98 if (!ASSERT_GE(prog_fd, 0, "prog_fd")) 99 goto cleanup; 100 101 err = bpf_prog_test_run_opts(prog_fd, &topts); 102 103 if (!ASSERT_OK(err, "test_run")) 104 goto cleanup; 105 106 break; 107 } 108 case SETUP_SKB_PROG_TP: 109 { 110 struct __sk_buff skb = {}; 111 struct bpf_object *obj; 112 int aux_prog_fd; 113 114 /* Just use its test_run to trigger kfree_skb tracepoint */ 115 err = bpf_prog_test_load("./test_pkt_access.bpf.o", BPF_PROG_TYPE_SCHED_CLS, 116 &obj, &aux_prog_fd); 117 if (!ASSERT_OK(err, "prog_load sched cls")) 118 goto cleanup; 119 120 LIBBPF_OPTS(bpf_test_run_opts, topts, 121 .data_in = &pkt_v4, 122 .data_size_in = sizeof(pkt_v4), 123 .ctx_in = &skb, 124 .ctx_size_in = sizeof(skb), 125 ); 126 127 link = bpf_program__attach(prog); 128 if (!ASSERT_OK_PTR(link, "bpf_program__attach")) 129 goto cleanup; 130 131 err = bpf_prog_test_run_opts(aux_prog_fd, &topts); 132 bpf_link__destroy(link); 133 134 if (!ASSERT_OK(err, "test_run")) 135 goto cleanup; 136 137 break; 138 } 139 case SETUP_XDP_PROG: 140 { 141 char data[5000]; 142 int err, prog_fd; 143 LIBBPF_OPTS(bpf_test_run_opts, opts, 144 .data_in = &data, 145 .data_size_in = sizeof(data), 146 .repeat = 1, 147 ); 148 149 prog_fd = bpf_program__fd(prog); 150 err = bpf_prog_test_run_opts(prog_fd, &opts); 151 152 if (!ASSERT_OK(err, "test_run")) 153 goto cleanup; 154 155 break; 156 } 157 } 158 159 ASSERT_EQ(skel->bss->err, 0, "err"); 160 161 cleanup: 162 dynptr_success__destroy(skel); 163 } 164 165 void test_dynptr(void) 166 { 167 int i; 168 169 for (i = 0; i < ARRAY_SIZE(success_tests); i++) { 170 if (!test__start_subtest(success_tests[i].prog_name)) 171 continue; 172 173 verify_success(success_tests[i].prog_name, success_tests[i].type); 174 } 175 176 RUN_TESTS(dynptr_fail); 177 } 178