tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
1// SPDX-License-Identifier: GPL-2.0
2
3#include <vmlinux.h>
4#include <errno.h>
5#include <bpf/bpf_helpers.h>
6
7#include "task_local_data.bpf.h"
8
9struct tld_keys {
10 tld_key_t value0;
11 tld_key_t value1;
12 tld_key_t value2;
13 tld_key_t value_not_exist;
14};
15
16struct test_tld_struct {
17 __u64 a;
18 __u64 b;
19 __u64 c;
20 __u64 d;
21};
22
23int test_value0;
24int test_value1;
25struct test_tld_struct test_value2;
26
27SEC("syscall")
28int task_main(void *ctx)
29{
30 struct tld_object tld_obj;
31 struct test_tld_struct *struct_p;
32 struct task_struct *task;
33 int err, *int_p;
34
35 task = bpf_get_current_task_btf();
36 err = tld_object_init(task, &tld_obj);
37 if (err)
38 return 1;
39
40 int_p = tld_get_data(&tld_obj, value0, "value0", sizeof(int));
41 if (int_p)
42 test_value0 = *int_p;
43 else
44 return 2;
45
46 int_p = tld_get_data(&tld_obj, value1, "value1", sizeof(int));
47 if (int_p)
48 test_value1 = *int_p;
49 else
50 return 3;
51
52 struct_p = tld_get_data(&tld_obj, value2, "value2", sizeof(struct test_tld_struct));
53 if (struct_p)
54 test_value2 = *struct_p;
55 else
56 return 4;
57
58 int_p = tld_get_data(&tld_obj, value_not_exist, "value_not_exist", sizeof(int));
59 if (int_p)
60 return 5;
61
62 return 0;
63}
64
65char _license[] SEC("license") = "GPL";