korg/linux
0
0
Fork 0
mirror of https://mirrors.bfsu.edu.cn/git/linux.git synced 2024-12-02 16:44:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

selftests/bpf: Add kprobe session verifier test for return value

Browse Source 
Making sure kprobe.session program can return only [0,1] values.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
Signed-off-by: Andrii Nakryiko <andrii@kernel.org>
Acked-by: Andrii Nakryiko <andrii@kernel.org>
Link: https://lore.kernel.org/bpf/20241108134544.480660-11-jolsa@kernel.org
This commit is contained in:
Jiri Olsa 2024-11-08 14:45:41 +01:00 committed by Andrii Nakryiko
parent 8c3a48b0d9
commit 504d21d905
2 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/testing/selftests/bpf/prog_tests/kprobe_multi_test.c
View File

@ -6,6 +6,7 @@
#include "kprobe_multi_override.skel.h"
#include "kprobe_multi_session.skel.h"
#include "kprobe_multi_session_cookie.skel.h"
#include "kprobe_multi_verifier.skel.h"
#include "bpf/libbpf_internal.h"
#include "bpf/hashmap.h"
@ -764,4 +765,5 @@ void test_kprobe_multi_test(void)
test_session_skel_api();
if (test__start_subtest("session_cookie"))
test_session_cookie_skel_api();
RUN_TESTS(kprobe_multi_verifier);
}

31
tools/testing/selftests/bpf/progs/kprobe_multi_verifier.c Normal file
View File

@ -0,0 +1,31 @@
// SPDX-License-Identifier: GPL-2.0
#include "vmlinux.h"
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
#include <bpf/usdt.bpf.h>
#include "bpf_misc.h"
char _license[] SEC("license") = "GPL";
SEC("kprobe.session")
__success
int kprobe_session_return_0(struct pt_regs *ctx)
{
return 0;
}
SEC("kprobe.session")
__success
int kprobe_session_return_1(struct pt_regs *ctx)
{
return 1;
}
SEC("kprobe.session")
__failure
__msg("At program exit the register R0 has smin=2 smax=2 should have been in [0, 1]")
int kprobe_session_return_2(struct pt_regs *ctx)
{
return 2;
}