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