1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (C) 2023 Yafang Shao <laoar.shao@gmail.com> */
3
4 #include <sys/types.h>
5 #include <unistd.h>
6 #include <test_progs.h>
7 #include "cgroup_helpers.h"
8 #include "test_cgroup1_hierarchy.skel.h"
9
bpf_cgroup1(struct test_cgroup1_hierarchy * skel)10 static void bpf_cgroup1(struct test_cgroup1_hierarchy *skel)
11 {
12 struct bpf_link *lsm_link, *fentry_link;
13 int err;
14
15 /* Attach LSM prog first */
16 lsm_link = bpf_program__attach_lsm(skel->progs.lsm_run);
17 if (!ASSERT_OK_PTR(lsm_link, "lsm_attach"))
18 return;
19
20 /* LSM prog will be triggered when attaching fentry */
21 fentry_link = bpf_program__attach_trace(skel->progs.fentry_run);
22 ASSERT_NULL(fentry_link, "fentry_attach_fail");
23
24 err = bpf_link__destroy(lsm_link);
25 ASSERT_OK(err, "destroy_lsm");
26 }
27
bpf_cgroup1_sleepable(struct test_cgroup1_hierarchy * skel)28 static void bpf_cgroup1_sleepable(struct test_cgroup1_hierarchy *skel)
29 {
30 struct bpf_link *lsm_link, *fentry_link;
31 int err;
32
33 /* Attach LSM prog first */
34 lsm_link = bpf_program__attach_lsm(skel->progs.lsm_s_run);
35 if (!ASSERT_OK_PTR(lsm_link, "lsm_attach"))
36 return;
37
38 /* LSM prog will be triggered when attaching fentry */
39 fentry_link = bpf_program__attach_trace(skel->progs.fentry_run);
40 ASSERT_NULL(fentry_link, "fentry_attach_fail");
41
42 err = bpf_link__destroy(lsm_link);
43 ASSERT_OK(err, "destroy_lsm");
44 }
45
bpf_cgroup1_invalid_id(struct test_cgroup1_hierarchy * skel)46 static void bpf_cgroup1_invalid_id(struct test_cgroup1_hierarchy *skel)
47 {
48 struct bpf_link *lsm_link, *fentry_link;
49 int err;
50
51 /* Attach LSM prog first */
52 lsm_link = bpf_program__attach_lsm(skel->progs.lsm_run);
53 if (!ASSERT_OK_PTR(lsm_link, "lsm_attach"))
54 return;
55
56 /* LSM prog will be triggered when attaching fentry */
57 fentry_link = bpf_program__attach_trace(skel->progs.fentry_run);
58 if (!ASSERT_OK_PTR(fentry_link, "fentry_attach_success"))
59 goto cleanup;
60
61 err = bpf_link__destroy(fentry_link);
62 ASSERT_OK(err, "destroy_lsm");
63
64 cleanup:
65 err = bpf_link__destroy(lsm_link);
66 ASSERT_OK(err, "destroy_fentry");
67 }
68
test_cgroup1_hierarchy(void)69 void test_cgroup1_hierarchy(void)
70 {
71 struct test_cgroup1_hierarchy *skel;
72 __u64 current_cgid;
73 int hid, err;
74
75 skel = test_cgroup1_hierarchy__open();
76 if (!ASSERT_OK_PTR(skel, "open"))
77 return;
78
79 skel->bss->target_pid = getpid();
80
81 err = bpf_program__set_attach_target(skel->progs.fentry_run, 0, "bpf_fentry_test1");
82 if (!ASSERT_OK(err, "fentry_set_target"))
83 goto destroy;
84
85 err = test_cgroup1_hierarchy__load(skel);
86 if (!ASSERT_OK(err, "load"))
87 goto destroy;
88
89 /* Setup cgroup1 hierarchy */
90 err = setup_cgroup_environment();
91 if (!ASSERT_OK(err, "setup_cgroup_environment"))
92 goto destroy;
93 err = setup_classid_environment();
94 if (!ASSERT_OK(err, "setup_classid_environment"))
95 goto cleanup_cgroup;
96
97 err = join_classid();
98 if (!ASSERT_OK(err, "join_cgroup1"))
99 goto cleanup;
100
101 current_cgid = get_classid_cgroup_id();
102 if (!ASSERT_GE(current_cgid, 0, "cgroup1 id"))
103 goto cleanup;
104
105 hid = get_cgroup1_hierarchy_id("net_cls");
106 if (!ASSERT_GE(hid, 0, "cgroup1 id"))
107 goto cleanup;
108 skel->bss->target_hid = hid;
109
110 if (test__start_subtest("test_cgroup1_hierarchy")) {
111 skel->bss->target_ancestor_cgid = current_cgid;
112 bpf_cgroup1(skel);
113 }
114
115 if (test__start_subtest("test_root_cgid")) {
116 skel->bss->target_ancestor_cgid = 1;
117 skel->bss->target_ancestor_level = 0;
118 bpf_cgroup1(skel);
119 }
120
121 if (test__start_subtest("test_invalid_level")) {
122 skel->bss->target_ancestor_cgid = 1;
123 skel->bss->target_ancestor_level = 1;
124 bpf_cgroup1_invalid_id(skel);
125 }
126
127 if (test__start_subtest("test_invalid_cgid")) {
128 skel->bss->target_ancestor_cgid = 0;
129 bpf_cgroup1_invalid_id(skel);
130 }
131
132 if (test__start_subtest("test_invalid_hid")) {
133 skel->bss->target_ancestor_cgid = 1;
134 skel->bss->target_ancestor_level = 0;
135 skel->bss->target_hid = -1;
136 bpf_cgroup1_invalid_id(skel);
137 }
138
139 if (test__start_subtest("test_invalid_cgrp_name")) {
140 skel->bss->target_hid = get_cgroup1_hierarchy_id("net_cl");
141 skel->bss->target_ancestor_cgid = current_cgid;
142 bpf_cgroup1_invalid_id(skel);
143 }
144
145 if (test__start_subtest("test_invalid_cgrp_name2")) {
146 skel->bss->target_hid = get_cgroup1_hierarchy_id("net_cls,");
147 skel->bss->target_ancestor_cgid = current_cgid;
148 bpf_cgroup1_invalid_id(skel);
149 }
150
151 if (test__start_subtest("test_sleepable_prog")) {
152 skel->bss->target_hid = hid;
153 skel->bss->target_ancestor_cgid = current_cgid;
154 bpf_cgroup1_sleepable(skel);
155 }
156
157 cleanup:
158 cleanup_classid_environment();
159 cleanup_cgroup:
160 cleanup_cgroup_environment();
161 destroy:
162 test_cgroup1_hierarchy__destroy(skel);
163 }
164