cljrs_runtime/env/env.rs
1//! Lexical environment: local frames, global namespace table, and current Env.
2
3use std::collections::{HashMap, HashSet};
4use std::sync::atomic::{AtomicBool, AtomicU8, Ordering};
5use std::sync::{Arc, Condvar, Mutex, RwLock};
6
7use crate::env::async_hook::AsyncRuntime;
8
9use crate::env::error::EvalResult;
10use crate::mode::{ExecutionMode, TierState};
11use cljrs_gc::{GcConfig, GcPtr};
12use cljrs_reader::Form;
13use cljrs_value::{CljxFn, Namespace, ReferClojureFilter, Value, Var};
14// ── RequireSpec / RequireRefer ─────────────────────────────────────────────────
15
16/// How symbols should be referred into the requiring namespace.
17#[derive(Debug, Clone)]
18pub enum RequireRefer {
19 None,
20 All,
21 Named(Vec<Arc<str>>),
22}
23
24/// A parsed `require` specification.
25#[derive(Debug, Clone)]
26pub struct RequireSpec {
27 pub ns: Arc<str>,
28 /// Present when the namespace symbol carried a `@<hash>` version suffix.
29 pub version: Option<Arc<str>>,
30 pub alias: Option<Arc<str>>,
31 pub refer: RequireRefer,
32}
33
34// ── Frame ─────────────────────────────────────────────────────────────────────
35
36/// One stack frame of local bindings (a single `let*`, `fn`, or `loop*` scope).
37pub struct Frame {
38 pub bindings: Vec<(Arc<str>, Value)>,
39}
40
41impl Default for Frame {
42 fn default() -> Self {
43 Self::new()
44 }
45}
46
47impl Frame {
48 pub fn new() -> Self {
49 Self {
50 bindings: Vec::new(),
51 }
52 }
53
54 pub fn bind(&mut self, name: Arc<str>, val: Value) {
55 // Shadow: push new binding; lookup searches from the end.
56 self.bindings.push((name, val));
57 }
58
59 pub fn lookup(&self, name: &str) -> Option<&Value> {
60 // Search in reverse order so later bindings shadow earlier ones.
61 tracing::trace!(target: "env", "lookup {}", name);
62 for (n, v) in self.bindings.iter().rev() {
63 if n.as_ref() == name {
64 return Some(v);
65 }
66 }
67 None
68 }
69}
70
71// ── GlobalEnv ─────────────────────────────────────────────────────────────────
72
73/// The global mutable store of all namespaces.
74pub struct GlobalEnv {
75 /// Process-unique identity of this runtime instance.
76 ///
77 /// Allocated from a counter, not derived from the `Arc`'s address: an
78 /// address is only unique while the allocation is live, so a dropped
79 /// runtime could hand its key to the next one and let it inherit stale
80 /// cross-defn IR. Used to scope per-instance registries.
81 id: u64,
82 pub namespaces: RwLock<HashMap<Arc<str>, GcPtr<Namespace>>>,
83 /// Directories to search when resolving namespace names to files.
84 pub source_paths: RwLock<Vec<std::path::PathBuf>>,
85 /// Namespaces that have been fully loaded from a file (idempotent guard).
86 pub loaded: Mutex<std::collections::HashSet<Arc<str>>>,
87 /// Namespaces currently being loaded, mapped to the thread loading them.
88 /// Used to detect true circular requires (same thread) vs concurrent loads
89 /// (different thread — those wait on `loading_done` instead of erroring).
90 pub loading: Mutex<HashMap<Arc<str>, std::thread::ThreadId>>,
91 /// Signalled whenever a namespace finishes loading (or fails).
92 pub loading_done: Condvar,
93 /// Built-in namespace sources embedded in the binary.
94 /// Checked by `load_ns` before falling back to source-path search.
95 pub builtin_sources: RwLock<HashMap<Arc<str>, &'static str>>,
96 /// GC configuration for automatic collection based on memory pressure.
97 pub gc_config: RwLock<Option<Arc<GcConfig>>>,
98 /// How this runtime executes function calls. Fixed when the runtime is
99 /// built; see [`crate::RuntimeBuilder::execution_mode`].
100 execution_mode: ExecutionMode,
101 /// Which tiers are live right now (see [`TierState`]). Starts at
102 /// [`TierState::TreeWalk`] — nothing can be lowered until `clojure.core`
103 /// exists — and is raised once to `execution_mode.target_tier()` when the
104 /// builder finishes bootstrapping.
105 tier_state: AtomicU8,
106 /// This runtime's Tier-1 and Tier-2 state: the lowered-IR cache, the JIT
107 /// counters and native-code tables, and the JIT backend attached to this
108 /// runtime. Instance state: two runtimes in one process never read,
109 /// evict, or invalidate each other's entries, and everything dies with
110 /// the runtime.
111 tiers: Arc<crate::tiered::tiers::Tiers>,
112 /// Optional async runtime registered by `cljrs-async`.
113 /// `None` when the library is not linked; `Some` after `cljrs_async::init`.
114 pub async_rt: RwLock<Option<Arc<dyn AsyncRuntime>>>,
115 /// Cache of values resolved at a specific commit.
116 /// Key format: `"<ns>/<name>@<commit>"` for individual vars,
117 /// or `"<ns>@<commit>"` for whole versioned namespaces.
118 pub version_cache: Mutex<HashMap<Arc<str>, Value>>,
119 /// Parsed `cljrs.edn` config, loaded once at startup.
120 pub deps_config: RwLock<Option<Arc<cljrs_project::config::DepsConfig>>>,
121 /// When true, every versioned-symbol or versioned-namespace resolution must
122 /// carry a valid commit signature (verified natively against `trusted_keys`)
123 /// before the historical code is executed. Off by default; enabled via
124 /// `--verify-commit-signatures` CLI flag or `:verify-commit-signatures true`
125 /// in `cljrs.edn`.
126 pub verify_commit_signatures: AtomicBool,
127 /// The git backend used by versioned resolution and signature checking, or
128 /// `None` in builds that carry no VCS implementation (wasm, or
129 /// `cljrs-runtime` without its default `deps` feature). With no provider,
130 /// source files are treated as living outside any repository and versioned
131 /// resolution can only use embedded (AOT) sources. See [`crate::env::vcs`].
132 vcs: RwLock<Option<Arc<dyn crate::env::vcs::VcsProvider>>>,
133 /// Session-scoped cache of commits that have already passed signature
134 /// verification this run, keyed by `(repo_root, commit_hash)`.
135 pub sig_verify_cache: Mutex<HashSet<(Arc<str>, Arc<str>)>>,
136 /// Pinned source texts fetched from git this session, keyed by
137 /// `"<ns>@<commit>"`. The AOT compiler embeds these in the produced
138 /// binary so versioned namespaces resolve without git at runtime.
139 pub versioned_sources: RwLock<HashMap<Arc<str>, Arc<str>>>,
140 /// When true (set by AOT harness main), versioned namespaces resolve
141 /// only from embedded builtin sources — never from git. A versioned
142 /// namespace that was not embedded at compile time fails with a clear
143 /// error instead of attempting a fetch.
144 pub versioned_offline: AtomicBool,
145 /// Provenance of native (Rust-backed) packages recorded at registration:
146 /// namespace → the git commit the package was built from. Consulted by
147 /// the versioned resolver's native HEAD fallback to detect pinned-commit
148 /// mismatches.
149 pub native_provenance: RwLock<HashMap<Arc<str>, Arc<str>>>,
150 /// When true, a pinned lookup of a native function whose recorded
151 /// provenance does not match the requested commit is an error instead of
152 /// a once-per-pin warning. CLI: `--enforce-native-versions`; cljrs.edn:
153 /// `:enforce-native-versions true`.
154 pub enforce_native_versions: AtomicBool,
155 /// Pinned-native mismatches already warned about this session
156 /// (key: `"<ns>@<commit>"`), so each pin warns at most once.
157 pub provenance_warned: Mutex<HashSet<Arc<str>>>,
158 /// Optional loader for **pinned native packages** (`:rust/load :dylib`),
159 /// installed by the CLI. Called by the versioned resolver with
160 /// `(globals, base_ns, commit)` before falling back to the HEAD native
161 /// binding; returns `Ok(true)` when it registered the package's pinned
162 /// implementations into the `"<base_ns>@<commit>"` namespace.
163 #[allow(clippy::type_complexity)]
164 pub pinned_native_loader: RwLock<Option<PinnedNativeLoader>>,
165 /// Optional loader for **native dependencies on the plain `require` path**
166 /// (`:rust/load :dylib`), installed by the CLI. Called by the
167 /// unversioned namespace loader with `(globals, ns)` when a `require`d
168 /// namespace has no Clojure source on the source path; returns `Ok(true)`
169 /// when it built the dep's crate at the pinned `:git/sha` and registered
170 /// the package's exports into the **unversioned** namespace, so a plain
171 /// `(require '[my.native.lib :as lib])` brings the native code in.
172 #[allow(clippy::type_complexity)]
173 pub native_require_loader: RwLock<Option<NativeRequireLoader>>,
174 /// Loaders for **AOT-compiled namespaces**, installed by the binary
175 /// produced by `cljrs compile`. Keyed by namespace name. When a plain
176 /// `require` resolves a namespace that has a registered loader, `load_ns`
177 /// invokes the loader instead of interpreting Clojure source: the loader
178 /// evaluates the namespace's small interpreted preamble (its `ns`/`require`
179 /// and macro definitions) and then calls the namespace's natively compiled
180 /// initializer, so the bulk of the namespace runs as machine code rather
181 /// than being tree-walked at startup.
182 #[allow(clippy::type_complexity)]
183 pub compiled_ns_loaders: RwLock<HashMap<Arc<str>, CompiledNsLoader>>,
184}
185
186/// Loader callback for an AOT-compiled namespace (see
187/// `GlobalEnv::compiled_ns_loaders`). Given the global env, it loads the
188/// namespace by running its interpreted preamble and its compiled initializer.
189pub type CompiledNsLoader = Arc<dyn Fn(&Arc<GlobalEnv>) -> EvalResult<()> + Send + Sync>;
190
191/// Loader callback for pinned native packages (see
192/// `GlobalEnv::pinned_native_loader`).
193pub type PinnedNativeLoader =
194 Arc<dyn Fn(&Arc<GlobalEnv>, &str, &str) -> EvalResult<bool> + Send + Sync>;
195
196/// Loader callback for native dependencies reached through a plain `require`
197/// (see `GlobalEnv::native_require_loader`).
198pub type NativeRequireLoader = Arc<dyn Fn(&Arc<GlobalEnv>, &str) -> EvalResult<bool> + Send + Sync>;
199
200/// Source of [`GlobalEnv::id`] values.
201static NEXT_GLOBAL_ENV_ID: std::sync::atomic::AtomicU64 = std::sync::atomic::AtomicU64::new(1);
202
203impl std::fmt::Debug for GlobalEnv {
204 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
205 write!(f, "GlobalEnv {{ ... }}")
206 }
207}
208
209impl GlobalEnv {
210 /// Create an empty global environment for the given execution mode.
211 ///
212 /// This is the *raw* constructor: no builtins, no bootstrap, no source
213 /// paths. Use [`crate::Runtime::builder`] unless you are the builder.
214 pub fn new(execution_mode: ExecutionMode) -> Arc<Self> {
215 let id = NEXT_GLOBAL_ENV_ID.fetch_add(1, Ordering::Relaxed);
216 Arc::new(Self {
217 id,
218 namespaces: RwLock::new(HashMap::new()),
219 source_paths: RwLock::new(Vec::new()),
220 loaded: Mutex::new(std::collections::HashSet::new()),
221 loading: Mutex::new(HashMap::new()),
222 loading_done: Condvar::new(),
223 builtin_sources: RwLock::new(HashMap::new()),
224 gc_config: RwLock::new(None),
225 execution_mode,
226 tier_state: AtomicU8::new(TierState::TreeWalk as u8),
227 tiers: crate::tiered::tiers::Tiers::new(id),
228 async_rt: RwLock::new(None),
229 version_cache: Mutex::new(HashMap::new()),
230 deps_config: RwLock::new(None),
231 verify_commit_signatures: AtomicBool::new(false),
232 vcs: RwLock::new(crate::env::vcs::default_provider()),
233 sig_verify_cache: Mutex::new(HashSet::new()),
234 versioned_sources: RwLock::new(HashMap::new()),
235 versioned_offline: AtomicBool::new(false),
236 native_provenance: RwLock::new(HashMap::new()),
237 enforce_native_versions: AtomicBool::new(false),
238 provenance_warned: Mutex::new(HashSet::new()),
239 pinned_native_loader: RwLock::new(None),
240 native_require_loader: RwLock::new(None),
241 compiled_ns_loaders: RwLock::new(HashMap::new()),
242 })
243 }
244
245 /// Replace the source path list.
246 pub fn set_source_paths(&self, paths: Vec<std::path::PathBuf>) {
247 *self.source_paths.write().unwrap() = paths;
248 }
249
250 /// Register an embedded namespace source (called by cljrs-stdlib at startup).
251 pub fn register_builtin_source(&self, ns: &str, src: &'static str) {
252 self.builtin_sources
253 .write()
254 .unwrap()
255 .insert(Arc::from(ns), src);
256 }
257
258 /// Look up an embedded source for a namespace, if one has been registered.
259 pub fn builtin_source(&self, ns: &str) -> Option<&'static str> {
260 self.builtin_sources.read().unwrap().get(ns).copied()
261 }
262
263 /// Register a loader for an AOT-compiled namespace (called by the harness
264 /// `main` of a binary produced by `cljrs compile`).
265 pub fn register_compiled_ns_loader(&self, ns: &str, loader: CompiledNsLoader) {
266 self.compiled_ns_loaders
267 .write()
268 .unwrap()
269 .insert(Arc::from(ns), loader);
270 }
271
272 /// Look up the loader for an AOT-compiled namespace, if one is registered.
273 pub fn compiled_ns_loader(&self, ns: &str) -> Option<CompiledNsLoader> {
274 self.compiled_ns_loaders.read().unwrap().get(ns).cloned()
275 }
276
277 /// Mark a namespace as fully loaded from a file.
278 pub fn mark_loaded(&self, ns: &str) {
279 self.loaded.lock().unwrap().insert(Arc::from(ns));
280 }
281
282 /// True if the namespace has already been loaded from a file.
283 pub fn is_loaded(&self, ns: &str) -> bool {
284 self.loaded.lock().unwrap().contains(ns)
285 }
286
287 /// Set the GC configuration for automatic memory pressure management.
288 pub fn set_gc_config(&self, config: Arc<GcConfig>) {
289 *self.gc_config.write().unwrap() = Some(config);
290 }
291
292 /// Get the GC configuration, if one has been set.
293 pub fn gc_config(&self) -> Option<Arc<GcConfig>> {
294 self.gc_config.read().unwrap().clone()
295 }
296
297 /// Resolve a short alias to a full namespace name in `current_ns`.
298 pub fn resolve_alias(&self, current_ns: &str, alias: &str) -> Option<Arc<str>> {
299 let map = self.namespaces.read().unwrap();
300 let ns = map.get(current_ns)?;
301 let aliases = ns.get().aliases.lock().unwrap();
302 aliases.get(alias).cloned()
303 }
304
305 /// Resolve an auto-resolved keyword name (the text after `::`) to its
306 /// fully-qualified `ns/name` form.
307 ///
308 /// `::kw` qualifies with `current_ns` directly; `::alias/kw` looks
309 /// `alias` up in `current_ns`'s alias table (populated by `(require
310 /// '[... :as alias])`) and qualifies with the resolved namespace.
311 pub fn resolve_auto_keyword(&self, current_ns: &str, name: &str) -> Result<String, String> {
312 match name.split_once('/') {
313 Some((alias, kw_name)) => match self.resolve_alias(current_ns, alias) {
314 Some(ns) => Ok(format!("{ns}/{kw_name}")),
315 None => Err(format!(
316 "invalid token: ::{name} (no such namespace alias: {alias})"
317 )),
318 },
319 None => Ok(format!("{current_ns}/{name}")),
320 }
321 }
322
323 /// Return the namespace with this name, creating it if it doesn't exist.
324 pub fn get_or_create_ns(&self, name: &str) -> GcPtr<Namespace> {
325 // Fast path: already exists.
326 {
327 let map = self.namespaces.read().unwrap();
328 if let Some(ns) = map.get(name) {
329 return ns.clone();
330 }
331 }
332 // Slow path: insert.
333 let mut map = self.namespaces.write().unwrap();
334 // Re-check after acquiring write lock.
335 if let Some(ns) = map.get(name) {
336 return ns.clone();
337 }
338 let ns = GcPtr::new(Namespace::new(name));
339 map.insert(Arc::from(name), ns.clone());
340 ns
341 }
342
343 /// Intern `name` with `val` in the given namespace, returning the Var.
344 pub fn intern(&self, ns_name: &str, name: Arc<str>, val: Value) -> GcPtr<Var> {
345 let ns = self.get_or_create_ns(ns_name);
346 let mut interns = ns.get().interns.lock().unwrap();
347 if let Some(var) = interns.get(&name) {
348 // Update existing var.
349 var.get().bind(val);
350 return var.clone();
351 }
352 let var = GcPtr::new(Var::new(ns_name, name.as_ref()));
353 var.get().bind(val);
354 interns.insert(name, var.clone());
355 var
356 }
357
358 /// Look up a Var in the named namespace (interns only).
359 pub fn lookup_var(&self, ns_name: &str, sym_name: &str) -> Option<GcPtr<Var>> {
360 let map = self.namespaces.read().unwrap();
361 let ns = map.get(ns_name)?;
362 let interns = ns.get().interns.lock().unwrap();
363 interns.get(sym_name).cloned()
364 }
365
366 /// Look up a value in `ns_name`: checks interns then refers.
367 /// Routes through the dynamic binding stack so `binding` overrides work.
368 pub fn lookup_in_ns(&self, ns_name: &str, sym_name: &str) -> Option<Value> {
369 let map = self.namespaces.read().unwrap();
370 let ns = map.get(ns_name)?;
371 let ns_ref = ns.get();
372 // Check interns first.
373 {
374 let interns = ns_ref.interns.lock().unwrap();
375 if let Some(var) = interns.get(sym_name) {
376 return crate::env::dynamics::deref_var(var);
377 }
378 }
379 // Then refers.
380 {
381 let refers = ns_ref.refers.lock().unwrap();
382 if let Some(var) = refers.get(sym_name) {
383 return crate::env::dynamics::deref_var(var);
384 }
385 }
386 None
387 }
388
389 /// Look up the raw Var (not its value) in `ns_name`: interns then refers.
390 pub fn lookup_var_in_ns(&self, ns_name: &str, sym_name: &str) -> Option<GcPtr<Var>> {
391 let map = self.namespaces.read().unwrap();
392 let ns = map.get(ns_name)?;
393 let ns_ref = ns.get();
394 {
395 let interns = ns_ref.interns.lock().unwrap();
396 if let Some(var) = interns.get(sym_name) {
397 return Some(var.clone());
398 }
399 }
400 {
401 let refers = ns_ref.refers.lock().unwrap();
402 if let Some(var) = refers.get(sym_name) {
403 return Some(var.clone());
404 }
405 }
406 None
407 }
408
409 /// Copy all interns from `src_ns` into `dst_ns` as refers.
410 ///
411 /// This is the *explicit* refer — `(:require [x :refer :all])` — so it is
412 /// never narrowed by `dst_ns`'s `(:refer-clojure ...)` filter, matching
413 /// `clojure.core/refer`: naming a namespace explicitly re-maps even names
414 /// an earlier `refer-clojure` left out. The automatic core refer every
415 /// namespace starts with goes through [`GlobalEnv::refer_core`] instead.
416 pub fn refer_all(&self, dst_ns: &str, src_ns: &str) {
417 let map = self.namespaces.read().unwrap();
418 let src = match map.get(src_ns) {
419 Some(ns) => ns.clone(),
420 None => return,
421 };
422 let dst = match map.get(dst_ns) {
423 Some(ns) => ns.clone(),
424 None => return,
425 };
426 let src_interns = src.get().interns.lock().unwrap();
427 let mut dst_refers = dst.get().refers.lock().unwrap();
428 for (name, var) in src_interns.iter() {
429 dst_refers.insert(name.clone(), var.clone());
430 }
431 }
432
433 /// Apply the automatic `clojure.core` refer that every namespace starts
434 /// with, narrowed by `dst_ns`'s `(:refer-clojure ...)` filter.
435 pub fn refer_core(&self, dst_ns: &str) {
436 self.refer_core_impl(dst_ns, false);
437 }
438
439 /// `replace`: drop the refers `dst_ns` already inherited from
440 /// `clojure.core` before re-referring, under the same lock — so installing
441 /// a filter after the namespace was pre-referred neither leaves stale
442 /// names behind nor exposes a window where core is only half-referred.
443 fn refer_core_impl(&self, dst_ns: &str, replace: bool) {
444 let map = self.namespaces.read().unwrap();
445 let src = match map.get("clojure.core") {
446 Some(ns) => ns.clone(),
447 None => return,
448 };
449 let dst = match map.get(dst_ns) {
450 Some(ns) => ns.clone(),
451 None => return,
452 };
453 // Lock order is filter → src interns → dst refers throughout.
454 let filter = dst.get().refer_clojure_filter.lock().unwrap();
455 let src_interns = src.get().interns.lock().unwrap();
456 let mut dst_refers = dst.get().refers.lock().unwrap();
457 if replace {
458 dst_refers.retain(|_, var| var.get().namespace.as_ref() != "clojure.core");
459 }
460 match filter.as_ref() {
461 Some(f) => {
462 for (name, var) in src_interns.iter() {
463 if let Some(local) = f.local_name(name) {
464 dst_refers.insert(local, var.clone());
465 }
466 }
467 }
468 None => {
469 for (name, var) in src_interns.iter() {
470 dst_refers.insert(name.clone(), var.clone());
471 }
472 }
473 }
474 }
475
476 /// Install `dst_ns`'s `(:refer-clojure ...)` filter (`None` removes any
477 /// previous one) and re-apply the automatic `clojure.core` refer under it.
478 ///
479 /// Refers already inherited from `clojure.core` are dropped first, so a
480 /// filter set after the namespace was pre-referred — the loader refers core
481 /// before it reads the file, and `ns` itself refers core before it reaches
482 /// the clause — still takes effect.
483 pub fn set_refer_clojure_filter(
484 &self,
485 dst_ns: &str,
486 filter: Option<ReferClojureFilter>,
487 ) -> Result<(), String> {
488 if let Some(f) = &filter {
489 self.validate_refer_clojure_filter(f)?;
490 }
491 let dst = self.get_or_create_ns(dst_ns);
492 {
493 let mut slot = dst.get().refer_clojure_filter.lock().unwrap();
494 // Nothing to install and nothing to undo: leave the refers alone.
495 if slot.is_none() && filter.is_none() {
496 return Ok(());
497 }
498 *slot = filter;
499 }
500 self.refer_core_impl(dst_ns, true);
501 Ok(())
502 }
503
504 /// Check a `(:refer-clojure ...)` filter against the names `clojure.core`
505 /// actually publishes, so a typo fails at the `ns` form rather than as an
506 /// unbound symbol somewhere further down the file.
507 ///
508 /// `:only` and `:rename` name specific vars and must resolve; `:exclude` is
509 /// subtractive and stays permissive (excluding a name core does not have is
510 /// harmless, and lets a file stay portable across core versions). Clojure
511 /// validates `:only` the same way but ignores an unresolvable `:rename`
512 /// key, since it only consults the rename map for names already in its
513 /// to-do list.
514 ///
515 /// Two names landing on the same local name is an error rather than a coin
516 /// flip: with `:rename {inc str}` both `inc` and core's own `str` want the
517 /// name `str`. Clojure warns and lets whichever one its intern table
518 /// yields last win; picking a winner by hash order here would make the
519 /// choice unstable from run to run.
520 fn validate_refer_clojure_filter(&self, filter: &ReferClojureFilter) -> Result<(), String> {
521 let map = self.namespaces.read().unwrap();
522 let Some(core) = map.get("clojure.core").cloned() else {
523 return Ok(());
524 };
525 drop(map);
526 let interns = core.get().interns.lock().unwrap();
527 // A runtime built without the core bootstrap has nothing to check
528 // against; do not fail every name.
529 if interns.is_empty() {
530 return Ok(());
531 }
532
533 for (opt, names) in [
534 ("only", filter.only.iter().flatten().collect::<Vec<_>>()),
535 ("rename", filter.rename.keys().collect::<Vec<_>>()),
536 ] {
537 let mut unknown: Vec<&str> = names
538 .into_iter()
539 .filter(|n| !interns.contains_key(*n))
540 .map(|n| n.as_ref())
541 .collect();
542 if !unknown.is_empty() {
543 unknown.sort_unstable();
544 return Err(format!(
545 ":refer-clojure :{opt} names {}, which clojure.core does not define",
546 unknown.join(", ")
547 ));
548 }
549 }
550
551 // local name → the core name referred under it.
552 let mut taken: HashMap<Arc<str>, Arc<str>> = HashMap::new();
553 let mut conflicts: Vec<(Arc<str>, Arc<str>, Arc<str>)> = Vec::new();
554 for name in interns.keys() {
555 let Some(local) = filter.local_name(name) else {
556 continue;
557 };
558 if let Some(prev) = taken.insert(local.clone(), name.clone()) {
559 let (a, b) = if prev.as_ref() <= name.as_ref() {
560 (prev, name.clone())
561 } else {
562 (name.clone(), prev)
563 };
564 conflicts.push((local, a, b));
565 }
566 }
567 if !conflicts.is_empty() {
568 conflicts.sort_unstable();
569 let (local, a, b) = &conflicts[0];
570 return Err(format!(
571 ":refer-clojure would refer both {a} and {b} as {local}; \
572 rename or exclude one of them"
573 ));
574 }
575 Ok(())
576 }
577
578 /// Names `ns_name` resolves to something other than `clojure.core`'s var
579 /// of the same name — what the IR lowerer must not inline as a builtin.
580 ///
581 /// Three things put a name in here: a `def` in `ns_name` itself, a refer
582 /// of that name from another namespace (or of a *different* core name
583 /// under it, via `:rename`), and a `(:refer-clojure ...)` filter that
584 /// leaves the name out of the automatic core refer. In each case an
585 /// unqualified call resolves to something that is not `clojure.core/name`
586 /// — or to nothing at all — and the tree-walking interpreter calls that,
587 /// so the lowered call site has to as well (issue #337).
588 ///
589 /// A name absent from the namespace's tables is *not* reported: with no
590 /// evidence to the contrary the lowerer keeps assuming core, which is what
591 /// the automatic core refer makes true for the overwhelming majority of
592 /// call sites (and keeps the lowerer's synthetic names, e.g. `case=`,
593 /// working in namespaces whose refers are not populated yet).
594 pub fn core_shadowed_names(&self, ns_name: &str) -> HashSet<Arc<str>> {
595 let mut out: HashSet<Arc<str>> = HashSet::new();
596 let (ns, core) = {
597 let map = self.namespaces.read().unwrap();
598 match map.get(ns_name) {
599 Some(ns) => (ns.clone(), map.get("clojure.core").cloned()),
600 None => return out,
601 }
602 };
603 let ns_ref = ns.get();
604
605 // A var bound here is core's only if it *is* core's var under its own
606 // name; core's own namespace passes this trivially for its interns.
607 let shadows = |name: &Arc<str>, var: &GcPtr<Var>| {
608 let v = var.get();
609 v.namespace.as_ref() != "clojure.core" || v.name != *name
610 };
611 for table in [&ns_ref.interns, &ns_ref.refers] {
612 let entries = table.lock().unwrap();
613 for (name, var) in entries.iter() {
614 if shadows(name, var) {
615 out.insert(name.clone());
616 }
617 }
618 }
619
620 // Names the `(:refer-clojure ...)` filter kept out of the automatic
621 // refer: nothing binds them here, so they are not core's either.
622 // Lock order is filter → core interns, as in `refer_core_impl`.
623 let filter = ns_ref.refer_clojure_filter.lock().unwrap();
624 if let (Some(filter), Some(core)) = (filter.as_ref(), core) {
625 let interns = core.get().interns.lock().unwrap();
626 for name in interns.keys() {
627 if filter.local_name(name).as_ref() != Some(name) {
628 out.insert(name.clone());
629 }
630 }
631 }
632 out
633 }
634
635 /// Copy selected interns from `src_ns` into `dst_ns` as refers.
636 pub fn refer_named(&self, dst_ns: &str, src_ns: &str, names: &[Arc<str>]) {
637 let map = self.namespaces.read().unwrap();
638 let src = match map.get(src_ns) {
639 Some(ns) => ns.clone(),
640 None => return,
641 };
642 let dst = match map.get(dst_ns) {
643 Some(ns) => ns.clone(),
644 None => return,
645 };
646 let src_interns = src.get().interns.lock().unwrap();
647 let mut dst_refers = dst.get().refers.lock().unwrap();
648 for name in names {
649 if let Some(var) = src_interns.get(name) {
650 // Use insert (not or_insert_with) so that an explicit
651 // `require :refer [name]` always overrides a previous refer
652 // (e.g. one inherited from clojure.core via refer-all).
653 // clojure.core.async's `into` intentionally shadows clojure.core/into;
654 // or_insert_with would silently drop the override.
655 dst_refers.insert(name.clone(), var.clone());
656 }
657 }
658 }
659
660 /// Register `alias` → `full_ns` in `current_ns`'s alias table.
661 pub fn add_alias(&self, current_ns: &str, alias: &str, full_ns: &str) {
662 let ns_ptr = self.get_or_create_ns(current_ns);
663 let mut aliases = ns_ptr.get().aliases.lock().unwrap();
664 aliases.insert(Arc::from(alias), Arc::from(full_ns));
665 }
666
667 /// Process-unique identity of this runtime instance.
668 #[inline(always)]
669 pub fn id(&self) -> u64 {
670 self.id
671 }
672
673 /// This runtime's Tier-1/Tier-2 state.
674 #[inline(always)]
675 pub fn tiers(&self) -> &Arc<crate::tiered::tiers::Tiers> {
676 &self.tiers
677 }
678
679 /// This runtime's cache of lowered IR.
680 #[inline(always)]
681 pub fn ir_cache(&self) -> &crate::tiered::ir_cache::IrCache {
682 self.tiers.ir_cache()
683 }
684
685 /// This runtime's JIT counters, profiles, and native-code tables.
686 #[inline(always)]
687 pub fn jit(&self) -> &crate::tiered::jit_state::JitState {
688 self.tiers.jit()
689 }
690
691 /// The JIT compiler attached to this runtime, if any.
692 ///
693 /// `None` when no JIT is linked or installed; callers then keep to the
694 /// interpreter tiers. Installed by `cljrs_compiler::jit::install`.
695 #[inline(always)]
696 pub fn jit_backend(&self) -> Option<Arc<dyn crate::tiered::backend::JitBackend>> {
697 self.tiers.jit().backend().cloned()
698 }
699
700 // ── Execution mode and tier state ────────────────────────────────────
701
702 /// How this runtime executes function calls.
703 #[inline(always)]
704 pub fn execution_mode(&self) -> ExecutionMode {
705 self.execution_mode
706 }
707
708 /// Which tiers are live right now.
709 #[inline(always)]
710 pub fn tier_state(&self) -> TierState {
711 TierState::from_u8(self.tier_state.load(Ordering::Acquire))
712 }
713
714 /// Raise the live tier state. Called once by the runtime builder after
715 /// the bootstrap completes; lowering the tier is not supported, so a
716 /// request below the current state is ignored.
717 pub fn set_tier_state(&self, tier: TierState) {
718 let _ = self.tier_state.fetch_max(tier as u8, Ordering::AcqRel);
719 }
720
721 /// True when IR may be lowered, cached, and interpreted. This is the
722 /// gate the old `compiler_ready` flag served.
723 #[inline(always)]
724 pub fn ir_enabled(&self) -> bool {
725 self.tier_state().ir_enabled()
726 }
727
728 // ── Evaluation entry points ──────────────────────────────────────────
729
730 /// Evaluate `form` in `env`.
731 #[inline(always)]
732 pub fn eval(&self, form: &Form, env: &mut Env) -> EvalResult {
733 crate::interp::eval::eval(form, env)
734 }
735
736 /// Call a Clojure function, taking the path this runtime's
737 /// [`ExecutionMode`] selects.
738 ///
739 /// This is the single function-call dispatch point: tree walk, tier-1 IR,
740 /// and JIT-native execution are all reached from here.
741 #[inline(always)]
742 pub fn call_cljrs_fn(&self, func: &CljxFn, args: &[Value], env: &mut Env) -> EvalResult {
743 match self.execution_mode {
744 ExecutionMode::TreeWalk => crate::interp::apply::call_cljrs_fn(func, args, env),
745 ExecutionMode::Tiered | ExecutionMode::TieredNoJit => {
746 crate::tiered::apply::call_cljrs_fn(func, args, env)
747 }
748 ExecutionMode::NoGcTransaction => crate::env::depth::call_cljrs_fn(func, args, env),
749 }
750 }
751
752 /// Notify the active tier that a new `fn*` was defined.
753 ///
754 /// In a tiered runtime with IR enabled this eagerly lowers the function
755 /// (when eager lowering is on); in every other mode it does nothing.
756 #[inline(always)]
757 pub fn on_fn_defined(&self, f: &CljxFn, env: &mut Env) {
758 if self.execution_mode.is_tiered() && self.ir_enabled() {
759 crate::tiered::ir_interp::eager_lower_fn(f, env);
760 }
761 }
762
763 /// Install an async runtime. Called once by `cljrs_async::init`.
764 /// Subsequent calls are silently ignored (first writer wins).
765 pub fn set_async_runtime(&self, rt: Arc<dyn AsyncRuntime>) {
766 let mut guard = self.async_rt.write().unwrap();
767 if guard.is_none() {
768 *guard = Some(rt);
769 }
770 }
771
772 /// Return the async runtime, if one has been registered.
773 pub fn async_runtime(&self) -> Option<Arc<dyn AsyncRuntime>> {
774 self.async_rt.read().unwrap().clone()
775 }
776
777 /// Return `(source_file, git_repo_root)` for the named namespace, if
778 /// both have been populated by the loader.
779 pub fn get_ns_git_context(&self, ns_name: &str) -> Option<(Arc<str>, Arc<str>)> {
780 let map = self.namespaces.read().unwrap();
781 let ns = map.get(ns_name)?;
782 let ns_ref = ns.get();
783 let file = ns_ref.source_file.lock().unwrap().clone()?;
784 let repo = ns_ref.git_repo_root.lock().unwrap().clone()?;
785 Some((file, repo))
786 }
787
788 /// Store a resolved versioned value in the cache.
789 /// Key: `"<ns>/<name>@<commit>"`.
790 pub fn cache_versioned(&self, ns: &str, name: &str, commit: &str, val: Value) {
791 let key: Arc<str> = Arc::from(format!("{ns}/{name}@{commit}"));
792 self.version_cache.lock().unwrap().insert(key, val);
793 }
794
795 /// Retrieve a previously resolved versioned value, if cached.
796 pub fn get_cached_versioned(&self, ns: &str, name: &str, commit: &str) -> Option<Value> {
797 let key = format!("{ns}/{name}@{commit}");
798 self.version_cache
799 .lock()
800 .unwrap()
801 .get(key.as_str())
802 .cloned()
803 }
804
805 /// Mark namespace `name@commit` as loaded in the standard loaded set.
806 pub fn cache_versioned_ns(&self, ns: &str, commit: &str) {
807 let key: Arc<str> = Arc::from(format!("{ns}@{commit}"));
808 self.version_cache.lock().unwrap().insert(key, Value::Nil);
809 }
810
811 /// Record the source text of a versioned namespace fetched from git.
812 /// Key: `"<ns>@<commit>"`. Consumed by the AOT compiler for embedding.
813 pub fn record_versioned_source(&self, versioned_ns: &str, src: &str) {
814 self.versioned_sources
815 .write()
816 .unwrap()
817 .insert(Arc::from(versioned_ns), Arc::from(src));
818 }
819
820 /// Snapshot of all versioned sources fetched this session, sorted by key.
821 pub fn versioned_sources_snapshot(&self) -> Vec<(Arc<str>, Arc<str>)> {
822 let map = self.versioned_sources.read().unwrap();
823 let mut entries: Vec<_> = map.iter().map(|(k, v)| (k.clone(), v.clone())).collect();
824 entries.sort_by(|a, b| a.0.cmp(&b.0));
825 entries
826 }
827
828 /// Restrict versioned-namespace resolution to embedded builtin sources
829 /// (no git). Called by AOT harness binaries, which embed every pinned
830 /// source discovered at compile time.
831 pub fn set_versioned_offline(&self, offline: bool) {
832 self.versioned_offline.store(offline, Ordering::Relaxed);
833 }
834
835 /// True when versioned namespaces may only come from embedded sources.
836 pub fn versioned_offline(&self) -> bool {
837 self.versioned_offline.load(Ordering::Relaxed)
838 }
839
840 /// Record the git commit a native (Rust-backed) package was built from.
841 /// Called at registration time (`Registry::set_provenance` or the
842 /// `register_provenance!` inventory entry in cljrs-interop).
843 pub fn set_native_provenance(&self, ns: &str, commit: &str) {
844 self.native_provenance
845 .write()
846 .unwrap()
847 .insert(Arc::from(ns), Arc::from(commit));
848 }
849
850 /// The recorded provenance commit for a native package's namespace.
851 pub fn native_provenance_for(&self, ns: &str) -> Option<Arc<str>> {
852 self.native_provenance.read().unwrap().get(ns).cloned()
853 }
854
855 /// Make pinned-native provenance mismatches hard errors.
856 pub fn set_enforce_native_versions(&self, enforce: bool) {
857 self.enforce_native_versions
858 .store(enforce, Ordering::Relaxed);
859 }
860
861 /// True when pinned-native provenance mismatches are errors.
862 pub fn enforce_native_versions(&self) -> bool {
863 self.enforce_native_versions.load(Ordering::Relaxed)
864 }
865
866 /// Install the pinned-native package loader (called once by
867 /// `cljrs::native::pinned::install`; first writer wins).
868 pub fn set_pinned_native_loader(&self, loader: PinnedNativeLoader) {
869 let mut guard = self.pinned_native_loader.write().unwrap();
870 if guard.is_none() {
871 *guard = Some(loader);
872 }
873 }
874
875 /// Install the native-dependency `require` loader (called once by
876 /// `cljrs::native::pinned::install`; first writer wins).
877 pub fn set_native_require_loader(&self, loader: NativeRequireLoader) {
878 let mut guard = self.native_require_loader.write().unwrap();
879 if guard.is_none() {
880 *guard = Some(loader);
881 }
882 }
883
884 /// The installed VCS backend, or `None` when this build has none (see
885 /// [`crate::env::vcs`]). Callers must degrade gracefully: "no provider"
886 /// means "this source file is not in a git repository".
887 pub fn vcs(&self) -> Option<Arc<dyn crate::env::vcs::VcsProvider>> {
888 self.vcs.read().unwrap().clone()
889 }
890
891 /// Replace the VCS backend. Lets an embedder that built without the
892 /// `deps` feature supply its own git implementation, or a sandboxed host
893 /// remove the default one (`None`) so no versioned resolution can reach
894 /// the filesystem's git history.
895 ///
896 /// Drops every cached signature verdict: those were reached by the
897 /// outgoing provider, against its trust set and its view of the
898 /// repository, and say nothing about what the incoming one would decide
899 /// for the same `(repo, commit)`. Keeping them would let a permissive
900 /// provider launder an approval for source a later provider serves.
901 pub fn set_vcs_provider(&self, provider: Option<Arc<dyn crate::env::vcs::VcsProvider>>) {
902 // Neither this nor `check_commit_signature` ever holds the `vcs` and
903 // `sig_verify_cache` locks at the same time, and the two reach for
904 // them in opposite orders — keep it that way, or the pair becomes a
905 // lock-order inversion.
906 *self.vcs.write().unwrap() = provider;
907 self.invalidate_signature_cache();
908 }
909
910 /// Forget every cached signature verdict, so the next
911 /// [`check_commit_signature`](Self::check_commit_signature) re-asks the
912 /// current provider. Called whenever the thing that produced those
913 /// verdicts changes: the provider itself, or its trusted-key set.
914 pub fn invalidate_signature_cache(&self) {
915 self.sig_verify_cache.lock().unwrap().clear();
916 }
917
918 /// If `:verify-commit-signatures` is enabled, verify that `commit` inside
919 /// `repo_root` carries a valid GPG or SSH signature.
920 ///
921 /// Returns `Ok(())` immediately when the feature is off. On the happy
922 /// path the result is cached per `(repo_root, commit)` so each commit is
923 /// only verified once per session. On failure returns
924 /// `EvalError::CommitSignatureVerificationFailed`.
925 ///
926 /// If verification is demanded but this build has no VCS provider, the
927 /// check fails: silently accepting an unverifiable commit would defeat the
928 /// flag the user explicitly turned on.
929 pub fn check_commit_signature(&self, repo_root: &str, commit: &str) -> EvalResult<()> {
930 if !self.verify_commit_signatures.load(Ordering::Relaxed) {
931 return Ok(());
932 }
933 let key = (Arc::<str>::from(repo_root), Arc::<str>::from(commit));
934 if self.sig_verify_cache.lock().unwrap().contains(&key) {
935 return Ok(());
936 }
937 let Some(vcs) = self.vcs() else {
938 return Err(crate::env::error::EvalError::Runtime(format!(
939 "commit-signature verification is enabled, but this build has no VCS \
940 provider to verify commit {commit} with (cljrs-runtime built without \
941 the `deps` feature)"
942 )));
943 };
944 vcs.verify_commit_signature(std::path::Path::new(repo_root), commit)
945 .map_err(|e| match e {
946 crate::env::vcs::SignatureFailure::Untrusted { commit, reason } => {
947 crate::env::error::EvalError::CommitSignatureVerificationFailed {
948 commit,
949 reason,
950 }
951 }
952 crate::env::vcs::SignatureFailure::Error(msg) => {
953 crate::env::error::EvalError::Runtime(msg)
954 }
955 })?;
956 self.sig_verify_cache.lock().unwrap().insert(key);
957 Ok(())
958 }
959
960 /// Build the trusted-signer key set from a parsed `cljrs.edn` config and
961 /// install it, so subsequent `check_commit_signature` calls verify against
962 /// it. Inline keys are parsed directly; `File` entries are read from disk.
963 /// Returns the number of keys loaded; warns (to stderr) on any key that
964 /// fails to load rather than aborting. Returns 0 when this build has no
965 /// VCS provider, since there is nothing that could consume the keys.
966 ///
967 /// Replacing the trust set invalidates the signature cache for the same
968 /// reason replacing the provider does: a verdict reached under the old
969 /// keys is not a verdict under the new ones. (In the normal flow this
970 /// runs at session start, before anything has been verified.)
971 pub fn load_trusted_signers(&self, config: &cljrs_project::config::DepsConfig) -> usize {
972 let Some(vcs) = self.vcs() else {
973 return 0;
974 };
975 let loaded = vcs.load_trusted_signers(&config.trusted_signers);
976 self.invalidate_signature_cache();
977 loaded
978 }
979}
980
981// ── Env ───────────────────────────────────────────────────────────────────────
982
983/// The full execution environment: a stack of local frames plus the global env.
984pub struct Env {
985 pub frames: Vec<Frame>,
986 pub current_ns: Arc<str>,
987 pub globals: Arc<GlobalEnv>,
988 /// When set, unversioned same-namespace symbol lookups implicitly resolve
989 /// at this commit hash instead of HEAD. Set by the versioned resolver when
990 /// evaluating a function body fetched from git history.
991 pub versioned_eval_commit: Option<Arc<str>>,
992 /// True when evaluating the body of an `^:async` function.
993 /// Set by `cljrs-async`; allows the `await` special form to know whether
994 /// to yield (async context) or block the OS thread (sync context).
995 pub is_async: bool,
996}
997
998impl Env {
999 pub fn new(globals: Arc<GlobalEnv>, ns: &str) -> Self {
1000 Self {
1001 frames: Vec::new(),
1002 current_ns: Arc::from(ns),
1003 globals,
1004 versioned_eval_commit: None,
1005 is_async: false,
1006 }
1007 }
1008
1009 /// Create an Env for evaluating source at a specific commit.
1010 pub fn new_versioned(globals: Arc<GlobalEnv>, ns: &str, commit: &str) -> Self {
1011 Self {
1012 versioned_eval_commit: Some(Arc::from(commit)),
1013 ..Self::new(globals, ns)
1014 }
1015 }
1016
1017 /// Create an Env pre-loaded with a function's closed-over bindings.
1018 pub fn with_closure(globals: Arc<GlobalEnv>, ns: &str, f: &CljxFn) -> Self {
1019 let mut env = Self::new(globals, ns);
1020 if !f.closed_over_names.is_empty() {
1021 env.push_frame();
1022 for (name, val) in f.closed_over_names.iter().zip(f.closed_over_vals.iter()) {
1023 env.bind(name.clone(), val.clone());
1024 }
1025 }
1026 env
1027 }
1028
1029 pub fn push_frame(&mut self) {
1030 self.frames.push(Frame::new());
1031 }
1032
1033 pub fn pop_frame(&mut self) {
1034 self.frames.pop();
1035 }
1036
1037 /// Bind `name` to `val` in the top frame.
1038 pub fn bind(&mut self, name: Arc<str>, val: Value) {
1039 if let Some(frame) = self.frames.last_mut() {
1040 frame.bind(name, val);
1041 }
1042 // If there are no frames, the binding is silently dropped.
1043 // Callers must push a frame first.
1044 }
1045
1046 /// Look up `name`: local frames (innermost first), then the current namespace.
1047 pub fn lookup(&self, name: &str) -> Option<Value> {
1048 tracing::trace!(target: "env", "lookup {} in {} frames", name, self.frames.len());
1049 for frame in self.frames.iter().rev() {
1050 if let Some(v) = frame.lookup(name) {
1051 return Some(v.clone());
1052 }
1053 }
1054 self.globals.lookup_in_ns(&self.current_ns, name)
1055 }
1056
1057 /// Look up `name` in local frames only — does **not** fall back to the
1058 /// global namespace. Used by the versioned resolver to check for local
1059 /// bindings before applying commit inheritance.
1060 pub fn lookup_local_frames(&self, name: &str) -> Option<Value> {
1061 for frame in self.frames.iter().rev() {
1062 if let Some(v) = frame.lookup(name) {
1063 return Some(v.clone());
1064 }
1065 }
1066 None
1067 }
1068
1069 /// Look up the Var object for `name` in the current namespace.
1070 pub fn lookup_var(&self, name: &str) -> Option<GcPtr<Var>> {
1071 self.globals.lookup_var_in_ns(&self.current_ns, name)
1072 }
1073
1074 /// Collect all current local bindings (all frames, innermost last).
1075 /// Used for closure capture.
1076 pub fn all_local_bindings(&self) -> (Vec<Arc<str>>, Vec<Value>) {
1077 let mut names = Vec::new();
1078 let mut vals = Vec::new();
1079 // Outermost first so inner frames override on lookup.
1080 for frame in &self.frames {
1081 for (n, v) in &frame.bindings {
1082 names.push(n.clone());
1083 vals.push(v.clone());
1084 }
1085 }
1086 (names, vals)
1087 }
1088
1089 /// Create a child Env for closure capture (same globals, same ns, captures locals).
1090 pub fn child(&self) -> Self {
1091 let (names, vals) = self.all_local_bindings();
1092 let mut child = Self::new(self.globals.clone(), &self.current_ns);
1093 child.is_async = self.is_async;
1094 if !names.is_empty() {
1095 child.push_frame();
1096 for (n, v) in names.into_iter().zip(vals) {
1097 child.bind(n, v);
1098 }
1099 }
1100 child
1101 }
1102
1103 #[inline(always)]
1104 pub fn eval(&mut self, form: &Form) -> EvalResult {
1105 let globals = self.globals.clone();
1106 globals.eval(form, self)
1107 }
1108
1109 #[inline(always)]
1110 pub fn call_cljrs_fn(&mut self, func: &CljxFn, args: &[Value]) -> EvalResult {
1111 let globals = self.globals.clone();
1112 globals.call_cljrs_fn(func, args, self)
1113 }
1114
1115 #[inline(always)]
1116 pub fn on_fn_defined(&mut self, func: &CljxFn) {
1117 let globals = self.globals.clone();
1118 globals.on_fn_defined(func, self);
1119 }
1120}