1use std::collections::{BTreeMap, HashMap, HashSet};
2use std::path::PathBuf;
3use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
4use std::sync::{Arc, Mutex};
5use std::time::Instant;
6
7use crate::chunk::{Chunk, ChunkRef, Constant};
8use crate::runtime_limits::RuntimeLimits;
9use crate::value::{
10 ModuleFunctionRegistry, VmAsyncBuiltinFn, VmBuiltinFn, VmClosure, VmEnv, VmError, VmMutex,
11 VmTaskHandle, VmValue,
12};
13use crate::BuiltinId;
14
15use super::debug::DebugHook;
16use super::modules::ModuleCache;
17use super::VmBuiltinMetadata;
18
19pub(crate) struct ResolvedLazyCallable {
33 pub(crate) exports: BTreeMap<String, Arc<VmClosure>>,
34 #[allow(dead_code)]
39 pub(crate) retained_module_graph: ModuleCache,
40}
41
42pub(crate) type LazyCallableResolution = Arc<ResolvedLazyCallable>;
43pub(crate) struct LazyCallableCacheSlot {
44 pub(crate) execution_guard: Option<Arc<harn_modules::package_execution::PackageExecutionGuard>>,
45 pub(crate) resolution: Arc<tokio::sync::OnceCell<LazyCallableResolution>>,
46}
47pub(crate) type LazyCallableModuleCache =
48 Arc<VmMutex<BTreeMap<PathBuf, Vec<LazyCallableCacheSlot>>>>;
49
50pub(crate) struct ScopeSpan(u64);
52
53impl ScopeSpan {
54 pub(crate) fn new(kind: crate::tracing::SpanKind, name: String) -> Self {
55 Self(crate::tracing::span_start(kind, name))
56 }
57}
58
59pub(crate) struct ExecutionDeadlineState {
64 origin: Instant,
65 deadline_offset: AtomicU64,
67 abandoned: AtomicBool,
71}
72
73impl ExecutionDeadlineState {
74 pub(crate) fn new(origin: Instant, deadline: Option<Instant>) -> Arc<Self> {
75 Arc::new(Self {
76 origin,
77 deadline_offset: AtomicU64::new(Self::encode(origin, deadline)),
78 abandoned: AtomicBool::new(false),
79 })
80 }
81
82 #[inline]
83 pub(crate) fn is_active(&self) -> bool {
84 self.deadline_offset.load(Ordering::Acquire) != 0
85 }
86
87 #[inline]
88 pub(crate) fn is_abandoned(&self) -> bool {
89 self.abandoned.load(Ordering::Acquire)
90 }
91
92 pub(crate) fn fork(&self) -> Arc<Self> {
93 let state = Self::new(self.origin, self.current());
94 state
95 .abandoned
96 .store(self.is_abandoned(), Ordering::Release);
97 state
98 }
99
100 pub(crate) fn current(&self) -> Option<Instant> {
101 let encoded = self.deadline_offset.load(Ordering::Acquire);
102 (encoded != 0)
103 .then(|| self.origin + std::time::Duration::from_nanos(encoded.saturating_sub(1)))
104 }
105
106 pub(crate) fn install(self: &Arc<Self>, deadline: Instant) -> ExecutionDeadlineGuard {
107 let previous = self.deadline_offset.load(Ordering::Acquire);
108 let requested = Self::encode(self.origin, Some(deadline));
109 let active = if previous == 0 {
110 requested
111 } else {
112 previous.min(requested)
113 };
114 self.deadline_offset.store(active, Ordering::Release);
115 ExecutionDeadlineGuard {
116 state: Arc::clone(self),
117 previous,
118 completed: false,
119 }
120 }
121
122 fn encode(origin: Instant, deadline: Option<Instant>) -> u64 {
123 deadline.map_or(0, |deadline| {
124 let nanos = deadline.saturating_duration_since(origin).as_nanos();
125 u64::try_from(nanos)
126 .unwrap_or(u64::MAX - 1)
127 .saturating_add(1)
128 })
129 }
130}
131
132pub(crate) struct ExecutionDeadlineGuard {
133 state: Arc<ExecutionDeadlineState>,
134 previous: u64,
135 completed: bool,
136}
137
138impl ExecutionDeadlineGuard {
139 pub(crate) fn complete(mut self) {
142 self.completed = true;
143 }
144}
145
146impl Drop for ExecutionDeadlineGuard {
147 fn drop(&mut self) {
148 self.state
149 .deadline_offset
150 .store(self.previous, Ordering::Release);
151 if !self.completed {
152 self.state.abandoned.store(true, Ordering::Release);
153 }
154 }
155}
156
157impl Drop for ScopeSpan {
158 fn drop(&mut self) {
159 crate::tracing::span_end(self.0);
160 }
161}
162
163#[derive(Clone)]
164pub(crate) struct LocalSlot {
165 pub(crate) value: VmValue,
166 pub(crate) initialized: bool,
167 pub(crate) synced: bool,
168}
169
170impl Drop for LocalSlot {
171 fn drop(&mut self) {
172 if crate::value::recursion::is_recursive_container(&self.value) {
180 crate::value::recursion::dismantle(std::mem::replace(&mut self.value, VmValue::Nil));
181 }
182 }
183}
184
185#[derive(Clone)]
186pub(crate) struct InterruptHandler {
187 pub(crate) handle: i64,
188 pub(crate) signals: Vec<String>,
189 pub(crate) once: bool,
190 pub(crate) graceful_timeout_ms: Option<u64>,
191 pub(crate) handler: VmValue,
192}
193
194pub(crate) struct CallFrame {
196 pub(crate) chunk: ChunkRef,
197 pub(crate) inline_cache_set: usize,
201 pub(crate) ip: usize,
202 pub(crate) stack_base: usize,
203 pub(crate) saved_env: VmEnv,
204 pub(crate) initial_env: Option<VmEnv>,
212 pub(crate) initial_local_slots: Option<Vec<LocalSlot>>,
213 pub(crate) saved_iterator_depth: usize,
215 pub(crate) fn_name: String,
217 pub(crate) argc: usize,
219 pub(crate) saved_source_dir: Option<std::path::PathBuf>,
222 pub(crate) module_functions: Option<ModuleFunctionRegistry>,
224 pub(crate) module_state: Option<crate::value::ModuleState>,
230 pub(crate) local_slots: Vec<LocalSlot>,
232 pub(crate) local_scope_base: usize,
234 pub(crate) local_scope_depth: usize,
236}
237
238pub(crate) struct InlineCacheSite {
239 pub(crate) cache_set: usize,
240 pub(crate) slot_count: usize,
241 pub(crate) slot: Option<usize>,
242}
243
244impl CallFrame {
245 #[inline]
246 pub(crate) fn inline_cache_site_for_previous_op(&self) -> InlineCacheSite {
247 let op_offset = self.ip.saturating_sub(1);
248 InlineCacheSite {
249 cache_set: self.inline_cache_set,
250 slot_count: self.chunk.inline_cache_slot_count(),
251 slot: self.chunk.inline_cache_slot(op_offset),
252 }
253 }
254}
255
256pub(crate) struct ExceptionHandler {
258 pub(crate) catch_ip: usize,
259 pub(crate) stack_depth: usize,
260 pub(crate) frame_depth: usize,
261 pub(crate) env_scope_depth: usize,
262 pub(crate) error_type: Option<crate::value::HarnStr>,
264}
265
266pub(crate) struct TaskScope {
269 pub(crate) task_ids: Vec<String>,
272 pub(crate) frame_depth: usize,
274 pub(crate) env_scope_depth: usize,
276}
277
278pub(crate) struct ProcessExitRequest {
282 code: Mutex<Option<i32>>,
283 requested: AtomicBool,
284}
285
286impl ProcessExitRequest {
287 fn new() -> Self {
288 Self {
289 code: Mutex::new(None),
290 requested: AtomicBool::new(false),
291 }
292 }
293
294 fn request(&self, code: i32) {
295 let mut recorded = self
296 .code
297 .lock()
298 .expect("process exit request lock poisoned");
299 if recorded.is_none() {
300 *recorded = Some(code);
301 self.requested.store(true, Ordering::Release);
302 }
303 }
304
305 fn code(&self) -> Option<i32> {
306 if !self.requested.load(Ordering::Acquire) {
307 return None;
308 }
309 *self
310 .code
311 .lock()
312 .expect("process exit request lock poisoned")
313 }
314}
315
316pub(crate) enum IterState {
318 Vec {
319 items: Arc<Vec<VmValue>>,
320 idx: usize,
321 },
322 Dict {
323 entries: Arc<crate::value::DictMap>,
324 keys: Vec<String>,
325 idx: usize,
326 },
327 Channel {
328 receiver: std::sync::Arc<tokio::sync::Mutex<tokio::sync::mpsc::Receiver<VmValue>>>,
329 close: std::sync::Arc<crate::value::VmChannelCloseState>,
330 },
331 Generator {
332 gen: Arc<crate::value::VmGenerator>,
333 },
334 Stream {
335 stream: Arc<crate::value::VmStream>,
336 },
337 Range {
341 next: i64,
342 end: i64,
343 inclusive: bool,
344 done: bool,
345 },
346 VmIter {
347 handle: crate::vm::iter::VmIterHandle,
348 },
349}
350
351#[derive(Clone)]
352pub(crate) enum VmBuiltinDispatch {
353 Sync(VmBuiltinFn),
354 Async(VmAsyncBuiltinFn),
355}
356
357#[derive(Clone)]
358pub(crate) struct VmBuiltinEntry {
359 pub(crate) name: Arc<str>,
360 pub(crate) dispatch: VmBuiltinDispatch,
361 pub(crate) recorded_effects: Option<&'static [harn_builtin_meta::EffectSpec]>,
365}
366
367pub struct Vm {
369 pub(crate) stack: Vec<VmValue>,
370 pub(crate) env: VmEnv,
371 pub(crate) output: String,
372 pub(crate) builtins: Arc<BTreeMap<String, VmBuiltinFn>>,
373 pub(crate) async_builtins: Arc<BTreeMap<String, VmAsyncBuiltinFn>>,
374 pub(crate) capability_methods:
378 Arc<BTreeMap<harn_builtin_meta::CapabilityId, BTreeMap<String, VmBuiltinDispatch>>>,
379 pub(crate) builtin_metadata: Arc<BTreeMap<String, VmBuiltinMetadata>>,
380 pub(crate) builtins_by_id: Arc<HashMap<BuiltinId, VmBuiltinEntry>>,
383 pub(crate) builtin_id_collisions: Arc<HashSet<BuiltinId>>,
386 pub(crate) iterators: Vec<IterState>,
388 pub(crate) frames: Vec<CallFrame>,
390 pub(crate) exception_handlers: Vec<ExceptionHandler>,
392 pub(crate) spawned_tasks: BTreeMap<String, VmTaskHandle>,
394 pub(crate) process_exit_request: Arc<ProcessExitRequest>,
396 pub(crate) sync_runtime: Arc<crate::synchronization::VmSyncRuntime>,
398 pub(crate) shared_state_runtime: Arc<crate::shared_state::VmSharedStateRuntime>,
400 pub(crate) inline_cache_sets: Vec<Vec<crate::chunk::InlineCacheEntry>>,
404 pub(crate) inline_cache_set_by_chunk: HashMap<u64, usize>,
405 pub(crate) pool_registry: Arc<crate::stdlib::pool::PoolRegistry>,
407 pub(crate) llm_mock_context: crate::llm::mock::LlmMockContext,
409 pub(crate) package_snapshot_registry: Arc<crate::stdlib::PackageSnapshotRegistry>,
413 pub(crate) wait_for_graph: Arc<crate::wait_for_graph::VmWaitForGraph>,
415 pub(crate) held_sync_guards: Vec<crate::synchronization::VmSyncHeldGuard>,
417 pub(crate) inherited_held_keys: Arc<Vec<crate::synchronization::VmSyncHeldKey>>,
425 pub(crate) task_scopes: Vec<TaskScope>,
431 pub(crate) task_counter: u64,
433 pub(crate) runtime_context_counter: u64,
435 pub(crate) runtime_context: crate::runtime_context::RuntimeContext,
437 pub(crate) deadlines: Vec<(Instant, usize)>,
439 pub(crate) execution_deadline: Arc<ExecutionDeadlineState>,
441 pub(crate) breakpoints: BTreeMap<String, std::collections::BTreeSet<usize>>,
446 pub(crate) function_breakpoints: std::collections::BTreeSet<String>,
452 pub(crate) pending_function_bp: Option<String>,
457 pub(crate) step_mode: bool,
459 pub(crate) step_frame_depth: usize,
461 pub(crate) stopped: bool,
463 pub(crate) last_line: usize,
465 pub(crate) source_dir: Option<std::path::PathBuf>,
467 pub(crate) package_execution_guard:
469 Option<Arc<harn_modules::package_execution::PackageExecutionGuard>>,
470 pub(crate) imported_paths: Vec<std::path::PathBuf>,
472 pub(crate) deferred_cyclic_imports: Vec<super::modules::DeferredCyclicImport>,
476 pub(crate) module_cache: ModuleCache,
478 pub(crate) prepared_module_cache: crate::PreparedModuleCache,
481 pub(crate) module_provenance: crate::module_artifact::ModuleProvenance,
484 pub(crate) module_phase_recorder: Option<super::ModulePhaseRecorder>,
486 pub(crate) lazy_callable_modules: LazyCallableModuleCache,
490 pub(crate) source_cache: Arc<BTreeMap<std::path::PathBuf, Arc<str>>>,
494 pub(crate) graph_link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
499 pub(crate) source_file: Option<String>,
501 pub(crate) source_text: Option<String>,
503 pub(crate) coverage: Option<crate::coverage::Coverage>,
506 pub(crate) bridge: Option<Arc<crate::bridge::HostBridge>>,
508 pub(crate) denied_builtins: Arc<HashSet<String>>,
510 pub(crate) cancel_token: Option<std::sync::Arc<std::sync::atomic::AtomicBool>>,
512 pub(crate) interrupt_signal_token: Option<std::sync::Arc<std::sync::Mutex<Option<String>>>>,
513 pub(crate) cancel_grace_instructions_remaining: Option<usize>,
518 pub(crate) interrupt_handlers: Vec<InterruptHandler>,
520 pub(crate) next_interrupt_handle: i64,
521 pub(crate) pending_interrupt_signal: Option<String>,
522 pub(crate) interrupted: bool,
523 pub(crate) dispatching_interrupt: bool,
524 pub(crate) interrupt_handler_deadline: Option<Instant>,
525 pub(crate) error_stack_trace: Vec<(String, usize, usize, Option<String>)>,
527 pub(crate) yield_sender: Option<tokio::sync::mpsc::Sender<Result<VmValue, VmError>>>,
530 pub(crate) project_root: Option<std::path::PathBuf>,
533 pub(crate) globals: Arc<crate::value::DictMap>,
536 pub(crate) root_harness: Option<VmValue>,
541 pub(crate) runtime_effects: crate::orchestration::RuntimeEffectState,
543 pub(crate) debug_hook: Option<parking_lot::Mutex<Box<DebugHook>>>,
545 pub(crate) runtime_limits: RuntimeLimits,
547}
548
549#[derive(Clone)]
557pub struct VmBaseline {
558 builtins: Arc<BTreeMap<String, VmBuiltinFn>>,
559 async_builtins: Arc<BTreeMap<String, VmAsyncBuiltinFn>>,
560 capability_methods:
561 Arc<BTreeMap<harn_builtin_meta::CapabilityId, BTreeMap<String, VmBuiltinDispatch>>>,
562 builtin_metadata: Arc<BTreeMap<String, VmBuiltinMetadata>>,
563 builtins_by_id: Arc<HashMap<BuiltinId, VmBuiltinEntry>>,
564 builtin_id_collisions: Arc<HashSet<BuiltinId>>,
565 source_dir: Option<std::path::PathBuf>,
566 source_file: Option<String>,
567 source_text: Option<String>,
568 project_root: Option<std::path::PathBuf>,
569 globals: Arc<crate::value::DictMap>,
570 root_harness: Option<VmValue>,
571 denied_builtins: Arc<HashSet<String>>,
572 prepared_module_cache: crate::PreparedModuleCache,
573 module_provenance: crate::module_artifact::ModuleProvenance,
574 graph_link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
577 runtime_limits: RuntimeLimits,
578}
579
580impl VmBaseline {
581 pub fn from_vm(vm: &Vm) -> Self {
582 Self {
583 builtins: Arc::clone(&vm.builtins),
584 async_builtins: Arc::clone(&vm.async_builtins),
585 capability_methods: Arc::clone(&vm.capability_methods),
586 builtin_metadata: Arc::clone(&vm.builtin_metadata),
587 builtins_by_id: Arc::clone(&vm.builtins_by_id),
588 builtin_id_collisions: Arc::clone(&vm.builtin_id_collisions),
589 source_dir: vm.source_dir.clone(),
590 source_file: vm.source_file.clone(),
591 source_text: vm.source_text.clone(),
592 project_root: vm.project_root.clone(),
593 globals: Arc::clone(&vm.globals),
594 root_harness: vm.root_harness.clone(),
595 denied_builtins: Arc::clone(&vm.denied_builtins),
596 prepared_module_cache: vm.prepared_module_cache.clone(),
597 module_provenance: vm.module_provenance,
598 graph_link_table: vm.graph_link_table.clone(),
599 runtime_limits: vm.runtime_limits,
600 }
601 }
602
603 pub fn instantiate(&self) -> Vm {
604 crate::initialize_runtime_assets();
605 let mut source_cache = BTreeMap::new();
606 if let (Some(file), Some(text)) = (&self.source_file, &self.source_text) {
607 source_cache.insert(std::path::PathBuf::from(file), Arc::from(text.as_str()));
608 }
609 if let Some(dir) = &self.source_dir {
610 crate::stdlib::set_thread_source_dir(dir);
611 }
612
613 let mut vm = Vm {
614 stack: Vec::with_capacity(256),
615 env: VmEnv::new(),
616 output: String::new(),
617 builtins: Arc::clone(&self.builtins),
618 async_builtins: Arc::clone(&self.async_builtins),
619 capability_methods: Arc::clone(&self.capability_methods),
620 builtin_metadata: Arc::clone(&self.builtin_metadata),
621 builtins_by_id: Arc::clone(&self.builtins_by_id),
622 builtin_id_collisions: Arc::clone(&self.builtin_id_collisions),
623 iterators: Vec::new(),
624 frames: Vec::new(),
625 exception_handlers: Vec::new(),
626 spawned_tasks: BTreeMap::new(),
627 process_exit_request: Arc::new(ProcessExitRequest::new()),
628 sync_runtime: Arc::new(crate::synchronization::VmSyncRuntime::new()),
629 shared_state_runtime: Arc::new(crate::shared_state::VmSharedStateRuntime::new()),
630 inline_cache_sets: Vec::new(),
631 inline_cache_set_by_chunk: HashMap::new(),
632 pool_registry: crate::stdlib::pool::new_pool_registry(),
633 llm_mock_context: crate::llm::mock::LlmMockContext::for_new_vm(),
634 package_snapshot_registry: Arc::new(Default::default()),
635 wait_for_graph: Arc::new(crate::wait_for_graph::VmWaitForGraph::new()),
636 held_sync_guards: Vec::new(),
637 inherited_held_keys: Arc::new(Vec::new()),
638 task_scopes: Vec::new(),
639 task_counter: 0,
640 runtime_context_counter: 0,
641 runtime_context: crate::runtime_context::RuntimeContext::root(),
642 deadlines: Vec::new(),
643 execution_deadline: super::execution::new_execution_deadline_state(None),
644 breakpoints: BTreeMap::new(),
645 function_breakpoints: std::collections::BTreeSet::new(),
646 pending_function_bp: None,
647 step_mode: false,
648 step_frame_depth: 0,
649 stopped: false,
650 last_line: 0,
651 source_dir: self.source_dir.clone(),
652 package_execution_guard: None,
653 imported_paths: Vec::new(),
654 deferred_cyclic_imports: Vec::new(),
655 module_cache: Arc::new(BTreeMap::new()),
656 prepared_module_cache: self.prepared_module_cache.clone(),
657 module_provenance: self.module_provenance,
658 module_phase_recorder: None,
659 lazy_callable_modules: Arc::new(crate::value::VmMutex::new(BTreeMap::new())),
660 source_cache: Arc::new(source_cache),
661 graph_link_table: self.graph_link_table.clone(),
662 source_file: self.source_file.clone(),
663 source_text: self.source_text.clone(),
664 coverage: crate::coverage::for_primary(self.source_file.as_deref()),
665 bridge: None,
666 denied_builtins: Arc::clone(&self.denied_builtins),
667 cancel_token: None,
668 interrupt_signal_token: None,
669 cancel_grace_instructions_remaining: None,
670 interrupt_handlers: Vec::new(),
671 next_interrupt_handle: 1,
672 pending_interrupt_signal: None,
673 interrupted: false,
674 dispatching_interrupt: false,
675 interrupt_handler_deadline: None,
676 error_stack_trace: Vec::new(),
677 yield_sender: None,
678 project_root: self.project_root.clone(),
679 globals: Arc::clone(&self.globals),
680 root_harness: self.root_harness.clone(),
681 runtime_effects: crate::orchestration::RuntimeEffectState::fresh(),
682 debug_hook: None,
683 runtime_limits: self.runtime_limits,
684 };
685
686 crate::stdlib::rebind_execution_state_builtins(&mut vm);
687 vm
688 }
689}
690
691impl Vm {
692 pub(crate) fn ensure_execution_available(&self) -> Result<(), VmError> {
693 if self.execution_deadline.is_abandoned() {
694 return Err(VmError::AbandonedExecution);
695 }
696 Ok(())
697 }
698
699 pub(crate) fn fresh_local_slots(chunk: &Chunk) -> Vec<LocalSlot> {
700 chunk
701 .local_slots
702 .iter()
703 .map(|_| LocalSlot {
704 value: VmValue::Nil,
705 initialized: false,
706 synced: false,
707 })
708 .collect()
709 }
710
711 pub(crate) fn bind_param_slots(
712 slots: &mut [LocalSlot],
713 func: &crate::chunk::CompiledFunction,
714 args: &[VmValue],
715 synced: bool,
716 ) {
717 Self::bind_param_slots_args(slots, func, &super::CallArgs::Slice(args), synced);
718 }
719
720 pub(crate) fn bind_param_slots_args(
721 slots: &mut [LocalSlot],
722 func: &crate::chunk::CompiledFunction,
723 args: &super::CallArgs<'_>,
724 synced: bool,
725 ) {
726 let param_count = func.params.len();
727 for (i, _param) in func.params.iter().enumerate() {
728 if i >= slots.len() {
729 break;
730 }
731 if func.has_rest_param && i == param_count - 1 {
732 let rest_args = args.to_vec_from(i);
733 slots[i].value = VmValue::List(std::sync::Arc::new(rest_args));
734 slots[i].initialized = true;
735 slots[i].synced = synced;
736 } else if let Some(arg) = args.get(i) {
737 slots[i].value = arg.clone();
738 slots[i].initialized = true;
739 slots[i].synced = synced;
740 }
741 }
742 }
743
744 pub(crate) fn visible_variables(&self) -> crate::value::DictMap {
745 let mut vars = self.env.all_variables();
746 let Some(frame) = self.frames.last() else {
747 return vars;
748 };
749 for (slot, info) in frame.local_slots.iter().zip(frame.chunk.local_slots.iter()) {
750 if slot.initialized && info.scope_depth <= frame.local_scope_depth {
751 vars.insert(crate::value::intern_key(&info.name), slot.value.clone());
752 }
753 }
754 vars
755 }
756
757 pub(crate) fn sync_current_frame_locals_to_env(&mut self) {
758 let frames = &mut self.frames;
759 let env = &mut self.env;
760 let Some(frame) = frames.last_mut() else {
761 return;
762 };
763 let local_scope_base = frame.local_scope_base;
764 let local_scope_depth = frame.local_scope_depth;
765 for (slot, info) in frame
766 .local_slots
767 .iter_mut()
768 .zip(frame.chunk.local_slots.iter())
769 {
770 if slot.initialized && !slot.synced && info.scope_depth <= local_scope_depth {
771 slot.synced = true;
772 let scope_idx = local_scope_base + info.scope_depth;
773 while env.scopes.len() <= scope_idx {
774 env.push_scope();
775 }
776 Arc::make_mut(&mut env.scopes[scope_idx].vars).insert(
780 info.name.clone(),
781 crate::value::Binding::Value {
782 value: slot.value.clone(),
783 mutable: info.mutable,
784 },
785 );
786 }
787 }
788 }
789
790 pub(crate) fn closure_call_env_for_current_frame(
791 &self,
792 closure: &crate::value::VmClosure,
793 ) -> VmEnv {
794 if closure.module_state().is_some() {
795 return closure.env.cloned_for_call();
796 }
797 let call_env = Self::closure_call_env(&self.env, closure);
798 if !closure.func.chunk.references_outer_names {
803 return call_env;
804 }
805 let mut call_env = call_env;
806 let Some(frame) = self.frames.last() else {
807 return call_env;
808 };
809 for (slot, info) in frame
810 .local_slots
811 .iter()
812 .zip(frame.chunk.local_slots.iter())
813 .filter(|(slot, info)| slot.initialized && info.scope_depth <= frame.local_scope_depth)
814 {
815 if matches!(slot.value, VmValue::Closure(_)) && !call_env.contains(&info.name) {
816 let _ = call_env.define(&info.name, slot.value.clone(), info.mutable);
817 }
818 }
819 call_env
820 }
821
822 pub(crate) fn active_local_slot_value(&self, name: &str) -> Option<VmValue> {
823 let frame = self.frames.last()?;
824 let idx = self.active_local_slot_index(name)?;
825 frame.local_slots.get(idx).map(|slot| slot.value.clone())
826 }
827
828 pub(crate) fn active_local_slot_index(&self, name: &str) -> Option<usize> {
833 let frame = self.frames.last()?;
834 for (idx, info) in frame.chunk.local_slots.iter().enumerate().rev() {
835 if info.name == name && info.scope_depth <= frame.local_scope_depth {
836 if let Some(slot) = frame.local_slots.get(idx) {
837 if slot.initialized {
838 return Some(idx);
839 }
840 }
841 }
842 }
843 None
844 }
845
846 pub(crate) fn assign_active_local_slot(
847 &mut self,
848 name: &str,
849 value: VmValue,
850 debug: bool,
851 ) -> Result<bool, VmError> {
852 let Some(frame) = self.frames.last_mut() else {
853 return Ok(false);
854 };
855 for (idx, info) in frame.chunk.local_slots.iter().enumerate().rev() {
856 if info.name == name && info.scope_depth <= frame.local_scope_depth {
857 if !debug && !info.mutable {
858 return Err(VmError::ImmutableAssignment(name.to_string()));
859 }
860 if let Some(slot) = frame.local_slots.get_mut(idx) {
861 crate::value::recursion::dismantle(std::mem::replace(&mut slot.value, value));
862 slot.initialized = true;
863 slot.synced = false;
864 return Ok(true);
865 }
866 }
867 }
868 Ok(false)
869 }
870
871 pub fn new() -> Self {
872 crate::initialize_runtime_assets();
873 Self {
874 stack: Vec::with_capacity(256),
875 env: VmEnv::new(),
876 output: String::new(),
877 builtins: Arc::new(BTreeMap::new()),
878 async_builtins: Arc::new(BTreeMap::new()),
879 capability_methods: Arc::new(BTreeMap::new()),
880 builtin_metadata: Arc::new(BTreeMap::new()),
881 builtins_by_id: Arc::new(HashMap::new()),
882 builtin_id_collisions: Arc::new(HashSet::new()),
883 iterators: Vec::new(),
884 frames: Vec::new(),
885 exception_handlers: Vec::new(),
886 spawned_tasks: BTreeMap::new(),
887 process_exit_request: Arc::new(ProcessExitRequest::new()),
888 sync_runtime: Arc::new(crate::synchronization::VmSyncRuntime::new()),
889 shared_state_runtime: Arc::new(crate::shared_state::VmSharedStateRuntime::new()),
890 inline_cache_sets: Vec::new(),
891 inline_cache_set_by_chunk: HashMap::new(),
892 pool_registry: crate::stdlib::pool::new_pool_registry(),
893 llm_mock_context: crate::llm::mock::LlmMockContext::for_new_vm(),
894 package_snapshot_registry: Arc::new(Default::default()),
895 wait_for_graph: Arc::new(crate::wait_for_graph::VmWaitForGraph::new()),
896 held_sync_guards: Vec::new(),
897 inherited_held_keys: Arc::new(Vec::new()),
898 task_scopes: Vec::new(),
899 task_counter: 0,
900 runtime_context_counter: 0,
901 runtime_context: crate::runtime_context::RuntimeContext::root(),
902 deadlines: Vec::new(),
903 execution_deadline: super::execution::new_execution_deadline_state(None),
904 breakpoints: BTreeMap::new(),
905 function_breakpoints: std::collections::BTreeSet::new(),
906 pending_function_bp: None,
907 step_mode: false,
908 step_frame_depth: 0,
909 stopped: false,
910 last_line: 0,
911 source_dir: None,
912 package_execution_guard: None,
913 imported_paths: Vec::new(),
914 deferred_cyclic_imports: Vec::new(),
915 module_cache: Arc::new(BTreeMap::new()),
916 prepared_module_cache: crate::PreparedModuleCache::default(),
917 module_provenance: crate::module_artifact::ModuleProvenance::User,
918 module_phase_recorder: None,
919 lazy_callable_modules: Arc::new(crate::value::VmMutex::new(BTreeMap::new())),
920 source_cache: Arc::new(BTreeMap::new()),
921 graph_link_table: None,
922 source_file: None,
923 source_text: None,
924 coverage: crate::coverage::for_primary(None),
925 bridge: None,
926 denied_builtins: Arc::new(HashSet::new()),
927 cancel_token: None,
928 interrupt_signal_token: None,
929 cancel_grace_instructions_remaining: None,
930 interrupt_handlers: Vec::new(),
931 next_interrupt_handle: 1,
932 pending_interrupt_signal: None,
933 interrupted: false,
934 dispatching_interrupt: false,
935 interrupt_handler_deadline: None,
936 error_stack_trace: Vec::new(),
937 yield_sender: None,
938 project_root: None,
939 globals: Arc::new(crate::value::DictMap::new()),
940 root_harness: None,
941 runtime_effects: crate::orchestration::RuntimeEffectState::fresh(),
942 debug_hook: None,
943 runtime_limits: RuntimeLimits::default(),
944 }
945 }
946
947 pub fn baseline(&self) -> VmBaseline {
948 VmBaseline::from_vm(self)
949 }
950
951 pub fn executed_effects(&self) -> Vec<crate::orchestration::EffectRecord> {
953 self.runtime_effects.snapshot()
954 }
955
956 pub fn clear_executed_effects(&mut self) {
958 self.runtime_effects.clear();
959 }
960
961 pub(crate) fn record_capability_effects(
962 &mut self,
963 capability: harn_builtin_meta::CapabilityId,
964 method: &str,
965 args: &[VmValue],
966 ) {
967 self.runtime_effects
968 .record_capability(capability, method, args);
969 }
970
971 pub(crate) fn record_builtin_contract_effects(&mut self, name: &str, args: &[VmValue]) {
972 let Some(entry) = crate::stdlib::recorded_effect_builtin_manifest_entry(name) else {
973 return;
974 };
975 self.record_builtin_effect_specs(entry.contract.effects, args);
976 }
977
978 pub(crate) fn record_builtin_effect_specs(
979 &mut self,
980 specs: &'static [harn_builtin_meta::EffectSpec],
981 args: &[VmValue],
982 ) {
983 self.runtime_effects.record_specs(specs, args);
984 }
985
986 pub fn set_prepared_module_cache(&mut self, cache: crate::PreparedModuleCache) {
989 self.prepared_module_cache = cache;
990 }
991
992 pub fn set_graph_link_table(
1000 &mut self,
1001 link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
1002 ) {
1003 self.graph_link_table = link_table;
1004 }
1005
1006 pub fn runtime_limits(&self) -> RuntimeLimits {
1008 self.runtime_limits
1009 }
1010
1011 pub fn runtime_limit_report(&self) -> crate::RuntimeLimitsReport {
1013 self.runtime_limits.report()
1014 }
1015
1016 #[inline]
1030 pub(crate) fn debugger_attached(&self) -> bool {
1031 self.debug_hook.is_some()
1032 || !self.breakpoints.is_empty()
1033 || !self.function_breakpoints.is_empty()
1034 }
1035
1036 pub fn set_bridge(&mut self, bridge: Arc<crate::bridge::HostBridge>) {
1038 self.bridge = Some(bridge);
1039 }
1040
1041 pub fn set_denied_builtins(&mut self, denied: HashSet<String>) {
1044 self.denied_builtins = Arc::new(denied);
1045 }
1046
1047 pub fn set_source_info(&mut self, file: &str, text: &str) {
1049 self.source_file = Some(file.to_string());
1050 self.source_text = Some(text.to_string());
1051 if let Some(cov) = self.coverage.as_mut() {
1052 cov.set_primary_file(file);
1053 }
1054 Arc::make_mut(&mut self.source_cache)
1055 .insert(std::path::PathBuf::from(file), Arc::from(text));
1056 }
1057
1058 pub fn start(&mut self, chunk: &Chunk) -> Result<(), VmError> {
1060 self.ensure_execution_available()?;
1061 let debugger = self.debugger_attached();
1068 let initial_env = if debugger {
1069 Some(self.env.clone())
1070 } else {
1071 None
1072 };
1073 let initial_local_slots = if debugger {
1074 Some(Self::fresh_local_slots(chunk))
1075 } else {
1076 None
1077 };
1078 let chunk = Arc::new(chunk.clone());
1079 let local_slots = Self::fresh_local_slots(&chunk);
1080 let inline_cache_set = self.inline_cache_set_index_for_chunk(&chunk);
1081 self.frames.push(CallFrame {
1082 chunk,
1083 inline_cache_set,
1084 ip: 0,
1085 stack_base: self.stack.len(),
1086 saved_env: self.env.clone(),
1087 initial_env,
1088 initial_local_slots,
1089 saved_iterator_depth: self.iterators.len(),
1090 fn_name: String::new(),
1091 argc: 0,
1092 saved_source_dir: None,
1093 module_functions: None,
1094 module_state: None,
1095 local_slots,
1096 local_scope_base: self.env.scope_depth().saturating_sub(1),
1097 local_scope_depth: 0,
1098 });
1099 Ok(())
1100 }
1101
1102 pub(crate) fn child_vm(&self) -> Vm {
1105 Vm {
1106 stack: Vec::with_capacity(64),
1107 env: self.env.clone(),
1108 output: String::new(),
1109 builtins: Arc::clone(&self.builtins),
1110 async_builtins: Arc::clone(&self.async_builtins),
1111 capability_methods: Arc::clone(&self.capability_methods),
1112 builtin_metadata: Arc::clone(&self.builtin_metadata),
1113 builtins_by_id: Arc::clone(&self.builtins_by_id),
1114 builtin_id_collisions: Arc::clone(&self.builtin_id_collisions),
1115 iterators: Vec::new(),
1116 frames: Vec::new(),
1117 exception_handlers: Vec::new(),
1118 spawned_tasks: BTreeMap::new(),
1119 process_exit_request: Arc::clone(&self.process_exit_request),
1120 sync_runtime: self.sync_runtime.clone(),
1121 shared_state_runtime: self.shared_state_runtime.clone(),
1122 inline_cache_sets: Vec::new(),
1123 inline_cache_set_by_chunk: HashMap::new(),
1124 pool_registry: self.pool_registry.clone(),
1125 llm_mock_context: self.llm_mock_context.clone(),
1126 package_snapshot_registry: self.package_snapshot_registry.clone(),
1127 wait_for_graph: self.wait_for_graph.clone(),
1128 held_sync_guards: Vec::new(),
1129 inherited_held_keys: Arc::new(Vec::new()),
1130 task_scopes: Vec::new(),
1131 task_counter: 0,
1132 runtime_context_counter: self.runtime_context_counter,
1133 runtime_context: self.runtime_context.clone(),
1134 deadlines: self.deadlines.clone(),
1135 execution_deadline: self.execution_deadline.fork(),
1136 breakpoints: BTreeMap::new(),
1137 function_breakpoints: std::collections::BTreeSet::new(),
1138 pending_function_bp: None,
1139 step_mode: false,
1140 step_frame_depth: 0,
1141 stopped: false,
1142 last_line: 0,
1143 source_dir: self.source_dir.clone(),
1144 package_execution_guard: self.package_execution_guard.clone(),
1145 imported_paths: Vec::new(),
1146 deferred_cyclic_imports: Vec::new(),
1147 module_cache: Arc::clone(&self.module_cache),
1148 prepared_module_cache: self.prepared_module_cache.clone(),
1149 module_provenance: self.module_provenance,
1150 module_phase_recorder: self.module_phase_recorder.clone(),
1151 lazy_callable_modules: Arc::clone(&self.lazy_callable_modules),
1152 source_cache: Arc::clone(&self.source_cache),
1153 graph_link_table: self.graph_link_table.clone(),
1154 source_file: self.source_file.clone(),
1155 source_text: self.source_text.clone(),
1156 coverage: crate::coverage::for_primary(self.source_file.as_deref()),
1157 bridge: self.bridge.clone(),
1158 denied_builtins: Arc::clone(&self.denied_builtins),
1159 cancel_token: self.cancel_token.clone(),
1160 interrupt_signal_token: self.interrupt_signal_token.clone(),
1161 cancel_grace_instructions_remaining: None,
1162 interrupt_handlers: Vec::new(),
1163 next_interrupt_handle: 1,
1164 pending_interrupt_signal: None,
1165 interrupted: self.interrupted,
1166 dispatching_interrupt: false,
1167 interrupt_handler_deadline: None,
1168 error_stack_trace: Vec::new(),
1169 yield_sender: None,
1170 project_root: self.project_root.clone(),
1171 globals: Arc::clone(&self.globals),
1172 root_harness: self.root_harness.clone(),
1173 runtime_effects: crate::orchestration::RuntimeEffectState::with_shared_recorder(
1174 Arc::clone(&self.runtime_effects.recorder),
1175 ),
1176 debug_hook: None,
1177 runtime_limits: self.runtime_limits,
1178 }
1179 }
1180
1181 pub(crate) fn child_vm_for_host(&self) -> Vm {
1184 self.child_vm()
1185 }
1186
1187 pub(crate) fn request_process_exit(&self, code: i32) {
1188 self.process_exit_request.request(code);
1189 }
1190
1191 pub(crate) fn requested_process_exit(&self) -> Option<i32> {
1192 self.process_exit_request.code()
1193 }
1194
1195 pub(crate) fn cancel_spawned_tasks(&mut self) {
1199 for (_, task) in std::mem::take(&mut self.spawned_tasks) {
1200 task.cancel_token
1201 .store(true, std::sync::atomic::Ordering::SeqCst);
1202 task.handle.abort();
1203 }
1204 }
1205
1206 pub fn set_source_dir(&mut self, dir: &std::path::Path) {
1209 let dir = crate::stdlib::process::normalize_context_path(dir);
1210 self.source_dir = Some(dir.clone());
1211 crate::stdlib::set_thread_source_dir(&dir);
1212 if self.project_root.is_none() {
1214 self.project_root = crate::stdlib::process::find_project_root(&dir);
1215 }
1216 }
1217
1218 pub fn set_project_root(&mut self, root: &std::path::Path) {
1221 self.project_root = Some(root.to_path_buf());
1222 }
1223
1224 pub(crate) fn explicit_project_root(&self) -> Option<&std::path::Path> {
1227 self.project_root.as_deref()
1228 }
1229
1230 pub fn project_root(&self) -> Option<&std::path::Path> {
1232 self.project_root.as_deref().or(self.source_dir.as_deref())
1233 }
1234
1235 pub fn set_global(&mut self, name: &str, value: VmValue) {
1238 Arc::make_mut(&mut self.globals).insert(crate::value::intern_key(name), value);
1239 }
1240
1241 pub fn global(&self, name: &str) -> Option<&VmValue> {
1243 self.globals.get(name)
1244 }
1245
1246 pub fn set_harness(&mut self, harness: crate::harness::Harness) {
1250 self.root_harness = Some(harness.into_vm_value());
1251 }
1252
1253 pub(crate) fn harness(&self) -> Option<&crate::harness::VmHarness> {
1254 match self.root_harness.as_ref() {
1255 Some(VmValue::Harness(handle)) => Some(handle),
1256 _ => None,
1257 }
1258 }
1259
1260 pub fn root_harness_value(&self) -> Option<VmValue> {
1263 self.root_harness.clone()
1264 }
1265
1266 pub fn output(&self) -> &str {
1268 &self.output
1269 }
1270
1271 pub fn take_output(&mut self) -> String {
1275 std::mem::take(&mut self.output)
1276 }
1277
1278 pub fn append_output(&mut self, text: &str) {
1282 self.output.push_str(text);
1283 }
1284
1285 pub(crate) fn pop(&mut self) -> Result<VmValue, VmError> {
1286 self.stack.pop().ok_or(VmError::StackUnderflow)
1287 }
1288
1289 pub(crate) fn peek(&self) -> Result<&VmValue, VmError> {
1290 self.stack.last().ok_or(VmError::StackUnderflow)
1291 }
1292
1293 pub(crate) fn const_str(c: &Constant) -> Result<&str, VmError> {
1294 match c {
1295 Constant::String(s) => Ok(s.as_str()),
1296 _ => Err(VmError::TypeError("expected string constant".into())),
1297 }
1298 }
1299
1300 pub(crate) fn release_sync_guards_for_current_scope(&mut self) {
1301 let depth = self.env.scope_depth();
1302 self.held_sync_guards
1303 .retain(|guard| guard.env_scope_depth < depth);
1304 self.cancel_task_scopes_where(|s| s.env_scope_depth >= depth);
1307 }
1308
1309 pub(crate) fn release_sync_guards_after_unwind(
1310 &mut self,
1311 frame_depth: usize,
1312 env_scope_depth: usize,
1313 ) {
1314 self.held_sync_guards.retain(|guard| {
1315 guard.frame_depth <= frame_depth && guard.env_scope_depth <= env_scope_depth
1316 });
1317 self.cancel_task_scopes_where(|s| {
1320 !(s.frame_depth <= frame_depth && s.env_scope_depth <= env_scope_depth)
1321 });
1322 }
1323
1324 pub(crate) fn release_sync_guards_for_frame(&mut self, frame_depth: usize) {
1325 self.held_sync_guards
1326 .retain(|guard| guard.frame_depth != frame_depth);
1327 self.cancel_task_scopes_where(|s| s.frame_depth == frame_depth);
1330 }
1331
1332 pub(crate) fn adopt_sync_permit_for_current_scope(
1333 &mut self,
1334 permit: crate::value::VmSyncPermitHandle,
1335 ) {
1336 if permit.is_released()
1337 || self
1338 .held_sync_guards
1339 .iter()
1340 .any(|guard| guard._permit.same_lease(&permit))
1341 {
1342 return;
1343 }
1344 self.held_sync_guards
1345 .push(crate::synchronization::VmSyncHeldGuard {
1346 _permit: permit,
1347 frame_depth: self.frames.len(),
1348 env_scope_depth: self.env.scope_depth(),
1349 });
1350 }
1351
1352 pub(crate) fn deregister_task_from_scopes(&mut self, id: &str) {
1355 for scope in &mut self.task_scopes {
1356 scope.task_ids.retain(|t| t != id);
1357 }
1358 }
1359
1360 fn cancel_task_scopes_where<F: Fn(&TaskScope) -> bool>(&mut self, doomed: F) {
1363 let mut i = 0;
1364 while i < self.task_scopes.len() {
1365 if doomed(&self.task_scopes[i]) {
1366 let scope = self.task_scopes.remove(i);
1367 for id in &scope.task_ids {
1368 if let Some(task) = self.spawned_tasks.remove(id) {
1369 task.cancel_token
1370 .store(true, std::sync::atomic::Ordering::SeqCst);
1371 task.handle.abort();
1372 }
1373 }
1374 } else {
1375 i += 1;
1376 }
1377 }
1378 }
1379
1380 pub(crate) fn held_permits_for(&self, kind: &str, key: &str) -> u32 {
1384 let own: u32 = self
1385 .held_sync_guards
1386 .iter()
1387 .filter(|guard| {
1388 !guard._permit.is_released()
1389 && guard._permit.kind() == kind
1390 && guard._permit.key() == key
1391 })
1392 .map(|guard| guard._permit.permits())
1393 .sum();
1394 let inherited: u32 = self
1395 .inherited_held_keys
1396 .iter()
1397 .filter(|held| held.kind == kind && held.key == key)
1398 .map(|held| held.permits)
1399 .sum();
1400 own + inherited
1401 }
1402
1403 pub(crate) fn combined_held_keys(&self) -> Vec<crate::synchronization::VmSyncHeldKey> {
1406 let mut keys: Vec<crate::synchronization::VmSyncHeldKey> = self
1407 .held_sync_guards
1408 .iter()
1409 .filter_map(|guard| crate::synchronization::VmSyncHeldKey::from_permit(&guard._permit))
1410 .collect();
1411 keys.extend(self.inherited_held_keys.iter().cloned());
1412 keys
1413 }
1414
1415 pub(crate) fn child_vm_inline(&self) -> Vm {
1421 let mut child = self.child_vm();
1422 child.inherited_held_keys = Arc::new(self.combined_held_keys());
1423 child
1424 }
1425}
1426
1427impl Drop for Vm {
1428 fn drop(&mut self) {
1429 if let Some(coverage) = self.coverage.take() {
1430 crate::coverage::merge_into_global(coverage);
1431 }
1432 self.cancel_spawned_tasks();
1433 }
1434}
1435
1436impl Default for Vm {
1437 fn default() -> Self {
1438 Self::new()
1439 }
1440}
1441
1442#[cfg(test)]
1443#[path = "state_tests.rs"]
1444mod tests;