// SPDX-License-Identifier: GPL-2.0+

#include <linux/kernel.h>
#include <linux/kprobes.h>
#include <linux/module.h>
#include <linux/kallsyms.h>
#include <asm/sections.h>

#include "decode-insn.h"
#include "simulate-insn.h"

/* Return:
 *   INSN_REJECTED     If instruction is one not allowed to kprobe,
 *   INSN_GOOD_NO_SLOT If instruction is supported but doesn't use its slot.
 */
enum probe_insn __kprobes
riscv_probe_decode_insn(probe_opcode_t *addr, struct arch_probe_insn *api)
{
        probe_opcode_t insn = *addr;

        /*
         * Reject instructions list:
         */
        RISCV_INSN_REJECTED(system,             insn);
        RISCV_INSN_REJECTED(fence,              insn);

        /*
         * Simulate instructions list:
         * TODO: the REJECTED ones below need to be implemented
         */
#ifdef CONFIG_RISCV_ISA_C
        RISCV_INSN_REJECTED(c_jal,              insn);
        RISCV_INSN_REJECTED(c_ebreak,           insn);

        RISCV_INSN_SET_SIMULATE(c_j,            insn);
        RISCV_INSN_SET_SIMULATE(c_jr,           insn);
        RISCV_INSN_SET_SIMULATE(c_jalr,         insn);
        RISCV_INSN_SET_SIMULATE(c_beqz,         insn);
        RISCV_INSN_SET_SIMULATE(c_bnez,         insn);
#endif

        RISCV_INSN_SET_SIMULATE(jal,            insn);
        RISCV_INSN_SET_SIMULATE(jalr,           insn);
        RISCV_INSN_SET_SIMULATE(auipc,          insn);
        RISCV_INSN_SET_SIMULATE(branch,         insn);

        return INSN_GOOD;
}