ab-riscv-interpreter 0.0.4

Composable and generic RISC-V interpreter
Documentation
//! Zcb compressed instruction execution

#[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 Rv64ZcbInstruction<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 {
            Self::CLbu { rd, rs1, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                let value = memory.read::<u8>(addr)?;
                regs.write(rd, u64::from(value));
            }
            Self::CLh { rd, rs1, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                let value = i64::from(memory.read::<i16>(addr)?);
                regs.write(rd, value.cast_unsigned());
            }
            Self::CLhu { rd, rs1, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                let value = memory.read::<u16>(addr)?;
                regs.write(rd, u64::from(value));
            }
            Self::CSb { rs1, rs2, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2) as u8)?;
            }
            Self::CSh { rs1, rs2, uimm } => {
                let addr = regs.read(rs1).wrapping_add(u64::from(uimm));
                memory.write(addr, regs.read(rs2) as u16)?;
            }
            Self::CZextB { rd } => {
                let value = regs.read(rd) & 0xff;
                regs.write(rd, value);
            }
            Self::CSextB { rd } => {
                let value = i64::from(regs.read(rd) as i8);
                regs.write(rd, value.cast_unsigned());
            }
            Self::CZextH { rd } => {
                let value = regs.read(rd) & 0xffff;
                regs.write(rd, value);
            }
            Self::CSextH { rd } => {
                let value = i64::from(regs.read(rd) as i16);
                regs.write(rd, value.cast_unsigned());
            }
            Self::CZextW { rd } => {
                let value = regs.read(rd) & 0xffff_ffff;
                regs.write(rd, value);
            }
            Self::CNot { rd } => {
                let value = !regs.read(rd);
                regs.write(rd, value);
            }
            Self::CMul { rd, rs2 } => {
                let value = regs.read(rd).wrapping_mul(regs.read(rs2));
                regs.write(rd, value);
            }
        }

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