@scriptc/compiler 0.1.1 → 0.1.3

package/dist/backend/llvm/emitter.js

@@ -62,16 +62,16 @@ import { InternalCompilerError } from "../../errors.js";
  * lazy-inflate representation as the C debugging backend.
  */
 import { deflateRawSync } from "node:zlib";
-import { endsWithJump, matchStringSelfConcat } from "../../ir/analysis.js";
+import { endsWithJump, matchStringSelfConcat, streamTypedRefEligible } from "../../ir/analysis.js";
 import { emitLibraryIdentityLines } from "../library-identity-markers.js";
-import { CAUGHT, ffiCallbackType, isFfiContextParam, isRefCounted, isUnitType, moduleEmbedsBuiltin, moduleEmbedsCompressedNpm, moduleUsesDynInvoke, moduleUsesFetch, moduleUsesFsWatch, moduleUsesHttpServer, moduleUsesNet, moduleUsesNodeTest, moduleUsesProcessEvents, moduleUsesStream, moduleUsesTls, moduleUsesTlsCa, NPM_COMPRESS_MIN, POINTER_KINDS, RUNTIME_EMITTER_CLASS, RUNTIME_ERROR_CLASSES, RUNTIME_STREAM_CLASSES, VOID } from "../../ir/ir.js";
+import { CAUGHT, ffiCallbackType, isDynTypedRefType, isFfiContextParam, isRefCounted, isUnitType, moduleEmbedsBuiltin, moduleEmbedsCompressedNpm, moduleUsesChildProcess, moduleUsesDynInvoke, moduleUsesFetch, moduleUsesFsWatch, moduleUsesHttpServer, moduleUsesNet, moduleUsesNodeTest, moduleUsesProcessEvents, moduleUsesStream, moduleUsesTls, moduleUsesTlsCa, NPM_COMPRESS_MIN, POINTER_KINDS, RUNTIME_EMITTER_CLASS, RUNTIME_ERROR_CLASSES, RUNTIME_STREAM_CLASSES, typeKey, VOID } from "../../ir/ir.js";
 import { matchIntegerBytesForLoop } from "../../ir/integer-loops.js";
 import { allocateFfiCallbackAdapters, hasForeignFfiCallback, hasRetainedFfiCallback } from "../ffi-callbacks.js";
 import { RUNTIME_ABI_MARKER } from "../runtime-abi.js";
 import { computeMayThrow } from "../c/may-throw.js";
 import { mangleArgPack, mangleAsyncSpawn, mangleClassObj, mangleFnClosure, mangleFunction, mangleGenDrop, mangleGenSpawn, mangleGlobal, mangleLocal, mangleRecordStruct, mangleTrampoline, mangleWrapper } from "../mangle.js";
 import { BlockBuilder } from "./blocks.js";
-import { f64Lit, ffiNativeTypeLl } from "./common.js";
+import { f64Lit, ffiNativeTypeLl, llvmCommentText } from "./common.js";
 import { emitLiteralExpr, emitOperatorExpr, emitStringExpr, emitContainerExpr, emitRecordExpr } from "./expr-primitives.js";
 import { emitControlExpr } from "./expr-control.js";
 import { emitCallExpr } from "./expr-calls.js";
@@ -80,7 +80,7 @@ import { emitIntrinsicExpr, emitSerializationExpr, emitAsyncExpr } from "./expr-
 import { emitJsInteropExpr, emitExpr } from "./expr-dispatch.js";
 import { emitJsMarshal, emitJsOp, emitJsExit, islandAdapter, islandTypedAdapter } from "./expr-island.js";
 import { dynKind, raceAdapterFor, genResultThunkFor, childExitThunkFor, childExitSignalThunkFor, childDataThunkFor, emitterFixedAdapter, wrapEmitterListener, unwrapNullableClosure, closeBindThunkFor, closeOverrideWrapFor } from "./expr-callbacks.js";
