mirror of
https://github.com/morluto/rea.git
synced 2026-11-14 10:43:30 +00:00
morluto
b3657c557f
* refactor(cli): run one-shot commands through one shared owner scripts/rea.mjs and the test CLI session worker each implemented the one-shot command sequence: argument normalization, output admission, configuration errors, version documents, the Amp scope refusal, and result delivery. The worker copy had drifted: it never rendered formatted version reports or the Amp refusal. Move the sequence into src/cliOneShot.ts. Both callers supply a host for their streams, exit and resource ownership. The launcher keeps closed-pipe status, worker shutdown and process exit; the session worker captures output per command. The cliOutput bundle entry is superseded and removed. The dispatcher test now replaces only the command runtime, so output admission runs production code. A new parity test compares the file session with a fresh process. * test(cli): bundle the one-shot entry in the runtime bundle fixture * perf(tests): run plain CLI commands on the file's shared process `cli.run` in `cliTest` files started a fresh compiled CLI for every command, about 0.45s and 260 MB each before any work. The file-scoped session already existed, but callers had to opt in through `runFileCli` or `sharedCliRun`, so 1,300 call sites still spawned. Make the fixture choose: `cli.run` uses the shared process when it reports the command exactly as a fresh process would, and starts one otherwise (stdin, explicit deadlines, MCP and MCP doctor routing, NODE_OPTIONS, or enabled logging). `freshProcess: true` keeps explicit process-level checks. The session worker now presents the launcher's entrypoint, so registration identity matches a fresh process. `runFileCli` and `sharedCliRun` are superseded and removed; their call sites use `cli.run`, and paired fresh/shared checks use `freshProcess`. * style(tests): format the managed member summary call site * test(cli): give each shared-session command an empty stdin The shared CLI process reads session frames from its own stdin. A command that reads stdin without supplied input, such as `compare-wasm-interfaces - --json`, read that pipe and hung, where a fresh process receives EOF. The worker now presents an ended, empty stdin to each command and restores its own afterwards. The parity test compared the routed path with itself after cli.run began choosing the session. It now compares a fresh process with the session directly, including the `mcp add` refusal and a command reading absent stdin. * test(cli): keep default command logging and launcher checks on fresh processes environment-serve and environment-build log at info when REA_LOG_LEVEL is unset, through a logger that writes to stderr directly. The shared CLI process captures only process.stderr writes, so routed runs lost those lines. Declare the default-logging commands beside the command names, have infoLogger read that set, and start a fresh process for them. The log-level precedence checks compared the shared process with itself after routing; their fresh side now passes freshProcess: true.
1433 lines
47 KiB
TypeScript
1433 lines
47 KiB
TypeScript
import { expect, it } from "vitest";
|
|
import { writeFile, unlink } from "node:fs/promises";
|
|
import { join } from "node:path";
|
|
import { createHash } from "node:crypto";
|
|
import { deflateSync } from "node:zlib";
|
|
import { toolContract } from "../../src/contracts/toolContracts.js";
|
|
import { gpuCodeObjectsSchema } from "../../src/domain/gpu/gpuCodeObjects.js";
|
|
import {
|
|
gpuCodeObjectsPageViewResultSchema,
|
|
gpuCodeObjectsSummaryViewResultSchema,
|
|
} from "../../src/domain/analysisView/analysisView.js";
|
|
import { parseEvidence, createEvidence } from "../../src/domain/evidence.js";
|
|
import { createTestTempDirectory } from "../fixtures/temporaryDirectory.js";
|
|
import { connectLocalToolsMcp } from "../fixtures/localToolsMcp.js";
|
|
import { parseMcpToolError } from "../fixtures/mcpToolError.js";
|
|
import { cliTest } from "../support/cli/cliFixture.js";
|
|
import {
|
|
cudaFixture,
|
|
peGpuFixture,
|
|
ccobFixture,
|
|
ptxFixture,
|
|
amdFixture,
|
|
amdNoteFixture,
|
|
cubinFixture,
|
|
hipFixture,
|
|
elfFixture,
|
|
llvmBitcodeFixture,
|
|
} from "../fixtures/gpu/codeObjects.js";
|
|
|
|
cliTest(
|
|
"reads CUDA/PTX and compressed HIP through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-cli-");
|
|
for (const [name, bytes] of [
|
|
["cuda", cudaFixture("zstd")],
|
|
["hip", ccobFixture()],
|
|
["amd", amdFixture()],
|
|
] as const) {
|
|
const path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
...(name === "amd"
|
|
? ["--targets", '["amdgcn-amd-amdhsa--gfx900"]']
|
|
: []),
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
expect(report.kernels).toHaveLength(name === "cuda" ? 2 : 1);
|
|
expect(report.kernels[0]!.parameter_bytes).toBe(12);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"reports an exhausted ELF symbol string scan as capacity instead of corruption",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-string-limit-");
|
|
const bytes = cubinFixture({ name: "k".repeat(65_536) });
|
|
const path = join(root, "long-symbol.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]).toMatchObject({
|
|
kind: "cubin",
|
|
inspection: "capacity",
|
|
});
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"inventories a complete CCOB stream containing ordinary CCOB payload text",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ccob-embedded-marker-");
|
|
const bundle = hipFixture(
|
|
Buffer.from("ordinary CCOB marker"),
|
|
"host-other",
|
|
),
|
|
compressed = deflateSync(bundle, { level: 0 });
|
|
expect(compressed.indexOf("CCOB")).toBeGreaterThan(0);
|
|
for (const version of [1, 2] as const) {
|
|
const width = version === 1 ? 20 : 24,
|
|
header = Buffer.from(
|
|
ccobFixture(version, "zlib", bundle).subarray(0, width),
|
|
);
|
|
if (version === 2) header.writeUInt32LE(width + compressed.length, 8);
|
|
const bytes = Buffer.concat([header, compressed]),
|
|
path = join(root, `v${version}`);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"inventory",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.coverage).toMatchObject({
|
|
discovery: "complete",
|
|
payloads: "not-requested",
|
|
});
|
|
expect(report.objects).toHaveLength(3);
|
|
expect(report.objects[0]).toMatchObject({
|
|
kind: "hip-bundle",
|
|
inspection: "complete",
|
|
encoded_size: compressed.length,
|
|
decoded_size: bundle.length,
|
|
sha256: createHash("sha256").update(bundle).digest("hex"),
|
|
location: { offset: width, size: compressed.length },
|
|
});
|
|
expect(
|
|
report.objects
|
|
.slice(1)
|
|
.map(({ bundle_id, inspection }) => [bundle_id, inspection]),
|
|
).toEqual([
|
|
["host-x86_64-unknown-linux", "complete"],
|
|
["host-other", "complete"],
|
|
]);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"distinguishes unsupported HIP bitcode from corrupted ELF through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-hip-bitcode-cli-"),
|
|
bitcode = llvmBitcodeFixture(),
|
|
corruptedElf = amdFixture(),
|
|
bundleId = "hip-amdgcn-amd-amdhsa--gfx900";
|
|
corruptedElf.writeBigUInt64LE(BigInt(corruptedElf.length + 1), 40);
|
|
for (const [name, payload, analysis, inspection] of [
|
|
["inventory", bitcode, "inventory", "inventory"],
|
|
["bitcode", bitcode, "kernels", "unsupported"],
|
|
["corrupt-elf", corruptedElf, "kernels", "malformed"],
|
|
] as const) {
|
|
const bytes = hipFixture(payload, bundleId),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
analysis,
|
|
"--targets",
|
|
'["gfx900"]',
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
const row = report.objects.find(({ kind }) => kind === "amdgpu")!;
|
|
expect(row).toMatchObject({
|
|
bundle_id: bundleId,
|
|
target: "gfx900",
|
|
payload_target: null,
|
|
inspection,
|
|
encoded_size: payload.length,
|
|
location: {
|
|
container_id: null,
|
|
offset: bytes.length - payload.length,
|
|
size: payload.length,
|
|
},
|
|
sha256:
|
|
analysis === "inventory"
|
|
? null
|
|
: createHash("sha256").update(payload).digest("hex"),
|
|
});
|
|
expect(report.coverage.payloads).toBe(
|
|
analysis === "inventory" ? "not-requested" : "partial",
|
|
);
|
|
expect(report.kernels).toEqual([]);
|
|
if (name === "bitcode") {
|
|
expect(row.limitations.join(" ")).toContain("LLVM bitcode");
|
|
expect(row.limitations.join(" ")).toContain(
|
|
"validity and inner target were not inspected",
|
|
);
|
|
}
|
|
}
|
|
const path = join(root, "raw-bitcode");
|
|
await writeFile(path, bitcode);
|
|
const raw = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(raw.exitCode).toBe(1);
|
|
expect(raw.json).toMatchObject({ code: "unsupported_target" });
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps embedded Clang bundle reads inside their native section",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-section-range-");
|
|
const magic = Buffer.from("__CLANG_OFFLOAD_BUNDLE__"),
|
|
truncated = magic,
|
|
complete = Buffer.concat([magic, Buffer.alloc(8)]);
|
|
const cases = [
|
|
["truncated-zero-neighbor", truncated, Buffer.alloc(8)],
|
|
[
|
|
"truncated-one-neighbor",
|
|
truncated,
|
|
Buffer.from([1, ...Array(7).fill(0)]),
|
|
],
|
|
["complete", complete, Buffer.alloc(8)],
|
|
] as const;
|
|
const reports = [];
|
|
for (const [name, bundle, adjacent] of cases) {
|
|
const bytes = elfFixture(62, [
|
|
{ name: ".hip_fatbin", bytes: bundle },
|
|
{ name: ".data", bytes: adjacent },
|
|
]),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.artifact.sha256).toBe(
|
|
createHash("sha256").update(bytes).digest("hex"),
|
|
);
|
|
reports.push(report);
|
|
}
|
|
const [zeroNeighbor, oneNeighbor, completeReport] = reports;
|
|
expect(zeroNeighbor!.coverage).toMatchObject({
|
|
discovery: "partial",
|
|
payloads: "partial",
|
|
});
|
|
expect(oneNeighbor!.coverage).toMatchObject({
|
|
discovery: "partial",
|
|
payloads: "partial",
|
|
});
|
|
expect(zeroNeighbor!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "malformed",
|
|
location: { size: 24, section_name: ".hip_fatbin" },
|
|
});
|
|
expect(oneNeighbor!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "malformed",
|
|
location: { size: 24, section_name: ".hip_fatbin" },
|
|
});
|
|
expect(completeReport!.coverage).toMatchObject({
|
|
discovery: "complete",
|
|
payloads: "complete",
|
|
});
|
|
expect(completeReport!.objects[0]).toMatchObject({
|
|
kind: "native-section",
|
|
inspection: "complete",
|
|
location: { size: 32, section_name: ".hip_fatbin" },
|
|
});
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"compares equivalent PTX targets and retains family-specific CUDA targets",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-target-spelling-");
|
|
for (const [name, target, payloadTarget, ptx, flags] of [
|
|
[
|
|
"compute",
|
|
"sm_80",
|
|
"compute_80",
|
|
ptxFixture.replace(".target sm_80", ".target compute_80"),
|
|
0,
|
|
],
|
|
[
|
|
"family",
|
|
"sm_100f",
|
|
"sm_100f",
|
|
ptxFixture.replace(".target sm_80", ".target sm_100f"),
|
|
0x200000,
|
|
],
|
|
] as const) {
|
|
const bytes = cudaFixture("none", [
|
|
{
|
|
kind: 1,
|
|
target: name === "compute" ? 80 : 100,
|
|
bytes: Buffer.from(ptx),
|
|
flags,
|
|
},
|
|
]),
|
|
path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]).toMatchObject({
|
|
target,
|
|
payload_target: payloadTarget,
|
|
target_consistency: "consistent",
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"reads CUDA V2 parameter records through the public CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-v2-parameter-"),
|
|
bytes = cudaFixture("none", [
|
|
{
|
|
kind: 2,
|
|
target: 80,
|
|
bytes: cubinFixture({ architecture: 80, v2ParameterSize: 20000 }),
|
|
},
|
|
]),
|
|
path = join(root, "large-parameter.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
parameter_count: 1,
|
|
parameter_bytes: 20000,
|
|
parameters: [{ ordinal: 0, offset: 0, size: 20000 }],
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"preserves HIP v3 feature target IDs for exact selection",
|
|
async ({ cli }) => {
|
|
const target = "amdgcn-amd-amdhsa--gfx906+xnack",
|
|
root = await createTestTempDirectory("rea-hip-v3-target-id-"),
|
|
payload = amdFixture(false, undefined, target);
|
|
payload.writeUInt32LE(0x2f, 48); // EF_AMDGPU_MACH_AMDGCN_GFX906
|
|
const bytes = hipFixture(payload, "hip-amdgcn-amd-amdhsa--gfx906+xnack"),
|
|
path = join(root, "feature.bundle");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--targets",
|
|
JSON.stringify(["gfx906+xnack"]),
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects.find((row) => row.kind === "amdgpu")!.target).toBe(
|
|
"gfx906+xnack",
|
|
);
|
|
expect(report.kernels).toHaveLength(1);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps PTX qualifier layouts and compact registers while excluding call prototypes",
|
|
async ({ cli }) => {
|
|
const source = ptxFixture
|
|
.replace(".u64 output", ".u64 .ptr .global output")
|
|
.replace(
|
|
".u32 value",
|
|
".u64 .ptr .global .align 4 ptr_arg, .param .u64 .ptr .align 1 increment_param_0, .param .u64 .ptr.global.align 4 joined_arg, .param .u64 .ptr .global.align 16 partly_joined_arg, .param .u64 .ptr.global .align 16 joined_space_arg, .param .b8 .align 8 payload[8], .param .u32 value",
|
|
)
|
|
.replace(".reg .b32", ".reg.b32")
|
|
.replace(
|
|
".reg .b64 %rd<5>;",
|
|
".reg .b64 %rd<5>; F: .callprototype _ (.reg.b32 _);",
|
|
),
|
|
root = await createTestTempDirectory("rea-ptx-callprototype-"),
|
|
path = join(root, "prototype.ptx");
|
|
await writeFile(path, source);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
parameter_count: 8,
|
|
parameter_bytes: 60,
|
|
parameters: [
|
|
{ name: "output", offset: 0, size: 8 },
|
|
{ name: "ptr_arg", offset: 8, size: 8 },
|
|
{ name: "increment_param_0", offset: 16, size: 8 },
|
|
{ name: "joined_arg", offset: 24, size: 8 },
|
|
{ name: "partly_joined_arg", offset: 32, size: 8 },
|
|
{ name: "joined_space_arg", offset: 40, size: 8 },
|
|
{ name: "payload", offset: 48, size: 8 },
|
|
{ name: "value", offset: 56, size: 4 },
|
|
],
|
|
registers: {
|
|
virtual: [
|
|
{ type: ".b32", count: 9 },
|
|
{ type: ".b64", count: 5 },
|
|
],
|
|
},
|
|
});
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
expect(report.ptx_artifacts[0]!.text).toBe(source);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"resolves CUBIN SHN_XINDEX symbols through their ELF extension table",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-cubin-shndx-cli-");
|
|
const bytes = cubinFixture({ extendedSectionIndex: true });
|
|
const path = join(root, "extended.cubin");
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
const sha256 = createHash("sha256").update(bytes).digest("hex");
|
|
expect(report.artifact.sha256).toBe(sha256);
|
|
expect(report.objects[0]!.sha256).toBe(sha256);
|
|
expect(report.kernels).toHaveLength(1);
|
|
expect(report.kernels[0]).toMatchObject({
|
|
name: "add_one",
|
|
parameter_count: 2,
|
|
parameter_bytes: 12,
|
|
shared_bytes: 1056,
|
|
sources: {
|
|
parameters: expect.objectContaining({ section_index: 65537 }),
|
|
shared: expect.objectContaining({ section_index: 65538 }),
|
|
},
|
|
});
|
|
|
|
const malformedBytes = Buffer.from(bytes);
|
|
const sectionTable = Number(malformedBytes.readBigUInt64LE(40));
|
|
const sectionCount = Number(
|
|
malformedBytes.readBigUInt64LE(sectionTable + 32),
|
|
);
|
|
let indexTable: number | undefined;
|
|
for (let index = 0; index < sectionCount; index++)
|
|
if (malformedBytes.readUInt32LE(sectionTable + index * 64 + 4) === 18) {
|
|
indexTable = index;
|
|
break;
|
|
}
|
|
expect(indexTable).toBeDefined();
|
|
malformedBytes.writeBigUInt64LE(4n, sectionTable + indexTable! * 64 + 32);
|
|
const malformedPath = join(root, "short-shndx.cubin");
|
|
await writeFile(malformedPath, malformedBytes);
|
|
const malformed = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
malformedPath,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(malformed.exitCode, malformed.stderr).toBe(0);
|
|
const malformedEvidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
malformed.json,
|
|
),
|
|
);
|
|
const malformedReport = gpuCodeObjectsSchema.parse(
|
|
malformedEvidence.normalized_result,
|
|
);
|
|
expect(malformedReport.objects[0]!.inspection).toBe("malformed");
|
|
expect(malformedReport.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"counts numeric percent-prefixed PTX register declarations",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-numeric-register-");
|
|
for (const register of ["%0", "%r0"] as const) {
|
|
const text = `.version 7.0\n.target sm_80\n.address_size 64\n.visible .entry kernel() {\n.reg .b32 ${register};\nmov.u32 ${register}, 0;\nret;\n}\n`;
|
|
const path = join(root, `${register.slice(1)}.ptx`);
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
).normalized_result,
|
|
);
|
|
expect(report.kernels).toMatchObject([
|
|
{
|
|
name: "kernel",
|
|
registers: { virtual: [{ type: ".b32", count: 1 }] },
|
|
},
|
|
]);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"preserves PTX target spelling and percent-prefixed entry names",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-identifiers-cli-");
|
|
const text = `.version 8.8\n.target sm_80\n.address_size 64\n.entry %kernel(.param .u32 input) {\nret;\n}\n`;
|
|
const path = join(root, "identifiers.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]!.target).toBe("sm_80");
|
|
expect(report.kernels).toEqual([
|
|
expect.objectContaining({
|
|
name: "%kernel",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
parameters: [
|
|
expect.objectContaining({
|
|
ordinal: 0,
|
|
name: "input",
|
|
offset: 0,
|
|
size: 4,
|
|
type: ".u32",
|
|
}),
|
|
],
|
|
}),
|
|
]);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"keeps entry-scoped PTX pragmas outside the body boundary",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-entry-pragma-");
|
|
const preamble = ".version 8.8\n.target sm_80\n.address_size 64\n",
|
|
body = "{ .reg .b32 %r<3>; .shared .align 4 .b8 scratch[16]; ret; }",
|
|
declarations = [
|
|
["plain", `.visible .entry k(.param .u32 input) ${body}`],
|
|
[
|
|
"entry-pragma",
|
|
`.visible .entry k(.param .u32 input) .pragma "nounroll"; ${body}`,
|
|
],
|
|
[
|
|
"body-pragma",
|
|
`.visible .entry k(.param .u32 input) { .pragma "nounroll"; .reg .b32 %r<3>; .shared .align 4 .b8 scratch[16]; ret; }`,
|
|
],
|
|
] as const;
|
|
const reports = new Map();
|
|
for (const [name, declaration] of declarations) {
|
|
const text = `${preamble}${declaration}\n`,
|
|
path = join(root, `${name}.ptx`);
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
reports.set(name, report);
|
|
}
|
|
const expected = {
|
|
name: "k",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
registers: { virtual: [{ type: ".b32", count: 3 }] },
|
|
shared_bytes: 16,
|
|
};
|
|
for (const [name, report] of reports) {
|
|
expect(report.kernels[0]).toMatchObject(expected);
|
|
expect(report.coverage.payloads).toBe("complete");
|
|
const declaration = declarations.find(
|
|
([caseName]) => caseName === name,
|
|
)![1];
|
|
expect(report.kernels[0]!.location.size).toBe(
|
|
Buffer.byteLength(declaration.slice(declaration.indexOf(".entry"))),
|
|
);
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"accepts the PTX compute_xx .target synonym verbatim",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-target-cli-");
|
|
const text = `.version 8.8\n.target compute_80\n.address_size 64\n.entry kernel() {\nret;\n}\n`;
|
|
const path = join(root, "target.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.objects[0]!.target).toBe("compute_80");
|
|
expect(report.ptx_artifacts[0]!.sha256).toBe(
|
|
createHash("sha256").update(text).digest("hex"),
|
|
);
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"extracts entry parameters declared in PTX 1.3 bodies",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-ptx-legacy-cli-");
|
|
const text = `.version 1.3\n.target sm_80\n.address_size 64\n.entry kernel {\n.param .u32 input;\nret;\n}\n`;
|
|
const path = join(root, "legacy.ptx");
|
|
await writeFile(path, text);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const evidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
response.json,
|
|
),
|
|
);
|
|
const report = gpuCodeObjectsSchema.parse(evidence.normalized_result);
|
|
expect(report.kernels).toEqual([
|
|
expect.objectContaining({
|
|
name: "kernel",
|
|
parameter_count: 1,
|
|
parameter_bytes: 4,
|
|
parameters: [
|
|
expect.objectContaining({
|
|
ordinal: 0,
|
|
name: "input",
|
|
offset: 0,
|
|
size: 4,
|
|
type: ".u32",
|
|
}),
|
|
],
|
|
}),
|
|
]);
|
|
const malformedPath = join(root, "truncated.ptx");
|
|
await writeFile(
|
|
malformedPath,
|
|
`.version 1.3\n.target sm_80\n.entry kernel {\n.param .u32 input\n`,
|
|
);
|
|
const malformed = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
malformedPath,
|
|
"--analysis",
|
|
"kernels",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(malformed.exitCode, malformed.stderr).toBe(0);
|
|
const malformedEvidence = parseEvidence(
|
|
toolContract("inspect_gpu_code_objects").outputSchema.parse(
|
|
malformed.json,
|
|
),
|
|
);
|
|
const malformedReport = gpuCodeObjectsSchema.parse(
|
|
malformedEvidence.normalized_result,
|
|
);
|
|
expect(malformedReport.objects[0]!.inspection).toBe("malformed");
|
|
expect(malformedReport.coverage.payloads).toBe("partial");
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"attributes malformed selector JSON to its actual CLI option",
|
|
async ({ cli }) => {
|
|
for (const name of ["object_ids", "targets", "view"]) {
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
"/not-opened-for-invalid-selectors",
|
|
`--${name}`,
|
|
"[",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode).not.toBe(0);
|
|
expect(response.json).toMatchObject({
|
|
code: "invalid_request",
|
|
details: {
|
|
operation: "inspect_gpu_code_objects",
|
|
issues: [
|
|
{
|
|
path: [name],
|
|
reason: "invalid_format",
|
|
message: `${name} must be valid JSON`,
|
|
},
|
|
],
|
|
},
|
|
});
|
|
}
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"publishes and enforces the lowercase --expected_sha256 digest",
|
|
async ({ cli }) => {
|
|
const schema = await cli.run({
|
|
arguments: ["inspect-gpu-code-objects", "--schema", "--json"],
|
|
});
|
|
expect(schema.exitCode, schema.stderr).toBe(0);
|
|
expect(schema.json).toMatchObject({
|
|
options: {
|
|
properties: { expected_sha256: { pattern: "^[a-f0-9]{64}$" } },
|
|
},
|
|
});
|
|
// PowerShell's Get-FileHash prints the digest in upper case.
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
"/not-opened-for-invalid-selectors",
|
|
"--expected_sha256",
|
|
"AB".repeat(32),
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode).toBe(1);
|
|
expect(response.json).toMatchObject({
|
|
code: "VALIDATION_ERROR",
|
|
fieldErrors: [expect.objectContaining({ path: "expected_sha256" })],
|
|
});
|
|
},
|
|
);
|
|
|
|
it("preserves incomplete decoder coverage through MCP target and object pages", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-partial-"),
|
|
{ call } = await connectLocalToolsMcp();
|
|
const cases = {
|
|
"section-names": () => {
|
|
const bytes = cubinFixture();
|
|
bytes.writeUInt16LE(0, 62);
|
|
return bytes;
|
|
},
|
|
symbols: () => {
|
|
const bytes = cubinFixture();
|
|
bytes.writeUInt32LE(1, Number(bytes.readBigUInt64LE(40)) + 3 * 64 + 4);
|
|
return bytes;
|
|
},
|
|
"later-amd-note": () => amdFixture(false, Buffer.alloc(3)),
|
|
"modern-amd-target-conflict": () =>
|
|
hipFixture(
|
|
amdFixture(false, undefined, "amdgpu9.00-amd-amdhsa--gfx900"),
|
|
"hipv4-amdgpu11.00-amd-amdhsa--gfx1100",
|
|
),
|
|
"target-conflict": () =>
|
|
cudaFixture("none", [
|
|
{
|
|
kind: 2,
|
|
target: 80,
|
|
bytes: cubinFixture({ architecture: 90 }),
|
|
},
|
|
]),
|
|
"register-conflict": () => cubinFixture({ textRegisters: 11 }),
|
|
"ptx-parameter": () =>
|
|
Buffer.from(ptxFixture.replace(".u64 output", ".u64 .vendor output")),
|
|
"ptx-shared": () =>
|
|
Buffer.from(ptxFixture.replace("scratch[32]", "scratch[]")),
|
|
};
|
|
for (const [missing, fixture] of Object.entries(cases)) {
|
|
const bytes = fixture(),
|
|
path = join(root, missing);
|
|
await writeFile(path, bytes);
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "targets" },
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const targets = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(targets.source_coverage.payloads).toBe("partial");
|
|
expect(targets.items).toEqual([
|
|
expect.objectContaining({
|
|
target:
|
|
missing === "later-amd-note"
|
|
? "amdgcn-amd-amdhsa--gfx900"
|
|
: missing === "modern-amd-target-conflict"
|
|
? "gfx1100"
|
|
: "sm_80",
|
|
kernel_count: missing === "symbols" ? 0 : 1,
|
|
kernel_coverage: "partial",
|
|
}),
|
|
]);
|
|
const objectReply = await call("inspect_analysis_view", {
|
|
source: {
|
|
kind: "retained-evidence",
|
|
evidence_id: targets.parent_evidence_id,
|
|
},
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection: "objects",
|
|
fields: ["kind"],
|
|
},
|
|
});
|
|
expect(objectReply.isError, JSON.stringify(objectReply)).not.toBe(true);
|
|
const objects = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(objectReply.structuredContent).normalized_result,
|
|
);
|
|
expect(objects.items).toEqual([
|
|
...(missing === "modern-amd-target-conflict"
|
|
? [expect.objectContaining({ kind: "host", inspection: "complete" })]
|
|
: []),
|
|
expect.objectContaining({
|
|
inspection: missing === "later-amd-note" ? "malformed" : "partial",
|
|
}),
|
|
]);
|
|
if (missing === "target-conflict")
|
|
expect(objects.items[0]).toMatchObject({
|
|
target: "sm_80",
|
|
payload_target: "sm_90",
|
|
target_consistency: "conflict",
|
|
});
|
|
if (missing === "modern-amd-target-conflict")
|
|
expect(objects.items[1]).toMatchObject({
|
|
kind: "amdgpu",
|
|
target: "gfx1100",
|
|
payload_target: "amdgpu9.00-amd-amdhsa--gfx900",
|
|
target_consistency: "conflict",
|
|
});
|
|
}
|
|
});
|
|
|
|
it("keeps unknown target discovery visible in the default MCP summary", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-unknown-target-"),
|
|
path = join(root, "cuda");
|
|
await writeFile(path, cubinFixture({ osAbi: 0x41, flags: 0x06016402 }));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", { path });
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const result = gpuCodeObjectsSummaryViewResultSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(result.summary.coverage.discovery).toBe("partial");
|
|
expect(result.summary.counts).toMatchObject({ objects: 1, targets: 0 });
|
|
const detailedReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "objects" },
|
|
});
|
|
expect(detailedReply.isError, JSON.stringify(detailedReply)).not.toBe(true);
|
|
const detailed = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(detailedReply.structuredContent).normalized_result,
|
|
);
|
|
expect(detailed.items).toHaveLength(1);
|
|
const source = {
|
|
kind: "retained-evidence",
|
|
evidence_id: detailed.parent_evidence_id,
|
|
} as const;
|
|
for (const collection of ["objects", "targets", "kernels"] as const) {
|
|
const filteredReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection,
|
|
targets: [""],
|
|
},
|
|
});
|
|
expect(filteredReply.isError, JSON.stringify(filteredReply)).not.toBe(true);
|
|
const filtered = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(filteredReply.structuredContent).normalized_result,
|
|
);
|
|
expect(filtered.items).toEqual([]);
|
|
}
|
|
});
|
|
|
|
it("preserves omitted AMD metadata in selected MCP object pages", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-metadata-columns-"),
|
|
path = join(root, "amd");
|
|
await writeFile(path, amdFixture());
|
|
const { call } = await connectLocalToolsMcp();
|
|
const inspected = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: { collection: "objects", fields: ["amd_target_metadata"] },
|
|
});
|
|
expect(inspected.isError, JSON.stringify(inspected)).not.toBe(true);
|
|
const full = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inspected.structuredContent).normalized_result,
|
|
);
|
|
expect(full.items[0]).toHaveProperty(
|
|
"amd_target_metadata.processor",
|
|
"gfx900",
|
|
);
|
|
const selected = await call("inspect_analysis_view", {
|
|
source: { kind: "retained-evidence", evidence_id: full.parent_evidence_id },
|
|
view: { kind: "gpu-code-objects", collection: "objects", fields: ["kind"] },
|
|
});
|
|
expect(selected.isError, JSON.stringify(selected)).not.toBe(true);
|
|
const page = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(selected.structuredContent).normalized_result,
|
|
);
|
|
expect(page.items).toHaveLength(1);
|
|
expect(page.items[0]).not.toHaveProperty("amd_target_metadata");
|
|
});
|
|
|
|
it("retains selected kernel facts and complete PTX for historical MCP pages after source removal", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-mcp-"),
|
|
path = join(root, "application");
|
|
await writeFile(path, cudaFixture("zstd"));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const inventoryReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
view: { collection: "objects", fields: ["kind", "target"] },
|
|
});
|
|
expect(inventoryReply.isError, JSON.stringify(inventoryReply)).not.toBe(true);
|
|
const inventory = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inventoryReply.structuredContent).normalized_result,
|
|
);
|
|
expect(inventory.items).toHaveLength(2);
|
|
expect(inventory.items[0]).not.toHaveProperty("sha256");
|
|
expect(inventory.items[0]).toHaveProperty("metadata_location");
|
|
const targetsReply = await call("inspect_analysis_view", {
|
|
source: {
|
|
kind: "retained-evidence",
|
|
evidence_id: inventory.parent_evidence_id,
|
|
},
|
|
view: { kind: "gpu-code-objects", collection: "targets" },
|
|
});
|
|
expect(targetsReply.isError, JSON.stringify(targetsReply)).not.toBe(true);
|
|
const targets = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(targetsReply.structuredContent).normalized_result,
|
|
);
|
|
expect(targets.items).toHaveLength(2);
|
|
expect(targets.items).toEqual(
|
|
expect.arrayContaining([
|
|
expect.objectContaining({
|
|
target: "sm_80",
|
|
object_count: 1,
|
|
kernel_count: 0,
|
|
kernel_coverage: "not-requested",
|
|
}),
|
|
expect.objectContaining({
|
|
target: "sm_90",
|
|
object_count: 1,
|
|
kernel_count: 0,
|
|
kernel_coverage: "not-requested",
|
|
}),
|
|
]),
|
|
);
|
|
const inspectedReply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
view: {
|
|
collection: "kernels",
|
|
names: ["add_one"],
|
|
fields: ["parameter_count", "registers"],
|
|
},
|
|
});
|
|
expect(inspectedReply.isError, JSON.stringify(inspectedReply)).not.toBe(true);
|
|
const inspected = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(inspectedReply.structuredContent).normalized_result,
|
|
);
|
|
expect(inspected.items).toHaveLength(2);
|
|
expect(inspected.items[0]).not.toHaveProperty("parameters");
|
|
expect(inspected.items[0]).toHaveProperty("sources");
|
|
const objectId =
|
|
"object_id" in inspected.items[0]!
|
|
? inspected.items[0]!.object_id
|
|
: undefined;
|
|
expect(objectId).toBeTypeOf("string");
|
|
await unlink(path);
|
|
const source = {
|
|
kind: "retained-evidence",
|
|
evidence_id: inspected.parent_evidence_id,
|
|
};
|
|
const textReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: {
|
|
kind: "gpu-code-objects",
|
|
collection: "ptx",
|
|
object_id: objectId,
|
|
offset: 0,
|
|
limit: 4096,
|
|
},
|
|
});
|
|
expect(textReply.isError, JSON.stringify(textReply)).not.toBe(true);
|
|
const text = gpuCodeObjectsPageViewResultSchema.parse(
|
|
parseEvidence(textReply.structuredContent).normalized_result,
|
|
);
|
|
expect(text.ptx!.text).toBe(ptxFixture);
|
|
expect(text.ptx!.sha256).toBe(
|
|
createHash("sha256").update(ptxFixture).digest("hex"),
|
|
);
|
|
const summaryReply = await call("inspect_analysis_view", {
|
|
source,
|
|
view: { kind: "summary" },
|
|
});
|
|
expect(summaryReply.isError, JSON.stringify(summaryReply)).not.toBe(true);
|
|
const summary = gpuCodeObjectsSummaryViewResultSchema.parse(
|
|
parseEvidence(summaryReply.structuredContent).normalized_result,
|
|
);
|
|
expect(summary.summary.counts).toMatchObject({
|
|
objects: 2,
|
|
kernels: 2,
|
|
ptx_artifacts: 1,
|
|
});
|
|
expect(summary.summary).not.toHaveProperty("ptx_artifacts");
|
|
});
|
|
|
|
it("rejects forged duplicate object IDs and mismatched Evidence subjects through MCP", async () => {
|
|
const root = await createTestTempDirectory("rea-gpu-import-"),
|
|
path = join(root, "cuda");
|
|
await writeFile(path, cudaFixture());
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const parent = parseEvidence(reply.structuredContent),
|
|
report = gpuCodeObjectsSchema.parse(parent.normalized_result);
|
|
const forgedHeaderReport = gpuCodeObjectsSchema.parse(
|
|
parent.normalized_result,
|
|
);
|
|
forgedHeaderReport.objects[1]!.cuda_target_header!.location.container_id =
|
|
forgedHeaderReport.objects[0]!.id;
|
|
const forgedHeader = createEvidence(undefined, parent.provider, {
|
|
operation: parent.operation,
|
|
parameters: parent.parameters,
|
|
result: forgedHeaderReport,
|
|
});
|
|
expect(
|
|
parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: forgedHeader },
|
|
view: { kind: "gpu-code-objects", collection: "objects" },
|
|
}),
|
|
).error,
|
|
).toMatchObject({ code: "invalid_request" });
|
|
report.objects[1]!.id = report.objects[0]!.id;
|
|
const forged = createEvidence(undefined, parent.provider, {
|
|
operation: "inspect_gpu_code_objects",
|
|
parameters: parent.parameters,
|
|
result: report,
|
|
});
|
|
const failure = parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: forged },
|
|
view: { kind: "gpu-code-objects", collection: "objects" },
|
|
}),
|
|
);
|
|
expect(failure.error).toMatchObject({ code: "invalid_request" });
|
|
const mismatched = createEvidence(
|
|
{ path: join(root, "other-file"), sha256: "f".repeat(64), format: "file" },
|
|
parent.provider,
|
|
{
|
|
operation: parent.operation,
|
|
parameters: parent.parameters,
|
|
result: parent.normalized_result,
|
|
},
|
|
);
|
|
const identityFailure = parseMcpToolError(
|
|
await call("inspect_analysis_view", {
|
|
source: { kind: "inline", evidence: mismatched },
|
|
view: { kind: "summary" },
|
|
}),
|
|
);
|
|
expect(identityFailure.error).toMatchObject({ code: "invalid_request" });
|
|
});
|
|
|
|
cliTest(
|
|
"uses DOS relocation ownership before GPU section discovery through the CLI",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-gpu-dos-relocations-cli-");
|
|
const path = join(root, "selected.exe");
|
|
const bytes = peGpuFixture();
|
|
bytes.writeUInt16LE(bytes.length % 512, 2);
|
|
bytes.writeUInt16LE(Math.ceil(bytes.length / 512), 4);
|
|
bytes.writeUInt16LE(2, 8);
|
|
await writeFile(path, bytes);
|
|
const args = [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"inventory",
|
|
"--json",
|
|
];
|
|
const valid = await cli.run({ arguments: args });
|
|
expect(valid.exitCode, valid.stderr).toBe(0);
|
|
bytes.writeUInt16LE(1, 6);
|
|
bytes.writeUInt16LE(4, 8);
|
|
bytes.writeUInt16LE(60, 24);
|
|
await writeFile(path, bytes);
|
|
const refused = await cli.run({ arguments: args });
|
|
expect(refused.exitCode).toBe(1);
|
|
expect(refused.json).toMatchObject({
|
|
code: "target_unavailable",
|
|
details: {
|
|
path,
|
|
constraint: "file_format_rejected",
|
|
reason: "MZ header does not declare a PE image",
|
|
},
|
|
});
|
|
},
|
|
);
|
|
|
|
cliTest(
|
|
"retains AMD target declarations and examined note coverage through CLI and MCP",
|
|
async ({ cli }) => {
|
|
const root = await createTestTempDirectory("rea-amd-target-sources-");
|
|
const { call } = await connectLocalToolsMcp();
|
|
const first = "amdgcn-amd-amdhsa--gfx900";
|
|
const second = "amdgcn-amd-amdhsa--gfx1100";
|
|
const legacyNote = Buffer.alloc(20);
|
|
legacyNote.writeUInt32LE(4, 0);
|
|
legacyNote.writeUInt32LE(4, 4);
|
|
legacyNote.writeUInt32LE(10, 8);
|
|
legacyNote.write("AMD\0", 12);
|
|
const headerConflict = amdFixture();
|
|
headerConflict.writeUInt32LE(0x141, 48);
|
|
for (const [name, bytes, declarations, coverage, effective] of [
|
|
[
|
|
"note-conflict",
|
|
amdFixture(false, amdNoteFixture(false, second)),
|
|
[first, second],
|
|
"complete",
|
|
null,
|
|
],
|
|
["header-conflict", headerConflict, [first], "complete", null],
|
|
[
|
|
"duplicate",
|
|
amdFixture(false, amdNoteFixture()),
|
|
[first, first],
|
|
"complete",
|
|
first,
|
|
],
|
|
["legacy-tail", amdFixture(false, legacyNote), [first], "partial", first],
|
|
[
|
|
"invalid-target",
|
|
amdFixture(false, undefined, 123),
|
|
[null],
|
|
"partial",
|
|
null,
|
|
],
|
|
[
|
|
"absent-target",
|
|
amdFixture(false, undefined, null),
|
|
[null],
|
|
"complete",
|
|
"gfx900",
|
|
],
|
|
[
|
|
"truncated-tail",
|
|
amdFixture(false, Buffer.alloc(3)),
|
|
[first],
|
|
"partial",
|
|
first,
|
|
],
|
|
] as const) {
|
|
const path = join(root, name);
|
|
await writeFile(path, bytes);
|
|
const response = await cli.run({
|
|
arguments: [
|
|
"inspect-gpu-code-objects",
|
|
path,
|
|
"--analysis",
|
|
"kernels",
|
|
"--detail",
|
|
"complete",
|
|
"--json",
|
|
],
|
|
});
|
|
expect(response.exitCode, response.stderr).toBe(0);
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
for (const value of [response.json, reply.structuredContent]) {
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
parseEvidence(value).normalized_result,
|
|
);
|
|
const row = report.objects[0]!;
|
|
expect(row).toMatchObject({
|
|
target: effective,
|
|
payload_target: effective,
|
|
});
|
|
const metadata = row.amd_target_metadata!;
|
|
expect(metadata).toMatchObject({
|
|
flags: name === "header-conflict" ? "0x00000141" : "0x0000012c",
|
|
processor: name === "header-conflict" ? "gfx1100" : "gfx900",
|
|
header_location: { container_id: row.id, offset: 0, size: 64 },
|
|
note_coverage: coverage,
|
|
});
|
|
expect(metadata.notes.map(({ target }) => target)).toEqual(
|
|
declarations,
|
|
);
|
|
expect(
|
|
new Set(metadata.notes.map(({ location }) => location.offset)).size,
|
|
).toBe(declarations.length);
|
|
for (const note of metadata.notes) {
|
|
expect(note.location.container_id).toBe(row.id);
|
|
expect(note.location.section_name).toBe(".note");
|
|
const source = bytes.subarray(
|
|
note.location.offset,
|
|
note.location.offset + note.location.size,
|
|
);
|
|
if (note.target !== null)
|
|
expect(source.includes(Buffer.from(note.target))).toBe(true);
|
|
expect(note.target_state).toBe(
|
|
name === "invalid-target"
|
|
? "invalid"
|
|
: name === "absent-target"
|
|
? "absent"
|
|
: "declared",
|
|
);
|
|
}
|
|
expect(report.kernels).toHaveLength(declarations.length);
|
|
if (coverage === "partial")
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
}
|
|
}
|
|
},
|
|
);
|
|
|
|
it("does not interpret non-AMD ELF flags as AMD processor declarations through MCP", async () => {
|
|
const root = await createTestTempDirectory("rea-amd-machine-");
|
|
const path = join(root, "wrong-machine");
|
|
const payload = amdFixture();
|
|
payload.writeUInt16LE(62, 18);
|
|
await writeFile(path, hipFixture(payload, "hip-amdgcn-amd-amdhsa--"));
|
|
const { call } = await connectLocalToolsMcp();
|
|
const reply = await call("inspect_gpu_code_objects", {
|
|
path,
|
|
analysis: "kernels",
|
|
detail: "complete",
|
|
});
|
|
expect(reply.isError, JSON.stringify(reply)).not.toBe(true);
|
|
const report = gpuCodeObjectsSchema.parse(
|
|
parseEvidence(reply.structuredContent).normalized_result,
|
|
);
|
|
expect(report.objects.find(({ kind }) => kind === "amdgpu")).toMatchObject({
|
|
target: null,
|
|
payload_target: null,
|
|
amd_target_metadata: null,
|
|
target_consistency: "unknown",
|
|
inspection: "malformed",
|
|
});
|
|
expect(report.kernels).toEqual([]);
|
|
expect(report.coverage.payloads).toBe("partial");
|
|
});
|