335 lines
10 KiB
Rust
335 lines
10 KiB
Rust
use fix_abstract_vm::{NixType, resolve_operand};
|
|
use fix_common::StringId;
|
|
use fix_error::Error;
|
|
use gc_arena::{Gc, RefLock};
|
|
use smallvec::SmallVec;
|
|
|
|
use crate::{
|
|
AttrSet, BytecodeReader, List, Step, StrictValue, Value, VmRuntimeCtx, VmRuntimeCtxExt,
|
|
};
|
|
|
|
impl<'gc> crate::Vm<'gc> {
|
|
#[inline(always)]
|
|
pub(crate) fn op_make_attrs(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let static_count = reader.read_u32() as usize;
|
|
let dynamic_count = reader.read_u32() as usize;
|
|
|
|
for i in 0..dynamic_count {
|
|
let depth = dynamic_count - 1 - i;
|
|
self.force_slot_to_pc(depth, reader, mc, reader.inst_start_pc())?;
|
|
}
|
|
|
|
let mut dyn_keys: SmallVec<[_; 2]> = SmallVec::with_capacity(dynamic_count);
|
|
for i in 0..dynamic_count {
|
|
let depth = dynamic_count - 1 - i;
|
|
let key_val = self.peek_forced(depth);
|
|
let key_sid = match ctx.get_string_id(key_val) {
|
|
Ok(id) => Some(id),
|
|
Err(NixType::Null) => None,
|
|
Err(got) => return self.finish_type_err(NixType::String, got),
|
|
};
|
|
dyn_keys.push(key_sid);
|
|
}
|
|
|
|
self.stack.truncate(self.stack.len() - dynamic_count);
|
|
|
|
let mut kv: SmallVec<[(crate::StringId, Value); 4]> =
|
|
SmallVec::with_capacity(static_count + dynamic_count);
|
|
|
|
for _ in 0..static_count {
|
|
let key = reader.read_string_id();
|
|
let val = resolve_operand(&reader.read_operand_data(ctx), mc, self);
|
|
let _span_id = reader.read_u32();
|
|
kv.push((key, val));
|
|
}
|
|
|
|
for key in dyn_keys {
|
|
let val = resolve_operand(&reader.read_operand_data(ctx), mc, self);
|
|
let _span_id = reader.read_u32();
|
|
if let Some(key) = key {
|
|
kv.push((key, val))
|
|
}
|
|
}
|
|
|
|
kv.sort_by_key(|(k, _)| *k);
|
|
let attrs = Gc::new(mc, AttrSet::from_sorted_unchecked(kv));
|
|
self.push(Value::new_gc(attrs));
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_make_empty_attrs(&mut self) -> Step {
|
|
self.push(self.empty_attrs);
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_select_static(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let _span_id = reader.read_u32();
|
|
let key = reader.read_string_id();
|
|
|
|
let attrset = self.force_and_retry::<Gc<AttrSet>>(reader, mc)?;
|
|
|
|
match attrset.lookup(key) {
|
|
Some(v) => {
|
|
self.push(v);
|
|
}
|
|
None => return self.select_skip(key, ctx, reader),
|
|
}
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_select_dynamic(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let _span_id = reader.read_u32();
|
|
|
|
let (attrset, key_val) = self.force_and_retry::<(Gc<AttrSet>, StrictValue)>(reader, mc)?;
|
|
|
|
let key_sid = match ctx.get_string_id(key_val) {
|
|
Ok(id) => id,
|
|
Err(got) => return self.finish_type_err(NixType::String, got),
|
|
};
|
|
|
|
match attrset.lookup(key_sid) {
|
|
Some(v) => {
|
|
self.push(v);
|
|
}
|
|
None => return self.select_skip(key_sid, ctx, reader),
|
|
}
|
|
Step::Continue(())
|
|
}
|
|
|
|
/// Skip the rest of a **Select** attrpath after a missing attribute.
|
|
/// Only recognises Select opcodes and jumps; encountering any other
|
|
/// opcode means we've reached the end of the select sequence and
|
|
/// should report the missing-attribute error.
|
|
fn select_skip(
|
|
&mut self,
|
|
key: StringId,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
) -> Step {
|
|
use fix_codegen::Op::*;
|
|
loop {
|
|
match reader.read_op() {
|
|
SelectStatic => {
|
|
reader.set_pc(reader.pc() + 4 + 4);
|
|
}
|
|
SelectDynamic => {
|
|
reader.set_pc(reader.pc() + 4);
|
|
}
|
|
JumpIfSelectSucceeded => {
|
|
reader.set_pc(reader.pc() + 4);
|
|
break Step::Continue(());
|
|
}
|
|
JumpIfSelectFailed => {
|
|
let offset = reader.read_i32();
|
|
reader.set_pc(((reader.pc() as isize) + (offset as isize)) as usize);
|
|
}
|
|
_ => {
|
|
let name = ctx.resolve_string(key);
|
|
return self
|
|
.finish_err(Error::eval_error(format!("attribute '{name}' missing")));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Skip the rest of a **HasAttr** attrpath after an intermediate
|
|
/// lookup failed. Only recognises HasAttr opcodes and jumps.
|
|
fn has_attr_skip(&mut self, reader: &mut BytecodeReader<'_>) -> Step {
|
|
use fix_codegen::Op::*;
|
|
loop {
|
|
match reader.read_op() {
|
|
HasAttrPathStatic => {
|
|
reader.set_pc(reader.pc() + 4 + 4);
|
|
}
|
|
HasAttrPathDynamic => {
|
|
reader.set_pc(reader.pc() + 4);
|
|
}
|
|
HasAttrStatic => {
|
|
reader.set_pc(reader.pc() + 4);
|
|
break Step::Continue(());
|
|
}
|
|
JumpIfSelectFailed => {
|
|
let offset = reader.read_i32();
|
|
reader.set_pc(((reader.pc() as isize) + (offset as isize)) as usize);
|
|
}
|
|
HasAttrDynamic => {
|
|
break Step::Continue(());
|
|
}
|
|
HasAttrResolve => {
|
|
reader.set_pc(reader.pc() - 1);
|
|
break Step::Continue(());
|
|
}
|
|
other => {
|
|
unreachable!("unexpected opcode {:?} in has_attr_skip", other as u8)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_has_attr_path_static(
|
|
&mut self,
|
|
_ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let _span_id = reader.read_u32();
|
|
let key = reader.read_string_id();
|
|
|
|
let current = self.force_and_retry::<StrictValue>(reader, mc)?;
|
|
|
|
match current
|
|
.as_gc::<AttrSet>()
|
|
.and_then(|attrs| attrs.lookup(key))
|
|
{
|
|
Some(v) => {
|
|
self.push(v);
|
|
}
|
|
None => return self.has_attr_skip(reader),
|
|
}
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_has_attr_path_dynamic(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let _span_id = reader.read_u32();
|
|
|
|
let (current, key_val) = self.force_and_retry::<(StrictValue, StrictValue)>(reader, mc)?;
|
|
|
|
let key_sid = match ctx.get_string_id(key_val) {
|
|
Ok(id) => id,
|
|
Err(got) => return self.finish_type_err(NixType::String, got),
|
|
};
|
|
|
|
match current
|
|
.as_gc::<AttrSet>()
|
|
.and_then(|attrs| attrs.lookup(key_sid))
|
|
{
|
|
Some(v) => {
|
|
self.push(v);
|
|
}
|
|
None => return self.has_attr_skip(reader),
|
|
}
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_jump_if_select_failed(&mut self, reader: &mut BytecodeReader<'_>) -> Step {
|
|
// No-op
|
|
let _offset = reader.read_i32();
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_jump_if_select_succeeded(&mut self, reader: &mut BytecodeReader<'_>) -> Step {
|
|
let offset = reader.read_i32();
|
|
reader.set_pc(((reader.pc() as isize) + (offset as isize)) as usize);
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_has_attr_static(
|
|
&mut self,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let key = reader.read_string_id();
|
|
let current = self.force_and_retry::<StrictValue>(reader, mc)?;
|
|
|
|
self.push(Value::new_inline(
|
|
current
|
|
.as_gc::<AttrSet>()
|
|
.and_then(|attrs| attrs.lookup(key))
|
|
.is_some(),
|
|
));
|
|
// Skip HasAttrResolve
|
|
reader.set_pc(reader.pc() + 1);
|
|
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_has_attr_dynamic(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let (current, dyn_key) = self.force_and_retry::<(StrictValue, StrictValue)>(reader, mc)?;
|
|
|
|
let key_sid = match ctx.get_string_id(dyn_key) {
|
|
Ok(id) => id,
|
|
Err(got) => return self.finish_type_err(NixType::String, got),
|
|
};
|
|
|
|
self.push(Value::new_inline(
|
|
current
|
|
.as_gc::<AttrSet>()
|
|
.and_then(|attrs| attrs.lookup(key_sid))
|
|
.is_some(),
|
|
));
|
|
// Skip HasAttrResolve
|
|
reader.set_pc(reader.pc() + 1);
|
|
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_has_attr_resolve(&mut self) -> Step {
|
|
// If we reach here, has_attr check has failed, push false (AttrSet is already popped)
|
|
self.push(Value::new_inline(false));
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_make_list(
|
|
&mut self,
|
|
ctx: &mut impl VmRuntimeCtx,
|
|
reader: &mut BytecodeReader<'_>,
|
|
mc: &gc_arena::Mutation<'gc>,
|
|
) -> Step {
|
|
let count = reader.read_u32() as usize;
|
|
let mut items: SmallVec<[Value; 4]> = SmallVec::with_capacity(count);
|
|
for _ in 0..count {
|
|
items.push(resolve_operand(&reader.read_operand_data(ctx), mc, self));
|
|
}
|
|
let list = Gc::new(
|
|
mc,
|
|
List {
|
|
inner: RefLock::new(items),
|
|
},
|
|
);
|
|
self.push(Value::new_gc(list));
|
|
Step::Continue(())
|
|
}
|
|
|
|
#[inline(always)]
|
|
pub(crate) fn op_make_empty_list(&mut self) -> Step {
|
|
self.push(self.empty_list);
|
|
Step::Continue(())
|
|
}
|
|
}
|