1f2644fb4SAndrii Nakryiko // SPDX-License-Identifier: GPL-2.0
2f2644fb4SAndrii Nakryiko /* Copyright (c) 2021 Facebook */
3f2644fb4SAndrii Nakryiko
4f2644fb4SAndrii Nakryiko #include <test_progs.h>
5f2644fb4SAndrii Nakryiko #include <sys/syscall.h>
6f2644fb4SAndrii Nakryiko #include "linked_funcs.skel.h"
7f2644fb4SAndrii Nakryiko
test_linked_funcs(void)8f2644fb4SAndrii Nakryiko void test_linked_funcs(void)
9f2644fb4SAndrii Nakryiko {
10f2644fb4SAndrii Nakryiko int err;
11f2644fb4SAndrii Nakryiko struct linked_funcs *skel;
12f2644fb4SAndrii Nakryiko
13f2644fb4SAndrii Nakryiko skel = linked_funcs__open();
14f2644fb4SAndrii Nakryiko if (!ASSERT_OK_PTR(skel, "skel_open"))
15f2644fb4SAndrii Nakryiko return;
16f2644fb4SAndrii Nakryiko
17*b82bb1ffSAndrii Nakryiko /* handler1 and handler2 are marked as SEC("?raw_tp/sys_enter") and
18*b82bb1ffSAndrii Nakryiko * are set to not autoload by default
19*b82bb1ffSAndrii Nakryiko */
20*b82bb1ffSAndrii Nakryiko bpf_program__set_autoload(skel->progs.handler1, true);
21*b82bb1ffSAndrii Nakryiko bpf_program__set_autoload(skel->progs.handler2, true);
22*b82bb1ffSAndrii Nakryiko
23f2644fb4SAndrii Nakryiko skel->rodata->my_tid = syscall(SYS_gettid);
24f2644fb4SAndrii Nakryiko skel->bss->syscall_id = SYS_getpgid;
25f2644fb4SAndrii Nakryiko
26f2644fb4SAndrii Nakryiko err = linked_funcs__load(skel);
27f2644fb4SAndrii Nakryiko if (!ASSERT_OK(err, "skel_load"))
28f2644fb4SAndrii Nakryiko goto cleanup;
29f2644fb4SAndrii Nakryiko
30f2644fb4SAndrii Nakryiko err = linked_funcs__attach(skel);
31f2644fb4SAndrii Nakryiko if (!ASSERT_OK(err, "skel_attach"))
32f2644fb4SAndrii Nakryiko goto cleanup;
33f2644fb4SAndrii Nakryiko
34f2644fb4SAndrii Nakryiko /* trigger */
35f2644fb4SAndrii Nakryiko syscall(SYS_getpgid);
36f2644fb4SAndrii Nakryiko
37f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_val1, 2000 + 2000, "output_val1");
38f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_ctx1, SYS_getpgid, "output_ctx1");
39f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_weak1, 42, "output_weak1");
40f2644fb4SAndrii Nakryiko
41f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_val2, 2 * 1000 + 2 * (2 * 1000), "output_val2");
42f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_ctx2, SYS_getpgid, "output_ctx2");
43f2644fb4SAndrii Nakryiko /* output_weak2 should never be updated */
44f2644fb4SAndrii Nakryiko ASSERT_EQ(skel->bss->output_weak2, 0, "output_weak2");
45f2644fb4SAndrii Nakryiko
46f2644fb4SAndrii Nakryiko cleanup:
47f2644fb4SAndrii Nakryiko linked_funcs__destroy(skel);
48f2644fb4SAndrii Nakryiko }
49