@scriptc/compiler 0.2.7 → 0.2.9

package/dist/backend/llvm/publish.js

@@ -0,0 +1,286 @@
+import { mapOf, STRING, typeKey } from "../../ir/ir.js";
+import { mangleClassStruct, mangleRecordStruct } from "../mangle.js";
+import { FN_ATTRS } from "./shapes.js";
+import { LlvmUnsupportedError } from "./unsupported.js";
+/** scr_pub_mark's object kinds (SCR_PUB_KIND_* in scr_runtime.h), for the RC audit. */
+const PUB_KIND_OBJECT = 4;
+const PUB_KIND_UNION = 5;
+const RUNTIME_WALKERS = {
+    string: "@scr_pub_str_fn",
+    bigint: "@scr_pub_leaf_fn",
+    dyn: "@scr_pub_dyn_fn",
+};
+export class PublishWalkers {
+    host;
+    walkers = new Map();
+    defsOut = [];
+    counter = 0;
+    constructor(host) {
+        this.host = host;
+    }
+    /** The definitions emitted so far (module-level LLVM text). */
+    defs() {
+        return this.defsOut;
+    }
+    /** Lines publishing the value `v` (an LLVM ptr of static type `t`), or
+     * none for types that hold no references. */
+    emitPublish(v, t, loc) {
+        const root = this.host.nullableArm(t) ?? t;
+        if (scalarRefusal(root)) {
+            // A Date (or process stream) is a scalar here, but Node.js refuses it.
+            this.host.declare(`declare void @scr_publish_refuse(ptr)`);
+            return [`  call void @scr_publish_refuse(ptr ${this.host.cstr(refusalMessage(root))})`];
+        }
+        const fn = this.walker(t, "the published value", loc);
+        if (fn === null)
+            return [];
+        this.host.declare(`declare void @scr_publish(ptr, ptr)`);
+        return [`  call void @scr_publish(ptr ${v}, ptr ${fn})`];
+    }
+    /** The walker symbol for values of type `t` (null: nothing to walk). */
+    walker(t, path, loc) {
+        const arm = this.host.nullableArm(t);
+        if (arm !== null)
+            return this.walker(arm, path, loc);
+        switch (t.kind) {
+            case "f64":
+            case "bool":
+            case "undefinedT":
+            case "nullT":
+            case "void":
+                return null;
+            case "string":
+            case "bigint":
+            case "dyn": {
+                const sym = RUNTIME_WALKERS[t.kind];
+                this.host.declare(`declare void ${sym}(ptr, ptr)`);
+                return sym;
+            }
+            case "object":
+            case "record":
+            case "array":
+            case "map":
+            case "set":
+            case "union":
+                break;
+            default:
+                return this.refusal(refusalMessage(t), `${typeKey(t)} at ${path}`);
+        }
+        const key = typeKey(t);
+        const known = this.walkers.get(key);
+        if (known !== undefined)
+            return known;
+        const sym = `@sc_pub_${this.counter++}`;
+        this.walkers.set(key, sym);
+        const body = this.body(t, sym, path, loc);
+        this.defsOut.push(...body, ``);
+        return sym;
+    }
+    refusals = new Map();
+    /** A walker that refuses the publish when a value is actually present:
+     * the Node.js implementation's TypeError, plus the static path for
+     * SCRIPTC_PUBLISH_TRACE. */
+    refusal(message, detail) {
+        const key = `${message}\0${detail}`;
+        const known = this.refusals.get(key);
+        if (known !== undefined)
+            return known;
+        const sym = `@sc_pub_${this.counter++}`;
+        this.refusals.set(key, sym);
+        this.host.declare(`declare void @scr_pub_refuse_at(ptr, ptr, ptr)`);
+        this.defsOut.push(`define internal void ${sym}(ptr %o, ptr %ctx) ${FN_ATTRS} { ; refuse ${llvmComment(detail)}`, `entry:`, `  call void @scr_pub_refuse_at(ptr %ctx, ptr ${this.host.cstr(message)}, ptr ${this.host.cstr(detail)})`, `  ret void`, `}`, ``);
+        return sym;
+    }
+    declareRuntime() {
+        this.host.declare(`declare zeroext i1 @scr_pub_mark(ptr, ptr, i1 zeroext, i32)`);
+        this.host.declare(`declare void @scr_pub_push(ptr, ptr, ptr)`);
+    }
+    /** `define` lines for the walker `sym` of `t`. */
+    body(t, sym, path, loc) {
+        this.declareRuntime();
+        const head = `define internal void ${sym}(ptr %o, ptr %ctx) ${FN_ATTRS} { ; publish ${llvmComment(typeKey(t))}`;
+        switch (t.kind) {
+            case "array": {
+                const elem = this.walker(t.elem, `${path}[element]`, loc);
+                if (scalarRefusal(t.elem))
+                    throw new LlvmUnsupportedError(`publish of ${typeKey(t)}`, loc);
+                this.host.declare(`declare void @scr_pub_arr(ptr, ptr, ptr)`);
+                return [
+                    head,
+                    `entry:`,
+                    `  call void @scr_pub_arr(ptr %o, ptr %ctx, ptr ${elem ?? "null"})`,
+                    `  ret void`,
+                    `}`,
+                ];
+            }
+            case "map":
+            case "set": {
+                const keyType = t.kind === "map" ? t.key : t.elem;
+                const k = this.walker(keyType, `${path}[key]`, loc);
+                const v = t.kind === "map" ? this.walker(t.value, `${path}[value]`, loc) : null;
+                if (scalarRefusal(keyType) || (t.kind === "map" && scalarRefusal(t.value)))
+                    throw new LlvmUnsupportedError(`publish of ${typeKey(t)}`, loc);
+                this.host.declare(`declare void @scr_pub_map(ptr, ptr, ptr, ptr)`);
+                return [
+                    head,
+                    `entry:`,
+                    `  call void @scr_pub_map(ptr %o, ptr %ctx, ptr ${k ?? "null"}, ptr ${v ?? "null"})`,
+                    `  ret void`,
+                    `}`,
+                ];
+            }
+            case "union": {
+                const def = this.host.unionsById.get(t.unionId);
+                if (!def)
+                    throw new LlvmUnsupportedError(`publish of an unknown union ${t.unionId}`, loc);
+                const arms = def.arms.map((arm, tag) => ({
+                    tag,
+                    fn: this.walker(arm, `${path} (as ${typeKey(arm)})`, loc),
+                }));
+                // Union arms are boxed pointers or plain scalars; a date arm would
+                // hold time-value bits that no walker may push.
+                if (def.arms.some((arm) => scalarRefusal(arm)))
+                    throw new LlvmUnsupportedError(`publish of ${typeKey(t)}`, loc);
+                const live = arms.filter((a) => a.fn !== null);
+                const lines = [
+                    head,
+                    `entry:`,
+                    // Union boxes always carry a collector header (scr_union_release).
+                    `  %m = call zeroext i1 @scr_pub_mark(ptr %ctx, ptr %o, i1 true, i32 ${PUB_KIND_UNION})`,
+                    `  br i1 %m, label %walk, label %done`,
+                    `walk:`,
+                ];
+                if (live.length === 0)
+                    lines.push(`  br label %done`);
+                else {
+                    lines.push(`  %tagp = getelementptr inbounds %ScrUnion, ptr %o, i64 0, i32 1`, `  %tag = load i32, ptr %tagp`, `  %slotp = getelementptr inbounds %ScrUnion, ptr %o, i64 0, i32 5`, `  %slot = load ptr, ptr %slotp`, `  switch i32 %tag, label %done [ ${live.map((a) => `i32 ${a.tag}, label %arm${a.tag}`).join(" ")} ]`);
+                    for (const a of live)
+                        lines.push(`arm${a.tag}:`, `  call void @scr_pub_push(ptr %ctx, ptr %slot, ptr ${a.fn})`, `  br label %done`);
+                }
+                lines.push(`done:`, `  ret void`, `}`);
+                return lines;
+            }
+            case "record": {
+                const shape = this.host.recordsById.get(t.shapeId);
+                if (!shape)
+                    throw new LlvmUnsupportedError(`publish of an unknown record ${t.shapeId}`, loc);
+                const members = [
+                    ...shape.fields.map((f, i) => ({ name: f.name, index: i + 1, type: f.type })),
+                    ...(shape.indexValue
+                        ? [
+                            {
+                                name: "[key: string]",
+                                index: shape.fields.length + 1,
+                                type: mapOf(STRING, shape.indexValue),
+                            },
+                        ]
+                        : []),
+                ];
+                return this.objectBody(head, mangleRecordStruct(shape.id), this.host.tracedShapes.has(`record:${shape.id}`), members, path, loc);
+            }
+            case "object":
+                return this.classBody(t.className, head, path, loc);
+            default:
+                throw new LlvmUnsupportedError(`publish of ${typeKey(t)}`, loc);
+        }
+    }
+    /** A class walker: the class itself, or a dispatch over every class of
+     * its subtree when it belongs to a hierarchy. */
+    classBody(className, head, path, loc) {
+        const info = this.host.classInfo(className);
+        if (!info.hierarchy)
+            return this.concreteClassBody(info, head, path, loc);
+        const members = this.host.classesInInterval(info.pre, info.post);
+        const cases = members.map((name) => {
+            const sub = this.host.classInfo(name);
+            const sym = `@sc_pub_${this.counter++}`;
+            const subHead = `define internal void ${sym}(ptr %o, ptr %ctx) ${FN_ATTRS} { ; publish class ${llvmComment(name)}`;
+            this.defsOut.push(...this.concreteClassBody(sub, subHead, `${path} (a ${name})`, loc), ``);
+            return { pre: sub.pre, sym, name };
+        });
+        const lines = [
+            head,
+            `entry:`,
+            `  %vtp = getelementptr inbounds ptr, ptr %o, i64 1`,
+            `  %vt = load ptr, ptr %vtp`,
+            `  %pre = load ${this.host.sizeType}, ptr %vt`,
+            `  switch ${this.host.sizeType} %pre, label %other [ ${cases.map((c) => `${this.host.sizeType} ${c.pre}, label %c${c.pre}`).join(" ")} ]`,
+        ];
+        for (const c of cases)
+            lines.push(`c${c.pre}: ; ${llvmComment(c.name)}`, `  call void ${c.sym}(ptr %o, ptr %ctx)`, `  ret void`);
+        lines.push(`other:`, `  unreachable`, `}`);
+        return lines;
+    }
+    concreteClassBody(info, head, path, loc) {
+        if (info.runtimeRooted) {
+            const refuse = this.refusal(info.def.name === "%Error" || info.def.base !== undefined
+                ? "Cannot publish a Error object"
+                : "Cannot publish this value", `${info.def.name} at ${path}`);
+            return [head, `entry:`, `  call void ${refuse}(ptr %o, ptr %ctx)`, `  ret void`, `}`];
+        }
+        return this.objectBody(head, mangleClassStruct(info.def.name), this.host.tracedShapes.has(`object:${info.def.name}`), info.fields, path, loc);
+    }
+    objectBody(head, struct, traced, members, path, loc) {
+        const lines = [
+            head,
+            `entry:`,
+            `  %m = call zeroext i1 @scr_pub_mark(ptr %ctx, ptr %o, i1 ${traced ? "true" : "false"}, i32 ${PUB_KIND_OBJECT})`,
+            `  br i1 %m, label %walk, label %done`,
+            `walk:`,
+        ];
+        let n = 0;
+        for (const member of members) {
+            const fn = this.walker(member.type, `${path}.${member.name}`, loc);
+            if (fn === null)
+                continue;
+            if (scalarRefusal(member.type)) {
+                // A scalar field (a date) always holds a value: refuse.
+                lines.push(`  call void ${fn}(ptr null, ptr %ctx) ; ${llvmComment(member.name)}`);
+                continue;
+            }
+            lines.push(`  %f${n} = getelementptr inbounds %${struct}, ptr %o, i64 0, i32 ${member.index}`, `  %v${n} = load ptr, ptr %f${n}`, `  call void @scr_pub_push(ptr %ctx, ptr %v${n}, ptr ${fn}) ; ${llvmComment(member.name)}`);
+            n++;
+        }
+        lines.push(`  br label %done`, `done:`, `  ret void`, `}`);
+        return lines;
+    }
+}
+/** Unpublishable types stored as scalars (dates are time values, process
+ * streams file descriptors): no pointer to push, and always a value. */
+function scalarRefusal(t) {
+    return t.kind === "date" || t.kind === "procStream";
+}
+/** The Node.js implementation's refusal for a value of type `t`. */
+function refusalMessage(t) {
+    switch (t.kind) {
+        case "func":
+            return "Cannot publish a function";
+        case "symbol":
+            return "Cannot publish a symbol";
+        case "regex":
+            return "Cannot publish a RegExp object";
+        case "date":
+            return "Cannot publish a Date object";
+        case "promise":
+            return "Cannot publish a Promise object";
+        case "bytes":
+            return `Cannot publish a ${BYTES_NAMES[t.elem] ?? "Uint8Array"} object`;
+        default:
+            return "Cannot publish this value";
+    }
+}
+const BYTES_NAMES = {
+    u8: "Uint8Array",
+    u8c: "Uint8ClampedArray",
+    i8: "Int8Array",
+    u16: "Uint16Array",
+    i16: "Int16Array",
+    u32: "Uint32Array",
+    i32: "Int32Array",
+    f32: "Float32Array",
+    f64: "Float64Array",
+};
+function llvmComment(text) {
+    return text.replace(/[\r\n]/g, " ");
+}
+//# sourceMappingURL=publish.js.map

