| #include <vmlinux.h> | |
| #include <bpf/bpf_tracing.h> | |
| #include "../test_kmods/bpf_testmod.h" | |
| #include "bpf_misc.h" | |
| char _license[] SEC("license") = "GPL"; | |
| __attribute__((nomerge)) extern void bpf_task_release(struct task_struct *p) __ksym; | |
| /* This is a test BPF program that uses struct_ops to access a referenced | |
| * kptr argument. This is a test for the verifier to ensure that it | |
| * 1) recognizes the task as a referenced object (i.e., ref_obj_id > 0), and | |
| * 2) the same reference can be acquired from multiple paths as long as it | |
| * has not been released. | |
| */ | |
| SEC("struct_ops/test_refcounted") | |
| int BPF_PROG(refcounted, int dummy, struct task_struct *task) | |
| { | |
| if (dummy == 1) | |
| bpf_task_release(task); | |
| else | |
| bpf_task_release(task); | |
| return 0; | |
| } | |
| SEC(".struct_ops.link") | |
| struct bpf_testmod_ops testmod_refcounted = { | |
| .test_refcounted = (void *)refcounted, | |
| }; | |