ab-riscv-interpreter 0.0.4

Composable and generic RISC-V interpreter
Documentation
//! RV64 Zca extension

#[cfg(test)]
mod tests;

use crate::{
    ExecutableInstruction, ExecutionError, ProgramCounter, RegisterFile, SystemInstructionHandler,
    VirtualMemory,
};
use ab_riscv_macros::instruction_execution;
use ab_riscv_primitives::prelude::*;
use core::ops::ControlFlow;

#[instruction_execution]
impl<Reg, Regs, ExtState, Memory, PC, InstructionHandler, CustomError>
    ExecutableInstruction<Regs, ExtState, Memory, PC, InstructionHandler, CustomError>
    for Rv64ZcaInstruction<Reg>
where
    Reg: Register<Type = u64>,
    Regs: RegisterFile<Reg>,
    Memory: VirtualMemory,
    PC: ProgramCounter<Reg::Type, Memory, CustomError>,
    InstructionHandler: SystemInstructionHandler<Reg, Regs, Memory, PC, CustomError>,
{
    #[inline(always)]
    fn execute(
        self,
        regs: &mut Regs,
        _ext_state: &mut ExtState,
        memory: &mut Memory,
        program_counter: &mut PC,
        system_instruction_handler: &mut InstructionHandler,
    ) -> Result<ControlFlow<()>, ExecutionError<Reg::Type, CustomError>> {
        match self {
            // Quadrant 00
            Self::CAddi4spn { rd, nzuimm } => {
                let sp_val = regs.read(Reg::SP);
                regs.write(rd, sp_val.wrapping_add(u64::from(nzuimm)));
            }
            Self::CLw { rd, rs1, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                let value = i64::from(memory.read::<i32>(addr)?);
                regs.write(rd, value.cast_unsigned());
            }
            Self::CLd { rd, rs1, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                let value = memory.read::<u64>(addr)?;
                regs.write(rd, value);
            }
            Self::CSw { rs1, rs2, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2) as u32)?;
            }
            Self::CSd { rs1, rs2, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2))?;
            }

            // Quadrant 01
            Self::CNop => {}
            Self::CAddi { rd, nzimm } => {
                let value = regs.read(rd).wrapping_add(i64::from(nzimm).cast_unsigned());
                regs.write(rd, value);
            }
            Self::CAddiw { rd, imm } => {
                let sum = (regs.read(rd) as i32).wrapping_add(i32::from(imm));
                regs.write(rd, i64::from(sum).cast_unsigned());
            }
            Self::CLi { rd, imm } => {
                regs.write(rd, i64::from(imm).cast_unsigned());
            }
            Self::CAddi16sp { nzimm } => {
                let value = regs
                    .read(Reg::SP)
                    .wrapping_add(i64::from(nzimm).cast_unsigned());
                regs.write(Reg::SP, value);
            }
            Self::CLui { rd, nzimm } => {
                regs.write(rd, i64::from(nzimm).cast_unsigned());
            }
            Self::CSrli { rd, shamt } => {
                let value = regs.read(rd) >> shamt;
                regs.write(rd, value);
            }
            Self::CSrai { rd, shamt } => {
                let value = regs.read(rd).cast_signed() >> shamt;
                regs.write(rd, value.cast_unsigned());
            }
            Self::CAndi { rd, imm } => {
                let value = regs.read(rd) & i64::from(imm).cast_unsigned();
                regs.write(rd, value);
            }
            Self::CSub { rd, rs2 } => {
                let value = regs.read(rd).wrapping_sub(regs.read(rs2));
                regs.write(rd, value);
            }
            Self::CXor { rd, rs2 } => {
                let value = regs.read(rd) ^ regs.read(rs2);
                regs.write(rd, value);
            }
            Self::COr { rd, rs2 } => {
                let value = regs.read(rd) | regs.read(rs2);
                regs.write(rd, value);
            }
            Self::CAnd { rd, rs2 } => {
                let value = regs.read(rd) & regs.read(rs2);
                regs.write(rd, value);
            }
            Self::CSubw { rd, rs2 } => {
                let diff = (regs.read(rd) as i32).wrapping_sub(regs.read(rs2) as i32);
                regs.write(rd, i64::from(diff).cast_unsigned());
            }
            Self::CAddw { rd, rs2 } => {
                let sum = (regs.read(rd) as i32).wrapping_add(regs.read(rs2) as i32);
                regs.write(rd, i64::from(sum).cast_unsigned());
            }
            Self::CJ { imm } => {
                let old_pc = program_counter.old_pc(size_of::<u16>() as u8);
                return program_counter
                    .set_pc(memory, old_pc.wrapping_add(i64::from(imm).cast_unsigned()))
                    .map_err(ExecutionError::from);
            }
            Self::CBeqz { rs1, imm } => {
                if regs.read(rs1) == 0 {
                    let old_pc = program_counter.old_pc(size_of::<u16>() as u8);
                    return program_counter
                        .set_pc(memory, old_pc.wrapping_add(i64::from(imm).cast_unsigned()))
                        .map_err(ExecutionError::from);
                }
            }
            Self::CBnez { rs1, imm } => {
                if regs.read(rs1) != 0 {
                    let old_pc = program_counter.old_pc(size_of::<u16>() as u8);
                    return program_counter
                        .set_pc(memory, old_pc.wrapping_add(i64::from(imm).cast_unsigned()))
                        .map_err(ExecutionError::from);
                }
            }

            // Quadrant 10
            Self::CSlli { rd, shamt } => {
                let value = regs.read(rd) << shamt;
                regs.write(rd, value);
            }
            Self::CLwsp { rd, uimm } => {
                let addr = regs.read(Reg::SP).wrapping_add(u64::from(uimm));
                let value = i64::from(memory.read::<i32>(addr)?);
                regs.write(rd, value.cast_unsigned());
            }
            Self::CLdsp { rd, uimm } => {
                let addr = regs.read(Reg::SP).wrapping_add(u64::from(uimm));
                let value = memory.read::<u64>(addr)?;
                regs.write(rd, value);
            }
            Self::CJr { rs1 } => {
                let target = regs.read(rs1) & !1;
                return program_counter
                    .set_pc(memory, target)
                    .map_err(ExecutionError::from);
            }
            Self::CMv { rd, rs2 } => {
                regs.write(rd, regs.read(rs2));
            }
            Self::CEbreak => {
                system_instruction_handler.handle_ebreak(regs, memory, program_counter.get_pc());
            }
            Self::CJalr { rs1 } => {
                let target = regs.read(rs1) & !1;
                let return_addr = program_counter.get_pc();
                regs.write(Reg::RA, return_addr);
                return program_counter
                    .set_pc(memory, target)
                    .map_err(ExecutionError::from);
            }
            Self::CAdd { rd, rs2 } => {
                let value = regs.read(rd).wrapping_add(regs.read(rs2));
                regs.write(rd, value);
            }
            Self::CSwsp { rs2, uimm } => {
                let addr = regs.read(Reg::SP).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2) as u32)?;
            }
            Self::CSdsp { rs2, uimm } => {
                let addr = regs.read(Reg::SP).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2))?;
            }
            Self::CUnimp => {
                let old_pc = program_counter.old_pc(size_of::<u16>() as u8);
                return Err(ExecutionError::IllegalInstruction { address: old_pc });
            }
        }

        Ok(ControlFlow::Continue(()))
    }
}