package/dist/backend/llvm/publish.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"publish.js","sourceRoot":"","sources":["../../../src/backend/llvm/publish.ts"],"names":[],"mappings":"AAwBA,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AA8BxD,uFAAuF;AACvF,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,cAAc,GAAG,CAAC,CAAC;AAEzB,MAAM,eAAe,GAAG;IACtB,MAAM,EAAE,iBAAiB;IACzB,MAAM,EAAE,kBAAkB;IAC1B,GAAG,EAAE,iBAAiB;CACd,CAAC;AAEX,MAAM,OAAO,cAAc;IAKI,IAAI;IAJhB,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC3C,OAAO,GAAa,EAAE,CAAC;IAChC,OAAO,GAAG,CAAC,CAAC;IAEpB,YAA6B,IAAiB;oBAAjB,IAAI;IAAgB,CAAC;IAElD,+DAA+D;IAC/D,IAAI;QACF,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;iDAC6C;IAC7C,WAAW,CAAC,CAAS,EAAE,CAAS,EAAE,GAAW;QAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,uEAAuE;YACvE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC;YAC3D,OAAO,CAAC,uCAAuC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAC1F,CAAC;QACD,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,qBAAqB,EAAE,GAAG,CAAC,CAAC;QACtD,IAAI,EAAE,KAAK,IAAI;YAAE,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;QACzD,OAAO,CAAC,gCAAgC,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAED,wEAAwE;IACxE,MAAM,CAAC,CAAS,EAAE,IAAY,EAAE,GAAW;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,GAAG,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QACrD,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,KAAK,KAAK,CAAC;YACX,KAAK,MAAM,CAAC;YACZ,KAAK,YAAY,CAAC;YAClB,KAAK,OAAO,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,IAAI,CAAC;YACd,KAAK,QAAQ,CAAC;YACd,KAAK,QAAQ,CAAC;YACd,KAAK,KAAK,EAAE,CAAC;gBACX,MAAM,GAAG,GAAG,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,GAAG,YAAY,CAAC,CAAC;gBACnD,OAAO,GAAG,CAAC;YACb,CAAC;YACD,KAAK,QAAQ,CAAC;YACd,KAAK,QAAQ,CAAC;YACd,KAAK,OAAO,CAAC;YACb,KAAK,KAAK,CAAC;YACX,KAAK,KAAK,CAAC;YACX,KAAK,OAAO;gBACV,MAAM;YACR;gBACE,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACtC,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,CAAC;QAC/B,OAAO,GAAG,CAAC;IACb,CAAC;IAEgB,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD;;gCAE4B;IACpB,OAAO,CAAC,OAAe,EAAE,MAAc;QAC7C,MAAM,GAAG,GAAG,GAAG,OAAO,KAAK,MAAM,EAAE,CAAC;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACtC,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC5B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;QACpE,IAAI,CAAC,OAAO,CAAC,IAAI,CACf,wBAAwB,GAAG,sBAAsB,QAAQ,eAAe,WAAW,CAAC,MAAM,CAAC,EAAE,EAC7F,QAAQ,EACR,gDAAgD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EACzG,YAAY,EACZ,GAAG,EACH,EAAE,CACH,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,6DAA6D,CAAC,CAAC;QACjF,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,2CAA2C,CAAC,CAAC;IACjE,CAAC;IAED,kDAAkD;IAC1C,IAAI,CAAC,CAAS,EAAE,GAAW,EAAE,IAAY,EAAE,GAAW;QAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,wBAAwB,GAAG,sBAAsB,QAAQ,gBAAgB,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChH,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,KAAK,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,GAAG,IAAI,WAAW,EAAE,GAAG,CAAC,CAAC;gBAC1D,IAAI,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;oBAAE,MAAM,IAAI,oBAAoB,CAAC,cAAc,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC3F,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,OAAO;oBACL,IAAI;oBACJ,QAAQ;oBACR,kDAAkD,IAAI,IAAI,MAAM,GAAG;oBACnE,YAAY;oBACZ,GAAG;iBACJ,CAAC;YACJ,CAAC;YACD,KAAK,KAAK,CAAC;YACX,KAAK,KAAK,EAAE,CAAC;gBACX,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAClD,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,IAAI,OAAO,EAAE,GAAG,CAAC,CAAC;gBACpD,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,GAAG,IAAI,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAChF,IAAI,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,IAAI,aAAa,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;oBACxE,MAAM,IAAI,oBAAoB,CAAC,cAAc,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAClE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,+CAA+C,CAAC,CAAC;gBACnE,OAAO;oBACL,IAAI;oBACJ,QAAQ;oBACR,kDAAkD,CAAC,IAAI,MAAM,SAAS,CAAC,IAAI,MAAM,GAAG;oBACpF,YAAY;oBACZ,GAAG;iBACJ,CAAC;YACJ,CAAC;YACD,KAAK,OAAO,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAChD,IAAI,CAAC,GAAG;oBAAE,MAAM,IAAI,oBAAoB,CAAC,+BAA+B,CAAC,CAAC,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC1F,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBACvC,GAAG;oBACH,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,QAAQ,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC;iBAC1D,CAAC,CAAC,CAAC;gBACJ,mEAAmE;gBACnE,gDAAgD;gBAChD,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;oBAC5C,MAAM,IAAI,oBAAoB,CAAC,cAAc,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBAClE,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC;gBAC/C,MAAM,KAAK,GAAG;oBACZ,IAAI;oBACJ,QAAQ;oBACR,mEAAmE;oBACnE,uEAAuE,cAAc,GAAG;oBACxF,sCAAsC;oBACtC,OAAO;iBACR,CAAC;gBACF,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;qBACjD,CAAC;oBACJ,KAAK,CAAC,IAAI,CACR,kEAAkE,EAClE,8BAA8B,EAC9B,mEAAmE,EACnE,gCAAgC,EAChC,oCAAoC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,GAAG,eAAe,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CACtG,CAAC;oBACF,KAAK,MAAM,CAAC,IAAI,IAAI;wBAClB,KAAK,CAAC,IAAI,CACR,MAAM,CAAC,CAAC,GAAG,GAAG,EACd,sDAAsD,CAAC,CAAC,EAAE,GAAG,EAC7D,kBAAkB,CACnB,CAAC;gBACN,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,CAAC;gBACvC,OAAO,KAAK,CAAC;YACf,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACd,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACnD,IAAI,CAAC,KAAK;oBACR,MAAM,IAAI,oBAAoB,CAAC,gCAAgC,CAAC,CAAC,OAAO,EAAE,EAAE,GAAG,CAAC,CAAC;gBACnF,MAAM,OAAO,GAAG;oBACd,GAAG,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC7E,GAAG,CAAC,KAAK,CAAC,UAAU;wBAClB,CAAC,CAAC;4BACE;gCACE,IAAI,EAAE,eAAe;gCACrB,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gCAC9B,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC;6BACtC;yBACF;wBACH,CAAC,CAAC,EAAE,CAAC;iBACR,CAAC;gBACF,OAAO,IAAI,CAAC,UAAU,CACpB,IAAI,EACJ,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC,EAC5B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,KAAK,CAAC,EAAE,EAAE,CAAC,EAChD,OAAO,EACP,IAAI,EACJ,GAAG,CACJ,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACtD;gBACE,MAAM,IAAI,oBAAoB,CAAC,cAAc,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED;qDACiD;IACzC,SAAS,CAAC,SAAiB,EAAE,IAAY,EAAE,IAAY,EAAE,GAAW;QAC1E,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACjC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,wBAAwB,GAAG,sBAAsB,QAAQ,sBAAsB,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;YACnH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,IAAI,OAAO,IAAI,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3F,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;QACrC,CAAC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG;YACZ,IAAI;YACJ,QAAQ;YACR,oDAAoD;YACpD,4BAA4B;YAC5B,iBAAiB,IAAI,CAAC,IAAI,CAAC,QAAQ,WAAW;YAC9C,YAAY,IAAI,CAAC,IAAI,CAAC,QAAQ,yBAAyB,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI;SAC1I,CAAC;QACF,KAAK,MAAM,CAAC,IAAI,KAAK;YACnB,KAAK,CAAC,IAAI,CACR,IAAI,CAAC,CAAC,GAAG,OAAO,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EACrC,eAAe,CAAC,CAAC,GAAG,oBAAoB,EACxC,YAAY,CACb,CAAC;QACJ,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC;QAC3C,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,iBAAiB,CACvB,IAAsB,EACtB,IAAY,EACZ,IAAY,EACZ,GAAW;QAEX,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CACzB,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,SAAS;gBACvD,CAAC,CAAC,+BAA+B;gBACjC,CAAC,CAAC,2BAA2B,EAC/B,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,OAAO,IAAI,EAAE,CAC9B,CAAC;YACF,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,eAAe,MAAM,oBAAoB,EAAE,YAAY,EAAE,GAAG,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CACpB,IAAI,EACJ,iBAAiB,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAChC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EACrD,IAAI,CAAC,MAAM,EACX,IAAI,EACJ,GAAG,CACJ,CAAC;IACJ,CAAC;IAEO,UAAU,CAChB,IAAY,EACZ,MAAc,EACd,MAAe,EACf,OAAwD,EACxD,IAAY,EACZ,GAAW;QAEX,MAAM,KAAK,GAAG;YACZ,IAAI;YACJ,QAAQ;YACR,6DAA6D,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,SAAS,eAAe,GAAG;YACjH,sCAAsC;YACtC,OAAO;SACR,CAAC;QACF,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,IAAI,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;YACnE,IAAI,EAAE,KAAK,IAAI;gBAAE,SAAS;YAC1B,IAAI,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,wDAAwD;gBACxD,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,0BAA0B,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClF,SAAS;YACX,CAAC;YACD,KAAK,CAAC,IAAI,CACR,OAAO,CAAC,8BAA8B,MAAM,wBAAwB,MAAM,CAAC,KAAK,EAAE,EAClF,OAAO,CAAC,sBAAsB,CAAC,EAAE,EACjC,6CAA6C,CAAC,SAAS,EAAE,OAAO,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAC3F,CAAC;YACF,CAAC,EAAE,CAAC;QACN,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,kBAAkB,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,CAAC;QAC3D,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED;wEACwE;AACxE,SAAS,aAAa,CAAC,CAAS;IAC9B,OAAO,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC;AACtD,CAAC;AAED,oEAAoE;AACpE,SAAS,cAAc,CAAC,CAAS;IAC/B,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,KAAK,MAAM;YACT,OAAO,2BAA2B,CAAC;QACrC,KAAK,QAAQ;YACX,OAAO,yBAAyB,CAAC;QACnC,KAAK,OAAO;YACV,OAAO,gCAAgC,CAAC;QAC1C,KAAK,MAAM;YACT,OAAO,8BAA8B,CAAC;QACxC,KAAK,SAAS;YACZ,OAAO,iCAAiC,CAAC;QAC3C,KAAK,OAAO;YACV,OAAO,oBAAoB,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,YAAY,SAAS,CAAC;QAC1E;YACE,OAAO,2BAA2B,CAAC;IACvC,CAAC;AACH,CAAC;AAED,MAAM,WAAW,GAA2B;IAC1C,EAAE,EAAE,YAAY;IAChB,GAAG,EAAE,mBAAmB;IACxB,EAAE,EAAE,WAAW;IACf,GAAG,EAAE,aAAa;IAClB,GAAG,EAAE,YAAY;IACjB,GAAG,EAAE,aAAa;IAClB,GAAG,EAAE,YAAY;IACjB,GAAG,EAAE,cAAc;IACnB,GAAG,EAAE,cAAc;CACpB,CAAC;AAEF,SAAS,WAAW,CAAC,IAAY;IAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;AACtC,CAAC"}

