722 lines
26 KiB
Rust
722 lines
26 KiB
Rust
use std::{cmp::Ordering, mem::replace};
|
|
|
|
use crate::{
|
|
parse, value::Primitive, AnnotatedError, Chunk, ChunkError, DustError, Identifier, Instruction,
|
|
Operation, Span, Type, Value, ValueError,
|
|
};
|
|
|
|
pub fn run(source: &str) -> Result<Option<Value>, DustError> {
|
|
let chunk = parse(source)?;
|
|
let mut vm = Vm::new(chunk);
|
|
|
|
vm.run()
|
|
.map_err(|error| DustError::Runtime { error, source })
|
|
}
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
pub struct Vm {
|
|
ip: usize,
|
|
chunk: Chunk,
|
|
stack: Vec<Register>,
|
|
}
|
|
|
|
impl Vm {
|
|
const STACK_LIMIT: usize = u16::MAX as usize;
|
|
|
|
pub fn new(chunk: Chunk) -> Self {
|
|
Self {
|
|
ip: 0,
|
|
chunk,
|
|
stack: Vec::new(),
|
|
}
|
|
}
|
|
|
|
pub fn run(&mut self) -> Result<Option<Value>, VmError> {
|
|
// DRY helper to get constant or register values for binary operations
|
|
fn get_arguments(
|
|
vm: &mut Vm,
|
|
instruction: Instruction,
|
|
position: Span,
|
|
) -> Result<(&Value, &Value), VmError> {
|
|
let left = if instruction.b_is_constant() {
|
|
vm.chunk.get_constant(instruction.b(), position)?
|
|
} else {
|
|
vm.get(instruction.b(), position)?
|
|
};
|
|
let right = if instruction.c_is_constant() {
|
|
vm.chunk.get_constant(instruction.c(), position)?
|
|
} else {
|
|
vm.get(instruction.c(), position)?
|
|
};
|
|
|
|
Ok((left, right))
|
|
}
|
|
|
|
while let Ok((instruction, position)) = self.read(Span(0, 0)).copied() {
|
|
log::info!(
|
|
"{} | {} | {} | {}",
|
|
self.ip - 1,
|
|
position,
|
|
instruction.operation(),
|
|
instruction
|
|
.disassembly_info(Some(&self.chunk))
|
|
.0
|
|
.unwrap_or_default()
|
|
);
|
|
|
|
match instruction.operation() {
|
|
Operation::Move => {
|
|
let to_register = instruction.a();
|
|
let from_register = instruction.b();
|
|
let value = self.take(from_register, to_register, position)?;
|
|
|
|
self.set(to_register, value, position)?;
|
|
}
|
|
Operation::Close => {
|
|
let from_register = instruction.b();
|
|
let to_register = instruction.c();
|
|
|
|
if self.stack.len() < to_register as usize {
|
|
return Err(VmError::StackUnderflow { position });
|
|
}
|
|
|
|
for register_index in from_register..to_register {
|
|
self.stack[register_index as usize] = Register::Empty;
|
|
}
|
|
}
|
|
Operation::LoadBoolean => {
|
|
let to_register = instruction.a();
|
|
let boolean = instruction.b_as_boolean();
|
|
let skip = instruction.c_as_boolean();
|
|
let value = Value::boolean(boolean);
|
|
|
|
self.set(to_register, value, position)?;
|
|
|
|
if skip {
|
|
self.ip += 1;
|
|
}
|
|
}
|
|
Operation::LoadConstant => {
|
|
let to_register = instruction.a();
|
|
let from_constant = instruction.b();
|
|
let jump = instruction.c_as_boolean();
|
|
|
|
self.set_constant(to_register, from_constant, position)?;
|
|
|
|
if jump {
|
|
self.ip += 1;
|
|
}
|
|
}
|
|
Operation::LoadList => {
|
|
let to_register = instruction.a();
|
|
let first_register = instruction.b();
|
|
let last_register = instruction.c();
|
|
|
|
let is_empty = to_register == first_register && first_register == last_register;
|
|
let item_type = if is_empty {
|
|
Type::Any
|
|
} else {
|
|
self.get(first_register, position)?.r#type()
|
|
};
|
|
let value = Value::list(first_register, last_register, item_type);
|
|
|
|
self.set(to_register, value, position)?;
|
|
}
|
|
Operation::DefineLocal => {
|
|
let from_register = instruction.a();
|
|
let to_local = instruction.b();
|
|
|
|
self.chunk.define_local(to_local, from_register, position)?;
|
|
}
|
|
Operation::GetLocal => {
|
|
let to_register = instruction.a();
|
|
let local_index = instruction.b();
|
|
let local = self.chunk.get_local(local_index, position)?;
|
|
|
|
self.set_pointer(to_register, local.register_index, position)?;
|
|
}
|
|
Operation::SetLocal => {
|
|
let register = instruction.a();
|
|
let local_index = instruction.b();
|
|
let local = self.chunk.get_local(local_index, position)?;
|
|
|
|
if !local.is_mutable {
|
|
return Err(VmError::CannotMutateImmutableLocal {
|
|
identifier: local.identifier.clone(),
|
|
position,
|
|
});
|
|
}
|
|
|
|
self.chunk.define_local(local_index, register, position)?;
|
|
}
|
|
Operation::Add => {
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let sum = left
|
|
.add(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), sum, position)?;
|
|
}
|
|
Operation::Subtract => {
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let difference = left
|
|
.subtract(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), difference, position)?;
|
|
}
|
|
Operation::Multiply => {
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let product = left
|
|
.multiply(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), product, position)?;
|
|
}
|
|
Operation::Divide => {
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let quotient = left
|
|
.divide(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), quotient, position)?;
|
|
}
|
|
Operation::Modulo => {
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let remainder = left
|
|
.modulo(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), remainder, position)?;
|
|
}
|
|
Operation::Test => {
|
|
let register = instruction.a();
|
|
let test_value = instruction.c_as_boolean();
|
|
let value = self.get(register, position)?;
|
|
let boolean = if let Value::Primitive(Primitive::Boolean(boolean)) = value {
|
|
*boolean
|
|
} else {
|
|
return Err(VmError::ExpectedBoolean {
|
|
found: value.clone(),
|
|
position,
|
|
});
|
|
};
|
|
|
|
if boolean != test_value {
|
|
self.ip += 1;
|
|
}
|
|
}
|
|
Operation::TestSet => {
|
|
let to_register = instruction.a();
|
|
let argument = instruction.b();
|
|
let test_value = instruction.c_as_boolean();
|
|
let borrowed_value = self.get(argument, position)?;
|
|
let boolean =
|
|
if let Value::Primitive(Primitive::Boolean(boolean)) = borrowed_value {
|
|
*boolean
|
|
} else {
|
|
return Err(VmError::ExpectedBoolean {
|
|
found: borrowed_value.clone(),
|
|
position,
|
|
});
|
|
};
|
|
|
|
if boolean == test_value {
|
|
let value = self.take(argument, to_register, position)?;
|
|
|
|
self.set(to_register, value, position)?;
|
|
} else {
|
|
self.ip += 1;
|
|
}
|
|
}
|
|
Operation::Equal => {
|
|
debug_assert_eq!(
|
|
self.chunk.get_instruction(self.ip, position)?.0.operation(),
|
|
Operation::Jump
|
|
);
|
|
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let equal_result = left
|
|
.equal(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
let boolean =
|
|
if let Value::Primitive(Primitive::Boolean(boolean)) = equal_result {
|
|
boolean
|
|
} else {
|
|
return Err(VmError::ExpectedBoolean {
|
|
found: equal_result.clone(),
|
|
position,
|
|
});
|
|
};
|
|
let compare_to = instruction.a_as_boolean();
|
|
|
|
if boolean == compare_to {
|
|
self.ip += 1;
|
|
} else {
|
|
let (jump, _) = *self.chunk.get_instruction(self.ip, position)?;
|
|
let jump_distance = jump.a();
|
|
let is_positive = jump.b_as_boolean();
|
|
let new_ip = if is_positive {
|
|
self.ip + jump_distance as usize
|
|
} else {
|
|
self.ip - jump_distance as usize
|
|
};
|
|
|
|
self.ip = new_ip;
|
|
}
|
|
}
|
|
Operation::Less => {
|
|
debug_assert_eq!(
|
|
self.chunk.get_instruction(self.ip, position)?.0.operation(),
|
|
Operation::Jump
|
|
);
|
|
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let less_result = left
|
|
.less_than(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
let boolean = if let Value::Primitive(Primitive::Boolean(boolean)) = less_result
|
|
{
|
|
boolean
|
|
} else {
|
|
return Err(VmError::ExpectedBoolean {
|
|
found: less_result.clone(),
|
|
position,
|
|
});
|
|
};
|
|
let compare_to = instruction.a_as_boolean();
|
|
|
|
if boolean == compare_to {
|
|
self.ip += 1;
|
|
} else {
|
|
let jump = self.chunk.get_instruction(self.ip, position)?.0;
|
|
let jump_distance = jump.a();
|
|
let is_positive = jump.b_as_boolean();
|
|
let new_ip = if is_positive {
|
|
self.ip + jump_distance as usize
|
|
} else {
|
|
self.ip - jump_distance as usize
|
|
};
|
|
|
|
self.ip = new_ip;
|
|
}
|
|
}
|
|
Operation::LessEqual => {
|
|
debug_assert_eq!(
|
|
self.chunk.get_instruction(self.ip, position)?.0.operation(),
|
|
Operation::Jump
|
|
);
|
|
|
|
let (left, right) = get_arguments(self, instruction, position)?;
|
|
let less_or_equal_result = left
|
|
.less_than_or_equal(right)
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
let boolean = if let Value::Primitive(Primitive::Boolean(boolean)) =
|
|
less_or_equal_result
|
|
{
|
|
boolean
|
|
} else {
|
|
return Err(VmError::ExpectedBoolean {
|
|
found: less_or_equal_result.clone(),
|
|
position,
|
|
});
|
|
};
|
|
let compare_to = instruction.a_as_boolean();
|
|
|
|
if boolean == compare_to {
|
|
self.ip += 1;
|
|
} else {
|
|
let jump = self.chunk.get_instruction(self.ip, position)?.0;
|
|
let jump_distance = jump.a();
|
|
let is_positive = jump.b_as_boolean();
|
|
let new_ip = if is_positive {
|
|
self.ip + jump_distance as usize
|
|
} else {
|
|
self.ip - jump_distance as usize
|
|
};
|
|
|
|
self.ip = new_ip;
|
|
}
|
|
}
|
|
Operation::Negate => {
|
|
let value = if instruction.b_is_constant() {
|
|
self.chunk.get_constant(instruction.b(), position)?
|
|
} else {
|
|
self.get(instruction.b(), position)?
|
|
};
|
|
let negated = value
|
|
.negate()
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), negated, position)?;
|
|
}
|
|
Operation::Not => {
|
|
let value = if instruction.b_is_constant() {
|
|
self.chunk.get_constant(instruction.b(), position)?
|
|
} else {
|
|
self.get(instruction.b(), position)?
|
|
};
|
|
let not = value
|
|
.not()
|
|
.map_err(|error| VmError::Value { error, position })?;
|
|
|
|
self.set(instruction.a(), not, position)?;
|
|
}
|
|
Operation::Jump => {
|
|
let offset = instruction.b();
|
|
let is_positive = instruction.c_as_boolean();
|
|
let new_ip = if is_positive {
|
|
self.ip + offset as usize
|
|
} else {
|
|
self.ip - (offset + 1) as usize
|
|
};
|
|
|
|
self.ip = new_ip;
|
|
}
|
|
Operation::Call => {
|
|
let to_register = instruction.a();
|
|
let function_index = instruction.b();
|
|
let value = self.get(function_index, position)?.clone();
|
|
let function = if let Value::Function(function) = value {
|
|
function
|
|
} else {
|
|
return Err(VmError::ExpectedFunction {
|
|
found: value,
|
|
position,
|
|
});
|
|
};
|
|
let mut function_vm = Vm::new(function.take_chunk());
|
|
|
|
for argument_index in function_index + 1..to_register {
|
|
let argument = self.get(argument_index, position)?.clone();
|
|
let top_of_stack = function_vm.stack.len() as u8;
|
|
|
|
function_vm.set(top_of_stack, argument, position)?;
|
|
}
|
|
|
|
let return_value = function_vm.run()?;
|
|
|
|
if let Some(value) = return_value {
|
|
self.set(function_index, value, position)?;
|
|
} else {
|
|
self.empty(function_index, position)?;
|
|
}
|
|
}
|
|
Operation::Return => {
|
|
let should_return_value = instruction.b_as_boolean();
|
|
|
|
return if should_return_value {
|
|
let top_of_stack = (self.stack.len() - 1) as u8;
|
|
let value = self.empty(top_of_stack, position)?;
|
|
|
|
Ok(Some(value))
|
|
} else {
|
|
Ok(None)
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(None)
|
|
}
|
|
|
|
fn set(&mut self, to_register: u8, value: Value, position: Span) -> Result<(), VmError> {
|
|
let length = self.stack.len();
|
|
let to_register = to_register as usize;
|
|
|
|
if length == Self::STACK_LIMIT {
|
|
return Err(VmError::StackOverflow { position });
|
|
}
|
|
|
|
match to_register.cmp(&length) {
|
|
Ordering::Less => {
|
|
log::trace!("Replace R{to_register} with {value}");
|
|
|
|
self.stack[to_register] = Register::Value(value);
|
|
|
|
Ok(())
|
|
}
|
|
Ordering::Equal => {
|
|
log::trace!("Push R{to_register} with {value}");
|
|
|
|
self.stack.push(Register::Value(value));
|
|
|
|
Ok(())
|
|
}
|
|
Ordering::Greater => Err(VmError::SkippedRegister {
|
|
index: to_register,
|
|
length,
|
|
position,
|
|
}),
|
|
}
|
|
}
|
|
|
|
fn set_pointer(
|
|
&mut self,
|
|
to_register: u8,
|
|
from_register: u8,
|
|
position: Span,
|
|
) -> Result<(), VmError> {
|
|
let length = self.stack.len();
|
|
let to_register = to_register as usize;
|
|
|
|
if length == Self::STACK_LIMIT {
|
|
return Err(VmError::StackOverflow { position });
|
|
}
|
|
|
|
match to_register.cmp(&length) {
|
|
Ordering::Less => {
|
|
log::trace!("Replace R{to_register} with R{from_register}");
|
|
|
|
self.stack[to_register] = Register::Pointer(from_register);
|
|
|
|
Ok(())
|
|
}
|
|
Ordering::Equal => {
|
|
log::trace!("Push R{to_register} with R{from_register}");
|
|
|
|
self.stack.push(Register::Pointer(from_register));
|
|
|
|
Ok(())
|
|
}
|
|
Ordering::Greater => Err(VmError::SkippedRegister {
|
|
index: to_register,
|
|
length,
|
|
position,
|
|
}),
|
|
}
|
|
}
|
|
|
|
fn set_constant(
|
|
&mut self,
|
|
index: u8,
|
|
constant_index: u8,
|
|
position: Span,
|
|
) -> Result<(), VmError> {
|
|
let length = self.stack.len();
|
|
let index = index as usize;
|
|
|
|
if length == Self::STACK_LIMIT {
|
|
return Err(VmError::StackOverflow { position });
|
|
}
|
|
|
|
if index == length {
|
|
log::trace!("Push register {index} with constant {constant_index}");
|
|
|
|
self.stack.push(Register::Constant(constant_index));
|
|
|
|
return Ok(());
|
|
}
|
|
|
|
if index < length {
|
|
log::trace!("Replace register {index} with constant {constant_index}");
|
|
|
|
self.stack[index] = Register::Constant(constant_index);
|
|
|
|
return Ok(());
|
|
}
|
|
|
|
Err(VmError::SkippedRegister {
|
|
index,
|
|
length: self.stack.len(),
|
|
position,
|
|
})
|
|
}
|
|
|
|
pub fn get(&self, index: u8, position: Span) -> Result<&Value, VmError> {
|
|
let index = index as usize;
|
|
let register = self
|
|
.stack
|
|
.get(index)
|
|
.ok_or_else(|| VmError::RegisterIndexOutOfBounds { index, position })?;
|
|
|
|
match register {
|
|
Register::Value(value) => Ok(value),
|
|
Register::Pointer(register_index) => {
|
|
let value = self.get(*register_index, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Constant(constant_index) => {
|
|
let value = self.chunk.get_constant(*constant_index, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Empty => Err(VmError::EmptyRegister { index, position }),
|
|
}
|
|
}
|
|
|
|
fn take(&mut self, index: u8, point_to: u8, position: Span) -> Result<Value, VmError> {
|
|
let index = index as usize;
|
|
let register = replace(&mut self.stack[index], Register::Pointer(point_to));
|
|
|
|
match register {
|
|
Register::Value(value) => Ok(value),
|
|
Register::Pointer(register_index) => {
|
|
let value = self.take(register_index, point_to, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Constant(constant_index) => {
|
|
let value = self.chunk.take_constant(constant_index, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Empty => Err(VmError::EmptyRegister { index, position }),
|
|
}
|
|
}
|
|
|
|
fn empty(&mut self, index: u8, position: Span) -> Result<Value, VmError> {
|
|
let index = index as usize;
|
|
|
|
if index >= self.stack.len() {
|
|
return Err(VmError::RegisterIndexOutOfBounds { index, position });
|
|
}
|
|
|
|
let register = replace(&mut self.stack[index], Register::Empty);
|
|
|
|
match register {
|
|
Register::Value(value) => Ok(value),
|
|
Register::Pointer(register_index) => {
|
|
let value = self.empty(register_index, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Constant(constant_index) => {
|
|
let value = self.chunk.take_constant(constant_index, position)?;
|
|
|
|
Ok(value)
|
|
}
|
|
Register::Empty => Err(VmError::EmptyRegister { index, position }),
|
|
}
|
|
}
|
|
|
|
fn read(&mut self, position: Span) -> Result<&(Instruction, Span), VmError> {
|
|
let current = self.chunk.get_instruction(self.ip, position)?;
|
|
|
|
self.ip += 1;
|
|
|
|
Ok(current)
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
enum Register {
|
|
Empty,
|
|
Value(Value),
|
|
Pointer(u8),
|
|
Constant(u8),
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub enum VmError {
|
|
CannotMutateImmutableLocal {
|
|
identifier: Identifier,
|
|
position: Span,
|
|
},
|
|
EmptyRegister {
|
|
index: usize,
|
|
position: Span,
|
|
},
|
|
ExpectedBoolean {
|
|
found: Value,
|
|
position: Span,
|
|
},
|
|
ExpectedFunction {
|
|
found: Value,
|
|
position: Span,
|
|
},
|
|
RegisterIndexOutOfBounds {
|
|
index: usize,
|
|
position: Span,
|
|
},
|
|
InvalidInstruction {
|
|
instruction: Instruction,
|
|
position: Span,
|
|
},
|
|
SkippedRegister {
|
|
index: usize,
|
|
length: usize,
|
|
position: Span,
|
|
},
|
|
StackOverflow {
|
|
position: Span,
|
|
},
|
|
StackUnderflow {
|
|
position: Span,
|
|
},
|
|
UndefinedVariable {
|
|
identifier: Identifier,
|
|
position: Span,
|
|
},
|
|
|
|
// Wrappers for foreign errors
|
|
Chunk(ChunkError),
|
|
Value {
|
|
error: ValueError,
|
|
position: Span,
|
|
},
|
|
}
|
|
|
|
impl From<ChunkError> for VmError {
|
|
fn from(error: ChunkError) -> Self {
|
|
Self::Chunk(error)
|
|
}
|
|
}
|
|
|
|
impl AnnotatedError for VmError {
|
|
fn title() -> &'static str {
|
|
"Runtime Error"
|
|
}
|
|
|
|
fn description(&self) -> &'static str {
|
|
match self {
|
|
Self::CannotMutateImmutableLocal { .. } => "Cannot mutate immutable variable",
|
|
Self::EmptyRegister { .. } => "Empty register",
|
|
Self::ExpectedBoolean { .. } => "Expected boolean",
|
|
Self::ExpectedFunction { .. } => "Expected function",
|
|
Self::RegisterIndexOutOfBounds { .. } => "Register index out of bounds",
|
|
Self::InvalidInstruction { .. } => "Invalid instruction",
|
|
Self::SkippedRegister { .. } => "Skipped register",
|
|
Self::StackOverflow { .. } => "Stack overflow",
|
|
Self::StackUnderflow { .. } => "Stack underflow",
|
|
Self::UndefinedVariable { .. } => "Undefined variable",
|
|
Self::Chunk(error) => error.description(),
|
|
Self::Value { .. } => "Value error",
|
|
}
|
|
}
|
|
|
|
fn details(&self) -> Option<String> {
|
|
match self {
|
|
Self::EmptyRegister { index, .. } => Some(format!("Register {index} is empty")),
|
|
Self::ExpectedFunction { found, .. } => Some(format!("{found} is not a function")),
|
|
Self::RegisterIndexOutOfBounds { index, .. } => {
|
|
Some(format!("Register {index} does not exist"))
|
|
}
|
|
Self::UndefinedVariable { identifier, .. } => {
|
|
Some(format!("{identifier} is not in scope"))
|
|
}
|
|
Self::Chunk(error) => error.details(),
|
|
Self::Value { error, .. } => Some(error.to_string()),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
fn position(&self) -> Span {
|
|
match self {
|
|
Self::CannotMutateImmutableLocal { position, .. } => *position,
|
|
Self::EmptyRegister { position, .. } => *position,
|
|
Self::ExpectedBoolean { position, .. } => *position,
|
|
Self::ExpectedFunction { position, .. } => *position,
|
|
Self::RegisterIndexOutOfBounds { position, .. } => *position,
|
|
Self::InvalidInstruction { position, .. } => *position,
|
|
Self::SkippedRegister { position, .. } => *position,
|
|
Self::StackUnderflow { position } => *position,
|
|
Self::StackOverflow { position } => *position,
|
|
Self::UndefinedVariable { position, .. } => *position,
|
|
Self::Chunk(error) => error.position(),
|
|
Self::Value { position, .. } => *position,
|
|
}
|
|
}
|
|
}
|