1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates.*/
3
4 #define _GNU_SOURCE
5 #include <unistd.h>
6 #include <sys/syscall.h>
7 #include <sys/types.h>
8 #include <test_progs.h>
9 #include <bpf/btf.h>
10 #include "rcu_read_lock.skel.h"
11 #include "cgroup_helpers.h"
12
13 static unsigned long long cgroup_id;
14
test_success(void)15 static void test_success(void)
16 {
17 struct rcu_read_lock *skel;
18 int err;
19
20 skel = rcu_read_lock__open();
21 if (!ASSERT_OK_PTR(skel, "skel_open"))
22 return;
23
24 skel->bss->target_pid = syscall(SYS_gettid);
25
26 bpf_program__set_autoload(skel->progs.get_cgroup_id, true);
27 bpf_program__set_autoload(skel->progs.task_succ, true);
28 bpf_program__set_autoload(skel->progs.two_regions, true);
29 bpf_program__set_autoload(skel->progs.non_sleepable_1, true);
30 bpf_program__set_autoload(skel->progs.non_sleepable_2, true);
31 bpf_program__set_autoload(skel->progs.task_trusted_non_rcuptr, true);
32 err = rcu_read_lock__load(skel);
33 if (!ASSERT_OK(err, "skel_load"))
34 goto out;
35
36 err = rcu_read_lock__attach(skel);
37 if (!ASSERT_OK(err, "skel_attach"))
38 goto out;
39
40 syscall(SYS_getpgid);
41
42 ASSERT_EQ(skel->bss->task_storage_val, 2, "task_storage_val");
43 ASSERT_EQ(skel->bss->cgroup_id, cgroup_id, "cgroup_id");
44 out:
45 rcu_read_lock__destroy(skel);
46 }
47
test_rcuptr_acquire(void)48 static void test_rcuptr_acquire(void)
49 {
50 struct rcu_read_lock *skel;
51 int err;
52
53 skel = rcu_read_lock__open();
54 if (!ASSERT_OK_PTR(skel, "skel_open"))
55 return;
56
57 skel->bss->target_pid = syscall(SYS_gettid);
58
59 bpf_program__set_autoload(skel->progs.task_acquire, true);
60 err = rcu_read_lock__load(skel);
61 if (!ASSERT_OK(err, "skel_load"))
62 goto out;
63
64 err = rcu_read_lock__attach(skel);
65 ASSERT_OK(err, "skel_attach");
66 out:
67 rcu_read_lock__destroy(skel);
68 }
69
70 static const char * const inproper_region_tests[] = {
71 "miss_lock",
72 "no_lock",
73 "miss_unlock",
74 "non_sleepable_rcu_mismatch",
75 "inproper_sleepable_helper",
76 "inproper_sleepable_kfunc",
77 "nested_rcu_region",
78 };
79
test_inproper_region(void)80 static void test_inproper_region(void)
81 {
82 struct rcu_read_lock *skel;
83 struct bpf_program *prog;
84 int i, err;
85
86 for (i = 0; i < ARRAY_SIZE(inproper_region_tests); i++) {
87 skel = rcu_read_lock__open();
88 if (!ASSERT_OK_PTR(skel, "skel_open"))
89 return;
90
91 prog = bpf_object__find_program_by_name(skel->obj, inproper_region_tests[i]);
92 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
93 goto out;
94 bpf_program__set_autoload(prog, true);
95 err = rcu_read_lock__load(skel);
96 ASSERT_ERR(err, "skel_load");
97 out:
98 rcu_read_lock__destroy(skel);
99 }
100 }
101
102 static const char * const rcuptr_misuse_tests[] = {
103 "task_untrusted_rcuptr",
104 "cross_rcu_region",
105 };
106
test_rcuptr_misuse(void)107 static void test_rcuptr_misuse(void)
108 {
109 struct rcu_read_lock *skel;
110 struct bpf_program *prog;
111 int i, err;
112
113 for (i = 0; i < ARRAY_SIZE(rcuptr_misuse_tests); i++) {
114 skel = rcu_read_lock__open();
115 if (!ASSERT_OK_PTR(skel, "skel_open"))
116 return;
117
118 prog = bpf_object__find_program_by_name(skel->obj, rcuptr_misuse_tests[i]);
119 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
120 goto out;
121 bpf_program__set_autoload(prog, true);
122 err = rcu_read_lock__load(skel);
123 ASSERT_ERR(err, "skel_load");
124 out:
125 rcu_read_lock__destroy(skel);
126 }
127 }
128
test_rcu_read_lock(void)129 void test_rcu_read_lock(void)
130 {
131 int cgroup_fd;
132
133 cgroup_fd = test__join_cgroup("/rcu_read_lock");
134 if (!ASSERT_GE(cgroup_fd, 0, "join_cgroup /rcu_read_lock"))
135 goto out;
136
137 cgroup_id = get_cgroup_id("/rcu_read_lock");
138 if (test__start_subtest("success"))
139 test_success();
140 if (test__start_subtest("rcuptr_acquire"))
141 test_rcuptr_acquire();
142 if (test__start_subtest("negative_tests_inproper_region"))
143 test_inproper_region();
144 if (test__start_subtest("negative_tests_rcuptr_misuse"))
145 test_rcuptr_misuse();
146 close(cgroup_fd);
147 out:;
148 }
149