package/dist/backend/llvm/rebound-parameters.d.ts

@@ -0,0 +1,12 @@
+import type { IrFunction, IrType } from "../../ir/ir.js";
+/** Parameters that keep the borrowed calling convention although the body
+ * rebinds them (`source = (source as LiteralType).regularType`). The
+ * incoming value stays the caller's borrow; every value the body assigns is
+ * owned through a separate owner slot, which each rebinding releases and
+ * replaces and which the function scope releases on every exit. Reads see
+ * the parameter slot, so whole-value uses retain as for any binding.
+ *
+ * Only plain statement assignments qualify: expression-position writes,
+ * loop and catch bindings, declarations and captures keep the ordinary
+ * owned parameter. Suspending bodies and environments use the owned ABI. */
+export declare function findReboundParameters(fn: IrFunction, plainReference: (type: IrType) => boolean, excluded: ReadonlySet<string>): ReadonlySet<string>;

package/dist/backend/llvm/rebound-parameters.js

@@ -0,0 +1,72 @@
+import { everyExprChild, everyStmtChild } from "../../ir/traverse.js";
+/** Parameters that keep the borrowed calling convention although the body
+ * rebinds them (`source = (source as LiteralType).regularType`). The
+ * incoming value stays the caller's borrow; every value the body assigns is
+ * owned through a separate owner slot, which each rebinding releases and
+ * replaces and which the function scope releases on every exit. Reads see
+ * the parameter slot, so whole-value uses retain as for any binding.
+ *
+ * Only plain statement assignments qualify: expression-position writes,
+ * loop and catch bindings, declarations and captures keep the ordinary
+ * owned parameter. Suspending bodies and environments use the owned ABI. */
+export function findReboundParameters(fn, plainReference, excluded) {
+    const result = new Set();
+    if (fn.async || fn.generator || fn.captures || fn.classCaptures)
+        return result;
+    const locals = new Map(fn.locals.map((local) => [local.id, local]));
+    const candidates = new Set(fn.params
+        .filter((param) => {
+        const local = locals.get(param.localId);
+        return (local !== undefined &&
+            !local.boxed &&
+            !local.tdz &&
+            !excluded.has(param.localId) &&
+            plainReference(param.type));
+    })
+        .map((param) => param.localId));
+    if (candidates.size === 0)
+        return result;
+    const rebound = new Set();
+    const invalid = new Set();
+    const expr = (e) => {
+        switch (e.kind) {
+            case "assignExpr":
+            case "incDec":
+                invalid.add(e.localId);
+                break;
+            case "closure":
+            case "classRef":
+                for (const id of e.captures ?? [])
+                    invalid.add(id);
+                break;
+        }
+        return everyExprChild(e, expr, (s) => stmt(s, true));
+    };
+    const stmt = (s, inExpression) => {
+        switch (s.kind) {
+            case "assign":
+                if (inExpression)
+                    invalid.add(s.localId);
+                else
+                    rebound.add(s.localId);
+                break;
+            case "varDecl":
+            case "forOf":
+            case "rethrow":
+                invalid.add(s.localId);
+                break;
+            case "tryCatch":
+                if (s.catchLocalId !== null)
+                    invalid.add(s.catchLocalId);
+                break;
+        }
+        return everyStmtChild(s, expr, (child) => stmt(child, inExpression));
+    };
+    for (const s of fn.body)
+        stmt(s, false);
+    for (const id of candidates)
+        if (rebound.has(id) && !invalid.has(id))
+            result.add(id);
+    return result;
+}
+//# sourceMappingURL=rebound-parameters.js.map

