Skip to main content

harn_vm/vm/
state.rs

1use std::collections::{BTreeMap, HashMap, HashSet};
2use std::path::PathBuf;
3use std::sync::atomic::{AtomicBool, 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;
16pub(crate) use super::execution_deadline::{ExecutionDeadlinePauseGuard, ExecutionDeadlineState};
17use super::modules::ModuleCache;
18use super::VmBuiltinMetadata;
19
20/// A lazy callable's resolved export set together with the module graph that
21/// was loaded to produce it.
22///
23/// The exported closures — and every function they can transitively reach
24/// through imports — hold only `Weak`s into their home module's function
25/// registry and module state. The child VM that first loaded the graph is
26/// normally their sole strong owner (via its `module_cache`), and it dies once
27/// the hook fire completes. A later child VM that hits this cache never
28/// re-imports the graph, so a transitively imported callee's sibling `pub fn`
29/// would fall through name resolution to host-bridge dispatch (`Undefined
30/// builtin`). Retaining the complete loaded graph here for the cache entry's
31/// lifetime keeps every transitively reachable registry and module state
32/// upgradeable at call time, for every fire — not just the first.
33pub(crate) struct ResolvedLazyCallable {
34    pub(crate) exports: BTreeMap<String, Arc<VmClosure>>,
35    /// Intentionally unread: retained solely to keep the loaded module graph's
36    /// function registries and module states alive for this cache entry (the
37    /// same liveness role [`crate::value::RetainedModuleScope`] plays for a
38    /// single retained closure, generalized across the whole import graph).
39    #[allow(dead_code)]
40    pub(crate) retained_module_graph: ModuleCache,
41}
42
43pub(crate) type LazyCallableResolution = Arc<ResolvedLazyCallable>;
44pub(crate) struct LazyCallableCacheSlot {
45    pub(crate) execution_guard: Option<Arc<harn_modules::package_execution::PackageExecutionGuard>>,
46    pub(crate) resolution: Arc<tokio::sync::OnceCell<LazyCallableResolution>>,
47}
48pub(crate) type LazyCallableModuleCache =
49    Arc<VmMutex<BTreeMap<PathBuf, Vec<LazyCallableCacheSlot>>>>;
50
51/// RAII guard that starts a tracing span on creation and ends it on drop.
52pub(crate) struct ScopeSpan(u64);
53
54impl ScopeSpan {
55    pub(crate) fn new(kind: crate::tracing::SpanKind, name: String) -> Self {
56        Self(crate::tracing::span_start(kind, name))
57    }
58}
59
60impl Drop for ScopeSpan {
61    fn drop(&mut self) {
62        crate::tracing::span_end(self.0);
63    }
64}
65
66#[derive(Clone)]
67pub(crate) struct LocalSlot {
68    pub(crate) value: VmValue,
69    pub(crate) initialized: bool,
70    pub(crate) synced: bool,
71}
72
73impl Drop for LocalSlot {
74    fn drop(&mut self) {
75        // Slot locals hold script values directly (e.g. a `let` bound to a
76        // deeply nested list). When a frame is torn down, the default
77        // recursive drop of such a value would overflow the native stack and
78        // abort the process. For the overwhelmingly common scalar slot this is
79        // a single `matches!` check and then the normal trivial drop; only a
80        // nested container is moved out and torn down iteratively, so hot
81        // frame teardown is unaffected.
82        if crate::value::recursion::is_recursive_container(&self.value) {
83            crate::value::recursion::dismantle(std::mem::replace(&mut self.value, VmValue::Nil));
84        }
85    }
86}
87
88#[derive(Clone)]
89pub(crate) struct InterruptHandler {
90    pub(crate) handle: i64,
91    pub(crate) signals: Vec<String>,
92    pub(crate) once: bool,
93    pub(crate) graceful_timeout_ms: Option<u64>,
94    pub(crate) handler: VmValue,
95}
96
97/// Call frame for function execution.
98pub(crate) struct CallFrame {
99    pub(crate) chunk: ChunkRef,
100    /// VM-local inline-cache set for this frame's chunk. Computed once at
101    /// frame entry so hot opcode dispatch can index cache feedback directly
102    /// instead of hashing the chunk id on every cached opcode.
103    pub(crate) inline_cache_set: usize,
104    pub(crate) ip: usize,
105    pub(crate) stack_base: usize,
106    pub(crate) saved_env: VmEnv,
107    /// Env snapshot captured at call-time, *after* argument binding. Used
108    /// by the debugger's `restartFrame` to rewind this frame to its
109    /// entry state (re-binding args from the original values) without
110    /// re-entering the call site. Cheap to clone because `VmEnv` is
111    /// already cloned into `saved_env` on every call. `None` for
112    /// scratch frames (evaluate, import init) where restart isn't
113    /// meaningful.
114    pub(crate) initial_env: Option<VmEnv>,
115    pub(crate) initial_local_slots: Option<Vec<LocalSlot>>,
116    /// Iterator stack depth to restore when this frame unwinds.
117    pub(crate) saved_iterator_depth: usize,
118    /// Function name for stack traces (empty for top-level pipeline); a
119    /// refcount-bump share of [`CompiledFunction::name`].
120    pub(crate) fn_name: crate::value::HarnStr,
121    /// Number of arguments actually passed by the caller (for default arg support).
122    pub(crate) argc: usize,
123    /// Saved VM_SOURCE_DIR to restore when this frame is popped.
124    /// Set when entering a closure that originated from an imported module.
125    pub(crate) saved_source_dir: Option<std::path::PathBuf>,
126    /// Module-local named functions available to symbolic calls within this frame.
127    pub(crate) module_functions: Option<ModuleFunctionRegistry>,
128    /// Shared module-level env for top-level `let` / `const` bindings of
129    /// this frame's originating module. Looked up after `self.env` and
130    /// before `self.globals` by `GetVar` / `SetVar`, giving each module
131    /// its own live static state that persists across calls. See the
132    /// `module_state` field on `VmClosure` for the full rationale.
133    pub(crate) module_state: Option<crate::value::ModuleState>,
134    /// Slot-indexed locals for compiler-resolved names in this frame.
135    pub(crate) local_slots: Vec<LocalSlot>,
136    /// Env scope index that corresponds to compiler local scope depth 0.
137    pub(crate) local_scope_base: usize,
138    /// Current compiler local scope depth, updated by PushScope/PopScope.
139    pub(crate) local_scope_depth: usize,
140}
141
142pub(crate) struct InlineCacheSite {
143    pub(crate) cache_set: usize,
144    pub(crate) slot_count: usize,
145    pub(crate) slot: Option<usize>,
146}
147
148impl CallFrame {
149    #[inline]
150    pub(crate) fn inline_cache_site_for_previous_op(&self) -> InlineCacheSite {
151        let op_offset = self.ip.saturating_sub(1);
152        InlineCacheSite {
153            cache_set: self.inline_cache_set,
154            slot_count: self.chunk.inline_cache_slot_count(),
155            slot: self.chunk.inline_cache_slot(op_offset),
156        }
157    }
158}
159
160/// Exception handler for try/catch.
161pub(crate) struct ExceptionHandler {
162    pub(crate) catch_ip: usize,
163    pub(crate) stack_depth: usize,
164    pub(crate) frame_depth: usize,
165    pub(crate) env_scope_depth: usize,
166    /// When present, this catch only handles errors whose enum_name matches.
167    pub(crate) error_type: Option<crate::value::HarnStr>,
168}
169
170/// A structured-concurrency nursery (`scope { }`). Tasks spawned while this
171/// scope is innermost record their id here; `TaskScopeExit` joins them.
172pub(crate) struct TaskScope {
173    /// Ids of tasks spawned in this scope that have not been explicitly
174    /// `await`ed away. Joined (normal exit) or cancelled (unwind) on close.
175    pub(crate) task_ids: Vec<String>,
176    /// Frame depth at which the scope was opened, for unwind pruning.
177    pub(crate) frame_depth: usize,
178    /// Env scope depth at open, for unwind pruning.
179    pub(crate) env_scope_depth: usize,
180}
181
182/// Terminal exit requested by any VM in one execution tree. A process exit is
183/// global control flow, so child VMs share this latch with their parent rather
184/// than relying on a task being explicitly awaited.
185pub(crate) struct ProcessExitRequest {
186    code: Mutex<Option<i32>>,
187    requested: AtomicBool,
188}
189
190impl ProcessExitRequest {
191    fn new() -> Self {
192        Self {
193            code: Mutex::new(None),
194            requested: AtomicBool::new(false),
195        }
196    }
197
198    fn request(&self, code: i32) {
199        let mut recorded = self
200            .code
201            .lock()
202            .expect("process exit request lock poisoned");
203        if recorded.is_none() {
204            *recorded = Some(code);
205            self.requested.store(true, Ordering::Release);
206        }
207    }
208
209    fn code(&self) -> Option<i32> {
210        if !self.requested.load(Ordering::Acquire) {
211            return None;
212        }
213        *self
214            .code
215            .lock()
216            .expect("process exit request lock poisoned")
217    }
218}
219
220/// Iterator state for for-in loops.
221pub(crate) enum IterState {
222    Vec {
223        items: Arc<Vec<VmValue>>,
224        idx: usize,
225    },
226    Dict {
227        entries: Arc<crate::value::DictMap>,
228        keys: Vec<crate::value::HarnStr>,
229        idx: usize,
230    },
231    Channel {
232        receiver: std::sync::Arc<tokio::sync::Mutex<tokio::sync::mpsc::Receiver<VmValue>>>,
233        close: std::sync::Arc<crate::value::VmChannelCloseState>,
234    },
235    Generator {
236        gen: Arc<crate::value::VmGenerator>,
237    },
238    Stream {
239        stream: Arc<crate::value::VmStream>,
240    },
241    /// Step through a lazy range without materializing a Vec.
242    /// Inclusive ranges keep `end` as an actual value so `i64::MAX to i64::MAX`
243    /// still yields one item instead of overflowing a one-past-end sentinel.
244    Range {
245        next: i64,
246        end: i64,
247        inclusive: bool,
248        done: bool,
249    },
250    VmIter {
251        handle: crate::vm::iter::VmIterHandle,
252    },
253}
254
255#[derive(Clone)]
256pub(crate) enum VmBuiltinDispatch {
257    Sync(VmBuiltinFn),
258    Async(VmAsyncBuiltinFn),
259}
260
261#[derive(Clone)]
262pub(crate) struct VmBuiltinEntry {
263    pub(crate) name: Arc<str>,
264    pub(crate) dispatch: VmBuiltinDispatch,
265    /// Static effect specs for source-visible capability/privileged builtins.
266    /// Keeping this beside the numeric dispatch entry makes pure direct calls
267    /// pay one `Option` branch instead of consulting the manifest by name.
268    pub(crate) recorded_effects: Option<&'static [harn_builtin_meta::EffectSpec]>,
269}
270
271/// The Harn bytecode virtual machine.
272pub struct Vm {
273    pub(crate) stack: Vec<VmValue>,
274    pub(crate) env: VmEnv,
275    pub(crate) output: String,
276    pub(crate) builtins: Arc<BTreeMap<String, VmBuiltinFn>>,
277    pub(crate) async_builtins: Arc<BTreeMap<String, VmAsyncBuiltinFn>>,
278    /// Host-supplied implementations keyed by the typed capability surface
279    /// they implement. Unlike ordinary builtins these names never enter the
280    /// source namespace: they are reachable only through a `Harness*` value.
281    pub(crate) capability_methods:
282        Arc<BTreeMap<harn_builtin_meta::CapabilityId, BTreeMap<String, VmBuiltinDispatch>>>,
283    pub(crate) builtin_metadata: Arc<BTreeMap<String, VmBuiltinMetadata>>,
284    /// Numeric side index for builtins. Name-keyed maps remain authoritative;
285    /// this index is the hot path for direct builtin bytecode and callback refs.
286    pub(crate) builtins_by_id: Arc<HashMap<BuiltinId, VmBuiltinEntry>>,
287    /// IDs with detected name collisions. Collided names safely fall back to
288    /// the authoritative name-keyed lookup path.
289    pub(crate) builtin_id_collisions: Arc<HashSet<BuiltinId>>,
290    /// Iterator state for for-in loops.
291    pub(crate) iterators: Vec<IterState>,
292    /// Call frame stack.
293    pub(crate) frames: Vec<CallFrame>,
294    /// Exception handler stack.
295    pub(crate) exception_handlers: Vec<ExceptionHandler>,
296    /// Spawned async task handles.
297    pub(crate) spawned_tasks: BTreeMap<String, VmTaskHandle>,
298    /// Shared terminal process-exit latch for this execution tree.
299    pub(crate) process_exit_request: Arc<ProcessExitRequest>,
300    /// Shared process-local synchronization primitives inherited by child VMs.
301    pub(crate) sync_runtime: Arc<crate::synchronization::VmSyncRuntime>,
302    /// Shared process-local cells, maps, and mailboxes inherited by child VMs.
303    pub(crate) shared_state_runtime: Arc<crate::shared_state::VmSharedStateRuntime>,
304    /// Per-isolate inline cache entries. `inline_cache_set_by_chunk` maps a
305    /// compiled chunk identity to an index in this vector at frame entry; the
306    /// dispatch loop uses the frame-local index for per-op reads/writes.
307    pub(crate) inline_cache_sets: Vec<Vec<crate::chunk::InlineCacheEntry>>,
308    pub(crate) inline_cache_set_by_chunk: HashMap<u64, usize>,
309    /// VM-scoped pool registry inherited by child VMs and scoped into Tokio tasks.
310    pub(crate) pool_registry: Arc<crate::stdlib::pool::PoolRegistry>,
311    /// Inline LLM fixtures and observations shared by this VM execution tree.
312    pub(crate) llm_mock_context: crate::llm::mock::LlmMockContext,
313    /// Reader leases opened by package_snapshot_open in this execution tree.
314    /// Child VMs share the registry; the final VM drop releases abandoned
315    /// leases without touching concurrent executions.
316    pub(crate) package_snapshot_registry: Arc<crate::stdlib::PackageSnapshotRegistry>,
317    /// Shared task/channel wait graph for this VM execution tree.
318    pub(crate) wait_for_graph: Arc<crate::wait_for_graph::VmWaitForGraph>,
319    /// Permits acquired by lexical synchronization blocks in this VM.
320    pub(crate) held_sync_guards: Vec<crate::synchronization::VmSyncHeldGuard>,
321    /// Locks held by an ancestor VM that is *suspended on this VM's execution*:
322    /// an inline async-builtin child runs while its parent is parked
323    /// mid-instruction still holding these permits. Re-acquiring more permits
324    /// than the primitive can grant is a provably-unresolvable self-deadlock, so
325    /// HARN-ORC-011 fires across the child boundary. Empty for new concurrent
326    /// tasks (`spawn`/`parallel`/triggers), where the parent keeps running and
327    /// blocking can be legitimately resolvable.
328    pub(crate) inherited_held_keys: Arc<Vec<crate::synchronization::VmSyncHeldKey>>,
329    /// Structured-concurrency nursery stack. Each `scope { }` block pushes a
330    /// `TaskScope`; tasks spawned while it is innermost register their id here.
331    /// On normal exit (`TaskScopeExit`) the scope's tasks are joined and the
332    /// first error propagates; on unwind they are cancelled. Modeled on
333    /// `held_sync_guards` (push on enter, prune/cancel on frame/handler exit).
334    pub(crate) task_scopes: Vec<TaskScope>,
335    /// Counter for generating unique task IDs.
336    pub(crate) task_counter: u64,
337    /// Counter for logical runtime-context task groups.
338    pub(crate) runtime_context_counter: u64,
339    /// Logical runtime task context visible through `runtime_context()`.
340    pub(crate) runtime_context: crate::runtime_context::RuntimeContext,
341    /// Active deadline stack: (deadline_instant, frame_depth).
342    pub(crate) deadlines: Vec<(Instant, usize)>,
343    /// Uncatchable wall-clock deadline imposed by the embedding host.
344    pub(crate) execution_deadline: Arc<ExecutionDeadlineState>,
345    /// Breakpoints, keyed by source-file path so a breakpoint at line N
346    /// in `auto.harn` doesn't also fire when execution hits line N in an
347    /// imported lib. The empty-string key is a wildcard used by callers
348    /// that don't track source paths (legacy `set_breakpoints` API).
349    pub(crate) breakpoints: BTreeMap<String, std::collections::BTreeSet<usize>>,
350    /// Function-name breakpoints. Any closure call whose
351    /// `CompiledFunction.name` matches an entry here raises a stop on
352    /// entry, regardless of the call site's file or line. Lets the IDE
353    /// break on `llm_call` / `host_run_pipeline` / any user pipeline
354    /// function without pinning down a source location first.
355    pub(crate) function_breakpoints: std::collections::BTreeSet<String>,
356    /// Latched on `push_closure_frame` when the callee's name matches
357    /// `function_breakpoints`; consumed by the next step so the stop is
358    /// reported with reason="function breakpoint" and the breakpoint
359    /// name available for the DAP `stopped` event.
360    pub(crate) pending_function_bp: Option<String>,
361    /// Whether the VM is in step mode.
362    pub(crate) step_mode: bool,
363    /// The frame depth at which stepping started (for step-over).
364    pub(crate) step_frame_depth: usize,
365    /// Whether the VM is currently stopped at a debug point.
366    pub(crate) stopped: bool,
367    /// Last source line executed (to detect line changes).
368    pub(crate) last_line: usize,
369    /// Source directory for resolving imports.
370    pub(crate) source_dir: Option<std::path::PathBuf>,
371    /// Installed-package identity retained for the active lazy pipeline.
372    pub(crate) package_execution_guard:
373        Option<Arc<harn_modules::package_execution::PackageExecutionGuard>>,
374    /// Modules currently being imported (cycle prevention).
375    pub(crate) imported_paths: Vec<std::path::PathBuf>,
376    /// Imports that hit an in-progress module (an import cycle) and so could
377    /// not bind inline. Drained by `flush_deferred_cyclic_imports` once the
378    /// involved modules finish loading.
379    pub(crate) deferred_cyclic_imports: Vec<super::modules::DeferredCyclicImport>,
380    /// Loaded module cache keyed by canonical or synthetic module path.
381    pub(crate) module_cache: ModuleCache,
382    /// Immutable hydrated module bytecode shared across fresh VM isolates.
383    /// Runtime closures, registries, state, and init execution are not cached.
384    pub(crate) prepared_module_cache: crate::PreparedModuleCache,
385    /// Authority provenance used for the next root module graph. This can only
386    /// move from ordinary user code to trusted host dispatch before loading.
387    pub(crate) module_provenance: crate::module_artifact::ModuleProvenance,
388    /// Optional timing recorder shared by this VM execution tree.
389    pub(crate) module_phase_recorder: Option<super::ModulePhaseRecorder>,
390    /// Lazy manifest modules initialized by any child in this execution tree.
391    /// The complete export set is retained so handlers and predicates from the
392    /// same module share one module state across repeated child invocations.
393    pub(crate) lazy_callable_modules: LazyCallableModuleCache,
394    /// Source text keyed by canonical or synthetic module path for debugger
395    /// retrieval. Entries share the bytes owned by
396    /// [`crate::module_source`] rather than copying them.
397    pub(crate) source_cache: Arc<BTreeMap<std::path::PathBuf, Arc<str>>>,
398    /// This spawn's import-graph link table, when the entry-chunk cache proved
399    /// one current. Lets module loading name a module's artifact from a recorded
400    /// digest instead of reading the file to recompute the same digest. `None`
401    /// means every module takes the ordinary read-and-hash path.
402    pub(crate) graph_link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
403    /// Verified, closed-program module templates for this execution tree.
404    /// Unlike the generic graph link table, absence is never interpreted as a
405    /// source-cache miss by the linked-program path.
406    pub(crate) linked_program_repository:
407        Option<Arc<crate::linked_program::LinkedProgramRepository>>,
408    /// Source file path for error reporting.
409    pub(crate) source_file: Option<String>,
410    /// Source text for error reporting.
411    pub(crate) source_text: Option<String>,
412    /// Line-coverage accumulator. `Some` only while a coverage session is
413    /// active (see [`crate::coverage`]); folded into the global report on drop.
414    pub(crate) coverage: Option<crate::coverage::Coverage>,
415    /// Optional bridge for delegating unknown builtins in bridge mode.
416    pub(crate) bridge: Option<Arc<crate::bridge::HostBridge>>,
417    /// Builtins denied by sandbox mode (`--deny` / `--allow` flags).
418    pub(crate) denied_builtins: Arc<HashSet<String>>,
419    /// Cancellation token for cooperative graceful shutdown (set by parent).
420    pub(crate) cancel_token: Option<std::sync::Arc<std::sync::atomic::AtomicBool>>,
421    pub(crate) interrupt_signal_token: Option<std::sync::Arc<std::sync::Mutex<Option<String>>>>,
422    /// Remaining instruction-boundary checks before a requested host
423    /// cancellation is forcefully raised. This gives `is_cancelled()` loops a
424    /// deterministic chance to return cleanly without letting non-cooperative
425    /// CPU-bound code run forever.
426    pub(crate) cancel_grace_instructions_remaining: Option<usize>,
427    /// User-visible interrupt handlers registered through `std/signal`.
428    pub(crate) interrupt_handlers: Vec<InterruptHandler>,
429    pub(crate) next_interrupt_handle: i64,
430    pub(crate) pending_interrupt_signal: Option<String>,
431    pub(crate) interrupted: bool,
432    pub(crate) dispatching_interrupt: bool,
433    pub(crate) interrupt_handler_deadline: Option<Instant>,
434    /// Captured stack trace from the most recent error (fn_name, line, col).
435    pub(crate) error_stack_trace: Vec<(String, usize, usize, Option<String>)>,
436    /// Yield channel sender for generator execution. When set, `Op::Yield`
437    /// sends values through this channel instead of being a no-op.
438    pub(crate) yield_sender: Option<tokio::sync::mpsc::Sender<Result<VmValue, VmError>>>,
439    /// Project root directory (detected via harn.toml).
440    /// Used as base directory for metadata, store, and checkpoint operations.
441    pub(crate) project_root: Option<std::path::PathBuf>,
442    /// Global constants (e.g. `pi`, `e`). Checked as a fallback in `GetVar`
443    /// after the environment, so user-defined variables can shadow them.
444    pub(crate) globals: Arc<crate::value::DictMap>,
445    /// Root authority injected by the embedder for the compiler-generated
446    /// `main(harness: Harness)` call. This is deliberately not an environment
447    /// binding or global: user code can receive it only through an explicit
448    /// parameter.
449    pub(crate) root_harness: Option<VmValue>,
450    /// Runtime effect receipt state (shared recorder + per-VM call memo).
451    pub(crate) runtime_effects: crate::orchestration::RuntimeEffectState,
452    /// Optional debugger hook invoked when execution advances to a new source line.
453    pub(crate) debug_hook: Option<parking_lot::Mutex<Box<DebugHook>>>,
454    /// Effective runtime ceilings for this VM execution.
455    pub(crate) runtime_limits: RuntimeLimits,
456}
457
458/// Reusable VM baseline for hosts that need many clean executions with the
459/// same stable builtin/source setup.
460///
461/// The baseline intentionally does not snapshot execution state. Each
462/// instantiation gets fresh stacks, frames, tasks, cancellation fields, sync
463/// primitives, shared cells/maps/mailboxes, and debug state. Builtin tables are
464/// shared through `Arc` until a per-execution rebind needs copy-on-write.
465#[derive(Clone)]
466pub struct VmBaseline {
467    builtins: Arc<BTreeMap<String, VmBuiltinFn>>,
468    async_builtins: Arc<BTreeMap<String, VmAsyncBuiltinFn>>,
469    capability_methods:
470        Arc<BTreeMap<harn_builtin_meta::CapabilityId, BTreeMap<String, VmBuiltinDispatch>>>,
471    builtin_metadata: Arc<BTreeMap<String, VmBuiltinMetadata>>,
472    builtins_by_id: Arc<HashMap<BuiltinId, VmBuiltinEntry>>,
473    builtin_id_collisions: Arc<HashSet<BuiltinId>>,
474    source_dir: Option<std::path::PathBuf>,
475    source_file: Option<String>,
476    source_text: Option<String>,
477    project_root: Option<std::path::PathBuf>,
478    globals: Arc<crate::value::DictMap>,
479    root_harness: Option<VmValue>,
480    denied_builtins: Arc<HashSet<String>>,
481    prepared_module_cache: crate::PreparedModuleCache,
482    module_provenance: crate::module_artifact::ModuleProvenance,
483    /// Carried so a VM rebuilt from this baseline keeps resolving modules
484    /// through the same link table the original spawn was handed.
485    graph_link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
486    linked_program_repository: Option<Arc<crate::linked_program::LinkedProgramRepository>>,
487    runtime_limits: RuntimeLimits,
488}
489
490impl VmBaseline {
491    pub fn from_vm(vm: &Vm) -> Self {
492        Self {
493            builtins: Arc::clone(&vm.builtins),
494            async_builtins: Arc::clone(&vm.async_builtins),
495            capability_methods: Arc::clone(&vm.capability_methods),
496            builtin_metadata: Arc::clone(&vm.builtin_metadata),
497            builtins_by_id: Arc::clone(&vm.builtins_by_id),
498            builtin_id_collisions: Arc::clone(&vm.builtin_id_collisions),
499            source_dir: vm.source_dir.clone(),
500            source_file: vm.source_file.clone(),
501            source_text: vm.source_text.clone(),
502            project_root: vm.project_root.clone(),
503            globals: Arc::clone(&vm.globals),
504            root_harness: vm.root_harness.clone(),
505            denied_builtins: Arc::clone(&vm.denied_builtins),
506            prepared_module_cache: vm.prepared_module_cache.clone(),
507            module_provenance: vm.module_provenance,
508            graph_link_table: vm.graph_link_table.clone(),
509            linked_program_repository: vm.linked_program_repository.clone(),
510            runtime_limits: vm.runtime_limits,
511        }
512    }
513
514    pub fn instantiate(&self) -> Vm {
515        crate::initialize_runtime_assets();
516        let mut source_cache = BTreeMap::new();
517        if let (Some(file), Some(text)) = (&self.source_file, &self.source_text) {
518            source_cache.insert(std::path::PathBuf::from(file), Arc::from(text.as_str()));
519        }
520        if let Some(dir) = &self.source_dir {
521            crate::stdlib::set_thread_source_dir(dir);
522        }
523
524        let mut vm = Vm {
525            stack: Vec::with_capacity(256),
526            env: VmEnv::new(),
527            output: String::new(),
528            builtins: Arc::clone(&self.builtins),
529            async_builtins: Arc::clone(&self.async_builtins),
530            capability_methods: Arc::clone(&self.capability_methods),
531            builtin_metadata: Arc::clone(&self.builtin_metadata),
532            builtins_by_id: Arc::clone(&self.builtins_by_id),
533            builtin_id_collisions: Arc::clone(&self.builtin_id_collisions),
534            iterators: Vec::new(),
535            frames: Vec::new(),
536            exception_handlers: Vec::new(),
537            spawned_tasks: BTreeMap::new(),
538            process_exit_request: Arc::new(ProcessExitRequest::new()),
539            sync_runtime: Arc::new(crate::synchronization::VmSyncRuntime::new()),
540            shared_state_runtime: Arc::new(crate::shared_state::VmSharedStateRuntime::new()),
541            inline_cache_sets: Vec::new(),
542            inline_cache_set_by_chunk: HashMap::new(),
543            pool_registry: crate::stdlib::pool::new_pool_registry(),
544            llm_mock_context: crate::llm::mock::LlmMockContext::for_new_vm(),
545            package_snapshot_registry: Arc::new(Default::default()),
546            wait_for_graph: Arc::new(crate::wait_for_graph::VmWaitForGraph::new()),
547            held_sync_guards: Vec::new(),
548            inherited_held_keys: Arc::new(Vec::new()),
549            task_scopes: Vec::new(),
550            task_counter: 0,
551            runtime_context_counter: 0,
552            runtime_context: crate::runtime_context::RuntimeContext::root(),
553            deadlines: Vec::new(),
554            execution_deadline: super::execution::new_execution_deadline_state(None),
555            breakpoints: BTreeMap::new(),
556            function_breakpoints: std::collections::BTreeSet::new(),
557            pending_function_bp: None,
558            step_mode: false,
559            step_frame_depth: 0,
560            stopped: false,
561            last_line: 0,
562            source_dir: self.source_dir.clone(),
563            package_execution_guard: None,
564            imported_paths: Vec::new(),
565            deferred_cyclic_imports: Vec::new(),
566            module_cache: Arc::new(BTreeMap::new()),
567            prepared_module_cache: self.prepared_module_cache.clone(),
568            module_provenance: self.module_provenance,
569            module_phase_recorder: None,
570            lazy_callable_modules: Arc::new(crate::value::VmMutex::new(BTreeMap::new())),
571            source_cache: Arc::new(source_cache),
572            graph_link_table: self.graph_link_table.clone(),
573            linked_program_repository: self.linked_program_repository.clone(),
574            source_file: self.source_file.clone(),
575            source_text: self.source_text.clone(),
576            coverage: crate::coverage::for_primary(self.source_file.as_deref()),
577            bridge: None,
578            denied_builtins: Arc::clone(&self.denied_builtins),
579            cancel_token: None,
580            interrupt_signal_token: None,
581            cancel_grace_instructions_remaining: None,
582            interrupt_handlers: Vec::new(),
583            next_interrupt_handle: 1,
584            pending_interrupt_signal: None,
585            interrupted: false,
586            dispatching_interrupt: false,
587            interrupt_handler_deadline: None,
588            error_stack_trace: Vec::new(),
589            yield_sender: None,
590            project_root: self.project_root.clone(),
591            globals: Arc::clone(&self.globals),
592            root_harness: self.root_harness.clone(),
593            runtime_effects: crate::orchestration::RuntimeEffectState::fresh(),
594            debug_hook: None,
595            runtime_limits: self.runtime_limits,
596        };
597
598        crate::stdlib::rebind_execution_state_builtins(&mut vm);
599        vm
600    }
601}
602
603impl Vm {
604    pub(crate) fn ensure_execution_available(&self) -> Result<(), VmError> {
605        if self.execution_deadline.is_abandoned() {
606            return Err(VmError::AbandonedExecution);
607        }
608        Ok(())
609    }
610
611    pub(crate) fn fresh_local_slots(chunk: &Chunk) -> Vec<LocalSlot> {
612        chunk
613            .local_slots
614            .iter()
615            .map(|_| LocalSlot {
616                value: VmValue::Nil,
617                initialized: false,
618                synced: false,
619            })
620            .collect()
621    }
622
623    pub(crate) fn bind_param_slots(
624        slots: &mut [LocalSlot],
625        func: &crate::chunk::CompiledFunction,
626        args: &[VmValue],
627        synced: bool,
628    ) {
629        Self::bind_param_slots_args(slots, func, &super::CallArgs::Slice(args), synced);
630    }
631
632    pub(crate) fn bind_param_slots_args(
633        slots: &mut [LocalSlot],
634        func: &crate::chunk::CompiledFunction,
635        args: &super::CallArgs<'_>,
636        synced: bool,
637    ) {
638        let param_count = func.params.len();
639        for (i, _param) in func.params.iter().enumerate() {
640            if i >= slots.len() {
641                break;
642            }
643            if func.has_rest_param && i == param_count - 1 {
644                let rest_args = args.to_vec_from(i);
645                slots[i].value = VmValue::List(std::sync::Arc::new(rest_args));
646                slots[i].initialized = true;
647                slots[i].synced = synced;
648            } else if let Some(arg) = args.get(i) {
649                slots[i].value = arg.clone();
650                slots[i].initialized = true;
651                slots[i].synced = synced;
652            }
653        }
654    }
655
656    pub(crate) fn visible_variables(&self) -> crate::value::DictMap {
657        let mut vars = self.env.all_variables();
658        let Some(frame) = self.frames.last() else {
659            return vars;
660        };
661        for (slot, info) in frame.local_slots.iter().zip(frame.chunk.local_slots.iter()) {
662            if slot.initialized && info.scope_depth <= frame.local_scope_depth {
663                vars.insert(crate::value::intern_key(&info.name), slot.value.clone());
664            }
665        }
666        vars
667    }
668
669    pub(crate) fn sync_current_frame_locals_to_env(&mut self) {
670        let frames = &mut self.frames;
671        let env = &mut self.env;
672        let Some(frame) = frames.last_mut() else {
673            return;
674        };
675        let local_scope_base = frame.local_scope_base;
676        let local_scope_depth = frame.local_scope_depth;
677        for (slot, info) in frame
678            .local_slots
679            .iter_mut()
680            .zip(frame.chunk.local_slots.iter())
681        {
682            if slot.initialized && !slot.synced && info.scope_depth <= local_scope_depth {
683                slot.synced = true;
684                let scope_idx = local_scope_base + info.scope_depth;
685                while env.scopes.len() <= scope_idx {
686                    env.push_scope();
687                }
688                // Slot-backed locals are never boxed captures (those resolve
689                // to `None` in the compiler and are defined straight into the
690                // env as `Cell`s), so syncing a slot always yields a `Value`.
691                Arc::make_mut(&mut env.scopes[scope_idx].vars).insert(
692                    info.name.clone(),
693                    crate::value::Binding::Value {
694                        value: slot.value.clone(),
695                        mutable: info.mutable,
696                    },
697                );
698            }
699        }
700    }
701
702    pub(crate) fn closure_call_env_for_current_frame(
703        &self,
704        closure: &crate::value::VmClosure,
705    ) -> VmEnv {
706        if closure.module_state().is_some() {
707            return closure.env.cloned_for_call();
708        }
709        let call_env = Self::closure_call_env(&self.env, closure);
710        // Same compile-time short-circuit as the env walk in
711        // `closure_call_env`: when the callee body never resolves an
712        // outer name through the env, injecting closure-typed *slot*
713        // locals from the caller's frame is wasted work too.
714        if !closure.func.chunk.references_outer_names {
715            return call_env;
716        }
717        let mut call_env = call_env;
718        let Some(frame) = self.frames.last() else {
719            return call_env;
720        };
721        for (slot, info) in frame
722            .local_slots
723            .iter()
724            .zip(frame.chunk.local_slots.iter())
725            .filter(|(slot, info)| slot.initialized && info.scope_depth <= frame.local_scope_depth)
726        {
727            if matches!(slot.value, VmValue::Closure(_)) && !call_env.contains(&info.name) {
728                let _ = call_env.define(&info.name, slot.value.clone(), info.mutable);
729            }
730        }
731        call_env
732    }
733
734    pub(crate) fn active_local_slot_value(&self, name: &str) -> Option<VmValue> {
735        let frame = self.frames.last()?;
736        let idx = self.active_local_slot_index(name)?;
737        frame.local_slots.get(idx).map(|slot| slot.value.clone())
738    }
739
740    /// Returns the slot index of an initialized active local with the given
741    /// name, walking from innermost to outermost scope. Used by legacy by-name
742    /// hot paths that still want to mutate the slot value in place without
743    /// paying a defensive `VmValue::clone` first.
744    pub(crate) fn active_local_slot_index(&self, name: &str) -> Option<usize> {
745        let frame = self.frames.last()?;
746        for (idx, info) in frame.chunk.local_slots.iter().enumerate().rev() {
747            if info.name == name && info.scope_depth <= frame.local_scope_depth {
748                if let Some(slot) = frame.local_slots.get(idx) {
749                    if slot.initialized {
750                        return Some(idx);
751                    }
752                }
753            }
754        }
755        None
756    }
757
758    pub(crate) fn assign_active_local_slot(
759        &mut self,
760        name: &str,
761        value: VmValue,
762        debug: bool,
763    ) -> Result<bool, VmError> {
764        let Some(frame) = self.frames.last_mut() else {
765            return Ok(false);
766        };
767        for (idx, info) in frame.chunk.local_slots.iter().enumerate().rev() {
768            if info.name == name && info.scope_depth <= frame.local_scope_depth {
769                if !debug && !info.mutable {
770                    return Err(VmError::ImmutableAssignment(name.to_string()));
771                }
772                if let Some(slot) = frame.local_slots.get_mut(idx) {
773                    crate::value::recursion::dismantle(std::mem::replace(&mut slot.value, value));
774                    slot.initialized = true;
775                    slot.synced = false;
776                    return Ok(true);
777                }
778            }
779        }
780        Ok(false)
781    }
782
783    pub fn new() -> Self {
784        crate::initialize_runtime_assets();
785        Self {
786            stack: Vec::with_capacity(256),
787            env: VmEnv::new(),
788            output: String::new(),
789            builtins: Arc::new(BTreeMap::new()),
790            async_builtins: Arc::new(BTreeMap::new()),
791            capability_methods: Arc::new(BTreeMap::new()),
792            builtin_metadata: Arc::new(BTreeMap::new()),
793            builtins_by_id: Arc::new(HashMap::new()),
794            builtin_id_collisions: Arc::new(HashSet::new()),
795            iterators: Vec::new(),
796            frames: Vec::new(),
797            exception_handlers: Vec::new(),
798            spawned_tasks: BTreeMap::new(),
799            process_exit_request: Arc::new(ProcessExitRequest::new()),
800            sync_runtime: Arc::new(crate::synchronization::VmSyncRuntime::new()),
801            shared_state_runtime: Arc::new(crate::shared_state::VmSharedStateRuntime::new()),
802            inline_cache_sets: Vec::new(),
803            inline_cache_set_by_chunk: HashMap::new(),
804            pool_registry: crate::stdlib::pool::new_pool_registry(),
805            llm_mock_context: crate::llm::mock::LlmMockContext::for_new_vm(),
806            package_snapshot_registry: Arc::new(Default::default()),
807            wait_for_graph: Arc::new(crate::wait_for_graph::VmWaitForGraph::new()),
808            held_sync_guards: Vec::new(),
809            inherited_held_keys: Arc::new(Vec::new()),
810            task_scopes: Vec::new(),
811            task_counter: 0,
812            runtime_context_counter: 0,
813            runtime_context: crate::runtime_context::RuntimeContext::root(),
814            deadlines: Vec::new(),
815            execution_deadline: super::execution::new_execution_deadline_state(None),
816            breakpoints: BTreeMap::new(),
817            function_breakpoints: std::collections::BTreeSet::new(),
818            pending_function_bp: None,
819            step_mode: false,
820            step_frame_depth: 0,
821            stopped: false,
822            last_line: 0,
823            source_dir: None,
824            package_execution_guard: None,
825            imported_paths: Vec::new(),
826            deferred_cyclic_imports: Vec::new(),
827            module_cache: Arc::new(BTreeMap::new()),
828            prepared_module_cache: crate::PreparedModuleCache::default(),
829            module_provenance: crate::module_artifact::ModuleProvenance::User,
830            module_phase_recorder: None,
831            lazy_callable_modules: Arc::new(crate::value::VmMutex::new(BTreeMap::new())),
832            source_cache: Arc::new(BTreeMap::new()),
833            graph_link_table: None,
834            linked_program_repository: None,
835            source_file: None,
836            source_text: None,
837            coverage: crate::coverage::for_primary(None),
838            bridge: None,
839            denied_builtins: Arc::new(HashSet::new()),
840            cancel_token: None,
841            interrupt_signal_token: None,
842            cancel_grace_instructions_remaining: None,
843            interrupt_handlers: Vec::new(),
844            next_interrupt_handle: 1,
845            pending_interrupt_signal: None,
846            interrupted: false,
847            dispatching_interrupt: false,
848            interrupt_handler_deadline: None,
849            error_stack_trace: Vec::new(),
850            yield_sender: None,
851            project_root: None,
852            globals: Arc::new(crate::value::DictMap::new()),
853            root_harness: None,
854            runtime_effects: crate::orchestration::RuntimeEffectState::fresh(),
855            debug_hook: None,
856            runtime_limits: RuntimeLimits::default(),
857        }
858    }
859
860    pub fn baseline(&self) -> VmBaseline {
861        VmBaseline::from_vm(self)
862    }
863
864    /// Typed, runtime-resolved effects attempted by this VM execution tree.
865    pub fn executed_effects(&self) -> Vec<crate::orchestration::EffectRecord> {
866        self.runtime_effects.snapshot()
867    }
868
869    /// Clear the execution-local effect recorder and per-VM memo.
870    pub fn clear_executed_effects(&mut self) {
871        self.runtime_effects.clear();
872    }
873
874    pub(crate) fn record_capability_effects(
875        &mut self,
876        capability: harn_builtin_meta::CapabilityId,
877        method: &str,
878        args: &[VmValue],
879    ) {
880        self.runtime_effects
881            .record_capability(capability, method, args);
882    }
883
884    pub(crate) fn record_builtin_contract_effects(&mut self, name: &str, args: &[VmValue]) {
885        let Some(entry) = crate::stdlib::recorded_effect_builtin_manifest_entry(name) else {
886            return;
887        };
888        self.record_builtin_effect_specs(entry.contract.effects, args);
889    }
890
891    pub(crate) fn record_builtin_effect_specs(
892        &mut self,
893        specs: &'static [harn_builtin_meta::EffectSpec],
894        args: &[VmValue],
895    ) {
896        self.runtime_effects.record_specs(specs, args);
897    }
898
899    /// Replace the scoped immutable module-template cache used by this VM.
900    /// Fresh runtime module state is still instantiated for every VM.
901    pub fn set_prepared_module_cache(&mut self, cache: crate::PreparedModuleCache) {
902        self.prepared_module_cache = cache;
903    }
904
905    /// Hand this VM the link table for the import graph it is about to run.
906    ///
907    /// Only an entry-chunk cache hit can supply one, because only that path holds
908    /// a manifest proving the recorded digests still describe the files on disk.
909    /// A table never changes what a module resolves to: one the table does not
910    /// name, or whose artifact is no longer on disk, is read and compiled as
911    /// usual.
912    pub fn set_graph_link_table(
913        &mut self,
914        link_table: Option<Arc<crate::context_manifest::GraphLinkTable>>,
915    ) {
916        self.graph_link_table = link_table;
917    }
918
919    /// Install the verified module-template repository carried by one linked
920    /// program. Templates remain scoped to this VM tree and never enter the
921    /// ordinary caller-independent cache.
922    pub fn set_linked_program_runtime(
923        &mut self,
924        runtime: &crate::linked_program::LinkedProgramRuntime,
925    ) {
926        self.linked_program_repository = Some(Arc::clone(&runtime.repository));
927        self.graph_link_table = None;
928    }
929
930    /// Return the effective runtime limit profile for this VM.
931    pub fn runtime_limits(&self) -> RuntimeLimits {
932        self.runtime_limits
933    }
934
935    /// Return a host/debug report describing the VM's effective runtime limits.
936    pub fn runtime_limit_report(&self) -> crate::RuntimeLimitsReport {
937        self.runtime_limits.report()
938    }
939
940    /// Returns true if any debugging affordance is active — DAP hook,
941    /// line breakpoints, or function breakpoints. Call-site code uses
942    /// this to decide whether to capture per-frame restart snapshots
943    /// (`initial_env`, `initial_local_slots`); without a debugger those
944    /// snapshots are dead weight, so skipping them removes two
945    /// allocations from every function call hot path.
946    ///
947    /// All three signals are stable across a function call's lifetime
948    /// (they're set before pipeline execution starts), so the gate is
949    /// consistent between frame creation and any later `restart_frame`
950    /// invocation. The three `is_empty` checks compile to a handful of
951    /// branch-predicted memory probes — cheaper than a single
952    /// `BTreeMap` clone, which is what we're avoiding.
953    #[inline]
954    pub(crate) fn debugger_attached(&self) -> bool {
955        self.debug_hook.is_some()
956            || !self.breakpoints.is_empty()
957            || !self.function_breakpoints.is_empty()
958    }
959
960    /// Set the bridge for delegating unknown builtins in bridge mode.
961    pub fn set_bridge(&mut self, bridge: Arc<crate::bridge::HostBridge>) {
962        self.bridge = Some(bridge);
963    }
964
965    /// Set builtins that are denied in sandbox mode.
966    /// When called, the given builtin names will produce a permission error.
967    pub fn set_denied_builtins(&mut self, mut denied: HashSet<String>) {
968        // A builtin may have a private registry name plus one or more public
969        // aliases. Normalize the whole family once so no dispatch path can
970        // escape policy merely by using another name for the same handler.
971        let denied_canonical_names = denied
972            .iter()
973            .filter_map(|name| crate::stdlib::builtin_manifest_entry(name))
974            .map(|entry| entry.canonical_name)
975            .collect::<HashSet<_>>();
976        if !denied_canonical_names.is_empty() {
977            denied.extend(
978                crate::stdlib::all_builtin_manifest()
979                    .iter()
980                    .filter(|entry| denied_canonical_names.contains(entry.canonical_name))
981                    .map(|entry| entry.name.to_string()),
982            );
983        }
984        self.denied_builtins = Arc::new(denied);
985    }
986
987    /// Set source info for error reporting (file path and source text).
988    pub fn set_source_info(&mut self, file: &str, text: &str) {
989        self.source_file = Some(file.to_string());
990        self.source_text = Some(text.to_string());
991        if let Some(cov) = self.coverage.as_mut() {
992            cov.set_primary_file(file);
993        }
994        Arc::make_mut(&mut self.source_cache)
995            .insert(std::path::PathBuf::from(file), Arc::from(text));
996    }
997
998    /// Initialize execution (push the initial frame).
999    pub fn start(&mut self, chunk: &Chunk) -> Result<(), VmError> {
1000        self.ensure_execution_available()?;
1001        // The top-level pipeline frame captures env at start so
1002        // restartFrame on the outermost frame rewinds to the
1003        // pre-pipeline state — basically "restart session" in
1004        // debugger terms. Skipped when no debugger is attached:
1005        // the snapshot is dead weight in that case and dominates
1006        // call-overhead bench numbers (~5-10%).
1007        let debugger = self.debugger_attached();
1008        let initial_env = if debugger {
1009            Some(self.env.clone())
1010        } else {
1011            None
1012        };
1013        let initial_local_slots = if debugger {
1014            Some(Self::fresh_local_slots(chunk))
1015        } else {
1016            None
1017        };
1018        let chunk = Arc::new(chunk.clone());
1019        let local_slots = Self::fresh_local_slots(&chunk);
1020        let inline_cache_set = self.inline_cache_set_index_for_chunk(&chunk);
1021        self.frames.push(CallFrame {
1022            chunk,
1023            inline_cache_set,
1024            ip: 0,
1025            stack_base: self.stack.len(),
1026            saved_env: self.env.clone(),
1027            initial_env,
1028            initial_local_slots,
1029            saved_iterator_depth: self.iterators.len(),
1030            fn_name: crate::value::HarnStr::new(),
1031            argc: 0,
1032            saved_source_dir: None,
1033            module_functions: None,
1034            module_state: None,
1035            local_slots,
1036            local_scope_base: self.env.scope_depth().saturating_sub(1),
1037            local_scope_depth: 0,
1038        });
1039        Ok(())
1040    }
1041
1042    /// Create a child VM that shares builtins and env but has fresh execution state.
1043    /// Used for parallel/spawn to fork the VM for concurrent tasks.
1044    pub(crate) fn child_vm(&self) -> Vm {
1045        Vm {
1046            stack: Vec::with_capacity(64),
1047            env: self.env.clone(),
1048            output: String::new(),
1049            builtins: Arc::clone(&self.builtins),
1050            async_builtins: Arc::clone(&self.async_builtins),
1051            capability_methods: Arc::clone(&self.capability_methods),
1052            builtin_metadata: Arc::clone(&self.builtin_metadata),
1053            builtins_by_id: Arc::clone(&self.builtins_by_id),
1054            builtin_id_collisions: Arc::clone(&self.builtin_id_collisions),
1055            iterators: Vec::new(),
1056            frames: Vec::new(),
1057            exception_handlers: Vec::new(),
1058            spawned_tasks: BTreeMap::new(),
1059            process_exit_request: Arc::clone(&self.process_exit_request),
1060            sync_runtime: self.sync_runtime.clone(),
1061            shared_state_runtime: self.shared_state_runtime.clone(),
1062            inline_cache_sets: Vec::new(),
1063            inline_cache_set_by_chunk: HashMap::new(),
1064            pool_registry: self.pool_registry.clone(),
1065            llm_mock_context: self.llm_mock_context.clone(),
1066            package_snapshot_registry: self.package_snapshot_registry.clone(),
1067            wait_for_graph: self.wait_for_graph.clone(),
1068            held_sync_guards: Vec::new(),
1069            inherited_held_keys: Arc::new(Vec::new()),
1070            task_scopes: Vec::new(),
1071            task_counter: 0,
1072            runtime_context_counter: self.runtime_context_counter,
1073            runtime_context: self.runtime_context.clone(),
1074            deadlines: self.deadlines.clone(),
1075            execution_deadline: self.execution_deadline.fork(),
1076            breakpoints: BTreeMap::new(),
1077            function_breakpoints: std::collections::BTreeSet::new(),
1078            pending_function_bp: None,
1079            step_mode: false,
1080            step_frame_depth: 0,
1081            stopped: false,
1082            last_line: 0,
1083            source_dir: self.source_dir.clone(),
1084            package_execution_guard: self.package_execution_guard.clone(),
1085            imported_paths: Vec::new(),
1086            deferred_cyclic_imports: Vec::new(),
1087            module_cache: Arc::clone(&self.module_cache),
1088            prepared_module_cache: self.prepared_module_cache.clone(),
1089            module_provenance: self.module_provenance,
1090            module_phase_recorder: self.module_phase_recorder.clone(),
1091            lazy_callable_modules: Arc::clone(&self.lazy_callable_modules),
1092            source_cache: Arc::clone(&self.source_cache),
1093            graph_link_table: self.graph_link_table.clone(),
1094            linked_program_repository: self.linked_program_repository.clone(),
1095            source_file: self.source_file.clone(),
1096            source_text: self.source_text.clone(),
1097            coverage: crate::coverage::for_primary(self.source_file.as_deref()),
1098            bridge: self.bridge.clone(),
1099            denied_builtins: Arc::clone(&self.denied_builtins),
1100            cancel_token: self.cancel_token.clone(),
1101            interrupt_signal_token: self.interrupt_signal_token.clone(),
1102            cancel_grace_instructions_remaining: None,
1103            interrupt_handlers: Vec::new(),
1104            next_interrupt_handle: 1,
1105            pending_interrupt_signal: None,
1106            interrupted: self.interrupted,
1107            dispatching_interrupt: false,
1108            interrupt_handler_deadline: None,
1109            error_stack_trace: Vec::new(),
1110            yield_sender: None,
1111            project_root: self.project_root.clone(),
1112            globals: Arc::clone(&self.globals),
1113            root_harness: self.root_harness.clone(),
1114            runtime_effects: crate::orchestration::RuntimeEffectState::with_shared_recorder(
1115                Arc::clone(&self.runtime_effects.recorder),
1116            ),
1117            debug_hook: None,
1118            runtime_limits: self.runtime_limits,
1119        }
1120    }
1121
1122    /// Create a child VM for external adapters that need to invoke Harn
1123    /// closures while sharing the parent's builtins, globals, and module state.
1124    pub(crate) fn child_vm_for_host(&self) -> Vm {
1125        self.child_vm()
1126    }
1127
1128    /// Return the cooperative cancellation/deadline sources that a blocking
1129    /// host adapter must install on its worker thread. The values are cloned
1130    /// once at the async-boundary; the token remains shared with the owning VM
1131    /// so `parallel` aborts and host cancellation reach an in-flight wait.
1132    pub(crate) fn interrupt_sources(
1133        &self,
1134    ) -> (
1135        Option<std::sync::Arc<std::sync::atomic::AtomicBool>>,
1136        Option<std::time::Instant>,
1137    ) {
1138        let scope_deadline = self.deadlines.last().map(|(deadline, _)| *deadline);
1139        let deadline = match (scope_deadline, self.interrupt_handler_deadline) {
1140            (Some(scope), Some(interrupt)) => Some(scope.min(interrupt)),
1141            (scope, interrupt) => scope.or(interrupt),
1142        };
1143        (self.cancel_token.clone(), deadline)
1144    }
1145
1146    pub(crate) fn request_process_exit(&self, code: i32) {
1147        self.process_exit_request.request(code);
1148    }
1149
1150    pub(crate) fn requested_process_exit(&self) -> Option<i32> {
1151        self.process_exit_request.code()
1152    }
1153
1154    /// Request cancellation for every outstanding child task owned by this VM
1155    /// and then abort the join handles. This prevents un-awaited spawned tasks
1156    /// from outliving their parent execution scope.
1157    pub(crate) fn cancel_spawned_tasks(&mut self) {
1158        for (_, task) in std::mem::take(&mut self.spawned_tasks) {
1159            task.cancel_token
1160                .store(true, std::sync::atomic::Ordering::SeqCst);
1161            task.handle.abort();
1162        }
1163    }
1164
1165    /// Set the source directory for import resolution and introspection.
1166    /// Also auto-detects the project root if not already set.
1167    pub fn set_source_dir(&mut self, dir: &std::path::Path) {
1168        let dir = crate::stdlib::process::normalize_context_path(dir);
1169        self.source_dir = Some(dir.clone());
1170        crate::stdlib::set_thread_source_dir(&dir);
1171        // Auto-detect project root if not explicitly set.
1172        if self.project_root.is_none() {
1173            self.project_root = crate::stdlib::process::find_project_root(&dir);
1174        }
1175    }
1176
1177    /// Explicitly set the project root directory.
1178    /// Used by ACP/CLI to override auto-detection.
1179    pub fn set_project_root(&mut self, root: &std::path::Path) {
1180        self.project_root = Some(root.to_path_buf());
1181    }
1182
1183    /// Get only the explicit or auto-detected project root, without falling
1184    /// back to `source_dir`.
1185    pub(crate) fn explicit_project_root(&self) -> Option<&std::path::Path> {
1186        self.project_root.as_deref()
1187    }
1188
1189    /// Get the project root directory, falling back to source_dir.
1190    pub fn project_root(&self) -> Option<&std::path::Path> {
1191        self.project_root.as_deref().or(self.source_dir.as_deref())
1192    }
1193
1194    /// Set a global constant (e.g. `pi`, `e`).
1195    /// Stored separately from the environment so user-defined variables can shadow them.
1196    pub fn set_global(&mut self, name: &str, value: VmValue) {
1197        Arc::make_mut(&mut self.globals).insert(crate::value::intern_key(name), value);
1198    }
1199
1200    /// Read a previously-installed ordinary global constant.
1201    pub fn global(&self, name: &str) -> Option<&VmValue> {
1202        self.globals.get(name)
1203    }
1204
1205    /// Install the script's root authority for the compiler-generated
1206    /// `main(harness: Harness)` call. It is kept out of all source-visible
1207    /// namespaces.
1208    pub fn set_harness(&mut self, harness: crate::harness::Harness) {
1209        self.root_harness = Some(harness.into_vm_value());
1210    }
1211
1212    pub(crate) fn harness(&self) -> Option<&crate::harness::VmHarness> {
1213        match self.root_harness.as_ref() {
1214            Some(VmValue::Harness(handle)) => Some(handle),
1215            _ => None,
1216        }
1217    }
1218
1219    /// Clone the embedder-owned root authority for an explicit exported
1220    /// function parameter. This does not install a source-visible binding.
1221    pub fn root_harness_value(&self) -> Option<VmValue> {
1222        self.root_harness.clone()
1223    }
1224
1225    /// Get the captured output.
1226    pub fn output(&self) -> &str {
1227        &self.output
1228    }
1229
1230    /// Drain and return the captured output, leaving the buffer empty.
1231    /// Used by the async-builtin dispatch path to forward closure output
1232    /// from a child VM back to its parent.
1233    pub fn take_output(&mut self) -> String {
1234        std::mem::take(&mut self.output)
1235    }
1236
1237    /// Append text to this VM's captured output. Used to forward output
1238    /// from child VMs (e.g. closures invoked via `call_closure_pub`)
1239    /// back into the parent stream.
1240    pub fn append_output(&mut self, text: &str) {
1241        self.output.push_str(text);
1242    }
1243
1244    pub(crate) fn pop(&mut self) -> Result<VmValue, VmError> {
1245        self.stack.pop().ok_or(VmError::StackUnderflow)
1246    }
1247
1248    pub(crate) fn peek(&self) -> Result<&VmValue, VmError> {
1249        self.stack.last().ok_or(VmError::StackUnderflow)
1250    }
1251
1252    pub(crate) fn const_str(c: &Constant) -> Result<&str, VmError> {
1253        match c {
1254            Constant::String(s) => Ok(s.as_str()),
1255            _ => Err(VmError::TypeError("expected string constant".into())),
1256        }
1257    }
1258
1259    pub(crate) fn release_sync_guards_for_current_scope(&mut self) {
1260        let depth = self.env.scope_depth();
1261        self.held_sync_guards
1262            .retain(|guard| guard.env_scope_depth < depth);
1263        // A `scope { }` torn down without a normal `TaskScopeExit` (break /
1264        // continue out of it) leaves a dangling nursery — cancel its tasks.
1265        self.cancel_task_scopes_where(|s| s.env_scope_depth >= depth);
1266    }
1267
1268    pub(crate) fn release_sync_guards_after_unwind(
1269        &mut self,
1270        frame_depth: usize,
1271        env_scope_depth: usize,
1272    ) {
1273        self.held_sync_guards.retain(|guard| {
1274            guard.frame_depth <= frame_depth && guard.env_scope_depth <= env_scope_depth
1275        });
1276        // Cancel nurseries opened above the catch handler (a `throw` unwound
1277        // past their `TaskScopeExit`).
1278        self.cancel_task_scopes_where(|s| {
1279            !(s.frame_depth <= frame_depth && s.env_scope_depth <= env_scope_depth)
1280        });
1281    }
1282
1283    pub(crate) fn release_sync_guards_for_frame(&mut self, frame_depth: usize) {
1284        self.held_sync_guards
1285            .retain(|guard| guard.frame_depth != frame_depth);
1286        // Cancel any nursery whose `scope {}` block belonged to the frame being
1287        // torn down (e.g. a `return` jumped past its `TaskScopeExit`).
1288        self.cancel_task_scopes_where(|s| s.frame_depth == frame_depth);
1289    }
1290
1291    pub(crate) fn adopt_sync_permit_for_current_scope(
1292        &mut self,
1293        permit: crate::value::VmSyncPermitHandle,
1294    ) {
1295        if permit.is_released()
1296            || self
1297                .held_sync_guards
1298                .iter()
1299                .any(|guard| guard._permit.same_lease(&permit))
1300        {
1301            return;
1302        }
1303        self.held_sync_guards
1304            .push(crate::synchronization::VmSyncHeldGuard {
1305                _permit: permit,
1306                frame_depth: self.frames.len(),
1307                env_scope_depth: self.env.scope_depth(),
1308            });
1309    }
1310
1311    /// Deregister a task id from every open nursery (it was explicitly
1312    /// `await`ed, so it must not be double-joined or cancelled at scope exit).
1313    pub(crate) fn deregister_task_from_scopes(&mut self, id: &str) {
1314        for scope in &mut self.task_scopes {
1315            scope.task_ids.retain(|t| t != id);
1316        }
1317    }
1318
1319    /// Cancel and remove every task scope matching `doomed`, aborting its bound
1320    /// tasks (used when a `scope {}` is torn down without a normal join).
1321    fn cancel_task_scopes_where<F: Fn(&TaskScope) -> bool>(&mut self, doomed: F) {
1322        let mut i = 0;
1323        while i < self.task_scopes.len() {
1324            if doomed(&self.task_scopes[i]) {
1325                let scope = self.task_scopes.remove(i);
1326                for id in &scope.task_ids {
1327                    if let Some(task) = self.spawned_tasks.remove(id) {
1328                        task.cancel_token
1329                            .store(true, std::sync::atomic::Ordering::SeqCst);
1330                        task.handle.abort();
1331                    }
1332                }
1333            } else {
1334                i += 1;
1335            }
1336        }
1337    }
1338
1339    /// Total live permits this VM already holds for `kind:key`. The held-set is
1340    /// tiny (bounded by lexical nesting and explicit sync acquisitions), so this
1341    /// scan is cheap and only runs on the rare blocking-acquire path.
1342    pub(crate) fn held_permits_for(&self, kind: &str, key: &str) -> u32 {
1343        let own: u32 = self
1344            .held_sync_guards
1345            .iter()
1346            .filter(|guard| {
1347                !guard._permit.is_released()
1348                    && guard._permit.kind() == kind
1349                    && guard._permit.key() == key
1350            })
1351            .map(|guard| guard._permit.permits())
1352            .sum();
1353        let inherited: u32 = self
1354            .inherited_held_keys
1355            .iter()
1356            .filter(|held| held.kind == kind && held.key == key)
1357            .map(|held| held.permits)
1358            .sum();
1359        own + inherited
1360    }
1361
1362    /// Every live sync permit held by this VM *and* its suspended ancestors: the
1363    /// transitive held-set seen by an inline child.
1364    pub(crate) fn combined_held_keys(&self) -> Vec<crate::synchronization::VmSyncHeldKey> {
1365        let mut keys: Vec<crate::synchronization::VmSyncHeldKey> = self
1366            .held_sync_guards
1367            .iter()
1368            .filter_map(|guard| crate::synchronization::VmSyncHeldKey::from_permit(&guard._permit))
1369            .collect();
1370        keys.extend(self.inherited_held_keys.iter().cloned());
1371        keys
1372    }
1373
1374    /// Clone a child VM for an **inline, same-task** execution (an async builtin
1375    /// awaited while this VM is parked, or a user closure that builtin runs and
1376    /// awaits). The child inherits this VM's transitive held-lock keys so a
1377    /// re-acquire of a parent-held lock is caught as a self-deadlock
1378    /// (HARN-ORC-011). Use plain `child_vm()` for new concurrent tasks.
1379    pub(crate) fn child_vm_inline(&self) -> Vm {
1380        let mut child = self.child_vm();
1381        child.inherited_held_keys = Arc::new(self.combined_held_keys());
1382        child.execution_deadline = Arc::clone(&self.execution_deadline);
1383        child
1384    }
1385}
1386
1387impl Drop for Vm {
1388    fn drop(&mut self) {
1389        if let Some(coverage) = self.coverage.take() {
1390            crate::coverage::merge_into_global(coverage);
1391        }
1392        self.cancel_spawned_tasks();
1393    }
1394}
1395
1396impl Default for Vm {
1397    fn default() -> Self {
1398        Self::new()
1399    }
1400}
1401
1402#[cfg(test)]
1403#[path = "state_tests.rs"]
1404mod tests;