summaryrefslogtreecommitdiff
path: root/tools/testing/selftests/bpf/progs/struct_ops_refcounted.c
blob: 76dcb6089d7f8e37de76aae48f1ed7d30279c97a (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#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) recongnizes 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,
};