crsn/crsn/src/builtin/exec.rs

use std::time::Duration;

use crate::asm::data::{Rd, RdData};
use crate::asm::data::literal::Addr;
use crate::asm::instr::Cond;
use crate::builtin::defs::{Barrier, BuiltinOp};
use crate::module::{EvalRes, OpTrait};
use crate::runtime::fault::Fault;
use crate::runtime::frame::StackFrame;
use crate::runtime::run_thread::{state::RunState, ThreadInfo};
use sexp::Sexp;
use sexp::Atom;
use std::str::FromStr;
use std::fmt::Display;
use crate::utils::A;

impl OpTrait for BuiltinOp {
    fn execute(&self, info: &ThreadInfo, state: &mut RunState) -> Result<EvalRes, Fault> {
        let program = &info.program;
        let mut res = EvalRes::default();
        match self {
            BuiltinOp::Nop => {}
            BuiltinOp::Halt => {
                trace!("HALT - stop execution");
                return Err(Fault::Halt);
            }
            BuiltinOp::Label(_) | BuiltinOp::FarLabel(_) | BuiltinOp::Routine(_) => {
                /* this is nop, but without any cost - just markers */
                res.cycles = 0;
            }
            BuiltinOp::Barrier {
                kind: Barrier::Open(lbl),
                ..
            } => {
                match program.find_far_label(lbl) {
                    Ok(pos) => {
                        res.cycles = 0;
                        state.set_pc(pos);
                    }
                    Err(e) => {
                        return Err(e);
                    }
                }
            }
            BuiltinOp::Barrier {
                msg, ..
            } => {
                return Err(Fault::FaultInstr {
                    msg: msg.clone().unwrap_or_else(|| "BARRIER".into())
                });
            }
            BuiltinOp::Fault(msg) => {
                return Err(Fault::FaultInstr {
                    msg: msg.clone().unwrap_or_else(|| "FAULT".into())
                });
            }
            BuiltinOp::FarJump(name) => {
                match program.find_far_label(name) {
                    Ok(pos) => {
                        state.set_pc(pos);
                    }
                    Err(e) => {
                        return Err(e);
                    }
                }
            }
            BuiltinOp::Jump(_name) => {
                panic!("jump not translated to skip by assembler!");
            }
            BuiltinOp::Call(name, args) => {
                match program.find_routine(&name) {
                    Ok(pos) => {
                        let mut values = Vec::with_capacity(args.len());
                        for arg in args {
                            values.push(state.read(*arg)?);
                        }

                        let mut frame2 = StackFrame::new(pos, &values);
                        std::mem::swap(&mut state.frame, &mut frame2);
                        state.call_stack.push(frame2);
                        res.advance = 0;
                    }
                    Err(e) => {
                        return Err(e);
                    }
                }
            }
            BuiltinOp::Ret(retvals) => {
                match state.call_stack.pop() {
                    Some(previous) => {
                        let mut values = Vec::with_capacity(retvals.len());
                        for arg in retvals {
                            values.push(state.read(*arg)?);
                        }
                        state.frame = previous;
                        state.frame.set_retvals(&values);
                    }
                    None => {
                        return Err(Fault::CallStackUnderflow);
                    }
                }
            }
            BuiltinOp::Skip(val) => {
                let steps = state.read(*val)?;
                res.advance = i64::from_ne_bytes(steps.to_ne_bytes());
                let pc = state.get_pc();
                program.validate_jump(pc, Addr((pc.0 as i64 + res.advance) as u64))?;
            }
            BuiltinOp::Move { dst, src } => {
                state.clear_status();
                let val = state.read(*src)?;
                state.update_status(val);
                state.write(*dst, val)?;
            }
            BuiltinOp::StoreStatus { dst } => {
                let packed = state.frame.status.store();
                state.write(*dst, packed)?;
            }
            BuiltinOp::LoadStatus { src } => {
                let x = state.read(*src)?;
                state.frame.status.load(x);
            }
            BuiltinOp::Sleep { micros } => {
                std::thread::sleep(Duration::from_micros(state.read(*micros)?))
            }
            BuiltinOp::Drop(obj) => {
                let x = state.read(Rd::new(RdData::Register(obj.reg())))?;

                trace!("Drop object: {:#x}", x);
                let mut dropped = false;
                for ex in info.program.extensions.iter() {
                    if ex.drop_obj(info, state, x)?.is_some() {
                        dropped = true;
                    }
                }
                if !dropped {
                    warn!("Object {:#x} to drop does not exist!", x);
                    state.set_flag(Cond::Invalid, true);
                }
            }
        }

        Ok(res)
    }

    fn to_sexp(&self) -> Sexp {
        match self {
            BuiltinOp::Nop => sexp::list(&[A("nop")]),
            BuiltinOp::Halt => sexp::list(&[A("halt")]),
            BuiltinOp::Sleep { micros } => sexp::list(&[A("sleep"), A(micros)]),
            BuiltinOp::Label(label) => sexp::list(&[A(label)]),
            BuiltinOp::Jump(label) => sexp::list(&[A("j"), A(label)]),
            BuiltinOp::FarLabel(label) => sexp::list(&[A("far"), A(label)]),
            BuiltinOp::FarJump(label) => sexp::list(&[A("fj"), A(label)]),
            BuiltinOp::Call(name, args) => {
                if args.is_empty() {
                    sexp::list(&[A("call"), A(name)])
                } else {
                    let mut inner = vec![A("call"), A(&name.name)];
                    inner.extend(args.iter().map(A));
                    sexp::list(&inner)
                }
            },
            BuiltinOp::Ret(values) => {
                if values.is_empty() {
                    sexp::list(&[A("ret")])
                } else {
                    let mut inner = vec![A("ret")];
                    inner.extend(values.iter().map(A));
                    sexp::list(&inner)
                }
            }
            BuiltinOp::Routine(name) => sexp::list(&[A("routine"), A(name)]),
            BuiltinOp::Skip(n) => sexp::list(&[A("skip"), A(n)]),
            BuiltinOp::Barrier { kind, msg } => {
                let mut inner = vec![];
                match kind {
                    Barrier::Open(label) => {
                        inner.push(A("barrier-open"));
                        inner.push(A(label));
                    }
                    Barrier::Close(label) => {
                        inner.push(A("barrier-close"));
                        inner.push(A(label));
                    }
                    Barrier::Standalone => {
                        inner.push(A("barrier"));

                        if let Some(msg) = msg {
                            inner.push(A(msg));
                        }
                    }
                }
                sexp::list(&inner)
            }
            BuiltinOp::Fault(msg) => {
                if let Some(msg) = msg {
                    sexp::list(&[A("fault"), A(msg)])
                } else {
                    sexp::list(&[A("fault")])
                }
            }
            BuiltinOp::Drop(obj) => sexp::list(&[A("drop"), A(obj)]),
            BuiltinOp::Move { dst, src } => sexp::list(&[A("ld"), A(dst), A(src)]),
            BuiltinOp::StoreStatus { dst } => sexp::list(&[A("sst"), A(dst)]),
            BuiltinOp::LoadStatus { src } => sexp::list(&[A("lst"), A(src)])
        }
    }
}