lemmascript 0.5.1 → 0.5.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md CHANGED
@@ -18,15 +18,24 @@ See the external case studies:
18
18
  - **[collab-todo-lemmascript](https://github.com/midspiral/collab-todo-lemmascript/)** — collaborative task management web app (React + Supabase) with a verified domain model. Single `domain.ts` imported directly by the UI, hooks, and edge functions — no adapter layer. 123 Dafny lemmas (120 in a separate `domain.proofs.dfy`): 16-conjunct invariant preserved across 25 single-project + 3 cross-project actions, NoOp completeness/soundness, initialization. Dafny only.
19
19
  - **[colorwheel-lemmascript](https://github.com/midspiral/colorwheel-lemmascript/)** — verified color palette generator with mood + harmony constraints. 31 Lean proofs + 18 behavioral properties, 115 Dafny lemmas (invariant preservation, commutativity, NoOp completeness).
20
20
  - **[clear-split-lemmascript](https://github.com/midspiral/clear-split-lemmascript/)** — greenfield verified expense splitting web app. Conservation theorem, invariant preservation, delta laws — all proven in both Lean (no sorry) and Dafny (56 lemmas).
21
+ - **[eventab-lemmascript](https://github.com/midspiral/eventab-lemmascript/)** — greenfield verified group bill-splitter, shipped as a single self-contained client-side page. Largest-remainder allocation in integer cents proves the money never leaks: per-person shares sum exactly to the tab, non-claimers pay nothing, and the settlement nets to zero even when payments are rounded to the dollar (the payer absorbs the difference). Dafny only.
21
22
  - **[github-star-checker-lemmascript](https://github.com/midspiral/github-star-checker-lemmascript/)** — small verified CLI that tracks GitHub star counts across repos and reports per-run deltas. Verifies: per-row diff correctness and `totalDiff == sumDiffs(rows)` (via an inductive `SumDiffs_append` lemma); three sign-classified extractors (gainers / losers / unchanged) with soundness, completeness, **ordered completeness** (gainers appear in the notification in the same order they were listed on the command line), and count/sum equalities against prefix-indexed `upTo` helpers; conservation theorem `decompose(r)` — the three splits partition every row exactly once, and `sumDiffs(increases) + sumDiffs(decreases) == totalDiff`. 33 Dafny VCs, 0 errors; proof additions include a head/tail bridge (`sumDiffs` ↔ `sumDiffsUpTo`) and two partition-on-n inductions. Dafny only.
22
23
  - **[equality-game-lemmascript](https://github.com/midspiral/equality-game-lemmascript/)** — greenfield verified arithmetic equality card game (React + Tailwind). Sound + complete decision procedure for "can these two card lists be combined into equal expressions": `canEqualize(L, R) ⟺ ∃ eL, eR. eval(eL) == eval(eR) ∧ multiset(leaves(eL)) == multiset(L) ∧ same for R`. Algorithm is subset-DP over a bitmask `m ∈ [1, 2^n − 1)`; the proof composes a `PopCount` upper/lower bound chain (with stdlib `LemmaDivDenominator` / `LemmaFundamentalDivModConverse`), a `splitLeft`/`splitRight` ↔ imperative-loop connection, a `WitnessCombine` lemma threading existential `Expr` witnesses through the cross-product loops, and a `ChooseMask` combinatorial constructor that, given any sub-multiset of `cards`, produces the realizing mask. Capped by `CompletenessFromMaskCoverage`. 753 verification conditions, 0 errors, 0 `assume`s, 0 axioms under `--isolate-assertions --verification-time-limit 180`. Dafny only.
23
24
  - **[talktimer-lemmascript](https://github.com/midspiral/talktimer-lemmascript/)** — verified talk timer React app, ported from a Dafny-only [`talktimer-lemmafit`](https://github.com/midspiral/talktimer-lemmafit/) twin. 17-variant `Action` state machine + verified `History` (undo/redo/preview/commitFrom) all in one `domain.ts` — the original Dafny's `Domain refines Kernel` abstract-module pattern inlined since LS has no abstract modules. 108 VCs in `domain.dfy` (invariant preservation) + 123 in `domain.proofs.dfy` (behavioral lemmas + Kernel round-trip). Dafny only.
24
25
  - **[node-casbin-lemmascript](https://github.com/midspiral/node-casbin-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [node-casbin](https://github.com/casbin/node-casbin). 5 functions verified, 217 existing tests pass. End-to-end correctness and order independence for all 4 effect modes in both Lean and Dafny (39 lemmas).
25
26
  - **[hono-lemmascript](https://github.com/midspiral/hono-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [hono](https://github.com/honojs/hono)'s security middleware. Four CVEs covered: IP restriction bypass ([CVE-2026-39409](https://github.com/honojs/hono/security/advisories/GHSA-3mpf-rcc7-5347)) and cookie name bypass ([CVE-2026-39410](https://github.com/honojs/hono/security/advisories/GHSA-r5rp-j6wh-rvv4)) — 51 Dafny lemmas, [cookie verification **in-place**](https://github.com/midspiral/hono-lemmascript/blob/lemmascript/src/utils/cookie.ts#L79); plus `serveStatic`'s URL-encoded directory traversal ([CVE-2024-32869](https://github.com/honojs/hono/security/advisories/GHSA-q5w7-8mq6-2hxq)) + repeated-slash bypass ([CVE-2026-39407](https://github.com/honojs/hono/security/advisories/GHSA-jw53-c2g8-vmwm)), proved as a *composition* — `decode(rawPath)` before `check(decoded)`, so a buggy implementation that reordered the steps would fail the proof. First use of `//@ assume` + `//@ havoc`-on-assign. Dafny only.
27
+ - **[hono-rate-limiter-with-lemmascript](https://github.com/midspiral/hono-rate-limiter-with-lemmascript)** — a greenfield verified feature shipped through a brownfield host's *middleware seam*: a standalone [Hono](https://hono.dev) rate-limit middleware whose admission decision is a machine-checked theorem. Every fixed-window "N per minute" limiter leaks the same way — a client fires `limit` requests just before a boundary and `limit` more just after, a 2× burst the docs swear can't happen. The verified core proves the crisp guarantee people actually want: over an integer-millisecond admission log, `admit` keeps `|log| ≤ limit` and window-faithful, and folding it over **any** monotone request stream (`run`) keeps the cumulative log `Spread` — so **no half-open window `(s, s+W]` ever admits more than `limit`** (`SlidingWindowBound`); the boundary burst is not mitigated but *impossible*. The naive cousin is refuted in the same file: `FixedWindowLeaks` exhibits a concrete trace where a fixed-window counter admits `2 × limit` in one sliding window while the verified `admit` rejects the overflow. An invariant-over-a-stateful-core proof (not a one-shot algorithm), wired into a live Hono server that gates real HTTP. 23 Dafny VCs, 0 errors; the clock's monotonicity and the per-key store's atomicity are the named trust boundary. Dafny only.
26
28
  - **[charmchat](https://github.com/CHARM-BDF/charmchat/blob/lemma/README_LemmaScript.md)** — brownfield verification of an AI agent orchestration backend. `isEmptyResult` (string emptiness predicate, 8 postconditions, <1s) and `topologicalSort` (Kahn's algorithm — memory safety, output bounds, completeness via acyclicity ranking witness, termination). Full completeness proof: 23 helper lemmas, 14 opaque ghost predicates, 115 loop invariants; 736 VCs verified under `--isolate-assertions --verification-time-limit 600`. Key technique: snapshot-based inner invariants (`ghost var originalRemDeps := remDeps`) replace the mid-iteration SEEN/UNSEEN split so preservation is frame reasoning against a ghost-constant rather than set-subtraction against mutating state. Dafny only.
27
- - **[xyflow-lemmascript](https://github.com/midspiral/xyflow-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [xyflow](https://github.com/xyflow/xyflow)'s core edge and geometry utilities. 9 functions verified: `addEdge` (dedup — never loses edges, adds at most one), `reconnectEdge` (semantic: under a unique-id precondition, the result is in-place — `|result| ≤ |edges|`, no insertion — *and* when a matching edge existed with non-empty new endpoints, the output contains an edge with those endpoints. Uses `//@ assume` to characterize destructuring, `find`, and the constructed edge), `connectionExists`, `getEdgeCenter` (midpoint correctness), `clamp` (bounds), `rectToBox`/`boxToRect` (field arithmetic), `getBoundsOfBoxes` (enclosure), `getOverlappingArea` (non-negative), `areSetsEqual` (subset + same size). 14 Dafny proof obligations. Dafny only.
29
+ - **[xyflow-lemmascript](https://github.com/midspiral/xyflow-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [xyflow](https://github.com/xyflow/xyflow)'s core edge and geometry utilities. 9 functions verified: `addEdge` (dedup — never loses edges, adds at most one), `reconnectEdge` (semantic: under a unique-id precondition, the result is in-place — `|result| ≤ |edges|`, no insertion — *and* when a matching edge existed with non-empty new endpoints, the output contains an edge with those endpoints. Uses `//@ assume` to characterize destructuring, `find`, and the constructed edge), `connectionExists`, `getEdgeCenter` (midpoint correctness), `clamp` (bounds), `rectToBox`/`boxToRect` (field arithmetic), `getBoundsOfBoxes` (enclosure), `getOverlappingArea` (non-negative), `areSetsEqual` (subset + same size). 14 Dafny proof obligations. Also adds a new verified feature — a **DAG connection gate**: `canReach` decides reachability soundly *and* completely, and `wouldCreateCycle` gates both new connections and edge reconnections so the graph is proven to stay acyclic, with `isAcyclic` (sound + complete) establishing the base case and a topological-rank witness giving a safe evaluation order (+29 obligations), shown live in a React Flow demo that refuses cycle-closing edges — extending the case study from verifying existing code to adding a verified feature. Dafny only.
28
30
  - **[rallly-lemmascript](https://github.com/midspiral/rallly-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [rallly](https://github.com/lukevella/rallly)'s meeting-poll Next.js app. 2 functions: `validateRedirectUrl` (in-place — open-redirect predicate; non-`undefined` outputs start with `/` but not `//`) and `scorePoll` (extracted ranking core — length preservation, score bounds, top-choice characterization, score-formula equality, within-poll monotonicity, tiebreaker injectivity). The injectivity proof surfaced a real spec-level constraint on the existing `(yes + ifNeedBe) * 1000 + yes` encoding: it overflows when an option has ≥ 1000 yes votes. 10 Dafny VCs, 0 errors. Drove four toolchain additions: `s.startsWith()`, `T | null` nullability, `\result` narrowing under `==>`, `Math.max(...arr)` spread. Dafny only.
29
31
  - **[opencode-lemmascript](https://github.com/midspiral/opencode-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield verification of [opencode](https://github.com/anomalyco/opencode)'s permission system and unified-diff patch parser. Highlights: (1) `Patch.parsePatch` carries conservation loop invariants over local ghost state — a parser bug here would silently corrupt user files when an AI applies a patch, and (2) the permission-engine work mechanically closes opencode bug #26514 (subagents bypassing Plan Mode's file-edit restrictions). 9 functions verified in-place, 0 errors. Dafny only.
32
+ - **[pi-lemmascript](https://github.com/midspiral/pi-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield, **in-place** verification of the context-compaction cut-point selector in **pi** (the [earendil-works](https://pi.dev) agent harness). When the context window fills, pi discards history before a chosen cut; a provider API rejects a retained prefix containing an orphaned `toolResult` (a tool result whose tool call was cut away). Both selector functions proven: the cut never lets the kept suffix *start with* — nor *split a tool-use/tool-result run* into — an orphaned tool result, even across the backward metadata snap. The no-orphan result forced the session tree's tool-pairing ordering into an explicit `requires`. 4 VCs, 0 errors. Drove five toolchain additions, headlined by an **opaque fall-through type**: a union LemmaScript can't discriminate (here an array-element union of unreachable imports) becomes a single opaque `type` — the field stays present so distinct values stay distinct, and with no constructor or tag predicate it can only be passed through, never unsoundly observed. Dafny only.
33
+ - **[balanced-match-lemmascript](https://github.com/midspiral/balanced-match-lemmascript/blob/lemmascript/README_LemmaScript.md)** — brownfield, **in-place** verification of [balanced-match](https://github.com/juliangruber/balanced-match), the ~70-line balanced-bracket finder pulled in by `npm`, `webpack`, and most of the JS tooling stack (1B+ downloads/month). The stack-based `range` core is verified by **refinement**: a pure recursive spec `range_spec` mirrors the loop one branch per recursive case, so the single equivalence `range == range_spec` transfers every property automatically — including an unconditional Dyck body-balance theorem for the interior of every returned pair. 2233 VCs, 0 errors under `--isolate-assertions` (registered on the `dafny-slow` track). Dafny only.
34
+ - **[guardians-lemmascript](https://github.com/midspiral/guardians-lemmascript)** — greenfield verification of the core safety argument behind [Guardians](https://github.com/metareflection/guardians) (Erik Meijer, "Guardians of the Agents", CACM Jan 2026), a generate-verify-execute checker for AI-agent workflows. Instead of verifying an app, it proves the agent *guardrail itself sound* — that a static taint/automaton check over the real recursive workflow AST can never admit an unsafe plan. Highlights: taint over **nested conditionals** as a sound branch-union over-approximation; per-source **provenance** with a join (a multi-input tool is tainted if *any* input was); **unbounded loops** discharged by a one-step pre-fixpoint (`sat = t0 ‖ bodyTaint(t0)`) that bounds taint over any iteration count without iterating to a fixpoint; and a **unified capstone** (`verifyWfSound`) — one clean verdict rules out, on *every* path, both a tainted-data-to-sink leak and a security-automaton error. 54 Dafny obligations, 0 errors. The verified cores are reached from a Guardians-style `Workflow`/`Policy` through a thin *unverified* adapter, differentially tested against the real Python Guardians (used as the oracle, not a porting target). Dafny only.
35
+ - **[quorum-lemmascript](https://github.com/midspiral/quorum-lemmascript)** — greenfield verified when2meet/Doodle-style group scheduler (React + Cloudflare Durable Objects), with one `domain.ts` running unchanged in the browser, the in-app query, and the server. The standout is that **the proof licenses the architecture**: `countFree` is a homomorphism from participant-list concatenation to integer addition (so the heatmap is order-independent) plus same-participant last-writer-wins convergence — which is exactly what makes the lock-free, no-login, *optimistic* multi-device backend safe, with the Durable Object and the browser applying the **same** verified `applyOp` (server-authoritatively, client-optimistically) with no rollback or operational transform. Also: heatmap is exactly the per-slot count and `isBest` exactly its argmax; monotonicity; invariant-preserving mutations + op-log `replay`; a sparse export codec round-trip; an in-app `whoIsFree(e, s)` whose length provably equals the cell's count; and a separate `grid.ts` proving the `(day, time) → slot` map in-range + injective — which makes specific-dates-vs-days-of-the-week pure shell labeling at zero proof cost; and full element-level permutation invariance (`heatmapPermInvariant` — the heatmap depends only on the *multiset* of participant rows), which drove the `perm(...)` spec predicate into LemmaScript itself. 100 Dafny VCs (90 + 10), 0 errors. The *aggregate* is proven; the React UI, WebSocket/DO I/O, and timezone labeling are the stated trust boundary. Dafny only.
36
+ - **[quota-lemmascript](https://github.com/midspiral/quota-lemmascript)** — greenfield verified booking app: providers publish a page of limited-capacity *featured slots*, signed-in users grab them (React + Cloudflare Durable Objects + D1). A deliberate **inverse** of [quorum-lemmascript](https://github.com/midspiral/quorum-lemmascript): where Quorum counts *up to* a threshold over data partitioned **per participant** (no conflicts ⇒ optimistic, lock-free, no rollback), Quota's bookings **contend** for shared inventory, so the load-bearing fact flips from a count to a **bound** — for every slot `j`, `confirmedCount(bookings, j) <= slots[j].capacity` — and so does the concurrency story: the *same* `domain.ts` runs in the browser and the Durable Object, but **server-authoritatively** (it never oversells under contention; no optimistic client apply). Proven: no overbooking (invariant preservation across `tryBook` / cancel), accept-iff-room, an **idempotent three-way `tryBook`** (a retry reads as success, not rejection — only "confirmed" mutates), cancellation frees seats, replay determinism, and full order-invariance of availability under contention — `confirmedCountPerm` / `hasRoomPermInvariant` show availability depends only on the *multiset* of the booking log (any reordering, not just a pairwise swap), via the same `perm(...)` predicate Quorum drove into LemmaScript. NDJSON export is built on the verified `confirmedOnly`. The counting kernel is Quorum's `countFree` re-pointed at bookings — total and precondition-free so it composes. 80 Dafny VCs, 0 errors. Trust boundary (stated plainly): auth, the React UI, WebSocket/DO/D1 I/O, email, slot date/time labeling, and abuse/rate-limiting. Dafny only.
37
+ - **[henri-lemmascript](https://github.com/midspiral/henri-lemmascript)** — a runnable TypeScript coding-agent CLI (a port of [henri](https://github.com/metareflection/henri/); multi-provider, Anthropic + AWS Bedrock) whose security- and protocol-critical core is verified and **imported directly by the live agent** — `decide()` gates every real tool call and the conversation invariant is asserted on every turn (it streams end-to-end against Bedrock). Three modules, 48 Dafny VCs, 0 errors. (1) **Permission gate**: soundness (`decide == Allow ⟺ isAllowed`), **path-traversal containment** — auto-allow-in-cwd can never resolve outside cwd, with `.`/`..` normalization proved in-core so the shell is trusted only to `resolve().split('/')` — grant monotonicity, and `rejectPrompts` is deny-only. (2) **Conversation protocol**: tool-call/result pairing plus the [pi-lemmascript](https://github.com/midspiral/pi-lemmascript)-style **no-orphaned-`tool_result`** property, proved as an invariant *preserved by the loop* — `wellFormed(msgs + [assistant(calls), tool(makeResults(calls))])` — not checked after the fact. (3) **Hook/config merge**: removal, **tool-name uniqueness — a fix** (henri concatenated hook tool lists with no dedup, so two hooks could shadow a name), order-independence, and additivity composed **cross-module** with the gate's monotonicity (merging only grows the allow-sets, which by P3 never revokes an `Allow`). The merge is verified **in place** via `//@ declare-type Tool { name: string }`, shadowing the real `Tool`'s function-valued `execute` so the actual `mergeTools(Tool[])` is the proof target rather than a parallel model. Dafny only.
38
+ - **[eslint-plugin-with-lemmascript](https://github.com/midspiral/eslint-plugin-with-lemmascript)** — a greenfield verified feature shipped through a brownfield host's *extension API*: a real, npm-published [ESLint](https://eslint.org) flat-config plugin whose rule decision is a machine-checked theorem. `no-forbidden-reach` enforces architecture boundaries — "the UI must never reach the DB layer" — through **any** import chain, catching the *laundered* `ui → service → db` violation that every one-hop incumbent (`import/no-restricted-paths`, `eslint-plugin-boundaries`, Nx module boundaries) silently passes. The verified core decides reachability soundly *and* completely (`reachesAny` / `violates`) and **constructs** the offending chain in proven code (`findReachPath` — a path-carrying BFS proven sound + complete by mirroring the frontier's endpoints in a ghost seq, so completeness reduces to the same closure argument as the reachability search; the chain printed in the lint error is therefore itself a *verified* import path, not a heuristic guess). The headline is a meta-theorem — `Domination` + `Strictness` — proving the transitive check **strictly dominates** one-hop checking: every direct violation is caught, *and* there provably exist laundered violations that direct-edge checks miss. 30 Dafny VCs, 0 errors. The reachability decision and the witness are proven; the import-graph extraction (which edges exist) is the stated trust boundary, and `dist/*.js` is `tsc`'s erasure of the verified source it ships alongside. Dafny only.
30
39
 
31
40
  ## Setup
32
41
 
@@ -71,9 +80,7 @@ npx lsc gen --backend=lean src/myModule.ts
71
80
  lake build
72
81
  ```
73
82
 
74
- ## What's Supported
75
-
76
- ### Annotations
83
+ ## Annotations
77
84
 
78
85
  ```typescript
79
86
  //@ requires arr.length > 0
@@ -83,6 +90,8 @@ lake build
83
90
  //@ type i nat
84
91
  ```
85
92
 
93
+ For the full surface, see [SPEC.md](SPEC.md).
94
+
86
95
  ## File Structure
87
96
 
88
97
  ### Dafny backend
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "lemmascript",
3
- "version": "0.5.1",
3
+ "version": "0.5.3",
4
4
  "description": "A verification toolchain for TypeScript — generates Lean 4 or Dafny from annotated TS",
5
5
  "type": "module",
6
6
  "engines": {
@@ -23,7 +23,7 @@
23
23
  },
24
24
  "devDependencies": {
25
25
  "@types/node": "^25.5.0",
26
- "tsx": "^4.0.0",
26
+ "tsx": "^4.22.4",
27
27
  "typescript": "^5.7.0"
28
28
  },
29
29
  "license": "MIT",
@@ -32,6 +32,7 @@
32
32
  "url": "https://github.com/midspiral/LemmaScript"
33
33
  },
34
34
  "keywords": [
35
+ "lemmascript",
35
36
  "verification",
36
37
  "typescript",
37
38
  "lean4",
@@ -0,0 +1,536 @@
1
+ /**
2
+ * auto-havoc pass — validation mode.
3
+ *
4
+ * Rewrites every *unmodellable* expression in the Typed IR to a `havoc` node,
5
+ * and drops unmodellable side-effecting statements. "Unmodellable" = the value
6
+ * derives from something outside LemmaScript's model (an `unknown`-typed
7
+ * receiver/result — e.g. `res.status()`, `req.body`, `JSON.parse(x)`,
8
+ * `process.env.X ?? y`, `uuidv4()` — or an anonymous object literal).
9
+ *
10
+ * This is the abstraction policy for verifying *validation* properties: the
11
+ * only thing that matters is that values reaching a contracted sink (an extern
12
+ * with `//@ requires sanitizer(x)`) are dominated by the sanitizer guard.
13
+ * Everything else is noise and is replaced by an arbitrary value.
14
+ *
15
+ * Soundness: `havoc` is a nondeterministic over-approximation — the verifier
16
+ * assumes nothing about a havoc'd value — so this can only make a proof FAIL
17
+ * (false positive), never spuriously pass. It uses havoc, never assume.
18
+ * (Trust boundary: a real sink hidden inside an unmodellable call is invisible;
19
+ * sinks must be the declared/annotated externs.)
20
+ *
21
+ * It runs after narrow, so any legitimate optional (`?.`, `??` on an
22
+ * optional-typed value) is already a `someMatch`; a `nullish`/`optChain` that
23
+ * survives to here is unmodellable and gets havoc'd.
24
+ */
25
+ /** A node whose own value can't be modeled and must become an arbitrary value.
26
+ * Leaves (var/literal/havoc) are never havoc'd — a reference to an already
27
+ * havoc'd variable is fine; only the *source* expression is replaced. */
28
+ function isBadNode(e) {
29
+ switch (e.kind) {
30
+ case "var":
31
+ case "num":
32
+ case "str":
33
+ case "bool":
34
+ case "havoc":
35
+ return false;
36
+ // A field/index is unmodellable when its *object* is opaque (`req.body`,
37
+ // `process.env.X`). It is NOT unmodellable just because its own type is
38
+ // `unknown`: a method reference on a known receiver (`path.includes`) has
39
+ // `unknown` type yet emits fine — the method dispatch happens at emit on
40
+ // the object's type, so havocing it would wreck a legitimate call.
41
+ case "field": return e.obj.ty.kind === "unknown";
42
+ case "index": return e.obj.ty.kind === "unknown";
43
+ // anonymous object literal — Dafny has no model for it
44
+ case "record": return e.ty.kind !== "user";
45
+ // a call / nullish / opt-chain / conditional whose *result* is opaque
46
+ default: return e.ty.kind === "unknown";
47
+ }
48
+ }
49
+ /** Does the subtree (including lambda bodies) contain anything unmodellable?
50
+ * Used to havoc a *whole* call like `xs.map(v => ({...}))` whose receiver is
51
+ * modellable but whose lambda produces an unmodellable value — havocing the
52
+ * lambda interior would leave a nondeterministic `*` inside a pure lambda. */
53
+ function containsBad(e) {
54
+ let found = false;
55
+ const visit = (n) => {
56
+ if (found)
57
+ return;
58
+ if (isBadNode(n)) {
59
+ found = true;
60
+ return;
61
+ }
62
+ forEachChildExpr(n, visit);
63
+ };
64
+ visit(e);
65
+ return found;
66
+ }
67
+ function mustHavoc(e) {
68
+ if (isBadNode(e))
69
+ return true;
70
+ if (e.kind === "call" && e.args.some(a => a.kind === "lambda" && a.body.some(s => stmtContainsBad(s))))
71
+ return true;
72
+ return false;
73
+ }
74
+ // ── structural recursion helpers ────────────────────────────
75
+ function forEachChildExpr(e, f) {
76
+ switch (e.kind) {
77
+ case "binop":
78
+ f(e.left);
79
+ f(e.right);
80
+ return;
81
+ case "unop":
82
+ f(e.expr);
83
+ return;
84
+ case "call":
85
+ f(e.fn);
86
+ e.args.forEach(f);
87
+ return;
88
+ case "index":
89
+ f(e.obj);
90
+ f(e.idx);
91
+ return;
92
+ case "field":
93
+ f(e.obj);
94
+ return;
95
+ case "record":
96
+ if (e.spread)
97
+ f(e.spread);
98
+ e.fields.forEach(fl => f(fl.value));
99
+ return;
100
+ case "arrayLiteral":
101
+ e.elems.forEach(f);
102
+ return;
103
+ case "conditional":
104
+ f(e.cond);
105
+ f(e.then);
106
+ f(e.else);
107
+ return;
108
+ case "optChain":
109
+ f(e.obj);
110
+ return;
111
+ case "nullish":
112
+ f(e.left);
113
+ f(e.right);
114
+ return;
115
+ case "forall":
116
+ case "exists":
117
+ f(e.body);
118
+ return;
119
+ case "someMatch":
120
+ f(e.scrutinee);
121
+ f(e.someBody);
122
+ f(e.noneBody);
123
+ return;
124
+ case "tagMatch":
125
+ f(e.scrutinee);
126
+ e.cases.forEach(c => f(c.body));
127
+ if (e.fallthrough)
128
+ f(e.fallthrough);
129
+ return;
130
+ case "lambda":
131
+ e.body.forEach(s => forEachChildExprInStmt(s, f));
132
+ return;
133
+ }
134
+ }
135
+ function forEachChildExprInStmt(s, f) {
136
+ switch (s.kind) {
137
+ case "let":
138
+ case "ghostLet":
139
+ f(s.init);
140
+ return;
141
+ case "assign":
142
+ case "ghostAssign":
143
+ f(s.value);
144
+ return;
145
+ case "return":
146
+ f(s.value);
147
+ return;
148
+ case "expr":
149
+ f(s.expr);
150
+ return;
151
+ case "assert":
152
+ f(s.expr);
153
+ return;
154
+ case "if":
155
+ f(s.cond);
156
+ s.then.forEach(st => forEachChildExprInStmt(st, f));
157
+ s.else.forEach(st => forEachChildExprInStmt(st, f));
158
+ return;
159
+ case "while":
160
+ f(s.cond);
161
+ s.body.forEach(st => forEachChildExprInStmt(st, f));
162
+ return;
163
+ case "forof":
164
+ f(s.iterable);
165
+ s.body.forEach(st => forEachChildExprInStmt(st, f));
166
+ return;
167
+ case "switch":
168
+ f(s.expr);
169
+ s.cases.forEach(c => c.body.forEach(st => forEachChildExprInStmt(st, f)));
170
+ s.defaultBody.forEach(st => forEachChildExprInStmt(st, f));
171
+ return;
172
+ case "someMatch":
173
+ f(s.scrutinee);
174
+ s.someBody.forEach(st => forEachChildExprInStmt(st, f));
175
+ s.noneBody.forEach(st => forEachChildExprInStmt(st, f));
176
+ return;
177
+ case "tagMatch":
178
+ f(s.scrutinee);
179
+ s.cases.forEach(c => c.body.forEach(st => forEachChildExprInStmt(st, f)));
180
+ s.fallthrough.forEach(st => forEachChildExprInStmt(st, f));
181
+ return;
182
+ }
183
+ }
184
+ function stmtContainsBad(s) {
185
+ let found = false;
186
+ forEachChildExprInStmt(s, e => { if (containsBad(e))
187
+ found = true; });
188
+ return found;
189
+ }
190
+ // ── contracted-call ("sink") detection ──────────────────────
191
+ // A call whose callee carries a `//@ requires` precondition. When we havoc an
192
+ // unmodellable expression we must NOT silently discard such a call nested
193
+ // inside it — its precondition would go unchecked (a false pass). Instead we
194
+ // hoist it to a discard statement so the precondition is still verified.
195
+ let _guarded = new Set();
196
+ // Every name LemmaScript models: local functions + all externs. A call to
197
+ // anything else with an unknown callee is "opaque" — abstracted by havoc.
198
+ let _known = new Set();
199
+ function calleeName(e) {
200
+ if (e.kind !== "call")
201
+ return null;
202
+ if (e.fn.kind === "var")
203
+ return e.fn.name;
204
+ if (e.fn.kind === "field")
205
+ return e.fn.field;
206
+ return null;
207
+ }
208
+ /** Root variable a receiver chain bottoms out at (`fs.x.y` → `fs`,
209
+ * `res.status(400).send` → `res`, `templateJson.variables.map` → `templateJson`). */
210
+ function rootVar(e) {
211
+ let cur = e;
212
+ for (;;) {
213
+ if (cur.kind === "var")
214
+ return cur.name;
215
+ if (cur.kind === "field")
216
+ cur = cur.obj;
217
+ else if (cur.kind === "call")
218
+ cur = cur.fn;
219
+ else if (cur.kind === "index")
220
+ cur = cur.obj;
221
+ else
222
+ return null;
223
+ }
224
+ }
225
+ /** Display name of an *opaque external* call this pass abstracts away — a call
226
+ * on an unknown receiver rooted at a module/global (`fs.readFileSync`,
227
+ * `JSON.parse`) or an unknown free function (`uuidv4`). Returns null for
228
+ * modelled calls (local functions, contracted/dispatched externs, methods on
229
+ * known receivers) and for calls rooted at a function-local/param (e.g.
230
+ * `templateJson.variables.map`, `res.status().send` — framework objects and
231
+ * havoc'd locals, not external sinks). These are the calls a completeness
232
+ * reviewer must confirm are not unguarded sinks. */
233
+ function opaqueCallName(e, locals) {
234
+ // Report exactly the calls the pass abstracts. A call is havoc'd iff its
235
+ // result type is unknown (the same `isBadNode`/`mustHavoc` criterion the pass
236
+ // uses) — keying on that, rather than re-matching the callee's syntax, makes
237
+ // the report complete: it can't miss an unusual form like `fs["read"](p)`.
238
+ if (e.kind !== "call" || e.ty.kind !== "unknown")
239
+ return null;
240
+ const root = rootVar(e.fn);
241
+ // Not an external sink: rooted at a parameter/local value (framework objects,
242
+ // havoc'd locals) or at a known function/extern (already modelled).
243
+ if (root && (locals.has(root) || _known.has(root)))
244
+ return null;
245
+ // Best display we can give the callee.
246
+ const f = e.fn;
247
+ if (f.kind === "field" && f.obj.kind === "var")
248
+ return `${f.obj.name}.${f.field}`;
249
+ if (f.kind === "field")
250
+ return f.field;
251
+ if (f.kind === "index" && f.obj.kind === "var" && f.idx.kind === "str")
252
+ return `${f.obj.name}.${f.idx.value}`;
253
+ if (f.kind === "var")
254
+ return f.name;
255
+ return root ?? "<computed call>";
256
+ }
257
+ /** Every binder in scope anywhere in a function body — params (seed) + let /
258
+ * forof / lambda parameters — so a call rooted at any of them is treated as a
259
+ * local value, not an external sink. A call rooted at a *free* name (a module
260
+ * or global like `fs`/`JSON`/`console`, or a free function like `uuidv4`) is
261
+ * what the report keeps. Still descends into lambdas so a sink hidden in a
262
+ * callback (`arr.forEach(x => fs.writeFileSync(x))`) is not missed. */
263
+ function localNames(stmts, seed) {
264
+ const out = new Set(seed);
265
+ const inExpr = (e) => {
266
+ if (e.kind === "lambda") {
267
+ e.params.forEach(p => out.add(p.name));
268
+ inStmts(e.body);
269
+ return;
270
+ }
271
+ forEachChildExpr(e, inExpr);
272
+ };
273
+ const inStmts = (ss) => {
274
+ for (const s of ss) {
275
+ if (s.kind === "let" || s.kind === "ghostLet")
276
+ out.add(s.name);
277
+ if (s.kind === "forof")
278
+ s.names.forEach(n => out.add(n));
279
+ forEachChildExprInStmt(s, inExpr);
280
+ if (s.kind === "if") {
281
+ inStmts(s.then);
282
+ inStmts(s.else);
283
+ }
284
+ else if (s.kind === "while" || s.kind === "forof")
285
+ inStmts(s.body);
286
+ else if (s.kind === "switch") {
287
+ s.cases.forEach(c => inStmts(c.body));
288
+ inStmts(s.defaultBody);
289
+ }
290
+ else if (s.kind === "someMatch") {
291
+ inStmts(s.someBody);
292
+ inStmts(s.noneBody);
293
+ }
294
+ else if (s.kind === "tagMatch") {
295
+ s.cases.forEach(c => inStmts(c.body));
296
+ inStmts(s.fallthrough);
297
+ }
298
+ }
299
+ };
300
+ inStmts(stmts);
301
+ return out;
302
+ }
303
+ /** All opaque external calls in a function body (deduped, sorted). */
304
+ function collectOpaqueCalls(stmts, locals) {
305
+ const found = new Set();
306
+ const visit = (e) => {
307
+ const n = opaqueCallName(e, locals);
308
+ if (n)
309
+ found.add(n);
310
+ forEachChildExpr(e, visit);
311
+ };
312
+ for (const s of stmts)
313
+ forEachChildExprInStmt(s, visit);
314
+ return [...found].sort();
315
+ }
316
+ /** Outermost contracted calls inside `e`'s subtree (excluding `e` itself, which
317
+ * is the havoc'd wrapper). Does not descend into lambdas — a sink there would
318
+ * reference the lambda's binder and can't be hoisted to the enclosing scope. */
319
+ function collectSinks(e) {
320
+ const out = [];
321
+ const walk = (n) => {
322
+ if (n.kind === "lambda")
323
+ return;
324
+ const name = calleeName(n);
325
+ if (name && _guarded.has(name)) {
326
+ out.push(n);
327
+ return;
328
+ } // outermost: don't descend
329
+ forEachChildExpr(n, walk);
330
+ };
331
+ forEachChildExpr(e, walk);
332
+ return out;
333
+ }
334
+ // ── rewrite ──────────────────────────────────────────────────
335
+ // `hoisted` accumulates discard statements (preserved contracted calls) to be
336
+ // emitted, in order, immediately before the statement being rewritten.
337
+ function rewriteExpr(e, hoisted) {
338
+ if (mustHavoc(e)) {
339
+ // Preserve any contracted call nested inside the discarded value so its
340
+ // precondition is still checked. transform lowers the `expr` discard to
341
+ // `var _ := call(...)` (and lifts any havoc'd args), so we lean on it.
342
+ for (const sink of collectSinks(e)) {
343
+ hoisted.push({ kind: "expr", expr: rewriteExpr(sink, hoisted) });
344
+ }
345
+ return { kind: "havoc", ty: e.ty };
346
+ }
347
+ switch (e.kind) {
348
+ case "var":
349
+ case "num":
350
+ case "str":
351
+ case "bool":
352
+ case "havoc": return e;
353
+ case "binop": return { ...e, left: rewriteExpr(e.left, hoisted), right: rewriteExpr(e.right, hoisted) };
354
+ case "unop": return { ...e, expr: rewriteExpr(e.expr, hoisted) };
355
+ case "call": return { ...e, fn: rewriteExpr(e.fn, hoisted), args: e.args.map(a => rewriteExpr(a, hoisted)) };
356
+ case "index": return { ...e, obj: rewriteExpr(e.obj, hoisted), idx: rewriteExpr(e.idx, hoisted) };
357
+ case "field": return { ...e, obj: rewriteExpr(e.obj, hoisted) };
358
+ case "record": return { ...e, spread: e.spread ? rewriteExpr(e.spread, hoisted) : null,
359
+ fields: e.fields.map(fl => ({ ...fl, value: rewriteExpr(fl.value, hoisted) })) };
360
+ case "arrayLiteral": return { ...e, elems: e.elems.map(el => rewriteExpr(el, hoisted)) };
361
+ case "conditional": return { ...e, cond: rewriteExpr(e.cond, hoisted), then: rewriteExpr(e.then, hoisted), else: rewriteExpr(e.else, hoisted) };
362
+ case "optChain": return { ...e, obj: rewriteExpr(e.obj, hoisted) };
363
+ case "nullish": return { ...e, left: rewriteExpr(e.left, hoisted), right: rewriteExpr(e.right, hoisted) };
364
+ case "forall":
365
+ case "exists": return { ...e, body: rewriteExpr(e.body, hoisted) };
366
+ case "someMatch": return { ...e, scrutinee: rewriteExpr(e.scrutinee, hoisted), someBody: rewriteExpr(e.someBody, hoisted), noneBody: rewriteExpr(e.noneBody, hoisted) };
367
+ case "tagMatch": return { ...e, scrutinee: rewriteExpr(e.scrutinee, hoisted),
368
+ cases: e.cases.map(c => ({ ...c, body: rewriteExpr(c.body, hoisted) })),
369
+ fallthrough: e.fallthrough ? rewriteExpr(e.fallthrough, hoisted) : null };
370
+ // A lambda body is its own statement scope; rewrite it independently
371
+ // (its hoists belong inside the lambda, not the enclosing statement).
372
+ case "lambda": return { ...e, body: rewriteStmts(e.body) };
373
+ }
374
+ }
375
+ /** A condition in a boolean context. If it touches anything unmodellable
376
+ * (`!fs.existsSync(...)`), the whole condition becomes a nondeterministic
377
+ * `bool` — havocing a subexpression would leave a non-bool `*` under `!`/`if`.
378
+ * A fully modellable guard (`!validPath(x)`, `s === ""`) is left intact, so
379
+ * the verifier still reasons about it. Contracted calls inside an havoc'd
380
+ * condition are still hoisted. */
381
+ function rewriteCond(e, hoisted) {
382
+ if (containsBad(e)) {
383
+ for (const sink of collectSinks(e)) {
384
+ hoisted.push({ kind: "expr", expr: rewriteExpr(sink, hoisted) });
385
+ }
386
+ return { kind: "havoc", ty: { kind: "bool" } };
387
+ }
388
+ return rewriteExpr(e, hoisted);
389
+ }
390
+ function rewriteStmts(stmts) {
391
+ const out = [];
392
+ for (const s of stmts) {
393
+ // A side-effecting statement whose value is unmodellable (`xs.forEach(...)`,
394
+ // `console.log(x)`) carries no verification content — drop it, but still
395
+ // preserve any contracted call nested inside it.
396
+ if (s.kind === "expr" && mustHavoc(s.expr)) {
397
+ const h = [];
398
+ for (const sink of collectSinks(s.expr)) {
399
+ h.push({ kind: "expr", expr: rewriteExpr(sink, h) });
400
+ }
401
+ out.push(...h);
402
+ continue;
403
+ }
404
+ out.push(...rewriteStmt(s));
405
+ }
406
+ return out;
407
+ }
408
+ function rewriteStmt(s) {
409
+ const h = [];
410
+ const rw = (e) => rewriteExpr(e, h);
411
+ switch (s.kind) {
412
+ case "let": {
413
+ const init = rw(s.init);
414
+ return [...h, { ...s, init }];
415
+ }
416
+ case "ghostLet": {
417
+ const init = rw(s.init);
418
+ return [...h, { ...s, init }];
419
+ }
420
+ case "assign": {
421
+ const value = rw(s.value);
422
+ return [...h, { ...s, value }];
423
+ }
424
+ case "ghostAssign": {
425
+ const value = rw(s.value);
426
+ return [...h, { ...s, value }];
427
+ }
428
+ case "return": {
429
+ const value = rw(s.value);
430
+ return [...h, { ...s, value }];
431
+ }
432
+ case "expr": {
433
+ const expr = rw(s.expr);
434
+ return [...h, { ...s, expr }];
435
+ }
436
+ case "assert": {
437
+ const expr = rw(s.expr);
438
+ return [...h, { ...s, expr }];
439
+ }
440
+ case "if": {
441
+ const cond = rewriteCond(s.cond, h);
442
+ return [...h, { ...s, cond, then: rewriteStmts(s.then), else: rewriteStmts(s.else) }];
443
+ }
444
+ case "while": {
445
+ const cond = rewriteCond(s.cond, h);
446
+ return [...h, { ...s, cond, body: rewriteStmts(s.body) }];
447
+ }
448
+ case "forof": {
449
+ const iterable = rw(s.iterable);
450
+ return [...h, { ...s, iterable, body: rewriteStmts(s.body) }];
451
+ }
452
+ case "switch": {
453
+ const expr = rw(s.expr);
454
+ return [...h, { ...s, expr,
455
+ cases: s.cases.map(c => ({ ...c, body: rewriteStmts(c.body) })),
456
+ defaultBody: rewriteStmts(s.defaultBody) }];
457
+ }
458
+ case "someMatch": {
459
+ const scrutinee = rw(s.scrutinee);
460
+ return [...h, { ...s, scrutinee, someBody: rewriteStmts(s.someBody), noneBody: rewriteStmts(s.noneBody) }];
461
+ }
462
+ case "tagMatch": {
463
+ const scrutinee = rw(s.scrutinee);
464
+ return [...h, { ...s, scrutinee,
465
+ cases: s.cases.map(c => ({ ...c, body: rewriteStmts(c.body) })),
466
+ fallthrough: rewriteStmts(s.fallthrough) }];
467
+ }
468
+ case "break":
469
+ case "continue":
470
+ case "throw": return [s];
471
+ }
472
+ }
473
+ /** Does any statement in the body contain a havoc expression? Havoc is only
474
+ * valid in Dafny `method`s (its `*` can't appear in a deterministic
475
+ * `function`), so a body we've havoc'd must be classified impure. */
476
+ function bodyHasHavoc(stmts) {
477
+ let found = false;
478
+ const visit = (e) => {
479
+ if (found)
480
+ return;
481
+ if (e.kind === "havoc") {
482
+ found = true;
483
+ return;
484
+ }
485
+ forEachChildExpr(e, visit);
486
+ };
487
+ for (const s of stmts)
488
+ forEachChildExprInStmt(s, visit);
489
+ return found;
490
+ }
491
+ /** Apply the auto-havoc abstraction to functions opted in via `//@ autohavoc`
492
+ * (file-level or per-function). A no-op when nothing is annotated. When any
493
+ * function opts in, also drop module constants whose initializer is
494
+ * unmodellable (e.g. `const fs = require("fs")` pulled in by reachability). */
495
+ export function autoHavocModule(mod) {
496
+ if (!mod.functions.some(f => f.autohavoc))
497
+ return mod;
498
+ // Names of calls carrying a `//@ requires` precondition — these must never be
499
+ // silently discarded by havoc (see collectSinks / rewriteExpr).
500
+ _guarded = new Set();
501
+ for (const e of mod.externs)
502
+ if (e.requires.length > 0) {
503
+ _guarded.add(e.qualified);
504
+ _guarded.add(e.flat);
505
+ }
506
+ for (const f of mod.functions)
507
+ if (f.requires.length > 0)
508
+ _guarded.add(f.name);
509
+ _known = new Set();
510
+ for (const f of mod.functions)
511
+ _known.add(f.name);
512
+ for (const e of mod.externs) {
513
+ _known.add(e.qualified);
514
+ _known.add(e.flat);
515
+ }
516
+ return {
517
+ ...mod,
518
+ constants: mod.constants.filter(c => !containsBad(c.value)),
519
+ functions: mod.functions.map(f => {
520
+ if (!f.autohavoc)
521
+ return f;
522
+ // Completeness transparency: surface the opaque external calls this pass
523
+ // abstracts away, so a forgotten sink (e.g. a raw `fs.readFileSync`) is
524
+ // visible rather than silently havoc'd behind a green check.
525
+ const locals = localNames(f.body, new Set(f.params.map(p => p.name)));
526
+ const opaque = collectOpaqueCalls(f.body, locals);
527
+ if (opaque.length > 0) {
528
+ console.error(`autohavoc: ${f.name} abstracts ${opaque.length} external call(s) — confirm none is an unguarded sink: ${opaque.join(", ")}`);
529
+ }
530
+ const body = rewriteStmts(f.body);
531
+ // Havoc forces Dafny `method` emission (`*` is invalid in a `function`).
532
+ const isPure = f.isPure && !bodyHasHavoc(body);
533
+ return { ...f, body, isPure, forcePure: f.forcePure && isPure };
534
+ }),
535
+ };
536
+ }