[PATCH bpf-next] bpf: fix build on architectures with special bpf_user_pt_regs_t

[Song Liu]

Architectures like s390, powerpc, arm64, riscv have speical definition of
bpf_user_pt_regs_t. So we need to cast the pointer before passing it to
bpf_get_stack(). This is similar to bpf_get_stack_tp().

Fixes: 03d42fd2d83f ("bpf: Separate bpf_get_[stack|stackid] for perf events 
BPF")
Reported-by: kernel test robot <l...@intel.com>
Signed-off-by: Song Liu <songliubrav...@fb.com>
---
 kernel/bpf/stackmap.c | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/kernel/bpf/stackmap.c b/kernel/bpf/stackmap.c
index 5beb2f8c23da1..4fd830a62be2d 100644
--- a/kernel/bpf/stackmap.c
+++ b/kernel/bpf/stackmap.c
@@ -678,6 +678,7 @@ const struct bpf_func_proto bpf_get_task_stack_proto = {
 BPF_CALL_4(bpf_get_stack_pe, struct bpf_perf_event_data_kern *, ctx,
           void *, buf, u32, size, u64, flags)
 {
+       struct pt_regs *regs = (struct pt_regs *)(ctx->regs);
        struct perf_event *event = ctx->event;
        struct perf_callchain_entry *trace;
        bool kernel, user;
@@ -685,7 +686,7 @@ BPF_CALL_4(bpf_get_stack_pe, struct 
bpf_perf_event_data_kern *, ctx,
        __u64 nr_kernel;
 
        if (!(event->attr.sample_type & __PERF_SAMPLE_CALLCHAIN_EARLY))
-               return __bpf_get_stack(ctx->regs, NULL, NULL, buf, size, flags);
+               return __bpf_get_stack(regs, NULL, NULL, buf, size, flags);
 
        if (unlikely(flags & ~(BPF_F_SKIP_FIELD_MASK | BPF_F_USER_STACK |
                               BPF_F_USER_BUILD_ID)))
@@ -705,8 +706,7 @@ BPF_CALL_4(bpf_get_stack_pe, struct 
bpf_perf_event_data_kern *, ctx,
                __u64 nr = trace->nr;
 
                trace->nr = nr_kernel;
-               err = __bpf_get_stack(ctx->regs, NULL, trace, buf,
-                                     size, flags);
+               err = __bpf_get_stack(regs, NULL, trace, buf, size, flags);
 
                /* restore nr */
                trace->nr = nr;
@@ -718,8 +718,7 @@ BPF_CALL_4(bpf_get_stack_pe, struct 
bpf_perf_event_data_kern *, ctx,
                        goto clear;
 
                flags = (flags & ~BPF_F_SKIP_FIELD_MASK) | skip;
-               err = __bpf_get_stack(ctx->regs, NULL, trace, buf,
-                                     size, flags);
+               err = __bpf_get_stack(regs, NULL, trace, buf, size, flags);
        }
        return err;
 
-- 
2.24.1