Bonobo
114 lines
3.3 KiB
TypeScript
114 lines
3.3 KiB
TypeScript
#!/usr/bin/env bun
|
|
|
|
import * as fs from "node:fs/promises";
|
|
import * as path from "node:path";
|
|
|
|
const repoRoot = path.join(import.meta.dir, "..");
|
|
const nativeDir = path.resolve(Bun.env.PI_NATIVE_VERIFY_DIR ?? path.join(repoRoot, "packages", "natives", "native"));
|
|
const defaultExpectedAddons = [
|
|
"linux-x64-modern",
|
|
"linux-x64-baseline",
|
|
"linux-arm64",
|
|
"darwin-x64-modern",
|
|
"darwin-x64-baseline",
|
|
"darwin-arm64",
|
|
"win32-x64-modern",
|
|
"win32-x64-baseline",
|
|
] as const;
|
|
const x64LinuxIsaContracts = [
|
|
{ addon: "linux-x64-baseline", filename: "pi_natives.linux-x64-baseline.node", label: "x86-64-v2" },
|
|
{ addon: "linux-x64-modern", filename: "pi_natives.linux-x64-modern.node", label: "x86-64-v3" },
|
|
] as const;
|
|
const AVX512_REGISTER_PATTERN = /\bzmm\d+\b|\bk[0-7]\b/;
|
|
|
|
export function hasAvx512Markers(disassembly: string): boolean {
|
|
return disassembly.split("\n").some(line => {
|
|
if (AVX512_REGISTER_PATTERN.test(line)) {
|
|
return true;
|
|
}
|
|
|
|
const columns = line.split("\t");
|
|
if (columns.length < 3) {
|
|
return false;
|
|
}
|
|
|
|
const bytes = columns[1]?.trim() ?? "";
|
|
const instruction = columns[2]?.trim() ?? "";
|
|
return bytes.startsWith("62 ") && /^[a-z]/i.test(instruction) && !instruction.startsWith(".byte");
|
|
});
|
|
}
|
|
|
|
function disassemble(binaryPath: string): string {
|
|
const objdumpPath = Bun.which("objdump");
|
|
if (!objdumpPath) {
|
|
throw new Error("objdump is required to verify linux-x64 native ISA contracts.");
|
|
}
|
|
|
|
const result = Bun.spawnSync([objdumpPath, "-d", binaryPath], { stdout: "pipe", stderr: "pipe" });
|
|
if (result.exitCode !== 0) {
|
|
const stderr = result.stderr.toString("utf-8").trim();
|
|
throw new Error(`objdump failed for ${binaryPath}${stderr ? `:\n${stderr}` : ""}`);
|
|
}
|
|
|
|
return result.stdout.toString("utf-8");
|
|
}
|
|
|
|
function resolveExpectedAddons(): string[] {
|
|
const configured = (Bun.env.PI_NATIVE_EXPECTED_ADDONS ?? "").trim();
|
|
if (!configured) {
|
|
return [...defaultExpectedAddons];
|
|
}
|
|
|
|
return configured.split(/[\s,]+/).filter(Boolean);
|
|
}
|
|
|
|
async function main(): Promise<void> {
|
|
const entries = await fs.readdir(nativeDir);
|
|
const expectedAddons = resolveExpectedAddons();
|
|
|
|
console.log(`Native addons downloaded from ${nativeDir}:`);
|
|
for (const entry of entries.sort((a, b) => a.localeCompare(b))) {
|
|
console.log(` ${entry}`);
|
|
}
|
|
console.log();
|
|
console.log(`Expected addons: ${expectedAddons.join(", ")}`);
|
|
|
|
const missingAddons = expectedAddons.filter((platform) => !entries.includes(`pi_natives.${platform}.node`));
|
|
if (missingAddons.length > 0) {
|
|
for (const platform of missingAddons) {
|
|
console.error(`MISSING pi_natives.${platform}.node`);
|
|
}
|
|
process.exit(1);
|
|
}
|
|
|
|
for (const platform of expectedAddons) {
|
|
console.log(`OK pi_natives.${platform}.node`);
|
|
}
|
|
|
|
const isaFailures: string[] = [];
|
|
for (const contract of x64LinuxIsaContracts) {
|
|
if (!expectedAddons.includes(contract.addon)) {
|
|
continue;
|
|
}
|
|
|
|
const binaryPath = path.join(nativeDir, contract.filename);
|
|
const disassembly = disassemble(binaryPath);
|
|
if (hasAvx512Markers(disassembly)) {
|
|
isaFailures.push(`${contract.filename} contains AVX-512 markers; ${contract.label} artifacts must stay below x86-64-v4.`);
|
|
continue;
|
|
}
|
|
console.log(`OK ${contract.filename} contains no AVX-512 markers`);
|
|
}
|
|
|
|
if (isaFailures.length > 0) {
|
|
for (const failure of isaFailures) {
|
|
console.error(failure);
|
|
}
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
if (import.meta.main) {
|
|
await main();
|
|
}
|