Adding selftest to check attaching rules for multi func programs.
- attach 2 programs:
fentry/bpf_fentry_test1
fexit/bpf_fentry_test2
- check that we can attach multi func program on top of them:
fentry.multi/bpf_fentry_test*
- check that we cannot attach another multi funct program
that does not cover the same BTF ids (one less):
fentry.multi/bpf_fentry_test[1-7]
fexit.multi/bpf_fentry_test[1-7]
- check that we can no longer attach standard trampoline
programs (below) on top of attached multi func program:
fentry/bpf_fentry_test1
fexit/bpf_fentry_test3
Because the supported wildcards do not allow us to
match just limited set of bpf_fentry_test*, adding
extra code to look it up in kernel's BTF.
Signed-off-by: Jiri Olsa <jolsa@xxxxxxxxxx>
---
.../bpf/prog_tests/multi_attach_check_test.c | 115 ++++++++++++++++++
.../selftests/bpf/progs/multi_attach_check.c | 36 ++++++
.../bpf/progs/multi_attach_check_extra1.c | 12 ++
.../bpf/progs/multi_attach_check_extra2.c | 12 ++
4 files changed, 175 insertions(+)
create mode 100644 tools/testing/selftests/bpf/prog_tests/multi_attach_check_test.c
create mode 100644 tools/testing/selftests/bpf/progs/multi_attach_check.c
create mode 100644 tools/testing/selftests/bpf/progs/multi_attach_check_extra1.c
create mode 100644 tools/testing/selftests/bpf/progs/multi_attach_check_extra2.c
diff --git a/tools/testing/selftests/bpf/prog_tests/multi_attach_check_test.c b/tools/testing/selftests/bpf/prog_tests/multi_attach_check_test.c
new file mode 100644
index 000000000000..32b23718437d
--- /dev/null
+++ b/tools/testing/selftests/bpf/prog_tests/multi_attach_check_test.c
@@ -0,0 +1,115 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <test_progs.h>
+#include <linux/btf_ids.h>
+#include "multi_attach_check.skel.h"
+#include "multi_attach_check_extra1.skel.h"
+#include "multi_attach_check_extra2.skel.h"
+#include <bpf/btf.h>
+
+static __u32 btf_ids[7];
+
+static int load_btf_ids(void)
+{
+ __u32 i, nr_types, cnt;
+ struct btf *btf;
+
+ btf = btf__load_vmlinux_btf();
+ if (!ASSERT_OK_PTR(btf, "btf__load_vmlinux_btf"))
+ return -1;
+
+ nr_types = btf__get_nr_types(btf);
+
+ for (i = 1, cnt = 0; i <= nr_types && cnt < 7; i++) {
+ const struct btf_type *t = btf__type_by_id(btf, i);
+ const char *name;
+
+ if (!btf_is_func(t))
+ continue;
+
+ name = btf__name_by_offset(btf, t->name_off);
+ if (!name)
+ continue;
+ if (strncmp(name, "bpf_fentry_test", sizeof("bpf_fentry_test") - 1))
+ continue;
+
+ btf_ids[cnt] = i;
+ cnt++;
+ }
+
+ btf__free(btf);
+ return ASSERT_EQ(cnt, 7, "bpf_fentry_test_cnt") ? 0 : -1;
+}
+
+void test_multi_attach_check_test(void)
+{
+ struct bpf_link *link1 = NULL, *link2 = NULL, *link3 = NULL;
+ DECLARE_LIBBPF_OPTS(bpf_link_create_opts, opts);
+ struct multi_attach_check_extra1 *skel_extra1 = NULL;
+ struct multi_attach_check_extra2 *skel_extra2 = NULL;
+ struct multi_attach_check *skel;
+ int link_fd, prog_fd;
+
+ /* Load/attach standard trampolines and on top of it multi
+ * func program. It should succeed.
+ */
+ skel = multi_attach_check__open_and_load();
+ if (!ASSERT_OK_PTR(skel, "multi_attach_check__load"))
+ return;
+
+ link1 = bpf_program__attach(skel->progs.test1);
+ if (!ASSERT_OK_PTR(link1, "multi_attach_check__test1_attach"))
+ goto cleanup;
+
+ link2 = bpf_program__attach(skel->progs.test2);
+ if (!ASSERT_OK_PTR(link2, "multi_attach_check__test2_attach"))
+ goto cleanup;
+
+ link3 = bpf_program__attach(skel->progs.test3);
+ if (!ASSERT_OK_PTR(link3, "multi_attach_check__test3_attach"))
+ goto cleanup;
+
+ if (!ASSERT_OK(load_btf_ids(), "load_btf_ids"))
+ goto cleanup;
+
+ /* There's 8 bpf_fentry_test* functions, get BTF ids for 7 of them
+ * and try to load/link multi func program with them. It should fail
+ * both for fentry.multi ...
+ */
+ opts.multi.btf_ids = btf_ids;
+ opts.multi.btf_ids_cnt = 7;
+
+ prog_fd = bpf_program__fd(skel->progs.test4);
+
+ link_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_FENTRY, &opts);
+ if (!ASSERT_LT(link_fd, 0, "bpf_link_create"))
+ goto cleanup;
+
+ close(link_fd);
+
+ /* ... and fexit.multi */
+ prog_fd = bpf_program__fd(skel->progs.test5);
+
+ link_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_FEXIT, &opts);
+ if (!ASSERT_LT(link_fd, 0, "bpf_link_create"))
+ goto cleanup;
+
+ close(link_fd);
+
+ /* Try to load/attach extra programs on top of multi func programs,
+ * it should fail for both fentry ...
+ */
+ skel_extra1 = multi_attach_check_extra1__open_and_load();
+ if (!ASSERT_ERR_PTR(skel_extra1, "multi_attach_check_extra1__load"))
+ multi_attach_check_extra1__destroy(skel_extra1);
+
+ /* ... and fexit */
+ skel_extra2 = multi_attach_check_extra2__open_and_load();
+ if (!ASSERT_ERR_PTR(skel_extra2, "multi_attach_check_extra2__load"))
+ multi_attach_check_extra2__destroy(skel_extra2);
+
+cleanup:
+ bpf_link__destroy(link1);
+ bpf_link__destroy(link2);
+ bpf_link__destroy(link3);
+ multi_attach_check__destroy(skel);
+}
diff --git a/tools/testing/selftests/bpf/progs/multi_attach_check.c b/tools/testing/selftests/bpf/progs/multi_attach_check.c
new file mode 100644
index 000000000000..cf0f25c69556
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/multi_attach_check.c
@@ -0,0 +1,36 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <linux/bpf.h>
+#include <bpf/bpf_helpers.h>
+#include <bpf/bpf_tracing.h>
+
+char _license[] SEC("license") = "GPL";
+
+SEC("fentry/bpf_fentry_test1")
+int BPF_PROG(test1, int a)
+{
+ return 0;
+}
+
+SEC("fexit/bpf_fentry_test2")
+int BPF_PROG(test2, int a, __u64 b, int ret)
+{
+ return 0;
+}
+
+SEC("fentry.multi/bpf_fentry_test*")
+int BPF_PROG(test3, __u64 a, __u64 b, __u64 c, __u64 d, __u64 e, __u64 f)
+{
+ return 0;
+}
+
+SEC("fentry.multi/bpf_fentry_test1-7")
+int BPF_PROG(test4, __u64 a, __u64 b, __u64 c, __u64 d, __u64 e, __u64 f, int ret)
+{
+ return 0;
+}
+
+SEC("fexit.multi/bpf_fentry_test1-7")
+int BPF_PROG(test5, __u64 a, __u64 b, __u64 c, __u64 d, __u64 e, __u64 f, int ret)
+{
+ return 0;
+}
diff --git a/tools/testing/selftests/bpf/progs/multi_attach_check_extra1.c b/tools/testing/selftests/bpf/progs/multi_attach_check_extra1.c
new file mode 100644
index 000000000000..c1d816a0206a
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/multi_attach_check_extra1.c
@@ -0,0 +1,12 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <linux/bpf.h>
+#include <bpf/bpf_helpers.h>
+#include <bpf/bpf_tracing.h>
+
+char _license[] SEC("license") = "GPL";
+
+SEC("fentry/bpf_fentry_test1")
+int BPF_PROG(test1, int a)
+{
+ return 0;
+}
diff --git a/tools/testing/selftests/bpf/progs/multi_attach_check_extra2.c b/tools/testing/selftests/bpf/progs/multi_attach_check_extra2.c
new file mode 100644
index 000000000000..cd66abb4b848
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/multi_attach_check_extra2.c
@@ -0,0 +1,12 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <linux/bpf.h>
+#include <bpf/bpf_helpers.h>
+#include <bpf/bpf_tracing.h>
+
+char _license[] SEC("license") = "GPL";
+
+SEC("fexit/bpf_fentry_test3")
+int BPF_PROG(test3, int a, __u64 b, int ret)
+{
+ return 0;
+}
--
2.31.1