package/dist/backend/llvm/rebound-parameters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rebound-parameters.js","sourceRoot":"","sources":["../../../src/backend/llvm/rebound-parameters.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtE;;;;;;;;;4EAS4E;AAC5E,MAAM,UAAU,qBAAqB,CACnC,EAAc,EACd,cAAyC,EACzC,QAA6B;IAE7B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACjC,IAAI,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,IAAI,EAAE,CAAC,aAAa;QAAE,OAAO,MAAM,CAAC;IAC/E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACpE,MAAM,UAAU,GAAG,IAAI,GAAG,CACxB,EAAE,CAAC,MAAM;SACN,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;QAChB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACxC,OAAO,CACL,KAAK,KAAK,SAAS;YACnB,CAAC,KAAK,CAAC,KAAK;YACZ,CAAC,KAAK,CAAC,GAAG;YACV,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC;YAC5B,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAC3B,CAAC;IACJ,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CACjC,CAAC;IACF,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,MAAM,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,IAAI,GAAG,CAAC,CAAS,EAAW,EAAE;QAClC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,KAAK,YAAY,CAAC;YAClB,KAAK,QAAQ;gBACX,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACvB,MAAM;YACR,KAAK,SAAS,CAAC;YACf,KAAK,UAAU;gBACb,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ,IAAI,EAAE;oBAAE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACnD,MAAM;QACV,CAAC;QACD,OAAO,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC;IACF,MAAM,IAAI,GAAG,CAAC,CAAS,EAAE,YAAqB,EAAW,EAAE;QACzD,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,KAAK,QAAQ;gBACX,IAAI,YAAY;oBAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;;oBACpC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAC5B,MAAM;YACR,KAAK,SAAS,CAAC;YACf,KAAK,OAAO,CAAC;YACb,KAAK,SAAS;gBACZ,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACvB,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,CAAC,YAAY,KAAK,IAAI;oBAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;gBACzD,MAAM;QACV,CAAC;QACD,OAAO,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC;IACF,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC,IAAI;QAAE,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxC,KAAK,MAAM,EAAE,IAAI,UAAU;QAAE,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACrF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/backend/llvm/rebound-parameters.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/backend/llvm/rebound-parameters.test.js