-import { streamDataAdapter, streamDoneFnFor, fsRenameThunkFor, streamCbThunkFor } from "./expr-stream-callbacks.js";
+import { streamDataAdapter, streamDoneFnFor, cryptoBytesThunkFor, fsRenameThunkFor, streamCbThunkFor } from "./expr-stream-callbacks.js";
 import { resolveThunkFor, tagInSet, arrPush, emitArrayCopyLoop, emitStrIntrinsic, emitArrIntrinsic, wrapNullable, emitMapNew, mapSet, emitMapLikeIntrinsic, emitSetNew } from "./expr-containers.js";
 import { emitBytesReceiver, emitIntegerLoopIndex, emitBytesIndex, emitBytesData, emitBytesLength, emitBytesGet, emitBytesU32, emitBytesSet, emitBytesIntrinsic } from "./expr-bytes.js";
 import { emitRegexIntrinsic, emitRecordKeyGet, keyedRecordReadInto } from "./expr-records.js";
@@ -202,8 +202,8 @@ class LlEmitter {
     /** ReadableStream.from adapters keep typed arrays by reference and box
      * one current element per pull. */
     streamFromArrayAdapters = new Map();
-    /** Identity-preserving static→dyn capsules used by Web APIs whose values
-     * remain directly observable (stream chunks and AbortSignal reasons). */
+    /** Identity-preserving static→dyn capsules for program classes and Web
+     * APIs whose values remain directly observable. */
     liveDynRefAdapters = new Map();
     /** Runtime-arm dispatchers for live union values. */
     liveDynUnionRefAdapters = new Map();
@@ -256,8 +256,8 @@ class LlEmitter {
      * `tryDepth` snapshots tryStack.length at region entry: a throw inside
      * a pending-return finally copy propagates OUT of the completing try
      * (past its own catch), so the copies emit under the truncated stack.
-     * break/continue never cross a finally (frontend fence + validator
-     * backstop), so return and the two tryCatch paths are the only copies. */
+     * break/continue use the same region snapshots through
+     * emitFinallysForJump. */
     finallyStack = [];
     /** Enclosing try contexts, innermost last — the compile-time unwind
      * targets (CEmitter.tryStack): a pending check or `throw` inside a try
@@ -692,6 +692,7 @@ class LlEmitter {
         // BEFORE the global releases (the C emitter's runExitListeners
         // ordering — the atexit half becomes an idempotent no-op).
         const usesEvents = moduleUsesProcessEvents(this.mod);
+        const usesChildProcess = moduleUsesChildProcess(this.mod);
         const usesFsWatch = moduleUsesFsWatch(this.mod);
         // Stream-surface programs fill the loop's stream hook (the deferred
         // next-tick emissions) and the emitter's post-registration flow kick
@@ -724,6 +725,8 @@ class LlEmitter {
         // Declared NOW — the extern block flushes before main assembles.
         if (usesEvents)
             this.declare(`declare void @scr_events_install()`);
+        if (usesChildProcess)
+            this.declare(`declare void @scr_child_dyn_install()`);
         if (usesFsWatch)
             this.declare(`declare void @scr_watch_install()`);
         if (this.ffiHasForeignCallback)
@@ -1114,7 +1117,7 @@ class LlEmitter {
         // Event-surface programs (signal/exit listeners) fill the loop's
         // nullable event hooks before %main — scr_events.c links only when
         // this line is emitted (native-toolchain.ts gates on the same predicate).
-        ...(usesEvents ? [`  call void @scr_events_install()`] : []), 
+        ...(usesEvents ? [`  call void @scr_events_install()`] : []), ...(usesChildProcess ? [`  call void @scr_child_dyn_install()`] : []), 
         // fs.watch programs fill the loop's watch hooks the same way —
         // scr_watch.c links only when this line is emitted.
         ...(usesFsWatch ? [`  call void @scr_watch_install()`] : []), ...(this.ffiHasForeignCallback ? [`  call void @scr_ffi_install()`] : []), ...(snapshotsTlsCa ? [`  call void @scr_tls_ca_install()`] : []), ...(usesFetch ? [`  call void @scr_fetch_install()`] : []), ...(embedsZlib ? [`  call void @scr_zlib_island_install()`] : []), ...(embedsNet ? [`  call void @scr_net_island_install()`] : []), ...(usesNet ? [`  call void @scr_net_install()`, `  call void @scr_net_dyn_install()`] : []), ...(usesHttp ? [`  call void @scr_http_dyn_install()`] : []), ...(usesStream ? [`  call void @scr_stream_install()`] : []), `  call void @scr_lib_init(i32 %argc, ptr %argv)`, ...(usesIsland
@@ -1537,7 +1540,7 @@ class LlEmitter {
             out.push(`define void @scr_wasi_coro_resume(ptr %handle) ${FN_ATTRS} {`, `entry:`, `  call void @llvm.coro.resume(ptr %handle)`, `  ret void`, `}`, `define void @scr_wasi_coro_destroy(ptr %handle) ${FN_ATTRS} {`, `entry:`, `  call void @llvm.coro.destroy(ptr %handle)`, `  ret void`, `}`, ``);
         }
         for (const fn of this.mod.functions) {
-            if (fn.async !== true)
+            if (fn.async !== true || fn.generator !== undefined)
                 continue;
             const { definitions, ret, tr, spawnParams, argPackLines } = this.emitArgPackAndTrampolinePrologue(fn);
             out.push(...definitions);
@@ -1708,6 +1711,27 @@ class LlEmitter {
                 tr.push(`  br label %done`, `done:`, `  ret void`, `}`, ``);
             }
             out.push(...tr);
+            let settleAsync = null;
+            if (fn.async) {
+                this.declare(`declare ptr @scr_async_gen_new(ptr, ptr, ptr, ptr)`);
+                const genT = {
+                    kind: "generator",
+                    async: true,
+                    yieldT: fn.generator.yieldT,
+                    retT: ret,
+                    nextT: fn.generator.nextT,
+                };
+                const resultT = fn.generator.resultType;
+                const build = this.genResultThunkFor(genT, resultT);
+                settleAsync = `${build}_async`;
+                const settleKey = `ags:${typeKey(genT)}`;
+                if (!this.resolveThunks.has(settleKey)) {
+                    this.resolveThunks.set(settleKey, settleAsync);
+                    const adapters = vAdapters(this, resultT);
+                    this.resolveThunkDefs.push(`define internal void @${settleAsync}(ptr %g, ptr %p) ${FN_ATTRS} {`, `entry:`, `  %r = call ptr @${build}(ptr %g)`, `  call void @scr_promise_fulfill_ref(ptr %p, ptr %r, ptr ${adapters.retain}, ptr ${adapters.release}, ptr ${traceArg(this, resultT)})`, `  ret void`, `}`, ``);
+                }
+                this.declare(`declare void @scr_promise_fulfill_ref(ptr, ptr, ptr, ptr, ptr)`);
+            }
             // The never-started teardown: drop the packed (+1) arguments.
             const dr = [
                 `define internal void @${mangleGenDrop(fn.name)}(ptr %ap) ${FN_ATTRS} {`,
@@ -1739,7 +1763,9 @@ class LlEmitter {
                 `entry:`,
                 ...argPackLines,
             ];
-            sp.push(`  %gg = call ptr @scr_gen_new(ptr @${mangleTrampoline(fn.name)}, ptr %ap, ptr @${mangleGenDrop(fn.name)})`, `  ret ptr %gg`, `}`, ``);
+            sp.push(settleAsync === null
+                ? `  %gg = call ptr @scr_gen_new(ptr @${mangleTrampoline(fn.name)}, ptr %ap, ptr @${mangleGenDrop(fn.name)})`
+                : `  %gg = call ptr @scr_async_gen_new(ptr @${mangleTrampoline(fn.name)}, ptr %ap, ptr @${mangleGenDrop(fn.name)}, ptr @${settleAsync})`, `  ret ptr %gg`, `}`, ``);
             out.push(...sp);
         }
         return out;
@@ -1867,6 +1893,34 @@ class LlEmitter {
         for (let i = this.scopes.length - 1; i >= scopeDepth; i--)
             this.releaseScope(this.scopes[i]);
     }
+    /** Run the finally regions an abrupt loop/block jump crosses, then
+     * release to the already-resolved target. This is the return path's
+     * completion walk without a parked result value. */
+    emitFinallysForJump(finallyDepth, frameDepth, scopeDepth) {
+        if (this.finallyStack.length <= finallyDepth) {
+            this.releaseForJump(frameDepth, scopeDepth);
+            return;
+        }
+        const savedFrames = this.frames;
+        const savedScopes = this.scopes;
+        const savedFinally = this.finallyStack;
+        const savedTry = this.tryStack;
+        for (let i = savedFinally.length - 1; i >= finallyDepth && !this.B.isTerminated(); i--) {
+            const fin = savedFinally[i];
+            this.releaseForJump(fin.frameDepth, fin.scopeDepth);
+            this.frames = this.frames.slice(0, fin.frameDepth);
+            this.scopes = this.scopes.slice(0, fin.scopeDepth);
+            this.finallyStack = savedFinally.slice(0, i);
+            this.tryStack = savedTry.slice(0, fin.tryDepth);
+            this.emitBlock(fin.body);
+        }
+        if (!this.B.isTerminated())
+            this.releaseForJump(frameDepth, scopeDepth);
+        this.frames = savedFrames;
+        this.scopes = savedScopes;
+        this.finallyStack = savedFinally;
+        this.tryStack = savedTry;
+    }
     /** THE unwind path at a point where an exception is pending: release
      * everything between here and the innermost try handler — or the whole
      * function — and branch to the handler / return a dummy value (never
@@ -1980,6 +2034,9 @@ class LlEmitter {
                 else if (arm.kind === "string") {
                     valueName = this.unionPeek(v.name);
                 }
+                else if (arm.kind === "bigint") {
+                    valueName = this.unionPeek(v.name);
+                }
                 const truthy = this.truthyOf(arm.kind, valueName, true);
                 B.line(`store i1 ${truthy}, ptr ${slot}`);
                 B.br(join);
@@ -2020,6 +2077,12 @@ class LlEmitter {
                 B.line(`${truthy} = icmp ne ${this.sizeType} ${len}, 0`);
                 return truthy;
             }
+            case "bigint": {
+                this.declare(`declare zeroext i1 @scr_bigint_truthy(ptr)`);
+                const truthy = B.tmp();
+                B.line(`${truthy} = call zeroext i1 @scr_bigint_truthy(ptr ${valueName})`);
+                return truthy;
+            }
             case "date":
                 return "true";
             case "array":
@@ -2040,8 +2103,11 @@ class LlEmitter {
             case "spawnRes":
             case "child":
             case "childStream":
+            case "childWriter":
             case "generator":
-            case "fsWatcher": {
+            case "fsWatcher":
+            case "cryptoHash":
+            case "cryptoHmac": {
                 if (unionArm)
                     return "true";
                 const truthy = B.tmp();
@@ -2362,10 +2428,10 @@ class LlEmitter {
      * fiber and returns the generator object) — CEmitter.callTargetC. */
     callTarget(fnName) {
         const fn = this.fnByName.get(fnName);
-        if (fn?.async === true)
-            return mangleAsyncSpawn(fnName);
         if (fn?.generator !== undefined)
             return mangleGenSpawn(fnName);
+        if (fn?.async === true)
+            return mangleAsyncSpawn(fnName);
         return mangleFunction(fnName);
     }
     /** Queue the current wasm coroutine and suspend it. The runtime decides
@@ -2495,7 +2561,7 @@ class LlEmitter {
             B.line(`call void @scr_wasi_coro_started(ptr ${handle})`);
             B.line(`${self} = call ptr @scr_fiber_self()`);
             this.currentWasiCoro = {
-                kind: fn.async === true ? "async" : "generator",
+                kind: fn.generator !== undefined ? "generator" : "async",
                 id,
                 handle,
                 self,
@@ -2857,7 +2923,7 @@ class LlEmitter {
                     for (const f of shape.fields) {
                         const lit = this.internLiteral(f.name);
                         const hit = B.tmp();
-                        B.line(`${hit} = call zeroext i1 @scr_str_eq(ptr ${key.name}, ptr ${lit}) ; ${f.name}`);
+                        B.line(`${hit} = call zeroext i1 @scr_str_eq(ptr ${key.name}, ptr ${lit}) ; ${llvmCommentText(f.name)}`);
                         const lh = B.newLabel("rks.h");
                         const ln = B.newLabel("rks.n");
                         B.condBr(hit, lh, ln);
@@ -2913,7 +2979,7 @@ class LlEmitter {
                     // typeEquals(f.type, iv) — the frontend fences everything else.
                     const lit = this.internLiteral(f.name);
                     const hit = B.tmp();
-                    B.line(`${hit} = call zeroext i1 @scr_str_eq(ptr ${key.name}, ptr ${lit}) ; ${f.name}`);
+                    B.line(`${hit} = call zeroext i1 @scr_str_eq(ptr ${key.name}, ptr ${lit}) ; ${llvmCommentText(f.name)}`);
                     const lh = B.newLabel("rks.h");
                     const ln = B.newLabel("rks.n");
                     B.condBr(hit, lh, ln);
@@ -2964,6 +3030,7 @@ class LlEmitter {
                     labels: s.labels,
                     frameDepth: this.frames.length,
                     scopeDepth: this.scopes.length,
+                    finallyDepth: this.finallyStack.length,
                 });
                 this.emitBlock(s.body);
                 this.jumpTargets.pop();
@@ -3003,6 +3070,7 @@ class LlEmitter {
                     ...(s.labels !== undefined && { labels: s.labels }),
                     frameDepth: this.frames.length,
                     scopeDepth: this.scopes.length,
+                    finallyDepth: this.finallyStack.length,
                 });
                 this.emitBlock(s.body);
                 this.jumpTargets.pop();
@@ -3024,6 +3092,7 @@ class LlEmitter {
                     ...(s.labels !== undefined && { labels: s.labels }),
                     frameDepth: this.frames.length,
                     scopeDepth: this.scopes.length,
+                    finallyDepth: this.finallyStack.length,
                 });
                 this.emitBlock(s.body);
                 this.jumpTargets.pop();
@@ -3085,6 +3154,7 @@ class LlEmitter {
                     ...(s.labels !== undefined && { labels: s.labels }),
                     frameDepth: this.frames.length,
                     scopeDepth: this.scopes.length,
+                    finallyDepth: this.finallyStack.length,
                 });
                 this.emitBlock(s.body);
                 this.jumpTargets.pop();
@@ -3156,6 +3226,7 @@ class LlEmitter {
                     ...(s.labels !== undefined && { labels: s.labels }),
                     frameDepth: this.frames.length,
                     scopeDepth: this.scopes.length,
+                    finallyDepth: this.finallyStack.length,
                 });
                 // The loop variable is a fresh const per iteration: its scope opens
                 // here, holds the (for ref elements: owned +1) current element, and
@@ -3216,8 +3287,9 @@ class LlEmitter {
                 }
                 if (!target)
                     throw new InternalCompilerError("llvm emitter bug: break target not found");
-                this.releaseForJump(target.frameDepth, target.scopeDepth);
-                B.terminate(`br label %${target.brkLabel}`);
+                this.emitFinallysForJump(target.finallyDepth, target.frameDepth, target.scopeDepth);
+                if (!B.isTerminated())
+                    B.terminate(`br label %${target.brkLabel}`);
                 break;
             }
             case "continue": {
@@ -3233,8 +3305,9 @@ class LlEmitter {
                 }
                 if (!target || target.contLabel === null)
                     throw new InternalCompilerError("llvm emitter bug: continue target not found");
-                this.releaseForJump(target.frameDepth, target.scopeDepth);
-                B.terminate(`br label %${target.contLabel}`);
+                this.emitFinallysForJump(target.finallyDepth, target.frameDepth, target.scopeDepth);
+                if (!B.isTerminated())
+                    B.terminate(`br label %${target.contLabel}`);
                 break;
             }
             case "return": {
@@ -3369,10 +3442,9 @@ class LlEmitter {
      *     <unwind>                   stash (it unwinds through the synthetic
      *   try.e:                       scope entry) — JS's semantics exactly
      *
-     * Returns inside tryBody/catchBody ride the finallyStack (inline copies
-     * at the return site — see `return`); break/continue never cross a
-     * finally and no jump leaves a finally body (frontend fence + validator
-     * backstop). */
+     * Abrupt completions inside tryBody/catchBody ride the finallyStack:
+     * returns inline copies at the return site, while break/continue use the
+     * shared region walk before branching to their resolved target. */
     emitTryCatch(s) {
         const B = this.B;
         const hasCatch = s.catchBody !== null;
@@ -3471,11 +3543,29 @@ class LlEmitter {
                 B.line(`${stash} = call ptr @scr_exc_take() ; stash across finally`);
                 B.line(`store ptr ${stash}, ptr ${stashSlot}`);
                 this.scopes.push([{ slot: stashSlot, type: CAUGHT }]);
+                const suppressHandler = s.suppressFinallyErrors
+                    ? {
+                        label: B.newLabel("try.fs"),
+                        used: false,
+                        frameDepth: this.frames.length,
+                        scopeDepth: this.scopes.length,
+                    }
+                    : null;
+                if (suppressHandler)
+                    this.tryStack.push(suppressHandler);
                 this.emitBlock(s.finallyBody);
+                if (suppressHandler)
+                    this.tryStack.pop();
                 this.scopes.pop(); // normal completion keeps the stash for the re-raise
                 B.line(`call void @scr_rethrow(ptr ${stash})`);
                 B.line(`call void @scr_caught_release(ptr ${stash})`);
                 this.emitUnwind();
+                if (suppressHandler?.used) {
+                    B.startBlock(suppressHandler.label);
+                    this.declare(`declare void @scr_exc_suppress(ptr)`);
+                    B.line(`call void @scr_exc_suppress(ptr ${stash}) ; consumes stash`);
+                    this.emitUnwind();
+                }
             }
             B.startBlock(endLabel);
         }
@@ -3546,6 +3636,7 @@ class LlEmitter {
             ...(s.labels !== undefined && { labels: s.labels }),
             frameDepth: this.frames.length,
             scopeDepth: this.scopes.length,
+            finallyDepth: this.finallyStack.length,
         });
         const scope = [];
         this.scopes.push(scope);
@@ -3731,6 +3822,9 @@ class LlEmitter {
     fsRenameThunkFor(cbT) {
         return fsRenameThunkFor(this.expressionContext(), cbT);
     }
+    cryptoBytesThunkFor(cbT) {
+        return cryptoBytesThunkFor(this.expressionContext(), cbT);
+    }
     streamCbThunkFor(kind, cbT) {
         return streamCbThunkFor(this.expressionContext(), kind, cbT);
     }
@@ -3809,6 +3903,19 @@ class LlEmitter {
     streamTypedRefCommitAdapter(t, snapshot) {
         return streamTypedRefCommitAdapter(this.expressionContext(), t, snapshot);
     }
+    liveDynRefAdapter(t) {
+        const key = typeKey(t);
+        const existing = this.liveDynRefAdapters.get(key);
+        if (existing)
+            return existing;
+        if (!streamTypedRefEligible(t) && !isDynTypedRefType(t)) {
+            throw new InternalCompilerError(`llvm emitter bug: live dyn ref of ${key}`);
+        }
+        const prefix = `sc_ldr_${this.liveDynRefAdapters.size}`;
+        const adapter = streamTypedRefMaterializeAdapter(this.expressionContext(), t, { prefix, adapters: new Map() }, `${prefix}_materialize`);
+        this.liveDynRefAdapters.set(key, adapter);
+        return adapter;
+    }
     liveDynUnionRefAdapter(t) {
         return liveDynUnionRefAdapter(this.expressionContext(), t);
     }