scriptc/packages/runtime/src/scr_alloc.test.ts

import { execFile } from "node:child_process";
import { mkdtemp, rm } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { promisify } from "node:util";
import { expect, test } from "vitest";

const exec = promisify(execFile);

// The size-class allocator is compiled out under AddressSanitizer, so the
// UBSan-only build is the one that exercises it; the ASan build pins the
// system-allocator bypass the sanitizer lane relies on.
// Worker executables (-DSCR_WORKERS) give each script thread its own arena;
// the ThreadSanitizer build checks the cross-thread handoff for races. The
// last build runs with glibc's memory tagging tunable set, which turns the
// thread-heap huge-page policy off.
test.each([
  ["size-class", ["-fsanitize=undefined"], {}],
  ["system", ["-fsanitize=address,undefined"], {}],
  ["size-class workers", ["-fsanitize=undefined", "-DSCR_WORKERS", "-pthread"], {}],
  ["size-class workers", ["-fsanitize=thread", "-DSCR_WORKERS", "-pthread"], {}],
  [
    "size-class workers",
    ["-fsanitize=undefined", "-DSCR_WORKERS", "-pthread"],
    { GLIBC_TUNABLES: "glibc.mem.tagging=3" },
  ],
])("small-object allocator contracts hold (%s, %j, %j)", async (mode, sanitize, extraEnv) => {
  const scratch = await mkdtemp(join(tmpdir(), "scriptc-alloc-"));
  try {
    const binary = join(scratch, "alloc");
    await exec("clang", [
      "-std=c11",
      "-O1",
      "-Wall",
      "-Wextra",
      // Linux runtime packs compile with _GNU_SOURCE, which exposes the
      // madvise advice the page policy uses; build the checks the same way.
      ...(process.platform === "linux" ? ["-D_GNU_SOURCE"] : []),
      ...sanitize,
      join(import.meta.dirname, "scr_alloc.test.c"),
      join(import.meta.dirname, "scr_alloc.c"),
      "-o",
      binary,
    ]);
    const result = await exec(binary, [], {
      env: {
        ...process.env,
        ASAN_OPTIONS: "halt_on_error=1",
        TSAN_OPTIONS: "halt_on_error=1",
        UBSAN_OPTIONS: "halt_on_error=1",
        ...extraEnv,
      },
    });
    expect(result.stdout).toBe(`allocator checks passed: ${mode}\n`);
    expect(result.stderr).toBe("");
  } finally {
    await rm(scratch, { recursive: true, force: true });
  }
});