@@ -0,0 +1,67 @@
+import { expect, test } from "vitest";
+import { F64 } from "../../ir/ir.js";
+import { findReboundParameters } from "./rebound-parameters.js";
+const loc = { file: "rebound.ts", start: 0, end: 0 };
+const NODE = { kind: "object", className: "Node" };
+const ref = (localId) => ({ kind: "varRef", localId, type: NODE, loc });
+const assign = (localId, value) => ({
+    kind: "assign",
+    localId,
+    value,
+    loc,
+});
+function fn(body, params) {
+    return {
+        name: "work",
+        params: params.map((p) => ({ localId: p.id, name: p.id, type: p.type ?? NODE })),
+        locals: params.map((p) => ({ id: p.id, name: p.id, type: p.type ?? NODE, mutable: true })),
+        body,
+        returnType: F64,
+        loc,
+    };
+}
+const plain = (type) => type.kind === "object";
+test("statement rebinding of reference parameters keeps the borrowed convention", () => {
+    const body = [
+        assign("a", ref("b")),
+        { kind: "if", cond: ref("a"), then: [assign("b", ref("a"))], else_: null, loc },
+    ];
+    const f = fn(body, [{ id: "a" }, { id: "b" }, { id: "untouched" }]);
+    expect(findReboundParameters(f, plain, new Set())).toEqual(new Set(["a", "b"]));
+    // Parameters already borrowed as walks are left to the walk analysis.
+    expect(findReboundParameters(f, plain, new Set(["a"]))).toEqual(new Set(["b"]));
+});
+test("expression writes, scalar types, captures and suspension keep the owned parameter", () => {
+    const sequence = {
+        kind: "seqExpr",
+        stmts: [assign("a", ref("b"))],
+        result: { kind: "numLit", value: 0, type: F64, loc },
+        type: F64,
+        loc,
+    };
+    const inExpression = fn([{ kind: "exprStmt", expr: sequence, loc }], [{ id: "a" }, { id: "b" }]);
+    expect(findReboundParameters(inExpression, plain, new Set()).size).toBe(0);
+    const scalar = fn([assign("n", { kind: "numLit", value: 1, type: F64, loc })], [{ id: "n", type: F64 }]);
+    expect(findReboundParameters(scalar, plain, new Set()).size).toBe(0);
+    const captured = fn([
+        assign("a", ref("b")),
+        {
+            kind: "exprStmt",
+            expr: {
+                kind: "closure",
+                name: "inner",
+                captures: ["a"],
+                type: F64,
+                loc,
+            },
+            loc,
+        },
+    ], [{ id: "a" }, { id: "b" }]);
+    expect(findReboundParameters(captured, plain, new Set()).size).toBe(0);
+    const suspending = {
+        ...fn([assign("a", ref("b"))], [{ id: "a" }, { id: "b" }]),
+        async: true,
+    };
+    expect(findReboundParameters(suspending, plain, new Set()).size).toBe(0);
+});
+//# sourceMappingURL=rebound-parameters.test.js.map

