// SPDX-License-Identifier: GPL-2.0
#include <linux/kernel.h>
#include <linux/stdarg.h>
#include <linux/string.h>
#include <linux/ctype.h>
#include <asm/stacktrace.h>
#include <asm/boot_data.h>
#include <asm/lowcore.h>
#include <asm/setup.h>
#include <asm/sclp.h>
#include <asm/uv.h>
#include "boot.h"

void print_stacktrace(unsigned long sp)
{
        struct stack_info boot_stack = { STACK_TYPE_TASK, (unsigned long)_stack_start,
                                         (unsigned long)_stack_end };
        bool first = true;

        boot_emerg("Call Trace:\n");
        while (!(sp & 0x7) && on_stack(&boot_stack, sp, sizeof(struct stack_frame))) {
                struct stack_frame *sf = (struct stack_frame *)sp;

                if (first)
                        boot_emerg("(sp:%016lx [<%016lx>] %pS)\n", sp, sf->gprs[8], (void *)sf->gprs[8]);
                else
                        boot_emerg(" sp:%016lx [<%016lx>] %pS\n", sp, sf->gprs[8], (void *)sf->gprs[8]);
                if (sf->back_chain <= sp)
                        break;
                sp = sf->back_chain;
                first = false;
        }
}

extern struct exception_table_entry __start___ex_table[];
extern struct exception_table_entry __stop___ex_table[];

static inline unsigned long extable_insn(const struct exception_table_entry *x)
{
        return (unsigned long)&x->insn + x->insn;
}

static bool ex_handler(struct pt_regs *regs)
{
        const struct exception_table_entry *ex;

        for (ex = __start___ex_table; ex < __stop___ex_table; ex++) {
                if (extable_insn(ex) != regs->psw.addr)
                        continue;
                if (ex->type != EX_TYPE_FIXUP)
                        return false;
                regs->psw.addr = extable_fixup(ex);
                return true;
        }
        return false;
}

void do_pgm_check(struct pt_regs *regs)
{
        struct psw_bits *psw = &psw_bits(regs->psw);
        unsigned long *gpregs = regs->gprs;

        if (ex_handler(regs))
                return;
        if (bootdebug)
                boot_rb_dump();
        boot_emerg("Linux version %s\n", kernel_version);
        if (!is_prot_virt_guest() && early_command_line[0])
                boot_emerg("Kernel command line: %s\n", early_command_line);
        boot_emerg("Kernel fault: interruption code %04x ilc:%d\n",
                   regs->int_code & 0xffff, regs->int_code >> 17);
        if (kaslr_enabled()) {
                boot_emerg("Kernel random base: %lx\n", __kaslr_offset);
                boot_emerg("Kernel random base phys: %lx\n", __kaslr_offset_phys);
        }
        boot_emerg("PSW : %016lx %016lx (%pS)\n",
                   regs->psw.mask, regs->psw.addr, (void *)regs->psw.addr);
        boot_emerg("      R:%x T:%x IO:%x EX:%x Key:%x M:%x W:%x P:%x AS:%x CC:%x PM:%x RI:%x EA:%x\n",
                   psw->per, psw->dat, psw->io, psw->ext, psw->key, psw->mcheck,
                   psw->wait, psw->pstate, psw->as, psw->cc, psw->pm, psw->ri, psw->eaba);
        boot_emerg("GPRS: %016lx %016lx %016lx %016lx\n", gpregs[0], gpregs[1], gpregs[2], gpregs[3]);
        boot_emerg("      %016lx %016lx %016lx %016lx\n", gpregs[4], gpregs[5], gpregs[6], gpregs[7]);
        boot_emerg("      %016lx %016lx %016lx %016lx\n", gpregs[8], gpregs[9], gpregs[10], gpregs[11]);
        boot_emerg("      %016lx %016lx %016lx %016lx\n", gpregs[12], gpregs[13], gpregs[14], gpregs[15]);
        print_stacktrace(gpregs[15]);
        boot_emerg("Last Breaking-Event-Address:\n");
        boot_emerg(" [<%016lx>] %pS\n", regs->last_break, (void *)regs->last_break);
        /* Convert to disabled wait PSW */
        psw->io = 0;
        psw->ext = 0;
        psw->wait = 1;
}