diff --git a/include/sbi/riscv_encoding.h b/include/sbi/riscv_encoding.h index 84d3c17b..64f0249b 100644 --- a/include/sbi/riscv_encoding.h +++ b/include/sbi/riscv_encoding.h @@ -86,6 +86,8 @@ #define HSTATUS_GVA _UL(0x00000040) #define HSTATUS_VSBE _UL(0x00000020) +#define MTVEC_MODE _UL(0x00000003) + #define MCAUSE_IRQ_MASK (_UL(1) << (__riscv_xlen - 1)) #define IRQ_S_SOFT 1 diff --git a/lib/sbi/sbi_trap.c b/lib/sbi/sbi_trap.c index e63a563b..f41db4d1 100644 --- a/lib/sbi/sbi_trap.c +++ b/lib/sbi/sbi_trap.c @@ -169,7 +169,7 @@ int sbi_trap_redirect(struct sbi_trap_regs *regs, csr_write(CSR_VSCAUSE, trap->cause); /* Set MEPC to VS-mode exception vector base */ - regs->mepc = csr_read(CSR_VSTVEC); + regs->mepc = csr_read(CSR_VSTVEC) & ~MTVEC_MODE; /* Set MPP to VS-mode */ regs->mstatus &= ~MSTATUS_MPP; @@ -204,7 +204,7 @@ int sbi_trap_redirect(struct sbi_trap_regs *regs, csr_write(CSR_SCAUSE, trap->cause); /* Set MEPC to S-mode exception vector base */ - regs->mepc = csr_read(CSR_STVEC); + regs->mepc = csr_read(CSR_STVEC) & ~MTVEC_MODE; /* Set MPP to S-mode */ regs->mstatus &= ~MSTATUS_MPP;