package/dist/backend/llvm/rebound-parameters.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rebound-parameters.test.js","sourceRoot":"","sources":["../../../src/backend/llvm/rebound-parameters.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,GAAG,EAA0D,MAAM,gBAAgB,CAAC;AAC7F,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAEhE,MAAM,GAAG,GAAG,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;AACrD,MAAM,IAAI,GAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;AAC3D,MAAM,GAAG,GAAG,CAAC,OAAe,EAAU,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;AACxF,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,KAAa,EAAU,EAAE,CAAC,CAAC;IAC1D,IAAI,EAAE,QAAQ;IACd,OAAO;IACP,KAAK;IACL,GAAG;CACJ,CAAC,CAAC;AAEH,SAAS,EAAE,CAAC,IAAc,EAAE,MAAuC;IACjE,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;QAChF,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1F,IAAI;QACJ,UAAU,EAAE,GAAG;QACf,GAAG;KACJ,CAAC;AACJ,CAAC;AAED,MAAM,KAAK,GAAG,CAAC,IAAY,EAAW,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC;AAEhE,IAAI,CAAC,2EAA2E,EAAE,GAAG,EAAE;IACrF,MAAM,IAAI,GAAG;QACX,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACrB,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAY;KAC1F,CAAC;IACF,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;IACpE,MAAM,CAAC,qBAAqB,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAChF,sEAAsE;IACtE,MAAM,CAAC,qBAAqB,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;AAClF,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,mFAAmF,EAAE,GAAG,EAAE;IAC7F,MAAM,QAAQ,GAAW;QACvB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE;QACpD,IAAI,EAAE,GAAG;QACT,GAAG;KACJ,CAAC;IACF,MAAM,YAAY,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IACjG,MAAM,CAAC,qBAAqB,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3E,MAAM,MAAM,GAAG,EAAE,CACf,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CACzB,CAAC;IACF,MAAM,CAAC,qBAAqB,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,EAAE,CACjB;QACE,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QACrB;YACE,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE;gBACJ,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,CAAC,GAAG,CAAC;gBACf,IAAI,EAAE,GAAG;gBACT,GAAG;aACiB;YACtB,GAAG;SACJ;KACF,EACD,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAC3B,CAAC;IACF,MAAM,CAAC,qBAAqB,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvE,MAAM,UAAU,GAAe;QAC7B,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,KAAK,EAAE,IAAI;KACZ,CAAC;IACF,MAAM,CAAC,qBAAqB,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAC3E,CAAC,CAAC,CAAC"}

