ft(interpreter): generalize binary operations

This commit is contained in:
2025-06-10 10:59:35 +09:00
parent 2b37884a60
commit 35fefb7625
3 changed files with 110 additions and 27 deletions

View File

@@ -27,30 +27,52 @@ impl Computer {
}
}
/// Adds two [`Argument`]s, saves it to `dest` and sets flags, if necessary.
pub fn add(&mut self, dest: ModRmTarget, src: Operand) {
let op = |lhs, rhs| lhs + rhs;
// TODO let flag_check = |_| ();
self.op(op, dest, src);
}
pub fn sub(&mut self, dest: ModRmTarget, src: Operand) {
let op = |lhs, rhs| lhs - rhs;
// TODO let flag_check = |_| ();
self.op(op, dest, src);
}
/// Applies a binary operator [`O`] to two [`Argument`]s, saves it to `dest`
/// and sets flags, if necessary.
/// A value can never be saved to an immediate Operand, so `dest` can only
/// be a [`ModRmTarget`].
pub fn add(&mut self, dest: ModRmTarget, src: Operand) {
fn op<O>(&mut self, op: O, dest: ModRmTarget, src: Operand)
where
O: Fn(ImmediateOperand, ImmediateOperand) -> ImmediateOperand,
// F: Fn(&mut Flags);
{
match src {
Operand::ModRmTarget(src_target) => self.add_modrm(dest, src_target),
Operand::Immediate(src_operand) => self.add_imm(dest, src_operand),
Operand::ModRmTarget(src_target) => self.op_modrm(op, dest, src_target),
Operand::Immediate(src_operand) => self.op_imm(op, dest, src_operand),
}
}
/// Adds two [`ModRmTarget`]s togeher, saves it to `dest` and sets flags, if necessary.
fn add_modrm(&mut self, dest: ModRmTarget, src: ModRmTarget) {
/// Applies a binary operator [`O`] to two [`ModRmTargets`]s, saves it to
/// dest and sets flags, if necessary.
fn op_modrm<O>(&mut self, op: O, dest: ModRmTarget, src: ModRmTarget)
where
O: Fn(ImmediateOperand, ImmediateOperand) -> ImmediateOperand,
{
match dest {
ModRmTarget::Memory(dest_memory_idx) => match src {
// mem, mem
ModRmTarget::Memory(_) => panic!("Cannot add Memory to Memory!"),
ModRmTarget::Memory(_) => {
panic!("Cannot apply binary operator from Memory to Memory!")
}
// mem, reg
ModRmTarget::Register(src_register) => {
let lhs = self.memory.read(&self.regs, dest_memory_idx);
let rhs = self.regs.read(src_register);
let result = lhs + rhs;
let result = op(lhs, rhs);
self.memory.write(&self.regs, dest_memory_idx, result);
// unsigned overflow
self.flags.of = result < rhs;
// flag_check(&mut self.flags);
}
},
ModRmTarget::Register(dest_register) => match src {
@@ -58,42 +80,42 @@ impl Computer {
ModRmTarget::Memory(src_memory_index) => {
let lhs = self.regs.read(dest_register);
let rhs = self.memory.read(&self.regs, src_memory_index);
let result = lhs + rhs;
let result = op(lhs, rhs);
self.regs.write(dest_register, result);
// unsigned overflow
self.flags.of = result < rhs;
// flag_check(&mut self.flags);
}
// reg, reg
ModRmTarget::Register(src_register) => {
let lhs = self.regs.read(dest_register);
let rhs = self.regs.read(src_register);
let result = lhs + rhs;
let result = op(lhs, rhs);
self.regs.write(dest_register, result);
// unsigned overflow
self.flags.of = result < rhs
// flag_check(&mut self.flags);
}
},
}
}
/// Adds an [`ImmediateOperand`] to a location pointed to by a
/// [`ModRmTarget`], saves it to dest and sets flags, if necessary.
fn add_imm(&mut self, dest: ModRmTarget, src: ImmediateOperand) {
/// Applies a binary operator [`O`] to [`ImmediateOperand`] and a location
/// pointed to by a [`ModRmTarget`], saves it to dest and sets flags,
/// if necessary.
fn op_imm<O>(&mut self, op: O, dest: ModRmTarget, src: ImmediateOperand)
where
O: Fn(ImmediateOperand, ImmediateOperand) -> ImmediateOperand,
// F: for<'F> Fn(&'F mut Flags),
{
match dest {
ModRmTarget::Memory(dest_mem) => {
let lhs = self.memory.read(&self.regs, dest_mem);
let result = lhs + src;
// unsigned overflow
self.flags.of = result < src;
let result = op(lhs, src);
self.memory.write(&self.regs, dest_mem, result);
// flag_check(&mut self.flags);
}
ModRmTarget::Register(dest_reg) => {
let lhs = self.regs.read(dest_reg);
let result = lhs + src;
// unsigned overflow
self.flags.of = result < src;
let result = op(lhs, src);
self.regs.write(dest_reg, result);
// flag_check(&mut self.flags);
}
}
}

View File

@@ -78,6 +78,29 @@ impl Interpreter {
ModRmTarget::Register(crate::register::Register::AX),
Operand::Immediate(ImmediateOperand::Word(src_word)),
),
/*
* SUB
*/
Mnemonic::SUB_FromReg(dest, src) => self
.computer
.sub(dest, Operand::ModRmTarget(ModRmTarget::Register(src))),
Mnemonic::SUB_ToReg(src, dest) => self
.computer
.sub(ModRmTarget::Register(dest), Operand::ModRmTarget(src)),
Mnemonic::SUB_Ib(dest, src) => self
.computer
.sub(dest, Operand::Immediate(ImmediateOperand::Byte(src))),
Mnemonic::SUB_Iv(dest, src) => self
.computer
.sub(dest, Operand::Immediate(ImmediateOperand::Word(src))),
Mnemonic::SUB_ALIb(src_byte) => self.computer.sub(
ModRmTarget::Register(crate::register::Register::AL),
Operand::Immediate(ImmediateOperand::Byte(src_byte)),
),
Mnemonic::SUB_AXIv(src_word) => self.computer.sub(
ModRmTarget::Register(crate::register::Register::AX),
Operand::Immediate(ImmediateOperand::Word(src_word)),
),
/*
* MOV

View File

@@ -7,7 +7,7 @@ use crate::register::SegmentRegister;
use crate::{disasm::DisasmError, register::Register};
use core::fmt;
use std::ops::{Add, Sub};
use std::ops::{Add, Div, Mul, Sub};
pub type Byte = u8; // b
pub type IByte = i8; // used for displacements of memory access
@@ -88,6 +88,44 @@ impl Sub for ImmediateOperand {
}
}
impl Mul for ImmediateOperand {
type Output = Self;
fn mul(self, other: Self) -> Self {
match self {
ImmediateOperand::Byte(lhsb) => match other {
ImmediateOperand::Byte(rhsb) => ImmediateOperand::Byte(lhsb.wrapping_mul(rhsb)),
_ => panic!("Cannot multiply Byte with Word"),
},
ImmediateOperand::Word(lhsw) => match other {
ImmediateOperand::Word(rhsw) => ImmediateOperand::Word(lhsw.wrapping_mul(rhsw)),
ImmediateOperand::Byte(rhsb) => {
ImmediateOperand::Word(lhsw.wrapping_mul(rhsb as Word))
}
},
}
}
}
impl Div for ImmediateOperand {
type Output = Self;
fn div(self, other: Self) -> Self {
match self {
ImmediateOperand::Byte(lhsb) => match other {
ImmediateOperand::Byte(rhsb) => ImmediateOperand::Byte(lhsb.wrapping_div(rhsb)),
_ => panic!("Cannot divide Byte with Word"),
},
ImmediateOperand::Word(lhsw) => match other {
ImmediateOperand::Word(rhsw) => ImmediateOperand::Word(lhsw.wrapping_div(rhsw)),
ImmediateOperand::Byte(rhsb) => {
ImmediateOperand::Word(lhsw.wrapping_div(rhsb as Word))
}
},
}
}
}
impl fmt::Display for ImmediateOperand {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {