// SPDX-License-Identifier: GPL-2.0
// Copyright (C) 2019 Hangzhou C-SKY Microsystems co.,ltd.

#include <linux/perf_event.h>
#include <linux/uaccess.h>

/* Kernel callchain */
struct stackframe {
        unsigned long fp;
        unsigned long lr;
};

static int unwind_frame_kernel(struct stackframe *frame)
{
        unsigned long low = (unsigned long)task_stack_page(current);
        unsigned long high = low + THREAD_SIZE;

        if (unlikely(frame->fp < low || frame->fp > high))
                return -EPERM;

        if (kstack_end((void *)frame->fp) || frame->fp & 0x3)
                return -EPERM;

        *frame = *(struct stackframe *)frame->fp;

        if (__kernel_text_address(frame->lr)) {
                int graph = 0;

                frame->lr = ftrace_graph_ret_addr(NULL, &graph, frame->lr,
                                NULL);
        }
        return 0;
}

static void notrace walk_stackframe(struct stackframe *fr,
                        struct perf_callchain_entry_ctx *entry)
{
        do {
                perf_callchain_store(entry, fr->lr);
        } while (unwind_frame_kernel(fr) >= 0);
}

/*
 * Get the return address for a single stackframe and return a pointer to the
 * next frame tail.
 */
static unsigned long user_backtrace(struct perf_callchain_entry_ctx *entry,
                        unsigned long fp, unsigned long reg_lr)
{
        struct stackframe buftail;
        unsigned long lr = 0;
        unsigned long __user *user_frame_tail = (unsigned long __user *)fp;

        /* Check accessibility of one struct frame_tail beyond */
        if (!access_ok(user_frame_tail, sizeof(buftail)))
                return 0;
        if (__copy_from_user_inatomic(&buftail, user_frame_tail,
                                      sizeof(buftail)))
                return 0;

        if (reg_lr != 0)
                lr = reg_lr;
        else
                lr = buftail.lr;

        fp = buftail.fp;
        perf_callchain_store(entry, lr);

        return fp;
}

/*
 * This will be called when the target is in user mode
 * This function will only be called when we use
 * "PERF_SAMPLE_CALLCHAIN" in
 * kernel/events/core.c:perf_prepare_sample()
 *
 * How to trigger perf_callchain_[user/kernel] :
 * $ perf record -e cpu-clock --call-graph fp ./program
 * $ perf report --call-graph
 *
 * On C-SKY platform, the program being sampled and the C library
 * need to be compiled with * -mbacktrace, otherwise the user
 * stack will not contain function frame.
 */
void perf_callchain_user(struct perf_callchain_entry_ctx *entry,
                         struct pt_regs *regs)
{
        unsigned long fp = 0;

        fp = regs->regs[4];
        perf_callchain_store(entry, regs->pc);

        /*
         * While backtrace from leaf function, lr is normally
         * not saved inside frame on C-SKY, so get lr from pt_regs
         * at the sample point. However, lr value can be incorrect if
         * lr is used as temp register
         */
        fp = user_backtrace(entry, fp, regs->lr);

        while (fp && !(fp & 0x3) && entry->nr < entry->max_stack)
                fp = user_backtrace(entry, fp, 0);
}

void perf_callchain_kernel(struct perf_callchain_entry_ctx *entry,
                           struct pt_regs *regs)
{
        struct stackframe fr;

        fr.fp = regs->regs[4];
        fr.lr = regs->lr;
        walk_stackframe(&fr, entry);
}