package/dist/backend/llvm/reference-effects.d.ts

@@ -1,11 +1,25 @@
-import { type IrExpr, type IrFunction, type IrStmt } from "../../ir/ir.js";
+import { type IrClassDef, type IrExpr, type IrFunction, type IrStmt } from "../../ir/ir.js";
 type Call = IrExpr & {
     kind: "call";
 };
+type VirtualCall = IrExpr & {
+    kind: "virtualCall";
+};
 /** These native operations may update lastIndex and allocate results, but
  * never invoke user code or remove an existing reference edge. Results own
  * their payloads independently. Callback replacement uses a different IR. */
 export declare function preservesRegexInputs(method: string): boolean;
+/** The module functions a virtual call can reach: the implementation the
+ * static class inherits or declares, plus every override in its subtree.
+ * Runtime classes keep their dispatch out of view, so they answer null. */
+export declare class VirtualTargets {
+    private readonly classes;
+    private readonly children;
+    private readonly cache;
+    constructor(classes: readonly IrClassDef[]);
+    targets(className: string, method: string): readonly string[] | null;
+    private compute;
+}
 /** Reference preservation is weaker than purity: scalar writes, allocation
  * and throwing are allowed. Caller owners and their reference edges must
  * survive until the consuming operation; callee-local rebinding is private.
@@ -17,8 +31,10 @@ export declare class ReferenceEffects {
     private readonly intrinsicCall;
     readonly functions: Set<string>;
     private readonly expressions;
-    constructor(functions: ReadonlyMap<string, IrFunction>, intrinsicCall: (call: Call) => boolean);
+    private readonly virtualTargets;
+    constructor(functions: ReadonlyMap<string, IrFunction>, intrinsicCall: (call: Call) => boolean, classes?: readonly IrClassDef[]);
     preserves(value: IrExpr): boolean;
+    virtualPreserves(call: VirtualCall): boolean;
     preservesScope(body: IrStmt[]): boolean;
 }
 export {};