github-mirror/linux
1
0
Fork 0
mirror of https://github.com/torvalds/linux.git synced 2026-09-22 04:34:03 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity Graph

selftests/bpf: Add get_preempt_count() support for RISC-V

Browse Source 
Currently, there is no RISC-V support for get_preempt_count() and
its fallback path always returns 0.

Add it so that bpf_in_interrupt(), bpf_in_nmi(), bpf_in_hardirq(),
bpf_in_serving_softirq(), and bpf_in_task() work for RISC-V as well.

Given that RISC-V has supported CONFIG_THREAD_INFO_IN_TASK since its
initial commit fbe934d69e ("RISC-V: Build Infrastructure") in 2017,
directly retrieve preempt_count from the thread_info embedded within
task_struct via bpf_get_current_task_btf().

This aligns the implementation with arm64, powerpc, and loongarch.

Tested on a RISC-V virtual machine.

Before:
  $ sudo ./test_progs -t exe_ctx
  ...
  #114     exe_ctx:FAIL
  Summary: 0/0 PASSED, 0 SKIPPED, 1 FAILED

After:
  $ sudo ./test_progs -t exe_ctx
  #114     exe_ctx:OK
  Summary: 1/0 PASSED, 0 SKIPPED, 0 FAILED

Signed-off-by: Tiezhu Yang <yangtiezhu@loongson.cn>
Tested-by: Pu Lehui <pulehui@huawei.com>
Reviewed-by: Pu Lehui <pulehui@huawei.com>
Link: https://lore.kernel.org/bpf/20260722022906.8778-1-yangtiezhu@loongson.cn
Signed-off-by: Kumar Kartikeya Dwivedi <memxor@gmail.com>
This commit is contained in:
Tiezhu Yang 2026-07-22 10:29:06 +08:00 committed by Kumar Kartikeya Dwivedi
parent 10416cd727
commit 9b19237a46
No known key found for this signature in database
GPG Key ID: 472D377B63542F83
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/testing/selftests/bpf/bpf_experimental.h
View File

@ -425,6 +425,8 @@ static inline int get_preempt_count(void)
return bpf_get_lowcore()->preempt_count;
#elif defined(bpf_target_loongarch)
return bpf_get_current_task_btf()->thread_info.preempt_count;
#elif defined(bpf_target_riscv)
return bpf_get_current_task_btf()->thread_info.preempt_count;
#endif
return 0;
}
@ -436,6 +438,7 @@ static inline int get_preempt_count(void)
* * powerpc64
* * s390x
* * loongarch
* * riscv
*/
static inline int bpf_in_interrupt(void)
{
@ -458,6 +461,7 @@ static inline int bpf_in_interrupt(void)
* * powerpc64
* * s390x
* * loongarch
* * riscv
*/
static inline int bpf_in_nmi(void)
{
@ -471,6 +475,7 @@ static inline int bpf_in_nmi(void)
* * powerpc64
* * s390x
* * loongarch
* * riscv
*/
static inline int bpf_in_hardirq(void)
{
@ -484,6 +489,7 @@ static inline int bpf_in_hardirq(void)
* * powerpc64
* * s390x
* * loongarch
* * riscv
*/
static inline int bpf_in_serving_softirq(void)
{
@ -505,6 +511,7 @@ static inline int bpf_in_serving_softirq(void)
* * powerpc64
* * s390x
* * loongarch
* * riscv
*/
static inline int bpf_in_task(void)
{