Commit a5533a6e authored by Menglong Dong's avatar Menglong Dong Committed by Alexei Starovoitov
Browse files

selftests/bpf: test bpf_get_func_* for fsession 



Test following bpf helper for fsession:
  bpf_get_func_arg()
  bpf_get_func_arg_cnt()
  bpf_get_func_ret()
  bpf_get_func_ip()

Signed-off-by: default avatarMenglong Dong <dongml2@chinatelecom.cn>
Link: https://lore.kernel.org/r/20260124062008.8657-12-dongml2@chinatelecom.cn


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent f7afef56

tools/testing/selftests/bpf/prog_tests/get_func_args_test.c

+1 −0
Original line number Diff line number Diff line
@@ -41,6 +41,7 @@ void test_get_func_args_test(void) 
	ASSERT_EQ(skel->bss->test4_result, 1, "test4_result");

	ASSERT_EQ(skel->bss->test5_result, 1, "test5_result");

	ASSERT_EQ(skel->bss->test6_result, 1, "test6_result");

	ASSERT_EQ(skel->bss->test7_result, 1, "test7_result");



cleanup:

	get_func_args_test__destroy(skel);

tools/testing/selftests/bpf/prog_tests/get_func_ip_test.c

+2 −0
Original line number Diff line number Diff line
@@ -46,6 +46,8 @@ static void test_function_entry(void) 
	ASSERT_EQ(skel->bss->test5_result, 1, "test5_result");

	ASSERT_EQ(skel->bss->test7_result, 1, "test7_result");

	ASSERT_EQ(skel->bss->test8_result, 1, "test8_result");

	ASSERT_EQ(skel->bss->test9_entry_result, 1, "test9_entry_result");

	ASSERT_EQ(skel->bss->test9_exit_result, 1, "test9_exit_result");



cleanup:

	get_func_ip_test__destroy(skel);

tools/testing/selftests/bpf/progs/get_func_args_test.c

+39 −1
Original line number Diff line number Diff line 
// SPDX-License-Identifier: GPL-2.0

#include <linux/bpf.h>

#include <vmlinux.h>

#include <bpf/bpf_helpers.h>

#include <bpf/bpf_tracing.h>

#include <errno.h>
@@ -165,3 +165,41 @@ int BPF_PROG(tp_test2) 


	return 0;

}



__u64 test7_result = 0;

#ifdef __TARGET_ARCH_x86

SEC("fsession/bpf_fentry_test1")

int BPF_PROG(test7)

{

	__u64 cnt = bpf_get_func_arg_cnt(ctx);

	__u64 a = 0, z = 0, ret = 0;

	__s64 err;



	test7_result = cnt == 1;



	/* valid arguments */

	err = bpf_get_func_arg(ctx, 0, &a);

	test7_result &= err == 0 && ((int) a == 1);



	/* not valid argument */

	err = bpf_get_func_arg(ctx, 1, &z);

	test7_result &= err == -EINVAL;



	if (bpf_session_is_return(ctx)) {

		err = bpf_get_func_ret(ctx, &ret);

		test7_result &= err == 0 && ret == 2;

	} else {

		err = bpf_get_func_ret(ctx, &ret);

		test7_result &= err == 0 && ret == 0;

	}



	return 0;

}

#else

SEC("fentry/bpf_fentry_test1")

int BPF_PROG(test7)

{

	test7_result = 1;

	return 0;

}

#endif

tools/testing/selftests/bpf/progs/get_func_ip_test.c

+23 −0
Original line number Diff line number Diff line
@@ -103,3 +103,26 @@ int BPF_URETPROBE(test8, int ret) 
	test8_result = (const void *) addr == (const void *) uprobe_trigger;

	return 0;

}



__u64 test9_entry_result = 0;

__u64 test9_exit_result = 0;

#ifdef __TARGET_ARCH_x86

SEC("fsession/bpf_fentry_test1")

int BPF_PROG(test9, int a)

{

	__u64 addr = bpf_get_func_ip(ctx);



	if (bpf_session_is_return(ctx))

		test9_exit_result = (const void *) addr == &bpf_fentry_test1;

	else

		test9_entry_result = (const void *) addr == &bpf_fentry_test1;

	return 0;

}

#else

SEC("fentry/bpf_fentry_test1")

int BPF_PROG(test9, int a)

{

	test9_entry_result = test9_exit_result = 1;

	return 0;

}

#endif